socket编程之解决流协议的粘包问题(一 )

流协议和粘包

这么说吧,TCP在传输数据的时候,是不区分边界的(数据和数据之间没有边界),因为是基于字节流,所以数据对TCP来说就是一大堆没有结构区别的字节块。那意味着什么?意味着TCP并不能对多个数据的整体的信息进行区分(打个比方:就像是你说一堆话没有标点符号全部连在一起,别人很可能弄错)或者对单个整体信息的错误区分(比如你要发送的整块数据被分成了好几块发送,然后这些数据在传输过程中可能由于网络原因,有的大数据块中被分片后的部分片段到了,可能由于接收端缓冲区满了,开始读取,而它们又没有边界之分,这时候就解释错误了)。那样就会使得我们要传输的信息可能粘连在一起,而被计算机解释错误。

而我们要怎么解决这种问题呢?

1.定长包 
2.在包的尾部加上\r\n等字符(ftp使用这种策略,如果包的内容中也包含\r\n,这时候需要用转义字符 \ 处理) 
3.包头加上包体长度(接收时先接收包头根据包头计算出包体的长度来接收包体) 
4.更复杂的应用层协议

首先第一种定长包:

有几种可能: 
1.包长小于1024,消息包最后面的那部分都用空白字节补全,凑齐1024字节这个长度,发送。 


由于这样的数据包都是定长度,不会出现像上面那样的一次发送多个具有不同数据结构的消息,也就意味这接受的数据包不能无脑地组合在一起被解释,因为单个数据包都是独立的数据结构。 

最大缺陷就是它并不能发挥TCP协议的高效性,而且极大地浪费了网络流量,很多无效数据在网络上传输,不推荐使用。

第二种在包的尾部加上\r,\n等字符

我们常用的ftp服务器就是这样的设计方式,在区分包内的\r和\n等字符时候使用转义字符\解决,但是这种限制了只能做某种特殊的服务,因为我们必须保证结束符的唯一性,在通常的数据传输中,由于用户的输入是不能限定的,什么样的字符都有,所以不能随便应用。

第三种:包头加上包体长度

这是我们应用得最多的一种传输的方式,保证了TCP的高效性,而且解决了粘包问题。 下面来模拟一下这个传输方式:其实就是封装一个发送和接受的函数,然后在接受数据的时候先对接受的包头数据分析,求出包头告诉我们的实际数据的长度,再进行二次接受,那就是我们要的数据

代码实现
服务器端:

#include<unistd.h>//通过接收包头来确定长度len
#include<sys/types.h>
#include<sys/socket.h>
#include<netinet/in.h>
#include<arpa/inet.h>

#include<stdio.h>
#include<stdlib.h>
#include<errno.h>
#include<string.h>
#define ERR_EXIT(m) \
    do \
    { \
        perror(m); \
        exit(EXIT_FAILURE); \
    } while(0)
struct packet
{
    int len;
    char buf[1024];//

};
ssize_t readn(int fd,void *buf,size_t count)
{
//ssize_t=int,size_t=unsigned int
//接收count个字节数
    size_t nleft=count;//剩余字节数
    ssize_t nread;//已经接收字节数
    char *bufp=(char *)buf;
    while(nleft>0)
    {
        if((nread=read(fd,bufp,nleft))<0)
        {
            if(errno==EINTR)//被中断
                continue;
            return -1;
        }
        else if(nread==0)//对等方关闭
            return count-nleft;//读到EOF,对方关闭
        bufp+=nread;
        nleft-=nread;
    }   
    return count;
}
ssize_t writen(int fd,void *buf,size_t count)
{
    size_t nleft=count;//剩余发送字节数
        ssize_t nwritten;//已经发送字节数
        char *bufp=(char *)buf;
        while(nleft>0)//一般而言,write缓冲区大于发送数据缓冲区,不阻塞
        {
                if((nwritten=write(fd,bufp,nleft))<0)
                {
                        if(errno==EINTR)//被中断
                                continue;
                        return -1;
                }
                else if(nwritten==0)//对等方关闭
                        continue;//读到EOF,对方关闭
                bufp+=nwritten;
                nleft-=nwritten;
        }
    return count;
}
void do_service(int conn)
{   
    struct packet recvbuf;
    int n;
        while(1){
                     memset(&recvbuf,0,sizeof(recvbuf));//初始化recvbuf
                     int ret=readn(conn,&recvbuf.len,4);//函数从打开的文件,设备中读取数据,返回读取的字节数。
             if(ret==-1)//四个字节先接收到len
             {
                ERR_EXIT("read");
             }
                     else if(ret<4)
             {
            printf("client_close\n");
            break;
             }
             n=ntohl(recvbuf.len);//转换成主机字节序
             ret=readn(conn,recvbuf.buf,n);//接收n个字节到recvbuf
             if(ret==-1)
                     {
                                ERR_EXIT("read");
                     }
                     else if(ret<n)
                     {
                        printf("client_close\n");
                        break;
                     }

             fputs(recvbuf.buf,stdout);//输出
                     writen(conn,&recvbuf,4+n);//buf中数据被复制到了TCP发送缓冲区
                }
}
int main(void)
{
    int listenfd;
    /*if((listenfd=socket(AF_INET,SOCK_STEAM,IPPOTO_TCP))<0) */
    if((listenfd=socket(AF_INET,SOCK_STREAM,0))<0)
        ERR_EXIT("socket");

    //IPV4地址结构
    struct sockaddr_in servaddr;
    memset(&servaddr,0,sizeof(servaddr));
    servaddr.sin_family=AF_INET;//地址家族
    servaddr.sin_port=htons(5188);//端口,主机转网络
    /*servaddr.sin_addr.s_addr=inet_addr("127.0.0.1");*/
    /*inet_aton("127.0.0.1",&servaddr.sin_addr);*/
    servaddr.sin_addr.s_addr=htonl(INADDR_ANY);
    int on=1;
    if(setsockopt(listenfd,SOL_SOCKET,SO_REUSEADDR,&on,sizeof(on))<0)
        ERR_EXIT("setsockopt");

    if(bind(listenfd,(struct sockaddr*)&servaddr,sizeof(servaddr))<0)
        ERR_EXIT("bind");


    if(listen(listenfd,SOMAXCONN)<0)
        ERR_EXIT("listen");

    struct sockaddr_in peeraddr;
    socklen_t peerlen =sizeof(peeraddr);//typedef int socklen_t
    int conn;//已连接套接字
    pid_t pid;
    while(1)
    {
        if((conn=accept(listenfd,(struct sockaddr*)&peeraddr,&peerlen))<0)
            ERR_EXIT("accept");

        printf("ip=%s port=%d\n",inet_ntoa(peeraddr.sin_addr),ntohs(peeraddr.sin_port));
        pid=fork();
        if(pid==-1)
            ERR_EXIT("fork");
        if(pid==0)
        {
            close(listenfd);
            do_service(conn);
            exit(EXIT_SUCCESS);//将子进程退出,要不它会fork()
        }
        else
            close(conn);
    }
    return 0;

}

