python sockerserver tcp 文件下载 udp

时间:2023-03-09 01:50:37
python  sockerserver  tcp 文件下载  udp
#tcp serverclass MyHandler(socketserver.BaseRequestHandler):    def handle(self):        # 通信循环        while True:            try:                cmd = self.request.recv(1024)                if len(cmd) == 0:                    break                else:                    cmd = cmd.decode(encoding='utf-8')                if cmd=='up':                    import os

                if cmd == 'get':                    path = 'file.txt'                    total_size = os.path.getsize(filename=path)                    import hashlib                    with open(path, 'rb') as f:                        for line in f:                            line                    hashlib_value = 'xxxx'                else:                    obj = subprocess.Popen(cmd,                                           shell=True,                                           stdout=subprocess.PIPE,                                           stderr=subprocess.PIPE)                    s_out = obj.stdout.read()                    s_err = obj.stderr.read()                    total_size = len(s_out) + len(s_err)                    path = None                    hashlib_value = '你大爷'

                header_dic = {                    'file_name': path,                    'total_size': total_size,                    'hashlib': hashlib_value                }                header_json = (json.dumps(header_dic)).encode(encoding='utf-8')                header_len = struct.pack('i', len(header_json))

                # print(header_len)                self.request.send(header_len)                # print(header_json)                self.request.send(header_json)

                if cmd == 'get':                    with open(path, 'rb') as f:                        for line in f:                            # print(line)                            self.request.send(line)                else:                    # print('cmd')                    self.request.send(s_out)                    self.request.send(s_err)

            except ConnectionResetError:                break

        self.request.close()

if __name__ == "__main__":    s = socketserver.ThreadingTCPServer(('127.0.0.1', 8080), MyHandler, bind_and_activate=True)    s.serve_forever()

#tcp clientimport socket, struct, json, hashlib

st = socket.socket(socket.AF_INET, socket.SOCK_STREAM)st.connect(('127.0.0.1', 8080))

def get_file(st):    data = st.recv(4)    data = struct.unpack('i', data)[0]    header_json = json.loads((st.recv(data)).decode('utf-8'))    print(header_json)    file_down = b''    encodeing = None    while header_json['total_size'] > 0:        msg = st.recv(1024)        header_json['total_size'] -= len(msg)        file_down += msg        try:            if encodeing == None:                print(msg.decode(encoding='utf-8'))            else:                print(msg.decode(encodeing='gbk'))        except Exception:            encodeing == False            print(msg.decode(encoding='gbk'))

    return file_down

while True:

    msg = input('请输入').strip()    if not len(msg): continue    try:        st.send(msg.encode('utf-8'))        file_down = get_file(st)

        if msg == 'get':            with open('save.txt', 'wb') as f:                f.write(file_down)            print('end')

    except ConnectionResetError:        break

st.close()

#udp server
import socketserverclass MyHandler(socketserver.BaseRequestHandler):    def handle(self):

        data = self.request[0]        self.request[1].sendto(data.upper(), self.client_address)

if __name__ == '__main__':    ss=socketserver.ThreadingUDPServer(('127.0.0.1',8080),MyHandler)    ss.serve_forever()
#udp clientfrom socket import *

st = socket(AF_INET, SOCK_DGRAM)while True:    msg = input('请输入').strip()    if not len(msg): print('不要发送空数据')    st.sendto(msg.encode(encoding='utf-8'), ('127.0.0.1', 8080))    print('发送了数据')    print(st.recvfrom(1024))

st.close()