mirror of
https://github.com/samba-team/samba.git
synced 2024-12-22 13:34:15 +03:00
3084928383
Boiler-plate replacement moving the (num_fds!=0) check down Signed-off-by: Volker Lendecke <vl@samba.org> Reviewed-by: Jeremy Allison <jra@samba.org>
136 lines
3.3 KiB
C
136 lines
3.3 KiB
C
/*
|
|
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/>.
|
|
*/
|
|
|
|
#include "includes.h"
|
|
#include "lib/util/server_id.h"
|
|
#include "messaging/messaging.h"
|
|
#include "messaging/messaging_internal.h"
|
|
|
|
#if defined(DEVELOPER) || defined(ENABLE_SELFTEST)
|
|
|
|
/*
|
|
* 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,
|
|
size_t num_fds,
|
|
int *fds,
|
|
DATA_BLOB *data)
|
|
{
|
|
int sig;
|
|
struct server_id_buf tmp;
|
|
|
|
if (num_fds != 0) {
|
|
DBG_WARNING("Received %zu fds, ignoring message\n", num_fds);
|
|
return;
|
|
}
|
|
|
|
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);
|
|
}
|
|
|
|
/*
|
|
* Cause the current process to sleep for a specified number of seconds
|
|
*/
|
|
static void do_sleep(struct imessaging_context *msg,
|
|
void *private_data,
|
|
uint32_t msg_type,
|
|
struct server_id src,
|
|
size_t num_fds,
|
|
int *fds,
|
|
DATA_BLOB *data)
|
|
{
|
|
unsigned int seconds;
|
|
struct server_id_buf tmp;
|
|
|
|
if (num_fds != 0) {
|
|
DBG_WARNING("Received %zu fds, ignoring message\n", num_fds);
|
|
return;
|
|
}
|
|
|
|
if (data->length != sizeof(seconds)) {
|
|
DBG_ERR("Process %s sent bogus sleep request\n",
|
|
server_id_str_buf(src, &tmp));
|
|
return;
|
|
}
|
|
|
|
seconds = *(unsigned int *)data->data;
|
|
DBG_ERR("Process %s requested a sleep of %u seconds\n",
|
|
server_id_str_buf(src, &tmp),
|
|
seconds);
|
|
sleep(seconds);
|
|
DBG_ERR("Restarting after %u second sleep requested by process %s\n",
|
|
seconds,
|
|
server_id_str_buf(src, &tmp));
|
|
}
|
|
|
|
/*
|
|
* 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;
|
|
}
|
|
|
|
status = imessaging_register(msg, NULL, MSG_SMB_SLEEP, do_sleep);
|
|
if (!NT_STATUS_IS_OK(status)) {
|
|
return status;
|
|
}
|
|
|
|
return NT_STATUS_OK;
|
|
}
|
|
|
|
#endif /* defined(DEVELOPER) || defined(ENABLE_SELFTEST) */
|