mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-11-19 05:58:09 +00:00
109 lines
2.8 KiB
C
109 lines
2.8 KiB
C
/*
|
|
* This file is part of the Trezor project, https://trezor.io/
|
|
*
|
|
* Copyright (c) SatoshiLabs
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include <trezor_rtl.h>
|
|
|
|
#include "py/builtin.h"
|
|
#include "py/compile.h"
|
|
#include "py/gc.h"
|
|
#include "py/mperrno.h"
|
|
#include "py/nlr.h"
|
|
#include "py/repl.h"
|
|
#include "py/runtime.h"
|
|
#include "py/stackctrl.h"
|
|
#include "shared/runtime/pyexec.h"
|
|
|
|
#include "ports/stm32/gccollect.h"
|
|
#include "ports/stm32/pendsv.h"
|
|
|
|
#include "rsod.h"
|
|
#include "rust_ui_common.h"
|
|
#include "systask.h"
|
|
#include "system.h"
|
|
|
|
#ifdef USE_SECP256K1_ZKP
|
|
#include "zkp_context.h"
|
|
#endif
|
|
|
|
int main(uint32_t cmd, void *arg) {
|
|
if (cmd == 1) {
|
|
systask_postmortem_t *info = (systask_postmortem_t *)arg;
|
|
rsod_gui(info);
|
|
system_exit(0);
|
|
}
|
|
|
|
screen_boot_stage_2();
|
|
|
|
#ifdef USE_SECP256K1_ZKP
|
|
ensure(sectrue * (zkp_context_init() == 0), NULL);
|
|
#endif
|
|
|
|
printf("CORE: Preparing stack\n");
|
|
// Stack limit should be less than real stack size, so we have a chance
|
|
// to recover from limit hit.
|
|
mp_stack_set_top(&_estack);
|
|
mp_stack_set_limit((char *)&_estack - (char *)&_sstack - 1024);
|
|
|
|
#if MICROPY_ENABLE_PYSTACK
|
|
static mp_obj_t pystack[1024];
|
|
mp_pystack_init(pystack, &pystack[MP_ARRAY_SIZE(pystack)]);
|
|
#endif
|
|
|
|
// GC init
|
|
printf("CORE: Starting GC\n");
|
|
gc_init(&_heap_start, &_heap_end);
|
|
|
|
// Interpreter init
|
|
printf("CORE: Starting interpreter\n");
|
|
mp_init();
|
|
mp_obj_list_init(mp_sys_argv, 0);
|
|
mp_obj_list_init(mp_sys_path, 0);
|
|
mp_obj_list_append(mp_sys_path, MP_OBJ_NEW_QSTR(MP_QSTR__dot_frozen));
|
|
|
|
// Execute the main script
|
|
printf("CORE: Executing main script\n");
|
|
pyexec_frozen_module("main.py");
|
|
|
|
// Clean up
|
|
printf("CORE: Main script finished, cleaning up\n");
|
|
mp_deinit();
|
|
|
|
// Python code shouldn't ever exit, avoid black screen if it does
|
|
error_shutdown("(PE)");
|
|
|
|
return 0;
|
|
}
|
|
|
|
// MicroPython default exception handler
|
|
|
|
void __attribute__((noreturn)) nlr_jump_fail(void *val) {
|
|
error_shutdown("(UE)");
|
|
}
|
|
|
|
// MicroPython builtin stubs
|
|
|
|
mp_import_stat_t mp_import_stat(const char *path) {
|
|
return MP_IMPORT_STAT_NO_EXIST;
|
|
}
|
|
|
|
mp_obj_t mp_builtin_open(uint n_args, const mp_obj_t *args, mp_map_t *kwargs) {
|
|
return mp_const_none;
|
|
}
|
|
MP_DEFINE_CONST_FUN_OBJ_KW(mp_builtin_open_obj, 1, mp_builtin_open);
|