Version 0.0.7

configure.ac

@@ -1,5 +1,5 @@
 AC_PREREQ([2.65])
-AC_INIT([glorytun], [0.0.6], [https://github.com/angt/glorytun/issues],
+AC_INIT([glorytun], [0.0.7], [https://github.com/angt/glorytun/issues],
         [glorytun], [https://github.com/angt/glorytun])
 AC_CONFIG_SRCDIR([src/common.h])
 AC_CONFIG_AUX_DIR([build-aux])

src/main.c

@@ -25,6 +25,8 @@
 #define GT_BUFFER_SIZE  (4*1024*1024)
 #define GT_TIMEOUT      (1000)
 #define GT_MTU_MAX      (1500)
+#define GT_TUNR_SIZE    (0x7FFF-16)
+#define GT_TUNW_SIZE    (0x7FFF)
 
 struct fdbuf {
     int fd;
@@ -636,7 +638,8 @@ int main (int argc, char **argv)
     long ka_count = -1;
     long ka_idle = -1;
     long ka_interval = -1;
-    long dscp_prio = 0x3F;
+    long prio_dscp = 46;
+    long prio_size = (GT_TUNR_SIZE*3)/4;
 
 #ifdef TCP_INFO
     struct {
@@ -652,6 +655,12 @@ int main (int argc, char **argv)
         { NULL },
     };
 
+    struct option prio_opts[] = {
+        { "dscp",  &prio_dscp, option_long },
+        { "size",  &prio_size, option_long },
+        { NULL },
+    };
+
     struct option opts[] = {
         { "listener",    NULL,         option_option },
         { "host",        &host,        option_str    },
@@ -663,7 +672,7 @@ int main (int argc, char **argv)
         { "multiqueue",  NULL,         option_option },
         { "keepalive",   ka_opts,      option_option },
         { "buffer-size", &buffer_size, option_long   },
-        { "dscp-prio",   &dscp_prio,   option_long   },
+        { "priority",    prio_opts,    option_option },
         { "daemon",      NULL,         option_option },
         { "debug",       NULL,         option_option },
         { "version",     NULL,         option_option },
@@ -688,6 +697,16 @@ int main (int argc, char **argv)
         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) {
         gt_log("libsodium initialization has failed!\n");
         return 1;
@@ -727,8 +746,8 @@ int main (int argc, char **argv)
 
     fd_set_nonblock(tun.fd);
 
-    buffer_setup(&tun.write, NULL, 0x7FFF);
-    buffer_setup(&tun.read, NULL, 0x7FFF-16);
+    buffer_setup(&tun.write, NULL, GT_TUNW_SIZE);
+    buffer_setup(&tun.read,  NULL, GT_TUNR_SIZE);
 
     buffer_setup(&sock.write, NULL, buffer_size);
     buffer_setup(&sock.read,  NULL, buffer_size);
@@ -874,9 +893,11 @@ int main (int argc, char **argv)
                         }
                     }
 
-                    if (ip_get_dscp(data, GT_MTU_MAX)==dscp_prio) {
+                    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;
@@ -891,10 +912,13 @@ int main (int argc, char **argv)
             if (blk_prio) {
                 uint8_t k = blk_read;
 
-                while (blk_prio && buffer_write_size(&tun.read)>8*GT_MTU_MAX) {
-                    while (!blks[k].prio)
+                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;
 
@@ -1009,8 +1033,13 @@ int main (int argc, char **argv)
 
     freeaddrinfo(ai);
 
+    free(blks);
+
     free(sock.write.data);
     free(sock.read.data);
 
+    free(tun.write.data);
+    free(tun.read.data);
+
     return 0;
 }

src/option.c

@@ -97,7 +97,7 @@ static int option_usage (struct option *opts, int slen)
     int len = slen;
 
     for (int k=0; opts[k].name; k++) {
-        if (len>60) {
+        if (len>slen+40) {
             gt_print("\n%*s", (int)slen, "");
             len = slen;
         }
@@ -130,8 +130,10 @@ int option (struct option *opts, int argc, char **argv)
 
     int slen = gt_print("usage: %s", argv[0]);
 
-    if (slen>40)
+    if (slen>40) {
         slen = 12;
+        gt_print("\n%*s", (int)slen, "");
+    }
 
     option_usage(opts, slen);