X-Git-Url: http://git.sameswireless.fr/l2tpns.git/blobdiff_plain/8d5653a922ae5934b2ca74fdf0e7a8018b550626..890bffb510b300b8ed55357c6d9e99f1c6de22e9:/autothrottle.c?ds=inline

diff --git a/autothrottle.c b/autothrottle.c
index 9096691..c427c46 100644
--- a/autothrottle.c
+++ b/autothrottle.c
@@ -4,102 +4,155 @@
 
 /* set up throttling based on RADIUS reply */
 
-char const *cvs_id = "$Id: autothrottle.c,v 1.12 2004-11-30 05:49:47 bodea Exp $";
+/*
+ * 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.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)))
+	{
+	    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);
+
+	    return PLUGIN_RET_OK;
+	}
+
+	type = *data->value;
+
+	while (*sp == ' ') sp++;
+	data->value = sp;
+
+	if ((rate = strtol(data->value, &sp, 10)) < 0 || *sp)
+	{
+	    f->log(3, f->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;
+	    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)
+{
+    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)
 	{
-		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')
-		{
-			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);
-		}
+	    *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 (!strcmp(data->key, "throttle"))
+	if (data->s->throttle_out)
 	{
-		if (!strcmp(data->value, "yes"))
-		{
-		    	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"))
-		{
-			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 = 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];
 	}
 
-	return PLUGIN_RET_OK;
+	*data->packet = p;
+    }
+
+    return PLUGIN_RET_OK;
 }
 
 int plugin_init(struct pluginfuncs *funcs)
 {
-	return ((p = funcs)) ? 1 : 0;
+    return ((f = funcs)) ? 1 : 0;
 }