Use seqence number context when serializing CryptoMessage.

1 files changed, 3 insertions, 3 deletions

diff --git a/examples/dhtd.hs b/examples/dhtd.hs
index 6161fbb4..5bc2b87a 100644
--- a/examples/dhtd.hs
+++ b/examples/dhtd.hs
@@ -99,7 +99,7 @@ import qualified Network.Tox.DHT.Transport as Tox
 import qualified Network.Tox.DHT.Handlers as Tox
 import qualified Network.Tox.Onion.Transport as Tox
 import qualified Network.Tox.Onion.Handlers as Tox
-import qualified Network.Tox.Crypto.Transport as Tox (CryptoMessage(..))
+import qualified Network.Tox.Crypto.Transport as Tox (CryptoMessage(..),putCryptoMessage,getCryptoMessage)
 import qualified Network.Tox.Crypto.Handlers as Tox
 import Data.Typeable
 import Network.Tox.ContactInfo as Tox
@@ -1407,7 +1407,7 @@ netcrypto (Just (DHT {..})) (Just mypubkey) h roster (Just tox) exes paramStr =
                                              when (them /= their_pub) $ do
                                                 atomically $ writeTVar whoAreThey their_pub
                                                 B.hPutStr myWrite ("\NUL\SOH" `B.append` S.encode (Tox.key2id their_pub))
-                                             B.hPutStr myWrite (S.encode msg)
+                                             B.hPutStr myWrite (S.runPut $ Tox.putCryptoMessage 0 msg)
                                              return (Just id)
                                 addHooks currentHooks typs = forM_ typs $ \typ -> modifyTVar (Tox.ncHooks session) (Map.insert typ (currentHooks typ ++ [makeHook session typ]))
                             case ranges of
@@ -1426,7 +1426,7 @@ netcrypto (Just (DHT {..})) (Just mypubkey) h roster (Just tox) exes paramStr =
                                 tid <- myThreadId
                                 let sidStr = printf "(%x)" (Tox.ncSessionId session)
                                 labelThread tid (exekey ++ ".forward" ++ sidStr)
-                                let myconduit = Conduit.sourceHandle myRead  .| conduitGet2 S.get -- :: ConduitT i CryptoMessage IO ()
+                                let myconduit = Conduit.sourceHandle myRead  .| conduitGet2 (Tox.getCryptoMessage 0) -- :: ConduitT i CryptoMessage IO ()
                                 Conduit.runConduit (myconduit .| awaitForever (\msg -> do
                                     let typ = toWord64 (getMessageType msg)
                                     mbSendIt <- liftIO $ atomically (lookupInRangeMap typ (Tox.ncOutHooks session))