Loading...
--- libmalloc/libmalloc-474.0.13/tests/madvise.c
+++ libmalloc/libmalloc-116/tests/madvise.c
@@ -6,12 +6,7 @@
#include <mach/vm_page_size.h>
#include <stdlib.h>
-#include "base.h"
-#include "platform.h"
-#include "nano_zone_common.h"
-#include "nano_zone.h"
-
-extern int
+extern boolean_t
malloc_engaged_nano(void);
#define T_EXPECT_BYTES(p, len, byte, msg, ...) do { \
@@ -81,9 +76,7 @@
T_EXPECT_POSIX_SUCCESS(munmap(mem, memsz), "munmap");
}
-// <rdar://problem/31844360> disable nano_subpage_madvise due to consistent
-// failures.
-#if 0
+#ifdef __LP64__
T_DECL(nano_subpage_madvise, "nano allocator madvise",
T_META_SYSCTL_INT("vm.madvise_free_debug=1"),
@@ -94,10 +87,16 @@
T_EXPECT_TRUE(malloc_engaged_nano(), "nano zone enabled");
void *ptr = malloc(128);
+#ifdef __arm64__
T_EXPECT_EQ_PTR(
- (void *)(((uintptr_t)ptr) >> (64-NANO_SIGNATURE_BITS)),
- (void *)NANOZONE_SIGNATURE,
+ (void *)(((uintptr_t)ptr) >> (64-36)), (void *)0x000000017ULL,
"malloc == nano allocation");
+#else
+ T_EXPECT_EQ_PTR(
+ (void *)(((uintptr_t)ptr) >> (64-20)), (void *)0x00006ULL,
+ "malloc == nano allocation");
+#endif
+
free(ptr);
const size_t granularity = 128;
@@ -292,92 +291,3 @@
}
}
#endif // #if 0
-
-static void
-test_aggressive_madvise(const vm_size_t total_size, const size_t granularity)
-{
- void *bank[total_size / granularity];
-
- for (int i = 0; i < (sizeof(bank)/sizeof(*bank)); i++) {
- bank[i] = malloc(granularity);
- memset(bank[i], 'A', granularity);
- }
-
- // free all allocations to force as much as possible to be madvise'd
- for (int i = 0; i < (sizeof(bank)/sizeof(*bank)); i++) {
- free(bank[i]);
- }
-
- // find the first page aligned allocation that has
- // continguous allocations after it that total an entire page
- void *ptr = NULL;
- void *next_ptr = NULL;
- int num_needed = -1;
- for (int i = 0; i < (sizeof(bank)/sizeof(*bank)); i++) {
- if (ptr) {
- if (bank[i] == next_ptr) {
- num_needed--;
-
- if (num_needed == 0) {
- // found an entire page that should be free
- break;
- }
- } else {
- // start search again because there was a non-contiguous allocation
- ptr = NULL;
- }
- }
-
- if (!ptr && round_page((uintptr_t)bank[i]) == (uintptr_t)bank[i]) {
- // start a new search at this page aligned allocation
- ptr = next_ptr = bank[i];
- num_needed = ((vm_kernel_page_size + granularity - 1) / granularity) - 1;
-
- if (num_needed == 0) {
- // granularity is greater than or equal to a page
- break;
- }
- }
-
- if (ptr) {
- next_ptr = (void *)((uintptr_t)next_ptr + granularity);
- }
- }
-
- T_ASSERT_NOTNULL(ptr, "expected pointer");
- T_ASSERT_EQ(num_needed, 0, "need entire page");
-
- // Check the entire page-aligned range is empty.
- T_EXPECT_BYTES(ptr, vm_page_size, 0x0, "madvise'd memory is all zeros");
-}
-
-T_DECL(tiny_aggressive_madvise, "tiny allocator free with MallocAggressiveMadvise=1",
- T_META_SYSCTL_INT("vm.madvise_free_debug=1"),
- T_META_ENVVAR("MallocNanoZone=0"),
- T_META_ENVVAR("MallocAggressiveMadvise=1"),
- T_META_ENABLED(CONFIG_AGGRESSIVE_MADVISE),
- T_META_ASROOT(YES))
-{
- test_aggressive_madvise(vm_page_size * 4, 16);
-}
-
-T_DECL(small_aggressive_madvise, "small allocator free with MallocAggressiveMadvise=1",
- T_META_SYSCTL_INT("vm.madvise_free_debug=1"),
- T_META_ENVVAR("MallocAggressiveMadvise=1"),
- T_META_ENABLED(CONFIG_AGGRESSIVE_MADVISE),
- T_META_ASROOT(YES))
-{
- test_aggressive_madvise(vm_page_size * 8, 1536);
-}
-
-T_DECL(medium_aggressive_madvise, "medium allocator free with MallocAggressiveMadvise=1",
- T_META_SYSCTL_INT("vm.madvise_free_debug=1"),
- T_META_ENVVAR("MallocMediumZone=1"),
- T_META_ENVVAR("MallocMediumActivationThreshold=1"),
- T_META_ENVVAR("MallocAggressiveMadvise=1"),
- T_META_ENABLED(CONFIG_MEDIUM_ALLOCATOR),
- T_META_ENABLED(CONFIG_AGGRESSIVE_MADVISE),
- T_META_ASROOT(YES))
-{
- test_aggressive_madvise(16 * 64 * 1024, 64 * 1024);
-}