pload.c
No OneTemporary
Actions

Size

9 KB

Referenced Files

None

Subscribers

None

pload.c
View Options

	/*
	* Copyright (c) 2008-2012 Red Hat, Inc.
	*
	* All rights reserved.
	*
	* Author: Steven Dake (sdake@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 <sys/types.h>
	#include <sys/socket.h>
	#include <sys/un.h>
	#include <sys/time.h>
	#include <sys/ioctl.h>
	#include <netinet/in.h>
	#include <sys/uio.h>
	#include <unistd.h>
	#include <fcntl.h>
	#include <stdlib.h>
	#include <stdio.h>
	#include <errno.h>
	#include <time.h>
	#include <netinet/in.h>
	#include <arpa/inet.h>
	#include <string.h>
	#include <assert.h>

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

	#include <corosync/swab.h>
	#include <corosync/corotypes.h>
	#include <corosync/corodefs.h>
	#include <corosync/mar_gen.h>
	#include <corosync/coroapi.h>
	#include <corosync/ipc_pload.h>
	#include <corosync/list.h>
	#include <corosync/logsys.h>

	#include "service.h"

	LOGSYS_DECLARE_SUBSYS ("PLOAD");

	enum pload_exec_message_req_types {
	MESSAGE_REQ_EXEC_PLOAD_START = 0,
	MESSAGE_REQ_EXEC_PLOAD_MCAST = 1
	};

	/*
	* Service Interfaces required by service_message_handler struct
	*/
	static char pload_exec_init_fn (struct corosync_api_v1 corosync_api);

	static void pload_confchg_fn (
	enum totem_configuration_type configuration_type,
	const unsigned int *member_list, size_t member_list_entries,
	const unsigned int *left_list, size_t left_list_entries,
	const unsigned int *joined_list, size_t joined_list_entries,
	const struct memb_ring_id *ring_id);

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

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

	static void req_exec_pload_start_endian_convert (void *msg);

	static void req_exec_pload_mcast_endian_convert (void *msg);

	static void message_handler_req_pload_start (void conn, const void msg);

	static int pload_lib_init_fn (void *conn);

	static int pload_lib_exit_fn (void *conn);

	static char buffer[1000000];

	static unsigned int msgs_delivered = 0;

	static unsigned int msgs_wanted = 0;

	static unsigned int msg_size = 0;

	static unsigned int msg_code = 1;

	static unsigned int msgs_sent = 0;


	static struct corosync_api_v1 *api;

	struct req_exec_pload_start {
	struct qb_ipc_request_header header;
	unsigned int msg_code;
	unsigned int msg_count;
	unsigned int msg_size;
	unsigned int time_interval;
	};

	struct req_exec_pload_mcast {
	struct qb_ipc_request_header header;
	unsigned int msg_code;
	};

	static struct corosync_lib_handler pload_lib_engine[] =
	{
	{ /* 0 */
	.lib_handler_fn = message_handler_req_pload_start,
	.flow_control = CS_LIB_FLOW_CONTROL_NOT_REQUIRED
	}
	};

	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
	}
	};

	struct corosync_service_engine pload_service_engine = {
	.name = "corosync profile loading service",
	.id = PLOAD_SERVICE,
	.priority = 1,
	.private_data_size = 0,
	.flow_control = CS_LIB_FLOW_CONTROL_REQUIRED,
	.lib_init_fn = pload_lib_init_fn,
	.lib_exit_fn = pload_lib_exit_fn,
	.lib_engine = pload_lib_engine,
	.lib_engine_count = sizeof (pload_lib_engine) / sizeof (struct corosync_lib_handler),
	.exec_engine = pload_exec_engine,
	.exec_engine_count = sizeof (pload_exec_engine) / sizeof (struct corosync_exec_handler),
	.confchg_fn = pload_confchg_fn,
	.exec_init_fn = pload_exec_init_fn,
	.exec_dump_fn = NULL,
	.sync_mode = CS_SYNC_V2
	};

	static DECLARE_LIST_INIT (confchg_notify);

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

	static char pload_exec_init_fn (struct corosync_api_v1 corosync_api)
	{
	#ifdef COROSYNC_SOLARIS
	logsys_subsys_init();
	#endif
	api = corosync_api;

	return NULL;
	}

	static void pload_confchg_fn (
	enum totem_configuration_type configuration_type,
	const unsigned int *member_list, size_t member_list_entries,
	const unsigned int *left_list, size_t left_list_entries,
	const unsigned int *joined_list, size_t joined_list_entries,
	const struct memb_ring_id *ring_id)
	{
	}

	static int pload_lib_init_fn (void *conn)
	{
	return (0);
	}

	static int pload_lib_exit_fn (void *conn)
	{
	return (0);
	}

	static void message_handler_req_pload_start (void conn, const void msg)
	{
	const struct req_lib_pload_start *req_lib_pload_start = msg;
	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_code = req_lib_pload_start->msg_code;
	req_exec_pload_start.msg_size = req_lib_pload_start->msg_size;
	req_exec_pload_start.msg_count = req_lib_pload_start->msg_count;
	req_exec_pload_start.time_interval = req_lib_pload_start->time_interval;
	iov.iov_base = (void *)&req_exec_pload_start;
	iov.iov_len = sizeof (struct req_exec_pload_start);

	msgs_delivered = 0;

	msgs_wanted = 0;

	msgs_sent = 0;

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

	static void req_exec_pload_start_endian_convert (void *msg)
	{
	}

	static void req_exec_pload_mcast_endian_convert (void *msg)
	{
	}

	static int 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++;
	msg_code++;
	}
	} while (msgs_sent < msgs_wanted);
	if (msgs_sent == msgs_wanted) {
	return (0);
	} else {
	return (-1);
	}
	}

	hdb_handle_t start_mcasting_handle;

	static void start_mcasting (void)
	{
	api->schedwrk_create (
	&start_mcasting_handle,
	send_message,
	&start_mcasting_handle);
	}

	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;

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

	start_mcasting ();
	}
	#ifndef timersub
	#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 */

	unsigned long long int tv1;
	unsigned long long int tv2;
	unsigned long long int tv_elapsed;
	int last_msg_no = 0;

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

	last_msg_no = pload_mcast->msg_code;
	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);
	}
	}

File Metadata

Mime Type: text/x-c
Expires: Mon, Dec 23, 10:27 PM (1 d, 8 h)
Storage Engine: blob
Storage Format: Raw Data
Storage Handle: 1129249
Default Alt Text: pload.c (9 KB)

pload.cNo OneTemporaryActions

pload.cView Options

File Metadata

Event Timeline

pload.c
No OneTemporary
Actions

pload.c
View Options