54 lines
1.6 KiB
Python
54 lines
1.6 KiB
Python
|
#!/usr/bin/python3
|
|||
|
|
# coding= utf-8
|
|||
|
|
|
|||
|
|
import socket
|
|||
|
|
import sys
|
|||
|
|
import struct
|
|||
|
|
import json
|
|||
|
|
|
|||
|
|
# 本机信息
|
|||
|
|
import time
|
|||
|
|
|
|||
|
|
host_ip = socket.gethostbyname(socket.gethostname())
|
|||
|
|
# 组播组IP和端口
|
|||
|
|
mcast_group_ip = '224.1.1.1'
|
|||
|
|
mcast_group_port = 2234
|
|||
|
|
|
|||
|
|
|
|||
|
|
def receiver():
|
|||
|
|
# 建立接收 udp socket
|
|||
|
|
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM, socket.IPPROTO_UDP)
|
|||
|
|
sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
|
|||
|
|
sock.setsockopt(socket.SOL_SOCKET, socket.SO_BROADCAST, 1)
|
|||
|
|
# linux能绑定网卡这里绑定组播IP地址不会服错，windows没法绑定网卡只能绑定本网卡IP地址
|
|||
|
|
if "linux" in sys.platform:
|
|||
|
|
# 绑定到的网卡名，如果自己的不是eth0则修改
|
|||
|
|
nic_name = 0
|
|||
|
|
# 监听的组播地址
|
|||
|
|
sock.setsockopt(socket.SOL_SOCKET, 25, nic_name)
|
|||
|
|
sock.bind((mcast_group_ip, mcast_group_port))
|
|||
|
|
else:
|
|||
|
|
sock.bind((host_ip, mcast_group_port))
|
|||
|
|
# 加入组播组
|
|||
|
|
mq_request = struct.pack("=4sl", socket.inet_aton(mcast_group_ip), socket.INADDR_ANY)
|
|||
|
|
sock.setsockopt(socket.IPPROTO_IP, socket.IP_ADD_MEMBERSHIP, mq_request)
|
|||
|
|
# 设置非阻塞
|
|||
|
|
sock.setblocking(True)
|
|||
|
|
while True:
|
|||
|
|
try:
|
|||
|
|
data, address = sock.recvfrom(4096)
|
|||
|
|
data2 = json.loads(data)
|
|||
|
|
# 若组播数据不正确
|
|||
|
|
if len(data2) < 3:
|
|||
|
|
return
|
|||
|
|
except socket.error as e:
|
|||
|
|
print(f"while receive message error occur:{e}")
|
|||
|
|
else:
|
|||
|
|
print("Receive Data!")
|
|||
|
|
print("FROM: ", address)
|
|||
|
|
print("DATA: ", data.decode('utf-8'))
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
receiver()
|