feat: add callback to handle memory.grow failures

Author: eloparco

core/iwasm/common/wasm_memory.c

@@ -8,6 +8,7 @@
 #include "../aot/aot_runtime.h"
 #include "bh_platform.h"
 #include "mem_alloc.h"
+#include "wasm_memory.h"
 
 #if WASM_ENABLE_SHARED_MEMORY != 0
 #include "../common/wasm_shared_memory.h"
@@ -24,6 +25,8 @@ static Memory_Mode memory_mode = MEMORY_MODE_UNKNOWN;
 
 static mem_allocator_t pool_allocator = NULL;
 
+static memory_grow_error_callback memory_grow_error_cb;
+
 #if WASM_MEM_ALLOC_WITH_USER_DATA != 0
 static void *allocator_user_data = NULL;
 static void *(*malloc_func)(void *user_data, unsigned int size) = NULL;
@@ -763,6 +766,12 @@ wasm_enlarge_memory_internal(WASMModuleInstance *module, uint32 inc_page_count)
 }
 #endif /* end of OS_ENABLE_HW_BOUND_CHECK */
 
+void
+set_memory_grow_error_callback(const memory_grow_error_callback callback)
+{
+    memory_grow_error_cb = callback;
+}
+
 bool
 wasm_enlarge_memory(WASMModuleInstance *module, uint32 inc_page_count)
 {
@@ -778,5 +787,8 @@ wasm_enlarge_memory(WASMModuleInstance *module, uint32 inc_page_count)
         shared_memory_unlock(module->memories[0]);
 #endif
 
+    if (!ret && memory_grow_error_cb)
+        memory_grow_error_cb(inc_page_count);
+
     return ret;
 }

core/iwasm/common/wasm_memory.h

@@ -24,6 +24,9 @@ wasm_runtime_memory_destroy();
 unsigned
 wasm_runtime_memory_pool_size();
 
+void
+set_memory_grow_error_callback(const memory_grow_error_callback callback);
+
 #ifdef __cplusplus
 }
 #endif

core/iwasm/include/wasm_export.h

@@ -1439,6 +1439,14 @@ WASM_RUNTIME_API_EXTERN bool
 wasm_runtime_is_import_global_linked(const char *module_name,
                                      const char *global_name);
 
+typedef void (*memory_grow_error_callback)(uint32_t inc_page_count);
+
+/**
+ * Setup callback invoked when memory.grow fails
+ */
+WASM_RUNTIME_API_EXTERN void
+set_memory_grow_error_callback(const memory_grow_error_callback callback);
+
 /* clang-format on */
 
 #ifdef __cplusplus