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

Add a check to wad.py that ensures the file being loaded is a WAD

Browse Source
This commit is contained in:
Campbell 2024-03-19 21:43:18 -04:00
parent 0cb37487b6
commit bfe937f58f
Signed by: NinjaCheetah
GPG Key ID: B547958AF96ED344
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/libWiiPy/wad.py
View File

@ -39,6 +39,15 @@ class WAD:
self.wad_meta_offset: int
# Load header data from WAD stream
with io.BytesIO(self.wad) as wad_data:
# Read the first 8 bytes of the file to ensure that it's a WAD. This will currently reject boot2 WADs, but
# this tool cannot handle them correctly right now anyway.
wad_data.seek(0x0)
wad_magic_bin = wad_data.read(8)
wad_magic_hex = binascii.hexlify(wad_magic_bin)
wad_magic = str(wad_magic_hex.decode())
if wad_magic != "0000002049730000":
raise TypeError("This does not appear to be a valid WAD file, or is a boot2 WAD, which is not currently"
"supported by this library.")
# ====================================================================================
# Get the sizes of each data region contained within the WAD.
# ====================================================================================