core/unix/main: Print memory info (peak memory) when TREZOR_LOG_MEMORY env var is set. Part of #194

5 years ago · 91007bafe9
parent 813beda90d
commit 91007bafe9
1 changed files with 5 additions and 1 deletions
--- a/core/embed/unix/main.c
+++ b/core/embed/unix/main.c
@ -653,7 +653,11 @@ MP_NOINLINE int main_(int argc, char **argv) {
  }

 #if MICROPY_PY_MICROPYTHON_MEM_INFO
-  if (mp_verbose_flag) {
+  char *env_str_trezor_log_memory = getenv("TREZOR_LOG_MEMORY");
+  if (!env_str_trezor_log_memory || atoi(env_str_trezor_log_memory) == 0) {
+    env_str_trezor_log_memory = NULL;
+  }
+  if (mp_verbose_flag || env_str_trezor_log_memory) {
    mp_micropython_mem_info(0, NULL);
  }
 #endif