Only one session at a time (per DHT node).

1 files changed, 5 insertions, 0 deletions

diff --git a/dht/src/Network/Tox/Session.hs b/dht/src/Network/Tox/Session.hs
index 53d63287..1b1c62c4 100644
--- a/dht/src/Network/Tox/Session.hs
+++ b/dht/src/Network/Tox/Session.hs
@@ -6,6 +6,7 @@ module Network.Tox.Session
     , SessionKey
     , Session(..)
     , sTheirUserKey
+    , sTheirDHTKey
     , sClose
     , handshakeH
     ) where
@@ -95,6 +96,10 @@ sTheirUserKey s = longTermKey $ runIdentity cookie
  where
     Cookie _ cookie = handshakeCookie (sReceivedHandshake s)
 
+sTheirDHTKey :: Session -> PublicKey
+sTheirDHTKey s = case handshakeCookie $ sReceivedHandshake s of
+    Cookie _ (Identity cd) -> dhtKey cd
+
 -- | Helper to close the 'Transport' associated with a session.
 sClose :: Session -> IO ()
 sClose s = do