fpe-0.1.3: test/ff1test.hs

{-# OPTIONS_GHC -Wno-name-shadowing -Wno-incomplete-uni-patterns #-}

import FPE.FF1 as FF1
import Crypto.Cipher.Types
import Crypto.Cipher.AES
import Crypto.Error
import Control.Monad (when)

import qualified Data.Vector.Unboxed as V
import qualified Data.ByteString as B


--  The first three samples from
--  https://csrc.nist.gov/CSRC/media/Projects/Cryptographic-Standards-and-Guidelines/documents/examples/FF1samples.pdf

key :: AES128
CryptoPassed key = cipherInit $ B.pack [
  0x2b, 0x7e, 0x15, 0x16, 0x28, 0xae, 0xd2, 0xa6,
  0xab, 0xf7, 0x15, 0x88, 0x09, 0xcf, 0x4f, 0x3c]

main :: IO ()
main = do
   let cipher = ecbEncrypt key
   let tweak = mempty
   let plain = V.fromList @Int [0..9]
   let crypt = FF1.encrypt cipher 10 tweak plain
   print crypt
   when (V.toList crypt /= [2,4,3,3,4,7,7,4,8,4]) $ error "bad encrypt"
   when (FF1.decrypt cipher 10 tweak crypt /= plain) $ error "bad decrypt"
   let tweak = B.pack [ 0x39, 0x38 .. 0x30]
   let crypt = FF1.encrypt cipher 10 tweak plain
   print crypt
   when (V.toList crypt /= [6,1,2,4,2,0,0,7,7,3]) $ error "bad tweak encrypt"
   let tweak = B.pack [0x37, 0x37, 0x37, 0x37, 0x70, 0x71, 0x72, 0x73, 0x37, 0x37, 0x37]
   let plain = V.fromList @Int [0..18]
   let crypt = FF1.encrypt cipher 36 tweak plain
   print crypt
   when (V.toList crypt /= [10,9,29,31,4,0,22,21,21,9,20,13,30,5,0,9,14,30,22])
     $ error "bad radix-36 encrypt"
   when (FF1.decrypt cipher 36 tweak crypt /= plain) $ error "bad radix-36 decrypt"