Virtual-World-RE/NeoGF
4
0
Fork 0
mirror of https://github.com/Virtual-World-RE/NeoGF.git synced 2024-11-15 12:45:34 +01:00
Code Issues Releases Wiki Activity

Update gcmtool.py

Browse Source 
added correct folder handling when using pack or unpack
This commit is contained in:
tmpz23 2021-12-19 17:22:28 +01:00 committed by GitHub
parent 18870c2478
commit 87a0c8c0df
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 13 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
gcmtool.py
View File

@ -13,7 +13,7 @@ DVD_MAGIC = b"\xC2\x33\x9F\x3D"
FST_TYPE_FILE = 0 FST_TYPE_FILE = 0
FST_TYPE_DIR = 1 FST_TYPE_DIR = 1
# BUG TO FIX : implement dest folder/file handling when given to unpack or pack.
###################################################################### ######################################################################
# Todo : add logging ; add extension check ; add --verbose ; add --disable-ignore # Todo : add logging ; add extension check ; add --verbose ; add --disable-ignore
# -> test it on random iso and check that it's the same than dolphin extract # -> test it on random iso and check that it's the same than dolphin extract
@ -33,7 +33,7 @@ class Dol:
# https://sudonull.com/post/68549-Gamecube-file-system-device # https://sudonull.com/post/68549-Gamecube-file-system-device
class GCM: class GCM:
def unpack(self, iso_path:Path): def unpack(self, iso_path:Path, folder_path:Path):
with iso_path.open("rb") as iso_file: with iso_path.open("rb") as iso_file:
bootbin_data = iso_file.read(0x440) bootbin_data = iso_file.read(0x440)
if bootbin_data[0x1c:0x20] != DVD_MAGIC: if bootbin_data[0x1c:0x20] != DVD_MAGIC:
@ -52,8 +52,8 @@ class GCM:
iso_file.seek(0x2440) iso_file.seek(0x2440)
apploaderimg_data = iso_file.read(apploader_size) apploaderimg_data = iso_file.read(apploader_size)
print( 0x2440 + apploader_size ) #print( 0x2440 + apploader_size )
print("fst.bin :", int.from_bytes(bootbin_data[0x424:0x428],"big", signed=False), int.from_bytes(bootbin_data[0x428:0x42c],"big", signed=False)) #print("fst.bin :", int.from_bytes(bootbin_data[0x424:0x428],"big", signed=False), int.from_bytes(bootbin_data[0x428:0x42c],"big", signed=False))
iso_file.seek( int.from_bytes(bootbin_data[0x424:0x428],"big", signed=False) ) iso_file.seek( int.from_bytes(bootbin_data[0x424:0x428],"big", signed=False) )
fstbin_data = iso_file.read( int.from_bytes(bootbin_data[0x428:0x42c],"big", signed=False) ) fstbin_data = iso_file.read( int.from_bytes(bootbin_data[0x428:0x42c],"big", signed=False) )
@ -63,7 +63,10 @@ class GCM:
dol = Dol() dol = Dol()
dolheader_data = iso_file.read(0x100) dolheader_data = iso_file.read(0x100)
bootdol_data = dolheader_data + iso_file.read( dol.getDolLen( dolheader_data ) - 0x100 ) bootdol_data = dolheader_data + iso_file.read( dol.getDolLen( dolheader_data ) - 0x100 )
print("dol :", dol_offset, dol.getDolLen( dolheader_data )) #print("dol :", dol_offset, dol.getDolLen( dolheader_data ))
if folder_path != Path("."):
base_path = folder_path
else:
base_path = Path(f"{bootbin_data[:4].decode('utf-8')}-{int.from_bytes(bootbin_data[6:7], 'little', signed=False):02}") base_path = Path(f"{bootbin_data[:4].decode('utf-8')}-{int.from_bytes(bootbin_data[6:7], 'little', signed=False):02}")
sys_path = base_path / "sys" sys_path = base_path / "sys"
sys_path.mkdir(parents=True, exist_ok=True) sys_path.mkdir(parents=True, exist_ok=True)
@ -116,7 +119,9 @@ class GCM:
with (currentdir_path / name).open("wb") as new_file: with (currentdir_path / name).open("wb") as new_file:
iso_file.seek(fileoffset) iso_file.seek(fileoffset)
new_file.write( iso_file.read(filesize) ) new_file.write( iso_file.read(filesize) )
print("file :", fileoffset, filesize) #print("file :", fileoffset, filesize)
def pack(self, folder_path:Path, iso_path:Path = None): def pack(self, folder_path:Path, iso_path:Path = None):
if iso_path == None: if iso_path == None:
iso_path = folder_path.parent / Path(folder_path.name).with_suffix(".iso") iso_path = folder_path.parent / Path(folder_path.name).with_suffix(".iso")
@ -174,14 +179,13 @@ class GCM:
else: else:
fileoffset = int.from_bytes(fstbin_data[i+4:i+8], "big", signed=False) fileoffset = int.from_bytes(fstbin_data[i+4:i+8], "big", signed=False)
filesize = int.from_bytes(fstbin_data[i+8:i+12], "big", signed=False) filesize = int.from_bytes(fstbin_data[i+8:i+12], "big", signed=False)
print(fileoffset, filesize) #print(fileoffset, filesize)
with (currentdir_path / name).open("rb") as new_file: with (currentdir_path / name).open("rb") as new_file:
iso_file.seek(fileoffset) iso_file.seek(fileoffset)
iso_file.write( new_file.read(filesize) ) iso_file.write( new_file.read(filesize) )
def get_argparser(): def get_argparser():
import argparse import argparse
parser = argparse.ArgumentParser(description='ISO/GCM packer & unpacker - [GameCube] v' + __version__) parser = argparse.ArgumentParser(description='ISO/GCM packer & unpacker - [GameCube] v' + __version__)
@ -210,6 +214,4 @@ if __name__ == '__main__':
gcm.pack( p_input, p_output ) gcm.pack( p_input, p_output )
elif args.unpack: elif args.unpack:
logging.info("### Unpack") logging.info("### Unpack")
if(p_output == Path(".")):
p_output = Path(p_input.name)
gcm.unpack( p_input, p_output ) gcm.unpack( p_input, p_output )