1 /**
2 * collectd - src/collectdctl.c
3 * Copyright (C) 2010 Håkon J Dugstad Johnsen
4 * Copyright (C) 2010 Sebastian Harl
5 *
6 * This program is free software; you can redistribute it and/or modify it
7 * under the terms of the GNU General Public License as published by the
8 * Free Software Foundation; only version 2 of the License is applicable.
9 *
10 * This program is distributed in the hope that it will be useful, but
11 * WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 * General Public License for more details.
14 *
15 * You should have received a copy of the GNU General Public License along
16 * with this program; if not, write to the Free Software Foundation, Inc.,
17 * 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
18 *
19 * Authors:
20 * Håkon J Dugstad Johnsen <hakon-dugstad.johnsen at telenor.com>
21 * Sebastian "tokkee" Harl <sh@tokkee.org>
22 **/
24 #if HAVE_CONFIG_H
25 # include "config.h"
26 #endif
28 #include "libcollectdclient/client.h"
30 #include <assert.h>
32 #include <errno.h>
34 #include <getopt.h>
36 #include <stdio.h>
37 #include <stdlib.h>
38 #include <string.h>
40 #include <unistd.h>
42 #define DEFAULT_SOCK LOCALSTATEDIR"/run/"PACKAGE_NAME"-unixsock"
44 extern char *optarg;
45 extern int optind;
47 static void exit_usage (const char *name, int status) {
48 fprintf ((status == 0) ? stdout : stderr,
49 "Usage: %s [options] <command> [cmd options]\n\n"
51 "Available options:\n"
52 " -s Path to collectd's UNIX socket.\n"
53 " Default: "DEFAULT_SOCK"\n"
55 "\n -h Display this help and exit.\n"
57 "\nAvailable commands:\n\n"
59 " * getval <identifier>\n"
60 " * flush [timeout=<seconds>] [plugin=<name>] [identifier=<id>]\n"
61 " * listval\n"
63 "\nIdentifiers:\n\n"
65 "An identifier has the following format:\n\n"
67 " [<hostname>/]<plugin>[-<plugin_instance>]/<type>[-<type_instance>]\n\n"
69 "Hostname defaults to the local hostname if omitted (e.g., uptime/uptime).\n"
70 "No error is returned if the specified identifier does not exist.\n"
72 "\nExample:\n\n"
74 " collectdctl flush plugin=rrdtool identifie=somehost/cpu-0/cpu-wait\n\n"
76 "Flushes all CPU wait RRD values of the first CPU of the local host.\n"
77 "I.e., writes all pending RRD updates of that data-source to disk.\n"
79 "\n"PACKAGE" "VERSION", http://collectd.org/\n"
80 "by Florian octo Forster <octo@verplant.org>\n"
81 "for contributions see `AUTHORS'\n"
82 , name);
83 exit (status);
84 }
86 /* Count the number of occurrences of the character 'chr'
87 * in the specified string. */
88 static int count_chars (const char *str, char chr) {
89 int count = 0;
91 while (*str != '\0') {
92 if (*str == chr) {
93 count++;
94 }
95 str++;
96 }
98 return count;
99 } /* count_chars */
101 static int parse_identifier (lcc_connection_t *c,
102 const char *value, lcc_identifier_t *ident)
103 {
104 char hostname[1024];
105 char ident_str[1024] = "";
106 int n_slashes;
108 int status;
110 n_slashes = count_chars (value, '/');
111 if (n_slashes == 1) {
112 /* The user has omitted the hostname part of the identifier
113 * (there is only one '/' in the identifier)
114 * Let's add the local hostname */
115 if (gethostname (hostname, sizeof (hostname)) != 0) {
116 fprintf (stderr, "ERROR: Failed to get local hostname: %s",
117 strerror (errno));
118 return (-1);
119 }
120 hostname[sizeof (hostname) - 1] = '\0';
122 snprintf (ident_str, sizeof (ident_str), "%s/%s", hostname, value);
123 ident_str[sizeof(ident_str) - 1] = '\0';
124 }
125 else {
126 strncpy (ident_str, value, sizeof (ident_str));
127 ident_str[sizeof (ident_str) - 1] = '\0';
128 }
130 status = lcc_string_to_identifier (c, ident, ident_str);
131 if (status != 0) {
132 fprintf (stderr, "ERROR: Failed to parse identifier ``%s'': %s.\n",
133 ident_str, lcc_strerror(c));
134 return (-1);
135 }
136 return (0);
137 } /* parse_identifier */
139 static int getval (lcc_connection_t *c, int argc, char **argv)
140 {
141 lcc_identifier_t ident;
143 size_t ret_values_num = 0;
144 gauge_t *ret_values = NULL;
145 char **ret_values_names = NULL;
147 int status;
148 size_t i;
150 assert (strcasecmp (argv[0], "getval") == 0);
152 if (argc != 2) {
153 fprintf (stderr, "ERROR: getval: Missing identifier.\n");
154 return (-1);
155 }
157 memset (&ident, 0, sizeof (ident));
158 status = parse_identifier (c, argv[1], &ident);
159 if (status != 0)
160 return (status);
162 #define BAIL_OUT(s) \
163 do { \
164 if (ret_values != NULL) \
165 free (ret_values); \
166 if (ret_values_names != NULL) { \
167 for (i = 0; i < ret_values_num; ++i) \
168 free (ret_values_names[i]); \
169 free (ret_values_names); \
170 } \
171 ret_values_num = 0; \
172 return (s); \
173 } while (0)
175 status = lcc_getval (c, &ident,
176 &ret_values_num, &ret_values, &ret_values_names);
177 if (status != 0)
178 BAIL_OUT (-1);
180 for (i = 0; i < ret_values_num; ++i)
181 printf ("%s=%e\n", ret_values_names[i], ret_values[i]);
182 BAIL_OUT (0);
183 } /* getval */
185 static int flush (lcc_connection_t *c, int argc, char **argv)
186 {
187 lcc_identifier_t ident;
188 lcc_identifier_t *identp = NULL;
190 char *plugin = NULL;
191 int timeout = -1;
193 int status;
194 int i;
196 assert (strcasecmp (argv[0], "flush") == 0);
198 for (i = 1; i < argc; ++i) {
199 char *key, *value;
201 key = argv[i];
202 value = strchr (argv[i], (int)'=');
204 if (! value) {
205 fprintf (stderr, "ERROR: flush: Invalid option ``%s''.\n", argv[i]);
206 return (-1);
207 }
209 *value = '\0';
210 ++value;
212 if (strcasecmp (key, "timeout") == 0) {
213 char *endptr = NULL;
215 timeout = strtol (value, &endptr, 0);
217 if (endptr == value) {
218 fprintf (stderr, "ERROR: Failed to parse timeout as number: %s.\n",
219 value);
220 return (-1);
221 }
222 else if ((endptr != NULL) && (*endptr != '\0')) {
223 fprintf (stderr, "WARNING: Ignoring trailing garbage after timeout: "
224 "%s.\n", endptr);
225 }
226 }
227 else if (strcasecmp (key, "plugin") == 0) {
228 plugin = value;
229 }
230 else if (strcasecmp (key, "identifier") == 0) {
231 int status;
233 memset (&ident, 0, sizeof (ident));
234 status = parse_identifier (c, value, &ident);
235 if (status != 0)
236 return (status);
237 identp = &ident;
238 }
239 }
241 status = lcc_flush (c, plugin, identp, timeout);
242 if (status != 0) {
243 fprintf (stderr, "ERROR: Flushing failed: %s.\n",
244 lcc_strerror (c));
245 return (-1);
246 }
248 return 0;
249 #undef BAIL_OUT
250 } /* flush */
252 static int listval (lcc_connection_t *c, int argc, char **argv)
253 {
254 lcc_identifier_t *ret_ident = NULL;
255 size_t ret_ident_num = 0;
257 int status;
258 size_t i;
260 assert (strcasecmp (argv[0], "listval") == 0);
262 if (argc != 1) {
263 fprintf (stderr, "ERROR: listval: Does not accept any arguments.\n");
264 return (-1);
265 }
267 #define BAIL_OUT(s) \
268 do { \
269 if (ret_ident != NULL) \
270 free (ret_ident); \
271 ret_ident_num = 0; \
272 return (s); \
273 } while (0)
275 status = lcc_listval (c, &ret_ident, &ret_ident_num);
276 if (status != 0)
277 BAIL_OUT (status);
279 for (i = 0; i < ret_ident_num; ++i) {
280 char id[1024];
282 status = lcc_identifier_to_string (c, id, sizeof (id), ret_ident + i);
283 if (status != 0) {
284 fprintf (stderr, "ERROR: listval: Failed to convert returned "
285 "identifier to a string.\n");
286 continue;
287 }
289 printf ("%s\n", id);
290 }
291 BAIL_OUT (0);
292 #undef BAIL_OUT
293 } /* listval */
295 int main (int argc, char **argv) {
296 char address[1024] = "unix:"DEFAULT_SOCK;
298 lcc_connection_t *c;
300 int status;
302 while (42) {
303 int c;
305 c = getopt (argc, argv, "s:h");
307 if (c == -1)
308 break;
310 switch (c) {
311 case 's':
312 snprintf (address, sizeof (address), "unix:%s", optarg);
313 address[sizeof (address) - 1] = '\0';
314 break;
315 case 'h':
316 exit_usage (argv[0], 0);
317 break;
318 default:
319 exit_usage (argv[0], 1);
320 }
321 }
323 if (optind >= argc) {
324 fprintf (stderr, "%s: missing command\n", argv[0]);
325 exit_usage (argv[0], 1);
326 }
328 c = NULL;
329 status = lcc_connect (address, &c);
330 if (status != 0) {
331 fprintf (stderr, "ERROR: Failed to connect to daemon at %s: %s.\n",
332 address, strerror (errno));
333 return (1);
334 }
336 if (strcasecmp (argv[optind], "getval") == 0)
337 status = getval (c, argc - optind, argv + optind);
338 else if (strcasecmp (argv[optind], "flush") == 0)
339 status = flush (c, argc - optind, argv + optind);
340 else if (strcasecmp (argv[optind], "listval") == 0)
341 status = listval (c, argc - optind, argv + optind);
342 else {
343 fprintf (stderr, "%s: invalid command: %s\n", argv[0], argv[optind]);
344 return (1);
345 }
347 LCC_DESTROY (c);
349 if (status != 0)
350 return (status);
351 return (0);
352 } /* main */
354 /* vim: set sw=2 ts=2 tw=78 expandtab : */