linux（一）------多進程併發服務器實現（fork）

本文實現一個多進程併發服務器
實現思路：
利用fork()函數實現每有一個client連接時就會創建一個子進程進程與client通信，父進程負責回收子進程的PCB
直接上代碼！

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

#define SERVER_PORT 8888
#define MAXLINE 8192

//進程回收函數
void do_sigchlid(int num)
{
    pid_t pid;
    while((pid = waitpid(-1, NULL, WNOHANG)) > 0);//回收任何子進程
    {
        printf("child died, pid = %d\n", pid);
    }
}

int main()
{
    int listenfd, connfd;
    pid_t pid;
    int len;
    char buf[MAXLINE];
    char str[256];
    int i;

    //創建監聽描述符
    listenfd = socket(AF_INET, SOCK_STREAM, 0);
    if(listenfd == -1)
    {
        perror("socket error");
        exit(1);
    }

    //she值端口複用
    int opt = 1;
    setsockopt(listenfd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));

    //綁定ip和端口
    struct sockaddr_in serveraddr;
    bzero(&serveraddr, sizeof(serveraddr));
    serveraddr.sin_port = htons(SERVER_PORT);
    serveraddr.sin_family = AF_INET;
    serveraddr.sin_addr.s_addr = htonl(INADDR_ANY);
    int ret = bind(listenfd, (struct sockaddr*)&serveraddr, sizeof(serveraddr));
    if(ret == -1)
    {
        perror("bind error");
        exit(1);
    }

    //設置監聽
    ret = listen(listenfd, 20);
    if(ret == -1)
    {
        perror("listen error");
        exit(1);
    }

    printf("Accepting connections ...\n");

    struct sockaddr_in clientaddr;
    socklen_t clientaddr_len = sizeof(clientaddr);
    //通信
    while(1)
    {
        //等待接收客戶端的連接
        connfd = accept(listenfd, (struct sockaddr*)&clientaddr, &clientaddr_len);

         printf("connected client ip:%s, port: %d\n",
                inet_ntop(AF_INET, &clientaddr.sin_addr, str, sizeof(str)), ntohs(clientaddr.sin_port));
        //有一個新的連接就創建一個進程
        pid  = fork();
        if(pid == 0)
        {
            //子進程
            //關閉監聽文件描述符
            close(listenfd);
            while(1)//與客戶端通信，機型數據處理
            {
                //讀數據
                len = read(connfd, buf, MAXLINE);
                if(len == 0)
                {
                    printf("客戶端斷開了連接...\n");
                    break;
                }
                for(i = 0; i < len; ++i)
                {
                    buf[i] = toupper(buf[i]);
                }
                //寫回客戶端
                write(STDOUT_FILENO, buf, len);//把buf寫到標準輸出中去
                write(connfd, buf, len);//把buf寫回客戶端
            }
            close(connfd);
            return 0;
        }
        else if(pid > 0)
        {
            //父進程，負責對子進程pcb資源的回收
            //使用信號來回收，子進程結束會給父進程發送SIGCHLD信號
            //默認父進程會屏蔽此信號，若父進程想回收子進程資源，
            //可以捕捉此信號

            //信號捕捉
            //註冊捕捉函數
            struct sigaction act;
            act.sa_flags = 0;
            act.sa_handler = do_sigchlid;//回調函數，通過sa_handler指針調用
            sigemptyset(&act.sa_mask);//將信號集初始化爲空
            sigaction(SIGCHLD, &act, NULL);
        }
    }
    return 0;
}