diff --git a/gcmtest.py b/gcmtest.py index e952b70..2005843 100644 --- a/gcmtest.py +++ b/gcmtest.py @@ -1,11 +1,11 @@ #!/usr/bin/env python3 from pathlib import Path -import hashlib import os import shutil +from time import time -__version__ = "0.0.4" +__version__ = "0.0.5" __author__ = "rigodron, algoflash, GGLinnk" __license__ = "MIT" __status__ = "developpement" @@ -42,15 +42,26 @@ def print_paths_differences(folder1_paths:list, folder2_paths:list): print(path) -# compare two files sha256 -def compare_sha256(file1_path:Path, file2_path:Path): - return hashlib.sha256( file1_path.read_bytes() ).hexdigest() == hashlib.sha256( file2_path.read_bytes() ).hexdigest() +# compare two files +def compare_files(file1_path:Path, file2_path:Path): + CLUSTER_LEN = 131072 + with file1_path.open("rb") as file1, file2_path.open("rb") as file2: + # Init + bytes1 = file1.read(CLUSTER_LEN) + bytes2 = file2.read(CLUSTER_LEN) + while bytes1 or bytes2: # continue if bytes1 and bytes2 have a len > 0 + if bytes1 != bytes2: + return False + bytes1 = file1.read(CLUSTER_LEN) + bytes2 = file2.read(CLUSTER_LEN) + return True # compare two GCM # -> raise an exception if there is a difference -def verify_GCM_sha256(folder1: Path, folder2: Path): - folder1_file_count = len(list(folder1.glob("**/*"))) +def compare_GCM(folder1: Path, folder2: Path): + folder1_root_paths = list((folder1 / "root").glob("**/*")) + folder1_file_count = len(folder1_root_paths) print(f"compare \"{folder1}\" - \"{folder2}\" ({folder1_file_count} files)") if folder1_file_count == 0: raise Exception(f"ERROR - EMPTY FOLDER: {folder1}") @@ -58,24 +69,39 @@ def verify_GCM_sha256(folder1: Path, folder2: Path): len1 = len(folder1.parts) len2 = len(folder2.parts) # 1. Compare names of filesystems - folder1_paths = [Path(*path.parts[len1:]) for path in (folder1/"root").glob('**/*')] - folder2_paths = [Path(*path.parts[len2:]) for path in (folder2/"root").glob('**/*')] + folder1_paths = [Path(*path.parts[len1:]) for path in folder1_root_paths] + folder2_paths = [Path(*path.parts[len2:]) for path in (folder2 / "root").glob('**/*')] if folder1_paths != folder2_paths: print_paths_differences(folder1_paths, folder2_paths) raise Exception(f"Folders \"{folder1}\" and \"{folder2}\" are different (not the same folders or files names).") # 2. Compare sys files content - if not compare_sha256(folder1/"sys"/"apploader.img", folder2/"sys"/"apploader.img"): + if not compare_files(folder1/"sys"/"apploader.img", folder2/"sys"/"apploader.img"): raise Exception(f"\"{folder1}/sys/apploader.bin\" and \"{folder2}/sys/apploader.bin\" are different.") - if not compare_sha256(folder1/"sys"/"boot.dol", folder2/"sys"/"boot.dol"): + if not compare_files(folder1/"sys"/"boot.dol", folder2/"sys"/"boot.dol"): raise Exception(f"\"{folder1}/sys/boot.bin\" and \"{folder2}/sys/boot.bin\" are different.") - for path1 in (folder1/"root").glob('**/*'): + for path1 in folder1_root_paths: if path1.is_file(): path2 = folder2/Path(*path1.parts[len1:]) - if not compare_sha256(path1, path2): + if not compare_files(path1, path2): raise Exception(f"\"{path1}/\" and \"{path2}\" are different.") +################################################## +# gcmtool.py commands wrappers +################################################## +def gcmtool_unpack(iso_path:Path, folder_path:Path): + if os.system(f"python gcmtool.py -u \"{iso_path}\" \"{folder_path}\"") != 0: + raise Exception("Error while unpacking gcm.") +def gcmtool_pack(folder_path:Path, iso_path:Path): + if os.system(f"python gcmtool.py -p \"{folder_path}\" \"{iso_path}\"") != 0: + raise Exception("Error while packing gcm.") +def gcmtool_rebuild_fst(folder_path:Path): + if os.system(f"python gcmtool.py -r \"{folder_path}\"") != 0: + raise Exception("Error while rebuilding FST.") + + +start = time() print("###############################################################################") print("# Checking tests folder") print("###############################################################################") @@ -91,12 +117,11 @@ print("######################################################################### unpack_path.mkdir(parents=True) for iso_path in roms_path.glob("*"): if iso_path.is_file(): - if os.system(f"python gcmtool.py -u \"{iso_path}\" \"{unpack_path}/{iso_path.name}\"") != 0: - raise Exception("Error while unpacking original gcm.") + gcmtool_unpack(iso_path, unpack_path / iso_path.name) # compare unpack_path dolphin_unpack_path for folder_path in unpack_path.glob("*"): - verify_GCM_sha256(folder_path, dolphin_unpack_path / folder_path.name) + compare_GCM(folder_path, dolphin_unpack_path / folder_path.name) print("###############################################################################") print("# TEST 2/3") @@ -104,22 +129,21 @@ print("# Comparing [unpack_path]->pack->unpack->[unpack2_path]") print("###############################################################################") # repack unpack_path repack_path repack_path.mkdir() -for folder_path in unpack_path.glob("*"): - if os.system(f"python gcmtool.py -p \"{folder_path}\" \"{repack_path}/{folder_path.name}\"") != 0: - raise Exception("Error while packing unpacked gcm.") +unpack_paths = list(unpack_path.glob("*")) +for folder_path in unpack_paths: + gcmtool_pack(folder_path, repack_path / folder_path.name) # unpack repack_path unpack2_path unpack2_path.mkdir() for iso_path in repack_path.glob("*"): - if os.system(f"python gcmtool.py -u \"{iso_path}\" \"{unpack2_path}/{iso_path.name}\"") != 0: - raise Exception("Error while unpacking repacked gcm.") + gcmtool_unpack(iso_path, unpack2_path / iso_path.name) # remove repack_path shutil.rmtree(repack_path) # compare unpack_path unpack2_path -for folder_path in unpack_path.glob("*"): - verify_GCM_sha256(folder_path, unpack2_path / folder_path.name) +for folder_path in unpack_paths: + compare_GCM(folder_path, unpack2_path / folder_path.name) # remove unpack2_path shutil.rmtree(unpack2_path) @@ -129,28 +153,26 @@ print("# TEST 3/3") print("# Comparing [unpack_path]->rebuild_fst->repack->unpack->[unpack2_path]") print("###############################################################################") # rebuild unpack_path FSTs -for folder_path in unpack_path.glob("*"): - if os.system(f"python gcmtool.py -r \"{folder_path}\"") != 0: - raise Exception("Error while rebuilding FST.") +unpack_paths = list(unpack_path.glob("*")) +for folder_path in unpack_paths: + gcmtool_rebuild_fst(folder_path) # repack unpack_path repack_path repack_path.mkdir() -for folder_path in unpack_path.glob("*"): - if os.system(f"python gcmtool.py -p \"{folder_path}\" \"{repack_path}/{folder_path.name}\"") != 0: - raise Exception("Error while packing gcm with new FST.") +for folder_path in unpack_paths: + gcmtool_pack(folder_path, repack_path / folder_path.name) # unpack repack_path unpack2_path unpack2_path.mkdir() for iso_path in repack_path.glob("*"): - if os.system(f"python gcmtool.py -u \"{iso_path}\" \"{unpack2_path}/{iso_path.name}\"") != 0: - raise Exception("Error while unpacking repacked gcm with new FST.") + gcmtool_unpack(iso_path, unpack2_path / iso_path.name) # remove repack_path shutil.rmtree(repack_path) # compare unpack_path unpack2_path for folder_path in unpack_path.glob("*"): - verify_GCM_sha256(folder_path, unpack2_path / folder_path.name) + compare_GCM(folder_path, unpack2_path / folder_path.name) # Remove tests folders print("###############################################################################") @@ -159,6 +181,7 @@ print("######################################################################### shutil.rmtree(unpack_path) shutil.rmtree(unpack2_path) +end = time() print("###############################################################################") -print("# All tests are OK") +print(f"# All tests are OK - elapsed time : {end - start}") print("###############################################################################")