1 from socket import *
2 #创建套接字
3 udp_server = socket(AF_INET,SOCK_DGRAM)
4
5 msg_server = ("",8877)
6 #绑定ip地址和端口
7 udp_server.bind(msg_server)
8
9 while True:
10 #接受消息,注意此处用的是 recvfrom()
11 msg_client = udp_server.recvfrom(1024)
12
13 print("新客户端已连接--->>>")
14
15 if len(msg_client) != 0:
16 #msg_client中有两个值,第一个表示收到的消息的内容,第二个表示客户端的IP地址和端口的信息
17 print("%s:%s" % (msg_client[1], msg_client[0].decode("utf-8")))
18 else:
19 break
20
21 udp_server.close()
from socket import *
sock = socket(AF_INET,SOCK_DGRAM)
addr_msg = ("192.168.1.104",8877)
#连接服务端
sock.connect(addr_msg)
while True:
msg_send = input("请输入要发送的信息:")
if msg_send =='q':
break
#注意此处用的是sendto() 其中第一个参数表示要发送的消息的内容,第二个参数表示服务端的IP地址和端口的元组
sock.sendto(msg_send.encode("utf-8"),addr_msg)
sock.close()
1 from socket import *
2 import struct
3 import os
4
5 def main():
6 file_name = input("请输入要下载的文件名:")
7 #建立一个UDP的套接字
8 udpsocket = socket(AF_INET,SOCK_DGRAM)
9 #创建一个tftp的下载请求 H表示第一个参数占用两个字节,d表示对应的参数为bite类型
10 # %ds 表示文件名占用的字符个数,如后面的octet占用了5个字符,所以写成5s
11 # octet是tftp传输的一种模式,它决定了传输数据的格式,还有其他的几种tftp传输的模式
12 #构造包使用的是struck中的pack ,其中的 H 表示使参数占用两个字节
13 request_header = struct.pack("!H%dsb5sb"%len(file_name),1,file_name,0,"octet",0)
14
15 server_msg = ("localhost",8888)
16 #发送一个下载文件的请求
17 udpsocket.sendto(request_header,server_msg)
18
19 f = open(file_name,"w")
20
21 # num表示的是数据块的编号,这里使得其初始值为0
22 num = 0
23 #此处建立一个标记,以确认传输过程中没发生错误
24 mask = True
25
26 while True:
27 # 接收服务端发来的数据包
28 response_data = udpsocket.recvfrom(1024)
29 recv_data, server_info = response_data
30
31 # 此处用struct中的unpack来解包 H 表示数据占用两个字节长度
32 # 这里的前两个字节中存储的是操作码
33 operation_num = struct.unpack("!H",recv_data[:2])
34 # 这里的第三个和第四个字节中存储的是数据块的编号
35 package_num = struct.unpack("!H",recv_data[2:4])
36
37 # 操作码为3时表示接收到的信息是服务端收到响应而发给客户端的数据包
38 if operation_num[0] == 3:
39 #这一次收到的值应该是在上一次收到的值的基础上加上一后的结果
40 num += 1
41 #此处表示一旦num的值超过了它2个字节所表示的值的范围,则让它从0开始再来重复一遍(主要是因为操作码只能占2个字节)
42 if num == 65535:
43 num = 0
44 # 进行块编号的确认
45 if num == package_num[0]: #此处取下标是因为此时的package_num表示的是一个元组,而块编号是元组中的第一个数据
46 # 将收到的具体数据写入到文件中
47 f.write(recv_data[4:])
48 #块编号的确认每次都要在之前的值上加1
49 num = package_num[0]
50 #构造一个响应包 操作码为4 发送的数据就是加上1后的块编号
51 ack_data = struct.pack("!HH",4,package_num[0])
52 #发送响应包
53 udpsocket.sendto(ack_data, server_info)
54
55 elif operation_num[0] == 5:
56 print("Error!")
57 #若发生错误信息,则将标记的值改成False , 以便后续操作删除这个文件
58 mask = False
59
60 # 一个数据包的最大长度为 4+512 = 516 位,如果数据包小于这个数值,则表明这已经是最后一次的数据传输了
61 if len(recv_data) < 516:
62 print("Cmplete!")
63 break
64 # mask标记的值为True则表示文件传输的过程中没有发生错误,就将本地接收到的文件关闭保存
65 if mask == True:
66 f.close()
67
68 else:
69 # mask标记的值为False则表示文件传输的过程中出现了错误信息,则将已经接收到的文件删除
70 os.remove(file_name)
71
72 if __name__ == "__main":
73 main()