unix: Add support for using the Python stack.

dpgeorge · dpgeorge · commit 357486d9b455 · 2017-12-11T13:49:09.000+11:00
diff --git a/ports/unix/main.c b/ports/unix/main.c
@@ -440,6 +440,11 @@ MP_NOINLINE int main_(int argc, char **argv) {
     gc_init(heap, heap + heap_size);
 #endif
 
+    #if MICROPY_ENABLE_PYSTACK
+    static mp_obj_t pystack[1024];
+    mp_pystack_init(pystack, &pystack[MP_ARRAY_SIZE(pystack)]);
+    #endif
+
     mp_init();
 
     char *home = getenv("HOME");
diff --git a/ports/unix/mpthreadport.c b/ports/unix/mpthreadport.c
@@ -66,6 +66,10 @@ STATIC void mp_thread_gc(int signo, siginfo_t *info, void *context) {
         // that we don't need the extra information, enough is captured by the
         // gc_collect_regs_and_stack function above
         //gc_collect_root((void**)context, sizeof(ucontext_t) / sizeof(uintptr_t));
+        #if MICROPY_ENABLE_PYSTACK
+        void **ptrs = (void**)(void*)MP_STATE_THREAD(pystack_start);
+        gc_collect_root(ptrs, (MP_STATE_THREAD(pystack_cur) - MP_STATE_THREAD(pystack_start)) / sizeof(void*));
+        #endif
         thread_signal_done = 1;
     }
 }

Original file line number	Diff line number	Diff line change
`@@ -66,6 +66,10 @@ STATIC void mp_thread_gc(int signo, siginfo_t info, void context) {`
`66`	`66`	`// that we don't need the extra information, enough is captured by the`
`67`	`67`	`// gc_collect_regs_and_stack function above`
`68`	`68`	`//gc_collect_root((void**)context, sizeof(ucontext_t) / sizeof(uintptr_t));`
	`69`	`+ #if MICROPY_ENABLE_PYSTACK`
	`70`	`+ void ptrs = (void)(void*)MP_STATE_THREAD(pystack_start);`
	`71`	`+ gc_collect_root(ptrs, (MP_STATE_THREAD(pystack_cur) - MP_STATE_THREAD(pystack_start)) / sizeof(void*));`
	`72`	`+ #endif`
`69`	`73`	`thread_signal_done = 1;`
`70`	`74`	`}`
`71`	`75`	`}`