unix网络编程（2）---客户端-服务器第二版

前一节讲述了最基本的客户端-服务器，本节讲述如何解决僵尸子进程回收问题。

我们知道，子进程退出后，父进程可以通过wait获取子进程的状态，而后清楚子进程剩余内核空间，从而避免僵尸进程造成的内存泄露。

对于wait函数，一般使用wait或waitpid两个函数，其区别主要是，waitpid可以指定pid获取子进程状态，并且waitpid具有非阻塞进程选项。我们这里使用waitpid函数。

#include <sys/wait.h>

pid_t waitpid(pid_t pid, int *statloc, int option);

若成功，返回子进程id。出错返回0或-1。

服务器代码如下：（客户端复用第一版）

#include <sys/socket.h>
#include <sys/wait.h>
#include <netinet/in.h>
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <stdlib.h>
#include <errno.h>

#define MAXLINE 1024

extern int errno;

void str_echo(int);

void sig_chld(int);

int main() {
    int sockfd;
    sockfd = socket(AF_INET, SOCK_STREAM, 0);

    struct sockaddr_in servaddr, cliaddr;
    bzero(&servaddr, sizeof(servaddr));
    servaddr.sin_family = AF_INET;
    servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
    servaddr.sin_port = htons(7070);
    bind(sockfd, (struct sockaddr *)&servaddr, sizeof(servaddr));
    listen(sockfd, 1024);
    signal(SIGCHLD, sig_chld);

    for (;;) {
        int connfd, childPid;
        socklen_t len = sizeof(cliaddr);
        connfd = accept(sockfd, (struct sockaddr *)&cliaddr, &len);

        if ((childPid = fork()) == 0) {
            close(sockfd);
            printf("connected with client.
");
            str_echo(connfd);
            exit(0);
        }
    }

    printf("server end!
");
    return 0;
}

void str_echo(int sockfd) {
    ssize_t n;
    char buf[MAXLINE];

again:

    while ((n = read(sockfd, buf, MAXLINE)) > 0) {
        printf("n:%ld
", n);
        write(sockfd, buf, n);
        bzero(buf, MAXLINE);

        if (n < 0 && errno == EINTR) {
            goto again;
        } else if (n < 0) {
            printf("str_echo:read error
");
        }
    }
}

void sig_chld(int signo) {
    pid_t pid;
    int stat;
    while ((pid = waitpid(-1, &stat, WNOHANG)) > 0) {
        printf("child %d terminated
", pid);
    }
    return;
}