Clean-up relay-client transport.

1 files changed, 2 insertions, 1 deletions

diff --git a/dht/src/Network/QueryResponse/TCP.hs b/dht/src/Network/QueryResponse/TCP.hs
index bad61727..de6e1eb7 100644
--- a/dht/src/Network/QueryResponse/TCP.hs
+++ b/dht/src/Network/QueryResponse/TCP.hs
@@ -168,6 +168,7 @@ acquireConnection mvar tcpcache stream addr bDoCon = do
 
 closeAll :: TCPCache (SessionProtocol x y) -> StreamHandshake addr x y -> IO ()
 closeAll tcpcache stream = do
+    dput XTCP "TCP.closeAll called."
     cache <- atomically $ swapTVar (lru tcpcache) MM.empty
     forM_ (MM.toList cache) $ \(MM.Binding (TCPAddress addr) r tm) -> do
         killSession r
@@ -188,5 +189,5 @@ tcpTransport maxcon stream = do
                                     mapM_ ($ y) msock
                                   `catchIOError` \e -> dput XTCP $ "TCP-send: " ++ show e
                                 labelThread t "tcp-send"
-        , closeTransport = closeAll tcpcache stream
+        , closeTransport = closeAll tcpcache stream >> putMVar msgvar Nothing
         }