在项目中,YS私有协议用到多播技术,在验证其安全特性时用到python去发送多播包,在此做个记录。
多播服务器用于向多播组发送多播数据包,其实现代码如下:
#coding:utf-8, import socket import time ANY = ‘0.0.0.0‘ SENDERPORT=1501 MCAST_ADDR = ‘224.168.2.9‘ MCAST_PORT = 1600 sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM, socket.IPPROTO_UDP) sock.bind((ANY,SENDERPORT)) #绑定发送端口到SENDERPORT,即此例的发送端口为1501 sock.setsockopt(socket.IPPROTO_IP, socket.IP_MULTICAST_TTL, 255) #设置使用多播发送 while 1: time.sleep(10) sock.sendto(‘Hello World‘, (MCAST_ADDR,MCAST_PORT) ); #将‘hello world‘发送到多播地址的指定端口,属于这个多播组的成员都可以收到这个信息
通过wireshark捕抓到的数据包如下:
多播客户端用于接收多播服务器发送的多播包,其代码如下:
#coding:utf-8, import socket import time ANY = ‘0.0.0.0‘ MCAST_ADDR = ‘224.168.2.9‘ MCAST_PORT = 1600 sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM, socket.IPPROTO_UDP) #创建UDP socket sock.setsockopt(socket.SOL_SOCKET,socket.SO_REUSEADDR,1) #允许端口复用 sock.bind((ANY,MCAST_PORT)) #绑定监听多播数据包的端口 sock.setsockopt(socket.IPPROTO_IP, socket.IP_MULTICAST_TTL, 255) #告诉内核这是一个多播类型的socket status = sock.setsockopt(socket.IPPROTO_IP, #告诉内核把自己加入指定的多播组,组地址由第三个参数指定 socket.IP_ADD_MEMBERSHIP, socket.inet_aton(MCAST_ADDR) + socket.inet_aton(ANY)); sock.setblocking(0) ts = time.time() while 1: try: data, addr = sock.recvfrom(1024) except socket.error, e: pass else: print "We got data!" print "FROM: ", addr print "DATA: ", data
多播客户端代码的关键是通过(socket.IP_ADD_MEMBERSHIP)把自己加入多播组以及监听在多播服务器指定的多播端口(MCAST_PORT),此外,多播服务器的目标地址必须是一个有效的多播地址,而不能是单播地址,比如:(localhost / INADDR_ANY / any valid IP)。