code: cleanup

Author: kimocoder

wifite/attack/all.py

@@ -55,10 +55,10 @@ def attack_multiple(cls, targets, session=None, session_mgr=None):
             if index % 5 == 0:  # Every 5 targets
                 from ..util.process import Process
                 Process.check_fd_limit()
-                
+
                 # Memory monitoring - especially important for infinite mode
                 MemoryMonitor.periodic_check(index)
-            
+
             # Infinite mode specific cleanup
             if infinite_monitor and index % 10 == 0:
                 infinite_monitor.on_target_complete()
@@ -111,25 +111,25 @@ def attack_single(cls, target, targets_remaining, session=None, session_mgr=None
 
         elif target.primary_encryption.startswith('WPA'): # Covers WPA, WPA2, WPA3
             # WPA can have multiple attack vectors:
-            
+
             # Check if this is a WPA3 target
             is_wpa3 = target.primary_encryption == 'WPA3' or \
-                     (hasattr(target, 'wpa3_info') and target.wpa3_info and 
+                     (hasattr(target, 'wpa3_info') and target.wpa3_info and
                       (hasattr(target.wpa3_info, 'has_wpa3') and target.wpa3_info.has_wpa3))
-            
+
             # For WPA3 targets, use specialized WPA3 attack if tools are available
             if is_wpa3 and WPA3ToolChecker.can_attack_wpa3():
                 # Use WPA3-specific attack module
                 attacks.append(AttackWPA3SAE(target))
-                
+
                 # For transition mode, also try standard WPA2 attacks as fallback
                 if hasattr(target, 'wpa3_info') and target.wpa3_info and target.wpa3_info.get('is_transition'):
                     if not Configuration.wps_only and not Configuration.use_pmkid_only:
                         # Add PMKID and WPA attacks as fallback for transition mode
                         if not Configuration.dont_use_pmkid:
                             attacks.append(AttackPMKID(target))
                         attacks.append(AttackWPA(target))
-            
+
             # WPS attacks (not applicable to pure WPA3)
             elif not is_wpa3 and \
                not Configuration.use_pmkid_only and \

wifite/attack/pmkid.py

@@ -66,7 +66,7 @@ def get_existing_pmkid_file(bssid):
         pmkid_pattern = os.path.join(Configuration.wpa_handshake_dir, 'pmkid_*.22000')
         files_found = glob.glob(pmkid_pattern)
         log_debug('AttackPMKID', f'Found {len(files_found)} PMKID file(s) to check')
-        
+
         for pmkid_filename in files_found:
             if not os.path.isfile(pmkid_filename):
                 continue
@@ -141,15 +141,15 @@ def run_hashcat(self):
             return False
 
         from ..util.process import Process
-        
+
         # Check tool availability - prioritize hcxdumptool, fallback to native
         use_native_pmkid = False
         dependencies = [
             HcxDumpTool.dependency_name,
             HcxPcapngTool.dependency_name
         ]
         missing_deps = [dep for dep in dependencies if not Process.exists(dep)]
-        
+
         if missing_deps:
             # Check if native PMKID capture is available as fallback
             if NATIVE_PMKID_AVAILABLE:
@@ -528,22 +528,22 @@ def dumptool_thread(self):
     def capture_pmkid_native(self):
         """
         Capture PMKID using native Scapy implementation.
-        
+
         This is a fallback method when hcxdumptool is not available.
         Uses ScapyPMKID to capture PMKID by sending auth frames and
         listening for EAPOL Message 1 responses.
-        
+
         Returns:
             Path to PMKID hash file (.22000) if captured, None otherwise
         """
         log_info('AttackPMKID', f'Starting native PMKID capture for {self.target.essid} ({self.target.bssid})')
-        
+
         if self.view:
             self.view.add_log("Starting native PMKID capture (Scapy)...")
             self.view.set_capture_tool("Native (Scapy)")
-        
+
         Color.pattack('PMKID', self.target, 'CAPTURE', 'Starting native capture...')
-        
+
         try:
             # Set interface to target channel
             from ..native.interface import NativeInterface
@@ -553,17 +553,17 @@ def capture_pmkid_native(self):
                     log_debug('AttackPMKID', f'Set channel to {self.target.channel}')
                 except Exception as e:
                     log_warning('AttackPMKID', f'Could not set channel: {e}')
-            
+
             # Capture PMKID with timeout
             timeout = Configuration.pmkid_timeout
             self.timer = Timer(timeout)
-            
+
             # Track progress for TUI
             def on_pmkid_captured(result):
                 log_info('AttackPMKID', f'Native capture found PMKID: {result.pmkid[:16]}...')
                 if self.view:
                     self.view.add_log(f'PMKID captured!')
-            
+
             # Use ScapyPMKID capture
             result = ScapyPMKID.capture(
                 interface=Configuration.interface,
@@ -574,7 +574,7 @@ def on_pmkid_captured(result):
                 channel=int(self.target.channel) if self.target.channel else None,
                 callback=on_pmkid_captured
             )
-            
+
             if result is None:
                 log_warning('AttackPMKID', 'Native PMKID capture failed: no PMKID received')
                 if self.view:
@@ -583,22 +583,22 @@ def on_pmkid_captured(result):
                 Color.pattack('PMKID', self.target, 'CAPTURE', '{R}Failed{O} to capture PMKID (native)\n')
                 Color.pl('')
                 return None
-            
+
             # Success - convert to hashcat format and save
             log_info('AttackPMKID', 'Native PMKID capture successful')
             if self.view:
                 self.view.update_pmkid_status(True, 1)
                 self.view.add_log("Successfully captured PMKID (native)!")
-            
+
             Color.clear_entire_line()
             Color.pattack('PMKID', self.target, 'CAPTURE', '{G}Captured PMKID{W} (native)')
-            
+
             # Generate hashcat 22000 format
             pmkid_hash = result.to_hashcat_22000()
-            
+
             # Save to file
             return self.save_pmkid(pmkid_hash)
-            
+
         except Exception as e:
             log_error('AttackPMKID', f'Native PMKID capture error: {e}', e)
             if self.view:

wifite/attack/wpa.py

@@ -21,27 +21,27 @@
 
 
 class AttackWPA(Attack):
-    
+
     # Maximum retry attempts for recoverable errors
     MAX_RETRY_ATTEMPTS = 3
     RETRY_DELAY_SECONDS = 2
-    
+
     def __init__(self, target):
         super(AttackWPA, self).__init__(target)
         self.clients = []
         self.crack_result = None
         self.success = False
-        
+
         # Interface assignment for dual interface support
         self.interface_assignment = None
         self.capture_interface = None  # Dedicated interface for handshake capture
         self.deauth_interface = None   # Dedicated interface for deauthentication
-        
+
         # Error recovery tracking
         self._retry_count = 0
         self._last_error = None
         self._recovered_from_error = False
-        
+
         # Initialize TUI view if in TUI mode
         self.view = None
         if OutputManager.is_tui_mode():
@@ -51,18 +51,18 @@ def __init__(self, target):
             except Exception:
                 # If TUI initialization fails, continue without it
                 self.view = None
-    
+
     @contextmanager
     def _error_recovery_context(self, operation_name: str):
         """
         Context manager for automatic error recovery during WPA attacks.
-        
+
         Handles common recoverable errors:
         - Interface going down
         - Channel switching failures
         - Resource exhaustion
         - Process crashes
-        
+
         Args:
             operation_name: Name of operation for logging
             

wifite/attack/wpa3.py

@@ -67,33 +67,33 @@ def run(self):
         4. Selects optimal attack strategy
         5. Executes the selected strategy
         6. Falls back to alternative strategies if needed
-        
+
         Returns:
             bool: True if attack succeeded, False otherwise
         """
         # Check for required WPA3 tools
         if not self._check_wpa3_tools():
             return False
-        
+
         # Start TUI view if available
         if self.view:
             self.view.start()
-        
+
         # Detect WPA3 capabilities
         self.wpa3_info = WPA3Detector.detect_wpa3_capability(self.target)
-        
+
         # Check for Dragonblood vulnerabilities
         self._check_dragonblood_vulnerability()
-        
+
         # Select attack strategy
         self.attack_strategy = WPA3AttackStrategy.select_strategy(
-            self.target, 
+            self.target,
             self.wpa3_info
         )
-        
+
         # Display strategy to user
         self._display_strategy()
-        
+
         # Execute strategy based on selection
         if self.attack_strategy == WPA3AttackStrategy.DOWNGRADE:
             return self._execute_downgrade_strategy()