From 84810cdbb0b99342bf49164bbeb9a71a169f2964 Mon Sep 17 00:00:00 2001
From: overcuriousity <overcuriousity@posteo.org>
Date: Sat, 13 Sep 2025 00:42:12 +0200
Subject: [PATCH] retreived scanner

---
 app.py                  | 1 -
 core/scanner.py         | 7 ++++---
 core/session_manager.py | 1 -
 3 files changed, 4 insertions(+), 5 deletions(-)

diff --git a/app.py b/app.py
index d50ef1b..2885167 100644
--- a/app.py
+++ b/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,
diff --git a/core/scanner.py b/core/scanner.py
index 7a0268d..3de0521 100644
--- a/core/scanner.py
+++ b/core/scanner.py
@@ -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:
diff --git a/core/session_manager.py b/core/session_manager.py
index 1ec0757..7d02866 100644
--- a/core/session_manager.py
+++ b/core/session_manager.py
@@ -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