客户端:

#include<unistd.h>
#include<sys/types.h>
#include<sys/socket.h>
#include<netinet/in.h>
#include<arpa/inet.h>

#include<stdio.h>
#include<stdlib.h>
#include<errno.h>
#include<string.h>
#define ERR_EXIT(m) \
    do \
    { \
        perror(m); \
        exit(EXIT_FAILURE); \
    } while(0)
struct packet
{
        int len;
        char buf[1024];//

};

ssize_t readn(int fd,void *buf,size_t count)
{
//ssize_t=int,size_t=unsigned int
//接收count个字节数
        size_t nleft=count;//剩余字节数
        ssize_t nread;//已经接收字节数
        char *bufp=(char *)buf;
        while(nleft>0)
        {
                if((nread=read(fd,bufp,nleft))<0)
                {
                        if(errno==EINTR)//被中断
                                continue;
                        return -1;
                }
                else if(nread==0)//对等方关闭
                        return count-nleft;//读到EOF,对方关闭
                bufp+=nread;
                nleft-=nread;
        }
        return count;
}
ssize_t writen(int fd,void *buf,size_t count)
{
        size_t nleft=count;//剩余发送字节数
        ssize_t nwritten;//已经发送字节数
        char *bufp=(char *)buf;
        while(nleft>0)//一般而言,write缓冲区大于发送数据缓冲区,不阻塞
        {
                if((nwritten=write(fd,bufp,nleft))<0)
                {
                        if(errno==EINTR)//被中断
                                continue;
                        return -1;
                }
                else if(nwritten==0)//对等方关闭
                        continue;//读到EOF,对方关闭
                bufp+=nwritten;
                nleft-=nwritten;
        }
        return count;
}
//发送定长包
int main()
{
    int sock;
    /*if((listenfd=socket(AF_INET,SOCK_STEAM,IPPOTO_TCP))<0) */
    if((sock=socket(AF_INET,SOCK_STREAM,0))<0)
        ERR_EXIT("socket");

    //IPV4地址结构
    struct sockaddr_in servaddr;
    memset(&servaddr,0,sizeof(servaddr));
    servaddr.sin_family=AF_INET;//地址家族
    servaddr.sin_port=htons(5188);//端口,主机转网络
    servaddr.sin_addr.s_addr=inet_addr("127.0.0.1");
    /*inet_aton("127.0.0.1",&servaddr.sin_addr);*/


    if(connect(sock,(struct sockaddr*)&servaddr,sizeof(servaddr))<0)
        ERR_EXIT("connect");

    struct packet sendbuf;
    struct packet recvbuf;
    memset(&sendbuf,0,sizeof(sendbuf));
    memset(&recvbuf,0,sizeof(recvbuf));
    int n;
    while(fgets(sendbuf.buf,sizeof(sendbuf.buf),stdin) !=NULL){
        n=strlen(sendbuf.buf);
        sendbuf.len=htonl(n);
        writen(sock,&sendbuf,4+n);//发送,4是头部大小,n是包体长度
        int ret=readn(sock,&recvbuf.len,4);//函数从打开的文件,设备中读取数据,返回读取的字.先接收4个字节
                if(ret==-1)
                {
                      ERR_EXIT("read");
                }
                else if(ret<4)
                {
                      printf("client_close\n");
                      break;
                }
                n=ntohl(recvbuf.len);//转换成主机字节序
                ret=readn(sock,recvbuf.buf,n);//接收n个字节到recvbuf
                if(ret==-1)
                {
                           ERR_EXIT("read");
                }
                else if(ret<n)
                {
                   printf("client_close\n");
                    break;
                }

        fputs(recvbuf.buf,stdout);
        memset(&sendbuf,0,sizeof(sendbuf));
        memset(&recvbuf,0,sizeof(recvbuf));
    }
    close(sock);
    return 0;


}
  • 0
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值