Code

collectd.spec: re-enable write_riemann on EL7, as build dep is now available in EPEL7
[collectd.git] / src / notify_nagios.c
1 /**
2  * collectd - src/notify_nagios.c
3  * Copyright (C) 2015       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  */
27 #include "collectd.h"
29 #include "common.h"
30 #include "plugin.h"
32 #define NAGIOS_OK 0
33 #define NAGIOS_WARNING 1
34 #define NAGIOS_CRITICAL 2
35 #define NAGIOS_UNKNOWN 3
37 #ifndef NAGIOS_COMMAND_FILE
38 #define NAGIOS_COMMAND_FILE "/usr/local/nagios/var/rw/nagios.cmd"
39 #endif
41 static char *nagios_command_file;
43 static int nagios_config(oconfig_item_t *ci) /* {{{ */
44 {
45   for (int i = 0; i < ci->children_num; i++) {
46     oconfig_item_t *child = ci->children + i;
48     if (strcasecmp("CommandFile", child->key) == 0)
49       cf_util_get_string(child, &nagios_command_file);
50     else
51       WARNING("notify_nagios plugin: Ignoring unknown config option \"%s\".",
52               child->key);
53   }
55   return 0;
56 } /* }}} nagios_config */
58 static int nagios_print(char const *buffer) /* {{{ */
59 {
60   char const *file = NAGIOS_COMMAND_FILE;
61   int fd;
62   int status;
63   struct flock lock = {0};
65   if (nagios_command_file != NULL)
66     file = nagios_command_file;
68   fd = open(file, O_WRONLY | O_APPEND);
69   if (fd < 0) {
70     char errbuf[1024];
71     status = errno;
72     ERROR("notify_nagios plugin: Opening \"%s\" failed: %s", file,
73           sstrerror(status, errbuf, sizeof(errbuf)));
74     return status;
75   }
77   lock.l_type = F_WRLCK;
78   lock.l_whence = SEEK_END;
80   status = fcntl(fd, F_GETLK, &lock);
81   if (status != 0) {
82     char errbuf[1024];
83     status = errno;
84     ERROR("notify_nagios plugin: Failed to acquire write lock on \"%s\": %s",
85           file, sstrerror(status, errbuf, sizeof(errbuf)));
86     close(fd);
87     return status;
88   }
90   status = (int)lseek(fd, 0, SEEK_END);
91   if (status == -1) {
92     char errbuf[1024];
93     status = errno;
94     ERROR("notify_nagios plugin: Seeking to end of \"%s\" failed: %s", file,
95           sstrerror(status, errbuf, sizeof(errbuf)));
96     close(fd);
97     return status;
98   }
100   status = (int)swrite(fd, buffer, strlen(buffer));
101   if (status != 0) {
102     char errbuf[1024];
103     status = errno;
104     ERROR("notify_nagios plugin: Writing to \"%s\" failed: %s", file,
105           sstrerror(status, errbuf, sizeof(errbuf)));
106     close(fd);
107     return status;
108   }
110   close(fd);
111   return status;
112 } /* }}} int nagios_print */
114 static int nagios_notify(const notification_t *n, /* {{{ */
115                          __attribute__((unused)) user_data_t *user_data) {
116   char svc_description[4 * DATA_MAX_NAME_LEN];
117   char buffer[4096];
118   int code;
119   int status;
121   status = format_name(svc_description, (int)sizeof(svc_description),
122                        /* host */ "", n->plugin, n->plugin_instance, n->type,
123                        n->type_instance);
124   if (status != 0) {
125     ERROR("notify_nagios plugin: Formatting service name failed.");
126     return status;
127   }
129   switch (n->severity) {
130   case NOTIF_OKAY:
131     code = NAGIOS_OK;
132     break;
133   case NOTIF_WARNING:
134     code = NAGIOS_WARNING;
135     break;
136   case NOTIF_FAILURE:
137     code = NAGIOS_CRITICAL;
138     break;
139   default:
140     code = NAGIOS_UNKNOWN;
141     break;
142   }
144   ssnprintf(buffer, sizeof(buffer),
145             "[%.0f] PROCESS_SERVICE_CHECK_RESULT;%s;%s;%d;%s\n",
146             CDTIME_T_TO_DOUBLE(n->time), n->host, &svc_description[1], code,
147             n->message);
149   return nagios_print(buffer);
150 } /* }}} int nagios_notify */
152 void module_register(void) {
153   plugin_register_complex_config("notify_nagios", nagios_config);
154   plugin_register_notification("notify_nagios", nagios_notify, NULL);
155 } /* void module_register (void) */
157 /* vim: set sw=2 sts=2 ts=8 et : */