[src/trunk]: src/tests/rump add a vm allocator test which tests that:

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

details:   https://anonhg.NetBSD.org/src/rev/281a65574035
branches:  trunk
changeset: 755670:281a65574035
user:      pooka <pooka%NetBSD.org@localhost>
date:      Mon Jun 14 21:06:09 2010 +0000

description:
add a vm allocator test which tests that:
* RUMP_MEMLIMIT works
* allocating memory with PR_NOWAIT will fail immediately if no memory
  is available and PR_WAITOK will wait for memory to be available

diffstat:

 tests/rump/kernspace/Makefile    |    4 +-
 tests/rump/kernspace/alloc.c     |  122 +++++++++++++++++++++++++++++++++++++++
 tests/rump/kernspace/kernspace.h |    3 +-
 tests/rump/rumpkern/t_vm.c       |   30 +++++++++-
 4 files changed, 155 insertions(+), 4 deletions(-)

diffs (212 lines):

diff -r 76d330102249 -r 281a65574035 tests/rump/kernspace/Makefile
--- a/tests/rump/kernspace/Makefile     Mon Jun 14 21:04:56 2010 +0000
+++ b/tests/rump/kernspace/Makefile     Mon Jun 14 21:06:09 2010 +0000
@@ -1,10 +1,10 @@
-#      $NetBSD: Makefile,v 1.2 2010/05/31 23:53:02 pooka Exp $
+#      $NetBSD: Makefile,v 1.3 2010/06/14 21:06:09 pooka Exp $
 #
 
 .include <bsd.own.mk>
 
 LIB=   kernspace
-SRCS=  thread.c busypage.c tsleep.c
+SRCS=  thread.c busypage.c tsleep.c alloc.c
 
 RUMPTOP=${NETBSDSRCDIR}/sys/rump
 
diff -r 76d330102249 -r 281a65574035 tests/rump/kernspace/alloc.c
--- /dev/null   Thu Jan 01 00:00:00 1970 +0000
+++ b/tests/rump/kernspace/alloc.c      Mon Jun 14 21:06:09 2010 +0000
@@ -0,0 +1,122 @@
+/*     $NetBSD: alloc.c,v 1.1 2010/06/14 21:06:09 pooka Exp $  */
+
+/*-
+ * Copyright (c) 2010 The NetBSD Foundation, Inc.
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ * 1. Redistributions of source code must retain the above copyright
+ *    notice, this list of conditions and the following disclaimer.
+ * 2. 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.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. 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 FOUNDATION 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 <sys/cdefs.h>
+#if !defined(lint)
+__RCSID("$NetBSD: alloc.c,v 1.1 2010/06/14 21:06:09 pooka Exp $");
+#endif /* !lint */
+
+#include <sys/param.h>
+#include <sys/condvar.h>
+#include <sys/kmem.h>
+#include <sys/kthread.h>
+#include <sys/mutex.h>
+#include <sys/pool.h>
+#include <sys/proc.h>
+
+#include <uvm/uvm.h>
+
+#include <rump/rumpuser.h>
+#include "kernspace.h"
+
+static void *store[32];
+static struct pool pp1, pp2;
+
+static kmutex_t mtx;
+static kcondvar_t kcv;
+static int curstat;
+
+static void
+hthr(void *arg)
+{
+       int i;
+
+       mutex_enter(&mtx);
+       curstat++;
+       cv_signal(&kcv);
+
+       while (curstat < 2)
+               cv_wait(&kcv, &mtx);
+       mutex_exit(&mtx);
+
+       /* try to guarantee that the sleep is triggered in PR_WAITOK */
+       while ((kernel_map->flags & VM_MAP_WANTVA) == 0)
+               kpause("take5", false, 1, NULL);
+
+       for (i = 0; i < __arraycount(store); i++) {
+               pool_put(&pp1, store[i]);
+       }
+
+       kthread_exit(0);
+}
+
+void
+rumptest_alloc(size_t thelimit)
+{
+       char *c;
+       int succ, i;
+
+       mutex_init(&mtx, MUTEX_DEFAULT, IPL_NONE);
+       cv_init(&kcv, "venailu");
+
+       kthread_create(PRI_NONE, KTHREAD_MPSAFE, NULL, hthr, NULL, NULL, "h");
+
+       pool_init(&pp1, 1024, 0, 0, 0, "vara-allas",
+           &pool_allocator_nointr, IPL_NONE);
+       pool_init(&pp2, 1024, 0, 0, 0, "allas",
+           &pool_allocator_nointr, IPL_NONE);
+       
+       for (i = 0; i < __arraycount(store); i++) {
+               store[i] = pool_get(&pp1, PR_NOWAIT);
+               if (store[i] == NULL) {
+                       panic("pool_get store failed");
+               }
+       }
+
+       /* wait until other thread runs */
+       mutex_enter(&mtx);
+       while (curstat == 0)
+               cv_wait(&kcv, &mtx);
+       mutex_exit(&mtx);
+
+       for (succ = 0;; succ++) {
+               if (succ * 1024 > thelimit)
+                       panic("managed to allocate over limit");
+               if ((c = pool_get(&pp2, PR_NOWAIT)) == NULL) {
+                       mutex_enter(&mtx);
+                       curstat++;
+                       cv_signal(&kcv);
+                       mutex_exit(&mtx);
+                       if (pool_get(&pp2, PR_WAITOK) == NULL)
+                               panic("pool get PR_WAITOK failed");
+                       break;
+               }
+               *c = 'a';
+       }
+}
diff -r 76d330102249 -r 281a65574035 tests/rump/kernspace/kernspace.h
--- a/tests/rump/kernspace/kernspace.h  Mon Jun 14 21:04:56 2010 +0000
+++ b/tests/rump/kernspace/kernspace.h  Mon Jun 14 21:06:09 2010 +0000
@@ -1,4 +1,4 @@
-/*     $NetBSD: kernspace.h,v 1.1 2010/05/31 23:32:51 pooka Exp $      */
+/*     $NetBSD: kernspace.h,v 1.2 2010/06/14 21:06:09 pooka Exp $      */
 
 /*-
  * Copyright (c) 2010 The NetBSD Foundation, Inc.
@@ -34,5 +34,6 @@
 void rumptest_threadjoin(void);
 void rumptest_thread(void);
 void rumptest_tsleep(void);
+void rumptest_alloc(size_t);
 
 #endif /* _TESTS_RUMP_KERNSPACE_KERNSPACE_H_ */
