arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-11-22 15:38:11 +00:00
Code Issues Releases Wiki Activity

feat(core/emu.py): erase profile when running emu.py -e

Browse Source
This commit is contained in:
matejcik 2021-04-07 11:09:18 +02:00 committed by matejcik
parent bb96cdc4e6
commit 25385af159
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
core/emu.py
View File

@ -29,6 +29,11 @@ PROFILING_WRAPPER = HERE / "prof" / "prof.py"
PROFILE_BASE = Path.home() / ".trezoremu" PROFILE_BASE = Path.home() / ".trezoremu"
TREZOR_STORAGE_FILES = (
"trezor.flash",
"trezor.sdcard",
)
def run_command_with_emulator(emulator, command): def run_command_with_emulator(emulator, command):
with emulator: with emulator:
@ -87,6 +92,7 @@ def _from_env(name):
@click.option("-c", "--command", "run_command", is_flag=True, help="Run command while emulator is running") @click.option("-c", "--command", "run_command", is_flag=True, help="Run command while emulator is running")
@click.option("-d", "--production/--no-production", default=_from_env("PYOPT"), help="Production mode (debuglink disabled)") @click.option("-d", "--production/--no-production", default=_from_env("PYOPT"), help="Production mode (debuglink disabled)")
@click.option("-D", "--debugger", is_flag=True, help="Run emulator in debugger (gdb/lldb)") @click.option("-D", "--debugger", is_flag=True, help="Run emulator in debugger (gdb/lldb)")
@click.option("-e", "--erase", is_flag=True, help="Erase profile before running")
@click.option("--executable", type=click.Path(exists=True, dir_okay=False), default=os.environ.get("MICROPYTHON"), help="Alternate emulator executable") @click.option("--executable", type=click.Path(exists=True, dir_okay=False), default=os.environ.get("MICROPYTHON"), help="Alternate emulator executable")
@click.option("-g", "--profiling/--no-profiling", default=_from_env("TREZOR_PROFILING"), help="Run with profiler wrapper") @click.option("-g", "--profiling/--no-profiling", default=_from_env("TREZOR_PROFILING"), help="Run with profiler wrapper")
@click.option("-G", "--alloc-profiling/--no-alloc-profiling", default=_from_env("TREZOR_MEMPERF"), help="Profile memory allocation (requires special micropython build)") @click.option("-G", "--alloc-profiling/--no-alloc-profiling", default=_from_env("TREZOR_MEMPERF"), help="Profile memory allocation (requires special micropython build)")
@ -110,6 +116,7 @@ def cli(
run_command, run_command,
production, production,
debugger, debugger,
erase,
executable, executable,
profiling, profiling,
alloc_profiling, alloc_profiling,
@ -198,6 +205,10 @@ def cli(
else: else:
profile_dir = Path("/var/tmp") profile_dir = Path("/var/tmp")
if erase:
for entry in TREZOR_STORAGE_FILES:
(profile_dir / entry).unlink(missing_ok=True)
if quiet: if quiet:
output = None output = None