// 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 <stdlib.h>
+#include <string.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 <stdlib.h>
-#include <malloc.h>
#include "l2tpns.h"
-extern char *radiussecret;
extern radiust *radius;
extern sessiont *session;
-extern ipt radiusserver[MAXRADSERVER]; // radius servers
extern u32 sessionid;
-extern u8 radiusfree;
-extern int radfd;
-extern u8 numradiusservers;
-extern char debug;
-extern char *tapdevice;
extern tbft *filter_buckets;
+extern struct configt *config;
#define DEVICE "tun0"
-unsigned long rl_rate = 0;
-int next_tbf = 1;
-
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[] = {
- "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",
- "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;
- if (!rl_rate) return;
-
log(2, 0, 0, 0, "Initializing HTB\n");
for (i = 0; commands[i] && *commands[i]; i++)
{
system(commands[i]);
}
log(2, 0, 0, 0, "Done initializing HTB\n");
-#endif
}
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);
-#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",
- filter_buckets[t].handle, rl_rate);
+ filter_buckets[t].handle, config->rl_rate);
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);
- 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;
- if (!rl_rate) return 0;
+ if (!config->rl_rate) return 0;
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;
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];
- if (!rl_rate) return;
+ if (!config->rl_rate) return;
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;
}
-#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);
- 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));
}
projects / l2tpns.git / blobdiff