回射客戶-服務器模型（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;
}