/*
   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) */