From d4c209fb9543019461bcf612da67708aeabcdce2 Mon Sep 17 00:00:00 2001 From: Joe Crayne Date: Sat, 25 Jan 2020 01:02:33 -0500 Subject: Ported dhtd to reworked QueryResponse design. --- dht/src/Network/Tox.hs | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) (limited to 'dht/src/Network/Tox.hs') diff --git a/dht/src/Network/Tox.hs b/dht/src/Network/Tox.hs index 6b39d57a..3dd1d48e 100644 --- a/dht/src/Network/Tox.hs +++ b/dht/src/Network/Tox.hs @@ -245,14 +245,15 @@ isLocalHost _ = False addVerbosity :: Transport err SockAddr ByteString -> Transport err SockAddr ByteString addVerbosity tr = - tr { awaitMessage = \kont -> awaitMessage tr $ \m -> do + tr { awaitMessage = do + (m,io) <- awaitMessage tr case m of - Arrival addr msg -> do + Arrival addr msg -> return $ (,) m $ do + io when (not (B.null msg || elem (B.head msg) [0,1,2,4,0x81,0x82,0x8c,0x8d])) $ do mapM_ (\x -> dput XMisc ( (show addr) ++ " --> " ++ x)) $ xxd 0 msg - _ -> return () - kont m + _ -> return (m,io) , sendMessage = \addr msg -> do when (not (B.null msg || elem (B.head msg) [0,1,2,4,0x81,0x8c,0x8d])) $ do mapM_ (\x -> dput XMisc ( (show addr) ++ " <-- " ++ x)) -- cgit v1.2.3