2 * collectd - src/utils_match.c
3 * Copyright (C) 2008 Florian octo Forster
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.
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.
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
20 * Florian octo Forster <octo at verplant.org>
27 #include "utils_match.h"
31 #define UTILS_MATCH_FLAGS_FREE_USER_DATA 0x01
38 int (*callback) (const char *str, char * const *matches, size_t matches_num,
46 static char *match_substr (const char *str, int begin, int end)
51 if ((begin < 0) || (end < 0) || (begin >= end))
53 if (end > (strlen (str) + 1))
55 ERROR ("utils_match: match_substr: `end' points after end of string.");
59 ret_len = end - begin;
60 ret = (char *) malloc (sizeof (char) * (ret_len + 1));
63 ERROR ("utils_match: match_substr: malloc failed.");
67 sstrncpy (ret, str + begin, ret_len + 1);
69 } /* char *match_substr */
71 static int default_callback (const char *str,
72 char * const *matches, size_t matches_num, void *user_data)
74 cu_match_value_t *data = (cu_match_value_t *) user_data;
76 if (data->ds_type & UTILS_MATCH_DS_TYPE_GAUGE)
84 value = strtod (matches[1], &endptr);
85 if (matches[1] == endptr)
88 if ((data->values_num == 0)
89 || (data->ds_type & UTILS_MATCH_CF_GAUGE_LAST))
91 data->value.gauge = value;
93 else if (data->ds_type & UTILS_MATCH_CF_GAUGE_AVERAGE)
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));
99 else if (data->ds_type & UTILS_MATCH_CF_GAUGE_MIN)
101 if (data->value.gauge > value)
102 data->value.gauge = value;
104 else if (data->ds_type & UTILS_MATCH_CF_GAUGE_MAX)
106 if (data->value.gauge < value)
107 data->value.gauge = value;
111 ERROR ("utils_match: default_callback: obj->ds_type is invalid!");
117 else if (data->ds_type & UTILS_MATCH_DS_TYPE_COUNTER)
122 if (data->ds_type & UTILS_MATCH_CF_COUNTER_INC)
124 data->value.counter++;
132 value = strtoll (matches[1], &endptr, 0);
133 if (matches[1] == endptr)
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;
142 ERROR ("utils_match: default_callback: obj->ds_type is invalid!");
150 ERROR ("utils_match: default_callback: obj->ds_type is invalid!");
155 } /* int default_callback */
160 cu_match_t *match_create_callback (const char *regex,
161 int (*callback) (const char *str,
162 char * const *matches, size_t matches_num, void *user_data),
168 DEBUG ("utils_match: match_create_callback: regex = %s", regex);
170 obj = (cu_match_t *) malloc (sizeof (cu_match_t));
173 memset (obj, '\0', sizeof (cu_match_t));
175 status = regcomp (&obj->regex, regex, REG_EXTENDED);
178 ERROR ("Compiling the regular expression \"%s\" failed.", regex);
183 obj->callback = callback;
184 obj->user_data = user_data;
187 } /* cu_match_t *match_create_callback */
189 cu_match_t *match_create_simple (const char *regex, int match_ds_type)
191 cu_match_value_t *user_data;
194 user_data = (cu_match_value_t *) malloc (sizeof (cu_match_value_t));
195 if (user_data == NULL)
197 memset (user_data, '\0', sizeof (cu_match_value_t));
198 user_data->ds_type = match_ds_type;
200 obj = match_create_callback (regex, default_callback, user_data);
207 obj->flags |= UTILS_MATCH_FLAGS_FREE_USER_DATA;
210 } /* cu_match_t *match_create_simple */
212 void match_destroy (cu_match_t *obj)
217 if (obj->flags & UTILS_MATCH_FLAGS_FREE_USER_DATA)
219 sfree (obj->user_data);
223 } /* void match_destroy */
225 int match_apply (cu_match_t *obj, const char *str)
228 regmatch_t re_match[32];
233 if ((obj == NULL) || (str == NULL))
236 status = regexec (&obj->regex, str,
237 STATIC_ARRAY_SIZE (re_match), re_match,
240 /* Regex did not match */
244 memset (matches, '\0', sizeof (matches));
245 for (matches_num = 0; matches_num < STATIC_ARRAY_SIZE (matches); matches_num++)
247 if ((re_match[matches_num].rm_so < 0)
248 || (re_match[matches_num].rm_eo < 0))
251 matches[matches_num] = match_substr (str,
252 re_match[matches_num].rm_so, re_match[matches_num].rm_eo);
253 if (matches[matches_num] == NULL)
262 ERROR ("utils_match: match_apply: match_substr failed.");
266 status = obj->callback (str, matches, matches_num, obj->user_data);
269 ERROR ("utils_match: match_apply: callback failed.");
273 for (i = 0; i < matches_num; i++)
279 } /* int match_apply */
281 void *match_get_user_data (cu_match_t *obj)
285 return (obj->user_data);
286 } /* void *match_get_user_data */
288 /* vim: set sw=2 sts=2 ts=8 : */