retreived scanner

This commit is contained in:
overcuriousity 2025-09-13 00:42:12 +02:00
parent d36fb7d814
commit 84810cdbb0
3 changed files with 4 additions and 5 deletions

1
app.py
View File

@ -387,7 +387,6 @@ def get_providers():
'requires_api_key': provider_name in ['shodan', 'virustotal'] 'requires_api_key': provider_name in ['shodan', 'virustotal']
} }
print(f"Returning provider info for session {user_session_id}: {list(provider_info.keys())}")
return jsonify({ return jsonify({
'success': True, 'success': True,
'providers': provider_info, 'providers': provider_info,

View File

@ -14,6 +14,7 @@ from utils.helpers import _is_valid_ip, _is_valid_domain
from providers.crtsh_provider import CrtShProvider from providers.crtsh_provider import CrtShProvider
from providers.dns_provider import DNSProvider from providers.dns_provider import DNSProvider
from providers.shodan_provider import ShodanProvider from providers.shodan_provider import ShodanProvider
from providers.virustotal_provider import VirusTotalProvider
class ScanStatus: class ScanStatus:
@ -66,7 +67,7 @@ class Scanner:
'dns': {'domains': True, 'ips': True}, 'dns': {'domains': True, 'ips': True},
'crtsh': {'domains': True, 'ips': False}, 'crtsh': {'domains': True, 'ips': False},
'shodan': {'domains': True, 'ips': True}, 'shodan': {'domains': True, 'ips': True},
'virustotal': {'domains': False, 'ips': False} # Disabled as requested 'virustotal': {'domains': True, 'ips': True}
} }
# Initialize providers with session config # Initialize providers with session config
@ -169,7 +170,7 @@ class Scanner:
'dns': DNSProvider, 'dns': DNSProvider,
'crtsh': CrtShProvider, 'crtsh': CrtShProvider,
'shodan': ShodanProvider, 'shodan': ShodanProvider,
# Skip virustotal as requested 'virustotal': VirusTotalProvider
} }
for provider_name, provider_class in provider_classes.items(): for provider_name, provider_class in provider_classes.items():
@ -662,7 +663,7 @@ class Scanner:
# Track failed state # Track failed state
self._update_provider_state(target, provider_name, 'failed', 0, str(e), start_time) self._update_provider_state(target, provider_name, 'failed', 0, str(e), start_time)
print(f"{provider_name} failed for {target}: {e}") print(f"{provider_name} failed for {target}: {e}")
raise return []
def _update_provider_state(self, target: str, provider_name: str, status: str, def _update_provider_state(self, target: str, provider_name: str, status: str,
results_count: int, error: str, start_time: datetime) -> None: results_count: int, error: str, start_time: datetime) -> None:

View File

@ -268,7 +268,6 @@ class SessionManager:
if scanner: if scanner:
# Ensure the scanner can check the Redis-based stop signal # Ensure the scanner can check the Redis-based stop signal
scanner.session_id = session_id scanner.session_id = session_id
print(f"Retrieved scanner for session {session_id} (status: {scanner.status})")
return scanner return scanner