/*
 * Copyright (c) 2008-2012 Red Hat, Inc.
 *
 * All rights reserved.
 *
 * Authors: Steven Dake (sdake@redhat.com)
 *          Fabio M. Di Nitto (fdinitto@redhat.com)
 *
 * This software licensed under BSD license, the text of which follows:
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 * - Redistributions of source code must retain the above copyright notice,
 *   this list of conditions and the following disclaimer.
 * - Redistributions in binary form must reproduce the above copyright notice,
 *   this list of conditions and the following disclaimer in the documentation
 *   and/or other materials provided with the distribution.
 * - Neither the name of the MontaVista Software, Inc. nor the names of its
 *   contributors may be used to endorse or promote products derived from this
 *   software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
 * THE POSSIBILITY OF SUCH DAMAGE.
 */

#include <config.h>

#include <qb/qblist.h>
#include <qb/qbutil.h>
#include <qb/qbipc_common.h>

#include <corosync/swab.h>
#include <corosync/corodefs.h>
#include <corosync/coroapi.h>
#include <corosync/icmap.h>
#include <corosync/logsys.h>

#include "service.h"
#include "util.h"

LOGSYS_DECLARE_SUBSYS ("PLOAD");

/*
 * Service Interfaces required by service_message_handler struct
 */
static struct corosync_api_v1 *api;

static char *pload_exec_init_fn (struct corosync_api_v1 *corosync_api);

/*
 * on wire / network bits
 */
enum pload_exec_message_req_types {
	MESSAGE_REQ_EXEC_PLOAD_START = 0,
	MESSAGE_REQ_EXEC_PLOAD_MCAST = 1
};

struct req_exec_pload_start {
	struct qb_ipc_request_header header;
	uint32_t msg_count;
	uint32_t msg_size;
};

struct req_exec_pload_mcast {
	struct qb_ipc_request_header header;
};

static void message_handler_req_exec_pload_start (const void *msg,
						  unsigned int nodeid);
static void req_exec_pload_start_endian_convert (void *msg);

static void message_handler_req_exec_pload_mcast (const void *msg,
						  unsigned int nodeid);
static void req_exec_pload_mcast_endian_convert (void *msg);

static struct corosync_exec_handler pload_exec_engine[] =
{
	{
		.exec_handler_fn 	= message_handler_req_exec_pload_start,
		.exec_endian_convert_fn	= req_exec_pload_start_endian_convert
	},
	{
		.exec_handler_fn 	= message_handler_req_exec_pload_mcast,
		.exec_endian_convert_fn	= req_exec_pload_mcast_endian_convert
	}
};

/*
 * internal bits and pieces
 */

/*
 * really unused buffer but we need to give something to iovec
 */
static char *buffer = NULL;

/*
 * wanted/size come from config
 * sent/delivered track the runtime status
 */
static uint32_t msgs_wanted = 0;
static uint32_t msg_size = 0;
static uint32_t msgs_sent = 0;
static uint32_t msgs_delivered = 0;

/*
 * bit flip to track if we are running or not and avoid multiple instances
 */
static uint8_t pload_started = 0;

/*
 * handle for scheduler
 */
static hdb_handle_t start_mcasting_handle;

/*
 * timing/profiling
 */
static unsigned long long int tv1;
static unsigned long long int tv2;
static unsigned long long int tv_elapsed;

/*
 * Service engine hooks
 */
struct corosync_service_engine pload_service_engine = {
	.name			= "corosync profile loading service",
	.id			= PLOAD_SERVICE,
	.priority		= 1,
	.flow_control		= CS_LIB_FLOW_CONTROL_REQUIRED,
	.exec_engine		= pload_exec_engine,
	.exec_engine_count	= sizeof (pload_exec_engine) / sizeof (struct corosync_exec_handler),
	.exec_init_fn		= pload_exec_init_fn
};

struct corosync_service_engine *pload_get_service_engine_ver0 (void)
{
	return (&pload_service_engine);
}

/*
 * internal use only functions
 */

/*
 * not all architectures / OSes define timersub in sys/time.h or time.h
 */

#ifndef timersub
#warning Using internal timersub definition. Check your include header files
#define timersub(a, b, result)					\
do {								\
	(result)->tv_sec = (a)->tv_sec - (b)->tv_sec;		\
	(result)->tv_usec = (a)->tv_usec - (b)->tv_usec;	\
	if ((result)->tv_usec < 0) {				\
		--(result)->tv_sec;				\
		(result)->tv_usec += 1000000;			\
	}							\
} while (0)
#endif /* timersub */

/*
 * tell all cluster nodes to start mcasting
 */
static void pload_send_start (uint32_t count, uint32_t size)
{
	struct req_exec_pload_start req_exec_pload_start;
	struct iovec iov;

	req_exec_pload_start.header.id = SERVICE_ID_MAKE (PLOAD_SERVICE, MESSAGE_REQ_EXEC_PLOAD_START);
	req_exec_pload_start.msg_count = count;
	req_exec_pload_start.msg_size = size;
	iov.iov_base = (void *)&req_exec_pload_start;
	iov.iov_len = sizeof (struct req_exec_pload_start);

	api->totem_mcast (&iov, 1, TOTEM_AGREED);
}

/*
 * send N empty data messages of size X
 */
