1*a1e26a70SApple OSS Distributions /* 2*a1e26a70SApple OSS Distributions * Copyright (c) 2022 Apple Inc. All rights reserved. 3*a1e26a70SApple OSS Distributions * 4*a1e26a70SApple OSS Distributions * @APPLE_OSREFERENCE_LICENSE_HEADER_START@ 5*a1e26a70SApple OSS Distributions * 6*a1e26a70SApple OSS Distributions * This file contains Original Code and/or Modifications of Original Code 7*a1e26a70SApple OSS Distributions * as defined in and that are subject to the Apple Public Source License 8*a1e26a70SApple OSS Distributions * Version 2.0 (the 'License'). You may not use this file except in 9*a1e26a70SApple OSS Distributions * compliance with the License. The rights granted to you under the License 10*a1e26a70SApple OSS Distributions * may not be used to create, or enable the creation or redistribution of, 11*a1e26a70SApple OSS Distributions * unlawful or unlicensed copies of an Apple operating system, or to 12*a1e26a70SApple OSS Distributions * circumvent, violate, or enable the circumvention or violation of, any 13*a1e26a70SApple OSS Distributions * terms of an Apple operating system software license agreement. 14*a1e26a70SApple OSS Distributions * 15*a1e26a70SApple OSS Distributions * Please obtain a copy of the License at 16*a1e26a70SApple OSS Distributions * http://www.opensource.apple.com/apsl/ and read it before using this file. 17*a1e26a70SApple OSS Distributions * 18*a1e26a70SApple OSS Distributions * The Original Code and all software distributed under the License are 19*a1e26a70SApple OSS Distributions * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER 20*a1e26a70SApple OSS Distributions * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES, 21*a1e26a70SApple OSS Distributions * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY, 22*a1e26a70SApple OSS Distributions * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT. 23*a1e26a70SApple OSS Distributions * Please see the License for the specific language governing rights and 24*a1e26a70SApple OSS Distributions * limitations under the License. 25*a1e26a70SApple OSS Distributions * 26*a1e26a70SApple OSS Distributions * @APPLE_OSREFERENCE_LICENSE_HEADER_END@ 27*a1e26a70SApple OSS Distributions */ 28*a1e26a70SApple OSS Distributions 29*a1e26a70SApple OSS Distributions #include <darwintest.h> 30*a1e26a70SApple OSS Distributions #include <TargetConditionals.h> 31*a1e26a70SApple OSS Distributions 32*a1e26a70SApple OSS Distributions #include <errno.h> 33*a1e26a70SApple OSS Distributions #include <stdio.h> 34*a1e26a70SApple OSS Distributions #include <stdlib.h> 35*a1e26a70SApple OSS Distributions #include <string.h> 36*a1e26a70SApple OSS Distributions #include <sys/mman.h> 37*a1e26a70SApple OSS Distributions 38*a1e26a70SApple OSS Distributions #include <mach/mach_error.h> 39*a1e26a70SApple OSS Distributions #include <mach/mach_init.h> 40*a1e26a70SApple OSS Distributions #include <mach/mach_vm.h> 41*a1e26a70SApple OSS Distributions 42*a1e26a70SApple OSS Distributions T_GLOBAL_META( 43*a1e26a70SApple OSS Distributions T_META_NAMESPACE("xnu.vm"), 44*a1e26a70SApple OSS Distributions T_META_RADAR_COMPONENT_NAME("xnu"), 45*a1e26a70SApple OSS Distributions T_META_RADAR_COMPONENT_VERSION("VM")); 46*a1e26a70SApple OSS Distributions 47*a1e26a70SApple OSS Distributions T_DECL(wire_copy_share, 48*a1e26a70SApple OSS Distributions "test VM object wired, copied and shared", T_META_TAG_VM_PREFERRED) 49*a1e26a70SApple OSS Distributions { 50*a1e26a70SApple OSS Distributions kern_return_t kr; 51*a1e26a70SApple OSS Distributions mach_vm_address_t vmaddr1, vmaddr2, vmaddr3; 52*a1e26a70SApple OSS Distributions mach_vm_size_t vmsize; 53*a1e26a70SApple OSS Distributions char *cp; 54*a1e26a70SApple OSS Distributions int i; 55*a1e26a70SApple OSS Distributions vm_prot_t cur_prot, max_prot; 56*a1e26a70SApple OSS Distributions int ret; 57*a1e26a70SApple OSS Distributions 58*a1e26a70SApple OSS Distributions /* allocate anonymous memory */ 59*a1e26a70SApple OSS Distributions vmaddr1 = 0; 60*a1e26a70SApple OSS Distributions vmsize = 32 * PAGE_SIZE; 61*a1e26a70SApple OSS Distributions kr = mach_vm_allocate( 62*a1e26a70SApple OSS Distributions mach_task_self(), 63*a1e26a70SApple OSS Distributions &vmaddr1, 64*a1e26a70SApple OSS Distributions vmsize, 65*a1e26a70SApple OSS Distributions VM_FLAGS_ANYWHERE); 66*a1e26a70SApple OSS Distributions T_QUIET; T_ASSERT_MACH_SUCCESS(kr, "vm_allocate()"); 67*a1e26a70SApple OSS Distributions 68*a1e26a70SApple OSS Distributions /* populate it */ 69*a1e26a70SApple OSS Distributions cp = (char *)(uintptr_t)vmaddr1; 70*a1e26a70SApple OSS Distributions for (i = 0; i < vmsize; i += PAGE_SIZE) { 71*a1e26a70SApple OSS Distributions cp[i] = i; 72*a1e26a70SApple OSS Distributions } 73*a1e26a70SApple OSS Distributions 74*a1e26a70SApple OSS Distributions /* wire one page */ 75*a1e26a70SApple OSS Distributions ret = mlock(cp, PAGE_SIZE); 76*a1e26a70SApple OSS Distributions T_QUIET; T_ASSERT_POSIX_SUCCESS(ret, "mlock()"); 77*a1e26a70SApple OSS Distributions 78*a1e26a70SApple OSS Distributions /* create a range to receive a copy */ 79*a1e26a70SApple OSS Distributions vmaddr2 = 0; 80*a1e26a70SApple OSS Distributions kr = mach_vm_allocate( 81*a1e26a70SApple OSS Distributions mach_task_self(), 82*a1e26a70SApple OSS Distributions &vmaddr2, 83*a1e26a70SApple OSS Distributions vmsize - PAGE_SIZE, 84*a1e26a70SApple OSS Distributions VM_FLAGS_ANYWHERE); 85*a1e26a70SApple OSS Distributions T_QUIET; T_ASSERT_MACH_SUCCESS(kr, "vm_allocate() for copy"); 86*a1e26a70SApple OSS Distributions 87*a1e26a70SApple OSS Distributions /* copy the rest of the original object */ 88*a1e26a70SApple OSS Distributions kr = mach_vm_copy( 89*a1e26a70SApple OSS Distributions mach_task_self(), 90*a1e26a70SApple OSS Distributions vmaddr1 + PAGE_SIZE, 91*a1e26a70SApple OSS Distributions vmsize - PAGE_SIZE, 92*a1e26a70SApple OSS Distributions vmaddr2); 93*a1e26a70SApple OSS Distributions T_QUIET; T_ASSERT_MACH_SUCCESS(kr, "vm_copy()"); 94*a1e26a70SApple OSS Distributions 95*a1e26a70SApple OSS Distributions /* share the whole thing */ 96*a1e26a70SApple OSS Distributions vmaddr3 = 0; 97*a1e26a70SApple OSS Distributions kr = mach_vm_remap( 98*a1e26a70SApple OSS Distributions mach_task_self(), 99*a1e26a70SApple OSS Distributions &vmaddr3, 100*a1e26a70SApple OSS Distributions vmsize, 101*a1e26a70SApple OSS Distributions 0, /* mask */ 102*a1e26a70SApple OSS Distributions VM_FLAGS_ANYWHERE, 103*a1e26a70SApple OSS Distributions mach_task_self(), 104*a1e26a70SApple OSS Distributions vmaddr1, 105*a1e26a70SApple OSS Distributions FALSE, /* copy */ 106*a1e26a70SApple OSS Distributions &cur_prot, 107*a1e26a70SApple OSS Distributions &max_prot, 108*a1e26a70SApple OSS Distributions VM_INHERIT_DEFAULT); 109*a1e26a70SApple OSS Distributions T_QUIET; T_ASSERT_MACH_SUCCESS(kr, "vm_remap()"); 110*a1e26a70SApple OSS Distributions } 111