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
This commit is contained in:
parent
771f2275a6
commit
e094e235ab
|
@ -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>
|
.include <bsd.own.mk>
|
||||||
|
|
||||||
LIB= kernspace
|
LIB= kernspace
|
||||||
SRCS= thread.c busypage.c tsleep.c
|
SRCS= thread.c busypage.c tsleep.c alloc.c
|
||||||
|
|
||||||
RUMPTOP=${NETBSDSRCDIR}/sys/rump
|
RUMPTOP=${NETBSDSRCDIR}/sys/rump
|
||||||
|
|
||||||
|
|
|
@ -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';
|
||||||
|
}
|
||||||
|
}
|
|
@ -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.
|
* Copyright (c) 2010 The NetBSD Foundation, Inc.
|
||||||
|
@ -34,5 +34,6 @@ void rumptest_busypage(void);
|
||||||
void rumptest_threadjoin(void);
|
void rumptest_threadjoin(void);
|
||||||
void rumptest_thread(void);
|
void rumptest_thread(void);
|
||||||
void rumptest_tsleep(void);
|
void rumptest_tsleep(void);
|
||||||
|
void rumptest_alloc(size_t);
|
||||||
|
|
||||||
#endif /* _TESTS_RUMP_KERNSPACE_KERNSPACE_H_ */
|
#endif /* _TESTS_RUMP_KERNSPACE_KERNSPACE_H_ */
|
||||||
|
|
|
@ -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.
|
* Copyright (c) 2010 The NetBSD Foundation, Inc.
|
||||||
|
@ -34,6 +34,8 @@
|
||||||
#include <rump/rump.h>
|
#include <rump/rump.h>
|
||||||
|
|
||||||
#include <atf-c.h>
|
#include <atf-c.h>
|
||||||
|
#include <stdio.h>
|
||||||
|
#include <stdlib.h>
|
||||||
|
|
||||||
#include "../../h_macros.h"
|
#include "../../h_macros.h"
|
||||||
#include "../kernspace/kernspace.h"
|
#include "../kernspace/kernspace.h"
|
||||||
|
@ -55,9 +57,35 @@ ATF_TC_BODY(busypage, tc)
|
||||||
rump_unschedule();
|
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_TCS(tp)
|
||||||
{
|
{
|
||||||
|
|
||||||
ATF_TP_ADD_TC(tp, busypage);
|
ATF_TP_ADD_TC(tp, busypage);
|
||||||
|
ATF_TP_ADD_TC(tp, uvmwait);
|
||||||
|
|
||||||
return atf_no_error();
|
return atf_no_error();
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue