src/utils_cmd_putnotif.c

   1 /**
   2  * collectd - src/utils_cms_putnotif.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  * Author:
  19  *   Florian octo Forster <octo at verplant.org>
  20  **/
  21
  22 #include "collectd.h"
  23 #include "common.h"
  24 #include "plugin.h"
  25
  26 #include "utils_parse_option.h"
  27
  28 #define print_to_socket(fh, ...) \
  29   if (fprintf (fh, __VA_ARGS__) < 0) { \
  30     char errbuf[1024]; \
  31     WARNING ("handle_putnotif: failed to write to socket #%i: %s", \
  32         fileno (fh), sstrerror (errno, errbuf, sizeof (errbuf))); \
  33     return -1; \
  34   }
  35
  36 static int set_option_severity (notification_t *n, const char *value)
  37 {
  38   if (strcasecmp (value, "Failure") == 0)
  39     n->severity = NOTIF_FAILURE;
  40   else if (strcasecmp (value, "Warning") == 0)
  41     n->severity = NOTIF_WARNING;
  42   else if (strcasecmp (value, "Okay") == 0)
  43     n->severity = NOTIF_OKAY;
  44   else
  45     return (-1);
  46
  47   return (0);
  48 } /* int set_option_severity */
  49
  50 static int set_option_time (notification_t *n, const char *value)
  51 {
  52   time_t tmp;
  53
  54   tmp = (time_t) atoi (value);
  55   if (tmp <= 0)
  56     return (-1);
  57
  58   n->time = tmp;
  59
  60   return (0);
  61 } /* int set_option_time */
  62
  63 static int set_option (notification_t *n, const char *option, const char *value)
  64 {
  65   if ((n == NULL) || (option == NULL) || (value == NULL))
  66     return (-1);
  67
  68   DEBUG ("utils_cmd_putnotif: set_option (option = %s, value = %s);",
  69       option, value);
  70
  71   /* Add a meta option in the form: <type>:<key> */
  72   if (option[0] != '\0' && option[1] == ':') {
  73     /* Refuse empty key */
  74     if (option[2] == '\0')
  75       return (1);
  76
  77     if (option[0] == 's')
  78       return plugin_notification_meta_add_string (n, option + 2, value);
  79     else
  80       return (1);
  81   }
  82
  83   if (strcasecmp ("severity", option) == 0)
  84     return (set_option_severity (n, value));
  85   else if (strcasecmp ("time", option) == 0)
  86     return (set_option_time (n, value));
  87   else if (strcasecmp ("message", option) == 0)
  88     sstrncpy (n->message, value, sizeof (n->message));
  89   else if (strcasecmp ("host", option) == 0)
  90     sstrncpy (n->host, value, sizeof (n->host));
  91   else if (strcasecmp ("plugin", option) == 0)
  92     sstrncpy (n->plugin, value, sizeof (n->plugin));
  93   else if (strcasecmp ("plugin_instance", option) == 0)
  94     sstrncpy (n->plugin_instance, value, sizeof (n->plugin_instance));
  95   else if (strcasecmp ("type", option) == 0)
  96     sstrncpy (n->type, value, sizeof (n->type));
  97   else if (strcasecmp ("type_instance", option) == 0)
  98     sstrncpy (n->type_instance, value, sizeof (n->type_instance));
  99   else
 100     return (1);
 101
 102   return (0);
 103 } /* int set_option */
 104
 105 int handle_putnotif (FILE *fh, char *buffer)
 106 {
 107   char *command;
 108   notification_t n;
 109   int status;
 110
 111   if ((fh == NULL) || (buffer == NULL))
 112     return (-1);
 113
 114   DEBUG ("utils_cmd_putnotif: handle_putnotif (fh = %p, buffer = %s);",
 115       (void *) fh, buffer);
 116
 117   command = NULL;
 118   status = parse_string (&buffer, &command);
 119   if (status != 0)
 120   {
 121     print_to_socket (fh, "-1 Cannot parse command.\n");
 122     return (-1);
 123   }
 124   assert (command != NULL);
 125
 126   if (strcasecmp ("PUTNOTIF", command) != 0)
 127   {
 128     print_to_socket (fh, "-1 Unexpected command: `%s'.\n", command);
 129     return (-1);
 130   }
 131
 132   memset (&n, '\0', sizeof (n));
 133
 134   status = 0;
 135   while (*buffer != 0)
 136   {
 137     char *key;
 138     char *value;
 139
 140     status = parse_option (&buffer, &key, &value);
 141     if (status != 0)
 142     {
 143       print_to_socket (fh, "-1 Malformed option.\n");
 144       break;
 145     }
 146
 147     status = set_option (&n, key, value);
 148     if (status != 0)
 149     {
 150       print_to_socket (fh, "-1 Error parsing option `%s'\n", key);
 151       break;
 152     }
 153   } /* for (i) */
 154
 155   /* Check for required fields and complain if anything is missing. */
 156   if ((status == 0) && (n.severity == 0))
 157   {
 158     print_to_socket (fh, "-1 Option `severity' missing.\n");
 159     status = -1;
 160   }
 161   if ((status == 0) && (n.time == 0))
 162   {
 163     print_to_socket (fh, "-1 Option `time' missing.\n");
 164     status = -1;
 165   }
 166   if ((status == 0) && (strlen (n.message) == 0))
 167   {
 168     print_to_socket (fh, "-1 No message or message of length 0 given.\n");
 169     status = -1;
 170   }
 171
 172   /* If status is still zero the notification is fine and we can finally
 173    * dispatch it. */
 174   if (status == 0)
 175   {
 176     plugin_dispatch_notification (&n);
 177     print_to_socket (fh, "0 Success\n");
 178   }
 179
 180   return (0);
 181 } /* int handle_putnotif */
 182
 183 /* vim: set shiftwidth=2 softtabstop=2 tabstop=8 : */