From ff93430c5df52518b8f404e3c0c17bef071cdbdd Mon Sep 17 00:00:00 2001
From: Carson Fleming <cflems@cflems.net>
Date: Tue, 28 Mar 2023 11:02:03 -0700
Subject: Fix some transition bugs; pty still hangs trying to pull streaming
 key

---
 pkd_stub.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'pkd_stub.py')

diff --git a/pkd_stub.py b/pkd_stub.py
index cac525c..695db77 100644
--- a/pkd_stub.py
+++ b/pkd_stub.py
@@ -283,7 +283,7 @@ def tcp_disconnect(sel, client):
     if client not in tcp_clients:
         return
 
-    sel.unregister(client['sock'])
+    sel.unregister(client['sock'].sock)
     client['sock'].close()
     client['alive'] = False
     idx = tcp_clients.index(client)
@@ -327,6 +327,7 @@ def tcp_unpty(sel, client, catchup=True, backtrack=0):
 
 def tcp_transport(sel, sock, client):
     global tcp_clients, privkey, bits
+
     if not client['alive']:
         return
     try:
-- 
cgit v1.2.3