#include "internal.h"
#if CONFIG_MEDIUM_ALLOCATOR
#pragma mark meta header helpers
static MALLOC_INLINE uint64_t
medium_sliding_madvise_granularity(magazine_t *magazine)
{
if (magazine->mag_num_bytes_in_objects == 0) {
return MEDIUM_MADVISE_MIN;
}
return MAX(MEDIUM_MADVISE_MIN, 1 << (64 -
__builtin_clzl(magazine->mag_num_bytes_in_objects >> MEDIUM_MADVISE_SHIFT)));
}
static MALLOC_INLINE void
medium_madvise_header_mark_dirty(msize_t *headers, msize_t index,
msize_t msize)
{
headers[index] = msize;
headers[index + msize - 1] = msize;
}
static MALLOC_INLINE void
medium_madvise_header_mark_clean(msize_t *headers, msize_t index,
msize_t msize)
{
headers[index] = msize | MEDIUM_IS_ADVISED;
headers[index + msize - 1] = msize | MEDIUM_IS_ADVISED;
}
static MALLOC_INLINE void
medium_madvise_header_mark_middle(msize_t *headers, msize_t index)
{
headers[index] = 0;
}
static MALLOC_INLINE msize_t
medium_madvise_header_dirty_len(msize_t *headers, msize_t index)
{
if (headers[index] & MEDIUM_IS_ADVISED) {
return 0;
}
return headers[index] & ~MEDIUM_IS_ADVISED;
}
static MALLOC_INLINE void
medium_meta_header_set_is_free(msize_t *meta_headers, msize_t index, msize_t msize)
{
meta_headers[index] = msize | MEDIUM_IS_FREE;
}
static MALLOC_INLINE void
medium_meta_header_set_not_free(msize_t *meta_headers, msize_t index)
{
meta_headers[index] &= ~MEDIUM_IS_FREE;
}
static MALLOC_INLINE void
medium_meta_header_set_in_use(msize_t *meta_headers, msize_t index, msize_t msize)
{
meta_headers[index] = msize;
}
static MALLOC_INLINE void
medium_meta_header_set_middle(msize_t *meta_headers, msize_t index)
{
meta_headers[index] = 0;
}
static MALLOC_INLINE MALLOC_ALWAYS_INLINE
mag_index_t
medium_mag_get_thread_index(void)
{
#if CONFIG_MEDIUM_USES_HYPER_SHIFT
if (os_likely(_os_cpu_number_override == -1)) {
return _os_cpu_number() >> hyper_shift;
} else {
return _os_cpu_number_override >> hyper_shift;
}
#else // CONFIG_MEDIUM_USES_HYPER_SHIFT
if (os_likely(_os_cpu_number_override == -1)) {
return _os_cpu_number();
} else {
return _os_cpu_number_override;
}
#endif // CONFIG_MEDIUM_USES_HYPER_SHIFT
}
#pragma mark in-place free list
static MALLOC_INLINE void
medium_inplace_checksum_ptr(rack_t *rack, inplace_linkage_s *linkage, void *ptr)
{
uintptr_t checksum = free_list_gen_checksum((uintptr_t)ptr ^ rack->cookie ^ (uintptr_t)rack);
linkage->checksum = checksum;
linkage->ptr = ptr;
}
static MALLOC_INLINE free_list_t
medium_inplace_unchecksum_ptr(rack_t *rack, inplace_linkage_s *linkage)
{
if (linkage->checksum != (uint8_t)free_list_gen_checksum((uintptr_t)linkage->ptr ^ rack->cookie ^ (uintptr_t)rack)) {
free_list_checksum_botch(rack, linkage, linkage->ptr);
__builtin_trap();
}
return (free_list_t){ .p = linkage->ptr };
}
static MALLOC_INLINE free_list_t
medium_inplace_free_entry_get_previous(rack_t *rack, medium_inplace_free_entry_t ptr)
{
return medium_inplace_unchecksum_ptr(rack, &ptr->previous);
}
static MALLOC_INLINE void
medium_inplace_free_entry_set_previous(rack_t *rack, medium_inplace_free_entry_t entry, free_list_t previous)
{
medium_inplace_checksum_ptr(rack, &entry->previous, previous.p);
}
static MALLOC_INLINE free_list_t
medium_inplace_free_entry_get_next(rack_t *rack, medium_inplace_free_entry_t ptr)
{
return medium_inplace_unchecksum_ptr(rack, &ptr->next);
}
static MALLOC_INLINE void
medium_inplace_free_entry_set_next(rack_t *rack, medium_inplace_free_entry_t entry, free_list_t next)
{
medium_inplace_checksum_ptr(rack, &entry->next, next.p);
}
#pragma mark OOB free list
static MALLOC_INLINE boolean_t
medium_needs_oob_free_entry(void *ptr, msize_t msize)
{
return ((trunc_page_quanta((uintptr_t)ptr) == (uintptr_t)ptr) &&
(MEDIUM_BYTES_FOR_MSIZE(msize) >= vm_kernel_page_size));
}
static MALLOC_INLINE boolean_t
medium_is_oob_free_entry(free_list_t ptr)
{
medium_region_t region = MEDIUM_REGION_FOR_PTR(ptr.p);
return (((uintptr_t)ptr.p >= (uintptr_t)®ion->medium_oob_free_entries[0]) &&
((uintptr_t)ptr.p < (uintptr_t)®ion->medium_oob_free_entries[MEDIUM_OOB_COUNT]));
}
static MALLOC_INLINE void
medium_oob_free_entry_set_previous(oob_free_entry_t oobe, free_list_t previous)
{
oobe->prev = (uintptr_t)previous.p;
}
static MALLOC_INLINE free_list_t
medium_oob_free_entry_get_previous(oob_free_entry_t oobe)
{
return (free_list_t){ .p = (void *)oobe->prev };
}
static MALLOC_INLINE void
medium_oob_free_entry_set_next(oob_free_entry_t oobe, free_list_t next)
{
oobe->next = (uintptr_t)next.p;
}
static MALLOC_INLINE free_list_t
medium_oob_free_entry_get_next(oob_free_entry_t oobe)
{
return (free_list_t){ .p = (void *)oobe->next };
}
static MALLOC_INLINE void *
medium_oob_free_entry_get_ptr(oob_free_entry_t oobe)
{
medium_region_t region = MEDIUM_REGION_FOR_PTR(oobe);
uint16_t block = oobe->ptr & ~MEDIUM_IS_OOB;
return (void *)((uintptr_t)region + (block << SHIFT_MEDIUM_QUANTUM));
}
static MALLOC_INLINE void
medium_oob_free_entry_set_ptr(oob_free_entry_t oobe, void *ptr)
{
oobe->ptr = MEDIUM_IS_OOB | (MEDIUM_OFFSET_FOR_PTR(ptr) >> SHIFT_MEDIUM_QUANTUM);
}
static MALLOC_INLINE void
medium_oob_free_entry_set_free(oob_free_entry_t oobe)
{
oobe->prev = ~0;
oobe->next = ~0;
oobe->ptr = 0;
}
static MALLOC_INLINE oob_free_entry_t
medium_oob_free_find_empty(void *ptr, msize_t msize)
{
medium_region_t region = MEDIUM_REGION_FOR_PTR(ptr);
for (int i=0; i < MEDIUM_OOB_COUNT; i++) {
if (region->medium_oob_free_entries[i].ptr == 0) {
return ®ion->medium_oob_free_entries[i];
}
}
#if DEBUG_MALLOC
malloc_report(ASL_LEVEL_INFO, "used all slots of OOB entries\n");
#endif
return NULL;
}
static MALLOC_INLINE oob_free_entry_t
medium_oob_free_find_ptr(void *ptr, msize_t msize)
{
medium_region_t region = MEDIUM_REGION_FOR_PTR(ptr);
for (int i=0; i < MEDIUM_OOB_COUNT; i++) {
oob_free_entry_t oob = ®ion->medium_oob_free_entries[i];
if (medium_oob_free_entry_get_ptr(oob) == ptr &&
oob->ptr & MEDIUM_IS_OOB) {
return ®ion->medium_oob_free_entries[i];
}
}
return NULL;
}
#pragma mark generic free list
static MALLOC_INLINE void
medium_free_list_set_previous(rack_t *rack, free_list_t entry, free_list_t previous)
{
if (medium_is_oob_free_entry(entry)) {
medium_oob_free_entry_set_previous(entry.oob, previous);
} else {
medium_inplace_free_entry_set_previous(rack, entry.medium_inplace, previous);
}
}
static MALLOC_INLINE free_list_t
medium_free_list_get_previous(rack_t *rack, free_list_t ptr)
{
MALLOC_ASSERT(ptr.p);
if (medium_is_oob_free_entry(ptr)) {
return medium_oob_free_entry_get_previous(ptr.oob);
} else {
return medium_inplace_free_entry_get_previous(rack, ptr.medium_inplace);
}
}
static MALLOC_INLINE void
medium_free_list_set_next(rack_t *rack, free_list_t entry, free_list_t next)
{
if (medium_is_oob_free_entry(entry)) {
medium_oob_free_entry_set_next(entry.oob, next);
} else {
medium_inplace_free_entry_set_next(rack, entry.medium_inplace, next);
}
}
static MALLOC_INLINE free_list_t
medium_free_list_get_next(rack_t *rack, free_list_t ptr)
{
MALLOC_ASSERT(ptr.p);
if (medium_is_oob_free_entry(ptr)) {
return medium_oob_free_entry_get_next(ptr.oob);
} else {
return medium_inplace_free_entry_get_next(rack, ptr.medium_inplace);
}
}
static MALLOC_INLINE void *
medium_free_list_get_ptr(rack_t *rack, free_list_t ptr)
{
if (!ptr.p) {
return NULL;
} else if (medium_is_oob_free_entry(ptr)) {
return medium_oob_free_entry_get_ptr(ptr.oob);
} else {
return (void *)ptr.p;
}
}
static MALLOC_INLINE free_list_t
medium_free_list_from_ptr(rack_t *rack, void *ptr, msize_t msize)
{
MALLOC_ASSERT(msize);
free_list_t entry;
entry.p = ptr;
if (medium_needs_oob_free_entry(ptr, msize)) {
oob_free_entry_t oobe = medium_oob_free_find_empty(ptr, msize);
if (oobe) {
medium_oob_free_entry_set_ptr(oobe, ptr);
entry.oob = oobe;
}
}
return entry;
}
static MALLOC_INLINE void
medium_free_mark_free(rack_t *rack, free_list_t entry, msize_t msize)
{
void *ptr = medium_free_list_get_ptr(rack, entry);
msize_t *meta_headers = MEDIUM_META_HEADER_FOR_PTR(ptr);
uintptr_t start_index = MEDIUM_META_INDEX_FOR_PTR(ptr);
uintptr_t end_index = MEDIUM_META_INDEX_FOR_PTR(ptr + MEDIUM_BYTES_FOR_MSIZE(msize) - 1);
MALLOC_ASSERT(start_index <= end_index);
medium_meta_header_set_is_free(meta_headers, start_index, msize);
medium_meta_header_set_is_free(meta_headers, end_index, msize);
}
static MALLOC_INLINE void
medium_free_mark_middle(rack_t *rack, free_list_t entry, msize_t msize)
{
void *ptr = medium_free_list_get_ptr(rack, entry);
msize_t *meta_headers = MEDIUM_META_HEADER_FOR_PTR(ptr);
uintptr_t start_index = MEDIUM_META_INDEX_FOR_PTR(ptr);
uintptr_t end_index = MEDIUM_META_INDEX_FOR_PTR(ptr + MEDIUM_BYTES_FOR_MSIZE(msize) - 1);
MALLOC_ASSERT(start_index <= end_index);
MALLOC_ASSERT((meta_headers[start_index] & ~MEDIUM_IS_FREE) == msize);
medium_meta_header_set_middle(meta_headers, start_index);
medium_meta_header_set_middle(meta_headers, end_index);
}
static MALLOC_INLINE void
medium_free_mark_unfree(rack_t *rack, free_list_t entry, msize_t msize)
{
void *ptr = medium_free_list_get_ptr(rack, entry);
msize_t *meta_headers = MEDIUM_META_HEADER_FOR_PTR(ptr);
uintptr_t start_index = MEDIUM_META_INDEX_FOR_PTR(ptr);
uintptr_t end_index = MEDIUM_META_INDEX_FOR_PTR(ptr +
MEDIUM_BYTES_FOR_MSIZE(msize) - 1);
MALLOC_ASSERT(start_index <= end_index);
medium_meta_header_set_not_free(meta_headers, start_index);
medium_meta_header_set_not_free(meta_headers, end_index);
}
static MALLOC_INLINE unsigned int
medium_free_list_count(rack_t *rack, free_list_t ptr)
{
unsigned int count = 0;
while (ptr.p) {
count++;
ptr = medium_free_list_get_next(rack, ptr);
}
return count;
}
static free_list_t
medium_free_list_add_ptr(rack_t *rack, magazine_t *medium_mag_ptr, void *ptr, msize_t msize)
{
grain_t slot = MEDIUM_FREE_SLOT_FOR_MSIZE(rack, msize);
free_list_t free_head = medium_mag_ptr->mag_free_list[slot];
free_list_t free_ptr = medium_free_list_from_ptr(rack, ptr, msize);
#if DEBUG_MALLOC
if (LOG(szone, ptr)) {
malloc_report(ASL_LEVEL_INFO, "in %s, ptr=%p, msize=%d\n", __FUNCTION__, ptr, msize);
}
if (((uintptr_t)ptr) & (MEDIUM_QUANTUM - 1)) {
malloc_zone_error(rack->debug_flags, true, "medium_free_list_add_ptr: Unaligned ptr %p\n", ptr);
}
if (!msize) {
malloc_zone_error(rack->debug_flags, true, "medium_free_list_add_ptr: msize=0, %p\n", ptr);
}
#endif
medium_free_list_set_previous(rack, free_ptr, (free_list_t){ .p = NULL });
medium_free_list_set_next(rack, free_ptr, free_head);
medium_free_mark_free(rack, free_ptr, msize);
if (medium_free_list_get_ptr(rack, free_head)) {
#if DEBUG_MALLOC
if (medium_free_list_get_previous(szone, free_head)) {
malloc_zone_error(rack->debug_flags, true, "medium_free_list_add_ptr: Internal invariant broken (free_head->previous != NULL)\n"
"ptr=%p slot=%d free_head=%p previous=%p\n", ptr, slot, (void *)free_head, free_head->previous.p);
}
if (!MEDIUM_PTR_IS_FREE(medium_free_list_get_ptr(szone, free_head))) {
malloc_zone_error(rack->debug_flags, true, "medium_free_list_add_ptr: Internal invariant broken (free_head is not a free pointer)\n"
"ptr=%p slot=%d free_head=%p\n", ptr, slot, (void *)medium_free_list_get_ptr(szone, free_head));
}
#endif
medium_free_list_set_previous(rack, free_head, free_ptr);
} else {
BITMAPN_SET(medium_mag_ptr->mag_bitmap, slot);
}
medium_mag_ptr->mag_free_list[slot] = free_ptr;
return free_ptr;
}
static void
medium_free_list_remove_ptr_no_clear(rack_t *rack, magazine_t *medium_mag_ptr, free_list_t entry, msize_t msize)
{
grain_t slot = MEDIUM_FREE_SLOT_FOR_MSIZE(rack, msize);
free_list_t next, previous;
#if DEBUG_MALLOC
if (LOG(szone, ptr)) {
malloc_report(ASL_LEVEL_INFO, "In %s, ptr=%p, msize=%d\n", __FUNCTION__, ptr, msize);
}
#endif
previous = medium_free_list_get_previous(rack, entry);
next = medium_free_list_get_next(rack, entry);
if (!medium_free_list_get_ptr(rack, previous)) {
#if DEBUG_MALLOC
if (medium_mag_ptr->mag_free_list[slot] != ptr) {
malloc_zone_error(rack->debug_flags, true,
"medium_free_list_remove_ptr_no_clear: Internal invariant broken (medium_mag_ptr->mag_free_list[slot])\n"
"ptr=%p slot=%d msize=%d medium_mag_ptr->mag_free_list[slot]=%p\n", ptr, slot, msize,
(void *)medium_mag_ptr->mag_free_list[slot]);
return;
}
#endif
medium_mag_ptr->mag_free_list[slot] = next;
if (!medium_free_list_get_ptr(rack, next)) {
BITMAPN_CLR(medium_mag_ptr->mag_bitmap, slot);
}
} else {
free_list_t prev_next = medium_free_list_get_next(rack, previous);
if (medium_free_list_get_ptr(rack, prev_next) != medium_free_list_get_ptr(rack, entry)) {
malloc_zone_error(rack->debug_flags, true,
"medium_free_list_remove_ptr_no_clear: Internal invariant broken (next ptr of prev) for %p, prev_next=%p\n",
medium_free_list_get_ptr(rack, entry), medium_free_list_get_ptr(rack, prev_next));
__builtin_unreachable(); }
medium_free_list_set_next(rack, previous, next);
}
if (medium_free_list_get_ptr(rack, next)) {
free_list_t next_prev = medium_free_list_get_previous(rack, next);
if (medium_free_list_get_ptr(rack, next_prev) != medium_free_list_get_ptr(rack, entry)) {
malloc_zone_error(rack->debug_flags, true,
"medium_free_list_remove_ptr_no_clear: Internal invariant broken (prev ptr of next) for %p, next_prev=%p\n",
medium_free_list_get_ptr(rack, entry), medium_free_list_get_ptr(rack, next_prev));
__builtin_unreachable(); }
medium_free_list_set_previous(rack, next, previous);
}
if (medium_is_oob_free_entry(entry)) {
medium_oob_free_entry_set_free(entry.oob);
}
}
static void
medium_free_list_remove_ptr(rack_t *rack, magazine_t *medium_mag_ptr, free_list_t entry, msize_t msize)
{
medium_free_mark_middle(rack, entry, msize);
medium_free_list_remove_ptr_no_clear(rack, medium_mag_ptr, entry, msize);
}
static free_list_t
medium_free_list_find_by_ptr(rack_t *rack, magazine_t *medium_mag_ptr, void *ptr, msize_t msize)
{
if (*MEDIUM_METADATA_FOR_PTR(ptr) == (MEDIUM_IS_FREE | msize)) {
if (medium_needs_oob_free_entry(ptr, msize)) {
medium_region_t region = MEDIUM_REGION_FOR_PTR(ptr);
for (int i=0; i<MEDIUM_OOB_COUNT; i++) {
if (!region->medium_oob_free_entries[i].ptr) {
continue;
}
if (medium_oob_free_entry_get_ptr(®ion->medium_oob_free_entries[i]) == ptr) {
return (free_list_t){ .oob = ®ion->medium_oob_free_entries[i] };
}
}
}
return (free_list_t){ .p = ptr };
}
malloc_zone_error(rack->debug_flags, true,
"medium_free_list_find_by_ptr: ptr is not free (ptr metadata !MEDIUM_IS_FREE), "
"ptr=%p msize=%d metadata=0x%x\n", ptr, msize, *MEDIUM_METADATA_FOR_PTR(ptr));
__builtin_trap();
}
void
medium_finalize_region(rack_t *rack, magazine_t *medium_mag_ptr)
{
void *last_block, *previous_block;
msize_t last_msize, previous_msize, last_index;
free_list_t previous;
if (medium_mag_ptr->mag_bytes_free_at_end) {
last_block = MEDIUM_REGION_END(medium_mag_ptr->mag_last_region) - medium_mag_ptr->mag_bytes_free_at_end;
last_msize = MEDIUM_MSIZE_FOR_BYTES(medium_mag_ptr->mag_bytes_free_at_end);
last_index = MEDIUM_META_INDEX_FOR_PTR(last_block);
previous_msize = MEDIUM_PREVIOUS_MSIZE(last_block);
if (last_index && (previous_msize <= last_index)) {
previous_block = (void *)((uintptr_t)last_block - MEDIUM_BYTES_FOR_MSIZE(previous_msize));
if (MEDIUM_PTR_IS_FREE(previous_block)) {
previous = medium_free_list_find_by_ptr(rack, medium_mag_ptr, previous_block, previous_msize);
medium_free_list_remove_ptr(rack, medium_mag_ptr, previous, previous_msize);
last_block = previous_block;
last_msize += previous_msize;
}
}
medium_free_list_add_ptr(rack, medium_mag_ptr, last_block, last_msize);
medium_mag_ptr->mag_bytes_free_at_end = 0;
}
#if CONFIG_ASLR_INTERNAL
free_list_t next;
if (medium_mag_ptr->mag_bytes_free_at_start) {
last_block = MEDIUM_REGION_ADDRESS(medium_mag_ptr->mag_last_region);
last_msize = MEDIUM_MSIZE_FOR_BYTES(medium_mag_ptr->mag_bytes_free_at_start);
void *next_block = (void *)((uintptr_t)last_block + medium_mag_ptr->mag_bytes_free_at_start);
if (MEDIUM_PTR_IS_FREE(next_block)) {
msize_t next_msize = MEDIUM_PTR_SIZE(next_block);
next = medium_free_list_find_by_ptr(rack, medium_mag_ptr, next_block, next_msize);
medium_free_list_remove_ptr(rack, medium_mag_ptr, next, next_msize);
last_msize += next_msize;
}
medium_free_list_add_ptr(rack, medium_mag_ptr, last_block, last_msize);
medium_mag_ptr->mag_bytes_free_at_start = 0;
}
#endif
medium_mag_ptr->mag_last_region = NULL;
}
int
medium_free_detach_region(rack_t *rack, magazine_t *medium_mag_ptr, region_t r)
{
unsigned char *ptr = MEDIUM_REGION_ADDRESS(r);
msize_t *meta_headers = MEDIUM_META_HEADER_FOR_PTR(ptr);
uintptr_t start = (uintptr_t)MEDIUM_REGION_ADDRESS(r);
uintptr_t current = start;
uintptr_t limit = (uintptr_t)MEDIUM_REGION_END(r);
int total_alloc = 0;
while (current < limit) {
unsigned index = MEDIUM_META_INDEX_FOR_PTR(current);
msize_t msize_and_free = meta_headers[index];
boolean_t is_free = msize_and_free & MEDIUM_IS_FREE;
msize_t msize = msize_and_free & ~MEDIUM_IS_FREE;
if (!msize) {
#if DEBUG_MALLOC
boolean_t is_free = msize_and_free & MEDIUM_IS_FREE;
malloc_report(ASL_LEVEL_ERR, "*** medium_free_detach_region error with %p: msize=%d is_free=%d\n", (void *)current, msize, is_free);
#endif
break;
}
if (is_free) {
free_list_t entry = medium_free_list_find_by_ptr(rack, medium_mag_ptr, (void *)current, msize);
medium_free_list_remove_ptr_no_clear(rack, medium_mag_ptr, entry, msize);
} else {
total_alloc++;
}
current += MEDIUM_BYTES_FOR_MSIZE(msize);
}
return total_alloc;
}
size_t
medium_free_reattach_region(rack_t *rack, magazine_t *medium_mag_ptr, region_t r)
{
unsigned char *ptr = MEDIUM_REGION_ADDRESS(r);
msize_t *meta_headers = MEDIUM_META_HEADER_FOR_PTR(ptr);
uintptr_t start = (uintptr_t)MEDIUM_REGION_ADDRESS(r);
uintptr_t current = start;
uintptr_t limit = (uintptr_t)MEDIUM_REGION_END(r);
size_t total_alloc = 0;
while (current < limit) {
unsigned index = MEDIUM_META_INDEX_FOR_PTR(current);
msize_t msize_and_free = meta_headers[index];
boolean_t is_free = msize_and_free & MEDIUM_IS_FREE;
msize_t msize = msize_and_free & ~MEDIUM_IS_FREE;
if (!msize) {
#if DEBUG_MALLOC
malloc_report(ASL_LEVEL_ERR, "*** medium_free_reattach_region error with %p: msize=%d is_free=%d\n", (void *)current, msize, is_free);
#endif
break;
}
if (is_free) {
medium_free_list_add_ptr(rack, medium_mag_ptr, (void *)current, msize);
} else {
total_alloc += MEDIUM_BYTES_FOR_MSIZE(msize);
}
current += MEDIUM_BYTES_FOR_MSIZE(msize);
}
return total_alloc;
}
typedef struct _medium_advisory_s {
struct _medium_advisory_s *next;
size_t size;
} medium_advisory_s, *medium_advisory_t;
void
medium_free_scan_madvise_free(rack_t *rack, magazine_t *depot_ptr, region_t r)
{
uintptr_t start = (uintptr_t)MEDIUM_REGION_ADDRESS(r);
uintptr_t current = start;
uintptr_t limit = (uintptr_t)MEDIUM_REGION_END(r);
msize_t *meta_headers = MEDIUM_META_HEADER_FOR_PTR(start);
msize_t *madv_headers = MEDIUM_MADVISE_HEADER_FOR_PTR(start);
int cnt = 0;
medium_advisory_t advisories = NULL;
while (current < limit) {
unsigned index = MEDIUM_META_INDEX_FOR_PTR(current);
msize_t alloc_msize = meta_headers[index] & ~MEDIUM_IS_FREE;
bool alloc_is_free = meta_headers[index] & MEDIUM_IS_FREE;
if (alloc_is_free && alloc_msize == MEDIUM_MAX_MSIZE) {
#if DEBUG_MALLOC
malloc_report(ASL_LEVEL_ERR, "*** medium_free_scan_madvise_free "
"first block is all free! %p: msize=%d is_free=%d\n",
(void *)current, alloc_msize, alloc_is_free);
#endif
uintptr_t pgLo = round_page_kernel(start + sizeof(medium_inplace_free_entry_s) +
sizeof(msize_t));
uintptr_t pgHi = trunc_page_kernel(start - sizeof(msize_t) +
(NUM_MEDIUM_BLOCKS << SHIFT_MEDIUM_QUANTUM));
if (pgLo < pgHi) {
medium_advisory_t mat = (medium_advisory_t)pgLo;
mat->next = advisories;
mat->size = pgHi - pgLo;
}
break;
}
if (!alloc_msize) {
#if DEBUG_MALLOC
malloc_report(ASL_LEVEL_ERR, "*** medium_free_scan_madvise_free "
"error with %p: msize=%d is_free=%d\n", (void *)current,
msize, is_free);
#endif
break;
}
if (alloc_is_free) {
msize_t advise_msize = madv_headers[index]
& ~MEDIUM_IS_ADVISED;
bool run_is_advised = madv_headers[index]
& MEDIUM_IS_ADVISED;
if (advise_msize != alloc_msize || !run_is_advised) {
uintptr_t pgLo = round_page_kernel(current +
sizeof(medium_inplace_free_entry_s) + sizeof(msize_t));
uintptr_t pgHi = trunc_page_kernel(current - sizeof(msize_t) +
MEDIUM_BYTES_FOR_MSIZE(alloc_msize));
if (pgLo < pgHi) {
medium_advisory_t mat = (medium_advisory_t)pgLo;
mat->next = advisories;
mat->size = pgHi - pgLo;
}
memset(&madv_headers[index], 0, sizeof(uint16_t) * alloc_msize);
medium_madvise_header_mark_clean(madv_headers, index,
alloc_msize);
}
}
current += MEDIUM_BYTES_FOR_MSIZE(alloc_msize);
}
if (cnt > 0) {
OSAtomicIncrement32Barrier(
&(REGION_TRAILER_FOR_MEDIUM_REGION(r)->pinned_to_depot));
SZONE_MAGAZINE_PTR_UNLOCK(depot_ptr);
while (advisories) {
uintptr_t addr = (uintptr_t)advisories;
size_t size = advisories->size;
advisories = advisories->next;
mvm_madvise_free(rack, r, addr, addr + size, NULL,
rack->debug_flags & MALLOC_DO_SCRIBBLE);
}
SZONE_MAGAZINE_PTR_LOCK(depot_ptr);
OSAtomicDecrement32Barrier(
&(REGION_TRAILER_FOR_MEDIUM_REGION(r)->pinned_to_depot));
}
}
static region_t
medium_find_msize_region(rack_t *rack, magazine_t *medium_mag_ptr, mag_index_t mag_index, msize_t msize)
{
void *ptr;
grain_t slot = MEDIUM_FREE_SLOT_FOR_MSIZE(rack, msize);
free_list_t *free_list = medium_mag_ptr->mag_free_list;
free_list_t *the_slot = free_list + slot;
free_list_t *limit;
unsigned bitmap;
CHECK_MAGAZINE_PTR_LOCKED(szone, medium_mag_ptr, __PRETTY_FUNCTION__);
ptr = medium_free_list_get_ptr(rack, *the_slot);
if (ptr) {
return MEDIUM_REGION_FOR_PTR(ptr);
}
unsigned idx = slot >> 5;
bitmap = 0;
unsigned mask = ~((1 << (slot & 31)) - 1);
for (; idx < MEDIUM_FREELIST_BITMAP_WORDS(rack); ++idx) {
bitmap = medium_mag_ptr->mag_bitmap[idx] & mask;
if (bitmap != 0) {
break;
}
mask = ~0U;
}
if ((bitmap == 0) && (idx == MEDIUM_FREELIST_BITMAP_WORDS(rack))) {
return NULL;
}
slot = BITMAP32_CTZ((&bitmap)) + (idx * 32);
limit = free_list + MEDIUM_FREE_SLOT_COUNT(rack) - 1;
free_list += slot;
if (free_list < limit) {
ptr = medium_free_list_get_ptr(rack, *free_list);
if (ptr) {
return MEDIUM_REGION_FOR_PTR(ptr);
} else {
#if DEBUG_MALLOC
malloc_report(ASL_LEVEL_ERR, "in medium_malloc_from_free_list(), mag_bitmap out of sync, slot=%d\n", slot);
#endif
}
}
ptr = medium_free_list_get_ptr(rack, *limit);
if (ptr) {
return MEDIUM_REGION_FOR_PTR(ptr);
}
return NULL;
}
static boolean_t
medium_get_region_from_depot(rack_t *rack, magazine_t *medium_mag_ptr, mag_index_t mag_index, msize_t msize)
{
magazine_t *depot_ptr = &(rack->magazines[DEPOT_MAGAZINE_INDEX]);
if (rack->num_magazines == 1) { return 0;
}
#if DEBUG_MALLOC
if (DEPOT_MAGAZINE_INDEX == mag_index) {
malloc_zone_error(rack->debug_flags, true, "medium_get_region_from_depot called for magazine index -1\n", NULL, NULL);
return 0;
}
#endif
SZONE_MAGAZINE_PTR_LOCK(depot_ptr);
region_trailer_t *node;
region_t sparse_region;
while (1) {
sparse_region = medium_find_msize_region(rack, depot_ptr, DEPOT_MAGAZINE_INDEX, msize);
if (NULL == sparse_region) { SZONE_MAGAZINE_PTR_UNLOCK(depot_ptr);
return 0;
}
node = REGION_TRAILER_FOR_MEDIUM_REGION(sparse_region);
if (0 >= node->pinned_to_depot) {
break;
}
SZONE_MAGAZINE_PTR_UNLOCK(depot_ptr);
yield();
SZONE_MAGAZINE_PTR_LOCK(depot_ptr);
}
recirc_list_extract(rack, depot_ptr, node);
int objects_in_use = medium_free_detach_region(rack, depot_ptr, sparse_region);
MAGAZINE_INDEX_FOR_MEDIUM_REGION(sparse_region) = mag_index;
node->pinned_to_depot = 0;
size_t bytes_inplay = medium_free_reattach_region(rack, medium_mag_ptr, sparse_region);
depot_ptr->mag_num_bytes_in_objects -= bytes_inplay;
depot_ptr->num_bytes_in_magazine -= MEDIUM_REGION_PAYLOAD_BYTES;
depot_ptr->mag_num_objects -= objects_in_use;
medium_mag_ptr->mag_num_bytes_in_objects += bytes_inplay;
medium_mag_ptr->num_bytes_in_magazine += MEDIUM_REGION_PAYLOAD_BYTES;
medium_mag_ptr->mag_num_objects += objects_in_use;
recirc_list_splice_first(rack, medium_mag_ptr, node);
SZONE_MAGAZINE_PTR_UNLOCK(depot_ptr);
MAGMALLOC_DEPOTREGION(MEDIUM_SZONE_FROM_RACK(rack), (int)mag_index, (void *)sparse_region, MEDIUM_REGION_SIZE,
(int)BYTES_USED_FOR_MEDIUM_REGION(sparse_region));
return 1;
}
#if CONFIG_MADVISE_PRESSURE_RELIEF
void
medium_madvise_pressure_relief(rack_t *rack)
{
mag_index_t mag_index;
magazine_t *medium_depot_ptr = &rack->magazines[DEPOT_MAGAZINE_INDEX];
for (mag_index = 0; mag_index < rack->num_magazines; mag_index++) {
size_t index;
for (index = 0; index < rack->region_generation->num_regions_allocated; ++index) {
SZONE_LOCK(MEDIUM_SZONE_FROM_RACK(rack));
region_t medium = rack->region_generation->hashed_regions[index];
if (!medium || medium == HASHRING_REGION_DEALLOCATED) {
SZONE_UNLOCK(MEDIUM_SZONE_FROM_RACK(rack));
continue;
}
magazine_t *mag_ptr = mag_lock_zine_for_region_trailer(rack->magazines,
REGION_TRAILER_FOR_MEDIUM_REGION(medium),
MAGAZINE_INDEX_FOR_MEDIUM_REGION(medium));
SZONE_UNLOCK(MEDIUM_SZONE_FROM_RACK(rack));
mag_index_t src_mag_index = MAGAZINE_INDEX_FOR_MEDIUM_REGION(medium);
if (src_mag_index == DEPOT_MAGAZINE_INDEX) {
SZONE_MAGAZINE_PTR_UNLOCK(mag_ptr);
continue;
}
if (REGION_TRAILER_FOR_MEDIUM_REGION(medium)->pinned_to_depot > 0) {
SZONE_MAGAZINE_PTR_UNLOCK(mag_ptr);
continue;
}
if (medium == mag_ptr->mag_last_region && (mag_ptr->mag_bytes_free_at_end || mag_ptr->mag_bytes_free_at_start)) {
medium_finalize_region(rack, mag_ptr);
}
recirc_list_extract(rack, mag_ptr, REGION_TRAILER_FOR_MEDIUM_REGION(medium));
int objects_in_use = medium_free_detach_region(rack, mag_ptr, medium);
SZONE_MAGAZINE_PTR_LOCK(medium_depot_ptr);
MAGAZINE_INDEX_FOR_MEDIUM_REGION(medium) = DEPOT_MAGAZINE_INDEX;
REGION_TRAILER_FOR_MEDIUM_REGION(medium)->pinned_to_depot = 0;
size_t bytes_inplay = medium_free_reattach_region(rack, medium_depot_ptr, medium);
mag_ptr->mag_num_bytes_in_objects -= bytes_inplay;
mag_ptr->num_bytes_in_magazine -= MEDIUM_REGION_PAYLOAD_BYTES;
mag_ptr->mag_num_objects -= objects_in_use;
SZONE_MAGAZINE_PTR_UNLOCK(mag_ptr);
medium_depot_ptr->mag_num_bytes_in_objects += bytes_inplay;
medium_depot_ptr->num_bytes_in_magazine += MEDIUM_REGION_PAYLOAD_BYTES;
medium_depot_ptr->mag_num_objects -= objects_in_use;
recirc_list_splice_last(rack, medium_depot_ptr, REGION_TRAILER_FOR_MEDIUM_REGION(medium));
medium_free_scan_madvise_free(rack, medium_depot_ptr, medium);
SZONE_MAGAZINE_PTR_UNLOCK(medium_depot_ptr);
}
}
}
#endif // CONFIG_MADVISE_PRESSURE_RELIEF
static MALLOC_INLINE void
medium_madvise_free_range_conditional_no_lock(rack_t *rack, magazine_t *mag_ptr,
size_t trigger_level, region_t region, free_list_t *fl, msize_t flmsz,
void *ptr, size_t sz)
{
region_trailer_t *node = REGION_TRAILER_FOR_MEDIUM_REGION(region);
msize_t *madvh = MEDIUM_MADVISE_HEADER_FOR_PTR(ptr);
msize_t trigger_msize = trigger_level >> SHIFT_MEDIUM_QUANTUM;
size_t free_header_size = sizeof(medium_inplace_free_entry_s) + sizeof(msize_t);
size_t free_trailer_size = sizeof(msize_t);
if (medium_is_oob_free_entry(*fl)) {
free_header_size = 0;
free_trailer_size = 0;
}
uintptr_t rangep = (uintptr_t)medium_free_list_get_ptr(rack, *fl);
msize_t range_idx = MEDIUM_META_INDEX_FOR_PTR(rangep);
msize_t range_msz = flmsz;
msize_t src_idx = MEDIUM_META_INDEX_FOR_PTR(ptr);
msize_t src_msz = MEDIUM_MSIZE_FOR_BYTES(sz);
msize_t src_end_idx = src_idx + src_msz - 1;
medium_madvise_header_mark_middle(madvh, src_idx);
medium_madvise_header_mark_middle(madvh, src_idx + src_msz - 1);
msize_t left_end_idx = src_idx - 1;
msize_t left_msz = src_idx - range_idx;
msize_t right_start_idx = src_idx + src_msz;
msize_t right_end_idx = range_idx + range_msz - 1;
msize_t right_msz = right_end_idx - right_start_idx + 1;
msize_t dirty_msz = src_msz;
size_t vote_force = 0;
msize_t left_dirty_msz = 0;
if (range_idx < src_idx) {
left_dirty_msz = medium_madvise_header_dirty_len(madvh, left_end_idx);
if (left_dirty_msz) {
dirty_msz += left_dirty_msz;
} else if (src_idx - range_idx > src_msz) {
vote_force++;
}
medium_madvise_header_mark_middle(madvh, range_idx);
medium_madvise_header_mark_middle(madvh, left_end_idx);
}
msize_t right_dirty_msz = 0;
if (right_end_idx < src_end_idx) {
right_dirty_msz = medium_madvise_header_dirty_len(madvh, right_start_idx);
if (right_dirty_msz) {
dirty_msz += right_dirty_msz;
} else if (right_end_idx - right_start_idx > src_msz) {
vote_force++;
}
medium_madvise_header_mark_middle(madvh, right_start_idx);
medium_madvise_header_mark_middle(madvh, right_end_idx);
}
uintptr_t safe_start_ptr = round_page_kernel(rangep + free_header_size);
uintptr_t safe_end_ptr = trunc_page_kernel(rangep +
MEDIUM_BYTES_FOR_MSIZE(range_msz) - free_trailer_size);
if ((vote_force == 2) || (dirty_msz >= trigger_msize)) {
uintptr_t lo = MAX(MEDIUM_PTR_FOR_META_INDEX(region, range_idx),
safe_start_ptr);
uintptr_t hi = MIN(MEDIUM_PTR_FOR_META_INDEX(region, range_idx) +
MEDIUM_BYTES_FOR_MSIZE(range_msz), safe_end_ptr);
if (range_idx < MEDIUM_META_INDEX_FOR_PTR(safe_start_ptr)) {
medium_madvise_header_mark_dirty(madvh, range_idx,
MEDIUM_META_INDEX_FOR_PTR(safe_start_ptr) - range_idx);
}
if (range_idx + range_msz > MEDIUM_META_INDEX_FOR_PTR(safe_end_ptr)) {
medium_madvise_header_mark_dirty(madvh,
MEDIUM_META_INDEX_FOR_PTR(safe_end_ptr) + 1, range_idx +
range_msz - MEDIUM_META_INDEX_FOR_PTR(safe_end_ptr));
}
medium_madvise_header_mark_clean(madvh,
MEDIUM_META_INDEX_FOR_PTR(lo),
MEDIUM_META_INDEX_FOR_PTR(hi) - MEDIUM_META_INDEX_FOR_PTR(lo));
medium_free_mark_unfree(rack, *fl, flmsz);
medium_free_list_remove_ptr_no_clear(rack, mag_ptr, *fl, flmsz);
OSAtomicIncrement32Barrier(&node->pinned_to_depot);
SZONE_MAGAZINE_PTR_UNLOCK(mag_ptr);
mvm_madvise_free(rack, region, lo, hi,
&rack->last_madvise, rack->debug_flags & MALLOC_DO_SCRIBBLE);
SZONE_MAGAZINE_PTR_LOCK(mag_ptr);
OSAtomicDecrement32Barrier(&node->pinned_to_depot);
*fl = medium_free_list_add_ptr(rack, mag_ptr, (void *)rangep, flmsz);
} else {
if (left_dirty_msz < left_msz) {
medium_madvise_header_mark_clean(madvh, range_idx,
left_msz - left_dirty_msz);
}
if (right_dirty_msz < right_msz) {
medium_madvise_header_mark_clean(madvh, right_start_idx +
right_dirty_msz, right_msz - right_dirty_msz);
}
medium_madvise_header_mark_dirty(madvh, src_idx - left_dirty_msz,
src_msz + right_dirty_msz + left_dirty_msz);
}
}
#if CONFIG_AGGRESSIVE_MADVISE || CONFIG_RECIRC_DEPOT
static MALLOC_INLINE void
medium_madvise_free_range_no_lock(rack_t *rack,
magazine_t *medium_mag_ptr,
size_t trigger_level,
region_t region,
free_list_t freee,
msize_t fmsize,
void *headptr,
size_t headsize)
{
void *ptr = medium_free_list_get_ptr(rack, freee);
region_trailer_t *node = REGION_TRAILER_FOR_MEDIUM_REGION(region);
size_t free_header_size = sizeof(medium_inplace_free_entry_s) + sizeof(msize_t);
if (medium_is_oob_free_entry(freee)) {
free_header_size = 0;
}
uintptr_t safe_ptr = (uintptr_t)ptr + free_header_size;
uintptr_t round_safe = round_page_kernel(safe_ptr);
uintptr_t safe_extent = (uintptr_t)ptr + MEDIUM_BYTES_FOR_MSIZE(fmsize);
uintptr_t trunc_extent = trunc_page_kernel(safe_extent);
if (round_safe + trigger_level < trunc_extent) {
uintptr_t lo = trunc_page_kernel((uintptr_t)headptr);
uintptr_t hi = round_page_kernel((uintptr_t)headptr + headsize + free_header_size);
uintptr_t free_lo = MAX(round_safe, lo);
uintptr_t free_hi = MIN(trunc_extent, hi);
if (free_lo < free_hi) {
medium_free_mark_unfree(rack, freee, fmsize);
medium_free_list_remove_ptr_no_clear(rack, medium_mag_ptr, freee, fmsize);
OSAtomicIncrement32Barrier(&(node->pinned_to_depot));
SZONE_MAGAZINE_PTR_UNLOCK(medium_mag_ptr);
mvm_madvise_free(rack, region, free_lo, free_hi, &rack->last_madvise, rack->debug_flags & MALLOC_DO_SCRIBBLE);
SZONE_MAGAZINE_PTR_LOCK(medium_mag_ptr);
OSAtomicDecrement32Barrier(&(node->pinned_to_depot));
medium_free_list_add_ptr(rack, medium_mag_ptr, ptr, fmsize);
}
}
}
#endif // CONFIG_AGGRESSIVE_MADVISE || CONFIG_RECIRC_DEPOT
#if CONFIG_RECIRC_DEPOT
static region_t
medium_free_try_depot_unmap_no_lock(rack_t *rack, magazine_t *depot_ptr, region_trailer_t *node)
{
if (0 < node->bytes_used || 0 < node->pinned_to_depot || depot_ptr->recirculation_entries < recirc_retained_regions) {
return NULL;
}
recirc_list_extract(rack, depot_ptr, node);
region_t sparse_region = MEDIUM_REGION_FOR_PTR(node);
int objects_in_use = medium_free_detach_region(rack, depot_ptr, sparse_region);
if (0 == objects_in_use) {
rgnhdl_t pSlot = hash_lookup_region_no_lock(rack->region_generation->hashed_regions,
rack->region_generation->num_regions_allocated,
rack->region_generation->num_regions_allocated_shift,
sparse_region);
if (NULL == pSlot) {
malloc_zone_error(rack->debug_flags, true, "medium_free_try_depot_unmap_no_lock hash lookup failed: %p\n", sparse_region);
return NULL;
}
*pSlot = HASHRING_REGION_DEALLOCATED;
depot_ptr->num_bytes_in_magazine -= MEDIUM_REGION_PAYLOAD_BYTES;
#ifdef __LP64___
OSAtomicIncrement64(&rack->num_regions_dealloc);
#else
OSAtomicIncrement32((int32_t *)&rack->num_regions_dealloc);
#endif
MAGMALLOC_DEALLOCREGION(MEDIUM_SZONE_FROM_RACK(rack), (void *)sparse_region, (int)MEDIUM_REGION_SIZE); return sparse_region;
} else {
malloc_zone_error(rack->debug_flags, true, "medium_free_try_depot_unmap_no_lock objects_in_use not zero: %d\n", objects_in_use);
return NULL;
}
}
static boolean_t
medium_free_do_recirc_to_depot(rack_t *rack, magazine_t *medium_mag_ptr, mag_index_t mag_index)
{
region_trailer_t *node = medium_mag_ptr->firstNode;
while (node && (!node->recirc_suitable || node->pinned_to_depot)) {
node = node->next;
}
if (NULL == node) {
#if DEBUG_MALLOC
malloc_report(ASL_LEVEL_ERR, "*** medium_free_do_recirc_to_depot end of list\n");
#endif
return TRUE; }
region_t sparse_region = MEDIUM_REGION_FOR_PTR(node);
if (sparse_region == medium_mag_ptr->mag_last_region &&
(medium_mag_ptr->mag_bytes_free_at_end || medium_mag_ptr->mag_bytes_free_at_start)) {
medium_finalize_region(rack, medium_mag_ptr);
}
recirc_list_extract(rack, medium_mag_ptr, node);
int objects_in_use = medium_free_detach_region(rack, medium_mag_ptr, sparse_region);
magazine_t *depot_ptr = &(rack->magazines[DEPOT_MAGAZINE_INDEX]);
SZONE_MAGAZINE_PTR_LOCK(depot_ptr);
MAGAZINE_INDEX_FOR_MEDIUM_REGION(sparse_region) = DEPOT_MAGAZINE_INDEX;
node->pinned_to_depot = 0;
size_t bytes_inplay = medium_free_reattach_region(rack, depot_ptr, sparse_region);
medium_mag_ptr->mag_num_bytes_in_objects -= bytes_inplay;
medium_mag_ptr->num_bytes_in_magazine -= MEDIUM_REGION_PAYLOAD_BYTES;
medium_mag_ptr->mag_num_objects -= objects_in_use;
SZONE_MAGAZINE_PTR_UNLOCK(medium_mag_ptr);
depot_ptr->mag_num_bytes_in_objects += bytes_inplay;
depot_ptr->num_bytes_in_magazine += MEDIUM_REGION_PAYLOAD_BYTES;
depot_ptr->mag_num_objects += objects_in_use;
recirc_list_splice_last(rack, depot_ptr, node);
MAGMALLOC_RECIRCREGION(MEDIUM_SZONE_FROM_RACK(rack), (int)mag_index, (void *)sparse_region, MEDIUM_REGION_SIZE,
(int)BYTES_USED_FOR_MEDIUM_REGION(sparse_region));
#if !CONFIG_AGGRESSIVE_MADVISE
medium_free_scan_madvise_free(rack, depot_ptr, sparse_region);
#endif
region_t r_dealloc = medium_free_try_depot_unmap_no_lock(rack, depot_ptr, node);
SZONE_MAGAZINE_PTR_UNLOCK(depot_ptr);
if (r_dealloc) {
mvm_deallocate_pages(r_dealloc, MEDIUM_REGION_SIZE, 0);
}
return FALSE; }
static MALLOC_INLINE boolean_t
medium_free_try_recirc_to_depot(rack_t *rack,
magazine_t *medium_mag_ptr,
mag_index_t mag_index,
region_t region,
free_list_t freee,
msize_t msize,
void *headptr,
size_t headsize)
{
region_trailer_t *node = REGION_TRAILER_FOR_MEDIUM_REGION(region);
size_t bytes_used = node->bytes_used;
if (rack->num_magazines == 1) {
return TRUE; } else if (DEPOT_MAGAZINE_INDEX != mag_index) {
if (bytes_used < DENSITY_THRESHOLD(MEDIUM_REGION_PAYLOAD_BYTES)) {
node->recirc_suitable = TRUE;
} else {
}
size_t a = medium_mag_ptr->num_bytes_in_magazine; size_t u = medium_mag_ptr->mag_num_bytes_in_objects;
if (a - u > ((3 * MEDIUM_REGION_PAYLOAD_BYTES) / 2) && u < DENSITY_THRESHOLD(a)) {
return medium_free_do_recirc_to_depot(rack, medium_mag_ptr, mag_index);
}
} else {
#if !CONFIG_AGGRESSIVE_MADVISE
medium_madvise_free_range_no_lock(rack, medium_mag_ptr,
vm_kernel_page_size, region, freee, msize, headptr, headsize);
#endif
if (0 < bytes_used || 0 < node->pinned_to_depot) {
} else {
region_t r_dealloc = medium_free_try_depot_unmap_no_lock(rack, medium_mag_ptr, node);
SZONE_MAGAZINE_PTR_UNLOCK(medium_mag_ptr);
if (r_dealloc) {
mvm_deallocate_pages(r_dealloc, MEDIUM_REGION_SIZE, 0);
}
return FALSE; }
}
return TRUE; }
#endif // CONFIG_RECIRC_DEPOT
static MALLOC_INLINE boolean_t
medium_free_no_lock(rack_t *rack, magazine_t *medium_mag_ptr, mag_index_t mag_index, region_t region, void *ptr, msize_t msize)
{
msize_t *meta_headers = MEDIUM_META_HEADER_FOR_PTR(ptr);
unsigned index = MEDIUM_META_INDEX_FOR_PTR(ptr);
size_t original_size = MEDIUM_BYTES_FOR_MSIZE(msize);
unsigned char *next_block = ((unsigned char *)ptr + original_size);
msize_t next_index = index + msize;
MALLOC_TRACE(TRACE_medium_free, (uintptr_t)rack, (uintptr_t)medium_mag_ptr, (uintptr_t)ptr, MEDIUM_BYTES_FOR_MSIZE(msize));
#if CONFIG_AGGRESSIVE_MADVISE || CONFIG_RECIRC_DEPOT
void *original_ptr = ptr;
#endif
#if DEBUG_MALLOC
if (!msize) {
malloc_zone_error(rack->debug_flags, true, "trying to free medium block "
"that is too small in medium_free_no_lock(), ptr=%p, msize=%d\n",
ptr, msize);
}
if (medium_madvise_header_dirty_len(MEDIUM_MADVISE_HEADER_FOR_PTR(ptr), index) == 0) {
malloc_zone_error(rack->debug_flags, true, "incoming freed block is not "
"marked as dirty in madvise metadata, ptr=%p, msize=%d\n",
ptr, msize);
}
#endif
if (index > 0 && (meta_headers[index - 1] & MEDIUM_IS_FREE)) {
msize_t previous_msize = meta_headers[index - 1] & ~MEDIUM_IS_FREE;
grain_t previous_index = index - previous_msize;
if (meta_headers[previous_index] == (previous_msize | MEDIUM_IS_FREE)) {
void *previous_ptr = (void *)((uintptr_t)ptr - MEDIUM_BYTES_FOR_MSIZE(previous_msize));
free_list_t previous = medium_free_list_find_by_ptr(rack, medium_mag_ptr, previous_ptr, previous_msize);
medium_free_list_remove_ptr(rack, medium_mag_ptr, previous, previous_msize);
ptr = previous_ptr;
medium_meta_header_set_middle(meta_headers, index); msize += previous_msize;
index -= previous_msize;
} else {
_os_set_crash_log_message("medium free list metadata inconsistency (headers[previous] != previous size)");
__builtin_trap();
}
}
if ((next_block < MEDIUM_REGION_END(region)) && (meta_headers[next_index] & MEDIUM_IS_FREE)) {
msize_t next_msize = meta_headers[next_index] & ~MEDIUM_IS_FREE;
free_list_t next = medium_free_list_find_by_ptr(rack, medium_mag_ptr, next_block, next_msize);
medium_free_list_remove_ptr(rack, medium_mag_ptr, next, next_msize);
msize += next_msize;
}
if (rack->debug_flags & MALLOC_DO_SCRIBBLE) {
if (!msize) {
malloc_zone_error(rack->debug_flags, true, "incorrect size information for %p - block header was damaged\n", ptr);
} else {
memset(ptr, SCRABBLE_BYTE, MEDIUM_BYTES_FOR_MSIZE(msize));
}
}
free_list_t freee = medium_free_list_add_ptr(rack, medium_mag_ptr, ptr, msize);
medium_mag_ptr->mag_num_bytes_in_objects -= original_size;
medium_mag_ptr->mag_num_objects--;
region_trailer_t *node = REGION_TRAILER_FOR_MEDIUM_REGION(region);
size_t bytes_used = node->bytes_used - original_size;
node->bytes_used = (unsigned int)bytes_used;
medium_madvise_free_range_conditional_no_lock(rack, medium_mag_ptr,
medium_sliding_madvise_granularity(medium_mag_ptr), region, &freee,
msize, original_ptr, original_size);
boolean_t needs_unlock = TRUE;
#if CONFIG_RECIRC_DEPOT
needs_unlock = medium_free_try_recirc_to_depot(rack, medium_mag_ptr, mag_index, region, freee, msize, original_ptr, original_size);
#endif
return needs_unlock;
}
static void *
medium_malloc_from_region_no_lock(rack_t *rack,
magazine_t *medium_mag_ptr,
mag_index_t mag_index,
msize_t msize,
void *aligned_address)
{
void *ptr;
if (medium_mag_ptr->mag_bytes_free_at_end || medium_mag_ptr->mag_bytes_free_at_start) {
medium_finalize_region(rack, medium_mag_ptr);
}
MAGAZINE_INDEX_FOR_MEDIUM_REGION(aligned_address) = mag_index;
rack_region_insert(rack, (region_t)aligned_address);
medium_mag_ptr->mag_last_region = aligned_address;
BYTES_USED_FOR_MEDIUM_REGION(aligned_address) = MEDIUM_BYTES_FOR_MSIZE(msize);
#if CONFIG_ASLR_INTERNAL
int offset_msize = malloc_entropy[1] & MEDIUM_ENTROPY_MASK;
#if DEBUG_MALLOC
if (getenv("MallocASLRForce")) {
offset_msize = strtol(getenv("MallocASLRForce"), NULL, 0) & MEDIUM_ENTROPY_MASK;
}
if (getenv("MallocASLRPrint")) {
malloc_report(ASL_LEVEL_INFO, "Region: %p offset: %d\n", aligned_address, offset_msize);
}
#endif
#else
int offset_msize = 0;
#endif
ptr = (void *)((uintptr_t)aligned_address +
MEDIUM_BYTES_FOR_MSIZE(offset_msize));
medium_meta_header_set_in_use(MEDIUM_META_HEADER_FOR_PTR(ptr),
offset_msize, msize);
medium_madvise_header_mark_dirty(MEDIUM_MADVISE_HEADER_FOR_PTR(ptr),
offset_msize, msize);
medium_mag_ptr->mag_num_objects++;
medium_mag_ptr->mag_num_bytes_in_objects += MEDIUM_BYTES_FOR_MSIZE(msize);
medium_mag_ptr->num_bytes_in_magazine += MEDIUM_REGION_PAYLOAD_BYTES;
msize_t trailing_offset = offset_msize + msize;
msize_t trailing_msize = NUM_MEDIUM_BLOCKS - msize - offset_msize;
medium_meta_header_set_in_use(MEDIUM_META_HEADER_FOR_PTR(ptr),
trailing_offset, trailing_msize);
medium_madvise_header_mark_clean(MEDIUM_MADVISE_HEADER_FOR_PTR(ptr),
trailing_offset, trailing_msize);
medium_mag_ptr->mag_bytes_free_at_end =
MEDIUM_BYTES_FOR_MSIZE(NUM_MEDIUM_BLOCKS - msize - offset_msize);
#if CONFIG_ASLR_INTERNAL
medium_mag_ptr->mag_bytes_free_at_start = MEDIUM_BYTES_FOR_MSIZE(offset_msize);
if (offset_msize) {
medium_meta_header_set_in_use(MEDIUM_META_HEADER_FOR_PTR(ptr), 0,
offset_msize);
medium_madvise_header_mark_clean(MEDIUM_MADVISE_HEADER_FOR_PTR(ptr), 0,
offset_msize);
}
#else
medium_mag_ptr->mag_bytes_free_at_start = 0;
#endif
recirc_list_splice_last(rack, medium_mag_ptr,
REGION_TRAILER_FOR_MEDIUM_REGION(aligned_address));
return ptr;
}
void *
medium_memalign(szone_t *szone, size_t alignment, size_t size, size_t span)
{
msize_t mspan = MEDIUM_MSIZE_FOR_BYTES(span + MEDIUM_QUANTUM - 1);
void *p = medium_malloc_should_clear(&szone->medium_rack, mspan, 0);
if (NULL == p) {
return NULL;
}
size_t offset = ((uintptr_t)p) & (alignment - 1); size_t pad = (0 == offset) ? 0 : alignment - offset;
msize_t msize = MEDIUM_MSIZE_FOR_BYTES(size + MEDIUM_QUANTUM - 1);
msize_t mpad = MEDIUM_MSIZE_FOR_BYTES(pad + MEDIUM_QUANTUM - 1);
msize_t mwaste = mspan - msize - mpad;
if (mpad > 0) {
void *q = (void *)(((uintptr_t)p) + pad);
magazine_t *medium_mag_ptr = mag_lock_zine_for_region_trailer(
szone->medium_rack.magazines,
REGION_TRAILER_FOR_MEDIUM_REGION(MEDIUM_REGION_FOR_PTR(p)),
MAGAZINE_INDEX_FOR_MEDIUM_REGION(MEDIUM_REGION_FOR_PTR(p)));
medium_meta_header_set_in_use(MEDIUM_META_HEADER_FOR_PTR(p), MEDIUM_META_INDEX_FOR_PTR(p), mpad);
medium_madvise_header_mark_dirty(MEDIUM_MADVISE_HEADER_FOR_PTR(p), MEDIUM_META_INDEX_FOR_PTR(p), mpad);
medium_meta_header_set_in_use(MEDIUM_META_HEADER_FOR_PTR(q), MEDIUM_META_INDEX_FOR_PTR(q), msize + mwaste);
medium_madvise_header_mark_dirty(MEDIUM_MADVISE_HEADER_FOR_PTR(q), MEDIUM_META_INDEX_FOR_PTR(q), msize + mwaste);
medium_mag_ptr->mag_num_objects++;
SZONE_MAGAZINE_PTR_UNLOCK(medium_mag_ptr);
free_medium(&szone->medium_rack, p, MEDIUM_REGION_FOR_PTR(p), MEDIUM_BYTES_FOR_MSIZE(mpad));
p = q; }
if (mwaste > 0) {
void *q = (void *)(((uintptr_t)p) + MEDIUM_BYTES_FOR_MSIZE(msize));
magazine_t *medium_mag_ptr = mag_lock_zine_for_region_trailer(szone->medium_rack.magazines,
REGION_TRAILER_FOR_MEDIUM_REGION(MEDIUM_REGION_FOR_PTR(p)),
MAGAZINE_INDEX_FOR_MEDIUM_REGION(MEDIUM_REGION_FOR_PTR(p)));
medium_meta_header_set_in_use(MEDIUM_META_HEADER_FOR_PTR(p), MEDIUM_META_INDEX_FOR_PTR(p), msize);
medium_madvise_header_mark_dirty(MEDIUM_MADVISE_HEADER_FOR_PTR(p), MEDIUM_META_INDEX_FOR_PTR(p), msize);
medium_meta_header_set_in_use(MEDIUM_META_HEADER_FOR_PTR(q), MEDIUM_META_INDEX_FOR_PTR(q), mwaste);
medium_madvise_header_mark_dirty(MEDIUM_MADVISE_HEADER_FOR_PTR(q), MEDIUM_META_INDEX_FOR_PTR(q), mwaste);
medium_mag_ptr->mag_num_objects++;
SZONE_MAGAZINE_PTR_UNLOCK(medium_mag_ptr);
free_medium(&szone->medium_rack, q, MEDIUM_REGION_FOR_PTR(q), MEDIUM_BYTES_FOR_MSIZE(mwaste));
}
return p; }
boolean_t
medium_claimed_address(rack_t *rack, void *ptr)
{
region_t r = medium_region_for_ptr_no_lock(rack, ptr);
return r && ptr < (void *)MEDIUM_REGION_END(r);
}
void *
medium_try_shrink_in_place(rack_t *rack, void *ptr, size_t old_size, size_t new_good_size)
{
msize_t new_msize = MEDIUM_MSIZE_FOR_BYTES(new_good_size);
msize_t mshrinkage = MEDIUM_MSIZE_FOR_BYTES(old_size) - new_msize;
if (mshrinkage) {
void *q = (void *)((uintptr_t)ptr + MEDIUM_BYTES_FOR_MSIZE(new_msize));
magazine_t *medium_mag_ptr = mag_lock_zine_for_region_trailer(rack->magazines,
REGION_TRAILER_FOR_MEDIUM_REGION(MEDIUM_REGION_FOR_PTR(ptr)),
MAGAZINE_INDEX_FOR_MEDIUM_REGION(MEDIUM_REGION_FOR_PTR(ptr)));
medium_meta_header_set_in_use(MEDIUM_META_HEADER_FOR_PTR(ptr), MEDIUM_META_INDEX_FOR_PTR(ptr), new_msize);
medium_madvise_header_mark_dirty(MEDIUM_MADVISE_HEADER_FOR_PTR(ptr), MEDIUM_META_INDEX_FOR_PTR(ptr), new_msize);
medium_meta_header_set_in_use(MEDIUM_META_HEADER_FOR_PTR(q), MEDIUM_META_INDEX_FOR_PTR(q), mshrinkage);
medium_madvise_header_mark_dirty(MEDIUM_MADVISE_HEADER_FOR_PTR(q), MEDIUM_META_INDEX_FOR_PTR(q), mshrinkage);
medium_mag_ptr->mag_num_objects++;
SZONE_MAGAZINE_PTR_UNLOCK(medium_mag_ptr);
free_medium(rack, q, MEDIUM_REGION_FOR_PTR(q), 0);
}
return ptr;
}
boolean_t
medium_try_realloc_in_place(rack_t *rack, void *ptr, size_t old_size, size_t new_size)
{
msize_t *meta_headers = MEDIUM_META_HEADER_FOR_PTR(ptr);
msize_t *madv_headers = MEDIUM_MADVISE_HEADER_FOR_PTR(ptr);
unsigned index;
msize_t old_msize, new_msize;
unsigned next_index;
void *next_block;
msize_t next_msize_and_free;
boolean_t is_free;
msize_t next_msize, leftover_msize;
void *leftover;
index = MEDIUM_META_INDEX_FOR_PTR(ptr);
old_msize = MEDIUM_MSIZE_FOR_BYTES(old_size);
new_msize = MEDIUM_MSIZE_FOR_BYTES(new_size + MEDIUM_QUANTUM - 1);
next_index = index + old_msize;
if (next_index >= NUM_MEDIUM_BLOCKS) {
return 0;
}
next_block = (char *)ptr + old_size;
#if DEBUG_MALLOC
if ((uintptr_t)next_block & (MEDIUM_QUANTUM - 1)) {
malloc_zone_error(rack->debug_flags, true, "internal invariant broken in realloc(next_block) for %p\n", next_block);
}
if (meta_headers[index] != old_msize) {
malloc_report(ASL_LEVEL_ERR, "*** medium_try_realloc_in_place incorrect old %d %d\n", meta_headers[index], old_msize);
}
#endif
magazine_t *medium_mag_ptr = mag_lock_zine_for_region_trailer(rack->magazines,
REGION_TRAILER_FOR_MEDIUM_REGION(MEDIUM_REGION_FOR_PTR(ptr)),
MAGAZINE_INDEX_FOR_MEDIUM_REGION(MEDIUM_REGION_FOR_PTR(ptr)));
if (DEPOT_MAGAZINE_INDEX == MAGAZINE_INDEX_FOR_MEDIUM_REGION(MEDIUM_REGION_FOR_PTR(ptr))) {
SZONE_MAGAZINE_PTR_UNLOCK(medium_mag_ptr);
return 0;
}
msize_t coalesced_msize = new_msize - old_msize;
#if CONFIG_MEDIUM_CACHE
void *last_free_ptr = medium_mag_ptr->mag_last_free;
msize_t last_free_msize = medium_mag_ptr->mag_last_free_msize;
if (last_free_ptr == next_block && old_msize + last_free_msize >= new_msize) {
leftover_msize = last_free_msize - coalesced_msize;
if (leftover_msize) {
medium_mag_ptr->mag_last_free_msize -= coalesced_msize;
medium_mag_ptr->mag_last_free += new_size - old_size;
medium_meta_header_set_in_use(meta_headers, index + new_msize, leftover_msize);
if (madv_headers[index] & MEDIUM_IS_ADVISED) {
medium_madvise_header_mark_clean(madv_headers, index + new_msize, leftover_msize);
} else {
medium_madvise_header_mark_dirty(madv_headers, index + new_msize, leftover_msize);
}
} else {
medium_mag_ptr->mag_last_free = NULL;
medium_mag_ptr->mag_last_free_msize = 0;
medium_mag_ptr->mag_last_free_rgn = NULL;
}
medium_meta_header_set_in_use(meta_headers, index, new_msize);
medium_madvise_header_mark_dirty(madv_headers, index, new_msize);
medium_meta_header_set_middle(meta_headers, next_index);
medium_madvise_header_mark_middle(madv_headers, next_index);
} else {
#endif // CONFIG_MEDIUM_CACHE
msize_t unused_msize = MEDIUM_MSIZE_FOR_BYTES(medium_mag_ptr->mag_bytes_free_at_end);
void *unused_start = MEDIUM_REGION_END(MEDIUM_REGION_FOR_PTR(ptr)) - medium_mag_ptr->mag_bytes_free_at_end;
if (medium_mag_ptr->mag_last_region == MEDIUM_REGION_FOR_PTR(ptr)
&& coalesced_msize < unused_msize && unused_start == ptr + old_size) {
medium_meta_header_set_in_use(meta_headers, index, new_msize);
medium_madvise_header_mark_dirty(madv_headers, index, new_msize);
medium_meta_header_set_middle(meta_headers, next_index);
medium_madvise_header_mark_middle(madv_headers, next_index);
medium_mag_ptr->mag_bytes_free_at_end -= MEDIUM_BYTES_FOR_MSIZE(coalesced_msize);
medium_meta_header_set_in_use(meta_headers, index + new_msize,
MEDIUM_MSIZE_FOR_BYTES(medium_mag_ptr->mag_bytes_free_at_end));
medium_madvise_header_mark_clean(madv_headers, index + new_msize,
MEDIUM_MSIZE_FOR_BYTES(medium_mag_ptr->mag_bytes_free_at_end));
} else {
next_msize_and_free = meta_headers[next_index];
is_free = next_msize_and_free & MEDIUM_IS_FREE;
if (!is_free) {
SZONE_MAGAZINE_PTR_UNLOCK(medium_mag_ptr);
return 0; }
next_msize = next_msize_and_free & ~MEDIUM_IS_FREE;
if (old_msize + next_msize < new_msize) {
SZONE_MAGAZINE_PTR_UNLOCK(medium_mag_ptr);
return 0; }
free_list_t freee = medium_free_list_find_by_ptr(rack, medium_mag_ptr, next_block, next_msize);
medium_free_list_remove_ptr(rack, medium_mag_ptr, freee, next_msize);
medium_meta_header_set_middle(meta_headers, next_index);
medium_madvise_header_mark_middle(madv_headers, next_index);
leftover_msize = old_msize + next_msize - new_msize;
if (leftover_msize) {
leftover = (unsigned char *)ptr + MEDIUM_BYTES_FOR_MSIZE(new_msize);
medium_free_list_add_ptr(rack, medium_mag_ptr, leftover, leftover_msize);
}
medium_meta_header_set_in_use(meta_headers, index, new_msize);
medium_madvise_header_mark_dirty(madv_headers, index, new_msize);
}
#if CONFIG_MEDIUM_CACHE
}
#endif // CONFIG_MEDIUM_CACHE
#if DEBUG_MALLOC
if (MEDIUM_BYTES_FOR_MSIZE(new_msize) > szone->large_threshold) {
malloc_report(ASL_LEVEL_ERR, "*** realloc in place for %p exceeded msize=%d\n", new_msize);
}
if (LOG(szone, ptr)) {
malloc_report(ASL_LEVEL_INFO, "in medium_try_realloc_in_place(), ptr=%p, msize=%d\n", ptr, *MEDIUM_METADATA_FOR_PTR(ptr));
}
#endif
medium_mag_ptr->mag_num_bytes_in_objects += MEDIUM_BYTES_FOR_MSIZE(new_msize - old_msize);
region_trailer_t *node = REGION_TRAILER_FOR_MEDIUM_REGION(MEDIUM_REGION_FOR_PTR(ptr));
size_t bytes_used = node->bytes_used + MEDIUM_BYTES_FOR_MSIZE(new_msize - old_msize);
node->bytes_used = (unsigned int)bytes_used;
if (bytes_used < DENSITY_THRESHOLD(MEDIUM_REGION_PAYLOAD_BYTES)) {
} else {
node->recirc_suitable = FALSE;
}
SZONE_MAGAZINE_PTR_UNLOCK(medium_mag_ptr);
CHECK(szone, __PRETTY_FUNCTION__);
return 1;
}
static char *medium_check_fail_msg = "check: incorrect medium region ";
#define MEDIUM_CHECK_FAIL(fmt, ...) \
malloc_zone_check_fail(medium_check_fail_msg, \
"%ld, counter=%d\n" fmt, region_index, counter, __VA_ARGS__);
boolean_t
medium_check_region(rack_t *rack, region_t region, size_t region_index,
unsigned counter)
{
unsigned char *ptr = MEDIUM_REGION_ADDRESS(region);
msize_t *meta_headers = MEDIUM_META_HEADER_FOR_PTR(ptr);
unsigned char *region_end = MEDIUM_REGION_END(region);
msize_t prev_free = 0;
unsigned index;
msize_t msize_and_free;
msize_t msize;
free_list_t free_head, previous, next;
msize_t *follower;
mag_index_t mag_index = MAGAZINE_INDEX_FOR_MEDIUM_REGION(MEDIUM_REGION_FOR_PTR(ptr));
magazine_t *medium_mag_ptr = &(rack->magazines[mag_index]);
CHECK_MAGAZINE_PTR_LOCKED(szone, medium_mag_ptr, __PRETTY_FUNCTION__);
if (region == medium_mag_ptr->mag_last_region) {
ptr += medium_mag_ptr->mag_bytes_free_at_start;
region_end -= medium_mag_ptr->mag_bytes_free_at_end;
}
while (ptr < region_end) {
index = MEDIUM_META_INDEX_FOR_PTR(ptr);
msize_and_free = meta_headers[index];
if (!(msize_and_free & MEDIUM_IS_FREE)) {
msize = msize_and_free;
if (!msize) {
MEDIUM_CHECK_FAIL("*** invariant broken: null msize ptr=%p num_medium_regions=%d end=%p\n", ptr,
(int)rack->num_regions, region_end);
return 0;
}
#if !CONFIG_RELAXED_INVARIANT_CHECKS
if (MEDIUM_BYTES_FOR_MSIZE(msize) > szone->large_threshold) {
MEDIUM_CHECK_FAIL("*** invariant broken for %p this medium msize=%d - size is too large\n", ptr, msize_and_free);
return 0;
}
#endif // CONFIG_RELAXED_INVARIANT_CHECKS
ptr += MEDIUM_BYTES_FOR_MSIZE(msize);
prev_free = 0;
} else {
msize = msize_and_free & ~MEDIUM_IS_FREE;
free_head = (free_list_t){ .p = ptr };
follower = (msize_t *)FOLLOWING_MEDIUM_PTR(ptr, msize);
if (!msize) {
MEDIUM_CHECK_FAIL("*** invariant broken for free block %p this msize=%d\n", ptr, msize);
return 0;
}
#if !CONFIG_RELAXED_INVARIANT_CHECKS
if (prev_free) {
MEDIUM_CHECK_FAIL("*** invariant broken for %p (2 free in a row)\n", ptr);
return 0;
}
#endif
if (medium_needs_oob_free_entry(ptr, msize)) {
oob_free_entry_t oob = medium_oob_free_find_ptr(ptr, msize);
if (oob) {
free_head.oob = oob;
}
}
previous = medium_free_list_get_previous(rack, free_head);
next = medium_free_list_get_next(rack, free_head);
if (previous.p && !MEDIUM_PTR_IS_FREE(medium_free_list_get_ptr(rack, previous))) {
MEDIUM_CHECK_FAIL("*** invariant broken for %p (previous %p is not a free pointer)\n", ptr, medium_free_list_get_ptr(rack, previous));
return 0;
}
if (next.p && !MEDIUM_PTR_IS_FREE(medium_free_list_get_ptr(rack, next))) {
MEDIUM_CHECK_FAIL("*** invariant broken for %p (next %p is not a free pointer)\n", ptr, medium_free_list_get_ptr(rack, next));
return 0;
}
if (MEDIUM_PREVIOUS_MSIZE(follower) != msize) {
MEDIUM_CHECK_FAIL("*** invariant broken for medium free %p followed by %p in region [%p-%p] "
"(end marker incorrect) should be %d; in fact %d\n",
ptr, follower, MEDIUM_REGION_ADDRESS(region), region_end, msize, MEDIUM_PREVIOUS_MSIZE(follower));
return 0;
}
ptr = (unsigned char *)follower;
prev_free = MEDIUM_IS_FREE;
}
}
return 1;
}
kern_return_t
medium_in_use_enumerator(task_t task,
void *context,
unsigned type_mask,
szone_t *szone,
memory_reader_t reader,
vm_range_recorder_t recorder)
{
size_t num_regions;
size_t index;
region_t *regions;
vm_range_t buffer[MAX_RECORDER_BUFFER];
unsigned count = 0;
kern_return_t err;
region_t region;
vm_range_t range;
vm_range_t admin_range;
vm_range_t ptr_range;
unsigned char *mapped_region;
msize_t *block_header;
unsigned block_index;
unsigned block_limit;
msize_t msize_and_free;
msize_t msize;
magazine_t *medium_mag_base = NULL;
region_hash_generation_t *srg_ptr;
err = reader(task, (vm_address_t)szone->medium_rack.region_generation, sizeof(region_hash_generation_t), (void **)&srg_ptr);
if (err) {
return err;
}
num_regions = srg_ptr->num_regions_allocated;
err = reader(task, (vm_address_t)srg_ptr->hashed_regions, sizeof(region_t) * num_regions, (void **)®ions);
if (err) {
return err;
}
if (type_mask & MALLOC_PTR_IN_USE_RANGE_TYPE) {
err = reader(task, (vm_address_t)(szone->medium_rack.magazines), szone->medium_rack.num_magazines * sizeof(magazine_t),
(void **)&medium_mag_base);
if (err) {
return err;
}
}
for (index = 0; index < num_regions; ++index) {
region = regions[index];
if (HASHRING_OPEN_ENTRY != region && HASHRING_REGION_DEALLOCATED != region) {
range.address = (vm_address_t)MEDIUM_REGION_ADDRESS(region);
range.size = MEDIUM_REGION_SIZE;
if (type_mask & MALLOC_ADMIN_REGION_RANGE_TYPE) {
admin_range.address = range.address + MEDIUM_METADATA_START;
admin_range.size = MEDIUM_METADATA_SIZE;
recorder(task, context, MALLOC_ADMIN_REGION_RANGE_TYPE, &admin_range, 1);
}
if (type_mask & (MALLOC_PTR_REGION_RANGE_TYPE | MALLOC_ADMIN_REGION_RANGE_TYPE)) {
ptr_range.address = range.address;
ptr_range.size = NUM_MEDIUM_BLOCKS * MEDIUM_QUANTUM;
recorder(task, context, MALLOC_PTR_REGION_RANGE_TYPE, &ptr_range, 1);
}
if (type_mask & MALLOC_PTR_IN_USE_RANGE_TYPE) {
vm_address_t mag_last_free = 0;
msize_t mag_last_free_msize = 0;
err = reader(task, range.address, range.size, (void **)&mapped_region);
if (err) {
return err;
}
mag_index_t mag_index = MAGAZINE_INDEX_FOR_MEDIUM_REGION(mapped_region);
magazine_t *medium_mag_ptr = medium_mag_base + mag_index;
if (DEPOT_MAGAZINE_INDEX != mag_index) {
mag_last_free = (uintptr_t)medium_mag_ptr->mag_last_free;
mag_last_free_msize = medium_mag_ptr->mag_last_free_msize;
} else {
for (mag_index = 0; mag_index < szone->medium_rack.num_magazines; mag_index++) {
if ((void *)range.address == (medium_mag_base + mag_index)->mag_last_free_rgn) {
mag_last_free = (uintptr_t)(medium_mag_base + mag_index)->mag_last_free;
mag_last_free_msize = (medium_mag_base + mag_index)->mag_last_free_msize;
}
}
}
block_header = (msize_t *)(mapped_region + MEDIUM_METADATA_START + sizeof(region_trailer_t));
block_index = 0;
block_limit = NUM_MEDIUM_BLOCKS;
if (region == medium_mag_ptr->mag_last_region) {
block_index += MEDIUM_MSIZE_FOR_BYTES(medium_mag_ptr->mag_bytes_free_at_start);
block_limit -= MEDIUM_MSIZE_FOR_BYTES(medium_mag_ptr->mag_bytes_free_at_end);
}
while (block_index < block_limit) {
msize_and_free = block_header[block_index];
msize = msize_and_free & ~MEDIUM_IS_FREE;
if (!(msize_and_free & MEDIUM_IS_FREE) &&
range.address + MEDIUM_BYTES_FOR_MSIZE(block_index) != mag_last_free) {
buffer[count].address = range.address + MEDIUM_BYTES_FOR_MSIZE(block_index);
buffer[count].size = MEDIUM_BYTES_FOR_MSIZE(msize);
count++;
if (count >= MAX_RECORDER_BUFFER) {
recorder(task, context, MALLOC_PTR_IN_USE_RANGE_TYPE, buffer, count);
count = 0;
}
}
if (!msize) {
return KERN_FAILURE; }
block_index += msize;
}
if (count) {
recorder(task, context, MALLOC_PTR_IN_USE_RANGE_TYPE, buffer, count);
count = 0;
}
}
}
}
return 0;
}
static void *
medium_malloc_from_free_list(rack_t *rack, magazine_t *medium_mag_ptr, mag_index_t mag_index, msize_t msize)
{
msize_t this_msize;
bool was_madvised;
grain_t slot = MEDIUM_FREE_SLOT_FOR_MSIZE(rack, msize);
free_list_t *free_list = medium_mag_ptr->mag_free_list;
free_list_t *the_slot = free_list + slot;
free_list_t *limit;
unsigned bitmap;
msize_t leftover_msize;
void *leftover_ptr;
void *ptr;
CHECK_MAGAZINE_PTR_LOCKED(szone, medium_mag_ptr, __PRETTY_FUNCTION__);
if (medium_free_list_get_ptr(rack, *the_slot)) {
ptr = medium_free_list_get_ptr(rack, *the_slot);
this_msize = msize;
medium_free_list_remove_ptr(rack, medium_mag_ptr, *the_slot, msize);
goto return_medium_alloc;
}
unsigned idx = slot >> 5;
bitmap = 0;
unsigned mask = ~((1 << (slot & 31)) - 1);
for (; idx < MEDIUM_FREELIST_BITMAP_WORDS(rack); ++idx) {
bitmap = medium_mag_ptr->mag_bitmap[idx] & mask;
if (bitmap != 0) {
break;
}
mask = ~0U;
}
if ((bitmap == 0) && (idx == MEDIUM_FREELIST_BITMAP_WORDS(rack))) {
goto try_medium_from_end;
}
slot = BITMAP32_CTZ((&bitmap)) + (idx * 32);
limit = free_list + MEDIUM_FREE_SLOT_COUNT(rack) - 1;
free_list += slot;
if ((ptr = medium_free_list_get_ptr(rack, *free_list))) {
this_msize = MEDIUM_PTR_SIZE(ptr);
was_madvised = (medium_madvise_header_dirty_len(
MEDIUM_MADVISE_HEADER_FOR_PTR(ptr), this_msize) == 0);
medium_free_list_remove_ptr(rack, medium_mag_ptr, *free_list, this_msize);
goto add_leftover_and_proceed;
}
#if DEBUG_MALLOC
malloc_report(ASL_LEVEL_ERR, "in medium_malloc_from_free_list(), mag_bitmap out of sync, slot=%d\n", slot);
#endif
try_medium_from_end:
if (medium_mag_ptr->mag_bytes_free_at_end >= MEDIUM_BYTES_FOR_MSIZE(msize)) {
ptr = MEDIUM_REGION_END(medium_mag_ptr->mag_last_region) -
medium_mag_ptr->mag_bytes_free_at_end;
medium_mag_ptr->mag_bytes_free_at_end -= MEDIUM_BYTES_FOR_MSIZE(msize);
if (medium_mag_ptr->mag_bytes_free_at_end) {
medium_meta_header_set_in_use(MEDIUM_META_HEADER_FOR_PTR(ptr),
MEDIUM_META_INDEX_FOR_PTR((unsigned char *)ptr + MEDIUM_BYTES_FOR_MSIZE(msize)),
MEDIUM_MSIZE_FOR_BYTES(medium_mag_ptr->mag_bytes_free_at_end));
medium_madvise_header_mark_clean(MEDIUM_MADVISE_HEADER_FOR_PTR(ptr),
MEDIUM_META_INDEX_FOR_PTR((unsigned char *)ptr + MEDIUM_BYTES_FOR_MSIZE(msize)),
MEDIUM_MSIZE_FOR_BYTES(medium_mag_ptr->mag_bytes_free_at_end));
}
this_msize = msize;
goto return_medium_alloc;
}
#if CONFIG_ASLR_INTERNAL
if (medium_mag_ptr->mag_bytes_free_at_start >= MEDIUM_BYTES_FOR_MSIZE(msize)) {
ptr = MEDIUM_REGION_ADDRESS(medium_mag_ptr->mag_last_region) +
medium_mag_ptr->mag_bytes_free_at_start -
MEDIUM_BYTES_FOR_MSIZE(msize);
medium_mag_ptr->mag_bytes_free_at_start -= MEDIUM_BYTES_FOR_MSIZE(msize);
if (medium_mag_ptr->mag_bytes_free_at_start) {
medium_meta_header_set_in_use(MEDIUM_META_HEADER_FOR_PTR(ptr), 0,
MEDIUM_MSIZE_FOR_BYTES(medium_mag_ptr->mag_bytes_free_at_start));
medium_madvise_header_mark_clean(MEDIUM_MADVISE_HEADER_FOR_PTR(ptr), 0,
MEDIUM_MSIZE_FOR_BYTES(medium_mag_ptr->mag_bytes_free_at_start));
}
this_msize = msize;
goto return_medium_alloc;
}
#endif
return NULL;
add_leftover_and_proceed:
if (this_msize > msize) {
leftover_msize = this_msize - msize;
leftover_ptr = (unsigned char *)ptr + MEDIUM_BYTES_FOR_MSIZE(msize);
#if DEBUG_MALLOC
if (LOG(szone, ptr)) {
malloc_report(ASL_LEVEL_INFO, "in medium_malloc_from_free_list(), adding leftover ptr=%p, this_msize=%d\n", ptr, this_msize);
}
#endif
medium_free_list_add_ptr(rack, medium_mag_ptr, leftover_ptr, leftover_msize);
if (was_madvised) {
medium_madvise_header_mark_clean(
MEDIUM_MADVISE_HEADER_FOR_PTR(leftover_ptr),
MEDIUM_META_INDEX_FOR_PTR(leftover_ptr), leftover_msize);
} else {
medium_madvise_header_mark_dirty(
MEDIUM_MADVISE_HEADER_FOR_PTR(leftover_ptr),
MEDIUM_META_INDEX_FOR_PTR(leftover_ptr), leftover_msize);
}
this_msize = msize;
}
return_medium_alloc:
medium_mag_ptr->mag_num_objects++;
medium_mag_ptr->mag_num_bytes_in_objects += MEDIUM_BYTES_FOR_MSIZE(this_msize);
region_trailer_t *node = REGION_TRAILER_FOR_MEDIUM_REGION(MEDIUM_REGION_FOR_PTR(ptr));
size_t bytes_used = node->bytes_used + MEDIUM_BYTES_FOR_MSIZE(this_msize);
node->bytes_used = (unsigned int)bytes_used;
if (bytes_used < DENSITY_THRESHOLD(MEDIUM_REGION_PAYLOAD_BYTES)) {
} else {
node->recirc_suitable = FALSE;
}
#if DEBUG_MALLOC
if (LOG(szone, ptr)) {
malloc_report(ASL_LEVEL_INFO, "in medium_malloc_from_free_list(), ptr=%p, this_msize=%d, msize=%d\n", ptr, this_msize, msize);
}
#endif
medium_meta_header_set_in_use(MEDIUM_META_HEADER_FOR_PTR(ptr),
MEDIUM_META_INDEX_FOR_PTR(ptr), this_msize);
medium_madvise_header_mark_dirty(MEDIUM_MADVISE_HEADER_FOR_PTR(ptr),
MEDIUM_META_INDEX_FOR_PTR(ptr), this_msize);
return ptr;
}
void *
medium_malloc_should_clear(rack_t *rack, msize_t msize, boolean_t cleared_requested)
{
void *ptr;
mag_index_t mag_index = medium_mag_get_thread_index() % rack->num_magazines;
magazine_t *medium_mag_ptr = &(rack->magazines[mag_index]);
MALLOC_TRACE(TRACE_medium_malloc, (uintptr_t)rack, MEDIUM_BYTES_FOR_MSIZE(msize), (uintptr_t)medium_mag_ptr, cleared_requested);
SZONE_MAGAZINE_PTR_LOCK(medium_mag_ptr);
#if CONFIG_MEDIUM_CACHE
ptr = medium_mag_ptr->mag_last_free;
if (medium_mag_ptr->mag_last_free_msize == msize) {
medium_mag_ptr->mag_last_free = NULL;
medium_mag_ptr->mag_last_free_msize = 0;
medium_mag_ptr->mag_last_free_rgn = NULL;
SZONE_MAGAZINE_PTR_UNLOCK(medium_mag_ptr);
CHECK(szone, __PRETTY_FUNCTION__);
if (cleared_requested) {
memset(ptr, 0, MEDIUM_BYTES_FOR_MSIZE(msize));
}
return ptr;
}
#endif
while (1) {
ptr = medium_malloc_from_free_list(rack, medium_mag_ptr, mag_index, msize);
if (ptr) {
SZONE_MAGAZINE_PTR_UNLOCK(medium_mag_ptr);
CHECK(szone, __PRETTY_FUNCTION__);
if (cleared_requested) {
memset(ptr, 0, MEDIUM_BYTES_FOR_MSIZE(msize));
}
return ptr;
}
if (medium_get_region_from_depot(rack, medium_mag_ptr, mag_index, msize)) {
ptr = medium_malloc_from_free_list(rack, medium_mag_ptr, mag_index, msize);
if (ptr) {
SZONE_MAGAZINE_PTR_UNLOCK(medium_mag_ptr);
CHECK(szone, __PRETTY_FUNCTION__);
if (cleared_requested) {
memset(ptr, 0, MEDIUM_BYTES_FOR_MSIZE(msize));
}
return ptr;
}
}
if (!medium_mag_ptr->alloc_underway) {
void *fresh_region;
medium_mag_ptr->alloc_underway = TRUE;
OSMemoryBarrier();
SZONE_MAGAZINE_PTR_UNLOCK(medium_mag_ptr);
fresh_region = mvm_allocate_pages_securely(MEDIUM_REGION_SIZE,
MEDIUM_BLOCKS_ALIGN, VM_MEMORY_MALLOC_MEDIUM,
rack->debug_flags);
SZONE_MAGAZINE_PTR_LOCK(medium_mag_ptr);
MAGMALLOC_ALLOCREGION(MEDIUM_SZONE_FROM_RACK(rack), (int)mag_index,
fresh_region, MEDIUM_REGION_SIZE);
if (!fresh_region) { medium_mag_ptr->alloc_underway = FALSE;
OSMemoryBarrier();
SZONE_MAGAZINE_PTR_UNLOCK(medium_mag_ptr);
return NULL;
}
ptr = medium_malloc_from_region_no_lock(rack, medium_mag_ptr,
mag_index, msize, fresh_region);
medium_mag_ptr->alloc_underway = FALSE;
OSMemoryBarrier();
SZONE_MAGAZINE_PTR_UNLOCK(medium_mag_ptr);
CHECK(szone, __PRETTY_FUNCTION__);
return ptr;
} else {
SZONE_MAGAZINE_PTR_UNLOCK(medium_mag_ptr);
yield();
SZONE_MAGAZINE_PTR_LOCK(medium_mag_ptr);
}
}
}
size_t
medium_size(rack_t *rack, const void *ptr)
{
if (medium_region_for_ptr_no_lock(rack, ptr)) {
if (MEDIUM_META_INDEX_FOR_PTR(ptr) >= NUM_MEDIUM_BLOCKS) {
return 0;
}
msize_t msize_and_free = *MEDIUM_METADATA_FOR_PTR(ptr);
if (msize_and_free & MEDIUM_IS_FREE) {
return 0;
}
#if CONFIG_MEDIUM_CACHE
{
mag_index_t mag_index = MAGAZINE_INDEX_FOR_MEDIUM_REGION(MEDIUM_REGION_FOR_PTR(ptr));
if (DEPOT_MAGAZINE_INDEX != mag_index) {
magazine_t *medium_mag_ptr = &(rack->magazines[mag_index]);
if (ptr == medium_mag_ptr->mag_last_free) {
return 0;
}
} else {
for (mag_index = 0; mag_index < rack->num_magazines; mag_index++) {
magazine_t *medium_mag_ptr = &(rack->magazines[mag_index]);
if (ptr == medium_mag_ptr->mag_last_free) {
return 0;
}
}
}
}
#endif
return MEDIUM_BYTES_FOR_MSIZE(msize_and_free);
}
return 0;
}
static MALLOC_NOINLINE void
free_medium_botch(rack_t *rack, void *ptr)
{
mag_index_t mag_index = MAGAZINE_INDEX_FOR_MEDIUM_REGION(MEDIUM_REGION_FOR_PTR(ptr));
magazine_t *medium_mag_ptr = &(rack->magazines[mag_index]);
SZONE_MAGAZINE_PTR_UNLOCK(medium_mag_ptr);
malloc_zone_error(rack->debug_flags, true, "double free for ptr %p\n", ptr);
}
void
free_medium(rack_t *rack, void *ptr, region_t medium_region, size_t known_size)
{
msize_t msize;
mag_index_t mag_index = MAGAZINE_INDEX_FOR_MEDIUM_REGION(MEDIUM_REGION_FOR_PTR(ptr));
magazine_t *medium_mag_ptr = &(rack->magazines[mag_index]);
if (known_size) {
msize = MEDIUM_MSIZE_FOR_BYTES(known_size + MEDIUM_QUANTUM - 1);
} else {
msize = MEDIUM_PTR_SIZE(ptr);
if (MEDIUM_PTR_IS_FREE(ptr)) {
free_medium_botch(rack, ptr);
return;
}
}
SZONE_MAGAZINE_PTR_LOCK(medium_mag_ptr);
#if CONFIG_MEDIUM_CACHE
if (DEPOT_MAGAZINE_INDEX != mag_index) {
void *ptr2 = medium_mag_ptr->mag_last_free; msize_t msize2 = medium_mag_ptr->mag_last_free_msize;
region_t rgn2 = medium_mag_ptr->mag_last_free_rgn;
if (ptr == ptr2) {
free_medium_botch(rack, ptr);
return;
}
if ((rack->debug_flags & MALLOC_DO_SCRIBBLE) && msize) {
memset(ptr, SCRABBLE_BYTE, MEDIUM_BYTES_FOR_MSIZE(msize));
}
uint64_t madv_window = medium_sliding_madvise_granularity(medium_mag_ptr);
if (MEDIUM_BYTES_FOR_MSIZE(msize) > madv_window) {
uintptr_t lo = round_page_kernel((uintptr_t)ptr);
uintptr_t hi = trunc_page_kernel((uintptr_t)ptr + MEDIUM_BYTES_FOR_MSIZE(msize));
SZONE_MAGAZINE_PTR_UNLOCK(medium_mag_ptr);
mvm_madvise_free(rack, medium_region, lo, hi, NULL, false);
SZONE_MAGAZINE_PTR_LOCK(medium_mag_ptr);
medium_madvise_header_mark_clean(MEDIUM_MADVISE_HEADER_FOR_PTR(ptr),
MEDIUM_META_INDEX_FOR_PTR(ptr), msize);
ptr2 = medium_mag_ptr->mag_last_free; msize2 = medium_mag_ptr->mag_last_free_msize;
rgn2 = medium_mag_ptr->mag_last_free_rgn;
}
medium_mag_ptr->mag_last_free = ptr;
medium_mag_ptr->mag_last_free_msize = msize;
medium_mag_ptr->mag_last_free_rgn = medium_region;
if (!ptr2) {
SZONE_MAGAZINE_PTR_UNLOCK(medium_mag_ptr);
CHECK(szone, __PRETTY_FUNCTION__);
return;
}
msize = msize2;
ptr = ptr2;
medium_region = rgn2;
}
#endif
region_trailer_t *trailer = REGION_TRAILER_FOR_MEDIUM_REGION(medium_region);
mag_index_t refreshed_index;
while (mag_index != (refreshed_index = trailer->mag_index)) {
SZONE_MAGAZINE_PTR_UNLOCK(medium_mag_ptr);
mag_index = refreshed_index;
medium_mag_ptr = &(rack->magazines[mag_index]);
SZONE_MAGAZINE_PTR_LOCK(medium_mag_ptr);
}
if (medium_free_no_lock(rack, medium_mag_ptr, mag_index, medium_region, ptr, msize)) {
SZONE_MAGAZINE_PTR_UNLOCK(medium_mag_ptr);
}
CHECK(szone, __PRETTY_FUNCTION__);
}
void
print_medium_free_list(rack_t *rack)
{
free_list_t ptr;
_SIMPLE_STRING b = _simple_salloc();
mag_index_t mag_index;
if (b) {
_simple_sappend(b, "medium free sizes:\n");
for (mag_index = -1; mag_index < rack->num_magazines; mag_index++) {
grain_t slot = 0;
_simple_sprintf(b, "\tMagazine %d: ", mag_index);
while (slot < MEDIUM_FREE_SLOT_COUNT(rack)) {
ptr = rack->magazines[mag_index].mag_free_list[slot];
if (medium_free_list_get_ptr(rack, ptr)) {
_simple_sprintf(b, "%s%y[%llu]; ", (slot == MEDIUM_FREE_SLOT_COUNT(rack) - 1) ? ">=" : "", (slot + 1) * MEDIUM_QUANTUM,
medium_free_list_count(rack, ptr));
}
slot++;
}
_simple_sappend(b, "\n");
}
malloc_report(MALLOC_REPORT_NOLOG | MALLOC_REPORT_NOPREFIX, "%s\n", _simple_string(b));
_simple_sfree(b);
}
}
void
print_medium_region(szone_t *szone, boolean_t verbose, region_t region, size_t bytes_at_start, size_t bytes_at_end)
{
unsigned counts[1024];
unsigned in_use = 0;
uintptr_t start = (uintptr_t)MEDIUM_REGION_ADDRESS(region);
uintptr_t current = start + bytes_at_start;
uintptr_t limit = (uintptr_t)MEDIUM_REGION_END(region) - bytes_at_end;
msize_t msize_and_free;
msize_t msize;
unsigned ci;
_SIMPLE_STRING b;
uintptr_t pgTot = 0;
uintptr_t advTot = 0;
if (region == HASHRING_REGION_DEALLOCATED) {
if ((b = _simple_salloc()) != NULL) {
_simple_sprintf(b, "Medium region [unknown address] was returned to the OS\n");
malloc_report(MALLOC_REPORT_NOLOG | MALLOC_REPORT_NOPREFIX, "%s\n", _simple_string(b));
_simple_sfree(b);
}
return;
}
memset(counts, 0, sizeof(counts));
while (current < limit) {
msize_and_free = *MEDIUM_METADATA_FOR_PTR(current);
msize = msize_and_free & ~MEDIUM_IS_FREE;
if (!msize) {
malloc_report(ASL_LEVEL_ERR, "*** error with %p: msize=%d\n", (void *)current, (unsigned)msize);
break;
}
if (!(msize_and_free & MEDIUM_IS_FREE)) {
if (msize < 1024) {
counts[msize]++;
}
in_use++;
} else {
uintptr_t pgLo = round_page_quanta(current + sizeof(medium_inplace_free_entry_s) + sizeof(msize_t));
uintptr_t pgHi = trunc_page_quanta(current + MEDIUM_BYTES_FOR_MSIZE(msize) - sizeof(msize_t));
if (pgLo < pgHi) {
pgTot += (pgHi - pgLo);
}
uintptr_t curAdv = current;
uintptr_t limAdv = current + MEDIUM_BYTES_FOR_MSIZE(msize);
while (curAdv < limAdv) {
msize_t adv = *(MEDIUM_MADVISE_HEADER_FOR_PTR(curAdv) +
MEDIUM_META_INDEX_FOR_PTR(curAdv));
uintptr_t bytes = MEDIUM_BYTES_FOR_MSIZE(adv & ~MEDIUM_IS_ADVISED);
if (adv & MEDIUM_IS_ADVISED) {
advTot += bytes;
}
curAdv += bytes;
}
}
current += MEDIUM_BYTES_FOR_MSIZE(msize);
}
if ((b = _simple_salloc()) != NULL) {
_simple_sprintf(b, "Medium region [%p-%p, %y] \t", (void *)start, MEDIUM_REGION_END(region), (int)MEDIUM_REGION_SIZE);
_simple_sprintf(b, "Magazine=%d \t", MAGAZINE_INDEX_FOR_MEDIUM_REGION(region));
_simple_sprintf(b, "Allocations in use=%d \t Bytes in use=%ly \t", in_use, BYTES_USED_FOR_MEDIUM_REGION(region));
if (bytes_at_end || bytes_at_start) {
_simple_sprintf(b, "Untouched=%ly ", bytes_at_end + bytes_at_start);
}
_simple_sprintf(b, "Advised=%ly ", advTot);
_simple_sprintf(b, "Dirty=%ly ", MEDIUM_REGION_PAYLOAD_BYTES -
bytes_at_start - bytes_at_end -
BYTES_USED_FOR_MEDIUM_REGION(region) - advTot);
if (verbose && in_use) {
_simple_sappend(b, "\n\tSizes in use: ");
for (ci = 0; ci < 1024; ci++) {
if (counts[ci]) {
_simple_sprintf(b, "%d[%d] ", MEDIUM_BYTES_FOR_MSIZE(ci), counts[ci]);
}
}
}
malloc_report(MALLOC_REPORT_NOLOG | MALLOC_REPORT_NOPREFIX, "%s\n", _simple_string(b));
_simple_sfree(b);
}
}
void
print_medium_region_vis(szone_t *szone, region_t region)
{
_SIMPLE_STRING b;
if (region == HASHRING_REGION_DEALLOCATED) {
if ((b = _simple_salloc()) != NULL) {
_simple_sprintf(b, "Medium region [unknown address] was returned "
"to the kernel\n");
malloc_report(MALLOC_REPORT_NOLOG | MALLOC_REPORT_NOPREFIX,
"%s\n", _simple_string(b));
_simple_sfree(b);
}
return;
}
msize_t *metah = MEDIUM_META_HEADER_FOR_PTR(region);
msize_t *madvh = MEDIUM_MADVISE_HEADER_FOR_PTR(region);
mag_index_t mag_idx = MAGAZINE_INDEX_FOR_MEDIUM_REGION(region);
magazine_t *mag = &szone->medium_rack.magazines[mag_idx];
msize_t cur_msz = 0;
bool is_free = false;
msize_t cur_adv_msz = 0;
bool is_advised = false;
if ((b = _simple_salloc()) == NULL) {
abort();
}
malloc_report(MALLOC_REPORT_NOLOG | MALLOC_REPORT_NOPREFIX,
"Medium region [%p-%p, %y, %y]\n", (void *)region,
MEDIUM_REGION_END(region), (int)MEDIUM_REGION_SIZE,
((medium_region_t)region)->trailer.bytes_used);
for (size_t x = 0; x < NUM_MEDIUM_BLOCKS; x++) {
bool is_last_rgn = mag->mag_last_region == region;
bool is_in_free_trailer = is_last_rgn && ((x << SHIFT_MEDIUM_QUANTUM) >
(MEDIUM_REGION_SIZE - mag->mag_bytes_free_at_end));
bool is_blk_start = false;
bool is_blk_end = cur_msz == 1;
bool is_adv_boundary = false;
if (cur_msz == 0) {
cur_msz = metah[x] & ~MEDIUM_IS_FREE;
is_free = metah[x] & MEDIUM_IS_FREE;
is_blk_start = true;
}
if (cur_adv_msz == 0 && madvh[x]) {
cur_adv_msz = madvh[x] & ~MEDIUM_IS_FREE;
is_advised = madvh[x] & MEDIUM_IS_FREE;
is_adv_boundary = true;
}
if (cur_adv_msz > 0) {
cur_adv_msz--;
}
const char *sigil = "?";
if (is_in_free_trailer) {
sigil = "_";
} else if (cur_msz > 0) {
if (is_free) {
if (is_advised) { sigil = "~"; }
else if (is_blk_start && cur_msz == 1) { sigil = "."; }
else if (is_blk_start) { sigil = "<"; }
else if (is_blk_end) { sigil = ">"; }
else if (is_adv_boundary) { sigil = "!"; }
else { sigil = "."; }
} else {
if (is_blk_start && cur_msz == 1) { sigil = "#"; }
else if (is_blk_start) { sigil = "["; }
else if (is_blk_end) { sigil = "]"; }
else { sigil = "#"; }
}
cur_msz--;
}
if (x > 0 && (x % 128) == 0) {
malloc_report(MALLOC_REPORT_NOLOG | MALLOC_REPORT_NOPREFIX, "%s\n",
_simple_string(b));
_simple_sfree(b);
if ((b = _simple_salloc()) == NULL) {
abort();
}
if (is_in_free_trailer) {
break;
}
}
_simple_sappend(b, sigil);
}
_simple_sfree(b);
}
static char *medium_freelist_fail_msg = "check: medium free list incorrect";
#define MEDIUM_FREELIST_FAIL(fmt, ...) \
malloc_zone_check_fail(medium_freelist_fail_msg, \
" (slot=%u), counter=%d\n" fmt, slot, counter, __VA_ARGS__);
boolean_t
medium_free_list_check(rack_t *rack, grain_t slot, unsigned counter)
{
mag_index_t mag_index;
for (mag_index = -1; mag_index < rack->num_magazines; mag_index++) {
magazine_t *medium_mag_ptr = &(rack->magazines[mag_index]);
SZONE_MAGAZINE_PTR_LOCK(medium_mag_ptr);
unsigned count = 0;
free_list_t current = rack->magazines[mag_index].mag_free_list[slot];
free_list_t previous = (free_list_t){ .p = NULL };
msize_t msize_and_free;
void *ptr = NULL;
while ((ptr = medium_free_list_get_ptr(rack, current))) {
msize_and_free = *MEDIUM_METADATA_FOR_PTR(ptr);
if (!(msize_and_free & MEDIUM_IS_FREE)) {
MEDIUM_FREELIST_FAIL("*** in-use ptr in free list slot=%u count=%d ptr=%p\n", slot, count, ptr);
SZONE_MAGAZINE_PTR_UNLOCK(medium_mag_ptr);
return 0;
}
if (((uintptr_t)ptr) & (MEDIUM_QUANTUM - 1)) {
MEDIUM_FREELIST_FAIL("*** unaligned ptr in free list slot=%u count=%d ptr=%p\n", slot, count, ptr);
SZONE_MAGAZINE_PTR_UNLOCK(medium_mag_ptr);
return 0;
}
if (!medium_region_for_ptr_no_lock(rack, ptr)) {
MEDIUM_FREELIST_FAIL("*** ptr not in szone slot=%d count=%d ptr=%p\n", slot, count, ptr);
SZONE_MAGAZINE_PTR_UNLOCK(medium_mag_ptr);
return 0;
}
if (medium_free_list_get_previous(rack, current).p != previous.p) {
MEDIUM_FREELIST_FAIL("*** previous incorrectly set slot=%u count=%d ptr=%p\n", slot, count, ptr);
SZONE_MAGAZINE_PTR_UNLOCK(medium_mag_ptr);
return 0;
}
previous = current;
current = medium_free_list_get_next(rack, current);
count++;
}
SZONE_MAGAZINE_PTR_UNLOCK(medium_mag_ptr);
}
return 1;
}
#endif // CONFIG_MEDIUM_ALLOCATOR