shared/runtime/pyexec: Set PYEXEC_FORCED_EXIT flag for SystemExit.

When MICROPY_PYEXEC_ENABLE_EXIT_CODE_HANDLING is enabled, SystemExit
now sets the PYEXEC_FORCED_EXIT flag in addition to the exit code. This
allows the REPL to properly detect and exit when SystemExit is raised,
while still preserving the exit code in the lower bits.

Enable MICROPY_PYEXEC_ENABLE_EXIT_CODE_HANDLING in Windows port.

Fixes repl_lock.py test which expects REPL to exit on SystemExit.

Signed-off-by: Andrew Leech <[email protected]>

Author: pi-anl

ports/windows/mpconfigport.h

@@ -164,6 +164,9 @@
 // Enable support for compile-only mode.
 #define MICROPY_PYEXEC_COMPILE_ONLY (1)
 
+// Enable handling of sys.exit() exit codes.
+#define MICROPY_PYEXEC_ENABLE_EXIT_CODE_HANDLING (1)
+
 #define MICROPY_ERROR_REPORTING     (MICROPY_ERROR_REPORTING_DETAILED)
 #define MICROPY_ERROR_PRINTER       (&mp_stderr_print)
 #define MICROPY_WARNINGS            (1)

shared/runtime/pyexec.c

@@ -178,6 +178,8 @@ static int parse_compile_execute(const void *source, mp_parse_input_kind_t input
             } else {
                 ret = PYEXEC_NORMAL_EXIT;
             }
+            // Set PYEXEC_FORCED_EXIT flag so REPL knows to exit
+            ret |= PYEXEC_FORCED_EXIT;
             #else
             ret = PYEXEC_FORCED_EXIT;
             #endif