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