基于python3实现socket文件传输和校验

时间:2021-05-22

基于socket的文件传输并进行MD5值校验,供大家参考,具体内容如下

文件传输分为两个类,一个是服务端,一个是客户端。

客户端发起发送文件或接收文件的请求,服务端收到请求后接收或发送文件,最后进行MD5值的校验

socket数据通过struct模块打包

需要发送文件到服务端时,调用sendFile函数,struct包内包含文件信息、文件大小、文件MD5等信息,服务端接收到文件后进行MD5值校验,校验成功后则返回成功

需要从服务器下载文件时,调用recvFile函数,收到文件后进行MD5校验

client类代码如下

import socketimport struct,osimport subprocess dataFormat='8s32s100s100sl' class fileClient(): def __init__(self,addr): self.addr = addr self.action = '' self.fileName = '' self.md5sum = '' self.clientfilePath = '' self.serverfilePath = '' self.size = 0 def struct_pack(self): ret = struct.pack(dataFormat,self.action.encode(),self.md5sum.encode(),self.clientfilePath.encode(), self.serverfilePath.encode(),self.size) return ret def struct_unpack(self,package): self.action,self.md5sum,self.clientfilePath,self.serverfilePath,self.size = struct.unpack(dataFormat,package) self.action = self.action.decode().strip('\x00') self.md5sum = self.md5sum.decode().strip('\x00') self.clientfilePath = self.clientfilePath.decode().strip('\x00') self.serverfilePath = self.serverfilePath.decode().strip('\x00') def sendFile(self,clientfile,serverfile): if not os.path.exists(clientfile): print('源文件/文件夹不存在') return "No such file or directory" self.action = 'upload' (status, output) = subprocess.getstatusoutput("md5sum " + clientfile + " | awk '{printf $1}'") if status == 0: self.md5sum = output else: return "md5sum error:"+status self.size = os.stat(clientfile).st_size self.serverfilePath = serverfile self.clientfilePath = clientfile ret = self.struct_pack() s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) try: s.connect(self.addr) s.send(ret) recv = s.recv(1024) if recv.decode() == 'dirNotExist': print("目标文件/文件夹不存在") return "No such file or directory" elif recv.decode() == 'ok': fo = open(clientfile, 'rb') while True: filedata = fo.read(1024) if not filedata: break s.send(filedata) fo.close() recv = s.recv(1024) if recv.decode() == 'ok': print("文件传输成功") s.close() return 0 else: s.close() return "md5sum error:md5sum is not correct!" except Exception as e: print(e) return "error:"+str(e) def recvFile(self,clientfile,serverfile): if not os.path.isdir(clientfile): filePath,fileName = os.path.split(clientfile) else: filePath = clientfile if not os.path.exists(filePath): print('本地目标文件/文件夹不存在') return "No such file or directory" self.action = 'download' self.clientfilePath = clientfile self.serverfilePath = serverfile s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) try: s.connect(self.addr) ret = self.struct_pack() s.send(ret) recv = s.recv(struct.calcsize(dataFormat)) self.struct_unpack(recv) if self.action.startswith("ok"): if os.path.isdir(clientfile): fileName = (os.path.split(serverfile))[1] clientfile = os.path.join(clientfile, fileName) self.recvd_size = 0 file = open(clientfile, 'wb') while not self.recvd_size == self.size: if self.size - self.recvd_size > 1024: rdata = s.recv(1024) self.recvd_size += len(rdata) else: rdata = s.recv(self.size - self.recvd_size) self.recvd_size = self.size file.write(rdata) file.close() print('\n等待校验...') (status, output) = subprocess.getstatusoutput("md5sum " + clientfile + " | awk '{printf $1}'") if output == self.md5sum: print("文件传输成功") else: print("文件校验不通过") (status, output) = subprocess.getstatusoutput("rm " + clientfile) elif self.action.startswith("nofile"): print('远程源文件/文件夹不存在') return "No such file or directory" except Exception as e: print(e) return "error:"+str(e)

