4 Copyright (C) Andrew Tridgell 2007
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/wait.h"
24 #include "system/dir.h"
25 #include "system/locale.h"
26 #include "system/time.h"
31 #include "lib/util/dlinklist.h"
32 #include "lib/util/debug.h"
33 #include "lib/util/samba_util.h"
35 #include "ctdb_private.h"
37 #include "common/rb_tree.h"
38 #include "common/system.h"
39 #include "common/common.h"
40 #include "common/logging.h"
43 static void ctdb_event_script_timeout(struct tevent_context
*ev
,
44 struct tevent_timer
*te
,
45 struct timeval t
, void *p
);
47 /* This is attached to the event script state. */
48 struct event_script_callback
{
49 struct event_script_callback
*next
, *prev
;
50 struct ctdb_context
*ctdb
;
52 /* Warning: this can free us! */
53 void (*fn
)(struct ctdb_context
*, int, void *);
57 struct ctdb_event_script_state
{
58 struct ctdb_context
*ctdb
;
59 struct event_script_callback
*callback
;
64 struct timeval timeout
;
67 struct ctdb_script_list_old
*scripts
;
70 static struct ctdb_script
*get_current_script(struct ctdb_event_script_state
*state
)
72 return &state
->scripts
->scripts
[state
->current
];
75 /* called from ctdb_logging when we have received output on STDERR from
76 * one of the eventscripts
78 static void log_event_script_output(const char *str
, uint16_t len
, void *p
)
80 struct ctdb_event_script_state
*state
81 = talloc_get_type(p
, struct ctdb_event_script_state
);
82 struct ctdb_script
*current
;
83 unsigned int slen
, min
;
85 /* We may have been aborted to run something else. Discard */
86 if (state
->scripts
== NULL
) {
90 current
= get_current_script(state
);
92 /* Append, but don't overfill buffer. It starts zero-filled. */
93 slen
= strlen(current
->output
);
94 min
= MIN(len
, sizeof(current
->output
) - slen
- 1);
96 memcpy(current
->output
+ slen
, str
, min
);
99 int32_t ctdb_control_get_event_script_status(struct ctdb_context
*ctdb
,
103 if (call_type
>= CTDB_EVENT_MAX
) {
107 if (ctdb
->last_status
[call_type
] == NULL
) {
108 /* If it's never been run, return nothing so they can tell. */
111 outdata
->dsize
= talloc_get_size(ctdb
->last_status
[call_type
]);
112 outdata
->dptr
= (uint8_t *)ctdb
->last_status
[call_type
];
117 /* To ignore directory entry return 0, else return non-zero */
118 static int script_filter(const struct dirent
*de
)
120 int namelen
= strlen(de
->d_name
);
122 /* Ignore . and .. */
127 /* Skip temporary files left behind by emacs */
128 if (de
->d_name
[namelen
-1] == '~') {
132 /* Filename should start with [0-9][0-9]. */
133 if (!isdigit(de
->d_name
[0]) || !isdigit(de
->d_name
[1]) ||
134 de
->d_name
[2] != '.') {
138 if (namelen
> MAX_SCRIPT_NAME
) {
145 /* Return true if OK, otherwise set errno. */
146 static bool check_executable(const char *dir
, const char *name
)
151 full
= talloc_asprintf(NULL
, "%s/%s", dir
, name
);
155 if (stat(full
, &st
) != 0) {
156 DEBUG(DEBUG_ERR
,("Could not stat event script %s: %s\n",
157 full
, strerror(errno
)));
162 if (!(st
.st_mode
& S_IXUSR
)) {
163 DEBUG(DEBUG_DEBUG
,("Event script %s is not executable. Ignoring this event script\n", full
));
173 static struct ctdb_script_list_old
*ctdb_get_script_list(
174 struct ctdb_context
*ctdb
,
177 struct dirent
**namelist
;
178 struct ctdb_script_list_old
*scripts
;
181 /* scan all directory entries and insert all valid scripts into the
184 count
= scandir(ctdb
->event_script_dir
, &namelist
, script_filter
, alphasort
);
186 DEBUG(DEBUG_CRIT
, ("Failed to read event script directory '%s' - %s\n",
187 ctdb
->event_script_dir
, strerror(errno
)));
191 /* Overallocates by one, but that's OK */
192 scripts
= talloc_zero_size(mem_ctx
,
194 + sizeof(scripts
->scripts
[0]) * count
);
195 if (scripts
== NULL
) {
196 DEBUG(DEBUG_ERR
, (__location__
" Failed to allocate scripts\n"));
199 scripts
->num_scripts
= count
;
201 for (i
= 0; i
< count
; i
++) {
202 struct ctdb_script
*s
= &scripts
->scripts
[i
];
204 if (strlcpy(s
->name
, namelist
[i
]->d_name
, sizeof(s
->name
)) >=
206 s
->status
= -ENAMETOOLONG
;
211 if (!check_executable(ctdb
->event_script_dir
,
212 namelist
[i
]->d_name
)) {
218 for (i
=0; i
<count
; i
++) {
226 /* There cannot be more than 10 arguments to command helper. */
227 #define MAX_HELPER_ARGS (10)
229 static bool child_helper_args(TALLOC_CTX
*mem_ctx
, struct ctdb_context
*ctdb
,
230 enum ctdb_event call
,
232 struct ctdb_script
*current
, int fd
,
233 int *argc
, const char ***argv
)
237 char *t
, *saveptr
, *opt
;
239 tmp
= talloc_array(mem_ctx
, const char *, 10+1);
240 if (tmp
== NULL
) goto failed
;
242 tmp
[0] = talloc_asprintf(tmp
, "%d", fd
);
243 tmp
[1] = talloc_asprintf(tmp
, "%s/%s", ctdb
->event_script_dir
, current
->name
);
244 tmp
[2] = talloc_asprintf(tmp
, "%s", ctdb_eventscript_call_names
[call
]);
247 /* Split options into individual arguments */
248 opt
= talloc_strdup(mem_ctx
, options
);
253 t
= strtok_r(opt
, " ", &saveptr
);
255 tmp
[n
++] = talloc_strdup(tmp
, t
);
256 if (n
> MAX_HELPER_ARGS
) {
259 t
= strtok_r(NULL
, " ", &saveptr
);
262 for (i
=0; i
<n
; i
++) {
263 if (tmp
[i
] == NULL
) {
268 /* Last argument should be NULL */
277 DEBUG(DEBUG_ERR
, (__location__
" too many arguments '%s' to eventscript '%s'\n",
278 options
, ctdb_eventscript_call_names
[call
]));
288 static void ctdb_event_script_handler(struct tevent_context
*ev
,
289 struct tevent_fd
*fde
,
290 uint16_t flags
, void *p
);
292 static char helper_prog
[PATH_MAX
+1] = "";
294 static int fork_child_for_script(struct ctdb_context
*ctdb
,
295 struct ctdb_event_script_state
*state
)
298 struct tevent_fd
*fde
;
299 struct ctdb_script
*current
= get_current_script(state
);
303 if (!ctdb_set_helper("event helper", helper_prog
, sizeof(helper_prog
),
305 CTDB_HELPER_BINDIR
, "ctdb_event_helper")) {
306 ctdb_die(ctdb
, __location__
307 " Unable to set event helper\n");
310 current
->start
= timeval_current();
314 DEBUG(DEBUG_ERR
, (__location__
" pipe failed for child eventscript process\n"));
318 /* Arguments for helper */
319 if (!child_helper_args(state
, ctdb
, state
->call
, state
->options
, current
,
320 state
->fd
[1], &argc
, &argv
)) {
321 DEBUG(DEBUG_ERR
, (__location__
" failed to create arguments for eventscript helper\n"));
328 if (!ctdb_vfork_with_logging(state
, ctdb
, current
->name
,
329 helper_prog
, argc
, argv
,
330 log_event_script_output
,
331 state
, &state
->child
)) {
342 set_close_on_exec(state
->fd
[0]);
344 /* Set ourselves up to be called when that's done. */
345 fde
= tevent_add_fd(ctdb
->ev
, state
, state
->fd
[0], TEVENT_FD_READ
,
346 ctdb_event_script_handler
, state
);
347 tevent_fd_set_auto_close(fde
);
353 Summarize status of this run of scripts.
355 static int script_status(struct ctdb_script_list_old
*scripts
)
359 for (i
= 0; i
< scripts
->num_scripts
; i
++) {
360 switch (scripts
->scripts
[i
].status
) {
364 /* Disabled or missing; that's OK. */
370 return scripts
->scripts
[i
].status
;
378 /* called when child is finished */
379 static void ctdb_event_script_handler(struct tevent_context
*ev
,
380 struct tevent_fd
*fde
,
381 uint16_t flags
, void *p
)
383 struct ctdb_event_script_state
*state
=
384 talloc_get_type(p
, struct ctdb_event_script_state
);
385 struct ctdb_script
*current
= get_current_script(state
);
386 struct ctdb_context
*ctdb
= state
->ctdb
;
390 DEBUG(DEBUG_ERR
,("Eventscript finished but ctdb is NULL\n"));
394 r
= sys_read(state
->fd
[0], ¤t
->status
, sizeof(current
->status
));
396 current
->status
= -errno
;
398 current
->status
= -EINTR
;
399 } else if (r
!= sizeof(current
->status
)) {
400 current
->status
= -EIO
;
403 current
->finished
= timeval_current();
404 /* valgrind gets overloaded if we run next script as it's still doing
405 * post-execution analysis, so kill finished child here. */
406 if (ctdb
->valgrinding
) {
407 ctdb_kill(ctdb
, state
->child
, SIGKILL
);
412 status
= script_status(state
->scripts
);
414 /* Aborted or finished all scripts? We're done. */
415 if (status
!= 0 || state
->current
+1 == state
->scripts
->num_scripts
) {
418 ("Eventscript %s %s finished with state %d\n",
419 ctdb_eventscript_call_names
[state
->call
],
420 state
->options
, status
));
427 /* Forget about that old fd. */
433 current
->status
= fork_child_for_script(ctdb
, state
);
434 if (current
->status
!= 0) {
435 /* This calls the callback. */
440 struct debug_hung_script_state
{
441 struct ctdb_context
*ctdb
;
443 enum ctdb_event call
;
446 static int debug_hung_script_state_destructor(struct debug_hung_script_state
*state
)
449 ctdb_kill(state
->ctdb
, state
->child
, SIGKILL
);
454 static void debug_hung_script_timeout(struct tevent_context
*ev
, struct tevent_timer
*te
,
455 struct timeval t
, void *p
)
457 struct debug_hung_script_state
*state
=
458 talloc_get_type(p
, struct debug_hung_script_state
);
463 static void debug_hung_script_done(struct tevent_context
*ev
, struct tevent_fd
*fde
,
464 uint16_t flags
, void *p
)
466 struct debug_hung_script_state
*state
=
467 talloc_get_type(p
, struct debug_hung_script_state
);
472 static void ctdb_run_debug_hung_script(struct ctdb_context
*ctdb
, struct debug_hung_script_state
*state
)
475 const char * debug_hung_script
= CTDB_ETCDIR
"/debug-hung-script.sh";
477 struct tevent_timer
*ttimer
;
478 struct tevent_fd
*tfd
;
483 DEBUG(DEBUG_ERR
,("Failed to create pipe fd for debug hung script\n"));
487 if (getenv("CTDB_DEBUG_HUNG_SCRIPT") != NULL
) {
488 debug_hung_script
= getenv("CTDB_DEBUG_HUNG_SCRIPT");
491 argv
= talloc_array(state
, const char *, 5);
493 argv
[0] = talloc_asprintf(argv
, "%d", fd
[1]);
494 argv
[1] = talloc_strdup(argv
, debug_hung_script
);
495 argv
[2] = talloc_asprintf(argv
, "%d", state
->child
);
496 argv
[3] = talloc_strdup(argv
, ctdb_eventscript_call_names
[state
->call
]);
499 for (i
=0; i
<4; i
++) {
500 if (argv
[i
] == NULL
) {
509 if (!ctdb_vfork_with_logging(state
, ctdb
, "Hung-script",
510 helper_prog
, 5, argv
, NULL
, NULL
, &pid
)) {
511 DEBUG(DEBUG_ERR
,("Failed to fork a child to track hung event script\n"));
521 ttimer
= tevent_add_timer(ctdb
->ev
, state
,
522 timeval_current_ofs(ctdb
->tunable
.script_timeout
, 0),
523 debug_hung_script_timeout
, state
);
524 if (ttimer
== NULL
) {
529 tfd
= tevent_add_fd(ctdb
->ev
, state
, fd
[0], TEVENT_FD_READ
,
530 debug_hung_script_done
, state
);
536 tevent_fd_set_auto_close(tfd
);
539 /* called when child times out */
540 static void ctdb_event_script_timeout(struct tevent_context
*ev
,
541 struct tevent_timer
*te
,
542 struct timeval t
, void *p
)
544 struct ctdb_event_script_state
*state
= talloc_get_type(p
, struct ctdb_event_script_state
);
545 struct ctdb_context
*ctdb
= state
->ctdb
;
546 struct ctdb_script
*current
= get_current_script(state
);
547 struct debug_hung_script_state
*debug_state
;
549 DEBUG(DEBUG_ERR
,("Event script '%s %s %s' timed out after %.1fs, pid: %d\n",
550 current
->name
, ctdb_eventscript_call_names
[state
->call
], state
->options
,
551 timeval_elapsed(¤t
->start
),
554 /* ignore timeouts for these events */
555 switch (state
->call
) {
556 case CTDB_EVENT_START_RECOVERY
:
557 case CTDB_EVENT_RECOVERED
:
558 case CTDB_EVENT_TAKE_IP
:
559 case CTDB_EVENT_RELEASE_IP
:
560 state
->scripts
->scripts
[state
->current
].status
= 0;
561 DEBUG(DEBUG_ERR
,("Ignoring hung script for %s call %d\n", state
->options
, state
->call
));
564 state
->scripts
->scripts
[state
->current
].status
= -ETIME
;
567 debug_state
= talloc_zero(ctdb
, struct debug_hung_script_state
);
568 if (debug_state
== NULL
) {
573 /* Save information useful for running debug hung script, so
574 * eventscript state can be freed.
576 debug_state
->ctdb
= ctdb
;
577 debug_state
->child
= state
->child
;
578 debug_state
->call
= state
->call
;
580 /* This destructor will actually kill the hung event script */
581 talloc_set_destructor(debug_state
, debug_hung_script_state_destructor
);
586 ctdb_run_debug_hung_script(ctdb
, debug_state
);
590 destroy an event script: kill it if ->child != 0.
592 static int event_script_destructor(struct ctdb_event_script_state
*state
)
595 struct event_script_callback
*callback
;
598 DEBUG(DEBUG_ERR
,(__location__
" Sending SIGTERM to child pid:%d\n", state
->child
));
600 if (ctdb_kill(state
->ctdb
, state
->child
, SIGTERM
) != 0) {
601 DEBUG(DEBUG_ERR
,("Failed to kill child process for eventscript, errno %s(%d)\n", strerror(errno
), errno
));
605 /* If we were the current monitor, we no longer are. */
606 if (state
->ctdb
->current_monitor
== state
) {
607 state
->ctdb
->current_monitor
= NULL
;
610 /* Save our scripts as the last executed status, if we have them.
611 * See ctdb_event_script_callback_v where we abort monitor event. */
612 if (state
->scripts
) {
613 talloc_free(state
->ctdb
->last_status
[state
->call
]);
614 state
->ctdb
->last_status
[state
->call
] = state
->scripts
;
615 if (state
->current
< state
->ctdb
->last_status
[state
->call
]->num_scripts
) {
616 state
->ctdb
->last_status
[state
->call
]->num_scripts
= state
->current
+1;
620 /* Use last status as result, or "OK" if none. */
621 if (state
->ctdb
->last_status
[state
->call
]) {
622 status
= script_status(state
->ctdb
->last_status
[state
->call
]);
627 state
->ctdb
->active_events
--;
628 if (state
->ctdb
->active_events
< 0) {
629 ctdb_fatal(state
->ctdb
, "Active events < 0");
632 /* This is allowed to free us; talloc will prevent double free anyway,
633 * but beware if you call this outside the destructor!
634 * the callback hangs off a different context so we walk the list
635 * of "active" callbacks until we find the one state points to.
636 * if we cant find it it means the callback has been removed.
638 for (callback
= state
->ctdb
->script_callbacks
; callback
!= NULL
; callback
= callback
->next
) {
639 if (callback
== state
->callback
) {
644 state
->callback
= NULL
;
647 /* Make sure destructor doesn't free itself! */
648 talloc_steal(NULL
, callback
);
649 callback
->fn(state
->ctdb
, status
, callback
->private_data
);
650 talloc_free(callback
);
656 static unsigned int count_words(const char *options
)
658 unsigned int words
= 0;
660 options
+= strspn(options
, " \t");
663 options
+= strcspn(options
, " \t");
664 options
+= strspn(options
, " \t");
669 static bool check_options(enum ctdb_event call
, const char *options
)
672 /* These all take no arguments. */
673 case CTDB_EVENT_INIT
:
674 case CTDB_EVENT_SETUP
:
675 case CTDB_EVENT_STARTUP
:
676 case CTDB_EVENT_START_RECOVERY
:
677 case CTDB_EVENT_RECOVERED
:
678 case CTDB_EVENT_MONITOR
:
679 case CTDB_EVENT_SHUTDOWN
:
680 case CTDB_EVENT_IPREALLOCATED
:
681 return count_words(options
) == 0;
683 case CTDB_EVENT_TAKE_IP
: /* interface, IP address, netmask bits. */
684 case CTDB_EVENT_RELEASE_IP
:
685 return count_words(options
) == 3;
687 case CTDB_EVENT_UPDATE_IP
: /* old interface, new interface, IP address, netmask bits. */
688 return count_words(options
) == 4;
691 DEBUG(DEBUG_ERR
,(__location__
"Unknown ctdb_event %u\n", call
));
696 static int remove_callback(struct event_script_callback
*callback
)
698 DLIST_REMOVE(callback
->ctdb
->script_callbacks
, callback
);
703 run the event script in the background, calling the callback when
706 static int ctdb_event_script_callback_v(struct ctdb_context
*ctdb
,
708 void (*callback
)(struct ctdb_context
*, int, void *),
710 enum ctdb_event call
,
711 const char *fmt
, va_list ap
)
713 struct ctdb_event_script_state
*state
;
715 if (ctdb
->recovery_mode
!= CTDB_RECOVERY_NORMAL
) {
716 /* we guarantee that only some specifically allowed event scripts are run
718 const enum ctdb_event allowed_calls
[] = {
721 CTDB_EVENT_START_RECOVERY
,
723 CTDB_EVENT_RELEASE_IP
,
724 CTDB_EVENT_IPREALLOCATED
,
727 for (i
=0;i
<ARRAY_SIZE(allowed_calls
);i
++) {
728 if (call
== allowed_calls
[i
]) break;
730 if (i
== ARRAY_SIZE(allowed_calls
)) {
731 DEBUG(DEBUG_ERR
,("Refusing to run event scripts call '%s' while in recovery\n",
732 ctdb_eventscript_call_names
[call
]));
737 /* Do not run new monitor events if some event is already
738 * running, unless the running event is a monitor event, in
739 * which case running a new one should cancel the old one. */
740 if (call
== CTDB_EVENT_MONITOR
&&
741 ctdb
->active_events
> 0 &&
742 ctdb
->current_monitor
== NULL
) {
743 if (callback
!= NULL
) {
744 callback(ctdb
, -ECANCELED
, private_data
);
749 /* Kill off any running monitor events to run this event. */
750 if (ctdb
->current_monitor
) {
751 struct ctdb_event_script_state
*ms
= talloc_get_type(ctdb
->current_monitor
, struct ctdb_event_script_state
);
753 /* Cancel current monitor callback state only if monitoring
754 * context ctdb->monitor->monitor_context has not been freed */
755 if (ms
->callback
!= NULL
&& !ctdb_stopped_monitoring(ctdb
)) {
756 ms
->callback
->fn(ctdb
, -ECANCELED
, ms
->callback
->private_data
);
757 talloc_free(ms
->callback
);
760 /* Discard script status so we don't save to last_status */
761 talloc_free(ctdb
->current_monitor
->scripts
);
762 ctdb
->current_monitor
->scripts
= NULL
;
763 talloc_free(ctdb
->current_monitor
);
764 ctdb
->current_monitor
= NULL
;
767 state
= talloc(ctdb
->event_script_ctx
, struct ctdb_event_script_state
);
768 CTDB_NO_MEMORY(ctdb
, state
);
770 /* The callback isn't done if the context is freed. */
771 state
->callback
= talloc(mem_ctx
, struct event_script_callback
);
772 CTDB_NO_MEMORY(ctdb
, state
->callback
);
773 DLIST_ADD(ctdb
->script_callbacks
, state
->callback
);
774 talloc_set_destructor(state
->callback
, remove_callback
);
775 state
->callback
->ctdb
= ctdb
;
776 state
->callback
->fn
= callback
;
777 state
->callback
->private_data
= private_data
;
781 state
->options
= talloc_vasprintf(state
, fmt
, ap
);
782 state
->timeout
= timeval_set(ctdb
->tunable
.script_timeout
, 0);
783 state
->scripts
= NULL
;
784 if (state
->options
== NULL
) {
785 DEBUG(DEBUG_ERR
, (__location__
" could not allocate state->options\n"));
789 if (!check_options(state
->call
, state
->options
)) {
790 DEBUG(DEBUG_ERR
, ("Bad eventscript options '%s' for '%s'\n",
792 ctdb_eventscript_call_names
[state
->call
]));
797 DEBUG(DEBUG_INFO
,(__location__
" Starting eventscript %s %s\n",
798 ctdb_eventscript_call_names
[state
->call
],
801 /* This is not a child of state, since we save it in destructor. */
802 state
->scripts
= ctdb_get_script_list(ctdb
, ctdb
);
803 if (state
->scripts
== NULL
) {
810 if (call
== CTDB_EVENT_MONITOR
) {
811 ctdb
->current_monitor
= state
;
814 talloc_set_destructor(state
, event_script_destructor
);
816 ctdb
->active_events
++;
819 if (state
->scripts
->num_scripts
== 0) {
820 callback(ctdb
, 0, private_data
);
825 state
->scripts
->scripts
[0].status
= fork_child_for_script(ctdb
, state
);
826 if (state
->scripts
->scripts
[0].status
!= 0) {
827 /* Callback is called from destructor, with fail result. */
832 if (!timeval_is_zero(&state
->timeout
)) {
833 tevent_add_timer(ctdb
->ev
, state
,
834 timeval_current_ofs(state
->timeout
.tv_sec
,
835 state
->timeout
.tv_usec
),
836 ctdb_event_script_timeout
, state
);
838 DEBUG(DEBUG_ERR
, (__location__
" eventscript %s %s called with no timeout\n",
839 ctdb_eventscript_call_names
[state
->call
],
848 run the event script in the background, calling the callback when
849 finished. If mem_ctx is freed, callback will never be called.
851 int ctdb_event_script_callback(struct ctdb_context
*ctdb
,
853 void (*callback
)(struct ctdb_context
*, int, void *),
855 enum ctdb_event call
,
856 const char *fmt
, ...)
862 ret
= ctdb_event_script_callback_v(ctdb
, mem_ctx
, callback
, private_data
, call
, fmt
, ap
);
869 struct callback_status
{
875 called when ctdb_event_script() finishes
877 static void event_script_callback(struct ctdb_context
*ctdb
, int status
, void *private_data
)
879 struct callback_status
*s
= (struct callback_status
*)private_data
;
885 run the event script, waiting for it to complete. Used when the caller
886 doesn't want to continue till the event script has finished.
888 int ctdb_event_script_args(struct ctdb_context
*ctdb
, enum ctdb_event call
,
889 const char *fmt
, ...)
893 struct callback_status status
= {
899 ret
= ctdb_event_script_callback_v(ctdb
, ctdb
,
900 event_script_callback
, &status
, call
, fmt
, ap
);
906 while (status
.done
== false && tevent_loop_once(ctdb
->ev
) == 0) /* noop */;
908 if (status
.status
== -ETIME
) {
909 DEBUG(DEBUG_ERR
, (__location__
" eventscript for '%s' timedout."
910 " Immediately banning ourself for %d seconds\n",
911 ctdb_eventscript_call_names
[call
],
912 ctdb
->tunable
.recovery_ban_period
));
914 /* Don't ban self if CTDB is starting up or shutting down */
915 if (call
!= CTDB_EVENT_INIT
&& call
!= CTDB_EVENT_SHUTDOWN
) {
920 return status
.status
;
923 int ctdb_event_script(struct ctdb_context
*ctdb
, enum ctdb_event call
)
925 /* GCC complains about empty format string, so use %s and "". */
926 return ctdb_event_script_args(ctdb
, call
, "%s", "");
929 struct eventscript_callback_state
{
930 struct ctdb_req_control_old
*c
;
934 called when a forced eventscript run has finished
936 static void run_eventscripts_callback(struct ctdb_context
*ctdb
, int status
,
939 const char *errmsg
= NULL
;
941 struct eventscript_callback_state
*state
=
942 talloc_get_type(private_data
, struct eventscript_callback_state
);
945 if (status
== -ECANCELED
) {
947 (__location__
" Eventscript cancelled\n"));
948 errmsg
= "cancelled";
951 (__location__
" Failed to run eventscripts\n"));
955 ctdb_request_control_reply(ctdb
, state
->c
, NULL
, status
, errmsg
);
956 /* This will free the struct ctdb_event_script_state we are in! */
962 /* Returns rest of string, or NULL if no match. */
963 static const char *get_call(const char *p
, enum ctdb_event
*call
)
967 /* Skip any initial whitespace. */
968 p
+= strspn(p
, " \t");
970 /* See if we match any. */
971 for (*call
= 0; *call
< CTDB_EVENT_MAX
; (*call
)++) {
972 len
= strlen(ctdb_eventscript_call_names
[*call
]);
973 if (strncmp(p
, ctdb_eventscript_call_names
[*call
], len
) == 0) {
974 /* If end of string or whitespace, we're done. */
975 if (strcspn(p
+ len
, " \t") == 0) {
984 A control to force running of the eventscripts from the ctdb client tool
986 int32_t ctdb_run_eventscripts(struct ctdb_context
*ctdb
,
987 struct ctdb_req_control_old
*c
,
988 TDB_DATA indata
, bool *async_reply
)
991 struct eventscript_callback_state
*state
;
993 enum ctdb_event call
;
995 /* Figure out what call they want. */
996 options
= get_call((const char *)indata
.dptr
, &call
);
998 DEBUG(DEBUG_ERR
, (__location__
" Invalid event name \"%s\"\n", (const char *)indata
.dptr
));
1002 if (ctdb
->recovery_mode
!= CTDB_RECOVERY_NORMAL
) {
1003 DEBUG(DEBUG_ERR
, (__location__
" Aborted running eventscript \"%s\" while in RECOVERY mode\n", indata
.dptr
));
1007 state
= talloc(ctdb
->event_script_ctx
, struct eventscript_callback_state
);
1008 CTDB_NO_MEMORY(ctdb
, state
);
1010 state
->c
= talloc_steal(state
, c
);
1012 DEBUG(DEBUG_NOTICE
,("Running eventscripts with arguments %s\n", indata
.dptr
));
1014 ret
= ctdb_event_script_callback(ctdb
,
1015 ctdb
, run_eventscripts_callback
, state
,
1016 call
, "%s", options
);
1019 DEBUG(DEBUG_ERR
,(__location__
" Failed to run eventscripts with arguments %s\n", indata
.dptr
));
1024 /* tell ctdb_control.c that we will be replying asynchronously */
1025 *async_reply
= true;
1032 int32_t ctdb_control_enable_script(struct ctdb_context
*ctdb
, TDB_DATA indata
)
1037 TALLOC_CTX
*tmp_ctx
= talloc_new(ctdb
);
1039 script
= (char *)indata
.dptr
;
1040 if (indata
.dsize
== 0) {
1041 DEBUG(DEBUG_ERR
,(__location__
" No script specified.\n"));
1042 talloc_free(tmp_ctx
);
1045 if (indata
.dptr
[indata
.dsize
- 1] != '\0') {
1046 DEBUG(DEBUG_ERR
,(__location__
" String is not null terminated.\n"));
1047 talloc_free(tmp_ctx
);
1050 if (index(script
,'/') != NULL
) {
1051 DEBUG(DEBUG_ERR
,(__location__
" Script name contains '/'. Failed to enable script %s\n", script
));
1052 talloc_free(tmp_ctx
);
1057 if (stat(ctdb
->event_script_dir
, &st
) != 0 &&
1059 DEBUG(DEBUG_CRIT
,("No event script directory found at '%s'\n", ctdb
->event_script_dir
));
1060 talloc_free(tmp_ctx
);
1065 filename
= talloc_asprintf(tmp_ctx
, "%s/%s", ctdb
->event_script_dir
, script
);
1066 if (filename
== NULL
) {
1067 DEBUG(DEBUG_ERR
,(__location__
" Failed to create script path\n"));
1068 talloc_free(tmp_ctx
);
1072 if (stat(filename
, &st
) != 0) {
1073 DEBUG(DEBUG_ERR
,("Could not stat event script %s. Failed to enable script.\n", filename
));
1074 talloc_free(tmp_ctx
);
1078 if (chmod(filename
, st
.st_mode
| S_IXUSR
) == -1) {
1079 DEBUG(DEBUG_ERR
,("Could not chmod %s. Failed to enable script.\n", filename
));
1080 talloc_free(tmp_ctx
);
1084 talloc_free(tmp_ctx
);
1088 int32_t ctdb_control_disable_script(struct ctdb_context
*ctdb
, TDB_DATA indata
)
1093 TALLOC_CTX
*tmp_ctx
= talloc_new(ctdb
);
1095 script
= (char *)indata
.dptr
;
1096 if (indata
.dsize
== 0) {
1097 DEBUG(DEBUG_ERR
,(__location__
" No script specified.\n"));
1098 talloc_free(tmp_ctx
);
1101 if (indata
.dptr
[indata
.dsize
- 1] != '\0') {
1102 DEBUG(DEBUG_ERR
,(__location__
" String is not null terminated.\n"));
1103 talloc_free(tmp_ctx
);
1106 if (index(script
,'/') != NULL
) {
1107 DEBUG(DEBUG_ERR
,(__location__
" Script name contains '/'. Failed to disable script %s\n", script
));
1108 talloc_free(tmp_ctx
);
1113 if (stat(ctdb
->event_script_dir
, &st
) != 0 &&
1115 DEBUG(DEBUG_CRIT
,("No event script directory found at '%s'\n", ctdb
->event_script_dir
));
1116 talloc_free(tmp_ctx
);
1121 filename
= talloc_asprintf(tmp_ctx
, "%s/%s", ctdb
->event_script_dir
, script
);
1122 if (filename
== NULL
) {
1123 DEBUG(DEBUG_ERR
,(__location__
" Failed to create script path\n"));
1124 talloc_free(tmp_ctx
);
1128 if (stat(filename
, &st
) != 0) {
1129 DEBUG(DEBUG_ERR
,("Could not stat event script %s. Failed to disable script.\n", filename
));
1130 talloc_free(tmp_ctx
);
1134 if (chmod(filename
, st
.st_mode
& ~(S_IXUSR
|S_IXGRP
|S_IXOTH
)) == -1) {
1135 DEBUG(DEBUG_ERR
,("Could not chmod %s. Failed to disable script.\n", filename
));
1136 talloc_free(tmp_ctx
);
1140 talloc_free(tmp_ctx
);