corosync/services/testquorum.c

/*
 * Copyright (c) 2008 Red Hat, Inc.
 *
 * All rights reserved.
 *
 * Author: Christine Caulfield (ccaulfie@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 Red Hat, 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 <assert.h>
#include <pwd.h>
#include <grp.h>
#include <sys/types.h>
#include <sys/poll.h>
#include <sys/uio.h>
#include <sys/mman.h>
#include <sys/socket.h>
#include <sys/un.h>
#include <sys/time.h>
#include <sys/resource.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <fcntl.h>
#include <stdlib.h>
#include <stdio.h>
#include <errno.h>
#include <signal.h>
#include <sched.h>
#include <time.h>

#include <corosync/engine/logsys.h>
#include <corosync/ipc_gen.h>
#include <corosync/lcr/lcr_comp.h>
#include <corosync/engine/coroapi.h>

#include <corosync/engine/quorum.h>

LOGSYS_DECLARE_SUBSYS ("TEST", LOG_INFO);

static void test_init(struct corosync_api_v1 *api, quorum_set_quorate_fn_t report);

/*
 * lcrso object definition
 */
static struct quorum_services_api_ver1 test_quorum_iface_ver0 = {
	.init				= test_init
};

static struct lcr_iface corosync_test_quorum_ver0[1] = {
	{
		.name			= "testquorum",
		.version		= 0,
		.versions_replace	= 0,
		.versions_replace_count	= 0,
		.dependencies		= 0,
		.dependency_count	= 0,
		.constructor		= NULL,
		.destructor		= NULL,
		.interfaces		= (void **)(void *)&test_quorum_iface_ver0,
	},
};

static struct lcr_comp test_quorum_comp_ver0 = {
	.iface_count			= 1,
	.ifaces				= corosync_test_quorum_ver0
};

__attribute__ ((constructor)) static void test_quorum_comp_register (void) {
	lcr_interfaces_set (&corosync_test_quorum_ver0[0], &test_quorum_iface_ver0);
	lcr_component_register (&test_quorum_comp_ver0);
}

/* -------------------------------------------------- */

static quorum_set_quorate_fn_t set_quorum;

static void key_change_notify(object_change_type_t change_type,
			      hdb_handle_t parent_object_handle,
			      hdb_handle_t object_handle,
			      void *object_name_pt, int object_name_len,
			      void *key_name_pt, int key_len,
			      void *key_value_pt, int key_value_len,
			      void *priv_data_pt)
{
	unsigned int members[1];
	struct memb_ring_id ring_id;

	memset(&ring_id, 0, sizeof(ring_id));

	/* If the 'quorum.quorate' key changes, then that changes quorum */
	if (strncmp(key_name_pt, "quorate", key_len) == 0) {
		set_quorum(members, 0, atoi(key_value_pt), &ring_id);
	}
}

static void quorum_callback(int quorate, void *context)
{
	log_printf(LOG_LEVEL_DEBUG, "quorum callback: quorate = %d\n", quorate);
}

static void test_init(struct corosync_api_v1 *api,
		      quorum_set_quorate_fn_t report)
{

	hdb_handle_t find_handle;
	hdb_handle_t quorum_handle = 0;

	set_quorum = report;

	/*
	 * Register for objdb changes on quorum { }
	 */
	api->object_find_create(OBJECT_PARENT_HANDLE, "quorum", strlen("quorum"), &find_handle);
        api->object_find_next(find_handle, &quorum_handle);
	api->object_find_destroy(find_handle);

	api->object_track_start(quorum_handle,
				1,
				key_change_notify,
				NULL, // object_create_notify
				NULL, // object_destroy_notify
				NULL, // object_reload_notify
				NULL); // priv_data

	/* Register for quorum changes too! */
	api->quorum_register_callback(quorum_callback, NULL);
}