From 2274c6c9610f9de70acbe8b16f421fe1438de369 Mon Sep 17 00:00:00 2001 From: Steven Dake Date: Tue, 28 Apr 2009 06:45:26 +0000 Subject: [PATCH] Add cpgverify program to test directory. git-svn-id: http://svn.fedorahosted.org/svn/corosync/trunk@2162 fd59a12c-fef9-0310-b244-a6a79926bd2f --- test/Makefile.am | 4 +- test/cpgverify.c | 158 +++++++++++++++++++++++++++++++++++++++++++++++ 2 files changed, 161 insertions(+), 1 deletion(-) create mode 100644 test/cpgverify.c diff --git a/test/Makefile.am b/test/Makefile.am index 783c38a5..f9852788 100644 --- a/test/Makefile.am +++ b/test/Makefile.am @@ -33,7 +33,7 @@ MAINTAINERCLEANFILES = Makefile.in INCLUDES = -I$(top_builddir)/include -I$(top_srcdir)/include -noinst_PROGRAMS = testevs evsbench evsverify testcpg testcpg2 cpgbench testconfdb \ +noinst_PROGRAMS = testevs evsbench evsverify cpgverify testcpg testcpg2 cpgbench testconfdb \ logsysbench logsysrec testquorum testvotequorum1 testvotequorum2 \ logsys_s logsys_t1 logsys_t2 testcpgzc cpgbenchzc testzcgc @@ -57,6 +57,8 @@ testvotequorum2_LDADD = -lvotequorum testvotequorum2_LDFLAGS = -L../lib evsverify_LDADD = -levs -ltotem_pg evsverify_LDFLAGS = -L../lib -L../exec +cpgverify_LDADD = -lcpg -ltotem_pg +cpgverify_LDFLAGS = -L../lib -L../exec evsbench_LDADD = -levs evsbench_LDFLAGS = -L../lib cpgbench_LDADD = -lcpg diff --git a/test/cpgverify.c b/test/cpgverify.c new file mode 100644 index 00000000..26b74005 --- /dev/null +++ b/test/cpgverify.c @@ -0,0 +1,158 @@ +/* + * Copyright (c) 2009 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 + +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include "../exec/crypto.h" + +struct my_msg { + unsigned int msg_size; + unsigned char sha1[20]; + unsigned char buffer[0]; +}; + +static int deliveries = 0; +static void cpg_deliver_fn ( + cpg_handle_t handle, + const struct cpg_name *group_name, + uint32_t nodeid, + uint32_t pid, + const void *m, + size_t msg_len) +{ + const struct my_msg *msg2 = m; + unsigned char sha1_compare[20]; + hash_state sha1_hash; + unsigned int i; + + + printf ("msg '%s'\n", msg2->buffer); + sha1_init (&sha1_hash); + sha1_process (&sha1_hash, msg2->buffer, msg2->msg_size); + sha1_done (&sha1_hash, sha1_compare); +printf ("SIZE %d HASH: ", msg2->msg_size); +for (i = 0; i < 20; i++) { +printf ("%x", sha1_compare[i]); +} +printf ("\n"); + if (memcmp (sha1_compare, msg2->sha1, 20) != 0) { + printf ("incorrect hash\n"); + exit (1); + } + deliveries++; +} + +static void cpg_confchg_fn ( + cpg_handle_t handle, + const struct cpg_name *group_name, + const struct cpg_address *member_list, size_t member_list_entries, + const struct cpg_address *left_list, size_t left_list_entries, + const struct cpg_address *joined_list, size_t joined_list_entries) +{ +} + +static cpg_callbacks_t callbacks = { + cpg_deliver_fn, + cpg_confchg_fn +}; + +static struct cpg_name group_name = { + .value = "cpg_bm", + .length = 6 +}; + + +static unsigned char buffer[200000]; +int main (void) +{ + cpg_handle_t handle; + cs_error_t result; + unsigned int i = 0, j; + struct my_msg msg; + hash_state sha1_hash; + struct iovec iov[2]; + int res; + + result = cpg_initialize (&handle, &callbacks); + if (result != CS_OK) { + printf ("Couldn't initialize CPG service %d\n", result); + exit (0); + } + + res = cpg_join (handle, &group_name); + if (res != CS_OK) { + printf ("cpg_join failed with result %d\n", res); + exit (1); + } + + iov[0].iov_base = &msg; + iov[0].iov_len = sizeof (struct my_msg); + iov[1].iov_base = buffer; + + /* + * Demonstrate cpg_mcast_joined + */ + for (i = 0; i < 1000000000; i++) { + msg.msg_size = 100 + rand() % 100000; + iov[1].iov_len = msg.msg_size; + for (j = 0; j < msg.msg_size; j++) { + buffer[j] = j; + } + sprintf ((char *)buffer, + "cpg_mcast_joined: This is message %12d", i); + sha1_init (&sha1_hash); + sha1_process (&sha1_hash, buffer, msg.msg_size); + sha1_done (&sha1_hash, msg.sha1); +try_again_one: + result = cpg_mcast_joined (handle, CPG_TYPE_AGREED, + iov, 2); + if (result == CS_ERR_TRY_AGAIN) { + goto try_again_one; + } + result = cpg_dispatch (handle, CS_DISPATCH_ALL); + } + + cpg_finalize (handle); + + return (0); +}