06點對點通信

注意要點：
（1）父子進程共享文件描述符,注意各自的職責
（2）注意避免殭屍進程的產生
（3）信號處理就是軟中斷

5p2pclient.c

#include <stdio.h>
#include <string.h>
#include <stdlib.h>


#include <sys/socket.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <netinet/ip.h> /* superset of previous */
#include <arpa/inet.h>
#include <errno.h>
#include <unistd.h>
#include <signal.h>



/*
	父進程從鍵盤接收數據,發給服務端
	
	子進程接收服務端數據,若檢測到服務端關閉,則發信號給
	父進程。
	子進程檢測出來對端關閉,就發信號給父進程退出,子進程也退出
*/

#if 1
//出現異常,使程序退出
void handle(int num)
{
	printf("recv num:%d \n", num);
	exit(0);
}

void test()
{
	int sockfd = 0;
	const char *serverip = "192.168.66.128";
	
	//安裝信號處理函數,信號具有繼承性
	signal(SIGUSR1, handle);
	
	//創建socket
	sockfd = socket(AF_INET, SOCK_STREAM, 0);
	if (sockfd == -1)
	{
		perror("socket()");
		exit(0);
	}
	//定義socket結構體 man 7 ip
	
	struct sockaddr_in srvsddr;
	srvsddr.sin_family = AF_INET;
	srvsddr.sin_port = htons(8001);//轉化爲網絡字節序

	srvsddr.sin_addr.s_addr = inet_addr(serverip);
	//進程-》內核
	if( connect(sockfd, (struct sockaddr *)&srvsddr,sizeof(srvsddr)) < 0)
	{
		perror("connect()");
		exit(0);
	}

	pid_t pid = 0;
	pid = fork();
	
	//父進程
	if (pid > 0) 
	{
		char sendbuf[1024] = {0};
		while( fgets(sendbuf, sizeof(sendbuf), stdin) != NULL )
		{
			write(sockfd, sendbuf, strlen(sendbuf));
			memset(sendbuf, 0, sizeof(sendbuf));	
		}	
	}
	else if(pid == 0)
	{
		char recvbuf[1024] = {0};
		int ret = 0;
		while(1)
		{
			ret = read(sockfd, recvbuf, sizeof(recvbuf));
			if (ret == 0) 
			{
				//如果在讀的過程中，對方已經關閉，tcpip協議會返回一個0數據包
				printf("對端關閉\n");
				break;
			}
			if (ret < 0) 
			{
				perror("讀數據失敗");
				exit(0);
			}
			fputs(recvbuf, stdout);//從服務器收到數據，打印屏幕
			memset(recvbuf, 0, sizeof(recvbuf));
		}
		//異常處理
		close(sockfd);
		//子進程退出之前向父進程發送信號
		kill(getppid(), SIGUSR1);
		exit(0);
	}
	
	close(sockfd);
	return;	
}

#endif


int main()
{
	test();
	return 0;
}

6p2psrv.c

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

/*
	服務端 父進程讀數據,檢測出對方關閉,也發信號給子進程
	退出,自己也退出
	
	子進程 從鍵盤讀數據,發送數據
	1、長連接與短連接客戶端說的算,服務器不會主動的close掉
	2、客戶端掛掉或者close()時,tcp/ip協議會偷偷的向服務端發送FIN數據段
	   服務端掛掉或者close()時,tcp/ip協議會偷偷的向客戶端發送FIN數據段
	
	ps -u hzmct
*/

#if 1

void handle(int num)
{
	printf("recv num:%d \n", num);
	exit(0);
}


void test()
{
	int sockfd = 0;
	int conn = 0;
	const char *serverip = "192.168.66.128";
	
	//創建socket
	sockfd = socket(AF_INET, SOCK_STREAM, 0);
	if (sockfd == -1)
	{
		perror("socket()");
		exit(0);
	}
	
	//定義socket結構體 man 7 ip
	struct sockaddr_in srvsddr;
	srvsddr.sin_family = AF_INET;
	srvsddr.sin_port = htons(8001);//轉化爲網絡字節序
	//第一種
	#if 0
	srvsddr.sin_addr.s_addr = inet_addr(serverip);
	#endif
	//第二種
	#if 0
	//srvsddr.sin_addr.s_addr = htonl(INADDR_ANY);//INADDR_ANY 就是0.0.0.0 不存在網絡字節序
	//srvaddr.sin_addr.s_addr = inet_addr(INADDR_ANY); //綁定本機的任意一個地址
	#endif
	//第三種
	//建議使用這種
	#if 1
	int ret;
	ret = inet_pton(AF_INET, serverip, &srvsddr.sin_addr);
	if (ret == 0)
	{
		perror("inet_pton()");
		exit(0);
	}	
	#endif
	
	//設置端口複用
	//使用SO_REUSEADDR選項可以使得不必等待TIME_WAIT狀態消失就可以重啓服務器
	int optval = 1;
    if( setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, &optval, sizeof(optval)) < 0)
	{
		perror("setsockopt()");
		exit(0);
	}
	
	//從進程->內核 將結構首地址和該結構大小都傳遞給了內核
	//內核知道需要從進程複製多少數據進來
	
	if(bind(sockfd, (struct sockaddr *)&srvsddr,sizeof(srvsddr)) <0 )
	{
		perror("bind()");
		exit(0);
	}
	
	//一但調用listen函數，這個套接字sockfd將變成被動套接字,只能接受連接，
	//不能主動的發送連接
	//listen 做了兩個隊列。。。。。。
	// 隊列由內核管理，一部分是完成三次握手的，一部分是沒有完成三次握手的。
	if(listen(sockfd, SOMAXCONN) < 0)
	{
		perror("listen()");
		exit(0);
	}
	
	struct sockaddr_in peeraddr;
	socklen_t peerlen = sizeof(peeraddr);//值-結果參數
	//accept 返回一個新的連接 ，這個新的連接是一個主動套接字
	//如果沒有客戶端到來,父進程將阻塞在這裏
	conn = accept(sockfd, (struct sockaddr *)&peeraddr, &peerlen);
	if (-1 == conn) 
	{
		perror("accept()");
		exit(0);
	}
	printf("客戶端的ip:%s port:%d\n", inet_ntoa(peeraddr.sin_addr), ntohs(peeraddr.sin_port));
			
	pid_t pid = fork();
	
	//子進程從鍵盤接收數據,發送數據
	if (pid == 0) 
	{	
		char sendbuf[1024] ={0};

		while( fgets(sendbuf, sizeof(sendbuf), stdin) != NULL )
		{
			write(conn, sendbuf, strlen(sendbuf));
			memset(sendbuf, 0, sizeof(sendbuf));	
		}
	}
	else if (pid > 0)
	{
		char recvbuf[1024] ={0};
		while(1)
		{
			memset(recvbuf, 0, sizeof(recvbuf));
			//如果在讀的過程中，對方已經關閉，tcpip協議會返回一個0數據包
			int ret = read(conn, recvbuf, sizeof(recvbuf));
			if (ret == 0) 
			{
				printf("對方已經關閉\n");
				break;
			}
			else if (ret < 0)
			{
				perror("讀取數據失敗\n");
				exit(0);
			}
			fputs(recvbuf, stdout);//服務器收到數據,打印到屏幕
		}
		//異常處理
		close(conn);
		//通知子進程退出
		kill(pid, SIGUSR1);
		sleep(1);
		
	}
		
	close(sockfd);
	return;	
}

#endif


int main()
{
	test();
	return 0;
}
/*
	長連接與短連接：客戶端說的算！
*/