Compare commits
5 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
8989138051 | ||
|
|
c2f76213cc | ||
|
|
6ed736327a | ||
|
|
e20be0ad97 | ||
|
|
2e7355bb92 |
@@ -1,5 +1,5 @@
|
|||||||
AC_PREREQ([2.65])
|
AC_PREREQ([2.65])
|
||||||
AC_INIT([glorytun], [0.0.7], [https://github.com/angt/glorytun/issues],
|
AC_INIT([glorytun], [0.0.8], [https://github.com/angt/glorytun/issues],
|
||||||
[glorytun], [https://github.com/angt/glorytun])
|
[glorytun], [https://github.com/angt/glorytun])
|
||||||
AC_CONFIG_SRCDIR([src/common.h])
|
AC_CONFIG_SRCDIR([src/common.h])
|
||||||
AC_CONFIG_AUX_DIR([build-aux])
|
AC_CONFIG_AUX_DIR([build-aux])
|
||||||
|
|||||||
@@ -14,9 +14,13 @@
|
|||||||
#define PALIGN(x) ((void *)ALIGN((size_t)(x)))
|
#define PALIGN(x) ((void *)ALIGN((size_t)(x)))
|
||||||
#define PALIGN_DOWN(x) ((void *)ALIGN_DOWN((size_t)(x)))
|
#define PALIGN_DOWN(x) ((void *)ALIGN_DOWN((size_t)(x)))
|
||||||
|
|
||||||
|
#define _1_(x) (__builtin_expect((x), 1))
|
||||||
|
#define _0_(x) (__builtin_expect((x), 0))
|
||||||
|
|
||||||
#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))
|
||||||
|
#define _align_(...) __attribute__((aligned(__VA_ARGS__)))
|
||||||
|
|
||||||
typedef struct buffer buffer_t;
|
typedef struct buffer buffer_t;
|
||||||
|
|
||||||
|
|||||||
73
src/main.c
73
src/main.c
@@ -35,9 +35,8 @@ struct fdbuf {
|
|||||||
};
|
};
|
||||||
|
|
||||||
struct blk {
|
struct blk {
|
||||||
uint8_t prio;
|
|
||||||
size_t size;
|
size_t size;
|
||||||
uint8_t data[GT_MTU_MAX];
|
uint8_t data[GT_MTU_MAX] _align_(16);
|
||||||
};
|
};
|
||||||
|
|
||||||
struct crypto_ctx {
|
struct crypto_ctx {
|
||||||
@@ -84,6 +83,7 @@ enum sk_opt {
|
|||||||
sk_keepintvl,
|
sk_keepintvl,
|
||||||
sk_congestion,
|
sk_congestion,
|
||||||
sk_defer_accept,
|
sk_defer_accept,
|
||||||
|
sk_quickack,
|
||||||
};
|
};
|
||||||
|
|
||||||
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)
|
||||||
@@ -123,6 +123,11 @@ static void sk_set (int fd, enum sk_opt opt, const void *val, socklen_t len)
|
|||||||
[sk_defer_accept] = { "TCP_DEFER_ACCEPT",
|
[sk_defer_accept] = { "TCP_DEFER_ACCEPT",
|
||||||
#ifdef TCP_DEFER_ACCEPT
|
#ifdef TCP_DEFER_ACCEPT
|
||||||
1, IPPROTO_TCP, TCP_DEFER_ACCEPT,
|
1, IPPROTO_TCP, TCP_DEFER_ACCEPT,
|
||||||
|
#endif
|
||||||
|
},
|
||||||
|
[sk_quickack] = { "TCP_QUICKACK",
|
||||||
|
#ifdef TCP_QUICKACK
|
||||||
|
1, IPPROTO_TCP, TCP_QUICKACK,
|
||||||
#endif
|
#endif
|
||||||
},
|
},
|
||||||
};
|
};
|
||||||
@@ -638,8 +643,6 @@ int main (int argc, char **argv)
|
|||||||
long ka_count = -1;
|
long ka_count = -1;
|
||||||
long ka_idle = -1;
|
long ka_idle = -1;
|
||||||
long ka_interval = -1;
|
long ka_interval = -1;
|
||||||
long prio_dscp = 46;
|
|
||||||
long prio_size = (GT_TUNR_SIZE*3)/4;
|
|
||||||
|
|
||||||
#ifdef TCP_INFO
|
#ifdef TCP_INFO
|
||||||
struct {
|
struct {
|
||||||
@@ -655,12 +658,6 @@ int main (int argc, char **argv)
|
|||||||
{ NULL },
|
{ NULL },
|
||||||
};
|
};
|
||||||
|
|
||||||
struct option prio_opts[] = {
|
|
||||||
{ "dscp", &prio_dscp, option_long },
|
|
||||||
{ "size", &prio_size, option_long },
|
|
||||||
{ NULL },
|
|
||||||
};
|
|
||||||
|
|
||||||
struct option opts[] = {
|
struct option opts[] = {
|
||||||
{ "listener", NULL, option_option },
|
{ "listener", NULL, option_option },
|
||||||
{ "host", &host, option_str },
|
{ "host", &host, option_str },
|
||||||
@@ -672,7 +669,7 @@ int main (int argc, char **argv)
|
|||||||
{ "multiqueue", NULL, option_option },
|
{ "multiqueue", NULL, option_option },
|
||||||
{ "keepalive", ka_opts, option_option },
|
{ "keepalive", ka_opts, option_option },
|
||||||
{ "buffer-size", &buffer_size, option_long },
|
{ "buffer-size", &buffer_size, option_long },
|
||||||
{ "priority", prio_opts, option_option },
|
{ "noquickack", NULL, option_option },
|
||||||
{ "daemon", NULL, option_option },
|
{ "daemon", NULL, option_option },
|
||||||
{ "debug", NULL, option_option },
|
{ "debug", NULL, option_option },
|
||||||
{ "version", NULL, option_option },
|
{ "version", NULL, option_option },
|
||||||
@@ -691,22 +688,13 @@ int main (int argc, char **argv)
|
|||||||
int delay = option_is_set(opts, "delay");
|
int delay = option_is_set(opts, "delay");
|
||||||
int debug = option_is_set(opts, "debug");
|
int debug = option_is_set(opts, "debug");
|
||||||
int keepalive = option_is_set(opts, "keepalive");
|
int keepalive = option_is_set(opts, "keepalive");
|
||||||
|
int noquickack = option_is_set(opts, "noquickack");
|
||||||
|
|
||||||
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 (prio_size < 0) {
|
|
||||||
prio_size = 0;
|
|
||||||
gt_log("priority size must be positive!\n");
|
|
||||||
}
|
|
||||||
|
|
||||||
if (prio_size > GT_TUNR_SIZE) {
|
|
||||||
prio_size = GT_TUNR_SIZE;
|
|
||||||
gt_log("priority size must be less than or equal to %zu\n", GT_TUNR_SIZE);
|
|
||||||
}
|
|
||||||
|
|
||||||
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;
|
||||||
@@ -737,7 +725,6 @@ int main (int argc, char **argv)
|
|||||||
|
|
||||||
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;
|
||||||
size_t blk_prio = 0;
|
|
||||||
uint8_t blk_read = 0;
|
uint8_t blk_read = 0;
|
||||||
uint8_t blk_write = 0;
|
uint8_t blk_write = 0;
|
||||||
|
|
||||||
@@ -893,13 +880,6 @@ int main (int argc, char **argv)
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if (ip_get_dscp(data, GT_MTU_MAX)==prio_dscp) {
|
|
||||||
blks[blk_write].prio = 1;
|
|
||||||
blk_prio++;
|
|
||||||
} else {
|
|
||||||
blks[blk_write].prio = 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
blks[blk_write++].size = r;
|
blks[blk_write++].size = r;
|
||||||
blk_count++;
|
blk_count++;
|
||||||
}
|
}
|
||||||
@@ -907,32 +887,6 @@ int main (int argc, char **argv)
|
|||||||
|
|
||||||
buffer_shift(&tun.read);
|
buffer_shift(&tun.read);
|
||||||
|
|
||||||
// XXX prio code needs a full rewrite :)
|
|
||||||
|
|
||||||
if (blk_prio) {
|
|
||||||
uint8_t k = blk_read;
|
|
||||||
|
|
||||||
while (blk_prio && buffer_read_size(&tun.read)<(size_t)prio_size) {
|
|
||||||
while (!blks[k].prio || !blks[k].size)
|
|
||||||
k++;
|
|
||||||
|
|
||||||
if (buffer_write_size(&tun.read)<blks[k].size)
|
|
||||||
break;
|
|
||||||
|
|
||||||
byte_cpy(tun.read.write, blks[k].data, blks[k].size);
|
|
||||||
tun.read.write += blks[k].size;
|
|
||||||
|
|
||||||
blks[k].size = 0;
|
|
||||||
blk_count--;
|
|
||||||
blk_prio--;
|
|
||||||
|
|
||||||
if (blk_read==k)
|
|
||||||
blk_read++;
|
|
||||||
|
|
||||||
k++;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
while (blk_count) {
|
while (blk_count) {
|
||||||
if (!blks[blk_read].size) {
|
if (!blks[blk_read].size) {
|
||||||
blk_read++;
|
blk_read++;
|
||||||
@@ -945,9 +899,6 @@ int main (int argc, char **argv)
|
|||||||
byte_cpy(tun.read.write, blks[blk_read].data, blks[blk_read].size);
|
byte_cpy(tun.read.write, blks[blk_read].data, blks[blk_read].size);
|
||||||
tun.read.write += blks[blk_read].size;
|
tun.read.write += blks[blk_read].size;
|
||||||
|
|
||||||
if (blks[blk_read].prio)
|
|
||||||
blk_prio--;
|
|
||||||
|
|
||||||
blks[blk_read++].size = 0;
|
blks[blk_read++].size = 0;
|
||||||
blk_count--;
|
blk_count--;
|
||||||
}
|
}
|
||||||
@@ -967,7 +918,8 @@ int main (int argc, char **argv)
|
|||||||
if (r>0)
|
if (r>0)
|
||||||
sock.write.read += r;
|
sock.write.read += r;
|
||||||
} else {
|
} else {
|
||||||
if (stop_loop) {
|
if (stop_loop && !(stop_loop>>2)) {
|
||||||
|
stop_loop |= (1<<2);
|
||||||
gt_log("%s: shutdown\n", sockname);
|
gt_log("%s: shutdown\n", sockname);
|
||||||
shutdown(sock.fd, SHUT_WR);
|
shutdown(sock.fd, SHUT_WR);
|
||||||
}
|
}
|
||||||
@@ -977,6 +929,9 @@ int main (int argc, char **argv)
|
|||||||
buffer_shift(&sock.read);
|
buffer_shift(&sock.read);
|
||||||
|
|
||||||
if (FD_ISSET(sock.fd, &rfds)) {
|
if (FD_ISSET(sock.fd, &rfds)) {
|
||||||
|
if (noquickack)
|
||||||
|
sk_set_int(sock.fd, sk_quickack, 0);
|
||||||
|
|
||||||
ssize_t r = fd_read(sock.fd, sock.read.write,
|
ssize_t r = fd_read(sock.fd, sock.read.write,
|
||||||
buffer_write_size(&sock.read));
|
buffer_write_size(&sock.read));
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user