diff options
Diffstat (limited to 'fluent-bit/lib/wasm-micro-runtime-WAMR-1.2.2/product-mini/platforms/windows/main.c')
-rw-r--r-- | fluent-bit/lib/wasm-micro-runtime-WAMR-1.2.2/product-mini/platforms/windows/main.c | 583 |
1 files changed, 0 insertions, 583 deletions
diff --git a/fluent-bit/lib/wasm-micro-runtime-WAMR-1.2.2/product-mini/platforms/windows/main.c b/fluent-bit/lib/wasm-micro-runtime-WAMR-1.2.2/product-mini/platforms/windows/main.c deleted file mode 100644 index 26fa7dcc9..000000000 --- a/fluent-bit/lib/wasm-micro-runtime-WAMR-1.2.2/product-mini/platforms/windows/main.c +++ /dev/null @@ -1,583 +0,0 @@ -/* - * Copyright (C) 2019 Intel Corporation. All rights reserved. - * SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception - */ - -#include <stdlib.h> -#include <string.h> - -#include "bh_platform.h" -#include "bh_read_file.h" -#include "wasm_export.h" - -static int app_argc; -static char **app_argv; - -#define MODULE_PATH ("--module-path=") - -/* clang-format off */ -static int -print_help() -{ - printf("Usage: iwasm [-options] wasm_file [args...]\n"); - printf("options:\n"); - printf(" -f|--function name Specify a function name of the module to run rather\n" - " than main\n"); -#if WASM_ENABLE_LOG != 0 - printf(" -v=n Set log verbose level (0 to 5, default is 2) larger\n" - " level with more log\n"); -#endif -#if WASM_ENABLE_INTERP != 0 - printf(" --interp Run the wasm app with interpreter mode\n"); -#endif -#if WASM_ENABLE_FAST_JIT != 0 - printf(" --fast-jit Run the wasm app with fast jit mode\n"); -#endif -#if WASM_ENABLE_JIT != 0 - printf(" --llvm-jit Run the wasm app with llvm jit mode\n"); -#endif -#if WASM_ENABLE_JIT != 0 && WASM_ENABLE_FAST_JIT != 0 && WASM_ENABLE_LAZY_JIT != 0 - printf(" --multi-tier-jit Run the wasm app with multi-tier jit mode\n"); -#endif - printf(" --stack-size=n Set maximum stack size in bytes, default is 64 KB\n"); - printf(" --heap-size=n Set maximum heap size in bytes, default is 16 KB\n"); -#if WASM_ENABLE_JIT != 0 - printf(" --llvm-jit-size-level=n Set LLVM JIT size level, default is 3\n"); - printf(" --llvm-jit-opt-level=n Set LLVM JIT optimization level, default is 3\n"); -#endif - printf(" --repl Start a very simple REPL (read-eval-print-loop) mode\n" - " that runs commands in the form of `FUNC ARG...`\n"); -#if WASM_ENABLE_LIBC_WASI != 0 - printf(" --env=<env> Pass wasi environment variables with \"key=value\"\n"); - printf(" to the program, for example:\n"); - printf(" --env=\"key1=value1\" --env=\"key2=value2\"\n"); - printf(" --dir=<dir> Grant wasi access to the given host directories\n"); - printf(" to the program, for example:\n"); - printf(" --dir=<dir1> --dir=<dir2>\n"); -#endif -#if WASM_ENABLE_MULTI_MODULE != 0 - printf(" --module-path=<path> Indicate a module search path. default is current\n" - " directory('./')\n"); -#endif -#if WASM_ENABLE_LIB_PTHREAD != 0 || WASM_ENABLE_LIB_WASI_THREADS != 0 - printf(" --max-threads=n Set maximum thread number per cluster, default is 4\n"); -#endif -#if WASM_ENABLE_DEBUG_INTERP != 0 - printf(" -g=ip:port Set the debug sever address, default is debug disabled\n"); - printf(" if port is 0, then a random port will be used\n"); -#endif - printf(" --version Show version information\n"); - return 1; -} -/* clang-format on */ - -static const void * -app_instance_main(wasm_module_inst_t module_inst) -{ - const char *exception; - - wasm_application_execute_main(module_inst, app_argc, app_argv); - if ((exception = wasm_runtime_get_exception(module_inst))) - printf("%s\n", exception); - return exception; -} - -static const void * -app_instance_func(wasm_module_inst_t module_inst, const char *func_name) -{ - wasm_application_execute_func(module_inst, func_name, app_argc - 1, - app_argv + 1); - /* The result of wasm function or exception info was output inside - wasm_application_execute_func(), here we don't output them again. */ - return wasm_runtime_get_exception(module_inst); -} - -/** - * Split a space separated strings into an array of strings - * Returns NULL on failure - * Memory must be freed by caller - * Based on: http://stackoverflow.com/a/11198630/471795 - */ -static char ** -split_string(char *str, int *count) -{ - char **res = NULL, **res1; - char *p, *next_token; - int idx = 0; - - /* split string and append tokens to 'res' */ - do { - p = strtok_s(str, " ", &next_token); - str = NULL; - res1 = res; - res = (char **)realloc(res1, sizeof(char *) * (uint32)(idx + 1)); - if (res == NULL) { - free(res1); - return NULL; - } - res[idx++] = p; - } while (p); - - /** - * Due to the function name, - * res[0] might contain a '\' to indicate a space - * func\name -> func name - */ - p = strchr(res[0], '\\'); - while (p) { - *p = ' '; - p = strchr(p, '\\'); - } - - if (count) { - *count = idx - 1; - } - return res; -} - -static void * -app_instance_repl(wasm_module_inst_t module_inst) -{ - char buffer[4096]; - char *cmd; - size_t n; - - while ((printf("webassembly> "), cmd = fgets(buffer, sizeof(buffer), stdin)) - != NULL) { - bh_assert(cmd); - n = strlen(cmd); - if (cmd[n - 1] == '\n') { - if (n == 1) - continue; - else - cmd[n - 1] = '\0'; - } - if (!strcmp(cmd, "__exit__")) { - printf("exit repl mode\n"); - break; - } - app_argv = split_string(cmd, &app_argc); - if (app_argv == NULL) { - LOG_ERROR("Wasm prepare param failed: split string failed.\n"); - break; - } - if (app_argc != 0) { - wasm_application_execute_func(module_inst, app_argv[0], - app_argc - 1, app_argv + 1); - } - free(app_argv); - } - - return NULL; -} - -#if WASM_ENABLE_LIBC_WASI != 0 -static bool -validate_env_str(char *env) -{ - char *p = env; - int key_len = 0; - - while (*p != '\0' && *p != '=') { - key_len++; - p++; - } - - if (*p != '=' || key_len == 0) - return false; - - return true; -} -#endif - -#if WASM_ENABLE_GLOBAL_HEAP_POOL != 0 -static char global_heap_buf[WASM_GLOBAL_HEAP_SIZE] = { 0 }; -#endif - -#if WASM_ENABLE_MULTI_MODULE != 0 -static char * -handle_module_path(const char *module_path) -{ - /* next character after '=' */ - return (strchr(module_path, '=')) + 1; -} - -static char *module_search_path = "."; -static bool -module_reader_callback(const char *module_name, uint8 **p_buffer, - uint32 *p_size) -{ - const char *format = "%s/%s.wasm"; - uint32 sz = (uint32)(strlen(module_search_path) + strlen("/") - + strlen(module_name) + strlen(".wasm") + 1); - char *wasm_file_name = BH_MALLOC(sz); - if (!wasm_file_name) { - return false; - } - - snprintf(wasm_file_name, sz, format, module_search_path, module_name); - - *p_buffer = (uint8 *)bh_read_file_to_buffer(wasm_file_name, p_size); - - wasm_runtime_free(wasm_file_name); - return *p_buffer != NULL; -} - -static void -moudle_destroyer(uint8 *buffer, uint32 size) -{ - if (!buffer) { - return; - } - - wasm_runtime_free(buffer); - buffer = NULL; -} -#endif /* WASM_ENABLE_MULTI_MODULE */ - -int -main(int argc, char *argv[]) -{ - int32 ret = -1; - char *wasm_file = NULL; - const char *func_name = NULL; - uint8 *wasm_file_buf = NULL; - uint32 wasm_file_size; - uint32 stack_size = 64 * 1024, heap_size = 16 * 1024; -#if WASM_ENABLE_JIT != 0 - uint32 llvm_jit_size_level = 3; - uint32 llvm_jit_opt_level = 3; -#endif - wasm_module_t wasm_module = NULL; - wasm_module_inst_t wasm_module_inst = NULL; - RunningMode running_mode = 0; - RuntimeInitArgs init_args; - char error_buf[128] = { 0 }; -#if WASM_ENABLE_LOG != 0 - int log_verbose_level = 2; -#endif - bool is_repl_mode = false; - bool is_xip_file = false; -#if WASM_ENABLE_LIBC_WASI != 0 - const char *dir_list[8] = { NULL }; - uint32 dir_list_size = 0; - const char *env_list[8] = { NULL }; - uint32 env_list_size = 0; -#endif -#if WASM_ENABLE_DEBUG_INTERP != 0 - char *ip_addr = NULL; - int instance_port = 0; -#endif - - /* Process options. */ - for (argc--, argv++; argc > 0 && argv[0][0] == '-'; argc--, argv++) { - if (!strcmp(argv[0], "-f") || !strcmp(argv[0], "--function")) { - argc--, argv++; - if (argc < 2) { - return print_help(); - } - func_name = argv[0]; - } -#if WASM_ENABLE_INTERP != 0 - else if (!strcmp(argv[0], "--interp")) { - running_mode = Mode_Interp; - } -#endif -#if WASM_ENABLE_FAST_JIT != 0 - else if (!strcmp(argv[0], "--fast-jit")) { - running_mode = Mode_Fast_JIT; - } -#endif -#if WASM_ENABLE_JIT != 0 - else if (!strcmp(argv[0], "--llvm-jit")) { - running_mode = Mode_LLVM_JIT; - } -#endif -#if WASM_ENABLE_JIT != 0 && WASM_ENABLE_FAST_JIT != 0 - else if (!strcmp(argv[0], "--multi-tier-jit")) { - running_mode = Mode_Multi_Tier_JIT; - } -#endif -#if WASM_ENABLE_LOG != 0 - else if (!strncmp(argv[0], "-v=", 3)) { - log_verbose_level = atoi(argv[0] + 3); - if (log_verbose_level < 0 || log_verbose_level > 5) - return print_help(); - } -#endif - else if (!strcmp(argv[0], "--repl")) { - is_repl_mode = true; - } - else if (!strncmp(argv[0], "--stack-size=", 13)) { - if (argv[0][13] == '\0') - return print_help(); - stack_size = atoi(argv[0] + 13); - } - else if (!strncmp(argv[0], "--heap-size=", 12)) { - if (argv[0][12] == '\0') - return print_help(); - heap_size = atoi(argv[0] + 12); - } -#if WASM_ENABLE_JIT != 0 - else if (!strncmp(argv[0], "--llvm-jit-size-level=", 22)) { - if (argv[0][22] == '\0') - return print_help(); - llvm_jit_size_level = atoi(argv[0] + 22); - if (llvm_jit_size_level < 1) { - printf("LLVM JIT size level shouldn't be smaller than 1, " - "setting it to 1\n"); - llvm_jit_size_level = 1; - } - else if (llvm_jit_size_level > 3) { - printf("LLVM JIT size level shouldn't be greater than 3, " - "setting it to 3\n"); - llvm_jit_size_level = 3; - } - } - else if (!strncmp(argv[0], "--llvm-jit-opt-level=", 21)) { - if (argv[0][21] == '\0') - return print_help(); - llvm_jit_opt_level = atoi(argv[0] + 21); - if (llvm_jit_opt_level < 1) { - printf("LLVM JIT opt level shouldn't be smaller than 1, " - "setting it to 1\n"); - llvm_jit_opt_level = 1; - } - else if (llvm_jit_opt_level > 3) { - printf("LLVM JIT opt level shouldn't be greater than 3, " - "setting it to 3\n"); - llvm_jit_opt_level = 3; - } - } -#endif -#if WASM_ENABLE_LIBC_WASI != 0 - else if (!strncmp(argv[0], "--dir=", 6)) { - if (argv[0][6] == '\0') - return print_help(); - if (dir_list_size >= sizeof(dir_list) / sizeof(char *)) { - printf("Only allow max dir number %d\n", - (int)(sizeof(dir_list) / sizeof(char *))); - return 1; - } - dir_list[dir_list_size++] = argv[0] + 6; - } - else if (!strncmp(argv[0], "--env=", 6)) { - char *tmp_env; - - if (argv[0][6] == '\0') - return print_help(); - if (env_list_size >= sizeof(env_list) / sizeof(char *)) { - printf("Only allow max env number %d\n", - (int)(sizeof(env_list) / sizeof(char *))); - return 1; - } - tmp_env = argv[0] + 6; - if (validate_env_str(tmp_env)) - env_list[env_list_size++] = tmp_env; - else { - printf("Wasm parse env string failed: expect \"key=value\", " - "got \"%s\"\n", - tmp_env); - return print_help(); - } - } -#endif /* WASM_ENABLE_LIBC_WASI */ -#if WASM_ENABLE_MULTI_MODULE != 0 - else if (!strncmp(argv[0], MODULE_PATH, strlen(MODULE_PATH))) { - module_search_path = handle_module_path(argv[0]); - if (!strlen(module_search_path)) { - return print_help(); - } - } -#endif -#if WASM_ENABLE_LIB_PTHREAD != 0 || WASM_ENABLE_LIB_WASI_THREADS != 0 - else if (!strncmp(argv[0], "--max-threads=", 14)) { - if (argv[0][14] == '\0') - return print_help(); - wasm_runtime_set_max_thread_num(atoi(argv[0] + 14)); - } -#endif -#if WASM_ENABLE_DEBUG_INTERP != 0 - else if (!strncmp(argv[0], "-g=", 3)) { - char *port_str = strchr(argv[0] + 3, ':'); - char *port_end; - if (port_str == NULL) - return print_help(); - *port_str = '\0'; - instance_port = strtoul(port_str + 1, &port_end, 10); - if (port_str[1] == '\0' || *port_end != '\0') - return print_help(); - ip_addr = argv[0] + 3; - } -#endif - else if (!strncmp(argv[0], "--version", 9)) { - uint32 major, minor, patch; - wasm_runtime_get_version(&major, &minor, &patch); - printf("iwasm %" PRIu32 ".%" PRIu32 ".%" PRIu32 "\n", major, minor, - patch); - return 0; - } - else - return print_help(); - } - - if (argc == 0) - return print_help(); - - wasm_file = argv[0]; - app_argc = argc; - app_argv = argv; - - memset(&init_args, 0, sizeof(RuntimeInitArgs)); - - init_args.running_mode = running_mode; -#if WASM_ENABLE_GLOBAL_HEAP_POOL != 0 - init_args.mem_alloc_type = Alloc_With_Pool; - init_args.mem_alloc_option.pool.heap_buf = global_heap_buf; - init_args.mem_alloc_option.pool.heap_size = sizeof(global_heap_buf); -#else - init_args.mem_alloc_type = Alloc_With_Allocator; - init_args.mem_alloc_option.allocator.malloc_func = malloc; - init_args.mem_alloc_option.allocator.realloc_func = realloc; - init_args.mem_alloc_option.allocator.free_func = free; -#endif - -#if WASM_ENABLE_JIT != 0 - init_args.llvm_jit_size_level = llvm_jit_size_level; - init_args.llvm_jit_opt_level = llvm_jit_opt_level; -#endif - -#if WASM_ENABLE_DEBUG_INTERP != 0 - init_args.instance_port = instance_port; - if (ip_addr) - strcpy(init_args.ip_addr, ip_addr); -#endif - - /* initialize runtime environment */ - if (!wasm_runtime_full_init(&init_args)) { - printf("Init runtime environment failed.\n"); - return -1; - } - -#if WASM_ENABLE_LOG != 0 - bh_log_set_verbose_level(log_verbose_level); -#endif - - /* load WASM byte buffer from WASM bin file */ - if (!(wasm_file_buf = - (uint8 *)bh_read_file_to_buffer(wasm_file, &wasm_file_size))) - goto fail1; - -#if WASM_ENABLE_AOT != 0 - if (wasm_runtime_is_xip_file(wasm_file_buf, wasm_file_size)) { - uint8 *wasm_file_mapped; - int map_prot = MMAP_PROT_READ | MMAP_PROT_WRITE | MMAP_PROT_EXEC; - int map_flags = MMAP_MAP_32BIT; - - if (!(wasm_file_mapped = - os_mmap(NULL, (uint32)wasm_file_size, map_prot, map_flags))) { - printf("mmap memory failed\n"); - wasm_runtime_free(wasm_file_buf); - goto fail1; - } - - bh_memcpy_s(wasm_file_mapped, wasm_file_size, wasm_file_buf, - wasm_file_size); - wasm_runtime_free(wasm_file_buf); - wasm_file_buf = wasm_file_mapped; - is_xip_file = true; - } -#endif - -#if WASM_ENABLE_MULTI_MODULE != 0 - wasm_runtime_set_module_reader(module_reader_callback, moudle_destroyer); -#endif - - /* load WASM module */ - if (!(wasm_module = wasm_runtime_load(wasm_file_buf, wasm_file_size, - error_buf, sizeof(error_buf)))) { - printf("%s\n", error_buf); - goto fail2; - } - -#if WASM_ENABLE_LIBC_WASI != 0 - wasm_runtime_set_wasi_args(wasm_module, dir_list, dir_list_size, NULL, 0, - env_list, env_list_size, argv, argc); -#endif - - /* instantiate the module */ - if (!(wasm_module_inst = - wasm_runtime_instantiate(wasm_module, stack_size, heap_size, - error_buf, sizeof(error_buf)))) { - printf("%s\n", error_buf); - goto fail3; - } - -#if WASM_ENABLE_DEBUG_INTERP != 0 - if (ip_addr != NULL) { - wasm_exec_env_t exec_env = - wasm_runtime_get_exec_env_singleton(wasm_module_inst); - uint32_t debug_port; - if (exec_env == NULL) { - printf("%s\n", wasm_runtime_get_exception(wasm_module_inst)); - goto fail4; - } - debug_port = wasm_runtime_start_debug_instance(exec_env); - if (debug_port == 0) { - printf("Failed to start debug instance\n"); - goto fail4; - } - } -#endif - - ret = 0; - if (is_repl_mode) { - app_instance_repl(wasm_module_inst); - } - else if (func_name) { - if (app_instance_func(wasm_module_inst, func_name)) { - /* got an exception */ - ret = 1; - } - } - else { - if (app_instance_main(wasm_module_inst)) { - /* got an exception */ - ret = 1; - } - } - -#if WASM_ENABLE_LIBC_WASI != 0 - if (ret == 0) { - /* wait for threads to finish and propagate wasi exit code. */ - ret = wasm_runtime_get_wasi_exit_code(wasm_module_inst); - if (wasm_runtime_get_exception(wasm_module_inst)) { - /* got an exception in spawned thread */ - ret = 1; - } - } -#endif - -#if WASM_ENABLE_DEBUG_INTERP != 0 -fail4: -#endif - /* destroy the module instance */ - wasm_runtime_deinstantiate(wasm_module_inst); - -fail3: - /* unload the module */ - wasm_runtime_unload(wasm_module); - -fail2: - /* free the file buffer */ - if (!is_xip_file) - wasm_runtime_free(wasm_file_buf); - else - os_munmap(wasm_file_buf, wasm_file_size); - -fail1: - /* destroy runtime environment */ - wasm_runtime_destroy(); - - return ret; -} |