多进程并发服务器

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

#define ERR_MSG(msg)  do{\
    fprintf(stderr, "__%s__ __%s__  __%d__ ", __FILE__, __func__, __LINE__);\
    perror(msg);\                                                                                                        
}while(0)

#define IP "192.168.50.92"  //ifconfig
#define PORT 6666               //端口号 1024~49151

int do_cli_msg(int newfd, struct sockaddr_in cin);

//一旦子进程退出，会给父进程发送一个17号信号
//进入handler函数，在handler函数中回收僵尸进程。
void handler(int sig)
{
    //若成功回收到僵尸进程则再收一次
    //直到没有收到僵尸进程，解除循环
    //==0， 有子进程，没有僵尸进程
    //==-1, 没有子进程，也没有僵尸进程
    while(waitpid(-1, NULL, WNOHANG) > 0);
    return;
}

int main(int argc, const char *argv[])
{
    if(signal(17, handler) == SIG_ERR)
    {
        ERR_MSG("signal");
        return -1;
    }
    printf("17号信号捕获成功\n");

    //创建流式套接字
    int sfd = socket(AF_INET, SOCK_STREAM, 0);
    if(sfd < 0)
    {
        ERR_MSG("socket");
        return -1;
    }
    printf("socket success sfd=%d\n", sfd);

    //允许端口快速被覆盖重用。
    int reuse = 1;
    if(setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) < 0)
    {
        ERR_MSG("setsockopt");
        return -1;
    }
    printf("允许端口快速被覆盖重用成功\n");



    //填充地址信息结构体，真实的地址信息结构体AF_INET: man 7 IP
    struct sockaddr_in sin;
    sin.sin_family      = AF_INET;      //必须填AF_INET;
    sin.sin_port        = htons(PORT);  //端口号的网络字节序1024~49151
    sin.sin_addr.s_addr = inet_addr(IP);//本机IP地址的网络字节序


    //绑定服务器的地址信息
    if(bind(sfd, (struct sockaddr*)&sin, sizeof(sin)) < 0)
    {
        ERR_MSG("bind");
        return -1;
    }
    printf("bind success __%d__\n", __LINE__);


    //将套接字设置为被动监听状态
    if(listen(sfd, 128) < 0)
    {
        ERR_MSG("listen");
        return -1;
    }
    printf("listen success __%d__\n", __LINE__);


    struct sockaddr_in cin;     //存储客户端地址信息
    socklen_t addrlen = sizeof(cin);

    int newfd = -1;
    pid_t cpid = 0;
    while(1)
    {
        //父进程专门运行客户端连接步骤 accept
        //生成新的文件描述符与客户端通信 
        newfd = accept(sfd, (struct sockaddr*)&cin, &addrlen);
        if(newfd < 0)
        {
            ERR_MSG("accept");
            return -1;
        }
        printf("[%s:%d] newfd=%d 客户端连接成功 __%d__\n", \
                inet_ntoa(cin.sin_addr), ntohs(cin.sin_port), newfd, __LINE__);

        //若能运行到当前位置，则代表有客户端连接成功，则需要创建一个子进程用于交互
        cpid = fork();
        if(0 == cpid)
        {

            //专门用于交互
            close(sfd);

            do_cli_msg(newfd, cin);         //该函数结束后代表客户端下线
            close(newfd);

            exit(0);                        //退出子进程
        }
        else if(cpid > 0)
        {
            close(newfd);                   //对于父进程而言，newfd没有用
        }
        else
        {
            ERR_MSG("fork");
            return -1;
        }

    }

    if(close(sfd) < 0)
    {
        ERR_MSG("close");
        return -1;
    }
    return 0;
}

int do_cli_msg(int newfd, struct sockaddr_in cin)
{
    char buf[128] = "";
    ssize_t res = 0;
    while(1)
    {
        bzero(buf, sizeof(buf));
        //接收
        res = recv(newfd, buf, sizeof(buf), 0);
        if(res < 0)
        {
            ERR_MSG("recv");
            return -1;
        }
        else if(0 == res)
        {
            printf("[%s:%d] newfd=%d 客户端下线 __%d__\n", \
                    inet_ntoa(cin.sin_addr), ntohs(cin.sin_port), newfd, __LINE__);
            break;
        }

        printf("[%s:%d] newfd=%d:%s __%d__\n", \
                inet_ntoa(cin.sin_addr), ntohs(cin.sin_port), newfd, buf, __LINE__);


        //发送
        strcat(buf, "*_*");

        if(send(newfd, buf, sizeof(buf), 0) < 0)
        {
            ERR_MSG("send");
            return -1;
        }
        printf("send success __%d__\n", __LINE__);
    }

    return 0;
}