libsent/src/adin/adin_tcpip.c

説明を見る。

/*
 * Copyright (c) 1991-2011 Kawahara Lab., Kyoto University
 * Copyright (c) 2000-2005 Shikano Lab., Nara Institute of Science and Technology
 * Copyright (c) 2005-2011 Julius project team, Nagoya Institute of Technology
 * All rights reserved
 */

#include <sent/stddefs.h>
#include <sent/adin.h>
#include <sent/tcpip.h>

static int adinnet_sd = -1;     
static int adinnet_asd = -1;    

#ifdef FORK_ADINNET
static pid_t child;             /* child process ID (0 if myself is child) */
#endif

boolean
adin_tcpip_standby(int freq, void *port_str)
{
  int port;

  port = atoi((char *)port_str);

  if ((adinnet_sd = ready_as_server(port)) < 0) {
    jlog("Error: adin_tcpip: cannot ready for server\n");
    return FALSE;
  }

  jlog("Stat: adin_tcpip: ready for server\n");

  return TRUE;
}

boolean
adin_tcpip_begin(char *pathname)
{
#ifdef FORK_ADINNET
    /***********************************/
    /*** server infinite loop here!! ***/
    /***********************************/
    for (;;) {
      /* wait connection */
      jlog("Stat: adin_tcpip: waiting connection...\n");
      if ((adinnet_asd = accept_from(adinnet_sd)) < 0) {
        return FALSE;
      }
      jlog("Stat: adin_tcpip: connected\n");
      /* fork self */
      child = fork();
      if (child < 0) {          /* error */
        jlog("Error: adin_tcpip: fork failed\n");
        return FALSE;
      }
      /* child thread should handle this request */
      if (child == 0) {         /* child thread */
        break;                  /* proceed */
      } else {                  /* parent thread */
        jlog("Stat: adin_tcpip: forked process [%d] handles this request\n", child);
      }
    }
#else  /* ~FORK_ADINNET */
    jlog("Stat: adin_tcpip: waiting connection...\n");
    if ((adinnet_asd = accept_from(adinnet_sd)) < 0) {
      return FALSE;
    }
    jlog("Stat: adin_tcpip: connected\n");
#endif /* FORK_ADINNET */

  return TRUE;
}

boolean
adin_tcpip_end()
{
    /* end of connection */
    close_socket(adinnet_asd);
#ifdef FORK_ADINNET
    /* terminate this child process here */
    jlog("Stat: adin_tcpip: connection end, child process now exit\n");
    exit(0);
#else
    /* wait for the next connection */
    jlog("Stat: adin_tcpip: connection end\n");
#endif

  return TRUE;
}

int
adin_tcpip_read(SP16 *buf, int sampnum)
{
  int cnt, ret;
  fd_set rfds;
  struct timeval tv;
  int status;

  /* check if some commands are waiting in queue */
  FD_ZERO(&rfds);
  FD_SET(adinnet_asd, &rfds);
  tv.tv_sec = 0;
  tv.tv_usec = 10000;           /* 10msec */
  status = select(adinnet_asd+1, &rfds, NULL, NULL, &tv);
  if (status < 0) {             /* error */
    jlog("Error: adin_tcpip: failed to poll socket\n");
    return -2;                  /* error return */
  }
  if (status > 0) {             /* there are some data */
    /* read one data segment, leave rest even if any for avoid blocking */
    ret = rd(adinnet_asd, (char *)buf, &cnt, sampnum * sizeof(SP16));
    if (ret == 0) {
      /* end of segment mark */
      return -3;
    }
    if (ret < 0) {
      /* end of input, mark */
      return -1;
    }
  } else {                      /* time out, no data */
    cnt = 0;
  }
  cnt /= sizeof(SP16);
#ifdef WORDS_BIGENDIAN
  swap_sample_bytes(buf, cnt);
#endif
  return cnt;
}

boolean
adin_tcpip_send_pause()
{
   int count;
   char com;
   /* send stop command to adinnet client */
   com = '0';
   count = wt(adinnet_asd, &com, 1);
   if (count < 0) jlog("Warning: adin_tcpip: cannot send pause command to client\n");
   jlog("Stat: adin_tcpip: sent pause command to client\n");
   return TRUE;
}

boolean
adin_tcpip_send_resume()
{
  int count;
  char com;
  int cnt, ret;
  fd_set rfds;
  struct timeval tv;
  int status;
  static char *tmpbuf = NULL;

  /* check if some commands are waiting in queue */
  count = 0;
  do {
    FD_ZERO(&rfds);
    FD_SET(adinnet_asd, &rfds);
    tv.tv_sec = 0;
    tv.tv_usec = 0;
    status = select(adinnet_asd+1, &rfds, NULL, NULL, &tv);
    if (status < 0) {           /* error */
      jlog("Error: adin_tcpip: failed to poll socket\n");
      return FALSE;                     /* error return */
    }
    if (status > 0) {           /* there are some data */
      if (tmpbuf == NULL) tmpbuf = (char *)mymalloc(MAXSPEECHLEN);
      ret = rd(adinnet_asd, tmpbuf, &cnt, MAXSPEECHLEN * sizeof(SP16));
    }
    if (cnt > 0) count += cnt;
  } while (status != 0);
  if (count > 0) {
    jlog("Stat: %d samples transfered while pause are flushed\n", count);
  }
    
  /* send resume command to adinnet client */
  com = '1';
  count = wt(adinnet_asd, &com, 1);
  if (count < 0) jlog("Warning: adin_tcpip: cannot send resume command to client\n");
  jlog("Stat: adin_tcpip: sent resume command to client\n");

  /* flush current buffer */
  return TRUE;
}

boolean
adin_tcpip_send_terminate()
{
   int count;
   char com;
   /* send terminate command to adinnet client */
   com = '2';
   count = wt(adinnet_asd, &com, 1);
   if (count < 0) jlog("Warning: adin_tcpip: cannot send terminate command to client\n");
   jlog("Stat: adin_tcpip: sent terminate command to client\n");
   return TRUE;
}

char *
adin_tcpip_input_name()
{
  return("network socket");
}