Compare commits
91 Commits
v0.0.88-mu
...
v0.0.95-mu
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
cb8db71e72 | ||
|
|
4fab60ea87 | ||
|
|
b4ec962a3b | ||
|
|
235250e49d | ||
|
|
016045b222 | ||
|
|
8403b0606d | ||
|
|
c685ecf124 | ||
|
|
f95aaef092 | ||
|
|
ff5b966866 | ||
|
|
44f75458d1 | ||
|
|
44dd8124b7 | ||
|
|
da44a9f55b | ||
|
|
4fa56178cf | ||
|
|
21df85cd94 | ||
|
|
e21c83602f | ||
|
|
7eecb435b9 | ||
|
|
1db79f77d7 | ||
|
|
3bfe9d6cd1 | ||
|
|
99b51efbb3 | ||
|
|
dd3ba523a4 | ||
|
|
d34a9a6ce8 | ||
|
|
bf83c83720 | ||
|
|
dcc460204d | ||
|
|
651b5c577c | ||
|
|
7ba60a5b61 | ||
|
|
40c6e3bc40 | ||
|
|
155fb32d61 | ||
|
|
b6ef00a0b3 | ||
|
|
fe96617f78 | ||
|
|
3c2423a38b | ||
|
|
7548ba2c27 | ||
|
|
22634d14fe | ||
|
|
e8262257fb | ||
|
|
e856c3c7d0 | ||
|
|
5b929a90b9 | ||
|
|
29297ff210 | ||
|
|
4b41b9baf5 | ||
|
|
04c93b6fe6 | ||
|
|
567e5d0193 | ||
|
|
0169a09469 | ||
|
|
3ce6c15a81 | ||
|
|
9ea48eaf66 | ||
|
|
f843b23365 | ||
|
|
dc9e19961a | ||
|
|
4047abee33 | ||
|
|
36e9a5d57d | ||
|
|
0c3c2ca28b | ||
|
|
ecdaeaf368 | ||
|
|
6ce92e43ac | ||
|
|
36ba7ddb7b | ||
|
|
975bb1610d | ||
|
|
660a6ace2b | ||
|
|
aeedc361c0 | ||
|
|
70086d295a | ||
|
|
69bcf664a8 | ||
|
|
6754ec0385 | ||
|
|
3b787a2043 | ||
|
|
8f35528fde | ||
|
|
f65ecac5fe | ||
|
|
086fa412ed | ||
|
|
8476332224 | ||
|
|
9dacd85713 | ||
|
|
ff83707581 | ||
|
|
a02839712e | ||
|
|
495138ffe4 | ||
|
|
e26ecf37a3 | ||
|
|
827876647f | ||
|
|
d1940692b2 | ||
|
|
d3307a22f8 | ||
|
|
93cefd6dba | ||
|
|
21718c8c14 | ||
|
|
aa54a72bbc | ||
|
|
32e6e7575a | ||
|
|
19eea3e96d | ||
|
|
75b2903ac2 | ||
|
|
0f5a6f5d98 | ||
|
|
ed90fdea02 | ||
|
|
520bd33cb3 | ||
|
|
361c695c5c | ||
|
|
bc5d622169 | ||
|
|
7859746f66 | ||
|
|
c3b8c44c68 | ||
|
|
297e93ed54 | ||
|
|
325575c6a7 | ||
|
|
3b1c9db3f8 | ||
|
|
557d3f7869 | ||
|
|
195908d379 | ||
|
|
eb5c6853c1 | ||
|
|
3c2481dc33 | ||
|
|
355040f576 | ||
|
|
bbf1c12f7a |
18
.build.sh
18
.build.sh
@@ -1,18 +0,0 @@
|
||||
#!/bin/sh
|
||||
|
||||
export CC="gcc -static"
|
||||
|
||||
git clone https://github.com/jedisct1/libsodium --depth=1 --branch stable
|
||||
cd libsodium || exit 1
|
||||
./autogen.sh && ./configure --enable-minimal --disable-shared --prefix=/usr && make install
|
||||
cd ..
|
||||
|
||||
./autogen.sh && ./configure && make
|
||||
[ -x glorytun ] || exit 1
|
||||
|
||||
mkdir -p deploy
|
||||
|
||||
cp glorytun deploy/glorytun-$(cat VERSION)-$(uname -m).debug.bin
|
||||
|
||||
strip -s glorytun
|
||||
cp glorytun deploy/glorytun-$(cat VERSION)-$(uname -m).bin
|
||||
2
.gitignore
vendored
2
.gitignore
vendored
@@ -11,3 +11,5 @@ build-aux
|
||||
.deps
|
||||
.dirstamp
|
||||
glorytun
|
||||
build*
|
||||
VERSION
|
||||
|
||||
3
.gitmodules
vendored
3
.gitmodules
vendored
@@ -1,3 +1,6 @@
|
||||
[submodule "mud"]
|
||||
path = mud
|
||||
url = https://github.com/angt/mud.git
|
||||
[submodule "argz"]
|
||||
path = argz
|
||||
url = https://github.com/angt/argz.git
|
||||
|
||||
38
Makefile.am
38
Makefile.am
@@ -5,25 +5,35 @@ bin_PROGRAMS = glorytun
|
||||
glorytun_CFLAGS = $(libsodium_CFLAGS)
|
||||
glorytun_LDADD = $(libsodium_LIBS)
|
||||
glorytun_SOURCES = \
|
||||
src/common.h \
|
||||
src/common.c \
|
||||
src/ip.h \
|
||||
src/str.h \
|
||||
src/main.c \
|
||||
src/option.c \
|
||||
src/option.h \
|
||||
src/tun.c \
|
||||
src/tun.h \
|
||||
src/db.c \
|
||||
src/db.h
|
||||
|
||||
glorytun_SOURCES += \
|
||||
argz/argz.c \
|
||||
argz/argz.h \
|
||||
mud/mud.c \
|
||||
mud/mud.h \
|
||||
mud/mud.c
|
||||
src/bench.c \
|
||||
src/bind.c \
|
||||
src/common.c \
|
||||
src/common.h \
|
||||
src/ctl.c \
|
||||
src/ctl.h \
|
||||
src/iface.c \
|
||||
src/iface.h \
|
||||
src/ip.h \
|
||||
src/keygen.c \
|
||||
src/main.c \
|
||||
src/path.c \
|
||||
src/set.c \
|
||||
src/show.c \
|
||||
src/str.h \
|
||||
src/tun.c \
|
||||
src/tun.h
|
||||
|
||||
EXTRA_DIST = \
|
||||
LICENSE \
|
||||
README.md \
|
||||
VERSION \
|
||||
autogen.sh \
|
||||
meson.build \
|
||||
mud/LICENSE \
|
||||
mud/README.md \
|
||||
systemd \
|
||||
version.sh
|
||||
|
||||
53
README.md
53
README.md
@@ -1,18 +1,55 @@
|
||||
# π₁(Glorytun)=0
|
||||
# Glorytun
|
||||
|
||||
Small, Simple and Stupid VPN over [mud](https://github.com/angt/mud).
|
||||
|
||||
#### Build and Install
|
||||
### Build and Install
|
||||
|
||||
Glorytun depends on [libsodium](https://github.com/jedisct1/libsodium) version >= 1.0.4.
|
||||
|
||||
To build and install the latest version:
|
||||
On Ubuntu, the following command should be sufficient:
|
||||
|
||||
$ git clone https://github.com/angt/glorytun --recursive --branch mud
|
||||
$ sudo apt-get install meson libsodium-dev pkg-config
|
||||
|
||||
Grab the latest release from github:
|
||||
|
||||
$ git clone https://github.com/angt/glorytun --recursive
|
||||
$ cd glorytun
|
||||
$ ./autogen.sh
|
||||
$ ./configure
|
||||
$ make
|
||||
# make install
|
||||
|
||||
To build and install the latest version with [meson](http://mesonbuild.com):
|
||||
|
||||
$ meson build
|
||||
$ sudo ninja -C build install
|
||||
|
||||
The more classical autotools suite is also available.
|
||||
|
||||
### Easy setup with systemd
|
||||
|
||||
Just call `glorytun-setup` and follow the instructions.
|
||||
|
||||
First, setup the server:
|
||||
|
||||
$ sudo glorytun-setup
|
||||
Config filename (tun0):
|
||||
Server ip (enter for server conf):
|
||||
Bind to port (5000):
|
||||
Server key (enter to generate a new one):
|
||||
Your new key: NEW_KEY
|
||||
Start glorytun now ? (enter to skip): y
|
||||
|
||||
Copy the new generated key and use it when configuring the client:
|
||||
|
||||
$ sudo glorytun-setup
|
||||
Config filename (tun0):
|
||||
Server ip (enter for server conf): SERVER_IP
|
||||
Server port (5000):
|
||||
Server key (enter to generate a new one): NEW_KEY
|
||||
Start glorytun now ? (enter to skip): y
|
||||
|
||||
You can check easily if it works by looking at your public ip.
|
||||
To stop the service:
|
||||
|
||||
$ sudo systemctl stop glorytun@tun0
|
||||
|
||||
---
|
||||
|
||||
For feature requests and bug reports, please create an [issue](https://github.com/angt/glorytun/issues).
|
||||
|
||||
1
argz
Submodule
1
argz
Submodule
Submodule argz added at 2d6a6d635f
@@ -8,13 +8,14 @@ AC_DEFINE_UNQUOTED([VERSION_MAJOR], [m4_esyscmd([./version.sh major])])
|
||||
AC_CONFIG_SRCDIR([src/common.h])
|
||||
AC_CONFIG_AUX_DIR([build-aux])
|
||||
AC_CONFIG_MACRO_DIR([m4])
|
||||
AM_INIT_AUTOMAKE([1.9 -Wall -Werror foreign tar-ustar subdir-objects])
|
||||
AM_INIT_AUTOMAKE([1.12 -Wall -Werror foreign tar-ustar subdir-objects])
|
||||
AM_DEP_TRACK
|
||||
AM_SILENT_RULES([yes])
|
||||
AM_PROG_CC_C_O
|
||||
AC_PROG_CC_C99
|
||||
AC_USE_SYSTEM_EXTENSIONS
|
||||
AC_SEARCH_LIBS([socket], [socket])
|
||||
AC_SEARCH_LIBS([fmin], [m])
|
||||
AC_CHECK_LIB([rt], [clock_gettime])
|
||||
AC_CHECK_FUNCS([clock_gettime])
|
||||
PKG_CHECK_MODULES([libsodium], [libsodium >= 1.0.4])
|
||||
|
||||
16
m4/pkg.m4
16
m4/pkg.m4
@@ -1,6 +1,6 @@
|
||||
dnl pkg.m4 - Macros to locate and utilise pkg-config. -*- Autoconf -*-
|
||||
dnl serial 11 (pkg-config-0.29)
|
||||
dnl
|
||||
# pkg.m4 - Macros to locate and utilise pkg-config. -*- Autoconf -*-
|
||||
# serial 12 (pkg-config-0.29.2)
|
||||
|
||||
dnl Copyright © 2004 Scott James Remnant <scott@netsplit.com>.
|
||||
dnl Copyright © 2012-2015 Dan Nicholson <dbn.lists@gmail.com>
|
||||
dnl
|
||||
@@ -41,7 +41,7 @@ dnl
|
||||
dnl See the "Since" comment for each macro you use to see what version
|
||||
dnl of the macros you require.
|
||||
m4_defun([PKG_PREREQ],
|
||||
[m4_define([PKG_MACROS_VERSION], [0.29])
|
||||
[m4_define([PKG_MACROS_VERSION], [0.29.2])
|
||||
m4_if(m4_version_compare(PKG_MACROS_VERSION, [$1]), -1,
|
||||
[m4_fatal([pkg.m4 version $1 or higher is required but ]PKG_MACROS_VERSION[ found])])
|
||||
])dnl PKG_PREREQ
|
||||
@@ -142,7 +142,7 @@ AC_ARG_VAR([$1][_CFLAGS], [C compiler flags for $1, overriding pkg-config])dnl
|
||||
AC_ARG_VAR([$1][_LIBS], [linker flags for $1, overriding pkg-config])dnl
|
||||
|
||||
pkg_failed=no
|
||||
AC_MSG_CHECKING([for $1])
|
||||
AC_MSG_CHECKING([for $2])
|
||||
|
||||
_PKG_CONFIG([$1][_CFLAGS], [cflags], [$2])
|
||||
_PKG_CONFIG([$1][_LIBS], [libs], [$2])
|
||||
@@ -152,11 +152,11 @@ and $1[]_LIBS to avoid the need to call pkg-config.
|
||||
See the pkg-config man page for more details.])
|
||||
|
||||
if test $pkg_failed = yes; then
|
||||
AC_MSG_RESULT([no])
|
||||
AC_MSG_RESULT([no])
|
||||
_PKG_SHORT_ERRORS_SUPPORTED
|
||||
if test $_pkg_short_errors_supported = yes; then
|
||||
$1[]_PKG_ERRORS=`$PKG_CONFIG --short-errors --print-errors --cflags --libs "$2" 2>&1`
|
||||
else
|
||||
else
|
||||
$1[]_PKG_ERRORS=`$PKG_CONFIG --print-errors --cflags --libs "$2" 2>&1`
|
||||
fi
|
||||
# Put the nasty error message in config.log where it belongs
|
||||
@@ -173,7 +173,7 @@ installed software in a non-standard prefix.
|
||||
_PKG_TEXT])[]dnl
|
||||
])
|
||||
elif test $pkg_failed = untried; then
|
||||
AC_MSG_RESULT([no])
|
||||
AC_MSG_RESULT([no])
|
||||
m4_default([$4], [AC_MSG_FAILURE(
|
||||
[The pkg-config script could not be found or is too old. Make sure it
|
||||
is in your PATH or set the PKG_CONFIG environment variable to the full
|
||||
|
||||
62
meson.build
Normal file
62
meson.build
Normal file
@@ -0,0 +1,62 @@
|
||||
project('glorytun', 'c',
|
||||
version: run_command('./version.sh').stdout(),
|
||||
license: 'BSD-3-Clause',
|
||||
default_options : [
|
||||
'buildtype=debugoptimized',
|
||||
'c_std=gnu99'
|
||||
]
|
||||
)
|
||||
|
||||
cc = meson.get_compiler('c')
|
||||
|
||||
prefix = get_option('prefix')
|
||||
bindir = join_paths(prefix, get_option('bindir'))
|
||||
|
||||
conf_data = configuration_data()
|
||||
conf_data.set('prefix', prefix)
|
||||
conf_data.set('bindir', bindir)
|
||||
|
||||
add_global_arguments('-DPACKAGE_VERSION="'+meson.project_version()+'"', language : 'c')
|
||||
add_global_arguments('-DPACKAGE_NAME="'+meson.project_name()+'"', language : 'c')
|
||||
|
||||
executable('glorytun', install: true,
|
||||
sources: [
|
||||
'argz/argz.c',
|
||||
'mud/mud.c',
|
||||
'src/bench.c',
|
||||
'src/bind.c',
|
||||
'src/common.c',
|
||||
'src/ctl.c',
|
||||
'src/iface.c',
|
||||
'src/keygen.c',
|
||||
'src/main.c',
|
||||
'src/path.c',
|
||||
'src/set.c',
|
||||
'src/show.c',
|
||||
'src/tun.c',
|
||||
],
|
||||
dependencies: [
|
||||
dependency('libsodium', version : '>=1.0.4'),
|
||||
cc.find_library('m', required : false)
|
||||
]
|
||||
)
|
||||
|
||||
systemd = dependency('systemd', required: false)
|
||||
|
||||
if systemd.found()
|
||||
systemdutildir = systemd.get_pkgconfig_variable('systemdutildir')
|
||||
configure_file(
|
||||
input: 'systemd/glorytun@.service.in',
|
||||
output: 'glorytun@.service',
|
||||
configuration: conf_data,
|
||||
install_dir: join_paths(systemdutildir, 'system')
|
||||
)
|
||||
install_data('systemd/glorytun.network',
|
||||
install_dir: join_paths(systemdutildir, 'network'))
|
||||
install_data('systemd/glorytun-client.network',
|
||||
install_dir: join_paths(systemdutildir, 'network'))
|
||||
install_data('systemd/glorytun-run',
|
||||
install_dir: bindir)
|
||||
install_data('systemd/glorytun-setup',
|
||||
install_dir: bindir)
|
||||
endif
|
||||
2
mud
2
mud
Submodule mud updated: f67eae0a7f...b809bf0785
159
src/bench.c
Normal file
159
src/bench.c
Normal file
@@ -0,0 +1,159 @@
|
||||
#include "common.h"
|
||||
|
||||
#include <math.h>
|
||||
#include <sodium.h>
|
||||
#include <string.h>
|
||||
#include <stdio.h>
|
||||
#include <sys/time.h>
|
||||
#include <time.h>
|
||||
#include <unistd.h>
|
||||
|
||||
#if defined __APPLE__
|
||||
#include <mach/mach_time.h>
|
||||
#endif
|
||||
|
||||
#include "../argz/argz.h"
|
||||
|
||||
#define STR_S(X) (((X) > 1) ? "s" : "")
|
||||
|
||||
static unsigned long long
|
||||
gt_now(void)
|
||||
{
|
||||
#if defined __APPLE__
|
||||
static mach_timebase_info_data_t mtid;
|
||||
if (!mtid.denom)
|
||||
mach_timebase_info(&mtid);
|
||||
return (mach_absolute_time() * mtid.numer / mtid.denom) / 1000ULL;
|
||||
#elif defined CLOCK_MONOTONIC
|
||||
struct timespec tv;
|
||||
clock_gettime(CLOCK_MONOTONIC, &tv);
|
||||
return tv.tv_sec * 1000000ULL + tv.tv_nsec / 1000ULL;
|
||||
#else
|
||||
struct timeval tv;
|
||||
gettimeofday(&tv, NULL);
|
||||
return tv.tv_sec * 1000000ULL + tv.tv_usec;
|
||||
#endif
|
||||
}
|
||||
|
||||
int
|
||||
gt_bench(int argc, char **argv)
|
||||
{
|
||||
unsigned long precision = 10;
|
||||
size_t bufsize = 64 * 1024;
|
||||
unsigned long duration = 1000;
|
||||
|
||||
struct argz bench_argz[] = {
|
||||
{"aes|chacha", NULL, NULL, argz_option},
|
||||
{"precision", "EXPONENT", &precision, argz_ulong},
|
||||
{"bufsize", "BYTES", &bufsize, argz_bytes},
|
||||
{"duration", "SECONDS", &duration, argz_time},
|
||||
{NULL}};
|
||||
|
||||
if (argz(bench_argz, argc, argv))
|
||||
return 1;
|
||||
|
||||
if (duration == 0 || bufsize == 0)
|
||||
return 0;
|
||||
|
||||
if (sodium_init() == -1) {
|
||||
gt_log("sodium init failed\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
duration /= 1000;
|
||||
|
||||
int term = isatty(1);
|
||||
int chacha = argz_is_set(bench_argz, "chacha");
|
||||
|
||||
if (!chacha && !crypto_aead_aes256gcm_is_available()) {
|
||||
gt_log("aes is not available on your platform\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
unsigned char *buf = calloc(1, bufsize + crypto_aead_aes256gcm_ABYTES);
|
||||
|
||||
if (!buf) {
|
||||
perror("calloc");
|
||||
return 1;
|
||||
}
|
||||
|
||||
unsigned char npub[crypto_aead_aes256gcm_NPUBBYTES];
|
||||
unsigned char key[crypto_aead_aes256gcm_KEYBYTES];
|
||||
|
||||
randombytes_buf(npub, sizeof(npub));
|
||||
randombytes_buf(key, sizeof(key));
|
||||
|
||||
if (term) {
|
||||
printf("\n");
|
||||
printf(" %-10s %s\n", "bench", chacha ? "chacha20poly1305" : "aes256gcm");
|
||||
printf(" %-10s %s\n", "libsodium", sodium_version_string());
|
||||
printf("\n");
|
||||
printf(" %-10s 2^(-%lu)\n", "precision", precision);
|
||||
printf(" %-10s %zu byte%s\n", "bufsize", bufsize, STR_S(bufsize));
|
||||
printf(" %-10s %lu second%s\n", "duration", duration, STR_S(duration));
|
||||
printf("\n");
|
||||
printf("------------------------------------------------------------\n");
|
||||
printf(" %3s %9s %14s %14s %14s\n", "2^n", "min", "avg", "max", "delta");
|
||||
printf("------------------------------------------------------------\n");
|
||||
}
|
||||
|
||||
for (int i = 0; !gt_quit && bufsize >> i; i++) {
|
||||
unsigned long long total_dt = 0ULL;
|
||||
size_t total_bytes = 0;
|
||||
double mbps = 0.0;
|
||||
double mbps_min = INFINITY;
|
||||
double mbps_max = 0.0;
|
||||
double mbps_dlt = INFINITY;
|
||||
|
||||
while (!gt_quit && mbps_dlt > ldexp(mbps, -precision)) {
|
||||
crypto_aead_aes256gcm_state ctx;
|
||||
|
||||
if (!chacha)
|
||||
crypto_aead_aes256gcm_beforenm(&ctx, key);
|
||||
|
||||
unsigned long long now = gt_now();
|
||||
double mbps_old = mbps;
|
||||
size_t bytes = 0;
|
||||
|
||||
gt_alarm = 0;
|
||||
alarm(duration);
|
||||
|
||||
while (!gt_quit && !gt_alarm) {
|
||||
if (chacha) {
|
||||
crypto_aead_chacha20poly1305_encrypt(
|
||||
buf, NULL, buf, 1ULL << i, NULL, 0, NULL, npub, key);
|
||||
} else {
|
||||
crypto_aead_aes256gcm_encrypt_afternm(
|
||||
buf, NULL, buf, 1ULL << i, NULL, 0, NULL, npub,
|
||||
(const crypto_aead_aes256gcm_state *)&ctx);
|
||||
}
|
||||
bytes += 1ULL << i;
|
||||
}
|
||||
|
||||
total_dt += gt_now() - now;
|
||||
total_bytes += bytes;
|
||||
|
||||
mbps = (total_bytes * 8.0) / total_dt;
|
||||
mbps_min = fmin(mbps_min, mbps);
|
||||
mbps_max = fmax(mbps_max, mbps);
|
||||
mbps_dlt = fabs(mbps_old - mbps);
|
||||
|
||||
if (term) {
|
||||
printf("\r %3i %9.2f Mbps %9.2f Mbps %9.2f Mbps %9.2e",
|
||||
i, mbps_min, mbps, mbps_max, mbps_dlt);
|
||||
fflush(stdout);
|
||||
}
|
||||
}
|
||||
|
||||
if (term) {
|
||||
printf("\n");
|
||||
} else {
|
||||
printf("%i %.2f %.2f %.2f\n", i, mbps_min, mbps, mbps_max);
|
||||
}
|
||||
}
|
||||
|
||||
printf("\n");
|
||||
free(buf);
|
||||
|
||||
return 0;
|
||||
}
|
||||
396
src/bind.c
Normal file
396
src/bind.c
Normal file
@@ -0,0 +1,396 @@
|
||||
#include "common.h"
|
||||
#include "ctl.h"
|
||||
#include "iface.h"
|
||||
#include "ip.h"
|
||||
#include "str.h"
|
||||
#include "tun.h"
|
||||
|
||||
#include <fcntl.h>
|
||||
#include <stdio.h>
|
||||
#include <sys/select.h>
|
||||
|
||||
#include "../argz/argz.h"
|
||||
#include "../mud/mud.h"
|
||||
|
||||
#ifndef O_CLOEXEC
|
||||
#define O_CLOEXEC 0
|
||||
#endif
|
||||
|
||||
static void
|
||||
fd_set_nonblock(int fd)
|
||||
{
|
||||
int ret;
|
||||
|
||||
if (fd == -1)
|
||||
return;
|
||||
|
||||
do {
|
||||
ret = fcntl(fd, F_GETFL, 0);
|
||||
} while (ret == -1 && errno == EINTR);
|
||||
|
||||
int flags = (ret == -1) ? 0 : ret;
|
||||
|
||||
do {
|
||||
ret = fcntl(fd, F_SETFL, flags | O_NONBLOCK);
|
||||
} while (ret == -1 && errno == EINTR);
|
||||
|
||||
if (ret == -1)
|
||||
perror("fcntl O_NONBLOCK");
|
||||
}
|
||||
|
||||
static int
|
||||
gt_setup_secretkey(struct mud *mud, const char *keyfile)
|
||||
{
|
||||
int fd;
|
||||
|
||||
do {
|
||||
fd = open(keyfile, O_RDONLY | O_CLOEXEC);
|
||||
} while (fd == -1 && errno == EINTR);
|
||||
|
||||
if (fd == -1) {
|
||||
perror("open keyfile");
|
||||
return -1;
|
||||
}
|
||||
|
||||
unsigned char key[32];
|
||||
char buf[2 * sizeof(key)];
|
||||
size_t size = 0;
|
||||
|
||||
while (size < sizeof(buf)) {
|
||||
ssize_t r = read(fd, &buf[size], sizeof(buf) - size);
|
||||
|
||||
if (r <= (ssize_t)0) {
|
||||
if (r && (errno == EAGAIN || errno == EINTR))
|
||||
continue;
|
||||
break;
|
||||
}
|
||||
|
||||
size += r;
|
||||
}
|
||||
|
||||
close(fd);
|
||||
|
||||
if (size != sizeof(buf)) {
|
||||
gt_log("unable to read secret key\n");
|
||||
return -1;
|
||||
}
|
||||
|
||||
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;
|
||||
}
|
||||
|
||||
static size_t
|
||||
gt_setup_mtu(struct mud *mud, const char *tun_name)
|
||||
{
|
||||
static size_t oldmtu = 0;
|
||||
size_t mtu = mud_get_mtu(mud);
|
||||
|
||||
if (mtu == oldmtu)
|
||||
return mtu;
|
||||
|
||||
gt_log("setup MTU to %zu on interface %s\n", mtu, tun_name);
|
||||
|
||||
if (iface_set_mtu(tun_name, mtu) == -1)
|
||||
perror("tun_set_mtu");
|
||||
|
||||
oldmtu = mtu;
|
||||
|
||||
return mtu;
|
||||
}
|
||||
|
||||
int
|
||||
gt_bind(int argc, char **argv)
|
||||
{
|
||||
struct sockaddr_storage bind_addr = { .ss_family = AF_INET };
|
||||
struct sockaddr_storage peer_addr = { 0 };
|
||||
unsigned short bind_port = 5000;
|
||||
unsigned short peer_port = bind_port;
|
||||
const char *dev = NULL;
|
||||
const char *keyfile = NULL;
|
||||
size_t bufsize = 64 * 1024 * 1024;
|
||||
size_t mtu = 1330;
|
||||
|
||||
struct argz toz[] = {
|
||||
{NULL, "IPADDR", &peer_addr, argz_addr},
|
||||
{NULL, "PORT", &peer_port, argz_ushort},
|
||||
{NULL}};
|
||||
|
||||
struct argz bindz[] = {
|
||||
{NULL, "IPADDR", &bind_addr, argz_addr},
|
||||
{NULL, "PORT", &bind_port, argz_ushort},
|
||||
{"to", NULL, &toz, argz_option},
|
||||
{"dev", "NAME", &dev, argz_str},
|
||||
{"mtu", "BYTES", &mtu, argz_bytes},
|
||||
{"keyfile", "FILE", &keyfile, argz_str},
|
||||
{"chacha", NULL, NULL, argz_option},
|
||||
{"persist", NULL, NULL, argz_option},
|
||||
{"bufsize", "BYTES", &bufsize, argz_bytes},
|
||||
{NULL}};
|
||||
|
||||
if (argz(bindz, argc, argv))
|
||||
return 1;
|
||||
|
||||
gt_set_port((struct sockaddr *)&bind_addr, bind_port);
|
||||
gt_set_port((struct sockaddr *)&peer_addr, peer_port);
|
||||
|
||||
unsigned char *buf = malloc(bufsize);
|
||||
|
||||
if (!buf) {
|
||||
perror("malloc");
|
||||
return 1;
|
||||
}
|
||||
|
||||
int chacha = argz_is_set(bindz, "chacha");
|
||||
int persist = argz_is_set(bindz, "persist");
|
||||
|
||||
struct mud *mud = mud_create((struct sockaddr *)&bind_addr);
|
||||
|
||||
if (!mud) {
|
||||
gt_log("couldn't create mud\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
if (str_empty(keyfile)) {
|
||||
if (mud_set_key(mud, NULL, 0)) {
|
||||
gt_log("couldn't generate a new key\n");
|
||||
return 1;
|
||||
}
|
||||
} else {
|
||||
if (gt_setup_secretkey(mud, keyfile))
|
||||
return 1;
|
||||
}
|
||||
|
||||
if (!chacha && mud_set_aes(mud)) {
|
||||
gt_log("AES is not available\n");
|
||||
chacha = 1;
|
||||
}
|
||||
|
||||
char tun_name[64];
|
||||
int tun_fd = tun_create(tun_name, sizeof(tun_name) - 1, dev);
|
||||
|
||||
if (tun_fd == -1) {
|
||||
gt_log("couldn't create tun device\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
mud_set_mtu(mud, mtu);
|
||||
mtu = gt_setup_mtu(mud, tun_name);
|
||||
|
||||
if (tun_set_persist(tun_fd, persist) == -1)
|
||||
perror("tun_set_persist");
|
||||
|
||||
if (peer_addr.ss_family) {
|
||||
if (mud_peer(mud, (struct sockaddr *)&peer_addr)) {
|
||||
perror("mud_peer");
|
||||
return 1;
|
||||
}
|
||||
}
|
||||
|
||||
int ctl_fd = ctl_create("/run/" PACKAGE_NAME, tun_name);
|
||||
|
||||
if (ctl_fd == -1) {
|
||||
perror("ctl_create");
|
||||
return 1;
|
||||
}
|
||||
|
||||
int mud_fd = mud_get_fd(mud);
|
||||
|
||||
fd_set_nonblock(tun_fd);
|
||||
fd_set_nonblock(mud_fd);
|
||||
fd_set_nonblock(ctl_fd);
|
||||
|
||||
gt_log("running...\n");
|
||||
|
||||
fd_set rfds;
|
||||
FD_ZERO(&rfds);
|
||||
|
||||
int last_fd = 1 + MAX(tun_fd, MAX(mud_fd, ctl_fd));
|
||||
|
||||
while (!gt_quit) {
|
||||
FD_SET(tun_fd, &rfds);
|
||||
FD_SET(mud_fd, &rfds);
|
||||
FD_SET(ctl_fd, &rfds);
|
||||
|
||||
if (select(last_fd, &rfds, NULL, NULL, NULL) == -1) {
|
||||
if (errno != EBADF)
|
||||
continue;
|
||||
perror("select");
|
||||
return 1;
|
||||
}
|
||||
|
||||
mtu = gt_setup_mtu(mud, tun_name);
|
||||
|
||||
if (FD_ISSET(ctl_fd, &rfds)) {
|
||||
struct ctl_msg req, res = {.reply = 1};
|
||||
struct sockaddr_storage ss;
|
||||
socklen_t sl = sizeof(ss);
|
||||
|
||||
ssize_t r = recvfrom(ctl_fd, &req, sizeof(req), 0,
|
||||
(struct sockaddr *)&ss, &sl);
|
||||
|
||||
if (r == (ssize_t)sizeof(req)) {
|
||||
res.type = req.type;
|
||||
|
||||
switch (req.type) {
|
||||
case CTL_NONE:
|
||||
break;
|
||||
case CTL_STATE:
|
||||
if (mud_set_state(mud, (struct sockaddr *)&req.path.addr, req.path.state))
|
||||
res.ret = errno;
|
||||
break;
|
||||
case CTL_PATH_STATUS:
|
||||
{
|
||||
unsigned count = 0;
|
||||
struct mud_path *paths = mud_get_paths(mud, &count);
|
||||
|
||||
if (!paths) {
|
||||
res.ret = errno;
|
||||
break;
|
||||
}
|
||||
|
||||
res.ret = EAGAIN;
|
||||
|
||||
for (unsigned i = 0; i < count; i++) {
|
||||
if (i && sendto(ctl_fd, &res, sizeof(res), 0,
|
||||
(const struct sockaddr *)&ss, sl) == -1)
|
||||
perror("sendto(ctl)");
|
||||
memcpy(&res.path_status, &paths[i], sizeof(struct mud_path));
|
||||
}
|
||||
|
||||
res.ret = 0;
|
||||
}
|
||||
break;
|
||||
case CTL_MTU:
|
||||
mud_set_mtu(mud, (size_t)req.mtu);
|
||||
mtu = gt_setup_mtu(mud, tun_name);
|
||||
res.mtu = mtu;
|
||||
break;
|
||||
case CTL_TC:
|
||||
if (mud_set_tc(mud, req.tc))
|
||||
res.ret = errno;
|
||||
break;
|
||||
case CTL_TIMEOUT:
|
||||
if (mud_set_send_timeout(mud, req.timeout))
|
||||
res.ret = errno;
|
||||
break;
|
||||
case CTL_TIMETOLERANCE:
|
||||
if (mud_set_time_tolerance(mud, req.timetolerance))
|
||||
res.ret = errno;
|
||||
break;
|
||||
case CTL_STATUS:
|
||||
res.status.mtu = mtu;
|
||||
res.status.chacha = chacha;
|
||||
res.status.bind = bind_addr;
|
||||
res.status.peer = peer_addr;
|
||||
break;
|
||||
}
|
||||
if (sendto(ctl_fd, &res, sizeof(res), 0,
|
||||
(const struct sockaddr *)&ss, sl) == -1)
|
||||
perror("sendto(ctl)");
|
||||
} else if (r == -1 && errno != EAGAIN) {
|
||||
perror("recvfrom(ctl)");
|
||||
}
|
||||
}
|
||||
|
||||
if (FD_ISSET(tun_fd, &rfds)) {
|
||||
size_t size = 0;
|
||||
|
||||
while (bufsize - size >= mtu) {
|
||||
const int r = tun_read(tun_fd, &buf[size], bufsize - size);
|
||||
|
||||
if (r <= 0 || r > mtu)
|
||||
break;
|
||||
|
||||
struct ip_common ic;
|
||||
|
||||
if (ip_get_common(&ic, &buf[size], r) || ic.size != r)
|
||||
break;
|
||||
|
||||
size += r;
|
||||
}
|
||||
|
||||
size_t p = 0;
|
||||
|
||||
while (p < size) {
|
||||
size_t q = p;
|
||||
int tc = 0;
|
||||
|
||||
while (q < size) {
|
||||
struct ip_common ic;
|
||||
|
||||
if ((ip_get_common(&ic, &buf[q], size - q)) ||
|
||||
(ic.size > size - q))
|
||||
break;
|
||||
|
||||
if (q + ic.size > p + mtu)
|
||||
break;
|
||||
|
||||
q += ic.size;
|
||||
|
||||
if (tc < (ic.tc & 0xFC))
|
||||
tc = ic.tc & 0xFC;
|
||||
}
|
||||
|
||||
if (p >= q)
|
||||
break;
|
||||
|
||||
int r = mud_send(mud, &buf[p], q - p, tc);
|
||||
|
||||
if (r == -1) {
|
||||
if (errno == EMSGSIZE) {
|
||||
mtu = gt_setup_mtu(mud, tun_name);
|
||||
} else if (errno != EAGAIN) {
|
||||
perror("mud_send");
|
||||
}
|
||||
}
|
||||
|
||||
p = q;
|
||||
}
|
||||
}
|
||||
|
||||
if (FD_ISSET(mud_fd, &rfds)) {
|
||||
size_t size = 0;
|
||||
|
||||
while (bufsize - size >= mtu) {
|
||||
const int r = mud_recv(mud, &buf[size], bufsize - size);
|
||||
|
||||
if (r <= 0) {
|
||||
if (r == -1 && errno != EAGAIN)
|
||||
perror("mud_recv");
|
||||
break;
|
||||
}
|
||||
|
||||
size += r;
|
||||
}
|
||||
|
||||
size_t p = 0;
|
||||
|
||||
while (p < size) {
|
||||
struct ip_common ic;
|
||||
|
||||
if ((ip_get_common(&ic, &buf[p], size - p)) ||
|
||||
(ic.size > size - p))
|
||||
break;
|
||||
|
||||
tun_write(tun_fd, &buf[p], ic.size);
|
||||
|
||||
p += ic.size;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if (gt_reload && tun_fd >= 0) {
|
||||
if (tun_set_persist(tun_fd, 1) == -1)
|
||||
perror("tun_set_persist");
|
||||
}
|
||||
|
||||
ctl_delete(ctl_fd);
|
||||
|
||||
return 0;
|
||||
}
|
||||
72
src/common.c
72
src/common.c
@@ -3,20 +3,6 @@
|
||||
#include <stdarg.h>
|
||||
#include <stdio.h>
|
||||
|
||||
int
|
||||
gt_print(const char *fmt, ...)
|
||||
{
|
||||
va_list ap;
|
||||
va_start(ap, fmt);
|
||||
int ret = vfprintf(stdout, fmt, ap);
|
||||
va_end(ap);
|
||||
|
||||
if (ret < 0)
|
||||
return 0;
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
void
|
||||
gt_log(const char *fmt, ...)
|
||||
{
|
||||
@@ -26,23 +12,6 @@ gt_log(const char *fmt, ...)
|
||||
va_end(ap);
|
||||
}
|
||||
|
||||
void
|
||||
gt_fatal(const char *fmt, ...)
|
||||
{
|
||||
va_list ap;
|
||||
va_start(ap, fmt);
|
||||
vfprintf(stderr, fmt, ap);
|
||||
va_end(ap);
|
||||
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
|
||||
void
|
||||
gt_na(const char *name)
|
||||
{
|
||||
gt_log("%s is not available on your platform\n", name);
|
||||
}
|
||||
|
||||
int
|
||||
gt_tohex(char *dst, size_t dst_size, const uint8_t *src, size_t src_size)
|
||||
{
|
||||
@@ -100,3 +69,44 @@ gt_fromhex(uint8_t *dst, size_t dst_size, const char *src, size_t src_size)
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
void
|
||||
gt_set_port(struct sockaddr *sa, uint16_t port)
|
||||
{
|
||||
switch (sa->sa_family) {
|
||||
case AF_INET:
|
||||
((struct sockaddr_in *)sa)->sin_port = htons(port);
|
||||
break;
|
||||
case AF_INET6:
|
||||
((struct sockaddr_in6 *)sa)->sin6_port = htons(port);
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
uint16_t
|
||||
gt_get_port(struct sockaddr *sa)
|
||||
{
|
||||
switch (sa->sa_family) {
|
||||
case AF_INET:
|
||||
return ntohs(((struct sockaddr_in *)sa)->sin_port);
|
||||
case AF_INET6:
|
||||
return ntohs(((struct sockaddr_in6 *)sa)->sin6_port);
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
int
|
||||
gt_toaddr(char *str, size_t size, struct sockaddr *sa)
|
||||
{
|
||||
switch (sa->sa_family) {
|
||||
case AF_INET:
|
||||
return -!inet_ntop(AF_INET,
|
||||
&((struct sockaddr_in *)sa)->sin_addr, str, size);
|
||||
case AF_INET6:
|
||||
return -!inet_ntop(AF_INET6,
|
||||
&((struct sockaddr_in6 *)sa)->sin6_addr, str, size);
|
||||
}
|
||||
|
||||
return -1;
|
||||
}
|
||||
|
||||
32
src/common.h
32
src/common.h
@@ -1,11 +1,26 @@
|
||||
#pragma once
|
||||
|
||||
#if defined __linux__ && !defined _GNU_SOURCE
|
||||
#define _GNU_SOURCE
|
||||
#endif
|
||||
|
||||
#include <stdlib.h>
|
||||
#include <stdint.h>
|
||||
#include <string.h>
|
||||
#include <unistd.h>
|
||||
#include <signal.h>
|
||||
#include <errno.h>
|
||||
|
||||
#include <arpa/inet.h>
|
||||
|
||||
#ifndef PACKAGE_NAME
|
||||
#define PACKAGE_NAME "glorytun"
|
||||
#endif
|
||||
|
||||
#ifndef PACKAGE_VERSION
|
||||
#define PACKAGE_VERSION "0.0.0"
|
||||
#endif
|
||||
|
||||
#define COUNT(x) (sizeof(x)/sizeof(x[0]))
|
||||
|
||||
#define ALIGN_SIZE (1<<4)
|
||||
@@ -35,10 +50,23 @@
|
||||
#undef MIN
|
||||
#define MIN(x,y) ({ __typeof__(x) X=(x); __typeof__(y) Y=(y); X < Y ? X : Y; })
|
||||
|
||||
extern volatile sig_atomic_t gt_alarm;
|
||||
extern volatile sig_atomic_t gt_reload;
|
||||
extern volatile sig_atomic_t gt_quit;
|
||||
|
||||
int gt_print (const char *, ...) _printf_(1,2);
|
||||
void gt_log (const char *, ...) _printf_(1,2);
|
||||
void gt_fatal (const char *, ...) _printf_(1,2) _noreturn_;
|
||||
void gt_na (const char *);
|
||||
|
||||
int gt_tohex (char *, size_t, const uint8_t *, size_t);
|
||||
int gt_fromhex (uint8_t *, size_t, const char *, size_t);
|
||||
|
||||
void gt_set_port (struct sockaddr *, uint16_t);
|
||||
uint16_t gt_get_port (struct sockaddr *);
|
||||
int gt_toaddr (char *, size_t, struct sockaddr *);
|
||||
|
||||
int gt_show (int, char **);
|
||||
int gt_bind (int, char **);
|
||||
int gt_path (int, char **);
|
||||
int gt_keygen (int, char **);
|
||||
int gt_bench (int, char **);
|
||||
int gt_set (int, char **);
|
||||
|
||||
183
src/ctl.c
Normal file
183
src/ctl.c
Normal file
@@ -0,0 +1,183 @@
|
||||
#include "common.h"
|
||||
#include "ctl.h"
|
||||
#include "str.h"
|
||||
|
||||
#include <stdio.h>
|
||||
#include <unistd.h>
|
||||
#include <dirent.h>
|
||||
#include <sys/socket.h>
|
||||
#include <sys/stat.h>
|
||||
#include <sys/un.h>
|
||||
|
||||
#define CTL_BIND_MAX 64
|
||||
|
||||
int
|
||||
ctl_reply(int fd, struct ctl_msg *res, struct ctl_msg *req)
|
||||
{
|
||||
if (fd == -1) {
|
||||
errno = EINVAL;
|
||||
return -1;
|
||||
}
|
||||
|
||||
if ((send(fd, req, sizeof(struct ctl_msg), 0) == -1) ||
|
||||
(recv(fd, res, sizeof(struct ctl_msg), 0) == -1))
|
||||
return -1;
|
||||
|
||||
if (res->type != req->type || !res->reply) {
|
||||
errno = EINTR;
|
||||
return -1;
|
||||
}
|
||||
|
||||
if (res->ret) {
|
||||
errno = res->ret;
|
||||
return -1;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static int
|
||||
ctl_setsun(struct sockaddr_un *dst, const char *dir, const char *file)
|
||||
{
|
||||
struct sockaddr_un sun = {
|
||||
.sun_family = AF_UNIX,
|
||||
};
|
||||
|
||||
const char *path[] = {dir, "/", file};
|
||||
const size_t len = sizeof(sun.sun_path) - 1;
|
||||
|
||||
if (str_cat(sun.sun_path, len, path, COUNT(path)) == len) {
|
||||
if (str_cat(NULL, len + 1, path, COUNT(path)) > len) {
|
||||
errno = EINVAL;
|
||||
return -1;
|
||||
}
|
||||
}
|
||||
|
||||
*dst = sun;
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static int
|
||||
ctl_bind(int fd, const char *dir, const char *file)
|
||||
{
|
||||
char tmp[32];
|
||||
struct sockaddr_un sun;
|
||||
|
||||
if (str_empty(file)) {
|
||||
for (int i = 0; i < CTL_BIND_MAX; i++) {
|
||||
if (snprintf(tmp, sizeof(tmp), ".%i", i) >= sizeof(tmp))
|
||||
return -1;
|
||||
|
||||
if (ctl_setsun(&sun, dir, tmp))
|
||||
return -1;
|
||||
|
||||
if (!bind(fd, (struct sockaddr *)&sun, sizeof(sun)))
|
||||
return 0;
|
||||
}
|
||||
} else {
|
||||
if (ctl_setsun(&sun, dir, file))
|
||||
return -1;
|
||||
|
||||
unlink(sun.sun_path);
|
||||
|
||||
if (!bind(fd, (struct sockaddr *)&sun, sizeof(sun)))
|
||||
return 0;
|
||||
}
|
||||
|
||||
return -1;
|
||||
}
|
||||
|
||||
void
|
||||
ctl_delete(int fd)
|
||||
{
|
||||
if (fd == -1)
|
||||
return;
|
||||
|
||||
struct sockaddr_storage ss;
|
||||
socklen_t sslen = sizeof(ss);
|
||||
|
||||
if ((getsockname(fd, (struct sockaddr *)&ss, &sslen) == 0) &&
|
||||
(ss.ss_family == AF_UNIX))
|
||||
unlink(((struct sockaddr_un *)&ss)->sun_path);
|
||||
|
||||
close(fd);
|
||||
}
|
||||
|
||||
int
|
||||
ctl_create(const char *dir, const char *file)
|
||||
{
|
||||
if (str_empty(dir)) {
|
||||
errno = EINVAL;
|
||||
return -1;
|
||||
}
|
||||
|
||||
if (mkdir(dir, 0700) == -1 && errno != EEXIST)
|
||||
return -1;
|
||||
|
||||
int fd = socket(AF_UNIX, SOCK_DGRAM, 0);
|
||||
|
||||
if (fd == -1)
|
||||
return -1;
|
||||
|
||||
if (ctl_bind(fd, dir, file)) {
|
||||
int err = errno;
|
||||
close(fd);
|
||||
errno = err;
|
||||
return -1;
|
||||
}
|
||||
|
||||
return fd;
|
||||
}
|
||||
|
||||
int
|
||||
ctl_connect(const char *dir, const char *file)
|
||||
{
|
||||
if (str_empty(dir)) {
|
||||
errno = EINVAL;
|
||||
return -1;
|
||||
}
|
||||
|
||||
if (!file) {
|
||||
DIR *dp = opendir(dir);
|
||||
|
||||
if (!dp)
|
||||
return -1;
|
||||
|
||||
struct dirent *d = NULL;
|
||||
|
||||
while (d = readdir(dp), d) {
|
||||
if (d->d_name[0] == '.')
|
||||
continue;
|
||||
|
||||
if (file) {
|
||||
closedir(dp);
|
||||
errno = ENOENT;
|
||||
return -1;
|
||||
}
|
||||
|
||||
file = &d->d_name[0];
|
||||
}
|
||||
|
||||
closedir(dp);
|
||||
}
|
||||
|
||||
struct sockaddr_un sun;
|
||||
|
||||
if (ctl_setsun(&sun, dir, file))
|
||||
return -1;
|
||||
|
||||
int fd = ctl_create(dir, NULL);
|
||||
|
||||
if (fd == -1)
|
||||
return -1;
|
||||
|
||||
if (connect(fd, (struct sockaddr *)&sun, sizeof(sun))) {
|
||||
int err = errno;
|
||||
ctl_delete(fd);
|
||||
errno = err;
|
||||
return -1;
|
||||
}
|
||||
|
||||
return fd;
|
||||
}
|
||||
43
src/ctl.h
Normal file
43
src/ctl.h
Normal file
@@ -0,0 +1,43 @@
|
||||
#pragma once
|
||||
|
||||
#include "../mud/mud.h"
|
||||
|
||||
#include <sys/socket.h>
|
||||
|
||||
enum ctl_type {
|
||||
CTL_NONE = 0,
|
||||
CTL_STATE,
|
||||
CTL_STATUS,
|
||||
CTL_MTU,
|
||||
CTL_TC,
|
||||
CTL_TIMEOUT,
|
||||
CTL_TIMETOLERANCE,
|
||||
CTL_PATH_STATUS,
|
||||
};
|
||||
|
||||
struct ctl_msg {
|
||||
enum ctl_type type;
|
||||
int reply, ret;
|
||||
union {
|
||||
struct {
|
||||
struct sockaddr_storage addr;
|
||||
enum mud_state state;
|
||||
} path;
|
||||
struct mud_path path_status;
|
||||
struct {
|
||||
size_t mtu;
|
||||
int chacha;
|
||||
struct sockaddr_storage bind;
|
||||
struct sockaddr_storage peer;
|
||||
} status;
|
||||
int mtu;
|
||||
int tc;
|
||||
unsigned long timeout;
|
||||
unsigned long timetolerance;
|
||||
};
|
||||
};
|
||||
|
||||
int ctl_create (const char *, const char *);
|
||||
int ctl_connect (const char *, const char *);
|
||||
int ctl_reply (int, struct ctl_msg *, struct ctl_msg *);
|
||||
void ctl_delete (int);
|
||||
158
src/db.c
158
src/db.c
@@ -1,158 +0,0 @@
|
||||
#include "common.h"
|
||||
|
||||
#include "db.h"
|
||||
#include "str.h"
|
||||
|
||||
#define CBIT(X) (1 & (intptr_t)(X))
|
||||
#define CBIT_PTR(X) (uint8_t *)(1 | (intptr_t)(X))
|
||||
#define CBIT_NODE(X) (struct node *)(1 ^ (intptr_t)(X))
|
||||
|
||||
struct node {
|
||||
uint8_t *child[2];
|
||||
uint32_t point;
|
||||
};
|
||||
|
||||
_pure_ static inline size_t
|
||||
db_size(const uint8_t *a)
|
||||
{
|
||||
return (a[0] ?: str_len((char *)a + 1)) + 1;
|
||||
}
|
||||
|
||||
_pure_ static inline size_t
|
||||
db_cmp(const uint8_t *a, const uint8_t *b)
|
||||
{
|
||||
const size_t size = a[0];
|
||||
|
||||
if (size != b[0])
|
||||
return 1;
|
||||
|
||||
if (!size) {
|
||||
size_t i = str_cmp((char *)a + 1, (char *)b + 1);
|
||||
return i ? i + 1 : 0;
|
||||
}
|
||||
|
||||
for (size_t i = 1; i <= size; i++) {
|
||||
if (a[i] != b[i])
|
||||
return i + 1;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
_pure_ static inline int
|
||||
db_dir(const uint32_t point, uint8_t *data, const size_t size)
|
||||
{
|
||||
const size_t pos = point >> 8;
|
||||
|
||||
if (pos >= size)
|
||||
return 0;
|
||||
|
||||
return ((point | data[pos]) & 255) == 255;
|
||||
}
|
||||
|
||||
uint8_t *
|
||||
db_search(uint8_t **p, uint8_t *data)
|
||||
{
|
||||
if (_0_(!*p))
|
||||
return NULL;
|
||||
|
||||
uint8_t *r = *p;
|
||||
const size_t size = db_size(data);
|
||||
|
||||
while (CBIT(r)) {
|
||||
struct node *node = CBIT_NODE(r);
|
||||
r = node->child[db_dir(node->point, data, size)];
|
||||
}
|
||||
|
||||
if (!db_cmp(r, data))
|
||||
return r;
|
||||
|
||||
return NULL;
|
||||
}
|
||||
|
||||
uint8_t *
|
||||
db_insert(uint8_t **p, uint8_t *data)
|
||||
{
|
||||
if (_0_(CBIT(data)))
|
||||
return NULL;
|
||||
|
||||
if (_0_(!*p)) {
|
||||
*p = data;
|
||||
return data;
|
||||
}
|
||||
|
||||
uint8_t *r = *p;
|
||||
size_t size = db_size(data);
|
||||
|
||||
while (CBIT(r)) {
|
||||
struct node *node = CBIT_NODE(r);
|
||||
r = node->child[db_dir(node->point, data, size)];
|
||||
}
|
||||
|
||||
const size_t diff = db_cmp(r, data);
|
||||
|
||||
if (_0_(!diff))
|
||||
return r;
|
||||
|
||||
const size_t pos = diff - 1;
|
||||
const uint8_t mask = ~((1u << 31) >> CLZ(r[pos] ^ data[pos]));
|
||||
const size_t point = (pos << 8) | mask;
|
||||
|
||||
while (CBIT(*p)) {
|
||||
struct node *node = CBIT_NODE(*p);
|
||||
|
||||
if (node->point > point)
|
||||
break;
|
||||
|
||||
p = node->child + db_dir(node->point, data, size);
|
||||
}
|
||||
|
||||
struct node *node = malloc(sizeof(struct node));
|
||||
|
||||
if (_0_(!node))
|
||||
return NULL;
|
||||
|
||||
const int dir = (mask | r[pos]) == 255;
|
||||
|
||||
node->child[dir] = *p;
|
||||
node->child[1 - dir] = data;
|
||||
node->point = point;
|
||||
|
||||
*p = CBIT_PTR(node);
|
||||
|
||||
return data;
|
||||
}
|
||||
|
||||
uint8_t *
|
||||
db_remove(uint8_t **p, uint8_t *data)
|
||||
{
|
||||
if (_0_(!*p))
|
||||
return NULL;
|
||||
|
||||
const size_t size = db_size(data);
|
||||
|
||||
uint8_t **p_old = NULL;
|
||||
struct node *node = NULL;
|
||||
int dir = 0;
|
||||
|
||||
while (CBIT(*p)) {
|
||||
p_old = p;
|
||||
node = CBIT_NODE(*p);
|
||||
dir = db_dir(node->point, data, size);
|
||||
p = node->child + dir;
|
||||
}
|
||||
|
||||
if (_0_(db_cmp(data, *p)))
|
||||
return NULL;
|
||||
|
||||
uint8_t *r = *p;
|
||||
|
||||
if (p_old) {
|
||||
*p_old = node->child[1 - dir];
|
||||
free(node);
|
||||
} else {
|
||||
*p = NULL;
|
||||
}
|
||||
|
||||
return r;
|
||||
}
|
||||
7
src/db.h
7
src/db.h
@@ -1,7 +0,0 @@
|
||||
#pragma once
|
||||
|
||||
#include <stdint.h>
|
||||
|
||||
uint8_t *db_search (uint8_t **, uint8_t *);
|
||||
uint8_t *db_insert (uint8_t **, uint8_t *);
|
||||
uint8_t *db_remove (uint8_t **, uint8_t *);
|
||||
36
src/iface.c
Normal file
36
src/iface.c
Normal file
@@ -0,0 +1,36 @@
|
||||
#include "common.h"
|
||||
#include "iface.h"
|
||||
#include "str.h"
|
||||
|
||||
#include <net/if.h>
|
||||
#include <sys/ioctl.h>
|
||||
|
||||
int
|
||||
iface_set_mtu(const char *dev_name, int mtu)
|
||||
{
|
||||
struct ifreq ifr = {
|
||||
.ifr_mtu = mtu,
|
||||
};
|
||||
|
||||
const size_t len = sizeof(ifr.ifr_name) - 1;
|
||||
|
||||
if (str_cpy(ifr.ifr_name, len, dev_name) == len) {
|
||||
if (str_len(dev_name, len + 1) > len) {
|
||||
errno = EINTR;
|
||||
return -1;
|
||||
}
|
||||
}
|
||||
|
||||
int fd = socket(AF_INET, SOCK_DGRAM, 0);
|
||||
|
||||
if (fd == -1)
|
||||
return -1;
|
||||
|
||||
int ret = ioctl(fd, SIOCSIFMTU, &ifr);
|
||||
|
||||
int err = errno;
|
||||
close(fd);
|
||||
errno = err;
|
||||
|
||||
return ret;
|
||||
}
|
||||
3
src/iface.h
Normal file
3
src/iface.h
Normal file
@@ -0,0 +1,3 @@
|
||||
#pragma once
|
||||
|
||||
int iface_set_mtu (const char *, int);
|
||||
40
src/ip.h
40
src/ip.h
@@ -19,6 +19,42 @@ ip_get_version(const uint8_t *data, size_t size)
|
||||
return data[0] >> 4;
|
||||
}
|
||||
|
||||
static inline uint32_t
|
||||
ip_read32(const uint8_t *src)
|
||||
{
|
||||
uint32_t ret = src[3];
|
||||
ret |= ((uint32_t)src[2]) << 8;
|
||||
ret |= ((uint32_t)src[1]) << 16;
|
||||
ret |= ((uint32_t)src[0]) << 24;
|
||||
return ret;
|
||||
}
|
||||
|
||||
static inline uint16_t
|
||||
ip_read16(const uint8_t *src)
|
||||
{
|
||||
uint16_t ret = src[1];
|
||||
ret |= ((uint16_t)src[0]) << 8;
|
||||
return ret;
|
||||
}
|
||||
|
||||
static inline size_t
|
||||
ip_get_mtu(struct ip_common *ic, const uint8_t *data, size_t size)
|
||||
{
|
||||
if (ic->hdr_size <= 0 || ic->hdr_size + 8 > size)
|
||||
return 0;
|
||||
|
||||
const uint8_t *p = &data[ic->hdr_size];
|
||||
|
||||
if (ic->version == 4 && ic->proto == 1 && p[0] == 3)
|
||||
return ip_read16(&p[6]);
|
||||
|
||||
// not tested..
|
||||
// if (ic->version == 6 && ic->proto == 58 && p[0] == 2)
|
||||
// return ip_read32(&p[4]);
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static inline int
|
||||
ip_get_common(struct ip_common *ic, const uint8_t *data, size_t size)
|
||||
{
|
||||
@@ -29,7 +65,7 @@ ip_get_common(struct ip_common *ic, const uint8_t *data, size_t size)
|
||||
ic->tc = data[1];
|
||||
ic->proto = data[9];
|
||||
ic->hdr_size = (data[0] & 0xF) << 2;
|
||||
ic->size = ((data[2] << 8) | data[3]);
|
||||
ic->size = ip_read16(&data[2]);
|
||||
if (ic->size >= 20)
|
||||
return 0;
|
||||
break;
|
||||
@@ -37,7 +73,7 @@ ip_get_common(struct ip_common *ic, const uint8_t *data, size_t size)
|
||||
ic->tc = ((data[0] & 0xF) << 4) | (data[1] >> 4);
|
||||
ic->proto = data[6];
|
||||
ic->hdr_size = 40;
|
||||
ic->size = ((data[4] << 8) | data[5]) + 40;
|
||||
ic->size = ip_read16(&data[4]) + 40;
|
||||
return 0;
|
||||
}
|
||||
|
||||
|
||||
22
src/keygen.c
Normal file
22
src/keygen.c
Normal file
@@ -0,0 +1,22 @@
|
||||
#include "common.h"
|
||||
|
||||
#include <sodium.h>
|
||||
#include <stdio.h>
|
||||
|
||||
int
|
||||
gt_keygen(int argc, char **argv)
|
||||
{
|
||||
if (sodium_init() == -1) {
|
||||
gt_log("sodium init failed\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
unsigned char key[32];
|
||||
randombytes_buf(key, sizeof(key));
|
||||
|
||||
char buf[2 * sizeof(key) + 1];
|
||||
gt_tohex(buf, sizeof(buf), key, sizeof(key));
|
||||
printf("%s\n", buf);
|
||||
|
||||
return 0;
|
||||
}
|
||||
479
src/main.c
479
src/main.c
@@ -1,93 +1,24 @@
|
||||
#include "common.h"
|
||||
|
||||
#include "db.h"
|
||||
#include "ip.h"
|
||||
#include "option.h"
|
||||
#include "str.h"
|
||||
#include "tun.h"
|
||||
|
||||
#include <fcntl.h>
|
||||
#include <inttypes.h>
|
||||
#include <limits.h>
|
||||
#include <poll.h>
|
||||
#include <signal.h>
|
||||
#include <stdio.h>
|
||||
#include <sys/socket.h>
|
||||
#include <sys/time.h>
|
||||
|
||||
#include <arpa/inet.h>
|
||||
#include <netdb.h>
|
||||
|
||||
#include "../mud/mud.h"
|
||||
|
||||
#ifndef O_CLOEXEC
|
||||
#define O_CLOEXEC 0
|
||||
#endif
|
||||
|
||||
#define GT_MTU(X) ((X)-28)
|
||||
|
||||
static struct {
|
||||
volatile sig_atomic_t quit;
|
||||
char *dev;
|
||||
char *keyfile;
|
||||
char *host;
|
||||
long port;
|
||||
struct {
|
||||
char *list;
|
||||
char *backup;
|
||||
long port;
|
||||
} bind;
|
||||
long mtu;
|
||||
long timeout;
|
||||
long time_tolerance;
|
||||
int ipv4;
|
||||
int ipv6;
|
||||
int mtu_auto;
|
||||
int chacha20;
|
||||
int version;
|
||||
struct {
|
||||
unsigned char *data;
|
||||
long size;
|
||||
} buf;
|
||||
} gt = {
|
||||
.port = 5000,
|
||||
.bind = {
|
||||
.port = 5000,
|
||||
},
|
||||
.mtu = 1500,
|
||||
.timeout = 5000,
|
||||
.ipv4 = 1,
|
||||
#ifdef __linux__
|
||||
.ipv6 = 1,
|
||||
#endif
|
||||
.buf = {
|
||||
.size = 64 * 1024,
|
||||
},
|
||||
};
|
||||
|
||||
static void
|
||||
fd_set_nonblock(int fd)
|
||||
{
|
||||
int ret;
|
||||
|
||||
do {
|
||||
ret = fcntl(fd, F_GETFL, 0);
|
||||
} while (ret == -1 && errno == EINTR);
|
||||
|
||||
int flags = (ret == -1) ? 0 : ret;
|
||||
|
||||
do {
|
||||
ret = fcntl(fd, F_SETFL, flags | O_NONBLOCK);
|
||||
} while (ret == -1 && errno == EINTR);
|
||||
|
||||
if (ret == -1)
|
||||
perror("fcntl O_NONBLOCK");
|
||||
}
|
||||
volatile sig_atomic_t gt_alarm;
|
||||
volatile sig_atomic_t gt_reload;
|
||||
volatile sig_atomic_t gt_quit;
|
||||
|
||||
static void
|
||||
gt_quit_handler(int sig)
|
||||
{
|
||||
gt.quit = 1;
|
||||
switch (sig) {
|
||||
case SIGALRM:
|
||||
gt_alarm = 1;
|
||||
return;
|
||||
case SIGHUP:
|
||||
gt_reload = 1; /* FALLTHRU */
|
||||
default:
|
||||
gt_quit = 1;
|
||||
}
|
||||
}
|
||||
|
||||
static void
|
||||
@@ -103,384 +34,60 @@ gt_set_signal(void)
|
||||
sigaction(SIGINT, &sa, NULL);
|
||||
sigaction(SIGQUIT, &sa, NULL);
|
||||
sigaction(SIGTERM, &sa, NULL);
|
||||
sigaction(SIGHUP, &sa, NULL);
|
||||
sigaction(SIGALRM, &sa, NULL);
|
||||
|
||||
sa.sa_handler = SIG_IGN;
|
||||
sigaction(SIGHUP, &sa, NULL);
|
||||
sigaction(SIGPIPE, &sa, NULL);
|
||||
sigaction(SIGUSR1, &sa, NULL);
|
||||
sigaction(SIGUSR2, &sa, NULL);
|
||||
}
|
||||
|
||||
static void
|
||||
gt_print_secretkey(struct mud *mud)
|
||||
{
|
||||
unsigned char key[32];
|
||||
size_t size = sizeof(key);
|
||||
|
||||
if (mud_get_key(mud, key, &size))
|
||||
return;
|
||||
|
||||
char buf[2 * sizeof(key) + 1];
|
||||
|
||||
gt_tohex(buf, sizeof(buf), key, size);
|
||||
gt_print("secret key: %s\n", buf);
|
||||
}
|
||||
|
||||
static int
|
||||
gt_setup_secretkey(struct mud *mud, char *keyfile)
|
||||
gt_version(int argc, char **argv)
|
||||
{
|
||||
int fd;
|
||||
|
||||
do {
|
||||
fd = open(keyfile, O_RDONLY | O_CLOEXEC);
|
||||
} while (fd == -1 && errno == EINTR);
|
||||
|
||||
if (fd == -1) {
|
||||
perror("open keyfile");
|
||||
return -1;
|
||||
}
|
||||
|
||||
unsigned char key[32];
|
||||
char buf[2 * sizeof(key)];
|
||||
size_t size = 0;
|
||||
|
||||
while (size < sizeof(buf)) {
|
||||
ssize_t r = read(fd, &buf[size], sizeof(buf) - size);
|
||||
|
||||
if (r <= (ssize_t)0) {
|
||||
if (r && (errno == EAGAIN || errno == EINTR))
|
||||
continue;
|
||||
break;
|
||||
}
|
||||
|
||||
size += r;
|
||||
}
|
||||
|
||||
close(fd);
|
||||
|
||||
if (size != sizeof(buf)) {
|
||||
gt_log("unable to read secret key\n");
|
||||
return -1;
|
||||
}
|
||||
|
||||
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));
|
||||
|
||||
printf(PACKAGE_VERSION "\n");
|
||||
return 0;
|
||||
}
|
||||
|
||||
static int
|
||||
gt_setup_option(int argc, char **argv)
|
||||
{
|
||||
// clang-format off
|
||||
|
||||
struct option opts[] = {
|
||||
{ "host", >.host, option_str },
|
||||
{ "port", >.port, option_long },
|
||||
{ "bind", >.bind.list, option_str },
|
||||
{ "bind-backup", >.bind.backup, option_str },
|
||||
{ "bind-port", >.bind.port, option_long },
|
||||
{ "dev", >.dev, option_str },
|
||||
{ "mtu", >.mtu, option_long },
|
||||
{ "mtu-auto", NULL, option_option },
|
||||
{ "keyfile", >.keyfile, option_str },
|
||||
{ "timeout", >.timeout, option_long },
|
||||
{ "time-tolerance", >.time_tolerance, option_long },
|
||||
{ "v4only", NULL, option_option },
|
||||
{ "v6only", NULL, option_option },
|
||||
{ "chacha20", NULL, option_option },
|
||||
{ "buf-size", >.buf.size, option_long },
|
||||
{ "version", NULL, option_option },
|
||||
{ NULL },
|
||||
};
|
||||
|
||||
// clang-format on
|
||||
|
||||
if (option(opts, argc, argv))
|
||||
return 1;
|
||||
|
||||
int v4only = option_is_set(opts, "v4only");
|
||||
int v6only = option_is_set(opts, "v6only");
|
||||
|
||||
if (v4only && v6only) {
|
||||
gt_log("v4only and v6only cannot be both set\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
if (gt.host && !option_is_set(opts, "keyfile")) {
|
||||
gt_log("keyfile option must be set\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
if ((int)gt.timeout <= 0) {
|
||||
gt_log("bad timeout\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
if (gt.buf.size <= 0) {
|
||||
gt_log("bad buf-size\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
if (v4only) {
|
||||
gt.ipv4 = 1;
|
||||
gt.ipv6 = 0;
|
||||
}
|
||||
|
||||
if (v6only) {
|
||||
gt.ipv4 = 0;
|
||||
gt.ipv6 = 1;
|
||||
}
|
||||
|
||||
gt.mtu_auto = option_is_set(opts, "mtu-auto");
|
||||
gt.chacha20 = option_is_set(opts, "chacha20");
|
||||
gt.version = option_is_set(opts, "version");
|
||||
|
||||
gt.buf.data = malloc(gt.buf.size);
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static void
|
||||
gt_setup_mtu(struct mud *mud, char *tun_name)
|
||||
{
|
||||
int mtu = mud_get_mtu(mud);
|
||||
|
||||
if (mtu == (int)gt.mtu)
|
||||
return;
|
||||
|
||||
gt.mtu = mtu;
|
||||
|
||||
gt_log("setup MTU to %i on interface %s\n", mtu, tun_name);
|
||||
|
||||
if (tun_set_mtu(tun_name, mtu) == -1)
|
||||
perror("tun_set_mtu");
|
||||
}
|
||||
|
||||
int
|
||||
main(int argc, char **argv)
|
||||
{
|
||||
gt_set_signal();
|
||||
|
||||
if (gt_setup_option(argc, argv))
|
||||
return 1;
|
||||
struct {
|
||||
char *name;
|
||||
char *help;
|
||||
int (*call)(int, char **);
|
||||
} cmd[] = {
|
||||
{"show", "show all running tunnels", gt_show},
|
||||
{"bench", "start a crypto bench", gt_bench},
|
||||
{"bind", "start a new tunnel", gt_bind},
|
||||
{"set", "change tunnel properties", gt_set},
|
||||
{"keygen", "generate a new secret key", gt_keygen},
|
||||
{"path", "manage paths", gt_path},
|
||||
{"version", "show version", gt_version},
|
||||
{NULL}};
|
||||
|
||||
if (gt.version) {
|
||||
gt_print(PACKAGE_STRING "\n");
|
||||
return 0;
|
||||
if (argc < 2)
|
||||
return gt_show(argc, argv);
|
||||
|
||||
for (int k = 0; cmd[k].name; k++) {
|
||||
if (!str_cmp(cmd[k].name, argv[1]))
|
||||
return cmd[k].call(argc - 1, argv + 1);
|
||||
}
|
||||
|
||||
int icmp_fd = -1;
|
||||
printf("unknown command `%s', available commands:\n\n", argv[1]);
|
||||
|
||||
if (gt.ipv4 && gt.mtu_auto) {
|
||||
icmp_fd = socket(AF_INET, SOCK_RAW, IPPROTO_ICMP);
|
||||
int len = 0;
|
||||
|
||||
if (icmp_fd == -1)
|
||||
gt_log("couldn't create ICMP socket\n");
|
||||
}
|
||||
for (int k = 0; cmd[k].name; k++)
|
||||
len = MAX(len, (int)str_len(cmd[k].name, 32));
|
||||
|
||||
char *tun_name = NULL;
|
||||
for (int k = 0; cmd[k].name; k++)
|
||||
printf(" %-*s %s\n", len, cmd[k].name, cmd[k].help);
|
||||
|
||||
int tun_fd = tun_create(gt.dev, &tun_name);
|
||||
printf("\n");
|
||||
|
||||
if (tun_fd == -1) {
|
||||
gt_log("couldn't create tun device\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
struct mud *mud = mud_create(gt.bind.port, gt.ipv4, gt.ipv6,
|
||||
!gt.chacha20, GT_MTU(gt.mtu));
|
||||
|
||||
if (!mud) {
|
||||
gt_log("couldn't create mud\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
if (str_empty(gt.keyfile)) {
|
||||
gt_print_secretkey(mud);
|
||||
} else {
|
||||
if (gt_setup_secretkey(mud, gt.keyfile))
|
||||
return 1;
|
||||
}
|
||||
|
||||
mud_set_send_timeout_msec(mud, gt.timeout);
|
||||
|
||||
if (gt.time_tolerance > 0)
|
||||
mud_set_time_tolerance_sec(mud, gt.time_tolerance);
|
||||
|
||||
if (gt.host && gt.port) {
|
||||
if (gt.bind.backup) {
|
||||
if (mud_peer(mud, gt.bind.backup, gt.host, gt.port, 1)) {
|
||||
perror("mud_peer (backup)");
|
||||
return 1;
|
||||
}
|
||||
}
|
||||
|
||||
if (gt.bind.list) {
|
||||
char tmp[1024];
|
||||
char *name = &tmp[0];
|
||||
|
||||
str_cpy(tmp, gt.bind.list, sizeof(tmp) - 1);
|
||||
|
||||
while (*name) {
|
||||
char *p = name;
|
||||
|
||||
while (*p && *p != ',')
|
||||
p++;
|
||||
|
||||
if (*p)
|
||||
*p++ = 0;
|
||||
|
||||
if (mud_peer(mud, name, gt.host, gt.port, 0)) {
|
||||
perror("mud_peer");
|
||||
return 1;
|
||||
}
|
||||
|
||||
name = p;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
gt_setup_mtu(mud, tun_name);
|
||||
|
||||
int mud_fd = mud_get_fd(mud);
|
||||
|
||||
fd_set_nonblock(tun_fd);
|
||||
fd_set_nonblock(mud_fd);
|
||||
|
||||
if (icmp_fd != -1)
|
||||
fd_set_nonblock(icmp_fd);
|
||||
|
||||
gt_log("running...\n");
|
||||
|
||||
fd_set rfds;
|
||||
FD_ZERO(&rfds);
|
||||
|
||||
int last_fd = 1 + MAX(tun_fd, MAX(mud_fd, icmp_fd));
|
||||
|
||||
while (!gt.quit) {
|
||||
FD_SET(tun_fd, &rfds);
|
||||
FD_SET(mud_fd, &rfds);
|
||||
|
||||
if (icmp_fd != -1)
|
||||
FD_SET(icmp_fd, &rfds);
|
||||
|
||||
if (select(last_fd, &rfds, NULL, NULL, NULL) == -1) {
|
||||
if (errno != EBADF)
|
||||
continue;
|
||||
perror("select");
|
||||
return 1;
|
||||
}
|
||||
|
||||
if (icmp_fd != -1 && FD_ISSET(icmp_fd, &rfds)) {
|
||||
struct sockaddr_storage ss;
|
||||
socklen_t sl = sizeof(ss);
|
||||
ssize_t r = recvfrom(icmp_fd, gt.buf.data, gt.buf.size, 0,
|
||||
(struct sockaddr *)&ss, &sl);
|
||||
if (r >= 8) {
|
||||
struct ip_common ic;
|
||||
if (!ip_get_common(&ic, gt.buf.data, r) && ic.proto == 1) {
|
||||
unsigned char *data = >.buf.data[ic.hdr_size];
|
||||
if (data[0] == 3) {
|
||||
int mtu = (data[6] << 8) | data[7];
|
||||
if (mtu) {
|
||||
gt_log("received MTU from ICMP: %i\n", mtu);
|
||||
mud_set_mtu(mud, GT_MTU(mtu));
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if (FD_ISSET(tun_fd, &rfds)) {
|
||||
size_t size = 0;
|
||||
|
||||
while (gt.buf.size - size >= gt.mtu) {
|
||||
const int r = tun_read(tun_fd, >.buf.data[size], gt.buf.size - size);
|
||||
|
||||
if (r <= 0 || r > gt.mtu)
|
||||
break;
|
||||
|
||||
struct ip_common ic;
|
||||
|
||||
if (ip_get_common(&ic, >.buf.data[size], r) || ic.size != r)
|
||||
break;
|
||||
|
||||
size += r;
|
||||
}
|
||||
|
||||
int p = 0;
|
||||
|
||||
while (p < size) {
|
||||
int tc = 0;
|
||||
int q = p;
|
||||
|
||||
while (q < size) {
|
||||
struct ip_common ic;
|
||||
|
||||
if ((ip_get_common(&ic, >.buf.data[q], size - q)) ||
|
||||
(ic.size > size - q))
|
||||
break;
|
||||
|
||||
if (q + ic.size > p + gt.mtu)
|
||||
break;
|
||||
|
||||
q += ic.size;
|
||||
|
||||
if (tc < (ic.tc & 0xFC))
|
||||
tc = ic.tc & 0xFC;
|
||||
}
|
||||
|
||||
if (p >= q)
|
||||
break;
|
||||
|
||||
int r = mud_send(mud, >.buf.data[p], q - p, tc);
|
||||
|
||||
if (r == -1 && errno == EMSGSIZE) {
|
||||
gt_setup_mtu(mud, tun_name);
|
||||
} else {
|
||||
if (r == -1 && errno != EAGAIN)
|
||||
perror("mud_send");
|
||||
}
|
||||
|
||||
p = q;
|
||||
}
|
||||
}
|
||||
|
||||
if (FD_ISSET(mud_fd, &rfds)) {
|
||||
size_t size = 0;
|
||||
|
||||
while (gt.buf.size - size >= gt.mtu) {
|
||||
const int r = mud_recv(mud, >.buf.data[size], gt.buf.size - size);
|
||||
|
||||
if (r <= 0) {
|
||||
if (r == -1 && errno != EAGAIN)
|
||||
perror("mud_recv");
|
||||
break;
|
||||
}
|
||||
|
||||
size += r;
|
||||
}
|
||||
|
||||
int p = 0;
|
||||
|
||||
while (p < size) {
|
||||
struct ip_common ic;
|
||||
|
||||
if ((ip_get_common(&ic, >.buf.data[p], size - p)) ||
|
||||
(ic.size > size - p))
|
||||
break;
|
||||
|
||||
tun_write(tun_fd, >.buf.data[p], ic.size);
|
||||
|
||||
p += ic.size;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return 0;
|
||||
return 1;
|
||||
}
|
||||
|
||||
146
src/option.c
146
src/option.c
@@ -1,146 +0,0 @@
|
||||
#include "common.h"
|
||||
|
||||
#include "option.h"
|
||||
#include "str.h"
|
||||
|
||||
int
|
||||
option_str(void *data, int argc, char **argv)
|
||||
{
|
||||
if (argc < 2 || str_empty(argv[1])) {
|
||||
gt_print("option `%s' need a string argument\n", argv[0]);
|
||||
return -1;
|
||||
}
|
||||
|
||||
memcpy(data, &argv[1], sizeof(argv[1]));
|
||||
|
||||
return 1;
|
||||
}
|
||||
|
||||
int
|
||||
option_long(void *data, int argc, char **argv)
|
||||
{
|
||||
if (argc < 2 || str_empty(argv[1])) {
|
||||
gt_print("option `%s' need an integer argument\n", argv[0]);
|
||||
return -1;
|
||||
}
|
||||
|
||||
errno = 0;
|
||||
char *end;
|
||||
long val = strtol(argv[1], &end, 0);
|
||||
|
||||
if (errno || argv[1] == end) {
|
||||
gt_print("argument `%s' is not a valid integer\n", argv[1]);
|
||||
return -1;
|
||||
}
|
||||
|
||||
memcpy(data, &val, sizeof(val));
|
||||
|
||||
return 1;
|
||||
}
|
||||
|
||||
int
|
||||
option_is_set(struct option *opts, const char *name)
|
||||
{
|
||||
for (int k = 0; opts[k].name; k++) {
|
||||
if (!str_cmp(opts[k].name, name))
|
||||
return opts[k].set;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
int
|
||||
option_option(void *data, int argc, char **argv)
|
||||
{
|
||||
if (!data)
|
||||
return 0;
|
||||
|
||||
struct option *opts = (struct option *)data;
|
||||
|
||||
for (int k = 0; opts[k].name; k++)
|
||||
opts[k].set = 0;
|
||||
|
||||
for (int i = 1; i < argc; i++) {
|
||||
int found = 0;
|
||||
|
||||
for (int k = 0; opts[k].name; k++) {
|
||||
if (str_cmp(opts[k].name, argv[i]))
|
||||
continue;
|
||||
|
||||
if (opts[k].set) {
|
||||
gt_print("option `%s' is already set\n", opts[k].name);
|
||||
return -1;
|
||||
}
|
||||
|
||||
int ret = opts[k].call(opts[k].data, argc - i, &argv[i]);
|
||||
|
||||
if (ret < 0)
|
||||
return -1;
|
||||
|
||||
opts[k].set = 1;
|
||||
|
||||
i += ret;
|
||||
found = 1;
|
||||
break;
|
||||
}
|
||||
|
||||
if (!found)
|
||||
return i - 1;
|
||||
}
|
||||
|
||||
return argc;
|
||||
}
|
||||
|
||||
static int
|
||||
option_usage(struct option *opts, int slen)
|
||||
{
|
||||
if (!opts)
|
||||
return 0;
|
||||
|
||||
int len = 0;
|
||||
|
||||
for (int k = 0; opts[k].name; k++) {
|
||||
if (len > 40) {
|
||||
gt_print("\n%*s", slen, "");
|
||||
len = 0;
|
||||
}
|
||||
|
||||
len += gt_print(" [%s", opts[k].name);
|
||||
|
||||
if (opts[k].call == option_option) {
|
||||
len += option_usage((struct option *)opts[k].data, slen + len);
|
||||
} else {
|
||||
len += gt_print(" ARG");
|
||||
}
|
||||
|
||||
len += gt_print("]");
|
||||
}
|
||||
|
||||
return len;
|
||||
}
|
||||
|
||||
int
|
||||
option(struct option *opts, int argc, char **argv)
|
||||
{
|
||||
int ret = option_option(opts, argc, argv);
|
||||
|
||||
if (ret == argc)
|
||||
return 0;
|
||||
|
||||
if (ret < 0 || ret + 1 >= argc)
|
||||
return 1;
|
||||
|
||||
gt_print("option `%s' is unknown\n", argv[ret + 1]);
|
||||
|
||||
int slen = gt_print("usage: %s", argv[0]);
|
||||
|
||||
if (slen > 40) {
|
||||
slen = 12;
|
||||
gt_print("\n%*s", slen, "");
|
||||
}
|
||||
|
||||
option_usage(opts, slen);
|
||||
gt_print("\n");
|
||||
|
||||
return 1;
|
||||
}
|
||||
15
src/option.h
15
src/option.h
@@ -1,15 +0,0 @@
|
||||
#pragma once
|
||||
|
||||
struct option {
|
||||
char *name;
|
||||
void *data;
|
||||
int (*call) (void *, int, char **);
|
||||
int set;
|
||||
};
|
||||
|
||||
int option_option (void *, int, char **);
|
||||
int option_str (void *, int, char **);
|
||||
int option_long (void *, int, char **);
|
||||
|
||||
int option_is_set (struct option *, const char *);
|
||||
int option (struct option *, int, char **);
|
||||
107
src/path.c
Normal file
107
src/path.c
Normal file
@@ -0,0 +1,107 @@
|
||||
#include "common.h"
|
||||
#include "ctl.h"
|
||||
#include "str.h"
|
||||
|
||||
#include <stdio.h>
|
||||
#include <sys/socket.h>
|
||||
|
||||
#include "../argz/argz.h"
|
||||
|
||||
int
|
||||
gt_path_status(int fd)
|
||||
{
|
||||
struct ctl_msg req = {
|
||||
.type = CTL_PATH_STATUS,
|
||||
}, res = {0};
|
||||
|
||||
if (send(fd, &req, sizeof(struct ctl_msg), 0) == -1)
|
||||
return -1;
|
||||
|
||||
do {
|
||||
if (recv(fd, &res, sizeof(struct ctl_msg), 0) == -1)
|
||||
return -1;
|
||||
|
||||
char bindstr[INET6_ADDRSTRLEN] = {0};
|
||||
char peerstr[INET6_ADDRSTRLEN] = {0};
|
||||
|
||||
if (gt_toaddr(bindstr, sizeof(bindstr),
|
||||
(struct sockaddr *)&res.path_status.local_addr) ||
|
||||
gt_toaddr(peerstr, sizeof(peerstr),
|
||||
(struct sockaddr *)&res.path_status.addr))
|
||||
return -2;
|
||||
|
||||
const char *statestr = NULL;
|
||||
|
||||
switch (res.path_status.state) {
|
||||
case MUD_UP: statestr = "UP"; break;
|
||||
case MUD_BACKUP: statestr = "BACKUP"; break;
|
||||
case MUD_DOWN: statestr = "DOWN"; break;
|
||||
default: return -2;
|
||||
}
|
||||
|
||||
printf("path %s\n"
|
||||
" bind: %s\n"
|
||||
" peer: %s port %"PRIu16"\n"
|
||||
" rtt: %.3f\n",
|
||||
statestr, bindstr, peerstr,
|
||||
gt_get_port((struct sockaddr *)&res.path_status.addr),
|
||||
res.path_status.rtt/(double)1e3);
|
||||
|
||||
} while (res.ret == EAGAIN);
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
int
|
||||
gt_path(int argc, char **argv)
|
||||
{
|
||||
const char *dev = NULL;
|
||||
|
||||
struct ctl_msg req = {
|
||||
.type = CTL_STATE,
|
||||
}, res = {0};
|
||||
|
||||
struct argz pathz[] = {
|
||||
{NULL, "IPADDR", &req.path.addr, argz_addr},
|
||||
{"dev", "NAME", &dev, argz_str},
|
||||
{"up|backup|down", NULL, NULL, argz_option},
|
||||
{NULL}};
|
||||
|
||||
if (argz(pathz, argc, argv))
|
||||
return 1;
|
||||
|
||||
int fd = ctl_connect("/run/" PACKAGE_NAME, dev);
|
||||
|
||||
if (fd == -1) {
|
||||
perror("path");
|
||||
ctl_delete(fd);
|
||||
return 1;
|
||||
}
|
||||
|
||||
int ret = 0;
|
||||
|
||||
if (!req.path.addr.ss_family) {
|
||||
ret = gt_path_status(fd);
|
||||
|
||||
if (ret == -2)
|
||||
gt_log("bad reply from server\n");
|
||||
} else {
|
||||
if (argz_is_set(pathz, "up")) {
|
||||
req.path.state = MUD_UP;
|
||||
} else if (argz_is_set(pathz, "backup")) {
|
||||
req.path.state = MUD_BACKUP;
|
||||
} else if (argz_is_set(pathz, "down")) {
|
||||
req.path.state = MUD_DOWN;
|
||||
}
|
||||
|
||||
if (req.path.state)
|
||||
ret = ctl_reply(fd, &res, &req);
|
||||
}
|
||||
|
||||
if (ret == -1)
|
||||
perror("path");
|
||||
|
||||
ctl_delete(fd);
|
||||
|
||||
return 0;
|
||||
}
|
||||
154
src/set.c
Normal file
154
src/set.c
Normal file
@@ -0,0 +1,154 @@
|
||||
#include "common.h"
|
||||
#include "ctl.h"
|
||||
#include "str.h"
|
||||
|
||||
#include <stdio.h>
|
||||
#include <sys/socket.h>
|
||||
|
||||
#include "../argz/argz.h"
|
||||
|
||||
static int
|
||||
gt_set_mtu(int fd, size_t mtu)
|
||||
{
|
||||
struct ctl_msg res, req = {
|
||||
.type = CTL_MTU,
|
||||
.mtu = mtu,
|
||||
};
|
||||
|
||||
int ret = ctl_reply(fd, &res, &req);
|
||||
|
||||
if (ret) {
|
||||
perror("set mtu");
|
||||
return 1;
|
||||
}
|
||||
|
||||
printf("mtu set to %i\n", res.mtu);
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static int
|
||||
gt_set_timeout(int fd, unsigned long timeout)
|
||||
{
|
||||
struct ctl_msg res, req = {
|
||||
.type = CTL_TIMEOUT,
|
||||
.timeout = timeout,
|
||||
};
|
||||
|
||||
int ret = ctl_reply(fd, &res, &req);
|
||||
|
||||
if (ret) {
|
||||
perror("set timeout");
|
||||
return 1;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static int
|
||||
gt_set_timetolerance(int fd, unsigned long timetolerance)
|
||||
{
|
||||
struct ctl_msg res, req = {
|
||||
.type = CTL_TIMETOLERANCE,
|
||||
.timetolerance = timetolerance,
|
||||
};
|
||||
|
||||
int ret = ctl_reply(fd, &res, &req);
|
||||
|
||||
if (ret) {
|
||||
perror("set timetolerance");
|
||||
return 1;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static int
|
||||
gt_set_tc(int fd, int tc)
|
||||
{
|
||||
struct ctl_msg res, req = {
|
||||
.type = CTL_TC,
|
||||
.tc = tc,
|
||||
};
|
||||
|
||||
int ret = ctl_reply(fd, &res, &req);
|
||||
|
||||
if (ret) {
|
||||
perror("set tc");
|
||||
return 1;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static int
|
||||
gt_argz_tc(void *data, int argc, char **argv)
|
||||
{
|
||||
if (argc < 1 || !argv[0])
|
||||
return -1;
|
||||
|
||||
int val = 0;
|
||||
const char *s = argv[0];
|
||||
|
||||
if ((s[0] == 'C') && (s[1] == 'S') &&
|
||||
(s[2] >= '0') && (s[2] <= '7') && !s[3]) {
|
||||
val = (s[2] - '0') << 3;
|
||||
} else if ((s[0] == 'A') && (s[1] == 'F') &&
|
||||
(s[2] >= '1') && (s[2] <= '4') &&
|
||||
(s[3] >= '1') && (s[3] <= '3') && !s[4]) {
|
||||
val = ((s[2] - '0') << 3) | ((s[3] - '0') << 1);
|
||||
} else if ((s[0] == 'E') && (s[1] == 'F') && !s[2]) {
|
||||
val = 46;
|
||||
} else return -1;
|
||||
|
||||
if (data)
|
||||
*(int *)data = val;
|
||||
|
||||
return 1;
|
||||
}
|
||||
|
||||
int
|
||||
gt_set(int argc, char **argv)
|
||||
{
|
||||
const char *dev = NULL;
|
||||
size_t mtu;
|
||||
int tc;
|
||||
unsigned long timetolerance;
|
||||
unsigned long timeout;
|
||||
|
||||
struct argz pathz[] = {
|
||||
{"dev", "NAME", &dev, argz_str},
|
||||
{"mtu", "BYTES", &mtu, argz_bytes},
|
||||
{"tc", "CS|AF|EF", &tc, gt_argz_tc},
|
||||
{"timeout", "SECONDS", &timeout, argz_time},
|
||||
{"timetolerance", "SECONDS", &timetolerance, argz_time},
|
||||
{NULL}};
|
||||
|
||||
if (argz(pathz, argc, argv))
|
||||
return 1;
|
||||
|
||||
int fd = ctl_connect("/run/" PACKAGE_NAME, dev);
|
||||
|
||||
if (fd == -1) {
|
||||
perror("set");
|
||||
return 1;
|
||||
}
|
||||
|
||||
int ret = 0;
|
||||
|
||||
if (argz_is_set(pathz, "mtu"))
|
||||
ret |= gt_set_mtu(fd, mtu);
|
||||
|
||||
if (argz_is_set(pathz, "tc"))
|
||||
ret |= gt_set_tc(fd, tc);
|
||||
|
||||
if (argz_is_set(pathz, "timeout"))
|
||||
ret |= gt_set_timeout(fd, timeout);
|
||||
|
||||
if (argz_is_set(pathz, "timetolerance"))
|
||||
ret |= gt_set_timetolerance(fd, timetolerance);
|
||||
|
||||
ctl_delete(fd);
|
||||
|
||||
return ret;
|
||||
}
|
||||
115
src/show.c
Normal file
115
src/show.c
Normal file
@@ -0,0 +1,115 @@
|
||||
#include "common.h"
|
||||
#include "ctl.h"
|
||||
#include "str.h"
|
||||
|
||||
#include "../argz/argz.h"
|
||||
|
||||
#include <stdio.h>
|
||||
#include <sys/socket.h>
|
||||
#include <dirent.h>
|
||||
#include <sys/un.h>
|
||||
#include <arpa/inet.h>
|
||||
|
||||
static int
|
||||
gt_show_dev_status(int fd, const char *dev)
|
||||
{
|
||||
struct ctl_msg res, req = {.type = CTL_STATUS};
|
||||
|
||||
if (ctl_reply(fd, &res, &req))
|
||||
return -1;
|
||||
|
||||
char bindstr[INET6_ADDRSTRLEN] = {0};
|
||||
char peerstr[INET6_ADDRSTRLEN] = {0};
|
||||
|
||||
if (gt_toaddr(bindstr, sizeof(bindstr),
|
||||
(struct sockaddr *)&res.status.bind))
|
||||
return -2;
|
||||
|
||||
int server = gt_toaddr(peerstr, sizeof(peerstr),
|
||||
(struct sockaddr *)&res.status.peer);
|
||||
|
||||
if (server) {
|
||||
printf("server %s:\n"
|
||||
" bind: %s port %"PRIu16"\n"
|
||||
" mtu: %zu\n"
|
||||
" cipher: %s\n",
|
||||
dev,
|
||||
bindstr, gt_get_port((struct sockaddr *)&res.status.bind),
|
||||
res.status.mtu,
|
||||
res.status.chacha ? "chacha20poly1305" : "aes256gcm");
|
||||
} else {
|
||||
printf("client %s:\n"
|
||||
" bind: %s port %"PRIu16"\n"
|
||||
" peer: %s port %"PRIu16"\n"
|
||||
" mtu: %zu\n"
|
||||
" cipher: %s\n",
|
||||
dev,
|
||||
bindstr, gt_get_port((struct sockaddr *)&res.status.bind),
|
||||
peerstr, gt_get_port((struct sockaddr *)&res.status.peer),
|
||||
res.status.mtu,
|
||||
res.status.chacha ? "chacha20poly1305" : "aes256gcm");
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static int
|
||||
gt_show_dev(const char *dev)
|
||||
{
|
||||
int fd = ctl_connect("/run/" PACKAGE_NAME, dev);
|
||||
|
||||
if (fd == -1) {
|
||||
perror(dev);
|
||||
return -1;
|
||||
}
|
||||
|
||||
int ret = gt_show_dev_status(fd, dev);
|
||||
|
||||
if (ret == -1)
|
||||
perror(dev);
|
||||
|
||||
if (ret == -2)
|
||||
gt_log("%s: bad reply from server\n", dev);
|
||||
|
||||
ctl_delete(fd);
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
int
|
||||
gt_show(int argc, char **argv)
|
||||
{
|
||||
const char *dev = NULL;
|
||||
|
||||
struct argz showz[] = {
|
||||
{"dev", "NAME", &dev, argz_str},
|
||||
{NULL}};
|
||||
|
||||
if (argz(showz, argc, argv))
|
||||
return 1;
|
||||
|
||||
if (dev) {
|
||||
gt_show_dev(dev);
|
||||
return 0;
|
||||
}
|
||||
|
||||
DIR *dp = opendir("/run/" PACKAGE_NAME);
|
||||
|
||||
if (!dp) {
|
||||
if (errno == ENOENT)
|
||||
return 0;
|
||||
perror("show");
|
||||
return 1;
|
||||
}
|
||||
|
||||
struct dirent *d = NULL;
|
||||
|
||||
while (d = readdir(dp), d) {
|
||||
if (d->d_name[0] != '.')
|
||||
gt_show_dev(d->d_name);
|
||||
}
|
||||
|
||||
closedir(dp);
|
||||
|
||||
return 0;
|
||||
}
|
||||
55
src/str.h
55
src/str.h
@@ -2,22 +2,6 @@
|
||||
|
||||
#include "common.h"
|
||||
|
||||
static inline size_t
|
||||
str_cpy(char *restrict dst, const char *restrict src, size_t len)
|
||||
{
|
||||
if (!dst || !src)
|
||||
return 0;
|
||||
|
||||
size_t i;
|
||||
|
||||
for (i = 0; i < len && src[i]; i++)
|
||||
dst[i] = src[i];
|
||||
|
||||
dst[i] = 0;
|
||||
|
||||
return i;
|
||||
}
|
||||
|
||||
_pure_ static inline int
|
||||
str_empty(const char *restrict str)
|
||||
{
|
||||
@@ -40,36 +24,39 @@ str_cmp(const char *restrict sa, const char *restrict sb)
|
||||
}
|
||||
|
||||
_pure_ static inline size_t
|
||||
str_len(const char *restrict str)
|
||||
str_len(const char *restrict str, size_t len)
|
||||
{
|
||||
if (!str)
|
||||
return 0;
|
||||
|
||||
return strlen(str);
|
||||
return strnlen(str, len);
|
||||
}
|
||||
|
||||
static inline char *
|
||||
str_cat(const char **strs, size_t count)
|
||||
static inline size_t
|
||||
str_cat(char *dst, size_t dst_len, const char **src, size_t count)
|
||||
{
|
||||
size_t size = 1;
|
||||
if (count && !src)
|
||||
return 0;
|
||||
|
||||
for (size_t i = 0; i < count; i++)
|
||||
size += str_len(strs[i]);
|
||||
size_t len = 0;
|
||||
|
||||
char *str = malloc(size);
|
||||
for (size_t i = 0; i < count && dst_len > len; i++) {
|
||||
size_t n = str_len(src[i], dst_len - len);
|
||||
|
||||
if (!str)
|
||||
return NULL;
|
||||
if (dst && n)
|
||||
memmove(&dst[len], src[i], n);
|
||||
|
||||
char *p = str;
|
||||
|
||||
for (size_t i = 0; i < count; i++) {
|
||||
size_t len = str_len(strs[i]);
|
||||
memcpy(p, strs[i], len);
|
||||
p += len;
|
||||
len += n;
|
||||
}
|
||||
|
||||
p[0] = 0;
|
||||
if (dst)
|
||||
dst[len] = 0;
|
||||
|
||||
return str;
|
||||
return len;
|
||||
}
|
||||
|
||||
static inline size_t
|
||||
str_cpy(char *dst, size_t dst_len, const char *src)
|
||||
{
|
||||
return str_cat(dst, dst_len, &src, 1);
|
||||
}
|
||||
|
||||
119
src/tun.c
119
src/tun.c
@@ -1,22 +1,20 @@
|
||||
#include "common.h"
|
||||
|
||||
#include "ip.h"
|
||||
#include "str.h"
|
||||
#include "tun.h"
|
||||
|
||||
#include <fcntl.h>
|
||||
#include <stdio.h>
|
||||
|
||||
#include <sys/ioctl.h>
|
||||
#include <sys/socket.h>
|
||||
#include <sys/uio.h>
|
||||
|
||||
#include <net/if.h>
|
||||
|
||||
#ifdef __linux__
|
||||
#define IFF_TUN 0x0001
|
||||
#define IFF_NO_PI 0x1000
|
||||
#define TUNSETIFF _IOW('T', 202, int)
|
||||
#define TUNSETPERSIST _IOW('T', 203, int)
|
||||
#endif
|
||||
|
||||
#ifdef __APPLE__
|
||||
@@ -32,17 +30,22 @@
|
||||
#ifdef __APPLE__
|
||||
|
||||
static int
|
||||
tun_create_by_id(char *name, size_t size, unsigned id)
|
||||
tun_create_by_id(char *name, size_t len, unsigned id)
|
||||
{
|
||||
int ret = snprintf(name, len + 1, "utun%u", id);
|
||||
|
||||
if (ret <= 0 || ret > len) {
|
||||
errno = EINVAL;
|
||||
return -1;
|
||||
}
|
||||
|
||||
int fd = socket(PF_SYSTEM, SOCK_DGRAM, SYSPROTO_CONTROL);
|
||||
|
||||
if (fd == -1)
|
||||
return -1;
|
||||
|
||||
struct ctl_info ci;
|
||||
|
||||
memset(&ci, 0, sizeof(ci));
|
||||
str_cpy(ci.ctl_name, UTUN_CONTROL_NAME, sizeof(ci.ctl_name) - 1);
|
||||
struct ctl_info ci = {0};
|
||||
str_cpy(ci.ctl_name, sizeof(ci.ctl_name) - 1, UTUN_CONTROL_NAME);
|
||||
|
||||
if (ioctl(fd, CTLIOCGINFO, &ci)) {
|
||||
int err = errno;
|
||||
@@ -66,13 +69,11 @@ tun_create_by_id(char *name, size_t size, unsigned id)
|
||||
return -1;
|
||||
}
|
||||
|
||||
snprintf(name, size, "utun%u", id);
|
||||
|
||||
return fd;
|
||||
}
|
||||
|
||||
static int
|
||||
tun_create_by_name(char *name, size_t size, char *dev_name)
|
||||
tun_create_by_name(char *name, size_t len, const char *dev_name)
|
||||
{
|
||||
unsigned id = 0;
|
||||
|
||||
@@ -81,7 +82,7 @@ tun_create_by_name(char *name, size_t size, char *dev_name)
|
||||
return -1;
|
||||
}
|
||||
|
||||
return tun_create_by_id(name, size, id);
|
||||
return tun_create_by_id(name, len, id);
|
||||
}
|
||||
|
||||
#else /* not __APPLE__ */
|
||||
@@ -89,25 +90,35 @@ tun_create_by_name(char *name, size_t size, char *dev_name)
|
||||
#ifdef __linux__
|
||||
|
||||
static int
|
||||
tun_create_by_name(char *name, size_t size, char *dev_name)
|
||||
tun_create_by_name(char *name, size_t len, const char *dev_name)
|
||||
{
|
||||
struct ifreq ifr = {
|
||||
.ifr_flags = IFF_TUN | IFF_NO_PI,
|
||||
};
|
||||
|
||||
const size_t ifr_len = sizeof(ifr.ifr_name) - 1;
|
||||
|
||||
if ((len < ifr_len) ||
|
||||
(str_len(dev_name, ifr_len + 1) > ifr_len)) {
|
||||
errno = EINVAL;
|
||||
return -1;
|
||||
}
|
||||
|
||||
int fd = open("/dev/net/tun", O_RDWR);
|
||||
|
||||
if (fd == -1)
|
||||
return -1;
|
||||
|
||||
struct ifreq ifr = {
|
||||
.ifr_flags = IFF_TUN | IFF_NO_PI,
|
||||
};
|
||||
|
||||
str_cpy(ifr.ifr_name, dev_name, IFNAMSIZ - 1);
|
||||
str_cpy(ifr.ifr_name, ifr_len, dev_name);
|
||||
|
||||
if (ioctl(fd, TUNSETIFF, &ifr)) {
|
||||
int err = errno;
|
||||
close(fd);
|
||||
errno = err;
|
||||
return -1;
|
||||
}
|
||||
|
||||
str_cpy(name, ifr.ifr_name, size - 1);
|
||||
str_cpy(name, len, ifr.ifr_name);
|
||||
|
||||
return fd;
|
||||
}
|
||||
@@ -115,46 +126,58 @@ tun_create_by_name(char *name, size_t size, char *dev_name)
|
||||
#else /* not __linux__ not __APPLE__ */
|
||||
|
||||
static int
|
||||
tun_create_by_name(char *name, size_t size, char *dev_name)
|
||||
tun_create_by_name(char *name, size_t len, const char *dev_name)
|
||||
{
|
||||
char path[64];
|
||||
char tmp[128];
|
||||
|
||||
snprintf(path, sizeof(path), "/dev/%s", dev_name);
|
||||
str_cpy(name, dev_name, size - 1);
|
||||
int ret = snprintf(tmp, sizeof(tmp), "/dev/%s", dev_name);
|
||||
|
||||
return open(path, O_RDWR);
|
||||
if (ret <= 0 || ret >= sizeof(tmp)) {
|
||||
errno = EINVAL;
|
||||
return -1;
|
||||
}
|
||||
|
||||
if (str_cpy(name, len, dev_name) == len) {
|
||||
if (str_len(dev_name, len + 1) > len) {
|
||||
errno = EINVAL;
|
||||
return -1;
|
||||
}
|
||||
}
|
||||
|
||||
return open(tmp, O_RDWR);
|
||||
}
|
||||
|
||||
#endif /* not __APPLE__ */
|
||||
|
||||
static int
|
||||
tun_create_by_id(char *name, size_t size, unsigned id)
|
||||
tun_create_by_id(char *name, size_t len, unsigned id)
|
||||
{
|
||||
char dev_name[64];
|
||||
char tmp[64];
|
||||
|
||||
snprintf(dev_name, sizeof(dev_name), "tun%u", id);
|
||||
int ret = snprintf(tmp, sizeof(tmp), "tun%u", id);
|
||||
|
||||
return tun_create_by_name(name, size, dev_name);
|
||||
if (ret <= 0 || ret >= sizeof(tmp)) {
|
||||
errno = EINVAL;
|
||||
return -1;
|
||||
}
|
||||
|
||||
return tun_create_by_name(name, len, tmp);
|
||||
}
|
||||
|
||||
#endif
|
||||
|
||||
int
|
||||
tun_create(char *dev_name, char **ret_name)
|
||||
tun_create(char *name, size_t len, const char *dev_name)
|
||||
{
|
||||
char name[64] = {0};
|
||||
int fd = -1;
|
||||
|
||||
if (str_empty(dev_name)) {
|
||||
for (unsigned id = 0; id < 32 && fd == -1; id++)
|
||||
fd = tun_create_by_id(name, sizeof(name), id);
|
||||
fd = tun_create_by_id(name, len, id);
|
||||
} else {
|
||||
fd = tun_create_by_name(name, sizeof(name), dev_name);
|
||||
fd = tun_create_by_name(name, len, dev_name);
|
||||
}
|
||||
|
||||
if (fd != -1 && ret_name)
|
||||
*ret_name = strdup(name);
|
||||
|
||||
return fd;
|
||||
}
|
||||
|
||||
@@ -239,24 +262,12 @@ tun_write(int fd, const void *data, size_t size)
|
||||
}
|
||||
|
||||
int
|
||||
tun_set_mtu(char *dev_name, int mtu)
|
||||
tun_set_persist(int fd, int on)
|
||||
{
|
||||
struct ifreq ifr = {
|
||||
.ifr_mtu = mtu,
|
||||
};
|
||||
|
||||
str_cpy(ifr.ifr_name, dev_name, IFNAMSIZ - 1);
|
||||
|
||||
int fd = socket(AF_INET, SOCK_DGRAM, 0);
|
||||
|
||||
if (fd == -1)
|
||||
return -1;
|
||||
|
||||
int ret = ioctl(fd, SIOCSIFMTU, &ifr);
|
||||
|
||||
int err = errno;
|
||||
close(fd);
|
||||
errno = err;
|
||||
|
||||
return ret;
|
||||
#ifdef TUNSETPERSIST
|
||||
return ioctl(fd, TUNSETPERSIST, on);
|
||||
#else
|
||||
errno = ENOSYS;
|
||||
return -1;
|
||||
#endif
|
||||
}
|
||||
|
||||
@@ -1,6 +1,6 @@
|
||||
#pragma once
|
||||
|
||||
int tun_create (char *, char **);
|
||||
int tun_read (int, void *, size_t);
|
||||
int tun_write (int, const void *, size_t);
|
||||
int tun_set_mtu (char *, int);
|
||||
int tun_create (char *, size_t, const char *);
|
||||
int tun_read (int, void *, size_t);
|
||||
int tun_write (int, const void *, size_t);
|
||||
int tun_set_persist (int, int);
|
||||
|
||||
10
systemd/glorytun-client.network
Normal file
10
systemd/glorytun-client.network
Normal file
@@ -0,0 +1,10 @@
|
||||
[Match]
|
||||
Name=gtc-*
|
||||
|
||||
[Network]
|
||||
Description=Glorytun client device
|
||||
DHCP=ipv4
|
||||
|
||||
[DHCP]
|
||||
CriticalConnection=yes
|
||||
RouteTable=200
|
||||
6
systemd/glorytun-run
Executable file
6
systemd/glorytun-run
Executable file
@@ -0,0 +1,6 @@
|
||||
#!/bin/sh
|
||||
|
||||
exec glorytun bind "$@" \
|
||||
$BIND $BIND_PORT \
|
||||
${DEV:+dev "$DEV"} \
|
||||
${HOST:+to "$HOST" "$PORT"}
|
||||
73
systemd/glorytun-setup
Executable file
73
systemd/glorytun-setup
Executable file
@@ -0,0 +1,73 @@
|
||||
#!/bin/sh
|
||||
|
||||
set -e
|
||||
|
||||
_ask() {
|
||||
printf "%s: " "$1"
|
||||
read -r "$2"
|
||||
}
|
||||
|
||||
_ask "Config filename (tun0)" NAME
|
||||
NAME=${NAME:-tun0}
|
||||
DIR="/etc/glorytun/$NAME"
|
||||
|
||||
if [ -d "$DIR" ]; then
|
||||
echo "This config already exit!"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
_ask "Server ip (enter for server conf)" HOST
|
||||
if [ -z "$HOST" ]; then
|
||||
_ask "Bind to port (5000)" BIND_PORT
|
||||
BIND_PORT=${BIND_PORT:-5000}
|
||||
else
|
||||
_ask "Server port (5000)" PORT
|
||||
PORT=${PORT:-5000}
|
||||
fi
|
||||
|
||||
BIND=0.0.0.0
|
||||
case "$HOST" in
|
||||
*:*) BIND=::
|
||||
esac
|
||||
|
||||
_ask "Server key (enter to generate a new one)" KEY
|
||||
if [ -z "$KEY" ]; then
|
||||
KEY=$(glorytun keygen)
|
||||
echo "Your new key: $KEY"
|
||||
fi
|
||||
|
||||
# install files
|
||||
mkdir -p "$DIR"
|
||||
|
||||
cat > "$DIR/env" <<EOF
|
||||
DEV=gt${HOST:+c}-$NAME
|
||||
HOST=$HOST
|
||||
PORT=$PORT
|
||||
BIND=$BIND
|
||||
BIND_PORT=$BIND_PORT
|
||||
OPTIONS=
|
||||
EOF
|
||||
|
||||
( umask 077; echo "$KEY" > "$DIR/key" )
|
||||
|
||||
[ "$HOST" ] && cat > "$DIR/post.sh" <<'EOF'
|
||||
#!/bin/sh
|
||||
|
||||
PREF=32765
|
||||
TABLE=200
|
||||
|
||||
# keep the current route to HOST
|
||||
SRC=$(ip route get "$HOST" | awk '/src/{getline;print $0}' RS=' ')
|
||||
ip rule add from "$SRC" table main pref "$((PREF-1))" || true
|
||||
glorytun path up "$SRC" dev "$DEV"
|
||||
|
||||
# forward everything else to the tunnel
|
||||
ip rule add from all table "$TABLE" pref "$PREF" || true
|
||||
EOF
|
||||
[ -f "$DIR/post.sh" ] && chmod u+x "$DIR/post.sh"
|
||||
|
||||
# start services
|
||||
_ask "Start glorytun now ? (enter to skip)" START
|
||||
case "$START" in y*|Y*)
|
||||
systemctl start glorytun@"$NAME" ;;
|
||||
esac
|
||||
14
systemd/glorytun.network
Normal file
14
systemd/glorytun.network
Normal file
@@ -0,0 +1,14 @@
|
||||
[Match]
|
||||
Name=gt-*
|
||||
|
||||
[Network]
|
||||
Description=Glorytun server device
|
||||
Address=0.0.0.0/24
|
||||
DHCPServer=yes
|
||||
IPMasquerade=yes
|
||||
|
||||
[DHCPServer]
|
||||
PoolOffset=2
|
||||
PoolSize=1
|
||||
EmitDNS=yes
|
||||
DNS=9.9.9.9
|
||||
14
systemd/glorytun@.service.in
Normal file
14
systemd/glorytun@.service.in
Normal file
@@ -0,0 +1,14 @@
|
||||
[Unit]
|
||||
Description=Glorytun on %I
|
||||
After=network.target
|
||||
|
||||
[Service]
|
||||
Type=simple
|
||||
Restart=always
|
||||
EnvironmentFile=/etc/glorytun/%i/env
|
||||
ExecStart=@bindir@/glorytun-run keyfile /etc/glorytun/%i/key $OPTIONS
|
||||
ExecStartPost=-/etc/glorytun/%i/post.sh
|
||||
CapabilityBoundingSet=CAP_NET_ADMIN
|
||||
|
||||
[Install]
|
||||
WantedBy=multi-user.target
|
||||
Reference in New Issue
Block a user