summaryrefslogtreecommitdiffstats
path: root/test/testvotequorum2.c
diff options
context:
space:
mode:
Diffstat (limited to 'test/testvotequorum2.c')
-rw-r--r--test/testvotequorum2.c236
1 files changed, 236 insertions, 0 deletions
diff --git a/test/testvotequorum2.c b/test/testvotequorum2.c
new file mode 100644
index 0000000..1e8c13f
--- /dev/null
+++ b/test/testvotequorum2.c
@@ -0,0 +1,236 @@
+/*
+ * Copyright (c) 2009-2014 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 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 CONTIBUTORS "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 <inttypes.h>
+#include <stdio.h>
+#include <stdint.h>
+#include <string.h>
+#include <stdlib.h>
+#include <unistd.h>
+#include <corosync/corotypes.h>
+#include <corosync/votequorum.h>
+
+static votequorum_handle_t handle;
+
+static votequorum_ring_id_t last_received_ring_id;
+static votequorum_ring_id_t ring_id_to_send;
+static int no_sent_old_ringid = 0;
+
+static int print_info(int ok_to_fail)
+{
+ struct votequorum_info info;
+ int err;
+
+ if ( (err=votequorum_getinfo(handle, VOTEQUORUM_QDEVICE_NODEID, &info)) != CS_OK) {
+ fprintf(stderr, "votequorum_getinfo error %d: %s\n", err, ok_to_fail?"OK":"FAILED");
+ return -1;
+ }
+ else {
+ printf("name %s\n", info.qdevice_name);
+ printf("qdevice votes %d\n", info.qdevice_votes);
+ if (info.flags & VOTEQUORUM_INFO_QDEVICE_ALIVE) {
+ printf("alive ");
+ }
+ if (info.flags & VOTEQUORUM_INFO_QDEVICE_CAST_VOTE) {
+ printf("cast-vote ");
+ }
+ if (info.flags & VOTEQUORUM_INFO_QDEVICE_MASTER_WINS) {
+ printf("master-wins");
+ }
+ printf("\n\n");
+ }
+ return 0;
+}
+
+static void votequorum_nodelist_notification_fn(
+ votequorum_handle_t vq_handle,
+ uint64_t context,
+ votequorum_ring_id_t ring_id,
+ uint32_t node_list_entries,
+ uint32_t node_list[])
+{
+
+ printf("votequorum nodelist notification called \n");
+ printf(" current ringid = (" CS_PRI_RING_ID ")\n", ring_id.nodeid, ring_id.seq);
+ printf("\n");
+
+ memcpy(&last_received_ring_id, &ring_id, sizeof(ring_id));
+ no_sent_old_ringid = 0;
+}
+
+static void usage(const char *command)
+{
+ printf("%s [-F <num>] [-p <num>] [-t <time>] [-n <name>] [-c] [-m]\n", command);
+ printf(" -F <num> Number of times to send old ringid before actual ringid is sent (for testing, default = 0)\n");
+ printf(" -p <num> Number of times to poll qdevice (default 0=infinte)\n");
+ printf(" -t <secs> Time (in seconds) to wait between polls (default=1)\n");
+ printf(" -n <name> Name of quorum device (default QDEVICE)\n");
+ printf(" -c Don't cast vote (default is to cast vote)\n");
+ printf(" -q Don't print device status every poll time (default=will print)\n");
+ printf(" -m Master wins (default no)\n");
+ printf(" -1 Print status once and exit\n");
+}
+
+int main(int argc, char *argv[])
+{
+ int ret = 0;
+ int cast_vote = 1, master_wins = 0;
+ int pollcount=0, polltime=1, quiet=0, once=0;
+ int send_old_ringid = 0;
+ int err;
+ int opt;
+ votequorum_callbacks_t callbacks;
+ const char *devicename = "QDEVICE";
+ const char *options = "F:n:p:t:cmq1h";
+
+ memset(&callbacks, 0, sizeof(callbacks));
+ callbacks.votequorum_nodelist_notify_fn = votequorum_nodelist_notification_fn;
+
+ while ((opt = getopt(argc, argv, options)) != -1) {
+ switch (opt) {
+ case 'm':
+ master_wins = 1;
+ break;
+ case 'c':
+ cast_vote = 0;
+ break;
+ case '1':
+ once = 1;
+ break;
+ case 'q':
+ quiet = 1;
+ break;
+ case 'F':
+ send_old_ringid = atoi(optarg)+1;
+ break;
+ case 'p':
+ pollcount = atoi(optarg)+1;
+ break;
+ case 'n':
+ devicename = strdup(optarg);
+ break;
+ case 't':
+ polltime = atoi(optarg);
+ break;
+ case 'h':
+ usage(argv[0]);
+ exit(0);
+ }
+ }
+
+ if ( (err=votequorum_initialize(&handle, &callbacks)) != CS_OK) {
+ fprintf(stderr, "votequorum_initialize FAILED: %d\n", err);
+ return -1;
+ }
+
+ if (quiet && once) {
+ fprintf(stderr, "setting both -q (quet) and -1 (once) makes no sense\n");
+ usage(argv[0]);
+ exit(1);
+ }
+
+ if (!quiet) {
+ print_info(1);
+ }
+ if (once) {
+ exit(0);
+ }
+
+ if (argc >= 2) {
+ if ( (err = votequorum_trackstart(handle, handle, CS_TRACK_CHANGES)) != CS_OK) {
+ fprintf(stderr, "votequorum_trackstart FAILED: %d\n", err);
+ ret = -1;
+ goto out;
+ }
+
+ if ( (err=votequorum_qdevice_register(handle, devicename)) != CS_OK) {
+ fprintf(stderr, "qdevice_register FAILED: %d\n", err);
+ ret = -1;
+ goto out;
+ }
+
+ if ( (err=votequorum_qdevice_master_wins(handle, devicename, master_wins)) != CS_OK) {
+ fprintf(stderr, "qdevice_master_wins FAILED: %d\n", err);
+ ret = -1;
+ goto out;
+ }
+
+
+ while (--pollcount) {
+ if (votequorum_dispatch(handle, CS_DISPATCH_ALL) != CS_OK) {
+ fprintf(stderr, "votequorum_dispatch error\n");
+ ret = -1;
+ goto out;
+ }
+
+ if (!quiet) print_info(0);
+
+ if (no_sent_old_ringid + 1 >= send_old_ringid) {
+ /*
+ * Finally send correct ringid
+ */
+ memcpy(&ring_id_to_send, &last_received_ring_id, sizeof(ring_id_to_send));
+ } else {
+ no_sent_old_ringid++;
+ }
+
+ if ((err=votequorum_qdevice_poll(handle, devicename, cast_vote, ring_id_to_send)) != CS_OK &&
+ err != CS_ERR_MESSAGE_ERROR) {
+ fprintf(stderr, "qdevice poll FAILED: %d\n", err);
+ ret = -1;
+ goto out;
+ }
+ if (err == CS_ERR_MESSAGE_ERROR) {
+ fprintf(stderr, "qdevice poll passed OLD ring_id\n");
+ }
+
+ if (!quiet) print_info(0);
+ sleep(polltime);
+ }
+ if ((err= votequorum_qdevice_unregister(handle, devicename)) != CS_OK) {
+ fprintf(stderr, "qdevice unregister FAILED: %d\n", err);
+ ret = -1;
+ goto out;
+ }
+ }
+
+ if (!quiet) print_info(1);
+
+out:
+ votequorum_finalize(handle);
+ return ret;
+}