shared/runtime/pyexec: Call mp_hal_stdio_mode_orig/raw as appropriate.

This ensures that ctrl-C works on the unix port when executing code at the
REPL.

Signed-off-by: Damien George <[email protected]>

Author: dpgeorge

ports/unix/main.c

@@ -191,7 +191,6 @@ static int do_repl(void) {
     int ret = 0;
     #if MICROPY_USE_READLINE == 1
     // use MicroPython supplied readline based repl
-    mp_hal_stdio_mode_raw();
     for (;;) {
         if (pyexec_mode_kind == PYEXEC_MODE_RAW_REPL) {
             if ((ret = pyexec_raw_repl()) != 0) {
@@ -203,7 +202,6 @@ static int do_repl(void) {
             }
         }
     }
-    mp_hal_stdio_mode_orig();
     #else
     // use simple readline
     mp_hal_stdout_tx_str(MICROPY_BANNER_NAME_AND_VERSION);

ports/unix/mphalport.h

@@ -48,6 +48,9 @@
         MP_THREAD_GIL_ENTER(); \
     } while (0)
 
+// The port provides `mp_hal_stdio_mode_raw()` and `mp_hal_stdio_mode_orig()`.
+#define MICROPY_HAL_HAS_STDIO_MODE_SWITCH (1)
+
 void mp_hal_set_interrupt_char(char c);
 
 #define mp_hal_stdio_poll unused // this is not implemented, nor needed

shared/runtime/pyexec.c

@@ -516,10 +516,20 @@ MP_REGISTER_ROOT_POINTER(vstr_t * repl_line);
 
 #else // MICROPY_REPL_EVENT_DRIVEN
 
+#if !MICROPY_HAL_HAS_STDIO_MODE_SWITCH
+// If the port doesn't need any stdio mode switching calls then provide trivial ones.
+static inline void mp_hal_stdio_mode_raw(void) {
+}
+static inline void mp_hal_stdio_mode_orig(void) {
+}
+#endif
+
 int pyexec_raw_repl(void) {
     vstr_t line;
     vstr_init(&line, 32);
 
+    mp_hal_stdio_mode_raw();
+
 raw_repl_reset:
     mp_hal_stdout_tx_str("raw REPL; CTRL-B to exit\r\n");
 
@@ -533,6 +543,7 @@ int pyexec_raw_repl(void) {
                 if (vstr_len(&line) == 2 && vstr_str(&line)[0] == CHAR_CTRL_E) {
                     int ret = do_reader_stdin(vstr_str(&line)[1]);
                     if (ret & PYEXEC_FORCED_EXIT) {
+                        mp_hal_stdio_mode_orig();
                         return ret;
                     }
                     vstr_reset(&line);
@@ -545,6 +556,7 @@ int pyexec_raw_repl(void) {
                 mp_hal_stdout_tx_str("\r\n");
                 vstr_clear(&line);
                 pyexec_mode_kind = PYEXEC_MODE_FRIENDLY_REPL;
+                mp_hal_stdio_mode_orig();
                 return 0;
             } else if (c == CHAR_CTRL_C) {
                 // clear line
@@ -565,10 +577,19 @@ int pyexec_raw_repl(void) {
             // exit for a soft reset
             mp_hal_stdout_tx_str("\r\n");
             vstr_clear(&line);
+            mp_hal_stdio_mode_orig();
             return PYEXEC_FORCED_EXIT;
         }
 
+        // Execute code with terminal mode switching for keyboard interrupt support.
+        // Switches to original mode (ISIG enabled) so Ctrl-C generates SIGINT during execution,
+        // then restores raw mode (ISIG disabled) for REPL input to support paste mode with arbitrary
+        // data including byte 0x03 without triggering interrupts.
+        mp_hal_stdio_mode_orig();
         int ret = parse_compile_execute(&line, MP_PARSE_FILE_INPUT, EXEC_FLAG_PRINT_EOF | EXEC_FLAG_SOURCE_IS_VSTR);
+        if (!(ret & PYEXEC_FORCED_EXIT)) {
+            mp_hal_stdio_mode_raw();
+        }
         if (ret & PYEXEC_FORCED_EXIT) {
             return ret;
         }
@@ -579,6 +600,8 @@ int pyexec_friendly_repl(void) {
     vstr_t line;
     vstr_init(&line, 32);
 
+    mp_hal_stdio_mode_raw();
+
 friendly_repl_reset:
     mp_hal_stdout_tx_str(MICROPY_BANNER_NAME_AND_VERSION);
     mp_hal_stdout_tx_str("; " MICROPY_BANNER_MACHINE);
@@ -634,6 +657,7 @@ int pyexec_friendly_repl(void) {
             mp_hal_stdout_tx_str("\r\n");
             vstr_clear(&line);
             pyexec_mode_kind = PYEXEC_MODE_RAW_REPL;
+            mp_hal_stdio_mode_orig();
             return 0;
         } else if (ret == CHAR_CTRL_B) {
             // reset friendly REPL
@@ -647,6 +671,7 @@ int pyexec_friendly_repl(void) {
             // exit for a soft reset
             mp_hal_stdout_tx_str("\r\n");
             vstr_clear(&line);
+            mp_hal_stdio_mode_orig();
             return PYEXEC_FORCED_EXIT;
         } else if (ret == CHAR_CTRL_E) {
             // paste mode
@@ -691,7 +716,15 @@ int pyexec_friendly_repl(void) {
             }
         }
 
+        // Execute code with terminal mode switching for keyboard interrupt support.
+        // Switches to original mode (ISIG enabled) so Ctrl-C generates SIGINT during execution,
+        // then restores raw mode (ISIG disabled) for REPL input to support paste mode with arbitrary
+        // data including byte 0x03 without triggering interrupts.
+        mp_hal_stdio_mode_orig();
         ret = parse_compile_execute(&line, parse_input_kind, EXEC_FLAG_ALLOW_DEBUGGING | EXEC_FLAG_IS_REPL | EXEC_FLAG_SOURCE_IS_VSTR);
+        if (!(ret & PYEXEC_FORCED_EXIT)) {
+            mp_hal_stdio_mode_raw();
+        }
         if (ret & PYEXEC_FORCED_EXIT) {
             return ret;
         }