static int pload_send_message (const void *arg)
{
	struct req_exec_pload_mcast req_exec_pload_mcast;
	struct iovec iov[2];
	unsigned int res;
	unsigned int iov_len = 1;

	req_exec_pload_mcast.header.id = SERVICE_ID_MAKE (PLOAD_SERVICE, MESSAGE_REQ_EXEC_PLOAD_MCAST);
	req_exec_pload_mcast.header.size = sizeof (struct req_exec_pload_mcast) + msg_size;

	iov[0].iov_base = (void *)&req_exec_pload_mcast;
	iov[0].iov_len = sizeof (struct req_exec_pload_mcast);
	if (msg_size > sizeof (req_exec_pload_mcast)) {
		iov[1].iov_base = &buffer;
		iov[1].iov_len = msg_size - sizeof (req_exec_pload_mcast);
		iov_len = 2;
	}

	do {
		res = api->totem_mcast (iov, iov_len, TOTEM_AGREED);
		if (res == -1) {
			break;
		} else {
			msgs_sent++;
		}
	} while (msgs_sent < msgs_wanted);

	if (msgs_sent == msgs_wanted) {
		return (0);
	} else {
		return (-1);
	}
}

/*
 * hook into icmap to read config at runtime
 * we do NOT start by default, ever!
 */
static void pload_read_config(
	int32_t event,
	const char *key_name,
	struct icmap_notify_value new_val,
	struct icmap_notify_value old_val,
	void *user_data)
{
	uint32_t pload_count = 1500000;
	uint32_t pload_size = 300;
	char *pload_start = NULL;

	icmap_get_uint32("pload.count", &pload_count);
	icmap_get_uint32("pload.size", &pload_size);

	if (pload_size > MESSAGE_SIZE_MAX) {
		pload_size = MESSAGE_SIZE_MAX;
		log_printf(LOGSYS_LEVEL_WARNING, "pload size limited to %u", pload_size);
	}

	if ((!pload_started) &&
	    (icmap_get_string("pload.start", &pload_start) == CS_OK)) {
		if (!strcmp(pload_start,
			    "i_totally_understand_pload_will_crash_my_cluster_and_kill_corosync_on_exit")) {
			buffer = malloc(pload_size);
			if (buffer) {
				log_printf(LOGSYS_LEVEL_WARNING, "Starting pload!");
				pload_send_start(pload_count,  pload_size);
			} else {
				log_printf(LOGSYS_LEVEL_WARNING,
					  "Unable to allocate pload buffer!");
			}
		}
		free(pload_start);
	}
}

/*
 * exec functions
 */
static char *pload_exec_init_fn (struct corosync_api_v1 *corosync_api)
{
	icmap_track_t pload_track = NULL;

	api = corosync_api;

	/*
	 * track changes to pload config and start only on demand
	 */
	if (icmap_track_add("pload.",
		ICMAP_TRACK_ADD | ICMAP_TRACK_DELETE | ICMAP_TRACK_MODIFY | ICMAP_TRACK_PREFIX,
		pload_read_config,
		NULL,
		&pload_track) != CS_OK) {
		return (char *)"Unable to setup pload config tracking!\n";
	}

	return NULL;
}

/*
 * network messages/onwire handlers
 */

static void req_exec_pload_start_endian_convert (void *msg)
{
	struct req_exec_pload_start *req_exec_pload_start = msg;

	req_exec_pload_start->msg_count = swab32(req_exec_pload_start->msg_count);
	req_exec_pload_start->msg_size = swab32(req_exec_pload_start->msg_size);
}

static void message_handler_req_exec_pload_start (
	const void *msg,
	unsigned int nodeid)
{
	const struct req_exec_pload_start *req_exec_pload_start = msg;

	/*
	 * don't start multiple instances
	 */
	if (pload_started) {
		return;
	}

	pload_started = 1;

	msgs_wanted = req_exec_pload_start->msg_count;
	msg_size = req_exec_pload_start->msg_size;

	api->schedwrk_create (
		&start_mcasting_handle,
		pload_send_message,
		&start_mcasting_handle);
}

static void req_exec_pload_mcast_endian_convert (void *msg)
{
}

static void message_handler_req_exec_pload_mcast (
	const void *msg,
	unsigned int nodeid)
{
	char log_buffer[1024];

	if (msgs_delivered == 0) {
		tv1 = qb_util_nano_current_get ();
	}
	msgs_delivered += 1;
	if (msgs_delivered == msgs_wanted) {
		tv2 = qb_util_nano_current_get ();
		tv_elapsed = tv2 - tv1;
		sprintf (log_buffer, "%5d Writes %d bytes per write %7.3f seconds runtime, %9.3f TP/S, %9.3f MB/S.",
			msgs_delivered,
			msg_size,
			(tv_elapsed / 1000000000.0),
			((float)msgs_delivered) /  (tv_elapsed / 1000000000.0),
			(((float)msgs_delivered) * ((float)msg_size) /
				(tv_elapsed / 1000000000.0)) / (1024.0 * 1024.0));
		log_printf (LOGSYS_LEVEL_NOTICE, "%s", log_buffer);
		log_printf (LOGSYS_LEVEL_WARNING, "Stopping corosync the hard way");
		if (buffer) {
			free(buffer);
			buffer = NULL;
		}
		exit(COROSYNC_DONE_PLOAD);
	}
}