src/utils_cmd_getthreshold.c

   1 /**
   2  * collectd - src/utils_cmd_getthreshold.c
   3  * Copyright (C) 2008,2009  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
  29 #include "common.h"
  30 #include "plugin.h"
  31
  32 #include "utils_avltree.h"
  33 #include "utils_threshold.h"
  34 #include "utils_parse_option.h" /* for `parse_string' */
  35 #include "utils_cmd_getthreshold.h"
  36
  37 #define print_to_socket(fh, ...) \
  38   if (fprintf (fh, __VA_ARGS__) < 0) { \
  39     char errbuf[1024]; \
  40     WARNING ("handle_getthreshold: failed to write to socket #%i: %s", \
  41         fileno (fh), sstrerror (errno, errbuf, sizeof (errbuf))); \
  42     return -1; \
  43   }
  44
  45 int handle_getthreshold (FILE *fh, char *buffer)
  46 {
  47   char *command;
  48   char *identifier;
  49   char *identifier_copy;
  50
  51   char *host;
  52   char *plugin;
  53   char *plugin_instance;
  54   char *type;
  55   char *type_instance;
  56
  57   threshold_t threshold;
  58
  59   int   status;
  60   size_t i;
  61
  62   if ((fh == NULL) || (buffer == NULL))
  63     return (-1);
  64
  65   DEBUG ("utils_cmd_getthreshold: handle_getthreshold (fh = %p, buffer = %s);",
  66       (void *) fh, buffer);
  67
  68   command = NULL;
  69   status = parse_string (&buffer, &command);
  70   if (status != 0)
  71   {
  72     print_to_socket (fh, "-1 Cannot parse command.\n");
  73     return (-1);
  74   }
  75   assert (command != NULL);
  76
  77   if (strcasecmp ("GETTHRESHOLD", command) != 0)
  78   {
  79     print_to_socket (fh, "-1 Unexpected command: `%s'.\n", command);
  80     return (-1);
  81   }
  82
  83   identifier = NULL;
  84   status = parse_string (&buffer, &identifier);
  85   if (status != 0)
  86   {
  87     print_to_socket (fh, "-1 Cannot parse identifier.\n");
  88     return (-1);
  89   }
  90   assert (identifier != NULL);
  91
  92   if (*buffer != 0)
  93   {
  94     print_to_socket (fh, "-1 Garbage after end of command: %s\n", buffer);
  95     return (-1);
  96   }
  97
  98   /* parse_identifier() modifies its first argument,
  99    * returning pointers into it */
 100   identifier_copy = sstrdup (identifier);
 101
 102   status = parse_identifier (identifier_copy, &host,
 103       &plugin, &plugin_instance,
 104       &type, &type_instance,
 105       /* default_host = */ NULL);
 106   if (status != 0)
 107   {
 108     DEBUG ("handle_getthreshold: Cannot parse identifier `%s'.", identifier);
 109     print_to_socket (fh, "-1 Cannot parse identifier `%s'.\n", identifier);
 110     sfree (identifier_copy);
 111     return (-1);
 112   }
 113
 114   value_list_t vl = {
 115     .values = NULL
 116   };
 117   sstrncpy (vl.host, host, sizeof (vl.host));
 118   sstrncpy (vl.plugin, plugin, sizeof (vl.plugin));
 119   if (plugin_instance != NULL)
 120     sstrncpy (vl.plugin_instance, plugin_instance, sizeof (vl.plugin_instance));
 121   sstrncpy (vl.type, type, sizeof (vl.type));
 122   if (type_instance != NULL)
 123     sstrncpy (vl.type_instance, type_instance, sizeof (vl.type_instance));
 124   sfree (identifier_copy);
 125
 126   status = ut_search_threshold (&vl, &threshold);
 127   if (status == ENOENT)
 128   {
 129     print_to_socket (fh, "-1 No threshold found for identifier %s\n",
 130         identifier);
 131     return (0);
 132   }
 133   else if (status != 0)
 134   {
 135     print_to_socket (fh, "-1 Error while looking up threshold: %i\n",
 136         status);
 137     return (-1);
 138   }
 139
 140   /* Lets count the number of lines we'll return. */
 141   i = 0;
 142   if (threshold.host[0] != 0)            i++;
 143   if (threshold.plugin[0] != 0)          i++;
 144   if (threshold.plugin_instance[0] != 0) i++;
 145   if (threshold.type[0] != 0)            i++;
 146   if (threshold.type_instance[0] != 0)   i++;
 147   if (threshold.data_source[0] != 0)     i++;
 148   if (!isnan (threshold.warning_min))    i++;
 149   if (!isnan (threshold.warning_max))    i++;
 150   if (!isnan (threshold.failure_min))    i++;
 151   if (!isnan (threshold.failure_max))    i++;
 152   if (threshold.hysteresis > 0.0)        i++;
 153   if (threshold.hits > 1)                i++;
 154
 155   /* Print the response */
 156   print_to_socket (fh, "%zu Threshold found\n", i);
 157
 158   if (threshold.host[0] != 0)
 159     print_to_socket (fh, "Host: %s\n", threshold.host)
 160   if (threshold.plugin[0] != 0)
 161     print_to_socket (fh, "Plugin: %s\n", threshold.plugin)
 162   if (threshold.plugin_instance[0] != 0)
 163     print_to_socket (fh, "Plugin Instance: %s\n", threshold.plugin_instance)
 164   if (threshold.type[0] != 0)
 165     print_to_socket (fh, "Type: %s\n", threshold.type)
 166   if (threshold.type_instance[0] != 0)
 167     print_to_socket (fh, "Type Instance: %s\n", threshold.type_instance)
 168   if (threshold.data_source[0] != 0)
 169     print_to_socket (fh, "Data Source: %s\n", threshold.data_source)
 170   if (!isnan (threshold.warning_min))
 171     print_to_socket (fh, "Warning Min: %g\n", threshold.warning_min)
 172   if (!isnan (threshold.warning_max))
 173     print_to_socket (fh, "Warning Max: %g\n", threshold.warning_max)
 174   if (!isnan (threshold.failure_min))
 175     print_to_socket (fh, "Failure Min: %g\n", threshold.failure_min)
 176   if (!isnan (threshold.failure_max))
 177     print_to_socket (fh, "Failure Max: %g\n", threshold.failure_max)
 178   if (threshold.hysteresis > 0.0)
 179     print_to_socket (fh, "Hysteresis: %g\n", threshold.hysteresis)
 180   if (threshold.hits > 1)
 181     print_to_socket (fh, "Hits: %i\n", threshold.hits)
 182
 183   return (0);
 184 } /* int handle_getthreshold */
 185
 186 /* vim: set sw=2 sts=2 ts=8 et : */