add Cisco-AVPairs to RADIUS accounting records via plugin_radius_account
This commit is contained in:
parent
620249a2af
commit
890bffb510
12 changed files with 668 additions and 584 deletions
202
autothrottle.c
202
autothrottle.c
|
|
@ -13,110 +13,146 @@
|
|||
* throttle=no
|
||||
*/
|
||||
|
||||
char const *cvs_id = "$Id: autothrottle.c,v 1.15 2005-10-11 07:59:09 bodea Exp $";
|
||||
char const *cvs_id = "$Id: autothrottle.c,v 1.16 2005-10-11 09:04:53 bodea Exp $";
|
||||
|
||||
int plugin_api_version = PLUGIN_API_VERSION;
|
||||
struct pluginfuncs *p;
|
||||
static struct pluginfuncs *f = 0;
|
||||
|
||||
#define THROTTLE_KEY "lcp:interface-config"
|
||||
|
||||
int plugin_radius_response(struct param_radius_response *data)
|
||||
{
|
||||
if (!strncmp(data->key, THROTTLE_KEY, sizeof(THROTTLE_KEY) - 1))
|
||||
if (!strncmp(data->key, THROTTLE_KEY, sizeof(THROTTLE_KEY) - 1))
|
||||
{
|
||||
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)))
|
||||
{
|
||||
char *sp = strchr(data->value, ' ');
|
||||
char type;
|
||||
int rate;
|
||||
f->log(3, f->get_id_by_session(data->s), data->s->tunnel,
|
||||
" Not throttling user (invalid type %.*s)\n",
|
||||
sp - data->value, data->value);
|
||||
|
||||
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')
|
||||
{
|
||||
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
|
||||
{
|
||||
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;
|
||||
}
|
||||
|
||||
if (!strcmp(data->key, "throttle"))
|
||||
type = *data->value;
|
||||
|
||||
while (*sp == ' ') sp++;
|
||||
data->value = sp;
|
||||
|
||||
if ((rate = strtol(data->value, &sp, 10)) < 0 || *sp)
|
||||
{
|
||||
char *e;
|
||||
int rate;
|
||||
f->log(3, f->get_id_by_session(data->s), data->s->tunnel,
|
||||
" Not throttling user (invalid rate %s)\n",
|
||||
data->value);
|
||||
|
||||
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 PLUGIN_RET_OK;
|
||||
}
|
||||
|
||||
return PLUGIN_RET_OK;
|
||||
if (type == 'i')
|
||||
{
|
||||
data->s->throttle_in = rate;
|
||||
f->log(3, f->get_id_by_session(data->s), data->s->tunnel,
|
||||
" Throttling user input to %dkb/s\n", rate);
|
||||
}
|
||||
else
|
||||
{
|
||||
data->s->throttle_out = rate;
|
||||
f->log(3, f->get_id_by_session(data->s), data->s->tunnel,
|
||||
" Throttling user output to %dkb/s\n", rate);
|
||||
}
|
||||
}
|
||||
else 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 = f->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)
|
||||
f->log(3, f->get_id_by_session(data->s), data->s->tunnel,
|
||||
" Throttling user to %dkb/s\n", rate);
|
||||
else
|
||||
f->log(3, f->get_id_by_session(data->s), data->s->tunnel,
|
||||
" Not throttling user\n");
|
||||
|
||||
data->s->throttle_in = data->s->throttle_out = rate;
|
||||
}
|
||||
|
||||
return PLUGIN_RET_OK;
|
||||
}
|
||||
|
||||
int plugin_radius_reset(struct param_radius_reset *data)
|
||||
{
|
||||
p->throttle(p->get_id_by_session(data->s), 0, 0);
|
||||
return PLUGIN_RET_OK;
|
||||
f->throttle(f->get_id_by_session(data->s), 0, 0);
|
||||
return PLUGIN_RET_OK;
|
||||
}
|
||||
|
||||
int plugin_radius_account(struct param_radius_account *data)
|
||||
{
|
||||
if (data->s->throttle_in || data->s->throttle_out)
|
||||
{
|
||||
uint8_t *p = *data->packet;
|
||||
int i = 1;
|
||||
|
||||
if (data->s->throttle_in)
|
||||
{
|
||||
*p = 26; // vendor-specific
|
||||
*(uint32_t *) (p + 2) = htonl(9); // Cisco
|
||||
p[6] = 1; // Cisco-AVPair
|
||||
p[7] = 2 + sprintf((char *) p + 8,
|
||||
"lcp:interface-config#%d=service-policy input %d", i++,
|
||||
data->s->throttle_in);
|
||||
|
||||
p[1] = p[7] + 6;
|
||||
p += p[1];
|
||||
}
|
||||
|
||||
if (data->s->throttle_out)
|
||||
{
|
||||
*p = 26; // vendor-specific
|
||||
*(uint32_t *) (p + 2) = htonl(9); // Cisco
|
||||
p[6] = 1; // Cisco-AVPair
|
||||
p[7] = 2 + sprintf((char *) p + 8,
|
||||
"lcp:interface-config#%d=service-policy output %d", i++,
|
||||
data->s->throttle_out);
|
||||
|
||||
p[1] = p[7] + 6;
|
||||
p += p[1];
|
||||
}
|
||||
|
||||
*data->packet = p;
|
||||
}
|
||||
|
||||
return PLUGIN_RET_OK;
|
||||
}
|
||||
|
||||
int plugin_init(struct pluginfuncs *funcs)
|
||||
{
|
||||
return ((p = funcs)) ? 1 : 0;
|
||||
return ((f = funcs)) ? 1 : 0;
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue