unix/main.c: Integrate atexit to be run at the start of exit process.

Signed-off-by: Andrew Leech <[email protected]>
andrewleech · Nov 24, 2024 · ff45348 · ff45348
1 parent 733b1d4
commit ff45348
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/ports/unix/main.c b/ports/unix/main.c
@@ -45,6 +45,7 @@
 #include "py/gc.h"
 #include "py/objstr.h"
 #include "py/cstack.h"
+#include "py/modatexit.h"
 #include "py/mphal.h"
 #include "py/mpthread.h"
 #include "extmod/misc.h"
@@ -748,6 +749,11 @@ MP_NOINLINE int main_(int argc, char **argv) {
         }
     }
 
+    #if MICROPY_PY_ATEXIT
+    int atexit_code = mp_atexit_execute();
+    ret = (atexit_code != 0) ? atexit_code : ret;
+    #endif
+
     #if MICROPY_PY_SYS_SETTRACE
     MP_STATE_THREAD(prof_trace_callback) = MP_OBJ_NULL;
     #endif