From 1c3f9d02050690e3c90ffe4ed2d41fa2cf797713 Mon Sep 17 00:00:00 2001
From: tmpz23 <28760271+tmpz23@users.noreply.github.com>
Date: Sun, 28 Nov 2021 09:38:36 +0100
Subject: [PATCH] Update pzztest.py

---
 pzztest.py | 36 +++++++++++++++++++++++++++++++++++-
 1 file changed, 35 insertions(+), 1 deletion(-)

diff --git a/pzztest.py b/pzztest.py
index af16106..14cff10 100644
--- a/pzztest.py
+++ b/pzztest.py
@@ -20,10 +20,19 @@ def verify_sha256(folder1: Path, folder2: Path):
                 invalid_files_count +=1
     print(f"Invalid files : {invalid_files_count}/{len(os.listdir(folder1))}")
 
+# compare le sha256 des deux fichiers passés en argument
+#     -> affiche le nom de fichier en cas de différence
+def verify_sha256_2(file1: Path, file2: Path):
+    with file1.open("rb") as f1, file2.open("rb") as f2:
+        if hashlib.sha256( f1.read() ).hexdigest() != hashlib.sha256( f2.read() ).hexdigest() :
+            return False
+    return True
+
 
 def get_argparser():
     parser = argparse.ArgumentParser(description='TEST TOOL')
     parser.add_argument('input_path',  metavar='INPUT',  help='')
+    parser.add_argument('output_path', metavar='OUTPUT', help='', nargs='?', default="")
 
     group = parser.add_mutually_exclusive_group(required=True)
     group.add_argument('-tdc',  '--test-decompress-compress',   action='store_true', help="")
@@ -44,6 +53,7 @@ def get_argparser():
         pzz : put all pzz in this folder
         then tip "pzztool.py -tcd pzz"
         The script will then check that tpls are correctly decompressed with their specific characteristics""")
+    group.add_argument('-tcp', '--test-compare-position',   action='store_true', help="compare plxxxx.pzz subfiles with plxxxx files inside afs_data.afs")
     return parser
 
 
@@ -123,4 +133,28 @@ if __name__ == '__main__':
                     print(f"Invalid TPL file length modulo 32 ({Path(p).stat().st_size % 32}) - {p}")
                     invalid_files_count += 1
         print(f"Invalid files : {invalid_files_count}/{total}")
- 
+    elif args.test_compare_position:
+        # FULL_AFS_FILE_DUMP contient tous les fichiers de l'afs_data.afs et pzzu le résultat de pzztool.py -bunpzz sur l'ensemble des pzz
+        # Comparaisons à effectuer :
+        # pzztest.py -tcp 0 data.bin
+        #     Les fichiers de l'afs_data sont parfois data2 / data3 ou absents
+        # pzztest.py -tcp 2 hit.bin
+        # pzztest.py -tcp 3 mot.bin
+        # pzztest.py -tcp 4 _mdl.arc
+        # pzztest.py -tcp 5 b_mdl.arc
+        # pzztest.py -tcp 6 g_mdl.arc
+        # pzztest.py -tcp 7 s_mdl.arc
+        # pzztest.py -tcp 8 c_mdl.arc
+        # pzztest.py -tcp 9 k_mdl.arc
+
+        for pzzpart_path in Path("pzzu").glob("**/00"+args.input_path+"*"):
+            file_path = Path("FULL_AFS_FILE_DUMP/"+pzzpart_path.parent.name+args.output_path)
+
+            if pzzpart_path.parent.name[:2] == "pl":
+                if not file_path.is_file():
+                    print(f"File doesn't exist : {file_path}")
+                elif pzzpart_path.stat().st_size == 0:
+                    print(f"File is empty : {pzzpart_path}")
+                else:
+                    if not verify_sha256_2(pzzpart_path, file_path):
+                        print(f"DIFFERENCE : {pzzpart_path} - {file_path}")