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