{-# LANGUAGE BangPatterns #-}
import Criterion.Main
import qualified Crypto.Hash.SHA256 as SHA256
import qualified Data.ByteString as B
import qualified Data.ByteString.Lazy as L
benchSize :: Int -> Benchmark
benchSize sz = bs `seq` bench msg (whnf SHA256.hash bs)
where
bs = B.replicate sz 0
msg = "bs-" ++ show sz
main :: IO ()
main = do
let !lbs64x256 = L.fromChunks $ replicate 4 (B.replicate 64 0)
!lbs64x4096 = L.fromChunks $ replicate 64 (B.replicate 64 0)
defaultMain
[ bgroup "cryptohash-sha256"
[ benchSize 0
, benchSize 8
, benchSize 32
, benchSize 64
, benchSize 128
, benchSize 256
, benchSize 1024
, benchSize 4096
, benchSize (128*1024)
, benchSize (1024*1024)
, benchSize (2*1024*1024)
, benchSize (4*1024*1024)
, L.length lbs64x256 `seq` bench "lbs64x256" (whnf SHA256.hashlazy lbs64x256)
, L.length lbs64x4096 `seq` bench "lbs64x4096" (whnf SHA256.hashlazy lbs64x4096)
]
]