src/utils_cmd_getval.c

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