Diff for /libaitio/src/aitio.c between versions 1.7.2.1 and 1.7.2.2

version 1.7.2.1, 2011/11/02 13:13:48 version 1.7.2.2, 2011/11/02 16:10:25
Line 592  ioCreatePIDFile(const char *csName, int ifExists) Line 592  ioCreatePIDFile(const char *csName, int ifExists)
  * @csFile = file for send   * @csFile = file for send
  * @sendLen = bytes to send, if 0 send all data   * @sendLen = bytes to send, if 0 send all data
  * @offset = start file offset   * @offset = start file offset
    * @sndbuf = SO_SNDBUF value, if 0 use default
  * return: 0 error, >0 ok, sended bytes   * return: 0 error, >0 ok, sended bytes
  */   */
 size_t  size_t
ioSendFile(int s, const char *csFile, size_t sendLen, off_t offset)ioSendFile(int s, const char *csFile, size_t sendLen, off_t offset, int sndbuf)
 {  {
         void *addr;          void *addr;
         int fd;          int fd;
        size_t len;        size_t len = 0;
         register size_t off = 0;
   
         if (!csFile)          if (!csFile)
                 return 0;                  return 0;
   
           if (sndbuf)
                   if (setsockopt(s, SOL_SOCKET, SO_SNDBUF, &sndbuf, sizeof sndbuf) == -1) {
                           LOGERR;
                           return 0;
                   }
   
         fd = open(csFile, O_RDONLY);          fd = open(csFile, O_RDONLY);
         if (fd == -1) {          if (fd == -1) {
                 LOGERR;                  LOGERR;
Line 625  ioSendFile(int s, const char *csFile, size_t sendLen,  Line 633  ioSendFile(int s, const char *csFile, size_t sendLen, 
         } else          } else
                 close(fd);                  close(fd);
   
        len = write(s, addr, sendLen);        while (off < sendLen && (len = write(s, addr + off, sendLen - off)) != -1)
                 off += len;
         if (len == -1) {          if (len == -1) {
                 LOGERR;                  LOGERR;
                 munmap(addr, sendLen);                  munmap(addr, sendLen);
                 return 0;                  return 0;
        }        } else
                 len = off;
   
         if (len != sendLen) {          if (len != sendLen) {
                 io_SetErr(ECANCELED, "Different sizes - request %u bytes, actually sended %u bytes\n",                   io_SetErr(ECANCELED, "Different sizes - request %u bytes, actually sended %u bytes\n", 
Line 648  ioSendFile(int s, const char *csFile, size_t sendLen,  Line 658  ioSendFile(int s, const char *csFile, size_t sendLen, 
  * @csFile = file for receive   * @csFile = file for receive
  * @recvLen = receive bytes   * @recvLen = receive bytes
  * @over = overwrite file if exists with mode like 0644   * @over = overwrite file if exists with mode like 0644
    * @rcvbuf = SO_RCVBUF value, if 0 use default
  * return: 0 error, >0 ok, received bytes   * return: 0 error, >0 ok, received bytes
  */   */
 size_t  size_t
ioRecvFile(int s, const char *csFile, size_t recvLen, int over)ioRecvFile(int s, const char *csFile, size_t recvLen, int over, int rcvbuf)
 {  {
         void *addr;          void *addr;
         int fd;          int fd;
        size_t len;        size_t len = 0;
         register size_t off = 0;
         struct pollfd pfd = { s, POLLIN | POLLPRI, 0 };
   
         if (!csFile || !recvLen)          if (!csFile || !recvLen)
                 return 0;                  return 0;
         if (!over && !access(csFile, F_OK))          if (!over && !access(csFile, F_OK))
                 return 0;                  return 0;
   
           if (rcvbuf)
                   if (setsockopt(s, SOL_SOCKET, SO_RCVBUF, &rcvbuf, sizeof rcvbuf) == -1) {
                           LOGERR;
                           return 0;
                   }
   
         fd = open(csFile, O_WRONLY | O_CREAT | O_TRUNC, over);          fd = open(csFile, O_WRONLY | O_CREAT | O_TRUNC, over);
         if (fd == -1) {          if (fd == -1) {
                 LOGERR;                  LOGERR;
Line 689  ioRecvFile(int s, const char *csFile, size_t recvLen,  Line 708  ioRecvFile(int s, const char *csFile, size_t recvLen, 
         } else          } else
                 close(fd);                  close(fd);
   
        len = read(s, addr, recvLen);        while (off < recvLen && poll(&pfd, 1, RECV_TIMEOUT) != -1)
                 while (off < recvLen && (len = read(s, addr + off, recvLen - off)) != -1)
                         off += len;
         if (len == -1) {          if (len == -1) {
                 LOGERR;                  LOGERR;
                 munmap(addr, recvLen);                  munmap(addr, recvLen);
                 unlink(csFile);                  unlink(csFile);
                 return 0;                  return 0;
        }        } else
                 len = off;
   
         if (len != recvLen)          if (len != recvLen)
                 io_SetErr(EAGAIN, "Different sizes - request %u bytes, actually received %u bytes\n",                   io_SetErr(EAGAIN, "Different sizes - request %u bytes, actually received %u bytes\n", 

Removed from v.1.7.2.1  
changed lines
  Added in v.1.7.2.2


FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>