Linux下p2p的聊天功能實現，linuxp2p聊天功能

Linux下p2p的聊天功能實現，linuxp2p聊天功能

Linux下p2p的聊天功能實現細節

Do one thing at a time, and do well.

今天閒著沒事，寫一個P2P的點對點的聊天功能的小程序，我覺得對網絡編程初學者的學習很有用的。二話不說，我先貼代碼吧。有幾個地方需要考慮清楚。我會在代碼的後面寫出來。代碼的下載文章的末尾。

server.c

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

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


void do_something(int conn)
{
     char recvbuf[1024];
     for(;;)
     {
     memset(recvbuf,0,sizeof(recvbuf));
     int ret = read(conn,recvbuf,sizeof(recvbuf));
     if(ret == 0)
     {
         printf("client closed!\n");
         break;
     }
     else if(ret == -1)
     {
         ERR_EXIT("read");
     }
     fputs(recvbuf,stdout);
     write(conn,recvbuf,ret);
     }
}
void handler(int sig)
{
    printf("recv a sig = %d\n",sig);
    exit(EXIT_SUCCESS);
}

int main()
{
    int listenfd;
    if((listenfd = socket(AF_INET,SOCK_STREAM,0)) < 0)
      ERR_EXIT("socket");
    
    int on = 1;
    int ret = setsockopt(listenfd,SOL_SOCKET,SO_REUSEADDR,&on,sizeof(on) );
     
    struct sockaddr_in servaddr;
    memset(&servaddr,0,sizeof(servaddr));
    servaddr.sin_family = AF_INET;
    servaddr.sin_port = htons(10001);
    servaddr.sin_addr.s_addr = htonl(INADDR_ANY);

    if(bind(listenfd,(struct sockaddr*)&servaddr,sizeof(servaddr)) < 0)
      ERR_EXIT("bind");
    
    if((listen(listenfd,SOMAXCONN)) < 0 )//主動套接字，變成被動套接字
      ERR_EXIT("listen");
     
    struct sockaddr_in peeraddr;
    socklen_t socklen = sizeof(peeraddr);
    int conn;
    pid_t  pid;
    if((conn = accept(listenfd,(struct sockaddr*)&peeraddr,&socklen)) < 0)// 獲得到是主動套接字
        ERR_EXIT("accept");
     printf("ip:%s   port:%d\n",inet_ntoa(peeraddr.sin_addr),ntohs(peeraddr.sin_port));       
         
        pid = fork();
        char sendbuf[1024] = {0};
        if(pid == -1) ERR_EXIT("pid");
        if(pid == 0)
        {
            signal(SIGUSR1,handler);
            while(fgets(sendbuf,sizeof(sendbuf),stdin) != NULL)
            {
                write(conn,sendbuf,strlen(sendbuf));                
                memset(sendbuf,0,sizeof(sendbuf));
            }
            printf("child closed\n");
            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");
                else if(ret == 0)
                { 
                  printf("peer close\n");
                  break;
                }
                fputs(recvbuf,stdout);
            }
            printf("kill parent!\n");
            kill(pid,SIGUSR1);
            exit(EXIT_SUCCESS);
            //do_something(conn);
        
        }
        close(conn);
        close(listenfd);
        exit(0);
}

#include <stdio.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <string.h>
#include <unistd.h>
#include <stdlib.h>
#include <arpa/inet.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 argc,char *argv[])
{
    int sockfd;
    if((sockfd = socket(AF_INET,SOCK_STREAM,0)) < 0)
      ERR_EXIT("socket");
    struct sockaddr_in servaddr;
    memset(&servaddr,0,sizeof(servaddr));
    servaddr.sin_family = AF_INET;
    servaddr.sin_port = htons(10001);
    servaddr.sin_addr.s_addr = inet_addr("127.0.0.1");

    if(connect(sockfd,(struct sockaddr*)&servaddr,sizeof(servaddr)) < 0)
      ERR_EXIT("connect");

    char sendbuf[1024] = {0};
    char recvbuf[1024] = {0};
    pid_t pid;
    pid = fork();
    if(pid == -1)
        ERR_EXIT("fork");
    if(pid == 0)
    {
        while(1)
        {
            memset(recvbuf,0,sizeof(recvbuf));
            int ret = read(sockfd,recvbuf,sizeof(recvbuf));
            if(ret == -1)
                ERR_EXIT("read");
            else if (ret == 0)
            {
                printf("peer closed\n");
                break;
            }
            fputs(recvbuf,stdout);
        }

        printf("child close\n");
        kill(pid,SIGUSR1);
        exit(EXIT_SUCCESS);
    }
    else
    {        
        signal(SIGUSR1,handler);
        while(fgets(sendbuf,sizeof(sendbuf),stdin) != NULL)
        {
            write (sockfd,sendbuf,strlen(sendbuf));  
            memset(sendbuf,0,sizeof(sendbuf));
        }
        printf("parent close!\n");
    }
    close(sockfd);
    exit(0);
}

實現的功能很簡單，但是需要注意的幾個細節：

1、C和S連接以後，當S關閉後，C仍然沒有關閉，我用到了信號的功能。

     實現的方法：當父進程關閉的時候，子進程也關閉，當子進程關閉的時候，把父進程也關閉了。

2、一個線程用來監聽，一個線程用來等待輸入。那麼，這是一個多線程的小程序。

　 創建一個子進程。C端，子進程進程監聽，父進程等待輸入。S端相反。

3、信號量的問題：

　 SIGUSR1：用戶自定義信號量，函數handler用來殺死進程，實現退出。

程序的測試：

總結：總的來說，這個還是很簡單的，也就是幾個函數是否靈活應用。注意，read函數如果沒有讀到，就會進入阻塞，如果收到一個0，代表對方關閉了程序，則退出程序。

代碼下載：GitHub

聲明：水平有限，如果有什麼地方寫錯了或者理解有誤，希望廣大網友指正。