C/C++ 实现UDP发送或接收组播消息，并可指定接收发送网卡

news/2024/5/18 14:41:30 标签: c语言, c++, udp, UDP组播/单播

一、发送端代码

#include <iostream>
#include <unistd.h>
#include <stdio.h>
#include <string.h>
#include <net/if.h>
#include <netinet/in.h>
#include <netdb.h>
#include <sys/ioctl.h>
#include "UDPOperation.h"
#include "GlobalVariable.h"
#include "Logger.h"
#include "EndException.h"
#include "BaseException.h"

UDPOperation::UDPOperation(char* remote_host, int remote_port, char* interface) : fd(-1)
{
    // 创建通信的套接字
    this->remote_host = remote_host;
    this->remote_port = remote_port;
    this->interface = interface;

    memset(&(this->cliaddr), 0, sizeof(sockaddr_in));
    this->cliaddr.sin_family = AF_INET;
    this->cliaddr.sin_port = htons(this->remote_port); // 接收端需要绑定remote_port端口
}

UDPOperation::~UDPOperation() {}

bool UDPOperation::create_udpsocket()
{
    this->fd = socket(AF_INET, SOCK_DGRAM, 0);
    if (this->fd == -1)
    {
        LOG_ERROR("Socket creation failed: %s", strerror(errno));
        throw EndException(errno, strerror(errno));
    }

    inet_pton(AF_INET, this->remote_host, &this->cliaddr.sin_addr.s_addr);

    hostent* host = gethostbyname(remote_host);
    unsigned long hostip = *(unsigned long *)host->h_addr;
    
    this->cliaddr.sin_addr.s_addr = hostip;
    
    unsigned char net = hostip & 0xff;
    if (net > 223 && net < 240)  // 如果是多播
    {   
        char numeric_ip[32] = "\0";
        get_ifaddr (numeric_ip);
        struct in_addr outputif;
        outputif.s_addr = inet_addr (numeric_ip);
        LOG_INFO("interface = %s, numeric_ip = %s", interface, numeric_ip);
        if (setsockopt(this->fd, IPPROTO_IP, IP_MULTICAST_IF, (char* ) &outputif, sizeof(struct in_addr)))
        {
            throw EndException(errno, strerror(errno));
        }
    }

    return true;
}

int UDPOperation::get_ifaddr(char* addr)
{
  int sock = socket (AF_INET, SOCK_DGRAM, 0);

  struct ifreq ifr;
  memset (&ifr, 0, sizeof (ifr));
  strcpy (ifr.ifr_name, interface);
  if (ioctl (sock, SIOCGIFADDR, &ifr) < 0) {
    close (sock);
    throw EndException(errno, strerror(errno));
    return 1;
  }
  
  strcpy (addr, inet_ntoa(((struct sockaddr_in* ) &(ifr.ifr_addr))->sin_addr));
  close (sock);

  return 0;
}

void UDPOperation::destory_udpsocket()
{
    close(this->fd);
}

bool UDPOperation::send_buffer(char *buffer)
{
    socklen_t len = sizeof(struct sockaddr_in);
    // 数据广播
    int t = sendto(this->fd, buffer, SEND_UDP_PER_TSPACKET_SIZE, 0, (struct sockaddr *)&cliaddr, len);
    if (t == -1)
    {
        LOG_ERROR("Socket send failed: %s", strerror(errno));
        throw BaseException(errno, strerror(errno));
    }
    return true;
}

二、接收端代码

#include <iostream>
#include <unistd.h>
#include <stdio.h>
#include <string.h>
#include <net/if.h>
#include <netinet/in.h>
#include <netdb.h>
#include <sys/ioctl.h>
#include "UDPOperation.h"
#include "GlobalVariable.h"
#include "Logger.h"
#include "EndException.h"
#include "BaseException.h"

UDPOperation::UDPOperation(std::string remote_host, int remote_port, char* interface) : fd(-1)
{
    // 创建通信的套接字
    this->remote_host = remote_host;
    this->remote_port = remote_port;
    this->interface = interface;

    memset(&(this->cliaddr), 0, sizeof(sockaddr_in));
    this->cliaddr.sin_family = AF_INET;
    this->cliaddr.sin_addr.s_addr = inet_addr(this->remote_host.c_str());
    this->cliaddr.sin_port = htons(this->remote_port); // 接收端需要绑定remote_port端口
}

