transocks.c

//
// Created by wong on 10/24/18.
//
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <signal.h>
#include <unistd.h>
#include <getopt.h>

#include "util.h"
#include "log.h"
#include "context.h"
#include "signal.h"
#include "listener.h"
#include "pump.h"


static transocks_global_env *globalEnv = NULL;

int main(int argc, char **argv) {
    int opt;

    char *listenerAddrPort = NULL;
    char *socks5AddrPort = NULL;
    char *pumpMethod = NULL;

    struct sockaddr_storage listener_ss;
    socklen_t listener_ss_size;
    struct sockaddr_storage socks5_ss;
    socklen_t socks5_ss_size;

    static struct option long_options[] = {
            {"listener-addr-port", required_argument, NULL, GETOPT_VAL_LISTENERADDRPORT},
            {"socks5-addr-port",   required_argument, NULL, GETOPT_VAL_SOCKS5ADDRPORT},
            {"pump-method",        optional_argument, NULL, GETOPT_VAL_PUMPMETHOD},
            {"help",               no_argument,       NULL, GETOPT_VAL_HELP},
            {NULL, 0,                                 NULL, 0}
    };

    while ((opt = getopt_long(argc, argv, "", long_options, NULL)) != -1) {
        switch (opt) {
            case GETOPT_VAL_LISTENERADDRPORT:
                listenerAddrPort = optarg;
                break;
            case GETOPT_VAL_SOCKS5ADDRPORT:
                socks5AddrPort = optarg;
                break;
            case GETOPT_VAL_PUMPMETHOD:
                pumpMethod = optarg;
                break;
            case '?':
            case 'h':
            case GETOPT_VAL_HELP:
                PRINTHELP_EXIT();
            default:
                PRINTHELP_EXIT();
        }
    }

    if (listenerAddrPort == NULL
        || socks5AddrPort == NULL) {
        PRINTHELP_EXIT();
    }

    if (pumpMethod == NULL) {
        pumpMethod = PUMPMETHOD_BUFFER;
    }

    // ignore SIGPIPE
    signal(SIGPIPE, SIG_IGN);

    if (transocks_parse_sockaddr_port(listenerAddrPort, (struct sockaddr *) &listener_ss, &listener_ss_size) != 0) {
        FATAL_WITH_HELPMSG("invalid listener address and port: %s", listenerAddrPort);
    }
    if (transocks_parse_sockaddr_port(socks5AddrPort, (struct sockaddr *) &socks5_ss, &socks5_ss_size) != 0) {
        FATAL_WITH_HELPMSG("invalid socks5 address and port: %s", socks5AddrPort);
    }

    // check if port exists
    if (!validateAddrPort(&listener_ss)) {
        FATAL_WITH_HELPMSG("fail to parse listener address port: %s", listenerAddrPort);
    }
    if (!validateAddrPort(&socks5_ss)) {
        FATAL_WITH_HELPMSG("fail to parse socks5 address port: %s", socks5AddrPort);
    }


    globalEnv = transocks_global_env_new();
    if (globalEnv == NULL) {
        goto bareExit;
    }
    if (signal_init(globalEnv) != 0) {
        goto shutdown;
    }
    memcpy(globalEnv->bind_addr, &listener_ss, sizeof(struct sockaddr_storage));
    globalEnv->bind_addr_len = listener_ss_size;
    memcpy(globalEnv->relay_addr, &socks5_ss, sizeof(struct sockaddr_storage));
    globalEnv->relay_addr_len = socks5_ss_size;

    if (listener_init(globalEnv) != 0) {
        goto shutdown;
    }
    globalEnv->pump_method_name = strdup(pumpMethod);
    if (globalEnv->pump_method_name == NULL) {
        goto shutdown;
    }

    if (transocks_pump_init(globalEnv) != 0) {
        goto shutdown;
    }

    LOGI("transocks-wong started");
    LOGI("using memory allocator: " TR_USED_MEM_ALLOCATOR);
    LOGI("using pumpmethod: %s", globalEnv->pump_method_name);

    // start event loop
    event_base_dispatch(globalEnv->eventBaseLoop);

    LOGI("exited event loop, shutting down..");

    shutdown:

    // exit gracefully
    transocks_drop_all_clients(globalEnv);
    // report intentional event loop break
    if (event_base_got_exit(globalEnv->eventBaseLoop)
        || event_base_got_break(globalEnv->eventBaseLoop)) {
        LOGE("exited event loop intentionally");
    }
    // we are done, bye
    TRANSOCKS_FREE(transocks_global_env_free, globalEnv);
    bareExit:
    return 0;
}