1 /**
2 * collectd - src/tail.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; only version 2 of the License is applicable.
8 *
9 * This program is distributed in the hope that it will be useful, but
10 * WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 * General Public License for more details.
13 *
14 * You should have received a copy of the GNU General Public License along
15 * with this program; if not, write to the Free Software Foundation, Inc.,
16 * 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
17 *
18 * Authors:
19 * Florian octo Forster <octo at verplant.org>
20 **/
22 #include "collectd.h"
23 #include "common.h"
24 #include "plugin.h"
25 #include "utils_tail_match.h"
27 /*
28 * <Plugin tail>
29 * <File "/var/log/exim4/mainlog">
30 * Instance "exim"
31 * <Match>
32 * Regex "S=([1-9][0-9]*)"
33 * ExcludeRegex "U=root.*S="
34 * DSType "CounterAdd"
35 * Type "ipt_bytes"
36 * Instance "total"
37 * </Match>
38 * </File>
39 * </Plugin>
40 */
42 struct ctail_config_match_s
43 {
44 char *regex;
45 char *excluderegex;
46 int flags;
47 char *type;
48 char *type_instance;
49 };
50 typedef struct ctail_config_match_s ctail_config_match_t;
52 cu_tail_match_t **tail_match_list = NULL;
53 size_t tail_match_list_num = 0;
55 static int ctail_config_add_match_dstype (ctail_config_match_t *cm,
56 oconfig_item_t *ci)
57 {
58 if ((ci->values_num != 1) || (ci->values[0].type != OCONFIG_TYPE_STRING))
59 {
60 WARNING ("tail plugin: `DSType' needs exactly one string argument.");
61 return (-1);
62 }
64 if (strncasecmp ("Gauge", ci->values[0].value.string, strlen ("Gauge")) == 0)
65 {
66 cm->flags = UTILS_MATCH_DS_TYPE_GAUGE;
67 if (strcasecmp ("GaugeAverage", ci->values[0].value.string) == 0)
68 cm->flags |= UTILS_MATCH_CF_GAUGE_AVERAGE;
69 else if (strcasecmp ("GaugeMin", ci->values[0].value.string) == 0)
70 cm->flags |= UTILS_MATCH_CF_GAUGE_MIN;
71 else if (strcasecmp ("GaugeMax", ci->values[0].value.string) == 0)
72 cm->flags |= UTILS_MATCH_CF_GAUGE_MAX;
73 else if (strcasecmp ("GaugeLast", ci->values[0].value.string) == 0)
74 cm->flags |= UTILS_MATCH_CF_GAUGE_LAST;
75 else
76 cm->flags = 0;
77 }
78 else if (strncasecmp ("Counter", ci->values[0].value.string, strlen ("Counter")) == 0)
79 {
80 cm->flags = UTILS_MATCH_DS_TYPE_COUNTER;
81 if (strcasecmp ("CounterSet", ci->values[0].value.string) == 0)
82 cm->flags |= UTILS_MATCH_CF_COUNTER_SET;
83 else if (strcasecmp ("CounterAdd", ci->values[0].value.string) == 0)
84 cm->flags |= UTILS_MATCH_CF_COUNTER_ADD;
85 else if (strcasecmp ("CounterInc", ci->values[0].value.string) == 0)
86 cm->flags |= UTILS_MATCH_CF_COUNTER_INC;
87 else
88 cm->flags = 0;
89 }
90 else if (strncasecmp ("Derive", ci->values[0].value.string, strlen ("Derive")) == 0)
91 {
92 cm->flags = UTILS_MATCH_DS_TYPE_DERIVE;
93 if (strcasecmp ("DeriveSet", ci->values[0].value.string) == 0)
94 cm->flags |= UTILS_MATCH_CF_DERIVE_SET;
95 else if (strcasecmp ("DeriveAdd", ci->values[0].value.string) == 0)
96 cm->flags |= UTILS_MATCH_CF_DERIVE_ADD;
97 else if (strcasecmp ("DeriveInc", ci->values[0].value.string) == 0)
98 cm->flags |= UTILS_MATCH_CF_DERIVE_INC;
99 else
100 cm->flags = 0;
101 }
102 else if (strncasecmp ("Absolute", ci->values[0].value.string, strlen ("Absolute")) == 0)
103 {
104 cm->flags = UTILS_MATCH_DS_TYPE_ABSOLUTE;
105 if (strcasecmp ("AbsoluteSet", ci->values[0].value.string) == 0)
106 cm->flags |= UTILS_MATCH_CF_ABSOLUTE_SET;
107 else
108 cm->flags = 0;
109 }
110 else
111 {
112 cm->flags = 0;
113 }
115 if (cm->flags == 0)
116 {
117 WARNING ("tail plugin: `%s' is not a valid argument to `DSType'.",
118 ci->values[0].value.string);
119 return (-1);
120 }
122 return (0);
123 } /* int ctail_config_add_match_dstype */
125 static int ctail_config_add_match (cu_tail_match_t *tm,
126 const char *plugin_instance, oconfig_item_t *ci)
127 {
128 ctail_config_match_t cm;
129 int status;
130 int i;
132 memset (&cm, '\0', sizeof (cm));
134 if (ci->values_num != 0)
135 {
136 WARNING ("tail plugin: Ignoring arguments for the `Match' block.");
137 }
139 status = 0;
140 for (i = 0; i < ci->children_num; i++)
141 {
142 oconfig_item_t *option = ci->children + i;
144 if (strcasecmp ("Regex", option->key) == 0)
145 status = cf_util_get_string (option, &cm.regex);
146 else if (strcasecmp ("ExcludeRegex", option->key) == 0)
147 status = cf_util_get_string (option, &cm.excluderegex);
148 else if (strcasecmp ("DSType", option->key) == 0)
149 status = ctail_config_add_match_dstype (&cm, option);
150 else if (strcasecmp ("Type", option->key) == 0)
151 status = cf_util_get_string (option, &cm.type);
152 else if (strcasecmp ("Instance", option->key) == 0)
153 status = cf_util_get_string (option, &cm.type_instance);
154 else
155 {
156 WARNING ("tail plugin: Option `%s' not allowed here.", option->key);
157 status = -1;
158 }
160 if (status != 0)
161 break;
162 } /* for (i = 0; i < ci->children_num; i++) */
164 while (status == 0)
165 {
166 if (cm.regex == NULL)
167 {
168 WARNING ("tail plugin: `Regex' missing in `Match' block.");
169 status = -1;
170 break;
171 }
173 if (cm.type == NULL)
174 {
175 WARNING ("tail plugin: `Type' missing in `Match' block.");
176 status = -1;
177 break;
178 }
180 if (cm.flags == 0)
181 {
182 WARNING ("tail plugin: `DSType' missing in `Match' block.");
183 status = -1;
184 break;
185 }
187 break;
188 } /* while (status == 0) */
190 if (status == 0)
191 {
192 status = tail_match_add_match_simple (tm, cm.regex, cm.excluderegex,
193 cm.flags, "tail", plugin_instance, cm.type, cm.type_instance);
195 if (status != 0)
196 {
197 ERROR ("tail plugin: tail_match_add_match_simple failed.");
198 }
199 }
201 sfree (cm.regex);
202 sfree (cm.excluderegex);
203 sfree (cm.type);
204 sfree (cm.type_instance);
206 return (status);
207 } /* int ctail_config_add_match */
209 static int ctail_config_add_file (oconfig_item_t *ci)
210 {
211 cu_tail_match_t *tm;
212 char *plugin_instance = NULL;
213 int num_matches = 0;
214 int status;
215 int i;
217 if ((ci->values_num != 1) || (ci->values[0].type != OCONFIG_TYPE_STRING))
218 {
219 WARNING ("tail plugin: `File' needs exactly one string argument.");
220 return (-1);
221 }
223 tm = tail_match_create (ci->values[0].value.string);
224 if (tm == NULL)
225 {
226 ERROR ("tail plugin: tail_match_create (%s) failed.",
227 ci->values[0].value.string);
228 return (-1);
229 }
231 status = 0;
232 for (i = 0; i < ci->children_num; i++)
233 {
234 oconfig_item_t *option = ci->children + i;
236 if (strcasecmp ("Match", option->key) == 0)
237 {
238 status = ctail_config_add_match (tm, plugin_instance, option);
239 if (status == 0)
240 num_matches++;
241 /* Be mild with failed matches.. */
242 status = 0;
243 }
244 else if (strcasecmp ("Instance", option->key) == 0)
245 status = cf_util_get_string (option, &plugin_instance);
246 else
247 {
248 WARNING ("tail plugin: Option `%s' not allowed here.", option->key);
249 status = -1;
250 }
252 if (status != 0)
253 break;
254 } /* for (i = 0; i < ci->children_num; i++) */
256 if (num_matches == 0)
257 {
258 ERROR ("tail plugin: No (valid) matches found for file `%s'.",
259 ci->values[0].value.string);
260 tail_match_destroy (tm);
261 return (-1);
262 }
263 else
264 {
265 cu_tail_match_t **temp;
267 temp = (cu_tail_match_t **) realloc (tail_match_list,
268 sizeof (cu_tail_match_t *) * (tail_match_list_num + 1));
269 if (temp == NULL)
270 {
271 ERROR ("tail plugin: realloc failed.");
272 tail_match_destroy (tm);
273 return (-1);
274 }
276 tail_match_list = temp;
277 tail_match_list[tail_match_list_num] = tm;
278 tail_match_list_num++;
279 }
281 return (0);
282 } /* int ctail_config_add_file */
284 static int ctail_config (oconfig_item_t *ci)
285 {
286 int i;
288 for (i = 0; i < ci->children_num; i++)
289 {
290 oconfig_item_t *option = ci->children + i;
292 if (strcasecmp ("File", option->key) == 0)
293 ctail_config_add_file (option);
294 else
295 {
296 WARNING ("tail plugin: Option `%s' not allowed here.", option->key);
297 }
298 } /* for (i = 0; i < ci->children_num; i++) */
300 return (0);
301 } /* int ctail_config */
303 static int ctail_init (void)
304 {
305 if (tail_match_list_num == 0)
306 {
307 WARNING ("tail plugin: File list is empty. Returning an error.");
308 return (-1);
309 }
311 return (0);
312 } /* int ctail_init */
314 static int ctail_read (void)
315 {
316 int success = 0;
317 size_t i;
319 for (i = 0; i < tail_match_list_num; i++)
320 {
321 int status;
323 status = tail_match_read (tail_match_list[i]);
324 if (status != 0)
325 {
326 ERROR ("tail plugin: tail_match_read[%zu] failed.", i);
327 }
328 else
329 {
330 success++;
331 }
332 }
334 if (success == 0)
335 return (-1);
336 return (0);
337 } /* int ctail_read */
339 static int ctail_shutdown (void)
340 {
341 size_t i;
343 for (i = 0; i < tail_match_list_num; i++)
344 {
345 tail_match_destroy (tail_match_list[i]);
346 tail_match_list[i] = NULL;
347 }
348 sfree (tail_match_list);
349 tail_match_list_num = 0;
351 return (0);
352 } /* int ctail_shutdown */
354 void module_register (void)
355 {
356 plugin_register_complex_config ("tail", ctail_config);
357 plugin_register_init ("tail", ctail_init);
358 plugin_register_read ("tail", ctail_read);
359 plugin_register_shutdown ("tail", ctail_shutdown);
360 } /* void module_register */
362 /* vim: set sw=2 sts=2 ts=8 : */