src/utils_cmd_getval.c

   1 /**
   2  * collectd - src/utils_cmd_getval.c
   3  * Copyright (C) 2008       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_cache.h"
  33 #include "utils_parse_option.h"
  34 #include "utils_cmd_getval.h"
  35
  36 #define print_to_socket(fh, ...) \
  37   do { \
  38     if (fprintf (fh, __VA_ARGS__) < 0) { \
  39       char errbuf[1024]; \
  40       WARNING ("handle_getval: failed to write to socket #%i: %s", \
  41           fileno (fh), sstrerror (errno, errbuf, sizeof (errbuf))); \
  42       return -1; \
  43     } \
  44     fflush(fh); \
  45   } while (0)
  46
  47 int handle_getval (FILE *fh, char *buffer)
  48 {
  49   char *command;
  50   char *identifier;
  51   char *identifier_copy;
  52
  53   char *hostname;
  54   char *plugin;
  55   char *plugin_instance;
  56   char *type;
  57   char *type_instance;
  58   gauge_t *values;
  59   size_t values_num;
  60
  61   const data_set_t *ds;
  62
  63   int   status;
  64   size_t i;
  65
  66   if ((fh == NULL) || (buffer == NULL))
  67     return (-1);
  68
  69   DEBUG ("utils_cmd_getval: handle_getval (fh = %p, buffer = %s);",
  70       (void *) fh, buffer);
  71
  72   command = NULL;
  73   status = parse_string (&buffer, &command);
  74   if (status != 0)
  75   {
  76     print_to_socket (fh, "-1 Cannot parse command.\n");
  77     return (-1);
  78   }
  79   assert (command != NULL);
  80
  81   if (strcasecmp ("GETVAL", command) != 0)
  82   {
  83     print_to_socket (fh, "-1 Unexpected command: `%s'.\n", command);
  84     return (-1);
  85   }
  86
  87   identifier = NULL;
  88   status = parse_string (&buffer, &identifier);
  89   if (status != 0)
  90   {
  91     print_to_socket (fh, "-1 Cannot parse identifier.\n");
  92     return (-1);
  93   }
  94   assert (identifier != NULL);
  95
  96   if (*buffer != 0)
  97   {
  98     print_to_socket (fh, "-1 Garbage after end of command: %s\n", buffer);
  99     return (-1);
 100   }
 101
 102   /* parse_identifier() modifies its first argument,
 103    * returning pointers into it */
 104   identifier_copy = sstrdup (identifier);
 105
 106   status = parse_identifier (identifier_copy, &hostname,
 107       &plugin, &plugin_instance,
 108       &type, &type_instance);
 109   if (status != 0)
 110   {
 111     DEBUG ("handle_getval: Cannot parse identifier `%s'.", identifier);
 112     print_to_socket (fh, "-1 Cannot parse identifier `%s'.\n", identifier);
 113     sfree (identifier_copy);
 114     return (-1);
 115   }
 116
 117   ds = plugin_get_ds (type);
 118   if (ds == NULL)
 119   {
 120     DEBUG ("handle_getval: plugin_get_ds (%s) == NULL;", type);
 121     print_to_socket (fh, "-1 Type `%s' is unknown.\n", type);
 122     sfree (identifier_copy);
 123     return (-1);
 124   }
 125
 126   values = NULL;
 127   values_num = 0;
 128   status = uc_get_rate_by_name (identifier, &values, &values_num);
 129   if (status != 0)
 130   {
 131     print_to_socket (fh, "-1 No such value\n");
 132     sfree (identifier_copy);
 133     return (-1);
 134   }
 135
 136   if (ds->ds_num != values_num)
 137   {
 138     ERROR ("ds[%s]->ds_num = %zu, "
 139         "but uc_get_rate_by_name returned %zu values.",
 140         ds->type, ds->ds_num, values_num);
 141     print_to_socket (fh, "-1 Error reading value from cache.\n");
 142     sfree (values);
 143     sfree (identifier_copy);
 144     return (-1);
 145   }
 146
 147   print_to_socket (fh, "%zu Value%s found\n", values_num,
 148       (values_num == 1) ? "" : "s");
 149   for (i = 0; i < values_num; i++)
 150   {
 151     print_to_socket (fh, "%s=", ds->ds[i].name);
 152     if (isnan (values[i]))
 153     {
 154       print_to_socket (fh, "NaN\n");
 155     }
 156     else
 157     {
 158       print_to_socket (fh, "%12e\n", values[i]);
 159     }
 160   }
 161
 162   sfree (values);
 163   sfree (identifier_copy);
 164
 165   return (0);
 166 } /* int handle_getval */
 167
 168 /* vim: set sw=2 sts=2 ts=8 : */