68 lines
2.2 KiB
C
68 lines
2.2 KiB
C
/* Copyright (C) 2015-2017 Jason A. Donenfeld <Jason@zx2c4.com>. All Rights Reserved. */
|
|
|
|
#include <stddef.h>
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
|
|
#include "subcommands.h"
|
|
|
|
const char *PROG_NAME;
|
|
|
|
static const struct {
|
|
const char *subcommand;
|
|
int (*function)(int, char**);
|
|
const char *description;
|
|
} subcommands[] = {
|
|
{ "show", show_main, "Shows the current configuration and device information" },
|
|
{ "showconf", showconf_main, "Shows the current configuration of a given WireGuard interface, for use with `setconf`" },
|
|
{ "set", set_main, "Change the current configuration, add peers, remove peers, or change peers" },
|
|
{ "setconf", setconf_main, "Applies a configuration file to a WireGuard interface" },
|
|
{ "addconf", setconf_main, "Appends a configuration file to a WireGuard interface" },
|
|
{ "genkey", genkey_main, "Generates a new private key and writes it to stdout" },
|
|
{ "genpsk", genkey_main, "Generates a new preshared key and writes it to stdout" },
|
|
{ "pubkey", pubkey_main, "Reads a private key from stdin and writes a public key to stdout" }
|
|
};
|
|
|
|
static void show_usage(FILE *file)
|
|
{
|
|
fprintf(file, "Usage: %s <cmd> [<args>]\n\n", PROG_NAME);
|
|
fprintf(file, "Available subcommands:\n");
|
|
for (size_t i = 0; i < sizeof(subcommands) / sizeof(subcommands[0]); ++i)
|
|
fprintf(file, " %s: %s\n", subcommands[i].subcommand, subcommands[i].description);
|
|
fprintf(file, "You may pass `--help' to any of these subcommands to view usage.\n");
|
|
}
|
|
|
|
int main(int argc, char *argv[])
|
|
{
|
|
char *tmp = NULL;
|
|
PROG_NAME = argv[0];
|
|
|
|
if (argc == 2 && (!strcmp(argv[1], "-h") || !strcmp(argv[1], "--help") || !strcmp(argv[1], "help"))) {
|
|
show_usage(stdout);
|
|
return 0;
|
|
}
|
|
|
|
if (argc == 1) {
|
|
char *new_argv[] = { "show", NULL };
|
|
return show_main(1, new_argv);
|
|
}
|
|
|
|
findsubcommand:
|
|
for (size_t i = 0; i < sizeof(subcommands) / sizeof(subcommands[0]); ++i) {
|
|
if (!strcmp(argv[1], subcommands[i].subcommand))
|
|
return subcommands[i].function(argc - 1, argv + 1);
|
|
}
|
|
|
|
/* Crude way of supporting "wg wg0 show..." */
|
|
if (!tmp && argc >= 3) {
|
|
tmp = argv[1];
|
|
argv[1] = argv[2];
|
|
argv[2] = tmp;
|
|
goto findsubcommand;
|
|
}
|
|
|
|
fprintf(stderr, "Invalid subcommand: `%s`\n", argv[1]);
|
|
show_usage(stderr);
|
|
return 1;
|
|
}
|