Use first known key for cookie response if wrong ip address.

  This isn't fool-proof, but it enables some testing scenarios
  where the ip address was not the expected one.

1 files changed, 2 insertions, 1 deletions

diff --git a/src/Network/Tox/DHT/Transport.hs b/src/Network/Tox/DHT/Transport.hs
index 7af8d408..f7feea3b 100644
--- a/src/Network/Tox/DHT/Transport.hs
+++ b/src/Network/Tox/DHT/Transport.hs
@@ -38,6 +38,7 @@ import Crypto.Tox hiding (encrypt,decrypt)
 import qualified Crypto.Tox as ToxCrypto
 import Network.QueryResponse
 
+import Control.Applicative
 import Control.Arrow
 import Control.Concurrent.STM
 import Control.Monad
@@ -108,7 +109,7 @@ parseDHTAddr crypto (msg,saddr)
     0x19 -> do
         cs <- atomically $ readTVar (pendingCookies crypto)
         let ni = fromMaybe (noReplyAddr saddr) $ do
-                    (cnt,key) <- lookup saddr cs
+                    (cnt,key) <- lookup saddr cs <|> listToMaybe (map snd cs)
                     either (const Nothing) Just $ nodeInfo (key2id key) saddr
         left $ fanGet bs getCookie (uncurry DHTCookie) (const $ ni)
     0x20 -> left $ fanGet bs getDHTReqest (uncurry DHTDHTRequest) (asymNodeInfo saddr . snd)