From 010ee3abba7bdca41364cffe9969308a7a641909 Mon Sep 17 00:00:00 2001 From: fred_nerk Date: Mon, 24 May 2004 04:28:41 +0000 Subject: [PATCH] Check return code when throttling users --- rl.c | 55 ++++++++++++++++++++++++++++++------------------------ throttle.c | 13 ++++++++----- 2 files changed, 39 insertions(+), 29 deletions(-) diff --git a/rl.c b/rl.c index 6abb78c..1692944 100644 --- a/rl.c +++ b/rl.c @@ -1,32 +1,32 @@ // L2TPNS Rate Limiting Stuff -// $Id: rl.c,v 1.3 2004/04/05 05:29:13 fred_nerk Exp $ +// $Id: rl.c,v 1.4 2004/05/24 04:28:41 fred_nerk Exp $ +#include +#include +#include #include +#include +#include #include #include -#include -#include +#include +#include +#include #include -#include -#include #include "l2tpns.h" extern radiust *radius; extern sessiont *session; extern u32 sessionid; -extern int radfd; extern tbft *filter_buckets; extern struct configt *config; #define DEVICE "tun0" -int next_tbf = 1; - void init_rl() { char *commands[] = { "tc qdisc add dev " DEVICE " root handle 1: htb", - "tc class add dev " DEVICE " parent 1: classid 1:1 htb rate 100mbit burst 300k", "tc filter del dev " DEVICE " protocol ip pref 1 fw", "iptables -t mangle -N throttle 2>&1 >/dev/null", "iptables -t mangle -F throttle 2>&1 >/dev/null", @@ -50,22 +50,31 @@ u16 rl_create_tbf() char cmd[2048]; 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); 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, 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); + 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; } @@ -76,12 +85,12 @@ u16 rl_get_tbf() 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; @@ -91,7 +100,6 @@ u16 rl_get_tbf() void rl_done_tbf(u16 t) { if (!t) return; - if (!config->rl_rate) return; log(2, 0, 0, 0, "Freeing up HTB %s\n", filter_buckets[t].handle); filter_buckets[t].in_use = 0; } @@ -106,9 +114,8 @@ void rl_destroy_tbf(u16 t) return; } snprintf(cmd, 2048, "tc qdisc del dev " DEVICE " handle %s", filter_buckets[t].handle); - system(cmd); - system("iptables -t mangle -D l2tpns -j throttle 2>&1 >/dev/null"); - system("iptables -t mangle -X throttle 2>&1 >/dev/null"); + 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)); } diff --git a/throttle.c b/throttle.c index 7744204..7244cfa 100644 --- a/throttle.c +++ b/throttle.c @@ -1,5 +1,5 @@ // L2TPNS Throttle Stuff -// $Id: throttle.c,v 1.2 2004/03/05 00:09:03 fred_nerk Exp $ +// $Id: throttle.c,v 1.3 2004/05/24 04:29:21 fred_nerk Exp $ #include #include @@ -19,7 +19,6 @@ extern radiust *radius; extern sessiont *session; extern u32 sessionid; -extern int radfd; extern tbft *filter_buckets; extern struct configt *config; @@ -41,12 +40,16 @@ int throttle_session(sessionidt s, int throttle) log(1, 0, s, session[s].tunnel, "Error creating a filtering bucket for user %s\n", session[s].user); return 0; } - log(2, 0, s, session[s].tunnel, "Throttling session %d for user %s\n", s, session[s].user); + log(2, 0, s, session[s].tunnel, "Throttling session %d for user %s (bucket %s)\n", s, session[s].user, filter_buckets[session[s].tbf].handle); snprintf(cmd, 2048, "iptables -t mangle -A throttle -d %s -j MARK --set-mark %d", inet_toa(ntohl(session[s].ip)), session[s].tbf); log(4, 0, s, session[s].tunnel, "Running %s\n", cmd); - system(cmd); + if (WEXITSTATUS(system(cmd)) != 0) + { + log(2, 0, s, session[s].tunnel, "iptables returned an error. Session is not throttled\n"); + return 0; + } } else { @@ -69,6 +72,6 @@ int throttle_session(sessionidt s, int throttle) } } session[s].throttle = throttle; - return 0; + return session[s].throttle; } -- 2.20.1