| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031 |
- /*
- * Copyright (C) 2019 Intel Corporation. All rights reserved.
- * SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
- */
- #include "bh_common.h"
- #include "bh_log.h"
- #include "wasm_export.h"
- #include "../interpreter/wasm.h"
- #include "../common/wasm_runtime_common.h"
- #include "thread_manager.h"
- #define WAMR_PTHREAD_KEYS_MAX 32
- #define get_module(exec_env) \
- wasm_exec_env_get_module(exec_env)
- #define get_module_inst(exec_env) \
- wasm_runtime_get_module_inst(exec_env)
- #define get_thread_arg(exec_env) \
- wasm_exec_env_get_thread_arg(exec_env)
- #define get_wasi_ctx(module_inst) \
- wasm_runtime_get_wasi_ctx(module_inst)
- #define validate_app_addr(offset, size) \
- wasm_runtime_validate_app_addr(module_inst, offset, size)
- #define validate_native_addr(addr, size) \
- wasm_runtime_validate_native_addr(module_inst, addr, size)
- #define addr_app_to_native(offset) \
- wasm_runtime_addr_app_to_native(module_inst, offset)
- #define addr_native_to_app(ptr) \
- wasm_runtime_addr_native_to_app(module_inst, ptr)
- extern bool
- wasm_runtime_call_indirect(wasm_exec_env_t exec_env,
- uint32 element_indices,
- uint32 argc, uint32 argv[]);
- enum {
- T_THREAD,
- T_MUTEX,
- T_COND,
- };
- enum thread_status_t {
- THREAD_INIT,
- THREAD_RUNNING,
- THREAD_CANCELLED,
- THREAD_EXIT,
- };
- enum mutex_status_t {
- MUTEX_CREATED,
- MUTEX_DESTROYED,
- };
- enum cond_status_t {
- COND_CREATED,
- COND_DESTROYED,
- };
- typedef struct ThreadKeyValueNode {
- bh_list_link l;
- wasm_exec_env_t exec_env;
- int32 thread_key_values[WAMR_PTHREAD_KEYS_MAX];
- } ThreadKeyValueNode;
- typedef struct KeyData {
- int32 destructor_func;
- bool is_created;
- } KeyData;
- typedef struct ClusterInfoNode {
- bh_list_link l;
- WASMCluster *cluster;
- HashMap *thread_info_map;
- /* Key data list */
- KeyData key_data_list[WAMR_PTHREAD_KEYS_MAX];
- korp_mutex key_data_list_lock;
- /* Every node contains the key value list for a thread */
- bh_list thread_list_head;
- bh_list *thread_list;
- } ClusterInfoNode;
- typedef struct ThreadInfoNode {
- wasm_exec_env_t parent_exec_env;
- wasm_exec_env_t exec_env;
- /* the id returned to app */
- uint32 handle;
- /* type can be [THREAD | MUTEX | CONDITION] */
- uint32 type;
- /* Thread status, this variable should be volatile
- as its value may be changed in different threads */
- volatile uint32 status;
- union {
- korp_tid thread;
- korp_mutex *mutex;
- korp_cond *cond;
- } u;
- } ThreadInfoNode;
- typedef struct {
- ThreadInfoNode *info_node;
- /* table elem index of the app's entry function */
- uint32 elem_index;
- /* arg of the app's entry function */
- void *arg;
- wasm_module_inst_t module_inst;
- } ThreadRoutineArgs;
- static bh_list cluster_info_list;
- static korp_mutex pthread_global_lock;
- static uint32 handle_id = 1;
- static void
- lib_pthread_destroy_callback(WASMCluster *cluster);
- static uint32
- thread_handle_hash(void *handle)
- {
- return (uint32)(uintptr_t)handle;
- }
- static bool
- thread_handle_equal(void *h1, void *h2)
- {
- return (uint32)(uintptr_t)h1 == (uint32)(uintptr_t)h2 ? true : false;
- }
- static void
- thread_info_destroy(void *node)
- {
- ThreadInfoNode *info_node = (ThreadInfoNode *)node;
- ThreadRoutineArgs *args;
- pthread_mutex_lock(&pthread_global_lock);
- if (info_node->type == T_THREAD) {
- args = get_thread_arg(info_node->exec_env);
- if (args) {
- wasm_runtime_free(args);
- }
- }
- else if (info_node->type == T_MUTEX) {
- if (info_node->status != MUTEX_DESTROYED)
- os_mutex_destroy(info_node->u.mutex);
- wasm_runtime_free(info_node->u.mutex);
- }
- else if (info_node->type == T_COND) {
- if (info_node->status != COND_DESTROYED)
- os_cond_destroy(info_node->u.cond);
- wasm_runtime_free(info_node->u.cond);
- }
- wasm_runtime_free(info_node);
- pthread_mutex_unlock(&pthread_global_lock);
- }
- bool
- lib_pthread_init()
- {
- if (0 != os_mutex_init(&pthread_global_lock))
- return false;
- bh_list_init(&cluster_info_list);
- if (!wasm_cluster_register_destroy_callback(
- lib_pthread_destroy_callback)) {
- os_mutex_destroy(&pthread_global_lock);
- return false;
- }
- return true;
- }
- void
- lib_pthread_destroy()
- {
- os_mutex_destroy(&pthread_global_lock);
- }
- static ClusterInfoNode*
- get_cluster_info(WASMCluster *cluster)
- {
- ClusterInfoNode *node;
- os_mutex_lock(&pthread_global_lock);
- node = bh_list_first_elem(&cluster_info_list);
- while (node) {
- if (cluster == node->cluster) {
- os_mutex_unlock(&pthread_global_lock);
- return node;
- }
- node = bh_list_elem_next(node);
- }
- os_mutex_unlock(&pthread_global_lock);
- return NULL;
- }
- static KeyData*
- key_data_list_lookup(wasm_exec_env_t exec_env, int32 key)
- {
- ClusterInfoNode *node;
- WASMCluster *cluster =
- wasm_exec_env_get_cluster(exec_env);
- if ((node = get_cluster_info(cluster))) {
- return (key >= 0 && key < WAMR_PTHREAD_KEYS_MAX
- && node->key_data_list[key].is_created)
- ? &(node->key_data_list[key]) : NULL;
- }
- return NULL;
- }
- /* Lookup the thread key value node for a thread,
- create a new one if failed
- This design will reduce the memory usage. If the thread doesn't use
- the local storage, it will not occupy memory space
- */
- static int32*
- key_value_list_lookup_or_create(wasm_exec_env_t exec_env,
- ClusterInfoNode *info, int32 key)
- {
- KeyData *key_node;
- ThreadKeyValueNode *data;
- /* Check if the key is valid */
- key_node = key_data_list_lookup(exec_env, key);
- if (!key_node) {
- return NULL;
- }
- /* Find key values node */
- data = bh_list_first_elem(info->thread_list);
- while (data) {
- if (data->exec_env == exec_env)
- return data->thread_key_values;
- data = bh_list_elem_next(data);
- }
- /* If not found, create a new node for this thread */
- if (!(data = wasm_runtime_malloc(sizeof(ThreadKeyValueNode))))
- return NULL;
- memset(data, 0, sizeof(ThreadKeyValueNode));
- data->exec_env = exec_env;
- if (bh_list_insert(info->thread_list, data) != 0) {
- wasm_runtime_free(data);
- return NULL;
- }
- return data->thread_key_values;
- }
- static void
- call_key_destructor(wasm_exec_env_t exec_env)
- {
- int32 i;
- uint32 destructor_index;
- KeyData *key_node;
- ThreadKeyValueNode *value_node;
- WASMCluster *cluster = wasm_exec_env_get_cluster(exec_env);
- ClusterInfoNode *info = get_cluster_info(cluster);
- if (!info) {
- return;
- }
- value_node = bh_list_first_elem(info->thread_list);
- while (value_node) {
- if (value_node->exec_env == exec_env)
- break;
- value_node = bh_list_elem_next(value_node);
- }
- /* This thread hasn't created key value node */
- if (!value_node)
- return;
- /* Destroy key values */
- for (i = 0; i < WAMR_PTHREAD_KEYS_MAX; i++) {
- if (value_node->thread_key_values[i] != 0) {
- int32 value = value_node->thread_key_values[i];
- os_mutex_lock(&info->key_data_list_lock);
- if ((key_node = key_data_list_lookup(exec_env, i)))
- destructor_index = key_node->destructor_func;
- else
- destructor_index = 0;
- os_mutex_unlock(&info->key_data_list_lock);
- /* reset key value */
- value_node->thread_key_values[i] = 0;
- /* Call the destructor func provided by app */
- if (destructor_index) {
- uint32 argv[1];
- argv[0] = value;
- wasm_runtime_call_indirect(exec_env,
- destructor_index,
- 1, argv);
- }
- }
- }
- bh_list_remove(info->thread_list, value_node);
- wasm_runtime_free(value_node);
- }
- static void
- destroy_thread_key_value_list(bh_list *list)
- {
- ThreadKeyValueNode *node, *next;
- /* There should be only one node for main thread */
- bh_assert(list->len <= 1);
- if (list->len) {
- node = bh_list_first_elem(list);
- while (node) {
- next = bh_list_elem_next(node);
- call_key_destructor(node->exec_env);
- node = next;
- }
- }
- }
- static ClusterInfoNode*
- create_cluster_info(WASMCluster *cluster)
- {
- ClusterInfoNode *node;
- bh_list_status ret;
- if (!(node = wasm_runtime_malloc(sizeof(ClusterInfoNode)))) {
- return NULL;
- }
- memset(node, 0, sizeof(WASMCluster));
- node->thread_list = &node->thread_list_head;
- ret = bh_list_init(node->thread_list);
- bh_assert(ret == BH_LIST_SUCCESS);
- if (os_mutex_init(&node->key_data_list_lock) != 0) {
- wasm_runtime_free(node);
- return NULL;
- }
- node->cluster = cluster;
- if (!(node->thread_info_map =
- bh_hash_map_create(32, true,
- (HashFunc)thread_handle_hash,
- (KeyEqualFunc)thread_handle_equal,
- NULL,
- thread_info_destroy))) {
- os_mutex_destroy(&node->key_data_list_lock);
- wasm_runtime_free(node);
- return NULL;
- }
- os_mutex_lock(&pthread_global_lock);
- ret = bh_list_insert(&cluster_info_list, node);
- bh_assert(ret == BH_LIST_SUCCESS);
- os_mutex_unlock(&pthread_global_lock);
- (void)ret;
- return node;
- }
- static bool
- destroy_cluster_info(WASMCluster *cluster)
- {
- ClusterInfoNode *node = get_cluster_info(cluster);
- if (node) {
- bh_hash_map_destroy(node->thread_info_map);
- destroy_thread_key_value_list(node->thread_list);
- os_mutex_destroy(&node->key_data_list_lock);
- /* Remove from the cluster info list */
- os_mutex_lock(&pthread_global_lock);
- bh_list_remove(&cluster_info_list, node);
- wasm_runtime_free(node);
- os_mutex_unlock(&pthread_global_lock);
- return true;
- }
- return false;
- }
- static void
- lib_pthread_destroy_callback(WASMCluster *cluster)
- {
- destroy_cluster_info(cluster);
- }
- static void
- delete_thread_info_node(ThreadInfoNode *thread_info)
- {
- ClusterInfoNode *node;
- bool ret;
- WASMCluster *cluster =
- wasm_exec_env_get_cluster(thread_info->exec_env);
- if ((node = get_cluster_info(cluster))) {
- ret = bh_hash_map_remove(node->thread_info_map,
- (void *)(uintptr_t)thread_info->handle,
- NULL, NULL);
- (void)ret;
- }
- thread_info_destroy(thread_info);
- }
- static bool
- append_thread_info_node(ThreadInfoNode *thread_info)
- {
- ClusterInfoNode *node;
- WASMCluster *cluster =
- wasm_exec_env_get_cluster(thread_info->exec_env);
- if (!(node = get_cluster_info(cluster))) {
- if (!(node = create_cluster_info(cluster))) {
- return false;
- }
- }
- if (!bh_hash_map_insert(node->thread_info_map,
- (void *)(uintptr_t)thread_info->handle,
- thread_info)) {
- return false;
- }
- return true;
- }
- static ThreadInfoNode*
- get_thread_info(wasm_exec_env_t exec_env, uint32 handle)
- {
- WASMCluster *cluster = wasm_exec_env_get_cluster(exec_env);
- ClusterInfoNode *info = get_cluster_info(cluster);
- if (!info) {
- return NULL;
- }
- return bh_hash_map_find(info->thread_info_map, (void *)(uintptr_t)handle);
- }
- static uint32
- allocate_handle()
- {
- uint32 id;
- os_mutex_lock(&pthread_global_lock);
- id = handle_id++;
- os_mutex_unlock(&pthread_global_lock);
- return id;
- }
- static void*
- pthread_start_routine(void *arg)
- {
- wasm_exec_env_t exec_env = (wasm_exec_env_t)arg;
- wasm_exec_env_t parent_exec_env;
- wasm_module_inst_t module_inst = get_module_inst(exec_env);
- ThreadRoutineArgs *routine_args = exec_env->thread_arg;
- ThreadInfoNode *info_node = routine_args->info_node;
- uint32 argv[1];
- parent_exec_env = info_node->parent_exec_env;
- os_mutex_lock(&parent_exec_env->wait_lock);
- info_node->exec_env = exec_env;
- info_node->u.thread = exec_env->handle;
- if (!append_thread_info_node(info_node)) {
- wasm_runtime_deinstantiate_internal(module_inst, true);
- delete_thread_info_node(info_node);
- os_cond_signal(&parent_exec_env->wait_cond);
- os_mutex_unlock(&parent_exec_env->wait_lock);
- return NULL;
- }
- info_node->status = THREAD_RUNNING;
- os_cond_signal(&parent_exec_env->wait_cond);
- os_mutex_unlock(&parent_exec_env->wait_lock);
- if (!validate_native_addr(routine_args->arg, sizeof(uint32))) {
- /* If there are exceptions, copy the exception to
- all other instance in this cluster */
- wasm_cluster_spread_exception(exec_env);
- wasm_runtime_deinstantiate_internal(module_inst, true);
- delete_thread_info_node(info_node);
- return NULL;
- }
- wasm_exec_env_set_thread_info(exec_env);
- argv[0] = addr_native_to_app(routine_args->arg);
- if(!wasm_runtime_call_indirect(exec_env,
- routine_args->elem_index,
- 1, argv)) {
- if (wasm_runtime_get_exception(module_inst))
- wasm_cluster_spread_exception(exec_env);
- }
- /* destroy pthread key values */
- call_key_destructor(exec_env);
- /* routine exit, destroy instance */
- wasm_runtime_deinstantiate_internal(module_inst, true);
- info_node->status = THREAD_EXIT;
- delete_thread_info_node(info_node);
- return (void *)(uintptr_t)argv[0];
- }
- static int
- pthread_create_wrapper(wasm_exec_env_t exec_env,
- uint32 *thread, /* thread_handle */
- const void *attr, /* not supported */
- uint32 elem_index, /* entry function */
- void *arg) /* arguments buffer */
- {
- wasm_module_t module = get_module(exec_env);
- wasm_module_inst_t new_module_inst = NULL;
- ThreadInfoNode *info_node = NULL;
- ThreadRoutineArgs *routine_args = NULL;
- uint32 thread_handle;
- int32 ret = -1;
- #if WASM_ENABLE_LIBC_WASI != 0
- wasm_module_inst_t module_inst = get_module_inst(exec_env);
- WASIContext *wasi_ctx = get_wasi_ctx(module_inst);
- #endif
- bh_assert(module);
- if (!(new_module_inst =
- wasm_runtime_instantiate_internal(module, true, 8192, 0,
- NULL, 0)))
- return -1;
- #if WASM_ENABLE_LIBC_WASI != 0
- if (wasi_ctx)
- wasm_runtime_set_wasi_ctx(new_module_inst, wasi_ctx);
- #endif
- if (!(info_node = wasm_runtime_malloc(sizeof(ThreadInfoNode))))
- goto fail;
- memset(info_node, 0, sizeof(ThreadInfoNode));
- thread_handle = allocate_handle();
- info_node->parent_exec_env = exec_env;
- info_node->handle = thread_handle;
- info_node->type = T_THREAD;
- info_node->status = THREAD_INIT;
- if (!(routine_args = wasm_runtime_malloc(sizeof(ThreadRoutineArgs))))
- goto fail;
- routine_args->arg = arg;
- routine_args->elem_index = elem_index;
- routine_args->info_node = info_node;
- routine_args->module_inst = new_module_inst;
- os_mutex_lock(&exec_env->wait_lock);
- ret = wasm_cluster_create_thread(exec_env, new_module_inst,
- pthread_start_routine,
- (void *)routine_args);
- if (ret != 0) {
- os_mutex_unlock(&exec_env->wait_lock);
- goto fail;
- }
- /* Wait for the thread routine to assign the exec_env to
- thread_info_node, otherwise the exec_env in the thread
- info node may be NULL in the next pthread API call */
- os_cond_wait(&exec_env->wait_cond, &exec_env->wait_lock);
- os_mutex_unlock(&exec_env->wait_lock);
- if (thread)
- *thread = thread_handle;
- return 0;
- fail:
- if (new_module_inst)
- wasm_runtime_deinstantiate_internal(new_module_inst, true);
- if (info_node)
- wasm_runtime_free(info_node);
- if (routine_args)
- wasm_runtime_free(routine_args);
- return ret;
- }
- static int32
- pthread_join_wrapper(wasm_exec_env_t exec_env, uint32 thread,
- int32 retval_offset) /* void **retval */
- {
- uint32 *ret;
- int32 join_ret;
- void **retval;
- ThreadInfoNode *node;
- wasm_module_inst_t module_inst;
- wasm_exec_env_t target_exec_env;
- node = get_thread_info(exec_env, thread);
- if (!node) {
- /* The thread has exited, return 0 to app */
- return 0;
- }
- target_exec_env = node->exec_env;
- bh_assert(target_exec_env != NULL);
- module_inst = get_module_inst(target_exec_env);
- /* validate addr before join thread, otherwise
- the module_inst may be freed */
- if (!validate_app_addr(retval_offset, sizeof(uint32))) {
- /* Join failed, but we don't want to terminate all threads,
- do not spread exception here */
- wasm_runtime_set_exception(module_inst, NULL);
- return -1;
- }
- retval = (void **)addr_app_to_native(retval_offset);
- join_ret = wasm_cluster_join_thread(target_exec_env, (void **)&ret);
- if (retval_offset != 0)
- *retval = (void*)ret;
- return join_ret;
- }
- static int32
- pthread_detach_wrapper(wasm_exec_env_t exec_env, uint32 thread)
- {
- ThreadInfoNode *node;
- wasm_exec_env_t target_exec_env;
- node = get_thread_info(exec_env, thread);
- if (!node)
- return 0;
- target_exec_env = node->exec_env;
- bh_assert(target_exec_env != NULL);
- return wasm_cluster_detach_thread(target_exec_env);
- }
- static int32
- pthread_cancel_wrapper(wasm_exec_env_t exec_env, uint32 thread)
- {
- ThreadInfoNode *node;
- wasm_exec_env_t target_exec_env;
- node = get_thread_info(exec_env, thread);
- if (!node)
- return 0;
- target_exec_env = node->exec_env;
- bh_assert(target_exec_env != NULL);
- return wasm_cluster_cancel_thread(target_exec_env);
- }
- static int32
- pthread_self_wrapper(wasm_exec_env_t exec_env)
- {
- ThreadRoutineArgs *args = get_thread_arg(exec_env);
- /* If thread_arg is NULL, it's the exec_env of the main thread,
- return id 0 to app */
- if (!args)
- return 0;
- return args->info_node->handle;
- }
- static void
- pthread_exit_wrapper(wasm_exec_env_t exec_env, int32 retval_offset)
- {
- wasm_module_inst_t module_inst = get_module_inst(exec_env);
- ThreadRoutineArgs *args = get_thread_arg(exec_env);
- /* Currently exit main thread is not allowed */
- if (!args)
- return;
- #ifdef OS_ENABLE_HW_BOUND_CHECK
- /* If hardware bound check enabled, don't deinstantiate module inst
- and thread info node here for AoT module, as they will be freed
- in pthread_start_routine */
- if (exec_env->jmpbuf_stack_top) {
- wasm_cluster_exit_thread(exec_env, (void *)(uintptr_t)retval_offset);
- }
- #endif
- /* destroy pthread key values */
- call_key_destructor(exec_env);
- /* routine exit, destroy instance */
- wasm_runtime_deinstantiate_internal(module_inst, true);
- delete_thread_info_node(args->info_node);
- wasm_cluster_exit_thread(exec_env, (void *)(uintptr_t)retval_offset);
- }
- static int32
- pthread_mutex_init_wrapper(wasm_exec_env_t exec_env, uint32 *mutex, void *attr)
- {
- korp_mutex *pmutex;
- ThreadInfoNode *info_node;
- if (!(pmutex = wasm_runtime_malloc(sizeof(korp_mutex)))) {
- return -1;
- }
- if (os_mutex_init(pmutex) != 0) {
- goto fail1;
- }
- if (!(info_node = wasm_runtime_malloc(sizeof(ThreadInfoNode))))
- goto fail2;
- memset(info_node, 0, sizeof(ThreadInfoNode));
- info_node->exec_env = exec_env;
- info_node->handle = allocate_handle();
- info_node->type = T_MUTEX;
- info_node->u.mutex = pmutex;
- info_node->status = MUTEX_CREATED;
- if (!append_thread_info_node(info_node))
- goto fail3;
- /* Return the mutex handle to app */
- if (mutex)
- *(uint32*)mutex = info_node->handle;
- return 0;
- fail3:
- delete_thread_info_node(info_node);
- fail2:
- os_mutex_destroy(pmutex);
- fail1:
- wasm_runtime_free(pmutex);
- return -1;
- }
- static int32
- pthread_mutex_lock_wrapper(wasm_exec_env_t exec_env, uint32 *mutex)
- {
- ThreadInfoNode* info_node = get_thread_info(exec_env, *mutex);
- if (!info_node || info_node->type != T_MUTEX)
- return -1;
- return os_mutex_lock(info_node->u.mutex);
- }
- static int32
- pthread_mutex_unlock_wrapper(wasm_exec_env_t exec_env, uint32 *mutex)
- {
- ThreadInfoNode* info_node = get_thread_info(exec_env, *mutex);
- if (!info_node || info_node->type != T_MUTEX)
- return -1;
- return os_mutex_unlock(info_node->u.mutex);
- }
- static int32
- pthread_mutex_destroy_wrapper(wasm_exec_env_t exec_env, uint32 *mutex)
- {
- int32 ret_val;
- ThreadInfoNode* info_node = get_thread_info(exec_env, *mutex);
- if (!info_node || info_node->type != T_MUTEX)
- return -1;
- ret_val = os_mutex_destroy(info_node->u.mutex);
- info_node->status = MUTEX_DESTROYED;
- delete_thread_info_node(info_node);
- return ret_val;
- }
- static int32
- pthread_cond_init_wrapper(wasm_exec_env_t exec_env, uint32 *cond, void *attr)
- {
- korp_cond *pcond;
- ThreadInfoNode *info_node;
- if (!(pcond = wasm_runtime_malloc(sizeof(korp_cond)))) {
- return -1;
- }
- if (os_cond_init(pcond) != 0) {
- goto fail1;
- }
- if (!(info_node = wasm_runtime_malloc(sizeof(ThreadInfoNode))))
- goto fail2;
- memset(info_node, 0, sizeof(ThreadInfoNode));
- info_node->exec_env = exec_env;
- info_node->handle = allocate_handle();
- info_node->type = T_COND;
- info_node->u.cond = pcond;
- info_node->status = COND_CREATED;
- if (!append_thread_info_node(info_node))
- goto fail3;
- /* Return the cond handle to app */
- if (cond)
- *(uint32*)cond = info_node->handle;
- return 0;
- fail3:
- delete_thread_info_node(info_node);
- fail2:
- os_cond_destroy(pcond);
- fail1:
- wasm_runtime_free(pcond);
- return -1;
- }
- static int32
- pthread_cond_wait_wrapper(wasm_exec_env_t exec_env, uint32 *cond, uint32 *mutex)
- {
- ThreadInfoNode *cond_info_node, *mutex_info_node;
- cond_info_node = get_thread_info(exec_env, *cond);
- if (!cond_info_node || cond_info_node->type != T_COND)
- return -1;
- mutex_info_node = get_thread_info(exec_env, *mutex);
- if (!mutex_info_node || mutex_info_node->type != T_MUTEX)
- return -1;
- return os_cond_wait(cond_info_node->u.cond, mutex_info_node->u.mutex);
- }
- /* Currently we don't support struct timespec in built-in libc,
- so the pthread_cond_timedwait use useconds instead
- */
- static int32
- pthread_cond_timedwait_wrapper(wasm_exec_env_t exec_env, uint32 *cond,
- uint32 *mutex, uint32 useconds)
- {
- ThreadInfoNode *cond_info_node, *mutex_info_node;
- cond_info_node = get_thread_info(exec_env, *cond);
- if (!cond_info_node || cond_info_node->type != T_COND)
- return -1;
- mutex_info_node = get_thread_info(exec_env, *mutex);
- if (!mutex_info_node || mutex_info_node->type != T_MUTEX)
- return -1;
- return os_cond_reltimedwait(cond_info_node->u.cond,
- mutex_info_node->u.mutex, useconds);
- }
- static int32
- pthread_cond_signal_wrapper(wasm_exec_env_t exec_env, uint32 *cond)
- {
- ThreadInfoNode* info_node = get_thread_info(exec_env, *cond);
- if (!info_node || info_node->type != T_COND)
- return -1;
- return os_cond_signal(info_node->u.cond);
- }
- static int32
- pthread_cond_destroy_wrapper(wasm_exec_env_t exec_env, uint32 *cond)
- {
- int32 ret_val;
- ThreadInfoNode* info_node = get_thread_info(exec_env, *cond);
- if (!info_node || info_node->type != T_COND)
- return -1;
- ret_val = os_cond_destroy(info_node->u.cond);
- info_node->status = COND_DESTROYED;
- delete_thread_info_node(info_node);
- return ret_val;
- }
- static int32
- pthread_key_create_wrapper(wasm_exec_env_t exec_env, int32 *key,
- int32 destructor_elem_index)
- {
- uint32 i;
- WASMCluster *cluster = wasm_exec_env_get_cluster(exec_env);
- ClusterInfoNode *info = get_cluster_info(cluster);
- if (!info) {
- /* The user may call pthread_key_create in main thread,
- in this case the cluster info hasn't been created */
- if (!(info = create_cluster_info(cluster))) {
- return -1;
- }
- }
- os_mutex_lock(&info->key_data_list_lock);
- for (i = 0; i < WAMR_PTHREAD_KEYS_MAX; i++) {
- if (!info->key_data_list[i].is_created) {
- break;
- }
- }
- if (i == WAMR_PTHREAD_KEYS_MAX) {
- os_mutex_unlock(&info->key_data_list_lock);
- return -1;
- }
- info->key_data_list[i].destructor_func = destructor_elem_index;
- info->key_data_list[i].is_created = true;
- *key = i;
- os_mutex_unlock(&info->key_data_list_lock);
- return 0;
- }
- static int32
- pthread_setspecific_wrapper(wasm_exec_env_t exec_env, int32 key,
- int32 value_offset)
- {
- WASMCluster *cluster = wasm_exec_env_get_cluster(exec_env);
- ClusterInfoNode *info = get_cluster_info(cluster);
- int32 *key_values;
- if (!info)
- return -1;
- os_mutex_lock(&info->key_data_list_lock);
- key_values = key_value_list_lookup_or_create(exec_env, info, key);
- if (!key_values) {
- os_mutex_unlock(&info->key_data_list_lock);
- return 0;
- }
- key_values[key] = value_offset;
- os_mutex_unlock(&info->key_data_list_lock);
- return 0;
- }
- static int32
- pthread_getspecific_wrapper(wasm_exec_env_t exec_env, int32 key)
- {
- WASMCluster *cluster = wasm_exec_env_get_cluster(exec_env);
- ClusterInfoNode *info = get_cluster_info(cluster);
- int32 ret, *key_values;
- if (!info)
- return -1;
- os_mutex_lock(&info->key_data_list_lock);
- key_values = key_value_list_lookup_or_create(exec_env, info, key);
- if (!key_values) {
- os_mutex_unlock(&info->key_data_list_lock);
- return 0;
- }
- ret = key_values[key];
- os_mutex_unlock(&info->key_data_list_lock);
- return ret;
- }
- static int32
- pthread_key_delete_wrapper(wasm_exec_env_t exec_env, int32 key)
- {
- KeyData *data;
- WASMCluster *cluster = wasm_exec_env_get_cluster(exec_env);
- ClusterInfoNode *info = get_cluster_info(cluster);
- if (!info)
- return -1;
- os_mutex_lock(&info->key_data_list_lock);
- data = key_data_list_lookup(exec_env, key);
- if (!data) {
- os_mutex_unlock(&info->key_data_list_lock);
- return -1;
- }
- memset(data, 0, sizeof(KeyData));
- os_mutex_unlock(&info->key_data_list_lock);
- return 0;
- }
- #define REG_NATIVE_FUNC(func_name, signature) \
- { #func_name, func_name##_wrapper, signature, NULL }
- static NativeSymbol native_symbols_lib_pthread[] = {
- REG_NATIVE_FUNC(pthread_create, "(**i*)i"),
- REG_NATIVE_FUNC(pthread_join, "(ii)i"),
- REG_NATIVE_FUNC(pthread_detach, "(i)i"),
- REG_NATIVE_FUNC(pthread_cancel, "(i)i"),
- REG_NATIVE_FUNC(pthread_self, "()i"),
- REG_NATIVE_FUNC(pthread_exit, "(i)"),
- REG_NATIVE_FUNC(pthread_mutex_init, "(**)i"),
- REG_NATIVE_FUNC(pthread_mutex_lock, "(*)i"),
- REG_NATIVE_FUNC(pthread_mutex_unlock, "(*)i"),
- REG_NATIVE_FUNC(pthread_mutex_destroy, "(*)i"),
- REG_NATIVE_FUNC(pthread_cond_init, "(**)i"),
- REG_NATIVE_FUNC(pthread_cond_wait, "(**)i"),
- REG_NATIVE_FUNC(pthread_cond_timedwait, "(**i)i"),
- REG_NATIVE_FUNC(pthread_cond_signal, "(*)i"),
- REG_NATIVE_FUNC(pthread_cond_destroy, "(*)i"),
- REG_NATIVE_FUNC(pthread_key_create, "(*i)i"),
- REG_NATIVE_FUNC(pthread_setspecific, "(ii)i"),
- REG_NATIVE_FUNC(pthread_getspecific, "(i)i"),
- REG_NATIVE_FUNC(pthread_key_delete, "(i)i"),
- };
- uint32
- get_lib_pthread_export_apis(NativeSymbol **p_lib_pthread_apis)
- {
- *p_lib_pthread_apis = native_symbols_lib_pthread;
- return sizeof(native_symbols_lib_pthread) / sizeof(NativeSymbol);
- }
|