socket编程之黏包

原理概述

　　上图是我在学习python的socket编程中遇到的黏包问题所画，以实例来说明这个高大上的黏包问题。

　　我们知道socket()实例中sendall()方法是无论数据有多大，一次性提交写入缓冲区（应用层）；再来看接收端，recv()方法有个参数为buffsize，没错buffsize就是套接口的发送缓冲区的大小了。所以数据大于SO_SNDBUF的就会被分块传输，问题就来了，当两次提交的数据都比较大，刚好第一次尾与第二次的首同一时间待在了SO_SNDBUF里，被接收到了，这就是黏包。

　　一句话：黏包最本质的原因就是接收方不知道接收的包有多大！

解决方法（应用层维护消息和消息边界）：

1. 定长包
2. 包尾加上 标记（FTP）
3. 包头增加包体长度。
4. 复杂的应用层协议。

实例

　　本实例多线程实例，实现的是客户端向服务端输入系统命令，服务器返回命令在本机上的执行结果。因为有些命令返回的结果是远大于1024的，所以可能出现黏包的问题。本实例的解决方案是第三条，server端每次向client端返回命令执行结果前，先发送包体大小并得到client端返回的确认信息，再发送数据，程序结构上避免了黏包问题。

TCPSocket服务端

#!/usr/bin/env python
#-*- coding:utf-8 -*-
import SocketServer
import os

class Myserver(SocketServer.BaseRequestHandler):

    def handle(self):
        conn = self.request
        print "Client from:",self.client_address
        conn.sendall("请输入您要查询的命令")
        flag = True
        while flag:
            data = conn.recv(1024)
            print "receive cmd: %s"%data
            if data == "exit":
                flag = False
            else:
                ret = os.popen(data).read().decode("gbk").encode("utf-8")
　　　　　　　　　#发送包大小
                conn.sendall(str(len(ret)))
                #收到客户端确认消息。
                scOK = conn.recv(1024)
　　　　　　　　　#发送包体内容
                conn.sendall(ret)

if __name__ == "__main__":
    server = SocketServer.ThreadingTCPServer(("localhost",8000),Myserver)
    server.serve_forever()

TCPSocket客户端

#!/usr/bin/env python
#-*- coding:utf-8 -*-
import socket
sk = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sk.connect(("127.0.0.1",8000))
sk.settimeout(5)
data = sk.recv(1024)
print "SocketServer: %s" % data
while True:
    reSize = 0
    msg = raw_input("Input:")
    sk.sendall(msg)
　　#接收包的大小
    totleSize = int(sk.recv(1024))
　　#收到包的大小后，给server端发送确认信息。
    sk.sendall("It is ok")
    while True:
        data = sk.recv(1024)
        reSize += len(data)
　　　　#当接收数据等于包的size后，跳出循环，停止 接收。
        if reSize == totleSize:
            print data
            break
        print data
    if msg == "exit":
        break
sk.close()