Use seqence number context when serializing CryptoMessage.

1 files changed, 3 insertions, 3 deletions

diff --git a/examples/atox.hs b/examples/atox.hs
index 7845d911..3bae5203 100644
--- a/examples/atox.hs
+++ b/examples/atox.hs
@@ -92,7 +92,7 @@ data SetCmd = SetME
     deriving (Eq,Bounded,Ord,Enum,Show)
 
 forkToxInputThread myRead = forkIO $ do
-    let myconduit = Conduit.sourceHandle myRead  .| conduitGet2 (S.get :: Get CryptoMessage) -- :: ConduitT i CryptoMessage IO ()
+    let myconduit = Conduit.sourceHandle myRead  .| conduitGet2 (getCryptoMessage 0 :: Get CryptoMessage) -- :: ConduitT i CryptoMessage IO ()
     Conduit.runConduit (myconduit .| Conduit.awaitForever handle)
  where
     handle (UpToN IPC (B.uncons -> Just (ord -> toEnum -> i,arg))) = updateState i arg
@@ -113,7 +113,7 @@ forkToxInputThread myRead = forkIO $ do
                                     modifyTVar' sMap (Map.insert key view)
 
     updateState AppendMsg arg
-             = case S.decode arg of
+             = case getCryptoMessage 0 `S.runGet` arg of
                  Left str  -> puts (packUtf8 str)
                  Right msg -> liftIO . atomically $ do
                                 me <- readTVar sMe
@@ -133,7 +133,7 @@ doit myReadFd myWriteFd = do
     terminalInputLoop myWrite
 
 hSend :: MonadIO m => Handle -> CryptoMessage -> m ()
-hSend h msg = liftIO $ B.hPutStrLn h (S.encode msg)
+hSend h msg = liftIO $ B.hPutStrLn h (S.runPut $ putCryptoMessage 0 msg)
 
 terminalInputLoop myWriteH = fix $ \loop -> do
     line <- B.getLine