Additional Fixes for Reverse Shell UI

Data/Agent/Roles/ReverseTunnel/tunnel_Powershell.py

@@ -179,5 +179,8 @@ class PowershellChannel:
                 pass
         # Include exit code in the close reason for debugging.
         exit_suffix = f" (exit={self._exit_code})" if self._exit_code is not None else ""
-        await self._send_close(code, (reason or "powershell_exit") + exit_suffix)
-        self.role._log(f"reverse_tunnel ps channel stopped channel={self.channel_id} reason={reason or 'exit'}")
+        close_reason = (reason or "powershell_exit") + exit_suffix
+        await self._send_close(code, close_reason)
+        self.role._log(
+            f"reverse_tunnel ps channel stopped channel={self.channel_id} reason={close_reason}"
+        )

Data/Agent/Roles/role_ReverseTunnel.py

@@ -521,6 +521,14 @@ class Role:
             self._log(f"reverse_tunnel connection failed tunnel_id={tunnel.tunnel_id}: {exc}", error=True)
             await self._emit_status({"tunnel_id": tunnel.tunnel_id, "agent_id": self.ctx.agent_id, "status": "error", "reason": "connect_failed"})
         finally:
+            try:
+                ws = tunnel.websocket
+                self._log(
+                    f"reverse_tunnel ws closing tunnel_id={tunnel.tunnel_id} "
+                    f"code={getattr(ws, 'close_code', None)} reason={getattr(ws, 'close_reason', None)}"
+                )
+            except Exception:
+                pass
             await self._shutdown_tunnel(tunnel)
 
     async def _pump_sender(self, tunnel: ActiveTunnel) -> None:
@@ -562,6 +570,7 @@ class Role:
                         f"reverse_tunnel websocket closed tunnel_id={tunnel.tunnel_id} "
                         f"code={ws.close_code} reason={ws.close_reason}"
                     )
+                    tunnel.stop_reason = ws.close_reason or "ws_closed"
                     break
         except asyncio.CancelledError:
             pass