WIP register saves

Author: tannewt

main.c

@@ -30,6 +30,7 @@
 #include "supervisor/cpu.h"
 #include "supervisor/filesystem.h"
 #include "supervisor/port.h"
+#include "supervisor/shared/cpu_regs.h"
 #include "supervisor/shared/reload.h"
 #include "supervisor/shared/safe_mode.h"
 #include "supervisor/shared/serial.h"
@@ -1135,9 +1136,14 @@ int __attribute__((used)) main(void) {
 void gc_collect(void) {
     gc_collect_start();
 
-    mp_uint_t regs[10];
+    // Load register values onto the stack. They get collected below with the rest of the stack.
+    size_t regs[SAVED_REGISTER_COUNT];
     mp_uint_t sp = cpu_get_regs_and_sp(regs);
 
+    // This naively collects all object references from an approximate stack
+    // range.
+    gc_collect_root((void **)sp, ((mp_uint_t)port_stack_get_top() - sp) / sizeof(mp_uint_t));
+
     // This collects root pointers from the VFS mount table. Some of them may
     // have lost their references in the VM even though they are mounted.
     gc_collect_root((void **)&MP_STATE_VM(vfs_mount_table), sizeof(mp_vfs_mount_t) / sizeof(mp_uint_t));
@@ -1170,9 +1176,6 @@ void gc_collect(void) {
     common_hal_wifi_gc_collect();
     #endif
 
-    // This naively collects all object references from an approximate stack
-    // range.
-    gc_collect_root((void **)sp, ((mp_uint_t)port_stack_get_top() - sp) / sizeof(mp_uint_t));
     gc_collect_end();
 }
 

ports/analog/Makefile

@@ -138,7 +138,7 @@ LINKERFILE = linking/$(MCU_VARIANT_LOWER)_cktpy.ld
 LDFLAGS += -nostartfiles -specs=nano.specs
 endif
 
-SRC_S += supervisor/cpu.s \
+SRC_S += $(TOP)/supervisor/shared/cpu_regs.S \
 			$(STARTUPFILE)
 
 # Needed to compile some MAX32 headers

ports/analog/supervisor/cpu.s

@@ -308,7 +308,7 @@ ifeq ($(CIRCUITPY_AUDIOBUSIO),1)
 SRC_C += peripherals/samd/i2s.c peripherals/samd/$(PERIPHERALS_CHIP_FAMILY)/i2s.c
 endif
 
-SRC_S = supervisor/$(CHIP_FAMILY)_cpu.s
+SRC_S = $(TOP)/supervisor/shared/cpu_regs.S
 
 OBJ = $(PY_O) $(SUPERVISOR_O) $(addprefix $(BUILD)/, $(SRC_C:.c=.o))
 OBJ += $(addprefix $(BUILD)/, $(SRC_ASF:.c=.o))

ports/atmel-samd/Makefile

@@ -110,7 +110,7 @@ LDFLAGS = \
 
 CFLAGS += -DCFG_TUSB_MCU=OPT_MCU_CXD56 -DCFG_TUD_MIDI_RX_BUFSIZE=512 -DCFG_TUD_CDC_RX_BUFSIZE=1024 -DCFG_TUD_MIDI_TX_BUFSIZE=512 -DCFG_TUD_CDC_TX_BUFSIZE=1024 -DCFG_TUD_MSC_BUFSIZE=512 $(CFLAGS_MOD)
 
-SRC_S = supervisor/cpu.s
+SRC_S = $(TOP)/supervisor/shared/cpu_regs.S
 
 SRC_C += \
 	background.c \