| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806 |
- /*
- * Copyright (C) 2019 Intel Corporation. All rights reserved.
- * SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
- */
- #include "wasm_runtime.h"
- #include "wasm_loader.h"
- #include "wasm_interp.h"
- #include "bh_common.h"
- #include "bh_log.h"
- #include "mem_alloc.h"
- #include "../common/wasm_runtime_common.h"
- #if WASM_ENABLE_SHARED_MEMORY != 0
- #include "../common/wasm_shared_memory.h"
- #endif
- static void
- set_error_buf(char *error_buf, uint32 error_buf_size, const char *string)
- {
- if (error_buf != NULL)
- snprintf(error_buf, error_buf_size, "%s", string);
- }
- WASMModule*
- wasm_load(const uint8 *buf, uint32 size,
- char *error_buf, uint32 error_buf_size)
- {
- return wasm_loader_load(buf, size, error_buf, error_buf_size);
- }
- WASMModule*
- wasm_load_from_sections(WASMSection *section_list,
- char *error_buf, uint32_t error_buf_size)
- {
- return wasm_loader_load_from_sections(section_list,
- error_buf, error_buf_size);
- }
- void
- wasm_unload(WASMModule *module)
- {
- wasm_loader_unload(module);
- }
- static void *
- runtime_malloc(uint64 size, char *error_buf, uint32 error_buf_size)
- {
- void *mem;
- if (size >= UINT32_MAX
- || !(mem = wasm_runtime_malloc((uint32)size))) {
- set_error_buf(error_buf, error_buf_size,
- "WASM module instantiate failed: "
- "allocate memory failed.");
- return NULL;
- }
- memset(mem, 0, (uint32)size);
- return mem;
- }
- #if WASM_ENABLE_MULTI_MODULE != 0
- static WASMModuleInstance *
- get_sub_module_inst(const WASMModuleInstance *parent_module_inst,
- const WASMModule *sub_module)
- {
- bh_list *sub_module_inst_list = parent_module_inst->sub_module_inst_list;
- WASMSubModInstNode *node = bh_list_first_elem(sub_module_inst_list);
- while (node && sub_module != node->module_inst->module) {
- node = bh_list_elem_next(node);
- }
- return node ? node->module_inst : NULL;
- }
- #endif
- /**
- * Destroy memory instances.
- */
- static void
- memories_deinstantiate(WASMModuleInstance *module_inst,
- WASMMemoryInstance **memories,
- uint32 count)
- {
- uint32 i;
- if (memories) {
- for (i = 0; i < count; i++)
- if (memories[i]) {
- #if WASM_ENABLE_MULTI_MODULE != 0
- if (memories[i]->owner != module_inst)
- continue;
- #endif
- #if WASM_ENABLE_SHARED_MEMORY != 0
- if (memories[i]->is_shared) {
- int32 ref_count =
- shared_memory_dec_reference(
- (WASMModuleCommon *)module_inst->module);
- bh_assert(ref_count >= 0);
- /* if the reference count is not zero,
- don't free the memory */
- if (ref_count > 0)
- continue;
- }
- #endif
- if (memories[i]->heap_handle) {
- mem_allocator_destroy(memories[i]->heap_handle);
- memories[i]->heap_handle = NULL;
- }
- wasm_runtime_free(memories[i]);
- }
- wasm_runtime_free(memories);
- }
- (void)module_inst;
- }
- static WASMMemoryInstance*
- memory_instantiate(WASMModuleInstance *module_inst,
- uint32 num_bytes_per_page,
- uint32 init_page_count, uint32 max_page_count,
- uint32 heap_size, uint32 flags,
- char *error_buf, uint32 error_buf_size)
- {
- WASMMemoryInstance *memory;
- uint64 heap_and_inst_size = offsetof(WASMMemoryInstance, base_addr) +
- (uint64)heap_size;
- uint64 total_size = heap_and_inst_size +
- num_bytes_per_page * (uint64)init_page_count;
- #if WASM_ENABLE_SHARED_MEMORY != 0
- bool is_shared_memory = flags & 0x02 ? true : false;
- /* shared memory */
- if (is_shared_memory) {
- WASMSharedMemNode *node =
- wasm_module_get_shared_memory(
- (WASMModuleCommon *)module_inst->module);
- /* If the memory of this module has been instantiated,
- return the memory instance directly */
- if (node) {
- uint32 ref_count;
- ref_count = shared_memory_inc_reference(
- (WASMModuleCommon *)module_inst->module);
- bh_assert(ref_count > 0);
- memory = (WASMMemoryInstance *)shared_memory_get_memory_inst(node);
- bh_assert(memory);
- (void)ref_count;
- return memory;
- }
- /* Allocate max page for shared memory */
- total_size = heap_and_inst_size +
- num_bytes_per_page * (uint64)max_page_count;
- }
- #endif
- /* Allocate memory space, addr data and global data */
- if (!(memory = runtime_malloc(total_size,
- error_buf, error_buf_size))) {
- return NULL;
- }
- memory->module_type = Wasm_Module_Bytecode;
- memory->num_bytes_per_page = num_bytes_per_page;
- memory->cur_page_count = init_page_count;
- memory->max_page_count = max_page_count;
- memory->heap_data = memory->base_addr;
- memory->memory_data = memory->heap_data + heap_size;
- #if WASM_ENABLE_SHARED_MEMORY != 0
- if (is_shared_memory) {
- memory->end_addr = memory->memory_data +
- num_bytes_per_page * memory->max_page_count;
- }
- else
- #endif
- {
- memory->end_addr = memory->memory_data +
- num_bytes_per_page * memory->cur_page_count;
- }
- bh_assert(memory->end_addr - (uint8*)memory == (uint32)total_size);
- /* Initialize heap */
- if (heap_size > 0
- && !(memory->heap_handle =
- mem_allocator_create(memory->heap_data, heap_size))) {
- wasm_runtime_free(memory);
- return NULL;
- }
- memory->heap_base_offset = -(int32)heap_size;
- #if WASM_ENABLE_SHARED_MEMORY != 0
- if (is_shared_memory) {
- memory->is_shared = true;
- if (!shared_memory_set_memory_inst(
- (WASMModuleCommon *)module_inst->module,
- (WASMMemoryInstanceCommon *)memory)) {
- set_error_buf(error_buf, error_buf_size,
- "Instantiate memory failed:"
- "allocate memory failed.");
- wasm_runtime_free(memory);
- return NULL;
- }
- }
- #endif
- return memory;
- }
- /**
- * Instantiate memories in a module.
- */
- static WASMMemoryInstance **
- memories_instantiate(const WASMModule *module,
- WASMModuleInstance *module_inst,
- uint32 heap_size, char *error_buf, uint32 error_buf_size)
- {
- WASMImport *import;
- uint32 mem_index = 0, i, memory_count =
- module->import_memory_count + module->memory_count;
- uint64 total_size;
- WASMMemoryInstance **memories, *memory;
- total_size = sizeof(WASMMemoryInstance*) * (uint64)memory_count;
- if (!(memories = runtime_malloc(total_size,
- error_buf, error_buf_size))) {
- return NULL;
- }
- /* instantiate memories from import section */
- import = module->import_memories;
- for (i = 0; i < module->import_memory_count; i++, import++) {
- uint32 num_bytes_per_page = import->u.memory.num_bytes_per_page;
- uint32 init_page_count = import->u.memory.init_page_count;
- uint32 max_page_count = import->u.memory.max_page_count;
- uint32 flags = import->u.memory.flags;
- uint32 actual_heap_size = heap_size;
- #if WASM_ENABLE_MULTI_MODULE != 0
- WASMMemoryInstance *memory_inst_linked = NULL;
- if (import->u.memory.import_module != NULL) {
- LOG_DEBUG("(%s, %s) is a memory of a sub-module",
- import->u.memory.module_name,
- import->u.memory.field_name);
- // TODO: how about native memory ?
- WASMModuleInstance *module_inst_linked =
- get_sub_module_inst(
- module_inst,
- import->u.memory.import_module);
- bh_assert(module_inst_linked);
- memory_inst_linked =
- wasm_lookup_memory(module_inst_linked,
- import->u.memory.field_name);
- bh_assert(memory_inst_linked);
- memories[mem_index++] = memory_inst_linked;
- memory = memory_inst_linked;
- }
- else
- #endif
- {
- if (!(memory = memories[mem_index++] = memory_instantiate(
- module_inst, num_bytes_per_page, init_page_count,
- max_page_count, actual_heap_size, flags,
- error_buf, error_buf_size))) {
- set_error_buf(error_buf, error_buf_size,
- "Instantiate memory failed: "
- "allocate memory failed.");
- memories_deinstantiate(
- module_inst,
- memories, memory_count);
- return NULL;
- }
- }
- }
- /* instantiate memories from memory section */
- for (i = 0; i < module->memory_count; i++) {
- if (!(memory = memories[mem_index++] =
- memory_instantiate(module_inst,
- module->memories[i].num_bytes_per_page,
- module->memories[i].init_page_count,
- module->memories[i].max_page_count,
- heap_size, module->memories[i].flags,
- error_buf, error_buf_size))) {
- set_error_buf(error_buf, error_buf_size,
- "Instantiate memory failed: "
- "allocate memory failed.");
- memories_deinstantiate(
- module_inst,
- memories, memory_count);
- return NULL;
- }
- #if WASM_ENABLE_MULTI_MODULE != 0
- memory->owner = module_inst;
- #endif
- }
- if (mem_index == 0) {
- /**
- * no import memory and define memory, but still need heap
- * for wasm code
- */
- if (!(memory = memories[mem_index++] =
- memory_instantiate(module_inst, 0, 0, 0, heap_size, 0,
- error_buf, error_buf_size))) {
- set_error_buf(error_buf, error_buf_size,
- "Instantiate memory failed: "
- "allocate memory failed.\n");
- memories_deinstantiate(module_inst, memories, memory_count);
- return NULL;
- }
- }
- bh_assert(mem_index == memory_count);
- (void)module_inst;
- return memories;
- }
- /**
- * Destroy table instances.
- */
- static void
- tables_deinstantiate(WASMTableInstance **tables, uint32 count)
- {
- uint32 i;
- if (tables) {
- for (i = 0; i < count; i++)
- if (tables[i])
- wasm_runtime_free(tables[i]);
- wasm_runtime_free(tables);
- }
- }
- /**
- * Instantiate tables in a module.
- */
- static WASMTableInstance **
- tables_instantiate(const WASMModule *module,
- WASMModuleInstance *module_inst,
- char *error_buf, uint32 error_buf_size)
- {
- WASMImport *import;
- uint32 table_index = 0, i, table_count =
- module->import_table_count + module->table_count;
- uint64 total_size = sizeof(WASMTableInstance*) * (uint64)table_count;
- WASMTableInstance **tables, *table;
- if (!(tables = runtime_malloc(total_size,
- error_buf, error_buf_size))) {
- return NULL;
- }
- /* instantiate tables from import section */
- import = module->import_tables;
- for (i = 0; i < module->import_table_count; i++, import++) {
- #if WASM_ENABLE_MULTI_MODULE != 0
- WASMTableInstance *table_inst_linked = NULL;
- WASMModuleInstance *module_inst_linked = NULL;
- if (import->u.table.import_module) {
- LOG_DEBUG("(%s, %s) is a table of a sub-module",
- import->u.table.module_name,
- import->u.memory.field_name);
- module_inst_linked =
- get_sub_module_inst(module_inst, import->u.table.import_module);
- bh_assert(module_inst_linked);
- table_inst_linked = wasm_lookup_table(module_inst_linked,
- import->u.table.field_name);
- bh_assert(table_inst_linked);
- total_size = offsetof(WASMTableInstance, base_addr);
- }
- else
- #endif
- {
- /* it is a built-in table */
- total_size = offsetof(WASMTableInstance, base_addr)
- + sizeof(uint32) * (uint64)import->u.table.init_size;
- }
- if (!(table = tables[table_index++] = runtime_malloc
- (total_size, error_buf, error_buf_size))) {
- tables_deinstantiate(tables, table_count);
- return NULL;
- }
- /* Set all elements to -1 to mark them as uninitialized elements */
- memset(table, -1, (uint32)total_size);
- #if WASM_ENABLE_MULTI_MODULE != 0
- table->table_inst_linked = table_inst_linked;
- if (table_inst_linked != NULL) {
- table->elem_type = table_inst_linked->elem_type;
- table->cur_size = table_inst_linked->cur_size;
- table->max_size = table_inst_linked->max_size;
- }
- else
- #endif
- {
- table->elem_type = import->u.table.elem_type;
- table->cur_size = import->u.table.init_size;
- table->max_size = import->u.table.max_size;
- }
- }
- /* instantiate tables from table section */
- for (i = 0; i < module->table_count; i++) {
- total_size = offsetof(WASMTableInstance, base_addr) +
- sizeof(uint32) * (uint64)module->tables[i].init_size;
- if (!(table = tables[table_index++] = runtime_malloc
- (total_size, error_buf, error_buf_size))) {
- tables_deinstantiate(tables, table_count);
- return NULL;
- }
- /* Set all elements to -1 to mark them as uninitialized elements */
- memset(table, -1, (uint32)total_size);
- table->elem_type = module->tables[i].elem_type;
- table->cur_size = module->tables[i].init_size;
- table->max_size = module->tables[i].max_size;
- #if WASM_ENABLE_MULTI_MODULE != 0
- table->table_inst_linked = NULL;
- #endif
- }
- bh_assert(table_index == table_count);
- (void)module_inst;
- return tables;
- }
- /**
- * Destroy function instances.
- */
- static void
- functions_deinstantiate(WASMFunctionInstance *functions, uint32 count)
- {
- if (functions) {
- wasm_runtime_free(functions);
- }
- }
- /**
- * Instantiate functions in a module.
- */
- static WASMFunctionInstance *
- functions_instantiate(const WASMModule *module,
- WASMModuleInstance *module_inst,
- char *error_buf, uint32 error_buf_size)
- {
- WASMImport *import;
- uint32 i, function_count =
- module->import_function_count + module->function_count;
- uint64 total_size = sizeof(WASMFunctionInstance) * (uint64)function_count;
- WASMFunctionInstance *functions, *function;
- if (!(functions = runtime_malloc(total_size,
- error_buf, error_buf_size))) {
- return NULL;
- }
- /* instantiate functions from import section */
- function = functions;
- import = module->import_functions;
- for (i = 0; i < module->import_function_count; i++, import++) {
- function->is_import_func = true;
- #if WASM_ENABLE_MULTI_MODULE != 0
- if (import->u.function.import_module) {
- LOG_DEBUG("(%s, %s) is a function of a sub-module",
- import->u.function.module_name,
- import->u.function.field_name);
- function->import_module_inst =
- get_sub_module_inst(module_inst,
- import->u.function.import_module);
- bh_assert(function->import_module_inst);
- WASMFunction *function_linked =
- import->u.function.import_func_linked;
- function->u.func = function_linked;
- function->import_func_inst =
- wasm_lookup_function(function->import_module_inst,
- import->u.function.field_name,
- NULL);
- bh_assert(function->import_func_inst);
- function->param_cell_num = function->u.func->param_cell_num;
- function->ret_cell_num = function->u.func->ret_cell_num;
- function->local_cell_num = function->u.func->local_cell_num;
- function->param_count =
- (uint16)function->u.func->func_type->param_count;
- function->local_count = (uint16)function->u.func->local_count;
- function->param_types = function->u.func->func_type->types;
- function->local_types = function->u.func->local_types;
- function->local_offsets = function->u.func->local_offsets;
- #if WASM_ENABLE_FAST_INTERP != 0
- function->const_cell_num = function->u.func->const_cell_num;
- #endif
- }
- else
- #endif /* WASM_ENABLE_MULTI_MODULE */
- {
- LOG_DEBUG("(%s, %s) is a function of native",
- import->u.function.module_name,
- import->u.function.field_name);
- function->u.func_import = &import->u.function;
- function->param_cell_num =
- import->u.function.func_type->param_cell_num;
- function->ret_cell_num =
- import->u.function.func_type->ret_cell_num;
- function->param_count =
- (uint16)function->u.func_import->func_type->param_count;
- function->param_types = function->u.func_import->func_type->types;
- function->local_cell_num = 0;
- function->local_count = 0;
- function->local_types = NULL;
- }
- function++;
- }
- /* instantiate functions from function section */
- for (i = 0; i < module->function_count; i++) {
- function->is_import_func = false;
- function->u.func = module->functions[i];
- function->param_cell_num = function->u.func->param_cell_num;
- function->ret_cell_num = function->u.func->ret_cell_num;
- function->local_cell_num = function->u.func->local_cell_num;
- function->param_count = (uint16)function->u.func->func_type->param_count;
- function->local_count = (uint16)function->u.func->local_count;
- function->param_types = function->u.func->func_type->types;
- function->local_types = function->u.func->local_types;
- function->local_offsets = function->u.func->local_offsets;
- #if WASM_ENABLE_FAST_INTERP != 0
- function->const_cell_num = function->u.func->const_cell_num;
- #endif
- function++;
- }
- bh_assert((uint32)(function - functions) == function_count);
- (void)module_inst;
- return functions;
- }
- /**
- * Destroy global instances.
- */
- static void
- globals_deinstantiate(WASMGlobalInstance *globals)
- {
- if (globals)
- wasm_runtime_free(globals);
- }
- /**
- * init_expr->u ==> init_val
- */
- static bool
- parse_init_expr(const InitializerExpression *init_expr,
- const WASMGlobalInstance *global_inst_array,
- uint32 boundary, WASMValue *init_val)
- {
- if (init_expr->init_expr_type == INIT_EXPR_TYPE_GET_GLOBAL) {
- uint32 target_global_index = init_expr->u.global_index;
- /**
- * a global gets the init value of another global
- */
- if (target_global_index >= boundary) {
- LOG_DEBUG("unknown target global, %d", target_global_index);
- return false;
- }
- /**
- * it will work if using WASMGlobalImport and WASMGlobal in
- * WASMModule, but will have to face complicated cases
- *
- * but we still have no sure the target global has been
- * initialized before
- */
- WASMValue target_value =
- global_inst_array[target_global_index].initial_value;
- bh_memcpy_s(init_val, sizeof(WASMValue), &target_value,
- sizeof(target_value));
- }
- else {
- bh_memcpy_s(init_val, sizeof(WASMValue), &init_expr->u,
- sizeof(init_expr->u));
- }
- return true;
- }
- /**
- * Instantiate globals in a module.
- */
- static WASMGlobalInstance *
- globals_instantiate(const WASMModule *module,
- WASMModuleInstance *module_inst,
- uint32 *p_global_data_size, char *error_buf,
- uint32 error_buf_size)
- {
- WASMImport *import;
- uint32 global_data_offset = 0;
- uint32 i, global_count =
- module->import_global_count + module->global_count;
- uint64 total_size = sizeof(WASMGlobalInstance) * (uint64)global_count;
- WASMGlobalInstance *globals, *global;
- if (!(globals = runtime_malloc(total_size,
- error_buf, error_buf_size))) {
- return NULL;
- }
- /* instantiate globals from import section */
- global = globals;
- import = module->import_globals;
- for (i = 0; i < module->import_global_count; i++, import++) {
- WASMGlobalImport *global_import = &import->u.global;
- global->type = global_import->type;
- global->is_mutable = global_import->is_mutable;
- #if WASM_ENABLE_MULTI_MODULE != 0
- if (global_import->import_module) {
- WASMModuleInstance *sub_module_inst = get_sub_module_inst(
- module_inst, global_import->import_module);
- bh_assert(sub_module_inst);
- WASMGlobalInstance *global_inst_linked =
- wasm_lookup_global(sub_module_inst, global_import->field_name);
- bh_assert(global_inst_linked);
- global->import_global_inst = global_inst_linked;
- global->import_module_inst = sub_module_inst;
- /**
- * although, actually don't need initial_value for an imported
- * global, we keep it here like a place holder because of
- * global-data and
- * (global $g2 i32 (global.get $g1))
- */
- WASMGlobal *linked_global = global_import->import_global_linked;
- InitializerExpression *linked_init_expr =
- &(linked_global->init_expr);
- bool ret = parse_init_expr(
- linked_init_expr,
- sub_module_inst->globals,
- sub_module_inst->global_count, &(global->initial_value));
- if (!ret) {
- set_error_buf(error_buf, error_buf_size,
- "Instantiate global failed: unknown global.");
- return NULL;
- }
- }
- else
- #endif
- {
- /* native globals share their initial_values in one module */
- global->initial_value = global_import->global_data_linked;
- }
- global->data_offset = global_data_offset;
- global_data_offset += wasm_value_type_size(global->type);
- global++;
- }
- /* instantiate globals from global section */
- for (i = 0; i < module->global_count; i++) {
- bool ret = false;
- uint32 global_count =
- module->import_global_count + module->global_count;
- InitializerExpression *init_expr = &(module->globals[i].init_expr);
- global->type = module->globals[i].type;
- global->is_mutable = module->globals[i].is_mutable;
- global->data_offset = global_data_offset;
- global_data_offset += wasm_value_type_size(global->type);
- /**
- * first init, it might happen that the target global instance
- * has not been initialize yet
- */
- if (init_expr->init_expr_type != INIT_EXPR_TYPE_GET_GLOBAL) {
- ret =
- parse_init_expr(init_expr, globals, global_count,
- &(global->initial_value));
- if (!ret) {
- set_error_buf(error_buf, error_buf_size,
- "Instantiate global failed: unknown global.");
- return NULL;
- }
- }
- global++;
- }
- bh_assert((uint32)(global - globals) == global_count);
- *p_global_data_size = global_data_offset;
- (void)module_inst;
- return globals;
- }
- static bool
- globals_instantiate_fix(WASMGlobalInstance *globals,
- const WASMModule *module,
- char *error_buf, uint32 error_buf_size)
- {
- WASMGlobalInstance *global = globals;
- uint32 i;
- uint32 global_count = module->import_global_count + module->global_count;
- /**
- * second init, only target global instances from global
- * (ignore import_global)
- * to fix skipped init_value in the previous round
- * hope two rounds are enough but how about a chain ?
- */
- for (i = 0; i < module->global_count; i++) {
- bool ret = false;
- InitializerExpression *init_expr = &module->globals[i].init_expr;
- if (init_expr->init_expr_type == INIT_EXPR_TYPE_GET_GLOBAL) {
- ret = parse_init_expr(init_expr, globals, global_count,
- &global->initial_value);
- if (!ret) {
- set_error_buf(error_buf, error_buf_size,
- "Instantiate global failed: unknown global.");
- return false;
- }
- }
- global++;
- }
- return true;
- }
- /**
- * Return export function count in module export section.
- */
- static uint32
- get_export_count(const WASMModule *module, uint8 kind)
- {
- WASMExport *export = module->exports;
- uint32 count = 0, i;
- for (i = 0; i < module->export_count; i++, export++)
- if (export->kind == kind)
- count++;
- return count;
- }
- /**
- * Destroy export function instances.
- */
- static void
- export_functions_deinstantiate(WASMExportFuncInstance *functions)
- {
- if (functions)
- wasm_runtime_free(functions);
- }
- /**
- * Instantiate export functions in a module.
- */
- static WASMExportFuncInstance*
- export_functions_instantiate(const WASMModule *module,
- WASMModuleInstance *module_inst,
- uint32 export_func_count,
- char *error_buf, uint32 error_buf_size)
- {
- WASMExportFuncInstance *export_funcs, *export_func;
- WASMExport *export = module->exports;
- uint32 i;
- uint64 total_size = sizeof(WASMExportFuncInstance) * (uint64)export_func_count;
- if (!(export_func = export_funcs = runtime_malloc
- (total_size, error_buf, error_buf_size))) {
- return NULL;
- }
- for (i = 0; i < module->export_count; i++, export++)
- if (export->kind == EXPORT_KIND_FUNC) {
- export_func->name = export->name;
- export_func->function = &module_inst->functions[export->index];
- export_func++;
- }
- bh_assert((uint32)(export_func - export_funcs) == export_func_count);
- return export_funcs;
- }
- #if WASM_ENABLE_MULTI_MODULE != 0
- static void
- export_globals_deinstantiate(WASMExportGlobInstance *globals)
- {
- if (globals)
- wasm_runtime_free(globals);
- }
- static WASMExportGlobInstance *
- export_globals_instantiate(const WASMModule *module,
- WASMModuleInstance *module_inst,
- uint32 export_glob_count, char *error_buf,
- uint32 error_buf_size)
- {
- WASMExportGlobInstance *export_globals, *export_global;
- WASMExport *export = module->exports;
- uint32 i;
- uint64 total_size = sizeof(WASMExportGlobInstance) * (uint64)export_glob_count;
- if (!(export_global = export_globals = runtime_malloc
- (total_size, error_buf, error_buf_size))) {
- return NULL;
- }
- for (i = 0; i < module->export_count; i++, export++)
- if (export->kind == EXPORT_KIND_GLOBAL) {
- export_global->name = export->name;
- export_global->global = &module_inst->globals[export->index];
- export_global++;
- }
- bh_assert((uint32)(export_global - export_globals) == export_glob_count);
- return export_globals;
- }
- #endif
- static bool
- execute_post_inst_function(WASMModuleInstance *module_inst)
- {
- WASMFunctionInstance *post_inst_func = NULL;
- WASMType *post_inst_func_type;
- uint32 i;
- for (i = 0; i < module_inst->export_func_count; i++)
- if (!strcmp(module_inst->export_functions[i].name, "__post_instantiate")) {
- post_inst_func = module_inst->export_functions[i].function;
- break;
- }
- if (!post_inst_func)
- /* Not found */
- return true;
- post_inst_func_type = post_inst_func->u.func->func_type;
- if (post_inst_func_type->param_count != 0
- || post_inst_func_type->result_count != 0)
- /* Not a valid function type, ignore it */
- return true;
- return wasm_create_exec_env_and_call_function(module_inst, post_inst_func,
- 0, NULL);
- }
- #if WASM_ENABLE_BULK_MEMORY != 0
- static bool
- execute_memory_init_function(WASMModuleInstance *module_inst)
- {
- WASMFunctionInstance *memory_init_func = NULL;
- WASMType *memory_init_func_type;
- uint32 i;
- for (i = 0; i < module_inst->export_func_count; i++)
- if (!strcmp(module_inst->export_functions[i].name, "__wasm_call_ctors")) {
- memory_init_func = module_inst->export_functions[i].function;
- break;
- }
- if (!memory_init_func)
- /* Not found */
- return true;
- memory_init_func_type = memory_init_func->u.func->func_type;
- if (memory_init_func_type->param_count != 0
- || memory_init_func_type->result_count != 0)
- /* Not a valid function type, ignore it */
- return true;
- return wasm_create_exec_env_and_call_function(module_inst,
- memory_init_func,
- 0, NULL);
- }
- #endif
- static bool
- execute_start_function(WASMModuleInstance *module_inst)
- {
- WASMFunctionInstance *func = module_inst->start_function;
- if (!func)
- return true;
- bh_assert(!func->is_import_func && func->param_cell_num == 0
- && func->ret_cell_num == 0);
- return wasm_create_exec_env_and_call_function(module_inst, func, 0, NULL);
- }
- #if WASM_ENABLE_MULTI_MODULE != 0
- static bool
- sub_module_instantiate(WASMModule *module, WASMModuleInstance *module_inst,
- uint32 stack_size, uint32 heap_size, char *error_buf,
- uint32 error_buf_size)
- {
- bh_list *sub_module_inst_list = module_inst->sub_module_inst_list;
- WASMRegisteredModule *sub_module_list_node =
- bh_list_first_elem(module->import_module_list);
- while (sub_module_list_node) {
- WASMModule *sub_module = (WASMModule*)sub_module_list_node->module;
- WASMModuleInstance *sub_module_inst = wasm_instantiate(
- sub_module, false, stack_size, heap_size, error_buf, error_buf_size);
- if (!sub_module_inst) {
- LOG_DEBUG("instantiate %s failed",
- sub_module_list_node->module_name);
- set_error_buf_v(error_buf, error_buf_size, "instantiate %s failed",
- sub_module_list_node->module_name);
- return false;
- }
- WASMSubModInstNode *sub_module_inst_list_node = runtime_malloc
- (sizeof(WASMSubModInstNode), error_buf, error_buf_size);
- if (!sub_module_inst_list_node) {
- LOG_DEBUG("Malloc WASMSubModInstNode failed, SZ:%d",
- sizeof(WASMSubModInstNode));
- wasm_deinstantiate(sub_module_inst, false);
- return false;
- }
- sub_module_inst_list_node->module_inst = sub_module_inst;
- sub_module_inst_list_node->module_name =
- sub_module_list_node->module_name;
- bh_list_status ret =
- bh_list_insert(sub_module_inst_list, sub_module_inst_list_node);
- bh_assert(BH_LIST_SUCCESS == ret);
- (void)ret;
- sub_module_list_node = bh_list_elem_next(sub_module_list_node);
- }
- return true;
- }
- static void
- sub_module_deinstantiate(WASMModuleInstance *module_inst)
- {
- bh_list *list = module_inst->sub_module_inst_list;
- WASMSubModInstNode *node = bh_list_first_elem(list);
- while (node) {
- WASMSubModInstNode *next_node = bh_list_elem_next(node);
- bh_list_remove(list, node);
- wasm_deinstantiate(node->module_inst, false);
- node = next_node;
- }
- }
- #endif
- /**
- * Instantiate module
- */
- WASMModuleInstance*
- wasm_instantiate(WASMModule *module, bool is_sub_inst,
- uint32 stack_size, uint32 heap_size,
- char *error_buf, uint32 error_buf_size)
- {
- WASMModuleInstance *module_inst;
- WASMGlobalInstance *globals = NULL, *global;
- uint32 global_count, global_data_size = 0, i;
- uint32 base_offset, length;
- uint8 *global_data, *global_data_end;
- #if WASM_ENABLE_MULTI_MODULE != 0
- bool ret = false;
- #endif
- if (!module)
- return NULL;
- /* Check heap size */
- heap_size = align_uint(heap_size, 8);
- if (heap_size > APP_HEAP_SIZE_MAX)
- heap_size = APP_HEAP_SIZE_MAX;
- /* Allocate the memory */
- if (!(module_inst = runtime_malloc(sizeof(WASMModuleInstance),
- error_buf, error_buf_size))) {
- return NULL;
- }
- memset(module_inst, 0, (uint32)sizeof(WASMModuleInstance));
- module_inst->module = module;
- #if WASM_ENABLE_MULTI_MODULE != 0
- module_inst->sub_module_inst_list =
- &module_inst->sub_module_inst_list_head;
- ret = sub_module_instantiate(module, module_inst, stack_size, heap_size,
- error_buf, error_buf_size);
- if (!ret) {
- LOG_DEBUG("build a sub module list failed");
- wasm_deinstantiate(module_inst, false);
- return NULL;
- }
- #endif
- /* Instantiate global firstly to get the mutable data size */
- global_count = module->import_global_count + module->global_count;
- if (global_count && !(globals = globals_instantiate(
- module,
- module_inst,
- &global_data_size, error_buf, error_buf_size))) {
- wasm_deinstantiate(module_inst, false);
- return NULL;
- }
- module_inst->global_count = global_count;
- module_inst->globals = globals;
- module_inst->memory_count =
- module->import_memory_count + module->memory_count;
- module_inst->table_count =
- module->import_table_count + module->table_count;
- module_inst->function_count =
- module->import_function_count + module->function_count;
- /* export */
- module_inst->export_func_count = get_export_count(module, EXPORT_KIND_FUNC);
- #if WASM_ENABLE_MULTI_MODULE != 0
- module_inst->export_tab_count = get_export_count(module, EXPORT_KIND_TABLE);
- module_inst->export_mem_count = get_export_count(module, EXPORT_KIND_MEMORY);
- module_inst->export_glob_count = get_export_count(module, EXPORT_KIND_GLOBAL);
- #endif
- if (global_count > 0) {
- if (!(module_inst->global_data = runtime_malloc
- (global_data_size, error_buf, error_buf_size))) {
- wasm_deinstantiate(module_inst, false);
- return NULL;
- }
- }
- /* Instantiate memories/tables/functions */
- if ((module_inst->memory_count > 0
- && !(module_inst->memories =
- memories_instantiate(module,
- module_inst,
- heap_size, error_buf, error_buf_size)))
- || (module_inst->table_count > 0
- && !(module_inst->tables =
- tables_instantiate(module,
- module_inst,
- error_buf, error_buf_size)))
- || (module_inst->function_count > 0
- && !(module_inst->functions =
- functions_instantiate(module,
- module_inst,
- error_buf, error_buf_size)))
- || (module_inst->export_func_count > 0
- && !(module_inst->export_functions = export_functions_instantiate(
- module, module_inst, module_inst->export_func_count,
- error_buf, error_buf_size)))
- #if WASM_ENABLE_MULTI_MODULE != 0
- || (module_inst->export_glob_count > 0
- && !(module_inst->export_globals = export_globals_instantiate(
- module, module_inst, module_inst->export_glob_count,
- error_buf, error_buf_size)))
- #endif
- ) {
- wasm_deinstantiate(module_inst, false);
- return NULL;
- }
- if (global_count > 0) {
- /**
- * since there might be some globals are not instantiate the first
- * instantiate round
- */
- if (!globals_instantiate_fix(globals, module,
- error_buf, error_buf_size)) {
- wasm_deinstantiate(module_inst, false);
- return NULL;
- }
- /* Initialize the global data */
- global_data = module_inst->global_data;
- global_data_end = global_data + global_data_size;
- global = globals;
- for (i = 0; i < global_count; i++, global++) {
- switch (global->type) {
- case VALUE_TYPE_I32:
- case VALUE_TYPE_F32:
- *(int32*)global_data = global->initial_value.i32;
- global_data += sizeof(int32);
- break;
- case VALUE_TYPE_I64:
- case VALUE_TYPE_F64:
- bh_memcpy_s(global_data, (uint32)(global_data_end - global_data),
- &global->initial_value.i64, sizeof(int64));
- global_data += sizeof(int64);
- break;
- default:
- bh_assert(0);
- }
- }
- bh_assert(global_data == global_data_end);
- }
- /* Initialize the memory data with data segment section */
- module_inst->default_memory =
- module_inst->memory_count ? module_inst->memories[0] : NULL;
- for (i = 0; i < module->data_seg_count; i++) {
- WASMMemoryInstance *memory = NULL;
- uint8 *memory_data = NULL;
- uint32 memory_size = 0;
- WASMDataSeg *data_seg = module->data_segments[i];
- #if WASM_ENABLE_BULK_MEMORY != 0
- if (data_seg->is_passive)
- continue;
- #endif
- /* has check it in loader */
- memory = module_inst->memories[data_seg->memory_index];
- bh_assert(memory);
- memory_data = memory->memory_data;
- bh_assert(memory_data);
- memory_size = memory->num_bytes_per_page * memory->cur_page_count;
- bh_assert(data_seg->base_offset.init_expr_type
- == INIT_EXPR_TYPE_I32_CONST
- || data_seg->base_offset.init_expr_type
- == INIT_EXPR_TYPE_GET_GLOBAL);
- if (data_seg->base_offset.init_expr_type
- == INIT_EXPR_TYPE_GET_GLOBAL) {
- bh_assert(data_seg->base_offset.u.global_index < global_count
- && globals[data_seg->base_offset.u.global_index].type
- == VALUE_TYPE_I32);
- data_seg->base_offset.u.i32 =
- globals[data_seg->base_offset.u.global_index]
- .initial_value.i32;
- }
- /* check offset since length might negative */
- base_offset = (uint32)data_seg->base_offset.u.i32;
- if (base_offset > memory_size) {
- LOG_DEBUG("base_offset(%d) > memory_size(%d)", base_offset,
- memory_size);
- set_error_buf(error_buf, error_buf_size,
- "data segment does not fit.");
- wasm_deinstantiate(module_inst, false);
- return NULL;
- }
- /* check offset + length(could be zero) */
- length = data_seg->data_length;
- if (base_offset + length > memory_size) {
- LOG_DEBUG("base_offset(%d) + length(%d) > memory_size(%d)",
- base_offset, length, memory_size);
- set_error_buf(
- error_buf, error_buf_size,
- "Instantiate module failed: data segment does not fit.");
- wasm_deinstantiate(module_inst, false);
- return NULL;
- }
- bh_memcpy_s(memory_data + base_offset, memory_size - base_offset,
- data_seg->data, length);
- }
- /* Initialize the table data with table segment section */
- module_inst->default_table =
- module_inst->table_count ? module_inst->tables[0] : NULL;
- for (i = 0; i < module->table_seg_count; i++) {
- WASMTableSeg *table_seg = module->table_segments + i;
- /* has check it in loader */
- WASMTableInstance *table = module_inst->tables[table_seg->table_index];
- bh_assert(table);
- uint32 *table_data = (uint32 *)table->base_addr;
- #if WASM_ENABLE_MULTI_MODULE != 0
- table_data = table->table_inst_linked
- ? (uint32 *)table->table_inst_linked->base_addr
- : table_data;
- #endif
- bh_assert(table_data);
- /* init vec(funcidx) */
- bh_assert(table_seg->base_offset.init_expr_type
- == INIT_EXPR_TYPE_I32_CONST
- || table_seg->base_offset.init_expr_type
- == INIT_EXPR_TYPE_GET_GLOBAL);
- if (table_seg->base_offset.init_expr_type
- == INIT_EXPR_TYPE_GET_GLOBAL) {
- bh_assert(table_seg->base_offset.u.global_index < global_count
- && globals[table_seg->base_offset.u.global_index].type
- == VALUE_TYPE_I32);
- table_seg->base_offset.u.i32 =
- globals[table_seg->base_offset.u.global_index].initial_value.i32;
- }
- /* check offset since length might negative */
- if ((uint32)table_seg->base_offset.u.i32 > table->cur_size) {
- LOG_DEBUG("base_offset(%d) > table->cur_size(%d)",
- table_seg->base_offset.u.i32, table->cur_size);
- set_error_buf(error_buf, error_buf_size,
- "elements segment does not fit");
- wasm_deinstantiate(module_inst, false);
- return NULL;
- }
- /* check offset + length(could be zero) */
- length = table_seg->function_count;
- if ((uint32)table_seg->base_offset.u.i32 + length > table->cur_size) {
- LOG_DEBUG("base_offset(%d) + length(%d)> table->cur_size(%d)",
- table_seg->base_offset.u.i32, length, table->cur_size);
- set_error_buf(error_buf, error_buf_size,
- "elements segment does not fit");
- wasm_deinstantiate(module_inst, false);
- return NULL;
- }
- /**
- * Check function index in the current module inst for now.
- * will check the linked table inst owner in future.
- * so loader check is enough
- */
- bh_memcpy_s(
- table_data + table_seg->base_offset.u.i32,
- (uint32)((table->cur_size - (uint32)table_seg->base_offset.u.i32)
- * sizeof(uint32)),
- table_seg->func_indexes, (uint32)(length * sizeof(uint32)));
- }
- #if WASM_ENABLE_LIBC_WASI != 0
- /* The sub-instance will get the wasi_ctx from main-instance */
- if (!is_sub_inst) {
- if (heap_size > 0
- && !wasm_runtime_init_wasi((WASMModuleInstanceCommon*)module_inst,
- module->wasi_args.dir_list,
- module->wasi_args.dir_count,
- module->wasi_args.map_dir_list,
- module->wasi_args.map_dir_count,
- module->wasi_args.env,
- module->wasi_args.env_count,
- module->wasi_args.argv,
- module->wasi_args.argc,
- error_buf, error_buf_size)) {
- wasm_deinstantiate(module_inst, false);
- return NULL;
- }
- }
- #endif
- if (module->start_function != (uint32)-1) {
- /* TODO: fix start function can be import function issue */
- if (module->start_function >= module->import_function_count)
- module_inst->start_function =
- &module_inst->functions[module->start_function];
- }
- /* module instance type */
- module_inst->module_type = Wasm_Module_Bytecode;
- /* Initialize the thread related data */
- if (stack_size == 0)
- stack_size = DEFAULT_WASM_STACK_SIZE;
- #if WASM_ENABLE_SPEC_TEST != 0
- if (stack_size < 48 *1024)
- stack_size = 48 * 1024;
- #endif
- module_inst->default_wasm_stack_size = stack_size;
- /* Execute __post_instantiate function */
- if (!execute_post_inst_function(module_inst)
- || !execute_start_function(module_inst)) {
- set_error_buf(error_buf, error_buf_size,
- module_inst->cur_exception);
- wasm_deinstantiate(module_inst, false);
- return NULL;
- }
- #if WASM_ENABLE_BULK_MEMORY != 0
- #if WASM_ENABLE_LIBC_WASI != 0
- if (!module->is_wasi_module) {
- #endif
- /* Only execute the memory init function for main instance because
- the data segments will be dropped once initialized.
- */
- if (!is_sub_inst) {
- if (!execute_memory_init_function(module_inst)) {
- set_error_buf(error_buf, error_buf_size,
- module_inst->cur_exception);
- wasm_deinstantiate(module_inst, false);
- return NULL;
- }
- }
- #if WASM_ENABLE_LIBC_WASI != 0
- }
- #endif
- #endif
- (void)global_data_end;
- return module_inst;
- }
- void
- wasm_deinstantiate(WASMModuleInstance *module_inst, bool is_sub_inst)
- {
- if (!module_inst)
- return;
- #if WASM_ENABLE_MULTI_MODULE != 0
- sub_module_deinstantiate(module_inst);
- #endif
- #if WASM_ENABLE_LIBC_WASI != 0
- /* Destroy wasi resource before freeing app heap, since some fields of
- wasi contex are allocated from app heap, and if app heap is freed,
- these fields will be set to NULL, we cannot free their internal data
- which may allocated from global heap. */
- /* Only destroy wasi ctx in the main module instance */
- if (!is_sub_inst)
- wasm_runtime_destroy_wasi((WASMModuleInstanceCommon*)module_inst);
- #endif
- if (module_inst->memory_count > 0)
- memories_deinstantiate(
- module_inst,
- module_inst->memories, module_inst->memory_count);
- tables_deinstantiate(module_inst->tables, module_inst->table_count);
- functions_deinstantiate(module_inst->functions, module_inst->function_count);
- globals_deinstantiate(module_inst->globals);
- export_functions_deinstantiate(module_inst->export_functions);
- #if WASM_ENABLE_MULTI_MODULE != 0
- export_globals_deinstantiate(module_inst->export_globals);
- #endif
- if (module_inst->global_data)
- wasm_runtime_free(module_inst->global_data);
- wasm_runtime_free(module_inst);
- }
- WASMFunctionInstance*
- wasm_lookup_function(const WASMModuleInstance *module_inst,
- const char *name, const char *signature)
- {
- uint32 i;
- for (i = 0; i < module_inst->export_func_count; i++)
- if (!strcmp(module_inst->export_functions[i].name, name))
- return module_inst->export_functions[i].function;
- (void)signature;
- return NULL;
- }
- #if WASM_ENABLE_MULTI_MODULE != 0
- WASMGlobalInstance *
- wasm_lookup_global(const WASMModuleInstance *module_inst, const char *name)
- {
- uint32 i;
- for (i = 0; i < module_inst->export_glob_count; i++)
- if (!strcmp(module_inst->export_globals[i].name, name))
- return module_inst->export_globals[i].global;
- return NULL;
- }
- WASMMemoryInstance *
- wasm_lookup_memory(const WASMModuleInstance *module_inst, const char *name)
- {
- /**
- * using a strong assumption that one module instance only has
- * one memory instance
- */
- (void)module_inst->export_memories;
- return module_inst->memories[0];
- }
- WASMTableInstance *
- wasm_lookup_table(const WASMModuleInstance *module_inst, const char *name)
- {
- /**
- * using a strong assumption that one module instance only has
- * one table instance
- */
- (void)module_inst->export_tables;
- return module_inst->tables[0];
- }
- #endif
- bool
- wasm_call_function(WASMExecEnv *exec_env,
- WASMFunctionInstance *function,
- unsigned argc, uint32 argv[])
- {
- WASMModuleInstance *module_inst = (WASMModuleInstance*)exec_env->module_inst;
- wasm_interp_call_wasm(module_inst, exec_env, function, argc, argv);
- return !wasm_get_exception(module_inst) ? true : false;
- }
- bool
- wasm_create_exec_env_and_call_function(WASMModuleInstance *module_inst,
- WASMFunctionInstance *func,
- unsigned argc, uint32 argv[])
- {
- WASMExecEnv *exec_env;
- bool ret;
- if (!(exec_env = wasm_exec_env_create(
- (WASMModuleInstanceCommon*)module_inst,
- module_inst->default_wasm_stack_size))) {
- wasm_set_exception(module_inst, "allocate memory failed.");
- return false;
- }
- /* set thread handle and stack boundary */
- wasm_exec_env_set_thread_info(exec_env);
- ret = wasm_call_function(exec_env, func, argc, argv);
- wasm_exec_env_destroy(exec_env);
- return ret;
- }
- void
- wasm_set_exception(WASMModuleInstance *module_inst,
- const char *exception)
- {
- if (exception)
- snprintf(module_inst->cur_exception,
- sizeof(module_inst->cur_exception),
- "Exception: %s", exception);
- else
- module_inst->cur_exception[0] = '\0';
- }
- const char*
- wasm_get_exception(WASMModuleInstance *module_inst)
- {
- if (module_inst->cur_exception[0] == '\0')
- return NULL;
- else
- return module_inst->cur_exception;
- }
- int32
- wasm_module_malloc(WASMModuleInstance *module_inst, uint32 size,
- void **p_native_addr)
- {
- WASMMemoryInstance *memory = module_inst->default_memory;
- uint8 *addr = mem_allocator_malloc(memory->heap_handle, size);
- if (!addr) {
- wasm_set_exception(module_inst, "out of memory");
- return 0;
- }
- if (p_native_addr)
- *p_native_addr = addr;
- return (int32)(addr - memory->memory_data);
- }
- void
- wasm_module_free(WASMModuleInstance *module_inst, int32 ptr)
- {
- if (ptr) {
- WASMMemoryInstance *memory = module_inst->default_memory;
- uint8 *addr = memory->memory_data + ptr;
- if (memory->heap_data < addr && addr < memory->memory_data)
- mem_allocator_free(memory->heap_handle, addr);
- }
- }
- int32
- wasm_module_dup_data(WASMModuleInstance *module_inst,
- const char *src, uint32 size)
- {
- char *buffer;
- int32 buffer_offset = wasm_module_malloc(module_inst, size,
- (void**)&buffer);
- if (buffer_offset != 0) {
- buffer = wasm_addr_app_to_native(module_inst, buffer_offset);
- bh_memcpy_s(buffer, size, src, size);
- }
- return buffer_offset;
- }
- bool
- wasm_validate_app_addr(WASMModuleInstance *module_inst,
- int32 app_offset, uint32 size)
- {
- WASMMemoryInstance *memory = module_inst->default_memory;
- int32 memory_data_size =
- (int32)(memory->num_bytes_per_page * memory->cur_page_count);
- /* integer overflow check */
- if (app_offset + (int32)size < app_offset) {
- goto fail;
- }
- if (memory->heap_base_offset <= app_offset
- && app_offset + (int32)size <= memory_data_size) {
- return true;
- }
- fail:
- wasm_set_exception(module_inst, "out of bounds memory access");
- return false;
- }
- bool
- wasm_validate_native_addr(WASMModuleInstance *module_inst,
- void *native_ptr, uint32 size)
- {
- uint8 *addr = (uint8*)native_ptr;
- WASMMemoryInstance *memory = module_inst->default_memory;
- int32 memory_data_size =
- (int32)(memory->num_bytes_per_page * memory->cur_page_count);
- if (addr + size < addr) {
- goto fail;
- }
- if (memory->heap_data <= addr
- && addr + size <= memory->memory_data + memory_data_size) {
- return true;
- }
- fail:
- wasm_set_exception(module_inst, "out of bounds memory access");
- return false;
- }
- void *
- wasm_addr_app_to_native(WASMModuleInstance *module_inst,
- int32 app_offset)
- {
- WASMMemoryInstance *memory = module_inst->default_memory;
- uint8 *addr = memory->memory_data + app_offset;
- int32 memory_data_size =
- (int32)(memory->num_bytes_per_page * memory->cur_page_count);
- if (memory->heap_data <= addr
- && addr < memory->memory_data + memory_data_size)
- return addr;
- return NULL;
- }
- int32
- wasm_addr_native_to_app(WASMModuleInstance *module_inst,
- void *native_ptr)
- {
- WASMMemoryInstance *memory = module_inst->default_memory;
- uint8 *addr = (uint8*)native_ptr;
- int32 memory_data_size =
- (int32)(memory->num_bytes_per_page * memory->cur_page_count);
- if (memory->heap_data <= addr
- && addr < memory->memory_data + memory_data_size)
- return (int32)(addr - memory->memory_data);
- return 0;
- }
- bool
- wasm_get_app_addr_range(WASMModuleInstance *module_inst,
- int32 app_offset,
- int32 *p_app_start_offset,
- int32 *p_app_end_offset)
- {
- WASMMemoryInstance *memory = module_inst->default_memory;
- int32 memory_data_size =
- (int32)(memory->num_bytes_per_page * memory->cur_page_count);
- if (memory->heap_base_offset <= app_offset
- && app_offset < memory_data_size) {
- if (p_app_start_offset)
- *p_app_start_offset = memory->heap_base_offset;
- if (p_app_end_offset)
- *p_app_end_offset = memory_data_size;
- return true;
- }
- return false;
- }
- bool
- wasm_get_native_addr_range(WASMModuleInstance *module_inst,
- uint8 *native_ptr,
- uint8 **p_native_start_addr,
- uint8 **p_native_end_addr)
- {
- WASMMemoryInstance *memory = module_inst->default_memory;
- uint8 *addr = (uint8*)native_ptr;
- int32 memory_data_size =
- (int32)(memory->num_bytes_per_page * memory->cur_page_count);
- if (memory->heap_data <= addr
- && addr < memory->memory_data + memory_data_size) {
- if (p_native_start_addr)
- *p_native_start_addr = memory->heap_data;
- if (p_native_end_addr)
- *p_native_end_addr = memory->memory_data + memory_data_size;
- return true;
- }
- return false;
- }
- bool
- wasm_enlarge_memory(WASMModuleInstance *module, uint32 inc_page_count)
- {
- WASMMemoryInstance *memory = module->default_memory, *new_memory;
- uint32 heap_size = memory->memory_data - memory->heap_data;
- uint32 total_size_old = memory->end_addr - (uint8*)memory;
- uint32 total_page_count = inc_page_count + memory->cur_page_count;
- uint64 total_size = offsetof(WASMMemoryInstance, base_addr)
- + (uint64)heap_size
- + memory->num_bytes_per_page * (uint64)total_page_count;
- void *heap_handle_old = memory->heap_handle;
- if (inc_page_count <= 0)
- /* No need to enlarge memory */
- return true;
- if (total_page_count < memory->cur_page_count /* integer overflow */
- || total_page_count > memory->max_page_count) {
- wasm_set_exception(module, "fail to enlarge memory.");
- return false;
- }
- if (total_size >= UINT32_MAX) {
- wasm_set_exception(module, "fail to enlarge memory.");
- return false;
- }
- #if WASM_ENABLE_SHARED_MEMORY != 0
- if (memory->is_shared) {
- /* For shared memory, we have reserved the maximum spaces during
- instantiate, only change the cur_page_count here */
- memory->cur_page_count = total_page_count;
- return true;
- }
- #endif
- if (heap_size > 0) {
- /* Destroy heap's lock firstly, if its memory is re-allocated,
- we cannot access its lock again. */
- mem_allocator_destroy_lock(memory->heap_handle);
- }
- if (!(new_memory = wasm_runtime_realloc(memory, (uint32)total_size))) {
- if (!(new_memory = wasm_runtime_malloc((uint32)total_size))) {
- if (heap_size > 0) {
- /* Restore heap's lock if memory re-alloc failed */
- mem_allocator_reinit_lock(memory->heap_handle);
- }
- wasm_set_exception(module, "fail to enlarge memory.");
- return false;
- }
- bh_memcpy_s((uint8*)new_memory, (uint32)total_size,
- (uint8*)memory, total_size_old);
- wasm_runtime_free(memory);
- }
- memset((uint8*)new_memory + total_size_old,
- 0, (uint32)total_size - total_size_old);
- if (heap_size > 0) {
- new_memory->heap_handle = (uint8*)heap_handle_old +
- ((uint8*)new_memory - (uint8*)memory);
- if (mem_allocator_migrate(new_memory->heap_handle,
- heap_handle_old) != 0) {
- wasm_set_exception(module, "fail to enlarge memory.");
- return false;
- }
- }
- new_memory->cur_page_count = total_page_count;
- new_memory->heap_data = new_memory->base_addr;
- new_memory->memory_data = new_memory->base_addr + heap_size;
- new_memory->end_addr = new_memory->memory_data +
- new_memory->num_bytes_per_page * total_page_count;
- module->memories[0] = module->default_memory = new_memory;
- return true;
- }
- bool
- wasm_call_indirect(WASMExecEnv *exec_env,
- uint32_t element_indices,
- uint32_t argc, uint32_t argv[])
- {
- WASMModuleInstance *module_inst = NULL;
- WASMTableInstance *table_inst = NULL;
- uint32_t function_indices = 0;
- WASMFunctionInstance *function_inst = NULL;
- module_inst =
- (WASMModuleInstance*)exec_env->module_inst;
- bh_assert(module_inst);
- table_inst = module_inst->default_table;
- if (!table_inst) {
- wasm_set_exception(module_inst, "unknown table");
- goto got_exception;
- }
- if (element_indices >= table_inst->cur_size) {
- wasm_set_exception(module_inst, "undefined element");
- goto got_exception;
- }
- /**
- * please be aware that table_inst->base_addr may point
- * to another module's table
- **/
- function_indices = ((uint32_t*)table_inst->base_addr)[element_indices];
- if (function_indices == 0xFFFFFFFF) {
- wasm_set_exception(module_inst, "uninitialized element");
- goto got_exception;
- }
- /**
- * we insist to call functions owned by the module itself
- **/
- if (function_indices >= module_inst->function_count) {
- wasm_set_exception(module_inst, "unknown function");
- goto got_exception;
- }
- function_inst = module_inst->functions + function_indices;
- wasm_interp_call_wasm(module_inst, exec_env, function_inst, argc, argv);
- return !wasm_get_exception(module_inst) ? true : false;
- got_exception:
- return false;
- }
- #if WASM_ENABLE_THREAD_MGR != 0
- bool
- wasm_set_aux_stack(WASMExecEnv *exec_env,
- uint32 start_offset, uint32 size)
- {
- WASMModuleInstance *module_inst =
- (WASMModuleInstance*)exec_env->module_inst;
- uint32 stack_top_idx =
- module_inst->module->llvm_aux_stack_global_index;
- uint32 data_end =
- module_inst->module->llvm_aux_data_end;
- uint32 stack_bottom =
- module_inst->module->llvm_aux_stack_bottom;
- bool is_stack_before_data =
- stack_bottom < data_end ? true : false;
- /* Check the aux stack space, currently we don't allocate space in heap */
- if ((is_stack_before_data && (size > start_offset))
- || ((!is_stack_before_data) && (start_offset - data_end < size)))
- return false;
- if ((stack_bottom != (uint32)-1) && (stack_top_idx != (uint32)-1)) {
- /* The aux stack top is a wasm global,
- set the initial value for the global */
- uint8 *global_addr =
- module_inst->global_data +
- module_inst->globals[stack_top_idx].data_offset;
- *(int32*)global_addr = start_offset;
- /* The aux stack boundary is a constant value,
- set the value to exec_env */
- exec_env->aux_stack_boundary = start_offset - size;
- return true;
- }
- return false;
- }
- bool
- wasm_get_aux_stack(WASMExecEnv *exec_env,
- uint32 *start_offset, uint32 *size)
- {
- WASMModuleInstance *module_inst =
- (WASMModuleInstance*)exec_env->module_inst;
- /* The aux stack information is resolved in loader
- and store in module */
- uint32 stack_bottom =
- module_inst->module->llvm_aux_stack_bottom;
- uint32 total_aux_stack_size =
- module_inst->module->llvm_aux_stack_size;
- if (stack_bottom != 0 && total_aux_stack_size != 0) {
- if (start_offset)
- *start_offset = stack_bottom;
- if (size)
- *size = total_aux_stack_size;
- return true;
- }
- return false;
- }
- #endif
|