feat: support scanning IP addresses

This commit is contained in:
Ahmed Allam
2025-11-14 01:35:27 +04:00
committed by Ahmed Allam
parent f1d2961779
commit 6abb53dc02
3 changed files with 23 additions and 4 deletions

View File

@@ -18,13 +18,14 @@ class StrixAgent(BaseAgent):
super().__init__(config)
async def execute_scan(self, scan_config: dict[str, Any]) -> dict[str, Any]:
async def execute_scan(self, scan_config: dict[str, Any]) -> dict[str, Any]: # noqa: PLR0912
user_instructions = scan_config.get("user_instructions", "")
targets = scan_config.get("targets", [])
repositories = []
local_code = []
urls = []
ip_addresses = []
for target in targets:
target_type = target["type"]
@@ -53,6 +54,8 @@ class StrixAgent(BaseAgent):
elif target_type == "web_application":
urls.append(details["target_url"])
elif target_type == "ip_address":
ip_addresses.append(details["target_ip"])
task_parts = []
@@ -74,6 +77,10 @@ class StrixAgent(BaseAgent):
task_parts.append("\n\nURLs:")
task_parts.extend(f"- {url}" for url in urls)
if ip_addresses:
task_parts.append("\n\nIP Addresses:")
task_parts.extend(f"- {ip}" for ip in ip_addresses)
task_description = " ".join(task_parts)
if user_instructions: