Sun Feb 23 07:00:00 CET 2003
[asterisk/asterisk.git] / agi / eagi-sphinx-test.c
1 /*
2  * Extended AGI test application
3  *
4  * Copyright (C) 2003, Digium, Inc.
5  */
6
7 #include <stdio.h>
8 #include <unistd.h>
9 #include <stdlib.h>
10 #include <errno.h>
11 #include <string.h>
12 #include <sys/select.h>
13 #include <fcntl.h>
14 #include <arpa/inet.h>
15 #include <sys/socket.h>
16 #include <netinet/in.h>
17 #include <netdb.h>
18
19 #define AUDIO_FILENO (STDERR_FILENO + 1)
20
21 #define SPHINX_HOST "192.168.1.108"
22 #define SPHINX_PORT 3460
23
24 static int sphinx_sock = -1;
25
26 static int connect_sphinx(void)
27 {
28         struct hostent *hp;
29         struct sockaddr_in sin;
30         int res;
31         hp = gethostbyname(SPHINX_HOST);
32         if (!hp) {
33                 fprintf(stderr, "Unable to resolve '%s'\n", SPHINX_HOST);
34                 return -1;
35         }
36         sphinx_sock = socket(PF_INET, SOCK_STREAM, 0);
37         if (sphinx_sock < 0) {
38                 fprintf(stderr, "Unable to allocate socket: %s\n", strerror(errno));
39                 return -1;
40         }
41         memset(&sin, 0, sizeof(sin));
42         sin.sin_family = AF_INET;
43         sin.sin_port = htons(SPHINX_PORT);
44         memcpy(&sin.sin_addr, hp->h_addr, sizeof(sin.sin_addr));
45         if (connect(sphinx_sock, &sin, sizeof(sin))) {
46                 fprintf(stderr, "Unable to connect on socket: %s\n", strerror(errno));
47                 close(sphinx_sock);
48                 sphinx_sock = -1;
49                 return -1;
50         }
51         res = fcntl(sphinx_sock, F_GETFL);
52         if ((res < 0) || (fcntl(sphinx_sock, F_SETFL, res | O_NONBLOCK) < 0)) {
53                 fprintf(stderr, "Unable to set flags on socket: %s\n", strerror(errno));
54                 close(sphinx_sock);
55                 sphinx_sock = -1;
56                 return -1;
57         }
58         return 0;
59 }
60
61 static int read_environment(void)
62 {
63         char buf[256];
64         char *val;
65         /* Read environment */
66         for(;;) {
67                 fgets(buf, sizeof(buf), stdin);
68                 if (feof(stdin))
69                         return -1;
70                 buf[strlen(buf) - 1] = '\0';
71                 /* Check for end of environment */
72                 if (!strlen(buf))
73                         return 0;
74                 val = strchr(buf, ':');
75                 if (!val) {
76                         fprintf(stderr, "Invalid environment: '%s'\n", buf);
77                         return -1;
78                 }
79                 *val = '\0';
80                 val++;
81                 val++;
82                 /* Skip space */
83                 fprintf(stderr, "Environment: '%s' is '%s'\n", buf, val);
84
85                 /* Load into normal environment */
86                 setenv(buf, val, 1);
87                 
88         }
89         /* Never reached */
90         return 0;
91 }
92
93 static char *wait_result(void)
94 {
95         fd_set fds;
96         int res;
97         int max;
98         static char astresp[256];
99         static char sphinxresp[256];
100         char audiobuf[4096];
101         for (;;) {
102                 FD_ZERO(&fds);
103                 FD_SET(STDIN_FILENO, &fds);
104                 FD_SET(AUDIO_FILENO, &fds);
105                 max = AUDIO_FILENO;
106                 if (sphinx_sock > -1) {
107                         FD_SET(sphinx_sock, &fds);
108                         if (sphinx_sock > max)
109                                 max = sphinx_sock;
110                 }
111                 /* Wait for *some* sort of I/O */
112                 res = select(max + 1, &fds, NULL, NULL, NULL);
113                 if (res < 0) {
114                         fprintf(stderr, "Error in select: %s\n", strerror(errno));
115                         return NULL;
116                 }
117                 if (FD_ISSET(STDIN_FILENO, &fds)) {
118                         fgets(astresp, sizeof(astresp), stdin);
119                         if (feof(stdin)) {
120                                 fprintf(stderr, "Got hungup on apparently\n");
121                                 return NULL;
122                         }
123                         astresp[strlen(astresp) - 1] = '\0';
124                         fprintf(stderr, "Ooh, got a response from Asterisk: '%s'\n", astresp);
125                         return astresp;
126                 }
127                 if (FD_ISSET(AUDIO_FILENO, &fds)) {
128                         res = read(AUDIO_FILENO, audiobuf, sizeof(audiobuf));
129                         if (res > 0) {
130                                 if (sphinx_sock > -1) 
131                                         write(sphinx_sock, audiobuf, res);
132                         }
133                 }
134                 if ((sphinx_sock > -1) && FD_ISSET(sphinx_sock, &fds)) {
135                         res = read(sphinx_sock, sphinxresp, sizeof(sphinxresp));
136                         if (res > 0) {
137                                 fprintf(stderr, "Oooh, Sphinx found a token: '%s'\n", sphinxresp);
138                                 return sphinxresp;
139                         } else if (res == 0) {
140                                 fprintf(stderr, "Hrm, lost sphinx, guess we're on our own\n");
141                                 close(sphinx_sock);
142                                 sphinx_sock = -1;
143                         }
144                 }
145         }
146                 
147 }
148
149 static char *run_command(char *command)
150 {
151         fprintf(stdout, "%s\n", command);
152         return wait_result();
153 }
154
155 static int run_script(void)
156 {
157         char *res;
158         res = run_command("STREAM FILE demo-enterkeywords 0123456789*#");
159         if (!res) {
160                 fprintf(stderr, "Failed to execute command\n");
161                 return -1;
162         }
163         fprintf(stderr, "1. Result is '%s'\n", res);
164         res = run_command("STREAM FILE demo-nomatch 0123456789*#");
165         if (!res) {
166                 fprintf(stderr, "Failed to execute command\n");
167                 return -1;
168         }
169         fprintf(stderr, "2. Result is '%s'\n", res);
170         res = run_command("SAY NUMBER 23452345 0123456789*#");
171         if (!res) {
172                 fprintf(stderr, "Failed to execute command\n");
173                 return -1;
174         }
175         fprintf(stderr, "3. Result is '%s'\n", res);
176         res = run_command("GET DATA demo-enterkeywords");
177         if (!res) {
178                 fprintf(stderr, "Failed to execute command\n");
179                 return -1;
180         }
181         fprintf(stderr, "4. Result is '%s'\n", res);
182         res = run_command("STREAM FILE auth-thankyou \"\"");
183         if (!res) {
184                 fprintf(stderr, "Failed to execute command\n");
185                 return -1;
186         }
187         fprintf(stderr, "5. Result is '%s'\n", res);
188         return 0;
189 }
190
191 int main(int argc, char *argv[])
192 {
193         char *tmp;
194         int ver = 0;
195         int subver = 0;
196         /* Setup stdin/stdout for line buffering */
197         setlinebuf(stdin);
198         setlinebuf(stdout);
199         if (read_environment()) {
200                 fprintf(stderr, "Failed to read environment: %s\n", strerror(errno));
201                 exit(1);
202         }
203         connect_sphinx();
204         tmp = getenv("agi_enhanced");
205         if (tmp) {
206                 if (sscanf(tmp, "%d.%d", &ver, &subver) != 2)
207                         ver = 0;
208         }
209         if (ver < 1) {
210                 fprintf(stderr, "No enhanced AGI services available.  Use EAGI, not AGI\n");
211                 exit(1);
212         }
213         if (run_script())
214                 return -1;
215         exit(0);
216 }