diff --git a/youtubeUnblock.c b/youtubeUnblock.c index 118553b..b8bc4a7 100644 --- a/youtubeUnblock.c +++ b/youtubeUnblock.c @@ -26,6 +26,9 @@ #include #include +#include +#include + #include "config.h" #include "mangle.h" @@ -59,34 +62,7 @@ struct config_t config = { .domains_strlen = sizeof(defaul_snistr), }; -const char* get_value(const char *option, const char *prefix) -{ - errno = 0; - - size_t prefix_len = strlen(prefix); - size_t option_len = strlen(option); - - if (option_len <= prefix_len || strncmp(prefix, option, prefix_len)) { - return NULL; - } - - return option + prefix_len; -} - -int parse_bool_option(const char *value) { - errno = 0; - if (strcmp(value, "1") == 0) { - return 1; - } - else if (strcmp(value, "0") == 0) { - return 0; - } - - errno = EINVAL; - return -1; -} - -long parse_numeric_option(const char* value) { +static long parse_numeric_option(const char* value) { errno = 0; char* end; @@ -99,135 +75,12 @@ long parse_numeric_option(const char* value) { return result; } -int parse_option(const char* option) { - const char* value; - int ret; - - if (!strcmp(option, "--no-gso")) { - config.use_gso = 0; - goto out; - } - - if (!strcmp(option, "--silent")) { - config.verbose = 0; - goto out; - } - - if ((value = get_value(option, "--sni-domains")) != 0) { - if (!value) { - goto err; - } - - if (strcmp(value, "all")) { - config.all_domains = 1; - } - - config.domains_str = value; - config.domains_strlen = strlen(value); - - goto out; - } - - if ((value = get_value(option, "--frag=")) != 0) { - if (!value) { - goto err; - } - - if (strcmp(value, "tcp") == 0) { - config.fragmentation_strategy = FRAG_STRAT_TCP; - } else if (strcmp(value, "ip") == 0) { - config.fragmentation_strategy = FRAG_STRAT_IP; - } else if (strcmp(value, "none") == 0) { - config.fragmentation_strategy = FRAG_STRAT_NONE; - } else { - goto err; - } - - goto out; - } - - if ((value = get_value(option, "--fake-sni=")) != 0) { - if (strcmp(value, "ack") == 0) { - config.fake_sni_strategy = FKSN_STRAT_ACK_SEQ; - } else if (strcmp(value, "ttl") == 0) { - config.fake_sni_strategy = FKSN_STRAT_TTL; - } - else if (strcmp(value, "none") == 0) { - config.fake_sni_strategy = FKSN_STRAT_NONE; - } else { - goto err; - } - - goto out; - } - - if ((value = get_value(option, "--seg2delay=")) != 0) { - long num = parse_numeric_option(value); - if (errno != 0 || - num < 0) - goto err; - - config.seg2_delay = num; - goto out; - } - - if ((value = get_value(option, "--threads=")) != 0) { - long num = parse_numeric_option(value); - if (errno != 0 || - num < 0 || - num > MAX_THREADS) { - errno = EINVAL; - goto err; - } - - config.threads = num; - goto out; - } - - if ((value = get_value(option, "--fake-sni-ttl=")) != 0) { - long num = parse_numeric_option(value); - if (errno != 0 || - num < 0 || - num > 255) { - goto err; - } - - config.fake_sni_ttl = num; - goto out; - } - -err: - errno = EINVAL; - return -1; -out: - errno = 0; - return 0; +static void print_version() { + printf("Youtube unblocker v0.0.1\n"); //TODO } -static int parse_args(int argc, const char *argv[]) { - int err; - char *end; - - if (argc < 2) { - errno = EINVAL; - goto errormsg_help; - } - - config.queue_start_num = parse_numeric_option(argv[1]); - if (errno != 0) goto errormsg_help; - - for (int i = 2; i < argc; i++) { - if (parse_option(argv[i])) { - printf("Invalid option %s\n", argv[i]); - goto errormsg_help; - } - } - - return 0; - -errormsg_help: - err = errno; - printf("Usage: %s [OPTIONS]\n", argv[0]); +static void print_usage(const char *argv0) { + printf("Usage: %s [ OPTIONS ] \n", argv0); printf("Options:\n"); printf("\t--sni-domains=|all\n"); printf("\t--fake-sni={ack,ttl,none}\n"); @@ -237,10 +90,131 @@ errormsg_help: printf("\t--threads=\n"); printf("\t--silent\n"); printf("\t--no-gso\n"); - errno = err; - if (errno == 0) errno = EINVAL; +} - return -1; +#define OPT_SNI_DOMAINS 1 +#define OPT_FAKE_SNI 2 +#define OPT_FAKE_SNI_TTL 3 +#define OPT_FRAG 4 +#define OPT_SEG2DELAY 5 +#define OPT_THREADS 6 +#define OPT_SILENT 7 +#define OPT_NO_GSO 8 + +static struct option long_opt[] = { + {"help", 0, 0, 'h'}, + {"version", 0, 0, 'v'}, + {"sni-domains", 1, 0, OPT_SNI_DOMAINS}, + {"fake-sni", 1, 0, OPT_FAKE_SNI}, + {"fake-sni-ttl", 1, 0, OPT_FAKE_SNI_TTL}, + {"frag", 1, 0, OPT_FRAG}, + {"seg2delay", 1, 0, OPT_SEG2DELAY}, + {"threads", 1, 0, OPT_THREADS}, + {"silent", 0, 0, OPT_SILENT}, + {"no-gso", 0, 0, OPT_NO_GSO}, + {0,0,0,0} + }; + +static int parse_args(int argc, const char *argv[]) { + int opt; + int optIdx; + long num; + + if (argc < 2) { + print_usage(argv[0]); + errno = EINVAL; + return -1; + } + + while ((opt = getopt_long(argc, argv, "hv", long_opt, &optIdx)) != -1) { + switch (opt) { + case 'h': + print_usage(argv[0]); + errno = 0; + return 1; + case 'v': + print_version(); + errno = 0; + return 1; + case OPT_SILENT: + config.verbose = 0; + break; + case OPT_NO_GSO: + config.use_gso = 0; + break; + case OPT_SNI_DOMAINS: + + if (strcmp(optarg, "all")) { + config.all_domains = 1; + } + config.domains_str = optarg; + config.domains_strlen = strlen(config.domains_str); + + break; + case OPT_FRAG: + if (strcmp(optarg, "tcp") == 0) { + config.fragmentation_strategy = FRAG_STRAT_TCP; + } else if (strcmp(optarg, "ip") == 0) { + config.fragmentation_strategy = FRAG_STRAT_IP; + } else if (strcmp(optarg, "none") == 0) { + config.fragmentation_strategy = FRAG_STRAT_NONE; + } else { + errno = EINVAL; + printf("Invalid option %s\n", long_opt[optIdx].name); + return -1; + } + break; + case OPT_FAKE_SNI: + if (strcmp(optarg, "ack") == 0) { + config.fake_sni_strategy = FKSN_STRAT_ACK_SEQ; + } else if (strcmp(optarg, "ttl") == 0) { + config.fake_sni_strategy = FKSN_STRAT_TTL; + } else if (strcmp(optarg, "none") == 0) { + config.fake_sni_strategy = FKSN_STRAT_NONE; + } else { + errno = EINVAL; + printf("Invalid option %s\n", long_opt[optIdx].name); + return -1; + } + break; + case OPT_SEG2DELAY: + num = parse_numeric_option(optarg); + if (errno != 0 || num < 0) { + printf("Invalid option %s\n", long_opt[optIdx].name); + return -1; + } + config.seg2_delay = num; + break; + case OPT_THREADS: + num = parse_numeric_option(optarg); + if (errno != 0 || num < 0 || num > MAX_THREADS) { + printf("Invalid option %s\n", long_opt[optIdx].name); + return -1; + } + config.threads = num; + break; + case OPT_FAKE_SNI_TTL: + num = parse_numeric_option(optarg); + if (errno != 0 || num < 0 || num > 255) { + printf("Invalid option %s\n", long_opt[optIdx].name); + return -1; + } + config.fake_sni_ttl = num; + break; + default: + printf("Invalid option %s\n", long_opt[optIdx].name); + errno = EINVAL; + return -1; + } + } + + config.queue_start_num = parse_numeric_option(argv[optind]); + if (errno != 0) { + printf("Invalid queue numbmer\n"); + return -1; + } + + return 0; } static int open_socket(struct mnl_socket **_nl) { @@ -788,8 +762,11 @@ void *init_queue_wrapper(void *qdconf) { int main(int argc, const char *argv[]) { if (parse_args(argc, argv)) { - perror("Unable to parse args"); - exit(EXIT_FAILURE); + if (errno) { + perror("Unable to parse args"); + exit(EXIT_FAILURE); + } + exit(EXIT_SUCCESS); } switch (config.fragmentation_strategy) {