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']
|
'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,
|
||||||
|
@ -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:
|
||||||
|
@ -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
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user