retreived scanner
This commit is contained in:
		
							parent
							
								
									d36fb7d814
								
							
						
					
					
						commit
						84810cdbb0
					
				
							
								
								
									
										1
									
								
								app.py
									
									
									
									
									
								
							
							
						
						
									
										1
									
								
								app.py
									
									
									
									
									
								
							@ -387,7 +387,6 @@ def get_providers():
 | 
			
		||||
                'requires_api_key': provider_name in ['shodan', 'virustotal']
 | 
			
		||||
            }
 | 
			
		||||
        
 | 
			
		||||
        print(f"Returning provider info for session {user_session_id}: {list(provider_info.keys())}")
 | 
			
		||||
        return jsonify({
 | 
			
		||||
            'success': True,
 | 
			
		||||
            'providers': provider_info,
 | 
			
		||||
 | 
			
		||||
@ -14,6 +14,7 @@ from utils.helpers import _is_valid_ip, _is_valid_domain
 | 
			
		||||
from providers.crtsh_provider import CrtShProvider
 | 
			
		||||
from providers.dns_provider import DNSProvider
 | 
			
		||||
from providers.shodan_provider import ShodanProvider
 | 
			
		||||
from providers.virustotal_provider import VirusTotalProvider
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class ScanStatus:
 | 
			
		||||
@ -66,7 +67,7 @@ class Scanner:
 | 
			
		||||
                'dns': {'domains': True, 'ips': True},
 | 
			
		||||
                'crtsh': {'domains': True, 'ips': False},
 | 
			
		||||
                'shodan': {'domains': True, 'ips': True},
 | 
			
		||||
                'virustotal': {'domains': False, 'ips': False}  # Disabled as requested
 | 
			
		||||
                'virustotal': {'domains': True, 'ips': True}
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            # Initialize providers with session config
 | 
			
		||||
@ -169,7 +170,7 @@ class Scanner:
 | 
			
		||||
            'dns': DNSProvider,
 | 
			
		||||
            'crtsh': CrtShProvider,
 | 
			
		||||
            'shodan': ShodanProvider,
 | 
			
		||||
            # Skip virustotal as requested
 | 
			
		||||
            'virustotal': VirusTotalProvider
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        for provider_name, provider_class in provider_classes.items():
 | 
			
		||||
@ -662,7 +663,7 @@ class Scanner:
 | 
			
		||||
            # Track failed state
 | 
			
		||||
            self._update_provider_state(target, provider_name, 'failed', 0, str(e), start_time)
 | 
			
		||||
            print(f"✗ {provider_name} failed for {target}: {e}")
 | 
			
		||||
            raise
 | 
			
		||||
            return []
 | 
			
		||||
 | 
			
		||||
    def _update_provider_state(self, target: str, provider_name: str, status: str, 
 | 
			
		||||
                              results_count: int, error: str, start_time: datetime) -> None:
 | 
			
		||||
 | 
			
		||||
@ -268,7 +268,6 @@ class SessionManager:
 | 
			
		||||
        if scanner:
 | 
			
		||||
            # Ensure the scanner can check the Redis-based stop signal
 | 
			
		||||
            scanner.session_id = session_id
 | 
			
		||||
            print(f"Retrieved scanner for session {session_id} (status: {scanner.status})")
 | 
			
		||||
        
 | 
			
		||||
        return scanner
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
		Loading…
	
	
			
			x
			
			
		
	
		Reference in New Issue
	
	Block a user