You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
60 lines
2.0 KiB
60 lines
2.0 KiB
import os
|
|
import socket
|
|
import zipfile
|
|
|
|
def receive_zip(HOST,PORT):
|
|
# 获取当前文件的绝对路径和目录路径
|
|
current_file_path = os.path.abspath(__file__)
|
|
current_directory = os.path.dirname(current_file_path)
|
|
|
|
# 创建一个socket对象
|
|
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as server_socket:
|
|
# 绑定IP和端口
|
|
server_socket.bind((HOST, PORT))
|
|
# 开始监听传入的连接
|
|
server_socket.listen()
|
|
|
|
print(f"等待客户端连接...")
|
|
conn, addr = server_socket.accept() # 接受客户端的连接
|
|
print(f"已连接：{addr}")
|
|
|
|
with conn:
|
|
# 从客户端接收数据
|
|
encrypted_data_path = os.path.join(current_directory, 'encrypted_data.zip')
|
|
with open(encrypted_data_path, 'wb') as file:
|
|
while True:
|
|
file_data = conn.recv(1024)
|
|
if not file_data:
|
|
break
|
|
file.write(file_data)
|
|
|
|
print("文件接收完成")
|
|
|
|
# 解压缩 ZIP 文件
|
|
output_folder = os.path.join(current_directory)
|
|
os.makedirs(output_folder, exist_ok=True)
|
|
|
|
with zipfile.ZipFile(encrypted_data_path, 'r') as zip_file:
|
|
zip_file.extractall(output_folder)
|
|
|
|
print("解压缩完成到目标文件夹：", output_folder)
|
|
|
|
def send_Bpubkey(HOST,PORT):
|
|
# 获取当前文件的绝对路径和目录路径
|
|
current_file_path = os.path.abspath(__file__)
|
|
current_directory = os.path.dirname(current_file_path)
|
|
|
|
# 文件路径
|
|
Bpubkey_path = os.path.join(current_directory, 'B_public.txt')
|
|
|
|
# 客户端连接服务器
|
|
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as client_socket:
|
|
client_socket.connect((HOST, PORT))
|
|
|
|
# 读取要传输的文件
|
|
with open(Bpubkey_path, 'rb') as file:
|
|
file_data = file.read()
|
|
|
|
# 发送文件数据
|
|
client_socket.sendall(file_data)
|
|
print("文件发送完成") |