import Crypto.Cipher.AES128
import Crypto.Cipher.AES
import Crypto.Classes
import Crypto.Modes (zeroIV)
import Criterion
import Criterion.Main
import System.Entropy
import Data.Serialize
import qualified Data.ByteString as B
main = do
let iv = zeroIV
ivV = IV (B.replicate 16 0)
pt <- getEntropy (2^12)
k <- buildKeyIO :: IO AESKey
let kV = initKey (B.pack [0..15])
defaultMain
[ bench "aes-ctr cipher-aes128" $ nf (fst . ctr k iv) pt
, bench "aes-ctr cipher-aes" $ nf (encryptCTR kV ivV) pt
, bench "aes-gcm cipher-aes" $ nf (encryptGCM kV ivV B.empty) pt]