2 * collectd - src/collectdctl.c
3 * Copyright (C) 2010 Håkon J Dugstad Johnsen
4 * Copyright (C) 2010 Sebastian Harl
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.
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.
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
20 * Håkon J Dugstad Johnsen <hakon-dugstad.johnsen at telenor.com>
21 * Sebastian "tokkee" Harl <sh@tokkee.org>
28 #include "libcollectdclient/client.h"
42 #define DEFAULT_SOCK LOCALSTATEDIR"/run/"PACKAGE_NAME"-unixsock"
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"
64 "An identifier has the following format:\n\n"
66 " [<hostname>/]<plugin>[-<plugin_instance>]/<type>[-<type_instance>]\n\n"
68 "Hostname defaults to the local hostname if omitted (e.g., uptime/uptime).\n"
69 "No error is returned if the specified identifier does not exist.\n"
73 " collectdctl flush plugin=rrdtool identifie=somehost/cpu-0/cpu-wait\n\n"
75 "Flushes all CPU wait RRD values of the first CPU of the local host.\n"
76 "I.e., writes all pending RRD updates of that data-source to disk.\n"
78 "\n"PACKAGE" "VERSION", http://collectd.org/\n"
79 "by Florian octo Forster <octo@verplant.org>\n"
80 "for contributions see `AUTHORS'\n"
85 /* Count the number of occurrences of the character 'chr'
86 * in the specified string. */
87 static int count_chars (const char *str, char chr) {
90 while (*str != '\0') {
100 static int parse_identifier (lcc_connection_t *c,
101 const char *value, lcc_identifier_t *ident)
104 char ident_str[1024] = "";
109 n_slashes = count_chars (value, '/');
110 if (n_slashes == 1) {
111 /* The user has omitted the hostname part of the identifier
112 * (there is only one '/' in the identifier)
113 * Let's add the local hostname */
114 if (gethostname (hostname, sizeof (hostname)) != 0) {
115 fprintf (stderr, "ERROR: Failed to get local hostname: %s",
119 hostname[sizeof (hostname) - 1] = '\0';
121 snprintf (ident_str, sizeof (ident_str), "%s/%s", hostname, value);
122 ident_str[sizeof(ident_str) - 1] = '\0';
125 strncpy (ident_str, value, sizeof (ident_str));
126 ident_str[sizeof (ident_str) - 1] = '\0';
129 status = lcc_string_to_identifier (c, ident, ident_str);
131 fprintf (stderr, "ERROR: Failed to parse identifier ``%s'': %s.\n",
132 ident_str, lcc_strerror(c));
136 } /* parse_identifier */
138 static int getval (lcc_connection_t *c, int argc, char **argv)
140 lcc_identifier_t ident;
142 size_t ret_values_num = 0;
143 gauge_t *ret_values = NULL;
144 char **ret_values_names = NULL;
149 assert (strcasecmp (argv[0], "getval") == 0);
152 fprintf (stderr, "ERROR: getval: Missing identifier.\n");
156 memset (&ident, 0, sizeof (ident));
157 status = parse_identifier (c, argv[1], &ident);
161 #define BAIL_OUT(s) \
163 if (ret_values != NULL) \
165 if (ret_values_names != NULL) { \
166 for (i = 0; i < ret_values_num; ++i) \
167 free (ret_values_names[i]); \
168 free (ret_values_names); \
170 ret_values_num = 0; \
174 status = lcc_getval (c, &ident,
175 &ret_values_num, &ret_values, &ret_values_names);
179 for (i = 0; i < ret_values_num; ++i)
180 printf ("%s=%e\n", ret_values_names[i], ret_values[i]);
184 static int flush (lcc_connection_t *c, int argc, char **argv)
186 lcc_identifier_t ident;
187 lcc_identifier_t *identp = NULL;
195 assert (strcasecmp (argv[0], "flush") == 0);
197 for (i = 1; i < argc; ++i) {
201 value = strchr (argv[i], (int)'=');
204 fprintf (stderr, "ERROR: flush: Invalid option ``%s''.\n", argv[i]);
211 if (strcasecmp (key, "timeout") == 0) {
214 timeout = strtol (value, &endptr, 0);
216 if (endptr == value) {
217 fprintf (stderr, "ERROR: Failed to parse timeout as number: %s.\n",
221 else if ((endptr != NULL) && (*endptr != '\0')) {
222 fprintf (stderr, "WARNING: Ignoring trailing garbage after timeout: "
226 else if (strcasecmp (key, "plugin") == 0) {
229 else if (strcasecmp (key, "identifier") == 0) {
232 memset (&ident, 0, sizeof (ident));
233 status = parse_identifier (c, value, &ident);
240 status = lcc_flush (c, plugin, identp, timeout);
242 fprintf (stderr, "ERROR: Flushing failed: %s.\n",
250 int main (int argc, char **argv) {
251 char address[1024] = "unix:"DEFAULT_SOCK;
260 c = getopt (argc, argv, "s:h");
267 snprintf (address, sizeof (address), "unix:%s", optarg);
268 address[sizeof (address) - 1] = '\0';
271 exit_usage (argv[0], 0);
274 exit_usage (argv[0], 1);
278 if (optind >= argc) {
279 fprintf (stderr, "%s: missing command\n", argv[0]);
280 exit_usage (argv[0], 1);
284 status = lcc_connect (address, &c);
286 fprintf (stderr, "ERROR: Failed to connect to daemon at %s: %s.\n",
287 address, strerror (errno));
291 if (strcasecmp (argv[optind], "getval") == 0)
292 status = getval (c, argc - optind, argv + optind);
293 else if (strcasecmp (argv[optind], "flush") == 0)
294 status = flush (c, argc - optind, argv + optind);
296 fprintf (stderr, "%s: invalid command: %s\n", argv[0], argv[optind]);
307 /* vim: set sw=2 ts=2 tw=78 expandtab : */