Rio包

最新推荐文章于 2024-02-28 11:33:56 发布

ningto.com

最新推荐文章于 2024-02-28 11:33:56 发布

阅读量2.8k

点赞数

分类专栏： c/c++ network 文章标签：网络

本文为博主原创文章，转载请注明来源(泞途 - 泊客网 https://ningto.com)。

本文链接：https://blog.csdn.net/tujiaw/article/details/7415162

版权

c/c++ 同时被 2 个专栏收录

122 篇文章 3 订阅

订阅专栏

network

19 篇文章 0 订阅

订阅专栏

// 读写网络套接字时，由于内部缓冲约束和较长的网络延迟会引起read和write返回不足值。
// Rio包可以解决上述问题进行健壮地读和写

#include <unistd.h>
#include <stdio.h>
#include <errno.h>

ssize_t rio_readn(int fd, void *usrbuf, size_t n)
{
    size_t nleft = n;
    ssize_t nread;
    char *bufp = usrbuf;

    while (nleft > 0)
    {
        if ((nread = read(fd, bufp, nleft)) < 0)
        {
            if (errno == EINTR)
            {
                nread = 0;
            }
            else
            {
                return -1;
            }
        }
        else if (nread == 0)
        {
            break;
        }

        nleft -= nread;
        bufp += nread;
    }

    return (n-nleft);
}

ssize_t rio_writen(int fd, void *usrbuf, size_t n)
{
    size_t nleft = n;
    ssize_t nwritten;
    char *bufp = usrbuf;

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

        nleft -= nwritten;
        bufp += nwritten;
    }

    return n;
}