执行shell命令的多线程多连接Socket

Server端：

import socket
import subprocess
import multiprocessing,threading
host = ''
port = 8000
buffersize = 1024 # 方便今后扩展
address = (host,port)   # 方便今后扩展

s = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
s.bind(address)

def cmd(conn,addr):
    while True:
        data = bytes.decode(conn.recv(buffersize))
        result = subprocess.check_output(data,shell=True)
        count = len(result)
        print('count:',count)
        conn.send(bytes(str(count),encoding='utf8'))
        while count > 0 :
            count2 = conn.send(result)
            count = count - count2
    conn.close()

while True:
    s.listen(10)
    conn,addr = s.accept()
    print('Connect by:',addr)
    process = threading.Thread(target=cmd,args=(conn,addr))  # 启动多线程来实现多个客户端同时连接，否则只能同时一个客户端连接。
    process.start()
s.close()

Client端：

import socket
'''
数据发送和接收要注意 encoding 和 decode
'''
host = '127.0.0.1'
port = 8000
address = (host,port) # 方便今后扩展
buffersize = 1024
s = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
s.connect(address)

while True:
    data = bytes(input('Please input cmd:'),encoding='utf8')
    if len(data.strip()) == 0:
        continue
    s.send(data)
    data_size = s.recv(buffersize)
    print('data_size',data_size)
    data2 = b''
    while True:
        if int(data_size) == 0:
            print('Done!')
            break
        data3 = s.recv(buffersize)
        # print('data3',len(data3))  # 调试代码
        data2 = data2 + data3
        # print('data3',data3)  # 调试代码
        # print('len data3',len(data3))  # 调试代码
        if len(data2) >= int(data_size):
            break
    print(data2.decode())

s.close()