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
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
|
/*
* Copyright (c) 2019 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.
*
*/
#include "commands.h"
#include "control-client.h"
#include <string.h>
#include <stdio.h>
static ControlClient *control_client;
GOptionEntry no_options[] =
{
{ NULL, 0, 0, G_OPTION_ARG_NONE, NULL, NULL }
};
gboolean
is_syslog_ng_running(void)
{
return control_client_connect(control_client);
}
gint
process_response_status(GString *response)
{
if (strncmp(response->str, "FAIL ", 5) == 0)
{
g_string_erase(response, 0, 5);
return 1;
}
else if (strncmp(response->str, "OK ", 3) == 0)
{
g_string_erase(response, 0, 3);
return 0;
}
return 0;
}
static gboolean
slng_send_cmd(const gchar *cmd)
{
if (!control_client_connect(control_client))
{
return FALSE;
}
if (control_client_send_command(control_client, cmd) < 0)
{
return FALSE;
}
return TRUE;
}
gint
slng_run_command(const gchar *command, CommandResponseHandlerFunc cb, gpointer user_data)
{
if (!slng_send_cmd(command))
return 1;
return control_client_read_reply(control_client, cb, user_data);
}
static gboolean
_is_response_empty(GString *response)
{
return (response == NULL || g_str_equal(response->str, ""));
}
static gint
_print_reply_to_stdout(GString *reply, gpointer user_data)
{
gboolean first_response = *((gboolean *)user_data);
gint retval = 0;
if (first_response)
{
if (_is_response_empty(reply))
retval = 1;
else retval = process_response_status(reply);
}
printf("%s", reply->str);
return retval;
}
gint
dispatch_command(const gchar *cmd)
{
gboolean first_response = TRUE;
gint retval = 0;
gchar *dispatchable_command = g_strdup_printf("%s\n", cmd);
retval = slng_run_command(dispatchable_command, _print_reply_to_stdout, &first_response);
secret_storage_wipe(dispatchable_command, strlen(dispatchable_command));
g_free(dispatchable_command);
return retval;
}
gint
run(const gchar *control_name, gint argc, gchar **argv, CommandDescriptor *mode, GOptionContext *ctx)
{
control_client = control_client_new(control_name);
gint result = mode->main(argc, argv, mode->mode, ctx);
control_client_free(control_client);
return result;
}
|
