python写数据库校验_基于python3实现socket文件传输和校验

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

来源：中文源码网    浏览： 次    日期：2018年9月2日

【下载文档:  基于python3实现socket文件传输和校验.txt 】

(友情提示:右键点上行txt文档名->目标另存为)

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

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

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

socket数据通过struct模块打包

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

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

client类代码如下

import socket

import struct,os

import 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 socket

import struct,os

import subprocess

import 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 fileSocket

import threading

import socketserver

import time

serverIp = '127.0.0.1'

serverPort = 19821

serverAddr = (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 = 19821

serverAddr = (serverIp,serverPort)

fileclient = fileSocket.fileClient(serverAddr)

fileclient.sendFile('fromClientPath/file','toServerPath/file')

fileclient.recvFile('toClientPath/file','fromServerPath/file')以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持中文源码网。

亲,试试微信扫码分享本页! *^_^*