目录
服务端向客户端发送消息,客户端回送文件(图片,文档等)
服务端
import socket
import os
import sys
import struct
def socket_service_image():
try:
# 创建一个tcp的socket对象
s = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
# 用socket选项设置地址重用
s.setsockopt(socket.SOL_SOCKET,socket.SO_REUSEADDR,1)
# 绑定ip,端口号
s.bind(("192.168.8.15", 6666))
# 监听连接请求,最多接受10个连接
s.listen(10)
except socket.error as msg:
print(msg)
# 退出程序
sys.exit()
print("Wait for Connection..........")
while True:
# 接收一个新的连接,并返回一个新的socket对象和客户端地址
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("\00")
# 新的文件名
new_filename = os.path.join("./",
"new_" + fn)
# 接收的文件大小为0
recvd_size = 0
# 以写入模式打开新文件
fp = open(new_filename,"wb")
while not recvd_size == filesize:
if filesize - recvd_size > 1024:
# 最多可以接收1024字节的文件数据
data = sock.recv(1024)
# 更新已接收的文件大小
recvd_size += len(data)
else:
data = sock.recv(1024)
recvd_size = filesize
# 将接收到的文件数据写入新文件
fp.write(data)
# 关闭新文件
fp.close()
# 关闭套接字
sock.close()
# 结束循环
break
if __name__ == "__main__":
socket_service_image()
客户端
import socket
import os
import sys
import struct
def sock_client_image():
while True:
try:
# 创建socket对象
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# 用socket选项设置地址重用
s.connect(('192.168.9.90', 6666))
except socket.error as msg:
print(msg)
print(sys.exit(1))
# 获取文件路径
filepath = input('input the file:')
# .pack()方法将文件名和文件大小打包成一个二进制字符串。b'128sq'是格式化字符串,表示使用128个字节存储文件名,再使用一个64位整数存储文件大小
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()
客户端向服务端发送消息,服务端回送文件(图片,文档等)
服务端
import os
import struct
from socket import *
from threading import Thread
def recv_data(client_socket, client_info):
while True:
# 接收客户端发送的数据,每次最多接收1024字节
recv_data = client_socket.recv(1024)
# 将接收到的字节数据解码为字符串,使用'gbk'编码格式
recv_content = recv_data.decode('gbk')
print(f'客户端:{recv_content}, 来自:{client_info}')
if recv_content == '图片':
# 图片路径
filepath = '3.jpg'
# .pack()方法将文件名和文件大小打包成一个二进制字符串。b'128sq'是格式化字符串,表示使用128个字节存储文件名,再使用一个64位整数存储文件大小
fhead = struct.pack(b'128sq', bytes(os.path.basename(filepath), encoding='utf-8'),
os.stat(filepath).st_size)
# 发送文件信息给客户端
client_socket.send(fhead)
fp = open(filepath, 'rb')
while True:
# 文件中读取最多1024字节的数据
data = fp.read(1024)
if not data:
print('{0} send over...'.format(filepath))
break
# 从图片文件中读取1024字节的数据
client_socket.send(data)
elif recv_content == 'end':
# 发送读取到的文件数据给客户端
client_socket.send('end'.encode('gbk'))
#
print('结束接收消息')
break
else:
# 发送结束标志给客户端
client_socket.send('你好,我是菜狗!'.encode('gbk'))
# 处理客户端连接
def handler(server_socket):
while True:
# 接受客户端连接请求,返回一个新的socket对象和客户端地址
client_socket, client_info = server_socket.accept()
# 创建一个新线程,调用recv_data函数处理客户端数据
t = Thread(target=recv_data, args=(client_socket, client_info))
# 启动
t.start()
# 等待新线程执行结束
t.join()
if __name__ == '__main__':
# 创建socket对象
server_socket = socket(AF_INET, SOCK_STREAM)
# 绑定ip,端口号
server_socket.bind(('192.168.9.36', 6551))
# 监听连接请求,最多接受10个连接
server_socket.listen(5)
print('等待链接!')
# 调用函数内的参数
handler(server_socket)
server_socket.close()
客户端
import os
import struct
from socket import *
from threading import Thread
def deal_image(sock):
while True:
# 计算图片文件的大小
file_size = struct.calcsize('128sq')
# 接收图片文件的头部信息
buf = sock.recv(file_size)
if buf:
# 解包接收到的头部信息,得到文件名和文件大小
filename, files = struct.unpack('128sq', buf)
# 解码文件名,并去除空字符
fn = filename.decode().strip('\x00')
# 新的文件名
new_filename = os.path.join('./', 'client_' + fn)
recvd_size = 0
fp = open(new_filename, 'wb')
while not recvd_size == files:
if files - recvd_size > 1024:
# 接收1024字节的文件数据
data = sock.recv(1024)
# 更新已接收的文件大小
recvd_size += len(data)
else:
data = sock.recv(1024)
# 接收完整文件
recvd_size = files
# 接收到的文件数据写入新文件
fp.write(data)
print('write over...')
# 关闭文件
fp.close()
# 关闭套接字
sock.close()
break
def send_data():
while True:
# 要发送的消息
msg = input('>')
# 将消息编码为gbk格式,并通过客户端套接字发送
client_socket.send(msg.encode('gbk'))
# 调用deal_image函数处理接收到的图片文件
deal_image(client_socket)
if msg == 'end':
print('结束发送消息')
break
if __name__ == '__main__':
# 创建客户端套接字,AF_INET为IPv4地址族,SOCK_STREAM为TCP协议
client_socket = socket(AF_INET, SOCK_STREAM)
# 绑定ip,端口号
client_socket.connect(('192.168.9.36', 6551))
# 创建一个线程
t = Thread(target=send_data)
# 启动线程
t.start()
# 等待线程结束
t.join()
client_socket.close()