ed/Odin
1
0
Fork 0
mirror of https://github.com/Ed94/Odin.git synced 2026-08-02 20:58:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

mem: Poison memory for Dynamic_Arena

Browse Source
This commit is contained in:
Feoramund
2025-06-15 12:34:16 -04:00
parent de3c322899
commit 78a10b462f
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
core/mem/allocators.odin
+7
View File
@@ -1734,6 +1734,7 @@ _dynamic_arena_cycle_new_block :: proc(a: ^Dynamic_Arena, loc := #caller_locatio
nil, nil,
0, 0,
) )
sanitizer.address_poison(data)
new_block = raw_data(data) new_block = raw_data(data)
} }
a.bytes_left = a.block_size a.bytes_left = a.block_size
@@ -1822,6 +1823,8 @@ dynamic_arena_alloc_bytes_non_zeroed :: proc(a: ^Dynamic_Arena, size: int, loc :
a.current_pos = ([^]byte)(a.current_pos)[n:] a.current_pos = ([^]byte)(a.current_pos)[n:]
a.bytes_left -= n a.bytes_left -= n
result := ([^]byte)(memory)[:size] result := ([^]byte)(memory)[:size]
ensure_poisoned(result)
sanitizer.address_unpoison(result)
return result, nil return result, nil
} }
@@ -1833,10 +1836,12 @@ the unused blocks.
*/ */
dynamic_arena_reset :: proc(a: ^Dynamic_Arena, loc := #caller_location) { dynamic_arena_reset :: proc(a: ^Dynamic_Arena, loc := #caller_location) {
if a.current_block != nil { if a.current_block != nil {
sanitizer.address_poison(a.current_block, a.block_size)
append(&a.unused_blocks, a.current_block, loc=loc) append(&a.unused_blocks, a.current_block, loc=loc)
a.current_block = nil a.current_block = nil
} }
for block in a.used_blocks { for block in a.used_blocks {
sanitizer.address_poison(block, a.block_size)
append(&a.unused_blocks, block, loc=loc) append(&a.unused_blocks, block, loc=loc)
} }
clear(&a.used_blocks) clear(&a.used_blocks)
@@ -1856,6 +1861,7 @@ the unused blocks.
dynamic_arena_free_all :: proc(a: ^Dynamic_Arena, loc := #caller_location) { dynamic_arena_free_all :: proc(a: ^Dynamic_Arena, loc := #caller_location) {
dynamic_arena_reset(a) dynamic_arena_reset(a)
for block in a.unused_blocks { for block in a.unused_blocks {
sanitizer.address_unpoison(block, a.block_size)
free(block, a.block_allocator, loc) free(block, a.block_allocator, loc)
} }
clear(&a.unused_blocks) clear(&a.unused_blocks)
@@ -1965,6 +1971,7 @@ dynamic_arena_resize_bytes_non_zeroed :: proc(
old_memory := raw_data(old_data) old_memory := raw_data(old_data)
old_size := len(old_data) old_size := len(old_data)
if old_size >= size { if old_size >= size {
sanitizer.address_poison(old_data[size:])
return byte_slice(old_memory, size), nil return byte_slice(old_memory, size), nil
} }
data, err := dynamic_arena_alloc_bytes_non_zeroed(a, size, loc) data, err := dynamic_arena_alloc_bytes_non_zeroed(a, size, loc)