feat(core): dump GC arena on OOM

Enabled for debug firmware and non-frozen emulator.

[no changelog]

Author: romanz

core/SConscript.firmware

@@ -396,7 +396,7 @@ ui.init_ui(TREZOR_MODEL, "firmware", RUST_UI_FEATURES)
 SOURCE_QSTR = SOURCE_MOD + SOURCE_MICROPYTHON + SOURCE_MICROPYTHON_SPEED
 
 if PYOPT == '0':
-    DEBUG_FLAGS = "-DMICROPY_OOM_CALLBACK=1"
+    DEBUG_FLAGS = "-DMICROPY_OOM_CALLBACK=1 -DSTATIC="
 else:
     DEBUG_FLAGS = "-DMICROPY_OOM_CALLBACK=0"
 

core/embed/rust/librust.h

@@ -2,7 +2,7 @@
 
 #include "librust_qstr.h"
 
-#ifdef TREZOR_EMULATOR
+#if !PYOPT
 mp_obj_t protobuf_debug_msg_type();
 mp_obj_t protobuf_debug_msg_def_type();
 #endif
@@ -11,6 +11,6 @@ extern mp_obj_module_t mp_module_trezorproto;
 extern mp_obj_module_t mp_module_trezorui_api;
 extern mp_obj_module_t mp_module_trezortranslate;
 
-#ifdef TREZOR_EMULATOR
+#if !PYOPT
 mp_obj_t ui_debug_layout_type();
 #endif

core/embed/upymod/modtrezorutils/modtrezorutils-meminfo.h

@@ -17,7 +17,7 @@
  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  */
 
-#if !TREZOR_EMULATOR || PYOPT
+#if PYOPT
 #define MEMINFO_DICT_ENTRIES /* empty */
 
 #else
@@ -38,6 +38,14 @@
 #include "embed/rust/librust.h"
 #include "embed/upymod/trezorobj.h"
 
+#if !TREZOR_EMULATOR
+#define fopen(path, mode) &mp_plat_print
+#define fprintf mp_printf
+#define fflush(f)
+#define fclose(f)
+#define FILE const mp_print_t
+#endif
+
 #define WORDS_PER_BLOCK ((MICROPY_BYTES_PER_GC_BLOCK) / MP_BYTES_PER_OBJ_WORD)
 #define BYTES_PER_BLOCK (MICROPY_BYTES_PER_GC_BLOCK)
 
@@ -199,7 +207,7 @@ void dump_short(FILE *out, mp_const_obj_t value) {
 
   } else if (mp_obj_is_small_int(value)) {
     static char num_buf[100];
-    snprintf(num_buf, 100, "%ld", MP_OBJ_SMALL_INT_VALUE(value));
+    snprintf(num_buf, 100, INT_FMT, MP_OBJ_SMALL_INT_VALUE(value));
     print_type(out, "smallint", num_buf, NULL, true);
 
   } else if (!VERIFY_PTR(value)) {
@@ -709,14 +717,14 @@ void dump_qstrdata(FILE *out) {
   }
 }
 
-/// def meminfo(filename: str) -> None:
-///     """Dumps map of micropython GC arena to a file.
-///     The JSON file can be decoded by analyze-memory-dump.py
-///     Only available in the emulator.
-///      """
-STATIC mp_obj_t mod_trezorutils_meminfo(mp_obj_t filename) {
-  size_t fn_len;
-  FILE *out = fopen(mp_obj_str_get_data(filename, &fn_len), "w");
+static void dump_meminfo_json(FILE *out) {
+  if (out == NULL) {
+#if TREZOR_EMULATOR
+    out = stdout;
+#else
+    out = &mp_plat_print;
+#endif
+  }
   fprintf(out, "[");
 
   // void **ptrs = (void **)(void *)&mp_state_ctx;
@@ -779,6 +787,19 @@ STATIC mp_obj_t mod_trezorutils_meminfo(mp_obj_t filename) {
   }
 
   gc_dump_alloc_table();
+}
+
+/// def meminfo(filename: str | None) -> None:
+///     """Dumps map of micropython GC arena to a file.
+///     The JSON file can be decoded by analyze-memory-dump.py
+///      """
+STATIC mp_obj_t mod_trezorutils_meminfo(mp_obj_t filename) {
+  size_t fn_len;
+  FILE *out = (filename == mp_const_none)
+                  ? NULL
+                  : fopen(mp_obj_str_get_data(filename, &fn_len), "w");
+  (void)fn_len;
+  dump_meminfo_json(out);
   return mp_const_none;
 }
 STATIC MP_DEFINE_CONST_FUN_OBJ_1(mod_trezorutils_meminfo_obj,

core/embed/upymod/modtrezorutils/modtrezorutils.c

@@ -284,6 +284,9 @@ STATIC MP_DEFINE_CONST_FUN_OBJ_0(mod_trezorutils_estimate_unused_stack_obj,
 #if MICROPY_OOM_CALLBACK
 static void gc_oom_callback(void) {
   gc_dump_info();
+#if BLOCK_ON_VCP
+  dump_meminfo_json(NULL);  // dump to stdout
+#endif
 }
 
 /// if __debug__: