Linux网络编程笔记(二）---P2P点对点聊天程序

先把程序和结果贴上，有时间我详细更新下细节

//服务器端
//P2Pserver.c
#include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <stdlib.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <stdio.h>
#include <errno.h>
#include <string.h>

void error_handling(char* message)
{
	fputs(message,stderr);
	fputc('\n',stderr);
	exit(1);
}
	

int main()
{
	int listenfd=socket(AF_INET,SOCK_STREAM,0);
	if(listenfd<0)
	error_handling("socket");
	
	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);
	
	int on=1;
	if(setsockopt(listenfd,SOL_SOCKET,SO_REUSEADDR,&on,sizeof(on))<0)
	error_handling("socket");

	if(bind(listenfd,(struct sockaddr*)&servaddr,sizeof(servaddr))<0)
	error_handling("bind");

	if(listen(listenfd,SOMAXCONN)<0)
	error_handling("listen");

	struct sockaddr_in peeraddr;
	socklen_t peerlen=sizeof(peeraddr);

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

	printf("IP=%s port=%d\n",inet_ntoa(peeraddr.sin_addr),ntohs(peeraddr.sin_port));//打印对等方的IP地址和端口号
	
	//char recvbuf[1024];
	//while(1)
	//{
	//	memset(recvbuf,0,sizeof(recvbuf));
	//	int ret=read(conn,recvbuf,sizeof(recvbuf));
	//	fputs(recvbuf,stdout);
	//	//write(conn,recvbuf,ret);   如果只是想实现简单的客户端发数据，服务器接受并且显示，则屏蔽这句
	//}
	
	pid_t pid;
        pid=fork();
        if(pid==-1)
          error_handling("fork failed");

        if(pid==0)    //子进程只管发数据
       {
	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]={0};
        while(1)
      {
        memset(recvbuf,0,sizeof(recvbuf));
         int ret=read(conn,recvbuf,sizeof(recvbuf));

        if(ret==-1)
          error_handling("read failed");

          else if(ret==0)
          {     printf("peer close\n");
                break;
           }
          
           fputs(recvbuf,stdout);
            } 
            exit(EXIT_SUCCESS);
         } 
         
         
 
close(conn);close(listenfd);
	return 0;
	}

	

//客户端
//P2Pclient.c
#include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <stdlib.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <stdio.h>
#include <errno.h>
#include <string.h>

void error_handling(char* message)
{
	fputs(message,stderr);
	fputc('\n',stderr);
	exit(1);
}

int main()
{
	int sock;
	sock=socket(PF_INET,SOCK_STREAM,IPPROTO_TCP);
	if(sock<0)
	error_handling("sock");

	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)
	error_handling("connect");
	
	//char sendbuf[1024]={0};
	//char recvbuf[1024]={0};
	
	//while(fgets(sendbuf,sizeof(sendbuf),stdin)!=NULL)
	//{
                //若保留这个while循环中的屏蔽掉的语句，实现的是回射客户/服务器
		//write(sock,sendbuf,strlen(sendbuf));   //不是sizeof的原因:sizeof是1024，strlen是recvbuf[]从输入流读了多少字节就取多少个
		//read(sock,recvbuf,sizeof(recvbuf));    如果只是想实现简单的客户端发数据，服务器接受并且显示，则屏蔽这几句
		//fputs(recvbuf,stdout);                  如果只是想实现简单的客户端发数据，服务器接受并且显示，则屏蔽这句  
		
		//memset(sendbuf,0,sizeof(sendbuf));
		//memset(recvbuf,0,sizeof(recvbuf));        如果只是想实现简单的客户端发数据，服务器接受并且显示，则屏蔽这句
	//}
	
         pid_t pid;
         pid=fork();
         if(pid==-1)
            error_handling("fork() failed");

        if(pid==0)
      {
         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]={0};
             while(1)
            {
               memset(recvbuf,0,sizeof(recvbuf));
               int ret=read(sock,recvbuf,sizeof(recvbuf));
               
               if(ret==-1)
          error_handling("read failed");

          else if(ret==0)
          {     printf("peer close\n");
                break;
           }
          
           fputs(recvbuf,stdout);
            } 
            exit(EXIT_SUCCESS);
         } 
 

	
	close(sock);
	return 0;
}
		

上图是客户端显示的“聊天记录”

下面是服务器端