NinjaCheetah/WiiPy
1
0
Fork 0
mirror of https://github.com/NinjaCheetah/WiiPy.git synced 2025-04-26 13:21:01 -04:00
Code Issues Packages Projects Releases Wiki Activity

Rewrote EmuNAND module to use libWiiPy's new EmuNAND class

Browse Source
This commit is contained in:
Campbell 2024-08-08 13:24:56 -04:00
parent 3062a739d6
commit fa6ba28dbe
Signed by: NinjaCheetah
GPG Key ID: B547958AF96ED344
1 changed files with 4 additions and 139 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

143
modules/title/emunand.py
View File

@ -1,110 +1,12 @@
# "modules/title/emunand.py" from WiiPy by NinjaCheetah # "modules/title/emunand.py" from WiiPy by NinjaCheetah
# https://github.com/NinjaCheetah/WiiPy # https://github.com/NinjaCheetah/WiiPy
import os
import pathlib import pathlib
import shutil
import libWiiPy import libWiiPy
class _EmuNANDStructure:
def __init__(self, emunand_root):
self.emunand_root: pathlib.Path = emunand_root
self.import_dir = self.emunand_root.joinpath("import")
self.meta_dir = self.emunand_root.joinpath("meta")
self.shared1_dir = self.emunand_root.joinpath("shared1")
self.shared2_dir = self.emunand_root.joinpath("shared2")
self.sys_dir = self.emunand_root.joinpath("sys")
self.ticket_dir = self.emunand_root.joinpath("ticket")
self.title_dir = self.emunand_root.joinpath("title")
self.tmp_dir = self.emunand_root.joinpath("tmp")
self.wfs_dir = self.emunand_root.joinpath("wfs")
self.import_dir.mkdir(exist_ok=True)
self.meta_dir.mkdir(exist_ok=True)
self.shared1_dir.mkdir(exist_ok=True)
self.shared2_dir.mkdir(exist_ok=True)
self.sys_dir.mkdir(exist_ok=True)
self.ticket_dir.mkdir(exist_ok=True)
self.title_dir.mkdir(exist_ok=True)
self.tmp_dir.mkdir(exist_ok=True)
self.wfs_dir.mkdir(exist_ok=True)
def _do_wad_install(emunand_struct: _EmuNANDStructure, title: libWiiPy.title.Title, skip_hash=False):
# Save the upper and lower portions of the Title ID, because these are used as target install directories.
tid_upper = title.tmd.title_id[:8]
tid_lower = title.tmd.title_id[8:]
# Tickets are installed as <tid_lower>.tik in /ticket/<tid_upper>/
ticket_dir = emunand_struct.ticket_dir.joinpath(tid_upper)
ticket_dir.mkdir(exist_ok=True)
open(ticket_dir.joinpath(tid_lower + ".tik"), "wb").write(title.wad.get_ticket_data())
# The TMD and normal contents are installed to /title/<tid_upper>/<tid_lower>/content/, with the tmd being named
# title.tmd and the contents being named <cid>.app.
title_dir = emunand_struct.title_dir.joinpath(tid_upper)
title_dir.mkdir(exist_ok=True)
title_dir = title_dir.joinpath(tid_lower)
title_dir.mkdir(exist_ok=True)
content_dir = title_dir.joinpath("content")
if content_dir.exists():
shutil.rmtree(content_dir) # Clear the content directory so old contents aren't left behind.
content_dir.mkdir(exist_ok=True)
open(content_dir.joinpath("title.tmd"), "wb").write(title.wad.get_tmd_data())
for content_file in range(0, title.tmd.num_contents):
if title.tmd.content_records[content_file].content_type == 1:
content_file_name = f"{title.tmd.content_records[content_file].content_id:08X}".lower()
open(content_dir.joinpath(content_file_name + ".app"), "wb").write(
title.get_content_by_index(content_file, skip_hash=skip_hash))
title_dir.joinpath("data").mkdir(exist_ok=True) # Empty directory used for save data for the title.
# Shared contents need to be installed to /shared1/, with incremental names determined by /shared1/content.map.
content_map_path = emunand_struct.shared1_dir.joinpath("content.map")
content_map = libWiiPy.title.SharedContentMap()
existing_hashes = []
if content_map_path.exists():
content_map.load(open(content_map_path, "rb").read())
for record in content_map.shared_records:
existing_hashes.append(record.content_hash)
for content_file in range(0, title.tmd.num_contents):
if title.tmd.content_records[content_file].content_type == 32769:
if title.tmd.content_records[content_file].content_hash not in existing_hashes:
content_file_name = content_map.add_content(title.tmd.content_records[content_file].content_hash)
open(emunand_struct.shared1_dir.joinpath(content_file_name + ".app"), "wb").write(
title.get_content_by_index(content_file, skip_hash=skip_hash))
open(emunand_struct.shared1_dir.joinpath("content.map"), "wb").write(content_map.dump())
# The "footer" or meta file is installed as title.met in /meta/<tid_upper>/<tid_lower>/. Only write this if meta
# is not nothing.
meta_data = title.wad.get_meta_data()
if meta_data != b'':
meta_dir = emunand_struct.meta_dir.joinpath(tid_upper)
meta_dir.mkdir(exist_ok=True)
meta_dir = meta_dir.joinpath(tid_lower)
meta_dir.mkdir(exist_ok=True)
open(meta_dir.joinpath("title.met"), "wb").write(title.wad.get_meta_data())
# The Title ID needs to be added to uid.sys, which is essentially a log of all titles that are installed or have
# ever been installed.
uid_sys_path = emunand_struct.sys_dir.joinpath("uid.sys")
uid_sys = libWiiPy.title.UidSys()
existing_tids = []
if uid_sys_path.exists():
uid_sys.load(open(uid_sys_path, "rb").read())
for entry in uid_sys.uid_entries:
existing_tids.append(entry.title_id)
else:
uid_sys.create()
existing_tids.append(uid_sys.uid_entries[0].title_id)
if title.tmd.title_id not in existing_tids:
uid_sys.add(title.tmd.title_id)
open(uid_sys_path, "wb").write(uid_sys.dump())
def handle_emunand_title(args): def handle_emunand_title(args):
emunand_path = pathlib.Path(args.emunand) emunand = libWiiPy.title.EmuNAND(args.emunand)
if args.skip_hash: if args.skip_hash:
skip_hash = True skip_hash = True
else: else:
@ -116,14 +18,6 @@ def handle_emunand_title(args):
if not input_path.exists(): if not input_path.exists():
raise FileNotFoundError(input_path) raise FileNotFoundError(input_path)
# Check if the EmuNAND path already exists, and ensure that it is a directory if it does.
if emunand_path.exists():
if emunand_path.is_file():
raise ValueError("A file already exists with the provided directory name!")
else:
emunand_path.mkdir()
emunand_struct = _EmuNANDStructure(emunand_path)
if input_path.is_dir(): if input_path.is_dir():
wad_files = list(input_path.glob("*.[wW][aA][dD]")) wad_files = list(input_path.glob("*.[wW][aA][dD]"))
@ -134,7 +28,7 @@ def handle_emunand_title(args):
title = libWiiPy.title.Title() title = libWiiPy.title.Title()
title.load_wad(open(wad, "rb").read()) title.load_wad(open(wad, "rb").read())
try: try:
_do_wad_install(emunand_struct, title, skip_hash) emunand.install_title(title, skip_hash=skip_hash)
wad_count += 1 wad_count += 1
except ValueError: except ValueError:
print(f"WAD {wad} could not be installed!") print(f"WAD {wad} could not be installed!")
@ -142,7 +36,7 @@ def handle_emunand_title(args):
else: else:
title = libWiiPy.title.Title() title = libWiiPy.title.Title()
title.load_wad(open(input_path, "rb").read()) title.load_wad(open(input_path, "rb").read())
_do_wad_install(emunand_struct, title, skip_hash) emunand.install_title(title, skip_hash=skip_hash)
print("Successfully installed WAD to EmuNAND!") print("Successfully installed WAD to EmuNAND!")
# Code for if the --uninstall argument was passed. # Code for if the --uninstall argument was passed.
@ -158,35 +52,6 @@ def handle_emunand_title(args):
if len(target_tid) != 16: if len(target_tid) != 16:
raise ValueError("Invalid Title ID! Title IDs must be 16 characters long.") raise ValueError("Invalid Title ID! Title IDs must be 16 characters long.")
# Setup required EmuNAND directories. emunand.uninstall_title(target_tid)
ticket_dir = emunand_path.joinpath("ticket")
title_dir = emunand_path.joinpath("title")
meta_dir = emunand_path.joinpath("meta")
# Save the upper and lower portions of the Title ID, because these are used as target install directories.
tid_upper = target_tid[:8]
tid_lower = target_tid[8:]
if not title_dir.joinpath(tid_upper).joinpath(tid_lower).exists():
print(f"Title with Title ID {target_tid} does not appear to be installed!")
# Begin by removing the Ticket, which is installed to /ticket/<tid_upper>/<tid_lower>.tik
if ticket_dir.joinpath(tid_upper).joinpath(tid_lower + ".tik").exists():
os.remove(ticket_dir.joinpath(tid_upper).joinpath(tid_lower + ".tik"))
# The TMD and contents are stored in /title/<tid_upper>/<tid_lower>/. Remove the TMD and all contents, but don't
# delete the entire directory if anything exists in data.
title_dir = title_dir.joinpath(tid_upper).joinpath(tid_lower)
if not title_dir.joinpath("data").exists():
shutil.rmtree(title_dir)
elif title_dir.joinpath("data").exists() and not os.listdir(title_dir.joinpath("data")):
shutil.rmtree(title_dir)
else:
# There are files in data, so we only want to delete the content directory.
shutil.rmtree(title_dir.joinpath("content"))
# On the off chance this title has a meta entry, delete that too.
if meta_dir.joinpath(tid_upper).joinpath(tid_lower).joinpath("title.met").exists():
shutil.rmtree(meta_dir.joinpath(tid_upper).joinpath(tid_lower))
print("Title uninstalled from EmuNAND!") print("Title uninstalled from EmuNAND!")