mirror of
https://github.com/simplex-chat/simplexmq.git
synced 2026-03-31 14:06:07 +00:00
99 lines
3.2 KiB
Haskell
99 lines
3.2 KiB
Haskell
{-# LANGUAGE OverloadedStrings #-}
|
|
|
|
module CoreTests.CryptoFileTests (cryptoFileTests) where
|
|
|
|
import AgentTests.FunctionalAPITests (runRight_)
|
|
import Control.Monad.Except
|
|
import Control.Monad.IO.Class
|
|
import Crypto.Random (getRandomBytes)
|
|
import qualified Data.ByteString.Lazy as LB
|
|
import GHC.IO.IOMode (IOMode (..))
|
|
import qualified Simplex.FileTransfer.Types as C
|
|
import Simplex.Messaging.Crypto.File (CryptoFile (..), FTCryptoError (..))
|
|
import qualified Simplex.Messaging.Crypto.File as CF
|
|
import System.Directory (getFileSize)
|
|
import Test.Hspec
|
|
|
|
cryptoFileTests :: Spec
|
|
cryptoFileTests = do
|
|
it "should write/read file" testWriteReadFile
|
|
it "should put/get file" testPutGetFile
|
|
it "should write/get file" testWriteGetFile
|
|
it "should put/read file" testPutReadFile
|
|
it "should fail reading empty or small file" testSmallFile
|
|
|
|
testFilePath :: FilePath
|
|
testFilePath = "tests/tmp/testcryptofile"
|
|
|
|
testWriteReadFile :: IO ()
|
|
testWriteReadFile = do
|
|
s <- LB.fromStrict <$> getRandomBytes 100000
|
|
file <- mkCryptoFile
|
|
runRight_ $ do
|
|
CF.writeFile file s
|
|
liftIO $ CF.getFileContentsSize file `shouldReturn` 100000
|
|
liftIO $ getFileSize testFilePath `shouldReturn` 100000 + fromIntegral C.authTagSize
|
|
s' <- CF.readFile file
|
|
liftIO $ s `shouldBe` s'
|
|
|
|
testPutGetFile :: IO ()
|
|
testPutGetFile = do
|
|
s <- LB.fromStrict <$> getRandomBytes 50000
|
|
s' <- LB.fromStrict <$> getRandomBytes 50000
|
|
file <- mkCryptoFile
|
|
runRight_ $ do
|
|
CF.withFile file WriteMode $ \h -> liftIO $ do
|
|
CF.hPut h s
|
|
CF.hPut h s'
|
|
CF.hPutTag h
|
|
liftIO $ CF.getFileContentsSize file `shouldReturn` 100000
|
|
liftIO $ getFileSize testFilePath `shouldReturn` 100000 + fromIntegral C.authTagSize
|
|
CF.withFile file ReadMode $ \h -> do
|
|
s1 <- liftIO $ CF.hGet h 30000
|
|
s2 <- liftIO $ CF.hGet h 40000
|
|
s3 <- liftIO $ CF.hGet h 30000
|
|
CF.hGetTag h
|
|
liftIO $ (s <> s') `shouldBe` LB.fromStrict (s1 <> s2 <> s3)
|
|
|
|
testWriteGetFile :: IO ()
|
|
testWriteGetFile = do
|
|
s <- LB.fromStrict <$> getRandomBytes 100000
|
|
file <- mkCryptoFile
|
|
runRight_ $ do
|
|
CF.writeFile file s
|
|
CF.withFile file ReadMode $ \h -> do
|
|
s' <- liftIO $ CF.hGet h 50000
|
|
s'' <- liftIO $ CF.hGet h 50000
|
|
CF.hGetTag h
|
|
liftIO $ runExceptT (CF.hGetTag h) `shouldReturn` Left FTCEInvalidAuthTag
|
|
liftIO $ s `shouldBe` LB.fromStrict (s' <> s'')
|
|
|
|
testPutReadFile :: IO ()
|
|
testPutReadFile = do
|
|
s <- LB.fromStrict <$> getRandomBytes 50000
|
|
s' <- LB.fromStrict <$> getRandomBytes 50000
|
|
file <- mkCryptoFile
|
|
runRight_ $ do
|
|
CF.withFile file WriteMode $ \h -> liftIO $ do
|
|
CF.hPut h s
|
|
CF.hPut h s'
|
|
runExceptT (CF.readFile file) `shouldReturn` Left FTCEInvalidAuthTag
|
|
runRight_ $ do
|
|
CF.withFile file WriteMode $ \h -> liftIO $ do
|
|
CF.hPut h s
|
|
CF.hPut h s'
|
|
CF.hPutTag h
|
|
s'' <- CF.readFile file
|
|
liftIO $ (s <> s') `shouldBe` s''
|
|
|
|
testSmallFile :: IO ()
|
|
testSmallFile = do
|
|
file <- mkCryptoFile
|
|
LB.writeFile testFilePath ""
|
|
runExceptT (CF.readFile file) `shouldReturn` Left FTCEInvalidFileSize
|
|
LB.writeFile testFilePath "123"
|
|
runExceptT (CF.readFile file) `shouldReturn` Left FTCEInvalidFileSize
|
|
|
|
mkCryptoFile :: IO CryptoFile
|
|
mkCryptoFile = CryptoFile testFilePath . Just <$> CF.randomArgs
|