return False;
}
-#ifndef DEVELOPER
+#if !defined(DEVELOPER) && !defined(ENABLE_SELFTEST)
fprintf(stderr, "Fault injection is only available in "
- "developer builds\n");
+ "developer and self test builds\n");
return False;
-#else /* DEVELOPER */
+#else /* DEVELOPER || ENABLE_SELFTEST */
{
int sig = 0;
return send_message(msg_ctx, pid, MSG_SMB_INJECT_FAULT,
&sig, sizeof(int));
}
-#endif /* DEVELOPER */
+#endif /* DEVELOPER || ENABLE_SELFTEST */
}
/* Force a browser election */
--- /dev/null
+/*
+ Unix SMB/CIFS implementation.
+
+ Handers for non core Samba internal messages
+
+ Handlers for messages that are only included in developer and self test
+ builds.
+
+ Copyright (C) Andrew Bartlett <abartlet@samba.org> 2018
+
+ This program is free software; you can redistribute it and/or modify
+ it under the terms of the GNU General Public License as published by
+ the Free Software Foundation; either version 3 of the License, or
+ (at your option) any later version.
+
+ This program 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 General Public License for more details.
+
+ You should have received a copy of the GNU General Public License
+ along with this program. If not, see <http://www.gnu.org/licenses/>.
+*/
+
+#if defined(DEVELOPER) || defined(ENABLE_SELFTEST)
+
+#include "includes.h"
+#include "lib/util/server_id.h"
+#include "messaging/messaging.h"
+#include "messaging/messaging_internal.h"
+
+/*
+ * Inject a fault into the currently running process
+ */
+static void do_inject_fault(struct imessaging_context *msg,
+ void *private_data,
+ uint32_t msg_type,
+ struct server_id src,
+ DATA_BLOB *data)
+{
+ int sig;
+ struct server_id_buf tmp;
+
+ if (data->length != sizeof(sig)) {
+ DBG_ERR("Process %s sent bogus signal injection request\n",
+ server_id_str_buf(src, &tmp));
+ return;
+ }
+
+ sig = *(int *)data->data;
+ if (sig == -1) {
+ DBG_ERR("Process %s requested an iternal failure, "
+ "calling exit(1)\n",
+ server_id_str_buf(src, &tmp));
+ exit(1);
+ }
+
+#if HAVE_STRSIGNAL
+ DBG_ERR("Process %s requested injection of signal %d (%s)\n",
+ server_id_str_buf(src, &tmp),
+ sig,
+ strsignal(sig));
+#else
+ DBG_ERR("Process %s requested injection of signal %d\n",
+ server_id_str_buf(src, &tmp),
+ sig);
+#endif
+
+ kill(getpid(), sig);
+}
+
+/*
+ * Register the extra messaging handlers
+ */
+NTSTATUS imessaging_register_extra_handlers(struct imessaging_context *msg)
+{
+ NTSTATUS status;
+
+ status = imessaging_register(
+ msg, NULL, MSG_SMB_INJECT_FAULT, do_inject_fault);
+ if (!NT_STATUS_IS_OK(status)) {
+ return status;
+ }
+
+ return NT_STATUS_OK;
+}
+
+#endif /* defined(DEVELOPER) || defined(ENABLE_SELFTEST) */