Add stderr hook

Author: EcmaXp

src/main.c

@@ -26,6 +26,7 @@ void mp_init_port() {
     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;
+    MP_STATE_PORT(stderr_hook_obj) = mp_const_none;
 }
 
 void do_str(const char *src, mp_parse_input_kind_t input_kind) {

src/modmachine.c

@@ -24,6 +24,14 @@ STATIC mp_obj_t machine_hook_signal(mp_obj_t hook_obj) {
 MP_DEFINE_CONST_FUN_OBJ_1(machine_hook_signal_obj, machine_hook_signal);
 
 
+STATIC mp_obj_t machine_hook_stdin(mp_obj_t hook_obj) {
+    MP_STATE_PORT(stdin_hook_obj) = hook_obj;
+    return hook_obj;
+}
+
+MP_DEFINE_CONST_FUN_OBJ_1(machine_hook_stdin_obj, machine_hook_stdin);
+
+
 STATIC mp_obj_t machine_hook_stdout(mp_obj_t hook_obj) {
     MP_STATE_PORT(stdout_hook_obj) = hook_obj;
     return hook_obj;
@@ -32,12 +40,12 @@ STATIC mp_obj_t machine_hook_stdout(mp_obj_t 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) {
-    MP_STATE_PORT(stdin_hook_obj) = hook_obj;
+STATIC mp_obj_t machine_hook_stderr(mp_obj_t hook_obj) {
+    MP_STATE_PORT(stderr_hook_obj) = hook_obj;
     return hook_obj;
 }
 
-MP_DEFINE_CONST_FUN_OBJ_1(machine_hook_stdin_obj, machine_hook_stdin);
+MP_DEFINE_CONST_FUN_OBJ_1(machine_hook_stderr_obj, machine_hook_stderr);
 
 
 STATIC const mp_rom_map_elem_t machine_module_globals_table[] = {
@@ -48,8 +56,9 @@ STATIC const mp_rom_map_elem_t machine_module_globals_table[] = {
 
         // hook
         {MP_ROM_QSTR(MP_QSTR_hook_signal),          MP_ROM_PTR(&machine_hook_signal_obj)},
-        {MP_ROM_QSTR(MP_QSTR_hook_stdout),          MP_ROM_PTR(&machine_hook_stdout_obj)},
         {MP_ROM_QSTR(MP_QSTR_hook_stdin),           MP_ROM_PTR(&machine_hook_stdin_obj)},
+        {MP_ROM_QSTR(MP_QSTR_hook_stdout),          MP_ROM_PTR(&machine_hook_stdout_obj)},
+        {MP_ROM_QSTR(MP_QSTR_hook_stderr),          MP_ROM_PTR(&machine_hook_stderr_obj)},
 };
 
 STATIC MP_DEFINE_CONST_DICT(machine_module_globals, machine_module_globals_table);

src/mpconfigport.h

@@ -236,6 +236,7 @@ extern const struct _mp_obj_type_t mp_type_SystemError;
     mp_obj_t signal_hook_obj; \
     mp_obj_t stdin_hook_obj; \
     mp_obj_t stdout_hook_obj; \
+    mp_obj_t stderr_hook_obj; \
     mp_obj_t object_hook_obj; \
 
 #define mp_type_fileio mp_type_vfs_openpie_fileio

src/mphalport.c

@@ -8,6 +8,16 @@
 #include "mphalport.h"
 #include "syscall.h"
 
+
+void mp_hal_stdio_tx_strn(mp_obj_t hook_obj, const char *str, mp_uint_t len) {
+    if (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(hook_obj, str_obj);
+    }
+}
+
 int mp_hal_stdin_rx_chr(void) {
     if (MP_STATE_PORT(stdin_hook_obj) == mp_const_none) {
         return 0;
@@ -22,12 +32,7 @@ void mp_hal_stdout_tx_str(const char *str) {
 }
 
 void mp_hal_stdout_tx_strn(const char *str, mp_uint_t len) {
-    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(MP_STATE_PORT(stdout_hook_obj), str_obj);
-    }
+    mp_hal_stdio_tx_strn(MP_STATE_PORT(stdout_hook_obj), str, len);
 }
 
 void mp_hal_stdout_tx_strn_cooked(const char *str, size_t len) {

src/openpie_vfs.c

@@ -121,9 +121,24 @@ STATIC mp_uint_t vfs_openpie_file_write(mp_obj_t o_in, const void *buf, mp_uint_
     mp_obj_vfs_openpie_file_t *o = MP_OBJ_TO_PTR(o_in);
     check_fd_is_open(o);
 #if MICROPY_PY_OS_DUPTERM
-    if (o->fd <= STDERR_FILENO && o->flag & 1) {
-        mp_hal_stdout_tx_strn(buf, size);
-        return size;
+    if (o->flag & 1) {
+        switch (o->fd) {
+            case STDIN_FILENO:
+                return 0;
+            case STDOUT_FILENO:
+                // mp_hal_stdout_tx_strn(buf, size);
+                mp_hal_stdio_tx_strn(MP_STATE_PORT(stdout_hook_obj), buf, size);
+                return size;
+            case STDERR_FILENO:
+            {
+                mp_obj_t stdio_hook_obj = MP_STATE_PORT(stderr_hook_obj);
+                if (stdio_hook_obj == NULL)
+                    stdio_hook_obj = MP_STATE_PORT(stdout_hook_obj);
+
+                mp_hal_stdio_tx_strn(stdio_hook_obj, buf, size);
+                return size;
+            }
+        }
     }
 #endif
     int written = 0;

src/umodules/machine.pyi

@@ -1,18 +1,23 @@
 # machine
 "functions related to the hardware"
+from typing import Callable, Tuple
 
 
 def debug(arg: str):
     pass
 
 
-def hook_stdin(func):
+def hook_signal(func: Callable[[Tuple[str, Tuple]], None]):
     pass
 
 
-def hook_stdout(func):
+def hook_stdin(func: Callable[[str], None]):
     pass
 
 
-def hook_signal(func):
+def hook_stdout(func: Callable[[str], None]):
+    pass
+
+
+def hook_stderr(func: Callable[[str], None]):
     pass