Merge branch 'collectd-5.8'
[collectd.git] / src / utils_cmd_putval.c
index 691f1f8..b5b9065 100644 (file)
 #include "collectd.h"
 
 #include "common.h"
-#include "plugin.h"
-
-#include "utils_cmds.h"
-#include "utils_cmd_putval.h"
-#include "utils_parse_option.h"
 #include "utils_cmd_putval.h"
 
 /*
  * private helper functions
  */
 
-static int set_option (value_list_t *vl, const char *key, const char *value)
-{
-       if ((vl == NULL) || (key == NULL) || (value == NULL))
-               return (-1);
+static int set_option(value_list_t *vl, const char *key, const char *value) {
+  if ((vl == NULL) || (key == NULL) || (value == NULL))
+    return -1;
 
-       if (strcasecmp ("interval", key) == 0)
-       {
-               double tmp;
-               char *endptr;
+  if (strcasecmp("interval", key) == 0) {
+    double tmp;
+    char *endptr;
 
-               endptr = NULL;
-               errno = 0;
-               tmp = strtod (value, &endptr);
+    endptr = NULL;
+    errno = 0;
+    tmp = strtod(value, &endptr);
 
-               if ((errno == 0) && (endptr != NULL)
-                               && (endptr != value) && (tmp > 0.0))
-                       vl->interval = DOUBLE_TO_CDTIME_T (tmp);
-       }
-       else
-               return (1);
+    if ((errno == 0) && (endptr != NULL) && (endptr != value) && (tmp > 0.0))
+      vl->interval = DOUBLE_TO_CDTIME_T(tmp);
+  } else
+    return 1;
 
-       return (0);
+  return 0;
 } /* int set_option */
 
 /*
  * public API
  */
 
-cmd_status_t cmd_parse_putval (size_t argc, char **argv,
-               cmd_putval_t *ret_putval, const cmd_options_t *opts,
-               cmd_error_handler_t *err)
-{
-       cmd_status_t result;
-
-       char *identifier;
-       char *hostname;
-       char *plugin;
-       char *plugin_instance;
-       char *type;
-       char *type_instance;
-       int   status;
-
-       char *identifier_copy;
-
-       const data_set_t *ds;
-       value_list_t vl = VALUE_LIST_INIT;
-
-       if ((ret_putval == NULL) || (opts == NULL))
-       {
-               errno = EINVAL;
-               cmd_error (CMD_ERROR, err, "Invalid arguments to cmd_parse_putval.");
-               return (CMD_ERROR);
-       }
-
-       if (argc < 2)
-       {
-               cmd_error (CMD_PARSE_ERROR, err,
-                               "Missing identifier and/or value-list.");
-               return (CMD_PARSE_ERROR);
-       }
-
-       identifier = argv[0];
-
-       /* parse_identifier() modifies its first argument, returning pointers into
-        * it; retain the old value for later. */
-       identifier_copy = sstrdup (identifier);
-
-       status = parse_identifier (identifier, &hostname,
-                       &plugin, &plugin_instance,
-                       &type, &type_instance,
-                       opts->identifier_default_host);
-       if (status != 0)
-       {
-               DEBUG ("cmd_handle_putval: Cannot parse identifier `%s'.",
-                               identifier_copy);
-               cmd_error (CMD_PARSE_ERROR, err, "Cannot parse identifier `%s'.",
-                               identifier_copy);
-               sfree (identifier_copy);
-               return (CMD_PARSE_ERROR);
-       }
-
-       if ((strlen (hostname) >= sizeof (vl.host))
-                       || (strlen (plugin) >= sizeof (vl.plugin))
-                       || ((plugin_instance != NULL)
-                               && (strlen (plugin_instance) >= sizeof (vl.plugin_instance)))
-                       || ((type_instance != NULL)
-                               && (strlen (type_instance) >= sizeof (vl.type_instance))))
-       {
-               cmd_error (CMD_PARSE_ERROR, err, "Identifier too long.");
-               sfree (identifier_copy);
-               return (CMD_PARSE_ERROR);
-       }
-
-       sstrncpy (vl.host, hostname, sizeof (vl.host));
-       sstrncpy (vl.plugin, plugin, sizeof (vl.plugin));
-       sstrncpy (vl.type, type, sizeof (vl.type));
-       if (plugin_instance != NULL)
-               sstrncpy (vl.plugin_instance, plugin_instance, sizeof (vl.plugin_instance));
-       if (type_instance != NULL)
-               sstrncpy (vl.type_instance, type_instance, sizeof (vl.type_instance));
-
-       ds = plugin_get_ds (type);
-       if (ds == NULL)
-       {
-               cmd_error (CMD_PARSE_ERROR, err, "1 Type `%s' isn't defined.", type);
-               sfree (identifier_copy);
-               return (CMD_PARSE_ERROR);
-       }
-
-       hostname = NULL;
-       plugin = NULL; plugin_instance = NULL;
-       type = NULL;   type_instance = NULL;
-
-       vl.values_len = ds->ds_num;
-       vl.values = malloc (vl.values_len * sizeof (*vl.values));
-       if (vl.values == NULL)
-       {
-               cmd_error (CMD_ERROR, err, "malloc failed.");
-               sfree (identifier_copy);
-               return (CMD_ERROR);
-       }
-
-       ret_putval->raw_identifier = identifier_copy;
-       if (ret_putval->raw_identifier == NULL)
-       {
-               cmd_error (CMD_ERROR, err, "malloc failed.");
-               cmd_destroy_putval (ret_putval);
-               sfree (vl.values);
-               return (CMD_ERROR);
-       }
-
-       /* All the remaining fields are part of the option list. */
-       result = CMD_OK;
-       for (size_t i = 1; i < argc; ++i)
-       {
-               value_list_t *tmp;
-
-               char *key   = NULL;
-               char *value = NULL;
-
-               status = cmd_parse_option (argv[i], &key, &value, err);
-               if (status == CMD_OK)
-               {
-                       assert (key != NULL);
-                       assert (value != NULL);
-                       set_option (&vl, key, value);
-                       continue;
-               }
-               else if (status != CMD_NO_OPTION)
-               {
-                       /* parse_option failed, buffer has been modified.
-                        * => we need to abort */
-                       result = status;
-                       break;
-               }
-               /* else: cmd_parse_option did not find an option; treat this as a
-                * value list. */
-
-               status = parse_values (argv[i], &vl, ds);
-               if (status != 0)
-               {
-                       cmd_error (CMD_PARSE_ERROR, err, "Parsing the values string failed.");
-                       result = CMD_PARSE_ERROR;
-                       break;
-               }
-
-               tmp = (value_list_t *) realloc (ret_putval->vl,
-                               (ret_putval->vl_num + 1) * sizeof(*ret_putval->vl));
-               if (tmp == NULL)
-               {
-                       cmd_error (CMD_ERROR, err, "realloc failed.");
-                       cmd_destroy_putval (ret_putval);
-                       result = CMD_ERROR;
-                       break;
-               }
-
-               ret_putval->vl = tmp;
-               ret_putval->vl_num++;
-               memcpy (&ret_putval->vl[ret_putval->vl_num - 1], &vl, sizeof (vl));
-       } /* while (*buffer != 0) */
-       /* Done parsing the options. */
-
-       if (result != CMD_OK)
-       {
-               if (ret_putval->vl_num == 0)
-                       sfree (vl.values);
-               cmd_destroy_putval (ret_putval);
-       }
-
-       return (result);
+cmd_status_t cmd_parse_putval(size_t argc, char **argv,
+                              cmd_putval_t *ret_putval,
+                              const cmd_options_t *opts,
+                              cmd_error_handler_t *err) {
+  cmd_status_t result;
+
+  char *identifier;
+  char *hostname;
+  char *plugin;
+  char *plugin_instance;
+  char *type;
+  char *type_instance;
+  int status;
+
+  char *identifier_copy;
+
+  const data_set_t *ds;
+  value_list_t vl = VALUE_LIST_INIT;
+
+  if ((ret_putval == NULL) || (opts == NULL)) {
+    errno = EINVAL;
+    cmd_error(CMD_ERROR, err, "Invalid arguments to cmd_parse_putval.");
+    return CMD_ERROR;
+  }
+
+  if (argc < 2) {
+    cmd_error(CMD_PARSE_ERROR, err, "Missing identifier and/or value-list.");
+    return CMD_PARSE_ERROR;
+  }
+
+  identifier = argv[0];
+
+  /* parse_identifier() modifies its first argument, returning pointers into
+   * it; retain the old value for later. */
+  identifier_copy = sstrdup(identifier);
+
+  status =
+      parse_identifier(identifier, &hostname, &plugin, &plugin_instance, &type,
+                       &type_instance, opts->identifier_default_host);
+  if (status != 0) {
+    DEBUG("cmd_handle_putval: Cannot parse identifier `%s'.", identifier_copy);
+    cmd_error(CMD_PARSE_ERROR, err, "Cannot parse identifier `%s'.",
+              identifier_copy);
+    sfree(identifier_copy);
+    return CMD_PARSE_ERROR;
+  }
+
+  if ((strlen(hostname) >= sizeof(vl.host)) ||
+      (strlen(plugin) >= sizeof(vl.plugin)) ||
+      ((plugin_instance != NULL) &&
+       (strlen(plugin_instance) >= sizeof(vl.plugin_instance))) ||
+      ((type_instance != NULL) &&
+       (strlen(type_instance) >= sizeof(vl.type_instance)))) {
+    cmd_error(CMD_PARSE_ERROR, err, "Identifier too long.");
+    sfree(identifier_copy);
+    return CMD_PARSE_ERROR;
+  }
+
+  sstrncpy(vl.host, hostname, sizeof(vl.host));
+  sstrncpy(vl.plugin, plugin, sizeof(vl.plugin));
+  sstrncpy(vl.type, type, sizeof(vl.type));
+  if (plugin_instance != NULL)
+    sstrncpy(vl.plugin_instance, plugin_instance, sizeof(vl.plugin_instance));
+  if (type_instance != NULL)
+    sstrncpy(vl.type_instance, type_instance, sizeof(vl.type_instance));
+
+  ds = plugin_get_ds(type);
+  if (ds == NULL) {
+    cmd_error(CMD_PARSE_ERROR, err, "1 Type `%s' isn't defined.", type);
+    sfree(identifier_copy);
+    return CMD_PARSE_ERROR;
+  }
+
+  hostname = NULL;
+  plugin = NULL;
+  plugin_instance = NULL;
+  type = NULL;
+  type_instance = NULL;
+
+  ret_putval->raw_identifier = identifier_copy;
+  if (ret_putval->raw_identifier == NULL) {
+    cmd_error(CMD_ERROR, err, "malloc failed.");
+    cmd_destroy_putval(ret_putval);
+    sfree(vl.values);
+    return CMD_ERROR;
+  }
+
+  /* All the remaining fields are part of the option list. */
+  result = CMD_OK;
+  for (size_t i = 1; i < argc; ++i) {
+    value_list_t *tmp;
+
+    char *key = NULL;
+    char *value = NULL;
+
+    status = cmd_parse_option(argv[i], &key, &value, err);
+    if (status == CMD_OK) {
+      assert(key != NULL);
+      assert(value != NULL);
+      set_option(&vl, key, value);
+      continue;
+    } else if (status != CMD_NO_OPTION) {
+      /* parse_option failed, buffer has been modified.
+       * => we need to abort */
+      result = status;
+      break;
+    }
+    /* else: cmd_parse_option did not find an option; treat this as a
+     * value list. */
+
+    vl.values_len = ds->ds_num;
+    vl.values = calloc(vl.values_len, sizeof(*vl.values));
+    if (vl.values == NULL) {
+      cmd_error(CMD_ERROR, err, "malloc failed.");
+      result = CMD_ERROR;
+      break;
+    }
+
+    status = parse_values(argv[i], &vl, ds);
+    if (status != 0) {
+      cmd_error(CMD_PARSE_ERROR, err, "Parsing the values string failed.");
+      result = CMD_PARSE_ERROR;
+      vl.values_len = 0;
+      sfree(vl.values);
+      break;
+    }
+
+    tmp = realloc(ret_putval->vl,
+                  (ret_putval->vl_num + 1) * sizeof(*ret_putval->vl));
+    if (tmp == NULL) {
+      cmd_error(CMD_ERROR, err, "realloc failed.");
+      cmd_destroy_putval(ret_putval);
+      result = CMD_ERROR;
+      vl.values_len = 0;
+      sfree(vl.values);
+      break;
+    }
+
+    ret_putval->vl = tmp;
+    ret_putval->vl_num++;
+    memcpy(&ret_putval->vl[ret_putval->vl_num - 1], &vl, sizeof(vl));
+
+    /* pointer is now owned by ret_putval->vl[] */
+    vl.values_len = 0;
+    vl.values = NULL;
+  } /* while (*buffer != 0) */
+  /* Done parsing the options. */
+
+  if (result != CMD_OK)
+    cmd_destroy_putval(ret_putval);
+
+  return result;
 } /* cmd_status_t cmd_parse_putval */
 
-void cmd_destroy_putval (cmd_putval_t *putval)
-{
-       if (putval == NULL)
-               return;
-
-       sfree (putval->raw_identifier);
-
-       for (size_t i = 0; i < putval->vl_num; ++i)
-       {
-               if (i == 0) /* values is shared between all entries */
-                       sfree (putval->vl[i].values);
-               meta_data_destroy (putval->vl[i].meta);
-               putval->vl[i].meta = NULL;
-       }
-       sfree (putval->vl);
-       putval->vl = NULL;
-       putval->vl_num = 0;
+void cmd_destroy_putval(cmd_putval_t *putval) {
+  if (putval == NULL)
+    return;
+
+  sfree(putval->raw_identifier);
+
+  for (size_t i = 0; i < putval->vl_num; ++i) {
+    sfree(putval->vl[i].values);
+    meta_data_destroy(putval->vl[i].meta);
+    putval->vl[i].meta = NULL;
+  }
+  sfree(putval->vl);
+  putval->vl = NULL;
+  putval->vl_num = 0;
 } /* void cmd_destroy_putval */
 
-cmd_status_t cmd_handle_putval (FILE *fh, char *buffer)
-{
-       cmd_error_handler_t err = { cmd_error_fh, fh };
-       cmd_t cmd;
+cmd_status_t cmd_handle_putval(FILE *fh, char *buffer) {
+  cmd_error_handler_t err = {cmd_error_fh, fh};
+  cmd_t cmd;
 
-       int status;
+  int status;
 
-       DEBUG ("utils_cmd_putval: cmd_handle_putval (fh = %p, buffer = %s);",
-                       (void *) fh, buffer);
+  DEBUG("utils_cmd_putval: cmd_handle_putval (fh = %p, buffer = %s);",
+        (void *)fh, buffer);
 
-       if ((status = cmd_parse (buffer, &cmd, NULL, &err)) != CMD_OK)
-               return (status);
-       if (cmd.type != CMD_PUTVAL)
-       {
-               cmd_error (CMD_UNKNOWN_COMMAND, &err, "Unexpected command: `%s'.",
-                               CMD_TO_STRING (cmd.type));
-               cmd_destroy (&cmd);
-               return (CMD_UNKNOWN_COMMAND);
-       }
+  if ((status = cmd_parse(buffer, &cmd, NULL, &err)) != CMD_OK)
+    return status;
+  if (cmd.type != CMD_PUTVAL) {
+    cmd_error(CMD_UNKNOWN_COMMAND, &err, "Unexpected command: `%s'.",
+              CMD_TO_STRING(cmd.type));
+    cmd_destroy(&cmd);
+    return CMD_UNKNOWN_COMMAND;
+  }
 
-       for (size_t i = 0; i < cmd.cmd.putval.vl_num; ++i)
-               plugin_dispatch_values (&cmd.cmd.putval.vl[i]);
+  for (size_t i = 0; i < cmd.cmd.putval.vl_num; ++i)
+    plugin_dispatch_values(&cmd.cmd.putval.vl[i]);
 
-       if (fh != stdout)
-               cmd_error (CMD_OK, &err, "Success: %i %s been dispatched.",
-                               (int)cmd.cmd.putval.vl_num,
-                               (cmd.cmd.putval.vl_num == 1) ? "value has" : "values have");
+  if (fh != stdout)
+    cmd_error(CMD_OK, &err, "Success: %i %s been dispatched.",
+              (int)cmd.cmd.putval.vl_num,
+              (cmd.cmd.putval.vl_num == 1) ? "value has" : "values have");
 
-       cmd_destroy (&cmd);
-       return (CMD_OK);
+  cmd_destroy(&cmd);
+  return CMD_OK;
 } /* int cmd_handle_putval */
 
-int cmd_create_putval (char *ret, size_t ret_len, /* {{{ */
-       const data_set_t *ds, const value_list_t *vl)
-{
-       char buffer_ident[6 * DATA_MAX_NAME_LEN];
-       char buffer_values[1024];
-       int status;
-
-       status = FORMAT_VL (buffer_ident, sizeof (buffer_ident), vl);
-       if (status != 0)
-               return (status);
-       escape_string (buffer_ident, sizeof (buffer_ident));
-
-       status = format_values (buffer_values, sizeof (buffer_values),
-                       ds, vl, /* store rates = */ 0);
-       if (status != 0)
-               return (status);
-       escape_string (buffer_values, sizeof (buffer_values));
-
-       ssnprintf (ret, ret_len,
-                       "PUTVAL %s interval=%.3f %s",
-                       buffer_ident,
-                       (vl->interval > 0)
-                       ? CDTIME_T_TO_DOUBLE (vl->interval)
-                       : CDTIME_T_TO_DOUBLE (plugin_get_interval ()),
-                       buffer_values);
-
-       return (0);
+int cmd_create_putval(char *ret, size_t ret_len, /* {{{ */
+                      const data_set_t *ds, const value_list_t *vl) {
+  char buffer_ident[6 * DATA_MAX_NAME_LEN];
+  char buffer_values[1024];
+  int status;
+
+  status = FORMAT_VL(buffer_ident, sizeof(buffer_ident), vl);
+  if (status != 0)
+    return status;
+  escape_string(buffer_ident, sizeof(buffer_ident));
+
+  status = format_values(buffer_values, sizeof(buffer_values), ds, vl,
+                         /* store rates = */ false);
+  if (status != 0)
+    return status;
+  escape_string(buffer_values, sizeof(buffer_values));
+
+  snprintf(ret, ret_len, "PUTVAL %s interval=%.3f %s", buffer_ident,
+           (vl->interval > 0) ? CDTIME_T_TO_DOUBLE(vl->interval)
+                              : CDTIME_T_TO_DOUBLE(plugin_get_interval()),
+           buffer_values);
+
+  return 0;
 } /* }}} int cmd_create_putval */