对 send 和 recv 的改造

// 发送数据

  ssize_t sendn(int fd, void *buf, size_t count, int flags)
  {
    size_t nleft = count;
    ssize_t nread;
    char *pbuf = (char*)buf;
    while(nleft > 0)
    {
        if ((nread = send(fd, pbuf, nleft, flags)) < 0)
        {
            if (errno == EINTR)//信号中断
            {
                continue;
            }
            return -1;
        }else if (nread == 0)//连接中断
        {
            return count - nleft;//返回发送的字节数
        }
        nleft -= nread;//计算剩余字节数
        pbuf += nread;//移动指针
        }
        return count;
  }
// 接收函数
ssize_t recvn(int fd, void* buf, size_t count, int flags)
{
    size_t nleft = count;
    ssize_t nwrite ;
    char *pbuf = (char*)buf;
    while(nleft > 0)
    {
        if ((nwrite = recv(fd, pbuf, nleft, flags)) < 0)
        {
            if (errno == EINTR)
            {
                continue;
            }
            return -1;
        }else if (nwrite == 0)
        {
            return count - nleft;
        }
        nleft -= nwrite;
        pbuf +=nwrite;
    }
    return count;
}

通过上面的改造发送数据必须使用定长包或者有 协议的包

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值