mirror of
https://github.com/zeldaret/botw
synced 2026-10-09 13:37:01 -04:00
Pistonight
124 lines
4.6 KiB
Python
Executable File
124 lines
4.6 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
from common.setup_venv import enter_venv
|
|
if __name__ == "__main__":
|
|
enter_venv()
|
|
|
|
import argparse
|
|
import hashlib
|
|
from pathlib import Path
|
|
import subprocess
|
|
import tempfile
|
|
import urllib.request
|
|
from typing import Optional
|
|
from common import setup_common as setup
|
|
|
|
|
|
TARGET_PATH = setup.get_target_path()
|
|
TARGET_ELF_PATH = setup.get_target_elf_path()
|
|
|
|
|
|
def _download_downgrade_to_v150_patch(version: str, dest: Path):
|
|
print(f">>>> downloading patch for {version}...")
|
|
urllib.request.urlretrieve(f"https://s.botw.link/v150_downgrade/{version}_to_v150.patch", dest)
|
|
|
|
|
|
def prepare_executable(original_nso: Optional[Path]):
|
|
COMPRESSED_V150_HASH = "898dc199301f7c419be5144bb5cb27e2fc346e22b27345ba3fb40c0060c2baf8"
|
|
UNCOMPRESSED_V150_HASH = "d9fa308d0ee7c0ab081c66d987523385e1afe06f66731bbfa32628438521c106"
|
|
COMPRESSED_V160_HASH = "15cfca7b89348956f85d945fade2e215a6af5991ed1071e181f97ca72f7ae20b"
|
|
UNCOMPRESSED_V160_HASH = "8a2fc8b1111a35a76fd2d53a8670599da4a7a9706a3d91215d30fd62149f00c1"
|
|
COMPRESSED_V190_HASH = "f6f0b20bb1f2b67d164c24871cd473147eef1d285f12e7bd492e3126c6022e27"
|
|
UNCOMPRESSED_V190_HASH = "6a30c162a2e6f885e0b1c0a6673a84795549a5ccdec1de8d3a741168ac4cba86"
|
|
|
|
# The uncompressed v1.5.0 main NSO.
|
|
TARGET_HASH = UNCOMPRESSED_V150_HASH
|
|
|
|
if TARGET_PATH.is_file() and hashlib.sha256(TARGET_PATH.read_bytes()).hexdigest() == TARGET_HASH and TARGET_ELF_PATH.is_file():
|
|
print(">>> NSO is already set up")
|
|
return
|
|
|
|
if original_nso is None:
|
|
setup.fail("please pass a path to the NSO (refer to the readme for more details)")
|
|
|
|
if not original_nso.is_file():
|
|
setup.fail(f"{original_nso} is not a file")
|
|
|
|
nso_data = original_nso.read_bytes()
|
|
nso_hash = hashlib.sha256(nso_data).hexdigest()
|
|
|
|
version_matrix = [
|
|
# SHA256 Compressed? Version
|
|
(UNCOMPRESSED_V150_HASH, False, "1.5.0"),
|
|
( COMPRESSED_V150_HASH, True, "1.5.0"),
|
|
(UNCOMPRESSED_V160_HASH, False, "1.6.0"),
|
|
( COMPRESSED_V160_HASH, True, "1.6.0"),
|
|
(UNCOMPRESSED_V190_HASH, False, "1.9.0"),
|
|
( COMPRESSED_V190_HASH, True, "1.9.0"),
|
|
]
|
|
|
|
for (expected_hash, is_compressed, version_str) in version_matrix:
|
|
if nso_hash != expected_hash:
|
|
continue
|
|
print(f">>> found {"compressed" if is_compressed else "uncompressed"} {version_str} NSO")
|
|
|
|
if version_str == "1.5.0":
|
|
if is_compressed:
|
|
setup._decompress_nso(original_nso, TARGET_PATH)
|
|
else:
|
|
TARGET_PATH.write_bytes(nso_data)
|
|
break
|
|
|
|
with tempfile.TemporaryDirectory() as tmpdir:
|
|
patch_path = Path(tmpdir) / "patch"
|
|
patch_version_str = "v" + version_str.replace(".","")
|
|
_download_downgrade_to_v150_patch(patch_version_str, patch_path)
|
|
if is_compressed:
|
|
decompressed_nso_path = Path(tmpdir) / f"{patch_version_str}.nso"
|
|
setup._decompress_nso(original_nso, decompressed_nso_path)
|
|
else:
|
|
decompressed_nso_path = original_nso
|
|
|
|
setup._apply_xdelta3_patch(decompressed_nso_path, patch_path, TARGET_PATH)
|
|
break
|
|
else:
|
|
setup.fail(f"unknown executable: {nso_hash}")
|
|
|
|
if not TARGET_PATH.is_file():
|
|
setup.fail("internal error while preparing executable (missing NSO); please report")
|
|
if hashlib.sha256(TARGET_PATH.read_bytes()).hexdigest() != TARGET_HASH:
|
|
setup.fail("internal error while preparing executable (wrong NSO hash); please report")
|
|
|
|
setup._convert_nso_to_elf(TARGET_PATH)
|
|
|
|
if not TARGET_ELF_PATH.is_file():
|
|
setup.fail("internal error while preparing executable (missing ELF); please report")
|
|
|
|
|
|
def create_build_dir():
|
|
build_dir = setup.ROOT / "build"
|
|
if build_dir.is_dir():
|
|
print(">>> build directory already exists: nothing to do")
|
|
return
|
|
|
|
subprocess.check_call(
|
|
"cmake -GNinja -DCMAKE_BUILD_TYPE=RelWithDebInfo -DCMAKE_TOOLCHAIN_FILE=toolchain/ToolchainNX64.cmake -DCMAKE_CXX_COMPILER_LAUNCHER=ccache -B build/".split(" "))
|
|
print(">>> created build directory")
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(
|
|
"setup.py", description="Set up the Breath of the Wild decompilation project")
|
|
parser.add_argument("original_nso", type=Path,
|
|
help="Path to the original NSO (1.5.0 or 1.9.0, compressed or not)", nargs="?")
|
|
args = parser.parse_args()
|
|
|
|
setup.install_viking()
|
|
prepare_executable(args.original_nso)
|
|
setup.set_up_compiler("4.0.1")
|
|
create_build_dir()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|