Socket编程---p2p模型

自己既能发送数据也能接收数据，既是服务端也是客户端

服务端代码：

#include<unistd.h>
#include<sys/stat.h>
#include<sys/wait.h>
#include<sys/types.h>
#include<fcntl.h>

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


/*
typedef void (*sighandler_t)(int);

sighandler_t signal(int signum, sighandler_t handler);
*/
//信号要出发的回调函数
void handle(int num){
	printf("recv num:%d,退出\n",num);
	exit(0);
}

int main(){
	int sockfd=0;
	signal(SIGUSR1,handle);//捕获SIGUSR1信号，并调用handle函数
	sockfd=socket(AF_INET,SOCK_STREAM,0);
	
	if(sockfd==-1){
		perror("fun socket\n");
		exit(0);
	}
	struct sockaddr_in serveraddr;
	serveraddr.sin_family=AF_INET;
	serveraddr.sin_port=htons(8001);
	serveraddr.sin_addr.s_addr=inet_addr("127.0.0.1");
	//serveraddr.sin_addr.s_addr=INADDR_ANY;//绑定本机的任意地址	
	int optval=1;
	
    if(setsockopt(sockfd, SOL_SOCKET,SO_REUSEADDR,&optval, sizeof(optval))<0){
		perror("setsockopt");
		exit(0);
	};

	if(bind(sockfd, (const struct sockaddr *)&serveraddr,sizeof(serveraddr))<0){
		perror("bind()");
		exit(0);
	};
	//一旦调用了listen这个函数，那么这个套接字sockfd将变成被动套接字；只能接受连接，不能主动发送连接
	//listen做了两个队列1、已经由客户端发出并达到服务器，服务器正在等待完成相应的tcp三次握手过程
	//2、已完成连接的队列
	if(listen(sockfd, SOMAXCONN)<0){
		perror("listen()");
		exit(0);
	};
	//
	//struct sockaddr perraddr;//通用ip
	//socklen_t perrlen;
	struct sockaddr_in perraddr;//tcpip 地址结构
	int perrlen=sizeof(perraddr);
	int conn=0;		
	//accept 接受一个新的连接，这个新的连接是一个主动套接字
	conn=accept(sockfd, (struct sockaddr *)&perraddr,&perrlen);	
	if(conn==-1){
		perror("fun accept");
		exit(0);
	}
	printf("perradd:%s\n perrport:%d\n",inet_ntoa(perraddr.sin_addr),ntohs(perraddr.sin_port));
	int pid=fork();
	if(pid>0){
		//父进程读取客户端的数据
		char revbuf[100]={0};
		while(1){
			int ret=read(conn,revbuf,sizeof(revbuf));
			if(ret==0){
				//如果在读的过程中，对方已经关闭，已返回tcpip协议，会返回一个o数据报
				printf("对方已关闭\n");
				//如果对方已经关闭，break,结束循环。
				break;
			}else if(ret<0){
				printf("读数据失败");
				exit(0);
			}
			fputs(revbuf,stdout);//服务器端收到数据，打印屏幕
			memset(revbuf,0,sizeof(revbuf));
		}	
		close(conn);//关闭连接
		kill(pid,SIGUSR1);//并向子进程发送信号，结束子进程
		sleep(1);
	}else if(pid==0){
		//char revbuf[1024]={0};
		char sendbuf[1024]={0};
		while(fgets(sendbuf,sizeof(sendbuf),stdin)!=NULL){
			//向客户端写数据
			write(conn,sendbuf,strlen(sendbuf));//服务端回发信息
			memset(sendbuf,0,sizeof(sendbuf));
		}
	}
	close(sockfd);	
	return 0;
}

客户端代码：

#include<unistd.h>
#include<sys/stat.h>
#include<sys/wait.h>
#include<sys/types.h>
#include<fcntl.h>

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

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

int main(){
	int sockfd=0;
	signal(SIGUSR1,handle);
	sockfd=socket(AF_INET,SOCK_STREAM,0);
	
	if(sockfd==-1){
		perror("fun socket\n");
		exit(0);
	}
	struct sockaddr_in serveraddr;
	serveraddr.sin_family=AF_INET;
	serveraddr.sin_port=htons(8001);
	serveraddr.sin_addr.s_addr=inet_addr("127.0.0.1");
	//serveraddr.sin_addr.s_addr=INADDR_ANY;//绑定本机的任意地址	
	
	//int connect(int sockfd, const struct sockaddr *addr,socklen_t addrlen);

	if(connect(sockfd, (const struct sockaddr *)&serveraddr,sizeof(serveraddr))<0){
		perror("fun connetc");
		exit(0);
	};
	int pid=fork();
	if(pid>0){//父进程从键盘接收数据
		//char revbuf[1024]={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){
		//子进程接收数据
		while(1){
			int ret=0;
			char revbuf[1024]={0};
			//从服务器读数据
			ret=read(sockfd,revbuf,sizeof(revbuf));
			if(ret<0){
				printf("read err\n");
				break;
			}
			if(ret==0){
				printf("read err\n");
				break;
			}
			fputs(revbuf,stdout);//从服务器收到数据，打印屏幕
			memset(revbuf,0,sizeof(revbuf));
		}
		close(sockfd);
		kill(getppid(),SIGUSR1);//非父进程发信号，退出进程。
		exit(0);
	}else{
		printf("进程创建失败\n");
	}
	close(sockfd);	
	return 0;
}

客户端先退出：

服务端先退出：