4 Copyright (C) Andrew Tridgell 2008
6 This program is free software; you can redistribute it and/or modify
7 it under the terms of the GNU General Public License as published by
8 the Free Software Foundation; either version 3 of the License, or
9 (at your option) any later version.
11 This program is distributed in the hope that it will be useful,
12 but WITHOUT ANY WARRANTY; without even the implied warranty of
13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 GNU General Public License for more details.
16 You should have received a copy of the GNU General Public License
17 along with this program; if not, see <http://www.gnu.org/licenses/>.
21 #include "system/filesys.h"
22 #include "system/network.h"
23 #include "system/syslog.h"
24 #include "system/time.h"
29 #include "lib/util/dlinklist.h"
30 #include "lib/util/debug.h"
31 #include "lib/util/blocking.h"
33 #include "ctdb_private.h"
34 #include "ctdb_client.h"
36 #include "common/system.h"
37 #include "common/common.h"
38 #include "common/logging.h"
40 const char *debug_extra
= "";
42 struct ctdb_log_backend
{
43 struct ctdb_log_backend
*prev
, *next
;
45 ctdb_log_setup_fn_t setup
;
48 struct ctdb_log_state
{
53 void (*logfn
)(const char *, uint16_t, void *);
55 struct ctdb_log_backend
*backends
;
58 /* Used by ctdb_set_child_logging() */
59 static struct ctdb_log_state
*log_state
;
61 void ctdb_log_register_backend(const char *prefix
, ctdb_log_setup_fn_t setup
)
63 struct ctdb_log_backend
*b
;
65 b
= talloc_zero(log_state
, struct ctdb_log_backend
);
67 printf("Failed to register backend \"%s\" - no memory\n",
75 DLIST_ADD_END(log_state
->backends
, b
);
79 /* Initialise logging */
80 bool ctdb_logging_init(TALLOC_CTX
*mem_ctx
, const char *logging
)
82 struct ctdb_log_backend
*b
;
85 log_state
= talloc_zero(mem_ctx
, struct ctdb_log_state
);
86 if (log_state
== NULL
) {
87 printf("talloc_zero failed\n");
92 ctdb_log_init_syslog();
94 for (b
= log_state
->backends
; b
!= NULL
; b
= b
->next
) {
95 size_t l
= strlen(b
->prefix
);
96 /* Exact match with prefix or prefix followed by ':' */
97 if (strncmp(b
->prefix
, logging
, l
) == 0 &&
98 (logging
[l
] == '\0' || logging
[l
] == ':')) {
99 ret
= b
->setup(mem_ctx
, logging
, "ctdbd");
103 printf("Log init for \"%s\" failed with \"%s\"\n",
104 logging
, strerror(ret
));
109 printf("Unable to find log backend for \"%s\"\n", logging
);
113 /* Note that do_debug always uses the global log state. */
114 static void write_to_log(struct ctdb_log_state
*log
,
115 const char *buf
, unsigned int len
)
117 if (script_log_level
<= DEBUGLEVEL
) {
118 if (log
!= NULL
&& log
->prefix
!= NULL
) {
119 dbgtext("%s: %*.*s\n", log
->prefix
, len
, len
, buf
);
121 dbgtext("%*.*s\n", len
, len
, buf
);
123 /* log it in the eventsystem as well */
124 if (log
&& log
->logfn
) {
125 log
->logfn(log
->buf
, len
, log
->logfn_private
);
131 called when log data comes in from a child process
133 static void ctdb_child_log_handler(struct tevent_context
*ev
,
134 struct tevent_fd
*fde
,
135 uint16_t flags
, void *private)
137 struct ctdb_log_state
*log
= talloc_get_type(private, struct ctdb_log_state
);
141 if (!(flags
& TEVENT_FD_READ
)) {
145 n
= sys_read(log
->pfd
, &log
->buf
[log
->buf_used
],
146 sizeof(log
->buf
) - log
->buf_used
);
150 if (log
!= log_state
) {
156 while (log
->buf_used
> 0 &&
157 (p
= memchr(log
->buf
, '\n', log
->buf_used
)) != NULL
) {
158 int n1
= (p
- log
->buf
)+1;
160 /* swallow \r from child processes */
161 if (n2
> 0 && log
->buf
[n2
-1] == '\r') {
164 write_to_log(log
, log
->buf
, n2
);
165 memmove(log
->buf
, p
+1, sizeof(log
->buf
) - n1
);
169 /* the buffer could have completely filled - unfortunately we have
170 no choice but to dump it out straight away */
171 if (log
->buf_used
== sizeof(log
->buf
)) {
172 write_to_log(log
, log
->buf
, log
->buf_used
);
177 static int log_context_destructor(struct ctdb_log_state
*log
)
179 /* Flush buffer in case it wasn't \n-terminated. */
180 if (log
->buf_used
> 0) {
181 write_to_log(log
, log
->buf
, log
->buf_used
);
187 * vfork + exec, redirecting child output to logging and specified callback.
189 struct ctdb_log_state
*ctdb_vfork_with_logging(TALLOC_CTX
*mem_ctx
,
190 struct ctdb_context
*ctdb
,
191 const char *log_prefix
,
194 const char **helper_argv
,
195 void (*logfn
)(const char *, uint16_t, void *),
196 void *logfn_private
, pid_t
*pid
)
199 struct ctdb_log_state
*log
;
200 struct tevent_fd
*fde
;
204 log
= talloc_zero(mem_ctx
, struct ctdb_log_state
);
205 CTDB_NO_MEMORY_NULL(ctdb
, log
);
207 log
->prefix
= log_prefix
;
209 log
->logfn_private
= logfn_private
;
212 DEBUG(DEBUG_ERR
, (__location__
" Failed to setup pipe for child logging\n"));
216 argv
= talloc_array(mem_ctx
, char *, helper_argc
+ 2);
218 DEBUG(DEBUG_ERR
, (__location__
"Failed to allocate memory for helper\n"));
221 argv
[0] = discard_const(helper
);
222 argv
[1] = talloc_asprintf(argv
, "%d", p
[1]);
223 if (argv
[1] == NULL
) {
224 DEBUG(DEBUG_ERR
, (__location__
"Failed to allocate memory for helper\n"));
229 for (i
=0; i
<helper_argc
; i
++) {
230 argv
[i
+2] = discard_const(helper_argv
[i
]);
241 DEBUG(DEBUG_ERR
, (__location__
"vfork failed for helper process\n"));
246 ctdb_track_child(ctdb
, *pid
);
249 set_close_on_exec(log
->pfd
);
250 talloc_set_destructor(log
, log_context_destructor
);
251 fde
= tevent_add_fd(ctdb
->ev
, log
, log
->pfd
, TEVENT_FD_READ
,
252 ctdb_child_log_handler
, log
);
253 tevent_fd_set_auto_close(fde
);
264 setup for logging of child process stdout
266 int ctdb_set_child_logging(struct ctdb_context
*ctdb
)
269 int old_stdout
, old_stderr
;
270 struct tevent_fd
*fde
;
272 if (log_state
->fd
== STDOUT_FILENO
) {
273 /* not needed for stdout logging */
277 /* setup a pipe to catch IO from subprocesses */
279 DEBUG(DEBUG_ERR
,(__location__
" Failed to setup for child logging pipe\n"));
283 /* We'll fail if stderr/stdout not already open; it's simpler. */
284 old_stdout
= dup(STDOUT_FILENO
);
285 old_stderr
= dup(STDERR_FILENO
);
286 if (old_stdout
< 0 || old_stderr
< 0) {
287 DEBUG(DEBUG_ERR
, ("Failed to dup stdout/stderr for child logging\n"));
290 if (dup2(p
[1], STDOUT_FILENO
) < 0 || dup2(p
[1], STDERR_FILENO
) < 0) {
291 int saved_errno
= errno
;
292 dup2(old_stdout
, STDOUT_FILENO
);
293 dup2(old_stderr
, STDERR_FILENO
);
300 printf(__location__
" dup2 failed: %s\n",
308 fde
= tevent_add_fd(ctdb
->ev
, log_state
, p
[0], TEVENT_FD_READ
,
309 ctdb_child_log_handler
, log_state
);
310 tevent_fd_set_auto_close(fde
);
312 log_state
->pfd
= p
[0];
314 DEBUG(DEBUG_DEBUG
, (__location__
" Created PIPE FD:%d for logging\n", p
[0]));
321 * set up a log handler to catch logging from TEVENT
323 static void ctdb_tevent_logging(void *private_data
,
324 enum tevent_debug_level level
,
326 va_list ap
) PRINTF_ATTRIBUTE(3, 0);
327 static void ctdb_tevent_logging(void *private_data
,
328 enum tevent_debug_level level
,
332 enum debug_level lvl
= DEBUG_CRIT
;
335 case TEVENT_DEBUG_FATAL
:
338 case TEVENT_DEBUG_ERROR
:
341 case TEVENT_DEBUG_WARNING
:
344 case TEVENT_DEBUG_TRACE
:
349 if (lvl
<= DEBUGLEVEL
) {
354 int ctdb_init_tevent_logging(struct ctdb_context
*ctdb
)
358 ret
= tevent_set_debug(ctdb
->ev
,