language-servers/roc
1
0
Fork 0
mirror of https://github.com/roc-lang/roc.git synced 2025-10-02 00:01:16 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 6

Handle freed values in refcount tests

Browse source
This commit is contained in:
Brian Carroll 2021-12-29 17:31:05 +00:00
parent 4d2e4d454b
commit 7aa3f77b3c
2 changed files with 32 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

11
compiler/test_gen/src/helpers/wasm.rs
View file

@ -300,8 +300,15 @@ where
let mut refcounts = Vec::with_capacity(num_refcounts);
for i in 0..num_refcounts {
let rc_encoded = refcount_ptrs[i].get().deref(memory).unwrap().get();
let rc = (rc_encoded - i32::MIN + 1) as u32;
let rc_ptr = refcount_ptrs[i].get();
let rc = if rc_ptr.offset() == 0 {
// RC pointer has been set to null, which means the value has been freed.
// In tests, we simply represent this as zero refcount.
0
} else {
let rc_encoded = rc_ptr.deref(memory).unwrap().get();
(rc_encoded - i32::MIN + 1) as u32
};
refcounts.push(rc);
}
Ok(refcounts)

26
compiler/test_gen/src/helpers/wasm_test_platform.c
View file

@ -35,6 +35,13 @@ size_t **init_refcount_test(size_t max_allocs)
abort();
#endif
size_t *data_ptr_to_rc_ptr(void *ptr, unsigned int alignment)
{
size_t alloc_addr = (size_t)ptr;
size_t rc_addr = alloc_addr + alignment - sizeof(size_t);
return (size_t *)rc_addr;
}
//--------------------------
void *roc_alloc(size_t size, unsigned int alignment)
@ -46,9 +53,7 @@ void *roc_alloc(size_t size, unsigned int alignment)
ASSERT(alignment >= sizeof(size_t));
ASSERT(rc_pointers_index < rc_pointers_len);
size_t alloc_addr = (size_t)allocated;
size_t rc_addr = alloc_addr + alignment - sizeof(size_t);
size_t *rc_ptr = (size_t *)rc_addr;
size_t *rc_ptr = data_ptr_to_rc_ptr(allocated, alignment);
rc_pointers[rc_pointers_index] = rc_ptr;
rc_pointers_index++;
}
@ -83,6 +88,21 @@ void *roc_realloc(void *ptr, size_t new_size, size_t old_size,
void roc_dealloc(void *ptr, unsigned int alignment)
{
// Null out the entry in the test array to signal that it was freed
// Then even if malloc reuses the space, everything still works
size_t *rc_ptr = data_ptr_to_rc_ptr(ptr, alignment);
int i = 0;
for (; i < rc_pointers_index; ++i)
{
if (rc_pointers[i] == rc_ptr)
{
rc_pointers[i] = NULL;
break;
}
}
int was_found = i < rc_pointers_index;
ASSERT(was_found);
#if ENABLE_PRINTF
printf("roc_dealloc deallocated %p with alignment %zd\n", ptr, alignment);
#endif