tcp大文件传输

tcp大文件传输

基本的tcp服务端与客户端

服务端

# 服务端
import socket

# 创建tcp套接字
tcp_server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# bind方法（地址元组） 设置服务器的地址和端口号
tcp_server.bind(("127.0.0.1", 7980))

# listen()方法设置监听,普通tcp套接字,变化为tcp监听套接字
tcp_server.listen(128)

# 套接字.accept() 获取双向连接和客户端地址
conn, client_addr = tcp_server.accept()

while True:

    # 使用双向链接对向，收消息。
    # 双向连接.recv(设置获取字节长度)
    # decode()字符串转字节格式
    msg = conn.recv(100).decode("utf8")

    if msg == "q":
    	# 关闭双向连接
        conn.close()
        break
        
    print(msg)
    # 发消息
    send_msg = input("服务端回复：")
    conn.send(send_msg.encode("utf8"))

tcp_server.close()

客户端

# 客户端
import socket

# 创建tcp套接字
tcp_client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

ip = "127.0.0.1"  # ip地址
port = 7980  # 端口号
server_add = (ip, port)

# 套接字.connect（服务器的地址元组）
tcp_client.connect(server_add)

while True:
    # 发消息：  套按字.send（消息内容）
    send_msg = input("客户端发送：")
    if send_msg == "exit":
        tcp_client.send('q'.encode("utf8"))
        # 关闭套接字
        tcp_client.close()
        break
    tcp_client.send(send_msg.encode("utf8"))

    # 收消息：   套按字.recv（接收长度）
    recv_data = tcp_client.recv(1024)
    print(recv_data.decode("utf8"))

传输大文件的tcp服务端与客户端

服务端

# 服务端
import socket
import struct

tcp_server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
tcp_server.bind(("127.0.0.1", 7980))

tcp_server.listen(128)

conn, client_addr = tcp_server.accept()
overall_len = 0  # 当前已接收文件的长度
# 开启文件操作
f = open("拷贝版.exe", "ab")
while True:
    print('服务端第一次收数据，接收发送长度')
    """
		struct模块
		rec = struct.pack("i", 整数)     > 打包整数类型，生成的是四个字节的bytes数据
		rec2 = struct.unpack("i", ret)     > 解包整数类型，获得的是一个元组。整数在零号索引位
		rec2是元组（整数，）
	"""
    rec = conn.recv(4)
    rec2 = struct.unpack("i", rec)
    print(rec2[0])
    overall_len += rec2[0]
    # 判断文件是否接收完毕
    if rec2[0] != 0:
        print('服务端第二次收数据，收正文内容')
        msg = conn.recv(rec2[0])
        f.write(msg)
    else:
        print("接收完成")
        break

f.close()
conn.close()
print("共接收长度为%d字节的数据,（%.2fKB,%.2fM）" % (overall_len, overall_len/1024, overall_len/1024/1024))
tcp_server.close()

客户端

# 客户端
import socket
import struct

# sock.SOCK_STREAM
tcp_client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

ip = "127.0.0.1"
port = 7980
server_add = (ip, port)

# 套接字.connect（服务器的地址）
tcp_client.connect(server_add)

# 发消息：  套按字.send（消息内容）
overall_len = 0
f = open("PowerWord.800.12012.exe", "rb")
while True:
	# 一次读取最大长度为1024字节的数据
	# 文件指针在已读取的数据之后，指针在哪就从哪里开始读取
    c = f.read(1024)
    print(c)  # 当前读取的数据
    # 发送将要发送的文件数据长度
    len_bytes = struct.pack("i", len(c))
    tcp_client.send(len_bytes)
    overall_len += len(c)
    # 如果将要发送的文件数据的长度为0则退出循环
    if len(c) == 0:
        print("结束")
        break

    tcp_client.send(c)
    print("发送了长度为%d的数据" % len(c))

print("共发送了长度为%d字节的数据,（%.2fKB,%.2fM）" % (overall_len, overall_len/1024, overall_len/1024/1024))
tcp_client.close()