tcpServer.h

Go to the documentation of this file.

 
#ifndef TCPSERVER_H
#define TCPSERVER_H
 
/* ============================================================
 *  プラットフォーム抽象化
 *  handle_client_session() が ClientFd / get_pid() / client_recv() 等を
 *  プラットフォームを意識せずに使えるようにします。
 * ============================================================ */
 
#ifndef _WIN32
 
    #include <sys/types.h>
    #include <unistd.h>

    typedef int     ClientFd;
    typedef pid_t   PidType;

    #define get_pid()                    ((PidType)getpid())
    #define client_recv(fd, buf, len)    read((fd), (buf), (len))
    #define client_send(fd, buf, len)    write((fd), (buf), (len))
    #define client_close(fd)             close(fd)
 
#else /* _WIN32 */
 
    #define WIN32_LEAN_AND_MEAN
    #include <windows.h>
    #include <winsock2.h>
    #include <ws2tcpip.h>

    typedef SOCKET  ClientFd;
    typedef DWORD   PidType;

    #define get_pid()                    ((PidType)GetCurrentProcessId())
    #define client_recv(fd, buf, len)    recv((fd), (buf), (int)(len), 0)
    #define client_send(fd, buf, len)    send((fd), (buf), (int)(len), 0)
    #define client_close(fd)             closesocket(fd)
 
#endif /* _WIN32 */
 
/* ============================================================
 *  共通定数
 * ============================================================ */

#define DEFAULT_PORT    8080
#define DEFAULT_WORKERS          4
#define DEFAULT_CONNS_PER_WORKER 1
#define BUFFER_SIZE              1024
 
/* ============================================================
 *  共通型定義
 * ============================================================ */

typedef enum {
    MODE_PREFORK = 0, 
    MODE_FORK    = 1  
} ServerMode;

typedef void (*ClientSessionFn)(ClientFd fd);

extern ClientSessionFn g_session_fn;
 
/* ============================================================
 *  関数プロトタイプ
 * ============================================================ */
 
/* --- 共通実装 (tcpServer.c) --- */

void handle_client_session(ClientFd fd);
 
/* --- プラットフォームフック (各プラットフォームファイルで実装) --- */

void platform_init(ClientSessionFn session_fn);

void platform_cleanup(void);

int dispatch_internal_args(int argc, char *argv[]);

void run_fork_server(int port);

void run_prefork_server(int port, int num_workers, int conns_per_worker);
 
#endif /* TCPSERVER_H */