前几天已经讲过了UDP协议下大文件传输的python实现代码,今天再实现TCP协议下大文件传输的python实现代码。
TCP和UDP的实现过程还是比较不一样的。
实现代码:
服务端:
import socket
import time
s = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
s.bind(('127.0.0.1',9999))
s.listen(5)
print('Waiting for connection...')
while True:
sock,addr = s.accept()
print('Accept new connection from %s:%s...' % addr)
if count == 0:
data1 = sock.recv(1024)
print(str(data1))
file_total_size = int(data1.decode())
received_size = 0
sock.send('received'.encode())
data = sock.recv(1024)
filepath = str(data.decode())
f = open(filepath, 'wb')
while received_size < file_total_size:
data = sock.recv(1024)
f.write(data)
received_size += len(data)
print('已接收 ',received_size,' Byte')
data = sock.recv(1024)
if data == b'end':
break
f.close()
s.close()
客户端:
import socket
import os
import time
filename = input('please enter the filename you want to send:\n')
filesize = str(os.path.getsize(filename))
fname1, fname2 = os.path.split(filename)
client_addr = ('127.0.0.1',9999)
f = open(filename,'rb')
count = 0
flag = 1
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
#建立连接:
s.connect(('127.0.0.1', 9999))
while True:
if count == 0:
s.send(filesize.encode())
start = time.time()
s.recv(1024)
s.send(fname2.encode())
for line in f:
s.send(line)
print('sending...')
s.send(b'end')
break
s.close
end = time.time()
print('cost' + str(round(end - start, 2)) + 's')
运行截图:
服务器端:
客户端:
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持脚本之家。