src/utils/latency/latency_config.c

   1 /**
   2  * collectd - src/utils_latency_config.c
   3  * Copyright (C) 2013-2016   Florian octo Forster
   4  *
   5  * Permission is hereby granted, free of charge, to any person obtaining a
   6  * copy of this software and associated documentation files (the "Software"),
   7  * to deal in the Software without restriction, including without limitation
   8  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
   9  * and/or sell copies of the Software, and to permit persons to whom the
  10  * Software is furnished to do so, subject to the following conditions:
  11  *
  12  * The above copyright notice and this permission notice shall be included in
  13  * all copies or substantial portions of the Software.
  14  *
  15  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  16  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  17  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  18  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  19  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  20  * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
  21  * DEALINGS IN THE SOFTWARE.
  22  *
  23  * Authors:
  24  *   Florian octo Forster <octo at collectd.org>
  25  *   Pavel Rochnyack <pavel2000 at ngs.ru>
  26  */
  27
  28 #include "collectd.h"
  29 #include "utils/common/common.h"
  30 #include "utils/latency/latency_config.h"
  31
  32 static int latency_config_add_percentile(latency_config_t *conf,
  33                                          oconfig_item_t *ci) {
  34   double percent;
  35   int status = cf_util_get_double(ci, &percent);
  36   if (status != 0)
  37     return status;
  38
  39   if ((percent <= 0.0) || (percent >= 100)) {
  40     P_ERROR("The value for \"%s\" must be between 0 and 100, "
  41             "exclusively.",
  42             ci->key);
  43     return ERANGE;
  44   }
  45
  46   double *tmp = realloc(conf->percentile,
  47                         sizeof(*conf->percentile) * (conf->percentile_num + 1));
  48   if (tmp == NULL) {
  49     P_ERROR("realloc failed.");
  50     return ENOMEM;
  51   }
  52   conf->percentile = tmp;
  53   conf->percentile[conf->percentile_num] = percent;
  54   conf->percentile_num++;
  55
  56   return 0;
  57 } /* int latency_config_add_percentile */
  58
  59 static int latency_config_add_bucket(latency_config_t *conf,
  60                                      oconfig_item_t *ci) {
  61   if ((ci->values_num != 2) || (ci->values[0].type != OCONFIG_TYPE_NUMBER) ||
  62       (ci->values[1].type != OCONFIG_TYPE_NUMBER)) {
  63     P_ERROR("\"%s\" requires exactly two numeric arguments.", ci->key);
  64     return EINVAL;
  65   }
  66
  67   if (ci->values[1].value.number &&
  68       ci->values[1].value.number <= ci->values[0].value.number) {
  69     P_ERROR("MIN must be less than MAX in \"%s\".", ci->key);
  70     return ERANGE;
  71   }
  72
  73   if (ci->values[0].value.number < 0) {
  74     P_ERROR("MIN must be greater then or equal to zero in \"%s\".", ci->key);
  75     return ERANGE;
  76   }
  77
  78   latency_bucket_t *tmp =
  79       realloc(conf->buckets, sizeof(*conf->buckets) * (conf->buckets_num + 1));
  80   if (tmp == NULL) {
  81     P_ERROR("realloc failed.");
  82     return ENOMEM;
  83   }
  84   conf->buckets = tmp;
  85   conf->buckets[conf->buckets_num].lower_bound =
  86       DOUBLE_TO_CDTIME_T(ci->values[0].value.number);
  87   conf->buckets[conf->buckets_num].upper_bound =
  88       DOUBLE_TO_CDTIME_T(ci->values[1].value.number);
  89   conf->buckets_num++;
  90
  91   return 0;
  92 } /* int latency_config_add_bucket */
  93
  94 int latency_config(latency_config_t *conf, oconfig_item_t *ci) {
  95   int status = 0;
  96
  97   for (int i = 0; i < ci->children_num; i++) {
  98     oconfig_item_t *child = ci->children + i;
  99
 100     if (strcasecmp("Percentile", child->key) == 0)
 101       status = latency_config_add_percentile(conf, child);
 102     else if (strcasecmp("Bucket", child->key) == 0)
 103       status = latency_config_add_bucket(conf, child);
 104     else if (strcasecmp("BucketType", child->key) == 0)
 105       status = cf_util_get_string(child, &conf->bucket_type);
 106     else
 107       P_WARNING("\"%s\" is not a valid option within a \"%s\" block.",
 108                 child->key, ci->key);
 109
 110     if (status != 0)
 111       return status;
 112   }
 113
 114   if ((status == 0) && (conf->percentile_num == 0) &&
 115       (conf->buckets_num == 0)) {
 116     P_ERROR("The \"%s\" block must contain at least one "
 117             "\"Percentile\" or \"Bucket\" option.",
 118             ci->key);
 119     return EINVAL;
 120   }
 121
 122   return 0;
 123 }
 124
 125 int latency_config_copy(latency_config_t *dst, const latency_config_t src) {
 126   *dst = (latency_config_t){
 127       .percentile_num = src.percentile_num, .buckets_num = src.buckets_num,
 128   };
 129
 130   dst->percentile = calloc(dst->percentile_num, sizeof(*dst->percentile));
 131   dst->buckets = calloc(dst->buckets_num, sizeof(*dst->buckets));
 132
 133   if ((dst->percentile == NULL) || (dst->buckets == NULL)) {
 134     latency_config_free(*dst);
 135     return ENOMEM;
 136   }
 137
 138   if (src.bucket_type != NULL) {
 139     dst->bucket_type = strdup(src.bucket_type);
 140     if (dst->bucket_type == NULL) {
 141       latency_config_free(*dst);
 142       return ENOMEM;
 143     }
 144   }
 145
 146   memmove(dst->percentile, src.percentile,
 147           dst->percentile_num * sizeof(*dst->percentile));
 148   memmove(dst->buckets, src.buckets, dst->buckets_num * sizeof(*dst->buckets));
 149
 150   return 0;
 151 } /* int latency_config_copy */
 152
 153 void latency_config_free(latency_config_t conf) {
 154   sfree(conf.percentile);
 155   sfree(conf.buckets);
 156   sfree(conf.bucket_type);
 157 } /* void latency_config_free */