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