projects / l2tpns.git / blobdiff
? search:
summary | shortlog | log | commit | commitdiff | tree
raw | inline | side by side
Foo
[l2tpns.git] / rl.c
diff --git a/rl.c b/rl.c
index 7b2084b..9f2f14a 100644 (file)
--- a/rl.c
+++ b/rl.c
@@ -1,57 +1,40 @@
 // L2TPNS Rate Limiting Stuff
 // L2TPNS Rate Limiting Stuff
-// $Id: rl.c,v 1.1 2003-12-16 07:07:39 fred_nerk Exp $
+// $Id: rl.c,v 1.4 2004-05-24 04:28:41 fred_nerk Exp $
 
 
+#include <arpa/inet.h>
+#include <errno.h>
+#include <malloc.h>
 #include <stdio.h>
 #include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
 #include <sys/file.h>
 #include <sys/stat.h>
 #include <sys/file.h>
 #include <sys/stat.h>
-#include <errno.h>
-#include <string.h>
+#include <sys/types.h>
+#include <sys/wait.h>
+#include <time.h>
 #include <unistd.h>
 #include <unistd.h>
-#include <stdlib.h>
-#include <malloc.h>
 #include "l2tpns.h"
 
 #include "l2tpns.h"
 
-extern char *radiussecret;
 extern radiust *radius;
 extern sessiont *session;
 extern radiust *radius;
 extern sessiont *session;
-extern ipt radiusserver[MAXRADSERVER]; // radius servers
 extern u32 sessionid;
 extern u32 sessionid;
-extern u8 radiusfree;
-extern int radfd;
-extern u8 numradiusservers;
-extern char debug;
-extern char *tapdevice;
 extern tbft *filter_buckets;
 extern tbft *filter_buckets;
+extern struct configt *config;
 
 #define DEVICE "tun0"
 
 
 #define DEVICE "tun0"
 
