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