使用C写socket收发UDP包，并发送UDP包到syslog中

秒速五厘米 2022-05-24 09:16 204阅读 0赞

# 一、UDP相关代码 #

在main函数中，先创建了一个线程来做UDP服务器，然后创建socket向UDP服务器发送和接收数据，最终直接使用给定的ip和端口号，向搭建好的rsyslog服务器发送数据，可以看到数据。为了省事，所有的东西都在一个文件中写完了，包括UDP服务器，UDP客户端。

代码如下：

#include <stdio.h>
    #include <stdlib.h>
    #include <string.h>
    
    #include <sys/socket.h>
    #include <stdio.h>
    #include <stdlib.h>
    #include <sys/types.h>
    #include <netinet/in.h>
    #include <arpa/inet.h>
    #include <string.h>
    #include <unistd.h>
    
    #define UDP_TEST_PORT       2234
    
    #define UDP_SERVER_IP       "127.0.0.1"
    
    #define MAX_LINE             1024
    
    /**********************************************************************
     * 功能描述：向给定ip和端口号发送udp字符串
     * 输入参数：	char *ip:ip地址
     * 			int port:端口号
     * 			char* str:发送的字符串
     * 输出参数：无
     * 返 回 值：	int:发送结果
     * 其它说明：无
     ***********************************************************************/
    int send_udp_str(char *ip, int port, char* str) {
    	int ret = 0;
    	if (str == NULL) {
    		ret = -1;
    		return ret;
    	}
    	struct sockaddr_in server;
    	int sockfd = 0;
    	int server_len = sizeof(struct sockaddr_in);
    
    	/* setup a socket，attention: must be SOCK_DGRAM */
    	if ((sockfd = socket(AF_INET, SOCK_DGRAM, 0)) < 0) {
    		perror("socket");
    		ret = -1;
    		return ret;
    	}
    
    	/*complete the struct: sockaddr_in*/
    	bzero(&server, sizeof(server));
    	server.sin_family = AF_INET;
    	server.sin_port = htons(port);
    	server.sin_addr.s_addr = inet_addr(ip);
    
    	/* send the string to server*/
    	if (sendto(sockfd, str, strlen(str), 0, (struct sockaddr *) &server,
    			server_len) < 0) {
    		printf("sendto error\n");
    		ret = -2;
    	}
    	close(sockfd);
    	return ret;
    }
    
    /**********************************************************************
     * 功能描述：接收udp字符串
     * 输入参数：	int sockfd:fd
    		struct sockaddr_in server
     * 		char* buf:接收的字符串存储空间，需要先分配好内存，如 char buf[1024];
     * 输出参数：无
     * 返 回 值：	int:结果
     * 其它说明：无
     ***********************************************************************/
    int recv_udp_str_by_fd(int sockfd, struct sockaddr_in server, char* buf) {
    	int ret = 0;
    	if (buf == NULL) {
    		ret = -1;
    		return ret;
    	}
    	memset(buf, 0, sizeof(buf));
    	int len = 0;
    
    	int server_len = sizeof(struct sockaddr_in);
    
    	/* recieve the string from server*/
    	len = recvfrom(sockfd, buf, 1024, 0, (struct sockaddr *) &server,
    			&server_len);
    	if (len < 0) {
    		printf("recvfrom error\n");
    		ret = -1;
    		return ret;
    	}
    //	printf("Received len = %d \n", len);
    
    
    	return ret;
    }
    
    /**********************************************************************
     * 功能描述：发送udp字符串
     * 输入参数：	int sockfd:fd
    		struct sockaddr_in server
     * 		char* str:发送的字符串
     * 输出参数：无
     * 返 回 值：	int:结果
     * 其它说明：无
     ***********************************************************************/
    int send_udp_str_by_fd(int sockfd, struct sockaddr_in server, char* str) {
    	int ret = 0;
    	if (str == NULL) {
    		ret = -1;
    		return ret;
    	}
    
    	int len = 0;
    
    	int server_len = sizeof(struct sockaddr_in);
    
    
    	/* send the string to server*/
    	if (sendto(sockfd, str, strlen(str), 0, (struct sockaddr *) &server,
    			server_len) < 0) {
    		printf("sendto error\n");
    		ret = -1;
    	}
    	return ret;
    }
    
    /**********************************************************************
     * 功能描述：开启一个UDP服务端
     * 输入参数：	char *ip:ip地址
     * 			int port:端口号
     * 输出参数：无
     * 返 回 值：	int:结果
     * 其它说明：无
     ***********************************************************************/
    int UDP_Server_main(char* ip, int port) {
    	//创建套接字
    	int sock = socket(AF_INET, SOCK_DGRAM, 0);
    	if (sock < 0) {
    		perror("socket");
    //		exit(1);
    		return -1;
    	}
    	//将套接字与ip地址和端口号进行绑定
    	struct sockaddr_in local;
    	local.sin_family = AF_INET;
    	local.sin_port = htons(port);
    	local.sin_addr.s_addr = inet_addr(ip);
    	if (bind(sock, (struct sockaddr*) &local, sizeof(local)) < 0) {
    		perror("bind");
    //		exit(2);
    		return -2;
    	}
    	char buf[1024];
    
    	struct sockaddr_in client;
    	socklen_t len = sizeof(client);
    	char* msg = "Have a goog day";
    	char msg_back[1024];
    	while (1) {
    		//读取数据
    		int r = recvfrom(sock, buf, sizeof(buf) - 1, 0,
    				(struct sockaddr*) &client, &len);
    		if (r < 0) {
    			perror("recvfrom");
    //			exit(3);
    		} else {
    			buf[r] = 0;
    			printf("[%s : %d]#  %s\n", inet_ntoa(client.sin_addr),
    					ntohs(client.sin_port), buf);
    
    			sprintf(msg_back, "%s: %s", msg, buf);
    			//回送数据
    			if (sendto(sock, msg_back, strlen(msg_back), 0,
    					(struct sockaddr*) &client, len) < 0) {
    				perror("sendto");
    //				exit(4);
    			}
    			if (strcmp(buf, "end") == 0) {
    				break;
    			}
    		}
    	}
    
    	printf("receive end from [%s : %d]\n", inet_ntoa(client.sin_addr),
    			ntohs(client.sin_port));
    	close(sock);
    	return 0;
    }
    // 开启UDPServer线程的函数，注意配置的ip和port与客户端一致
    int UDP_Server_main_thread() {
    	printf("UDP_Server_main_thread begin\n");
    	int ret = UDP_Server_main(UDP_SERVER_IP, UDP_TEST_PORT);
    	printf("UDP_Server_main_thread end ret = %d\n", ret);
    	return ret;
    }
    
    int main(int argC, char* arg[]) {
    	int ret = 0;
    	char *ipAddr = UDP_SERVER_IP;
    	int port = UDP_TEST_PORT;
    	char sendStr[MAX_LINE] = {0}; //默认发送串
    	char buf[MAX_LINE] = {0}; //接收缓冲区
    
    	pthread_t tid;
    	int err;
    	//----------------创建UDP服务端线程----------------
    	err = pthread_create(&tid, NULL, UDP_Server_main_thread, NULL);
    	if (err != 0) {
    		perror(" fail to create thread ");
    		return -1;
    	}
    	sleep(1);
    
    	// ----------------新建socket，发送UDP包到服务端线程----------------
    	struct sockaddr_in server;
    	/*complete the struct: sockaddr_in*/
    	bzero(&server, sizeof(server));
    	server.sin_family = AF_INET;
    	server.sin_port = htons(port);
    	server.sin_addr.s_addr = inet_addr(ipAddr);
    
    	int sockfd = 0;
    	/* setup a socket，attention: must be SOCK_DGRAM */
    	if ((sockfd = socket(AF_INET, SOCK_DGRAM, 0)) < 0) {
    		perror("socket");
    		err = -1;
    		return err;
    	}
    
    	int count = 1;
    	while (1) {
    		sprintf(sendStr, "client-%d|aaa", count);
    		send_udp_str_by_fd(sockfd, server, sendStr);
    		recv_udp_str_by_fd(sockfd, server, buf);
    		printf("Receive from server: %s\n", buf);
    		count +=10;
    		if(count > 100)
    			break;
    	}
    
    	send_udp_str(ipAddr, port, "aaaaaatest_send");
    
    	send_udp_str(ipAddr, port, "bbbbb");
    
    	send_udp_str(ipAddr, port, "end");
    
    	pthread_join(tid, NULL);
    
    
    	// -------------发送UDP包到syslog服务端----------------
    	ipAddr = "192.168.3.39";
    	port = 514;
    	char tmpStr[4096];
    
    	int count1 = 1;
    	while(count1 <= 10)
    	{
    		sprintf(tmpStr, "hahaha，to syslog-%d", count1);
    		ret = send_udp_str(ipAddr, port, tmpStr);
    		printf("count = %d; ret = %d\n", count1, ret);
    		sleep(1);
    		count1++;
    	}
    	return 0;
    
    }

