1 /**
2 * collectd - src/utils_match.c
3 * Copyright (C) 2008 Florian octo Forster
4 *
5 * This program is free software; you can redistribute it and/or modify it
6 * under the terms of the GNU General Public License as published by the
7 * Free Software Foundation; either version 2 of the License, or (at your
8 * option) any later version.
9 *
10 * This program is distributed in the hope that it will be useful, but
11 * WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 * General Public License for more details.
14 *
15 * You should have received a copy of the GNU General Public License along
16 * with this program; if not, write to the Free Software Foundation, Inc.,
17 * 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
18 *
19 * Authors:
20 * Florian octo Forster <octo at verplant.org>
21 **/
23 #include "collectd.h"
24 #include "common.h"
25 #include "plugin.h"
27 #include "utils_match.h"
29 #include <regex.h>
31 #define UTILS_MATCH_FLAGS_FREE_USER_DATA 0x01
33 struct cu_match_s
34 {
35 regex_t regex;
36 int flags;
38 int (*callback) (const char *str, char * const *matches, size_t matches_num,
39 void *user_data);
40 void *user_data;
41 };
43 /*
44 * Private functions
45 */
46 static char *match_substr (const char *str, int begin, int end)
47 {
48 char *ret;
49 size_t ret_len;
51 if ((begin < 0) || (end < 0) || (begin >= end))
52 return (NULL);
53 if ((size_t) end > (strlen (str) + 1))
54 {
55 ERROR ("utils_match: match_substr: `end' points after end of string.");
56 return (NULL);
57 }
59 ret_len = end - begin;
60 ret = (char *) malloc (sizeof (char) * (ret_len + 1));
61 if (ret == NULL)
62 {
63 ERROR ("utils_match: match_substr: malloc failed.");
64 return (NULL);
65 }
67 sstrncpy (ret, str + begin, ret_len + 1);
68 return (ret);
69 } /* char *match_substr */
71 static int default_callback (const char __attribute__((unused)) *str,
72 char * const *matches, size_t matches_num, void *user_data)
73 {
74 cu_match_value_t *data = (cu_match_value_t *) user_data;
76 if (data->ds_type & UTILS_MATCH_DS_TYPE_GAUGE)
77 {
78 gauge_t value;
79 char *endptr = NULL;
81 if (matches_num < 2)
82 return (-1);
84 value = strtod (matches[1], &endptr);
85 if (matches[1] == endptr)
86 return (-1);
88 if ((data->values_num == 0)
89 || (data->ds_type & UTILS_MATCH_CF_GAUGE_LAST))
90 {
91 data->value.gauge = value;
92 }
93 else if (data->ds_type & UTILS_MATCH_CF_GAUGE_AVERAGE)
94 {
95 double f = ((double) data->values_num)
96 / ((double) (data->values_num + 1));
97 data->value.gauge = (data->value.gauge * f) + (value * (1.0 - f));
98 }
99 else if (data->ds_type & UTILS_MATCH_CF_GAUGE_MIN)
100 {
101 if (data->value.gauge > value)
102 data->value.gauge = value;
103 }
104 else if (data->ds_type & UTILS_MATCH_CF_GAUGE_MAX)
105 {
106 if (data->value.gauge < value)
107 data->value.gauge = value;
108 }
109 else
110 {
111 ERROR ("utils_match: default_callback: obj->ds_type is invalid!");
112 return (-1);
113 }
115 data->values_num++;
116 }
117 else if (data->ds_type & UTILS_MATCH_DS_TYPE_COUNTER)
118 {
119 counter_t value;
120 char *endptr = NULL;
122 if (data->ds_type & UTILS_MATCH_CF_COUNTER_INC)
123 {
124 data->value.counter++;
125 data->values_num++;
126 return (0);
127 }
129 if (matches_num < 2)
130 return (-1);
132 value = strtoll (matches[1], &endptr, 0);
133 if (matches[1] == endptr)
134 return (-1);
136 if (data->ds_type & UTILS_MATCH_CF_COUNTER_SET)
137 data->value.counter = value;
138 else if (data->ds_type & UTILS_MATCH_CF_COUNTER_ADD)
139 data->value.counter += value;
140 else
141 {
142 ERROR ("utils_match: default_callback: obj->ds_type is invalid!");
143 return (-1);
144 }
146 data->values_num++;
147 }
148 else if (data->ds_type & UTILS_MATCH_DS_TYPE_DERIVE)
149 {
150 derive_t value;
151 char *endptr = NULL;
153 if (data->ds_type & UTILS_MATCH_CF_DERIVE_INC)
154 {
155 data->value.counter++;
156 data->values_num++;
157 return (0);
158 }
160 if (matches_num < 2)
161 return (-1);
163 value = strtoll (matches[1], &endptr, 0);
164 if (matches[1] == endptr)
165 return (-1);
167 if (data->ds_type & UTILS_MATCH_CF_DERIVE_SET)
168 data->value.derive = value;
169 else if (data->ds_type & UTILS_MATCH_CF_DERIVE_ADD)
170 data->value.derive += value;
171 else
172 {
173 ERROR ("utils_match: default_callback: obj->ds_type is invalid!");
174 return (-1);
175 }
177 data->values_num++;
178 }
179 else if (data->ds_type & UTILS_MATCH_DS_TYPE_ABSOLUTE)
180 {
181 absolute_t value;
182 char *endptr = NULL;
184 if (matches_num < 2)
185 return (-1);
187 value = strtoll (matches[1], &endptr, 0);
188 if (matches[1] == endptr)
189 return (-1);
191 if (data->ds_type & UTILS_MATCH_CF_ABSOLUTE_SET)
192 data->value.absolute = value;
193 else
194 {
195 ERROR ("utils_match: default_callback: obj->ds_type is invalid!");
196 return (-1);
197 }
199 data->values_num++;
200 }
201 else
202 {
203 ERROR ("utils_match: default_callback: obj->ds_type is invalid!");
204 return (-1);
205 }
207 return (0);
208 } /* int default_callback */
210 /*
211 * Public functions
212 */
213 cu_match_t *match_create_callback (const char *regex,
214 int (*callback) (const char *str,
215 char * const *matches, size_t matches_num, void *user_data),
216 void *user_data)
217 {
218 cu_match_t *obj;
219 int status;
221 DEBUG ("utils_match: match_create_callback: regex = %s", regex);
223 obj = (cu_match_t *) malloc (sizeof (cu_match_t));
224 if (obj == NULL)
225 return (NULL);
226 memset (obj, '\0', sizeof (cu_match_t));
228 status = regcomp (&obj->regex, regex, REG_EXTENDED);
229 if (status != 0)
230 {
231 ERROR ("Compiling the regular expression \"%s\" failed.", regex);
232 sfree (obj);
233 return (NULL);
234 }
236 obj->callback = callback;
237 obj->user_data = user_data;
239 return (obj);
240 } /* cu_match_t *match_create_callback */
242 cu_match_t *match_create_simple (const char *regex, int match_ds_type)
243 {
244 cu_match_value_t *user_data;
245 cu_match_t *obj;
247 user_data = (cu_match_value_t *) malloc (sizeof (cu_match_value_t));
248 if (user_data == NULL)
249 return (NULL);
250 memset (user_data, '\0', sizeof (cu_match_value_t));
251 user_data->ds_type = match_ds_type;
253 obj = match_create_callback (regex, default_callback, user_data);
254 if (obj == NULL)
255 {
256 sfree (user_data);
257 return (NULL);
258 }
260 obj->flags |= UTILS_MATCH_FLAGS_FREE_USER_DATA;
262 return (obj);
263 } /* cu_match_t *match_create_simple */
265 void match_destroy (cu_match_t *obj)
266 {
267 if (obj == NULL)
268 return;
270 if (obj->flags & UTILS_MATCH_FLAGS_FREE_USER_DATA)
271 {
272 sfree (obj->user_data);
273 }
275 sfree (obj);
276 } /* void match_destroy */
278 int match_apply (cu_match_t *obj, const char *str)
279 {
280 int status;
281 regmatch_t re_match[32];
282 char *matches[32];
283 size_t matches_num;
284 size_t i;
286 if ((obj == NULL) || (str == NULL))
287 return (-1);
289 status = regexec (&obj->regex, str,
290 STATIC_ARRAY_SIZE (re_match), re_match,
291 /* eflags = */ 0);
293 /* Regex did not match */
294 if (status != 0)
295 return (0);
297 memset (matches, '\0', sizeof (matches));
298 for (matches_num = 0; matches_num < STATIC_ARRAY_SIZE (matches); matches_num++)
299 {
300 if ((re_match[matches_num].rm_so < 0)
301 || (re_match[matches_num].rm_eo < 0))
302 break;
304 matches[matches_num] = match_substr (str,
305 re_match[matches_num].rm_so, re_match[matches_num].rm_eo);
306 if (matches[matches_num] == NULL)
307 {
308 status = -1;
309 break;
310 }
311 }
313 if (status != 0)
314 {
315 ERROR ("utils_match: match_apply: match_substr failed.");
316 }
317 else
318 {
319 status = obj->callback (str, matches, matches_num, obj->user_data);
320 if (status != 0)
321 {
322 ERROR ("utils_match: match_apply: callback failed.");
323 }
324 }
326 for (i = 0; i < matches_num; i++)
327 {
328 sfree (matches[i]);
329 }
331 return (status);
332 } /* int match_apply */
334 void *match_get_user_data (cu_match_t *obj)
335 {
336 if (obj == NULL)
337 return (NULL);
338 return (obj->user_data);
339 } /* void *match_get_user_data */
341 /* vim: set sw=2 sts=2 ts=8 : */