Compare commits

...

28 Commits

Author SHA1 Message Date
angt
35a9bf27df Add state INITIALIZED 2015-12-31 16:07:36 +01:00
angt
ba0af8cc20 Define VERSION_MAJOR and use it in handshake 2015-12-29 18:31:23 +01:00
angt
ec85be5c6a Code cleanup 2015-12-29 12:59:55 +01:00
angt
fe989851ab Print more debug info 2015-12-29 12:58:39 +01:00
angt
46842dd200 Write state after the close() 2015-12-28 07:18:00 +01:00
angt
e5eb30598d Code cleanup 2015-12-24 17:38:08 +01:00
angt
c81ad0a7c6 One more time 2015-12-24 17:35:01 +01:00
angt
97641d6dda Reserve exclam for very bad message 2015-12-24 17:32:32 +01:00
angt
e48dac775c The statefile option needs an absolute path and a fifo 2015-12-24 17:29:41 +01:00
angt
6282f36ac7 Add statefile option 2015-12-24 16:53:50 +01:00
angt
cf022af4a9 Simplify the db by merging size and mask 2015-12-24 13:13:43 +01:00
angt
597c586657 Don't free data in db_remove() 2015-12-23 22:26:53 +01:00
angt
fc7f9aa0c8 Add a very simple low cost database 2015-12-23 17:11:20 +01:00
angt
7492f977b6 Define a simple proto 2015-12-21 12:34:24 +01:00
angt
38b7333533 Refuse to start the client without keyfile 2015-12-18 16:07:14 +01:00
angt
1ab854f058 Code cleanup 2015-12-18 11:57:20 +01:00
angt
c59def90fb Try to use SO_ACCEPTFILTER to defer accept on non-linux platforms 2015-12-17 23:20:31 +01:00
angt
347a3ecce7 Try to open tunX instead of glorytun by default in Linux 2015-12-17 17:06:04 +01:00
angt
98d13ef510 Print the tun name, not the path 2015-12-17 17:01:31 +01:00
angt
35b95001c4 Warn only when tun_create() fail 2015-12-17 14:07:14 +01:00
angt
75c12b36d6 Code cleanup 2015-12-17 13:59:40 +01:00
angt
db01c8b33f Update README.md 2015-12-17 08:51:07 +01:00
angt
fb4f6f6cb9 Do the md5sum after deploy 2015-12-16 11:27:53 +01:00
angt
a86aea431d Let travis do the md5sum 2015-12-16 08:35:20 +01:00
angt
1a128a6d92 Print readable ip in debug mode 2015-12-15 17:32:03 +01:00
angt
ddae22a3d9 Use a new random secret key without keyfile 2015-12-15 09:07:44 +01:00
angt
893de45272 Accept lower-case in fromhex() 2015-12-15 08:28:10 +01:00
angt
78ba4c9a59 Do not call ip_get_version() again and again 2015-12-14 18:37:50 +01:00
13 changed files with 483 additions and 150 deletions

View File

@@ -30,3 +30,6 @@ deploy:
tags: true tags: true
repo: angt/glorytun repo: angt/glorytun
condition: "${TRAVIS_OS_NAME}-${CC} == linux-gcc" condition: "${TRAVIS_OS_NAME}-${CC} == linux-gcc"
after_deploy:
- md5sum glorytun-*.tar.gz

View File

@@ -13,7 +13,9 @@ glorytun_SOURCES = \
src/option.c \ src/option.c \
src/option.h \ src/option.h \
src/tun.c \ src/tun.c \
src/tun.h src/tun.h \
src/db.c \
src/db.h
EXTRA_DIST = \ EXTRA_DIST = \
LICENSE \ LICENSE \

View File

