crypton-1.1.4: tests/KAT_CMAC.hs

module KAT_CMAC (tests) where

import Crypto.Cipher.AES (AES128, AES192, AES256)
import Crypto.Cipher.TripleDES (DES_EDE2, DES_EDE3)
import Crypto.Cipher.Types (
    BlockCipher,
    Cipher,
    blockSize,
    cipherInit,
    ecbEncrypt,
 )
import Crypto.Error (eitherCryptoError)
import qualified Crypto.MAC.CMAC as CMAC

import Imports

import qualified Data.ByteArray as B
import qualified Data.ByteString as BS
import Data.Char (digitToInt)

hxs :: String -> ByteString
hxs = BS.pack . rec'
  where
    dtoW8 = fromIntegral . digitToInt
    rec' (' ' : xs) = rec' xs
    rec' (x : y : xs) = dtoW8 x * 16 + dtoW8 y : rec' xs
    rec' [_] = error "hxs: invalid hex pattern."
    rec' [] = []

unsafeCipher :: Cipher k => ByteString -> k
unsafeCipher = either (error . show) id . eitherCryptoError . cipherInit

ecb0 :: BlockCipher k => k -> ByteString
ecb0 k = ecbEncrypt k $ BS.replicate (blockSize k) 0

{- Test vectors from NIST data-sheet
   (AES128-CMAC, AES192-CMAC, AES256-CMAC, Three Key TDEA, Two Key TDEA)
   http://csrc.nist.gov/publications/nistpubs/800-38B/Updated_CMAC_Examples.pdf
   The data of AES128-CMAC is same as them in RFC4493.
 -}

msg512 :: ByteString
msg512 =
    hxs $
        "6bc1bee2 2e409f96 e93d7e11 7393172a"
            ++ "ae2d8a57 1e03ac9c 9eb76fac 45af8e51"
            ++ "30c81c46 a35ce411 e5fbc119 1a0a52ef"
            ++ "f69f2445 df4f9b17 ad2b417b e66c3710"

msg320 :: ByteString
msg320 = BS.take 40 msg512

msg256 :: ByteString
msg256 = BS.take 32 msg512

msg160 :: ByteString
msg160 = BS.take 20 msg512

msg128 :: ByteString
msg128 = BS.take 16 msg512

msg64 :: ByteString
msg64 = BS.take 8 msg512

msg0 :: ByteString
msg0 = BS.empty

bsCMAC :: BlockCipher k => k -> ByteString -> ByteString
bsCMAC k = B.convert . CMAC.cmac k

gAES128 :: TestTree
gAES128 =
    igroup
        "aes128"
        [ ecb0 aes128key @?= hxs "7df76b0c 1ab899b3 3e42f047 b91b546f"
        , aes128k1 @?= hxs "fbeed618 35713366 7c85e08f 7236a8de"
        , aes128k2 @?= hxs "f7ddac30 6ae266cc f90bc11e e46d513b"
        , bsCMAC aes128key msg0
            @?= hxs "bb1d6929 e9593728 7fa37d12 9b756746"
        , bsCMAC aes128key msg128
            @?= hxs "070a16b4 6b4d4144 f79bdd9d d04a287c"
        , bsCMAC aes128key msg320
            @?= hxs "dfa66747 de9ae630 30ca3261 1497c827"
        , bsCMAC aes128key msg512
            @?= hxs "51f0bebf 7e3b9d92 fc497417 79363cfe"
        ]
  where
    aes128key :: AES128
    aes128key =
        unsafeCipher $
            hxs
                "2b7e1516 28aed2a6 abf71588 09cf4f3c"

    aes128k1, aes128k2 :: ByteString
    (aes128k1, aes128k2) = CMAC.subKeys aes128key

gAES192 :: TestTree
gAES192 =
    igroup
        "aes192"
        [ ecb0 aes192key @?= hxs "22452d8e 49a8a593 9f7321ce ea6d514b"
        , aes192k1 @?= hxs "448a5b1c 93514b27 3ee6439d d4daa296"
        , aes192k2 @?= hxs "8914b639 26a2964e 7dcc873b a9b5452c"
        , bsCMAC aes192key msg0
            @?= hxs "d17ddf46 adaacde5 31cac483 de7a9367"
        , bsCMAC aes192key msg128
            @?= hxs "9e99a7bf 31e71090 0662f65e 617c5184"
        , bsCMAC aes192key msg320
            @?= hxs "8a1de5be 2eb31aad 089a82e6 ee908b0e"
        , bsCMAC aes192key msg512
            @?= hxs "a1d5df0e ed790f79 4d775896 59f39a11"
        ]
  where
    aes192key :: AES192
    aes192key =
        unsafeCipher . hxs $
            "8e73b0f7 da0e6452 c810f32b 809079e5"
                ++ "62f8ead2 522c6b7b"

    aes192k1, aes192k2 :: ByteString
    (aes192k1, aes192k2) = CMAC.subKeys aes192key

