/*
 * Copyright (c) 2009 Red Hat, Inc.
 *
 * All rights reserved.
 *
 * Author: Steven Dake (sdake@redhat.com)
 *
 * libqb is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 2.1 of the License, or
 * (at your option) any later version.
 *
 * libqb 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 Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with libqb.  If not, see <http://www.gnu.org/licenses/>.
 */
#include "os_base.h"
#include <signal.h>

#include <qb/qblog.h>
#include <qb/qbutil.h>
#include <qb/qbipcc.h>

#define ITERATIONS 10000
pid_t mypid;
int32_t blocking = QB_TRUE;
int32_t events = QB_FALSE;
int32_t verbose = 0;
static qb_ipcc_connection_t *conn;
#define MAX_MSG_SIZE (8192*128)
static qb_util_stopwatch_t *sw;

static void bm_finish(const char *operation, int32_t size)
{
	float ops_per_sec;
	float mbs_per_sec;
	float elapsed;

	qb_util_stopwatch_stop(sw);
	elapsed = qb_util_stopwatch_sec_elapsed_get(sw);
	ops_per_sec = ((float)ITERATIONS) / elapsed;
	mbs_per_sec = ((((float)ITERATIONS) * size) / elapsed) / (1024.0 * 1024.0);

	qb_log(LOG_INFO, "write size, %d, OPs/sec, %9.3f, MB/sec, %9.3f",
	       size, ops_per_sec, mbs_per_sec);
}

struct my_req {
	struct qb_ipc_request_header hdr;
	char message[1024 * 1024];
};

static struct my_req request;

static int32_t bmc_send_nozc(uint32_t size)
{
	struct qb_ipc_response_header res_header;
	int32_t res;

	request.hdr.id = QB_IPC_MSG_USER_START + 3;
	request.hdr.size = sizeof(struct qb_ipc_request_header) + size;

repeat_send:
	res = qb_ipcc_send(conn, &request, request.hdr.size);
	if (res < 0) {
		if (res == -EAGAIN) {
			goto repeat_send;
		} else if (res == -EINVAL || res == -EINTR || res == -ENOTCONN) {
			qb_perror(LOG_ERR, "qb_ipcc_send");
			return -1;
		} else {
			errno = -res;
			qb_perror(LOG_ERR, "qb_ipcc_send");
			goto repeat_send;
		}
	}

	if (blocking) {
		res = qb_ipcc_recv(conn,
				&res_header,
				sizeof(struct qb_ipc_response_header), -1);
		if (res == -EINTR) {
			return -1;
		}
		if (res < 0) {
			qb_perror(LOG_ERR, "qb_ipcc_recv");
		}
		assert(res == sizeof(struct qb_ipc_response_header));
		assert(res_header.id == 13);
		assert(res_header.size == sizeof(struct qb_ipc_response_header));
	}
	if (events) {
		res = qb_ipcc_event_recv(conn,
				&res_header,
				sizeof(struct qb_ipc_response_header), -1);
		if (res == -EINTR) {
			return -1;
		}
		if (res < 0) {
			qb_perror(LOG_ERR, "qb_ipcc_event_recv");
		}
		assert(res == sizeof(struct qb_ipc_response_header));
		assert(res_header.id == 13);
		assert(res_header.size == sizeof(struct qb_ipc_response_header));
	}
	return 0;
}

struct qb_ipc_request_header *global_zcb_buffer;

static void show_usage(const char *name)
{
	qb_log(LOG_INFO, "usage: \n");
	qb_log(LOG_INFO, "%s <options>\n", name);
	qb_log(LOG_INFO, "\n");
	qb_log(LOG_INFO, "  options:\n");
	qb_log(LOG_INFO, "\n");
	qb_log(LOG_INFO, "  -n             non-blocking ipc (default blocking)\n");
	qb_log(LOG_INFO, "  -e             receive events\n");
	qb_log(LOG_INFO, "  -v             verbose\n");
	qb_log(LOG_INFO, "  -h             show this help text\n");
	qb_log(LOG_INFO, "\n");
}

static void sigterm_handler(int32_t num)
{
	qb_log(LOG_INFO, "bmc: %s(%d)\n", __func__, num);
	qb_ipcc_disconnect(conn);
	exit(0);
}

int32_t
main(int32_t argc, char *argv[])
{
	const char *options = "nevh";
	int32_t opt;
	int32_t i, j;
	size_t size;

	mypid = getpid();

	qb_log_init("bmc", LOG_USER, LOG_EMERG);
	qb_log_ctl(QB_LOG_SYSLOG, QB_LOG_CONF_ENABLED, QB_FALSE);
	qb_log_filter_ctl(QB_LOG_STDERR, QB_LOG_FILTER_ADD,
			  QB_LOG_FILTER_FILE, "*", LOG_INFO);
	qb_log_ctl(QB_LOG_STDERR, QB_LOG_CONF_ENABLED, QB_TRUE);

	while ((opt = getopt(argc, argv, options)) != -1) {
		switch (opt) {
		case 'n':
			blocking = QB_FALSE;
			break;
		case 'e':
			events = QB_TRUE;
			break;
		case 'v':
			verbose++;
			break;
		case 'h':
		default:
			show_usage(argv[0]);
			exit(0);
			break;
		}
	}

	signal(SIGINT, sigterm_handler);
	signal(SIGILL, sigterm_handler);
	signal(SIGTERM, sigterm_handler);
	conn = qb_ipcc_connect("bm1", MAX_MSG_SIZE);
	if (conn == NULL) {
		qb_perror(LOG_ERR, "qb_ipcc_connect");
		exit(1);
	}

	sw =  qb_util_stopwatch_create();
	size = QB_MAX(sizeof(struct qb_ipc_request_header), 64);
	for (j = 0; j < 20; j++) {
		if (size >= MAX_MSG_SIZE)
			break;
		qb_util_stopwatch_start(sw);
		for (i = 0; i < ITERATIONS; i++) {
			if (bmc_send_nozc(size) == -1) {
				break;
			}
		}
		bm_finish("send_nozc", size);
		size *= 2;
	}

	qb_ipcc_disconnect(conn);
	return EXIT_SUCCESS;
}