
#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()