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"
32 #include "lib/util/time.h"
34 #include "ctdb_private.h"
35 #include "ctdb_client.h"
37 #include "common/system.h"
38 #include "common/common.h"
39 #include "common/logging.h"
41 const char *debug_extra
= "";
43 struct ctdb_log_backend
{
44 struct ctdb_log_backend
*prev
, *next
;
46 ctdb_log_setup_fn_t setup
;
49 struct ctdb_log_state
{
54 void (*logfn
)(const char *, uint16_t, void *);
56 struct ctdb_log_backend
*backends
;
59 /* Used by ctdb_set_child_logging() */
60 static struct ctdb_log_state
*log_state
;
62 void ctdb_log_register_backend(const char *prefix
, ctdb_log_setup_fn_t setup
)
64 struct ctdb_log_backend
*b
;
66 b
= talloc_zero(log_state
, struct ctdb_log_backend
);
68 printf("Failed to register backend \"%s\" - no memory\n",
76 DLIST_ADD_END(log_state
->backends
, b
);
80 /* Initialise logging */
81 bool ctdb_logging_init(TALLOC_CTX
*mem_ctx
, const char *logging
)
83 struct ctdb_log_backend
*b
;
86 log_state
= talloc_zero(mem_ctx
, struct ctdb_log_state
);
87 if (log_state
== NULL
) {
88 printf("talloc_zero failed\n");
93 ctdb_log_init_syslog();
95 for (b
= log_state
->backends
; b
!= NULL
; b
= b
->next
) {
96 size_t l
= strlen(b
->prefix
);
97 /* Exact match with prefix or prefix followed by ':' */
98 if (strncmp(b
->prefix
, logging
, l
) == 0 &&
99 (logging
[l
] == '\0' || logging
[l
] == ':')) {
100 ret
= b
->setup(mem_ctx
, logging
, "ctdbd");
104 printf("Log init for \"%s\" failed with \"%s\"\n",
105 logging
, strerror(ret
));
110 printf("Unable to find log backend for \"%s\"\n", logging
);
114 /* Note that do_debug always uses the global log state. */
115 static void write_to_log(struct ctdb_log_state
*log
,
116 const char *buf
, unsigned int len
)
118 if (script_log_level
<= DEBUGLEVEL
) {
119 if (log
!= NULL
&& log
->prefix
!= NULL
) {
120 dbgtext("%s: %*.*s\n", log
->prefix
, len
, len
, buf
);
122 dbgtext("%*.*s\n", len
, len
, buf
);
124 /* log it in the eventsystem as well */
125 if (log
&& log
->logfn
) {
126 log
->logfn(log
->buf
, len
, log
->logfn_private
);
132 called when log data comes in from a child process
134 static void ctdb_child_log_handler(struct tevent_context
*ev
,
135 struct tevent_fd
*fde
,
136 uint16_t flags
, void *private)
138 struct ctdb_log_state
*log
= talloc_get_type(private, struct ctdb_log_state
);
142 if (!(flags
& TEVENT_FD_READ
)) {
146 n
= sys_read(log
->pfd
, &log
->buf
[log
->buf_used
],
147 sizeof(log
->buf
) - log
->buf_used
);
151 if (log
!= log_state
) {
157 while (log
->buf_used
> 0 &&
158 (p
= memchr(log
->buf
, '\n', log
->buf_used
)) != NULL
) {
159 int n1
= (p
- log
->buf
)+1;
161 /* swallow \r from child processes */
162 if (n2
> 0 && log
->buf
[n2
-1] == '\r') {
165 write_to_log(log
, log
->buf
, n2
);
166 memmove(log
->buf
, p
+1, sizeof(log
->buf
) - n1
);
170 /* the buffer could have completely filled - unfortunately we have
171 no choice but to dump it out straight away */
172 if (log
->buf_used
== sizeof(log
->buf
)) {
173 write_to_log(log
, log
->buf
, log
->buf_used
);
178 static int log_context_destructor(struct ctdb_log_state
*log
)
180 /* Flush buffer in case it wasn't \n-terminated. */
181 if (log
->buf_used
> 0) {
182 write_to_log(log
, log
->buf
, log
->buf_used
);
188 * vfork + exec, redirecting child output to logging and specified callback.
190 struct ctdb_log_state
*ctdb_vfork_with_logging(TALLOC_CTX
*mem_ctx
,
191 struct ctdb_context
*ctdb
,
192 const char *log_prefix
,
195 const char **helper_argv
,
196 void (*logfn
)(const char *, uint16_t, void *),
197 void *logfn_private
, pid_t
*pid
)
200 struct ctdb_log_state
*log
;
201 struct tevent_fd
*fde
;
204 struct timeval before
;
207 log
= talloc_zero(mem_ctx
, struct ctdb_log_state
);
208 CTDB_NO_MEMORY_NULL(ctdb
, log
);
210 log
->prefix
= log_prefix
;
212 log
->logfn_private
= logfn_private
;
215 DEBUG(DEBUG_ERR
, (__location__
" Failed to setup pipe for child logging\n"));
219 argv
= talloc_array(mem_ctx
, char *, helper_argc
+ 2);
221 DEBUG(DEBUG_ERR
, (__location__
"Failed to allocate memory for helper\n"));
224 argv
[0] = discard_const(helper
);
225 argv
[1] = talloc_asprintf(argv
, "%d", p
[1]);
226 if (argv
[1] == NULL
) {
227 DEBUG(DEBUG_ERR
, (__location__
"Failed to allocate memory for helper\n"));
232 for (i
=0; i
<helper_argc
; i
++) {
233 argv
[i
+2] = discard_const(helper_argv
[i
]);
236 before
= timeval_current();
246 DEBUG(DEBUG_ERR
, (__location__
"vfork failed for helper process\n"));
251 delta_t
= timeval_elapsed(&before
);
253 DEBUG(DEBUG_WARNING
, ("vfork() took %lf seconds\n", delta_t
));
256 ctdb_track_child(ctdb
, *pid
);
259 set_close_on_exec(log
->pfd
);
260 talloc_set_destructor(log
, log_context_destructor
);
261 fde
= tevent_add_fd(ctdb
->ev
, log
, log
->pfd
, TEVENT_FD_READ
,
262 ctdb_child_log_handler
, log
);
263 tevent_fd_set_auto_close(fde
);
274 setup for logging of child process stdout
276 int ctdb_set_child_logging(struct ctdb_context
*ctdb
)
279 int old_stdout
, old_stderr
;
280 struct tevent_fd
*fde
;
282 if (log_state
->fd
== STDOUT_FILENO
) {
283 /* not needed for stdout logging */
287 /* setup a pipe to catch IO from subprocesses */
289 DEBUG(DEBUG_ERR
,(__location__
" Failed to setup for child logging pipe\n"));
293 /* We'll fail if stderr/stdout not already open; it's simpler. */
294 old_stdout
= dup(STDOUT_FILENO
);
295 if (old_stdout
< 0) {
296 DEBUG(DEBUG_ERR
, ("Failed to dup stdout for child logging\n"));
299 old_stderr
= dup(STDERR_FILENO
);
300 if (old_stderr
< 0) {
301 DEBUG(DEBUG_ERR
, ("Failed to dup stderr for child logging\n"));
305 if (dup2(p
[1], STDOUT_FILENO
) < 0 || dup2(p
[1], STDERR_FILENO
) < 0) {
306 int saved_errno
= errno
;
307 dup2(old_stdout
, STDOUT_FILENO
);
308 dup2(old_stderr
, STDERR_FILENO
);
315 printf(__location__
" dup2 failed: %s\n",
323 fde
= tevent_add_fd(ctdb
->ev
, log_state
, p
[0], TEVENT_FD_READ
,
324 ctdb_child_log_handler
, log_state
);
325 tevent_fd_set_auto_close(fde
);
327 log_state
->pfd
= p
[0];
329 DEBUG(DEBUG_DEBUG
, (__location__
" Created PIPE FD:%d for logging\n", p
[0]));
336 * set up a log handler to catch logging from TEVENT
338 static void ctdb_tevent_logging(void *private_data
,
339 enum tevent_debug_level level
,
341 va_list ap
) PRINTF_ATTRIBUTE(3, 0);
342 static void ctdb_tevent_logging(void *private_data
,
343 enum tevent_debug_level level
,
347 enum debug_level lvl
= DEBUG_CRIT
;
350 case TEVENT_DEBUG_FATAL
:
353 case TEVENT_DEBUG_ERROR
:
356 case TEVENT_DEBUG_WARNING
:
359 case TEVENT_DEBUG_TRACE
:
364 if (lvl
<= DEBUGLEVEL
) {
369 int ctdb_init_tevent_logging(struct ctdb_context
*ctdb
)
373 ret
= tevent_set_debug(ctdb
->ev
,