{-# LANGUAGE OverloadedStrings #-}

module Poly1305 (tests) where

import qualified Data.ByteString as B

import qualified Data.ByteString.Char8 as B ()

import Imports

import Crypto.Error

import qualified Crypto.MAC.Poly1305 as Poly1305

import qualified Data.ByteArray as B (convert)

instance Show Poly1305.Auth where

    show _ = "Auth"

data Chunking = Chunking Int Int

    deriving (Show,Eq)

instance Arbitrary Chunking where

    arbitrary = Chunking <$> choose (1,34) <*> choose (1,2048)

tests = testGroup "Poly1305"

    [ testCase "V0" $

        let key = "\x85\xd6\xbe\x78\x57\x55\x6d\x33\x7f\x44\x52\xfe\x42\xd5\x06\xa8\x01\x03\x80\x8a\xfb\x0d\xb2\xfd\x4a\xbf\xf6\xaf\x41\x49\xf5\x1b" :: ByteString

            msg = "Cryptographic Forum Research Group" :: ByteString

            tag = "\xa8\x06\x1d\xc1\x30\x51\x36\xc6\xc2\x2b\x8b\xaf\x0c\x01\x27\xa9" :: ByteString

         in tag @=? B.convert (Poly1305.auth key msg)

    , testProperty "Chunking" $ \(Chunking chunkLen totalLen) ->

        let key = B.replicate 32 0

            msg = B.pack $ take totalLen $ concat (replicate 10 [1..255])

         in Poly1305.auth key msg == Poly1305.finalize (foldr (flip Poly1305.update) (throwCryptoError $ Poly1305.initialize key) (chunks chunkLen msg))

    ]

  where

        chunks i bs

            | B.length bs < i = [bs]

            | otherwise       = let (b1,b2) = B.splitAt i bs in b1 : chunks i b2