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