Update mud

Required on BSD. Doesn't break anything elsewhere.

src/common.c

@@ -1,22 +1,24 @@
 #include "common.h"
 
-#include <stdio.h>
 #include <stdarg.h>
+#include <stdio.h>
 
-int gt_print (const char *fmt, ...)
+int
+gt_print(const char *fmt, ...)
 {
     va_list ap;
     va_start(ap, fmt);
     int ret = vfprintf(stdout, fmt, ap);
     va_end(ap);
 
-    if (ret<0)
+    if (ret < 0)
         return 0;
 
     return ret;
 }
 
-void gt_log (const char *fmt, ...)
+void
+gt_log(const char *fmt, ...)
 {
     va_list ap;
     va_start(ap, fmt);
@@ -24,7 +26,8 @@ void gt_log (const char *fmt, ...)
     va_end(ap);
 }
 
-void gt_fatal (const char *fmt, ...)
+void
+gt_fatal(const char *fmt, ...)
 {
     va_list ap;
     va_start(ap, fmt);
@@ -34,24 +37,26 @@ void gt_fatal (const char *fmt, ...)
     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);
 }
 
-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;
 
-    if _0_(((dst_size-1)/2)<src_size)
+    if (_0_(((dst_size - 1) / 2) < src_size))
         return -1;
 
     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;
@@ -59,37 +64,38 @@ int gt_tohex (char *dst, size_t dst_size, const uint8_t *src, size_t src_size)
     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;
 }
 
-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;
 
-    if _0_(dst_size<(src_size/2))
+    if (_0_(dst_size < (src_size / 2)))
         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 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;
 
-        *dst++ = (a<<4)|b;
+        *dst++ = (a << 4) | b;
     }
 
     return 0;

src/db.c

@@ -3,56 +3,57 @@
 #include "db.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 {
     uint8_t *child[2];
     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];
 
-    if (size!=b[0])
+    if (size != b[0])
         return 1;
 
     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;
 }
 
-_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 ((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;
 
     uint8_t *r = *p;
@@ -69,12 +70,13 @@ uint8_t *db_search (uint8_t **p, uint8_t *data)
     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;
 
-    if _0_(!*p) {
+    if (_0_(!*p)) {
         *p = 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);
 
-    if _0_(!diff)
+    if (_0_(!diff))
         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)) {
         struct node *node = CBIT_NODE(*p);
 
-        if (node->point>point)
+        if (node->point > point)
             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));
 
-    if _0_(!node)
+    if (_0_(!node))
         return NULL;
 
-    const int dir = (mask|r[pos])==255;
+    const int dir = (mask | r[pos]) == 255;
 
     node->child[dir] = *p;
-    node->child[1-dir] = data;
+    node->child[1 - dir] = data;
     node->point = point;
 
     *p = CBIT_PTR(node);
