STM-based awaitMessage.

1 files changed, 3 insertions, 1 deletions

diff --git a/dht/src/Network/Tox.hs b/dht/src/Network/Tox.hs
index 61a1d117..4b66cfc8 100644
--- a/dht/src/Network/Tox.hs
+++ b/dht/src/Network/Tox.hs
@@ -222,10 +222,12 @@ isLocalHost _                       = False
 addVerbosity :: Transport err SockAddr ByteString -> Transport err SockAddr ByteString
 addVerbosity tr =
     tr { awaitMessage = \kont -> awaitMessage tr $ \m -> do
-            forM_ m $ mapM_ $ \(msg,addr) -> do
+            case m of
+              Arrival addr msg -> do
                 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
        , sendMessage = \addr msg -> do
             when (not (B.null msg || elem (B.head msg) [0,1,2,4,0x81,0x8c,0x8d])) $ do