6. C/S实现对等通信

                   服务器只能被动接收客户端的信息，而不能主动发送信息给客户端，如果我们想实现对等通信，即P2P，可以在服务器程序用使用两个进程，一个进程接收用户的输入并发送给客户端，另一个进程被动接收客户端的消息并打印出来，此进程当read 返回0 时得知客户端已经关闭需要退出进程，此时尚有另一个进程未退出，可以通过在退出前发送消息给它，在消息处理函数中退出。当然客户端也必须使用双进程，原理与服务器程序相同。

服务器程序

// p2pser.c 

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

#define ERR_EXIT(m) \
		do{ \
			perror(m); \
			exit(EXIT_FAILURE); \
			}while(0)


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

int main()
{
	int listenfd; 
	if( (listenfd = socket(PF_INET,SOCK_STREAM,IPPROTO_TCP) ) < 0)
			// listenfd = socket(AF_INET,SOCK_STREAM,0)
			ERR_EXIT("socket error");
	
	struct sockaddr_in servaddr;
	memset(&servaddr,0,sizeof(servaddr));
	servaddr.sin_family = AF_INET;
	servaddr.sin_port = htons(5188);
	servaddr.sin_addr.s_addr = htonl(INADDR_ANY);// INADDR_ANY，这个宏表示本地的任意IP地址
	//servaddr.sin_addr.s_addr = inet_addr("127.0.0.1");
	//inet_aton("127.0.0.1",&servaddr.sin_addr);
	

	int on = 1;
	if( setsockopt(listenfd,SOL_SOCKET,SO_REUSEADDR,&on,sizeof(on)) < 0)
		ERR_EXIT("setsockopt err");

	if( bind( listenfd,(struct sockaddr*)&servaddr,sizeof(servaddr)) < 0)
		ERR_EXIT("bind err");
	if( listen(listenfd,SOMAXCONN) < 0)  //  INADDR_ANY，这个宏表示本地的任意IP地址
			ERR_EXIT("lesten err");

	struct sockaddr_in peeraddr; // 传出参数
	socklen_t peerlen = sizeof(peeraddr);// 传入传出参数,必须有初始值
	int conn; // 已经连接套接字(变为主动套接字,可以主动connect)
	if( (conn = accept(listenfd,(struct sockaddr*)&peeraddr,&peerlen)) < 0)// 3次握手完成
			ERR_EXIT("accept err");
		//通过peeraddr打印连接上来的客户端ip和端口号
			printf("recv connect ip=%s ,port = %d \n",inet_ntoa(peeraddr.sin_addr),ntohs(peeraddr.sin_port));
	
	pid_t pid;
	pid = fork();
	if(pid == -1)
		ERR_EXIT("fork error");
		
	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));
		}
		exit(EXIT_SUCCESS);
	}
	else// 父进程负责接收
	{
		char recvbuf[1024];
		while(1)
		{
			memset(recvbuf,0,sizeof(recvbuf));
			int ret = read(conn,recvbuf,sizeof(recvbuf));
			if( ret == -1)
				ERR_EXIT("read err");
			else if(ret == 0)
			{
				printf("peer close\n");
				break;
			} 
			fputs(recvbuf,stdout);
		}
		kill(pid,SIGUSR1);// 父进程退出时发送信号给子进程
		exit(EXIT_SUCCESS);
	
	}
	
}

客户端程序

/// p2plic.c

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

#define ERR_EXIT(m) \
		do{ \
			perror(m); \
			exit(EXIT_FAILURE); \
		}while(0)

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


int main()
{
	int sock;
	if( (sock = socket(PF_INET,SOCK_STREAM,IPPROTO_TCP)) < 0)
			ERR_EXIT("sock err");
	struct sockaddr_in servaddr;
	memset(&servaddr,0,sizeof(servaddr));

	servaddr.sin_family = AF_INET;
	servaddr.sin_port = htons(5188);
	servaddr.sin_addr.s_addr = inet_addr("127.0.0.1"); 
	
	if( connect(sock,(struct sockaddr*)&servaddr,sizeof(servaddr)) < 0)
			ERR_EXIT("connect error");
			
	pid_t pid;
	pid = fork();
	if(pid == -1)
		ERR_EXIT("fork err");
	if(pid == 0)
	{
		signal(SIGUSR1,handle);
		char sendbuf[1024] = {0};
		while(fgets(sendbuf,sizeof(sendbuf),stdin) != NULL)
		{
			write(sock,sendbuf,strlen(sendbuf));
			memset(sendbuf,0,sizeof(sendbuf));
		}
		exit(EXIT_SUCCESS);
		
	}
	else//父进程接收到数据,打印
	{
		char recvbuf[1024];
		while(1)
		{
			memset(recvbuf,0,sizeof(recvbuf));
			int ret = read(sock,recvbuf,sizeof(recvbuf));
			if(ret < 0)
				ERR_EXIT("ret err");
			else if(ret == 0)
			{
				printf("peer close \n");
				break;
			}
			fputs(recvbuf,stdout);
		}
		kill(pid,SIGUSR1);
		exit(EXIT_SUCCESS);
	}		
			
	return 0;
}