一直以为在阻塞的tcp socket上使用read/recv读取的数据长度一定和指定的读取长度一致,但是实际测试时发现往往返回的长度都比指定长度短,查找资料发现其实是一直误解了这个函数。 
  引用《UNIX网络编程 卷一 套接字联网API》3.9中的说法: 
字节流套接口(如tcp套接口)上的read和write函数所表现的行为不同于通常的文件IO。字节流套接口上的读或写输入或输出的字节数可能比要求的数量少,但这不是错误状况,原因是内核中套接口的缓冲区可能已达到了极限。此时所需的是调用者再次调用read或write函数,以输入或输出剩余的字节。 

  书中给出了readn和writen函数解决这个问题。 

ssize_t      /* Read "n" bytes from a descriptor. */  
readn(int fd, void *vptr, size_t n)  
{  
 size_t nleft;  
 ssize_t nread;  
 char *ptr;  
  
 ptr = vptr;  
 nleft = n;  
 while (nleft > 0) {  
  if ( (nread = read(fd, ptr, nleft)) < 0) {  
   if (errno == EINTR)  
    nread = 0;  /* and call read() again */  
   else  
    return(-1);  
  } else if (nread == 0)  
   break;    /* EOF */  
  
  nleft -= nread;  
  ptr   += nread;  
 }  
 return(n - nleft);  /* return >= 0 */  
}  

 

ssize_t      /* Write "n" bytes to a descriptor. */  
writen(int fd, const void *vptr, size_t n)  
{  
 size_t  nleft;  
 ssize_t  nwritten;  
 const char *ptr;  
  
 ptr = vptr;  
 nleft = n;  
 while (nleft > 0) {  
  if ( (nwritten = write(fd, ptr, nleft)) <= 0) {  
   if (nwritten < 0 && errno == EINTR)  
    nwritten = 0;  /* and call write() again */  
   else  
    return(-1);   /* error */  
  }  
  
  nleft -= nwritten;  
  ptr   += nwritten;  
 }  
 return(n);  
}  

在13.3中提供了另外一个选择: 
  在recv中,可以使用MSG_WAITALL标志保证要求读取的字节数。即使使用了MSG_WAITALL标志,如果发生了下列情况:(a)捕获一个信号(b)连接被终止,或(c)在套接口上发生错误,这个函数返回的字节数仍会比请求的少。 
  而且要注意的是这个标志并不是每个版本的recv都会实现的。

可自定义数据包头 定义几个字节的数据包长度位 先接受长度 再接受数据包 结合recv的MSG_PEEK来使用

MSG_PEEK可以参考:linux socket recv函数 MSG_PEEK标志介绍  https://blog.csdn.net/whatday/article/details/96100572

 

 

Logo

更多推荐