Add an option to set source address for a BGP session.
[l2tpns.git] / autothrottle.c
index 1a37809..2bc4c99 100644 (file)
  * lcp:interface-config#2=service-policy output N
  *
  * throttle=N
- * throttle=yes (use throttle_rate from config)
+ * throttle=yes (use throttle_speed from config)
  * throttle=no
  */
 
-char const *cvs_id = "$Id: autothrottle.c,v 1.14 2005/05/13 05:28:16 bodea Exp $";
+char const *cvs_id = "$Id: autothrottle.c,v 1.17 2006/05/18 14:40:31 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;
-
-               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);
-               }
+           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 (!strcmp(data->key, "throttle"))
+       if (type == 'i')
        {
-               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;
+           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)
+       {
+           *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;
+    return PLUGIN_RET_OK;
 }
 
 int plugin_init(struct pluginfuncs *funcs)
 {
-       return ((p = funcs)) ? 1 : 0;
+    return ((f = funcs)) ? 1 : 0;
 }