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