linux udp套接字编程获取报文源地址和源端口(二)

之前项目中涉及udp套接字编程，其中一个要求是获取客户端发过来报文的端口和ip地址，功能很简单，只是对这一块不很熟。之前使用的方法是通过调用recvmsg这个接口，并通过参数msg里面的msg_name来获取客户端地址,如下

ssize_t recvmsg(int sockfd, struct msghdr *msg, int flags);     

         //省略次要代码        
         msg.msg_name = &saddr;     // 存储报文来源地址
         msg.msg_namelen = addr_len;
         msg.msg_iov = &iov[0];
         msg.msg_iovlen = 1;
         msg.msg_control = cmh;
         msg.msg_controllen = sizeof(buff);    
         iov[0].iov_base = &buffer;
         iov[0].iov_len = sizeof(buffer);

后来boss说这代码太难看，建议我直接使用recvfrom这个接口，这下代码只需一行就可以获取到客户端地址和端口信息了。果然经验丰富

ssize_t recvfrom(int sockfd, void *buf, size_t len, int flags,
                 struct sockaddr *src_addr, socklen_t *addrlen);

一个简单使用例子：

/*
 *  Description : linux udp套接字获取报文源地址和源端口(二)
 *  Date        ：20180605
 *  Author      ：mason
 *  Mail        : mrsonko@126.com
 *
 */
#include <netinet/in.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <arpa/inet.h>
#include <sys/errno.h>
#include <sys/uio.h> 
#include <stdint.h>  
#include <stdlib.h>
#include <unistd.h>  
#include <stdint.h>
#include <string.h>
#include <stdio.h>

#define BUFFER_SIZE 512
#define log(fmt, arg...) printf("[udptest] %s:%d "fmt, __FUNCTION__, __LINE__, ##arg)

#ifndef NIPQUAD
#define NIPQUAD(addr) \
	((unsigned char *)&addr)[0], \
	((unsigned char *)&addr)[1], \
	((unsigned char *)&addr)[2], \
	((unsigned char *)&addr)[3]
#endif

void main() {
    int on = 1;
    int sockfd;
    int recv_len, ret, addr_len;
    char buffer[512] = {0};
    struct sockaddr_in saddr;
    struct sockaddr_in local_addr;

    // 创建UDP套接字
    sockfd = socket(AF_INET, SOCK_DGRAM, 0);
    if (sockfd == -1) {
        log("create socket fail \r\n");
        return ;
    }    

    // 设置监听地址
    local_addr.sin_family = AF_INET;
    local_addr.sin_addr.s_addr = INADDR_ANY; 
    local_addr.sin_port = htons(3500);
    addr_len = sizeof(struct sockaddr_in);

    // 设置SO_REUSEADDR属性, 地址复用    
    if ((ret = setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, (char *)&on, sizeof(on))) != 0) {
        log("setsockopt reuseaddr fail, ret : %d,error : %d \r\n", ret, errno);
        close(sockfd);
        return ;
    }

    // 绑定本地监听地址
    if (0 != bind(sockfd, (struct sockaddr *)&local_addr, sizeof(struct sockaddr_in))) {
        printf("bind local listening addr fail，errno : %d \r\n", errno);
        close(sockfd);
        return ;
    }   

    log("Start receiving message: \n");
    while(1) {
        //接收数据并将客户端的地址和端口信息存储到saddr结构体中
        recv_len = recvfrom(sockfd, buffer, BUFFER_SIZE, 0, (struct sockaddr *)&saddr, &addr_len);
        if(recv_len != -1)
            log("receive msg from : %u:%u:%u:%u:%hu，msg : %s \r\n",
            NIPQUAD(saddr.sin_addr), ntohs(saddr.sin_port), buffer);

        //重置
        memset(buffer, 0, BUFFER_SIZE); 
        memset(&saddr, 0, addr_len);
    }
}

Makefile:

udpsaddr:
	$(CC) -o udpsaddr udpsaddr.c
clean:
	@rm -rf *.o udpsaddr

也可以直接在linux命令行上敲：

gcc -o udpsaddr udpsaddr.c

测试结果：

github代码下载：

git@github.com:FuYuanDe/udp.git