基于linux的预先创建子进程池的服务器程序设计

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

#define PORT 8080
#define MAX_CLIENTS 10
#define MAX_WORKERS 5

void worker_process(int client_socket) {
    char buffer[1024];
    int valread = read(client_socket, buffer, sizeof(buffer));
    printf("Received: %s\n", buffer);
    send(client_socket, "Hello from server", 17, 0);
    close(client_socket);
}

int main() {
    int server_fd, new_socket;
    struct sockaddr_in address;
    int addrlen = sizeof(address);

    // 创建socket
    if ((server_fd = socket(AF_INET, SOCK_STREAM, 0)) == 0) {
        perror("socket failed");
        exit(EXIT_FAILURE);
    }

    // 绑定socket
    address.sin_family = AF_INET;
    address.sin_addr.s_addr = INADDR_ANY;
    address.sin_port = htons(PORT);
    if (bind(server_fd, (struct sockaddr *)&address, sizeof(address)) < 0) {
        perror("bind failed");
        exit(EXIT_FAILURE);
    }

    // 监听
    if (listen(server_fd, MAX_CLIENTS) < 0) {
        perror("listen");
        exit(EXIT_FAILURE);
    }

    pid_t workers[MAX_WORKERS];
    for (int i = 0; i < MAX_WORKERS; i++) {
        workers[i] = fork();
        if (workers[i] == 0) {
            while (1) {
                if ((new_socket = accept(server_fd, (struct sockaddr *)&address, (socklen_t*)&addrlen)) < 0) {
                    perror("accept");
                    exit(EXIT_FAILURE);
                }
                worker_process(new_socket);
            }
        }
    }

    for (int i = 0; i < MAX_WORKERS; i++) {
        wait(NULL);
    }

    return 0;
}

#include <sys/resource.h>

void set_resource_limits() {
    struct rlimit rl;
    rl.rlim_cur = 1024 * 1024 * 50; // 50 MB
    rl.rlim_max = 1024 * 1024 * 100; // 100 MB
    setrlimit(RLIMIT_AS, &rl);
}