readn和writen函数一览------顺便说说recv函数的MSG_WAITALL参数

时间:2022-11-15 23:57:31

        我博客中很少只给出某些代码的片段, 基本都是能够直接运行的完整代码, 本文我们只给片段(很简单), 下面来看看:

ssize_t readn(int fd, void *buf, int n)
{
	size_t nleft = n;   // left的意思是“剩下”, 而非“左边”
	char *bufptr = buf;
	ssize_t nread;

	while(nleft > 0)
	{
		if((nread = read(fd, bufptr, n)) < 0)
		{
			if(errno == EINTR) 	// 遇到中断
			{ 
				continue;   // 或者用 nread = 0;
			}
			else
			{
				return -1;  // 真正错误
			}

		}
		else if(nread == 0) // 对端关闭
		{
			break;
		}

		nleft -= nread;
		bufptr += nread;
	}

	return (n - nleft);
}
        说白了, 就一直读, 读到指定的字节数为止。 其实, 在recv函数中, 最后参数如果是MSG_WAITALL,  那么就可以用recv函数一行代码, 替代上述所有代码, 棒棒哒。 一直读取到n个字节, 在实际中, 是经常用到的方法, 太常见了。


        再看:

ssize_t writen(int fd, void *buf, int n)
{
	size_t nleft = n;
	char *bufptr = buf;
	ssize_t nwrite;

	while(nleft > 0)
	{
		if((nwrite = write(fd, bufptr, n)) <= 0)
		{
			if(errno == EINTR)
			{
				nwrite = 0;
			}
			else
			{
				return -1;
			}
		}

		nleft -= nwrite;
		bufptr += nwrite;
	}

	return n;
}
       说白了, 就一直写, 写到指定的字节数为止。


        不多说。