# 二、Ubuntu的rsyslog服务器搭建过程： #

## 1 安装rsyslog ##

$ sudo apt-get install rsyslog

直接一步步安装就好

## 2 修改配置文件，打开接收514端口数据 ##

$ sudo vim /etc/rsyslog.conf

修改内容

先找到514端口相关内容，然后取消注释（删除前面的\#），改为如下：

# provides UDP syslog reception
    module(load="imudp")
    input(type="imudp" port="514")
    
    # provides TCP syslog reception
    module(load="imtcp")
    input(type="imtcp" port="514")

查看结果：

$ sudo netstat -tulpn | grep rsyslog
    tcp        0      0 0.0.0.0:514             0.0.0.0:*               LISTEN      2786/rsyslogd       
    tcp6       0      0 :::514                  :::*                    LISTEN      2786/rsyslogd       
    udp        0      0 0.0.0.0:514             0.0.0.0:*                           2786/rsyslogd       
    udp6       0      0 :::514                  :::*                                2786/rsyslogd

可以看到tcp和udp的514端口都打开了。

## 3 重启rsyslog服务 ##

$ sudo service rsyslog restart

## 4 修改启动配置文件 ##

$ sudo vim /etc/default/rsyslog

改为如下：

# Options for rsyslogd
    # -x disables DNS lookups for remote messages
    # See rsyslogd(8) for more details
    RSYSLOGD_OPTIONS="-r"

