l2tpns/autothrottle.c

118 lines
2.9 KiB
C

#include <string.h>
#include "l2tpns.h"
#include "plugin.h"
/* set up throttling based on RADIUS reply */
char const *cvs_id = "$Id: autothrottle.c,v 1.11 2004-11-30 00:46:36 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)
{
char *t;
int i = 0;
int rate;
if (strncmp(data->key, THROTTLE_KEY, sizeof(THROTTLE_KEY) - 1) == 0)
{
char *pt;
if (strncmp(data->value, "serv", 4))
return PLUGIN_RET_OK;
pt = strdup(data->value);
while ((t = strsep(&pt, " ")) != NULL)
{
if (strcmp(t, "serv") == 0)
i = 1;
else if (strcmp(t, "o") == 0 && i == 1)
i = 3;
else if (strcmp(t, "i") == 0 && i == 1)
i = 2;
else if (i > 1 )
{
if ((rate = strtol(t, (char **)NULL, 10)) < 0 )
{
p->log(3, p->get_id_by_session(data->s), data->s->tunnel,
"Syntax Error: rate is not a number %s\n", t);
free(pt);
return PLUGIN_RET_OK;
}
switch (i)
{
case 2: // output
data->s->throttle_out = rate;
free(pt);
p->log(3, p->get_id_by_session(data->s), data->s->tunnel,
" Set output throttle rate %dkb/s\n", rate);
return PLUGIN_RET_OK;
case 3: //input
data->s->throttle_in = rate;
free(pt);
p->log(3, p->get_id_by_session(data->s), data->s->tunnel,
" Set input throttle rate %dkb/s\n", rate);
return PLUGIN_RET_OK;
default:
p->log(1, p->get_id_by_session(data->s), data->s->tunnel,
"Syntax error in rate limit AV pair: %s=%s\n", data->key, data->value);
free(pt);
return PLUGIN_RET_OK;
}
}
}
p->log(3, p->get_id_by_session(data->s), data->s->tunnel,
"Unknown lcp:interface-config AV pair %s=%s\n",data->key, data->value);
free(pt);
return PLUGIN_RET_OK;
}
else if (strcmp(data->key, "throttle") == 0)
{
if (strcmp(data->value, "yes") == 0)
{
unsigned long *rate = p->getconfig("throttle_speed", UNSIGNED_LONG);
if (rate)
{
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 (throttle_speed=0)\n");
data->s->throttle_in = data->s->throttle_out = *rate;
}
else
p->log(1, p->get_id_by_session(data->s), data->s->tunnel,
"Not throttling user (can't get throttle_speed)\n");
}
else if (strcmp(data->value, "no") == 0)
{
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 = 0;
}
}
p->log(4, p->get_id_by_session(data->s), data->s->tunnel,
"autothrottle module ignoring AV pair %s=%s\n",
data->key, data->value);
return PLUGIN_RET_OK;
}
int plugin_init(struct pluginfuncs *funcs)
{
return ((p = funcs)) ? 1 : 0;
}