国产探花免费观看_亚洲丰满少妇自慰呻吟_97日韩有码在线_资源在线日韩欧美_一区二区精品毛片,辰东完美世界有声小说,欢乐颂第一季,yy玄幻小说排行榜完本

首頁 > 編程 > Python > 正文

基于python3實現(xiàn)socket文件傳輸和校驗

2020-01-04 14:48:38
字體:
供稿:網(wǎng)友

基于socket的文件傳輸并進(jìn)行MD5值校驗,供大家參考,具體內(nèi)容如下

文件傳輸分為兩個類,一個是服務(wù)端,一個是客戶端。

客戶端發(fā)起發(fā)送文件或接收文件的請求,服務(wù)端收到請求后接收或發(fā)送文件,最后進(jìn)行MD5值的校驗

socket數(shù)據(jù)通過struct模塊打包

需要發(fā)送文件到服務(wù)端時,調(diào)用sendFile函數(shù),struct包內(nèi)包含文件信息、文件大小、文件MD5等信息,服務(wù)端接收到文件后進(jìn)行MD5值校驗,校驗成功后則返回成功

需要從服務(wù)器下載文件時,調(diào)用recvFile函數(shù),收到文件后進(jìn)行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("目標(biāo)文件/文件夾不存在")    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('本地目標(biāo)文件/文件夾不存在')   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('遠(yuǎn)程源文件/文件夾不存在')    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()

調(diào)用server,并開啟服務(wù)

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

調(diào)用client,發(fā)送/接受文件

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')

以上就是本文的全部內(nèi)容,希望對大家的學(xué)習(xí)有所幫助,也希望大家多多支持VEVB武林網(wǎng)。


注:相關(guān)教程知識閱讀請移步到python教程頻道。
發(fā)表評論 共有條評論
用戶名: 密碼:
驗證碼: 匿名發(fā)表
主站蜘蛛池模板: 富蕴县| 治县。| 五台县| 太白县| 金沙县| 旬阳县| 保康县| 淮滨县| 淅川县| 巴彦淖尔市| 景宁| 满洲里市| 扶风县| 桦甸市| 兰州市| 琼结县| 贡山| 西和县| 莱芜市| 定日县| 罗田县| 东宁县| 北川| 镇平县| 百色市| 宁海县| 五寨县| 阳山县| 郸城县| 颍上县| 武平县| 五华县| 商都县| 永胜县| 无锡市| 桓仁| 巴林左旗| 滨海县| 常州市| 陈巴尔虎旗| 金堂县|