moriarty-project 0.1.26__py3-none-any.whl → 0.1.27__py3-none-any.whl

moriarty/modules/domain_scanner.py

@@ -76,6 +76,7 @@ class DomainScanner:
         stealth_level: int = 0,
         threads: int = 10,
         timeout: int = 30,
+        ports_profile: str = "quick",
         verbose: bool = False,
     ):
         self.target = target
@@ -83,6 +84,7 @@ class DomainScanner:
         self.stealth_level = stealth_level
         self.threads = threads
         self.timeout = timeout
+        self.ports_profile = ports_profile.lower()
         self.verbose = verbose
         self.result = ScanResult(target=target)
         self.stealth = None
@@ -257,8 +259,11 @@ class DomainScanner:
         try:
             from moriarty.modules.port_scanner_nmap import PortScanner
 
-            # Define o perfil baseado no timeout
-            profile = "extended" if self.timeout > 45 else "quick"
+            # Usa o perfil de portas fornecido ou define baseado no timeout se não especificado
+            profile = self.ports_profile if hasattr(self, 'ports_profile') else ("extended" if self.timeout > 45 else "quick")
+            
+            # Log do perfil que será usado
+            console.print(f"  🔧 Usando perfil de portas: [bold]{profile}[/]")
             
             # Cria o scanner com as configurações apropriadas
             scanner = PortScanner(
@@ -272,7 +277,7 @@ class DomainScanner:
             results = await scanner.scan()
             
             if not results:
-                console.print("ℹ️  Nenhuma porta aberta encontrada.")
+                console.print("  ℹ️  Nenhuma porta aberta encontrada.")
                 self.result.port_details = []
                 self.result.open_ports = []
                 return
@@ -395,24 +400,24 @@ class DomainScanner:
         """Executa crawler leve para inventário de rotas."""
         console.print("\n[bold red]▶ Web Crawler[/bold red]")
         try:
-            console.print("🔍 Iniciando configuração do Web Crawler...")
+            console.print("  🔍 Iniciando configuração do Web Crawler...")
             from moriarty.modules.web_crawler import WebCrawler
 
             base_url = self._default_base_url()
-            console.print(f"🌐 URL base: {base_url}")
+            console.print(f"  🌐 URL base: {base_url}")
             
             # Configurações do crawler
             max_pages = max(50, self.threads * 10)
             max_depth = 2
             concurrency = max(5, self.threads)
             
-            console.print(f"⚙️  Configurações: max_pages={max_pages}, max_depth={max_depth}, concurrency={concurrency}")
+            console.print(f"  ⚙️  Configurações: max_pages={max_pages}, max_depth={max_depth}, concurrency={concurrency}")
             
             try:
-                console.print("🚀 Iniciando crawler...")
+                console.print("  🚀 Iniciando crawler...")
                 try:
                     # Log de debug: Verificando se o WebCrawler pode ser instanciado
-                    console.print("🔧 Instanciando WebCrawler...")
+                    console.print("  🔧 Instanciando WebCrawler...")
                     crawler = WebCrawler(
                         base_url=base_url,
                         max_pages=max_pages,
@@ -421,16 +426,16 @@ class DomainScanner:
                         follow_subdomains=False,
                         stealth=self.stealth,
                     )
-                    console.print("✅ WebCrawler instanciado com sucesso!")
+                    console.print("  ✅ WebCrawler instanciado com sucesso!")
                 except Exception as e:
-                    console.print(f"❌ [red]Erro ao instanciar WebCrawler: {str(e)}[/red]")
+                    console.print(f"  ❌ [red]Erro ao instanciar WebCrawler: {str(e)}[/red]")
                     logger.error("webcrawler.init_error", error=str(e), exc_info=True)
                     raise
                 
                 try:
-                    console.print("🔄 Executando varredura...")
+                    console.print("  🔄 Executando varredura...")
                     pages = await crawler.crawl()
-                    console.print(f"✅ Varredura concluída! {len(pages)} páginas encontradas.")
+                    console.print(f"  ✅ Varredura concluída! {len(pages)} páginas encontradas.")
                     
                     # Log detalhado das páginas encontradas
                     for i, (url, page) in enumerate(pages.items(), 1):
@@ -438,7 +443,7 @@ class DomainScanner:
                         if page.error:
                             console.print(f"    [red]Erro: {page.error}[/red]")
                 except Exception as e:
-                    console.print(f"❌ [red]Erro durante o crawler: {str(e)}[/red]")
+                    console.print(f"  ❌ [red]Erro durante o crawler: {str(e)}[/red]")
                     logger.error("webcrawler.crawl_error", error=str(e), exc_info=True)
                     raise
                 
@@ -452,15 +457,15 @@ class DomainScanner:
                         }
                         for url, page in pages.items()
                     }
-                    console.print("📊 Dados do crawl processados com sucesso!")
+                    console.print("  📊 Dados do crawl processados com sucesso!")
                 except Exception as e:
-                    console.print(f"❌ [red]Erro ao processar resultados do crawl: {str(e)}[/red]")
+                    console.print(f"  ❌ [red]Erro ao processar resultados do crawl: {str(e)}[/red]")
                     logger.error("webcrawler.process_error", error=str(e), exc_info=True)
                     raise
 
                 try:
                     extracted = self._extract_targets_from_crawl(pages)
-                    console.print(f"🔗 {len(extracted)} alvos extraídos para fuzzing")
+                    console.print(f"  🔗 {len(extracted)} alvos extraídos para fuzzing")
                     
                     for definition in extracted:
                         self._register_web_target(