Compare commits
1 Commits
v0.0.40-mu
...
v0.0.41-mu
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
2f290dbf85 |
2
mud
2
mud
Submodule mud updated: f0ad4bbdd3...7e5c487951
49
src/main.c
49
src/main.c
@@ -20,8 +20,6 @@
|
||||
#include <arpa/inet.h>
|
||||
#include <netdb.h>
|
||||
|
||||
#include <sodium.h>
|
||||
|
||||
#include "mud.h"
|
||||
|
||||
#ifndef O_CLOEXEC
|
||||
@@ -29,10 +27,10 @@
|
||||
#endif
|
||||
|
||||
static struct {
|
||||
int timeout;
|
||||
volatile sig_atomic_t quit;
|
||||
volatile sig_atomic_t info;
|
||||
uint8_t key[crypto_generichash_KEYBYTES];
|
||||
int timeout;
|
||||
int state_fd;
|
||||
} gt;
|
||||
|
||||
static void fd_set_nonblock (int fd)
|
||||
@@ -185,16 +183,16 @@ static size_t fd_write_all (int fd, const void *data, size_t size)
|
||||
return done;
|
||||
}
|
||||
|
||||
static int gt_setup_secretkey (char *keyfile)
|
||||
static int gt_setup_secretkey (struct mud *mud, char *keyfile)
|
||||
{
|
||||
const size_t size = sizeof(gt.key);
|
||||
unsigned char key[32];
|
||||
|
||||
if (str_empty(keyfile)) {
|
||||
char buf[2*size+1];
|
||||
char buf[2*sizeof(key)+1];
|
||||
|
||||
randombytes_buf(gt.key, size);
|
||||
gt_tohex(buf, sizeof(buf), gt.key, size);
|
||||
state("SECRETKEY", buf);
|
||||
mud_get_key(mud, key, sizeof(key));
|
||||
gt_tohex(buf, sizeof(buf), key, sizeof(key));
|
||||
state_send(gt.state_fd, "SECRETKEY", buf);
|
||||
|
||||
return 0;
|
||||
}
|
||||
@@ -210,21 +208,23 @@ static int gt_setup_secretkey (char *keyfile)
|
||||
return -1;
|
||||
}
|
||||
|
||||
char key[2*size];
|
||||
size_t r = fd_read_all(fd, key, sizeof(key));
|
||||
char buf[2*sizeof(key)];
|
||||
size_t r = fd_read_all(fd, buf, sizeof(buf));
|
||||
|
||||
close(fd);
|
||||
|
||||
if (r!=sizeof(key)) {
|
||||
if (r!=sizeof(buf)) {
|
||||
gt_log("unable to read secret key\n");
|
||||
return -1;
|
||||
}
|
||||
|
||||
if (gt_fromhex(gt.key, size, key, sizeof(key))) {
|
||||
if (gt_fromhex(key, sizeof(key), buf, sizeof(buf))) {
|
||||
gt_log("secret key is not valid\n");
|
||||
return -1;
|
||||
}
|
||||
|
||||
mud_set_key(mud, key, sizeof(key));
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
@@ -283,7 +283,7 @@ int main (int argc, char **argv)
|
||||
return 1;
|
||||
}
|
||||
|
||||
if (!option_is_set(opts, "keyfile")) {
|
||||
if (host && !option_is_set(opts, "keyfile")) {
|
||||
gt_log("keyfile option must be set\n");
|
||||
return 1;
|
||||
}
|
||||
@@ -293,12 +293,9 @@ int main (int argc, char **argv)
|
||||
return 1;
|
||||
}
|
||||
|
||||
if (sodium_init()==-1) {
|
||||
gt_log("libsodium initialization has failed\n");
|
||||
return 1;
|
||||
}
|
||||
gt.state_fd = state_create(statefile);
|
||||
|
||||
if (state_init(statefile))
|
||||
if (statefile && gt.state_fd==-1)
|
||||
return 1;
|
||||
|
||||
char *tun_name = NULL;
|
||||
@@ -312,9 +309,6 @@ int main (int argc, char **argv)
|
||||
|
||||
fd_set_nonblock(tun_fd);
|
||||
|
||||
if (gt_setup_secretkey(keyfile))
|
||||
return 1;
|
||||
|
||||
struct mud *mud = mud_create(bind_port, v4, v6);
|
||||
|
||||
if (!mud) {
|
||||
@@ -322,7 +316,8 @@ int main (int argc, char **argv)
|
||||
return 1;
|
||||
}
|
||||
|
||||
mud_set_key(mud, gt.key, sizeof(gt.key));
|
||||
if (gt_setup_secretkey(mud, keyfile))
|
||||
return 1;
|
||||
|
||||
mud_set_send_timeout_msec(mud, gt.timeout);
|
||||
|
||||
@@ -356,7 +351,7 @@ int main (int argc, char **argv)
|
||||
|
||||
int mud_fd = mud_get_fd(mud);
|
||||
|
||||
state("INITIALIZED", tun_name);
|
||||
state_send(gt.state_fd, "INITIALIZED", tun_name);
|
||||
|
||||
fd_set rfds;
|
||||
FD_ZERO(&rfds);
|
||||
@@ -389,12 +384,12 @@ int main (int argc, char **argv)
|
||||
|
||||
if (mud_is_up(mud)) {
|
||||
if (!started) {
|
||||
state("STARTED", tun_name);
|
||||
state_send(gt.state_fd, "STARTED", tun_name);
|
||||
started = 1;
|
||||
}
|
||||
} else {
|
||||
if (started) {
|
||||
state("STOPPED", tun_name);
|
||||
state_send(gt.state_fd, "STOPPED", tun_name);
|
||||
started = 0;
|
||||
}
|
||||
}
|
||||
|
||||
41
src/state.c
41
src/state.c
@@ -7,16 +7,14 @@
|
||||
#include <fcntl.h>
|
||||
#include <sys/stat.h>
|
||||
|
||||
static int state_fd = -1;
|
||||
|
||||
int state_init (const char *filename)
|
||||
int state_create (const char *filename)
|
||||
{
|
||||
if (str_empty(filename))
|
||||
return 0;
|
||||
return -1;
|
||||
|
||||
state_fd = open(filename, O_WRONLY);
|
||||
int fd = open(filename, O_WRONLY);
|
||||
|
||||
if (state_fd==-1) {
|
||||
if (fd==-1) {
|
||||
if (errno!=EINTR)
|
||||
perror("open");
|
||||
return -1;
|
||||
@@ -24,38 +22,41 @@ int state_init (const char *filename)
|
||||
|
||||
struct stat st = {0};
|
||||
|
||||
if (fstat(state_fd, &st)==-1) {
|
||||
if (fstat(fd, &st)==-1) {
|
||||
perror("fstat");
|
||||
close(state_fd);
|
||||
state_fd = -1;
|
||||
close(fd);
|
||||
return -1;
|
||||
}
|
||||
|
||||
if (!S_ISFIFO(st.st_mode)) {
|
||||
gt_log("`%s' is not a fifo\n", filename);
|
||||
close(state_fd);
|
||||
state_fd = -1;
|
||||
close(fd);
|
||||
return -1;
|
||||
}
|
||||
|
||||
return 0;
|
||||
return fd;
|
||||
}
|
||||
|
||||
void state (const char *state, const char *info)
|
||||
void state_send (int fd, const char *state, const char *info)
|
||||
{
|
||||
if (str_empty(state))
|
||||
return;
|
||||
|
||||
if (fd==-1) {
|
||||
gt_print("%s %s\n", state, info);
|
||||
return;
|
||||
}
|
||||
|
||||
const char *strs[] = { state, " ", info, "\n" };
|
||||
char *str = str_cat(strs, COUNT(strs));
|
||||
|
||||
if (!str)
|
||||
if (!str) {
|
||||
perror("str_cat");
|
||||
return;
|
||||
|
||||
if (state_fd==-1) {
|
||||
gt_print("%s", str);
|
||||
} else {
|
||||
if (write(state_fd, str, str_len(str))==-1 && errno!=EINTR)
|
||||
perror("write");
|
||||
}
|
||||
|
||||
if (write(fd, str, str_len(str))==-1 && errno!=EINTR)
|
||||
perror("write");
|
||||
|
||||
free(str);
|
||||
}
|
||||
|
||||
@@ -1,4 +1,4 @@
|
||||
#pragma once
|
||||
|
||||
int state_init (const char *);
|
||||
void state (const char *, const char *);
|
||||
int state_create (const char *);
|
||||
void state_send (int, const char *, const char *);
|
||||
|
||||
Reference in New Issue
Block a user