1
0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-12-05 05:58:23 +00:00
trezor-firmware/tools/snippets/eth_defs_unpack.py

108 lines
3.4 KiB
Python
Raw Normal View History

#!/usr/bin/env python3
# This file is part of the Trezor project.
#
# Copyright (C) 2012-2022 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>.
from __future__ import annotations
import click
import requests
import zipfile
from pathlib import Path
from trezorlib import definitions, merkle_tree
ZIP_FILENAME = "definitions-sparse.zip"
TOPDIRS = ("chain-id", "slip44")
class SparseZipSource(definitions.Source):
def __init__(self, zip: Path | zipfile.ZipFile) -> None:
if isinstance(zip, Path):
self.zip = zipfile.ZipFile(zip)
else:
self.zip = zip
# extract signature
self.signature = self.read_bytes("signature.dat")
self.root_hash = self.read_bytes("root.dat")
# construct a Merkle tree
entries = []
for name in self.zip.namelist():
if name.startswith("chain-id/"):
entries.append(self.read_bytes(name))
entries.sort()
self.merkle_tree = merkle_tree.MerkleTree(entries)
if self.root_hash != self.merkle_tree.get_root_hash():
raise ValueError("Failed to reconstruct the correct Merkle tree")
def read_bytes(self, path: str | Path) -> bytes:
with self.zip.open(str(path)) as f:
return f.read()
def fetch_path(self, *components: str) -> bytes | None:
path = "/".join(components)
data = self.read_bytes(path)
proof = self.merkle_tree.get_proof(data)
proof_bytes = definitions.ProofFormat.build(proof)
return data + proof_bytes + self.signature
@click.command()
@click.option(
"-z",
"--definitions-zip",
type=click.Path(exists=True, dir_okay=False, resolve_path=True, path_type=Path),
help="Local zip file with stored definitions.",
)
@click.argument(
"outdir",
type=click.Path(resolve_path=True, file_okay=False, writable=True, path_type=Path),
)
def unpack_definitions(definitions_zip: Path, outdir: Path) -> None:
"""Script that unpacks and completes (insert missing Merkle Tree proofs
into the definitions) the Ethereum definitions (networks and tokens).
If no local zip is provided, the latest one will be downloaded from trezor.io.
"""
if definitions_zip is None:
result = requests.get(definitions.DEFS_BASE_URL + ZIP_FILENAME)
result.raise_for_status()
zip = zipfile.ZipFile(result.raw)
else:
zip = zipfile.ZipFile(definitions_zip)
source = SparseZipSource(zip)
if not outdir.exists():
outdir.mkdir()
for name in zip.namelist():
if name == "signature.dat" or not name.endswith(".dat"):
continue
local_path = outdir / name
local_path.parent.mkdir(parents=True, exist_ok=True)
data = source.fetch_path(name)
assert data is not None, f"Could not read data for: {name}"
local_path.write_bytes(data)
if __name__ == "__main__":
unpack_definitions()