ppad-poly1305-0.4.3: bench/Main.hs

{-# OPTIONS_GHC -fno-warn-orphans #-}
{-# LANGUAGE BangPatterns #-}
{-# LANGUAGE OverloadedStrings #-}

module Main where

import Control.DeepSeq
import Criterion.Main
import qualified Crypto.MAC.Poly1305 as Poly1305
import qualified Data.ByteString as BS
import qualified Data.ByteString.Base16 as B16
import Data.Maybe (fromJust)

instance NFData Poly1305.MAC where
  rnf (Poly1305.MAC b) = rnf b

main :: IO ()
main = defaultMain [
    suite
  ]

msg :: BS.ByteString
msg = fromJust . B16.decode $
  "4c616469657320616e642047656e746c656d656e206f662074686520636c617373206f66202739393a204966204920636f756c64206f6666657220796f75206f6e6c79206f6e652074697020666f7220746865206675747572652c2073756e73637265656e20776f756c642062652069742e"

key_small :: BS.ByteString
key_small = fromJust . B16.decode $
  "0000000000000000000000000000000000000000000000000000000000000003"

key_mid :: BS.ByteString
key_mid = fromJust . B16.decode $
  "8888888888888888888888888888888888888888888888888888888888888883"

key_big :: BS.ByteString
key_big = fromJust . B16.decode $
  "fffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff3"

msg_1k :: BS.ByteString
msg_1k = BS.replicate 1024 0x42

msg_4k :: BS.ByteString
msg_4k = BS.replicate 4096 0x42

suite :: Benchmark
suite =
  let !mac0 = fromJust (Poly1305.mac key_big msg)
      !mac1 = fromJust (Poly1305.mac key_big msg_1k)
      !mac2 = fromJust (Poly1305.mac key_big msg)
  in  bgroup "ppad-poly1305" [
        bench "mac (small key)" $ nf (Poly1305.mac key_small) msg
      , bench "mac (mid key)" $ nf (Poly1305.mac key_mid) msg
      , bench "mac (big key)" $ nf (Poly1305.mac key_big) msg
      , bench "mac (1024B msg)" $ nf (Poly1305.mac key_big) msg_1k
      , bench "mac (4096B msg)" $ nf (Poly1305.mac key_big) msg_4k
      , bgroup "MAC comparison" [
          bench "mac, unequal" $ whnf (mac0 ==) mac1
        , bench "mac, equal" $ whnf (mac0 ==) mac2
        ]
      ]