Much more helpful help.

This commit is contained in:
Ian Gulliver
2016-02-17 16:04:11 -08:00
parent 5fd120baec
commit 8e2e7b2562
5 changed files with 88 additions and 53 deletions

125
adsbus.c
View File

@@ -14,18 +14,9 @@
#include "stats.h" #include "stats.h"
static bool add_dump(char *format) {
struct serializer *serializer = client_get_serializer(format);
if (!serializer) {
fprintf(stderr, "Unknown dump format: %s\n", format);
return false;
}
client_add(1, serializer);
return true;
}
static void print_usage(char *argv[]) { static void print_usage(char *argv[]) {
fprintf(stderr, fprintf(stderr,
"\n"
"Usage: %s [OPTION]...\n" "Usage: %s [OPTION]...\n"
"\n" "\n"
"Options:\n" "Options:\n"
@@ -35,6 +26,65 @@ static void print_usage(char *argv[]) {
"\t--incoming=[HOST/]PORT\n" "\t--incoming=[HOST/]PORT\n"
"\t--listen=FORMAT=[HOST/]PORT\n" "\t--listen=FORMAT=[HOST/]PORT\n"
, argv[0]); , argv[0]);
backend_print_usage();
client_print_usage();
}
static bool add_dump(char *arg) {
struct serializer *serializer = client_get_serializer(arg);
if (!serializer) {
fprintf(stderr, "Unknown --dump=FORMAT: %s\n", arg);
return false;
}
client_add(1, serializer);
return true;
}
static bool add_backend(char *arg) {
char *port = strrchr(arg, '/');
if (!port) {
fprintf(stderr, "Invalid --backend=HOST/PORT (missing \"/\"): %s\n", arg);
return false;
}
*(port++) = '\0';
backend_new(arg, port);
return true;
}
static bool add_incoming(char *arg){
char *port = strrchr(arg, '/');
if (port) {
*(port++) = '\0';
incoming_new(arg, port, backend_new_fd, NULL);
} else {
incoming_new(NULL, arg, backend_new_fd, NULL);
}
return true;
}
static bool add_listener(char *arg) {
char *host_port = strchr(arg, '=');
if (!host_port) {
fprintf(stderr, "Invalid --listener=FORMAT=[HOST/]PORT (missing \"=\"): %s\n", arg);
return false;
}
*(host_port++) = '\0';
struct serializer *serializer = client_get_serializer(arg);
if (!serializer) {
fprintf(stderr, "Unknown --listener=FORMAT=[HOST/]PORT format: %s\n", arg);
return false;
}
char *port = strrchr(host_port, '/');
if (port) {
*(port++) = '\0';
incoming_new(host_port, port, client_add_wrapper, serializer);
} else {
incoming_new(NULL, host_port, client_add_wrapper, serializer);
}
return true;
} }
static bool parse_opts(int argc, char *argv[]) { static bool parse_opts(int argc, char *argv[]) {
@@ -47,26 +97,15 @@ static bool parse_opts(int argc, char *argv[]) {
}; };
int opt; int opt;
char *delim1, *delim2;
while ((opt = getopt_long_only(argc, argv, "", long_options, NULL)) != -1) { while ((opt = getopt_long_only(argc, argv, "", long_options, NULL)) != -1) {
bool (*handler)(char *) = NULL;
switch (opt) { switch (opt) {
case 'b': case 'b':
// It would be really nice if libc had a standard way to split host:port. handler = add_backend;
delim1 = strrchr(optarg, '/');
if (delim1 == NULL) {
print_usage(argv);
return false;
}
*delim1 = '\0';
delim1++;
backend_new(optarg, delim1);
break; break;
case 'd': case 'd':
if (!add_dump(optarg)) { handler = add_dump;
return false;
}
break; break;
case 'h': case 'h':
@@ -74,44 +113,24 @@ static bool parse_opts(int argc, char *argv[]) {
return false; return false;
case 'i': case 'i':
delim1 = strrchr(optarg, '/'); handler = add_incoming;
if (delim1 == NULL) {
incoming_new(NULL, optarg, backend_new_fd, NULL);
} else {
*delim1 = '\0';
delim1++;
incoming_new(optarg, delim1, backend_new_fd, NULL);
}
break; break;
case 'l': case 'l':
delim1 = strchr(optarg, '='); handler = add_listener;
if (delim1 == NULL) {
print_usage(argv);
return false;
}
*delim1 = '\0';
delim1++;
struct serializer *serializer = client_get_serializer(optarg);
if (!serializer) {
fprintf(stderr, "Unknown format: %s\n", optarg);
return false;
}
delim2 = strrchr(delim1, '/');
if (delim2 == NULL) {
incoming_new(NULL, delim1, client_add_wrapper, serializer);
} else {
*delim2 = '\0';
delim2++;
incoming_new(delim1, delim2, client_add_wrapper, serializer);
}
break; break;
default: default:
print_usage(argv); print_usage(argv);
return false; return false;
} }
if (handler) {
if (!handler(optarg)) {
print_usage(argv);
return false;
}
}
} }
if (optind != argc) { if (optind != argc) {

View File

@@ -178,3 +178,10 @@ static bool backend_autodetect_parse(struct backend *backend, struct packet *pac
} }
return false; return false;
} }
void backend_print_usage() {
fprintf(stderr, "\nSupported input formats (autodetected):\n");
for (int i = 0; i < NUM_PARSERS; i++) {
fprintf(stderr, "\t%s\n", parsers[i].name);
}
}

View File

@@ -23,3 +23,4 @@ struct backend {
void backend_new(char *, char *); void backend_new(char *, char *);
void backend_new_fd(int, void *); void backend_new_fd(int, void *);
void backend_print_usage();

View File

@@ -138,3 +138,10 @@ void client_write(struct packet *packet) {
} }
} }
} }
void client_print_usage() {
fprintf(stderr, "\nSupported output formats:\n");
for (int i = 0; i < NUM_SERIALIZERS; i++) {
fprintf(stderr, "\t%s\n", serializers[i].name);
}
}

View File

@@ -8,3 +8,4 @@ struct serializer *client_get_serializer(char *);
void client_add(int, struct serializer *); void client_add(int, struct serializer *);
void client_add_wrapper(int, void *); void client_add_wrapper(int, void *);
void client_write(struct packet *); void client_write(struct packet *);
void client_print_usage();