调用fork()后Socket服务器无法退出，需服务指定客户端后终止

嘿，这个坑我踩过！咱们来一步步搞清楚为什么加了fork()之后服务器就停不下来，然后把代码改得符合你的需求——服务2个客户端后就退出。

原代码的问题出在哪？ #

咱们先拆解原代码的逻辑：

• 父进程调用fork()后，子进程会完全复制父进程的所有上下文，包括count变量和循环的执行状态。这意味着子进程会接着父进程的循环继续跑，也会去调用accept()，等于同时有多个进程在等待客户端连接，自然停不下来。
• 子进程处理完客户端（你的代码里只是打了个printf）之后，没有退出，反而继续进入下一次循环，无限等待新的客户端，完全不受父进程的count < 2限制。
• 另外父进程没回收子进程的资源，会产生僵尸进程，不过这是次要问题，核心还是子进程没退出导致的无限循环。

修正后的代码 #

#include <stdio.h>
#include <unistd.h>
#include <sys/wait.h>
#include <sys/socket.h>

// 假设你已经完成了server_socket的创建、绑定、监听等初始化步骤

int main() {
    int server_socket; // 已初始化的监听套接字
    int client_socket;
    pid_t pid;
    int count;

    for(count = 0; count < 2; count++) {
        // 等待客户端连接
        client_socket = accept(server_socket, NULL, NULL);
        if(client_socket == -1) {
            perror("Failed to accept client");
            continue;
        }

        pid = fork();
        if(pid == -1) {
            perror("Failed to fork process");
            close(client_socket);
            continue;
        } else if(pid == 0) {
            // 子进程：处理客户端请求
            printf("from Server\n");
            
            // 子进程不需要监听套接字，关闭节省资源
            close(server_socket);
            // 处理完客户端后关闭连接
            close(client_socket);
            // 子进程必须退出，不能继续执行父进程的循环！
            _exit(0);
        } else {
            // 父进程：关闭客户端套接字（子进程已经复制了这个文件描述符）
            close(client_socket);
            // 非阻塞回收子进程，避免僵尸进程，不影响父进程继续接受连接
            waitpid(pid, NULL, WNOHANG);
        }
    }

    // 父进程等待所有子进程处理完毕，再关闭监听套接字
    while(wait(NULL) > 0);
    close(server_socket);

    return 0;
}

关键修改点说明 #

• 子进程必须退出：在子进程处理完客户端逻辑后，一定要调用_exit(0)（用_exit而不是exit是为了避免子进程刷新父进程的缓冲区，更安全），这样子进程就不会继续执行循环，只有父进程在控制服务客户端的数量。
• 子进程关闭监听套接字：子进程不需要继续监听新连接，关闭server_socket可以避免不必要的资源占用。
• 父进程回收子进程：用waitpid（带WNOHANG选项）或者wait回收子进程的退出状态，防止产生僵尸进程。最后用while(wait(NULL) > 0)确保所有子进程都被回收完再退出。
• 增加错误处理：给accept和fork加上错误处理，让代码更健壮，遇到问题能快速定位。

这样修改后，父进程会严格循环2次，接受2个客户端连接，每个客户端交给子进程处理后，子进程立即退出，父进程回收资源，最后等所有子进程处理完就关闭服务器套接字，完美实现你的需求！

内容的提问来源于stack exchange，提问作者Ankit Negi