Compare commits
6 Commits
v0.0.72-mu
...
v0.0.77-mu
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
3675820555 | ||
|
|
3ba1b914d6 | ||
|
|
1dbf3fcc80 | ||
|
|
6167145454 | ||
|
|
97ae583707 | ||
|
|
0d97df6b50 |
2
mud
2
mud
Submodule mud updated: e1fd7f8b0d...ef260f90de
60
src/common.c
60
src/common.c
@@ -1,22 +1,24 @@
|
|||||||
#include "common.h"
|
#include "common.h"
|
||||||
|
|
||||||
#include <stdio.h>
|
|
||||||
#include <stdarg.h>
|
#include <stdarg.h>
|
||||||
|
#include <stdio.h>
|
||||||
|
|
||||||
int gt_print (const char *fmt, ...)
|
int
|
||||||
|
gt_print(const char *fmt, ...)
|
||||||
{
|
{
|
||||||
va_list ap;
|
va_list ap;
|
||||||
va_start(ap, fmt);
|
va_start(ap, fmt);
|
||||||
int ret = vfprintf(stdout, fmt, ap);
|
int ret = vfprintf(stdout, fmt, ap);
|
||||||
va_end(ap);
|
va_end(ap);
|
||||||
|
|
||||||
if (ret<0)
|
if (ret < 0)
|
||||||
return 0;
|
return 0;
|
||||||
|
|
||||||
return ret;
|
return ret;
|
||||||
}
|
}
|
||||||
|
|
||||||
void gt_log (const char *fmt, ...)
|
void
|
||||||
|
gt_log(const char *fmt, ...)
|
||||||
{
|
{
|
||||||
va_list ap;
|
va_list ap;
|
||||||
va_start(ap, fmt);
|
va_start(ap, fmt);
|
||||||
@@ -24,7 +26,8 @@ void gt_log (const char *fmt, ...)
|
|||||||
va_end(ap);
|
va_end(ap);
|
||||||
}
|
}
|
||||||
|
|
||||||
void gt_fatal (const char *fmt, ...)
|
void
|
||||||
|
gt_fatal(const char *fmt, ...)
|
||||||
{
|
{
|
||||||
va_list ap;
|
va_list ap;
|
||||||
va_start(ap, fmt);
|
va_start(ap, fmt);
|
||||||
@@ -34,24 +37,26 @@ void gt_fatal (const char *fmt, ...)
|
|||||||
exit(EXIT_FAILURE);
|
exit(EXIT_FAILURE);
|
||||||
}
|
}
|
||||||
|
|
||||||
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)
|
||||||
{
|
{
|
||||||
if _0_(!dst_size)
|
if (_0_(!dst_size))
|
||||||
return -1;
|
return -1;
|
||||||
|
|
||||||
if _0_(((dst_size-1)/2)<src_size)
|
if (_0_(((dst_size - 1) / 2) < src_size))
|
||||||
return -1;
|
return -1;
|
||||||
|
|
||||||
static const char tbl[] = "0123456789ABCDEF";
|
static const char tbl[] = "0123456789ABCDEF";
|
||||||
|
|
||||||
for (size_t i=0; i<src_size; i++) {
|
for (size_t i = 0; i < src_size; i++) {
|
||||||
*dst++ = tbl[0xF&(src[i]>>4)];
|
*dst++ = tbl[0xF & (src[i] >> 4)];
|
||||||
*dst++ = tbl[0xF&(src[i])];
|
*dst++ = tbl[0xF & (src[i])];
|
||||||
}
|
}
|
||||||
|
|
||||||
*dst = 0;
|
*dst = 0;
|
||||||
@@ -59,37 +64,38 @@ int gt_tohex (char *dst, size_t dst_size, const uint8_t *src, size_t src_size)
|
|||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
_const_
|
_const_ static inline int
|
||||||
static inline int fromhex (const char c)
|
fromhex(const char c)
|
||||||
{
|
{
|
||||||
if (c>='0' && c<='9')
|
if (c >= '0' && c <= '9')
|
||||||
return c-'0';
|
return c - '0';
|
||||||
|
|
||||||
if (c>='A' && c<='F')
|
if (c >= 'A' && c <= 'F')
|
||||||
return c-'A'+10;
|
return c - 'A' + 10;
|
||||||
|
|
||||||
if (c>='a' && c<='f')
|
if (c >= 'a' && c <= 'f')
|
||||||
return c-'a'+10;
|
return c - 'a' + 10;
|
||||||
|
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
int gt_fromhex (uint8_t *dst, size_t dst_size, const char *src, size_t src_size)
|
int
|
||||||
|
gt_fromhex(uint8_t *dst, size_t dst_size, const char *src, size_t src_size)
|
||||||
{
|
{
|
||||||
if _0_(src_size&1)
|
if (_0_(src_size & 1))
|
||||||
return -1;
|
return -1;
|
||||||
|
|
||||||
if _0_(dst_size<(src_size/2))
|
if (_0_(dst_size < (src_size / 2)))
|
||||||
return -1;
|
return -1;
|
||||||
|
|
||||||
for (size_t i=0; i<src_size; i+=2) {
|
for (size_t i = 0; i < src_size; i += 2) {
|
||||||
const int a = fromhex(src[i]);
|
const int a = fromhex(src[i]);
|
||||||
const int b = fromhex(src[i+1]);
|
const int b = fromhex(src[i + 1]);
|
||||||
|
|
||||||
if _0_(a==-1 || b==-1)
|
if (_0_(a == -1 || b == -1))
|
||||||
return -1;
|
return -1;
|
||||||
|
|
||||||
*dst++ = (a<<4)|b;
|
*dst++ = (a << 4) | b;
|
||||||
}
|
}
|
||||||
|
|
||||||
return 0;
|
return 0;
|
||||||
|
|||||||
79
src/db.c
79
src/db.c
@@ -3,56 +3,57 @@
|
|||||||
#include "db.h"
|
#include "db.h"
|
||||||
#include "str.h"
|
#include "str.h"
|
||||||
|
|
||||||
#define CBIT(X) (1&(intptr_t)(X))
|
#define CBIT(X) (1 & (intptr_t)(X))
|
||||||
#define CBIT_PTR(X) (uint8_t *)(1|(intptr_t)(X))
|
#define CBIT_PTR(X) (uint8_t *)(1 | (intptr_t)(X))
|
||||||
#define CBIT_NODE(X) (struct node *)(1^(intptr_t)(X))
|
#define CBIT_NODE(X) (struct node *)(1 ^ (intptr_t)(X))
|
||||||
|
|
||||||
struct node {
|
struct node {
|
||||||
uint8_t *child[2];
|
uint8_t *child[2];
|
||||||
uint32_t point;
|
uint32_t point;
|
||||||
};
|
};
|
||||||
|
|
||||||
_pure_
|
_pure_ static inline size_t
|
||||||
static inline size_t db_size (const uint8_t *a)
|
db_size(const uint8_t *a)
|
||||||
{
|
{
|
||||||
return (a[0]?:str_len((char *)a+1))+1;
|
return (a[0] ?: str_len((char *)a + 1)) + 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
_pure_
|
_pure_ static inline size_t
|
||||||
static inline size_t db_cmp (const uint8_t *a, const uint8_t *b)
|
db_cmp(const uint8_t *a, const uint8_t *b)
|
||||||
{
|
{
|
||||||
const size_t size = a[0];
|
const size_t size = a[0];
|
||||||
|
|
||||||
if (size!=b[0])
|
if (size != b[0])
|
||||||
return 1;
|
return 1;
|
||||||
|
|
||||||
if (!size) {
|
if (!size) {
|
||||||
size_t i = str_cmp((char *)a+1, (char *)b+1);
|
size_t i = str_cmp((char *)a + 1, (char *)b + 1);
|
||||||
return i?i+1:0;
|
return i ? i + 1 : 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
for (size_t i=1; i<=size; i++) {
|
for (size_t i = 1; i <= size; i++) {
|
||||||
if (a[i]!=b[i])
|
if (a[i] != b[i])
|
||||||
return i+1;
|
return i + 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
_pure_
|
_pure_ static inline int
|
||||||
static inline int db_dir (const uint32_t point, uint8_t *data, const size_t size)
|
db_dir(const uint32_t point, uint8_t *data, const size_t size)
|
||||||
{
|
{
|
||||||
const size_t pos = point>>8;
|
const size_t pos = point >> 8;
|
||||||
|
|
||||||
if (pos>=size)
|
if (pos >= size)
|
||||||
return 0;
|
return 0;
|
||||||
|
|
||||||
return ((point|data[pos])&255)==255;
|
return ((point | data[pos]) & 255) == 255;
|
||||||
}
|
}
|
||||||
|
|
||||||
uint8_t *db_search (uint8_t **p, uint8_t *data)
|
uint8_t *
|
||||||
|
db_search(uint8_t **p, uint8_t *data)
|
||||||
{
|
{
|
||||||
if _0_(!*p)
|
if (_0_(!*p))
|
||||||
return NULL;
|
return NULL;
|
||||||
|
|
||||||
uint8_t *r = *p;
|
uint8_t *r = *p;
|
||||||
@@ -69,12 +70,13 @@ uint8_t *db_search (uint8_t **p, uint8_t *data)
|
|||||||
return NULL;
|
return NULL;
|
||||||
}
|
}
|
||||||
|
|
||||||
uint8_t *db_insert (uint8_t **p, uint8_t *data)
|
uint8_t *
|
||||||
|
db_insert(uint8_t **p, uint8_t *data)
|
||||||
{
|
{
|
||||||
if _0_(CBIT(data))
|
if (_0_(CBIT(data)))
|
||||||
return NULL;
|
return NULL;
|
||||||
|
|
||||||
if _0_(!*p) {
|
if (_0_(!*p)) {
|
||||||
*p = data;
|
*p = data;
|
||||||
return data;
|
return data;
|
||||||
}
|
}
|
||||||
@@ -89,31 +91,31 @@ uint8_t *db_insert (uint8_t **p, uint8_t *data)
|
|||||||
|
|
||||||
const size_t diff = db_cmp(r, data);
|
const size_t diff = db_cmp(r, data);
|
||||||
|
|
||||||
if _0_(!diff)
|
if (_0_(!diff))
|
||||||
return r;
|
return r;
|
||||||
|
|
||||||
const size_t pos = diff-1;
|
const size_t pos = diff - 1;
|
||||||
const uint8_t mask = ~((1u<<31)>>CLZ(r[pos]^data[pos]));
|
const uint8_t mask = ~((1u << 31) >> CLZ(r[pos] ^ data[pos]));
|
||||||
const size_t point = (pos<<8)|mask;
|
const size_t point = (pos << 8) | mask;
|
||||||
|
|
||||||
while (CBIT(*p)) {
|
while (CBIT(*p)) {
|
||||||
struct node *node = CBIT_NODE(*p);
|
struct node *node = CBIT_NODE(*p);
|
||||||
|
|
||||||
if (node->point>point)
|
if (node->point > point)
|
||||||
break;
|
break;
|
||||||
|
|
||||||
p = node->child+db_dir(node->point, data, size);
|
p = node->child + db_dir(node->point, data, size);
|
||||||
}
|
}
|
||||||
|
|
||||||
struct node *node = malloc(sizeof(struct node));
|
struct node *node = malloc(sizeof(struct node));
|
||||||
|
|
||||||
if _0_(!node)
|
if (_0_(!node))
|
||||||
return NULL;
|
return NULL;
|
||||||
|
|
||||||
const int dir = (mask|r[pos])==255;
|
const int dir = (mask | r[pos]) == 255;
|
||||||
|
|
||||||
node->child[dir] = *p;
|
node->child[dir] = *p;
|
||||||
node->child[1-dir] = data;
|
node->child[1 - dir] = data;
|
||||||
node->point = point;
|
node->point = point;
|
||||||
|
|
||||||
*p = CBIT_PTR(node);
|
*p = CBIT_PTR(node);
|
||||||
@@ -121,9 +123,10 @@ uint8_t *db_insert (uint8_t **p, uint8_t *data)
|
|||||||
return data;
|
return data;
|
||||||
}
|
}
|
||||||
|
|
||||||
uint8_t *db_remove (uint8_t **p, uint8_t *data)
|
uint8_t *
|
||||||
|
db_remove(uint8_t **p, uint8_t *data)
|
||||||
{
|
{
|
||||||
if _0_(!*p)
|
if (_0_(!*p))
|
||||||
return NULL;
|
return NULL;
|
||||||
|
|
||||||
const size_t size = db_size(data);
|
const size_t size = db_size(data);
|
||||||
@@ -136,16 +139,16 @@ uint8_t *db_remove (uint8_t **p, uint8_t *data)
|
|||||||
p_old = p;
|
p_old = p;
|
||||||
node = CBIT_NODE(*p);
|
node = CBIT_NODE(*p);
|
||||||
dir = db_dir(node->point, data, size);
|
dir = db_dir(node->point, data, size);
|
||||||
p = node->child+dir;
|
p = node->child + dir;
|
||||||
}
|
}
|
||||||
|
|
||||||
if _0_(db_cmp(data, *p))
|
if (_0_(db_cmp(data, *p)))
|
||||||
return NULL;
|
return NULL;
|
||||||
|
|
||||||
uint8_t *r = *p;
|
uint8_t *r = *p;
|
||||||
|
|
||||||
if (p_old) {
|
if (p_old) {
|
||||||
*p_old = node->child[1-dir];
|
*p_old = node->child[1 - dir];
|
||||||
free(node);
|
free(node);
|
||||||
} else {
|
} else {
|
||||||
*p = NULL;
|
*p = NULL;
|
||||||
|
|||||||
175
src/main.c
175
src/main.c
@@ -1,19 +1,19 @@
|
|||||||
#include "common.h"
|
#include "common.h"
|
||||||
|
|
||||||
#include "buffer.h"
|
#include "buffer.h"
|
||||||
#include "ip.h"
|
|
||||||
#include "str.h"
|
|
||||||
#include "option.h"
|
|
||||||
#include "tun.h"
|
|
||||||
#include "db.h"
|
#include "db.h"
|
||||||
|
#include "ip.h"
|
||||||
|
#include "option.h"
|
||||||
#include "state.h"
|
#include "state.h"
|
||||||
|
#include "str.h"
|
||||||
|
#include "tun.h"
|
||||||
|
|
||||||
|
#include <fcntl.h>
|
||||||
#include <inttypes.h>
|
#include <inttypes.h>
|
||||||
#include <limits.h>
|
#include <limits.h>
|
||||||
#include <stdio.h>
|
|
||||||
#include <signal.h>
|
|
||||||
#include <poll.h>
|
#include <poll.h>
|
||||||
#include <fcntl.h>
|
#include <signal.h>
|
||||||
|
#include <stdio.h>
|
||||||
#include <sys/socket.h>
|
#include <sys/socket.h>
|
||||||
#include <sys/time.h>
|
#include <sys/time.h>
|
||||||
|
|
||||||
@@ -33,25 +33,27 @@ static struct {
|
|||||||
int state_fd;
|
int state_fd;
|
||||||
} gt;
|
} gt;
|
||||||
|
|
||||||
static void fd_set_nonblock (int fd)
|
static void
|
||||||
|
fd_set_nonblock(int fd)
|
||||||
{
|
{
|
||||||
int ret;
|
int ret;
|
||||||
|
|
||||||
do {
|
do {
|
||||||
ret = fcntl(fd, F_GETFL, 0);
|
ret = fcntl(fd, F_GETFL, 0);
|
||||||
} while (ret==-1 && errno==EINTR);
|
} while (ret == -1 && errno == EINTR);
|
||||||
|
|
||||||
int flags = (ret==-1)?0:ret;
|
int flags = (ret == -1) ? 0 : ret;
|
||||||
|
|
||||||
do {
|
do {
|
||||||
ret = fcntl(fd, F_SETFL, flags|O_NONBLOCK);
|
ret = fcntl(fd, F_SETFL, flags | O_NONBLOCK);
|
||||||
} while (ret==-1 && errno==EINTR);
|
} while (ret == -1 && errno == EINTR);
|
||||||
|
|
||||||
if (ret==-1)
|
if (ret == -1)
|
||||||
perror("fcntl O_NONBLOCK");
|
perror("fcntl O_NONBLOCK");
|
||||||
}
|
}
|
||||||
|
|
||||||
static void gt_sa_handler (int sig)
|
static void
|
||||||
|
gt_sa_handler(int sig)
|
||||||
{
|
{
|
||||||
switch (sig) {
|
switch (sig) {
|
||||||
case SIGINT:
|
case SIGINT:
|
||||||
@@ -65,7 +67,8 @@ static void gt_sa_handler (int sig)
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
static void gt_set_signal (void)
|
static void
|
||||||
|
gt_set_signal(void)
|
||||||
{
|
{
|
||||||
struct sigaction sa = {
|
struct sigaction sa = {
|
||||||
.sa_flags = 0,
|
.sa_flags = 0,
|
||||||
@@ -84,15 +87,16 @@ static void gt_set_signal (void)
|
|||||||
sigaction(SIGPIPE, &sa, NULL);
|
sigaction(SIGPIPE, &sa, NULL);
|
||||||
}
|
}
|
||||||
|
|
||||||
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 ((fd==-1) || !size)
|
if ((fd == -1) || !size)
|
||||||
return -1;
|
return -1;
|
||||||
|
|
||||||
ssize_t ret = read(fd, data, size);
|
ssize_t ret = read(fd, data, size);
|
||||||
|
|
||||||
if (ret==-1) {
|
if (ret == -1) {
|
||||||
if (errno==EAGAIN || errno==EINTR)
|
if (errno == EAGAIN || errno == EINTR)
|
||||||
return -1;
|
return -1;
|
||||||
|
|
||||||
if (errno)
|
if (errno)
|
||||||
@@ -104,18 +108,19 @@ static ssize_t fd_read (int fd, void *data, size_t size)
|
|||||||
return ret;
|
return ret;
|
||||||
}
|
}
|
||||||
|
|
||||||
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 ((fd==-1) || !size)
|
if ((fd == -1) || !size)
|
||||||
return -1;
|
return -1;
|
||||||
|
|
||||||
ssize_t ret = write(fd, data, size);
|
ssize_t ret = write(fd, data, size);
|
||||||
|
|
||||||
if (ret==-1) {
|
if (ret == -1) {
|
||||||
if (errno==EAGAIN || errno==EINTR)
|
if (errno == EAGAIN || errno == EINTR)
|
||||||
return -1;
|
return -1;
|
||||||
|
|
||||||
if (errno==EPIPE || errno==ECONNRESET)
|
if (errno == EPIPE || errno == ECONNRESET)
|
||||||
return 0;
|
return 0;
|
||||||
|
|
||||||
if (errno)
|
if (errno)
|
||||||
@@ -127,17 +132,18 @@ static ssize_t fd_write (int fd, const void *data, size_t size)
|
|||||||
return ret;
|
return ret;
|
||||||
}
|
}
|
||||||
|
|
||||||
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;
|
||||||
|
|
||||||
while (done<size) {
|
while (done < size) {
|
||||||
ssize_t ret = fd_read(fd, (uint8_t *)data+done, size-done);
|
ssize_t ret = fd_read(fd, (uint8_t *)data + done, size - done);
|
||||||
|
|
||||||
if (!ret)
|
if (!ret)
|
||||||
break;
|
break;
|
||||||
|
|
||||||
if (ret<0) {
|
if (ret < 0) {
|
||||||
struct pollfd pollfd = {
|
struct pollfd pollfd = {
|
||||||
.fd = fd,
|
.fd = fd,
|
||||||
.events = POLLIN,
|
.events = POLLIN,
|
||||||
@@ -155,17 +161,18 @@ static size_t fd_read_all (int fd, void *data, size_t size)
|
|||||||
return done;
|
return done;
|
||||||
}
|
}
|
||||||
|
|
||||||
static size_t fd_write_all (int fd, const void *data, size_t size)
|
static size_t
|
||||||
|
fd_write_all(int fd, const void *data, size_t size)
|
||||||
{
|
{
|
||||||
size_t done = 0;
|
size_t done = 0;
|
||||||
|
|
||||||
while (done<size) {
|
while (done < size) {
|
||||||
ssize_t ret = fd_write(fd, (const uint8_t *)data+done, size-done);
|
ssize_t ret = fd_write(fd, (const uint8_t *)data + done, size - done);
|
||||||
|
|
||||||
if (!ret)
|
if (!ret)
|
||||||
break;
|
break;
|
||||||
|
|
||||||
if (ret<0) {
|
if (ret < 0) {
|
||||||
struct pollfd pollfd = {
|
struct pollfd pollfd = {
|
||||||
.fd = fd,
|
.fd = fd,
|
||||||
.events = POLLOUT,
|
.events = POLLOUT,
|
||||||
@@ -183,12 +190,13 @@ static size_t fd_write_all (int fd, const void *data, size_t size)
|
|||||||
return done;
|
return done;
|
||||||
}
|
}
|
||||||
|
|
||||||
static int gt_setup_secretkey (struct mud *mud, char *keyfile)
|
static int
|
||||||
|
gt_setup_secretkey(struct mud *mud, char *keyfile)
|
||||||
{
|
{
|
||||||
unsigned char key[32];
|
unsigned char key[32];
|
||||||
|
|
||||||
if (str_empty(keyfile)) {
|
if (str_empty(keyfile)) {
|
||||||
char buf[2*sizeof(key)+1];
|
char buf[2 * sizeof(key) + 1];
|
||||||
size_t size = sizeof(key);
|
size_t size = sizeof(key);
|
||||||
|
|
||||||
if (mud_get_key(mud, key, &size))
|
if (mud_get_key(mud, key, &size))
|
||||||
@@ -203,20 +211,20 @@ static int gt_setup_secretkey (struct mud *mud, char *keyfile)
|
|||||||
int fd;
|
int fd;
|
||||||
|
|
||||||
do {
|
do {
|
||||||
fd = open(keyfile, O_RDONLY|O_CLOEXEC);
|
fd = open(keyfile, O_RDONLY | O_CLOEXEC);
|
||||||
} while (fd==-1 && errno==EINTR);
|
} while (fd == -1 && errno == EINTR);
|
||||||
|
|
||||||
if (fd==-1) {
|
if (fd == -1) {
|
||||||
perror("open keyfile");
|
perror("open keyfile");
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
char buf[2*sizeof(key)];
|
char buf[2 * sizeof(key)];
|
||||||
size_t r = fd_read_all(fd, buf, sizeof(buf));
|
size_t r = fd_read_all(fd, buf, sizeof(buf));
|
||||||
|
|
||||||
close(fd);
|
close(fd);
|
||||||
|
|
||||||
if (r!=sizeof(buf)) {
|
if (r != sizeof(buf)) {
|
||||||
gt_log("unable to read secret key\n");
|
gt_log("unable to read secret key\n");
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
@@ -231,7 +239,8 @@ static int gt_setup_secretkey (struct mud *mud, char *keyfile)
|
|||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
int main (int argc, char **argv)
|
int
|
||||||
|
main(int argc, char **argv)
|
||||||
{
|
{
|
||||||
gt_set_signal();
|
gt_set_signal();
|
||||||
|
|
||||||
@@ -260,6 +269,7 @@ int main (int argc, char **argv)
|
|||||||
#endif
|
#endif
|
||||||
|
|
||||||
struct option opts[] = {
|
struct option opts[] = {
|
||||||
|
// clang-format off
|
||||||
{ "host", &host, option_str },
|
{ "host", &host, option_str },
|
||||||
{ "port", &port, option_long },
|
{ "port", &port, option_long },
|
||||||
{ "bind", &bind_list, option_str },
|
{ "bind", &bind_list, option_str },
|
||||||
@@ -267,6 +277,7 @@ int main (int argc, char **argv)
|
|||||||
{ "bind-port", &bind_port, option_long },
|
{ "bind-port", &bind_port, option_long },
|
||||||
{ "dev", &dev, option_str },
|
{ "dev", &dev, option_str },
|
||||||
{ "mtu", &mtu, option_long },
|
{ "mtu", &mtu, option_long },
|
||||||
|
{ "mtu-auto", NULL, option_option },
|
||||||
{ "keyfile", &keyfile, option_str },
|
{ "keyfile", &keyfile, option_str },
|
||||||
{ "statefile", &statefile, option_str },
|
{ "statefile", &statefile, option_str },
|
||||||
{ "timeout", >.timeout, option_long },
|
{ "timeout", >.timeout, option_long },
|
||||||
@@ -276,13 +287,14 @@ int main (int argc, char **argv)
|
|||||||
{ "chacha20", NULL, option_option },
|
{ "chacha20", NULL, option_option },
|
||||||
{ "version", NULL, option_option },
|
{ "version", NULL, option_option },
|
||||||
{ NULL },
|
{ NULL },
|
||||||
|
// clang-format on
|
||||||
};
|
};
|
||||||
|
|
||||||
if (option(opts, argc, argv))
|
if (option(opts, argc, argv))
|
||||||
return 1;
|
return 1;
|
||||||
|
|
||||||
if (option_is_set(opts, "version")) {
|
if (option_is_set(opts, "version")) {
|
||||||
gt_print(PACKAGE_STRING"\n");
|
gt_print(PACKAGE_STRING "\n");
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -307,35 +319,35 @@ int main (int argc, char **argv)
|
|||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (gt.timeout<=0 || gt.timeout>INT_MAX) {
|
if (gt.timeout <= 0 || gt.timeout > INT_MAX) {
|
||||||
gt_log("bad timeout\n");
|
gt_log("bad timeout\n");
|
||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
int icmp_fd = -1;
|
int icmp_fd = -1;
|
||||||
|
|
||||||
if (v4) {
|
if (v4 && option_is_set(opts, "mtu-auto")) {
|
||||||
icmp_fd = socket(AF_INET, SOCK_RAW, IPPROTO_ICMP);
|
icmp_fd = socket(AF_INET, SOCK_RAW, IPPROTO_ICMP);
|
||||||
|
|
||||||
if (icmp_fd==-1)
|
if (icmp_fd == -1)
|
||||||
gt_log("couldn't create ICMP socket\n");
|
gt_log("couldn't create ICMP socket\n");
|
||||||
}
|
}
|
||||||
|
|
||||||
gt.state_fd = state_create(statefile);
|
gt.state_fd = state_create(statefile);
|
||||||
|
|
||||||
if (statefile && gt.state_fd==-1)
|
if (statefile && gt.state_fd == -1)
|
||||||
return 1;
|
return 1;
|
||||||
|
|
||||||
char *tun_name = NULL;
|
char *tun_name = NULL;
|
||||||
|
|
||||||
int tun_fd = tun_create(dev, &tun_name);
|
int tun_fd = tun_create(dev, &tun_name);
|
||||||
|
|
||||||
if (tun_fd==-1) {
|
if (tun_fd == -1) {
|
||||||
gt_log("couldn't create tun device\n");
|
gt_log("couldn't create tun device\n");
|
||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (tun_set_mtu(tun_name, mtu)==-1) {
|
if (tun_set_mtu(tun_name, mtu) == -1) {
|
||||||
perror("tun_set_mtu");
|
perror("tun_set_mtu");
|
||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
@@ -371,12 +383,12 @@ int main (int argc, char **argv)
|
|||||||
char tmp[1024];
|
char tmp[1024];
|
||||||
char *name = &tmp[0];
|
char *name = &tmp[0];
|
||||||
|
|
||||||
str_cpy(tmp, bind_list, sizeof(tmp)-1);
|
str_cpy(tmp, bind_list, sizeof(tmp) - 1);
|
||||||
|
|
||||||
while (*name) {
|
while (*name) {
|
||||||
char *p = name;
|
char *p = name;
|
||||||
|
|
||||||
while (*p && *p!=',')
|
while (*p && *p != ',')
|
||||||
p++;
|
p++;
|
||||||
|
|
||||||
if (*p)
|
if (*p)
|
||||||
@@ -401,35 +413,36 @@ int main (int argc, char **argv)
|
|||||||
fd_set rfds;
|
fd_set rfds;
|
||||||
FD_ZERO(&rfds);
|
FD_ZERO(&rfds);
|
||||||
|
|
||||||
unsigned char buf[8*1024];
|
unsigned char buf[8 * 1024];
|
||||||
|
|
||||||
while (!gt.quit) {
|
while (!gt.quit) {
|
||||||
FD_SET(tun_fd, &rfds);
|
FD_SET(tun_fd, &rfds);
|
||||||
FD_SET(mud_fd, &rfds);
|
FD_SET(mud_fd, &rfds);
|
||||||
|
|
||||||
if (icmp_fd!=-1)
|
if (icmp_fd != -1)
|
||||||
FD_SET(icmp_fd, &rfds);
|
FD_SET(icmp_fd, &rfds);
|
||||||
|
|
||||||
if _0_(select(mud_fd+1, &rfds, NULL, NULL, NULL)==-1) {
|
if (select(mud_fd + 1, &rfds, NULL, NULL, NULL) == -1) {
|
||||||
if (errno==EINTR)
|
if (errno == EINTR)
|
||||||
continue;
|
continue;
|
||||||
perror("select");
|
perror("select");
|
||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (icmp_fd!=-1 && FD_ISSET(icmp_fd, &rfds)) {
|
if (icmp_fd != -1 && FD_ISSET(icmp_fd, &rfds)) {
|
||||||
struct sockaddr_storage ss;
|
struct sockaddr_storage ss;
|
||||||
socklen_t sl = sizeof(ss);
|
socklen_t sl = sizeof(ss);
|
||||||
ssize_t r = recvfrom(icmp_fd, buf, sizeof(buf), 0, (struct sockaddr *)&ss, &sl);
|
ssize_t r = recvfrom(icmp_fd, buf, sizeof(buf), 0,
|
||||||
if (r>=8) {
|
(struct sockaddr *)&ss, &sl);
|
||||||
|
if (r >= 8) {
|
||||||
struct ip_common ic;
|
struct ip_common ic;
|
||||||
if (!ip_get_common(&ic, buf, r) && ic.proto==1) {
|
if (!ip_get_common(&ic, buf, r) && ic.proto == 1) {
|
||||||
unsigned char *data = &buf[ic.hdr_size];
|
unsigned char *data = &buf[ic.hdr_size];
|
||||||
if (data[0]==3) {
|
if (data[0] == 3) {
|
||||||
int new_mtu = (data[6]<<8)|data[7];
|
int new_mtu = (data[6] << 8) | data[7];
|
||||||
if (new_mtu) {
|
if (new_mtu) {
|
||||||
gt_log("received MTU from ICMP: %i\n", new_mtu);
|
gt_log("received MTU from ICMP: %i\n", new_mtu);
|
||||||
mud_set_mtu(mud, new_mtu-50); // XXX
|
mud_set_mtu(mud, new_mtu - 50); // XXX
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -439,15 +452,15 @@ int main (int argc, char **argv)
|
|||||||
if (FD_ISSET(tun_fd, &rfds)) {
|
if (FD_ISSET(tun_fd, &rfds)) {
|
||||||
size_t size = 0;
|
size_t size = 0;
|
||||||
|
|
||||||
while (sizeof(buf)-size>mtu) {
|
while (sizeof(buf) - size > mtu) {
|
||||||
const ssize_t r = tun_read(tun_fd, &buf[size], sizeof(buf)-size);
|
const ssize_t r = tun_read(tun_fd, &buf[size],
|
||||||
|
sizeof(buf) - size);
|
||||||
if (r<=0)
|
if (r <= 0)
|
||||||
break;
|
break;
|
||||||
|
|
||||||
struct ip_common ic;
|
struct ip_common ic;
|
||||||
|
|
||||||
if (ip_get_common(&ic, &buf[size], r) || ic.size!=r)
|
if (ip_get_common(&ic, &buf[size], r) || ic.size != r)
|
||||||
break;
|
break;
|
||||||
|
|
||||||
size += r;
|
size += r;
|
||||||
@@ -455,42 +468,43 @@ int main (int argc, char **argv)
|
|||||||
|
|
||||||
int p = 0;
|
int p = 0;
|
||||||
|
|
||||||
while (p<size) {
|
while (p < size) {
|
||||||
int tc = 0;
|
int tc = 0;
|
||||||
int q = p;
|
int q = p;
|
||||||
|
|
||||||
while (q<size) {
|
while (q < size) {
|
||||||
struct ip_common ic;
|
struct ip_common ic;
|
||||||
|
|
||||||
if (ip_get_common(&ic, &buf[q], size-q) || ic.size>size-q) {
|
if ((ip_get_common(&ic, &buf[q], size - q)) ||
|
||||||
|
(ic.size > size - q)) {
|
||||||
size = q;
|
size = q;
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (q+ic.size>p+mtu)
|
if (q + ic.size > p + mtu)
|
||||||
break;
|
break;
|
||||||
|
|
||||||
q += ic.size;
|
q += ic.size;
|
||||||
|
|
||||||
if (tc<(ic.tc&0xFC))
|
if (tc < (ic.tc & 0xFC))
|
||||||
tc = ic.tc&0xFC;
|
tc = ic.tc & 0xFC;
|
||||||
}
|
}
|
||||||
|
|
||||||
int r = mud_send(mud, &buf[p], q-p, tc);
|
int r = mud_send(mud, &buf[p], q - p, tc);
|
||||||
|
|
||||||
if (r==-1 && errno==EMSGSIZE) {
|
if (r == -1 && errno == EMSGSIZE) {
|
||||||
int new_mtu = mud_get_mtu(mud);
|
int new_mtu = mud_get_mtu(mud);
|
||||||
|
|
||||||
if (new_mtu!=mtu) {
|
if (new_mtu != mtu) {
|
||||||
mtu = new_mtu;
|
mtu = new_mtu;
|
||||||
|
|
||||||
gt_log("MTU changed: %li\n", mtu);
|
gt_log("MTU changed: %li\n", mtu);
|
||||||
|
|
||||||
if (tun_set_mtu(tun_name, mtu)==-1)
|
if (tun_set_mtu(tun_name, mtu) == -1)
|
||||||
perror("tun_set_mtu");
|
perror("tun_set_mtu");
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
if (r==-1 && errno!=EAGAIN)
|
if (r == -1 && errno != EAGAIN)
|
||||||
perror("mud_send");
|
perror("mud_send");
|
||||||
|
|
||||||
p = q;
|
p = q;
|
||||||
@@ -502,18 +516,19 @@ int main (int argc, char **argv)
|
|||||||
while (1) {
|
while (1) {
|
||||||
const int size = mud_recv(mud, buf, sizeof(buf));
|
const int size = mud_recv(mud, buf, sizeof(buf));
|
||||||
|
|
||||||
if (size<=0) {
|
if (size <= 0) {
|
||||||
if (size==-1 && errno!=EAGAIN)
|
if (size == -1 && errno != EAGAIN)
|
||||||
perror("mud_recv");
|
perror("mud_recv");
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
|
|
||||||
int p = 0;
|
int p = 0;
|
||||||
|
|
||||||
while (p<size) {
|
while (p < size) {
|
||||||
struct ip_common ic;
|
struct ip_common ic;
|
||||||
|
|
||||||
if (ip_get_common(&ic, &buf[p], size-p) || ic.size>size-p)
|
if ((ip_get_common(&ic, &buf[p], size - p)) ||
|
||||||
|
(ic.size > size - p))
|
||||||
break;
|
break;
|
||||||
|
|
||||||
tun_write(tun_fd, &buf[p], ic.size);
|
tun_write(tun_fd, &buf[p], ic.size);
|
||||||
|
|||||||
54
src/option.c
54
src/option.c
@@ -3,9 +3,10 @@
|
|||||||
#include "option.h"
|
#include "option.h"
|
||||||
#include "str.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 || str_empty(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;
|
||||||
}
|
}
|
||||||
@@ -15,9 +16,10 @@ int option_str (void *data, int argc, char **argv)
|
|||||||
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 || str_empty(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;
|
||||||
}
|
}
|
||||||
@@ -26,7 +28,7 @@ int option_long (void *data, int argc, char **argv)
|
|||||||
char *end;
|
char *end;
|
||||||
long val = strtol(argv[1], &end, 0);
|
long val = strtol(argv[1], &end, 0);
|
||||||
|
|
||||||
if (errno || argv[1]==end) {
|
if (errno || argv[1] == end) {
|
||||||
gt_print("argument `%s' is not a valid integer\n", argv[1]);
|
gt_print("argument `%s' is not a valid integer\n", argv[1]);
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
@@ -36,9 +38,10 @@ int option_long (void *data, int argc, char **argv)
|
|||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
int option_is_set (struct option *opts, const char *name)
|
int
|
||||||
|
option_is_set(struct option *opts, const char *name)
|
||||||
{
|
{
|
||||||
for (int k=0; opts[k].name; k++) {
|
for (int k = 0; opts[k].name; k++) {
|
||||||
if (!str_cmp(opts[k].name, name))
|
if (!str_cmp(opts[k].name, name))
|
||||||
return opts[k].set;
|
return opts[k].set;
|
||||||
}
|
}
|
||||||
@@ -46,20 +49,21 @@ int option_is_set (struct option *opts, const char *name)
|
|||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
int option_option (void *data, int argc, char **argv)
|
int
|
||||||
|
option_option(void *data, int argc, char **argv)
|
||||||
{
|
{
|
||||||
if (!data)
|
if (!data)
|
||||||
return 0;
|
return 0;
|
||||||
|
|
||||||
struct option *opts = (struct option *)data;
|
struct option *opts = (struct option *)data;
|
||||||
|
|
||||||
for (int k=0; opts[k].name; k++)
|
for (int k = 0; opts[k].name; k++)
|
||||||
opts[k].set = 0;
|
opts[k].set = 0;
|
||||||
|
|
||||||
for (int i=1; i<argc; i++) {
|
for (int i = 1; i < argc; i++) {
|
||||||
int found = 0;
|
int found = 0;
|
||||||
|
|
||||||
for (int k=0; opts[k].name; k++) {
|
for (int k = 0; opts[k].name; k++) {
|
||||||
if (str_cmp(opts[k].name, argv[i]))
|
if (str_cmp(opts[k].name, argv[i]))
|
||||||
continue;
|
continue;
|
||||||
|
|
||||||
@@ -68,9 +72,9 @@ int option_option (void *data, int argc, char **argv)
|
|||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
int ret = opts[k].call(opts[k].data, argc-i, &argv[i]);
|
int ret = opts[k].call(opts[k].data, argc - i, &argv[i]);
|
||||||
|
|
||||||
if (ret<0)
|
if (ret < 0)
|
||||||
return -1;
|
return -1;
|
||||||
|
|
||||||
opts[k].set = 1;
|
opts[k].set = 1;
|
||||||
@@ -81,29 +85,30 @@ int option_option (void *data, int argc, char **argv)
|
|||||||
}
|
}
|
||||||
|
|
||||||
if (!found)
|
if (!found)
|
||||||
return i-1;
|
return i - 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
return argc;
|
return argc;
|
||||||
}
|
}
|
||||||
|
|
||||||
static int option_usage (struct option *opts, int slen)
|
static int
|
||||||
|
option_usage(struct option *opts, int slen)
|
||||||
{
|
{
|
||||||
if (!opts)
|
if (!opts)
|
||||||
return 0;
|
return 0;
|
||||||
|
|
||||||
int len = 0;
|
int len = 0;
|
||||||
|
|
||||||
for (int k=0; opts[k].name; k++) {
|
for (int k = 0; opts[k].name; k++) {
|
||||||
if (len>40) {
|
if (len > 40) {
|
||||||
gt_print("\n%*s", slen, "");
|
gt_print("\n%*s", slen, "");
|
||||||
len = 0;
|
len = 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
len += gt_print(" [%s", opts[k].name);
|
len += gt_print(" [%s", opts[k].name);
|
||||||
|
|
||||||
if (opts[k].call==option_option) {
|
if (opts[k].call == option_option) {
|
||||||
len += option_usage((struct option *)opts[k].data, slen+len);
|
len += option_usage((struct option *)opts[k].data, slen + len);
|
||||||
} else {
|
} else {
|
||||||
len += gt_print(" ARG");
|
len += gt_print(" ARG");
|
||||||
}
|
}
|
||||||
@@ -114,21 +119,22 @@ static int option_usage (struct option *opts, int slen)
|
|||||||
return len;
|
return len;
|
||||||
}
|
}
|
||||||
|
|
||||||
int option (struct option *opts, int argc, char **argv)
|
int
|
||||||
|
option(struct option *opts, int argc, char **argv)
|
||||||
{
|
{
|
||||||
int ret = option_option(opts, argc, argv);
|
int ret = option_option(opts, argc, argv);
|
||||||
|
|
||||||
if (ret==argc)
|
if (ret == argc)
|
||||||
return 0;
|
return 0;
|
||||||
|
|
||||||
if (ret<0 || ret+1>=argc)
|
if (ret < 0 || ret + 1 >= argc)
|
||||||
return 1;
|
return 1;
|
||||||
|
|
||||||
gt_print("option `%s' is unknown\n", argv[ret+1]);
|
gt_print("option `%s' is unknown\n", argv[ret + 1]);
|
||||||
|
|
||||||
int slen = gt_print("usage: %s", argv[0]);
|
int slen = gt_print("usage: %s", argv[0]);
|
||||||
|
|
||||||
if (slen>40) {
|
if (slen > 40) {
|
||||||
slen = 12;
|
slen = 12;
|
||||||
gt_print("\n%*s", slen, "");
|
gt_print("\n%*s", slen, "");
|
||||||
}
|
}
|
||||||
|
|||||||
20
src/state.c
20
src/state.c
@@ -3,26 +3,27 @@
|
|||||||
#include "state.h"
|
#include "state.h"
|
||||||
#include "str.h"
|
#include "str.h"
|
||||||
|
|
||||||
#include <stdio.h>
|
|
||||||
#include <fcntl.h>
|
#include <fcntl.h>
|
||||||
|
#include <stdio.h>
|
||||||
#include <sys/stat.h>
|
#include <sys/stat.h>
|
||||||
|
|
||||||
int state_create (const char *filename)
|
int
|
||||||
|
state_create(const char *filename)
|
||||||
{
|
{
|
||||||
if (str_empty(filename))
|
if (str_empty(filename))
|
||||||
return -1;
|
return -1;
|
||||||
|
|
||||||
int fd = open(filename, O_WRONLY);
|
int fd = open(filename, O_WRONLY);
|
||||||
|
|
||||||
if (fd==-1) {
|
if (fd == -1) {
|
||||||
if (errno!=EINTR)
|
if (errno != EINTR)
|
||||||
perror("open");
|
perror("open");
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
struct stat st = {0};
|
struct stat st = {0};
|
||||||
|
|
||||||
if (fstat(fd, &st)==-1) {
|
if (fstat(fd, &st) == -1) {
|
||||||
perror("fstat");
|
perror("fstat");
|
||||||
close(fd);
|
close(fd);
|
||||||
return -1;
|
return -1;
|
||||||
@@ -37,17 +38,18 @@ int state_create (const char *filename)
|
|||||||
return fd;
|
return fd;
|
||||||
}
|
}
|
||||||
|
|
||||||
void state_send (int fd, const char *state, const char *info)
|
void
|
||||||
|
state_send(int fd, const char *state, const char *info)
|
||||||
{
|
{
|
||||||
if (str_empty(state))
|
if (str_empty(state))
|
||||||
return;
|
return;
|
||||||
|
|
||||||
if (fd==-1) {
|
if (fd == -1) {
|
||||||
gt_print("%s %s\n", state, info);
|
gt_print("%s %s\n", state, info);
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
const char *strs[] = { state, " ", info, "\n" };
|
const char *strs[] = {state, " ", info, "\n"};
|
||||||
char *str = str_cat(strs, COUNT(strs));
|
char *str = str_cat(strs, COUNT(strs));
|
||||||
|
|
||||||
if (!str) {
|
if (!str) {
|
||||||
@@ -55,7 +57,7 @@ void state_send (int fd, const char *state, const char *info)
|
|||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (write(fd, str, str_len(str))==-1 && errno!=EINTR)
|
if (write(fd, str, str_len(str)) == -1 && errno != EINTR)
|
||||||
perror("write");
|
perror("write");
|
||||||
|
|
||||||
free(str);
|
free(str);
|
||||||
|
|||||||
99
src/tun.c
99
src/tun.c
@@ -1,11 +1,11 @@
|
|||||||
#include "common.h"
|
#include "common.h"
|
||||||
|
|
||||||
#include "tun.h"
|
|
||||||
#include "str.h"
|
|
||||||
#include "ip.h"
|
#include "ip.h"
|
||||||
|
#include "str.h"
|
||||||
|
#include "tun.h"
|
||||||
|
|
||||||
#include <stdio.h>
|
|
||||||
#include <fcntl.h>
|
#include <fcntl.h>
|
||||||
|
#include <stdio.h>
|
||||||
|
|
||||||
#include <sys/ioctl.h>
|
#include <sys/ioctl.h>
|
||||||
#include <sys/socket.h>
|
#include <sys/socket.h>
|
||||||
@@ -18,9 +18,9 @@
|
|||||||
#endif
|
#endif
|
||||||
|
|
||||||
#ifdef __APPLE__
|
#ifdef __APPLE__
|
||||||
#include <sys/sys_domain.h>
|
|
||||||
#include <sys/kern_control.h>
|
|
||||||
#include <net/if_utun.h>
|
#include <net/if_utun.h>
|
||||||
|
#include <sys/kern_control.h>
|
||||||
|
#include <sys/sys_domain.h>
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#if defined(__APPLE__) || defined(__OpenBSD__)
|
#if defined(__APPLE__) || defined(__OpenBSD__)
|
||||||
@@ -29,17 +29,18 @@
|
|||||||
|
|
||||||
#ifdef __APPLE__
|
#ifdef __APPLE__
|
||||||
|
|
||||||
static int tun_create_by_id (char *name, size_t size, unsigned id)
|
static int
|
||||||
|
tun_create_by_id(char *name, size_t size, unsigned id)
|
||||||
{
|
{
|
||||||
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;
|
||||||
|
|
||||||
struct ctl_info ci;
|
struct ctl_info ci;
|
||||||
|
|
||||||
memset(&ci, 0, sizeof(ci));
|
memset(&ci, 0, sizeof(ci));
|
||||||
str_cpy(ci.ctl_name, UTUN_CONTROL_NAME, sizeof(ci.ctl_name)-1);
|
str_cpy(ci.ctl_name, UTUN_CONTROL_NAME, sizeof(ci.ctl_name) - 1);
|
||||||
|
|
||||||
if (ioctl(fd, CTLIOCGINFO, &ci)) {
|
if (ioctl(fd, CTLIOCGINFO, &ci)) {
|
||||||
close(fd);
|
close(fd);
|
||||||
@@ -51,7 +52,7 @@ static int tun_create_by_id (char *name, size_t size, unsigned id)
|
|||||||
.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 = id+1,
|
.sc_unit = id + 1,
|
||||||
};
|
};
|
||||||
|
|
||||||
if (connect(fd, (struct sockaddr *)&sc, sizeof(sc))) {
|
if (connect(fd, (struct sockaddr *)&sc, sizeof(sc))) {
|
||||||
@@ -64,11 +65,12 @@ static int tun_create_by_id (char *name, size_t size, unsigned id)
|
|||||||
return fd;
|
return fd;
|
||||||
}
|
}
|
||||||
|
|
||||||
static int tun_create_by_name (char *name, size_t size, char *dev_name)
|
static int
|
||||||
|
tun_create_by_name(char *name, size_t size, char *dev_name)
|
||||||
{
|
{
|
||||||
unsigned id = 0;
|
unsigned id = 0;
|
||||||
|
|
||||||
if (sscanf(dev_name, "utun%u", &id)!=1)
|
if (sscanf(dev_name, "utun%u", &id) != 1)
|
||||||
return -1;
|
return -1;
|
||||||
|
|
||||||
return tun_create_by_id(name, size, id);
|
return tun_create_by_id(name, size, id);
|
||||||
@@ -78,44 +80,47 @@ static int tun_create_by_name (char *name, size_t size, char *dev_name)
|
|||||||
|
|
||||||
#ifdef __linux__
|
#ifdef __linux__
|
||||||
|
|
||||||
static int tun_create_by_name (char *name, size_t size, char *dev_name)
|
static int
|
||||||
|
tun_create_by_name(char *name, size_t size, char *dev_name)
|
||||||
{
|
{
|
||||||
int fd = open("/dev/net/tun", O_RDWR);
|
int fd = open("/dev/net/tun", O_RDWR);
|
||||||
|
|
||||||
if (fd==-1)
|
if (fd == -1)
|
||||||
return -1;
|
return -1;
|
||||||
|
|
||||||
struct ifreq ifr = {
|
struct ifreq ifr = {
|
||||||
.ifr_flags = IFF_TUN|IFF_NO_PI,
|
.ifr_flags = IFF_TUN | IFF_NO_PI,
|
||||||
};
|
};
|
||||||
|
|
||||||
str_cpy(ifr.ifr_name, dev_name, IFNAMSIZ-1);
|
str_cpy(ifr.ifr_name, dev_name, IFNAMSIZ - 1);
|
||||||
|
|
||||||
if (ioctl(fd, TUNSETIFF, &ifr)) {
|
if (ioctl(fd, TUNSETIFF, &ifr)) {
|
||||||
close(fd);
|
close(fd);
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
str_cpy(name, ifr.ifr_name, size-1);
|
str_cpy(name, ifr.ifr_name, size - 1);
|
||||||
|
|
||||||
return fd;
|
return fd;
|
||||||
}
|
}
|
||||||
|
|
||||||
#else /* not __linux__ not __APPLE__ */
|
#else /* not __linux__ not __APPLE__ */
|
||||||
|
|
||||||
static int tun_create_by_name (char *name, size_t size, char *dev_name)
|
static int
|
||||||
|
tun_create_by_name(char *name, size_t size, char *dev_name)
|
||||||
{
|
{
|
||||||
char path[64];
|
char path[64];
|
||||||
|
|
||||||
snprintf(path, sizeof(path), "/dev/%s", dev_name);
|
snprintf(path, sizeof(path), "/dev/%s", dev_name);
|
||||||
str_cpy(name, dev_name, size-1);
|
str_cpy(name, dev_name, size - 1);
|
||||||
|
|
||||||
return open(path, O_RDWR);
|
return open(path, O_RDWR);
|
||||||
}
|
}
|
||||||
|
|
||||||
#endif /* not __APPLE__ */
|
#endif /* not __APPLE__ */
|
||||||
|
|
||||||
static int tun_create_by_id (char *name, size_t size, unsigned id)
|
static int
|
||||||
|
tun_create_by_id(char *name, size_t size, unsigned id)
|
||||||
{
|
{
|
||||||
char dev_name[64];
|
char dev_name[64];
|
||||||
|
|
||||||
@@ -126,25 +131,27 @@ static int tun_create_by_id (char *name, size_t size, unsigned id)
|
|||||||
|
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
int tun_create (char *dev_name, char **ret_name)
|
int
|
||||||
|
tun_create(char *dev_name, char **ret_name)
|
||||||
{
|
{
|
||||||
char name[64] = {0};
|
char name[64] = {0};
|
||||||
int fd = -1;
|
int fd = -1;
|
||||||
|
|
||||||
if (str_empty(dev_name)) {
|
if (str_empty(dev_name)) {
|
||||||
for (unsigned id=0; id<32 && fd==-1; id++)
|
for (unsigned id = 0; id < 32 && fd == -1; id++)
|
||||||
fd = tun_create_by_id(name, sizeof(name), id);
|
fd = tun_create_by_id(name, sizeof(name), id);
|
||||||
} else {
|
} else {
|
||||||
fd = tun_create_by_name(name, sizeof(name), dev_name);
|
fd = tun_create_by_name(name, sizeof(name), dev_name);
|
||||||
}
|
}
|
||||||
|
|
||||||
if (fd!=-1 && ret_name)
|
if (fd != -1 && ret_name)
|
||||||
*ret_name = strdup(name);
|
*ret_name = strdup(name);
|
||||||
|
|
||||||
return fd;
|
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)
|
||||||
return -1;
|
return -1;
|
||||||
@@ -153,8 +160,14 @@ ssize_t tun_read (int fd, void *data, size_t size)
|
|||||||
uint32_t family;
|
uint32_t family;
|
||||||
|
|
||||||
struct iovec iov[2] = {
|
struct iovec iov[2] = {
|
||||||
{ .iov_base = &family, .iov_len = sizeof(family) },
|
{
|
||||||
{ .iov_base = data, .iov_len = size }
|
.iov_base = &family,
|
||||||
|
.iov_len = sizeof(family),
|
||||||
|
},
|
||||||
|
{
|
||||||
|
.iov_base = data,
|
||||||
|
.iov_len = size,
|
||||||
|
},
|
||||||
};
|
};
|
||||||
|
|
||||||
ssize_t ret = readv(fd, iov, 2);
|
ssize_t ret = readv(fd, iov, 2);
|
||||||
@@ -162,8 +175,8 @@ ssize_t tun_read (int fd, void *data, size_t size)
|
|||||||
ssize_t ret = read(fd, data, size);
|
ssize_t ret = read(fd, data, size);
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
if (ret==-1) {
|
if (ret == -1) {
|
||||||
if (errno==EAGAIN || errno==EINTR)
|
if (errno == EAGAIN || errno == EINTR)
|
||||||
return -1;
|
return -1;
|
||||||
|
|
||||||
if (errno)
|
if (errno)
|
||||||
@@ -173,16 +186,17 @@ ssize_t tun_read (int fd, void *data, size_t size)
|
|||||||
}
|
}
|
||||||
|
|
||||||
#ifdef GT_BSD_TUN
|
#ifdef GT_BSD_TUN
|
||||||
if (ret<(ssize_t) sizeof(family))
|
if (ret < (ssize_t)sizeof(family))
|
||||||
return 0;
|
return 0;
|
||||||
|
|
||||||
return ret-sizeof(family);
|
return ret - sizeof(family);
|
||||||
#else
|
#else
|
||||||
return ret;
|
return ret;
|
||||||
#endif
|
#endif
|
||||||
}
|
}
|
||||||
|
|
||||||
ssize_t tun_write (int fd, const void *data, size_t size)
|
ssize_t
|
||||||
|
tun_write(int fd, const void *data, size_t size)
|
||||||
{
|
{
|
||||||
if (!size)
|
if (!size)
|
||||||
return -1;
|
return -1;
|
||||||
@@ -202,8 +216,14 @@ ssize_t tun_write (int fd, const void *data, size_t size)
|
|||||||
}
|
}
|
||||||
|
|
||||||
struct iovec iov[2] = {
|
struct iovec iov[2] = {
|
||||||
{ .iov_base = &family, .iov_len = sizeof(family) },
|
{
|
||||||
{ .iov_base = (void *) data, .iov_len = size },
|
.iov_base = &family,
|
||||||
|
.iov_len = sizeof(family),
|
||||||
|
},
|
||||||
|
{
|
||||||
|
.iov_base = (void *)data,
|
||||||
|
.iov_len = size,
|
||||||
|
},
|
||||||
};
|
};
|
||||||
|
|
||||||
ssize_t ret = writev(fd, iov, 2);
|
ssize_t ret = writev(fd, iov, 2);
|
||||||
@@ -211,8 +231,8 @@ ssize_t tun_write (int fd, const void *data, size_t size)
|
|||||||
ssize_t ret = write(fd, data, size);
|
ssize_t ret = write(fd, data, size);
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
if (ret==-1) {
|
if (ret == -1) {
|
||||||
if (errno==EAGAIN || errno==EINTR)
|
if (errno == EAGAIN || errno == EINTR)
|
||||||
return -1;
|
return -1;
|
||||||
|
|
||||||
if (errno)
|
if (errno)
|
||||||
@@ -222,26 +242,27 @@ ssize_t tun_write (int fd, const void *data, size_t size)
|
|||||||
}
|
}
|
||||||
|
|
||||||
#ifdef GT_BSD_TUN
|
#ifdef GT_BSD_TUN
|
||||||
if (ret<(ssize_t) sizeof(family))
|
if (ret < (ssize_t)sizeof(family))
|
||||||
return 0;
|
return 0;
|
||||||
|
|
||||||
return ret-sizeof(family);
|
return ret - sizeof(family);
|
||||||
#else
|
#else
|
||||||
return ret;
|
return ret;
|
||||||
#endif
|
#endif
|
||||||
}
|
}
|
||||||
|
|
||||||
int tun_set_mtu (char *dev_name, int mtu)
|
int
|
||||||
|
tun_set_mtu(char *dev_name, int mtu)
|
||||||
{
|
{
|
||||||
struct ifreq ifr = {
|
struct ifreq ifr = {
|
||||||
.ifr_mtu = mtu,
|
.ifr_mtu = mtu,
|
||||||
};
|
};
|
||||||
|
|
||||||
str_cpy(ifr.ifr_name, dev_name, IFNAMSIZ-1);
|
str_cpy(ifr.ifr_name, dev_name, IFNAMSIZ - 1);
|
||||||
|
|
||||||
int fd = socket(AF_INET, SOCK_DGRAM, 0);
|
int fd = socket(AF_INET, SOCK_DGRAM, 0);
|
||||||
|
|
||||||
if (fd==-1)
|
if (fd == -1)
|
||||||
return -1;
|
return -1;
|
||||||
|
|
||||||
int ret = ioctl(fd, SIOCSIFMTU, &ifr);
|
int ret = ioctl(fd, SIOCSIFMTU, &ifr);
|
||||||
|
|||||||
Reference in New Issue
Block a user