From c40a839e2d770efe8330757f764385c4c2ffb86e Mon Sep 17 00:00:00 2001 From: Igor Peshansky Date: Fri, 2 Sep 2016 17:47:23 -0400 Subject: [PATCH] Allow replacing within and deleting metadata keys. --- src/target_replace.c | 253 ++++++++++++++++++++++++++++++++++++++++++- 1 file changed, 251 insertions(+), 2 deletions(-) diff --git a/src/target_replace.c b/src/target_replace.c index 40a6fec9..975acbce 100644 --- a/src/target_replace.c +++ b/src/target_replace.c @@ -43,6 +43,17 @@ struct tr_action_s tr_action_t *next; }; +struct tr_meta_data_action_s; +typedef struct tr_meta_data_action_s tr_meta_data_action_t; +struct tr_meta_data_action_s +{ + char *key; + regex_t re; + char *replacement; + + tr_meta_data_action_t *next; +}; + struct tr_data_s { tr_action_t *host; @@ -50,6 +61,7 @@ struct tr_data_s tr_action_t *plugin_instance; /* tr_action_t *type; */ tr_action_t *type_instance; + tr_meta_data_action_t *meta; }; typedef struct tr_data_s tr_data_t; @@ -85,6 +97,21 @@ static void tr_action_destroy (tr_action_t *act) /* {{{ */ sfree (act); } /* }}} void tr_action_destroy */ +static void tr_meta_data_action_destroy (tr_meta_data_action_t *act) /* {{{ */ +{ + if (act == NULL) + return; + + sfree (act->key); + regfree (&act->re); + sfree (act->replacement); + + if (act->next != NULL) + tr_meta_data_action_destroy (act->next); + + sfree (act); +} /* }}} void tr_meta_data_action_destroy */ + static int tr_config_add_action (tr_action_t **dest, /* {{{ */ const oconfig_item_t *ci, int may_be_empty) { @@ -153,6 +180,108 @@ static int tr_config_add_action (tr_action_t **dest, /* {{{ */ return (0); } /* }}} int tr_config_add_action */ +static int tr_config_add_meta_action (tr_meta_data_action_t **dest, /* {{{ */ + const oconfig_item_t *ci, int should_delete) +{ + tr_meta_data_action_t *act; + int status; + + if (dest == NULL) + return (-EINVAL); + + if (should_delete) + { + if ((ci->values_num != 2) + || (ci->values[0].type != OCONFIG_TYPE_STRING) + || (ci->values[1].type != OCONFIG_TYPE_STRING)) + { + ERROR ("Target `replace': The `%s' option requires exactly two string " + "arguments.", ci->key); + return (-1); + } + } + else + { + if ((ci->values_num != 3) + || (ci->values[0].type != OCONFIG_TYPE_STRING) + || (ci->values[1].type != OCONFIG_TYPE_STRING) + || (ci->values[2].type != OCONFIG_TYPE_STRING)) + { + ERROR ("Target `replace': The `%s' option requires exactly three string " + "arguments.", ci->key); + return (-1); + } + } + + if (strlen (ci->values[0].value.string) == 0) + { + ERROR ("Target `replace': The `%s' option does not accept empty string as " + "first argument.", ci->key); + return (-1); + } + + act = calloc (1, sizeof (*act)); + if (act == NULL) + { + ERROR ("tr_config_add_meta_action: calloc failed."); + return (-ENOMEM); + } + + act->key = NULL; + act->replacement = NULL; + + act->key = tr_strdup (ci->values[0].value.string); + if (act->key == NULL) + { + ERROR ("tr_config_add_meta_action: tr_strdup failed."); + sfree (act); + return (-ENOMEM); + } + + status = regcomp (&act->re, ci->values[1].value.string, REG_EXTENDED); + if (status != 0) + { + char errbuf[1024] = ""; + + /* regerror assures null termination. */ + regerror (status, &act->re, errbuf, sizeof (errbuf)); + ERROR ("Target `replace': Compiling the regular expression `%s' " + "failed: %s.", + ci->values[1].value.string, errbuf); + sfree (act->key); + sfree (act); + return (-EINVAL); + } + + if (!should_delete) { + act->replacement = tr_strdup (ci->values[2].value.string); + if (act->replacement == NULL) + { + ERROR ("tr_config_add_meta_action: tr_strdup failed."); + sfree (act->key); + regfree (&act->re); + sfree (act); + return (-ENOMEM); + } + } + + /* Insert action at end of list. */ + if (*dest == NULL) + *dest = act; + else + { + tr_meta_data_action_t *prev; + + prev = *dest; + while (prev->next != NULL) + prev = prev->next; + + prev->next = act; + } + + return (0); +} /* }}} int tr_config_add_meta_action */ + static int tr_action_invoke (tr_action_t *act_head, /* {{{ */ char *buffer_in, size_t buffer_in_size, int may_be_empty) { @@ -215,6 +344,112 @@ static int tr_action_invoke (tr_action_t *act_head, /* {{{ */ return (0); } /* }}} int tr_action_invoke */ +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; + + value_type = meta_data_type (*dest, act->key); + if (value_type == 0) /* not found */ + continue; + if (value_type != MD_TYPE_STRING) + { + ERROR ("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) + 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); + continue; + } + + if (act->replacement != NULL) + { + meta_data_t *result; + + 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); + 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); + 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); + return (meta_data_status); + } + + meta_data_clone_merge (dest, result); + meta_data_destroy (result); + } + else /* no replacement; delete the key */ + { + DEBUG ("target_replace plugin: tr_meta_data_action_invoke: " + "deleting `%s'", act->key); + meta_data_delete (*dest, act->key); + } + } /* for (act = act_head; act != NULL; act = act->next) */ + + return (0); +} /* }}} int tr_meta_data_action_invoke */ + static int tr_destroy (void **user_data) /* {{{ */ { tr_data_t *data; @@ -231,6 +466,7 @@ static int tr_destroy (void **user_data) /* {{{ */ tr_action_destroy (data->plugin_instance); /* tr_action_destroy (data->type); */ tr_action_destroy (data->type_instance); + tr_meta_data_action_destroy (data->meta); sfree (data); return (0); @@ -253,6 +489,7 @@ static int tr_create (const oconfig_item_t *ci, void **user_data) /* {{{ */ data->plugin_instance = NULL; /* data->type = NULL; */ data->type_instance = NULL; + data->meta = NULL; status = 0; for (int i = 0; i < ci->children_num; i++) @@ -277,6 +514,12 @@ static int tr_create (const oconfig_item_t *ci, void **user_data) /* {{{ */ else if (strcasecmp ("TypeInstance", child->key) == 0) status = tr_config_add_action (&data->type_instance, child, /* may be empty = */ 1); + else if (strcasecmp ("MetaData", child->key) == 0) + status = tr_config_add_meta_action (&data->meta, child, + /* should delete = */ 0); + else if (strcasecmp ("DeleteMetaData", child->key) == 0) + status = tr_config_add_meta_action (&data->meta, child, + /* should delete = */ 1); else { ERROR ("Target `replace': The `%s' configuration option is not understood " @@ -295,7 +538,8 @@ static int tr_create (const oconfig_item_t *ci, void **user_data) /* {{{ */ && (data->plugin == NULL) && (data->plugin_instance == NULL) /* && (data->type == NULL) */ - && (data->type_instance == NULL)) + && (data->type_instance == NULL) + && (data->meta == NULL)) { ERROR ("Target `replace': You need to set at least one of `Host', " "`Plugin', `PluginInstance' or `TypeInstance'."); @@ -330,13 +574,18 @@ static int tr_invoke (const data_set_t *ds, value_list_t *vl, /* {{{ */ return (-EINVAL); } + if (data->meta != NULL) + { + tr_meta_data_action_invoke (data->meta, &(vl->meta)); + } + #define HANDLE_FIELD(f,e) \ if (data->f != NULL) \ tr_action_invoke (data->f, vl->f, sizeof (vl->f), e) HANDLE_FIELD (host, 0); HANDLE_FIELD (plugin, 0); HANDLE_FIELD (plugin_instance, 1); - /* HANDLE_FIELD (type); */ + /* HANDLE_FIELD (type, 0); */ HANDLE_FIELD (type_instance, 1); return (FC_TARGET_CONTINUE); -- 2.30.2