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

基于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)