Loading...
tests/magazine_small_test.c libmalloc-657.60.21 libmalloc-140.40.1
--- libmalloc/libmalloc-657.60.21/tests/magazine_small_test.c
+++ libmalloc/libmalloc-140.40.1/tests/magazine_small_test.c
@@ -1,5 +1,5 @@
 //
-//  magazine_small_test.c
+//  magazine_tiny_test.c
 //  libmalloc
 //
 //  Created by Matt Wright on 8/22/16.
@@ -11,22 +11,18 @@
 #include "../src/magazine_small.c"
 #include "magazine_testing.h"
 
-// Stubs
-bool aggressive_madvise_enabled = DEFAULT_AGGRESSIVE_MADVISE_ENABLED;
-malloc_zero_policy_t malloc_zero_policy = MALLOC_ZERO_POLICY_DEFAULT;
-
-T_GLOBAL_META(T_META_RUN_CONCURRENTLY(true), T_META_TAG_VM_PREFERRED);
-
 static inline void
-small_test_rack_setup(rack_t *rack)
+test_rack_setup(rack_t *rack)
 {
-	test_rack_setup(rack, RACK_TYPE_SMALL);
+	memset(rack, 'a', sizeof(rack));
+	rack_init(rack, RACK_TYPE_SMALL, 1, 0);
+	T_QUIET; T_ASSERT_NOTNULL(rack->magazines, "magazine initialisation");
 }
 
 T_DECL(basic_small_alloc, "small rack init and alloc")
 {
 	struct rack_s rack;
-	small_test_rack_setup(&rack);
+	test_rack_setup(&rack);
 
 	void *ptr = small_malloc_should_clear(&rack, SMALL_MSIZE_FOR_BYTES(512), false);
 	T_ASSERT_NOTNULL(ptr, "allocation");
@@ -38,10 +34,10 @@
 	T_ASSERT_EQ((int)sz, 512, "size == 32");
 }
 
