wasm-micro-runtime/samples/native-stack-overflow/src/native_impl.c
YAMAMOTO Takashi d6e8d224ce
Add native-stack-overflow sample (#3321)
This is a test code to examine native stack overflow detection logic.

The current output on my environment (macOS amd64):

```shell
====== Interpreter
 stack size   | fail?  | leak?  | exception
---------------------------------------------------------------------------
    0 - 14704 | failed | leaked | Exception: native stack overflow
14704 - 17904 | failed |     ok | Exception: native stack overflow
17904 - 24576 |     ok |     ok |

====== AOT
 stack size   | fail?  | leak?  | exception
---------------------------------------------------------------------------
    0 - 18176 | failed | leaked | Exception: native stack overflow
18176 - 24576 |     ok |     ok |

====== AOT WAMR_DISABLE_HW_BOUND_CHECK=1
 stack size   | fail?  | leak?  | exception
---------------------------------------------------------------------------
    0 -  1968 | failed |     ok | Exception: native stack overflow
 1968 - 24576 |     ok |     ok |
```

This is a preparation to work on relevant issues, including:

https://github.com/bytecodealliance/wasm-micro-runtime/issues/3325
https://github.com/bytecodealliance/wasm-micro-runtime/issues/3320
https://github.com/bytecodealliance/wasm-micro-runtime/issues/3314
https://github.com/bytecodealliance/wasm-micro-runtime/issues/3297
2024-04-19 15:49:09 +08:00

72 lines
1.7 KiB
C

/*
* Copyright (C) 2024 Midokura Japan KK. All rights reserved.
* SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
*/
#include <inttypes.h>
#include <stdio.h>
#include "wasm_export.h"
#include "bh_platform.h"
/*
* this "nest" var has two purposes:
* - prevent tail-call optimization
* - detect possible resource leak
*/
unsigned int nest = 0;
ptrdiff_t prev_diff = 0;
uint32_t
call_indirect(wasm_exec_env_t exec_env, uint32_t funcidx, uint32_t x)
{
uint32_t argv[1] = {
x,
};
uint32_t argc = 1;
if (!wasm_runtime_call_indirect(exec_env, funcidx, argc, argv)) {
/* failed */
return 0;
}
return argv[0];
}
uint32_t
host_consume_stack_and_call_indirect(wasm_exec_env_t exec_env, uint32_t funcidx,
uint32_t x, uint32_t stack)
{
void *boundary = os_thread_get_stack_boundary();
void *fp = __builtin_frame_address(0);
ptrdiff_t diff = fp - boundary;
if (diff > stack) {
prev_diff = diff;
nest++;
uint32_t ret =
host_consume_stack_and_call_indirect(exec_env, funcidx, x, stack);
nest--;
return ret;
}
return call_indirect(exec_env, funcidx, x);
}
static uint32_t
consume_stack1(wasm_exec_env_t exec_env, void *base, uint32_t stack)
{
void *fp = __builtin_frame_address(0);
ptrdiff_t diff = (unsigned char *)base - (unsigned char *)fp;
assert(diff > 0);
char buf[16];
memset_s(buf, sizeof(buf), 0, sizeof(buf));
if (diff > stack) {
return diff;
}
return consume_stack1(exec_env, base, stack);
}
uint32_t
host_consume_stack(wasm_exec_env_t exec_env, uint32_t stack)
{
void *base = __builtin_frame_address(0);
return consume_stack1(exec_env, base, stack);
}