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