Loading...
tests/posix_memalign_test.c libmalloc-521.120.7 libmalloc-792.80.2
--- libmalloc/libmalloc-521.120.7/tests/posix_memalign_test.c
+++ libmalloc/libmalloc-792.80.2/tests/posix_memalign_test.c
@@ -11,14 +11,38 @@
 #include <stdio.h>
 #include <stdlib.h>
 
+#include "../src/platform.h"
+
+#if !MALLOC_TARGET_EXCLAVES
 #include <mach/mach.h>
+#endif // !MALLOC_TARGET_EXCLAVES
 
 #include <malloc/malloc.h>
 #include <malloc_private.h>
 
 
-T_GLOBAL_META(T_META_RUN_CONCURRENTLY(true), T_META_TAG_XZONE);
-
+union memtag_ptr {
+	uint64_t value;
+
+	struct {
+		uint64_t ptr_bits : 56;
+		uint64_t ptr_tag : 4;
+		uint64_t ptr_upper : 4;
+	};
+};
+
+static uint8_t *
+memtag_strip_address(uint8_t *tagged_addr)
+{
+	union memtag_ptr p = {
+			.value = (uint64_t)tagged_addr,
+	};
+	return (uint8_t *)p.ptr_bits;
+}
+
+T_GLOBAL_META(T_META_RUN_CONCURRENTLY(true));
+
+#if !MALLOC_TARGET_EXCLAVES
 struct t_recorder_ctx {
 	void *ptr;
 	size_t size;
@@ -54,6 +78,8 @@
 static void
 check_pointer_is_enumerated(void *ptr, size_t size)
 {
+	// Under MTE, `ptr` is tagged, but the enumerator reports canonical addresses.
+	ptr = memtag_strip_address(ptr);
 
 	vm_address_t *zones;
 	unsigned zone_count;
@@ -81,6 +107,7 @@
 	}
 	T_QUIET; T_FAIL("pointer %p not enumerated in any zone", ptr);
 }
+#endif // !MALLOC_TARGET_EXCLAVES
 
 static inline void *
 t_posix_memalign(size_t alignment, size_t size, bool scribble, bool enumerate)
@@ -94,9 +121,11 @@
 	T_QUIET; T_EXPECT_LE(size, allocated_size, "allocation size");
 
 	T_QUIET; T_EXPECT_TRUE(malloc_claimed_address(ptr), "should be claimed");
+#if !MALLOC_TARGET_EXCLAVES
 	if (enumerate) {
 		check_pointer_is_enumerated(ptr, size);
 	}
+#endif // !MALLOC_TARGET_EXCLAVES
 
 	if (scribble) {
 		// Scribble memory pointed to by `ptr` to make sure we're not using that
@@ -108,7 +137,8 @@
 	return ptr;
 }
 
-T_DECL(posix_memalign_free, "posix_memalign all power of two alignments <= 4096")
+T_DECL(posix_memalign_free, "posix_memalign all power of two alignments <= 4096",
+	   T_META_TAG_VM_PREFERRED, T_META_TAG_ALL_ALLOCATORS)
 {
 	for (size_t alignment = sizeof(void*); alignment < 4096; alignment *= 2) {
 		bool enumerate = true;
@@ -121,7 +151,9 @@
 	}
 }
 
-T_DECL(posix_memalign_alignment_not_a_power_of_2, "posix_memalign should return EINVAL if alignment is not a power of 2")
+T_DECL(posix_memalign_alignment_not_a_power_of_2,
+	   "posix_memalign should return EINVAL if alignment is not a power of 2",
+	   T_META_TAG_VM_PREFERRED, T_META_TAG_ALL_ALLOCATORS)
 {
 	{
 		void *ptr = NULL;
@@ -138,7 +170,9 @@
 	}
 }
 
-T_DECL(posix_memalign_alignment_not_a_multiple_of_voidstar, "posix_memalign should return EINVAL if alignment is not a multiple of sizeof(void*)")
+T_DECL(posix_memalign_alignment_not_a_multiple_of_voidstar,
+	   "posix_memalign should return EINVAL if alignment is not a multiple of sizeof(void*)",
+	   T_META_TAG_VM_PREFERRED, T_META_TAG_ALL_ALLOCATORS)
 {
 	void *ptr = NULL;
 	const size_t alignment = sizeof(void*)+1;
@@ -147,7 +181,9 @@
 	T_QUIET; T_ASSERT_EQ(result, EINVAL, "posix_memalign should return EINVAL");
 }
 
-T_DECL(posix_memalign_allocate_size_0, "posix_memalign should return something that can be passed to free() when size is 0")
+T_DECL(posix_memalign_allocate_size_0,
+       "posix_memalign should return something that can be passed to free() when size is 0",
+	   T_META_TAG_VM_PREFERRED, T_META_TAG_ALL_ALLOCATORS)
 {
 	void *ptr = NULL;
 	int result = posix_memalign(&ptr, 8, 0);
@@ -156,7 +192,8 @@
 }
 
 #if defined(__LP64__)
-T_DECL(posix_memalign_large, "posix_memalign large power of two alignments")
+T_DECL(posix_memalign_large, "posix_memalign large power of two alignments",
+		T_META_TAG_VM_NOT_PREFERRED, T_META_TAG_ALL_ALLOCATORS)
 {
 	// 64GB on macOS, 64MB on embedded
 	uint64_t max_alignment = TARGET_OS_OSX ? UINT64_C(68719476736) : UINT64_C(67108864);
@@ -169,3 +206,19 @@
 	T_END;
 }
 #endif // __LP64__
+
+T_DECL(posix_memalign_single_page_large,
+		"Allocate tiny blocks with large alignment",
+		T_META_TAG_VM_PREFERRED, T_META_TAG_XZONE_ONLY)
+{
+	void *ptr1 = t_posix_memalign(32768, 1, false, true);
+	T_ASSERT_NOTNULL(ptr1, "Allocated aligned ptr %p", ptr1);
+	void *ptr2 = t_posix_memalign(16*1024, 48 * 1024, false, true);
+	T_ASSERT_NOTNULL(ptr2, "Allocated aligned ptr %p", ptr2);
+	void *ptr3 = t_posix_memalign(32768, 1, false, true);
+	T_ASSERT_NOTNULL(ptr3, "Allocated aligned ptr %p", ptr3);
+
+	free(ptr1);
+	free(ptr2);
+	free(ptr3);
+}