Python的socket的客户端和服务端

1、服务端

import socket
import json
import subprocess

# 1、创建socket对象
server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)  # 流式协议（tcp)
# 数据报协议UDP:socket.SOCK_DGRAM
# 2、绑定地址
server.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
server.bind(("127.0.0.1", 8080))
# 3、监听连接请求（开始营业） 半连接池：5
server.listen(5)

while True:
    conn, addr = server.accept()
    print("客户端IP+端口：", addr)
    # 5、 数据传输
    while True:
        try:
            cmd = conn.recv(1024)
        except:
            break
        cmd = cmd.decode("utf-8")
        print("客户端发过来的数据：", cmd)
        # 执行终端命令
        obj = subprocess.Popen(cmd.decode("utf-8"),
                               shell=True,
                               # 命令执行的正确结果就会给这个管道
                               stdout=subprocess.PIPE,
                               stderr=subprocess.PIPE
                               )
        out_res = obj.stdout.read()
        err_res = obj.stderr.read()
        data_size = len(out_res) + len(err_res)
        # 头部数据，指定8个字节
        # header = bytes(str(data_size), 'utf-8').zfill(8)
        header = {
            'file_name': 'a.mp4',
            'file_size': data_size,
            'md5': "fdjalfj"
        }
        header_json = json.dumps(header)
        header_bytes = header_json.encode('utf-8')
        header_h = bytes(str(len(header_bytes)), 'utf-8').zfill(4)

        conn.send(header_h)
        conn.send(header_bytes)
        conn.send(out_res)
        conn.send(err_res)

server.close()

2、客户端

import socket
import json

# 1、创建socket对象
client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)  # 流式协议（tcp)
# 2、 建连接
client.connect(("127.0.0.1", 8080))

# 3、传输数据
while True:
    cmd = input("请输入终端命令>>>").strip()
    if not cmd:
        continue
    client.send(cmd.encode('utf-8'))
    # 先拿到头部长度
    header_size = int(client.recv(4).decode("utf-8"))
    header_json = client.recv(header_size).decode('utf-8')
    header = json.loads(header_json)
    print(header)

    data_size = header_size['file_size']
    recv_size = 0
    # 粘包问题
    data = b''
    while recv_size < data_size:
        res = client.recv(1024)
        recv_size += len(res)
        data += res
    print(data.decode("utf-8"))
    print(len(data))

# 4、 关闭连结
client.close()