X-Git-Url: http://git.sameswireless.fr/l2tpns.git/blobdiff_plain/3eaa7ad4067f8aad7114dcb49d8e74668aee6912..620249a2af0b84d3d7eda1b74fb4c14266962c88:/autothrottle.c

diff --git a/autothrottle.c b/autothrottle.c
index 4770ae8..051b9a8 100644
--- a/autothrottle.c
+++ b/autothrottle.c
@@ -4,7 +4,16 @@
 
 /* set up throttling based on RADIUS reply */
 
-char const *cvs_id = "$Id: autothrottle.c,v 1.13 2004-11-30 07:14:45 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.15 2005-10-11 07:59:09 bodea Exp $";
 
 int plugin_api_version = PLUGIN_API_VERSION;
 struct pluginfuncs *p;
@@ -69,36 +78,44 @@ int plugin_radius_response(struct param_radius_response *data)
 
 	if (!strcmp(data->key, "throttle"))
 	{
-		if (!strcmp(data->value, "yes"))
+		char *e;
+		int rate;
+
+		if ((rate = strtol(data->value, &e, 10)) < 0 || *e)
 		{
-		    	unsigned long *rate = p->getconfig("throttle_speed", UNSIGNED_LONG);
-			if (rate)
+			rate = -1;
+			if (!strcmp(data->value, "yes"))
 			{
-				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;
+				unsigned long *ts = p->getconfig("throttle_speed", UNSIGNED_LONG);
+				if (ts)
+					rate = *ts;
 			}
-			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"))
+				rate = 0;
 		}
-		else if (!strcmp(data->value, "no"))
-		{
+
+		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 = 0;
-		}
+		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;
+}
+
 int plugin_init(struct pluginfuncs *funcs)
 {
 	return ((p = funcs)) ? 1 : 0;