如何用Tokio模拟多设备UDP通信并获取正确客户端地址？

我有一个可接收多个独立设备连接的服务，希望在运行该服务的同一机器上模拟这些设备。需要一个模拟程序，能监听服务发送的消息，并以对应接收设备的IP地址进行回复。

当前使用的Rust代码如下：

use anyhow::Result;
use std::net::{Ipv4Addr, SocketAddrV4};
use tokio::net::UdpSocket;

#[tokio::main]
async fn main() -> Result<()> {
    let local_addr = SocketAddrV4::new(Ipv4Addr::new(0, 0, 0, 0), 14551);
    let socket = UdpSocket::bind(local_addr).await?;

    let mut buf = [0; 1024];
    loop {
        let (len, addr) = socket.recv_from(&mut buf).await?;

        println!(
            "Received packet from {} to {}: {:?}",
            addr.ip(),
            socket.local_addr()?.ip(),
            &buf[..len],
        );
    }
}

运行后输出：

Received packet from 192.168.1.27 to 0.0.0.0: [253, 9, 0, 0, 156, 0, 255, 0, 0, 0, 0, 0, 0, 0, 6, 8, 0, 0, 3, 173, 1]

疑问：如何获取服务发送消息时的目标IP（而非0.0.0.0）？是否可以通过tokio实现，还是必须手动解析接收的帧并注入响应帧？

解决方案 #

1. 为什么当前显示0.0.0.0 #

你绑定的是0.0.0.0:14551，这表示监听所有可用网络接口的14551端口。socket.local_addr()?.ip()返回的是绑定的通配地址，而非数据包实际的目标IP。

2. 获取UDP数据包的目标IP（Tokio实现方式） #

Tokio默认的UdpSocket::recv_from仅返回发送方地址，要获取目标IP，需借助操作系统的socket扩展选项。通过socket2库开启IP_RECVDSTADDR（IPv4）或IPV6_RECVDSTADDR（IPv6）选项，即可拿到数据包的准确目标地址，无需手动解析帧。

步骤1：添加依赖

在Cargo.toml中加入以下依赖：

[dependencies]
anyhow = "1.0"
tokio = { version = "1.0", features = ["full"] }
socket2 = "0.5"

步骤2：修改代码实现

use anyhow::Result;
use socket2::{Domain, Protocol, Socket, Type};
use std::net::{Ipv4Addr, SocketAddr, SocketAddrV4};
use tokio::net::UdpSocket;

#[tokio::main]
async fn main() -> Result<()> {
    // 创建底层Socket并配置选项
    let socket = Socket::new(Domain::IPV4, Type::DGRAM, Some(Protocol::UDP))?;
    socket.set_reuse_address(true)?;
    socket.set_reuse_port(true)?;
    // 开启接收目标地址的功能
    socket.set_ip_recv_dstaddr(true)?;

    // 绑定到通配地址和指定端口
    let local_addr = SocketAddrV4::new(Ipv4Addr::new(0, 0, 0, 0), 14551).into();
    socket.bind(&local_addr)?;

    // 转换为Tokio的UdpSocket
    let socket = UdpSocket::from_std(socket.into())?;

    let mut buf = [0; 1024];
    loop {
        // 使用socket2的方法获取发送方、目标方地址
        let (len, src_addr, dst_addr) = socket2::Socket::from(socket.as_ref())
            .recv_from_with_dst_addr(&mut buf)?;

        println!(
            "Received packet from {} to {}: {:?}",
            src_addr.ip(),
            dst_addr.ip(),
            &buf[..len],
        );

        // 模拟设备回复：以目标IP作为回复的源IP（需系统支持绑定多IP）
        let response = b"Simulated device response";
        socket.send_to(response, src_addr).await?;
    }
}

3. 模拟多设备的实践建议 #

• 若机器有多个虚拟网卡IP，可为每个模拟设备绑定独立IP，服务发送到对应IP时，模拟程序可精准识别并回复。
• 若只有单物理IP，可通过不同端口区分设备（需服务支持按端口识别设备），或使用SO_BINDTODEVICE选项绑定到不同虚拟网络设备。

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