UI: fix PMKID passive UI support

Author: kimocoder

wifite/attack/pmkid.py

@@ -27,7 +27,7 @@ def __init__(self, target):
         self.pcapng_file = Configuration.temp('pmkid.pcapng')
         self.success = False
         self.timer = None
-        
+
         # Initialize TUI view if in TUI mode
         self.view = None
         if OutputManager.is_tui_mode():
@@ -137,7 +137,7 @@ def run_hashcat(self):
                 Color.pl('{+} {D}Checking for existing PMKID for BSSID: {C}%s{W}' % self.target.bssid)
             if self.view:
                 self.view.add_log("Checking for existing PMKID hash...")
-            
+
             pmkid_file = AttackPMKID.get_existing_pmkid_file(self.target.bssid)
             if pmkid_file is not None:
                 if self.view:
@@ -169,7 +169,7 @@ def run_hashcat(self):
         if WpaSecUploader.should_upload():
             if self.view:
                 self.view.add_log("Checking wpa-sec upload configuration...")
-            
+
             # Use the pcapng file if it exists, otherwise skip upload
             # Note: If only .22000 hash file exists, that's fine - wpa-sec doesn't accept hash files anyway
             if os.path.exists(self.pcapng_file):
@@ -187,8 +187,8 @@ def run_hashcat(self):
             if self.view:
                 self.view.add_log("Skipping crack phase (--skip-crack flag)")
             return self._extracted_from_run_hashcat_44(
-                '{+} Not cracking pmkid because {C}skip-crack{W} was used{W}'
-            )
+                '{+} Not cracking pmkid because {C}skip-crack{W} was used{W}')
+
         # Crack it.
         if Process.exists(Hashcat.dependency_name):
             try:
@@ -217,15 +217,15 @@ def run(self):
         if self.view:
             self.view.start()
             self.view.set_attack_type("PMKID Attack")
-            
+
             # Handle hidden ESSID
             essid_display = self.target.essid if self.target.essid else "<hidden ESSID>"
             self.view.add_log(f"Starting PMKID attack on {essid_display} ({self.target.bssid})")
-            
+
             # Handle invalid channel
             channel_display = self.target.channel if self.target.channel and str(self.target.channel) != '-1' else "unknown"
             self.view.add_log(f"Channel: {channel_display}")
-        
+
         if self.do_airCRACK:
             self.run_aircrack()
         else:
@@ -379,7 +379,7 @@ def capture_pmkid(self):
                 elapsed = Configuration.pmkid_timeout - self.timer.remaining()
                 self.view.update_pmkid_status(False, attempts)
                 self.view.add_log(f"Waiting for PMKID... ({int(elapsed)}s / {Configuration.pmkid_timeout}s)")
-            
+
             Color.pattack('PMKID', self.target, 'CAPTURE', 'Waiting for PMKID ({C}%s{W})' % str(self.timer))
             time.sleep(1)
 
@@ -397,7 +397,7 @@ def capture_pmkid(self):
         if self.view:
             self.view.update_pmkid_status(True, attempts)
             self.view.add_log("Successfully captured PMKID!")
-        
+
         Color.clear_entire_line()
         Color.pattack('PMKID', self.target, 'CAPTURE', '{G}Captured PMKID{W}')
         return self.save_pmkid(pmkid_hash)
@@ -424,7 +424,7 @@ def crack_pmkid_file(self, pmkid_file):
                 self.view.set_attack_type("PMKID Crack")
                 self.view.add_log(f"Starting PMKID crack with wordlist: {Configuration.wordlist}")
                 self.view.add_log("Running hashcat...")
-            
+
             Color.clear_entire_line()
             Color.pattack('PMKID', self.target, 'CRACK', 'Cracking PMKID using {C}%s{W} ...\n' % Configuration.wordlist)
             key = Hashcat.crack_pmkid(pmkid_file)
@@ -454,7 +454,7 @@ def _extracted_from_crack_pmkid_file_31(self, key, pmkid_file):
                     'Status': 'SUCCESS'
                 }
             })
