xxx_hook_obj 가 gc 되는 버그 고침

Author: EcmaXp

src/main.c

@@ -21,6 +21,13 @@ void debug_printer(void *self, const char *buf, size_t len) {
 
 const struct _mp_print_t debug_print = {NULL, debug_printer};
 
+void mp_init_port() {
+    MP_STATE_PORT(object_hook_obj) = mp_const_none;
+    MP_STATE_PORT(signal_hook_obj) = mp_const_none;
+    MP_STATE_PORT(stdin_hook_obj) = mp_const_none;
+    MP_STATE_PORT(stdout_hook_obj) = mp_const_none;
+}
+
 void do_str(const char *src, mp_parse_input_kind_t input_kind) {
     nlr_buf_t nlr;
     if (nlr_push(&nlr) == 0) {
@@ -61,6 +68,7 @@ int main(int argc, char **argv) {
         gc_init(&_ram_start, ((void *)&_ram_start) + ram_size);
 
         mp_init();
+        mp_init_port();
         mp_obj_list_init(mp_sys_path, 0);
         mp_obj_list_append(mp_sys_path, MP_OBJ_NEW_QSTR(MP_QSTR_));
         mp_obj_list_init(mp_sys_argv, 0);

src/modmachine.c

@@ -4,10 +4,6 @@
 #include "extmod/machine_mem.h"
 #include "syscall.h"
 
-mp_obj_t signal_hook_obj = mp_const_none;
-mp_obj_t stdout_hook_obj = mp_const_none;
-mp_obj_t stdin_hook_obj = mp_const_none;
-
 
 mp_obj_t machine_debug(mp_obj_t obj) {
     size_t len = 0;
@@ -21,23 +17,23 @@ MP_DEFINE_CONST_FUN_OBJ_1(machine_debug_obj, machine_debug);
 
 
 STATIC mp_obj_t machine_hook_signal(mp_obj_t hook_obj) {
-    signal_hook_obj = hook_obj;
+    MP_STATE_PORT(signal_hook_obj) = hook_obj;
     return hook_obj;
 }
 
 MP_DEFINE_CONST_FUN_OBJ_1(machine_hook_signal_obj, machine_hook_signal);
 
 
 STATIC mp_obj_t machine_hook_stdout(mp_obj_t hook_obj) {
-    stdout_hook_obj = hook_obj;
+    MP_STATE_PORT(stdout_hook_obj) = hook_obj;
     return hook_obj;
 }
 
 MP_DEFINE_CONST_FUN_OBJ_1(machine_hook_stdout_obj, machine_hook_stdout);
 
 
 STATIC mp_obj_t machine_hook_stdin(mp_obj_t hook_obj) {
-    stdin_hook_obj = hook_obj;
+    MP_STATE_PORT(stdin_hook_obj) = hook_obj;
     return hook_obj;
 }
 

src/moduvalue.c

@@ -8,9 +8,6 @@
 mp_obj_t wrap_result(int code);
 
 
-mp_obj_t object_hook_obj = mp_const_none;
-
-
 STATIC mp_obj_t uvalue_dispose(mp_obj_t raw_value_obj) {
     uvalue_obj_t *self = MP_OBJ_TO_PTR(raw_value_obj);
     if (self->value != mp_const_none) {
@@ -66,16 +63,16 @@ mp_obj_t uvalue_new(mp_obj_t value) {
 
 mp_obj_t oc_create_value(mp_obj_t raw_value_obj) {
     mp_obj_t value_obj = uvalue_new(raw_value_obj);
-    if (object_hook_obj == mp_const_none) {
+    if (MP_STATE_PORT(object_hook_obj) == mp_const_none) {
         return value_obj;
     }
 
-    return mp_call_function_1(object_hook_obj, value_obj);
+    return mp_call_function_1(MP_STATE_PORT(object_hook_obj), value_obj);
 }
 
 
 STATIC mp_obj_t uvalue_hook_value(mp_obj_t hook_obj) {
-    object_hook_obj = hook_obj;
+    MP_STATE_PORT(object_hook_obj) = hook_obj;
     return hook_obj;
 }
 

src/mpconfigport.h

@@ -232,7 +232,11 @@ extern const struct _mp_obj_type_t mp_type_SystemError;
 #define MICROPY_PORT_CONSTANTS
 
 #define MICROPY_PORT_ROOT_POINTERS \
-    const char *readline_hist[8];
+    const char *readline_hist[8]; \
+    mp_obj_t signal_hook_obj; \
+    mp_obj_t stdin_hook_obj; \
+    mp_obj_t stdout_hook_obj; \
+    mp_obj_t object_hook_obj; \
 
 #define mp_type_fileio mp_type_vfs_openpie_fileio
 #define mp_type_textio mp_type_vfs_openpie_textio

src/mphalport.c

@@ -4,18 +4,15 @@
 
 #include "py/runtime.h"
 #include "py/objstr.h"
+#include "machine.h"
 #include "mphalport.h"
 #include "syscall.h"
 
-extern mp_obj_t signal_hook_obj;
-extern mp_obj_t stdin_hook_obj;
-extern mp_obj_t stdout_hook_obj;
-
 int mp_hal_stdin_rx_chr(void) {
-    if (stdin_hook_obj == mp_const_none) {
+    if (MP_STATE_PORT(stdin_hook_obj) == mp_const_none) {
         return 0;
     } else {
-        mp_obj_t obj = mp_call_function_0(stdin_hook_obj);
+        mp_obj_t obj = mp_call_function_0(MP_STATE_PORT(stdin_hook_obj));
         return (int) mp_obj_get_int(obj);
     }
 }
@@ -25,11 +22,11 @@ void mp_hal_stdout_tx_str(const char *str) {
 }
 
 void mp_hal_stdout_tx_strn(const char *str, mp_uint_t len) {
-    if (stdout_hook_obj == mp_const_none) {
+    if (MP_STATE_PORT(stdout_hook_obj) == mp_const_none) {
         __syscall2(SYS_DEBUG, (int)str, (int)len);
     } else {
         mp_obj_t str_obj = mp_obj_new_str(str, len);
-        mp_call_function_1(stdout_hook_obj, str_obj);
+        mp_call_function_1(MP_STATE_PORT(stdout_hook_obj), str_obj);
     }
 }
 
@@ -44,7 +41,9 @@ void mp_hal_delay_ms(mp_uint_t ms) {
         mp_handle_pending();
         int32_t ticks = (ms - elapsed) / (1000 / 20);
         mp_obj_t tick_obj = mp_obj_new_int(ticks < 0? 0 : ticks);
-        mp_call_function_1(signal_hook_obj, tick_obj);
+        if (MP_STATE_PORT(signal_hook_obj) == mp_const_none)
+            __fatal_error("signal_hook is None");
+        mp_call_function_1(MP_STATE_PORT(signal_hook_obj), tick_obj);
         elapsed = mp_hal_ticks_ms() - start;
     }
 }