mirror of
https://github.com/kazu-yamamoto/crypton.git
synced 2024-10-06 00:37:20 +03:00
78 lines
2.9 KiB
Haskell
78 lines
2.9 KiB
Haskell
{-# LANGUAGE OverloadedStrings #-}
|
|
module KAT_AES (tests) where
|
|
|
|
import Imports
|
|
import BlockCipher
|
|
import Crypto.Cipher.Types
|
|
import qualified Crypto.Cipher.AES as AES
|
|
|
|
import qualified KAT_AES.KATECB as KATECB
|
|
import qualified KAT_AES.KATCBC as KATCBC
|
|
import qualified KAT_AES.KATXTS as KATXTS
|
|
import qualified KAT_AES.KATGCM as KATGCM
|
|
import qualified KAT_AES.KATOCB3 as KATOCB3
|
|
|
|
{-
|
|
instance Show AES.AES where
|
|
show _ = "AES"
|
|
instance Arbitrary AES.AESIV where
|
|
arbitrary = AES.aesIV_ . B.pack <$> replicateM 16 arbitrary
|
|
instance Arbitrary AES.AES where
|
|
arbitrary = AES.initAES . B.pack <$> replicateM 16 arbitrary
|
|
-}
|
|
|
|
toKatECB (k,p,c) = KAT_ECB { ecbKey = k, ecbPlaintext = p, ecbCiphertext = c }
|
|
toKatCBC (k,iv,p,c) = KAT_CBC { cbcKey = k, cbcIV = iv, cbcPlaintext = p, cbcCiphertext = c }
|
|
toKatXTS (k1,k2,iv,p,_,c) = KAT_XTS { xtsKey1 = k1, xtsKey2 = k2, xtsIV = iv, xtsPlaintext = p, xtsCiphertext = c }
|
|
toKatAEAD mode (k,iv,h,p,c,taglen,tag) =
|
|
KAT_AEAD { aeadMode = mode
|
|
, aeadKey = k
|
|
, aeadIV = iv
|
|
, aeadHeader = h
|
|
, aeadPlaintext = p
|
|
, aeadCiphertext = c
|
|
, aeadTaglen = taglen
|
|
, aeadTag = tag
|
|
}
|
|
toKatGCM = toKatAEAD AEAD_GCM
|
|
toKatOCB = toKatAEAD AEAD_OCB
|
|
|
|
kats128 = defaultKATs
|
|
{ kat_ECB = map toKatECB KATECB.vectors_aes128_enc
|
|
, kat_CBC = map toKatCBC KATCBC.vectors_aes128_enc
|
|
, kat_CFB = [ KAT_CFB { cfbKey = "\x2b\x7e\x15\x16\x28\xae\xd2\xa6\xab\xf7\x15\x88\x09\xcf\x4f\x3c"
|
|
, cfbIV = "\xC8\xA6\x45\x37\xA0\xB3\xA9\x3F\xCD\xE3\xCD\xAD\x9F\x1C\xE5\x8B"
|
|
, cfbPlaintext = "\x30\xc8\x1c\x46\xa3\x5c\xe4\x11\xe5\xfb\xc1\x19\x1a\x0a\x52\xef"
|
|
, cfbCiphertext = "\x26\x75\x1f\x67\xa3\xcb\xb1\x40\xb1\x80\x8c\xf1\x87\xa4\xf4\xdf"
|
|
}
|
|
]
|
|
, kat_XTS = map toKatXTS KATXTS.vectors_aes128_enc
|
|
, kat_AEAD = map toKatGCM KATGCM.vectors_aes128_enc ++
|
|
map toKatOCB KATOCB3.vectors_aes128_enc
|
|
}
|
|
|
|
kats192 = defaultKATs
|
|
{ kat_ECB = map toKatECB KATECB.vectors_aes192_enc
|
|
, kat_CBC = map toKatCBC KATCBC.vectors_aes192_enc
|
|
}
|
|
|
|
kats256 = defaultKATs
|
|
{ kat_ECB = map toKatECB KATECB.vectors_aes256_enc
|
|
, kat_CBC = map toKatCBC KATCBC.vectors_aes256_enc
|
|
, kat_XTS = map toKatXTS KATXTS.vectors_aes256_enc
|
|
, kat_AEAD = map toKatGCM KATGCM.vectors_aes256_enc
|
|
}
|
|
|
|
tests = testGroup "AES"
|
|
[ testBlockCipher kats128 (undefined :: AES.AES128)
|
|
, testBlockCipher kats192 (undefined :: AES.AES192)
|
|
, testBlockCipher kats256 (undefined :: AES.AES256)
|
|
{-
|
|
, testProperty "genCtr" $ \(key, iv1) ->
|
|
let (bs1, iv2) = AES.genCounter key iv1 32
|
|
(bs2, iv3) = AES.genCounter key iv2 32
|
|
(bsAll, iv3') = AES.genCounter key iv1 64
|
|
in (B.concat [bs1,bs2] == bsAll && iv3 == iv3')
|
|
-}
|
|
]
|