1 /**
2 * collectd - src/target_notification.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 **/
27 #include "collectd.h"
29 #include "common.h"
30 #include "filter_chain.h"
31 #include "utils_cache.h"
32 #include "utils_subst.h"
34 struct tn_data_s
35 {
36 int severity;
37 char *message;
38 };
39 typedef struct tn_data_s tn_data_t;
41 static int tn_config_add_severity (tn_data_t *data, /* {{{ */
42 const oconfig_item_t *ci)
43 {
44 if ((ci->values_num != 1)
45 || (ci->values[0].type != OCONFIG_TYPE_STRING))
46 {
47 ERROR ("Target `notification': The `%s' option requires exactly one string "
48 "argument.", ci->key);
49 return (-1);
50 }
52 if ((strcasecmp ("FAILURE", ci->values[0].value.string) == 0)
53 || (strcasecmp ("CRITICAL", ci->values[0].value.string) == 0))
54 data->severity = NOTIF_FAILURE;
55 else if ((strcasecmp ("WARNING", ci->values[0].value.string) == 0)
56 || (strcasecmp ("WARN", ci->values[0].value.string) == 0))
57 data->severity = NOTIF_WARNING;
58 else if (strcasecmp ("OKAY", ci->values[0].value.string) == 0)
59 data->severity = NOTIF_OKAY;
60 else
61 {
62 WARNING ("Target `notification': Unknown severity `%s'. "
63 "Will use `FAILURE' instead.",
64 ci->values[0].value.string);
65 data->severity = NOTIF_FAILURE;
66 }
68 return (0);
69 } /* }}} int tn_config_add_severity */
71 static int tn_config_add_string (char **dest, /* {{{ */
72 const oconfig_item_t *ci)
73 {
74 char *temp;
76 if (dest == NULL)
77 return (-EINVAL);
79 if ((ci->values_num != 1)
80 || (ci->values[0].type != OCONFIG_TYPE_STRING))
81 {
82 ERROR ("Target `notification': The `%s' option requires exactly one string "
83 "argument.", ci->key);
84 return (-1);
85 }
87 if (ci->values[0].value.string[0] == 0)
88 {
89 ERROR ("Target `notification': The `%s' option does not accept empty strings.",
90 ci->key);
91 return (-1);
92 }
94 temp = sstrdup (ci->values[0].value.string);
95 if (temp == NULL)
96 {
97 ERROR ("tn_config_add_string: sstrdup failed.");
98 return (-1);
99 }
101 free (*dest);
102 *dest = temp;
104 return (0);
105 } /* }}} int tn_config_add_string */
107 static int tn_destroy (void **user_data) /* {{{ */
108 {
109 tn_data_t *data;
111 if (user_data == NULL)
112 return (-EINVAL);
114 data = *user_data;
115 if (data == NULL)
116 return (0);
118 sfree (data->message);
119 sfree (data);
121 return (0);
122 } /* }}} int tn_destroy */
124 static int tn_create (const oconfig_item_t *ci, void **user_data) /* {{{ */
125 {
126 tn_data_t *data;
127 int status;
128 int i;
130 data = calloc (1, sizeof (*data));
131 if (data == NULL)
132 {
133 ERROR ("tn_create: calloc failed.");
134 return (-ENOMEM);
135 }
137 data->message = NULL;
138 data->severity = 0;
140 status = 0;
141 for (i = 0; i < ci->children_num; i++)
142 {
143 oconfig_item_t *child = ci->children + i;
145 if (strcasecmp ("Message", child->key) == 0)
146 status = tn_config_add_string (&data->message, child);
147 else if (strcasecmp ("Severity", child->key) == 0)
148 status = tn_config_add_severity (data, child);
149 else
150 {
151 ERROR ("Target `notification': The `%s' configuration option is not understood "
152 "and will be ignored.", child->key);
153 status = 0;
154 }
156 if (status != 0)
157 break;
158 }
160 /* Additional sanity-checking */
161 while (status == 0)
162 {
163 if ((data->severity != NOTIF_FAILURE)
164 && (data->severity != NOTIF_WARNING)
165 && (data->severity != NOTIF_OKAY))
166 {
167 DEBUG ("Target `notification': Setting "
168 "the default severity `WARNING'.");
169 data->severity = NOTIF_WARNING;
170 }
172 if (data->message == NULL)
173 {
174 ERROR ("Target `notification': No `Message' option has been specified. "
175 "Without it, the `Notification' target is useless.");
176 status = -1;
177 }
179 break;
180 }
182 if (status != 0)
183 {
184 tn_destroy ((void *) &data);
185 return (status);
186 }
188 *user_data = data;
189 return (0);
190 } /* }}} int tn_create */
192 static int tn_invoke (const data_set_t *ds, value_list_t *vl, /* {{{ */
193 notification_meta_t __attribute__((unused)) **meta, void **user_data)
194 {
195 tn_data_t *data;
196 notification_t n = { 0 };
197 char temp[NOTIF_MAX_MSG_LEN];
199 gauge_t *rates;
200 int rates_failed;
202 size_t i;
204 if ((ds == NULL) || (vl == NULL) || (user_data == NULL))
205 return (-EINVAL);
207 data = *user_data;
208 if (data == NULL)
209 {
210 ERROR ("Target `notification': Invoke: `data' is NULL.");
211 return (-EINVAL);
212 }
214 /* Initialize the structure. */
215 n.severity = data->severity;
216 n.time = cdtime ();
217 sstrncpy (n.message, data->message, sizeof (n.message));
218 sstrncpy (n.host, vl->host, sizeof (n.host));
219 sstrncpy (n.plugin, vl->plugin, sizeof (n.plugin));
220 sstrncpy (n.plugin_instance, vl->plugin_instance,
221 sizeof (n.plugin_instance));
222 sstrncpy (n.type, vl->type, sizeof (n.type));
223 sstrncpy (n.type_instance, vl->type_instance,
224 sizeof (n.type_instance));
225 n.meta = NULL;
227 #define REPLACE_FIELD(t,v) \
228 if (subst_string (temp, sizeof (temp), n.message, t, v) != NULL) \
229 sstrncpy (n.message, temp, sizeof (n.message));
230 REPLACE_FIELD ("%{host}", n.host);
231 REPLACE_FIELD ("%{plugin}", n.plugin);
232 REPLACE_FIELD ("%{plugin_instance}", n.plugin_instance);
233 REPLACE_FIELD ("%{type}", n.type);
234 REPLACE_FIELD ("%{type_instance}", n.type_instance);
236 rates_failed = 0;
237 rates = NULL;
238 for (i = 0; i < ds->ds_num; i++)
239 {
240 char template[DATA_MAX_NAME_LEN];
241 char value_str[DATA_MAX_NAME_LEN];
243 ssnprintf (template, sizeof (template), "%%{ds:%s}", ds->ds[i].name);
245 if (ds->ds[i].type != DS_TYPE_GAUGE)
246 {
247 if ((rates == NULL) && (rates_failed == 0))
248 {
249 rates = uc_get_rate (ds, vl);
250 if (rates == NULL)
251 rates_failed = 1;
252 }
253 }
255 /* If this is a gauge value, use the current value. */
256 if (ds->ds[i].type == DS_TYPE_GAUGE)
257 ssnprintf (value_str, sizeof (value_str),
258 GAUGE_FORMAT, (double) vl->values[i].gauge);
259 /* If it's a counter, try to use the current rate. This may fail, if the
260 * value has been renamed. */
261 else if (rates != NULL)
262 ssnprintf (value_str, sizeof (value_str),
263 GAUGE_FORMAT, (double) rates[i]);
264 /* Since we don't know any better, use the string `unknown'. */
265 else
266 sstrncpy (value_str, "unknown", sizeof (value_str));
268 REPLACE_FIELD (template, value_str);
269 }
270 sfree (rates);
272 plugin_dispatch_notification (&n);
274 return (FC_TARGET_CONTINUE);
275 } /* }}} int tn_invoke */
277 void module_register (void)
278 {
279 target_proc_t tproc = { 0 };
281 tproc.create = tn_create;
282 tproc.destroy = tn_destroy;
283 tproc.invoke = tn_invoke;
284 fc_register_target ("notification", tproc);
285 } /* module_register */
287 /* vim: set sw=2 sts=2 tw=78 et fdm=marker : */