Tox: Added timestamps to dhtkey and sockaddr information.

1 files changed, 4 insertions, 3 deletions

diff --git a/Connection/Tox/Threads.hs b/Connection/Tox/Threads.hs
index 5602fa40..a5a839f9 100644
--- a/Connection/Tox/Threads.hs
+++ b/Connection/Tox/Threads.hs
@@ -32,6 +32,7 @@ import Control.Concurrent.Lifted
 import GHC.Conc                  (labelThread)
 #endif
 
+import Control.Arrow
 import Control.Concurrent.STM
 import Control.Monad
 import Data.Function
@@ -124,15 +125,15 @@ persueContact getPolicy getStatus PersueContactMethods{..} statusVar = do
         $ \retryAfterTimeout ->
             orElse (do
                     readTVar statusVar >>= check . (/= InProgress AcquiringIPAddress)
-                    theirDhtKey <- DHT.dhtpk <$> retryUntilJust (contactKeyPacket contact)
+                    (stamp_theirDhtKey,theirDhtKey) <- second DHT.dhtpk <$> retryUntilJust (contactKeyPacket contact)
                     -- We don't have an IP address yet.
                     maybe (return ()) (const retry) =<< readTVar (contactLastSeenAddr contact)
                     return $ do -- AcquiringIPAddress
                         atomically $ writeTVar statusVar (InProgress AcquiringIPAddress)
                         retryAfterTimeout 0)
                    (do
-                    theirDhtKey <- DHT.dhtpk <$> retryUntilJust (contactKeyPacket contact)
-                    saddr <- retryUntilJust (contactLastSeenAddr contact)
+                    (stamp_theirDhtKey,theirDhtKey) <- second DHT.dhtpk <$> retryUntilJust (contactKeyPacket contact)
+                    (stamp_saddr,saddr) <- retryUntilJust (contactLastSeenAddr contact)
                     ni <- either (const retry) return $ nodeInfo (key2id theirDhtKey) saddr
                     return $ do
                     -- AcquiringCookie