linux网络编程（9）UDP服务器与客户端代码

刺骨的言语ヽ痛彻心扉 2022-05-22 11:47 154阅读 0赞

服务器代码：

#include <sys/socket.h>
    #include <unistd.h>
    #include <signal.h>
    #include <string.h>
    #include <stdio.h>
    #include <time.h>
    #include <stdlib.h>
    #include <netinet/in.h>
    #include<arpa/inet.h>
    
    int sockfd;
    
    void sig_handler(int signo)
    {
    	if (signo == SIGINT)
    	{
    		printf("server close\n");
    		close(sockfd);
    		exit(1);
    	}
    }
    
    
    void out_addr(struct sockaddr_in * clientaddr)
    {
    	char ip[16];
    	int port;
    	memset(ip, 0, sizeof(ip));
    	inet_ntop(AF_INET, &clientaddr->sin_addr.s_addr, ip, sizeof(ip));
    	port = ntohs(clientaddr->sin_port);
    	printf("client: %s(%d)\n", ip, port);
    }
    
    void do_service()
    {
    	struct sockaddr_in clientaddr;
    	socklen_t len = sizeof(clientaddr);
    	char buffer[1024];
    	memset(buffer, 0, sizeof(buffer));
    	//接收客户端的数据报文
    	if (recvfrom(sockfd, buffer, sizeof(buffer), 0, (struct sockaddr*)&clientaddr, &len) < 0)
    	{
    		perror("recvfrom error");
    	}
    	else
    	{
    		out_addr(&clientaddr);
    		printf("client send into: %s \n", buffer);
    
    		//向客户端发送数据报文
    		long int t = time(0);
    		char *ptr = ctime(&t);
    		size_t size = strlen(ptr) * sizeof(char);
    		if (sendto(sockfd, ptr, size, 0, (struct sockaddr*)&clientaddr, len) < 0)
    		{
    			perror("sendto error");
    		}
    	}
    }
    
    int main(int argc, char* argv[])
    {
    	if (argc < 2)
    	{
    		printf("usage: %s port \n", argv[0]);
    		exit(1);
    	}
    
    	if (signal(SIGINT, sig_handler) == SIG_ERR)
    	{
    		perror("signal sigint error");
    		exit(1);
    	}
    
    	/*1.创建socket*/
    	sockfd = socket(AF_INET, SOCK_DGRAM, 0);
    	if (sockfd < 0)
    	{
    		perror("socket error");
    		exit(1);
    	}
    
    	int ret;
    	int opt = 1;
    	//设置套接字选项SO_REUSEADDR，1.保证停用的端口可以立即使用；2.在此程序不被终止的情况下，新启动的程序绑定相同的端口也会
            //成功,不过此程序端口会失效。
    	if ((ret = setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt))) < 0)
    	{
    		perror("setsockopt error");
    		exit(1);
    	}
    
    	/*2.调用bind函数对socket和地址进行绑定*/
    	struct sockaddr_in serveraddr;
    	memset(&serveraddr, 0, sizeof(serveraddr));
    	serveraddr.sin_family = AF_INET; //IPv4
    	serveraddr.sin_port = htons(atoi(argv[1]));//port
    	serveraddr.sin_addr.s_addr = INADDR_ANY;//ip
    	if (bind(sockfd, (struct sockaddr*)&serveraddr, sizeof(serveraddr)) < 0)
    	{
    		perror("bind error");
    		exit(1);
    	}
    
    	/*3.与客户端进行数据通信*/
    	while (1)
    	{
    		do_service();
    	}
    
    
    
    	return 0;
    }

客户端代码：

#include <sys/socket.h>
    #include <unistd.h>
    #include <signal.h>
    #include <string.h>
    #include <stdio.h>
    #include <time.h>
    #include <stdlib.h>
    #include <netinet/in.h>
    #include <arpa/inet.h>
    #include <memory.h>
    
    
    int main(int argc, char* argv[])
    {
    	if (argc < 3)
    	{
    		printf("usage: %s ip port\n", argv[0]);
    		exit(1);
    	}
    
    	/*1.创建socket*/
    	int sockfd = socket(AF_INET, SOCK_DGRAM, 0);
    	if (sockfd < 0)
    	{
    		perror("socket error");
    		exit(1);
    	}
    
    	/*2.调用recvfrom 、sendto等函数和服务器双向通信*/
    	struct sockaddr_in serveraddr;
    	memset(&serveraddr, 0, sizeof(serveraddr));
    	serveraddr.sin_family = AF_INET;
    	serveraddr.sin_port = htons(atoi(argv[2]));
    	inet_pton(AF_INET, argv[1], &serveraddr.sin_addr.s_addr);
    
    	//调用connect并不会像TCP编程中的三次握手，只是在内核在记录服务器端的一些地址信息（包括IP、端口等）
    	//这样后面即使不调用sendto指定服务器的地址，而只调用send也可以向服务器发送数据。
    	//建议调用connect，这样会保证客户端在接收的时候，只接收到来自指定服务器端的数据，而不会收到其它服务器的数据。
    	if (connect(sockfd, (struct sockaddr*)&serveraddr, sizeof(serveraddr)) < 0)
    	{
    		perror("sendto error");;
    		exit(1);
    	}
    
    	char buffer[1024] = "hello linux";
    	/*
    	if (send(sockfd, buffer, sizeof(buffer), 0) < 0)
    	{
    		perror("sendto error");
    		exit(1);
    	}
    	*/
    	//向服务器发送数据报文
    	if (sendto(sockfd, buffer, sizeof(buffer), 0, (struct sockaddr*)&serveraddr, sizeof(serveraddr)) < 0)
    	{
    		perror("sendto error");
    		exit(1);
    	}
    	else
    	{
    		//接收服务器端发送的数据报文
    		memset(buffer, 0, sizeof(buffer));
    		if (recv(sockfd, buffer, sizeof(buffer), 0) < 0)
    		{
    			perror("recv error");
    			exit(0);
    		}
    		else
    		{
    			printf("%s", buffer);
    		}
    	}
    
    	return 0;
    
    }

测试：

开启了两个终端，上面的终端运行服务器程序，下面的终端运行客户端程序。

![70][]

注意：

调用connect并不会像TCP编程中的三次握手，只是在内核中记录服务器端的一些地址信息（包括IP、端口等），这样后面即使不调用sendto指定服务器的地址，而只调用send也可以向指定的服务器发送数据。  
但是，建议调用connect，这样会保证客户端在接收的时候，只接收到来自指定服务器端的数据，而不会收到其它服务器的数据。

[70]: /images/20220522/8deba03784e9483da3e82c1f0669e1b1.png