t_mlock.c revision 272343
1147072Sbrooks/* $NetBSD: t_mlock.c,v 1.5 2014/02/26 20:49:26 martin Exp $ */ 2147072Sbrooks 3147072Sbrooks/*- 4147072Sbrooks * Copyright (c) 2012 The NetBSD Foundation, Inc. 5147072Sbrooks * All rights reserved. 6147072Sbrooks * 7147072Sbrooks * This code is derived from software contributed to The NetBSD Foundation 8147072Sbrooks * by Jukka Ruohonen. 9147072Sbrooks * 10147072Sbrooks * Redistribution and use in source and binary forms, with or without 11147072Sbrooks * modification, are permitted provided that the following conditions 12147072Sbrooks * are met: 13147072Sbrooks * 1. Redistributions of source code must retain the above copyright 14147072Sbrooks * notice, this list of conditions and the following disclaimer. 15147072Sbrooks * 2. Redistributions in binary form must reproduce the above copyright 16147072Sbrooks * notice, this list of conditions and the following disclaimer in the 17147072Sbrooks * documentation and/or other materials provided with the distribution. 18147072Sbrooks * 19147072Sbrooks * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS 20147072Sbrooks * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED 21147072Sbrooks * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 22147072Sbrooks * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS 23147072Sbrooks * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 24147072Sbrooks * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 25147072Sbrooks * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 26147072Sbrooks * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 27147072Sbrooks * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 28147072Sbrooks * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 29147072Sbrooks * POSSIBILITY OF SUCH DAMAGE. 30147072Sbrooks */ 31147072Sbrooks#include <sys/cdefs.h> 32147072Sbrooks__RCSID("$NetBSD: t_mlock.c,v 1.5 2014/02/26 20:49:26 martin Exp $"); 33147072Sbrooks 34147072Sbrooks#include <sys/mman.h> 35147072Sbrooks#include <sys/resource.h> 36147072Sbrooks#include <sys/sysctl.h> 37147072Sbrooks#include <sys/wait.h> 38147072Sbrooks 39147072Sbrooks#include <errno.h> 40147072Sbrooks#include <atf-c.h> 41147072Sbrooks#include <stdint.h> 42147072Sbrooks#include <stdio.h> 43149399Sbrooks#include <stdlib.h> 44149399Sbrooks#include <unistd.h> 45149399Sbrooks 46147072Sbrooksstatic long page = 0; 47147072Sbrooks 48147072SbrooksATF_TC(mlock_clip); 49147072SbrooksATF_TC_HEAD(mlock_clip, tc) 50147072Sbrooks{ 51147072Sbrooks atf_tc_set_md_var(tc, "descr", "Test with mlock(2) that UVM only " 52147072Sbrooks "clips if the clip address is within the entry (PR kern/44788)"); 53147072Sbrooks} 54147072Sbrooks 55147072SbrooksATF_TC_BODY(mlock_clip, tc) 56147072Sbrooks{ 57147072Sbrooks void *buf; 58147072Sbrooks 59147072Sbrooks buf = malloc(page); 60147072Sbrooks ATF_REQUIRE(buf != NULL); 61147072Sbrooks 62147072Sbrooks if (page < 1024) 63147072Sbrooks atf_tc_skip("page size too small"); 64147072Sbrooks 65147072Sbrooks for (size_t i = page; i >= 1; i = i - 1024) { 66147072Sbrooks (void)mlock(buf, page - i); 67147072Sbrooks (void)munlock(buf, page - i); 68147072Sbrooks } 69147072Sbrooks 70147072Sbrooks free(buf); 71147072Sbrooks} 72147072Sbrooks 73147072SbrooksATF_TC(mlock_err); 74147072SbrooksATF_TC_HEAD(mlock_err, tc) 75147072Sbrooks{ 76147072Sbrooks atf_tc_set_md_var(tc, "descr", 77147072Sbrooks "Test error conditions in mlock(2) and munlock(2)"); 78147072Sbrooks} 79147072Sbrooks 80147072SbrooksATF_TC_BODY(mlock_err, tc) 81147072Sbrooks{ 82147072Sbrooks unsigned long vmin = 0; 83147072Sbrooks size_t len = sizeof(vmin); 84147072Sbrooks void *invalid_ptr; 85147072Sbrooks int null_errno = ENOMEM; /* error expected for NULL */ 86147072Sbrooks 87147072Sbrooks if (sysctlbyname("vm.minaddress", &vmin, &len, NULL, 0) != 0) 88147072Sbrooks atf_tc_fail("failed to read vm.minaddress"); 89147072Sbrooks 90147072Sbrooks if (vmin > 0) 91147072Sbrooks null_errno = EINVAL; /* NULL is not inside user VM */ 92147072Sbrooks 93147072Sbrooks errno = 0; 94147072Sbrooks ATF_REQUIRE_ERRNO(null_errno, mlock(NULL, page) == -1); 95147072Sbrooks 96147072Sbrooks errno = 0; 97147072Sbrooks ATF_REQUIRE_ERRNO(null_errno, mlock((char *)0, page) == -1); 98147072Sbrooks 99147072Sbrooks errno = 0; 100147072Sbrooks ATF_REQUIRE_ERRNO(EINVAL, mlock((char *)-1, page) == -1); 101147072Sbrooks 102147072Sbrooks errno = 0; 103147072Sbrooks ATF_REQUIRE_ERRNO(null_errno, munlock(NULL, page) == -1); 104147072Sbrooks 105147072Sbrooks errno = 0; 106147072Sbrooks ATF_REQUIRE_ERRNO(null_errno, munlock((char *)0, page) == -1); 107147072Sbrooks 108147072Sbrooks errno = 0; 109147072Sbrooks ATF_REQUIRE_ERRNO(EINVAL, munlock((char *)-1, page) == -1); 110147072Sbrooks 111147072Sbrooks /* 112147072Sbrooks * Try to create a pointer to an unmapped page - first after current 113147072Sbrooks * brk will likely do. 114147072Sbrooks */ 115147072Sbrooks invalid_ptr = (void*)(((uintptr_t)sbrk(0)+page) & ~(page-1)); 116147072Sbrooks printf("testing with (hopefully) invalid pointer %p\n", invalid_ptr); 117147072Sbrooks 118147072Sbrooks errno = 0; 119228614Sdim ATF_REQUIRE_ERRNO(ENOMEM, mlock(invalid_ptr, page) == -1); 120147072Sbrooks 121147072Sbrooks errno = 0; 122147072Sbrooks ATF_REQUIRE_ERRNO(ENOMEM, munlock(invalid_ptr, page) == -1); 123147072Sbrooks} 124147072Sbrooks 125147072SbrooksATF_TC(mlock_limits); 126147072SbrooksATF_TC_HEAD(mlock_limits, tc) 127147072Sbrooks{ 128228614Sdim atf_tc_set_md_var(tc, "descr", "Test system limits with mlock(2)"); 129228614Sdim} 130147072Sbrooks 131147072SbrooksATF_TC_BODY(mlock_limits, tc) 132228615Sdim{ 133147072Sbrooks struct rlimit res; 134147072Sbrooks void *buf; 135147072Sbrooks pid_t pid; 136147072Sbrooks int sta; 137147072Sbrooks 138147072Sbrooks buf = malloc(page); 139147072Sbrooks ATF_REQUIRE(buf != NULL); 140147072Sbrooks 141147072Sbrooks pid = fork(); 142147072Sbrooks ATF_REQUIRE(pid >= 0); 143147072Sbrooks 144147072Sbrooks if (pid == 0) { 145147072Sbrooks 146147072Sbrooks for (ssize_t i = page; i >= 2; i -= 100) { 147147072Sbrooks 148147072Sbrooks res.rlim_cur = i - 1; 149147072Sbrooks res.rlim_max = i - 1; 150147072Sbrooks 151147072Sbrooks (void)fprintf(stderr, "trying to lock %zd bytes " 152147072Sbrooks "with %zu byte limit\n", i, (size_t)res.rlim_cur); 153147072Sbrooks 154147072Sbrooks if (setrlimit(RLIMIT_MEMLOCK, &res) != 0) 155147072Sbrooks _exit(EXIT_FAILURE); 156147072Sbrooks 157147072Sbrooks errno = 0; 158147072Sbrooks 159147072Sbrooks if (mlock(buf, i) != -1 || errno != EAGAIN) { 160147072Sbrooks (void)munlock(buf, i); 161147072Sbrooks _exit(EXIT_FAILURE); 162147072Sbrooks } 163147072Sbrooks } 164147072Sbrooks 165147072Sbrooks _exit(EXIT_SUCCESS); 166147072Sbrooks } 167147072Sbrooks 168147072Sbrooks (void)wait(&sta); 169147072Sbrooks 170147072Sbrooks if (WIFEXITED(sta) == 0 || WEXITSTATUS(sta) != EXIT_SUCCESS) 171147072Sbrooks atf_tc_fail("mlock(2) locked beyond system limits"); 172147072Sbrooks 173147072Sbrooks free(buf); 174147072Sbrooks} 175147072Sbrooks 176147072SbrooksATF_TC(mlock_mmap); 177147072SbrooksATF_TC_HEAD(mlock_mmap, tc) 178147072Sbrooks{ 179147072Sbrooks atf_tc_set_md_var(tc, "descr", "Test mlock(2)-mmap(2) interaction"); 180147072Sbrooks} 181147072Sbrooks 182147072SbrooksATF_TC_BODY(mlock_mmap, tc) 183147072Sbrooks{ 184147072Sbrooks static const int flags = MAP_ANON | MAP_PRIVATE | MAP_WIRED; 185147072Sbrooks void *buf; 186147072Sbrooks 187147072Sbrooks /* 188147072Sbrooks * Make a wired RW mapping and check that mlock(2) 189147072Sbrooks * does not fail for the (already locked) mapping. 190147072Sbrooks */ 191147072Sbrooks buf = mmap(NULL, page, PROT_READ | PROT_WRITE, flags, -1, 0); 192147072Sbrooks 193147072Sbrooks ATF_REQUIRE(buf != MAP_FAILED); 194147072Sbrooks ATF_REQUIRE(mlock(buf, page) == 0); 195147072Sbrooks ATF_REQUIRE(munlock(buf, page) == 0); 196147072Sbrooks ATF_REQUIRE(munmap(buf, page) == 0); 197147072Sbrooks ATF_REQUIRE(munlock(buf, page) != 0); 198147072Sbrooks 199147072Sbrooks /* 200147072Sbrooks * But it should be impossible to mlock(2) a PROT_NONE mapping. 201147072Sbrooks */ 202147072Sbrooks buf = mmap(NULL, page, PROT_NONE, flags, -1, 0); 203147072Sbrooks 204147072Sbrooks ATF_REQUIRE(buf != MAP_FAILED); 205147072Sbrooks ATF_REQUIRE(mlock(buf, page) != 0); 206147072Sbrooks ATF_REQUIRE(munmap(buf, page) == 0); 207147072Sbrooks} 208147072Sbrooks 209147072SbrooksATF_TC(mlock_nested); 210147072SbrooksATF_TC_HEAD(mlock_nested, tc) 211147072Sbrooks{ 212147072Sbrooks atf_tc_set_md_var(tc, "descr", 213147072Sbrooks "Test that consecutive mlock(2) calls succeed"); 214147072Sbrooks} 215147072Sbrooks 216147072SbrooksATF_TC_BODY(mlock_nested, tc) 217147072Sbrooks{ 218147072Sbrooks const size_t maxiter = 100; 219147072Sbrooks void *buf; 220147072Sbrooks 221147072Sbrooks buf = malloc(page); 222147072Sbrooks ATF_REQUIRE(buf != NULL); 223147072Sbrooks 224147072Sbrooks for (size_t i = 0; i < maxiter; i++) 225147072Sbrooks ATF_REQUIRE(mlock(buf, page) == 0); 226147072Sbrooks 227147072Sbrooks ATF_REQUIRE(munlock(buf, page) == 0); 228147072Sbrooks free(buf); 229147072Sbrooks} 230147072Sbrooks 231147072SbrooksATF_TP_ADD_TCS(tp) 232147072Sbrooks{ 233147072Sbrooks 234147072Sbrooks page = sysconf(_SC_PAGESIZE); 235147072Sbrooks ATF_REQUIRE(page >= 0); 236147072Sbrooks 237147072Sbrooks ATF_TP_ADD_TC(tp, mlock_clip); 238147072Sbrooks ATF_TP_ADD_TC(tp, mlock_err); 239147072Sbrooks ATF_TP_ADD_TC(tp, mlock_limits); 240147072Sbrooks ATF_TP_ADD_TC(tp, mlock_mmap); 241147072Sbrooks ATF_TP_ADD_TC(tp, mlock_nested); 242147072Sbrooks 243147072Sbrooks return atf_no_error(); 244147072Sbrooks} 245147072Sbrooks