Merge branch 'collectd-5.7'
[collectd.git] / src / match_value.c
1 /**
2  * collectd - src/match_value.c
3  * Copyright (C) 2008       Florian 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 Forster <octo at collectd.org>
25  **/
26
27 /*
28  * This module allows to filter and rewrite value lists based on
29  * Perl-compatible regular expressions.
30  */
31
32 #include "collectd.h"
33
34 #include "common.h"
35 #include "filter_chain.h"
36 #include "utils_cache.h"
37
38 #define SATISFY_ALL 0
39 #define SATISFY_ANY 1
40
41 /*
42  * private data types
43  */
44 struct mv_match_s;
45 typedef struct mv_match_s mv_match_t;
46 struct mv_match_s {
47   gauge_t min;
48   gauge_t max;
49   int invert;
50   int satisfy;
51
52   char **data_sources;
53   size_t data_sources_num;
54 };
55
56 /*
57  * internal helper functions
58  */
59 static void mv_free_match(mv_match_t *m) /* {{{ */
60 {
61   if (m == NULL)
62     return;
63
64   if (m->data_sources != NULL) {
65     for (size_t i = 0; i < m->data_sources_num; ++i)
66       free(m->data_sources[i]);
67     free(m->data_sources);
68   }
69
70   free(m);
71 } /* }}} void mv_free_match */
72
73 static int mv_config_add_satisfy(mv_match_t *m, /* {{{ */
74                                  oconfig_item_t *ci) {
75   if ((ci->values_num != 1) || (ci->values[0].type != OCONFIG_TYPE_STRING)) {
76     ERROR("`value' match: `%s' needs exactly one string argument.", ci->key);
77     return -1;
78   }
79
80   if (strcasecmp("All", ci->values[0].value.string) == 0)
81     m->satisfy = SATISFY_ALL;
82   else if (strcasecmp("Any", ci->values[0].value.string) == 0)
83     m->satisfy = SATISFY_ANY;
84   else {
85     ERROR("`value' match: Passing `%s' to the `%s' option is invalid. "
86           "The argument must either be `All' or `Any'.",
87           ci->values[0].value.string, ci->key);
88     return -1;
89   }
90
91   return 0;
92 } /* }}} int mv_config_add_satisfy */
93
94 static int mv_config_add_data_source(mv_match_t *m, /* {{{ */
95                                      oconfig_item_t *ci) {
96   size_t new_data_sources_num;
97   char **temp;
98
99   /* Check number of arbuments. */
100   if (ci->values_num < 1) {
101     ERROR("`value' match: `%s' needs at least one argument.", ci->key);
102     return -1;
103   }
104
105   /* Check type of arguments */
106   for (int i = 0; i < ci->values_num; i++) {
107     if (ci->values[i].type == OCONFIG_TYPE_STRING)
108       continue;
109
110     ERROR("`value' match: `%s' accepts only string arguments "
111           "(argument %i is a %s).",
112           ci->key, i + 1,
113           (ci->values[i].type == OCONFIG_TYPE_BOOLEAN) ? "truth value"
114                                                        : "number");
115     return -1;
116   }
117
118   /* Allocate space for the char pointers */
119   new_data_sources_num = m->data_sources_num + ((size_t)ci->values_num);
120   temp = realloc(m->data_sources, new_data_sources_num * sizeof(char *));
121   if (temp == NULL) {
122     ERROR("`value' match: realloc failed.");
123     return -1;
124   }
125   m->data_sources = temp;
126
127   /* Copy the strings, allocating memory as needed. */
128   for (int i = 0; i < ci->values_num; i++) {
129     /* If we get here, there better be memory for us to write to. */
130     assert(m->data_sources_num < new_data_sources_num);
131
132     size_t j = m->data_sources_num;
133     m->data_sources[j] = sstrdup(ci->values[i].value.string);
134     if (m->data_sources[j] == NULL) {
135       ERROR("`value' match: sstrdup failed.");
136       continue;
137     }
138     m->data_sources_num++;
139   }
140
141   return 0;
142 } /* }}} int mv_config_add_data_source */
143
144 static int mv_config_add_gauge(gauge_t *ret_value, /* {{{ */
145                                oconfig_item_t *ci) {
146
147   if ((ci->values_num != 1) || (ci->values[0].type != OCONFIG_TYPE_NUMBER)) {
148     ERROR("`value' match: `%s' needs exactly one numeric argument.", ci->key);
149     return -1;
150   }
151
152   *ret_value = ci->values[0].value.number;
153
154   return 0;
155 } /* }}} int mv_config_add_gauge */
156
157 static int mv_config_add_boolean(int *ret_value, /* {{{ */
158                                  oconfig_item_t *ci) {
159
160   if ((ci->values_num != 1) || (ci->values[0].type != OCONFIG_TYPE_BOOLEAN)) {
161     ERROR("`value' match: `%s' needs exactly one boolean argument.", ci->key);
162     return -1;
163   }
164
165   if (ci->values[0].value.boolean)
166     *ret_value = 1;
167   else
168     *ret_value = 0;
169
170   return 0;
171 } /* }}} int mv_config_add_boolean */
172
173 static int mv_create(const oconfig_item_t *ci, void **user_data) /* {{{ */
174 {
175   mv_match_t *m;
176   int status;
177
178   m = calloc(1, sizeof(*m));
179   if (m == NULL) {
180     ERROR("mv_create: calloc failed.");
181     return -ENOMEM;
182   }
183
184   m->min = NAN;
185   m->max = NAN;
186   m->invert = 0;
187   m->satisfy = SATISFY_ALL;
188   m->data_sources = NULL;
189   m->data_sources_num = 0;
190
191   status = 0;
192   for (int i = 0; i < ci->children_num; i++) {
193     oconfig_item_t *child = ci->children + i;
194
195     if (strcasecmp("Min", child->key) == 0)
196       status = mv_config_add_gauge(&m->min, child);
197     else if (strcasecmp("Max", child->key) == 0)
198       status = mv_config_add_gauge(&m->max, child);
199     else if (strcasecmp("Invert", child->key) == 0)
200       status = mv_config_add_boolean(&m->invert, child);
201     else if (strcasecmp("Satisfy", child->key) == 0)
202       status = mv_config_add_satisfy(m, child);
203     else if (strcasecmp("DataSource", child->key) == 0)
204       status = mv_config_add_data_source(m, child);
205     else {
206       ERROR("`value' match: The `%s' configuration option is not "
207             "understood and will be ignored.",
208             child->key);
209       status = 0;
210     }
211
212     if (status != 0)
213       break;
214   }
215
216   /* Additional sanity-checking */
217   while (status == 0) {
218     if (isnan(m->min) && isnan(m->max)) {
219       ERROR("`value' match: Neither minimum nor maximum are defined. "
220             "This match will be ignored.");
221       status = -1;
222     }
223
224     break;
225   }
226
227   if (status != 0) {
228     mv_free_match(m);
229     return status;
230   }
231
232   *user_data = m;
233   return 0;
234 } /* }}} int mv_create */
235
236 static int mv_destroy(void **user_data) /* {{{ */
237 {
238   if ((user_data != NULL) && (*user_data != NULL))
239     mv_free_match(*user_data);
240   return 0;
241 } /* }}} int mv_destroy */
242
243 static int mv_match(const data_set_t *ds, const value_list_t *vl, /* {{{ */
244                     notification_meta_t __attribute__((unused)) * *meta,
245                     void **user_data) {
246   mv_match_t *m;
247   gauge_t *values;
248   int status;
249
250   if ((user_data == NULL) || (*user_data == NULL))
251     return -1;
252
253   m = *user_data;
254
255   values = uc_get_rate(ds, vl);
256   if (values == NULL) {
257     ERROR("`value' match: Retrieving the current rate from the cache "
258           "failed.");
259     return -1;
260   }
261
262   status = FC_MATCH_NO_MATCH;
263
264   for (size_t i = 0; i < ds->ds_num; i++) {
265     int value_matches = 0;
266
267     /* Check if this data source is relevant. */
268     if (m->data_sources != NULL) {
269       size_t j;
270
271       for (j = 0; j < m->data_sources_num; j++)
272         if (strcasecmp(ds->ds[i].name, m->data_sources[j]) == 0)
273           break;
274
275       /* No match, ignore this data source. */
276       if (j >= m->data_sources_num)
277         continue;
278     }
279
280     DEBUG("`value' match: current = %g; min = %g; max = %g; invert = %s;",
281           values[i], m->min, m->max, m->invert ? "true" : "false");
282
283     if ((!isnan(m->min) && (values[i] < m->min)) ||
284         (!isnan(m->max) && (values[i] > m->max)))
285       value_matches = 0;
286     else
287       value_matches = 1;
288
289     if (m->invert) {
290       if (value_matches)
291         value_matches = 0;
292       else
293         value_matches = 1;
294     }
295
296     if (value_matches != 0) {
297       status = FC_MATCH_MATCHES;
298       if (m->satisfy == SATISFY_ANY)
299         break;
300     } else {
301       status = FC_MATCH_NO_MATCH;
302       if (m->satisfy == SATISFY_ALL)
303         break;
304     }
305   } /* for (i = 0; i < ds->ds_num; i++) */
306
307   free(values);
308   return status;
309 } /* }}} int mv_match */
310
311 void module_register(void) {
312   match_proc_t mproc = {0};
313
314   mproc.create = mv_create;
315   mproc.destroy = mv_destroy;
316   mproc.match = mv_match;
317   fc_register_match("value", mproc);
318 } /* module_register */