tcpServer.c

Go to the documentation of this file.

 
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
 
#include <console-util.h>
 
#include "tcpServer.h"
 
/* doxygen コメントは、ヘッダに記載 */
void handle_client_session(ClientFd fd) {
    char buffer[BUFFER_SIZE];
    int  n;
 
    printf("[PID %lu] クライアント処理開始\n", (unsigned long)get_pid());
 
    while ((n = (int)client_recv(fd, buffer, sizeof(buffer) - 1)) > 0) {
        buffer[n] = '\0';
        printf("[PID %lu] 受信: %s", (unsigned long)get_pid(), buffer);
        client_send(fd, buffer, (size_t)n);
    }
 
    printf("[PID %lu] クライアント切断\n", (unsigned long)get_pid());
    client_close(fd);
}

static void parse_args(int argc, char *argv[],
                       ServerMode *mode, int *port, int *workers,
                       int *conns_per_worker) {
    for (int i = 1; i < argc; i++) {
        if (strcmp(argv[i], "--mode") == 0 && i + 1 < argc) {
            i++;
            if (strcmp(argv[i], "fork") == 0) {
                *mode = MODE_FORK;
            } else if (strcmp(argv[i], "prefork") == 0) {
                *mode = MODE_PREFORK;
            }
        } else if (strcmp(argv[i], "--port") == 0 && i + 1 < argc) {
            *port = atoi(argv[++i]);
        } else if (strcmp(argv[i], "--workers") == 0 && i + 1 < argc) {
            *workers = atoi(argv[++i]);
        } else if (strcmp(argv[i], "--conns-per-worker") == 0 && i + 1 < argc) {
            *conns_per_worker = atoi(argv[++i]);
        }
    }
}

int main(int argc, char *argv[]) {
    console_init();
    platform_init(handle_client_session);
 
    if (dispatch_internal_args(argc, argv)) {
        platform_cleanup();
        console_dispose();
        return 0;
    }
 
    ServerMode mode             = MODE_PREFORK;
    int        port             = DEFAULT_PORT;
    int        workers          = DEFAULT_WORKERS;
    int        conns_per_worker = DEFAULT_CONNS_PER_WORKER;
 
    parse_args(argc, argv, &mode, &port, &workers, &conns_per_worker);
 
    if (mode == MODE_FORK) {
        run_fork_server(port);
    } else {
        run_prefork_server(port, workers, conns_per_worker);
    }
 
    platform_cleanup();
    console_dispose();
    return 0;
}