@@ -121,9 +123,10 @@ uint8_t *db_insert (uint8_t **p, uint8_t *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;
 
     const size_t size = db_size(data);
@@ -136,16 +139,16 @@ uint8_t *db_remove (uint8_t **p, uint8_t *data)
         p_old = p;
         node = CBIT_NODE(*p);
         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;
 
     uint8_t *r = *p;
 
     if (p_old) {
-        *p_old = node->child[1-dir];
+        *p_old = node->child[1 - dir];
         free(node);
     } else {
         *p = NULL;

src/main.c

@@ -1,19 +1,19 @@
 #include "common.h"
 
 #include "buffer.h"
-#include "ip.h"
-#include "str.h"
-#include "option.h"
-#include "tun.h"
 #include "db.h"
+#include "ip.h"
+#include "option.h"
 #include "state.h"
+#include "str.h"
+#include "tun.h"
 
+#include <fcntl.h>
 #include <inttypes.h>
 #include <limits.h>
-#include <stdio.h>
-#include <signal.h>
 #include <poll.h>
-#include <fcntl.h>
+#include <signal.h>
+#include <stdio.h>
 #include <sys/socket.h>
 #include <sys/time.h>
 
@@ -33,25 +33,27 @@ static struct {
     int state_fd;
 } gt;
 
-static void fd_set_nonblock (int fd)
+static void
+fd_set_nonblock(int fd)
 {
     int ret;
 
     do {
         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 {
-        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");
 }
 
-static void gt_sa_handler (int sig)
+static void
+gt_sa_handler(int sig)
 {
     switch (sig) {
     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 = {
         .sa_flags = 0,
@@ -84,15 +87,16 @@ static void gt_set_signal (void)
     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;
 
     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;
 
         if (errno)
@@ -104,18 +108,19 @@ static ssize_t fd_read (int fd, void *data, size_t size)
     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;
 
     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;
 
-        if (errno==EPIPE || errno==ECONNRESET)
+        if (errno == EPIPE || errno == ECONNRESET)
             return 0;
 
         if (errno)
@@ -127,17 +132,18 @@ static ssize_t fd_write (int fd, const void *data, size_t size)
     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;
 
-    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)
             break;
 
-        if (ret<0) {
+        if (ret < 0) {
             struct pollfd pollfd = {
                 .fd = fd,
                 .events = POLLIN,
@@ -155,17 +161,18 @@ static size_t fd_read_all (int fd, void *data, size_t size)
     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;
 
-    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)
             break;
 
-        if (ret<0) {
+        if (ret < 0) {
             struct pollfd pollfd = {
                 .fd = fd,
                 .events = POLLOUT,
@@ -183,12 +190,13 @@ static size_t fd_write_all (int fd, const void *data, size_t size)
     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];
 
     if (str_empty(keyfile)) {
-        char buf[2*sizeof(key)+1];
+        char buf[2 * sizeof(key) + 1];
         size_t size = sizeof(key);
 
         if (mud_get_key(mud, key, &size))
@@ -203,20 +211,20 @@ static int gt_setup_secretkey (struct mud *mud, char *keyfile)
     int fd;
 
     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");
         return -1;
     }
 
-    char buf[2*sizeof(key)];
+    char buf[2 * sizeof(key)];
     size_t r = fd_read_all(fd, buf, sizeof(buf));
 
     close(fd);
 
-    if (r!=sizeof(buf)) {
+    if (r != sizeof(buf)) {
         gt_log("unable to read secret key\n");
         return -1;
     }
@@ -231,7 +239,8 @@ static int gt_setup_secretkey (struct mud *mud, char *keyfile)
     return 0;
 }
 
-int main (int argc, char **argv)
+int
+main(int argc, char **argv)
 {
     gt_set_signal();
 
@@ -239,6 +248,7 @@ int main (int argc, char **argv)
     long port = 5000;
 
     char *bind_list = NULL;
+    char *bind_backup = NULL;
     long bind_port = 5000;
 
     char *dev = NULL;
@@ -259,14 +269,15 @@ int main (int argc, char **argv)
 #endif
 
     struct option opts[] = {
+        // clang-format off
         { "host",           &host,           option_str    },
         { "port",           &port,           option_long   },
         { "bind",           &bind_list,      option_str    },
+        { "bind-backup",    &bind_backup,    option_str    },
         { "bind-port",      &bind_port,      option_long   },
         { "dev",            &dev,            option_str    },
         { "mtu",            &mtu,            option_long   },
         { "keyfile",        &keyfile,        option_str    },
-        { "multiqueue",     NULL,            option_option },
         { "statefile",      &statefile,      option_str    },
         { "timeout",        &gt.timeout,     option_long   },
         { "time-tolerance", &time_tolerance, option_long   },
@@ -275,13 +286,14 @@ int main (int argc, char **argv)
         { "chacha20",       NULL,            option_option },
         { "version",        NULL,            option_option },
         {  NULL                                            },
+        // clang-format on
     };
 
     if (option(opts, argc, argv))
         return 1;
 
     if (option_is_set(opts, "version")) {
-        gt_print(PACKAGE_STRING"\n");
+        gt_print(PACKAGE_STRING "\n");
         return 0;
     }
 
@@ -306,30 +318,44 @@ int main (int argc, char **argv)
         return 1;
     }
 
-    if (gt.timeout<=0 || gt.timeout>INT_MAX) {
+    if (gt.timeout <= 0 || gt.timeout > INT_MAX) {
         gt_log("bad timeout\n");
         return 1;
     }
 
+    int icmp_fd = -1;
+
+    if (v4) {
+        icmp_fd = socket(AF_INET, SOCK_RAW, IPPROTO_ICMP);
+
+        if (icmp_fd == -1)
+            gt_log("couldn't create ICMP socket\n");
+    }
+
     gt.state_fd = state_create(statefile);
 
-    if (statefile && gt.state_fd==-1)
+    if (statefile && gt.state_fd == -1)
         return 1;
 
     char *tun_name = NULL;
 
-    int tun_fd = tun_create(dev, &tun_name, option_is_set(opts, "multiqueue"));
+    int tun_fd = tun_create(dev, &tun_name);
 
-    if (tun_fd==-1) {
+    if (tun_fd == -1) {
         gt_log("couldn't create tun device\n");
         return 1;
     }
 
+    if (tun_set_mtu(tun_name, mtu) == -1) {
+        perror("tun_set_mtu");
+        return 1;
+    }
+
     fd_set_nonblock(tun_fd);
 
     int chacha = option_is_set(opts, "chacha20");
 
-    struct mud *mud = mud_create(bind_port, v4, v6, !chacha);
+    struct mud *mud = mud_create(bind_port, v4, v6, !chacha, mtu);
 
     if (!mud) {
         gt_log("couldn't create mud\n");
@@ -344,165 +370,173 @@ int main (int argc, char **argv)
     if (time_tolerance > 0)
         mud_set_time_tolerance_sec(mud, time_tolerance);
 
-    if (host && port && bind_list) {
+    if (host && port) {
+        if (bind_backup) {
+            if (mud_peer(mud, bind_backup, host, port, 1)) {
+                perror("mud_peer (backup)");
+                return 1;
+            }
+        }
+
+        if (bind_list) {
             char tmp[1024];
             char *name = &tmp[0];
 
-        size_t size = str_cpy(tmp, bind_list, sizeof(tmp)-1);
+            str_cpy(tmp, bind_list, sizeof(tmp) - 1);
 
-        for (size_t i=0; i<size; i++) {
+            while (*name) {
-            if (tmp[i]!=',')
+                char *p = name;
-                continue;
 
-            tmp[i] = 0;
+                while (*p && *p != ',')
+                    p++;
 
-            if (mud_peer(mud, name, host, port))
+                if (*p)
+                    *p++ = 0;
+
+                if (mud_peer(mud, name, host, port, 0)) {
+                    perror("mud_peer");
                     return 1;
-
-            name = &tmp[i+1];
                 }
 
-        if (name[0] && mud_peer(mud, name, host, port))
+                name = p;
-            return 1;
+            }
+        }
     }
 
     int mud_fd = mud_get_fd(mud);
 
+    fd_set_nonblock(mud_fd);
+
     state_send(gt.state_fd, "INITIALIZED", tun_name);
 
     fd_set rfds;
     FD_ZERO(&rfds);
 
-    int started = 0;
+    unsigned char buf[8 * 1024];
-
-    struct {
-        unsigned char *buf;
-    } send, recv;
-
-    send.buf = malloc(2*mtu);
-    recv.buf = malloc(mtu);
-
-    size_t send_size = 0;
-    size_t send_limit = 0;
-    int send_tc = 0;
-    int send_next_tc = 0;
 
     while (!gt.quit) {
         FD_SET(tun_fd, &rfds);
-
-        if (mud_can_pull(mud)) {
         FD_SET(mud_fd, &rfds);
-        } else {
-            FD_CLR(mud_fd, &rfds);
-        }
 
-        struct timeval timeout = {
+        if (icmp_fd != -1)
-            .tv_usec = 100000,
+            FD_SET(icmp_fd, &rfds);
-        };
 
-        if (mud_can_push(mud) || send_size)
+        if (select(mud_fd + 1, &rfds, NULL, NULL, NULL) == -1) {
-            timeout.tv_usec = 1000;
+            if (errno == EINTR)
-
-        if _0_(select(mud_fd+1, &rfds, NULL, NULL, &timeout)==-1) {
-            if (errno==EINTR)
                 continue;
             perror("select");
             return 1;
         }
 
-        if (mud_is_up(mud)) {
+        if (icmp_fd != -1 && FD_ISSET(icmp_fd, &rfds)) {
-            if (!started) {
+            struct sockaddr_storage ss;
-                state_send(gt.state_fd, "STARTED", tun_name);
+            socklen_t sl = sizeof(ss);
-                started = 1;
+            ssize_t r = recvfrom(icmp_fd, buf, sizeof(buf), 0,
+                                 (struct sockaddr *)&ss, &sl);
+            if (r >= 8) {
+                struct ip_common ic;
+                if (!ip_get_common(&ic, buf, r) && ic.proto == 1) {
+                    unsigned char *data = &buf[ic.hdr_size];
+                    if (data[0] == 3) {
+                        int new_mtu = (data[6] << 8) | data[7];
+                        if (new_mtu) {
+                            gt_log("received MTU from ICMP: %i\n", new_mtu);
+                            mud_set_mtu(mud, new_mtu - 50); // XXX
+                        }
+                    }
                 }
-        } else {
-            if (started) {
-                state_send(gt.state_fd, "STOPPED", tun_name);
-                started = 0;
             }
         }
 
         if (FD_ISSET(tun_fd, &rfds)) {
-            while (send_size<mtu) {
+            size_t size = 0;
-                const ssize_t r = tun_read(tun_fd, send.buf+send_size, mtu);
 
-                if (r<=0) {
+            while (sizeof(buf) - size > mtu) {
-                    gt.quit |= !r;
+                const ssize_t r = tun_read(tun_fd, &buf[size],
+                                           sizeof(buf) - size);
+                if (r <= 0)
                     break;
-                }
 
                 struct ip_common ic;
 
-                if (ip_get_common(&ic, send.buf+send_size, mtu) || ic.size!=r) {
+                if (ip_get_common(&ic, &buf[size], r) || ic.size != r)
-                    gt_log("packet dropped: malformed\n");
+                    break;
-                    continue;
+
+                size += r;
             }
 
-                send_size += r;
+            int p = 0;
 
-                int update_tc = (ic.tc&0xFC)>(send_tc&0xFC);
+            while (p < size) {
+                int tc = 0;
+                int q = p;
 
-                if (send_size<=mtu) {
+                while (q < size) {
-                    send_limit = send_size;
+                    struct ip_common ic;
-                    if ((ic.tc&0xFC)>(send_tc&0xFC))
+
-                        send_tc = ic.tc;
+                    if ((ip_get_common(&ic, &buf[q], size - q)) ||
+                        (ic.size > size - q)) {
+                        size = q;
+                        break;
+                    }
+
+                    if (q + ic.size > p + mtu)
+                        break;
+
+                    q += ic.size;
+
+                    if (tc < (ic.tc & 0xFC))
+                        tc = ic.tc & 0xFC;
+                }
+
+                int r = mud_send(mud, &buf[p], q - p, tc);
+
+                if (r == -1 && errno == EMSGSIZE) {
+                    int new_mtu = mud_get_mtu(mud);
+
+                    if (new_mtu != mtu) {
+                        mtu = new_mtu;
+
+                        gt_log("MTU changed: %li\n", mtu);
+
+                        if (tun_set_mtu(tun_name, mtu) == -1)
+                            perror("tun_set_mtu");
+                    }
                 } else {
-                    if ((ic.tc&0xFC)>(send_next_tc&0xFC))
+                    if (r == -1 && errno != EAGAIN)
-                        send_next_tc = ic.tc;
-                }
-            }
-        }
-
-        if (send_limit) {
-            int r = mud_send(mud, send.buf, send_limit, send_tc);
-            if (r==-1 && errno!=EAGAIN) {
                         perror("mud_send");
-            } else if (r==send_limit) {
-                if (send_size>send_limit)
-                    memmove(&send.buf[0], &send.buf[send_limit], send_size-send_limit);
-                send_size -= send_limit;
-                send_limit = send_size;
-                send_tc = send_next_tc;
-                send_next_tc = 0;
-            }
-        }
 
-        if (mud_push(mud)==-1 && errno!=EAGAIN)
+                    p = q;
-            perror("mud_push");
+                }
+            }
+        }
 
         if (FD_ISSET(mud_fd, &rfds)) {
-            if (mud_pull(mud)==-1 && errno!=EAGAIN)
+            while (1) {
-                perror("mud_pull");
+                const int size = mud_recv(mud, buf, sizeof(buf));
-        }
 
-        while (!gt.quit) {
+                if (size <= 0) {
-            const int size = mud_recv(mud, recv.buf, mtu);
+                    if (size == -1 && errno != EAGAIN)
-
-            if (size<=0) {
-                if (size==-1 && errno!=EAGAIN)
                         perror("mud_recv");
                     break;
                 }
 
                 int p = 0;
 
-            while (p<size) {
+                while (p < size) {
                     struct ip_common ic;
 
-                if (ip_get_common(&ic, recv.buf+p, size-p) || ic.size>size-p)
+                    if ((ip_get_common(&ic, &buf[p], size - p)) ||
+                        (ic.size > size - p))
                         break;
 
-                const ssize_t r = tun_write(tun_fd, recv.buf+p, ic.size);
+                    tun_write(tun_fd, &buf[p], ic.size);
-
-                if (r<=0) {
-                    gt.quit |= !r;
-                    break;
-                }
 
                     p += ic.size;
                 }
             }
         }
+    }
 
     return 0;
 }

src/option.c

@@ -3,9 +3,10 @@
 #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 || str_empty(argv[1])) {
+    if (argc < 2 || str_empty(argv[1])) {
         gt_print("option `%s' need a string argument\n", argv[0]);
         return -1;
     }
@@ -15,9 +16,10 @@ int option_str (void *data, int argc, char **argv)
     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]);
         return -1;
     }
@@ -26,7 +28,7 @@ int option_long (void *data, int argc, char **argv)
     char *end;
     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]);
         return -1;
     }
@@ -36,9 +38,10 @@ int option_long (void *data, int argc, char **argv)
     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))
             return opts[k].set;
     }
@@ -46,20 +49,21 @@ int option_is_set (struct option *opts, const char *name)
     return 0;
 }
 
-int option_option (void *data, int argc, char **argv)
+int
+option_option(void *data, int argc, char **argv)
 {
     if (!data)
         return 0;
 
     struct option *opts = (struct option *)data;
 
-    for (int k=0; opts[k].name; k++)
+    for (int k = 0; opts[k].name; k++)
         opts[k].set = 0;
 
-    for (int i=1; i<argc; i++) {
+    for (int i = 1; i < argc; i++) {
         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]))
                 continue;
 
@@ -68,9 +72,9 @@ int option_option (void *data, int argc, char **argv)
                 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;
 
             opts[k].set = 1;
@@ -81,29 +85,30 @@ int option_option (void *data, int argc, char **argv)
         }
 
         if (!found)
-            return i-1;
+            return i - 1;
     }
 
     return argc;
 }
 
-static int option_usage (struct option *opts, int slen)
+static int
+option_usage(struct option *opts, int slen)
 {
     if (!opts)
         return 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, "");
             len = 0;
         }
 
         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 {
             len += gt_print(" ARG");
         }
@@ -114,21 +119,22 @@ static int option_usage (struct option *opts, int slen)
     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);
 
-    if (ret==argc)
+    if (ret == argc)
         return 0;
 
-    if (ret<0 || ret+1>=argc)
+    if (ret < 0 || ret + 1 >= argc)
         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]);
 
-    if (slen>40) {
+    if (slen > 40) {
         slen = 12;
         gt_print("\n%*s", slen, "");
     }

src/state.c

@@ -3,26 +3,27 @@
 #include "state.h"
 #include "str.h"
 
-#include <stdio.h>
 #include <fcntl.h>
+#include <stdio.h>
 #include <sys/stat.h>
 
-int state_create (const char *filename)
+int
+state_create(const char *filename)
 {
     if (str_empty(filename))
         return -1;
 
     int fd = open(filename, O_WRONLY);
 
-    if (fd==-1) {
+    if (fd == -1) {
-        if (errno!=EINTR)
+        if (errno != EINTR)
             perror("open");
         return -1;
     }
 
     struct stat st = {0};
 
-    if (fstat(fd, &st)==-1) {
+    if (fstat(fd, &st) == -1) {
         perror("fstat");
         close(fd);
         return -1;
@@ -37,17 +38,18 @@ int state_create (const char *filename)
     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))
         return;
 
-    if (fd==-1) {
+    if (fd == -1) {
         gt_print("%s %s\n", state, info);
         return;
     }
 
-    const char *strs[] = { state, " ", info, "\n" };
+    const char *strs[] = {state, " ", info, "\n"};
     char *str = str_cat(strs, COUNT(strs));
 
     if (!str) {
@@ -55,7 +57,7 @@ void state_send (int fd, const char *state, const char *info)
         return;
     }
 
-    if (write(fd, str, str_len(str))==-1 && errno!=EINTR)
+    if (write(fd, str, str_len(str)) == -1 && errno != EINTR)
         perror("write");
 
     free(str);

src/tun.c

@@ -1,25 +1,26 @@
 #include "common.h"
 
-#include "tun.h"
-#include "str.h"
 #include "ip.h"
+#include "str.h"
+#include "tun.h"
 
-#include <stdio.h>
 #include <fcntl.h>
+#include <stdio.h>
 
 #include <sys/ioctl.h>
 #include <sys/socket.h>
 #include <sys/uio.h>
 
+#include <net/if.h>
+
 #ifdef __linux__
-#include <linux/if.h>
 #include <linux/if_tun.h>
 #endif
 
 #ifdef __APPLE__
-#include <sys/sys_domain.h>
-#include <sys/kern_control.h>
 #include <net/if_utun.h>
+#include <sys/kern_control.h>
+#include <sys/sys_domain.h>
 #endif
 
 #if defined(__APPLE__) || defined(__OpenBSD__)
@@ -28,17 +29,18 @@
 
 #ifdef __APPLE__
 
-static int tun_create_by_id (char *name, size_t size, unsigned id, _unused_ int mq)
+static int
+tun_create_by_id(char *name, size_t size, unsigned id)
 {
     int fd = socket(PF_SYSTEM, SOCK_DGRAM, SYSPROTO_CONTROL);
 
-    if (fd==-1)
+    if (fd == -1)
         return -1;
 
     struct ctl_info ci;
 
     memset(&ci, 0, sizeof(ci));
-    str_cpy(ci.ctl_name, UTUN_CONTROL_NAME, sizeof(ci.ctl_name)-1);
+    str_cpy(ci.ctl_name, UTUN_CONTROL_NAME, sizeof(ci.ctl_name) - 1);
 
     if (ioctl(fd, CTLIOCGINFO, &ci)) {
         close(fd);
@@ -50,7 +52,7 @@ static int tun_create_by_id (char *name, size_t size, unsigned id, _unused_ int
         .sc_len = sizeof(sc),
         .sc_family = AF_SYSTEM,
         .ss_sysaddr = AF_SYS_CONTROL,
-        .sc_unit = id+1,
+        .sc_unit = id + 1,
     };
 
     if (connect(fd, (struct sockaddr *)&sc, sizeof(sc))) {
@@ -63,98 +65,93 @@ static int tun_create_by_id (char *name, size_t size, unsigned id, _unused_ int
     return fd;
 }
 
-static int tun_create_by_name (char *name, size_t size, char *dev_name, int mq)
+static int
+tun_create_by_name(char *name, size_t size, char *dev_name)
 {
     unsigned id = 0;
 
-    if (sscanf(dev_name, "utun%u", &id)!=1)
+    if (sscanf(dev_name, "utun%u", &id) != 1)
         return -1;
 
-    return tun_create_by_id(name, size, id, mq);
+    return tun_create_by_id(name, size, id);
 }
 
 #else /* not __APPLE__ */
 
 #ifdef __linux__
 
-static int tun_create_by_name (char *name, size_t size, char *dev_name, int mq)
+static int
+tun_create_by_name(char *name, size_t size, char *dev_name)
 {
     int fd = open("/dev/net/tun", O_RDWR);
 
-    if (fd==-1)
+    if (fd == -1)
         return -1;
 
     struct ifreq ifr = {
-        .ifr_flags = IFF_TUN|IFF_NO_PI,
+        .ifr_flags = IFF_TUN | IFF_NO_PI,
     };
 
-    if (mq) {
+    str_cpy(ifr.ifr_name, dev_name, IFNAMSIZ - 1);
-#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);
+    str_cpy(name, ifr.ifr_name, size - 1);
 
     return fd;
 }
 
 #else /* not __linux__ not __APPLE__ */
 
-static int tun_create_by_name (char *name, size_t size, char *dev_name, _unused_ int mq)
+static int
+tun_create_by_name(char *name, size_t size, char *dev_name)
 {
     char path[64];
 
     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);
 }
 
 #endif /* not __APPLE__ */
 
-static int tun_create_by_id (char *name, size_t size, unsigned id, int mq)
+static int
+tun_create_by_id(char *name, size_t size, unsigned id)
 {
     char dev_name[64];
 
     snprintf(dev_name, sizeof(dev_name), "tun%u", id);
 
-    return tun_create_by_name(name, size, dev_name, mq);
+    return tun_create_by_name(name, size, dev_name);
 }
 
 #endif
 
-int tun_create (char *dev_name, char **ret_name, int mq)
+int
+tun_create(char *dev_name, char **ret_name)
 {
     char name[64] = {0};
     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++)
+        for (unsigned id = 0; id < 32 && fd == -1; id++)
-            fd = tun_create_by_id(name, sizeof(name), id, mq);
+            fd = tun_create_by_id(name, sizeof(name), id);
     } else {
-        fd = tun_create_by_name(name, sizeof(name), dev_name, mq);
+        fd = tun_create_by_name(name, sizeof(name), dev_name);
     }
 
-    if (fd!=-1 && ret_name)
+    if (fd != -1 && ret_name)
         *ret_name = strdup(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)
         return -1;
@@ -163,8 +160,14 @@ ssize_t tun_read (int fd, void *data, size_t size)
     uint32_t family;
 
     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);
@@ -172,8 +175,8 @@ ssize_t tun_read (int fd, void *data, size_t size)
     ssize_t ret = read(fd, data, size);
 #endif
 
-    if (ret==-1) {
+    if (ret == -1) {
-        if (errno==EAGAIN || errno==EINTR)
+        if (errno == EAGAIN || errno == EINTR)
             return -1;
 
         if (errno)
@@ -183,16 +186,17 @@ ssize_t tun_read (int fd, void *data, size_t size)
     }
 
 #ifdef GT_BSD_TUN
-    if (ret<(ssize_t) sizeof(family))
+    if (ret < (ssize_t)sizeof(family))
         return 0;
 
-    return ret-sizeof(family);
+    return ret - sizeof(family);
 #else
     return ret;
 #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)
         return -1;
@@ -212,8 +216,14 @@ ssize_t tun_write (int fd, const void *data, size_t size)
     }
 
     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);
@@ -221,8 +231,8 @@ ssize_t tun_write (int fd, const void *data, size_t size)
     ssize_t ret = write(fd, data, size);
 #endif
 
-    if (ret==-1) {
+    if (ret == -1) {
-        if (errno==EAGAIN || errno==EINTR)
+        if (errno == EAGAIN || errno == EINTR)
             return -1;
 
         if (errno)
@@ -232,11 +242,34 @@ ssize_t tun_write (int fd, const void *data, size_t size)
     }
 
 #ifdef GT_BSD_TUN
-    if (ret<(ssize_t) sizeof(family))
+    if (ret < (ssize_t)sizeof(family))
         return 0;
 
-    return ret-sizeof(family);
+    return ret - sizeof(family);
 #else
     return ret;
 #endif
 }
+
+int
+tun_set_mtu(char *dev_name, int mtu)
+{
+    struct ifreq ifr = {
+        .ifr_mtu = mtu,
+    };
+
+    str_cpy(ifr.ifr_name, dev_name, IFNAMSIZ - 1);
+
+    int fd = socket(AF_INET, SOCK_DGRAM, 0);
+
+    if (fd == -1)
+        return -1;
+
+    int ret = ioctl(fd, SIOCSIFMTU, &ifr);
+
+    int err = errno;
+    close(fd);
+    errno = err;
+
+    return ret;
+}

src/tun.h

@@ -2,6 +2,7 @@
 
 #include <unistd.h>
 
-int     tun_create (char *, char **, int);
+int     tun_create  (char *, char **);
 ssize_t tun_read    (int, void *, size_t);
 ssize_t tun_write   (int, const void *, size_t);
+int     tun_set_mtu (char *, int);