Python UDP组播接收端解码异常及字符数限制问题求助

问题现象 #

编写Python UDP组播程序时，发送端运行正常，但接收端出现解码问题：

• 输入超过两个字符时，触发ValueError: too many values to unpack (expected 2)
• 输入两个字符时，触发AttributeError: 'int' object has no attribute 'decode'

错误信息 #

输入超过两个字符时：

Traceback (most recent call last):
  File "/home/felix/PycharmProjects/Multicast UDP programming/client1.py", line 27, in <module>
    data, addr = sock.recv(1024)
ValueError: too many values to unpack (expected 2)

输入两个字符时：

Traceback (most recent call last):
  File "/home/felix/PycharmProjects/Multicast UDP programming/client1.py", line 33, in <module>
    msg = data.decode()
AttributeError: 'int' object has no attribute 'decode'

相关代码 #

Sender.py #

import socket

group = '224.1.1.1'
port = 15000

# 数据包经过2跳后不再转发
ttl = 2

# 创建socket
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM, socket.IPPROTO_UDP)

# 设置为阻塞模式
sock.setblocking(True)

# 设置组播TTL
sock.setsockopt(socket.IPPROTO_IP, socket.IP_MULTICAST_TTL, ttl)

# 获取用户输入
input_msg = input("Enter your message: ")

# 发送数据包
sock.sendto(input_msg.encode(), (group, port))

sock.close()

Receiver.py #

import socket
import struct

cast_grp = socket.gethostbyname('224.1.1.1')
cast_port = 15000

# 创建UDP socket
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM, socket.IPPROTO_UDP)
sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)

# 绑定到组播端口
sock.bind(('', cast_port))

# 加入组播组
mreq = struct.pack("4sl", socket.inet_aton(cast_grp), socket.INADDR_ANY)
sock.setsockopt(socket.IPPROTO_IP, socket.IP_ADD_MEMBERSHIP, mreq)

# 设置为阻塞模式
sock.setblocking(True)

# 禁用组播回环
sock.setsockopt(socket.IPPROTO_IP, socket.IP_MULTICAST_LOOP, 0)

while True:
    # 接收数据包
    try:
        data, addr = sock.recv(1024)
    except socket.error:
        # 无数据时继续
        continue

    # 解码数据
    msg = data.decode()

    # 打印消息
    print(f"{addr}:{msg}")
    break

# 关闭socket
sock.close()

问题原因 #

核心错误出在接收端的sock.recv(1024)调用：

• UDP套接字的recv()方法仅返回接收到的字节数据，不会返回发送方地址。
• 当尝试将其解包为data, addr时，Python会把字节串视为可迭代对象拆分：
  • 若字节串长度大于2，拆分出的元素数量超过预期，触发too many values to unpack错误；
  • 若字节串长度恰好为2，会拆成两个int类型的单个字节，此时data是int值，调用decode()自然触发属性错误。

UDP接收数据的正确方式是使用recvfrom()方法，它会返回包含(data, addr)的元组，其中addr是发送方的地址信息。

修复后的Receiver.py代码 #

import socket
import struct

cast_grp = socket.gethostbyname('224.1.1.1')
cast_port = 15000

# 创建UDP socket
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM, socket.IPPROTO_UDP)
sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)

# 绑定到组播端口
sock.bind(('', cast_port))

# 加入组播组
mreq = struct.pack("4sl", socket.inet_aton(cast_grp), socket.INADDR_ANY)
sock.setsockopt(socket.IPPROTO_IP, socket.IP_ADD_MEMBERSHIP, mreq)

# 设置为阻塞模式
sock.setblocking(True)

# 禁用组播回环
sock.setsockopt(socket.IPPROTO_IP, socket.IP_MULTICAST_LOOP, 0)

while True:
    # 使用recvfrom替代recv，获取数据和发送方地址
    try:
        data, addr = sock.recvfrom(1024)
    except socket.error:
        # 无数据时继续
        continue

    # 解码数据
    msg = data.decode()

    # 打印消息
    print(f"{addr}:{msg}")
    break

# 关闭socket
sock.close()

内容的提问来源于stack exchange，提问作者Felix Njuguna