diff --git a/pzztest.py b/pzztest.py index b4e9f07..ecf6cda 100644 --- a/pzztest.py +++ b/pzztest.py @@ -1,12 +1,27 @@ import argparse -from pathlib import Path -import pzztool import hashlib import os +from pathlib import Path +import pzztool +import shutil TPL_MAGIC_FILE = b"\x00\x20\xAF\x30" # http://virtualre.rf.gd/index.php/TPL_(Format_de_fichier) + + +# compare le sha256 de chaque PZZ des dossiers passés en argument +# -> affiche le nom de fichier en cas de différence +def verify_sha256(folder1: Path, folder2: Path): + invalid_files_count = 0 + for pzz_file_name in os.listdir(folder1): + with open(folder1 / pzz_file_name, "rb") as f1, open(folder2 / pzz_file_name, "rb") as f2: + if hashlib.sha256( f1.read() ).hexdigest() != hashlib.sha256( f2.read() ).hexdigest() : + print(f"ERROR - INVALID FILE : {pzz_file_name}") + invalid_files_count +=1 + print(f"Invalid files : {invalid_files_count}/{len(os.listdir(folder1))}") + + def get_argparser(): parser = argparse.ArgumentParser(description='TEST TOOL') parser.add_argument('input_path', metavar='INPUT', help='') @@ -15,17 +30,22 @@ def get_argparser(): group.add_argument('-tdc', '--test-decompress-compress', action='store_true', help="") group.add_argument('-tbup', '--test-batch-unpack-pack', action='store_true', help=""" -tbup source_pzz_folder - source_pzz_folder : put all pzz in this folder - pzzu : will be created with all unpacked pzz from pzz folder - pzz2 : will be created with all packed pzz from pzzu folder - then it will print the file name when file sha256 is different between source_pzz_folder and pzz2 folder""") + source_pzz_folder : put all pzz in this folder + pzzu : will be created with all unpacked pzz from pzz folder + pzz2 : will be created with all packed pzz from pzzu folder + print file_name when sha256 is different between source_pzz_folder and pzz2 folder""") + group.add_argument('-tbunpzzpzz', '--test-batch-unpzz-pzz', action='store_true', help=""" + -tbunpzzpzz source_pzz_folder + source_pzz_folder : put all pzz in this folder + pzzu : will be created with all unpzz pzz from pzz folder + pzz2 : will be created with all pzz(pzz_folder) from pzzu folder + print file_name when sha256 is different between source_pzz_folder and pzz2 folder""") group.add_argument('-tctplh', '--test-check-tpl-headers', action='store_true', help="-tctplh afs_data_folder : check all files headers in the afs_data and print those who have the tpl magicfile") group.add_argument('-td', '--test-decompress', action='store_true', help=""" pzz : put all pzz in this folder then tip "pzztool.py -td" decompressed_tpl_files : will be created with all decompressed files from pzzu having the tpl header The script will then check that tpls are correctly decompressed with their specific characteristics""") - return parser @@ -54,18 +74,34 @@ if __name__ == '__main__': for invalid in listofinvalid: print(invalid) elif args.test_batch_unpack_pack: - # compare le sha256 de chaque PZZ du dossier passé en argument et pzz2 puis affiche le nom de fichier en cas de différence print("# TEST : BATCH UNPACK PACK") - os.system(f"python pzztool.py -bu {args.input_path} pzzu -v") - os.system("python pzztool.py -bp pzzu pzz2 -v") - invalid_files_count = 0 - for pzz_file_name in os.listdir(p_input): - with open(p_input / pzz_file_name, "rb") as f1, open(f"pzz2/{pzz_file_name}", "rb") as f2: - if hashlib.sha256( f1.read() ).hexdigest() != hashlib.sha256( f2.read() ).hexdigest() : - print(f"ERROR - INVALID FILE : {pzz_file_name}") - invalid_files_count +=1 - print(f"Invalid files : {invalid_files_count}/{len(os.listdir(p_input))}") + os.system(f"python pzztool.py -bu {args.input_path} pzzu") + os.system("python pzztool.py -bp pzzu pzz2") + verify_sha256(p_input, Path("pzz2")) + elif args.test_batch_unpzz_pzz: + os.system(f"python pzztool.py -bunpzz {args.input_path} pzzu") + os.system("python pzztool.py -bpzz pzzu pzz2") + verify_sha256(p_input, Path("pzz2")) + + # Clean du dossier pzz2 généré par le script + shutil.rmtree("pzz2") + + """ + si pzz : U -> decomp / testé sur les fichiers car l'unpzz décompresse par défaut + si pzz : U -> comp / à tester + si pzz : C -> decomp / testé sur les fichiers car l'unpzz décompresse par défaut + si pzz : C -> comp / à tester + """ + # On parcours tous les dossiers : si U -> comp ; si C -> comp : compression de tous les fichiers peu importe le type + for pzz_folder in os.listdir("pzzu"): + for pzz_file_part_name in os.listdir("pzzu/"+pzz_folder): + # créé un nouveau fichier compressé, à côté de l'original + os.system(f"python pzztool.py -c pzzu/{pzz_folder}/{pzz_file_part_name}") + # supprime l'original + os.remove(f"pzzu/{pzz_folder}/{pzz_file_part_name}") + os.system("python pzztool.py -bpzz pzzu pzz2") + verify_sha256(p_input, Path("pzz2")) elif args.test_check_tpl_headers: # Démontre que SEUL les TPLs ont ce magicfile # TEST OK