-        
+
         Color.clear_entire_line()
         Color.pattack('PMKID', self.target, 'CRACKED', '{C}Key: {G}%s{W}' % key)
         self.crack_result = CrackResultPMKID(self.target.bssid, self.target.essid,

wifite/ui/attack_view.py

@@ -185,7 +185,7 @@ def _render_target_info(self) -> Panel:
         """
         # Create target info table
         table = Table(show_header=False, box=None, padding=(0, 1))
-        table.add_column("Label", style="bold cyan", width=12)
+        table.add_column("Label", style="bold white", width=12)
         table.add_column("Value", style="white")
 
         # ESSID
@@ -216,25 +216,25 @@ def _render_target_info(self) -> Panel:
 
         # Create a second column for encryption and power
         info_text = Text()
-        info_text.append("Encryption: ", style="bold cyan")
+        info_text.append("Encryption: ", style="bold white")
         info_text.append(enc_badge)
         info_text.append("  |  ", style="bright_black")
-        info_text.append("Power: ", style="bold cyan")
+        info_text.append("Power: ", style="bold white")
 
         # Power level
         power_dbm = self.target.power - 100 if self.target.power > 0 else self.target.power
         info_text.append(SignalStrengthBar.render(power_dbm))
 
         # Add resume indicator to title if this is a resumed session
-        title_text = f"[bold cyan]Target: {essid}[/bold cyan]"
+        title_text = f"[bold white]Target: {essid}[/bold white]"
         if self.session:
             title_text = f"[bold magenta]RESUMED[/bold magenta] | {title_text}"
 
         return Panel(
             table,
             title=title_text,
             subtitle=info_text,
-            border_style="cyan",
+            border_style="white",
             padding=(0, 1)
         )
 
@@ -288,7 +288,7 @@ def _render_footer(self) -> Panel:
 
         return Panel(
             Align.center(footer),
-            border_style="blue",
+            border_style="white",
             padding=(0, 1)
         )
 
@@ -713,7 +713,7 @@ def update_capture_status(self, networks_detected: int = None, pmkids_captured:
 
         # Build metrics
         metrics = {
-            'Networks Detected': f'[cyan]{self.networks_detected}[/cyan]',
+            'Networks Detected': f'[white]{self.networks_detected}[/white]',
             'PMKIDs Captured': f'[bold green]{self.pmkids_captured}[/bold green]' if self.pmkids_captured > 0 else f'[dim]{self.pmkids_captured}[/dim]',
             'Capture File Size': f'[yellow]{size_str}[/yellow]',
             'Last Extraction': f'[white]{extraction_str}[/white]',
@@ -785,7 +785,7 @@ def _render_target_info(self) -> Panel:
         """
         # For passive mode, show general capture info instead of specific target
         table = Table(show_header=False, box=None, padding=(0, 1))
-        table.add_column("Label", style="bold cyan", width=18)
+        table.add_column("Label", style="bold white", width=18)
         table.add_column("Value", style="white")
 
         table.add_row("Mode:", "Passive PMKID Capture")
@@ -805,14 +805,14 @@ def _render_target_info(self) -> Panel:
         else:
             table.add_row("Duration Limit:", "Infinite (Ctrl+C to stop)")
 
-        title_text = "[bold cyan]Passive PMKID Sniffing[/bold cyan]"
+        title_text = "[bold white]Passive PMKID Sniffing[/bold white]"
         if self.session:
             title_text = f"[bold magenta]RESUMED[/bold magenta] | {title_text}"
 
         return Panel(
             table,
             title=title_text,
-            border_style="cyan",
+            border_style="white",
             padding=(0, 1)
         )
 
@@ -1247,7 +1247,7 @@ def _update_display(self):
                 # Calculate deauths per minute
                 elapsed = time.time() - self.attack_start_time
                 deauths_per_min = (self.deauths_sent / elapsed * 60) if elapsed > 0 else 0
-                metrics['Deauths Sent'] = f'[cyan]{self.deauths_sent:,}[/cyan] ([dim]{deauths_per_min:.1f}/min[/dim])'
+                metrics['Deauths Sent'] = f'[white]{self.deauths_sent:,}[/white] ([dim]{deauths_per_min:.1f}/min[/dim])'
             else:
                 metrics['Deauths Sent'] = f'[dim]0[/dim]'
 
@@ -1262,7 +1262,7 @@ def _update_display(self):
 
         # Add timing metrics if available
         if self.time_to_first_client is not None:
-            metrics['Time to 1st Client'] = f'[cyan]{self.time_to_first_client:.1f}s[/cyan]'
+            metrics['Time to 1st Client'] = f'[white]{self.time_to_first_client:.1f}s[/white]'
 
         if self.time_to_first_credential is not None:
             metrics['Time to 1st Cred'] = f'[yellow]{self.time_to_first_credential:.1f}s[/yellow]'
@@ -1310,7 +1310,7 @@ def _format_phase_indicator(self, phase: str) -> str:
         elif phase == "Failed":
             return f"[bold red]{indicator}[/bold red] {phase}"
         elif phase == "Running":
-            return f"[bold cyan]{indicator}[/bold cyan] {phase}"
+            return f"[bold white]{indicator}[/bold white] {phase}"
         elif phase in ["Stopping", "Cleaning up"]:
             return f"[yellow]{indicator}[/yellow] {phase}"
         else:
@@ -1365,7 +1365,7 @@ def _render_progress(self) -> Panel:
 
             # Create client table
             client_table = Table(show_header=True, box=None, padding=(0, 1))
-            client_table.add_column("MAC Address", style="cyan", width=17)
+            client_table.add_column("MAC Address", style="white", width=17)
             client_table.add_column("IP Address", style="green", width=15)
             client_table.add_column("Hostname", style="yellow")
             client_table.add_column("Duration", style="white", width=8)
@@ -1386,7 +1386,7 @@ def _render_progress(self) -> Panel:
             return Panel(
                 Group(base_panel.renderable, Text(), Text("Connected Clients:", style="bold"), client_table),
                 title="[bold]Progress[/bold]",
-                border_style="blue"
+                border_style="white"
             )
 
         return base_panel
@@ -1437,7 +1437,7 @@ def update_attack_statistics(self, deauth_count, disassoc_count,
                 'Deauth Frames': f'[red]{deauth_count:,}[/red]',
                 'Disassoc Frames': f'[yellow]{disassoc_count:,}[/yellow]',
                 'Total Attacks': f'[bold red]{total_attacks:,}[/bold red]',
-                'Networks Attacked': f'[cyan]{len(networks)}[/cyan]',
+                'Networks Attacked': f'[white]{len(networks)}[/white]',
                 'Unique Attackers': f'[magenta]{len(attackers)}[/magenta]'
             }
         })
