我博客中很少只给出某些代码的片段, 基本都是能够直接运行的完整代码, 本文我们只给片段(很简单), 下面来看看:
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;
}
说白了, 就一直写, 写到指定的字节数为止。
不多说。