{-# LANGUAGE OverloadedStrings, CPP #-}
module KAT (katTests) where
import Test.Framework.Providers.QuickCheck2 (testProperty)
import qualified Data.ByteString as B
import qualified Data.ByteString.Char8 as BC
import Data.Word
import qualified Crypto.Cipher.Blowfish as Blowfish
import qualified Crypto.Cipher.Camellia as Camellia
import qualified Crypto.Cipher.RC4 as RC4
encryptStream fi fc key plaintext = B.unpack $ snd $ fc (fi $ B.pack key) plaintext
encryptBlock fi fc key plaintext = B.unpack $ fc (fi $ B.pack key) plaintext
unright (Right r) = r
unright (Left e) = error e
wordify :: [Char] -> [Word8]
wordify = map (toEnum . fromEnum)
vectors_rc4 =
[ (wordify "Key", "Plaintext", [ 0xBB,0xF3,0x16,0xE8,0xD9,0x40,0xAF,0x0A,0xD3 ])
, (wordify "Wiki", "pedia", [ 0x10,0x21,0xBF,0x04,0x20 ])
, (wordify "Secret", "Attack at dawn", [ 0x45,0xA0,0x1F,0x64,0x5F,0xC3,0x5B,0x38,0x35,0x52,0x54,0x4B,0x9B,0xF5 ])
]
vectors_camellia128 =
[
( replicate 16 0
, B.replicate 16 0
, [0x3d,0x02,0x80,0x25,0xb1,0x56,0x32,0x7c,0x17,0xf7,0x62,0xc1,0xf2,0xcb,0xca,0x71]
)
, ( [0x01,0x23,0x45,0x67,0x89,0xab,0xcd,0xef,0xfe,0xdc,0xba,0x98,0x76,0x54,0x32,0x10]
, B.pack [0x01,0x23,0x45,0x67,0x89,0xab,0xcd,0xef,0xfe,0xdc,0xba,0x98,0x76,0x54,0x32,0x10]
, [0x67,0x67,0x31,0x38,0x54,0x96,0x69,0x73,0x08,0x57,0x06,0x56,0x48,0xea,0xbe,0x43]
)
]
vectors_camellia192 =
[
( [0x01,0x23,0x45,0x67,0x89,0xab,0xcd,0xef,0xfe,0xdc,0xba,0x98,0x76,0x54,0x32,0x10,0x00,0x11,0x22,0x33,0x44,0x55,0x66,0x77]
, B.pack [0x01,0x23,0x45,0x67,0x89,0xab,0xcd,0xef,0xfe,0xdc,0xba,0x98,0x76,0x54,0x32,0x10]
,[0xb4,0x99,0x34,0x01,0xb3,0xe9,0x96,0xf8,0x4e,0xe5,0xce,0xe7,0xd7,0x9b,0x09,0xb9]
)
]
vectors_camellia256 =
[
( [0x01,0x23,0x45,0x67,0x89,0xab,0xcd,0xef,0xfe,0xdc,0xba,0x98,0x76,0x54,0x32,0x10
,0x00,0x11,0x22,0x33,0x44,0x55,0x66,0x77,0x88,0x99,0xaa,0xbb,0xcc,0xdd,0xee,0xff]
, B.pack [0x01,0x23,0x45,0x67,0x89,0xab,0xcd,0xef,0xfe,0xdc,0xba,0x98,0x76,0x54,0x32,0x10]
, [0x9a,0xcc,0x23,0x7d,0xff,0x16,0xd7,0x6c,0x20,0xef,0x7c,0x91,0x9e,0x3a,0x75,0x09]
)
]
vectors_blowfish =
[
( replicate 8 0
, B.replicate 8 0
, [0x4e,0xf9,0x97,0x45,0x61,0x98,0xDD,0x78]
)
, ( replicate 8 255
, B.replicate 8 255
, [0x51,0x86,0x6F,0xD5,0xB8,0x5E,0xCB,0x8A]
)
, ( [0x7C,0xA1,0x10,0x45,0x4A,0x1A,0x6E,0x57]
, B.pack [0x01,0xA1,0xD6,0xD0,0x39,0x77,0x67,0x42]
, [0x59,0xC6,0x82,0x45,0xEB,0x05,0x28,0x2B]
)
]
vectors =
[ ("RC4", vectors_rc4, encryptStream RC4.initCtx RC4.combine)
-- Blowfish implementation
, ("Blowfish", vectors_blowfish, encryptBlock (unright . Blowfish.initKey) Blowfish.encrypt)
-- Camellia implementation
, ("Camellia", vectors_camellia128, encryptBlock (unright . Camellia.initKey128) Camellia.encrypt)
]
katTests = map makeTests vectors
where makeTests (name, v, f) = testProperty name (and $ map makeTest v)
where makeTest (key,plaintext,expected) = assertEq expected $ f key plaintext
assertEq expected got
| expected == got = True
| otherwise = error ("expected: " ++ show expected ++ " got: " ++ show got)