From 7ad2e6ace34a6d417ec1b84472fa409d5451137b Mon Sep 17 00:00:00 2001 From: Scott Shawcroft Date: Mon, 22 Oct 2018 17:57:28 -0700 Subject: Add stack validity check and raise an error when it happens. The backtrace cannot be given because it relies on the validity of the qstr data structures on the heap which may have been corrupted. In fact, it still can crash hard when the bytecode itself is overwritten. To fix, we'd need a way to skip gathering the backtrace completely. This also increases the default stack size on M4s so it can accomodate the stack needed by ASF4s nvm API. --- supervisor/shared/stack.c | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) (limited to 'supervisor/shared/stack.c') diff --git a/supervisor/shared/stack.c b/supervisor/shared/stack.c index 6b74742eb..8f4da4ce6 100755 --- a/supervisor/shared/stack.c +++ b/supervisor/shared/stack.c @@ -27,6 +27,7 @@ #include "stack.h" #include "py/mpconfig.h" +#include "py/runtime.h" #include "supervisor/cpu.h" extern uint32_t _estack; @@ -37,6 +38,8 @@ supervisor_allocation* stack_alloc = NULL; #define EXCEPTION_STACK_SIZE 1024 +#define STACK_CANARY_VALUE 0x017829ef + void allocate_stack(void) { mp_uint_t regs[10]; mp_uint_t sp = cpu_get_regs_and_sp(regs); @@ -50,6 +53,19 @@ void allocate_stack(void) { } else { current_stack_size = next_stack_size; } + *stack_alloc->ptr = STACK_CANARY_VALUE; +} + +inline bool stack_ok(void) { + return *stack_alloc->ptr == STACK_CANARY_VALUE; +} + +inline void assert_heap_ok(void) { + if (!stack_ok()) { + asm("nop"); + while(true) {} + mp_raise_RuntimeError(translate("Stack clobbered heap.")); + } } void stack_init(void) { @@ -58,6 +74,7 @@ void stack_init(void) { void stack_resize(void) { if (next_stack_size == current_stack_size) { + *stack_alloc->ptr = STACK_CANARY_VALUE; return; } free_memory(stack_alloc); -- cgit v1.2.3