UDP组播

import socket
import sys
import time

DSTPORT = 11111     # 报文的目的端口
SRCPORT = 22222     # 报文的源端口
SRCADDR = '172.16.170.4'   # 发送报文的网卡 ip
MULTICAST = '239.0.0.1'    # 组播组 ip


class MultiCastSend:

    def __init__(self):
        self.socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
        self.multicast_group = (MULTICAST, DSTPORT)
        self.socket.setsockopt(socket.IPPROTO_IP, socket.IP_MULTICAST_TTL, 255)
        self.socket.setsockopt(socket.IPPROTO_IP, socket.IP_ADD_MEMBERSHIP,
                               socket.inet_aton(MULTICAST) + socket.inet_aton(SRCADDR))

    def send_multicast_data(self, data: bytes):
        self.socket.sendto(data, self.multicast_group)

    def close(self):
        self.socket.close()


if __name__=="__main__":
    send_obj = MultiCastSend()
    content = b'\x01\x02\x03\x04'
    i = 0
    while True:
        if i <= 100:
            send_obj.send_multicast_data(content)
            print("send successful")
            i += 1
            time.sleep(2)
        else:
            print("100 次循环结束")
            send_obj.close()
            sys.exit(0)

import socket
import struct

class MultiCastRecv:


    def __init__(self,multicastip:str,port:int,timeout=5):
        self.socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
        multicast_group = multicastip
        address = ('', port)
        group = socket.inet_aton(multicast_group)
        mreq = struct.pack('4sL', group, socket.INADDR_ANY)
        test_socket = self.socket
        test_socket.setsockopt(socket.IPPROTO_IP, socket.IP_ADD_MEMBERSHIP, mreq)
        test_socket.bind(address)
        test_socket.settimeout(timeout)
        self.data = None


    def recv_data(self):
         self.data,addr = self.socket.recvfrom(4096)


if __name__ == "__main__":
    recv_obj = MultiCastRecv('239.0.0.1', 11111)
    while True:
        recv_obj.recv_data()
        content = recv_obj.data
        print(f'Recv data is {content}')