在Python 3中获取接收数据包源IP地址的方法求助

嘿，我刚好折腾过类似的需求，给你分享一个亲测有效的实现方案，应该能完美解决你的问题！

核心思路其实很简单：用UDP套接字监听多播组，接收数据包时通过recvfrom()方法直接获取发送方的源IP和端口，然后用这个地址回复单播，同时把源IP存入集合（自动去重）来维护发送者列表。

完整代码实现 #

import socket
import struct

# 配置多播参数
MULTICAST_GROUP = '224.0.0.1'  # 常用的测试多播地址，可根据需求修改
MULTICAST_PORT = 5000
LOCAL_INTERFACE = ''  # 监听所有本地接口，多网卡环境可指定具体IP

# 用集合存储发送者IP，自动去重
sender_ips = set()

def main():
    # 创建UDP套接字
    sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    
    # 设置地址复用，避免端口被占用的问题
    sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
    
    # 绑定到指定端口，监听所有接口
    sock.bind((LOCAL_INTERFACE, MULTICAST_PORT))
    
    # 加入多播组，这一步是接收多播消息的关键
    mreq = struct.pack('4s4s', socket.inet_aton(MULTICAST_GROUP), socket.inet_aton(LOCAL_INTERFACE))
    sock.setsockopt(socket.IPPROTO_IP, socket.IP_ADD_MEMBERSHIP, mreq)
    
    print(f"开始监听多播组 {MULTICAST_GROUP}:{MULTICAST_PORT}... 按Ctrl+C终止程序")
    
    try:
        while True:
            # 接收数据，返回（数据内容，发送方地址）
            data, sender_addr = sock.recvfrom(1024)  # bufsize可根据消息大小调整
            sender_ip, sender_port = sender_addr
            
            # 打印收到的消息
            print(f"\n收到来自 {sender_ip}:{sender_port} 的消息: {data.decode('utf-8')}")
            
            # 记录发送者IP
            sender_ips.add(sender_ip)
            print(f"当前已记录的发送者IP列表: {list(sender_ips)}")
            
            # 发送单播回复
            response = f"已收到你的消息，来自 {sender_ip} 的请求已处理！".encode('utf-8')
            sock.sendto(response, sender_addr)
            print(f"已向 {sender_ip}:{sender_port} 发送回复")
            
    except KeyboardInterrupt:
        print("\n程序终止，最终记录的发送者IP列表:")
        for idx, ip in enumerate(sender_ips, 1):
            print(f"{idx}. {ip}")
    finally:
        # 退出多播组并关闭套接字
        sock.setsockopt(socket.IPPROTO_IP, socket.IP_DROP_MEMBERSHIP, mreq)
        sock.close()

if __name__ == "__main__":
    main()

关键细节说明 #

• 获取源IP：recvfrom()方法会返回两个值，第二个值sender_addr是一个元组(源IP, 源端口)，直接提取第一个元素就能拿到发送者的IP地址。
• 去重记录IP：用集合set()存储IP，自动避免重复记录同一个发送者的IP。如果需要保留发送顺序，可以改用列表+手动去重的方式。
• 多播组加入：必须通过IP_ADD_MEMBERSHIP选项让套接字加入指定的多播组，否则无法接收多播消息。多网卡环境下，要把LOCAL_INTERFACE设为对应网卡的IP。
• 单播回复：直接调用sendto()方法，把回复内容和发送方的地址传进去，就能实现精准的单播回复。

注意事项 #

• 确保本地防火墙允许UDP的MULTICAST_PORT端口通信，否则可能收不到多播消息。
• 如果发送的消息包含非文本内容，要调整解码/编码方式，比如使用data直接处理二进制数据。
• 可以根据实际需求修改bufsize（recvfrom()的参数），如果消息较大，建议设置为4096或更大。

内容的提问来源于stack exchange，提问作者Nika Benashvili