mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-11-25 17:09:44 +00:00
41e07769e2
Allowing tests to specify the profiling wrapper over the emulator it is starting. [no changelog]
125 lines
3.8 KiB
Python
125 lines
3.8 KiB
Python
# This file is part of the Trezor project.
|
|
#
|
|
# Copyright (C) 2012-2019 SatoshiLabs and contributors
|
|
#
|
|
# This library is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Lesser General Public License version 3
|
|
# as published by the Free Software Foundation.
|
|
#
|
|
# This library 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 Lesser General Public License for more details.
|
|
#
|
|
# You should have received a copy of the License along with this library.
|
|
# If not, see <https://www.gnu.org/licenses/lgpl-3.0.html>.
|
|
|
|
import tempfile
|
|
from collections import defaultdict
|
|
from pathlib import Path
|
|
from typing import Dict, List, Optional, Sequence, Tuple
|
|
|
|
from trezorlib._internal.emulator import CoreEmulator, Emulator, LegacyEmulator
|
|
|
|
ROOT = Path(__file__).resolve().parent.parent
|
|
BINDIR = ROOT / "tests" / "emulators"
|
|
|
|
LOCAL_BUILD_PATHS = {
|
|
"core": ROOT / "core" / "build" / "unix" / "trezor-emu-core",
|
|
"legacy": ROOT / "legacy" / "firmware" / "trezor.elf",
|
|
}
|
|
|
|
CORE_SRC_DIR = ROOT / "core" / "src"
|
|
|
|
ENV = {"SDL_VIDEODRIVER": "dummy"}
|
|
|
|
|
|
def check_version(tag: str, version_tuple: Tuple[int, int, int]) -> None:
|
|
if tag is not None and tag.startswith("v") and len(tag.split(".")) == 3:
|
|
version = ".".join(str(i) for i in version_tuple)
|
|
if tag[1:] != version:
|
|
raise RuntimeError(f"Version mismatch: tag {tag} reports version {version}")
|
|
|
|
|
|
def filename_from_tag(gen: str, tag: str) -> Path:
|
|
return BINDIR / f"trezor-emu-{gen}-{tag}"
|
|
|
|
|
|
def get_tags() -> Dict[str, List[str]]:
|
|
files = list(BINDIR.iterdir())
|
|
if not files:
|
|
raise ValueError(
|
|
"No files found. Use download_emulators.sh to download emulators."
|
|
)
|
|
|
|
result = defaultdict(list)
|
|
for f in sorted(files):
|
|
try:
|
|
# example: "trezor-emu-core-v2.1.1" or "trezor-emu-core-v2.1.1-46ab42fw"
|
|
_, _, gen, tag = f.name.split("-", maxsplit=3)
|
|
result[gen].append(tag)
|
|
except ValueError:
|
|
pass
|
|
return result
|
|
|
|
|
|
ALL_TAGS = get_tags()
|
|
|
|
|
|
class EmulatorWrapper:
|
|
def __init__(
|
|
self,
|
|
gen: str,
|
|
tag: Optional[str] = None,
|
|
storage: Optional[bytes] = None,
|
|
port: Optional[int] = None,
|
|
headless: bool = True,
|
|
auto_interact: bool = True,
|
|
main_args: Sequence[str] = ("-m", "main"),
|
|
) -> None:
|
|
if tag is not None:
|
|
executable = filename_from_tag(gen, tag)
|
|
else:
|
|
executable = LOCAL_BUILD_PATHS[gen]
|
|
|
|
if not executable.exists():
|
|
raise ValueError(f"emulator executable not found: {executable}")
|
|
|
|
self.profile_dir = tempfile.TemporaryDirectory()
|
|
if executable == LOCAL_BUILD_PATHS["core"]:
|
|
workdir = CORE_SRC_DIR
|
|
else:
|
|
workdir = None
|
|
|
|
if gen == "legacy":
|
|
self.emulator = LegacyEmulator(
|
|
executable,
|
|
self.profile_dir.name,
|
|
storage=storage,
|
|
headless=headless,
|
|
auto_interact=auto_interact,
|
|
)
|
|
elif gen == "core":
|
|
self.emulator = CoreEmulator(
|
|
executable,
|
|
self.profile_dir.name,
|
|
storage=storage,
|
|
workdir=workdir,
|
|
port=port,
|
|
headless=headless,
|
|
auto_interact=auto_interact,
|
|
main_args=main_args,
|
|
)
|
|
else:
|
|
raise ValueError(
|
|
f"Unrecognized gen - {gen} - only 'core' and 'legacy' supported"
|
|
)
|
|
|
|
def __enter__(self) -> Emulator:
|
|
self.emulator.start()
|
|
return self.emulator
|
|
|
|
def __exit__(self, exc_type, exc_value, traceback) -> None:
|
|
self.emulator.stop()
|
|
self.profile_dir.cleanup()
|