加入收藏 | 设为首页 | 会员中心 | 我要投稿 李大同 (https://www.lidatong.com.cn/)- 科技、建站、经验、云计算、5G、大数据,站长网!
当前位置: 首页 > 综合聚焦 > 服务器 > Linux > 正文

Linux上的UDP connect()和recv()

发布时间:2020-12-14 01:48:56 所属栏目:Linux 来源:网络整理
导读:根据connect(2)手册页 If the socket sockfd is of type SOCK_DGRAM then serv_addr is the address to which datagrams are sent by default, and the only address from which datagrams are received . If the socket is of type SOCK_STREAM or SOCK_SEQ
根据connect(2)手册页

If the socket sockfd is of type SOCK_DGRAM then serv_addr is the address to which datagrams are sent by default,and the only address from which datagrams are received. If the socket is of type SOCK_STREAM or SOCK_SEQPACKET,this call attempts to make a connection to the socket that is bound to the address specified by serv_addr.

我正在尝试过滤来自同一端口上广播的两个不同多播组的数据包,我认为connect()已经完成了这项工作,但我无法使其工作.事实上,当我将它添加到我的程序时,我没有收到任何数据包.更多信息在这个thread.

这是我设置连接参数的方式:

memset(&mc_addr,sizeof(mc_addr));
mc_addr.sin_family = AF_INET;
mc_addr.sin_addr.s_addr = inet_addr(multicast_addr);
mc_addr.sin_port = htons(multicast_port);
printf("Connecting...n");
if( connect(sd,(struct sockaddr*)&mc_addr,sizeof(mc_addr)) < 0 ) {
  perror("connect");
  return -1;
}

printf("Receiving...n");
while( (len = recv(sd,msg_buf,sizeof(msg_buf),0)) > 0 )
  printf("Received %d bytesn",len);

解决方法

您的程序(可能)有以下问题:

>你应该使用bind()而不是connect(),和
>你缺少setsockopt(…,IP_ADD_MEMBERSHIP,…).

这是一个接收多播的示例程序.它使用recvfrom(),而不是recv(),但它是相同的,除了你还得到每个接收数据包的源地址.

要从多个多播组接收,您有三个选项.

第一个选项:为每个多播组使用单独的套接字,并将每个套接字bind()绑定到多播地址.这是最简单的选择.

第二个选项:为每个多播组使用单独的套接字,bind()每个套接字INADDR_ANY,并使用套接字过滤器过滤掉除一个多播组之外的所有组播组.

因为您已绑定到INADDR_ANY,您可能仍会获得其他多播组的数据包.可以使用内核的套接字过滤器过滤掉它们:

#include <stdint.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#include <linux/filter.h>

/**
 * Adds a Linux socket filter to a socket so that only IP
 * packets with the given destination IP address will pass.
 * dst_addr is in network byte order.
 */
int add_ip_dst_filter (int fd,uint32_t dst_addr)
{
    uint16_t hi = ntohl(dst_addr) >> 16;
    uint16_t lo = ntohl(dst_addr) & 0xFFFF;

    struct sock_filter filter[] = {
        BPF_STMT(BPF_LD + BPF_H + BPF_ABS,SKF_NET_OFF + 16),// A <- IP dst high
        BPF_JUMP(BPF_JMP + BPF_JEQ + BPF_K,hi,3),// if A != hi,goto ignore
        BPF_STMT(BPF_LD + BPF_H + BPF_ABS,SKF_NET_OFF + 18),// A <- IP dst low
        BPF_JUMP(BPF_JMP + BPF_JEQ + BPF_K,lo,1),// if A != lo,goto ignore
        BPF_STMT(BPF_RET + BPF_K,65535),// accept
        BPF_STMT(BPF_RET + BPF_K,0)                          // ignore
    };

    struct sock_fprog fprog = {
        .len = sizeof(filter) / sizeof(filter[0]),.filter = filter
    };

    return setsockopt(fd,SOL_SOCKET,SO_ATTACH_FILTER,&fprog,sizeof(fprog));
}

第三种选择:使用单个套接字接收所有多播组的多播.

在这种情况下,您应该为每个组执行IP_ADD_MEMBERSHIP.这样您就可以在单个套接字上获取所有数据包.

但是,您需要额外的代码来确定接收到的数据包发往哪个组播组.要做到这一点,你必须:

>使用recvmsg()接收数据包并读取IP_PKTINFO或等效的辅助数据消息.但是,为了让recvmsg()给你这个消息,你首先必须这样做
>使用setsockopt()启用IP_PKTINFO辅助数据消息的接收.

您需要做的确切事情取决于IP协议版本和操作系统.这是我如何做到的(未经测试的IPv6代码):enabling PKTINFO和reading the option.

这是一个接收多播的简单程序,它演示了第一个选项(绑定到多播地址).

#include <stdio.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>

#define MAXBUFSIZE 65536

int main (int argc,char **argv)
{
    if (argc != 4) {
        printf("Usage: %s <group address> <port> <interface address>n",argv[0]);
        return 1;
    }

    int sock,status,socklen;
    char buffer[MAXBUFSIZE+1];
    struct sockaddr_in saddr;
    struct ip_mreq imreq;

    // set content of struct saddr and imreq to zero
    memset(&saddr,sizeof(struct sockaddr_in));
    memset(&imreq,sizeof(struct ip_mreq));

    // open a UDP socket
    sock = socket(PF_INET,SOCK_DGRAM,0);
    if (sock < 0) {
        perror("socket failed!");
        return 1;
    }

    // join group
    imreq.imr_multiaddr.s_addr = inet_addr(argv[1]);
    imreq.imr_interface.s_addr = inet_addr(argv[3]);
    status = setsockopt(sock,IPPROTO_IP,(const void *)&imreq,sizeof(struct ip_mreq));

    saddr.sin_family = PF_INET;
    saddr.sin_port = htons(atoi(argv[2]));
    saddr.sin_addr.s_addr = inet_addr(argv[1]);
    status = bind(sock,(struct sockaddr *)&saddr,sizeof(struct sockaddr_in));
    if (status < 0) {
        perror("bind failed!");
        return 1;
    }

    // receive packets from socket
    while (1) {
        socklen = sizeof(saddr);
        status = recvfrom(sock,buffer,MAXBUFSIZE,&socklen);
        if (status < 0) {
            printf("recvfrom failed!n");
            return 1;
        }

        buffer[status] = '';
        printf("Received: '%s'n",buffer);
    }
}

(编辑:李大同)

【声明】本站内容均来自网络,其相关言论仅代表作者个人观点,不代表本站立场。若无意侵犯到您的权利,请及时与联系站长删除相关内容!

    推荐文章
      热点阅读