回射客户-服务器模型（3）

前面连接的C/S模型只是单方面的客户端发送数据，服务器端接收数据。那么本节将根据前两节已有的经验知识来实现一个简单的点对点聊天程序。

注：这里只实现了以个服务器与一个客户端的相互收发数据。

下面这篇文章对整个回射客户服务器的模型建立过程有一个说明，我这里就不再做过多讲解。由于刚刚接触网络编程，中间可能会有纰漏，还请大家多多指正。

点击打开链接

服务器端：p2psrv.c

#include <stdio.h>
#include <stdlib.h>
#include <errno.h>
#include <arpa/inet.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <unistd.h>
#include <signal.h>
#include <string.h>

void ERR_EXIT(const char* err)
{
	perror(err);
	exit(EXIT_SUCCESS);
}

void handler(int sig)
{
	printf("recv a sig = %d\n", sig);
	exit(EXIT_SUCCESS);
}

int main(void)
{
	//1.创建套接字
	int listenfd;
	if((listenfd = socket(AF_INET, SOCK_STREAM, 0)) < 0)
		ERR_EXIT("socket create error");

	//2.初始化地址
	struct sockaddr_in servaddr;
	memset(&servaddr, 0, sizeof(servaddr));
	servaddr.sin_family = AF_INET;
	servaddr.sin_port = htons(5188);
	servaddr.sin_addr.s_addr = INADDR_ANY;

	//3.在绑定之前先初始化SO_REUSEADDR选项
	//目的是让服务器在TIME_WAIT状态下仍能开启
	int on;
	if(setsockopt(listenfd, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on)) < 0)
		ERR_EXIT("setsockopt error");

	//4.绑定
	if(bind(listenfd, (struct sockaddr*)&servaddr, sizeof(servaddr)) < 0)
		ERR_EXIT("bind error");

	//5.监听
	if(listen(listenfd, SOMAXCONN) < 0)
		ERR_EXIT("listen error");

	//6.接收连接请求
	//先给对方指定一个地址结构
	struct sockaddr_in peeraddr;
	socklen_t peerlen = sizeof(peeraddr);
	memset(&peeraddr, 0, sizeof(peeraddr));

	int conn;
	if((conn = accept(listenfd, (struct sockaddr*)&peeraddr, &peerlen)) < 0)
		ERR_EXIT("accept error");

	printf("IP=%s port=%d\n", inet_ntoa(peeraddr.sin_addr), peeraddr.sin_port);
	
	//7.双方开始进行通信
	//这里我们需要fork出一个子进程，使得父进程用来接收数据，子进程用来发送数据
	close(listenfd);
	pid_t pid = fork();
	if(pid < 0)
	{
		ERR_EXIT("fork error");
	}
	else if(pid == 0)
	{
		//子进程处理，发送数据
		signal(SIGUSR1, handler);
		char sendbuf[1024] = {0};
		while(fgets(sendbuf, sizeof(sendbuf), stdin) != NULL)
		{
			write(conn, sendbuf, strlen(sendbuf));
			memset(sendbuf, 0, sizeof(sendbuf));
		}
		close(conn);
		exit(EXIT_SUCCESS);	
	}
	else
	{
		//父进程处理，接收数据
		char recvbuf[1024];
		while(1)
		{
			memset(recvbuf, 0, sizeof(recvbuf));
			//要考虑到客户端关闭的情况
			int ret = read(conn, recvbuf, sizeof(recvbuf));
			if(ret == -1)
				perror("read error");
			else if(ret == 0)
			{
				printf("peer close\n");
				break;
			}
			fputs(recvbuf, stdout);
		}
		//父进程退出的同时也要通知子进程也退出
		//给子进程一个信号，退出
		close(conn);
		kill(pid, SIGUSR1);
		usleep(1);
		exit(EXIT_SUCCESS);
	}
	return 0;
}

客户端：p2pcli.c

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <errno.h>
#include <arpa/inet.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <unistd.h>
#include <signal.h>

void ERR_EXIT(const char* err)
{
	perror(err);
	exit(EXIT_SUCCESS);
}

void handler(int sig)
{
	printf("recv a sig = %d\n", sig);
	exit(EXIT_SUCCESS);
}

int main()
{
	//1.创建套接字
	int sock;
	if((sock = socket(AF_INET, SOCK_STREAM, 0)) < 0)
		ERR_EXIT("socket create error");

	//2.指定对等方的地址
	struct sockaddr_in servaddr;
	memset(&servaddr, 0, sizeof(servaddr));
	servaddr.sin_family = AF_INET;
	servaddr.sin_port = htons(5188);
	inet_aton("127.0.0.1", &servaddr.sin_addr);

	//3.建立连接
	if(connect(sock, (struct sockaddr*)&servaddr, sizeof(servaddr)) < 0)	
		ERR_EXIT("connect error");

	//4.双方开始进行通信
	//需要fork出一个子进程用于接收数据
	pid_t pid = fork();
	if(pid < 0)
	{
		ERR_EXIT("fork error");
	}
	else if(pid == 0)
	{
		//子进程处理，接收数据
		char recvbuf[1024];
		while(1)
		{
			//考虑到对等方关闭的情况
			memset(recvbuf, 0, sizeof(recvbuf));
			int ret = read(sock, recvbuf, sizeof(recvbuf));
			if(ret == -1)
				ERR_EXIT("read error");
			else if(ret == 0)
			{
				printf("peer close\n");
				break;
			}
			fputs(recvbuf, stdout);
		}
		close(sock);
		//给父进程一个信号，退出
		kill(getppid(), SIGUSR1);
		exit(EXIT_SUCCESS);
	}
	else
	{
		signal(SIGUSR1, handler);
		//父进程处理，发送数据
		char sendbuf[1024] = {0};
		while(fgets(sendbuf, sizeof(sendbuf), stdin) != NULL)
		{
			write(sock, sendbuf, strlen(sendbuf));
			memset(sendbuf, 0, sizeof(sendbuf));
		}
		close(sock);
		exit(EXIT_SUCCESS);
	}
	return 0;
}