#include <string.h>
-#include <malloc.h>
-#include <stdlib.h>
-#include <sys/wait.h>
-#include <sys/types.h>
#include "l2tpns.h"
#include "plugin.h"
-#include "control.h"
-int __plugin_api_version = 1;
+/* 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.14 2005/05/13 05:28:16 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, 0, 0, 0, " Throttling user\n");
- data->s->throttle = 1;
+ 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;
}
- else if (strcmp(data->value, "no") == 0)
+
+ if (type == 'i')
{
- p->log(3, 0, 0, 0, " Not throttling user\n");
- data->s->throttle = 0;
+ 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);
}
}
+
+ 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 PLUGIN_RET_OK;
}
{
return ((p = funcs)) ? 1 : 0;
}
-
-void plugin_done()
-{
-}
-