之前項目中涉及udp套接字編程，其中一個要求是獲取客戶端發(fā)過來報文的端口和ip地址，功能很簡單，只是對這一塊不很熟。之前使用的方法是通過調(diào)用recvmsg這個接口，并通過參數(shù)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這個接口，這下代碼只需一行就可以獲取到客戶端地址和端口信息了。果然經(jīng)驗豐富

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] #endifvoid 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;// 創(chuàng)建UDP套接字sockfd = socket(AF_INET, SOCK_DGRAM, 0);if (sockfd == -1) {log("create socket fail \r\n");return ;} // 設(shè)置監(jiān)聽地址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);// 設(shè)置SO_REUSEADDR屬性, 地址復(fù)用 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 ;}// 綁定本地監(jiān)聽地址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) {//接收數(shù)據(jù)并將客戶端的地址和端口信息存儲到saddr結(jié)構(gòu)體中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

測試結(jié)果：

github代碼下載：

git@github.com:FuYuanDe/udp.git

總結(jié)

以上是生活随笔為你收集整理的linux udp套接字编程获取报文源地址和源端口(二)的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。