Rust and C 多播案例

下面将分别用 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 实现

/* ./recv.c */
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <fcntl.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <string.h>
#include <errno.h>
#include <sys/types.h>
 
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:%s\n", times, buff);  
		//sleep(2);   
	}  
	  
	// 退出广播组 
	err = setsockopt(sockfd, IPPROTO_IP, IP_DROP_MEMBERSHIP,&mreq, sizeof(mreq));  
		  
	close(sockfd); 
 
	return 0;  
}

/* ./send.c */
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <fcntl.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <string.h>
#include <errno.h>
#include <sys/types.h>
 
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 代码，每个细节都可以自己把握，舒服！！！