《UNIX网络编程 卷1》 笔记： readn和writen函数

字节流套接字上调用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错误，这时应该忽略这个错误并继续被中断的系统调用。