Atlas/Atlas-Install
1
0
Fork 0
mirror of https://github.com/Faraphel/Atlas-Install.git synced 2025-07-05 12:18:21 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

added patch_data to str.py, allowing for main.dol modification

Browse source
This commit is contained in:
Faraphel 2022-07-14 16:41:05 +02:00
parent f11c8a5de2
commit ce30419f8c
1 changed files with 32 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

33
source/wt/str.py
View file

@ -1,3 +1,34 @@
from source.wt import *
tools_path = tools_szs_dir / ("wstrt.exe" if system == "win64" else "wstrt")
tools_path = tools_szs_dir / ("wstrt.exe" if system == "win64" else "wstrt")
_tools_run = get_tools_run_function(tools_path)
_tools_run_popen = get_tools_run_popen_function(tools_path)
@better_wt_error(tools_path)
def patch_data(dol_data: bytes, region: int = None, https: str = None, domain: str = None,
sections: list[Path] = None) -> bytes:
"""
Patch a main.dol file content
:param https: can be RESTORE, HTTP, DOMAIN, SAKE0 or SAKE1. Allow for modifying url to custom server
:param domain: if https is set to DOMAIN, url to the custom server
:param sections: sections that can be added to manage cheat
:param region: optional region for the game
:param dol_data: main.dol file content
:return: patched main.dol file content
"""
args = []
if region is not None: args.extend(["--region", region])
if https is not None: args.extend(["--https", https])
if domain is not None: args.extend(["--domain", domain])
for section in sections if sections is not None else []:
args.extend(["--add-section", section])
process = _tools_run_popen("PATCH", "-", "--DEST", "-", "--clean-dol", "add-lecode", *args)
stdout, _ = process.communicate(input=dol_data)
if process.returncode != 0:
raise WTError(tools_path, process.returncode)
return stdout