服务端代码 socketReceive.py：
#!/usr/bin/env python # -*- coding=utf-8 -*- """ file: socketReceive.py socket service """ import re import socket import threading import time import sys import os import struct def socket_service(): try: s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) s.bind(('127.0.0.1', 9000)) s.listen(10) except socket.error as msg: print(msg); sys.exit(1) print('Waiting client connection...'); while 1: conn, addr = s.accept() t = threading.Thread(target=deal_data, args=(conn, addr)) t.start() def deal_data(conn, addr): print('Accept new connection from {0}'.format(addr)); #conn.settimeout(500) conn.send(b'Hi, Welcome to the server!') while 1: fileinfo_size = struct.calcsize('128si2s') buf = conn.recv(fileinfo_size) if buf: filename, filesize ,pwd = struct.unpack('128si2s', buf); print("pwd : "+ bytes.decode(pwd)); #删除byte转为str后的\x00 用strip也可以 newFileName = bytes.decode(filename).rstrip('\x00'); new_filename = os.path.join('d:/', '' +newFileName); print('file new name is {0}, filesize if {1}'.format(new_filename,filesize)); recvd_size = 0 # 定义已接收文件的大小 fp = open(new_filename, 'wb') print('start receiving...'); while not recvd_size == filesize: if filesize - recvd_size > 1024: data = conn.recv(1024) recvd_size += len(data) else: data = conn.recv(filesize - recvd_size) recvd_size = filesize fp.write(data) fp.close() print('end receive...'); conn.close() break if __name__ == '__main__': socket_service()客户端代码 socketSend.py：
#!/usr/bin/env python # -*- coding=utf-8 -*- """ file: socketSend.py socket client """ import socket import os import sys import struct ''' 字符串和字节互转 ''' def convertStrOrByte(): # bytes object byte = b"example" # str object str = "example" # str to bytes 字符串转字节 bytes(str, encoding="utf8") # bytes to str 字节转字符串 str(bytes, encoding="utf-8") # an alternative method # str to bytes 字符串转为字节 str.encode(str) # bytes to str 字节转为字符串 bytes.decode(bytes) def socket_client(): try: s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) s.connect(('127.0.0.1', 9000)) except socket.error as msg: print(msg); sys.exit(1) print(s.recv(1024)); while 1: #filepath = input('please input file path: ') filepath = "C:/Users/Administrator/Desktop/sort.exe"; if os.path.isfile(filepath): # 定义定义文件信息。128s表示文件名为128bytes长，l表示一个int或log文件类型，在此为文件大小 2s 代表的是2个字节的字符串长度 fileinfo_size = struct.calcsize('128si2s') # 定义文件头信息，包含文件名和文件大小 fhead = struct.pack('128si2s', bytes(os.path.basename(filepath).encode('utf-8')),os.stat(filepath).st_size,b'pw'); s.send(fhead) print('client filepath: {0}'.format(filepath)) fp = open(filepath, 'rb') while 1: data = fp.read(1024) if not data: print('{0} file send over...'.format(filepath)); break s.send(data) s.close() break if __name__ == '__main__': socket_client()运行效果：
服务端端是可以一直接受上传请求的。
1 、 判定密码是否正确 2、 有的时候我们上传的文件可能需要放到不同的路径下,可以写个配置。
全部读取配置，新建一个test.properties文件
#密码 pwd=pw #sort.exe传到哪个路径 sort.exe=d:/test/ #默认路径 defaultDir=d:/服务端代码：
#!/usr/bin/env python # -*- coding=utf-8 -*- """ file: socketReceive.py socket service """ import re import socket import threading import time import sys import os import struct #读取配置的依赖 import re import os import tempfile def socket_service(): try: s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) s.bind(('127.0.0.1', 9000)) s.listen(10) except socket.error as msg: print(msg); sys.exit(1) print('Waiting client connection...'); while 1: conn, addr = s.accept() t = threading.Thread(target=deal_data, args=(conn, addr)) t.start() def deal_data(conn, addr): print('Accept new connection from {0}'.format(addr)); #conn.settimeout(500) conn.send(b'Hi, Welcome to the server!') while 1: fileinfo_size = struct.calcsize('128si2s') buf = conn.recv(fileinfo_size) if buf: filename, filesize ,pwd = struct.unpack('128si2s', buf); print("pwd : "+ bytes.decode(pwd)); #判定密码是否正确 file_path = './test.properties' property = Properties(file_path); # 读取文件 if property.get("pwd") == bytes.decode(pwd): print("password validate success!") else: print("password validate error") break; #删除byte转为str后的\x00 用strip也可以 newFileName = bytes.decode(filename).rstrip('\x00'); #得到文件路径前缀 dirPrefix = property.get(newFileName); if dirPrefix == None or dirPrefix == "": dirPrefix = property.get("defaultDir"); new_filename = os.path.join(dirPrefix, '' +newFileName); print('file new name is {0}, filesize if {1}'.format(new_filename,filesize)); recvd_size = 0 # 定义已接收文件的大小 fp = open(new_filename, 'wb') print('start receiving...'); while not recvd_size == filesize: if filesize - recvd_size > 1024: data = conn.recv(1024) recvd_size += len(data) else: data = conn.recv(filesize - recvd_size) recvd_size = filesize fp.write(data) fp.close() print('end receive...'); conn.close() break ''' zhouzhongqing 2018年5月2日16:17:13 读取properties文件 ''' class Properties: def __init__(self, file_name): self.file_name = file_name self.properties = {} try: fopen = open(self.file_name, 'r') for line in fopen: line = line.strip() if line.find('=') > 0 and not line.startswith('#'): strs = line.split('=') self.properties[strs[0].strip()] = strs[1].strip() except Exception: #raise Exception print("read file exception ..."); else: fopen.close(); def has_key(self, key): return key in self.properties def get(self, key, default_value=''): if key in self.properties: return self.properties[key] return default_value def put(self, key, value): self.properties[key] = value replace_property(self.file_name, key + '=.*', key + '=' + value, True) def parse(file_name): return Properties(file_name) def replace_property(file_name, from_regex, to_str, append_on_not_exists=True): tmpfile = tempfile.TemporaryFile() if os.path.exists(file_name): r_open = open(file_name, 'r') pattern = re.compile(r'' + from_regex) found = None for line in r_open: if pattern.search(line) and not line.strip().startswith('#'): found = True line = re.sub(from_regex, to_str, line) tmpfile.write(bytes(line, encoding="utf8")) if not found and append_on_not_exists: tmpfile.write( bytes('\n' + to_str ,encoding="utf-8")) r_open.close() tmpfile.seek(0) content = tmpfile.read() if os.path.exists(file_name): os.remove(file_name) w_open = open(file_name, 'w') w_open.write(str(content,encoding="utf-8")) w_open.close() tmpfile.close() else: print("file %s not found" % file_name); if __name__ == '__main__': socket_service()客户端代码：
#!/usr/bin/env python # -*- coding=utf-8 -*- """ file: socketSend.py socket client """ import socket import os import sys import struct ''' 字符串和字节互转 ''' def convertStrOrByte(): # bytes object byte = b"example" # str object str = "example" # str to bytes 字符串转字节 bytes(str, encoding="utf8") # bytes to str 字节转字符串 str(bytes, encoding="utf-8") # an alternative method # str to bytes 字符串转为字节 str.encode(str) # bytes to str 字节转为字符串 bytes.decode(bytes) def socket_client(): try: s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) s.connect(('127.0.0.1', 9000)) except socket.error as msg: print(msg); sys.exit(1) print(s.recv(1024)); while 1: #filepath = input('please input file path: ') filepath = "C:/Users/Administrator/Desktop/sort.exe"; if os.path.isfile(filepath): # 定义定义文件信息。128s表示文件名为128bytes长，l表示一个int或log文件类型，在此为文件大小 2s 代表的是2个字节的字符串长度 fileinfo_size = struct.calcsize('128si2s') # 定义文件头信息，包含文件名和文件大小 fhead = struct.pack('128si2s', bytes(os.path.basename(filepath).encode('utf-8')),os.stat(filepath).st_size,b'pw'); s.send(fhead) print('client filepath: {0}'.format(filepath)) fp = open(filepath, 'rb') while 1: data = fp.read(1024) if not data: print('{0} file send over...'.format(filepath)); break s.send(data) s.close() break if __name__ == '__main__': socket_client()当然这里面应该还有很多不完善的地方,有问题的地方希望多多指正。
