fix: fix pickle and pyc inject code

tests/final_tests_util.py

@@ -145,28 +145,37 @@ backdoors = [
 ]
 
 
-def inject_pickle_backdoor(path: str) -> None:
+def inject_pickle_backdoor(root_path: str) -> None:
     """
     Generate a pickle backdoor and insert it into the specified path.
 
     Args:
         path (str): The path to the repository to insert the backdoor into.
     """
-    for i, backdoor in enumerate(backdoors):
+    all_path = [str(p) for p in Path(root_path).glob("*") if p.is_dir()]
-        filename = os.path.join(path, f"backdoor{i}.pickle")
+    paths = random.sample(all_path, random.randrange(1, len(all_path)))
+    for path in paths:
+        backdoor_id = random.randrange(0, len(backdoors))
+        backdoor = backdoors[backdoor_id]
+        filename = os.path.join(path, f"backdoor{backdoor_id}.pickle")
         with open(filename, "wb") as f:
             pickle.dump(backdoor, f)
 
 
-def inject_pyc_backdoor(path: str) -> None:
+def inject_pyc_backdoor(root_path: str) -> None:
     """
     Generate a pyc backdoor and insert it into the specified path.
 
     Args:
         path (str): The path to the repository to insert the backdoor into.
     """
-    for i, backdoor in enumerate(backdoors):
+    all_path = [str(p) for p in Path(root_path).glob("*") if p.is_dir()]
-        filename = os.path.join(path, f"backdoor{i}.pyc")
+    paths = random.sample(all_path, random.randrange(1, len(all_path)))
+
+    for path in paths:
+        backdoor_id = random.randrange(0, len(backdoors))
+        backdoor = backdoors[backdoor_id]
+        filename = os.path.join(path, f"backdoor{backdoor_id}.pyc")
 
         # Compile the string to a code object
         code = compile(backdoor, filename, "exec")