libsent/src/net/rdwt.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/tcpip.h>

#define         BUFSZ   4096    ///< Buffer size


int
rd(int fd, char *data, int *len, int maxlen)
{
  int count=0;
  int tmpbytes, tmplen;
  int toread = sizeof(int), ret, off = 0;
  
  while (toread > 0) {
    ret = 
#ifdef WINSOCK
      recv(fd,((char *)len) + off, toread, 0);
#else
      read(fd,((char *)len) + off, toread);
#endif
    if (ret <= 0) {  
      if (ret < 0) jlog("Error: rdwt: failed to read data at %d / %d\n", count, len);
      return(-1);
    }
    toread -= ret;
    off += ret;
  }
  if (*len > maxlen) {
    jlog("Error: rdwt: transfer data length exceeded: %d (>%d)\n", len, maxlen);
    return(-1);
  }
  while (count<(*len)){

    tmplen = (*len) - count;
    if (tmplen > BUFSZ) tmplen = BUFSZ;
    if ((tmpbytes =
#ifdef WINSOCK
         recv(fd,data+count,tmplen,0)
#else
         read(fd,data+count,tmplen)
#endif
         ) < 0) {
      jlog("Error: rdwt: failed to read data at %d / %d\n",count, len);
      return(count);
    }
    count += tmpbytes;
  }
  return(count);
}

int
wt(int fd, char *data, int len)
{
  int tmpbytes;

  /* len == 0 is used to tell end of segment ack */
  if ((tmpbytes=
#ifdef WINSOCK
       send(fd,(char *)&len,sizeof(int),0)
#else
       write(fd,(char *)&len,sizeof(int))
#endif
       ) != sizeof(int)) {
    /*jlog( "failed to write num\n");*/
    return(-1);
  }
  if (len > 0) {
    if ((tmpbytes=
#ifdef WINSOCK
         send(fd,data,len,0)
#else
         write(fd,data,len)
#endif
         ) < 0) {
      jlog("Error: rdwt: failed to write data (%d bytes)\n",len);
      return(-1);
    }
  } else {
    tmpbytes = 0;
  }
  return(tmpbytes);
}