C语言多服务器多播组网框架

使用技术:

UDP ipv4 禁止本地回环 允许端口复用 超时等待 限制跳点 Echo

功能描述:

任意endpoint可主动发送多播,也可以收到信息后自动多播

前期准备:

开启操作系统多播广播功能,关闭系统防火墙,使用物理路由还需在路由开启广播多播功能

发送端和接收端运行不分先后 可无限复制多个接收端模拟集群

发送端:

cpp 复制代码
#define _GNU_SOURCE
#include <stdio.h>
#include <sys/socket.h>
#include <sys/un.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <sys/select.h>
#include <string.h>
#include <unistd.h>
#include <stdint.h>

#define MULTICAST_GROUP "239.0.0.1"
#define MULTICAST_PORT 55501
//发送端
int main()
{
    int server_sockfd;
    struct sockaddr_in server_sockaddr, multi_sockaddr, recv_sockaddr;
    memset(&server_sockaddr, 0, sizeof(server_sockaddr));
    memset(&multi_sockaddr, 0, sizeof(multi_sockaddr));
    memset(&recv_sockaddr, 0, sizeof(recv_sockaddr));
    socklen_t multi_sockaddr_len = sizeof(multi_sockaddr);
    socklen_t server_sockaddr_len = sizeof(server_sockaddr);
    socklen_t recv_sockaddr_len = sizeof(recv_sockaddr);
    ssize_t send_bytes, recv_bytes;
    char send_buf[1024] = "multicast from ubuntu";
    char recv_buf[1024] = {0};
    char recv_sockaddr_ip[INET_ADDRSTRLEN] = {0};
    // 用于加入多播网络
    struct ip_mreq mreq = {0};
    // 用于超时等待
    struct timeval tv = {0};
    tv.tv_sec = 1;
    tv.tv_usec = 0;
    // ipv4 udp
    server_sockfd = socket(AF_INET, SOCK_DGRAM, 0);
    if (server_sockfd == -1)
    {
        perror("socket");
    }
    // 地址端口复用
    int optval = 1;
    setsockopt(server_sockfd, SOL_SOCKET, SO_REUSEADDR, &optval, sizeof(optval));
    // 超时等待
    setsockopt(server_sockfd, SOL_SOCKET, SO_RCVTIMEO, &tv, sizeof(tv));
    // 限制跳点
    uint8_t ttl = 1;
    setsockopt(server_sockfd, IPPROTO_IP, IP_MULTICAST_TTL, &ttl, sizeof(ttl));
    // 禁止本地回环
    uint8_t loop = 0;
    setsockopt(server_sockfd, IPPROTO_IP, IP_MULTICAST_LOOP, &loop, sizeof(loop));
    // 绑定接收端口,与多播端口一致,监听任意地址
    server_sockaddr.sin_family = AF_INET;
    server_sockaddr.sin_port = htons(MULTICAST_PORT);
    server_sockaddr.sin_addr.s_addr = htonl(INADDR_ANY);
    if ((bind(server_sockfd, (struct sockaddr *)&server_sockaddr, server_sockaddr_len)) == -1)
    {
        perror("bind");
    }
    // 加入多播组
    inet_pton(AF_INET, MULTICAST_GROUP, &mreq.imr_multiaddr.s_addr);
    mreq.imr_interface.s_addr = htonl(INADDR_ANY);
    setsockopt(server_sockfd, IPPROTO_IP, IP_ADD_MEMBERSHIP, &mreq, sizeof(mreq));
    // 设置发送地址为多播组
    multi_sockaddr.sin_family = AF_INET;
    multi_sockaddr.sin_port = htons(MULTICAST_PORT);
    inet_pton(AF_INET, MULTICAST_GROUP, &multi_sockaddr.sin_addr.s_addr);
    while (1)
    {
        sleep(1);
        send_bytes = sendto(server_sockfd, send_buf, strlen(send_buf),
                            0, (struct sockaddr *)&multi_sockaddr, multi_sockaddr_len);
        if (send_bytes == -1)
        {
            perror("sendto");
        }
        recv_bytes = recvfrom(server_sockfd, recv_buf, sizeof(recv_buf),
                              0, (struct sockaddr *)&recv_sockaddr, &recv_sockaddr_len);
        if (recv_bytes == -1)
        {
            perror("recvfrom");
        }
        inet_ntop(AF_INET, &recv_sockaddr.sin_addr, recv_sockaddr_ip, sizeof(recv_sockaddr_ip));
        // 收到消息则打印日志
        if (recv_bytes > 0)
        {
            printf("INET_ADDR : %s PORT : %d Message : %s\n", recv_sockaddr_ip, ntohs(recv_sockaddr.sin_port), recv_buf);
            memset(recv_buf, 0, sizeof(recv_buf));
            memset(&recv_sockaddr, 0, sizeof(recv_sockaddr));
        }
    }
    close(server_sockfd);
    return 0;
}

接收端:

cpp 复制代码
#define _GNU_SOURCE
#include <stdio.h>
#include <sys/socket.h>
#include <sys/un.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <string.h>
#include <unistd.h>
#include <stdint.h>

