funstuff

Author: CyberRoute

core/arp_scanner.py

@@ -87,6 +87,7 @@ def __init__(self, packet_summary: str, packet_text: str, parent=None):
         self.setCentralWidget(container)
 
     def append_token(self, token: str):
+        """Append a streamed token to the output, clearing the placeholder on first call."""
         if self._output.toPlainText() == "Analysing…":
             self._output.clear()
         cursor = self._output.textCursor()
@@ -96,13 +97,16 @@ def append_token(self, token: str):
         self._output.ensureCursorVisible()
 
     def set_analysing(self):
+        """Reset the output area to the 'analysing' placeholder state."""
         self._output.setPlainText("Analysing…")
         self._copy_button.setEnabled(False)
 
     def set_error(self, msg: str):
+        """Display an error message in the output area."""
         self._output.setPlainText(f"Error: {msg}")
 
     def set_done(self):
+        """Mark analysis as complete and enable the copy button."""
         self._copy_button.setEnabled(True)
 
     def _copy_to_clipboard(self):
@@ -176,6 +180,44 @@ def __init__(  # pylint: disable=too-many-arguments,too-many-positional-argument
         self._save_pcap_button.setEnabled(False)
         layout.addWidget(self._save_pcap_button)
 
+        self._build_packet_panel(layout)
+
+        self._captured_packets = []  # newest first, mirrors list order
+        self._ollama_thread: OllamaThread | None = None
+        self._llm_window: LlmAnalysisWindow | None = None
+
+        central_widget = QWidget()
+        central_widget.setLayout(layout)
+        self.setCentralWidget(central_widget)
+        self.resize(680, 850)
+
+        self._load_ollama_models()
+
+    @Slot(bool)
+    def _toggle_mitm(self, checked):
+        if checked:
+            self._mitm = MitmThread(self.interface, self.ip_address, self.gateway_ip)
+            self._mitm.statusChanged.connect(self._on_status)
+            self._mitm.packetCaptured.connect(self._on_packet)
+            self._mitm.stopped.connect(self._on_mitm_stopped)
+            self._mitm.start()
+            self._mitm_button.setText("Stop MITM")
+            self._status_label.setStyleSheet("color: #ff8800; font-weight: bold")
+        else:
+            self._mitm_button.setEnabled(False)
+            if self._mitm:
+                self._mitm.stop()  # async — _on_mitm_stopped will reset UI
+
+    @Slot()
+    def _on_mitm_stopped(self):
+        self._mitm = None
+        self._mitm_button.setText("Start MITM")
+        self._mitm_button.setChecked(False)
+        self._mitm_button.setEnabled(True)
+        self._status_label.setStyleSheet("color: grey")
+
+    def _build_packet_panel(self, layout: QVBoxLayout):
+        """Build the packet list/detail/LLM controls and add them to *layout*."""
         mono = QFont()
         mono.setStyleHint(QFont.StyleHint.Monospace)
         mono.setPointSize(9)
@@ -194,7 +236,6 @@ def __init__(  # pylint: disable=too-many-arguments,too-many-positional-argument
         self._analyse_button.setEnabled(False)
         self._analyse_button.clicked.connect(self._analyse_packet)
 
-        # Stack: packet list | packet detail | [analyse btn + llm output]
         pkt_splitter = QSplitter(Qt.Vertical)
         pkt_splitter.addWidget(self._packet_list)
         pkt_splitter.addWidget(self._packet_detail)
@@ -206,7 +247,6 @@ def __init__(  # pylint: disable=too-many-arguments,too-many-positional-argument
         )
         self._user_context.returnPressed.connect(self._analyse_packet)
 
-        # Model selector row
         self._model_combo = QComboBox()
         self._model_combo.setPlaceholderText("Select Ollama model…")
         self._refresh_models_button = QPushButton("↻")
@@ -225,40 +265,6 @@ def __init__(  # pylint: disable=too-many-arguments,too-many-positional-argument
         layout.addLayout(model_row)
         layout.addWidget(self._analyse_button)
 
-        self._captured_packets = []  # newest first, mirrors list order
-        self._ollama_thread: OllamaThread | None = None
-        self._llm_window: LlmAnalysisWindow | None = None
-
-        central_widget = QWidget()
-        central_widget.setLayout(layout)
-        self.setCentralWidget(central_widget)
-        self.resize(680, 850)
-
-        self._load_ollama_models()
-
-    @Slot(bool)
-    def _toggle_mitm(self, checked):
-        if checked:
-            self._mitm = MitmThread(self.interface, self.ip_address, self.gateway_ip)
-            self._mitm.statusChanged.connect(self._on_status)
-            self._mitm.packetCaptured.connect(self._on_packet)
-            self._mitm.stopped.connect(self._on_mitm_stopped)
-            self._mitm.start()
-            self._mitm_button.setText("Stop MITM")
-            self._status_label.setStyleSheet("color: #ff8800; font-weight: bold")
-        else:
-            self._mitm_button.setEnabled(False)
-            if self._mitm:
-                self._mitm.stop()  # async — _on_mitm_stopped will reset UI
-
-    @Slot()
-    def _on_mitm_stopped(self):
-        self._mitm = None
-        self._mitm_button.setText("Start MITM")
-        self._mitm_button.setChecked(False)
-        self._mitm_button.setEnabled(True)
-        self._status_label.setStyleSheet("color: grey")
-
     def _load_ollama_models(self):
         """Populate the model combo box with models available on the local Ollama server."""
         models = fetch_ollama_models()
@@ -684,8 +690,6 @@ def _shutdown(self):
         if self.arp_scanner_thread is not None and self.arp_scanner_thread.isRunning():
             self.arp_scanner_thread.quit()
             self.arp_scanner_thread.wait()
-        from PySide6.QtWidgets import QApplication  # pylint: disable=E0611
-
         QApplication.quit()
 
 
@@ -744,7 +748,10 @@ def run(self):
     def _scan_network(self, src_ip, network):
         hosts = [str(ip) for ip in network.hosts() if str(ip) != src_ip]
         if self.use_native:
-            print(f"[scan] native ARP scanner — {len(hosts)} hosts  timeout={self.timeout}s  workers={_RESOLVE_WORKERS}")
+            print(
+                f"[scan] native ARP scanner — {len(hosts)} hosts"
+                f"  timeout={self.timeout}s  workers={_RESOLVE_WORKERS}"
+            )
             return self._run_native_scan(src_ip, hosts)
         print(f"Using Scapy ARP scanner — {len(hosts)} hosts")
         return self._run_scapy_scan(hosts)