dhtBootstrap

1 files changed, 7 insertions, 2 deletions

diff --git a/src/Network/BitTorrent/MainlineDHT.hs b/src/Network/BitTorrent/MainlineDHT.hs
index 3e7a0eda..47d5b3e1 100644
--- a/src/Network/BitTorrent/MainlineDHT.hs
+++ b/src/Network/BitTorrent/MainlineDHT.hs
@@ -538,7 +538,12 @@ mkNodeInfo nid addr = NodeInfo
     , nodePort = fromMaybe 0 $ sockAddrPort addr
     }
 
-newClient :: SwarmsDatabase -> SockAddr -> IO (MainlineClient, Routing)
+newClient :: SwarmsDatabase -> SockAddr
+    -> IO ( MainlineClient
+          , Routing
+          , [NodeInfo] -> [NodeInfo] -> IO ()
+          , [NodeInfo] -> [NodeInfo] -> IO ()
+          )
 newClient swarms addr = do
     udp <- udpTransport addr
     nid <- NodeId <$> getRandomBytes 20
@@ -651,7 +656,7 @@ newClient swarms addr = do
     refresh_thread4 <- forkPollForRefresh $ refresher4 routing
     refresh_thread6 <- forkPollForRefresh $ refresher6 routing
 
-    return (client, routing)
+    return (client, routing, bootstrap (refresher4 routing), bootstrap (refresher6 routing))
 
 -- | Modifies a purely random 'NodeId' to one that is related to a given
 -- routable address in accordance with BEP 42.