static int tr_meta_data_action_invoke(/* {{{ */
tr_meta_data_action_t *act_head,
meta_data_t **dest) {
int status;
regmatch_t matches[8] = {[0] = {0}};
if (act_head == NULL)
return (-EINVAL);
if ((*dest) == NULL) /* nothing to do */
return (0);
for (tr_meta_data_action_t *act = act_head; act != NULL; act = act->next) {
char temp[DATA_MAX_NAME_LEN];
char *subst_status;
int value_type;
int meta_data_status;
char *value;
meta_data_t *result;
value_type = meta_data_type(*dest, act->key);
if (value_type == 0) /* not found */
continue;
if (value_type != MD_TYPE_STRING) {
WARNING("Target `replace': Attempting replace on metadata key `%s', "
"which isn't a string.",
act->key);
continue;
}
meta_data_status = meta_data_get_string(*dest, act->key, &value);
if (meta_data_status != 0) {
ERROR("Target `replace': Unable to retrieve metadata value for `%s'.",
act->key);
return (meta_data_status);
}
DEBUG("target_replace plugin: tr_meta_data_action_invoke: `%s' "
"old value = `%s'",
act->key, value);
status = regexec(&act->re, value, STATIC_ARRAY_SIZE(matches), matches,
/* flags = */ 0);
if (status == REG_NOMATCH) {
sfree(value);
continue;
} else if (status != 0) {
char errbuf[1024] = "";
regerror(status, &act->re, errbuf, sizeof(errbuf));
ERROR("Target `replace': Executing a regular expression failed: %s.",
errbuf);
sfree(value);
continue;
}
if (act->replacement == NULL) {
/* no replacement; delete the key */
DEBUG("target_replace plugin: tr_meta_data_action_invoke: "
"deleting `%s'",
act->key);
meta_data_delete(*dest, act->key);
sfree(value);
continue;
}
subst_status = subst(temp, sizeof(temp), value, (size_t)matches[0].rm_so,
(size_t)matches[0].rm_eo, act->replacement);
if (subst_status == NULL) {
ERROR("Target `replace': subst (value = %s, start = %zu, end = %zu, "
"replacement = %s) failed.",
value, (size_t)matches[0].rm_so, (size_t)matches[0].rm_eo,
act->replacement);
sfree(value);
continue;
}
DEBUG("target_replace plugin: tr_meta_data_action_invoke: `%s' "
"value `%s' -> `%s'",
act->key, value, temp);
if ((result = meta_data_create()) == NULL) {
ERROR("Target `replace': failed to create metadata for `%s'.", act->key);
sfree(value);
return (-ENOMEM);
}
meta_data_status = meta_data_add_string(result, act->key, temp);
if (meta_data_status != 0) {
ERROR("Target `replace': Unable to set metadata value for `%s'.",
act->key);
meta_data_destroy(result);
sfree(value);
return (meta_data_status);
}
meta_data_clone_merge(dest, result);
meta_data_destroy(result);
sfree(value);
} /* for (act = act_head; act != NULL; act = act->next) */
return (0);
} /* }}} int tr_meta_data_action_invoke */
static int ts_create(const oconfig_item_t *ci, void **user_data) /* {{{ */
{
ts_data_t *data;
int status;
data = calloc(1, sizeof(*data));
if (data == NULL) {
ERROR("ts_create: calloc failed.");
return (-ENOMEM);
}
data->host = NULL;
data->plugin = NULL;
data->plugin_instance = NULL;
/* data->type = NULL; */
data->type_instance = NULL;
data->meta = NULL;
data->meta_delete = NULL;
status = 0;
for (int i = 0; i < ci->children_num; i++) {
oconfig_item_t *child = ci->children + i;
if ((strcasecmp("Host", child->key) == 0) ||
(strcasecmp("Hostname", child->key) == 0))
status = ts_config_add_string(&data->host, child,
/* may be empty = */ 0);
else if (strcasecmp("Plugin", child->key) == 0)
status = ts_config_add_string(&data->plugin, child,
/* may be empty = */ 0);
else if (strcasecmp("PluginInstance", child->key) == 0)
status = ts_config_add_string(&data->plugin_instance, child,
/* may be empty = */ 1);
#if 0
else if (strcasecmp ("Type", child->key) == 0)
status = ts_config_add_string (&data->type, child,
/* may be empty = */ 0);
#endif
else if (strcasecmp("TypeInstance", child->key) == 0)
status = ts_config_add_string(&data->type_instance, child,
/* may be empty = */ 1);
else if (strcasecmp("MetaData", child->key) == 0)
status = ts_config_add_meta(&data->meta, child,
/* may be empty = */ 1);
else if (strcasecmp("DeleteMetaData", child->key) == 0)
status = ts_config_add_meta_delete(&data->meta_delete, child);
else {
ERROR("Target `set': The `%s' configuration option is not understood "
"and will be ignored.",
child->key);
status = 0;
}
if (status != 0)
break;
}
/* Additional sanity-checking */
while (status == 0) {
if ((data->host == NULL) && (data->plugin == NULL) &&
(data->plugin_instance == NULL)
/* && (data->type == NULL) */
&& (data->type_instance == NULL) && (data->meta == NULL) &&
(data->meta_delete == NULL)) {
ERROR("Target `set': You need to set at least one of `Host', "
"`Plugin', `PluginInstance', `TypeInstance', "
"`MetaData', or `DeleteMetaData'.");
status = -1;
}
if (data->meta != NULL) {
/* If data->meta_delete is NULL, this loop is a no-op. */
for (ts_key_list_t *l = data->meta_delete; l != NULL; l = l->next) {
if (meta_data_type(data->meta, l->key) != 0) {
/* MetaData and DeleteMetaData for the same key. */
ERROR("Target `set': Can only have one of `MetaData' or "
"`DeleteMetaData' for any given key.");
status = -1;
}
}
}
break;
}
if (status != 0) {
ts_destroy((void *)&data);
return (status);
}
*user_data = data;
return (0);
} /* }}} int ts_create */
static int meta_data_keys_to_json(char *buffer, size_t buffer_size, /* {{{ */
meta_data_t *meta, char **keys,
size_t keys_num) {
size_t offset = 0;
int status;
buffer[0] = 0;
#define BUFFER_ADD(...) \
do { \
status = ssnprintf(buffer + offset, buffer_size - offset, __VA_ARGS__); \
if (status < 1) \
return (-1); \
else if (((size_t)status) >= (buffer_size - offset)) \
return (-ENOMEM); \
else \
offset += ((size_t)status); \
} while (0)
for (size_t i = 0; i < keys_num; ++i) {
int type;
char *key = keys[i];
type = meta_data_type(meta, key);
if (type == MD_TYPE_STRING) {
char *value = NULL;
if (meta_data_get_string(meta, key, &value) == 0) {
char temp[512] = "";
status = json_escape_string(temp, sizeof(temp), value);
sfree(value);
if (status != 0)
return status;
BUFFER_ADD(",\"%s\":%s", key, temp);
}
} else if (type == MD_TYPE_SIGNED_INT) {
int64_t value = 0;
if (meta_data_get_signed_int(meta, key, &value) == 0)
BUFFER_ADD(",\"%s\":%" PRIi64, key, value);
} else if (type == MD_TYPE_UNSIGNED_INT) {
uint64_t value = 0;
if (meta_data_get_unsigned_int(meta, key, &value) == 0)
BUFFER_ADD(",\"%s\":%" PRIu64, key, value);
} else if (type == MD_TYPE_DOUBLE) {
double value = 0.0;
if (meta_data_get_double(meta, key, &value) == 0)
BUFFER_ADD(",\"%s\":%f", key, value);
} else if (type == MD_TYPE_BOOLEAN) {
_Bool value = 0;
if (meta_data_get_boolean(meta, key, &value) == 0)
BUFFER_ADD(",\"%s\":%s", key, value ? "true" : "false");
}
} /* for (keys) */
if (offset == 0)
return (ENOENT);
buffer[0] = '{'; /* replace leading ',' */
BUFFER_ADD("}");
#undef BUFFER_ADD
return (0);
} /* }}} int meta_data_keys_to_json */