UDPOperation::~UDPOperation() {}

bool UDPOperation::create_udpsocket()
{
    this->fd = socket(AF_INET, SOCK_DGRAM, 0);
    if (this->fd == -1)
    {
        LOG_ERROR("Socket creation failed: %s", strerror(errno));
        throw EndException(errno, strerror(errno));
    }

    // 设置socket选项，允许重用地址  
    int reuse = 1;  
    if (setsockopt(this->fd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) < 0) {  
        LOG_ERROR("Error setting socket option: %s", strerror(errno));
        throw EndException(errno, strerror(errno)); 
    }  

    struct sockaddr_in local_addr;      //local address
    memset(&local_addr, 0, sizeof(local_addr));
    local_addr.sin_family = AF_INET;
    local_addr.sin_addr.s_addr = inet_addr("0.0.0.0");   // 设定本地监听必须是0.0.0.0 这里是关键！
    local_addr.sin_port = htons(remote_port);             //this port must be the group port
    //建立本地捆绑（主机地址/端口号）
    if (bind(fd, (struct sockaddr*)&local_addr, sizeof(local_addr)) != 0)
    {
        LOG_ERROR("Error binding socket: %s", strerror(errno));
        throw EndException(errno, strerror(errno)); 
    }
    
    // 如果是组播 加入组播
    int net = stoi(remote_host.substr(0, remote_host.find('.')));
    if (net >= 224 && net <= 239)
    {
        struct ip_mreq mreq;
        mreq.imr_multiaddr.s_addr = inet_addr(this->remote_host.c_str());
        if(strlen(interface) == 0){
            mreq.imr_interface.s_addr = htonl(INADDR_ANY);                //任意接口接收组播信息
        }else{
            char numeric_ip[32] = "\0";
            get_ifaddr (numeric_ip);
            LOG_INFO("interface = %s, numeric_ip = %s", interface, numeric_ip);
            mreq.imr_interface.s_addr = inet_addr(numeric_ip);    //指定新接口接收组播信息
        }

        if (setsockopt(fd, IPPROTO_IP, IP_ADD_MEMBERSHIP, &mreq, sizeof(mreq)) != 0) {
            LOG_ERROR("Error setting socket option for multicast: %s", strerror(errno));
            throw EndException(errno, strerror(errno));
        }
    }
    return true;
}

void UDPOperation::destory_udpsocket()
{
    close(this->fd);
}

int UDPOperation::recv_buffer(uint8_t *buffer, int size)
{
    socklen_t len = sizeof(struct sockaddr_in);
    int bytes_received = recvfrom(this->fd, buffer, size, 0, (struct sockaddr *)&this->cliaddr, &len);  
    if (bytes_received < 0) {  
        LOG_ERROR("Error receiving data: %s", strerror(errno));
        throw BaseException(errno, strerror(errno)); 
    }  
    return bytes_received;
}

int UDPOperation::get_ifaddr(char* addr)
{
    int sock = socket (AF_INET, SOCK_DGRAM, 0);

    struct ifreq ifr;
    memset (&ifr, 0, sizeof (ifr));
    strcpy (ifr.ifr_name, interface);
    if (ioctl (sock, SIOCGIFADDR, &ifr) < 0) {
        close (sock);
        throw EndException(errno, strerror(errno));
        return 1;
    }

    strcpy (addr, inet_ntoa(((struct sockaddr_in* ) &(ifr.ifr_addr))->sin_addr));
    close (sock);

    return 0;
}

udp_219">三、若udp组播接收不到数据可能是如下原因

# 2. 看系统有没有过滤组播包：
# 2.1 看接受组播的网卡是否过滤了：
cat /proc/sys/net/ipv4/conf/en4/rp_filter
# 如果是0， good。
# 2.2 看all网卡是否过滤了：
cat /proc/sys/net/ipv4/conf/all/rp_filter
# 如果是0， good。
# 这两个值都必须是0，才行！如果不是0，这样修改：
# 2.3 临时修改取消过滤：
sudo sysctl -w net.ipv4.conf.en4.rp_filter=0
sudo sysctl -w net.ipv4.conf.all.rp_filter=0
# 2.4 永久修改取消过滤（重启亦有效）：
sudo vi /etc/sysctl.conf
# 改为：
net.ipv4.conf.default.rp_filter=0
net.ipv4.conf.all.rp_filter=0

