Loading...
--- libmalloc/libmalloc-521.120.7/src/early_malloc.c
+++ libmalloc/libmalloc-657.80.3/src/early_malloc.c
@@ -69,7 +69,7 @@
static_assert(CHAR_BIT == 8, "CHAR_BIT is 8");
static_assert(powerof2(MFM_ALLOC_SIZE_MAX), "MFM_ALLOC_SIZE_MAX is a power of 2");
-static_assert(powerof2(MFM_BLOCK_SIZE_MAX), "MFM_ALLOC_SIZE_MAX is a power of 2");
+static_assert(powerof2(MFM_BLOCK_SIZE_MAX), "MFM_BLOCK_SIZE_MAX is a power of 2");
#if !MALLOC_TARGET_EXCLAVES
#define MFM_INTERNAL_CRASH(code, msg) ({ \
@@ -158,7 +158,6 @@
static struct mfm_arena *mfm_arena;
-
#pragma mark validation and helper functions
/*!
@@ -531,6 +530,7 @@
#endif
}
+
/*!
* @function __mfm_block_insert_head()
*
@@ -544,15 +544,18 @@
struct mfm_block *blk)
{
uint64_t head, offs, next;
+ struct mfm_block *next_blk;
head = __mfm_block_offset(arena, hblk);
next = __mfm_block_next(hblk);
offs = __mfm_block_offset(arena, blk);
+ next_blk = &arena->mfm_base[next];
+
blk->mfmb_prev = head;
__mfm_block_set_next(blk, next);
__mfm_block_set_next(hblk, offs);
- arena->mfm_base[next].mfmb_prev = offs;
+ next_blk->mfmb_prev = offs;
}
/*!
@@ -565,11 +568,15 @@
__mfm_block_remove(struct mfm_arena *arena, struct mfm_block *blk)
{
uint64_t next, prev;
+ struct mfm_block *next_blk, *prev_blk;
+
next = __mfm_block_next(blk);
prev = blk->mfmb_prev;
- arena->mfm_base[next].mfmb_prev = prev;
- __mfm_block_set_next(&arena->mfm_base[prev], next);
+ next_blk = &arena->mfm_base[next];
+ prev_blk = &arena->mfm_base[prev];
+ next_blk->mfmb_prev = prev;
+ __mfm_block_set_next(prev_blk, next);
__builtin_bzero(blk, sizeof(struct mfm_block));
}
@@ -638,6 +645,7 @@
#if MALLOC_TARGET_EXCLAVES
plat_map_t map = {0};
#endif // MALLOC_TARGET_EXCLAVES
+
#if MALLOC_TARGET_EXCLAVES
arena = mvm_allocate_pages_plat(MFM_ARENA_SIZE, 0, MALLOC_NO_POPULATE,
@@ -671,6 +679,8 @@
mach_vm_address_t vm_addr = (mach_vm_address_t)arena;
mach_vm_size_t vm_size = (mach_vm_size_t)MFM_ARENA_SIZE;
int alloc_flags = VM_FLAGS_OVERWRITE | VM_MAKE_TAG(VM_MEMORY_MALLOC_TINY);
+
+
kern_return_t kr = mach_vm_map(mach_task_self(), &vm_addr, vm_size,
/* mask */ 0, alloc_flags, MEMORY_OBJECT_NULL, /* offset */ 0,
/* copy */ false, VM_PROT_DEFAULT, VM_PROT_ALL, VM_INHERIT_DEFAULT);
@@ -708,6 +718,7 @@
{
struct mfm_arena *arena = os_atomic_load(&mfm_arena, dependency);
size_t index;
+
if (!__mfm_address_owned(arena, ptr)) {
return 0ul;
@@ -809,6 +820,7 @@
#endif
__mfm_unlock(arena);
+
return ptr;
}
@@ -817,20 +829,23 @@
{
struct mfm_arena *arena = os_atomic_load(&mfm_arena, dependency);
size_t index, size;
+ void *addr = ptr;
#if MFM_TRACE
dprintf(STDERR_FILENO, "{ -1, %p },\n", ptr);
#endif
- if (!__mfm_address_owned(arena, ptr)) {
+
+ if (!__mfm_address_owned(arena, addr)) {
MFM_INTERNAL_CRASH(ptr, "not MFM owned");
}
- index = __mfm_block_index(arena, ptr);
+ index = __mfm_block_index(arena, addr);
if (!__mfm_block_is_allocated(arena, index)) {
MFM_CLIENT_CRASH(ptr, "not an allocated block");
}
size = __mfm_block_size(arena, index);
+
bzero(ptr, MFM_QUANTUM * size);
@@ -881,6 +896,7 @@
mfm_claimed_address(void *ptr)
{
struct mfm_arena *arena = os_atomic_load(&mfm_arena, dependency);
+
return __mfm_address_owned(arena, ptr);
}