eagi-sphinx-test.c

Go to the documentation of this file.

/*
 * Extended AGI test application
 *
 * This code is released into public domain
 * without any warranty of any kind.
 *
 */
 
/*! \file
 * Extended AGI test application
 *
 * This code is released into public domain
 * without any warranty of any kind.
 *
 * \ingroup agi
 */
 
#include <stdio.h>
#include <unistd.h>
#include <stdlib.h>
#include <errno.h>
#include <string.h>
#include <sys/select.h>
#include <fcntl.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <netdb.h>
 
#include "asterisk.h"
 
#include "asterisk/compat.h"
 
#define AUDIO_FILENO (STDERR_FILENO + 1)
 
#define SPHINX_HOST "192.168.1.108"
#define SPHINX_PORT 3460
 
static int sphinx_sock = -1;
 
static int connect_sphinx(void)
{
    struct hostent *hp;
    struct sockaddr_in sin;
    int res;
    hp = gethostbyname(SPHINX_HOST);
    if (!hp) {
        fprintf(stderr, "Unable to resolve '%s'\n", SPHINX_HOST);
        return -1;
    }
    sphinx_sock = socket(PF_INET, SOCK_STREAM, 0);
    if (sphinx_sock < 0) {
        fprintf(stderr, "Unable to allocate socket: %s\n", strerror(errno));
        return -1;
    }
    memset(&sin, 0, sizeof(sin));
    sin.sin_family = AF_INET;
    sin.sin_port = htons(SPHINX_PORT);
    memcpy(&sin.sin_addr, hp->h_addr, sizeof(sin.sin_addr));
    if (connect(sphinx_sock, (struct sockaddr *)&sin, sizeof(sin))) {
        fprintf(stderr, "Unable to connect on socket: %s\n", strerror(errno));
        close(sphinx_sock);
        sphinx_sock = -1;
        return -1;
    }
    res = fcntl(sphinx_sock, F_GETFL);
    if ((res < 0) || (fcntl(sphinx_sock, F_SETFL, res | O_NONBLOCK) < 0)) {
        fprintf(stderr, "Unable to set flags on socket: %s\n", strerror(errno));
        close(sphinx_sock);
        sphinx_sock = -1;
        return -1;
    }
    return 0;
}
 
static int read_environment(void)
{
    char buf[256];
    char *val;
    /* Read environment */
    for(;;) {
        if (!fgets(buf, sizeof(buf), stdin)) {
            return -1;
        }
        if (feof(stdin))
            return -1;
        buf[strlen(buf) - 1] = '\0';
        /* Check for end of environment */
        if (!strlen(buf))
            return 0;
        val = strchr(buf, ':');
        if (!val) {
            fprintf(stderr, "Invalid environment: '%s'\n", buf);
            return -1;
        }
        *val = '\0';
        val++;
        val++;
        /* Skip space */
        fprintf(stderr, "Environment: '%s' is '%s'\n", buf, val);
 
        /* Load into normal environment */
        setenv(buf, val, 1);
 
    }
    /* Never reached */
    return 0;
}
 
static char *wait_result(void)
{
    fd_set fds;
    int res;
    int max;
    static char astresp[256];
    static char sphinxresp[256];
    char audiobuf[4096];
    for (;;) {
        FD_ZERO(&fds);
        FD_SET(STDIN_FILENO, &fds);
        FD_SET(AUDIO_FILENO, &fds);
        max = AUDIO_FILENO;
        if (sphinx_sock > -1) {
            FD_SET(sphinx_sock, &fds);
            if (sphinx_sock > max)
                max = sphinx_sock;
        }
        /* Wait for *some* sort of I/O */
        res = select(max + 1, &fds, NULL, NULL, NULL);
        if (res < 0) {
            fprintf(stderr, "Error in select: %s\n", strerror(errno));
            return NULL;
        }
        if (FD_ISSET(STDIN_FILENO, &fds)) {
            if (!fgets(astresp, sizeof(astresp), stdin)) {
                return NULL;
            }
            if (feof(stdin)) {
                fprintf(stderr, "Got hungup on apparently\n");
                return NULL;
            }
            astresp[strlen(astresp) - 1] = '\0';
            fprintf(stderr, "Ooh, got a response from Asterisk: '%s'\n", astresp);
            return astresp;
        }
        if (FD_ISSET(AUDIO_FILENO, &fds)) {
            res = read(AUDIO_FILENO, audiobuf, sizeof(audiobuf));
            if ((res > 0) && (sphinx_sock > -1)) {
                if (write(sphinx_sock, audiobuf, res) < 0) {
                    fprintf(stderr, "write() failed: %s\n", strerror(errno));
                }
            }
        }
        if ((sphinx_sock > -1) && FD_ISSET(sphinx_sock, &fds)) {
            res = read(sphinx_sock, sphinxresp, sizeof(sphinxresp));
            if (res > 0) {
                fprintf(stderr, "Oooh, Sphinx found a token: '%s'\n", sphinxresp);
                return sphinxresp;
            } else if (res == 0) {
                fprintf(stderr, "Hrm, lost sphinx, guess we're on our own\n");
                close(sphinx_sock);
                sphinx_sock = -1;
            }
        }
    }
 
}
 
static char *run_command(char *command)
{
    fprintf(stdout, "%s\n", command);
    return wait_result();
}
 
static int run_script(void)
{
    char *res;
    res = run_command("STREAM FILE demo-enterkeywords 0123456789*#");
    if (!res) {
        fprintf(stderr, "Failed to execute command\n");
        return -1;
    }
    fprintf(stderr, "1. Result is '%s'\n", res);
    res = run_command("STREAM FILE demo-nomatch 0123456789*#");
    if (!res) {
        fprintf(stderr, "Failed to execute command\n");
        return -1;
    }
    fprintf(stderr, "2. Result is '%s'\n", res);
    res = run_command("SAY NUMBER 23452345 0123456789*#");
    if (!res) {
        fprintf(stderr, "Failed to execute command\n");
        return -1;
    }
    fprintf(stderr, "3. Result is '%s'\n", res);
    res = run_command("GET DATA demo-enterkeywords");
    if (!res) {
        fprintf(stderr, "Failed to execute command\n");
        return -1;
    }
    fprintf(stderr, "4. Result is '%s'\n", res);
    res = run_command("STREAM FILE auth-thankyou \"\"");
    if (!res) {
        fprintf(stderr, "Failed to execute command\n");
        return -1;
    }
    fprintf(stderr, "5. Result is '%s'\n", res);
    return 0;
}
 
int main(int argc, char *argv[])
{
    char *tmp;
    int ver = 0;
    int subver = 0;
    /* Setup stdin/stdout for line buffering */
    setlinebuf(stdin);
    setlinebuf(stdout);
    if (read_environment()) {
        fprintf(stderr, "Failed to read environment: %s\n", strerror(errno));
        exit(1);
    }
    connect_sphinx();
    tmp = getenv("agi_enhanced");
    if (tmp) {
        if (sscanf(tmp, "%30d.%30d", &ver, &subver) != 2)
            ver = 0;
    }
    if (ver < 1) {
        fprintf(stderr, "No enhanced AGI services available.  Use EAGI, not AGI\n");
        exit(1);
    }
    if (run_script())
        return -1;
    exit(0);
}