rp_filter参数详细介绍：
rp_filter参数有三个值，0、1、2，具体含义：

0：不开启源地址校验。

1：开启严格的反向路径校验。对每个进来的数据包，校验其反向路径是否是最佳路径。如果反向路径不是最佳路径， 则直接丢弃该数据包。

2：开启松散的反向路径校验。对每个进来的数据包，校验其源地址是否可达，即反向路径是否能通（通过任意网口），如果反向路径不同，则直接丢弃该数据包。


http://www.niftyadmin.cn/n/5140031.html

相关文章

查看局域网内另外一个电脑屏幕

查看局域网内另外一个电脑屏幕

查看局域网内另外一个电脑屏幕是一个相对简单但实用的技术。在局域网中&#xff0c;我们可以使用远程桌面、网络发现和共享、软件等技术来实现这一目标。 今天重点讲解一下&#xff0c;如何通过域之盾软件来查看另一个电脑屏幕&#xff1a; 1、部署软件&#xff0c;安装提示一…
阅读更多...
CondConv 动态卷积学习笔记 （附代码）

CondConv 动态卷积学习笔记 （附代码）

论文地址:https://arxiv.org/abs/1904.04971 代码地址&#xff1a;https://github.com/tensorflow/tpu/tree/master/models/official/efficientnet/condconv 1.是什么&#xff1f; CondConv是一种条件参数卷积&#xff0c;也称为动态卷积&#xff0c;它是一种即插即用的模块&…
阅读更多...
解决远程连接数据库缓慢的问题【图文】【非常详细】

解决远程连接数据库缓慢的问题【图文】【非常详细】

问题概述 当我们远程访问数据库&#xff0c;遇到连接不上或者连接等待时间较长&#xff0c;问题大概率就出在数据库远程链接解析的问题&#xff0c;就是在MySQL的配置文件中增加如下配置参数&#xff1a; [mysqld] skip-name-resolve 具体操作如下 解决步骤 打开mysql所在文…
阅读更多...
驱动开发11 编写iic驱动-读取温湿度数据

驱动开发11 编写iic驱动-读取温湿度数据

头文件 head.h #ifndef __HEAD_H__ #define __HEAD_H__ #define GET_HUM _IOR(m, 1, int) #define GET_TEM _IOR(m, 0, int) #endif 应用程序 si7006.c #include <stdlib.h> #include <stdio.h> #include <sys/types.h> #include <sys/stat.h> #inc…
阅读更多...
实现每栏中间穿插一个低于外部盒子高度的分割线

实现每栏中间穿插一个低于外部盒子高度的分割线

效果&#xff1a; <!DOCTYPE html> <html lang"en"><head><meta charset"UTF-8"><meta http-equiv"X-UA-Compatible" content"IEedge"><meta name"viewport" content"widthdevice-wi…
阅读更多...
torch.gather函数用法

torch.gather函数用法

torch.gather 函数用于从输入张量中收集&#xff08;或选择&#xff09;指定位置的元素&#xff0c;然后将它们放入一个新的张量中。这对于根据索引从输入张量中检索值非常有用。torch.gather 的用法如下&#xff1a; torch.gather(input, dim, index, outNone) 参数说明&…
阅读更多...
第一次写python的小程序

第一次写python的小程序

功能&#xff1a;遍历目录下的所有txt文件将*和-之间的时间戳转换为rtc时间 # -*- coding: UTF-8 -*-import sys import re import os import shutil import argparse import datetime import chardetdef get_encoding(file):with open(file, rb) as f:data f.read()return c…
阅读更多...
再谈Java类和对象

再谈Java类和对象

&#x1f388;个人主页&#xff1a;https://blog.csdn.net/m0_73920844?typeblog &#x1f302;c/java领域新星创作者 &#x1f389;欢迎&#x1f44d;点赞✍评论❤️收藏 &#x1f61b;&#x1f61b;&#x1f61b;希望我的文章能对你有所帮助&#xff0c;有不足的地方还请各位…
阅读更多...
最新文章

Copyright @ 2022~2023