基于 UDP 实现多播数据传输

一、发送端

多播发送端与UDP客户端的处理流程类似：

#include <arpa/inet.h>
#include <netinet/in.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <unistd.h>

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#define BUF_MAX_SIZE 1024

int main(int argc, char *argv[]) {
    int send_sockfd;                   // 发送方套接字描述符
    char buffer[BUF_MAX_SIZE];         // 数据缓冲区
    struct sockaddr_in multicast_addr; // 多播组套接字信息

    /* 参数检查 */
    if (argc != 3) {
        printf("Command format : %s <multicast IP> <port>!\n", argv[0]);
        exit(1);
    }

    /* 获取发送方套接字文件描述符 */
    if ((send_sockfd = socket(AF_INET, SOCK_DGRAM, 0)) == -1) {
        perror("socket error");
        exit(1);
    }

    /* 初始化多播组套接字信息 */
    memset(&multicast_addr, 0, sizeof(struct sockaddr_in));
    multicast_addr.sin_family = AF_INET;                 // 使用IPv4
    multicast_addr.sin_addr.s_addr = inet_addr(argv[1]); // 指定多播组IP
    multicast_addr.sin_port = htons(atoi(argv[2]));      // 对端口进行字节序转化

    /* 设定发送方发送的数据包的TTL值 */
    int ttl = 32;
    setsockopt(send_sockfd, IPPROTO_IP, IP_MULTICAST_TTL, (void *) &ttl, sizeof(ttl));

    /* 循环向多播组中发送消息 */
    while (1) {
        memset(buffer, 0, BUF_MAX_SIZE);
        printf("[sender] Input multicast message : ");
        scanf("%[^\n]%*c", buffer);
        sendto(send_sockfd, buffer, strlen(buffer), 0, (struct sockaddr *) &multicast_addr, sizeof(multicast_addr));
    }

    return 0;
}

---

二、接收端

多播接收端与UDP服务端的处理流程类似：

#include <arpa/inet.h>
#include <netinet/in.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <unistd.h>

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#define BUF_MAX_SIZE 1024

int main(int argc, char *argv[]) {
    int recv_sockfd;                   // 接收方套接字描述符
    char buffer[BUF_MAX_SIZE];         // 数据缓冲区
    struct sockaddr_in multicast_addr; // 多播组套接字信息
    struct ip_mreq multicast_group;    // 多播组信息

    /* 参数检查 */
    if (argc != 3) {
        printf("Command format : %s <multicast IP> <port>!\n", argv[0]);
        exit(1);
    }

    /* 获取接收方套接字文件描述符 */
    if ((recv_sockfd = socket(AF_INET, SOCK_DGRAM, 0)) < 0) {
        perror("socket error");
        exit(1);
    }

    /* 初始化多播组套接字信息 */
    memset(&multicast_addr, 0, sizeof(struct sockaddr_in));
    multicast_addr.sin_family = AF_INET;                 // 使用IPv4
    multicast_addr.sin_addr.s_addr = inet_addr(argv[1]); // 指定多播组IP
    multicast_addr.sin_port = htons(atoi(argv[2]));      // 对端口进行字节序转化

    /* 绑定多播组套接字信息 */
    if (bind(recv_sockfd, (struct sockaddr *) (&multicast_addr), sizeof(struct sockaddr_in)) < 0) {
        perror("bind error");
        exit(1);
    }

    /* 将本机加入多播组 */
    multicast_group.imr_multiaddr.s_addr = inet_addr(argv[1]); // 多播组需要使用D类IP地址，其范围为224.0.0.0-239.255.255.255
    multicast_group.imr_interface.s_addr = INADDR_ANY;         // 加入多播组的主机地址信息
    setsockopt(recv_sockfd, IPPROTO_IP, IP_ADD_MEMBERSHIP, (void *) &multicast_addr, sizeof(multicast_addr));

    /* 循环接收多播消息 */
    while (1) {
        memset(buffer, 0, BUF_MAX_SIZE);
        recvfrom(recv_sockfd, buffer, sizeof(buffer), 0, NULL, 0);
        printf("[receiver] multicast message : %s\n", buffer);
    }

    return 0;
}

---

三、运行结果

atreus@MacBook-Pro % gcc sender.c -o sender
atreus@MacBook-Pro % ./sender 224.0.0.1 1521     
[sender] Input multicast message : a
[sender] Input multicast message : b
[sender] Input multicast message : c
[sender] Input multicast message : ^C
atreus@MacBook-Pro % 

atreus@MacBook-Pro % gcc receiver.c -o receiver
atreus@MacBook-Pro % ./receiver 224.0.0.1 1521 
[receiver] multicast message : a
[receiver] multicast message : b
[receiver] multicast message : c
^C
atreus@MacBook-Pro %