1. struct是什么?
在Python中,struct
模块提供了一种处理字节序列的方式。它允许你执行打包(pack)和解包(unpack)操作,用于将数据转换为字节流以及从字节流中提取数据。这对于处理二进制数据、网络通信和与底层系统进行交互非常有用。
服务端:
# 导入模块
import socket
import os
import sys
import struct
# 定义了一个名为 socket_service_image 的函数,用于执行文件接收的服务器逻辑。
def socket_service_image():
# 捕获异常
try:
# 创建TCP套接字
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# 设置套接字选项,允许重新使用地址。
s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
# 绑定IP地址和端口号
s.bind(("192.168.10.138", 6666))
# 设置最大监听数量
s.listen(5)
# 如果在绑定过程中发生错误,捕获异常。
except socket.error as msg:
# 打印绑定错误的消息。
print(msg)
# 退出程序并返回状态码1,表示绑定失败。
sys.exit(1)
# 打印等待连接的消息。
print("wait for Connection...........................")
# 无限循环,等待客户端连接。
while True:
# 接受客户端连接,并返回一个新的套接字对象和客户端地址。
sock, addr = s.accept()
# 定义了一个名为 deal_image 的函数,用于处理接收到的客户端连接。
deal_image(sock, addr)
def deal_image(sock, addr):
# 打印接受到客户端连接的消息,显示客户端的地址信息。
print("Accept connection from {0}".format(addr))
# 无限循环,接收客户端发送的文件数据。
while True:
# 计算文件头部信息的字节大小。
fileinfo_size = struct.calcsize('128sq')
# 接收文件头部信息。
buf = sock.recv(fileinfo_size)
# 判断 如果接收到了文件头部信息。
if buf:
# 解包文件头部信息,得到文件名和文件大小。
filename, filesize = struct.unpack('128sq', buf)
# 解码文件名并去除末尾的空字符。
fn = filename.decode().strip('\x00')
# 生成保存文件的新路径和文件名。
new_filename = os.path.join('./', 'new_' + fn)
# 初始化已接收的文件大小为0。
recvd_size = 0
# 以二进制写入模式打开新文件。
fp = open(new_filename, 'wb')
# 循环接收文件数据,直到接收完整个文件。
while not recvd_size == filesize:
# 如果剩余文件大小大于1024字节。
if filesize - recvd_size > 1024:
# 接收1024字节的文件数据。
data = sock.recv(1024)
# 增加已接收的文件大小。
recvd_size += len(data)
# 如果剩余文件大小不足1024字节。
else:
# 接收剩余的文件数据。
data = sock.recv(1024)
# 更新已接收的文件大小。
recvd_size = filesize
# 写入文件数据。
fp.write(data)
# 关闭文件。
fp.close()
# 关闭套接字连接。
sock.close()
# 跳出循环。
break
# 这个条件判断语句用于检查是否直接执行了当前脚本。
if __name__ == '__main__':
# 调用 socket_service_image 函数,开始执行服务器程序逻辑。
socket_service_image()
客户端:
# 导入所需的模块。
import socket
import os
import sys
import struct
# 定义了一个名为 sock_client_image 的函数,用于执行文件发送的客户端逻辑。
def sock_client_image():
# 无限循环,确保客户端可以持续发送文件。
while True:
# 捕获异常
try:
# 创建一个TCP套接字对象。
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# 监听IP和端口号
s.connect(("192.168.56.106", 6696))
# 如果连接过程中发生错误,捕获异常。
except socket.error as msg:
# 打印连接错误的消息。
print(msg)
# 退出程序并返回状态码1,表示连接失败。
print(sys.exit(1))
# 获取用户输入的文件路径。
filepath = input("input the file:")
# 使用 struct.pack() 函数打包文件名和文件大小,将其作为文件头部信息发送给服务器。
# 文件名通过 os.path.basename(filepath) 获取,文件大小通过 os.stat(filepath).st_size 获取。
fhead = struct.pack(b'128sq', bytes(os.path.basename(filepath), encoding='utf-8'), os.stat(filepath).st_size)
# 将文件头部信息发送到服务器。
s.send(fhead)
# 以二进制只读模式打开用户指定的文件。
fp = open(filepath, 'rb')
# 循环读取文件内容并发送给服务器。
while True:
# 从文件中读取1024字节的数据。
data = fp.read(1024)
# 如果没有读取到数据,表示文件已经发送完毕。
if not data:
# 打印文件发送完成的消息。
print('{0} send over...'.format(filepath))
# 跳出循环
break
s.send(data)
# 关闭套接字连接
s.close()
# 这个条件判断语句用于检查是否直接执行了当前脚本。
if __name__ == '__main__':
# 调用 sock_client_image 函数,开始执行客户端程序逻辑。
sock_client_image()