Compare commits
39 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
30502e05ec | ||
|
|
905673ecec | ||
|
|
cfbb77c710 | ||
|
|
13dcdc5afb | ||
|
|
56ec81bf92 | ||
|
|
f1024381e0 | ||
|
|
3f3b21f08d | ||
|
|
40dd5ce18a | ||
|
|
5efb379251 | ||
|
|
2eb6521f9c | ||
|
|
bfd81e9666 | ||
|
|
8fa2322314 | ||
|
|
8982f27220 | ||
|
|
a5f97fcc8c | ||
|
|
35a9bf27df | ||
|
|
ba0af8cc20 | ||
|
|
ec85be5c6a | ||
|
|
fe989851ab | ||
|
|
46842dd200 | ||
|
|
e5eb30598d | ||
|
|
c81ad0a7c6 | ||
|
|
97641d6dda | ||
|
|
e48dac775c | ||
|
|
6282f36ac7 | ||
|
|
cf022af4a9 | ||
|
|
597c586657 | ||
|
|
fc7f9aa0c8 | ||
|
|
7492f977b6 | ||
|
|
38b7333533 | ||
|
|
1ab854f058 | ||
|
|
c59def90fb | ||
|
|
347a3ecce7 | ||
|
|
98d13ef510 | ||
|
|
35b95001c4 | ||
|
|
75c12b36d6 | ||
|
|
db01c8b33f | ||
|
|
fb4f6f6cb9 | ||
|
|
a86aea431d | ||
|
|
1a128a6d92 |
@@ -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
|
||||||
|
|||||||
2
LICENSE
2
LICENSE
@@ -1,4 +1,4 @@
|
|||||||
Copyright (c) 2015, angt
|
Copyright (c) 2015-2016, angt
|
||||||
All rights reserved.
|
All rights reserved.
|
||||||
|
|
||||||
Redistribution and use in source and binary forms, with or without
|
Redistribution and use in source and binary forms, with or without
|
||||||
|
|||||||
10
Makefile.am
10
Makefile.am
@@ -6,17 +6,21 @@ glorytun_CFLAGS = $(libsodium_CFLAGS)
|
|||||||
glorytun_LDADD = $(libsodium_LIBS)
|
glorytun_LDADD = $(libsodium_LIBS)
|
||||||
glorytun_SOURCES = \
|
glorytun_SOURCES = \
|
||||||
src/common.h \
|
src/common.h \
|
||||||
src/common-static.h \
|
|
||||||
src/common.c \
|
src/common.c \
|
||||||
src/ip-static.h \
|
src/buffer.h \
|
||||||
|
src/ip.h \
|
||||||
|
src/str.h \
|
||||||
src/main.c \
|
src/main.c \
|
||||||
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 \
|
||||||
README.md \
|
README.md \
|
||||||
|
VERSION \
|
||||||
autogen.sh \
|
autogen.sh \
|
||||||
version.sh
|
version.sh
|
||||||
|
|||||||
13
README.md
13
README.md
@@ -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 [...]
|
|
||||||
|
|||||||
@@ -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])
|
||||||
|
|||||||
63
src/buffer.h
Normal file
63
src/buffer.h
Normal file
@@ -0,0 +1,63 @@
|
|||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include "common.h"
|
||||||
|
|
||||||
|
typedef struct buffer buffer_t;
|
||||||
|
|
||||||
|
struct buffer {
|
||||||
|
uint8_t *data;
|
||||||
|
uint8_t *read;
|
||||||
|
uint8_t *write;
|
||||||
|
uint8_t *end;
|
||||||
|
};
|
||||||
|
|
||||||
|
static inline void buffer_setup (buffer_t *buffer, void *data, size_t size)
|
||||||
|
{
|
||||||
|
if (!data)
|
||||||
|
data = malloc(ALIGN(size));
|
||||||
|
|
||||||
|
buffer->data = data;
|
||||||
|
buffer->read = data;
|
||||||
|
buffer->write = data;
|
||||||
|
buffer->end = data;
|
||||||
|
buffer->end += size;
|
||||||
|
}
|
||||||
|
|
||||||
|
static inline void buffer_format (buffer_t *buffer)
|
||||||
|
{
|
||||||
|
buffer->write = buffer->data;
|
||||||
|
buffer->read = buffer->data;
|
||||||
|
}
|
||||||
|
|
||||||
|
_pure_
|
||||||
|
static inline size_t buffer_size (buffer_t *buffer)
|
||||||
|
{
|
||||||
|
return buffer->end-buffer->data;
|
||||||
|
}
|
||||||
|
|
||||||
|
_pure_
|
||||||
|
static inline size_t buffer_write_size (buffer_t *buffer)
|
||||||
|
{
|
||||||
|
return buffer->end-buffer->write;
|
||||||
|
}
|
||||||
|
|
||||||
|
_pure_
|
||||||
|
static inline size_t buffer_read_size (buffer_t *buffer)
|
||||||
|
{
|
||||||
|
return buffer->write-buffer->read;
|
||||||
|
}
|
||||||
|
|
||||||
|
static inline void buffer_shift (buffer_t *buffer)
|
||||||
|
{
|
||||||
|
if (buffer->read==buffer->write) {
|
||||||
|
buffer_format(buffer);
|
||||||
|
} else {
|
||||||
|
const uint8_t *src = PALIGN_DOWN(buffer->read);
|
||||||
|
const size_t size = ALIGN(buffer->write-src);
|
||||||
|
if (buffer->data+size<src) {
|
||||||
|
memcpy(buffer->data, src, size);
|
||||||
|
buffer->read -= src-buffer->data;
|
||||||
|
buffer->write -= src-buffer->data;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -1,147 +0,0 @@
|
|||||||
#pragma once
|
|
||||||
|
|
||||||
#include "common.h"
|
|
||||||
|
|
||||||
#include <unistd.h>
|
|
||||||
#include <errno.h>
|
|
||||||
|
|
||||||
static inline void byte_set (void *dst, const char value, size_t size)
|
|
||||||
{
|
|
||||||
if (!dst)
|
|
||||||
return;
|
|
||||||
|
|
||||||
char *restrict d = dst;
|
|
||||||
|
|
||||||
while (size--)
|
|
||||||
*d++ = value;
|
|
||||||
}
|
|
||||||
|
|
||||||
static inline void byte_cpy (void *dst, const void *src, size_t size)
|
|
||||||
{
|
|
||||||
if (!dst || !src)
|
|
||||||
return;
|
|
||||||
|
|
||||||
char *restrict d = dst;
|
|
||||||
const char *restrict s = src;
|
|
||||||
|
|
||||||
while (size--)
|
|
||||||
*d++ = *s++;
|
|
||||||
}
|
|
||||||
|
|
||||||
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_cmp (const char *restrict sa, const char *restrict sb)
|
|
||||||
{
|
|
||||||
if (!sa || !sb)
|
|
||||||
return 1;
|
|
||||||
|
|
||||||
while (*sa==*sb++)
|
|
||||||
if (!*sa++)
|
|
||||||
return 0;
|
|
||||||
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
_pure_
|
|
||||||
static inline size_t str_len (const char *restrict str)
|
|
||||||
{
|
|
||||||
if (!str)
|
|
||||||
return 0;
|
|
||||||
|
|
||||||
size_t i = 0;
|
|
||||||
|
|
||||||
while (str[i])
|
|
||||||
i++;
|
|
||||||
|
|
||||||
return i;
|
|
||||||
}
|
|
||||||
|
|
||||||
static inline char *str_cat (const char *const strs[], size_t count)
|
|
||||||
{
|
|
||||||
size_t size = 1;
|
|
||||||
|
|
||||||
for (size_t i=0; i<count; i++)
|
|
||||||
size += str_len(strs[i]);
|
|
||||||
|
|
||||||
char *str = malloc(size);
|
|
||||||
|
|
||||||
if (!str)
|
|
||||||
return NULL;
|
|
||||||
|
|
||||||
char *p = str;
|
|
||||||
|
|
||||||
for (size_t i=0; i<count; i++) {
|
|
||||||
size_t len = str_len(strs[i]);
|
|
||||||
byte_cpy(p, strs[i], len);
|
|
||||||
p += len;
|
|
||||||
}
|
|
||||||
|
|
||||||
p[0] = 0;
|
|
||||||
|
|
||||||
return str;
|
|
||||||
}
|
|
||||||
|
|
||||||
static inline void buffer_setup (buffer_t *buffer, void *data, size_t size)
|
|
||||||
{
|
|
||||||
if (!data)
|
|
||||||
data = malloc(ALIGN(size));
|
|
||||||
|
|
||||||
buffer->data = data;
|
|
||||||
buffer->read = data;
|
|
||||||
buffer->write = data;
|
|
||||||
buffer->end = data;
|
|
||||||
buffer->end += size;
|
|
||||||
}
|
|
||||||
|
|
||||||
static inline void buffer_format (buffer_t *buffer)
|
|
||||||
{
|
|
||||||
buffer->write = buffer->data;
|
|
||||||
buffer->read = buffer->data;
|
|
||||||
}
|
|
||||||
|
|
||||||
_pure_
|
|
||||||
static inline size_t buffer_size (buffer_t *buffer)
|
|
||||||
{
|
|
||||||
return buffer->end-buffer->data;
|
|
||||||
}
|
|
||||||
|
|
||||||
_pure_
|
|
||||||
static inline size_t buffer_write_size (buffer_t *buffer)
|
|
||||||
{
|
|
||||||
return buffer->end-buffer->write;
|
|
||||||
}
|
|
||||||
|
|
||||||
_pure_
|
|
||||||
static inline size_t buffer_read_size (buffer_t *buffer)
|
|
||||||
{
|
|
||||||
return buffer->write-buffer->read;
|
|
||||||
}
|
|
||||||
|
|
||||||
static inline void buffer_shift (buffer_t *buffer)
|
|
||||||
{
|
|
||||||
if (buffer->read==buffer->write) {
|
|
||||||
buffer_format(buffer);
|
|
||||||
} else {
|
|
||||||
const uint8_t *src = PALIGN_DOWN(buffer->read);
|
|
||||||
const size_t size = ALIGN(buffer->write-src);
|
|
||||||
if (buffer->data+size<src) {
|
|
||||||
byte_cpy(buffer->data, src, size);
|
|
||||||
buffer->read -= src-buffer->data;
|
|
||||||
buffer->write -= src-buffer->data;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -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)
|
||||||
|
|||||||
14
src/common.h
14
src/common.h
@@ -2,6 +2,9 @@
|
|||||||
|
|
||||||
#include <stdlib.h>
|
#include <stdlib.h>
|
||||||
#include <stdint.h>
|
#include <stdint.h>
|
||||||
|
#include <string.h>
|
||||||
|
#include <unistd.h>
|
||||||
|
#include <errno.h>
|
||||||
|
|
||||||
#define COUNT(x) (sizeof(x)/sizeof(x[0]))
|
#define COUNT(x) (sizeof(x)/sizeof(x[0]))
|
||||||
|
|
||||||
@@ -17,6 +20,8 @@
|
|||||||
#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 CLZ(x) (__builtin_clz(x))
|
||||||
|
|
||||||
#define _printf_(A,B) __attribute__ ((format(printf,A,B)))
|
#define _printf_(A,B) __attribute__ ((format(printf,A,B)))
|
||||||
#define _noreturn_ __attribute__ ((noreturn))
|
#define _noreturn_ __attribute__ ((noreturn))
|
||||||
#define _unused_ __attribute__ ((unused))
|
#define _unused_ __attribute__ ((unused))
|
||||||
@@ -24,15 +29,6 @@
|
|||||||
#define _const_ __attribute__ ((const))
|
#define _const_ __attribute__ ((const))
|
||||||
#define _align_(...) __attribute__ ((aligned(__VA_ARGS__)))
|
#define _align_(...) __attribute__ ((aligned(__VA_ARGS__)))
|
||||||
|
|
||||||
typedef struct buffer buffer_t;
|
|
||||||
|
|
||||||
struct buffer {
|
|
||||||
uint8_t *data;
|
|
||||||
uint8_t *read;
|
|
||||||
uint8_t *write;
|
|
||||||
uint8_t *end;
|
|
||||||
};
|
|
||||||
|
|
||||||
int gt_print (const char *, ...) _printf_(1,2);
|
int gt_print (const char *, ...) _printf_(1,2);
|
||||||
void gt_log (const char *, ...) _printf_(1,2);
|
void gt_log (const char *, ...) _printf_(1,2);
|
||||||
void gt_fatal (const char *, ...) _printf_(1,2) _noreturn_;
|
void gt_fatal (const char *, ...) _printf_(1,2) _noreturn_;
|
||||||
|
|||||||
155
src/db.c
Normal file
155
src/db.c
Normal file
@@ -0,0 +1,155 @@
|
|||||||
|
#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
Normal file
7
src/db.h
Normal 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 *);
|
||||||
@@ -1,57 +0,0 @@
|
|||||||
#pragma once
|
|
||||||
|
|
||||||
#include <stdint.h>
|
|
||||||
|
|
||||||
_pure_
|
|
||||||
static inline int ip_get_version (const uint8_t *data, size_t size)
|
|
||||||
{
|
|
||||||
if (size<20) // XXX
|
|
||||||
return -1; // XXX
|
|
||||||
|
|
||||||
return data[0]>>4;
|
|
||||||
}
|
|
||||||
|
|
||||||
_pure_
|
|
||||||
static inline ssize_t ip_get_size (const int ip_version, const uint8_t *data, size_t size)
|
|
||||||
{
|
|
||||||
switch (ip_version) {
|
|
||||||
case 4:
|
|
||||||
return ((data[2]<<8)|data[3]);
|
|
||||||
case 6:
|
|
||||||
return ((data[4]<<8)|data[5])+40;
|
|
||||||
case -1:
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
_pure_
|
|
||||||
static inline ssize_t ip_get_proto (const int ip_version, const uint8_t *data, size_t size)
|
|
||||||
{
|
|
||||||
switch (ip_version) {
|
|
||||||
case 4:
|
|
||||||
return data[9];
|
|
||||||
case 6:
|
|
||||||
return data[6];
|
|
||||||
case -1:
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
_pure_
|
|
||||||
static inline ssize_t ip_get_hdr_size (const int ip_version, const uint8_t *data, size_t size)
|
|
||||||
{
|
|
||||||
switch (ip_version) {
|
|
||||||
case 4:
|
|
||||||
return (data[0]&0xF)<<2;
|
|
||||||
case 6:
|
|
||||||
return 40;
|
|
||||||
case -1:
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
39
src/ip.h
Normal file
39
src/ip.h
Normal file
@@ -0,0 +1,39 @@
|
|||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <stdint.h>
|
||||||
|
|
||||||
|
struct ip_common {
|
||||||
|
uint8_t version;
|
||||||
|
uint8_t proto;
|
||||||
|
uint8_t hdr_size;
|
||||||
|
uint16_t size;
|
||||||
|
};
|
||||||
|
|
||||||
|
_pure_
|
||||||
|
static inline uint8_t ip_get_version (const uint8_t *data, size_t size)
|
||||||
|
{
|
||||||
|
if (size<20)
|
||||||
|
return 0;
|
||||||
|
|
||||||
|
return data[0]>>4;
|
||||||
|
}
|
||||||
|
|
||||||
|
static inline int ip_get_common (struct ip_common *ic, const uint8_t *data, size_t size)
|
||||||
|
{
|
||||||
|
ic->version = ip_get_version(data, size);
|
||||||
|
|
||||||
|
switch (ic->version) {
|
||||||
|
case 4:
|
||||||
|
ic->proto = data[9];
|
||||||
|
ic->hdr_size = (data[0]&0xF)<<2;
|
||||||
|
ic->size = ((data[2]<<8)|data[3]);
|
||||||
|
return 0;
|
||||||
|
case 6:
|
||||||
|
ic->proto = data[6];
|
||||||
|
ic->hdr_size = 40;
|
||||||
|
ic->size = ((data[4]<<8)|data[5])+40;
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
436
src/main.c
436
src/main.c
@@ -1,35 +1,45 @@
|
|||||||
|
#include "common.h"
|
||||||
|
|
||||||
|
#include "buffer.h"
|
||||||
|
#include "ip.h"
|
||||||
|
#include "str.h"
|
||||||
|
#include "option.h"
|
||||||
|
#include "tun.h"
|
||||||
|
#include "db.h"
|
||||||
|
|
||||||
#include <inttypes.h>
|
#include <inttypes.h>
|
||||||
#include <limits.h>
|
#include <limits.h>
|
||||||
#include <stdio.h>
|
#include <stdio.h>
|
||||||
#include <signal.h>
|
#include <signal.h>
|
||||||
#include <poll.h>
|
#include <poll.h>
|
||||||
|
#include <fcntl.h>
|
||||||
#include <sys/time.h>
|
#include <sys/time.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>
|
||||||
|
|
||||||
#include <sodium.h>
|
#include <sodium.h>
|
||||||
|
|
||||||
#include "common-static.h"
|
|
||||||
#include "ip-static.h"
|
|
||||||
|
|
||||||
#include "option.h"
|
|
||||||
#include "tun.h"
|
|
||||||
|
|
||||||
#ifndef O_CLOEXEC
|
#ifndef O_CLOEXEC
|
||||||
#define O_CLOEXEC 0
|
#define O_CLOEXEC 0
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#define GT_BUFFER_SIZE (4*1024*1024)
|
#define GT_BUFFER_SIZE (64*1024)
|
||||||
#define GT_TIMEOUT (5000)
|
#define GT_TIMEOUT (5000)
|
||||||
#define GT_MTU_MAX (1500)
|
#define GT_MTU_MAX (1500)
|
||||||
#define GT_TUNR_SIZE (0x7FFF-16)
|
#define GT_TUNR_SIZE (0x7FFF-16)
|
||||||
@@ -41,11 +51,6 @@ struct fdbuf {
|
|||||||
buffer_t write;
|
buffer_t write;
|
||||||
};
|
};
|
||||||
|
|
||||||
struct blk {
|
|
||||||
size_t size;
|
|
||||||
uint8_t data[GT_MTU_MAX] _align_(16);
|
|
||||||
};
|
|
||||||
|
|
||||||
struct crypto_ctx {
|
struct crypto_ctx {
|
||||||
struct {
|
struct {
|
||||||
crypto_aead_aes256gcm_state state;
|
crypto_aead_aes256gcm_state state;
|
||||||
@@ -92,7 +97,9 @@ enum sk_opt {
|
|||||||
sk_keepintvl,
|
sk_keepintvl,
|
||||||
sk_congestion,
|
sk_congestion,
|
||||||
sk_defer_accept,
|
sk_defer_accept,
|
||||||
|
sk_acceptfilter,
|
||||||
sk_quickack,
|
sk_quickack,
|
||||||
|
sk_user_timeout,
|
||||||
};
|
};
|
||||||
|
|
||||||
static void sk_set (int fd, enum sk_opt opt, const void *val, socklen_t len)
|
static void sk_set (int fd, enum sk_opt opt, const void *val, socklen_t len)
|
||||||
@@ -137,6 +144,16 @@ 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
|
||||||
|
},
|
||||||
|
[sk_user_timeout] = { "TCP_USER_TIMEOUT",
|
||||||
|
#ifdef TCP_USER_TIMEOUT
|
||||||
|
1, IPPROTO_TCP, TCP_USER_TIMEOUT,
|
||||||
#endif
|
#endif
|
||||||
},
|
},
|
||||||
};
|
};
|
||||||
@@ -159,21 +176,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;
|
||||||
}
|
}
|
||||||
@@ -255,7 +274,7 @@ static char *sk_get_name (int fd)
|
|||||||
|
|
||||||
static struct addrinfo *ai_create (const char *host, const char *port, int listener)
|
static struct addrinfo *ai_create (const char *host, const char *port, int listener)
|
||||||
{
|
{
|
||||||
if (!port || !port[0]) {
|
if (str_empty(port)) {
|
||||||
gt_log("port is not valid\n");
|
gt_log("port is not valid\n");
|
||||||
return NULL;
|
return NULL;
|
||||||
}
|
}
|
||||||
@@ -327,7 +346,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 +366,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 +387,15 @@ static ssize_t fd_write (int fd, const void *data, size_t size)
|
|||||||
return ret;
|
return ret;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
static void state_write (int fd, const char *str)
|
||||||
|
{
|
||||||
|
if (fd==-1) {
|
||||||
|
gt_print("state: %s", str);
|
||||||
|
} else {
|
||||||
|
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,34 +519,172 @@ static int gt_decrypt (struct crypto_ctx *ctx, buffer_t *dst, buffer_t *src)
|
|||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
static void gt_print_hdr (const int ip_version, 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 ssize_t ip_proto = ip_get_proto(ip_version, data, ip_size);
|
const size_t lim = size&~1u;
|
||||||
const ssize_t ip_hdr_size = ip_get_hdr_size(ip_version, data, ip_size);
|
|
||||||
|
|
||||||
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)
|
||||||
|
sum += data[size-1]<<8;
|
||||||
|
|
||||||
|
return sum;
|
||||||
|
}
|
||||||
|
|
||||||
|
_const_
|
||||||
|
static inline uint16_t sum16_final (uint32_t sum)
|
||||||
|
{
|
||||||
|
sum = (sum>>16)+(sum&0xFFFF);
|
||||||
|
return ~(sum+(sum>>16));
|
||||||
|
}
|
||||||
|
|
||||||
|
struct tcp_entry {
|
||||||
|
uint8_t key[37];
|
||||||
|
struct {
|
||||||
|
uint32_t seq;
|
||||||
|
uint32_t ack;
|
||||||
|
size_t count;
|
||||||
|
struct timeval time;
|
||||||
|
} data[2];
|
||||||
|
};
|
||||||
|
|
||||||
|
static int tcp_entry_set_key (struct tcp_entry *te, struct ip_common *ic, uint8_t *data)
|
||||||
|
{
|
||||||
|
uint8_t *key = &te->key[1];
|
||||||
|
size_t size = 0;
|
||||||
|
|
||||||
|
switch (ic->version) {
|
||||||
case 4:
|
case 4:
|
||||||
gt_tohex(ip_src, sizeof(ip_src), &data[12], 4);
|
size = 8;
|
||||||
gt_tohex(ip_dst, sizeof(ip_dst), &data[16], 4);
|
memcpy(key, &data[12], 8);
|
||||||
break;
|
break;
|
||||||
case 6:
|
case 6:
|
||||||
gt_tohex(ip_src, sizeof(ip_src), &data[9], 16);
|
size = 32;
|
||||||
gt_tohex(ip_dst, sizeof(ip_dst), &data[25], 16);
|
memcpy(key, &data[9], 32);
|
||||||
break;
|
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);
|
memcpy(&key[size], &data[ic->hdr_size], 4);
|
||||||
|
te->key[0] = size+4;
|
||||||
|
|
||||||
if (ip_hdr_size<=0)
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
static int tcp_entry_set_key_rev (struct tcp_entry *te, struct ip_common *ic, uint8_t *data)
|
||||||
|
{
|
||||||
|
uint8_t *key = &te->key[1];
|
||||||
|
size_t size = 0;
|
||||||
|
|
||||||
|
switch (ic->version) {
|
||||||
|
case 4:
|
||||||
|
size = 8;
|
||||||
|
memcpy(key, &data[12+4], 4);
|
||||||
|
memcpy(key+4, &data[12], 4);
|
||||||
|
break;
|
||||||
|
case 6:
|
||||||
|
size = 32;
|
||||||
|
memcpy(key, &data[9+16], 16);
|
||||||
|
memcpy(key+16, &data[9], 16);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
memcpy(&key[size], &data[ic->hdr_size+2], 2);
|
||||||
|
memcpy(&key[size+2], &data[ic->hdr_size], 2);
|
||||||
|
te->key[0] = size+4;
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
static int gt_track (uint8_t **db, struct ip_common *ic, uint8_t *data, int rev)
|
||||||
|
{
|
||||||
|
if (ic->proto!=IPPROTO_TCP)
|
||||||
|
return 0;
|
||||||
|
|
||||||
|
if (!ic->hdr_size)
|
||||||
|
return 1;
|
||||||
|
|
||||||
|
struct tcp_entry entry;
|
||||||
|
|
||||||
|
if (rev) {
|
||||||
|
tcp_entry_set_key_rev(&entry, ic, data);
|
||||||
|
} else {
|
||||||
|
tcp_entry_set_key(&entry, ic, data);
|
||||||
|
}
|
||||||
|
|
||||||
|
struct tcphdr tcp;
|
||||||
|
memcpy(&tcp, &data[ic->hdr_size], sizeof(tcp));
|
||||||
|
tcp.th_seq = ntohl(tcp.th_seq);
|
||||||
|
tcp.th_ack = ntohl(tcp.th_ack);
|
||||||
|
|
||||||
|
struct tcp_entry *r_entry = (void *)db_search(db, entry.key);
|
||||||
|
|
||||||
|
if (!r_entry) {
|
||||||
|
r_entry = calloc(1, sizeof(entry));
|
||||||
|
|
||||||
|
if (!r_entry)
|
||||||
|
return 1;
|
||||||
|
|
||||||
|
memcpy(r_entry->key, entry.key, sizeof(entry.key));
|
||||||
|
|
||||||
|
if (!db_insert(db, r_entry->key)) {
|
||||||
|
free(r_entry);
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
gt_print("new tcp entry\n");
|
||||||
|
} else {
|
||||||
|
gt_print("old_seq:%u\told_ack:%u\tcount:%zu\n",
|
||||||
|
r_entry->data[rev].seq,
|
||||||
|
r_entry->data[rev].ack,
|
||||||
|
r_entry->data[rev].count);
|
||||||
|
}
|
||||||
|
|
||||||
|
r_entry->data[rev].seq = tcp.th_seq;
|
||||||
|
r_entry->data[rev].ack = tcp.th_ack;
|
||||||
|
r_entry->data[rev].count++;
|
||||||
|
gettimeofday(&r_entry->data[rev].time, NULL);
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
static void gt_print_hdr (struct ip_common *ic, uint8_t *data)
|
||||||
|
{
|
||||||
|
if (!ic->hdr_size)
|
||||||
return;
|
return;
|
||||||
|
|
||||||
if (ip_proto==6) {
|
uint32_t sum = ic->proto+ic->size-ic->hdr_size;
|
||||||
|
|
||||||
|
char ip_src[INET6_ADDRSTRLEN];
|
||||||
|
char ip_dst[INET6_ADDRSTRLEN];
|
||||||
|
|
||||||
|
switch (ic->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;
|
||||||
|
}
|
||||||
|
|
||||||
|
uint8_t *const packet = &data[ic->hdr_size];
|
||||||
|
|
||||||
|
if (ic->proto==IPPROTO_TCP) {
|
||||||
struct tcphdr tcp;
|
struct tcphdr tcp;
|
||||||
|
|
||||||
byte_cpy(&tcp, &data[ip_hdr_size], sizeof(tcp));
|
memcpy(&tcp, packet, 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, &packet[sizeof(tcp)], ic->size-ic->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);
|
||||||
@@ -526,27 +692,31 @@ static void gt_print_hdr (const int ip_version, uint8_t *data, size_t ip_size, c
|
|||||||
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:%hhu\tsrc:%s.%u\tdst:%s.%u\tseq:%u\tack:%u\twin:%u\tsize:%u\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,
|
ic->proto, ip_src, tcp.th_sport, ip_dst, tcp.th_dport,
|
||||||
|
tcp.th_seq, tcp.th_ack, tcp.th_win, ic->size-ic->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 (ic->proto==IPPROTO_UDP) {
|
||||||
struct udphdr udp;
|
struct udphdr udp;
|
||||||
|
|
||||||
byte_cpy(&udp, &data[ip_hdr_size], sizeof(udp));
|
memcpy(&udp, packet, sizeof(udp));
|
||||||
|
|
||||||
udp.uh_sport = ntohs(udp.uh_sport);
|
udp.uh_sport = ntohs(udp.uh_sport);
|
||||||
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:%hhu\tsrc:%s.%u\tdst:%s.%u\tsize:%u\n",
|
||||||
sockname, udp.uh_sport, udp.uh_dport, udp.uh_ulen);
|
ic->proto, ip_src, udp.uh_sport, ip_dst, udp.uh_dport, udp.uh_ulen-8);
|
||||||
|
} else {
|
||||||
|
gt_print("proto:%hhu\tsrc:%s\tdst:%s\tsize:%hu\n",
|
||||||
|
ic->proto, ip_src, ip_dst, ic->size);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -554,7 +724,7 @@ 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);
|
||||||
|
|
||||||
if (!keyfile) {
|
if (str_empty(keyfile)) {
|
||||||
char buf[2*size+1];
|
char buf[2*size+1];
|
||||||
|
|
||||||
randombytes_buf(ctx->skey, size);
|
randombytes_buf(ctx->skey, size);
|
||||||
@@ -596,10 +766,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];
|
||||||
@@ -611,10 +784,14 @@ static int gt_setup_crypto (struct crypto_ctx *ctx, int fd, int listener)
|
|||||||
|
|
||||||
crypto_generichash_state state;
|
crypto_generichash_state state;
|
||||||
|
|
||||||
|
memset(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);
|
||||||
|
|
||||||
|
memcpy(&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));
|
||||||
|
|
||||||
@@ -624,6 +801,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));
|
||||||
|
|
||||||
@@ -669,8 +849,8 @@ static int gt_setup_crypto (struct crypto_ctx *ctx, int fd, int listener)
|
|||||||
sodium_memzero(shared, sizeof(shared));
|
sodium_memzero(shared, sizeof(shared));
|
||||||
sodium_memzero(key, sizeof(key));
|
sodium_memzero(key, sizeof(key));
|
||||||
|
|
||||||
byte_cpy(ctx->read.nonce, data_r, nonce_size);
|
memcpy(ctx->read.nonce, data_r, nonce_size);
|
||||||
byte_cpy(ctx->write.nonce, data_w, nonce_size);
|
memcpy(ctx->write.nonce, data_w, nonce_size);
|
||||||
|
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
@@ -681,9 +861,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;
|
||||||
|
|
||||||
@@ -696,6 +877,8 @@ int main (int argc, char **argv)
|
|||||||
long retry_const = 0;
|
long retry_const = 0;
|
||||||
long retry_limit = 1000000;
|
long retry_limit = 1000000;
|
||||||
|
|
||||||
|
long user_timeout = 0;
|
||||||
|
|
||||||
struct option ka_opts[] = {
|
struct option ka_opts[] = {
|
||||||
{ "count", &ka_count, option_long },
|
{ "count", &ka_count, option_long },
|
||||||
{ "idle", &ka_idle, option_long },
|
{ "idle", &ka_idle, option_long },
|
||||||
@@ -725,6 +908,8 @@ 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 },
|
||||||
|
{ "timeout", &user_timeout, option_long },
|
||||||
{ "debug", NULL, option_option },
|
{ "debug", NULL, option_option },
|
||||||
{ "version", NULL, option_option },
|
{ "version", NULL, option_option },
|
||||||
{ NULL },
|
{ NULL },
|
||||||
@@ -746,14 +931,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) {
|
||||||
|
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;
|
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;
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -772,16 +969,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) {
|
||||||
return 1;
|
gt_log("couldn't create tun device\n");
|
||||||
|
|
||||||
struct blk *blks = calloc(256, sizeof(struct blk));
|
|
||||||
size_t blk_count = 0;
|
|
||||||
uint8_t blk_read = 0;
|
|
||||||
uint8_t blk_write = 0;
|
|
||||||
|
|
||||||
if (!blks)
|
|
||||||
return 1;
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
fd_set_nonblock(tun.fd);
|
fd_set_nonblock(tun.fd);
|
||||||
|
|
||||||
@@ -821,7 +1012,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;
|
||||||
|
uint8_t *db = NULL;
|
||||||
|
|
||||||
|
state_write(state_fd, "INITIALIZED\n");
|
||||||
|
|
||||||
while (!gt_close) {
|
while (!gt_close) {
|
||||||
if (retry_count>=0 && retry>=retry_count+1) {
|
if (retry_count>=0 && retry>=retry_count+1) {
|
||||||
@@ -850,7 +1068,7 @@ int main (int argc, char **argv)
|
|||||||
|
|
||||||
char *sockname = sk_get_name(sock.fd);
|
char *sockname = sk_get_name(sock.fd);
|
||||||
|
|
||||||
if (!sockname) {
|
if (str_empty(sockname)) {
|
||||||
close(sock.fd);
|
close(sock.fd);
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
@@ -873,6 +1091,9 @@ int main (int argc, char **argv)
|
|||||||
sk_set_int(sock.fd, sk_keepintvl, ka_interval);
|
sk_set_int(sock.fd, sk_keepintvl, ka_interval);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if (user_timeout>0 && user_timeout<=INT_MAX)
|
||||||
|
sk_set_int(sock.fd, sk_user_timeout, user_timeout);
|
||||||
|
|
||||||
sk_set(sock.fd, sk_congestion, congestion, str_len(congestion));
|
sk_set(sock.fd, sk_congestion, congestion, str_len(congestion));
|
||||||
|
|
||||||
switch (gt_setup_crypto(&ctx, sock.fd, listener)) {
|
switch (gt_setup_crypto(&ctx, sock.fd, listener)) {
|
||||||
@@ -888,7 +1109,7 @@ int main (int argc, char **argv)
|
|||||||
|
|
||||||
retry = 0;
|
retry = 0;
|
||||||
|
|
||||||
gt_log("%s: running\n", sockname);
|
state_write(state_fd, "STARTED\n");
|
||||||
|
|
||||||
fd_set rfds;
|
fd_set rfds;
|
||||||
FD_ZERO(&rfds);
|
FD_ZERO(&rfds);
|
||||||
@@ -908,7 +1129,9 @@ int main (int argc, char **argv)
|
|||||||
goto restart;
|
goto restart;
|
||||||
FD_CLR(tun.fd, &rfds);
|
FD_CLR(tun.fd, &rfds);
|
||||||
} else {
|
} else {
|
||||||
if (!blks[blk_write].size) {
|
buffer_shift(&tun.read);
|
||||||
|
|
||||||
|
if (buffer_write_size(&tun.read)>=GT_MTU_MAX) {
|
||||||
FD_SET(tun.fd, &rfds);
|
FD_SET(tun.fd, &rfds);
|
||||||
} else {
|
} else {
|
||||||
FD_CLR(tun.fd, &rfds);
|
FD_CLR(tun.fd, &rfds);
|
||||||
@@ -939,68 +1162,53 @@ int main (int argc, char **argv)
|
|||||||
// gettimeofday(&now, NULL);
|
// gettimeofday(&now, NULL);
|
||||||
|
|
||||||
if (FD_ISSET(tun.fd, &rfds)) {
|
if (FD_ISSET(tun.fd, &rfds)) {
|
||||||
while (!blks[blk_write].size) {
|
while (1) {
|
||||||
uint8_t *data = blks[blk_write].data;
|
const size_t size = buffer_write_size(&tun.read);
|
||||||
const ssize_t r = tun_read(tun.fd, data, GT_MTU_MAX);
|
|
||||||
|
if (size<GT_MTU_MAX)
|
||||||
|
break;
|
||||||
|
|
||||||
|
const ssize_t r = tun_read(tun.fd, tun.read.write, GT_MTU_MAX);
|
||||||
|
|
||||||
if (r<=0) {
|
if (r<=0) {
|
||||||
gt_close |= !r;
|
gt_close |= !r;
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
|
|
||||||
const int ip_version = ip_get_version(data, GT_MTU_MAX);
|
struct ip_common ic;
|
||||||
const ssize_t ip_size = ip_get_size(ip_version, data, GT_MTU_MAX);
|
|
||||||
|
|
||||||
if _0_(ip_size<=0)
|
if (ip_get_common(&ic, tun.read.write, GT_MTU_MAX))
|
||||||
continue;
|
continue;
|
||||||
|
|
||||||
if _0_(ip_size!=r) {
|
if _0_(ic.size!=r) {
|
||||||
char tmp[2*GT_MTU_MAX+1];
|
char tmp[2*GT_MTU_MAX+1];
|
||||||
gt_tohex(tmp, sizeof(tmp), data, r);
|
gt_tohex(tmp, sizeof(tmp), tun.read.write, r);
|
||||||
gt_log("%s: DUMP %zi %s\n", sockname, r, tmp);
|
gt_log("%s: DUMP %zi %s\n", sockname, r, tmp);
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
|
|
||||||
if _0_(debug)
|
if _0_(debug) {
|
||||||
gt_print_hdr(ip_version, data, ip_size, sockname);
|
gt_print_hdr(&ic, tun.read.write);
|
||||||
|
|
||||||
blks[blk_write++].size = r;
|
if (gt_track(&db, &ic, tun.read.write, 0))
|
||||||
blk_count++;
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
tun.read.write += r;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
while (1) {
|
if _1_(!stop_loop)
|
||||||
buffer_shift(&tun.read);
|
|
||||||
|
|
||||||
if _0_(!stop_loop) {
|
|
||||||
for (; blk_count; blk_read++) {
|
|
||||||
const size_t size = blks[blk_read].size;
|
|
||||||
|
|
||||||
if (!size || buffer_write_size(&tun.read)<size)
|
|
||||||
break;
|
|
||||||
|
|
||||||
byte_cpy(tun.read.write, blks[blk_read].data, size);
|
|
||||||
tun.read.write += size;
|
|
||||||
|
|
||||||
blks[blk_read].size = 0;
|
|
||||||
blk_count--;
|
|
||||||
}
|
|
||||||
|
|
||||||
gt_encrypt(&ctx, &sock.write, &tun.read);
|
gt_encrypt(&ctx, &sock.write, &tun.read);
|
||||||
}
|
|
||||||
|
|
||||||
if (!buffer_read_size(&sock.write))
|
|
||||||
break;
|
|
||||||
|
|
||||||
|
if (buffer_read_size(&sock.write)) {
|
||||||
const ssize_t r = fd_write(sock.fd, sock.write.read,
|
const ssize_t r = fd_write(sock.fd, sock.write.read,
|
||||||
buffer_read_size(&sock.write));
|
buffer_read_size(&sock.write));
|
||||||
|
|
||||||
if (r>0) {
|
if (r>0) {
|
||||||
sock.write.read += r;
|
sock.write.read += r;
|
||||||
} else {
|
} else if (!r) {
|
||||||
if (!r)
|
|
||||||
stop_loop |= (1<<2);
|
stop_loop |= (1<<2);
|
||||||
break;
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -1038,18 +1246,26 @@ int main (int argc, char **argv)
|
|||||||
|
|
||||||
size_t size = buffer_read_size(&tun.write);
|
size_t size = buffer_read_size(&tun.write);
|
||||||
|
|
||||||
const int ip_version = ip_get_version(tun.write.read, size);
|
if (!size)
|
||||||
ssize_t ip_size = ip_get_size(ip_version, tun.write.read, size);
|
break;
|
||||||
|
|
||||||
if _0_(!ip_size) {
|
struct ip_common ic;
|
||||||
|
|
||||||
|
if (ip_get_common(&ic, tun.write.read, size) || ic.size>size) {
|
||||||
gt_log("%s: bad packet!\n", sockname);
|
gt_log("%s: bad packet!\n", sockname);
|
||||||
goto restart;
|
goto restart;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (ip_size<0 || (size_t)ip_size>size)
|
if _0_(debug) {
|
||||||
break;
|
gt_print_hdr(&ic, tun.write.read);
|
||||||
|
|
||||||
ssize_t r = tun_write(tun.fd, tun.write.read, ip_size);
|
if (gt_track(&db, &ic, tun.write.read, 1)) {
|
||||||
|
tun.write.read += ic.size;
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
ssize_t r = tun_write(tun.fd, tun.write.read, ic.size);
|
||||||
|
|
||||||
if (r>0) {
|
if (r>0) {
|
||||||
tun.write.read += r;
|
tun.write.read += r;
|
||||||
@@ -1070,12 +1286,12 @@ int main (int argc, char **argv)
|
|||||||
close(sock.fd);
|
close(sock.fd);
|
||||||
sock.fd = -1;
|
sock.fd = -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
state_write(state_fd, "STOPPED\n");
|
||||||
}
|
}
|
||||||
|
|
||||||
freeaddrinfo(ai);
|
freeaddrinfo(ai);
|
||||||
|
|
||||||
free(blks);
|
|
||||||
|
|
||||||
free(sock.write.data);
|
free(sock.write.data);
|
||||||
free(sock.read.data);
|
free(sock.read.data);
|
||||||
|
|
||||||
|
|||||||
14
src/option.c
14
src/option.c
@@ -1,25 +1,23 @@
|
|||||||
#include "common-static.h"
|
#include "common.h"
|
||||||
|
|
||||||
#include <stdio.h>
|
|
||||||
#include <stdlib.h>
|
|
||||||
|
|
||||||
#include "option.h"
|
#include "option.h"
|
||||||
|
#include "str.h"
|
||||||
|
|
||||||
int option_str (void *data, int argc, char **argv)
|
int option_str (void *data, int argc, char **argv)
|
||||||
{
|
{
|
||||||
if (argc<2 || !argv[1]) {
|
if (argc<2 || str_empty(argv[1])) {
|
||||||
gt_print("option `%s' need a string argument\n", argv[0]);
|
gt_print("option `%s' need a string argument\n", argv[0]);
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
byte_cpy(data, &argv[1], sizeof(argv[1]));
|
memcpy(data, &argv[1], sizeof(argv[1]));
|
||||||
|
|
||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
int option_long (void *data, int argc, char **argv)
|
int option_long (void *data, int argc, char **argv)
|
||||||
{
|
{
|
||||||
if (argc<2 || !argv[1]) {
|
if (argc<2 || str_empty(argv[1])) {
|
||||||
gt_print("option `%s' need an integer argument\n", argv[0]);
|
gt_print("option `%s' need an integer argument\n", argv[0]);
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
@@ -33,7 +31,7 @@ int option_long (void *data, int argc, char **argv)
|
|||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
byte_cpy(data, &val, sizeof(val));
|
memcpy(data, &val, sizeof(val));
|
||||||
|
|
||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
|||||||
73
src/str.h
Normal file
73
src/str.h
Normal file
@@ -0,0 +1,73 @@
|
|||||||
|
#pragma once
|
||||||
|
|
||||||
|
#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)
|
||||||
|
{
|
||||||
|
return !str || !str[0];
|
||||||
|
}
|
||||||
|
|
||||||
|
_pure_
|
||||||
|
static inline size_t str_cmp (const char *restrict sa, const char *restrict sb)
|
||||||
|
{
|
||||||
|
if (!sa || !sb)
|
||||||
|
return 1;
|
||||||
|
|
||||||
|
size_t i = 0;
|
||||||
|
|
||||||
|
while (sa[i]==sb[i])
|
||||||
|
if (!sa[i++])
|
||||||
|
return 0;
|
||||||
|
|
||||||
|
return i+1;
|
||||||
|
}
|
||||||
|
|
||||||
|
_pure_
|
||||||
|
static inline size_t str_len (const char *restrict str)
|
||||||
|
{
|
||||||
|
if (!str)
|
||||||
|
return 0;
|
||||||
|
|
||||||
|
return strlen(str);
|
||||||
|
}
|
||||||
|
|
||||||
|
static inline char *str_cat (const char *const strs[], size_t count)
|
||||||
|
{
|
||||||
|
size_t size = 1;
|
||||||
|
|
||||||
|
for (size_t i=0; i<count; i++)
|
||||||
|
size += str_len(strs[i]);
|
||||||
|
|
||||||
|
char *str = malloc(size);
|
||||||
|
|
||||||
|
if (!str)
|
||||||
|
return NULL;
|
||||||
|
|
||||||
|
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;
|
||||||
|
}
|
||||||
|
|
||||||
|
p[0] = 0;
|
||||||
|
|
||||||
|
return str;
|
||||||
|
}
|
||||||
159
src/tun.c
159
src/tun.c
@@ -1,7 +1,8 @@
|
|||||||
#include "common-static.h"
|
#include "common.h"
|
||||||
#include "ip-static.h"
|
|
||||||
|
|
||||||
#include "tun.h"
|
#include "tun.h"
|
||||||
|
#include "str.h"
|
||||||
|
#include "ip.h"
|
||||||
|
|
||||||
#include <stdio.h>
|
#include <stdio.h>
|
||||||
#include <fcntl.h>
|
#include <fcntl.h>
|
||||||
@@ -22,61 +23,26 @@
|
|||||||
#endif
|
#endif
|
||||||
|
|
||||||
#if defined(__APPLE__) || defined(__OpenBSD__)
|
#if defined(__APPLE__) || defined(__OpenBSD__)
|
||||||
# define GT_BSD_TUN 1
|
#define GT_BSD_TUN
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#ifdef __linux__
|
#ifdef __APPLE__
|
||||||
int tun_create (char *name, int multiqueue)
|
|
||||||
|
static int tun_create_by_id (char *name, size_t size, unsigned id, _unused_ int mq)
|
||||||
{
|
{
|
||||||
int fd = open("/dev/net/tun", O_RDWR);
|
|
||||||
|
|
||||||
if (fd<0) {
|
|
||||||
perror("open /dev/net/tun");
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
|
|
||||||
struct ifreq ifr = {
|
|
||||||
.ifr_flags = IFF_TUN|IFF_NO_PI,
|
|
||||||
};
|
|
||||||
|
|
||||||
if (multiqueue) {
|
|
||||||
#ifdef IFF_MULTI_QUEUE
|
|
||||||
ifr.ifr_flags |= IFF_MULTI_QUEUE;
|
|
||||||
#else
|
|
||||||
gt_na("IFF_MULTI_QUEUE");
|
|
||||||
#endif
|
|
||||||
}
|
|
||||||
|
|
||||||
str_cpy(ifr.ifr_name, name, IFNAMSIZ-1);
|
|
||||||
|
|
||||||
int ret = ioctl(fd, TUNSETIFF, &ifr);
|
|
||||||
|
|
||||||
if (ret<0) {
|
|
||||||
perror("ioctl TUNSETIFF");
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
|
|
||||||
gt_log("tun name: %s\n", ifr.ifr_name);
|
|
||||||
|
|
||||||
return fd;
|
|
||||||
}
|
|
||||||
#elif defined(__APPLE__)
|
|
||||||
int tun_create (_unused_ char *name, _unused_ int mq)
|
|
||||||
{
|
|
||||||
for (unsigned dev_id = 0; dev_id < 32; dev_id++) {
|
|
||||||
struct ctl_info ci;
|
|
||||||
|
|
||||||
byte_set(&ci, 0, sizeof(ci));
|
|
||||||
str_cpy(ci.ctl_name, UTUN_CONTROL_NAME, sizeof(ci.ctl_name)-1);
|
|
||||||
|
|
||||||
int fd = socket(PF_SYSTEM, SOCK_DGRAM, SYSPROTO_CONTROL);
|
int fd = socket(PF_SYSTEM, SOCK_DGRAM, SYSPROTO_CONTROL);
|
||||||
|
|
||||||
if (fd==-1)
|
if (fd==-1)
|
||||||
return -1;
|
return -1;
|
||||||
|
|
||||||
if (ioctl(fd, CTLIOCGINFO, &ci)==-1) {
|
struct ctl_info ci;
|
||||||
|
|
||||||
|
memset(&ci, 0, sizeof(ci));
|
||||||
|
str_cpy(ci.ctl_name, UTUN_CONTROL_NAME, sizeof(ci.ctl_name)-1);
|
||||||
|
|
||||||
|
if (ioctl(fd, CTLIOCGINFO, &ci)) {
|
||||||
close(fd);
|
close(fd);
|
||||||
continue;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
struct sockaddr_ctl sc = {
|
struct sockaddr_ctl sc = {
|
||||||
@@ -84,43 +50,110 @@ int tun_create (_unused_ char *name, _unused_ int mq)
|
|||||||
.sc_len = sizeof(sc),
|
.sc_len = sizeof(sc),
|
||||||
.sc_family = AF_SYSTEM,
|
.sc_family = AF_SYSTEM,
|
||||||
.ss_sysaddr = AF_SYS_CONTROL,
|
.ss_sysaddr = AF_SYS_CONTROL,
|
||||||
.sc_unit = dev_id+1,
|
.sc_unit = id+1,
|
||||||
};
|
};
|
||||||
|
|
||||||
if (connect(fd, (struct sockaddr *)&sc, sizeof(sc))==-1) {
|
if (connect(fd, (struct sockaddr *)&sc, sizeof(sc))) {
|
||||||
close(fd);
|
close(fd);
|
||||||
continue;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
gt_log("tun name: /dev/utun%u\n", dev_id);
|
snprintf(name, size, "utun%u", id);
|
||||||
|
|
||||||
return fd;
|
return fd;
|
||||||
}
|
}
|
||||||
|
|
||||||
return -1;
|
static int tun_create_by_name (char *name, size_t size, char *dev_name, int mq)
|
||||||
}
|
|
||||||
#else
|
|
||||||
int tun_create (_unused_ char *name, _unused_ int mq)
|
|
||||||
{
|
{
|
||||||
for (unsigned dev_id = 0; dev_id < 32; dev_id++) {
|
unsigned id = 0;
|
||||||
char dev_path[11];
|
|
||||||
|
|
||||||
snprintf(dev_path, sizeof(dev_path), "/dev/tun%u", dev_id);
|
if (sscanf(dev_name, "utun%u", &id)!=1)
|
||||||
|
return -1;
|
||||||
|
|
||||||
int fd = open(dev_path, O_RDWR);
|
return tun_create_by_id(name, size, id, mq);
|
||||||
|
}
|
||||||
|
|
||||||
|
#else /* not __APPLE__ */
|
||||||
|
|
||||||
|
#ifdef __linux__
|
||||||
|
|
||||||
|
static int tun_create_by_name (char *name, size_t size, char *dev_name, int mq)
|
||||||
|
{
|
||||||
|
int fd = open("/dev/net/tun", O_RDWR);
|
||||||
|
|
||||||
if (fd==-1)
|
if (fd==-1)
|
||||||
continue;
|
return -1;
|
||||||
|
|
||||||
gt_log("tun name: %s\n", dev_path);
|
struct ifreq ifr = {
|
||||||
|
.ifr_flags = IFF_TUN|IFF_NO_PI,
|
||||||
|
};
|
||||||
|
|
||||||
|
if (mq) {
|
||||||
|
#ifdef IFF_MULTI_QUEUE
|
||||||
|
ifr.ifr_flags |= IFF_MULTI_QUEUE;
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
str_cpy(ifr.ifr_name, dev_name, IFNAMSIZ-1);
|
||||||
|
|
||||||
|
if (ioctl(fd, TUNSETIFF, &ifr)) {
|
||||||
|
close(fd);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
str_cpy(name, ifr.ifr_name, size-1);
|
||||||
|
|
||||||
return fd;
|
return fd;
|
||||||
}
|
}
|
||||||
|
|
||||||
return -1;
|
#else /* not __linux__ not __APPLE__ */
|
||||||
|
|
||||||
|
static int tun_create_by_name (char *name, size_t size, char *dev_name, _unused_ int mq)
|
||||||
|
{
|
||||||
|
char path[64];
|
||||||
|
|
||||||
|
snprintf(path, sizeof(path), "/dev/%s", dev_name);
|
||||||
|
str_cpy(name, dev_name, size-1);
|
||||||
|
|
||||||
|
return open(path, O_RDWR);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#endif /* not __APPLE__ */
|
||||||
|
|
||||||
|
static int tun_create_by_id (char *name, size_t size, unsigned id, int mq)
|
||||||
|
{
|
||||||
|
char dev_name[64];
|
||||||
|
|
||||||
|
snprintf(dev_name, sizeof(dev_name), "tun%u", id);
|
||||||
|
|
||||||
|
return tun_create_by_name(name, size, dev_name, mq);
|
||||||
|
}
|
||||||
|
|
||||||
#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)
|
||||||
|
|||||||
10
version.sh
10
version.sh
@@ -3,7 +3,11 @@
|
|||||||
[ -z "${VERSION}" ] && VERSION=`git describe --tags --always 2>/dev/null` \
|
[ -z "${VERSION}" ] && VERSION=`git describe --tags --always 2>/dev/null` \
|
||||||
&& VERSION=${VERSION#v}
|
&& VERSION=${VERSION#v}
|
||||||
|
|
||||||
[ -z "${VERSION}" ] && VERSION=`basename \`pwd\`` \
|
[ -z "${VERSION}" ] && VERSION=`cat VERSION 2>/dev/null`
|
||||||
&& VERSION=${VERSION#*-}
|
|
||||||
|
|
||||||
printf ${VERSION}
|
[ -z "${VERSION}" ] && VERSION=0.0.0
|
||||||
|
|
||||||
|
[ "$1" = "major" ] && printf ${VERSION%%.*} \
|
||||||
|
&& exit 0
|
||||||
|
|
||||||
|
printf ${VERSION} | tee VERSION
|
||||||
|
|||||||
Reference in New Issue
Block a user