Python組播

Python組播

import time
import socket
import threading
import os

# 組播組IP和端口
ANY = "0.0.0.0"
mcast_group_ip = '239.255.0.181'
mcast_group_port = 38090

def thread_recv(sock):
    while True:
        try:
            message, addr = sock.recvfrom(1024)
            print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())}: Rece{addr}: {message.decode()}')
        except :
            print("while receive message error occur")


def multicast():
    # 建立發送socket，和正常UDP數據包沒區別
    sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM, socket.IPPROTO_UDP)
    # 允許端口複用
    sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
    # 綁定監聽多播數據包的端口
    sock.bind((ANY, mcast_group_port))
    # 聲明該socket爲多播類型
    sock.setsockopt(socket.IPPROTO_IP, socket.IP_MULTICAST_TTL, 255)
    # 加入多播組，組地址由第三個參數制定
    sock.setsockopt(
        socket.IPPROTO_IP,
        socket.IP_ADD_MEMBERSHIP,
        socket.inet_aton(mcast_group_ip) + socket.inet_aton(ANY)
    )
    # 設置是否收到自己發送的消息
    sock.setsockopt(socket.IPPROTO_IP, socket.IP_MULTICAST_LOOP, 0)
    threading.Thread(target=thread_recv, args=(sock,)).start()
    # 每十秒發送一遍消息
    while True:
        try:
            message = "python multicast msg!"
            # 發送寫法和正常UDP數據包的還是完全沒區別
            # 猜測只可能是網卡自己在識別到目的ip是組播地址後，自動將目的mac地址設爲多播mac地址
            sock.sendto(message.encode(), (mcast_group_ip, mcast_group_port))
            print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())}: msg send finish.')
            time.sleep(10)
        except KeyboardInterrupt:
            print('exit.')
            sock.close()
            os._exit(0)

            

if __name__ == "__main__":
    multicast()