Update TCP kademlia table.

1 files changed, 10 insertions, 0 deletions

diff --git a/OnionRouter.hs b/OnionRouter.hs
index 0e0b5afb..55ba9c28 100644
--- a/OnionRouter.hs
+++ b/OnionRouter.hs
@@ -191,6 +191,13 @@ newOnionRouter crypto perror = do
             , trampolineIds = ti
             , trampolineCount = tc
             , tcpKademliaClient = tcp
+                { TCP.tcpClient =
+                    let c = TCP.tcpClient tcp
+                    in c { clientNet = addHandler perror (handleMessage c)
+                                        $ onInbound (updateTCP refresher)
+                                        $ clientNet c
+                         }
+                }
             , tcpBucketRefresher = refresher
             , routeLog = rlog
             , routeThread = error "forkRouteBuilder not invoked (missing onion route builder thread)."
@@ -199,6 +206,9 @@ newOnionRouter crypto perror = do
             }
     return or
 
+updateTCP :: BucketRefresher NodeId TCP.NodeInfo -> TCP.NodeInfo -> p -> IO ()
+updateTCP refresher addr x = insertNode (refreshKademlia refresher) addr
+
 selectGateway :: TVar (R.BucketList TCP.NodeInfo) -> NodeInfo -> STM (Maybe TCP.NodeInfo)
 selectGateway tbl ni = do
     ns <- kclosest TCP.tcpSpace 2 (nodeId ni) <$> readTVar tbl