基于python3实现socket文件传输和校验
作者:m4_Sean 发布时间:2023-06-30 08:49:57
标签: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')
来源:https://blog.csdn.net/qq_20468343/article/details/77938802
0
投稿
猜你喜欢
- 导读:由于banner一般用于专题类网站,在门户网站的二级页面,用户进来之前,在首页已经对主题有一定的了解和认识,所以banner的作用是在
- Python通过yield提供了对协程的基本支持,但是不完全。而第三方的gevent为Python提供了比较完善的协程支持。gevent是第
- 一、安装一个基于Python的强大的信号库,它既支持简单的对象到对象通信,也支持针对多个对象进行组播支持注册全局命名信号,支持自定义命名信号
- 1、矩阵加法使用a = np.random.random((3,3))b = np.random.randint(0,9,(3,3))ad
- 当我们的程序报错时,解释器会将整个异常的堆栈信息全部输出出来,举个例子:def foo(): &
- 安装 Tesseract OCRTesseract OCR 是一款由 Google 团队开发的开源 OCR(Optical Characte
- 本文实例讲述了Python函数装饰器实现方法。分享给大家供大家参考,具体如下:编写函数装饰器这里主要介绍编写函数装饰器的相关内容。跟踪调用如
- 浏览网页的时候经常会碰到一些不认识的英文单词,或者想知道一些中文单词的翻译,这时候再去找翻译软件或者翻译网站就有些麻烦了。因此我做了一个“中
- 一。初识单元测试1)定义:单元:函数或者是类单元测试:测试类或者函数python内置的单元测试框架:unittest2)单元测试的意义好处:
- 一般语言都提供了按字典排序的API,比如跟微信公众平台对接时就需要用到字典排序。按字典排序有很多种算法,最容易想到的就是字符串搜索的方式,但
- python基础pandas的drop()用法做数据处理得时候用到了pandas,体验不错,记录如下:import pandas as pd
- SQLSRV驱动程序允许您创建一个结果集,其中包含可以根据游标类型以任何顺序访问的行。本主题将讨论客户端(缓冲)和服务器端(非缓冲)游标及其
- 1.使用jobsName.ini文件保存要创建job的名字jobs1jobs2jobs32.使用Jenkins创建job时自动生成的conf
- 首先我们供给一台主机要进行的步骤就是对其主机端口的扫描,查看其中开放的端口。我们首先创建一个TCP的全连接的扫描器。我们使用socket来创
- 下面的示例看看这三个函数的具体的区别,其中var_dump和var_export比较少用,但他们两者又很相似。所以可以看看:<?php
- 在 settings.py 中添加以下内容:LOGGING = { 'version': 1,
- 工作中需要根据某个应用程序具体吃了多少内存来决定执行某些操作,所以需要写个小工具来模拟应用程序使用内存情况,下面是我写的一个Python脚本
- 最近,QQ的办公版本——TIM进行了一次更新升级。本次更新升级大幅修改了界面的样式,看起来更加的清爽、简洁和高效了。这种界面州的先生还是比较
- 自执行函数:自动执行的函数。它在被解释时就已经在运行了。一般函数都是在被调用时才会执行的。 自执行函数的一般格式:(function() {
- 写在之前我们都知道 Python 中内置了许多标准的数据结构,比如列表,元组,字典等。与此同时标准库还提供了一些额外的数据结构,我们可以基于