@@ -1472,7 +1472,7 @@ def _render_target_info(self) -> Panel:
         from ..config import Configuration
 
         table = Table(show_header=False, box=None, padding=(0, 1))
-        table.add_column("Label", style="bold cyan", width=18)
+        table.add_column("Label", style="bold white", width=18)
         table.add_column("Value", style="white")
 
         table.add_row("Mode:", "Attack Monitoring")
@@ -1532,7 +1532,7 @@ def _render_progress(self) -> Panel:
         # Add network list
         if len(self.networks_under_attack) > 0:
             sections.append(Text())
-            sections.append(Text("Networks Under Attack:", style="bold cyan"))
+            sections.append(Text("Networks Under Attack:", style="bold white"))
             sections.append(self._render_network_list())
 
         # Add attacker list
@@ -1544,7 +1544,7 @@ def _render_progress(self) -> Panel:
         return Panel(
             Group(*sections),
             title="[bold]Attack Monitoring[/bold]",
-            border_style="blue"
+            border_style="white"
         )
 
     def _render_event_log(self) -> Table:
@@ -1557,7 +1557,7 @@ def _render_event_log(self) -> Table:
         event_table = Table(show_header=True, box=None, padding=(0, 1))
         event_table.add_column("Time", style="white", width=8)
         event_table.add_column("Type", style="white", width=8)
-        event_table.add_column("Target", style="cyan", width=25)
+        event_table.add_column("Target", style="white", width=25)
         event_table.add_column("Attacker", style="magenta", width=17)
 
         # Show last 10 events
@@ -1594,7 +1594,7 @@ def _render_network_list(self) -> Table:
             Rich Table with network attack statistics
         """
         network_table = Table(show_header=True, box=None, padding=(0, 1))
-        network_table.add_column("ESSID", style="cyan", width=20)
+        network_table.add_column("ESSID", style="white", width=20)
         network_table.add_column("BSSID", style="white", width=17)
         network_table.add_column("Attacks", style="red", width=8, justify="right")
         network_table.add_column("Last Seen", style="yellow", width=8)
@@ -1640,7 +1640,7 @@ def _render_attacker_list(self) -> Table:
         attacker_table = Table(show_header=True, box=None, padding=(0, 1))
         attacker_table.add_column("Attacker MAC", style="magenta", width=17)
         attacker_table.add_column("Attacks", style="red", width=8, justify="right")
-        attacker_table.add_column("Targets", style="cyan", width=8, justify="right")
+        attacker_table.add_column("Targets", style="white", width=8, justify="right")
 
         # Sort attackers by attack count (descending) and take top 10
         sorted_attackers = sorted(

wifite/wifite.py

@@ -51,11 +51,11 @@ def __init__(self):
 
         # Automatic cleanup of old session files on startup
         self.cleanup_old_sessions()
-        
+
         # Initialize interface assignment storage
         self.interface_assignment = None
         self.available_interfaces = []
-        
+
         # Initialize interface manager for state tracking and cleanup (Task 10.4)
         from .util.interface_manager import InterfaceManager
         self.interface_manager = InterfaceManager()
@@ -1008,9 +1008,20 @@ def passive_pmkid_capture(self):
             Color.pl('')
 
         try:
+            # Create TUI controller if not in classic mode
+            tui_controller = None
+            if Configuration.use_tui:
+                try:
+                    from .ui.tui import TUIController
+                    tui_controller = TUIController()
+                except (ImportError, Exception) as e:
+                    # Fall back to classic mode if TUI fails to load
+                    Color.pl('{!} {O}Warning: TUI mode failed to load, using classic mode{W}')
+                    if Configuration.verbose > 0:
+                        Color.pl('{!} {O}TUI Error: %s{W}' % str(e))
+                    Configuration.use_tui = False
+
             # Create and run passive PMKID attack
-            # Pass TUI controller if TUI mode is enabled
-            tui_controller = self.tui_controller if Configuration.use_tui else None
             attack = AttackPassivePMKID(tui_controller=tui_controller)
             attack.run()