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