Merge pull request #557 from AutomationSolutionz/Security_Actions_Issue

Security Action File Generation Issue for windows

Author: Antu7

Framework/Built_In_Automation/Security/BuiltInFunctions.py

@@ -56,13 +56,19 @@ def port_scaning_nmap(data_set: list) -> str:
 def server_scaning_wapiti(data_set: list) -> str:
     target = next(item[2] for item in data_set if item[0] == 'wapiti')
     wapiti_action = next(item[2] for item in data_set if item[0] == 'verbosity')
+
     if not target.startswith(("http://", "https://")):
         target = "http://" + target
+
     command = ["wapiti", f"-v={wapiti_action}", "-u", target]
 
+    # Set UTF-8 encoding
+    env = os.environ.copy()
+    env["PYTHONUTF8"] = "1"
+
     try:
         print(command)
-        result = subprocess.run(command, capture_output=True, text=True, check=True)
+        result = subprocess.run(command, capture_output=True, text=True, check=True, env=env)
         print("Command Output:", result.stdout)
 
         for line in result.stdout.splitlines():
@@ -76,7 +82,7 @@ def server_scaning_wapiti(data_set: list) -> str:
         security_report_dir = Path(ConfigModule.get_config_value("sectionOne", "test_case_folder", temp_config)) / 'security_report' / 'wapiti'
         os.makedirs(security_report_dir, exist_ok=True)
         destination_path = security_report_dir / os.path.basename(report_path)
-        # Move the report file to the new location
+
         shutil.move(report_path, destination_path)
         print(f"Report moved to {destination_path}")
 
@@ -101,7 +107,6 @@ def server_scaning_arachni(data_set: list) -> str:
         print("***** Arachni setup failed. *****")
         return "zeuz_failed"
 
-
 def server_scaning_nikto(data_set: list) -> str:
     """
     Runs Nikto scan if Perl is installed, otherwise provides installation instructions based on the platform.

Framework/Built_In_Automation/Security/nmap_scan.py

@@ -14,8 +14,8 @@ def run_nmap(ip, output_dir=None):
     xml_output_file = os.path.join(output_dir, f"nmap_scan_{ip}.xml")
     normal_output_file = os.path.join(output_dir, f"nmap_scan_{ip}.txt")
 
-    command = f"nmap -sV --script vuln -oX {xml_output_file} -oN {normal_output_file} {ip}"
-    process = subprocess.Popen(command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+    process = subprocess.Popen(["nmap", "-sV", "--script", "vuln", "-oX", xml_output_file, "-oN", normal_output_file, ip], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+
     spinner = ['|', '/', '-', '\\']
     start_time = datetime.now()
     scanning = True

node_cli.py

@@ -147,7 +147,7 @@ def main():
 
     kill_old_process(Path.cwd().parent / "pid.txt")
     check_min_python_version(min_python_version="3.11", show_warning=True)
-    install_missing_modules()
+    update_outdated_modules()
     monkeypatch_fromisoformat()
     start_server()