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

服务器代码：

#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，保证停用的端口可以立即使用
	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;

}

测试：

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

注意：

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