1 files changed, 8 insertions, 4 deletions

diff --git a/dht/src/Network/Tox.hs b/dht/src/Network/Tox.hs
index 0a6cccaa..6b39d57a 100644
--- a/dht/src/Network/Tox.hs
+++ b/dht/src/Network/Tox.hs
@@ -73,7 +73,7 @@ import qualified Network.Tox.Onion.Transport as Onion
 import Network.Tox.RelayPinger
 import System.Global6
 import Network.Tox.Transport
-import Network.Tox.TCP (tcpClient, ViaRelay(..))
+import Network.Tox.TCP (tcpClient, ViaRelay(..), RelayClient)
 import Network.Tox.Onion.Routes
 import Network.Tox.ContactInfo
 import Text.XXD
@@ -214,8 +214,8 @@ data Tox extra  = Tox
 
 
 -- | Create a DHTPublicKey packet to send to a remote contact.
-getContactInfo :: Tox extra -> IO DHT.DHTPublicKey
-getContactInfo Tox{toxCryptoKeys,toxRouting,toxOnionRoutes} = join $ atomically $ do
+getContactInfo :: Maybe (RelayClient,PublicKey) -> Tox extra -> IO DHT.DHTPublicKey
+getContactInfo mthem Tox{toxCryptoKeys,toxRouting,toxOnionRoutes} = join $ atomically $ do
     (rcnt,relays) <- currentRelays (tcpRelayPinger toxOnionRoutes)
     r4 <- readTVar $ DHT.routing4 toxRouting
     r6 <- readTVar $ DHT.routing6 toxRouting
@@ -227,12 +227,16 @@ getContactInfo Tox{toxCryptoKeys,toxRouting,toxOnionRoutes} = join $ atomically
         n6s = R.kclosest DHT.toxSpace 4 self r6
         ns = filter (DHT.isGlobal . nodeIP) [n4,n6]
               ++ concat (zipWith (\a b -> [a,b]) n4s n6s)
+        sending_ns = take 4 $ relays ++ map TCP.fromUDPNode ns
     return $ do
+        forM_ mthem $ \(tcp,theirDHTKey) ->
+            forM_ (filter (\n -> TCP.tcpPort n /= 0) sending_ns) $ \ni -> do
+                Multi.tcpConnectionRequest tcp theirDHTKey ni
         timestamp <- round . (* 1000000) <$> getPOSIXTime
         return DHT.DHTPublicKey
             { dhtpkNonce = timestamp
             , dhtpk      = id2key self
-            , dhtpkNodes = DHT.SendNodes $ take 4 $ relays ++ map TCP.fromUDPNode ns
+            , dhtpkNodes = DHT.SendNodes sending_ns
             }
 
 isLocalHost :: SockAddr -> Bool