email plugin: stop linking with libpthread
[collectd.git] / src / tail.c
1 /**
2  * collectd - src/tail.c
3  * Copyright (C) 2008       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  **/
26
27 #include "collectd.h"
28 #include "common.h"
29 #include "plugin.h"
30 #include "utils_tail_match.h"
31
32 /*
33  *  <Plugin tail>
34  *    <File "/var/log/exim4/mainlog">
35  *      Instance "exim"
36  *      Interval 60
37  *      <Match>
38  *        Regex "S=([1-9][0-9]*)"
39  *        ExcludeRegex "U=root.*S="
40  *        DSType "CounterAdd"
41  *        Type "ipt_bytes"
42  *        Instance "total"
43  *      </Match>
44  *    </File>
45  *  </Plugin>
46  */
47
48 struct ctail_config_match_s
49 {
50   char *regex;
51   char *excluderegex;
52   int flags;
53   char *type;
54   char *type_instance;
55   cdtime_t interval;
56 };
57 typedef struct ctail_config_match_s ctail_config_match_t;
58
59 static cu_tail_match_t **tail_match_list = NULL;
60 static size_t tail_match_list_num = 0;
61 static cdtime_t tail_match_list_intervals[255];
62
63 static int ctail_config_add_match_dstype (ctail_config_match_t *cm,
64     oconfig_item_t *ci)
65 {
66   if ((ci->values_num != 1) || (ci->values[0].type != OCONFIG_TYPE_STRING))
67   {
68     WARNING ("tail plugin: `DSType' needs exactly one string argument.");
69     return (-1);
70   }
71
72   if (strncasecmp ("Gauge", ci->values[0].value.string, strlen ("Gauge")) == 0)
73   {
74     cm->flags = UTILS_MATCH_DS_TYPE_GAUGE;
75     if (strcasecmp ("GaugeAverage", ci->values[0].value.string) == 0)
76       cm->flags |= UTILS_MATCH_CF_GAUGE_AVERAGE;
77     else if (strcasecmp ("GaugeMin", ci->values[0].value.string) == 0)
78       cm->flags |= UTILS_MATCH_CF_GAUGE_MIN;
79     else if (strcasecmp ("GaugeMax", ci->values[0].value.string) == 0)
80       cm->flags |= UTILS_MATCH_CF_GAUGE_MAX;
81     else if (strcasecmp ("GaugeLast", ci->values[0].value.string) == 0)
82       cm->flags |= UTILS_MATCH_CF_GAUGE_LAST;
83     else if (strcasecmp ("GaugeInc", ci->values[0].value.string) == 0)
84       cm->flags |= UTILS_MATCH_CF_GAUGE_INC;
85     else if (strcasecmp ("GaugeAdd", ci->values[0].value.string) == 0)
86       cm->flags |= UTILS_MATCH_CF_GAUGE_ADD;
87     else
88       cm->flags = 0;
89   }
90   else if (strncasecmp ("Counter", ci->values[0].value.string, strlen ("Counter")) == 0)
91   {
92     cm->flags = UTILS_MATCH_DS_TYPE_COUNTER;
93     if (strcasecmp ("CounterSet", ci->values[0].value.string) == 0)
94       cm->flags |= UTILS_MATCH_CF_COUNTER_SET;
95     else if (strcasecmp ("CounterAdd", ci->values[0].value.string) == 0)
96       cm->flags |= UTILS_MATCH_CF_COUNTER_ADD;
97     else if (strcasecmp ("CounterInc", ci->values[0].value.string) == 0)
98       cm->flags |= UTILS_MATCH_CF_COUNTER_INC;
99     else
100       cm->flags = 0;
101   }
102   else if (strncasecmp ("Derive", ci->values[0].value.string, strlen ("Derive")) == 0)
103   {
104     cm->flags = UTILS_MATCH_DS_TYPE_DERIVE;
105     if (strcasecmp ("DeriveSet", ci->values[0].value.string) == 0)
106       cm->flags |= UTILS_MATCH_CF_DERIVE_SET;
107     else if (strcasecmp ("DeriveAdd", ci->values[0].value.string) == 0)
108       cm->flags |= UTILS_MATCH_CF_DERIVE_ADD;
109     else if (strcasecmp ("DeriveInc", ci->values[0].value.string) == 0)
110       cm->flags |= UTILS_MATCH_CF_DERIVE_INC;
111     else
112       cm->flags = 0;
113   }
114   else if (strncasecmp ("Absolute", ci->values[0].value.string, strlen ("Absolute")) == 0)
115   {
116     cm->flags = UTILS_MATCH_DS_TYPE_ABSOLUTE;
117     if (strcasecmp ("AbsoluteSet", ci->values[0].value.string) == 0)
118       cm->flags |= UTILS_MATCH_CF_ABSOLUTE_SET;
119     else
120       cm->flags = 0;
121   }
122   else
123   {
124     cm->flags = 0;
125   }
126
127   if (cm->flags == 0)
128   {
129     WARNING ("tail plugin: `%s' is not a valid argument to `DSType'.",
130         ci->values[0].value.string);
131     return (-1);
132   }
133
134   return (0);
135 } /* int ctail_config_add_match_dstype */
136
137 static int ctail_config_add_match (cu_tail_match_t *tm,
138     const char *plugin_instance, oconfig_item_t *ci, cdtime_t interval)
139 {
140   ctail_config_match_t cm;
141   int status;
142   int i;
143
144   memset (&cm, '\0', sizeof (cm));
145
146   if (ci->values_num != 0)
147   {
148     WARNING ("tail plugin: Ignoring arguments for the `Match' block.");
149   }
150
151   status = 0;
152   for (i = 0; i < ci->children_num; i++)
153   {
154     oconfig_item_t *option = ci->children + i;
155
156     if (strcasecmp ("Regex", option->key) == 0)
157       status = cf_util_get_string (option, &cm.regex);
158     else if (strcasecmp ("ExcludeRegex", option->key) == 0)
159       status = cf_util_get_string (option, &cm.excluderegex);
160     else if (strcasecmp ("DSType", option->key) == 0)
161       status = ctail_config_add_match_dstype (&cm, option);
162     else if (strcasecmp ("Type", option->key) == 0)
163       status = cf_util_get_string (option, &cm.type);
164     else if (strcasecmp ("Instance", option->key) == 0)
165       status = cf_util_get_string (option, &cm.type_instance);
166     else
167     {
168       WARNING ("tail plugin: Option `%s' not allowed here.", option->key);
169       status = -1;
170     }
171
172     if (status != 0)
173       break;
174   } /* for (i = 0; i < ci->children_num; i++) */
175
176   while (status == 0)
177   {
178     if (cm.regex == NULL)
179     {
180       WARNING ("tail plugin: `Regex' missing in `Match' block.");
181       status = -1;
182       break;
183     }
184
185     if (cm.type == NULL)
186     {
187       WARNING ("tail plugin: `Type' missing in `Match' block.");
188       status = -1;
189       break;
190     }
191
192     if (cm.flags == 0)
193     {
194       WARNING ("tail plugin: `DSType' missing in `Match' block.");
195       status = -1;
196       break;
197     }
198
199     break;
200   } /* while (status == 0) */
201
202   if (status == 0)
203   {
204     status = tail_match_add_match_simple (tm, cm.regex, cm.excluderegex,
205         cm.flags, "tail", plugin_instance, cm.type, cm.type_instance, interval);
206
207     if (status != 0)
208     {
209       ERROR ("tail plugin: tail_match_add_match_simple failed.");
210     }
211   }
212
213   sfree (cm.regex);
214   sfree (cm.excluderegex);
215   sfree (cm.type);
216   sfree (cm.type_instance);
217
218   return (status);
219 } /* int ctail_config_add_match */
220
221 static int ctail_config_add_file (oconfig_item_t *ci)
222 {
223   cu_tail_match_t *tm;
224   cdtime_t interval = 0;
225   char *plugin_instance = NULL;
226   int num_matches = 0;
227   int i;
228
229   if ((ci->values_num != 1) || (ci->values[0].type != OCONFIG_TYPE_STRING))
230   {
231     WARNING ("tail plugin: `File' needs exactly one string argument.");
232     return (-1);
233   }
234
235   tm = tail_match_create (ci->values[0].value.string);
236   if (tm == NULL)
237   {
238     ERROR ("tail plugin: tail_match_create (%s) failed.",
239         ci->values[0].value.string);
240     return (-1);
241   }
242
243   for (i = 0; i < ci->children_num; i++)
244   {
245     oconfig_item_t *option = ci->children + i;
246     int status = 0;
247
248     if (strcasecmp ("Instance", option->key) == 0)
249       status = cf_util_get_string (option, &plugin_instance);
250     else if (strcasecmp ("Interval", option->key) == 0)
251       cf_util_get_cdtime (option, &interval);
252     else if (strcasecmp ("Match", option->key) == 0)
253     {
254       status = ctail_config_add_match (tm, plugin_instance, option, interval);
255       if (status == 0)
256         num_matches++;
257       /* Be mild with failed matches.. */
258       status = 0;
259     }
260     else
261     {
262       status = -1;
263     }
264
265     if (status != 0)
266       break;
267   } /* for (i = 0; i < ci->children_num; i++) */
268
269   sfree (plugin_instance);
270
271   if (num_matches == 0)
272   {
273     ERROR ("tail plugin: No (valid) matches found for file `%s'.",
274         ci->values[0].value.string);
275     tail_match_destroy (tm);
276     return (-1);
277   }
278   else
279   {
280     cu_tail_match_t **temp;
281
282     temp = realloc (tail_match_list,
283         sizeof (cu_tail_match_t *) * (tail_match_list_num + 1));
284     if (temp == NULL)
285     {
286       ERROR ("tail plugin: realloc failed.");
287       tail_match_destroy (tm);
288       return (-1);
289     }
290
291     tail_match_list = temp;
292     tail_match_list[tail_match_list_num] = tm;
293     tail_match_list_intervals[tail_match_list_num] = interval;
294     tail_match_list_num++;
295   }
296
297   return (0);
298 } /* int ctail_config_add_file */
299
300 static int ctail_config (oconfig_item_t *ci)
301 {
302   int i;
303
304   for (i = 0; i < ci->children_num; i++)
305   {
306     oconfig_item_t *option = ci->children + i;
307
308     if (strcasecmp ("File", option->key) == 0)
309       ctail_config_add_file (option);
310     else
311     {
312       WARNING ("tail plugin: Option `%s' not allowed here.", option->key);
313     }
314   } /* for (i = 0; i < ci->children_num; i++) */
315
316   return (0);
317 } /* int ctail_config */
318
319 static int ctail_read (user_data_t *ud)
320 {
321   int status;
322
323   status = tail_match_read ((cu_tail_match_t *)ud->data);
324   if (status != 0)
325   {
326     ERROR ("tail plugin: tail_match_read failed.");
327     return (-1);
328   }
329
330   return (0);
331 } /* int ctail_read */
332
333 static int ctail_init (void)
334 {
335   char str[255];
336   user_data_t ud;
337   size_t i;
338
339   if (tail_match_list_num == 0)
340   {
341     WARNING ("tail plugin: File list is empty. Returning an error.");
342     return (-1);
343   }
344
345   memset(&ud, '\0', sizeof(ud));
346
347   for (i = 0; i < tail_match_list_num; i++)
348   {
349     ud.data = (void *)tail_match_list[i];
350     ssnprintf(str, sizeof(str), "tail-%zu", i);
351     plugin_register_complex_read (NULL, str, ctail_read, tail_match_list_intervals[i], &ud);
352   }
353
354   return (0);
355 } /* int ctail_init */
356
357 static int ctail_shutdown (void)
358 {
359   size_t i;
360
361   for (i = 0; i < tail_match_list_num; i++)
362   {
363     tail_match_destroy (tail_match_list[i]);
364     tail_match_list[i] = NULL;
365   }
366   sfree (tail_match_list);
367   tail_match_list_num = 0;
368
369   return (0);
370 } /* int ctail_shutdown */
371
372 void module_register (void)
373 {
374   plugin_register_complex_config ("tail", ctail_config);
375   plugin_register_init ("tail", ctail_init);
376   plugin_register_shutdown ("tail", ctail_shutdown);
377 } /* void module_register */
378
379 /* vim: set sw=2 sts=2 ts=8 : */