@@ -1,8 +1,12 @@
# Glorytun # Glorytun
Small, Simple and Stupid **TCP** VPN. Small, Simple and Stupid TCP VPN.
**Work In Progress:** Do not touch! This code will probably format your harddisk! #### Work In Progress
This code will probably format your harddisk!
#### Build and Install
Glorytun depends on [libsodium](https://github.com/jedisct1/libsodium) version >= 1.0.4 Glorytun depends on [libsodium](https://github.com/jedisct1/libsodium) version >= 1.0.4
and needs an AES-NI capable CPU. and needs an AES-NI capable CPU.
@@ -16,7 +20,4 @@ To build and install the latest version:
$ make $ make
# make install # make install
To create and use a new secret key: For feature requests and bug reports, please create an [issue](https://github.com/angt/glorytun/issues).
$ hexdump -n 32 /dev/urandom -e '"%X"' > glorytun.key
# glorytun keyfile glorytun.key [...]

View File

@@ -4,6 +4,7 @@ AC_INIT([glorytun],
[https://github.com/angt/glorytun/issues], [https://github.com/angt/glorytun/issues],
[glorytun], [glorytun],
[https://github.com/angt/glorytun]) [https://github.com/angt/glorytun])
AC_DEFINE_UNQUOTED([VERSION_MAJOR], [m4_esyscmd([./version.sh major])])
AC_CONFIG_SRCDIR([src/common.h]) AC_CONFIG_SRCDIR([src/common.h])
AC_CONFIG_AUX_DIR([build-aux]) AC_CONFIG_AUX_DIR([build-aux])
AC_CONFIG_MACRO_DIR([m4]) AC_CONFIG_MACRO_DIR([m4])

View File

@@ -18,7 +18,7 @@ static inline void byte_set (void *dst, const char value, size_t size)
static inline void byte_cpy (void *dst, const void *src, size_t size) static inline void byte_cpy (void *dst, const void *src, size_t size)
{ {
if (!dst || !src) if (!dst)
return; return;
char *restrict d = dst; char *restrict d = dst;
@@ -44,16 +44,24 @@ static inline size_t str_cpy (char *restrict dst, const char *restrict src, size
} }
_pure_ _pure_
static inline int str_cmp (const char *restrict sa, const char *restrict sb) static inline int str_empty (const char *restrict str)
{
return !str || !str[0];
}
_pure_
static inline size_t str_cmp (const char *restrict sa, const char *restrict sb)
{ {
if (!sa || !sb) if (!sa || !sb)
return 1; return 1;
while (*sa==*sb++) size_t i = 0;
if (!*sa++)
return 0;
return 1; while (sa[i]==sb[i])
if (!sa[i++])
return 0;
return i+1;
} }
_pure_ _pure_

View File

@@ -36,7 +36,7 @@ void gt_fatal (const char *fmt, ...)
void gt_na (const char *name) void gt_na (const char *name)
{ {
gt_log("%s is not available on your platform!\n", 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) int gt_tohex (char *dst, size_t dst_size, const uint8_t *src, size_t src_size)
@@ -68,6 +68,9 @@ static inline int fromhex (const char c)
if (c>='A' && c<='F') if (c>='A' && c<='F')
return c-'A'+10; return c-'A'+10;
if (c>='a' && c<='f')
return c-'a'+10;
return -1; return -1;
} }

View File

@@ -17,12 +17,14 @@
#define _1_(x) (__builtin_expect((x), 1)) #define _1_(x) (__builtin_expect((x), 1))
#define _0_(x) (__builtin_expect((x), 0)) #define _0_(x) (__builtin_expect((x), 0))
#define _printf_(A,B) __attribute__((format(printf,A,B))) #define CLZ(x) (__builtin_clz(x))
#define _noreturn_ __attribute__((noreturn))
#define _unused_ __attribute__((unused)) #define _printf_(A,B) __attribute__ ((format(printf,A,B)))
#define _pure_ __attribute__((pure)) #define _noreturn_ __attribute__ ((noreturn))
#define _const_ __attribute__((const)) #define _unused_ __attribute__ ((unused))
#define _align_(...) __attribute__((aligned(__VA_ARGS__))) #define _pure_ __attribute__ ((pure))
#define _const_ __attribute__ ((const))
#define _align_(...) __attribute__ ((aligned(__VA_ARGS__)))
typedef struct buffer buffer_t; typedef struct buffer buffer_t;

153
src/db.c Normal file
View File

@@ -0,0 +1,153 @@
#include "db.h"
#include "common-static.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 Normal file
View File

@@ -0,0 +1,7 @@
#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 *);

View File

@@ -12,9 +12,9 @@ static inline int ip_get_version (const uint8_t *data, size_t size)
} }
_pure_ _pure_
static inline ssize_t ip_get_size (const uint8_t *data, size_t size) static inline ssize_t ip_get_size (const int ip_version, const uint8_t *data, size_t size)
{ {
switch (ip_get_version(data, size)) { switch (ip_version) {
case 4: case 4:
return ((data[2]<<8)|data[3]); return ((data[2]<<8)|data[3]);
case 6: case 6:
@@ -27,9 +27,9 @@ static inline ssize_t ip_get_size (const uint8_t *data, size_t size)
} }
_pure_ _pure_
static inline ssize_t ip_get_proto (const uint8_t *data, size_t size) static inline ssize_t ip_get_proto (const int ip_version, const uint8_t *data, size_t size)
{ {
switch (ip_get_version(data, size)) { switch (ip_version) {
case 4: case 4:
return data[9]; return data[9];
case 6: case 6:
@@ -42,9 +42,9 @@ static inline ssize_t ip_get_proto (const uint8_t *data, size_t size)
} }
_pure_ _pure_
static inline ssize_t ip_get_hdr_size (const uint8_t *data, size_t size) static inline ssize_t ip_get_hdr_size (const int ip_version, const uint8_t *data, size_t size)
{ {
switch (ip_get_version(data, size)) { switch (ip_version) {
case 4: case 4:
return (data[0]&0xF)<<2; return (data[0]&0xF)<<2;
case 6: case 6:

View File

@@ -2,18 +2,26 @@
#include <limits.h> #include <limits.h>
#include <stdio.h> #include <stdio.h>
#include <signal.h> #include <signal.h>
#include <string.h>
#include <poll.h> #include <poll.h>
#include <sys/time.h> #include <sys/time.h>
#include <sys/fcntl.h> #include <sys/fcntl.h>
#include <sys/stat.h>
#ifndef __FAVOR_BSD #ifndef __FAVOR_BSD
#define __FAVOR_BSD 1 #define __FAVOR_BSD
#define GT_FAKE_BSD
#endif #endif
#include <netinet/in.h> #include <netinet/in.h>
#include <netinet/tcp.h> #include <netinet/tcp.h>
#include <netinet/udp.h> #include <netinet/udp.h>
#ifdef GT_FAKE_BSD
#undef GT_FAKE_BSD
#undef __FAVOR_BSD
#endif
#include <arpa/inet.h> #include <arpa/inet.h>
#include <netdb.h> #include <netdb.h>
@@ -92,6 +100,7 @@ enum sk_opt {
sk_keepintvl, sk_keepintvl,
sk_congestion, sk_congestion,
sk_defer_accept, sk_defer_accept,
sk_acceptfilter,
sk_quickack, sk_quickack,
}; };
@@ -137,6 +146,11 @@ static void sk_set (int fd, enum sk_opt opt, const void *val, socklen_t len)
[sk_quickack] = { "TCP_QUICKACK", [sk_quickack] = { "TCP_QUICKACK",
#ifdef TCP_QUICKACK #ifdef TCP_QUICKACK
1, IPPROTO_TCP, TCP_QUICKACK, 1, IPPROTO_TCP, TCP_QUICKACK,
#endif
},
[sk_acceptfilter] = { "SO_ACCEPTFILTER",
#ifdef SO_ACCEPTFILTER
1, SOL_SOCKET, SO_ACCEPTFILTER,
#endif #endif
}, },
}; };
@@ -159,21 +173,23 @@ static int sk_listen (int fd, struct addrinfo *ai)
{ {
sk_set_int(fd, sk_reuseaddr, 1); sk_set_int(fd, sk_reuseaddr, 1);
int ret = bind(fd, ai->ai_addr, ai->ai_addrlen); if (bind(fd, ai->ai_addr, ai->ai_addrlen)==-1) {
if (ret==-1) {
perror("bind"); perror("bind");
return -1; return -1;
} }
ret = listen(fd, 8); if (listen(fd, 8)==-1) {
if (ret==-1) {
perror("listen"); perror("listen");
return -1; return -1;
} }
#ifdef __linux__
sk_set_int(fd, sk_defer_accept, GT_TIMEOUT/1000); sk_set_int(fd, sk_defer_accept, GT_TIMEOUT/1000);
#else
char data[256] = {0};
str_cpy(data, "dataready", sizeof(data)-1);
sk_set(fd, sk_acceptfilter, &data, sizeof(data));
#endif
return 0; return 0;
} }
@@ -327,7 +343,7 @@ static void gt_set_signal (void)
static ssize_t fd_read (int fd, void *data, size_t size) static ssize_t fd_read (int fd, void *data, size_t size)
{ {
if (!size) if ((fd==-1) || !size)
return -1; return -1;
ssize_t ret = read(fd, data, size); ssize_t ret = read(fd, data, size);
@@ -347,7 +363,7 @@ static ssize_t fd_read (int fd, void *data, size_t size)
static ssize_t fd_write (int fd, const void *data, size_t size) static ssize_t fd_write (int fd, const void *data, size_t size)
{ {
if (!size) if ((fd==-1) || !size)
return -1; return -1;
ssize_t ret = write(fd, data, size); ssize_t ret = write(fd, data, size);
@@ -368,6 +384,11 @@ static ssize_t fd_write (int fd, const void *data, size_t size)
return ret; return ret;
} }
static ssize_t fd_write_str (int fd, const char *str)
{
return fd_write(fd, str, str_len(str));
}
static size_t fd_read_all (int fd, void *data, size_t size) static size_t fd_read_all (int fd, void *data, size_t size)
{ {
size_t done = 0; size_t done = 0;
@@ -491,53 +512,83 @@ static int gt_decrypt (struct crypto_ctx *ctx, buffer_t *dst, buffer_t *src)
return 0; return 0;
} }
static void gt_print_hdr (uint8_t *data, size_t ip_size, const char *sockname) _pure_
static inline uint32_t sum16 (uint32_t sum, const uint8_t *data, const size_t size)
{ {
const int ip_version = ip_get_version(data, GT_MTU_MAX); const size_t lim = size&~1u;
const ssize_t ip_proto = ip_get_proto(data, GT_MTU_MAX);
const ssize_t ip_hdr_size = ip_get_hdr_size(data, GT_MTU_MAX);
char ip_src[33]; for (size_t i=0; i<lim; i+=2)
char ip_dst[33]; sum += (data[i]<<8)|data[i+1];
switch (ip_version) { if (size&1)
case 4: sum += data[size-1]<<8;
gt_tohex(ip_src, sizeof(ip_src), &data[12], 4);
gt_tohex(ip_dst, sizeof(ip_dst), &data[16], 4);
break;
case 6:
gt_tohex(ip_src, sizeof(ip_src), &data[9], 16);
gt_tohex(ip_dst, sizeof(ip_dst), &data[25], 16);
break;
}
gt_log("%s: version=%i size=%zi proto=%zi src=%s dst=%s\n", sockname, ip_version, ip_size, ip_proto, ip_src, ip_dst); return sum;
}
if (ip_hdr_size<=0) _const_
static inline uint16_t sum16_final (uint32_t sum)
{
sum = (sum>>16)+(sum&0xFFFF);
return ~(sum+(sum>>16));
}
static void gt_print_hdr (const int ip_version, uint8_t *data, size_t ip_size)
{
const ssize_t ip_proto = ip_get_proto(ip_version, data, ip_size);
const ssize_t ip_hdr_size = ip_get_hdr_size(ip_version, data, ip_size);
if (ip_proto<0 || ip_hdr_size<=0)
return; return;
if (ip_proto==6) { uint32_t sum = (size_t)ip_proto+ip_size-(size_t)ip_hdr_size;
char ip_src[INET6_ADDRSTRLEN];
char ip_dst[INET6_ADDRSTRLEN];
switch (ip_version) {
case 4:
inet_ntop(AF_INET, &data[12], ip_src, sizeof(ip_src));
inet_ntop(AF_INET, &data[16], ip_dst, sizeof(ip_dst));
sum = sum16(sum, &data[12], 2*4);
break;
case 6:
inet_ntop(AF_INET6, &data[9], ip_src, sizeof(ip_src));
inet_ntop(AF_INET6, &data[25], ip_dst, sizeof(ip_dst));
sum = sum16(sum, &data[9], 2*16); // XXX
break;
}
if (ip_proto==IPPROTO_TCP) {
struct tcphdr tcp; struct tcphdr tcp;
byte_cpy(&tcp, &data[ip_hdr_size], sizeof(tcp)); byte_cpy(&tcp, &data[ip_hdr_size], sizeof(tcp));
uint16_t tcp_sum = ntohs(tcp.th_sum);
tcp.th_sum = 0;
sum = sum16(sum, (uint8_t *)&tcp, sizeof(tcp));
sum = sum16(sum, &data[ip_hdr_size+sizeof(tcp)], ip_size-ip_hdr_size-sizeof(tcp));
uint16_t computed_sum = sum16_final(sum);
tcp.th_sport = ntohs(tcp.th_sport); tcp.th_sport = ntohs(tcp.th_sport);
tcp.th_dport = ntohs(tcp.th_dport); tcp.th_dport = ntohs(tcp.th_dport);
tcp.th_seq = ntohl(tcp.th_seq); tcp.th_seq = ntohl(tcp.th_seq);
tcp.th_ack = ntohl(tcp.th_ack); tcp.th_ack = ntohl(tcp.th_ack);
tcp.th_win = ntohs(tcp.th_win); tcp.th_win = ntohs(tcp.th_win);
gt_log("%s: tcp src=%u dst=%u seq=%u ack=%u win=%u %c%c%c%c%c%c\n", gt_print("proto:%zi\tsrc:%s.%u\tdst:%s.%u\tseq:%u\tack:%u\twin:%u\tsize:%zu\tflags:%c%c%c%c%c%c\tsum:%i\n",
sockname, tcp.th_sport, tcp.th_dport, tcp.th_seq, tcp.th_ack, tcp.th_win, ip_proto, ip_src, tcp.th_sport, ip_dst, tcp.th_dport,
tcp.th_seq, tcp.th_ack, tcp.th_win, ip_size-ip_hdr_size+tcp.th_off*4,
(tcp.th_flags&TH_FIN) ?'F':'.', (tcp.th_flags&TH_FIN) ?'F':'.',
(tcp.th_flags&TH_SYN) ?'S':'.', (tcp.th_flags&TH_SYN) ?'S':'.',
(tcp.th_flags&TH_RST) ?'R':'.', (tcp.th_flags&TH_RST) ?'R':'.',
(tcp.th_flags&TH_PUSH)?'P':'.', (tcp.th_flags&TH_PUSH)?'P':'.',
(tcp.th_flags&TH_ACK) ?'A':'.', (tcp.th_flags&TH_ACK) ?'A':'.',
(tcp.th_flags&TH_URG) ?'U':'.'); (tcp.th_flags&TH_URG) ?'U':'.',
} (computed_sum==tcp_sum));
if (ip_proto==17) { } else if (ip_proto==IPPROTO_UDP) {
struct udphdr udp; struct udphdr udp;
byte_cpy(&udp, &data[ip_hdr_size], sizeof(udp)); byte_cpy(&udp, &data[ip_hdr_size], sizeof(udp));
@@ -546,8 +597,11 @@ static void gt_print_hdr (uint8_t *data, size_t ip_size, const char *sockname)
udp.uh_dport = ntohs(udp.uh_dport); udp.uh_dport = ntohs(udp.uh_dport);
udp.uh_ulen = ntohs(udp.uh_ulen); udp.uh_ulen = ntohs(udp.uh_ulen);
gt_log("%s: udp src=%u dst=%u len=%u\n", gt_print("proto:%zi\tsrc:%s.%u\tdst:%s.%u\tsize:%u\n",
sockname, udp.uh_sport, udp.uh_dport, udp.uh_ulen); ip_proto, ip_src, udp.uh_sport, ip_dst, udp.uh_dport, udp.uh_ulen-8);
} else {
gt_print("proto:%zi\tsrc:%s\tdst:%s\tsize:%zu\n",
ip_proto, ip_src, ip_dst, ip_size);
} }
} }
@@ -555,10 +609,16 @@ static int gt_setup_secretkey (struct crypto_ctx *ctx, char *keyfile)
{ {
const size_t size = sizeof(ctx->skey); const size_t size = sizeof(ctx->skey);
byte_set(ctx->skey, 1, size); if (!keyfile) {
char buf[2*size+1];
randombytes_buf(ctx->skey, size);
gt_tohex(buf, sizeof(buf), ctx->skey, size);
gt_print("new secret key: %s\n", buf);
if (!keyfile)
return 0; return 0;
}
int fd; int fd;
@@ -591,10 +651,13 @@ static int gt_setup_secretkey (struct crypto_ctx *ctx, char *keyfile)
static int gt_setup_crypto (struct crypto_ctx *ctx, int fd, int listener) static int gt_setup_crypto (struct crypto_ctx *ctx, int fd, int listener)
{ {
const uint8_t gt[] = {'G', 'T', VERSION_MAJOR, 0 };
const size_t size = 96;
const size_t hash_size = 32;
const size_t nonce_size = crypto_aead_aes256gcm_NPUBBYTES; const size_t nonce_size = crypto_aead_aes256gcm_NPUBBYTES;
const size_t public_size = crypto_scalarmult_SCALARBYTES; const size_t public_size = crypto_scalarmult_SCALARBYTES;
const size_t hash_size = crypto_generichash_BYTES;
const size_t size = nonce_size + public_size + hash_size;
uint8_t secret[crypto_scalarmult_SCALARBYTES]; uint8_t secret[crypto_scalarmult_SCALARBYTES];
uint8_t shared[crypto_scalarmult_BYTES]; uint8_t shared[crypto_scalarmult_BYTES];
@@ -606,10 +669,14 @@ static int gt_setup_crypto (struct crypto_ctx *ctx, int fd, int listener)
crypto_generichash_state state; crypto_generichash_state state;
byte_set(data_w, 0, size);
randombytes_buf(data_w, nonce_size); randombytes_buf(data_w, nonce_size);
randombytes_buf(secret, sizeof(secret)); randombytes_buf(secret, sizeof(secret));
crypto_scalarmult_base(&data_w[nonce_size], secret); crypto_scalarmult_base(&data_w[nonce_size], secret);
byte_cpy(&data_w[size-hash_size-sizeof(gt)], gt, sizeof(gt));
crypto_generichash(&data_w[size-hash_size], hash_size, crypto_generichash(&data_w[size-hash_size], hash_size,
data_w, size-hash_size, ctx->skey, sizeof(ctx->skey)); data_w, size-hash_size, ctx->skey, sizeof(ctx->skey));
@@ -619,6 +686,9 @@ static int gt_setup_crypto (struct crypto_ctx *ctx, int fd, int listener)
if (fd_read_all(fd, data_r, size)!=size) if (fd_read_all(fd, data_r, size)!=size)
return -1; return -1;
if (memcmp(&data_r[size-hash_size-sizeof(gt)], gt, sizeof(gt)))
return -2;
crypto_generichash(hash, hash_size, crypto_generichash(hash, hash_size,
data_r, size-hash_size, ctx->skey, sizeof(ctx->skey)); data_r, size-hash_size, ctx->skey, sizeof(ctx->skey));
@@ -676,9 +746,10 @@ int main (int argc, char **argv)
char *host = NULL; char *host = NULL;
char *port = "5000"; char *port = "5000";
char *dev = PACKAGE_NAME; char *dev = NULL;
char *keyfile = NULL; char *keyfile = NULL;
char *congestion = NULL; char *congestion = NULL;
char *statefile = NULL;
long buffer_size = GT_BUFFER_SIZE; long buffer_size = GT_BUFFER_SIZE;
@@ -720,6 +791,7 @@ int main (int argc, char **argv)
{ "noquickack", NULL, option_option }, { "noquickack", NULL, option_option },
{ "retry", &retry_opts, option_option }, { "retry", &retry_opts, option_option },
{ "daemon", NULL, option_option }, { "daemon", NULL, option_option },
{ "statefile", &statefile, option_str },
{ "debug", NULL, option_option }, { "debug", NULL, option_option },
{ "version", NULL, option_option }, { "version", NULL, option_option },
{ NULL }, { NULL },
@@ -741,14 +813,26 @@ int main (int argc, char **argv)
if (buffer_size < 2048) { if (buffer_size < 2048) {
buffer_size = 2048; buffer_size = 2048;
gt_log("buffer size must be greater than 2048!\n"); gt_log("buffer size must be greater than 2048\n");
} }
if (!listener && !option_is_set(opts, "retry")) if (!listener) {
retry_count = 0; if (!option_is_set(opts, "keyfile")) {
gt_log("keyfile option must be set\n");
return 1;
}
if (!option_is_set(opts, "retry"))
retry_count = 0;
}
if (statefile && statefile[0]!='/') {
gt_log("statefile must be an absolute path\n");
return 1;
}
if (sodium_init()==-1) { if (sodium_init()==-1) {
gt_log("libsodium initialization has failed!\n"); gt_log("libsodium initialization has failed\n");
return 1; return 1;
} }
@@ -757,11 +841,6 @@ int main (int argc, char **argv)
return 1; return 1;
} }
struct crypto_ctx ctx;
if (gt_setup_secretkey(&ctx, keyfile))
return 1;
struct addrinfo *ai = ai_create(host, port, listener); struct addrinfo *ai = ai_create(host, port, listener);
if (!ai) if (!ai)
@@ -772,8 +851,10 @@ int main (int argc, char **argv)
tun.fd = tun_create(dev, option_is_set(opts, "multiqueue")); tun.fd = tun_create(dev, option_is_set(opts, "multiqueue"));
if (tun.fd==-1) if (tun.fd==-1) {
gt_log("couldn't create tun device\n");
return 1; return 1;
}
struct blk *blks = calloc(256, sizeof(struct blk)); struct blk *blks = calloc(256, sizeof(struct blk));
size_t blk_count = 0; size_t blk_count = 0;
@@ -800,6 +881,11 @@ int main (int argc, char **argv)
return 1; return 1;
} }
struct crypto_ctx ctx;
if (gt_setup_secretkey(&ctx, keyfile))
return 1;
if (option_is_set(opts, "daemon")) { if (option_is_set(opts, "daemon")) {
switch (fork()) { switch (fork()) {
case -1: case -1:
@@ -816,8 +902,34 @@ int main (int argc, char **argv)
chdir("/"); chdir("/");
} }
int state_fd = -1;
if (statefile) {
state_fd = open(statefile, O_WRONLY);
if (state_fd==-1) {
if (errno!=EINTR)
perror("open statefile");
return 1;
}
struct stat st = {0};
if (fstat(state_fd, &st)==-1) {
perror("stat statefile");
return 1;
}
if (!S_ISFIFO(st.st_mode)) {
gt_log("`%s' is not a fifo\n", statefile);
return 1;
}
}
long retry = 0; long retry = 0;
fd_write_str(state_fd, "INITIALIZED");
while (!gt_close) { while (!gt_close) {
if (retry_count>=0 && retry>=retry_count+1) { if (retry_count>=0 && retry>=retry_count+1) {
gt_log("couldn't %s (%d attempt%s)\n", listener?"listen":"connect", gt_log("couldn't %s (%d attempt%s)\n", listener?"listen":"connect",
@@ -883,7 +995,7 @@ int main (int argc, char **argv)
retry = 0; retry = 0;
gt_log("%s: running\n", sockname); fd_write_str(state_fd, "STARTED");
fd_set rfds; fd_set rfds;
FD_ZERO(&rfds); FD_ZERO(&rfds);
@@ -943,7 +1055,8 @@ int main (int argc, char **argv)
break; break;
} }
const ssize_t ip_size = ip_get_size(data, GT_MTU_MAX); const int ip_version = ip_get_version(data, GT_MTU_MAX);
const ssize_t ip_size = ip_get_size(ip_version, data, GT_MTU_MAX);
if _0_(ip_size<=0) if _0_(ip_size<=0)
continue; continue;
@@ -956,7 +1069,7 @@ int main (int argc, char **argv)
} }
if _0_(debug) if _0_(debug)
gt_print_hdr(data, ip_size, sockname); gt_print_hdr(ip_version, data, ip_size);
blks[blk_write++].size = r; blks[blk_write++].size = r;
blk_count++; blk_count++;
@@ -1031,7 +1144,9 @@ int main (int argc, char **argv)
} }
size_t size = buffer_read_size(&tun.write); size_t size = buffer_read_size(&tun.write);
ssize_t ip_size = ip_get_size(tun.write.read, size);
const int ip_version = ip_get_version(tun.write.read, size);
ssize_t ip_size = ip_get_size(ip_version, tun.write.read, size);
if _0_(!ip_size) { if _0_(!ip_size) {
gt_log("%s: bad packet!\n", sockname); gt_log("%s: bad packet!\n", sockname);
@@ -1044,6 +1159,8 @@ int main (int argc, char **argv)
ssize_t r = tun_write(tun.fd, tun.write.read, ip_size); ssize_t r = tun_write(tun.fd, tun.write.read, ip_size);
if (r>0) { if (r>0) {
if _0_(debug)
gt_print_hdr(ip_version, tun.write.read, ip_size);
tun.write.read += r; tun.write.read += r;
} else { } else {
gt_close |= !r; gt_close |= !r;
@@ -1062,6 +1179,8 @@ int main (int argc, char **argv)
close(sock.fd); close(sock.fd);
sock.fd = -1; sock.fd = -1;
} }
fd_write_str(state_fd, "STOPPED");
} }
freeaddrinfo(ai); freeaddrinfo(ai);

172
src/tun.c
View File

@@ -11,116 +11,148 @@
#include <sys/uio.h> #include <sys/uio.h>
#ifdef __linux__ #ifdef __linux__
# include <linux/if.h> #include <linux/if.h>
# include <linux/if_tun.h> #include <linux/if_tun.h>
#endif #endif
#ifdef __APPLE__ #ifdef __APPLE__
# include <sys/sys_domain.h> #include <sys/sys_domain.h>
# include <sys/kern_control.h> #include <sys/kern_control.h>
# include <net/if_utun.h> #include <net/if_utun.h>
#endif #endif
#if defined(__APPLE__) || defined(__OpenBSD__) #if defined(__APPLE__) || defined(__OpenBSD__)
# define GT_BSD_TUN 1 #define GT_BSD_TUN
#endif #endif
#ifdef __APPLE__
static int tun_create_by_id (char *name, size_t size, unsigned id, _unused_ int mq)
{
int fd = socket(PF_SYSTEM, SOCK_DGRAM, SYSPROTO_CONTROL);
if (fd==-1)
return -1;
struct ctl_info ci;
byte_set(&ci, 0, sizeof(ci));
str_cpy(ci.ctl_name, UTUN_CONTROL_NAME, sizeof(ci.ctl_name)-1);
if (ioctl(fd, CTLIOCGINFO, &ci)) {
close(fd);
return -1;
}
struct sockaddr_ctl sc = {
.sc_id = ci.ctl_id,
.sc_len = sizeof(sc),
.sc_family = AF_SYSTEM,
.ss_sysaddr = AF_SYS_CONTROL,
.sc_unit = id+1,
};
if (connect(fd, (struct sockaddr *)&sc, sizeof(sc))) {
close(fd);
return -1;
}
snprintf(name, size, "utun%u", id);
return fd;
}
static int tun_create_by_name (char *name, size_t size, char *dev_name, int mq)
{
unsigned id = 0;
if (sscanf(dev_name, "utun%u", &id)!=1)
return -1;
return tun_create_by_id(name, size, id, mq);
}
#else /* not __APPLE__ */
#ifdef __linux__ #ifdef __linux__
int tun_create (char *name, int multiqueue)
static int tun_create_by_name (char *name, size_t size, char *dev_name, int mq)
{ {
int fd = open("/dev/net/tun", O_RDWR); int fd = open("/dev/net/tun", O_RDWR);
if (fd<0) { if (fd==-1)
perror("open /dev/net/tun");
return -1; return -1;
}
struct ifreq ifr = { struct ifreq ifr = {
.ifr_flags = IFF_TUN|IFF_NO_PI, .ifr_flags = IFF_TUN|IFF_NO_PI,
}; };
if (multiqueue) { if (mq) {
#ifdef IFF_MULTI_QUEUE #ifdef IFF_MULTI_QUEUE
ifr.ifr_flags |= IFF_MULTI_QUEUE; ifr.ifr_flags |= IFF_MULTI_QUEUE;
#else
gt_na("IFF_MULTI_QUEUE");
#endif #endif
} }
str_cpy(ifr.ifr_name, name, IFNAMSIZ-1); str_cpy(ifr.ifr_name, dev_name, IFNAMSIZ-1);
int ret = ioctl(fd, TUNSETIFF, &ifr); if (ioctl(fd, TUNSETIFF, &ifr)) {
close(fd);
if (ret<0) {
perror("ioctl TUNSETIFF");
return -1; return -1;
} }
gt_log("tun name: %s\n", ifr.ifr_name); str_cpy(name, ifr.ifr_name, size-1);
return fd; return fd;
} }
#elif defined(__APPLE__)
int tun_create (_unused_ char *name, _unused_ int mq) #else /* not __linux__ not __APPLE__ */
static int tun_create_by_name (char *name, size_t size, char *dev_name, _unused_ int mq)
{ {
for (unsigned dev_id = 0; dev_id < 32; dev_id++) { char path[64];
struct ctl_info ci;
byte_set(&ci, 0, sizeof(ci)); snprintf(path, sizeof(path), "/dev/%s", dev_name);
str_cpy(ci.ctl_name, UTUN_CONTROL_NAME, sizeof(ci.ctl_name)-1); str_cpy(name, dev_name, size-1);
int fd = socket(PF_SYSTEM, SOCK_DGRAM, SYSPROTO_CONTROL); return open(path, O_RDWR);
if (fd==-1)
return -1;
if (ioctl(fd, CTLIOCGINFO, &ci)==-1) {
close(fd);
continue;
}
struct sockaddr_ctl sc = {
.sc_id = ci.ctl_id,
.sc_len = sizeof(sc),
.sc_family = AF_SYSTEM,
.ss_sysaddr = AF_SYS_CONTROL,
.sc_unit = dev_id+1,
};
if (connect(fd, (struct sockaddr *)&sc, sizeof(sc))==-1) {
close(fd);
continue;
}
gt_log("tun name: /dev/utun%u\n", dev_id);
return fd;
}
return -1;
} }
#else
int tun_create (_unused_ char *name, _unused_ int mq) #endif /* not __APPLE__ */
static int tun_create_by_id (char *name, size_t size, unsigned id, int mq)
{ {
for (unsigned dev_id = 0; dev_id < 32; dev_id++) { char dev_name[64];
char dev_path[11];
snprintf(dev_path, sizeof(dev_path), "/dev/tun%u", dev_id); snprintf(dev_name, sizeof(dev_name), "tun%u", id);
int fd = open(dev_path, O_RDWR); return tun_create_by_name(name, size, dev_name, mq);
if (fd==-1)
continue;
gt_log("tun name: %s\n", dev_path);
return fd;
}
return -1;
} }
#endif #endif
int tun_create (char *dev_name, int mq)
{
char name[64];
int fd = -1;
#ifndef IFF_MULTI_QUEUE
if (mq)
gt_na("IFF_MULTI_QUEUE");
#endif
if (str_empty(dev_name)) {
for (unsigned id=0; id<32 && fd==-1; id++)
fd = tun_create_by_id(name, sizeof(name), id, mq);
} else {
fd = tun_create_by_name(name, sizeof(name), dev_name, mq);
}
if (fd!=-1)
gt_print("tun name: %s\n", name);
return fd;
}
ssize_t tun_read (int fd, void *data, size_t size) ssize_t tun_read (int fd, void *data, size_t size)
{ {
if (!size) if (!size)

View File

@@ -6,4 +6,6 @@
[ -z "${VERSION}" ] && VERSION=`basename \`pwd\`` \ [ -z "${VERSION}" ] && VERSION=`basename \`pwd\`` \
&& VERSION=${VERSION#*-} && VERSION=${VERSION#*-}
[ "$1" = "major" ] && VERSION=${VERSION%%.*}
printf ${VERSION} printf ${VERSION}