您的位置：首页 > 脚本大全 > > 正文

python处理tcp包（Python3使用TCP编写一个简易的文件下载器功能）

更多时间：2021-10-04 01:18:59 类别：脚本大全浏览量：2362

python处理tcp包

Python3使用TCP编写一个简易的文件下载器功能

利用python3来实现tcp协议，和udp类似。udp应用于及时通信，而tcp协议用来传送文件、命令等操作，因为这些数据不允许丢失，否则会造成文件错误或命令混乱。下面代码就是模拟客户端通过命令行操作服务器。客户端输入命令，服务器执行并且返回结果。

tcp（transmission control protocol 传输控制协议）:是一种面向连接的、可靠的、基于字节流的传输层通信协议，由ietf的rfc 793定义。

使用tcp编写一个简易的文件下载器要求：需编写文件下载器服务端和文件下载器客户端

服务器端代码：

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67 import socket

import os

import threading

# 处理客户端请求下载文件的操作（从主线程提出来的代码）

def deal_client_request(ip_port, service_client_socket):

# 连接成功后，输出“客户端连接成功”和客户端的ip和端口

print("客户端连接成功", ip_port)

# 接收客户端的请求信息

file_name = service_client_socket.recv(1024)

# 解码

file_name_data = file_name.decode("utf-8")

# 判断文件是否存在

if os.path.exists(file_name_data):

#输出文件字节数

fsize = os.path.getsize(file_name_data)

#转化为兆单位

fmb = fsize/float(1024*1024)

#要传输的文件信息

senddata = "文件名：%s 文件大小：%.2fmb"%(file_name_data,fmb)

#发送和打印文件信息

service_client_socket.send(senddata.encode("utf-8"))

print("请求文件名：%s 文件大小：%.2f mb"%(file_name_data,fmb))

#接受客户是否需要下载

options = service_client_socket.recv(1024)

if options.decode("utf-8") == "y":

# 打开文件

with open(file_name_data, "rb") as f:

#　计算总数据包数目

nums = fsize/1024

#　当前传输的数据包数目

cnum = 0

while true:

file_data = f.read(1024)

cnum = cnum + 1

jindu = cnum/nums*100

print("当前已下载：%.2f%%"%jindu,end = "\r")

if file_data:

# 只要读取到数据，就向客户端进行发送

service_client_socket.send(file_data)

# 数据读完，退出循环

else:

print("请求的文件数据发送完成")

break

else:

print("下载取消！")

else:

print("下载的文件不存在！")

# 关闭服务当前客户端的套接字

service_client_socket.close()

if __name__ == '__main__':

# 把工作目录切换到data目录下

os.chdir("./data")

# 创建套接字

tcp_server_socket = socket.socket(socket.af_inet, socket.sock_stream)

# 绑定端口号

tcp_server_socket.bind(("", 3356))

# 设置监听，将主动套接字变为被动套接字

tcp_server_socket.listen(128)

# 循环调用accept，可以支持多个客户端同时连接，和多个客户端同时下载文件

while true:

service_client_socket, ip_port = tcp_server_socket.accept()

# 连接成功后打印套接字号

#print(id(service_client_socket))

# 创建子线程

sub_thread = threading.thread(target=deal_client_request, args=(ip_port, service_client_socket))

# 启动子线程

sub_thread.start()

客户端代码：

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50 # 多任务文件下载器客户端

import socket

if __name__ == '__main__':

# 创建套接字

tcp_client_socket = socket.socket(socket.af_inet, socket.sock_stream)

# 和服务端连接

server_ip = input("输入服务器ip：")

tcp_client_socket.connect((server_ip, 3356))

# 发送下载文件的请求

file_name = input("请输入要下载的文件名：")

# 编码

file_name_data = file_name.encode("utf-8")

# 发送文件下载请求数据

tcp_client_socket.send(file_name_data)

#　接收要下载的文件信息

file_info = tcp_client_socket.recv(1024)

# 文件信息解码

info_decode = file_info.decode("utf-8")

print(info_decode)

#获取文件大小

fileszie = float(info_decode.split('：')[2].split('mb')[0])

fileszie2 = fileszie*1024

# 是否下载？输入ｙ　确认　输入ｑ取消

opts = input("是否下载？(y 确认　q 取消)")

if opts == 'q':

print("下载取消！程序退出")

else:

print("正在下载　》》》")

#向服务器确认正在下载

tcp_client_socket.send(b'y')

# 把数据写入到文件里

with open("./" + file_name, "wb") as file:

#目前接收到的数据包数目

cnum = 0

while true:

# 循环接收文件数据

file_data = tcp_client_socket.recv(1024)

# 接收到数据

if file_data:

# 写入数据

file.write(file_data)

cnum = cnum+1

jindu =cnum/fileszie2*100

print("当前已下载：%.2f%%"%jindu,end = "\r")

# 接收完成

else:

print("下载结束！")

break

# 关闭套接字

tcp_client_socket.close()