-T_DECL(basic_small_teardown, "small rack init, alloc, teardown")
+T_DECL(basic_tiny_teardown, "small rack init, alloc, teardown")
 {
 	struct rack_s rack;
-	small_test_rack_setup(&rack);
+	test_rack_setup(&rack);
 
 	void *ptr = small_malloc_should_clear(&rack, TINY_MSIZE_FOR_BYTES(512), false);
 	T_ASSERT_NOTNULL(ptr, "allocation");
@@ -61,7 +57,7 @@
 T_DECL(basic_small_free, "small free")
 {
 	struct rack_s rack;
-	small_test_rack_setup(&rack);
+	test_rack_setup(&rack);
 
 	void *ptr = small_malloc_should_clear(&rack, SMALL_MSIZE_FOR_BYTES(512), false);
 	T_ASSERT_NOTNULL(ptr, "allocation");
@@ -76,7 +72,7 @@
 T_DECL(basic_small_shrink, "small rack shrink")
 {
 	struct rack_s rack;
-	small_test_rack_setup(&rack);
+	test_rack_setup(&rack);
 
 	void *ptr = small_malloc_should_clear(&rack, SMALL_MSIZE_FOR_BYTES(1024), false);
 	T_ASSERT_NOTNULL(ptr, "allocation");
@@ -93,71 +89,43 @@
 T_DECL(basic_small_realloc_in_place, "small rack realloc in place")
 {
 	struct rack_s rack;
-	small_test_rack_setup(&rack);
+	test_rack_setup(&rack);
 
-	// Allocate two blocks and free the second, then try to realloc() the first.
-	// This should extend in-place using the one-level death row cache that's
-	// occupied by the second block.
 	void *ptr = small_malloc_should_clear(&rack, SMALL_MSIZE_FOR_BYTES(512), false);
 	T_ASSERT_NOTNULL(ptr, "allocation");
 
 	size_t sz = small_size(&rack, ptr);
 	T_ASSERT_EQ((int)sz, 512, "size == 512");
 
+	// Due to <rdar://problem/28032870>, we can't just do alloc+realloc here because
+	// the allocator won't see the remainder of the region as free (the
+	// mag_bytes_free_at_end section is marked as busy and realloc doesn't check
+	// for it).
+
+	// On top of that, <rdar://problem/28033016> means we can't do
+	// alloc+alloc+free+realloc because the free'd block gets put in the small cache
+	// and the realloc path *also* doesn't check that. That means to actually do a
+	// fast-path realloc you need to:
+
+	//   1. alloc three times (to get the blocks out of the free_at_end region)
+	//   2. free the logically consecutive block
+	//   3. free any other block, to get the consecutive block out of the small cache
+
 	void *ptr2 = small_malloc_should_clear(&rack, SMALL_MSIZE_FOR_BYTES(512), false);
 	T_ASSERT_NOTNULL(ptr2, "allocation 2");
 	T_ASSERT_EQ_PTR(ptr2, (void *)((uintptr_t)ptr + 512), "sequential allocations");
 
+	void *ptr3 = small_malloc_should_clear(&rack, SMALL_MSIZE_FOR_BYTES(512), false);
+	T_ASSERT_NOTNULL(ptr3, "allocation 3");
+
 	free_small(&rack, ptr2, SMALL_REGION_FOR_PTR(ptr2), 0);
+	free_small(&rack, ptr3, SMALL_REGION_FOR_PTR(ptr3), 0);
 
-	// Attempt to realloc up to 1024 bytes, this should happen in place
-	// because of the death-row cache.
+	// attempt to realloc up to 1024 bytes, this should happen in place
+	// as the rack is otherwise completely empty
 	boolean_t reallocd = small_try_realloc_in_place(&rack, ptr, sz, 1024);
 	T_ASSERT_TRUE(reallocd, "realloced");
 
 	size_t nsz = small_size(&rack, ptr);
 	T_ASSERT_EQ((int)nsz, 1024, "realloc size == 1024");
-
-	// Try another realloc(). This should extend in place because the rest of
-	// the rack is empty.
-	reallocd = small_try_realloc_in_place(&rack, ptr, nsz, 2048);
-	T_ASSERT_TRUE(reallocd, "realloced #2");
-	nsz = small_size(&rack, ptr);
-	T_ASSERT_EQ((int)nsz, 2048, "realloc size == 2048");
-
-	free_small(&rack, ptr, SMALL_REGION_FOR_PTR(ptr), 0);
 }
-
-T_DECL(small_free_deallocate, "check small regions deallocate when empty")
-{
-	struct rack_s rack;
-	memset(&rack, 'a', sizeof(rack));
-	rack_init(&rack, RACK_TYPE_SMALL, 1, 0);
-	// force recirc to be released when empty
-	recirc_retained_regions = 0;
-
-	magazine_t *mag = &rack.magazines[0];
-	T_ASSERT_EQ(mag->mag_last_region, NULL, "no regions before allocation");
-
-	void *ptr = small_malloc_should_clear(&rack, 1, false);
-	T_ASSERT_NE(ptr, NULL, "allocate");
-
-	region_t region = SMALL_REGION_FOR_PTR(ptr);
-	mag_index_t mag_index = MAGAZINE_INDEX_FOR_SMALL_REGION(region);
-	magazine_t *dest_mag = &(rack.magazines[mag_index]);
-	T_ASSERT_GE(mag_index, -1, "assert ptr not in recirc");
-
-	// Is to pass the recirc discriminant in small_free_try_recirc_to_depot
-	// which normally prevents recirc unless 1.5x of a region has been
-	// allocated in the magazine as a whole.
-	dest_mag->num_bytes_in_magazine = (5 * SMALL_HEAP_SIZE);
-
-	SZONE_MAGAZINE_PTR_LOCK(dest_mag);
-	small_free_no_lock(&rack, dest_mag, mag_index, region, ptr, 1);
-
-	T_ASSERT_EQ(mag->mag_last_region, NULL, "no regions after last free");
-
-	magazine_t *depot = &rack.magazines[DEPOT_MAGAZINE_INDEX];
-	T_ASSERT_EQ(depot->mag_num_objects, 0, "no objects in depot after last free");
-	T_ASSERT_EQ(depot->num_bytes_in_magazine, 0ul, "no region in depot after last free");
-}