server类代码如下

import socketimport struct,osimport subprocessimport socketserver dataFormat='8s32s100s100sl' class fileServer(socketserver.StreamRequestHandler): def struct_pack(self): ret = struct.pack(dataFormat, self.action.encode(), self.md5sum.encode(), self.clientfilePath.encode(), self.serverfilePath.encode(), self.size) return ret def struct_unpack(self, package): self.action, self.md5sum, self.clientfilePath, self.serverfilePath, self.size = struct.unpack(dataFormat, package) self.action = self.action.decode().strip('\x00') self.md5sum = self.md5sum.decode().strip('\x00') self.clientfilePath = self.clientfilePath.decode().strip('\x00') self.serverfilePath = self.serverfilePath.decode().strip('\x00') def handle(self): print('connected from:', self.client_address) fileinfo_size = struct.calcsize(dataFormat) self.buf = self.request.recv(fileinfo_size) if self.buf: self.struct_unpack(self.buf) print("get action:"+self.action) if self.action.startswith("upload"): try: if os.path.isdir(self.serverfilePath): fileName = (os.path.split(self.clientfilePath))[1] self.serverfilePath = os.path.join(self.serverfilePath, fileName) filePath,fileName = os.path.split(self.serverfilePath) if not os.path.exists(filePath): self.request.send(str.encode('dirNotExist')) else: self.request.send(str.encode('ok')) recvd_size = 0 file = open(self.serverfilePath, 'wb') while not recvd_size == self.size: if self.size - recvd_size > 1024: rdata = self.request.recv(1024) recvd_size += len(rdata) else: rdata = self.request.recv(self.size - recvd_size) recvd_size = self.size file.write(rdata) file.close() (status, output) = subprocess.getstatusoutput("md5sum " + self.serverfilePath + " | awk '{printf $1}'") if output == self.md5sum: self.request.send(str.encode('ok')) else: self.request.send(str.encode('md5sum error')) except Exception as e: print(e) finally: self.request.close() elif self.action.startswith("download"): try: if os.path.exists(self.serverfilePath): (status, output) = subprocess.getstatusoutput("md5sum " + self.serverfilePath + " | awk '{printf $1}'") if status == 0: self.md5sum = output self.action = 'ok' self.size = os.stat(self.serverfilePath).st_size ret = self.struct_pack() self.request.send(ret) fo = open(self.serverfilePath, 'rb') while True: filedata = fo.read(1024) if not filedata: break self.request.send(filedata) fo.close() else: self.action = 'nofile' ret = self.struct_pack() self.request.send(ret) except Exception as e: print(e) finally: self.request.close()

调用server,并开启服务

import fileSocketimport threadingimport socketserverimport time serverIp = '127.0.0.1'serverPort = 19821serverAddr = (serverIp,serverPort) class fileServerth(threading.Thread): def __init__(self): threading.Thread.__init__(self) self.create_time = time.time() self.local = threading.local() def run(self): print("fileServer is running...") fileserver.serve_forever() fileserver = socketserver.ThreadingTCPServer(serverAddr, fileSocket.fileServer)fileserverth = fileServerth()fileserverth.start()

调用client,发送/接受文件

import fileSocket serverIp = '127.0.0.1'serverPort = 19821serverAddr = (serverIp,serverPort) fileclient = fileSocket.fileClient(serverAddr)fileclient.sendFile('fromClientPath/file','toServerPath/file')fileclient.recvFile('toClientPath/file','fromServerPath/file')

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持。

声明:本页内容来源网络,仅供用户参考;我单位不保证亦不表示资料全面及准确无误,也不保证亦不表示这些资料为最新信息,如因任何原因,本网内容或者用户因倚赖本网内容造成任何损失或损害,我单位将不会负任何法律责任。如涉及版权问题,请提交至online#300.cn邮箱联系删除。

相关文章