## 5 查看结果 ##

$ tail -f /var/log/syslog

运行UDP程序，可以看到结果：

$ tail -f /var/log/syslog
    May 24 15:56:26 192.168.3.51 hahaha，to syslog-1
    May 24 15:56:27 192.168.3.51 hahaha，to syslog-2
    May 24 15:56:28 192.168.3.51 hahaha，to syslog-3
    May 24 15:56:29 192.168.3.51 hahaha，to syslog-4
    May 24 15:56:30 192.168.3.51 hahaha，to syslog-5
    May 24 15:56:31 192.168.3.51 hahaha，to syslog-6
    May 24 15:56:32 192.168.3.51 hahaha，to syslog-7
    May 24 15:56:33 192.168.3.51 hahaha，to syslog-8
    May 24 15:56:34 192.168.3.51 hahaha，to syslog-9
    May 24 15:56:35 192.168.3.51 hahaha，to syslog-10

UDP参考： [http://velep.com/archives/934.html][http_velep.com_archives_934.html]

Ubuntu的rsyslog服务器搭建参考：[https://blog.csdn.net/martin\_ywz/article/details/52523094][https_blog.csdn.net_martin_ywz_article_details_52523094]

[http_velep.com_archives_934.html]: http://velep.com/archives/934.html
[https_blog.csdn.net_martin_ywz_article_details_52523094]: https://blog.csdn.net/martin_ywz/article/details/52523094