Fix websockets server shutdown error

- Removed incorrect 'closed' attribute check on websockets Server object
- Added proper exception handling for server shutdown
- Fixed 'Server' object has no attribute 'closed' error during shutdown
- Server now shuts down cleanly without throwing exceptions
parent 286f91f0
...@@ -693,10 +693,12 @@ async def main(): ...@@ -693,10 +693,12 @@ async def main():
print("\nShutting down WebSocket SSH Daemon...") print("\nShutting down WebSocket SSH Daemon...")
# Close WebSocket server if it's still running # Close WebSocket server
if not ws_server.closed: try:
ws_server.close() ws_server.close()
await ws_server.wait_closed() await ws_server.wait_closed()
except Exception as e:
if debug: print(f"[DEBUG] Error closing WebSocket server: {e}")
# Cancel cleanup task # Cancel cleanup task
if not cleanup_coro.done(): if not cleanup_coro.done():
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment