mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-11-30 19:38:15 +00:00
128 lines
3.4 KiB
Python
128 lines
3.4 KiB
Python
from __future__ import annotations
|
|
|
|
import subprocess
|
|
import zlib
|
|
from pathlib import Path
|
|
|
|
HERE = Path(__file__).parent.resolve()
|
|
|
|
# go up from site_scons to core/
|
|
PROJECT_ROOT = HERE.parent.resolve()
|
|
|
|
|
|
def get_version(file: str) -> str:
|
|
major = 0
|
|
minor = 0
|
|
patch = 0
|
|
|
|
file_path = PROJECT_ROOT / file
|
|
with open(file_path, "r") as f:
|
|
for line in f:
|
|
if line.startswith("#define VERSION_MAJOR "):
|
|
major = line.split("VERSION_MAJOR")[1].strip()
|
|
if line.startswith("#define VERSION_MINOR "):
|
|
minor = line.split("VERSION_MINOR")[1].strip()
|
|
if line.startswith("#define VERSION_PATCH "):
|
|
patch = line.split("VERSION_PATCH")[1].strip()
|
|
return f"{major}.{minor}.{patch}"
|
|
|
|
|
|
def get_git_revision_hash() -> str:
|
|
return subprocess.check_output(["git", "rev-parse", "HEAD"]).decode("ascii").strip()
|
|
|
|
|
|
def get_git_revision_short_hash() -> str:
|
|
return (
|
|
subprocess.check_output(["git", "rev-parse", "--short", "HEAD"])
|
|
.decode("ascii")
|
|
.strip()
|
|
)
|
|
|
|
|
|
def get_git_modified() -> bool:
|
|
return (
|
|
subprocess.check_output(["git", "diff", "--name-status"])
|
|
.decode("ascii")
|
|
.strip()
|
|
!= ""
|
|
)
|
|
|
|
|
|
def get_defs_for_cmake(defs: list[str | tuple[str, str]]) -> list[str]:
|
|
result: list[str] = []
|
|
for d in defs:
|
|
if type(d) is tuple:
|
|
val = d[1].replace('"', '\\"').replace("(", "\\(").replace(")", "\\)")
|
|
result.append(f'{d[0]}="{val}"')
|
|
else:
|
|
result.append(d)
|
|
return result
|
|
|
|
|
|
def _compress(data: bytes) -> bytes:
|
|
z = zlib.compressobj(level=9, wbits=-10)
|
|
return z.compress(data) + z.flush()
|
|
|
|
|
|
def get_bindgen_defines(defines: list[str | tuple[str, str]], paths: list[str]) -> str:
|
|
rest_defs = []
|
|
for d in defines:
|
|
if type(d) is tuple:
|
|
d = f"-D{d[0]}={d[1]}"
|
|
else:
|
|
d = f"-D{d}"
|
|
rest_defs.append(d)
|
|
for d in paths:
|
|
rest_defs.append(f"-I../../{d}")
|
|
|
|
return ",".join(rest_defs)
|
|
|
|
|
|
def embed_compressed_binary(obj_program, env, section, target_, file, build, symbol):
|
|
_in = f"embedded_{section}.bin.deflated"
|
|
|
|
def redefine_sym(suffix):
|
|
src = (
|
|
f"_binary_build_{build}_"
|
|
+ _in.replace("/", "_").replace(".", "_")
|
|
+ "_"
|
|
+ suffix
|
|
)
|
|
dest = f"_deflated_{symbol}_{suffix}"
|
|
return f" --redefine-sym {src}={dest}"
|
|
|
|
def compress_action(target, source, env):
|
|
srcf = Path(str(source[0]))
|
|
dstf = Path(str(target[0]))
|
|
compressed = _compress(srcf.read_bytes())
|
|
dstf.write_bytes(compressed)
|
|
return 0
|
|
|
|
compress = env.Command(target=_in, source=file, action=compress_action)
|
|
|
|
obj_program.extend(
|
|
env.Command(
|
|
target=target_,
|
|
source=_in,
|
|
action="$OBJCOPY -I binary -O elf32-littlearm -B arm"
|
|
f" --rename-section .data=.{section}"
|
|
+ redefine_sym("start")
|
|
+ redefine_sym("end")
|
|
+ redefine_sym("size")
|
|
+ " $SOURCE $TARGET",
|
|
)
|
|
)
|
|
|
|
env.Depends(obj_program, compress)
|
|
|
|
|
|
def embed_raw_binary(obj_program, env, section, target_, file):
|
|
obj_program.extend(
|
|
env.Command(
|
|
target=target_,
|
|
source=file,
|
|
action="$OBJCOPY -I binary -O elf32-littlearm -B arm"
|
|
f" --rename-section .data=.{section}" + " $SOURCE $TARGET",
|
|
)
|
|
)
|