X-Git-Url: https://git.octo.it/?a=blobdiff_plain;f=src%2Futils_cmd_flush.c;h=a01d2343c9cb46f904dd295988f10fa9731418ca;hb=49ac868d2346c77335272b8a643bca295db9feaa;hp=4e7526be5a014df57b87a71d1aa136da07ab75c3;hpb=c1318b06b424523d3704fcdfd16d3adb25d17d18;p=collectd.git diff --git a/src/utils_cmd_flush.c b/src/utils_cmd_flush.c index 4e7526be..a01d2343 100644 --- a/src/utils_cmd_flush.c +++ b/src/utils_cmd_flush.c @@ -1,56 +1,36 @@ /** * collectd - src/utils_cmd_flush.c - * Copyright (C) 2008 Sebastian Harl - * Copyright (C) 2008 Florian Forster + * Copyright (C) 2008 Sebastian Harl + * Copyright (C) 2008 Florian Forster * - * This program is free software; you can redistribute it and/or modify it - * under the terms of the GNU General Public License as published by the - * Free Software Foundation; only version 2 of the License is applicable. + * Permission is hereby granted, free of charge, to any person obtaining a + * copy of this software and associated documentation files (the "Software"), + * to deal in the Software without restriction, including without limitation + * the rights to use, copy, modify, merge, publish, distribute, sublicense, + * and/or sell copies of the Software, and to permit persons to whom the + * Software is furnished to do so, subject to the following conditions: * - * This program is distributed in the hope that it will be useful, but - * WITHOUT ANY WARRANTY; without even the implied warranty of - * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU - * General Public License for more details. + * The above copyright notice and this permission notice shall be included in + * all copies or substantial portions of the Software. * - * You should have received a copy of the GNU General Public License along - * with this program; if not, write to the Free Software Foundation, Inc., - * 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA + * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR + * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, + * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE + * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER + * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING + * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER + * DEALINGS IN THE SOFTWARE. * * Authors: * Sebastian "tokkee" Harl - * Florian "octo" Forster + * Florian "octo" Forster **/ #include "collectd.h" #include "common.h" #include "plugin.h" #include "utils_parse_option.h" - -#define print_to_socket(fh, ...) \ - do { \ - if (fprintf (fh, __VA_ARGS__) < 0) { \ - char errbuf[1024]; \ - WARNING ("handle_flush: failed to write to socket #%i: %s", \ - fileno (fh), sstrerror (errno, errbuf, sizeof (errbuf))); \ - return -1; \ - } \ - fflush(fh); \ - } while (0) - -static int add_to_array (char ***array, int *array_num, char *value) -{ - char **temp; - - temp = (char **) realloc (*array, sizeof (char *) * (*array_num + 1)); - if (temp == NULL) - return (-1); - - *array = temp; - (*array)[*array_num] = value; - (*array_num)++; - - return (0); -} /* int add_to_array */ +#include "utils_cmd_flush.h" int handle_flush (FILE *fh, char *buffer) { @@ -59,11 +39,24 @@ int handle_flush (FILE *fh, char *buffer) double timeout = 0.0; char **plugins = NULL; - int plugins_num = 0; + size_t plugins_num = 0; char **identifiers = NULL; - int identifiers_num = 0; + size_t identifiers_num = 0; - int i; + size_t i; + +#define PRINT_TO_SOCK(fh, ...) \ + do { \ + if (fprintf (fh, __VA_ARGS__) < 0) { \ + char errbuf[1024]; \ + WARNING ("handle_flush: failed to write to socket #%i: %s", \ + fileno (fh), sstrerror (errno, errbuf, sizeof (errbuf))); \ + strarray_free (plugins, plugins_num); \ + strarray_free (identifiers, identifiers_num); \ + return -1; \ + } \ + fflush(fh); \ + } while (0) if ((fh == NULL) || (buffer == NULL)) return (-1); @@ -73,7 +66,7 @@ int handle_flush (FILE *fh, char *buffer) if (strncasecmp ("FLUSH", buffer, strlen ("FLUSH")) != 0) { - print_to_socket (fh, "-1 Cannot parse command.\n"); + PRINT_TO_SOCK (fh, "-1 Cannot parse command.\n"); return (-1); } buffer += strlen ("FLUSH"); @@ -89,34 +82,30 @@ int handle_flush (FILE *fh, char *buffer) status = parse_option (&buffer, &opt_key, &opt_value); if (status != 0) { - print_to_socket (fh, "-1 Parsing options failed.\n"); - sfree (plugins); - sfree (identifiers); + PRINT_TO_SOCK (fh, "-1 Parsing options failed.\n"); + strarray_free (plugins, plugins_num); + strarray_free (identifiers, identifiers_num); return (-1); } if (strcasecmp ("plugin", opt_key) == 0) - { - add_to_array (&plugins, &plugins_num, opt_value); - } + strarray_add (&plugins, &plugins_num, opt_value); else if (strcasecmp ("identifier", opt_key) == 0) - { - add_to_array (&identifiers, &identifiers_num, opt_value); - } + strarray_add (&identifiers, &identifiers_num, opt_value); else if (strcasecmp ("timeout", opt_key) == 0) { char *endptr; - + errno = 0; endptr = NULL; timeout = strtod (opt_value, &endptr); if ((endptr == opt_value) || (errno != 0) || (!isfinite (timeout))) { - print_to_socket (fh, "-1 Invalid value for option `timeout': " + PRINT_TO_SOCK (fh, "-1 Invalid value for option `timeout': " "%s\n", opt_value); - sfree (plugins); - sfree (identifiers); + strarray_free (plugins, plugins_num); + strarray_free (identifiers, identifiers_num); return (-1); } else if (timeout < 0.0) @@ -126,34 +115,29 @@ int handle_flush (FILE *fh, char *buffer) } else { - print_to_socket (fh, "-1 Cannot parse option %s\n", opt_key); - sfree (plugins); - sfree (identifiers); + PRINT_TO_SOCK (fh, "-1 Cannot parse option %s\n", opt_key); + strarray_free (plugins, plugins_num); + strarray_free (identifiers, identifiers_num); return (-1); } } /* while (*buffer != 0) */ - /* Add NULL entries for `any plugin' and/or `any value' if nothing was - * specified. */ - if (plugins_num == 0) - add_to_array (&plugins, &plugins_num, NULL); - - if (identifiers_num == 0) - add_to_array (&identifiers, &identifiers_num, NULL); - - for (i = 0; i < plugins_num; i++) + for (i = 0; (i == 0) || (i < plugins_num); i++) { - char *plugin; + char *plugin = NULL; int j; - plugin = plugins[i]; + if (plugins_num != 0) + plugin = plugins[i]; - for (j = 0; j < identifiers_num; j++) + for (j = 0; (j == 0) || (j < identifiers_num); j++) { - char *identifier; + char *identifier = NULL; int status; - identifier = identifiers[j]; + if (identifiers_num != 0) + identifier = identifiers[j]; + status = plugin_flush (plugin, DOUBLE_TO_CDTIME_T (timeout), identifier); @@ -164,20 +148,13 @@ int handle_flush (FILE *fh, char *buffer) } } - if ((success + error) > 0) - { - print_to_socket (fh, "0 Done: %i successful, %i errors\n", - success, error); - } - else - { - plugin_flush (NULL, timeout, NULL); - print_to_socket (fh, "0 Done\n"); - } + PRINT_TO_SOCK (fh, "0 Done: %i successful, %i errors\n", + success, error); - sfree (plugins); - sfree (identifiers); + strarray_free (plugins, plugins_num); + strarray_free (identifiers, identifiers_num); return (0); +#undef PRINT_TO_SOCK } /* int handle_flush */ /* vim: set sw=4 ts=4 tw=78 noexpandtab : */