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