[src/trunk]: src/tests/rump/rumpkern test for rumpclient reconnect feature

[pooka <pooka%NetBSD.org@localhost>]

details:   https://anonhg.NetBSD.org/src/rev/868f9a02aa45
branches:  trunk
changeset: 761336:868f9a02aa45
user:      pooka <pooka%NetBSD.org@localhost>
date:      Mon Jan 24 17:51:29 2011 +0000

description:
test for rumpclient reconnect feature

diffstat:

 tests/rump/rumpkern/h_client/Makefile     |    4 +-
 tests/rump/rumpkern/h_client/h_reconcli.c |  120 ++++++++++++++++++++++++++++++
 tests/rump/rumpkern/t_sp.sh               |   13 +++-
 3 files changed, 135 insertions(+), 2 deletions(-)

diffs (188 lines):

diff -r 40ef8b11e092 -r 868f9a02aa45 tests/rump/rumpkern/h_client/Makefile
--- a/tests/rump/rumpkern/h_client/Makefile     Mon Jan 24 17:47:51 2011 +0000
+++ b/tests/rump/rumpkern/h_client/Makefile     Mon Jan 24 17:51:29 2011 +0000
@@ -1,4 +1,4 @@
-#      $NetBSD: Makefile,v 1.4 2011/01/06 07:00:28 pooka Exp $
+#      $NetBSD: Makefile,v 1.5 2011/01/24 17:51:29 pooka Exp $
 #
 
 .include <bsd.own.mk>
@@ -6,6 +6,7 @@
 TESTSDIR=      ${TESTSBASE}/rump/rumpkern/h_client
 
 TESTS_C+=      h_forkcli
+TESTS_C+=      h_reconcli
 TESTS_C+=      h_sigcli
 TESTS_C+=      h_simplecli
 TESTS_C+=      h_stresscli
@@ -14,6 +15,7 @@
 
 LDADD+=                        -lrumpclient
 LDADD.h_stresscli=     -lpthread
+LDADD.h_reconcli=      -lpthread
 
 WARNS= 4
 NOMAN=
diff -r 40ef8b11e092 -r 868f9a02aa45 tests/rump/rumpkern/h_client/h_reconcli.c
--- /dev/null   Thu Jan 01 00:00:00 1970 +0000
+++ b/tests/rump/rumpkern/h_client/h_reconcli.c Mon Jan 24 17:51:29 2011 +0000
@@ -0,0 +1,120 @@
+/*     $NetBSD: h_reconcli.c,v 1.1 2011/01/24 17:51:29 pooka Exp $     */
+
+#include <sys/types.h>
+#include <sys/sysctl.h>
+
+#include <rump/rumpclient.h>
+#include <rump/rump_syscalls.h>
+
+#include <err.h>
+#include <pthread.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+
+static volatile int quit, riseandwhine;
+
+static pthread_mutex_t closermtx;
+static pthread_cond_t closercv;
+
+static void *
+closer(void *arg)
+{
+
+       pthread_mutex_lock(&closermtx);
+       while (!quit) {
+               while (!riseandwhine)
+                       pthread_cond_wait(&closercv, &closermtx);
+               riseandwhine = 0;
+               pthread_mutex_unlock(&closermtx);
+
+               /* try to catch a random slot */
+               usleep(random() % 100000);
+
+               /*
+                * wide-angle disintegration beam, but takes care
+                * of the client rumpkernel communication socket.
+                */
+               closefrom(3);
+
+               pthread_mutex_lock(&closermtx);
+       }
+       pthread_mutex_unlock(&closermtx);
+
+       return NULL;
+}
+
+static const int hostnamemib[] = { CTL_KERN, KERN_HOSTNAME };
+static char goodhostname[128];
+
+static void *
+worker(void *arg)
+{
+       char hostnamebuf[128];
+       size_t blen;
+
+       pthread_mutex_lock(&closermtx);
+       while (!quit) {
+               pthread_mutex_unlock(&closermtx);
+               if (rump_sys_getpid() == -1)
+                       err(1, "getpid");
+
+               blen = sizeof(hostnamebuf);
+               memset(hostnamebuf, 0, sizeof(hostnamebuf));
+               if (rump_sys___sysctl(hostnamemib, __arraycount(hostnamemib),
+                   hostnamebuf, &blen, NULL, 0) == -1)
+                       err(1, "sysctl");
+               if (strcmp(hostnamebuf, goodhostname) != 0)
+                       exit(1);
+               pthread_mutex_lock(&closermtx);
+               riseandwhine = 1;
+               pthread_cond_signal(&closercv);
+       }
+       riseandwhine = 1;
+       pthread_cond_signal(&closercv);
+       pthread_mutex_unlock(&closermtx);
+
+       return NULL;
+}
+
+int
+main(int argc, char *argv[])
+{
+       pthread_t pt, w1, w2, w3, w4;
+       size_t blen;
+       int timecount;
+
+       if (argc != 2)
+               errx(1, "need timecount");
+       timecount = atoi(argv[1]);
+       if (timecount <= 0)
+               errx(1, "invalid timecount %d\n", timecount);
+
+       srandom(time(NULL));
+
+       if (rumpclient_init() == -1)
+               err(1, "init");
+
+       blen = sizeof(goodhostname);
+       if (rump_sys___sysctl(hostnamemib, __arraycount(hostnamemib),
+           goodhostname, &blen, NULL, 0) == -1)
+               err(1, "sysctl");
+
+       pthread_create(&pt, NULL, closer, NULL);
+       pthread_create(&w1, NULL, worker, NULL);
+       pthread_create(&w2, NULL, worker, NULL);
+       pthread_create(&w3, NULL, worker, NULL);
+       pthread_create(&w4, NULL, worker, NULL);
+
+       sleep(timecount);
+       quit = 1;
+
+       pthread_join(pt, NULL);
+       pthread_join(w1, NULL);
+       pthread_join(w2, NULL);
+       pthread_join(w3, NULL);
+       pthread_join(w4, NULL);
+
+       exit(0);
+}
diff -r 40ef8b11e092 -r 868f9a02aa45 tests/rump/rumpkern/t_sp.sh
--- a/tests/rump/rumpkern/t_sp.sh       Mon Jan 24 17:47:51 2011 +0000
+++ b/tests/rump/rumpkern/t_sp.sh       Mon Jan 24 17:51:29 2011 +0000
@@ -1,4 +1,4 @@
-#      $NetBSD: t_sp.sh,v 1.9 2011/01/14 13:23:15 pooka Exp $
+#      $NetBSD: t_sp.sh,v 1.10 2011/01/24 17:51:29 pooka Exp $
 #
 # Copyright (c) 2010 The NetBSD Foundation, Inc.
 # All rights reserved.
@@ -49,6 +49,7 @@
 test_case fork_fakeauth fork fakeauth
 test_case sigsafe sigsafe sigsafe
 test_case signal signal
+test_case reconnect reconnect
 
 basic()
 {
@@ -92,6 +93,15 @@
        atf_check -s signal:27 $(atf_get_srcdir)/h_client/h_simplecli block
 }
 
+reconnect()
+{
+
+
+       export RUMP_SERVER=unix://commsock
+       atf_check -s exit:0 rump_server ${RUMP_SERVER}
+       atf_check -s exit:0 $(atf_get_srcdir)/h_client/h_reconcli 2
+}
+
 atf_init_test_cases()
 {
 
@@ -104,4 +114,5 @@
        atf_add_test_case fork_fakeauth
        atf_add_test_case sigsafe
        atf_add_test_case signal
+       atf_add_test_case reconnect
 }