下面将分别用 Rust 和 C 实现一个 Linux 上的多播案例：发送端循环发送十次数据，接受端循环接收十次数据并打印到屏幕。
Rust 实现// ./send/src/main.rs
use std::net::UdpSocket;
use std::thread;
fn main() {
let socket = UdpSocket::bind("127.0.0.1:9999").unwrap();
const COUNT: usize = 100;
for i in 0..10u8 {
let buf = [i + 0x41; COUNT];
socket.send_to(&buf, "127.0.0.1:8888").unwrap();
}
thread::sleep(std::time::Duration::from_millis(1000));
}
// ./recv/src/main.rs
use std::net::{UdpSocket, Ipv4Addr};
fn main() {
let socket = UdpSocket::bind("127.0.0.1:8888").unwrap();
let multicast_addr = Ipv4Addr::new(234, 2, 2, 2);
let inter = Ipv4Addr::new(0,0,0,0);
socket.join_multicast_v4(&multicast_addr, &inter).unwrap();
let mut buf = [0u8; 65536];
for _ in 0..10 {
let (amt, src) = socket.recv_from(&mut buf).unwrap();
println!("{}: {:?}", amt, src);
}
socket.leave_multicast_v4(&multicast_addr, &inter).unwrap();
}
C 实现
#include#include #include #include #include #include #include #include #include #include int main(int argc, char*argv[]) { int sockfd; // 套接字文件描述符 struct sockaddr_in local_addr; // 本地地址 int err = -1; char group[16] = "224.0.1.88"; // 多播组 IP sockfd = socket(AF_INET, SOCK_DGRAM, 0); //建立套接字 if (sockfd == -1) { perror("socket()"); return -1; } // 初始化地址 memset(&local_addr, 0, sizeof(local_addr)); local_addr.sin_family = AF_INET; local_addr.sin_addr.s_addr = htonl(INADDR_ANY); local_addr.sin_port = htons(8000); // 绑定socket err = bind(sockfd,(struct sockaddr*)&local_addr, sizeof(local_addr)); if(err < 0) { perror("bind()"); return -2; } struct ip_mreq mreq; // 多播地址结构体 // 加入多播组，相当于创建一个QQ群，某人加入此群 mreq.imr_multiaddr.s_addr = inet_addr(group); // 多播地址，类似于 QQ 群号 mreq.imr_interface.s_addr = htonl(INADDR_ANY);// 将本机加入多播组，类似于某人加入此群 // 加入多播组 err = setsockopt(sockfd, IPPROTO_IP, IP_ADD_MEMBERSHIP,&mreq, sizeof(mreq)); if (err < 0) { perror("setsockopt():IP_ADD_MEMBERSHIP"); return -4; } int times = 0; int addr_len = 0; char buff[256] = {0}; int n = 0; // 循环接收广播组的消息，5次后退出 for(times = 0; times<10; times++) { addr_len = sizeof(local_addr); memset(buff, 0, sizeof(buff)); // 接收数据 n = recvfrom(sockfd, buff, sizeof(buff), 0,(struct sockaddr*)&local_addr, &addr_len); if( n== -1) { perror("recvfrom()"); } printf("Recv %dst message from server:%sn", times, buff); //sleep(2); } // 退出广播组 err = setsockopt(sockfd, IPPROTO_IP, IP_DROp_MEMBERSHIP,&mreq, sizeof(mreq)); close(sockfd); return 0; }
#include运行结果#include #include #include #include #include #include #include #include #include int main(int argc, char*argv) { int sockfd; // 套接字文件描述符 struct sockaddr_in dest_addr; // 目标ip char buf[] = "multicast test!"; sockfd = socket(AF_INET, SOCK_DGRAM, 0); // 建立套接字 if (sockfd == -1) { perror("socket()"); return -1; } // 初始化目标 ip 信息 memset(&dest_addr, 0, sizeof(dest_addr)); dest_addr.sin_family = AF_INET; dest_addr.sin_addr.s_addr = inet_addr("224.0.1.88"); // 目的地址，为多播地址 dest_addr.sin_port = htons(8000); // 多播服务器的端口也是 8000 // 向多播地址发送数据 for(int times = 0; times<10; times++) { int n = sendto(sockfd, buf, strlen(buf), 0,(struct sockaddr*)&dest_addr, sizeof(dest_addr)); if( n < 0) { perror("sendto()"); return -2; } //sleep(1); } return 0; }
先执行接收程序，在执行发送程序。发送端没有任何打印，只看接收端。Rust 执行结果如下：
[wlb@Arch recv]$ cargo run
Compiling recv v0.1.0 (/home/wlb/Documents/code/rust/mutilcast/recv)
Finished dev [unoptimized + debuginfo] target(s) in 0.18s
Running `target/debug/recv`
100: 127.0.0.1:9999
100: 127.0.0.1:9999
100: 127.0.0.1:9999
100: 127.0.0.1:9999
100: 127.0.0.1:9999
100: 127.0.0.1:9999
100: 127.0.0.1:9999
100: 127.0.0.1:9999
100: 127.0.0.1:9999
100: 127.0.0.1:9999
[wlb@Arch recv]$
C 执行结果如下：
[wlb@Arch recv]$ ./recv1 Recv 0st message from server:multicast test! Recv 1st message from server:multicast test! Recv 2st message from server:multicast test! Recv 3st message from server:multicast test! Recv 4st message from server:multicast test! Recv 5st message from server:multicast test! Recv 6st message from server:multicast test! Recv 7st message from server:multicast test! Recv 8st message from server:multicast test! Recv 9st message from server:multicast test! [wlb@Arch recv]$
Rust 和 C 都实现了这个案例，并成功运行。
Rust 的实现使用了更少的代码，可读性也很高。它的开发效率应该不低，项目维护也会更容易。加上它的零成本抽象，程序执行起来并不会比 C 慢多少。所以个人觉得 Rust 还是值得一学的，况且 Linux 内核中已经加入了 Rust 的代码，未来可期吧。
不过我还是喜欢 C 代码，每个细节都可以自己把握，舒服！！！
