Tox: Added timestamps to dhtkey and sockaddr information.

1 files changed, 6 insertions, 4 deletions

diff --git a/src/Network/Tox/DHT/Handlers.hs b/src/Network/Tox/DHT/Handlers.hs
index 6a80b4a4..7e986fdd 100644
--- a/src/Network/Tox/DHT/Handlers.hs
+++ b/src/Network/Tox/DHT/Handlers.hs
@@ -117,8 +117,8 @@ classify client msg = fromMaybe (IsUnknown "unknown")
 data NodeInfoCallback = NodeInfoCallback
     { interestingNodeId :: NodeId
     , listenerId        :: Int
-    , observedAddress   :: NodeInfo -> STM ()
-    , rumoredAddress    :: SockAddr -> NodeInfo -> STM ()
+    , observedAddress   :: POSIXTime -> NodeInfo -> STM ()
+    , rumoredAddress    :: POSIXTime -> SockAddr -> NodeInfo -> STM ()
     }
 
 data Routing = Routing
@@ -405,22 +405,24 @@ getNodes client cbvar nid addr = do
     -- hPutStrLn stderr $ show addr ++ " -sendnodes-> " ++ show reply
     forM_ (join reply) $ \(SendNodes ns) -> 
         forM_ ns $ \n -> do
+            now <- getPOSIXTime
             atomically $ do
                 mcbs <- HashMap.lookup (nodeId n) <$> readTVar cbvar
                 forM_ mcbs $ \cbs -> do
                     forM_ cbs $ \cb -> do
-                        rumoredAddress cb (nodeAddr addr) n
+                        rumoredAddress cb now (nodeAddr addr) n
     return $ fmap unwrapNodes $ join reply
 
 updateRouting :: Client -> Routing -> OnionRouter -> NodeInfo -> Message -> IO ()
 updateRouting client routing orouter naddr msg
     | PacketKind 0x21 <- msgType msg = dput XLan "(tox)updateRouting: ignoring lan discovery" -- ignore lan discovery
     | otherwise                      = do
+    now <- getPOSIXTime
     atomically $ do
         m <- HashMap.lookup (nodeId naddr) <$> readTVar (nodesOfInterest routing)
         forM_ m $ mapM_ $ \NodeInfoCallback{interestingNodeId,observedAddress} -> do
             when (interestingNodeId == nodeId naddr)
-                $ observedAddress naddr
+                $ observedAddress now naddr
     case prefer4or6 naddr Nothing of
         Want_IP4 -> updateTable client naddr orouter (committee4 routing) (refresher4 routing)
         Want_IP6 -> updateTable client naddr orouter (committee6 routing) (refresher6 routing)