src/policer.c

   1 /* policer.c - Leaky bucket
   2  * Copyright 2000-2002 srvx Development Team
   3  *
   4  * This file is part of srvx.
   5  *
   6  * srvx is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation; either version 2 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with srvx; if not, write to the Free Software Foundation,
  18  * Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA.
  19  */
  20
  21 #include "common.h"
  22 #include "policer.h"
  23
  24 /* This policer uses the "leaky bucket" (GCRA) algorithm. */
  25
  26 struct policer_params {
  27     double bucket_size;
  28     double drain_rate;
  29 };
  30
  31 struct policer_params *
  32 policer_params_new(void)
  33 {
  34     struct policer_params *params = malloc(sizeof(struct policer_params));
  35     params->bucket_size = 0.0;
  36     params->drain_rate = 0.0;
  37     return params;
  38 }
  39
  40 int
  41 policer_params_set(struct policer_params *params, const char *param, const char *value)
  42 {
  43     if (!irccasecmp(param, "size")) {
  44         params->bucket_size = strtod(value, NULL);
  45     } else if (!irccasecmp(param, "drain-rate")) {
  46         params->drain_rate = strtod(value, NULL);
  47     } else {
  48         return 0;
  49     }
  50     return 1;
  51 }
  52
  53 void
  54 policer_params_delete(struct policer_params *params)
  55 {
  56     free(params);
  57 }
  58
  59 int
  60 policer_conforms(struct policer *pol, time_t reqtime, double weight)
  61 {
  62     int res;
  63     pol->level -= pol->params->drain_rate * (reqtime - pol->last_req);
  64     if (pol->level < 0.0) pol->level = 0.0;
  65     res = pol->level < pol->params->bucket_size;
  66     pol->level += weight;
  67     pol->last_req = reqtime;
  68     return res;
  69 }