addtoNonce24: unsafeDupablePerformIO at def, not call sites.

1 files changed, 4 insertions, 7 deletions

diff --git a/src/Network/Tox/Crypto/Handlers.hs b/src/Network/Tox/Crypto/Handlers.hs
index 6e1623de..13ee0ed2 100644
--- a/src/Network/Tox/Crypto/Handlers.hs
+++ b/src/Network/Tox/Crypto/Handlers.hs
@@ -24,7 +24,6 @@ import qualified Data.PacketQueue as PQ
 import Data.Serialize             as S
 import Data.Word
 import qualified Data.Word64Map as W64
-import GHC.Conc (unsafeIOToSTM)
 import qualified Data.Set         as Set
 import qualified Data.Array.Unboxed as A
 import SensibleDir
@@ -36,7 +35,6 @@ import System.Directory
 import System.Random -- for ping fuzz
 import Control.Concurrent
 import GHC.Conc (labelThread)
-import System.IO.Unsafe(unsafeDupablePerformIO {- unsafeIOToSTM -})
 import PingMachine
 
 -- util, todo: move to another module
@@ -341,7 +339,7 @@ freshCryptoSession  sessions
             f <- lookupNonceFunction crypto newsession theirSessionKey
             atomically $ do
                 n24 <- readTVar ncMyPacketNonce0
-                let n24plus1 = unsafeDupablePerformIO (incrementNonce24 n24)
+                let n24plus1 = incrementNonce24 n24
                 writeTVar ncMyPacketNonce0 n24plus1
                 return (return (f n24, n24, ncOutgoingIdMap0))
     pktoq <- atomically $ PQ.newOutGoing pktq ncToWire toWireIO 0 (outboundQueueCapacity sessions) 0
@@ -488,8 +486,8 @@ cryptoNetHandler sessions addr (NetCrypto (CryptoPacket nonce16 encrypted)) = do
             theirBaseNonce <- atomically $ readTVar ncTheirBaseNonce
             -- Try to decrypt message
             let diff :: Word16
-                diff = nonce16 - fromIntegral (last2Bytes theirBaseNonce) -- truncating to Word16
-            tempNonce <- addtoNonce24 theirBaseNonce (fromIntegral diff) -- expanding to Word
+                diff      = nonce16 - fromIntegral (last2Bytes theirBaseNonce) -- truncating to Word16
+                tempNonce = addtoNonce24 theirBaseNonce (fromIntegral diff)    -- expanding to Word
             lr <- fmap join $ sequence $ do -- Either Monad --
                         pubkey <- maybeToEither ncTheirSessionPublic
                         Right $ do -- IO Monad
@@ -508,8 +506,7 @@ cryptoNetHandler sessions addr (NetCrypto (CryptoPacket nonce16 encrypted)) = do
                     when (diff > 2 * dATA_NUM_THRESHOLD)$
                         atomically $ do
                             y <- readTVar ncTheirBaseNonce
-                            -- all because Storable forces IO...
-                            x <- unsafeIOToSTM $ addtoNonce24 y (fromIntegral dATA_NUM_THRESHOLD)
+                            let x = addtoNonce24 y (fromIntegral dATA_NUM_THRESHOLD)
                             writeTVar ncTheirBaseNonce y
                     -- then set session confirmed,
                     atomically $ writeTVar ncState Confirmed