python tcp 粘包问题解决、文件下载等

from socket import *

#以下是关于tcp：服务端 和 客户端的小例子#服务端socket_server = socket(AF_INET, SOCK_STREAM)
socket_server.bind(('127.0.0.1', 8080))print('进行监听')socket_server.listen(5)
while True:    print('进行连接')    conn, address = socket_server.accept()    print('连接成功')
    while True:        data=conn.recv(1024)        if not len(data):break        conn.send(data.upper())        print(data.decode(encoding='utf-8'))        conn.close()
socket_server.close()
from socket import *import struct
socket_client = socket(AF_INET, SOCK_STREAM)print('进行连接')socket_client.connect(('127.0.0.1', 8080))print('连接成功')while True:    msg = input('请输入')    if not len(msg): continue    socket_client.send(msg.encode(encoding='utf-8'))    data =socket_client.recv(1024)    print(data.decode(encoding='utf-8'))
socket_client.close()
#远程执行命令的小例子#服务端import socketimport osimport subprocessphone = socket.socket(socket.AF_INET, socket.SOCK_STREAM)print('进行绑定，监听')phone.bind(('127.0.0.1', 8080))phone.listen(5)
def cmd(cmdlist):    obj=subprocess.Popen(    cmdlist,    shell=True,    stdout=subprocess.PIPE,    stderr=subprocess.PIPE    )
    stdout_res=obj.stdout.read()    stderr_res=obj.stderr.read()
    return stdout_res+stderr_res#假如不写成方法，这里可以分成两次发送过去，更加节省内存
while True:    conn, add = phone.accept()    print('建立连接')
    while True:        try:            msg = conn.recv(1024)            if not len(msg):                break
            print('执行命令')            print(msg)            #os.system(msg.decode(encoding='utf-8'))
            stdout_res=cmd(msg.decode(encoding='utf-8'))            print('发送消息')            conn.send(stdout_res)        except Exception:            break
    conn.close()phone.close()
#客户端import socketst = socket.socket(socket.AF_INET, socket.SOCK_STREAM)st.connect(('127.0.0.1', 8080))while True:    msg=input('请输入cmd 命令').strip()    print('发送消息')    if not len(msg):continue
    st.send(msg.encode(encoding='utf-8'))    data=st.recv(1024)    print(data.decode('gbk'))
st.close()

#获取服务端内容小例子

from socket import *import structimport subprocessimport json
 
st = socket(AF_INET, SOCK_STREAM)print('绑定并且监听')st.bind(('127.0.0.1', 8080))st.listen(5)print('绑定监听成功')
 
while True:    conn, _ = st.accept()    print('与客户端建立连接成功')
 
    while True:        try:            cmd = conn.recv(1024)            if not len(cmd): break            cmd = cmd.decode(encoding='utf-8')            if cmd != 'get':                obj_sub = subprocess.Popen(cmd,                                           shell=True,                                           stderr=subprocess.PIPE,                                           stdout=subprocess.PIPE)                sub_out = obj_sub.stdout.read()                sub_err = obj_sub.stderr.read()
 
                total_size = len(sub_err) + len(sub_out)                file_path = '为空'            else:                file_path = 'get.txt'                import os
 
                total_size = os.path.getsize(file_path)
 
            header_dic = {                'file_name': file_path,                'total_size': total_size,                'hashlib': '还没有有开始写呢'            }
 
            header_json = json.dumps(header_dic)            header_bytes = header_json.encode(encoding='utf-8')
 
            conn.send(struct.pack('i', len(header_bytes)))  # 将文件头的长度发送给客户端
 
            conn.send(header_bytes)  # 将文件头的内容发给客户端            if cmd == 'get':                with open(file_path, 'rb') as f:
 
                    for line in f:                        conn.send(line)            else:                conn.send(sub_out)  # 正常内容                conn.send(sub_err)  # 错误消息
 
        except ConnectionResetError:            break
 
conn.close()
 
st.close()

from socket import *import structimport subprocessimport json
 
st = socket(AF_INET, SOCK_STREAM)st.connect(('127.0.0.1', 8080))
 
print('与客户端建立连接')
 
while True:    msg = input('请输入cmd命令:').strip()    if not len(msg):        continue    st.send(msg.encode(encoding='utf-8'))
 
    header_struct_len = st.recv(4)
 
    header_len = (struct.unpack('i', header_struct_len))[0]
 
    header_json_dic = st.recv(header_len)  # 拿到文件头    header_dic = header_json_dic.decode(encoding='utf-8')    header_dic = json.loads(header_dic)    print(header_dic)    res = b''    while header_dic['total_size'] > 0:        data = st.recv(1024)        header_dic['total_size'] -= len(data)        res += data        # print(data.decode(encoding='gbk'))    print('msg')    if msg == 'get':        print(res.decode(encoding='utf-8'))        #print(res)        print(type(res))
 
    else:        print(res.decode(encoding='gbk'))
 
st.close()print('error')