X-Git-Url: http://git.sameswireless.fr/l2tpns.git/blobdiff_plain/c2f98ee93148952298986946a26be28cab46eb5b..024ba5b35b10ce4008cec64602f427c1614fd729:/autothrottle.c diff --git a/autothrottle.c b/autothrottle.c index 4d985a2..be7233b 100644 --- a/autothrottle.c +++ b/autothrottle.c @@ -1,42 +1,116 @@ #include -#include -#include -#include -#include #include "l2tpns.h" #include "plugin.h" -#include "control.h" -int __plugin_api_version = 1; -struct pluginfuncs p; +/* set up throttling based on RADIUS reply */ + +/* + * lcp:interface-config#1=service-policy input N + * lcp:interface-config#2=service-policy output N + * + * throttle=N + * throttle=yes (use throttle_rate from config) + * throttle=no + */ + +char const *cvs_id = "$Id: autothrottle.c,v 1.13.2.1 2005/05/16 05:00:50 bodea Exp $"; + +int plugin_api_version = PLUGIN_API_VERSION; +struct pluginfuncs *p; + +#define THROTTLE_KEY "lcp:interface-config" int plugin_radius_response(struct param_radius_response *data) { - if (strcmp(data->key, "throttle") == 0) + if (!strncmp(data->key, THROTTLE_KEY, sizeof(THROTTLE_KEY) - 1)) { - if (strcmp(data->value, "yes") == 0) + char *sp = strchr(data->value, ' '); + char type; + int rate; + + if (!sp || sp - data->value < 4 || + strncmp("service-policy", data->value, sp - data->value)) + return PLUGIN_RET_OK; + + while (*sp == ' ') sp++; + data->value = sp; + + if (!(sp = strchr(data->value, ' ')) || + (strncmp("input", data->value, sp - data->value) && + strncmp("output", data->value, sp - data->value))) + { + p->log(3, p->get_id_by_session(data->s), data->s->tunnel, + " Not throttling user (invalid type %.*s)\n", + sp - data->value, data->value); + + return PLUGIN_RET_OK; + } + + type = *data->value; + + while (*sp == ' ') sp++; + data->value = sp; + + if ((rate = strtol(data->value, &sp, 10)) < 0 || *sp) + { + p->log(3, p->get_id_by_session(data->s), data->s->tunnel, + " Not throttling user (invalid rate %s)\n", + data->value); + + return PLUGIN_RET_OK; + } + + if (type == 'i') { - p.log(3, 0, 0, 0, " Throttling user\n"); - data->s->throttle = 1; + data->s->throttle_in = rate; + p->log(3, p->get_id_by_session(data->s), data->s->tunnel, + " Throttling user input to %dkb/s\n", + rate); } - else if (strcmp(data->value, "no") == 0) + else { - p.log(3, 0, 0, 0, " Not throttling user\n"); - data->s->throttle = 0; + data->s->throttle_out = rate; + p->log(3, p->get_id_by_session(data->s), data->s->tunnel, + " Throttling user output to %dkb/s\n", + rate); } } - return PLUGIN_RET_OK; -} -int plugin_init(struct pluginfuncs *funcs) -{ - if (!funcs) return 0; - memcpy(&p, funcs, sizeof(p)); + if (!strcmp(data->key, "throttle")) + { + char *e; + int rate; + + if ((rate = strtol(data->value, &e, 10)) < 0 || *e) + { + rate = -1; + if (!strcmp(data->value, "yes")) + { + unsigned long *ts = p->getconfig("throttle_speed", UNSIGNED_LONG); + if (ts) + rate = *ts; + } + else if (!strcmp(data->value, "no")) + rate = 0; + } + + if (rate < 0) + return PLUGIN_RET_OK; + + if (rate) + p->log(3, p->get_id_by_session(data->s), data->s->tunnel, + " Throttling user to %dkb/s\n", rate); + else + p->log(3, p->get_id_by_session(data->s), data->s->tunnel, + " Not throttling user\n"); + + data->s->throttle_in = data->s->throttle_out = rate; + } - return 1; + return PLUGIN_RET_OK; } -void plugin_done() +int plugin_init(struct pluginfuncs *funcs) { + return ((p = funcs)) ? 1 : 0; } -