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

Merge pull request #6 from NinjaCheetah/breakdown_wad

Browse Source 
Extract individual app contents from WAD
This commit is contained in:
Lillian Skinner 2024-03-01 00:00:30 -05:00 committed by GitHub
commit 80c37b9b55
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 27 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
.gitignore vendored
View File

@ -163,4 +163,10 @@ cython_debug/
# Allows me to keep TMD files in my repository folder for testing without accidentally publishing them # Allows me to keep TMD files in my repository folder for testing without accidentally publishing them
*.tmd *.tmd
*.wad *.wad
out_prod/
remakewad.pl
# Also awful macOS files
*._*
*.DS_Store

23
src/libWiiPy/tmd.py
View File

@ -5,6 +5,7 @@
import io import io
import binascii import binascii
import struct
from dataclasses import dataclass from dataclasses import dataclass
from typing import List from typing import List
@ -14,9 +15,9 @@ class ContentRecord:
"""Creates a content record object that contains the details of a content contained in a title.""" """Creates a content record object that contains the details of a content contained in a title."""
cid: int # Content ID cid: int # Content ID
index: int # Index in the list of contents index: int # Index in the list of contents
content_type: int # normal: 0x0001; dlc: 0x4001; shared: 0x8001 content_type: int # Normal: 0x0001, DLC: 0x4001, Shared: 0x8001
content_size: int content_size: int
content_hash: bytearray # SHA1 hash content content_hash: bytes # SHA1 hash content
class TMD: class TMD:
@ -41,7 +42,7 @@ class TMD:
self.title_version: int # The version of the associated title. self.title_version: int # The version of the associated title.
self.num_contents: int # The number of contents contained in the associated title. self.num_contents: int # The number of contents contained in the associated title.
self.boot_index: int self.boot_index: int
self.content_record: List[ContentRecord] self.content_records: List[ContentRecord] = []
# Load data from TMD file # Load data from TMD file
with io.BytesIO(self.tmd) as tmddata: with io.BytesIO(self.tmd) as tmddata:
# ==================================================================================== # ====================================================================================
@ -104,6 +105,14 @@ class TMD:
# Content index in content list that contains the boot file # Content index in content list that contains the boot file
tmddata.seek(0x1E0) tmddata.seek(0x1E0)
self.boot_index = tmddata.read(2) self.boot_index = tmddata.read(2)
# Get content records for the number of contents in num_contents.
for content in range(0, self.num_contents):
tmddata.seek(0x1E4 + (36 * content))
content_record_hdr = struct.unpack(">LHH4x4s20s", tmddata.read(36))
self.content_records.append(
ContentRecord(int(content_record_hdr[0]), int(content_record_hdr[1]),
int(content_record_hdr[2]), int.from_bytes(content_record_hdr[3]),
binascii.hexlify(content_record_hdr[4])))
def get_title_id(self): def get_title_id(self):
"""Returns the TID of the TMD's associated title.""" """Returns the TID of the TMD's associated title."""
@ -186,3 +195,11 @@ class TMD:
def get_num_contents(self): def get_num_contents(self):
"""Returns the number of contents listed in the TMD.""" """Returns the number of contents listed in the TMD."""
return self.num_contents return self.num_contents
def get_content_record(self, record):
"""Returns the content record at the specified index."""
if record < self.num_contents:
return self.content_records[record]
else:
raise IndexError("Invalid content record! TMD lists '" + str(self.num_contents - 1) +
"' contents but index was '" + str(record) + "'!")