j3d1/tinc
1
0
Fork 0
Code Issues 1 Pull requests Releases Wiki Activity

Put every command in its own function.

Browse source
This commit is contained in:
Guus Sliepen 2012-07-15 15:46:16 +02:00
parent a444ec3964
commit dd102efd24
1 changed files with 411 additions and 320 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

731
src/tincctl.c
View file

@ -46,6 +46,14 @@ static char controlcookie[1024];
char *netname = NULL; char *netname = NULL;
char *confbase = NULL; char *confbase = NULL;
// Horrible global variables...
static int pid = 0;
static int fd = -1;
static char line[4096];
static int code;
static int req;
static int result;
#ifdef HAVE_MINGW #ifdef HAVE_MINGW
static struct WSAData wsa_state; static struct WSAData wsa_state;
#endif #endif
@ -59,6 +67,16 @@ static struct option const long_options[] = {
{NULL, 0, NULL, 0} {NULL, 0, NULL, 0}
}; };
static void version(void) {
printf("%s version %s (built %s %s, protocol %d.%d)\n", PACKAGE,
VERSION, __DATE__, __TIME__, PROT_MAJOR, PROT_MINOR);
printf("Copyright (C) 1998-2012 Ivo Timmermans, Guus Sliepen and others.\n"
"See the AUTHORS file for a complete list.\n\n"
"tinc comes with ABSOLUTELY NO WARRANTY. This is free software,\n"
"and you are welcome to redistribute it under certain conditions;\n"
"see the file COPYING for details.\n");
}
static void usage(bool status) { static void usage(bool status) {
if(status) if(status)
fprintf(stderr, "Try `%s --help\' for more information.\n", fprintf(stderr, "Try `%s --help\' for more information.\n",
@ -542,179 +560,28 @@ static bool remove_service(void) {
} }
#endif #endif
int main(int argc, char *argv[]) { static bool connect_tincd() {
int fd; if(fd >= 0)
int result;
char host[128];
char port[128];
int pid;
program_name = argv[0];
if(!parse_options(argc, argv))
return 1;
make_names();
if(show_version) {
printf("%s version %s (built %s %s, protocol %d.%d)\n", PACKAGE,
VERSION, __DATE__, __TIME__, PROT_MAJOR, PROT_MINOR);
printf("Copyright (C) 1998-2012 Ivo Timmermans, Guus Sliepen and others.\n"
"See the AUTHORS file for a complete list.\n\n"
"tinc comes with ABSOLUTELY NO WARRANTY. This is free software,\n"
"and you are welcome to redistribute it under certain conditions;\n"
"see the file COPYING for details.\n");
return 0;
}
if(show_help) {
usage(false);
return 0;
}
if(optind >= argc) {
fprintf(stderr, "Not enough arguments.\n");
usage(true);
return 1;
}
// First handle commands that don't involve connecting to a running tinc daemon.
if(!strcasecmp(argv[optind], "generate-rsa-keys")) {
return !rsa_keygen(optind < argc - 1 ? atoi(argv[optind + 1]) : 2048);
}
if(!strcasecmp(argv[optind], "generate-ecdsa-keys")) {
return !ecdsa_keygen();
}
if(!strcasecmp(argv[optind], "generate-keys")) {
return !(rsa_keygen(optind < argc - 1 ? atoi(argv[optind + 1]) : 2048) && ecdsa_keygen());
}
if(!strcasecmp(argv[optind], "start")) {
int i, j;
char *c;
char *slash = strrchr(argv[0], '/');
#ifdef HAVE_MINGW
if ((c = strrchr(argv[0], '\\')) > slash)
slash = c;
#endif
if (slash++) {
c = xmalloc((slash - argv[0]) + sizeof("tincd"));
sprintf(c, "%.*stincd", (int)(slash - argv[0]), argv[0]);
}
else
c = "tincd";
argv[0] = c;
for(i = j = 1; argv[i]; ++i)
if (i != optind && strcmp(argv[i], "--") != 0)
argv[j++] = argv[i];
argv[j] = NULL;
execvp(c, argv);
fprintf(stderr, "Could not start %s: %s\n", c, strerror(errno));
return 1;
}
if(!strcasecmp(argv[optind], "init")) {
char *filename = NULL;
xasprintf(&filename, "%s/tinc.conf", confbase);
if(!access(confbase, F_OK)) {
fprintf(stderr, "Configuration file %s already exists!\n", filename);
return 1;
}
if(optind >= argc - 1) {
if(isatty(0) && isatty(1)) {
char buf[1024];
fprintf(stdout, "Enter the Name you want your tinc node to have: ");
fflush(stdout);
if(!fgets(buf, sizeof buf, stdin)) {
fprintf(stderr, "Error while reading stdin: %s\n", strerror(errno));
return 1;
}
int len = strlen(buf);
if(len)
buf[--len] = 0;
if(!len) {
fprintf(stderr, "No name given!\n");
return 1;
}
name = strdup(buf);
} else {
fprintf(stderr, "No Name given!\n");
return 1;
}
} else {
name = strdup(argv[optind + 1]);
if(!*name) {
fprintf(stderr, "No Name given!\n");
return 1;
}
}
for(int i = 0; i < strlen(name); i++) {
if(!isalnum(name[i]) && name[i] != '_') {
fprintf(stderr, "Invalid Name! Only a-z, A-Z, 0-9 and _ are allowed characters.\n");
return 1;
}
}
if(mkdir(CONFDIR, 0755) && errno != EEXIST) {
fprintf(stderr, "Could not create directory %s: %s\n", CONFDIR, strerror(errno));
return 1;
}
if(mkdir(confbase, 0755) && errno != EEXIST) {
fprintf(stderr, "Could not create directory %s: %s\n", confbase, strerror(errno));
return 1;
}
char *filename2 = NULL;
xasprintf(&filename2, "%s/hosts", confbase);
if(mkdir(filename2, 0755) && errno != EEXIST) {
fprintf(stderr, "Could not create directory %s: %s\n", filename, strerror(errno));
return 1;
}
FILE *f = fopen(filename, "w");
if(!f) {
fprintf(stderr, "Could not create file %s: %s\n", filename, strerror(errno));
return 1;
}
fprintf(f, "Name = %s\n", name);
fclose(f);
fclose(stdin);
if(!rsa_keygen(2048) || !ecdsa_keygen())
return false;
return true; return true;
}
/*
* Now handle commands that do involve connecting to a running tinc daemon.
* Authenticate the server by ensuring the parent directory can be
* traversed only by root. Note this is not totally race-free unless all
* ancestors are writable only by trusted users, which we don't verify.
*/
FILE *f = fopen(pidfilename, "r"); FILE *f = fopen(pidfilename, "r");
if(!f) { if(!f) {
fprintf(stderr, "Could not open pid file %s: %s\n", pidfilename, strerror(errno)); fprintf(stderr, "Could not open pid file %s: %s\n", pidfilename, strerror(errno));
return 1; return false;
} }
char host[128];
char port[128];
if(fscanf(f, "%20d %1024s %128s port %128s", &pid, controlcookie, host, port) != 4) { if(fscanf(f, "%20d %1024s %128s port %128s", &pid, controlcookie, host, port) != 4) {
fprintf(stderr, "Could not parse pid file %s\n", pidfilename); fprintf(stderr, "Could not parse pid file %s\n", pidfilename);
return 1; return false;
} }
#ifdef HAVE_MINGW #ifdef HAVE_MINGW
if(WSAStartup(MAKEWORD(2, 2), &wsa_state)) { if(WSAStartup(MAKEWORD(2, 2), &wsa_state)) {
fprintf(stderr, "System call `%s' failed: %s", "WSAStartup", winerror(GetLastError())); fprintf(stderr, "System call `%s' failed: %s", "WSAStartup", winerror(GetLastError()));
return 1; return false;
} }
#endif #endif
@ -729,13 +596,13 @@ int main(int argc, char *argv[]) {
if(getaddrinfo(host, port, &hints, &res) || !res) { if(getaddrinfo(host, port, &hints, &res) || !res) {
fprintf(stderr, "Cannot resolve %s port %s: %s", host, port, strerror(errno)); fprintf(stderr, "Cannot resolve %s port %s: %s", host, port, strerror(errno));
return 1; return false;
} }
fd = socket(res->ai_family, SOCK_STREAM, IPPROTO_TCP); fd = socket(res->ai_family, SOCK_STREAM, IPPROTO_TCP);
if(fd < 0) { if(fd < 0) {
fprintf(stderr, "Cannot create TCP socket: %s\n", sockstrerror(sockerrno)); fprintf(stderr, "Cannot create TCP socket: %s\n", sockstrerror(sockerrno));
return 1; return false;
} }
#ifdef HAVE_MINGW #ifdef HAVE_MINGW
@ -748,210 +615,434 @@ int main(int argc, char *argv[]) {
if(connect(fd, res->ai_addr, res->ai_addrlen) < 0) { if(connect(fd, res->ai_addr, res->ai_addrlen) < 0) {
fprintf(stderr, "Cannot connect to %s port %s: %s\n", host, port, sockstrerror(sockerrno)); fprintf(stderr, "Cannot connect to %s port %s: %s\n", host, port, sockstrerror(sockerrno));
return 1; return false;
} }
freeaddrinfo(res); freeaddrinfo(res);
char line[4096];
char data[4096]; char data[4096];
int code, version, req; int version;
if(!recvline(fd, line, sizeof line) || sscanf(line, "%d %s %d", &code, data, &version) != 3 || code != 0) { if(!recvline(fd, line, sizeof line) || sscanf(line, "%d %s %d", &code, data, &version) != 3 || code != 0) {
fprintf(stderr, "Cannot read greeting from control socket: %s\n", fprintf(stderr, "Cannot read greeting from control socket: %s\n", sockstrerror(sockerrno));
sockstrerror(sockerrno)); return false;
return 1;
} }
sendline(fd, "%d ^%s %d", ID, controlcookie, TINC_CTL_VERSION_CURRENT); sendline(fd, "%d ^%s %d", ID, controlcookie, TINC_CTL_VERSION_CURRENT);
if(!recvline(fd, line, sizeof line) || sscanf(line, "%d %d %d", &code, &version, &pid) != 3 || code != 4 || version != TINC_CTL_VERSION_CURRENT) { if(!recvline(fd, line, sizeof line) || sscanf(line, "%d %d %d", &code, &version, &pid) != 3 || code != 4 || version != TINC_CTL_VERSION_CURRENT) {
fprintf(stderr, "Could not fully establish control socket connection\n"); fprintf(stderr, "Could not fully establish control socket connection\n");
return 1; return false;
} }
if(!strcasecmp(argv[optind], "pid")) { return true;
printf("%d\n", pid); }
return 0;
}
if(!strcasecmp(argv[optind], "stop")) {
static int cmd_start(int argc, char *argv[]) {
int i, j;
char *c;
char *slash = strrchr(argv[0], '/');
#ifdef HAVE_MINGW
if ((c = strrchr(argv[0], '\\')) > slash)
slash = c;
#endif
if (slash++) {
c = xmalloc((slash - argv[0]) + sizeof("tincd"));
sprintf(c, "%.*stincd", (int)(slash - argv[0]), argv[0]);
}
else
c = "tincd";
argv[0] = c;
for(i = j = 1; argv[i]; ++i)
if (i != optind && strcmp(argv[i], "--") != 0)
argv[j++] = argv[i];
argv[j] = NULL;
execvp(c, argv);
fprintf(stderr, "Could not start %s: %s\n", c, strerror(errno));
return 1;
}
static int cmd_stop(int argc, char *argv[]) {
#ifndef HAVE_MINGW #ifndef HAVE_MINGW
sendline(fd, "%d %d", CONTROL, REQ_STOP); if(!connect_tincd())
if(!recvline(fd, line, sizeof line) || sscanf(line, "%d %d %d", &code, &req, &result) != 3 || code != CONTROL || req != REQ_STOP || result) { return 1;
fprintf(stderr, "Could not stop tinc daemon\n");
return 1; sendline(fd, "%d %d", CONTROL, REQ_STOP);
} if(!recvline(fd, line, sizeof line) || sscanf(line, "%d %d %d", &code, &req, &result) != 3 || code != CONTROL || req != REQ_STOP || result) {
fprintf(stderr, "Could not stop tinc daemon.\n");
return 1;
}
#else #else
if(!remove_service()) if(!remove_service())
return 1; return 1;
#endif #endif
return 0; return 0;
} }
if(!strcasecmp(argv[optind], "reload")) { static int cmd_restart(int argc, char *argv[]) {
sendline(fd, "%d %d", CONTROL, REQ_RELOAD); return cmd_stop(argc, argv) ?: cmd_start(argc, argv);
if(!recvline(fd, line, sizeof line) || sscanf(line, "%d %d %d", &code, &req, &result) != 3 || code != CONTROL || req != REQ_RELOAD || result) { }
fprintf(stderr, "Could not reload tinc daemon\n");
return 1;
}
return 0;
}
if(!strcasecmp(argv[optind], "restart")) { static int cmd_reload(int argc, char *argv[]) {
sendline(fd, "%d %d", CONTROL, REQ_RESTART); if(!connect_tincd())
if(!recvline(fd, line, sizeof line) || sscanf(line, "%d %d %d", &code, &req, &result) != 3 || code != CONTROL || req != REQ_RESTART || result) { return 1;
fprintf(stderr, "Could not restart tinc daemon\n");
return 1;
}
return 0;
}
if(!strcasecmp(argv[optind], "retry")) { sendline(fd, "%d %d", CONTROL, REQ_RELOAD);
sendline(fd, "%d %d", CONTROL, REQ_RETRY); if(!recvline(fd, line, sizeof line) || sscanf(line, "%d %d %d", &code, &req, &result) != 3 || code != CONTROL || req != REQ_RELOAD || result) {
if(!recvline(fd, line, sizeof line) || sscanf(line, "%d %d %d", &code, &req, &result) != 3 || code != CONTROL || req != REQ_RETRY || result) { fprintf(stderr, "Could not reload configuration.\n");
fprintf(stderr, "Could not retry outgoing connections\n");
return 1;
}
return 0;
}
if(!strcasecmp(argv[optind], "dump")) {
if(argc < optind + 2) {
fprintf(stderr, "Not enough arguments.\n");
usage(true);
return 1;
}
bool do_graph = false;
if(!strcasecmp(argv[optind+1], "nodes"))
sendline(fd, "%d %d", CONTROL, REQ_DUMP_NODES);
else if(!strcasecmp(argv[optind+1], "edges"))
sendline(fd, "%d %d", CONTROL, REQ_DUMP_EDGES);
else if(!strcasecmp(argv[optind+1], "subnets"))
sendline(fd, "%d %d", CONTROL, REQ_DUMP_SUBNETS);
else if(!strcasecmp(argv[optind+1], "connections"))
sendline(fd, "%d %d", CONTROL, REQ_DUMP_CONNECTIONS);
else if(!strcasecmp(argv[optind+1], "graph")) {
sendline(fd, "%d %d", CONTROL, REQ_DUMP_NODES);
sendline(fd, "%d %d", CONTROL, REQ_DUMP_EDGES);
do_graph = true;
printf("digraph {\n");
} else {
fprintf(stderr, "Unknown dump type '%s'.\n", argv[optind+1]);
usage(true);
return 1;
}
while(recvline(fd, line, sizeof line)) {
char node1[4096], node2[4096];
int n = sscanf(line, "%d %d %s to %s", &code, &req, node1, node2);
if(n == 2) {
if(do_graph && req == REQ_DUMP_NODES)
continue;
else {
if(do_graph)
printf("}\n");
return 0;
}
}
if(n < 2)
break;
if(!do_graph)
printf("%s\n", line + 5);
else {
if(req == REQ_DUMP_NODES)
printf(" %s [label = \"%s\"];\n", node1, node1);
else
printf(" %s -> %s;\n", node1, node2);
}
}
fprintf(stderr, "Error receiving dump\n");
return 1; return 1;
} }
if(!strcasecmp(argv[optind], "purge")) { return 0;
sendline(fd, "%d %d", CONTROL, REQ_PURGE);
if(!recvline(fd, line, sizeof line) || sscanf(line, "%d %d %d", &code, &req, &result) != 3 || code != CONTROL || req != REQ_PURGE || result) { }
fprintf(stderr, "Could not purge tinc daemon\n");
return 1; static int cmd_dump(int argc, char *argv[]) {
} if(argc != 2) {
return 0; fprintf(stderr, "Invalid number of arguments.\n");
usage(true);
return 1;
} }
if(!strcasecmp(argv[optind], "debug")) { bool do_graph = false;
int debuglevel, origlevel;
if(argc != optind + 2) { if(!strcasecmp(argv[1], "nodes"))
fprintf(stderr, "Invalid arguments.\n"); sendline(fd, "%d %d", CONTROL, REQ_DUMP_NODES);
return 1; else if(!strcasecmp(argv[1], "edges"))
} sendline(fd, "%d %d", CONTROL, REQ_DUMP_EDGES);
debuglevel = atoi(argv[optind+1]); else if(!strcasecmp(argv[1], "subnets"))
sendline(fd, "%d %d", CONTROL, REQ_DUMP_SUBNETS);
sendline(fd, "%d %d %d", CONTROL, REQ_SET_DEBUG, debuglevel); else if(!strcasecmp(argv[1], "connections"))
if(!recvline(fd, line, sizeof line) || sscanf(line, "%d %d %d", &code, &req, &origlevel) != 3 || code != CONTROL || req != REQ_SET_DEBUG) { sendline(fd, "%d %d", CONTROL, REQ_DUMP_CONNECTIONS);
fprintf(stderr, "Could not purge tinc daemon\n"); else if(!strcasecmp(argv[1], "graph")) {
return 1; sendline(fd, "%d %d", CONTROL, REQ_DUMP_NODES);
} sendline(fd, "%d %d", CONTROL, REQ_DUMP_EDGES);
do_graph = true;
fprintf(stderr, "Old level %d, new level %d\n", origlevel, debuglevel); } else {
return 0; fprintf(stderr, "Unknown dump type '%s'.\n", argv[1]);
usage(true);
return 1;
} }
if(!strcasecmp(argv[optind], "connect")) { if(!connect_tincd())
if(argc != optind + 2) { return 1;
fprintf(stderr, "Invalid arguments.\n");
return 1;
}
char *name = argv[optind + 1];
sendline(fd, "%d %d %s", CONTROL, REQ_CONNECT, name); if(do_graph)
if(!recvline(fd, line, sizeof line) || sscanf(line, "%d %d %d", &code, &req, &result) != 3 || code != CONTROL || req != REQ_CONNECT || result) { printf("digraph {\n");
fprintf(stderr, "Could not connect to %s\n", name);
return 1; while(recvline(fd, line, sizeof line)) {
char node1[4096], node2[4096];
int n = sscanf(line, "%d %d %s to %s", &code, &req, node1, node2);
if(n == 2) {
if(do_graph && req == REQ_DUMP_NODES)
continue;
else {
if(do_graph)
printf("}\n");
return 0;
}
}
if(n < 2)
break;
if(!do_graph)
printf("%s\n", line + 5);
else {
if(req == REQ_DUMP_NODES)
printf(" %s [label = \"%s\"];\n", node1, node1);
else
printf(" %s -> %s;\n", node1, node2);
} }
return 0;
} }
if(!strcasecmp(argv[optind], "disconnect")) { fprintf(stderr, "Error receiving dump.\n");
if(argc != optind + 2) { return 1;
fprintf(stderr, "Invalid arguments.\n"); }
return 1;
}
char *name = argv[optind + 1];
sendline(fd, "%d %d %s", CONTROL, REQ_DISCONNECT, name); static int cmd_purge(int argc, char *argv[]) {
if(!recvline(fd, line, sizeof line) || sscanf(line, "%d %d %d", &code, &req, &result) != 3 || code != CONTROL || req != REQ_DISCONNECT || result) { if(!connect_tincd())
fprintf(stderr, "Could not disconnect %s\n", name); return 1;
return 1;
} sendline(fd, "%d %d", CONTROL, REQ_PURGE);
return 0; if(!recvline(fd, line, sizeof line) || sscanf(line, "%d %d %d", &code, &req, &result) != 3 || code != CONTROL || req != REQ_PURGE || result) {
fprintf(stderr, "Could not purge old information.\n");
return 1;
} }
return 0;
}
static int cmd_debug(int argc, char *argv[]) {
if(argc != 2) {
fprintf(stderr, "Invalid number of arguments.\n");
return 1;
}
if(!connect_tincd())
return 1;
int debuglevel = atoi(argv[1]);
int origlevel;
sendline(fd, "%d %d %d", CONTROL, REQ_SET_DEBUG, debuglevel);
if(!recvline(fd, line, sizeof line) || sscanf(line, "%d %d %d", &code, &req, &origlevel) != 3 || code != CONTROL || req != REQ_SET_DEBUG) {
fprintf(stderr, "Could not set debug level.\n");
return 1;
}
fprintf(stderr, "Old level %d, new level %d.\n", origlevel, debuglevel);
return 0;
}
static int cmd_retry(int argc, char *argv[]) {
if(!connect_tincd())
return 1;
sendline(fd, "%d %d", CONTROL, REQ_RETRY);
if(!recvline(fd, line, sizeof line) || sscanf(line, "%d %d %d", &code, &req, &result) != 3 || code != CONTROL || req != REQ_RETRY || result) {
fprintf(stderr, "Could not retry outgoing connections.\n");
return 1;
}
return 0;
}
static int cmd_connect(int argc, char *argv[]) {
if(argc != 2) {
fprintf(stderr, "Invalid number of arguments.\n");
return 1;
}
if(!connect_tincd())
return 1;
sendline(fd, "%d %d %s", CONTROL, REQ_CONNECT, argv[1]);
if(!recvline(fd, line, sizeof line) || sscanf(line, "%d %d %d", &code, &req, &result) != 3 || code != CONTROL || req != REQ_CONNECT || result) {
fprintf(stderr, "Could not connect to %s.\n", argv[1]);
return 1;
}
return 0;
}
static int cmd_disconnect(int argc, char *argv[]) {
if(argc != 2) {
fprintf(stderr, "Invalid number of arguments.\n");
return 1;
}
if(!connect_tincd())
return 1;
sendline(fd, "%d %d %s", CONTROL, REQ_DISCONNECT, argv[1]);
if(!recvline(fd, line, sizeof line) || sscanf(line, "%d %d %d", &code, &req, &result) != 3 || code != CONTROL || req != REQ_DISCONNECT || result) {
fprintf(stderr, "Could not disconnect %s.\n", argv[1]);
return 1;
}
return 0;
}
static int cmd_top(int argc, char *argv[]) {
#ifdef HAVE_CURSES #ifdef HAVE_CURSES
if(!strcasecmp(argv[optind], "top")) { if(!connect_tincd())
top(fd); return 1;
return 0;
} top(fd);
return 0;
#else
fprintf(stderr, "This version of tincctl was compiled without support for the curses library.\n");
return 1;
#endif #endif
}
if(!strcasecmp(argv[optind], "pcap")) { static int cmd_pcap(int argc, char *argv[]) {
pcap(fd, stdout, optind < argc - 1 ? atoi(argv[optind + 1]) : 0); if(!connect_tincd())
return 1;
pcap(fd, stdout, argc > 1 ? atoi(argv[1]) : 0);
return 0;
}
static int cmd_log(int argc, char *argv[]) {
if(!connect_tincd())
return 1;
logcontrol(fd, stdout, argc > 1 ? atoi(argv[1]) : -1);
return 0;
}
static int cmd_pid(int argc, char *argv[]) {
if(!connect_tincd())
return 1;
printf("%d\n", pid);
return 0;
}
static int cmd_init(int argc, char *argv[]) {
char *tinc_conf = NULL;
xasprintf(&tinc_conf, "%s/tinc.conf", confbase);
if(!access(confbase, F_OK)) {
fprintf(stderr, "Configuration file %s already exists!\n", tinc_conf);
return 1;
}
if(optind >= argc - 1) {
if(isatty(0) && isatty(1)) {
char buf[1024];
fprintf(stdout, "Enter the Name you want your tinc node to have: ");
fflush(stdout);
if(!fgets(buf, sizeof buf, stdin)) {
fprintf(stderr, "Error while reading stdin: %s\n", strerror(errno));
return 1;
}
int len = strlen(buf);
if(len)
buf[--len] = 0;
if(!len) {
fprintf(stderr, "No name given!\n");
return 1;
}
name = strdup(buf);
} else {
fprintf(stderr, "No Name given!\n");
return 1;
}
} else {
name = strdup(argv[optind + 1]);
if(!*name) {
fprintf(stderr, "No Name given!\n");
return 1;
}
}
for(int i = 0; i < strlen(name); i++) {
if(!isalnum(name[i]) && name[i] != '_') {
fprintf(stderr, "Invalid Name! Only a-z, A-Z, 0-9 and _ are allowed characters.\n");
return 1;
}
}
if(mkdir(CONFDIR, 0755) && errno != EEXIST) {
fprintf(stderr, "Could not create directory %s: %s\n", CONFDIR, strerror(errno));
return 1;
}
if(mkdir(confbase, 0755) && errno != EEXIST) {
fprintf(stderr, "Could not create directory %s: %s\n", confbase, strerror(errno));
return 1;
}
char *hosts_dir = NULL;
xasprintf(&hosts_dir, "%s/hosts", confbase);
if(mkdir(hosts_dir, 0755) && errno != EEXIST) {
fprintf(stderr, "Could not create directory %s: %s\n", hosts_dir, strerror(errno));
return 1;
}
FILE *f = fopen(tinc_conf, "w");
if(!f) {
fprintf(stderr, "Could not create file %s: %s\n", tinc_conf, strerror(errno));
return 1;
}
fprintf(f, "Name = %s\n", name);
fclose(f);
fclose(stdin);
if(!rsa_keygen(2048) || !ecdsa_keygen())
return false;
return true;
}
static int cmd_generate_keys(int argc, char *argv[]) {
return !(rsa_keygen(argc > 1 ? atoi(argv[1]) : 2048) && ecdsa_keygen());
}
static int cmd_generate_rsa_keys(int argc, char *argv[]) {
return !rsa_keygen(argc > 1 ? atoi(argv[1]) : 2048);
}
static int cmd_generate_ecdsa_keys(int argc, char *argv[]) {
return !ecdsa_keygen();
}
static int cmd_help(int argc, char *argv[]) {
usage(false);
return 0;
}
static int cmd_version(int argc, char *argv[]) {
version();
return 0;
}
static const struct {
const char *command;
int (*function)(int argc, char *argv[]);
} commands[] = {
{"start", cmd_start},
{"stop", cmd_stop},
{"restart", cmd_restart},
{"reload", cmd_reload},
{"dump", cmd_dump},
{"purge", cmd_purge},
{"debug", cmd_debug},
{"retry", cmd_retry},
{"connect", cmd_connect},
{"disconnect", cmd_disconnect},
{"top", cmd_top},
{"pcap", cmd_pcap},
{"log", cmd_log},
{"pid", cmd_pid},
{"init", cmd_init},
{"generate-keys", cmd_generate_keys},
{"generate-rsa-keys", cmd_generate_rsa_keys},
{"generate-ecdsa-keys", cmd_generate_ecdsa_keys},
{"help", cmd_help},
{"version", cmd_version},
{NULL, NULL},
};
int main(int argc, char *argv[]) {
program_name = argv[0];
if(!parse_options(argc, argv))
return 1;
make_names();
if(show_version) {
version();
return 0; return 0;
} }
if(!strcasecmp(argv[optind], "log")) { if(show_help) {
logcontrol(fd, stdout, optind < argc - 1 ? atoi(argv[optind + 1]) : -1); usage(false);
return 0; return 0;
} }
if(optind >= argc) {
fprintf(stderr, "No command given.\n");
usage(true);
return 1;
}
for(int i = 0; commands[i].command; i++) {
if(!strcasecmp(argv[optind], commands[i].command))
return commands[i].function(argc - optind, argv + optind);
}
fprintf(stderr, "Unknown command `%s'.\n", argv[optind]); fprintf(stderr, "Unknown command `%s'.\n", argv[optind]);
usage(true); usage(true);
return 1;
close(fd);
return 0;
} }