Multi Transports: TCP for DHT/Cookies/Handshakes.

1 files changed, 10 insertions, 0 deletions

diff --git a/dht/src/Network/Tox/DHT/Transport.hs b/dht/src/Network/Tox/DHT/Transport.hs
index 0583c9a3..ff743f29 100644
--- a/dht/src/Network/Tox/DHT/Transport.hs
+++ b/dht/src/Network/Tox/DHT/Transport.hs
@@ -33,8 +33,10 @@ module Network.Tox.DHT.Transport
     , dhtMessageType
     , asymNodeInfo
     , putMessage -- Convenient for serializing DHTLanDiscovery
+    , toxSpace
     ) where
 
+import qualified Network.Kademlia.Routing as R
 import Network.Tox.NodeId
 import qualified Network.Tox.TCP.NodeId as TCP
 import Crypto.Tox hiding (encrypt,decrypt)
@@ -464,3 +466,11 @@ transcode f (DHTCookieRequest asym)     = DHTCookieRequest     $ asym { asymmDat
 transcode f (DHTCookie n dta)           = DHTCookie n          $ f n  $ Left dta
 transcode f (DHTDHTRequest pubkey asym) = DHTDHTRequest pubkey $ asym { asymmData = f (asymmNonce asym) (Right asym) }
 transcode f (DHTLanDiscovery nid)       = DHTLanDiscovery nid
+
+toxSpace :: R.KademliaSpace NodeId NodeInfo
+toxSpace = R.KademliaSpace
+    { R.kademliaLocation = nodeId
+    , R.kademliaTestBit  = testNodeIdBit
+    , R.kademliaXor      = xorNodeId
+    , R.kademliaSample   = sampleNodeId
+    }