109 lines
3.7 KiB
C
109 lines
3.7 KiB
C
#include "../lib.h"
|
|
#include "../wasm_export.h"
|
|
#include "bh_platform.h"
|
|
|
|
#include "__WASM_ARRAY_FILE__"
|
|
#include <stdio.h>
|
|
|
|
void fail_start_trace(void) { printf("Start Trace Marker.\n"); }
|
|
void fail_stop_trace(void) { printf("Stop Trace Marker.\n"); }
|
|
void fail_marker_positive(void) { printf("Positive Marker.\n"); }
|
|
void fail_marker_negative(void) { printf("Negative Marker.\n"); }
|
|
void fail_marker_detected(void) { printf("Detected Marker.\n"); }
|
|
|
|
void fail_start_trace_wasm(wasm_exec_env_t exec_env) { fail_start_trace(); }
|
|
void fail_stop_trace_wasm(wasm_exec_env_t exec_env) { fail_stop_trace(); }
|
|
void fail_marker_positive_wasm(wasm_exec_env_t exec_env) {
|
|
fail_marker_positive();
|
|
}
|
|
void fail_marker_negative_wasm(wasm_exec_env_t exec_env) {
|
|
fail_marker_negative();
|
|
}
|
|
void fail_marker_detected_wasm(wasm_exec_env_t exec_env) {
|
|
fail_marker_detected();
|
|
}
|
|
|
|
#define STACK_SIZE (4 * 1024)
|
|
#define HEAP_SIZE STACK_SIZE
|
|
#define RUNTIME_POOL_SIZE 4 * STACK_SIZE
|
|
|
|
int main(int argc, char *argv[]) {
|
|
char error_buf[128];
|
|
|
|
/* initialize the wasm runtime */
|
|
static char global_heap_buf[RUNTIME_POOL_SIZE];
|
|
static RuntimeInitArgs init_args;
|
|
memset(&init_args, 0, sizeof(RuntimeInitArgs));
|
|
|
|
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);
|
|
init_args.max_thread_num = 1;
|
|
if (!wasm_runtime_full_init(&init_args)) {
|
|
return 1;
|
|
}
|
|
|
|
// Export fail symbols as native functions, so they can be "called" from wasm
|
|
static NativeSymbol native_symbols[] = {
|
|
{"fail_start_trace", fail_start_trace_wasm, "()"},
|
|
{"fail_stop_trace", fail_stop_trace_wasm, "()"},
|
|
{"fail_marker_positive", fail_marker_positive_wasm, "()"},
|
|
{"fail_marker_negative", fail_marker_negative_wasm, "()"},
|
|
{"fail_marker_detected", fail_marker_detected_wasm, "()"},
|
|
};
|
|
int n_native_symbols = sizeof(native_symbols) / sizeof(NativeSymbol);
|
|
if (!wasm_runtime_register_natives("env", native_symbols, n_native_symbols)) {
|
|
printf("Failed to register native symbols.\n");
|
|
return 1;
|
|
}
|
|
|
|
/* parse the WASM file from buffer and create a WASM module */
|
|
wasm_module_t module = wasm_runtime_load(__WASM_ARRAY__, __WASM_ARRAY_LEN__,
|
|
error_buf, sizeof(error_buf));
|
|
if (!module) {
|
|
printf("Failed to parse module from WASM file: \"%s\"\n", error_buf);
|
|
return 1;
|
|
}
|
|
|
|
/* create an instance of the WASM module (WASM linear memory is ready) */
|
|
wasm_module_inst_t module_inst = wasm_runtime_instantiate(
|
|
module, STACK_SIZE, HEAP_SIZE, error_buf, sizeof(error_buf));
|
|
if (!module_inst) {
|
|
printf("Failed to instantiate WASM module: \"%s\"\n", error_buf);
|
|
return 1;
|
|
}
|
|
|
|
// Single exec_env for all functions (probably'd need a single one per
|
|
// thread?)
|
|
wasm_exec_env_t exec_env =
|
|
wasm_runtime_create_exec_env(module_inst, STACK_SIZE);
|
|
if (!exec_env) {
|
|
printf("Failed to create WASM execution environment.\n");
|
|
return 1;
|
|
}
|
|
|
|
// Call wasm function
|
|
uint32_t args[1] = {0};
|
|
wasm_function_inst_t func =
|
|
wasm_runtime_lookup_function(module_inst, "wasm_module");
|
|
if (!func) {
|
|
printf("Failed to obtain WASM function instance.\n");
|
|
return 1;
|
|
}
|
|
printf("Calling WASM function.\n");
|
|
if (!wasm_runtime_call_wasm(exec_env, func, 0, args)) {
|
|
printf("Failed to call WASM function.\n");
|
|
return 1;
|
|
}
|
|
|
|
// In case wasm_module returns a value
|
|
uint32_t retval = args[0];
|
|
printf("WASM function returned %d.\n", retval);
|
|
|
|
printf("Destroying WAMR objects.\n");
|
|
wasm_runtime_destroy_exec_env(exec_env);
|
|
wasm_runtime_deinstantiate(module_inst);
|
|
wasm_runtime_unload(module);
|
|
wasm_runtime_destroy();
|
|
}
|