Python Socket 文件传输实现:基于 struct 打包文件名与大小
本文介绍如何使用 Python 的 socket 和 struct 模块实现简单的文件传输功能。通过 struct 打包文件名和文件大小信息,确保服务端能准确解析文件头,避免传输错误。代码包含完整的客户端和服务端示例,演示了基于 TCP 的文件发送与接收流程,适用于学习网络编程基础。
Socket 文件传输原理
利用 socket 结合 struct 模块,可以实现客户端向服务端发送文件的功能。struct 用于处理二进制数据的打包与解包,确保文件元数据(如文件名和大小)能准确传输。
服务端实现代码
服务端代码主要监听指定 IP 和端口,接收客户端连接后,使用 struct 解析文件头信息。文件头包含文件名(12字节)和文件大小(长整型)。解析后,服务端根据文件名保存文件,并循环接收剩余数据块,直到文件传输完成。若连接超时,则关闭连接。
客户端实现代码
客户端代码负责获取用户输入的文件路径,检查文件是否存在。若文件存在,客户端使用 struct 将文件名和文件大小打包成二进制数据发送给服务端。随后,客户端以固定大小(如1024字节)循环读取文件内容并发送,直到文件全部发送完毕,最后提示传输成功。
# -*- coding: UTF-8 -*-
import socket, time, socketserver, struct, os, _thread
host = '127.0.0.1'
port = 12307
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) # 定义socket类型
s.bind((host, port)) # 绑定需要监听的Ip和端口号,tuple格式
s.listen(1)
def conn_thread(connection, address):
while True:
try:
connection.settimeout(600)
fileinfo_size = struct.calcsize('12sl')#12s表示12个字符,l表示一个长整型数
buf = connection.recv(fileinfo_size)
if buf: # 如果不加这个if,第一个文件传输完成后会自动走到下一句,需要拿到文件大小信息才可以继续执行
filename, filesize = struct.unpack('12sl', buf)
filename_f = filename.decode("utf-8").strip('\00') # C语言中'\0'是一个ASCII码为0的字符,在python中表示占一个位置得空字符
filenewname = os.path.join('e:\\', os.path.basename(filename_f))
print(u'文件名称:%s , 文件大小: %s' % (filenewname, filesize))
recvd_size = 0 # 定义接收了的文件大小
file = open(filenewname, 'wb')
print(u"开始传输文件内容")
while not recvd_size == filesize:
if filesize - recvd_size > 1024:
rdata = connection.recv(1024)
recvd_size += len(rdata)
else:
rdata = connection.recv(filesize - recvd_size)
recvd_size = filesize
file.write(rdata)
file.close()
print('receive done')
# connection.close()
except socket.timeout:
connection.close()
while True:
print(u"开始进入监听状态")
connection, address = s.accept()
print('Connected by ', address)
# thread = threading.Thread(target=conn_thread,args=(connection,address)) #使用threading也可以
# thread.start()
_thread.start_new_thread(conn_thread, (connection, address))
s.close()
运行效果说明
运行服务端后,控制台会显示监听状态及连接信息。当客户端发送文件时,服务端会打印接收到的文件名和文件大小,并显示传输进度或结果。
# -*- coding: UTF-8 -*-
import socket, os, struct
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.connect(('127.0.0.1', 12307))
while True:
filepath = input('请输入要传输的文件绝对路径:\r\n')
print(type(filepath))
print(len(filepath.encode("utf-8")))
if os.path.isfile(filepath):
#fileinfo_size = struct.calcsize('20sl') # 定义打包规则
# 定义文件头信息,包含文件名和文件大小
fhead = struct.pack('12sl', filepath.encode("utf-8"), os.stat(filepath).st_size)
print(os.stat(filepath).st_size)
s.send(fhead)
print (u'文件路径: ', filepath)
# with open(filepath,'rb') as fo: 这样发送文件有问题,发送完成后还会发一些东西过去
fo = open(filepath, 'rb')
while True:
filedata = fo.read(1024)
if not filedata:
break
s.send(filedata)
fo.close()
print (u'传输成功')
# s.close()

客户端在输入有效文件路径后,会显示文件路径及大小信息。传输完成后,控制台会输出“传输成功”的提示,表明文件已完整发送至服务端。

以上即为基于 Python Socket 和 struct 模块实现文件传输的完整示例。该方案通过明确的文件头定义,解决了单纯发送文件数据可能导致的解析混乱问题,适合用于简单的内网文件传输场景。希望本文内容能对读者有所帮助,欢迎继续支持本站获取更多技术资源。
本站声明
本文内容仅供参考,不代表本站观点。转载或引用请注明出处。如发现内容问题,请通过联系我们页面反馈。