-unsigned long rl_rate = 0;
-int next_tbf = 1;
-
 void init_rl()
 {
 void init_rl()
 {
-#ifdef TC_TBF
-    system("tc qdisc del dev " DEVICE " root");
-    system("tc qdisc add dev " DEVICE " root handle 1: cbq avpkt 10000 bandwidth 100mbit");
-    system("tc filter del dev " DEVICE " protocol ip pref 1 fw");
-    system("iptables -t mangle -N throttle 2>&1 > /dev/null");
-    system("iptables -t mangle -F throttle");
-    system("iptables -t mangle -A l2tpns -j throttle");
-#endif
-#ifdef TC_HTB
     char *commands[] = {
     char *commands[] = {
-       "tc qdisc add dev " DEVICE " root handle 1: htb default 1",
-       "tc class add dev " DEVICE " parent 1: classid 1:1 htb rate 100mbit burst 300k",
+       "tc qdisc add dev " DEVICE " root handle 1: htb",
        "tc filter del dev " DEVICE " protocol ip pref 1 fw",
        "tc filter del dev " DEVICE " protocol ip pref 1 fw",
-       "iptables -t mangle -N throttle 2>&1 > /dev/null",
-       "iptables -t mangle -F throttle",
-       "iptables -t mangle -A l2tpns -j throttle",
+       "iptables -t mangle -N throttle 2>&1 >/dev/null",
+       "iptables -t mangle -F throttle 2>&1 >/dev/null",
+       "iptables -t mangle -A l2tpns -j throttle 2>&1 >/dev/null",
        NULL
     };
     int i;
 
        NULL
     };
     int i;
 
-    if (!rl_rate) return;
-
     log(2, 0, 0, 0, "Initializing HTB\n");
     for (i = 0; commands[i] && *commands[i]; i++)
     {
     log(2, 0, 0, 0, "Initializing HTB\n");
     for (i = 0; commands[i] && *commands[i]; i++)
     {
@@ -59,67 +42,55 @@ void init_rl()
        system(commands[i]);
     }
     log(2, 0, 0, 0, "Done initializing HTB\n");
        system(commands[i]);
     }
     log(2, 0, 0, 0, "Done initializing HTB\n");
-#endif
 }
 
 u16 rl_create_tbf()
 {
     u16 t;
     char cmd[2048];
 }
 
 u16 rl_create_tbf()
 {
     u16 t;
     char cmd[2048];
-    if (!rl_rate) return 0;
+    if (!config->rl_rate) return 0;
 
 
-    if (next_tbf >= MAXSESSION) return 0;
-    t = next_tbf++;
+    t = ++config->next_tbf;
+    if (config->next_tbf >= MAXSESSION) return 0;
     snprintf(filter_buckets[t].handle, 9, "1:%d0", t);
 
     snprintf(filter_buckets[t].handle, 9, "1:%d0", t);
 
-#ifdef TC_TBF
-    log(2, 0, 0, 0, "Creating new tbf %s\n", filter_buckets[t].handle);
-    snprintf(cmd, 2048, "tc class add dev " DEVICE " parent 1: classid 1:%d cbq bandwidth 100Mbit rate 100Mbit "
-           "weight 1 prio 8 allot 1514 cell 8 maxburst 20 avpkt 1000 bounded isolated",
-           t);
-    log(3, 0, 0, 0, "%s\n", cmd);
-    system(cmd);
-
-    snprintf(cmd, 2048, "tc qdisc add dev " DEVICE " parent 1:%d handle %s tbf rate %dkbit buffer 1600 limit 3000",
-           t, filter_buckets[t].handle, rl_rate);
-    log(3, 0, 0, 0, "%s\n", cmd);
-    system(cmd);
-
-    snprintf(cmd, 2048, "tc filter add dev " DEVICE " protocol ip parent 1:0 prio 1 handle %d fw flowid 1:%d",
-           t, t);
-    log(3, 0, 0, 0, "%s\n", cmd);
-    system(cmd);
-#endif
-#ifdef TC_HTB
     log(2, 0, 0, 0, "Creating new htb %s\n", filter_buckets[t].handle);
     snprintf(cmd, 2048, "tc class add dev " DEVICE " parent 1: classid %s htb rate %lukbit burst 15k",
     log(2, 0, 0, 0, "Creating new htb %s\n", filter_buckets[t].handle);
     snprintf(cmd, 2048, "tc class add dev " DEVICE " parent 1: classid %s htb rate %lukbit burst 15k",
-           filter_buckets[t].handle, rl_rate);
+           filter_buckets[t].handle, config->rl_rate);
     log(3, 0, 0, 0, "%s\n", cmd);
     log(3, 0, 0, 0, "%s\n", cmd);
-    system(cmd);
+    if (WEXITSTATUS(system(cmd)) != 0)
+    {
+       memset(filter_buckets[t].handle, 0, sizeof(filter_buckets[t].handle));
+       log(0, 0, 0, 0, "tc returned an error creating a token bucket\n");
+       return 0;
+    }
 
     snprintf(cmd, 2048, "tc filter add dev " DEVICE " protocol ip parent 1:0 prio 1 handle %d fw flowid %s",
            t, filter_buckets[t].handle);
     log(3, 0, 0, 0, "%s\n", cmd);
 
     snprintf(cmd, 2048, "tc filter add dev " DEVICE " protocol ip parent 1:0 prio 1 handle %d fw flowid %s",
            t, filter_buckets[t].handle);
     log(3, 0, 0, 0, "%s\n", cmd);
-    system(cmd);
-#endif
+    if (WEXITSTATUS(system(cmd)) != 0)
+    {
+       memset(filter_buckets[t].handle, 0, sizeof(filter_buckets[t].handle));
+       log(0, 0, 0, 0, "tc returned an error creating a filter\n");
+       return 0;
+    }
 
 
-    next_tbf++;
     return t;
 }
 
 u16 rl_get_tbf()
 {
     int i;
     return t;
 }
 
 u16 rl_get_tbf()
 {
     int i;
-    if (!rl_rate) return 0;
+    if (!config->rl_rate) return 0;
 
     for (i = 1; i < MAXSESSION; i++)
     {
 
     for (i = 1; i < MAXSESSION; i++)
     {
-       if (!filter_buckets[i].in_use && *filter_buckets[i].handle)
-       {
-           filter_buckets[i].in_use = 1;
-           log(2, 0, 0, 0, "Returning tbf %s\n", filter_buckets[i].handle);
-           return i;
-       }
+       if (!*filter_buckets[i].handle) continue;
+       if (filter_buckets[i].in_use) continue;
+
+       filter_buckets[i].in_use = 1;
+       log(2, 0, 0, 0, "Returning tbf %s\n", filter_buckets[i].handle);
+       return i;
     }
     i = rl_create_tbf();
     if (i) filter_buckets[i].in_use = 1;
     }
     i = rl_create_tbf();
     if (i) filter_buckets[i].in_use = 1;
@@ -129,30 +100,22 @@ u16 rl_get_tbf()
 void rl_done_tbf(u16 t)
 {
     if (!t) return;
 void rl_done_tbf(u16 t)
 {
     if (!t) return;
-    if (!rl_rate) return;
-    log(2, 0, 0, 0, "Freeing up TBF %s\n", filter_buckets[t].handle);
+    log(2, 0, 0, 0, "Freeing up HTB %s\n", filter_buckets[t].handle);
     filter_buckets[t].in_use = 0;
 }
 
 void rl_destroy_tbf(u16 t)
 {
     char cmd[2048];
     filter_buckets[t].in_use = 0;
 }
 
 void rl_destroy_tbf(u16 t)
 {
     char cmd[2048];
-    if (!rl_rate) return;
+    if (!config->rl_rate) return;
     if (filter_buckets[t].in_use)
     {
     if (filter_buckets[t].in_use)
     {
-       log(0, 0, 0, 0, "Trying to destroy an in-use TBF %s\n", filter_buckets[t].handle);
+       log(0, 0, 0, 0, "Trying to destroy an in-use HTB %s\n", filter_buckets[t].handle);
        return;
     }
        return;
     }
-#ifdef TC_TBF
-    snprintf(cmd, 2048, "tc qdisc del dev " DEVICE " handle %s", filter_buckets[t].handle);
-    system(cmd);
-#endif
-#ifdef TC_HTB
     snprintf(cmd, 2048, "tc qdisc del dev " DEVICE " handle %s", filter_buckets[t].handle);
     snprintf(cmd, 2048, "tc qdisc del dev " DEVICE " handle %s", filter_buckets[t].handle);
-    system(cmd);
-#endif
-    system("iptables -t mangle -D l2tpns -j throttle");
-    system("iptables -t mangle -X throttle");
+    if (WEXITSTATUS(system(cmd)) != 0)
+       log(0, 0, 0, 0, "tc returned an error deleting a token bucket\n");
     memset(filter_buckets[t].handle, 0, sizeof(filter_buckets[t].handle));
 }
 
     memset(filter_buckets[t].handle, 0, sizeof(filter_buckets[t].handle));
 }
 
for cloning use: "git clone git://git.sameswireless.fr/l2tpns.git" ; For add to your sources.list: "deb http://package.sameswifi.fr/ jessie main"