字节流套接字上调用read或write读取或写入的字节数可能比请求的数量少,这是正常的现象。原因在于内核中用于套接字的缓冲区可能已达到了极限(例如套接字的接收缓冲区可能没有数据或发送缓冲区没有可用空间),此时所需的是调用者再次调用read或write函数,以读取或写入剩余的字节。
我们实现的从套接字上读取n字节数据的readn函数和写入n字节数据到套接字的writen函数代码如下:
#include "unp.h"
/*从套接字中读取n字节数据*/
ssize_t 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;
else /*错误*/
return -1;
} else if (nread == 0) /*遇到EOF*/
break;
nleft -= nread;
ptr += nread;
}
return n - nleft;
}
/*包裹函数*/
ssize_t Readn(int fd, void *vptr, size_t n)
{
if (readn(fd, vptr, n) == -1)
err_sys("readn error");
}
/*写入n字节数据到套接字*/
ssize_t writen(int fd, const void *vptr, size_t n)
{
size_t nleft;
ssize_t nwriten;
const char *ptr;
ptr = vptr;
nleft = n;
while (nleft > 0) {
if ((nwriten = write(fd, ptr, nleft)) <= 0) {
if (nwriten < 0 && errno == EINTR) /*重试*/
nwriten = 0;
else /*错误*/
return -1;
}
nleft -= nwriten;
ptr += nwriten;
}
return n;
}
/*包裹函数*/
ssize_t Writen(int fd, const void *vptr, size_t n)
{
if (writen(fd, vptr, n) == -1)
err_sys("writen error");
}
注:read或write系统调用在执行过程中被信号打断时会产生EINTR错误,这时应该忽略这个错误并继续被中断的系统调用。