#define MULTICAST_GROUP "239.0.0.1"
#define MULTICAST_PORT 55501
//接收端
int main()
{
    int server_sockfd;
    struct sockaddr_in server_sockaddr, multi_sockaddr, recv_sockaddr;
    memset(&server_sockaddr, 0, sizeof(server_sockaddr));
    memset(&multi_sockaddr, 0, sizeof(multi_sockaddr));
    memset(&recv_sockaddr, 0, sizeof(recv_sockaddr));
    socklen_t multi_sockaddr_len = sizeof(multi_sockaddr);
    socklen_t server_sockaddr_len = sizeof(server_sockaddr);
    socklen_t recv_sockaddr_len = sizeof(recv_sockaddr);
    ssize_t send_bytes, recv_bytes;
    char send_buf[1024] = "multicast from kali";
    char recv_buf[1024] = {0};
    char recv_sockaddr_ip[INET_ADDRSTRLEN] = {0};
    struct ip_mreq mreq = {0};
    server_sockfd = socket(AF_INET, SOCK_DGRAM, 0);
    if (server_sockfd == -1)
    {
        perror("socket");
    }
    // 禁止本地回环
    uint8_t loop = 0;
    setsockopt(server_sockfd, IPPROTO_IP, IP_MULTICAST_LOOP, &loop, sizeof(loop));
    // 允许端口复用
    int optval = 1;
    setsockopt(server_sockfd, SOL_SOCKET, SO_REUSEADDR, &optval, sizeof(optval));
    // 绑定地址为监听任意,绑定监听端口为多播端口
    server_sockaddr.sin_family = AF_INET;
    server_sockaddr.sin_addr.s_addr = htonl(INADDR_ANY);
    server_sockaddr.sin_port = htons(MULTICAST_PORT);
    if ((bind(server_sockfd, (struct sockaddr *)&server_sockaddr, server_sockaddr_len)) == -1)
    {
        perror("bind");
    }

    // 加入多播组
    mreq.imr_multiaddr.s_addr = inet_addr(MULTICAST_GROUP);
    mreq.imr_interface.s_addr = htonl(INADDR_ANY);
    setsockopt(server_sockfd, IPPROTO_IP, IP_ADD_MEMBERSHIP, &mreq, sizeof(mreq));
    // 向多播组发送消息用
    multi_sockaddr.sin_family = AF_INET;
    multi_sockaddr.sin_port = htons(MULTICAST_PORT);
    inet_pton(AF_INET, MULTICAST_GROUP, &multi_sockaddr.sin_addr.s_addr);
    while (1)
    {
        recv_bytes = recvfrom(server_sockfd, recv_buf, sizeof(recv_buf),
                              0, (struct sockaddr *)&recv_sockaddr, &recv_sockaddr_len);
        if (recv_bytes == -1)
        {
            perror("recvfrom");
        }
        inet_ntop(AF_INET, &recv_sockaddr.sin_addr, recv_sockaddr_ip, sizeof(recv_sockaddr_ip));
        // 如果收到了消息就打印
        if (recv_bytes > 0)
        {
            printf("INET_ADDR : %s PORT : %d Message : %s\n", recv_sockaddr_ip, ntohs(recv_sockaddr.sin_port), recv_buf);
            memset(recv_buf, 0, sizeof(recv_buf));
            memset(&recv_sockaddr, 0, sizeof(recv_sockaddr));
        }
        sleep(1);
        // echo一条消息,也可主动发起
        send_bytes = sendto(server_sockfd, send_buf, strlen(send_buf),
                            0, (struct sockaddr *)&multi_sockaddr, multi_sockaddr_len);
        if (send_bytes == -1)
        {
            perror("sendto");
        }
    }
    close(server_sockfd);
    return 0;
}
相关推荐
迎風吹頭髮2 小时前
UNIX下C语言编程与实践58-UNIX TCP 连接处理:accept 函数与新套接字创建
c语言·网络·unix
Predestination王瀞潞4 小时前
IO操作(Num22)
开发语言·c++
IT成长日记4 小时前
【Nginx开荒攻略】Nginx虚拟主机配置:从域名、端口到IP的完整指南
linux·运维·服务器·nginx·虚拟主机
宋恩淇要努力5 小时前
C++继承
开发语言·c++
迎風吹頭髮6 小时前
UNIX下C语言编程与实践53-UNIX 共享内存控制:shmctl 函数与共享内存管理
服务器·c语言·unix
迎風吹頭髮6 小时前
Linux内核架构浅谈8-Linux内核与UNIX的传承:设计思想与特性差异
linux·运维·架构
猫头虎6 小时前
如何查看局域网内IP冲突问题?如何查看局域网IP环绕问题?arp -a命令如何使用?
网络·python·网络协议·tcp/ip·开源·pandas·pip
黑马金牌编程6 小时前
Linux 服务器常见的性能调优
linux·运维·服务器·性能优化
jieyu11196 小时前
网络、主机安全扫描工具
linux·安全·系统安全
沿着路走到底6 小时前
python 基础
开发语言·python