diff -r 76d330102249 -r 281a65574035 tests/rump/rumpkern/t_vm.c
--- a/tests/rump/rumpkern/t_vm.c        Mon Jun 14 21:04:56 2010 +0000
+++ b/tests/rump/rumpkern/t_vm.c        Mon Jun 14 21:06:09 2010 +0000
@@ -1,4 +1,4 @@
-/*     $NetBSD: t_vm.c,v 1.1 2010/05/31 23:36:12 pooka Exp $   */
+/*     $NetBSD: t_vm.c,v 1.2 2010/06/14 21:06:09 pooka Exp $   */
 
 /*-
  * Copyright (c) 2010 The NetBSD Foundation, Inc.
@@ -34,6 +34,8 @@
 #include <rump/rump.h>
 
 #include <atf-c.h>
+#include <stdio.h>
+#include <stdlib.h>
 
 #include "../../h_macros.h"
 #include "../kernspace/kernspace.h"
@@ -55,9 +57,35 @@
        rump_unschedule();
 }
 
+ATF_TC(uvmwait);
+ATF_TC_HEAD(uvmwait, tc)
+{
+
+       atf_tc_set_md_var(tc, "descr", "Tests that uvm_wait works");
+       atf_tc_set_md_var(tc, "timeout", "10");
+}
+
+#define UVMWAIT_LIMIT 1024*1024
+ATF_TC_BODY(uvmwait, tc)
+{
+       char buf[64];
+
+       /* limit rump kernel memory */
+       snprintf(buf, sizeof(buf), "%d", UVMWAIT_LIMIT);
+       setenv("RUMP_MEMLIMIT", buf, 1);
+
+       rump_init();
+
+       rump_schedule();
+       rumptest_alloc(UVMWAIT_LIMIT);
+       rump_unschedule();
+}
+
 ATF_TP_ADD_TCS(tp)
 {
+
        ATF_TP_ADD_TC(tp, busypage);
+       ATF_TP_ADD_TC(tp, uvmwait);
 
        return atf_no_error();
 }