[apple/xnu.git] / tests / phys_footprint_interval_max.c

/*
 * Copyright (c) 2018 Apple Inc. All rights reserved.
 *
 * @APPLE_LICENSE_HEADER_START@
 *
 * This file contains Original Code and/or Modifications of Original Code
 * as defined in and that are subject to the Apple Public Source License
 * Version 2.0 (the 'License'). You may not use this file except in
 * compliance with the License. Please obtain a copy of the License at
 * http://www.opensource.apple.com/apsl/ and read it before using this
 * file.
 *
 * The Original Code and all software distributed under the License are
 * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
 * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
 * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
 * Please see the License for the specific language governing rights and
 * limitations under the License.
 *
 * @APPLE_LICENSE_HEADER_END@
 */

#include <darwintest.h>

#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <mach/mach_vm.h>
#include <mach/mach_init.h>
#include <sys/resource.h>
#include <libproc.h>
#include <libproc_internal.h>
#include <TargetConditionals.h>

T_GLOBAL_META(T_META_RUN_CONCURRENTLY(true));

#define ALLOC_SIZE_LARGE 5*1024*1024
#define ALLOC_SIZE_SMALL 2*1024*1024

int proc_rlimit_control(pid_t pid, int flavor, void *arg);

T_DECL(phys_footprint_interval_max,
    "Validate physical footprint interval tracking")
{
	int ret;
	struct rusage_info_v4 ru;
	mach_vm_address_t addr = (mach_vm_address_t)NULL;

	ret = proc_pid_rusage(getpid(), RUSAGE_INFO_V4, (rusage_info_t *)&ru);
	T_QUIET;
	T_ASSERT_POSIX_SUCCESS(ret, "proc_pid_rusage");
	T_ASSERT_EQ(ru.ri_lifetime_max_phys_footprint, ru.ri_interval_max_phys_footprint,
	    "Max footprint and interval footprint are equal prior to dirtying memory");

	ret = mach_vm_allocate(mach_task_self(), &addr, (mach_vm_size_t)ALLOC_SIZE_LARGE, VM_FLAGS_ANYWHERE);
	T_QUIET;
	T_ASSERT_MACH_SUCCESS(ret, "mach_vm_allocate(ALLOC_SIZE_LARGE)");

	memset((void *)addr, 0xab, ALLOC_SIZE_LARGE);

	ret = proc_pid_rusage(getpid(), RUSAGE_INFO_V4, (rusage_info_t *)&ru);
	T_QUIET;
	T_ASSERT_POSIX_SUCCESS(ret, "proc_pid_rusage");
	T_ASSERT_EQ(ru.ri_lifetime_max_phys_footprint, ru.ri_interval_max_phys_footprint,
	    "Max footprint and interval footprint are equal after dirtying large memory region");

	mach_vm_deallocate(mach_task_self(), addr, (mach_vm_size_t)ALLOC_SIZE_LARGE);

	ret = proc_pid_rusage(getpid(), RUSAGE_INFO_V4, (rusage_info_t *)&ru);
	T_QUIET;
	T_ASSERT_POSIX_SUCCESS(ret, "proc_pid_rusage");
	T_ASSERT_EQ(ru.ri_lifetime_max_phys_footprint, ru.ri_interval_max_phys_footprint,
	    "Max footprint and interval footprint are still equal after freeing large memory region");

	ret = proc_reset_footprint_interval(getpid());
	T_ASSERT_POSIX_SUCCESS(ret, "proc_reset_footprint_interval()");

	ret = proc_pid_rusage(getpid(), RUSAGE_INFO_V4, (rusage_info_t *)&ru);
	T_QUIET;
	T_ASSERT_POSIX_SUCCESS(ret, "proc_pid_rusage");
	T_ASSERT_GT(ru.ri_lifetime_max_phys_footprint, ru.ri_interval_max_phys_footprint,
	    "Max footprint is greater than interval footprint after resetting interval");

	ret = mach_vm_allocate(mach_task_self(), &addr, (mach_vm_size_t)ALLOC_SIZE_SMALL, VM_FLAGS_ANYWHERE);
	T_QUIET;
	T_ASSERT_MACH_SUCCESS(ret, "mach_vm_allocate(ALLOC_SIZE_SMALL)");
	memset((void *)addr, 0xab, ALLOC_SIZE_SMALL);

	ret = proc_pid_rusage(getpid(), RUSAGE_INFO_V4, (rusage_info_t *)&ru);
	T_QUIET;
	T_ASSERT_POSIX_SUCCESS(ret, "proc_pid_rusage");
	T_ASSERT_GT(ru.ri_lifetime_max_phys_footprint, ru.ri_interval_max_phys_footprint,
	    "Max footprint is still greater than interval footprint after dirtying small memory region");
}
Commit	Line	Data
d9a64523 A	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
cb323159 A	36	T_GLOBAL_META(T_META_RUN_CONCURRENTLY(true));
cb323159 A	37
d9a64523 A	38	#define ALLOC_SIZE_LARGE 510241024
	39	#define ALLOC_SIZE_SMALL 210241024
	40
	41	int proc_rlimit_control(pid_t pid, int flavor, void *arg);
	42
	43	T_DECL(phys_footprint_interval_max,
0a7de745	44	"Validate physical footprint interval tracking")
d9a64523 A	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,
0a7de745	54	"Max footprint and interval footprint are equal prior to dirtying memory");
d9a64523 A	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,
0a7de745	66	"Max footprint and interval footprint are equal after dirtying large memory region");
d9a64523 A	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,
0a7de745	74	"Max footprint and interval footprint are still equal after freeing large memory region");
d9a64523 A	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,
0a7de745	83	"Max footprint is greater than interval footprint after resetting interval");
d9a64523 A	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,
0a7de745	94	"Max footprint is still greater than interval footprint after dirtying small memory region");
d9a64523	95	}