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 rn");
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 rn", 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 rn", 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 rn",
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

最后

以上就是鲤鱼马里奥为你收集整理的linux udp套接字编程获取报文源地址和源端口(二)的全部内容，希望文章能够帮你解决linux udp套接字编程获取报文源地址和源端口(二)所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。