1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112
|
/*
* Copyright (c) 2020 Balabit
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.
*
* This library 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
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this library; if not, write to the Free Software
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
*
* As an additional exemption you are allowed to compile & link against the
* OpenSSL libraries as published by the OpenSSL project. See the file
* COPYING for details.
*
*/
typedef struct _IterateState
{
TFSimpleFuncState super;
GMutex mutex;
GString *current;
LogMessageValueType current_type;
LogTemplate *template;
} IterateState;
static gboolean
tf_iterate_prepare(LogTemplateFunction *self, gpointer s, LogTemplate *parent, gint argc, gchar *argv[],
GError **error)
{
IterateState *state = (IterateState *)s;
GOptionContext *ctx = g_option_context_new("iterate");
if (!g_option_context_parse(ctx, &argc, &argv, error))
{
g_option_context_free(ctx);
return FALSE;
}
if (argc != 3)
{
g_set_error(error, LOG_TEMPLATE_ERROR, LOG_TEMPLATE_ERROR_COMPILE,
"Wrong number of arguments. Example: $(iterate template initial-value).\n");
g_option_context_free(ctx);
return FALSE;
}
state->template = log_template_new(configuration, "iterate");
if (!log_template_compile(state->template, argv[1], error))
{
log_template_unref(state->template);
state->template = NULL;
g_option_context_free(ctx);
return FALSE;
}
state->current = g_string_new(argv[2]);
g_option_context_free(ctx);
g_mutex_init(&state->mutex);
return TRUE;
}
static void
update_current(LogTemplateFunction *self, IterateState *state, LogMessage *msg)
{
gchar *current_value = g_strdup(state->current->str);
g_string_assign(state->current, "");
LogTemplateEvalOptions options = {NULL, LTZ_LOCAL, 0, current_value, LM_VT_STRING};
log_template_format_value_and_type(state->template, msg, &options, state->current, &state->current_type);
g_free(current_value);
}
static void
tf_iterate_call(LogTemplateFunction *self, gpointer s, const LogTemplateInvokeArgs *args, GString *result,
LogMessageValueType *type)
{
IterateState *state = (IterateState *)s;
g_mutex_lock(&state->mutex);
g_string_append(result, state->current->str);
*type = state->current_type;
update_current(self, state, args->messages[0]);
g_mutex_unlock(&state->mutex);
}
static void
tf_iterate_free_state(gpointer s)
{
IterateState *state = (IterateState *)s;
log_template_unref(state->template);
state->template = NULL;
g_string_free(state->current, TRUE);
state->current = NULL;
tf_simple_func_free_state(&state->super);
g_mutex_clear(&state->mutex);
}
TEMPLATE_FUNCTION(IterateState, tf_iterate, tf_iterate_prepare, NULL,
tf_iterate_call, tf_iterate_free_state, NULL);
|