1*aca3beaaSApple OSS Distributions #include <darwintest.h> 2*aca3beaaSApple OSS Distributions 3*aca3beaaSApple OSS Distributions #include <errno.h> 4*aca3beaaSApple OSS Distributions #include <stdio.h> 5*aca3beaaSApple OSS Distributions #include <stdlib.h> 6*aca3beaaSApple OSS Distributions #include <string.h> 7*aca3beaaSApple OSS Distributions 8*aca3beaaSApple OSS Distributions #include <sys/mman.h> 9*aca3beaaSApple OSS Distributions 10*aca3beaaSApple OSS Distributions #include <mach/mach_error.h> 11*aca3beaaSApple OSS Distributions #include <mach/mach_init.h> 12*aca3beaaSApple OSS Distributions #include <mach/mach_vm.h> 13*aca3beaaSApple OSS Distributions 14*aca3beaaSApple OSS Distributions #define ANON_MAX_PAGES 0xFFFFFFFFULL 15*aca3beaaSApple OSS Distributions #define ANON_MAX_SIZE (ANON_MAX_PAGES * vm_kernel_page_size) 16*aca3beaaSApple OSS Distributions 17*aca3beaaSApple OSS Distributions T_DECL(anon_max_size, "Test an ALLOC_MAX_SIZE allocation", 18*aca3beaaSApple OSS Distributions T_META_NAMESPACE("xnu.vm"), 19*aca3beaaSApple OSS Distributions T_META_RADAR_COMPONENT_NAME("xnu"), 20*aca3beaaSApple OSS Distributions T_META_RADAR_COMPONENT_VERSION("VM"), 21*aca3beaaSApple OSS Distributions T_META_REQUIRES_SYSCTL_EQ("kern.development", 1)) 22*aca3beaaSApple OSS Distributions { 23*aca3beaaSApple OSS Distributions kern_return_t kr; 24*aca3beaaSApple OSS Distributions mach_vm_address_t vm_addr; 25*aca3beaaSApple OSS Distributions mach_vm_size_t vm_size; 26*aca3beaaSApple OSS Distributions unsigned char *cp; 27*aca3beaaSApple OSS Distributions int ret; 28*aca3beaaSApple OSS Distributions unsigned char vec; 29*aca3beaaSApple OSS Distributions 30*aca3beaaSApple OSS Distributions /* allocate the largest anonymous size possible */ 31*aca3beaaSApple OSS Distributions vm_size = ANON_MAX_SIZE; 32*aca3beaaSApple OSS Distributions /* truncate to avoid going over actual max size when rounding up */ 33*aca3beaaSApple OSS Distributions vm_size &= ~PAGE_MASK; 34*aca3beaaSApple OSS Distributions vm_addr = 0; 35*aca3beaaSApple OSS Distributions kr = mach_vm_allocate(mach_task_self(), 36*aca3beaaSApple OSS Distributions &vm_addr, 37*aca3beaaSApple OSS Distributions vm_size, 38*aca3beaaSApple OSS Distributions VM_FLAGS_ANYWHERE | VM_FLAGS_PURGABLE); 39*aca3beaaSApple OSS Distributions if (kr == KERN_NO_SPACE) { 40*aca3beaaSApple OSS Distributions T_SKIP("not enough address space..."); 41*aca3beaaSApple OSS Distributions } 42*aca3beaaSApple OSS Distributions T_QUIET; T_ASSERT_MACH_SUCCESS(kr, "vm_allocate(0x%llx)", ANON_MAX_SIZE); 43*aca3beaaSApple OSS Distributions 44*aca3beaaSApple OSS Distributions /* dirty the first and last pages */ 45*aca3beaaSApple OSS Distributions cp = (unsigned char *)(uintptr_t)vm_addr; 46*aca3beaaSApple OSS Distributions cp[0] = 'a'; 47*aca3beaaSApple OSS Distributions cp[vm_size - 1] = 'z'; 48*aca3beaaSApple OSS Distributions 49*aca3beaaSApple OSS Distributions /* trigger the VM compressor for that VM object */ 50*aca3beaaSApple OSS Distributions ret = madvise(cp, (size_t)vm_size, MADV_PAGEOUT); 51*aca3beaaSApple OSS Distributions T_QUIET; T_ASSERT_POSIX_SUCCESS(ret, "madvise(MADV_PAGEOUT)"); 52*aca3beaaSApple OSS Distributions 53*aca3beaaSApple OSS Distributions /* wait for the pages to be (asynchronously) compressed */ 54*aca3beaaSApple OSS Distributions T_QUIET; T_LOG("waiting for first page to be paged out..."); 55*aca3beaaSApple OSS Distributions do { 56*aca3beaaSApple OSS Distributions ret = mincore(&cp[0], 1, (char *)&vec); 57*aca3beaaSApple OSS Distributions T_QUIET; T_ASSERT_POSIX_SUCCESS(ret, "mincore(1st)"); 58*aca3beaaSApple OSS Distributions } while (vec & MINCORE_INCORE); 59*aca3beaaSApple OSS Distributions T_QUIET; T_LOG("waiting for last page to be paged out..."); 60*aca3beaaSApple OSS Distributions do { 61*aca3beaaSApple OSS Distributions ret = mincore(&cp[vm_size - 1], 1, (char *)&vec); 62*aca3beaaSApple OSS Distributions T_QUIET; T_ASSERT_POSIX_SUCCESS(ret, "mincore(last)"); 63*aca3beaaSApple OSS Distributions } while (vec & MINCORE_INCORE); 64*aca3beaaSApple OSS Distributions 65*aca3beaaSApple OSS Distributions /* trigger pageins and check the contents */ 66*aca3beaaSApple OSS Distributions T_QUIET; T_ASSERT_EQ(cp[0], 'a', "first page intact"); 67*aca3beaaSApple OSS Distributions T_QUIET; T_ASSERT_EQ(cp[vm_size - 1], 'z', "last page intact"); 68*aca3beaaSApple OSS Distributions 69*aca3beaaSApple OSS Distributions /* success */ 70*aca3beaaSApple OSS Distributions return; 71*aca3beaaSApple OSS Distributions } 72