【Linux网络编程】组播代码

head.h

//head.h
#include <stdio.h>
#include <unistd.h>
#include <errno.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <stdlib.h>
#include <fcntl.h>
#include <strings.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <string.h>

客户端（接收方）

//客户端（接收方）
#include "head.h"

#define errlog(err) do{perror(err); exit(1);} while(0)
#define N 32

int main(int argc, const char *argv[])
{
    int sockfd;
    struct sockaddr_in groupcastaddr;//广播地址变量
    socklen_t len = sizeof(groupcastaddr);
    char buf[N] = {0};

    //设定通信标准，基于UDP的广播发送
    sockfd = socket(AF_INET, SOCK_DGRAM, 0);
    if(sockfd < 0)
    {
        errlog("fail to socket");
    }

    //加入到组播地址中
    struct ip_mreq mreq;
    socklen_t mreq_len = sizeof(mreq); 
    mreq.imr_multiaddr.s_addr = inet_addr(argv[1]); //组播地址
    mreq.imr_interface.s_addr = htonl(INADDR_ANY);//如果你只有一个网卡，本地IP地址可以填写 INADDR_ANY，
                                                  //即从默认的网口接收组播数据包，如果你有不止一块网卡，并且它们连接了不同的网络，
                                                  //你必须指定某个网卡的IP地址，从那个网络接口接收组播数据包。
    //添加本地IP到组播地址中
    setsockopt(sockfd, IPPROTO_IP, IP_ADD_MEMBERSHIP, &mreq, mreq_len);



    //2、填充信息结构体
    groupcastaddr.sin_family = AF_INET;
    groupcastaddr.sin_port = htons(atoi(argv[2]));
    groupcastaddr.sin_addr.s_addr = htonl(INADDR_ANY);//必须将IP地址设置为 INADDR_ANY ，因为组播数据包的目的IP地址不确定，
                                                      //是由IP地址与MAC地址组合而来，因此接收端必须有能力接收来自任何IP地址的数据包
    //inet_aton("192.168.8.129", &groupcastaddr.sin_addr);

    //3、绑定信息结构体
    if(bind(sockfd, (struct sockaddr*)&groupcastaddr, len) < 0)
    {
        errlog("fail to bind");
    }
    while(1)
    {

        //从局域网中接收数据
        recvfrom(sockfd, buf, N, 0, (struct sockaddr*)&groupcastaddr, &len);
        if(strncmp(buf, "quit", 4) == 0)
        {
            //setsockopt(sockfd, IPPROTO_IP, IP_DROP_MEMBERSHIP, &mreq, mreq_len);
            break;
        }
        printf("--> %s \n", buf);
    }



    close(sockfd);
    return 0;
}

服务器（发送方）

//服务器（发送方）
#include "head.h"

#define errlog(err) do{perror(err); exit(1);} while(0)
#define N 32

int main(int argc, const char *argv[])
{
    int sockfd;
    struct sockaddr_in groupcastaddr;//广播地址变量
    socklen_t len = sizeof(groupcastaddr);
    char buf[N] = {0};

    //1、设定通信标准，基于UDP的广播发送
    sockfd = socket(AF_INET, SOCK_DGRAM, 0);
    if(sockfd < 0)
    {
        errlog("fail to socket");
    }
    //2、填充网络信息结构体
    groupcastaddr.sin_family = AF_INET;
    groupcastaddr.sin_port = htons(atoi(argv[2]));//设置端口号
    groupcastaddr.sin_addr.s_addr = inet_addr(argv[1]);//设置组播地址


    while(1)
    {
        fgets(buf, N, stdin);
        buf[strlen(buf) - 1] = '\0';

        //发送数据到组播地址中
        sendto(sockfd, buf, N, 0, (struct sockaddr*)&groupcastaddr, len);
    }

    close(sockfd);
    return 0;
}