1 /* 2 * Copyright (c) 2018 Apple Inc. All rights reserved. 3 * 4 * @APPLE_LICENSE_HEADER_START@ 5 * 6 * This file contains Original Code and/or Modifications of Original Code 7 * as defined in and that are subject to the Apple Public Source License 8 * Version 2.0 (the 'License'). You may not use this file except in 9 * compliance with the License. Please obtain a copy of the License at 10 * http://www.opensource.apple.com/apsl/ and read it before using this 11 * file. 12 * 13 * The Original Code and all software distributed under the License are 14 * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER 15 * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES, 16 * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY, 17 * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT. 18 * Please see the License for the specific language governing rights and 19 * limitations under the License. 20 * 21 * @APPLE_LICENSE_HEADER_END@ 22 */ 23 24 #include <darwintest.h> 25 26 #include <stdlib.h> 27 #include <unistd.h> 28 #include <string.h> 29 #include <mach/mach_vm.h> 30 #include <mach/mach_init.h> 31 #include <sys/resource.h> 32 #include <libproc.h> 33 #include <libproc_internal.h> 34 #include <TargetConditionals.h> 35 36 T_GLOBAL_META(T_META_RUN_CONCURRENTLY(true)); 37 38 #define ALLOC_SIZE_LARGE 5*1024*1024 39 #define ALLOC_SIZE_SMALL 2*1024*1024 40 41 int proc_rlimit_control(pid_t pid, int flavor, void *arg); 42 43 T_DECL(phys_footprint_interval_max, 44 "Validate physical footprint interval tracking") 45 { 46 int ret; 47 struct rusage_info_v4 ru; 48 mach_vm_address_t addr = (mach_vm_address_t)NULL; 49 50 ret = proc_pid_rusage(getpid(), RUSAGE_INFO_V4, (rusage_info_t *)&ru); 51 T_QUIET; 52 T_ASSERT_POSIX_SUCCESS(ret, "proc_pid_rusage"); 53 T_ASSERT_EQ(ru.ri_lifetime_max_phys_footprint, ru.ri_interval_max_phys_footprint, 54 "Max footprint and interval footprint are equal prior to dirtying memory"); 55 56 ret = mach_vm_allocate(mach_task_self(), &addr, (mach_vm_size_t)ALLOC_SIZE_LARGE, VM_FLAGS_ANYWHERE); 57 T_QUIET; 58 T_ASSERT_MACH_SUCCESS(ret, "mach_vm_allocate(ALLOC_SIZE_LARGE)"); 59 60 memset((void *)addr, 0xab, ALLOC_SIZE_LARGE); 61 62 ret = proc_pid_rusage(getpid(), RUSAGE_INFO_V4, (rusage_info_t *)&ru); 63 T_QUIET; 64 T_ASSERT_POSIX_SUCCESS(ret, "proc_pid_rusage"); 65 T_ASSERT_EQ(ru.ri_lifetime_max_phys_footprint, ru.ri_interval_max_phys_footprint, 66 "Max footprint and interval footprint are equal after dirtying large memory region"); 67 68 mach_vm_deallocate(mach_task_self(), addr, (mach_vm_size_t)ALLOC_SIZE_LARGE); 69 70 ret = proc_pid_rusage(getpid(), RUSAGE_INFO_V4, (rusage_info_t *)&ru); 71 T_QUIET; 72 T_ASSERT_POSIX_SUCCESS(ret, "proc_pid_rusage"); 73 T_ASSERT_EQ(ru.ri_lifetime_max_phys_footprint, ru.ri_interval_max_phys_footprint, 74 "Max footprint and interval footprint are still equal after freeing large memory region"); 75 76 ret = proc_reset_footprint_interval(getpid()); 77 T_ASSERT_POSIX_SUCCESS(ret, "proc_reset_footprint_interval()"); 78 79 ret = proc_pid_rusage(getpid(), RUSAGE_INFO_V4, (rusage_info_t *)&ru); 80 T_QUIET; 81 T_ASSERT_POSIX_SUCCESS(ret, "proc_pid_rusage"); 82 T_ASSERT_GT(ru.ri_lifetime_max_phys_footprint, ru.ri_interval_max_phys_footprint, 83 "Max footprint is greater than interval footprint after resetting interval"); 84 85 ret = mach_vm_allocate(mach_task_self(), &addr, (mach_vm_size_t)ALLOC_SIZE_SMALL, VM_FLAGS_ANYWHERE); 86 T_QUIET; 87 T_ASSERT_MACH_SUCCESS(ret, "mach_vm_allocate(ALLOC_SIZE_SMALL)"); 88 memset((void *)addr, 0xab, ALLOC_SIZE_SMALL); 89 90 ret = proc_pid_rusage(getpid(), RUSAGE_INFO_V4, (rusage_info_t *)&ru); 91 T_QUIET; 92 T_ASSERT_POSIX_SUCCESS(ret, "proc_pid_rusage"); 93 T_ASSERT_GT(ru.ri_lifetime_max_phys_footprint, ru.ri_interval_max_phys_footprint, 94 "Max footprint is still greater than interval footprint after dirtying small memory region"); 95 } 96