File: tf-iterate.c

package info (click to toggle)
syslog-ng 4.8.1-6
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 20,456 kB
  • sloc: ansic: 177,631; python: 13,035; cpp: 11,611; makefile: 7,012; sh: 5,147; java: 3,651; xml: 3,344; yacc: 1,377; lex: 599; perl: 193; awk: 190; objc: 162
file content (112 lines) | stat: -rw-r--r-- 3,405 bytes parent folder | download | duplicates (3)
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);