gAES256 :: TestTree
gAES256 =
    igroup
        "aes256"
        [ ecb0 aes256key @?= hxs "e568f681 94cf76d6 174d4cc0 4310a854"
        , aes256k1 @?= hxs "cad1ed03 299eedac 2e9a9980 8621502f"
        , aes256k2 @?= hxs "95a3da06 533ddb58 5d353301 0c42a0d9"
        , bsCMAC aes256key msg0
            @?= hxs "028962f6 1b7bf89e fc6b551f 4667d983"
        , bsCMAC aes256key msg128
            @?= hxs "28a7023f 452e8f82 bd4bf28d 8c37c35c"
        , bsCMAC aes256key msg320
            @?= hxs "aaf3d8f1 de5640c2 32f5b169 b9c911e6"
        , bsCMAC aes256key msg512
            @?= hxs "e1992190 549f6ed5 696a2c05 6c315410"
        ]
  where
    aes256key :: AES256
    aes256key =
        unsafeCipher . hxs $
            "603deb10 15ca71be 2b73aef0 857d7781"
                ++ "1f352c07 3b6108d7 2d9810a3 0914dff4"

    aes256k1, aes256k2 :: ByteString
    (aes256k1, aes256k2) = CMAC.subKeys aes256key

gTDEA3 :: TestTree
gTDEA3 =
    igroup
        "Three Key TDEA"
        [ ecb0 tdea3key @?= hxs "c8cc74e9 8a7329a2"
        , tdea3k1 @?= hxs "9198e9d3 14e6535f"
        , tdea3k2 @?= hxs "2331d3a6 29cca6a5"
        , bsCMAC tdea3key msg0
            @?= hxs "b7a688e1 22ffaf95"
        , bsCMAC tdea3key msg64
            @?= hxs "8e8f2931 36283797"
        , bsCMAC tdea3key msg160
            @?= hxs "743ddbe0 ce2dc2ed"
        , bsCMAC tdea3key msg256
            @?= hxs "33e6b109 2400eae5"
        ]
  where
    tdea3key :: DES_EDE3
    tdea3key =
        unsafeCipher . hxs $
            "8aa83bf8 cbda1062"
                ++ "0bc1bf19 fbb6cd58"
                ++ "bc313d4a 371ca8b5"

    tdea3k1, tdea3k2 :: ByteString
    (tdea3k1, tdea3k2) = CMAC.subKeys tdea3key

gTDEA2 :: TestTree
gTDEA2 =
    igroup
        "Two Key TDEA"
        [ ecb0 tdea2key @?= hxs "c7679b9f 6b8d7d7a"
        , tdea2k1 @?= hxs "8ecf373e d71afaef"
        , tdea2k2 @?= hxs "1d9e6e7d ae35f5c5"
        , bsCMAC tdea2key msg0
            @?= hxs "bd2ebf9a 3ba00361"
        , bsCMAC tdea2key msg64
            @?= hxs "4ff2ab81 3c53ce83"
        , bsCMAC tdea2key msg160
            @?= hxs "62dd1b47 1902bd4e"
        , bsCMAC tdea2key msg256
            @?= hxs "31b1e431 dabc4eb8"
        ]
  where
    tdea2key :: DES_EDE2
    tdea2key =
        unsafeCipher . hxs $
            "4cf15134 a2850dd5"
                ++ "8a3d10ba 80570d38"

    tdea2k1, tdea2k2 :: ByteString
    (tdea2k1, tdea2k2) = CMAC.subKeys tdea2key

igroup :: TestName -> [Assertion] -> TestTree
igroup nm = testGroup nm . zipWith (flip ($)) [1 ..] . map icase
  where
    icase c i = testCase (show (i :: Int)) c

nistVectors :: TestTree
nistVectors =
    testGroup
        "KAT - NIST test vectors"
        [gAES128, gAES192, gAES256, gTDEA3, gTDEA2]

tests :: TestTree
tests =
    testGroup
        "CMAC"
        [nistVectors]