mirror of
https://github.com/Waujito/youtubeUnblock.git
synced 2024-12-22 06:15:31 +00:00
parsing args with getopt
This commit is contained in:
zabbius
parent
7d60fd8854
commit
916d575920
297
youtubeUnblock.c
297
youtubeUnblock.c
@ -26,6 +26,9 @@
|
|||||||
#include <pthread.h>
|
#include <pthread.h>
|
||||||
#include <sys/socket.h>
|
#include <sys/socket.h>
|
||||||
|
|
||||||
|
#include <unistd.h>
|
||||||
|
#include <getopt.h>
|
||||||
|
|
||||||
#include "config.h"
|
#include "config.h"
|
||||||
#include "mangle.h"
|
#include "mangle.h"
|
||||||
|
|
||||||
@ -59,34 +62,7 @@ struct config_t config = {
|
|||||||
.domains_strlen = sizeof(defaul_snistr),
|
.domains_strlen = sizeof(defaul_snistr),
|
||||||
};
|
};
|
||||||
|
|
||||||
const char* get_value(const char *option, const char *prefix)
|
static long parse_numeric_option(const char* value) {
|
||||||
{
|
|
||||||
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) {
|
|
||||||
errno = 0;
|
errno = 0;
|
||||||
|
|
||||||
char* end;
|
char* end;
|
||||||
@ -99,135 +75,12 @@ long parse_numeric_option(const char* value) {
|
|||||||
return result;
|
return result;
|
||||||
}
|
}
|
||||||
|
|
||||||
int parse_option(const char* option) {
|
static void print_version() {
|
||||||
const char* value;
|
printf("Youtube unblocker v0.0.1\n"); //TODO
|
||||||
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 int parse_args(int argc, const char *argv[]) {
|
static void print_usage(const char *argv0) {
|
||||||
int err;
|
printf("Usage: %s [ OPTIONS ] <queue_num> \n", argv0);
|
||||||
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 <queue_num> [OPTIONS]\n", argv[0]);
|
|
||||||
printf("Options:\n");
|
printf("Options:\n");
|
||||||
printf("\t--sni-domains=<comma separated domain list>|all\n");
|
printf("\t--sni-domains=<comma separated domain list>|all\n");
|
||||||
printf("\t--fake-sni={ack,ttl,none}\n");
|
printf("\t--fake-sni={ack,ttl,none}\n");
|
||||||
@ -237,10 +90,131 @@ errormsg_help:
|
|||||||
printf("\t--threads=<threads number>\n");
|
printf("\t--threads=<threads number>\n");
|
||||||
printf("\t--silent\n");
|
printf("\t--silent\n");
|
||||||
printf("\t--no-gso\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) {
|
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[]) {
|
int main(int argc, const char *argv[]) {
|
||||||
if (parse_args(argc, argv)) {
|
if (parse_args(argc, argv)) {
|
||||||
perror("Unable to parse args");
|
if (errno) {
|
||||||
exit(EXIT_FAILURE);
|
perror("Unable to parse args");
|
||||||
|
exit(EXIT_FAILURE);
|
||||||
|
}
|
||||||
|
exit(EXIT_SUCCESS);
|
||||||
}
|
}
|
||||||
|
|
||||||
switch (config.fragmentation_strategy) {
|
switch (config.fragmentation_strategy) {
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user