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