1 /*
2 * Copyright (c) 2018 Balabit
3 * Copyright (c) 2018 László Várady <laszlo.varady@balabit.com>
4 *
5 * This program is free software; you can redistribute it and/or modify it
6 * under the terms of the GNU General Public License version 2 as published
7 * by the Free Software Foundation, or (at your option) any later version.
8 *
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
13 *
14 * You should have received a copy of the GNU General Public License
15 * along with this program; if not, write to the Free Software
16 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
17 *
18 * As an additional exemption you are allowed to compile & link against the
19 * OpenSSL libraries as published by the OpenSSL project. See the file
20 * COPYING for details.
21 *
22 */
23
24 #include "msg-generator.h"
25 #include "msg-generator-source.h"
26
27 struct MsgGeneratorSourceDriver
28 {
29 LogSrcDriver super;
30 MsgGeneratorSourceOptions source_options;
31 MsgGeneratorSource *source;
32 };
33
34 static gboolean
msg_generator_sd_init(LogPipe * s)35 msg_generator_sd_init(LogPipe *s)
36 {
37 MsgGeneratorSourceDriver *self = (MsgGeneratorSourceDriver *) s;
38 GlobalConfig *cfg = log_pipe_get_config(s);
39
40 if (!log_src_driver_init_method(s))
41 return FALSE;
42
43 self->source = msg_generator_source_new(cfg);
44
45 msg_generator_source_options_init(&self->source_options, cfg, self->super.super.group);
46 msg_generator_source_set_options(self->source, &self->source_options, self->super.super.id, NULL, FALSE, FALSE,
47 self->super.super.super.expr_node);
48
49 log_pipe_append((LogPipe *) self->source, &self->super.super.super);
50
51 if (!msg_generator_source_init(self->source))
52 {
53 msg_generator_source_free(self->source);
54 self->source = NULL;
55 return FALSE;
56 }
57
58 return TRUE;
59 }
60
61 static gboolean
msg_generator_sd_deinit(LogPipe * s)62 msg_generator_sd_deinit(LogPipe *s)
63 {
64 MsgGeneratorSourceDriver *self = (MsgGeneratorSourceDriver *) s;
65
66 msg_generator_source_deinit(self->source);
67 msg_generator_source_free(self->source);
68 self->source = NULL;
69
70 return log_src_driver_deinit_method(s);
71 }
72
73 static void
msg_generator_sd_free(LogPipe * s)74 msg_generator_sd_free(LogPipe *s)
75 {
76 MsgGeneratorSourceDriver *self = (MsgGeneratorSourceDriver *) s;
77 g_hash_table_unref(self->source_options.name_value);
78 msg_generator_source_options_destroy(&self->source_options);
79 log_src_driver_free(s);
80 }
81
82 MsgGeneratorSourceOptions *
msg_generator_sd_get_source_options(LogDriver * s)83 msg_generator_sd_get_source_options(LogDriver *s)
84 {
85 MsgGeneratorSourceDriver *self = (MsgGeneratorSourceDriver *) s;
86
87 return &self->source_options;
88 }
89
90 LogDriver *
msg_generator_sd_new(GlobalConfig * cfg)91 msg_generator_sd_new(GlobalConfig *cfg)
92 {
93 MsgGeneratorSourceDriver *self = g_new0(MsgGeneratorSourceDriver, 1);
94 log_src_driver_init_instance(&self->super, cfg);
95 msg_generator_source_options_defaults(&self->source_options);
96 self->source_options.name_value = g_hash_table_new_full(g_str_hash, g_str_equal, g_free,
97 (GDestroyNotify) log_template_unref);
98 self->super.super.super.init = msg_generator_sd_init;
99 self->super.super.super.deinit = msg_generator_sd_deinit;
100 self->super.super.super.free_fn = msg_generator_sd_free;
101
102 return &self->super.super;
103 }
104