HevConf: Add support for custom cycle of UDP STUN check.

Author: muink

README.md

@@ -51,6 +51,7 @@ Options:
  -d                  run as daemon
  -i <interface>      network interface or IP address
  -k <interval>       seconds between each keep-alive
+ -c <count>          UDP STUN check cycle (every <count> intervals)
  -s <addr>[:port]    domain name or address of STUN server
  -h <addr>[:port]    domain name or address of HTTP server
  -e <path>           script path for notify mapped address

src/hev-conf.c

@@ -21,6 +21,7 @@
 static int mode = SOCK_STREAM;
 static int type = AF_INET;
 static int keep;
+static unsigned int ucount;
 static int dmon;
 static int tmsec;
 static int bport[3];
@@ -52,6 +53,7 @@ hev_conf_help (void)
         " -d                  run as daemon\n"
         " -i <interface>      network interface or IP address\n"
         " -k <interval>       seconds between each keep-alive\n"
+        " -c <count>          UDP STUN check cycle (every <count> intervals)\n"
         " -s <addr>[:port]    domain name or address of STUN server\n"
         " -h <addr>[:port]    domain name or address of HTTP server\n"
         " -e <path>           script path for notify mapped address\n"
@@ -94,6 +96,9 @@ hev_conf_init (int argc, char *argv[])
         case 'k':
             keep = strtoul (optarg, NULL, 10) * 1000;
             break;
+        case 'c':
+            ucount = strtoul (optarg, NULL, 10);
+            break;
         case 's':
             sscanf (optarg, "%255[^:]:%5[0123456789]", stun, sport);
             break;
@@ -143,6 +148,10 @@ hev_conf_init (int argc, char *argv[])
         keep *= 1000;
     }
 
+    if (ucount <= 0) {
+        ucount = 10;
+    }
+
     if (!bport[0]) {
         bport[0] = 0;
     }
@@ -183,6 +192,12 @@ hev_conf_keep (void)
     return keep;
 }
 
+unsigned int
+hev_conf_ucount (void)
+{
+    return ucount;
+}
+
 const char *
 hev_conf_stun (void)
 {

src/hev-conf.h

@@ -57,6 +57,15 @@ int hev_conf_type (void);
  */
 int hev_conf_keep (void);
 
+/**
+ * hev_conf_ucount:
+ *
+ * Get the UDP STUN check cycle value.
+ *
+ * Returns: returns integer number.
+ */
+unsigned int hev_conf_ucount (void);
+
 /**
  * hev_conf_stun:
  *

src/hev-unsk.c

@@ -28,6 +28,7 @@ static struct sockaddr_storage saddr;
 static struct sockaddr_storage daddr;
 static HevTask *task;
 static int timeout;
+static unsigned int cycle;
 
 static void
 stun_handler (void)
@@ -54,7 +55,7 @@ unsk_keep_alive (void)
         }
 
         fd = hev_ufwd_fd ();
-        if ((fd < 0) || (++n >= 10)) {
+        if ((fd < 0) || (++n >= cycle)) {
             hev_stun_run ((struct sockaddr *)&saddr, stun_handler);
             n = 0;
             continue;
@@ -100,6 +101,7 @@ unsk_run (void)
     iface = hev_conf_iface ();
     mark = hev_conf_mark ();
     timeout = hev_conf_keep ();
+    cycle = hev_conf_ucount ();
 
     fd = hev_sock_client_base (type, SOCK_DGRAM, addr, port, stun, sport, iface,
                                mark, &saddr, &daddr);