/************************************************************//**
*
*	@file: memory_impl.c
*	@author: Martin Fouilleul
*	@date: 17/04/2023
*
*****************************************************************/

#include"orca_app.h"

void* wasm_memory_resize_callback(void* p, size_t new_size, size_t old_size, void* userData)
{
	wasm_memory* memory = (wasm_memory*)userData;

	if(memory->committed >= new_size)
	{
		return(memory->ptr);
	}
	else if(memory->committed < memory->reserved)
	{
		u32 commitSize = new_size - memory->committed;

		mem_base_allocator* allocator = mem_base_allocator_default();
		mem_base_commit(allocator, memory->ptr + memory->committed, commitSize);
		memory->committed += commitSize;
		return(memory->ptr);
	}
	else
	{
		DEBUG_ASSERT(0, "Out of memory");
		return(0);
	}
	return NULL;
}

void wasm_memory_free_callback(void* p, size_t size, void* userData)
{
	wasm_memory* memory = (wasm_memory*)userData;

	mem_base_allocator* allocator = mem_base_allocator_default();
	mem_base_release(allocator, memory->ptr, memory->reserved);
	memset(memory, 0, sizeof(wasm_memory));
}

extern u32 orca_mem_grow(u64 size)
{
	const size_t PAGE_SIZE = 65536;

	orca_runtime* runtime = orca_runtime_get();
	wasm_memory* memory = &runtime->wasmMemory;

	size = AlignUpOnPow2(size, PAGE_SIZE);
	// u64 totalSize = size + m3_GetMemorySize(runtime->m3Runtime);
	// bb_slice memory = bb_module_instance_mem_all(runtime->bbModuleInst);

	u32 addr = memory->committed;

	//NOTE: call resize memory, which will call our custom resize callback... this is a bit involved because
	//      wasm3 doesn't allow resizing the memory directly
	// M3Result res = ResizeMemory(runtime->m3Runtime, totalSize/PAGE_SIZE);
	bb_module_instance_mem_grow(runtime->bbModuleInst, size/PAGE_SIZE);

	return(addr);
}

void* wasm_memory_offset_to_ptr(wasm_memory* memory, u32 offset)
{
	DEBUG_ASSERT(offset < bb_module_instance_mem_all(orca_runtime_get()->bbModuleInst).length, "Wasm offset exceeds memory length")
	return memory->ptr + offset;
	// M3MemoryHeader* header = (M3MemoryHeader*)(memory->ptr);
	// DEBUG_ASSERT(offset < header->length, "Wasm offset exceeds memory length")
	// return memory->ptr + sizeof(M3MemoryHeader) + offset;
}