TCP并发服务器

/*****************************************
        TCP并发服务器模型
    同时处理多种客户端请求
******************************************/

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

typedef struct sockaddr SA;

#define  N  64

int main(int argc, char *argv[])
{
    int listenfd, connfd;
    char buf[N];
    pid_t pid;
    struct sockaddr_in servaddr;

    if (argc < 3)
    {
        printf("Usage : %s <ip> <port>\n", argv[0]);
        return -1;
    }

    // XXX int socket(int domain, int type, int protocol);
    if ((listenfd = socket(PF_INET, SOCK_STREAM, 0)) < 0)
    {
        perror("fail to socket");
        exit(-1);
    }

    bzero(&servaddr, sizeof(servaddr));
    servaddr.sin_family = PF_INET;
    servaddr.sin_port = htons(atoi(argv[2]));
    servaddr.sin_addr.s_addr = inet_addr(argv[1]);
    
    // XXX int bind(int sockfd, const struct sockaddr *addr, socklen_t addrlen);
    if (bind(listenfd, (SA *)&servaddr, sizeof(servaddr)) < 0)
    {
        perror("fail to bind");
        exit(-1);
    }

    if (listen(listenfd, 5) < 0)
    {
        perror("fail to listen");
        exit(-1);
    }

    signal(SIGCHLD, SIG_IGN); \\忽略SIGCHLD信号，这样子进程结束的资源回收由子进程自行处理，避免僵尸进程
    while ( 1 )
    {
        if ((connfd = accept(listenfd, NULL, NULL)) < 0)
        {
            perror("fail to accept");
            exit(-1);
        }
        if ((pid = fork()) < 0) {perror("fork"); return -1;}
        if (pid == 0) //child process
        {
            while (recv(connfd, buf, N, 0) > 0)
            {
                printf("recv---%s", buf);
                send(connfd, buf, N, 0);
            }
            close(connfd);
            printf("disconnect...\n");
            exit(0);
        }
        else //father process
        {
            close(connfd);
        }
    }

    return 0;
}