系统级IO

#include<cstddef>

// unixIO
ssize_t read(int fd, void *buf, size_t n);      //若成功则为读的字节数,若EOF则为0,若出错则为-1
ssize_t write(int fd, const void *buf, size_t n);

//Robust IO

//不带缓存版本
ssize_t rio_readn(int fd, void *userbuf, size_t n)
{
    size_t nleft = n;
    ssize_t nread;
    char *bufp = (char*)userbuf;


    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 *userbuf, size_t n)
{
    size_t nleft = n;
    ssize_t nwrite;
    char *bufp = (char*)userbuf;


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


        nleft -= nwrite;
        bufp += nwrite;
    }

    return n;

}
//带缓存版本
static const int RIO_BUFSIZE = 8192;

struct rio_t
{
    int     rio_fd;             //文件描述符
    int     rio_cnt;            //未读字节数
    char    *rio_bufptr;        //下一个未读字节
    char    rio_buf[RIO_BUFSIZE];//内部缓冲
};

void rio_readinitb(rio_t *rp, int fd)
{
    rp->rio_fd = fd;
    rp->rio_cnt = 0;
    rp->rio_bufptr = rp->rio_buf;
}

ssize_t rio_read(rio_t *rp, void *userbuf, size_t n)
{


    while(rp->rio_cnt <= 0)
    {
        rp->rio_cnt = read(rp->rio_fd, rp->rio_buf, sizeof(rp->rio_buf));

        if(rp->rio_cnt < 0)
        {
            if(errno != EINTR)
                return -1;
        }
        else if (rp->rio_cnt == 0)
            return 0;
        else
            rp->rio_bufptr = rp->rio_buf;
    }



    size_t cnt = n > rp->rio_cnt ? rp->rio_cnt : n;


    memcpy(userbuf, rp->rio_bufptr, cnt);

    rp->rio_bufptr += cnt;
    rp->rio_cnt -= cnt;

    return cnt;
}
//读文本行,复制到内存位置userbuf,最后在结尾添NULL(0)
ssize_t rio_readlineb(rio_t *rp, void *userbuf, size_t maxlen)
{
    ssize_t readn;
    char c;
    char *buf = (char*)userbuf;
    int i;
    for(i = 1; i < maxlen; ++i)
    {
        readn = rio_read(rp, &c, 1);
        if(readn < 0)
            return -1;
        else if (readn == 0)
        {
            if(i == 1)
                return 0;
            else
                break;
        }
        else
        {
            *buf++ = c;
            if(c == '\n')
            {
                ++i;
                break;
            }
        }



    }

    *buf = 0;
    return i-1;

}
//读二进制文件和文本行
ssize_t rio_readnb(rio_t *rp, void *userbuf, size_t n)
{
    size_t left = n;
    ssize_t readn;
    char *buf = (char*)userbuf;
    while(left > 0)
    {
        readn = rio_read(rp, buf, left);

        if(readn == 0)
            break;
        else if (readn < 0)
            return -1;
        else
        {

        }

        left -= readn;
        buf += readn;
    }
    return (n - left);
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值