{-# LANGUAGE FlexibleInstances #-}
module Crypto.Cipher.DES.Primitive (encrypt, decrypt, Block(..)) where
import Data.Word
import Data.Bits
newtype Block = Block Word64
type Rotation = Int
type Key = Word64
type Bits4 = [Bool]
type Bits6 = [Bool]
type Bits32 = [Bool]
type Bits48 = [Bool]
type Bits56 = [Bool]
type Bits64 = [Bool]
desXor :: [Bool] -> [Bool] -> [Bool]
desXor :: [Bool] -> [Bool] -> [Bool]
desXor a :: [Bool]
a b :: [Bool]
b = (Bool -> Bool -> Bool) -> [Bool] -> [Bool] -> [Bool]
forall a b c. (a -> b -> c) -> [a] -> [b] -> [c]
zipWith (\x :: Bool
x y :: Bool
y -> (Bool -> Bool
not Bool
x Bool -> Bool -> Bool
&& Bool
y) Bool -> Bool -> Bool
|| (Bool
x Bool -> Bool -> Bool
&& Bool -> Bool
not Bool
y)) [Bool]
a [Bool]
b
desRotate :: [Bool] -> Int -> [Bool]
desRotate :: [Bool] -> Int -> [Bool]
desRotate bits :: [Bool]
bits rot :: Int
rot = Int -> [Bool] -> [Bool]
forall a. Int -> [a] -> [a]
drop Int
rot' [Bool]
bits [Bool] -> [Bool] -> [Bool]
forall a. [a] -> [a] -> [a]
++ Int -> [Bool] -> [Bool]
forall a. Int -> [a] -> [a]
take Int
rot' [Bool]
bits
where rot' :: Int
rot' = Int
rot Int -> Int -> Int
forall a. Integral a => a -> a -> a
`mod` [Bool] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Bool]
bits
bitify :: Word64 -> Bits64
bitify :: Word64 -> [Bool]
bitify w :: Word64
w = (Int -> Bool) -> [Int] -> [Bool]
forall a b. (a -> b) -> [a] -> [b]
map (\b :: Int
b -> Word64
w Word64 -> Word64 -> Word64
forall a. Bits a => a -> a -> a
.&. (Word64 -> Int -> Word64
forall a. Bits a => a -> Int -> a
shiftL 1 Int
b) Word64 -> Word64 -> Bool
forall a. Eq a => a -> a -> Bool
/= 0) [63,62..0]
unbitify :: Bits64 -> Word64
unbitify :: [Bool] -> Word64
unbitify bs :: [Bool]
bs = (Word64 -> Bool -> Word64) -> Word64 -> [Bool] -> Word64
forall (t :: * -> *) b a.
Foldable t =>
(b -> a -> b) -> b -> t a -> b
foldl (\i :: Word64
i b :: Bool
b -> if Bool
b then 1 Word64 -> Word64 -> Word64
forall a. Num a => a -> a -> a
+ Word64 -> Int -> Word64
forall a. Bits a => a -> Int -> a
shiftL Word64
i 1 else Word64 -> Int -> Word64
forall a. Bits a => a -> Int -> a
shiftL Word64
i 1) 0 [Bool]
bs
initial_permutation :: Bits64 -> Bits64
initial_permutation :: [Bool] -> [Bool]
initial_permutation mb :: [Bool]
mb = (Int -> Bool) -> [Int] -> [Bool]
forall a b. (a -> b) -> [a] -> [b]
map ([Bool] -> Int -> Bool
forall a. [a] -> Int -> a
(!!) [Bool]
mb) [Int]
i
where i :: [Int]
i = [57, 49, 41, 33, 25, 17, 9, 1, 59, 51, 43, 35, 27, 19, 11, 3,
61, 53, 45, 37, 29, 21, 13, 5, 63, 55, 47, 39, 31, 23, 15, 7,
56, 48, 40, 32, 24, 16, 8, 0, 58, 50, 42, 34, 26, 18, 10, 2,
60, 52, 44, 36, 28, 20, 12, 4, 62, 54, 46, 38, 30, 22, 14, 6]
key_transformation :: Bits64 -> Bits56
key_transformation :: [Bool] -> [Bool]
key_transformation kb :: [Bool]
kb = (Int -> Bool) -> [Int] -> [Bool]
forall a b. (a -> b) -> [a] -> [b]
map ([Bool] -> Int -> Bool
forall a. [a] -> Int -> a
(!!) [Bool]
kb) [Int]
i
where i :: [Int]
i = [56, 48, 40, 32, 24, 16, 8, 0, 57, 49, 41, 33, 25, 17,
9, 1, 58, 50, 42, 34, 26, 18, 10, 2, 59, 51, 43, 35,
62, 54, 46, 38, 30, 22, 14, 6, 61, 53, 45, 37, 29, 21,
13, 5, 60, 52, 44, 36, 28, 20, 12, 4, 27, 19, 11, 3]
des_enc :: Block -> Key -> Block
des_enc :: Block -> Word64 -> Block
des_enc = [Int] -> Block -> Word64 -> Block
do_des [1,2,4,6,8,10,12,14,15,17,19,21,23,25,27,28]
des_dec :: Block -> Key -> Block
des_dec :: Block -> Word64 -> Block
des_dec = [Int] -> Block -> Word64 -> Block
do_des [28,27,25,23,21,19,17,15,14,12,10,8,6,4,2,1]
do_des :: [Rotation] -> Block -> Key -> Block
do_des :: [Int] -> Block -> Word64 -> Block
do_des rots :: [Int]
rots (Block m :: Word64
m) k :: Word64
k = Word64 -> Block
Block (Word64 -> Block) -> Word64 -> Block
forall a b. (a -> b) -> a -> b
$ [Int] -> ([Bool], [Bool]) -> [Bool] -> Word64
des_work [Int]
rots (Int -> [Bool] -> ([Bool], [Bool])
forall a. Int -> [a] -> ([a], [a])
takeDrop 32 [Bool]
mb) [Bool]
kb
where kb :: [Bool]
kb = [Bool] -> [Bool]
key_transformation ([Bool] -> [Bool]) -> [Bool] -> [Bool]
forall a b. (a -> b) -> a -> b
$ Word64 -> [Bool]
bitify Word64
k
mb :: [Bool]
mb = [Bool] -> [Bool]
initial_permutation ([Bool] -> [Bool]) -> [Bool] -> [Bool]
forall a b. (a -> b) -> a -> b
$ Word64 -> [Bool]
bitify Word64
m
des_work :: [Rotation] -> (Bits32, Bits32) -> Bits56 -> Word64
des_work :: [Int] -> ([Bool], [Bool]) -> [Bool] -> Word64
des_work [] (ml :: [Bool]
ml, mr :: [Bool]
mr) _ = [Bool] -> Word64
unbitify ([Bool] -> Word64) -> [Bool] -> Word64
forall a b. (a -> b) -> a -> b
$ [Bool] -> [Bool]
final_perm ([Bool] -> [Bool]) -> [Bool] -> [Bool]
forall a b. (a -> b) -> a -> b
$ ([Bool]
mr [Bool] -> [Bool] -> [Bool]
forall a. [a] -> [a] -> [a]
++ [Bool]
ml)
des_work (r :: Int
r:rs :: [Int]
rs) mb :: ([Bool], [Bool])
mb kb :: [Bool]
kb = [Int] -> ([Bool], [Bool]) -> [Bool] -> Word64
des_work [Int]
rs ([Bool], [Bool])
mb' [Bool]
kb
where mb' :: ([Bool], [Bool])
mb' = Int -> ([Bool], [Bool]) -> [Bool] -> ([Bool], [Bool])
do_round Int
r ([Bool], [Bool])
mb [Bool]
kb
do_round :: Rotation -> (Bits32, Bits32) -> Bits56 -> (Bits32, Bits32)
do_round :: Int -> ([Bool], [Bool]) -> [Bool] -> ([Bool], [Bool])
do_round r :: Int
r (ml :: [Bool]
ml, mr :: [Bool]
mr) kb :: [Bool]
kb = ([Bool]
mr, [Bool]
m')
where kb' :: [Bool]
kb' = [Bool] -> Int -> [Bool]
get_key [Bool]
kb Int
r
comp_kb :: [Bool]
comp_kb = [Bool] -> [Bool]
compression_permutation [Bool]
kb'
expa_mr :: [Bool]
expa_mr = [Bool] -> [Bool]
expansion_permutation [Bool]
mr
res :: [Bool]
res = [Bool]
comp_kb [Bool] -> [Bool] -> [Bool]
`desXor` [Bool]
expa_mr
res' :: [([Bool], [Bool])]
res' = [([Bool], [Bool])] -> [([Bool], [Bool])]
forall a. [a] -> [a]
tail ([([Bool], [Bool])] -> [([Bool], [Bool])])
-> [([Bool], [Bool])] -> [([Bool], [Bool])]
forall a b. (a -> b) -> a -> b
$ (([Bool], [Bool]) -> ([Bool], [Bool]))
-> ([Bool], [Bool]) -> [([Bool], [Bool])]
forall a. (a -> a) -> a -> [a]
iterate (Int -> ([Bool], [Bool]) -> ([Bool], [Bool])
forall a a. Int -> (a, [a]) -> ([a], [a])
trans 6) ([], [Bool]
res)
trans :: Int -> (a, [a]) -> ([a], [a])
trans n :: Int
n (_, b :: [a]
b) = (Int -> [a] -> [a]
forall a. Int -> [a] -> [a]
take Int
n [a]
b, Int -> [a] -> [a]
forall a. Int -> [a] -> [a]
drop Int
n [a]
b)
res_s :: [Bool]
res_s = [[Bool]] -> [Bool]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
concat ([[Bool]] -> [Bool]) -> [[Bool]] -> [Bool]
forall a b. (a -> b) -> a -> b
$ (([Bool] -> [Bool]) -> ([Bool], [Bool]) -> [Bool])
-> [[Bool] -> [Bool]] -> [([Bool], [Bool])] -> [[Bool]]
forall a b c. (a -> b -> c) -> [a] -> [b] -> [c]
zipWith (\f :: [Bool] -> [Bool]
f (x :: [Bool]
x,_) -> [Bool] -> [Bool]
f [Bool]
x) [[Bool] -> [Bool]
s_box_1, [Bool] -> [Bool]
s_box_2,
[Bool] -> [Bool]
s_box_3, [Bool] -> [Bool]
s_box_4,
[Bool] -> [Bool]
s_box_5, [Bool] -> [Bool]
s_box_6,
[Bool] -> [Bool]
s_box_7, [Bool] -> [Bool]
s_box_8] [([Bool], [Bool])]
res'
res_p :: [Bool]
res_p = [Bool] -> [Bool]
p_box [Bool]
res_s
m' :: [Bool]
m' = [Bool]
res_p [Bool] -> [Bool] -> [Bool]
`desXor` [Bool]
ml
get_key :: Bits56 -> Rotation -> Bits56
get_key :: [Bool] -> Int -> [Bool]
get_key kb :: [Bool]
kb r :: Int
r = [Bool]
kb'
where (kl :: [Bool]
kl, kr :: [Bool]
kr) = Int -> [Bool] -> ([Bool], [Bool])
forall a. Int -> [a] -> ([a], [a])
takeDrop 28 [Bool]
kb
kb' :: [Bool]
kb' = [Bool] -> Int -> [Bool]
desRotate [Bool]
kl Int
r [Bool] -> [Bool] -> [Bool]
forall a. [a] -> [a] -> [a]
++ [Bool] -> Int -> [Bool]
desRotate [Bool]
kr Int
r
compression_permutation :: Bits56 -> Bits48
compression_permutation :: [Bool] -> [Bool]
compression_permutation kb :: [Bool]
kb = (Int -> Bool) -> [Int] -> [Bool]
forall a b. (a -> b) -> [a] -> [b]
map ([Bool] -> Int -> Bool
forall a. [a] -> Int -> a
(!!) [Bool]
kb) [Int]
i
where i :: [Int]
i = [13, 16, 10, 23, 0, 4, 2, 27, 14, 5, 20, 9,
22, 18, 11, 3, 25, 7, 15, 6, 26, 19, 12, 1,
40, 51, 30, 36, 46, 54, 29, 39, 50, 44, 32, 47,
43, 48, 38, 55, 33, 52, 45, 41, 49, 35, 28, 31]
expansion_permutation :: Bits32 -> Bits48
expansion_permutation :: [Bool] -> [Bool]
expansion_permutation mb :: [Bool]
mb = (Int -> Bool) -> [Int] -> [Bool]
forall a b. (a -> b) -> [a] -> [b]
map ([Bool] -> Int -> Bool
forall a. [a] -> Int -> a
(!!) [Bool]
mb) [Int]
i
where i :: [Int]
i = [31, 0, 1, 2, 3, 4, 3, 4, 5, 6, 7, 8,
7, 8, 9, 10, 11, 12, 11, 12, 13, 14, 15, 16,
15, 16, 17, 18, 19, 20, 19, 20, 21, 22, 23, 24,
23, 24, 25, 26, 27, 28, 27, 28, 29, 30, 31, 0]
s_box :: [[Word8]] -> Bits6 -> Bits4
s_box :: [[Word8]] -> [Bool] -> [Bool]
s_box s :: [[Word8]]
s [a :: Bool
a,b :: Bool
b,c :: Bool
c,d :: Bool
d,e :: Bool
e,f :: Bool
f] = Int -> Word8 -> [Bool]
to_bool 4 (Word8 -> [Bool]) -> Word8 -> [Bool]
forall a b. (a -> b) -> a -> b
$ ([[Word8]]
s [[Word8]] -> Int -> [Word8]
forall a. [a] -> Int -> a
!! Int
row) [Word8] -> Int -> Word8
forall a. [a] -> Int -> a
!! Int
col
where row :: Int
row = [Int] -> Int
forall (t :: * -> *) a. (Foldable t, Num a) => t a -> a
sum ([Int] -> Int) -> [Int] -> Int
forall a b. (a -> b) -> a -> b
$ (Bool -> Int -> Int) -> [Bool] -> [Int] -> [Int]
forall a b c. (a -> b -> c) -> [a] -> [b] -> [c]
zipWith Bool -> Int -> Int
numericise [Bool
a,Bool
f] [1, 0]
col :: Int
col = [Int] -> Int
forall (t :: * -> *) a. (Foldable t, Num a) => t a -> a
sum ([Int] -> Int) -> [Int] -> Int
forall a b. (a -> b) -> a -> b
$ (Bool -> Int -> Int) -> [Bool] -> [Int] -> [Int]
forall a b c. (a -> b -> c) -> [a] -> [b] -> [c]
zipWith Bool -> Int -> Int
numericise [Bool
b,Bool
c,Bool
d,Bool
e] [3, 2, 1, 0]
numericise :: Bool -> Int -> Int
numericise :: Bool -> Int -> Int
numericise = (\x :: Bool
x y :: Int
y -> if Bool
x then 2Int -> Int -> Int
forall a b. (Num a, Integral b) => a -> b -> a
^Int
y else 0)
to_bool :: Int -> Word8 -> [Bool]
to_bool :: Int -> Word8 -> [Bool]
to_bool 0 _ = []
to_bool n :: Int
n i :: Word8
i = ((Word8
i Word8 -> Word8 -> Word8
forall a. Bits a => a -> a -> a
.&. 8) Word8 -> Word8 -> Bool
forall a. Eq a => a -> a -> Bool
== 8)Bool -> [Bool] -> [Bool]
forall a. a -> [a] -> [a]
:Int -> Word8 -> [Bool]
to_bool (Int
nInt -> Int -> Int
forall a. Num a => a -> a -> a
-1) (Word8 -> Int -> Word8
forall a. Bits a => a -> Int -> a
shiftL Word8
i 1)
s_box _ _ = [Char] -> [Bool]
forall a. HasCallStack => [Char] -> a
error "DES: internal error bits6 more than 6 elements"
s_box_1 :: Bits6 -> Bits4
s_box_1 :: [Bool] -> [Bool]
s_box_1 = [[Word8]] -> [Bool] -> [Bool]
s_box [[Word8]]
i
where i :: [[Word8]]
i = [[14, 4, 13, 1, 2, 15, 11, 8, 3, 10, 6, 12, 5, 9, 0, 7],
[ 0, 15, 7, 4, 14, 2, 13, 1, 10, 6, 12, 11, 9, 5, 3, 8],
[ 4, 1, 14, 8, 13, 6, 2, 11, 15, 12, 9, 7, 3, 10, 5, 0],
[15, 12, 8, 2, 4, 9, 1, 7, 5, 11, 3, 14, 10, 0, 6, 13]]
s_box_2 :: Bits6 -> Bits4
s_box_2 :: [Bool] -> [Bool]
s_box_2 = [[Word8]] -> [Bool] -> [Bool]
s_box [[Word8]]
i
where i :: [[Word8]]
i = [[15, 1, 8, 14, 6, 11, 3, 4, 9, 7, 2, 13, 12, 0, 5, 10],
[3, 13, 4, 7, 15, 2, 8, 14, 12, 0, 1, 10, 6, 9, 11, 5],
[0, 14, 7, 11, 10, 4, 13, 1, 5, 8, 12, 6, 9, 3, 2, 15],
[13, 8, 10, 1, 3, 15, 4, 2, 11, 6, 7, 12, 0, 5, 14, 9]]
s_box_3 :: Bits6 -> Bits4
s_box_3 :: [Bool] -> [Bool]
s_box_3 = [[Word8]] -> [Bool] -> [Bool]
s_box [[Word8]]
i
where i :: [[Word8]]
i = [[10, 0, 9, 14 , 6, 3, 15, 5, 1, 13, 12, 7, 11, 4, 2, 8],
[13, 7, 0, 9, 3, 4, 6, 10, 2, 8, 5, 14, 12, 11, 15, 1],
[13, 6, 4, 9, 8, 15, 3, 0, 11, 1, 2, 12, 5, 10, 14, 7],
[1, 10, 13, 0, 6, 9, 8, 7, 4, 15, 14, 3, 11, 5, 2, 12]]
s_box_4 :: Bits6 -> Bits4
s_box_4 :: [Bool] -> [Bool]
s_box_4 = [[Word8]] -> [Bool] -> [Bool]
s_box [[Word8]]
i
where i :: [[Word8]]
i = [[7, 13, 14, 3, 0, 6, 9, 10, 1, 2, 8, 5, 11, 12, 4, 15],
[13, 8, 11, 5, 6, 15, 0, 3, 4, 7, 2, 12, 1, 10, 14, 9],
[10, 6, 9, 0, 12, 11, 7, 13, 15, 1, 3, 14, 5, 2, 8, 4],
[3, 15, 0, 6, 10, 1, 13, 8, 9, 4, 5, 11, 12, 7, 2, 14]]
s_box_5 :: Bits6 -> Bits4
s_box_5 :: [Bool] -> [Bool]
s_box_5 = [[Word8]] -> [Bool] -> [Bool]
s_box [[Word8]]
i
where i :: [[Word8]]
i = [[2, 12, 4, 1, 7, 10, 11, 6, 8, 5, 3, 15, 13, 0, 14, 9],
[14, 11, 2, 12, 4, 7, 13, 1, 5, 0, 15, 10, 3, 9, 8, 6],
[4, 2, 1, 11, 10, 13, 7, 8, 15, 9, 12, 5, 6, 3, 0, 14],
[11, 8, 12, 7, 1, 14, 2, 13, 6, 15, 0, 9, 10, 4, 5, 3]]
s_box_6 :: Bits6 -> Bits4
s_box_6 :: [Bool] -> [Bool]
s_box_6 = [[Word8]] -> [Bool] -> [Bool]
s_box [[Word8]]
i
where i :: [[Word8]]
i = [[12, 1, 10, 15, 9, 2, 6, 8, 0, 13, 3, 4, 14, 7, 5, 11],
[10, 15, 4, 2, 7, 12, 9, 5, 6, 1, 13, 14, 0, 11, 3, 8],
[9, 14, 15, 5, 2, 8, 12, 3, 7, 0, 4, 10, 1, 13, 11, 6],
[4, 3, 2, 12, 9, 5, 15, 10, 11, 14, 1, 7, 6, 0, 8, 13]]
s_box_7 :: Bits6 -> Bits4
s_box_7 :: [Bool] -> [Bool]
s_box_7 = [[Word8]] -> [Bool] -> [Bool]
s_box [[Word8]]
i
where i :: [[Word8]]
i = [[4, 11, 2, 14, 15, 0, 8, 13, 3, 12, 9, 7, 5, 10, 6, 1],
[13, 0, 11, 7, 4, 9, 1, 10, 14, 3, 5, 12, 2, 15, 8, 6],
[1, 4, 11, 13, 12, 3, 7, 14, 10, 15, 6, 8, 0, 5, 9, 2],
[6, 11, 13, 8, 1, 4, 10, 7, 9, 5, 0, 15, 14, 2, 3, 12]]
s_box_8 :: Bits6 -> Bits4
s_box_8 :: [Bool] -> [Bool]
s_box_8 = [[Word8]] -> [Bool] -> [Bool]
s_box [[Word8]]
i
where i :: [[Word8]]
i = [[13, 2, 8, 4, 6, 15, 11, 1, 10, 9, 3, 14, 5, 0, 12, 7],
[1, 15, 13, 8, 10, 3, 7, 4, 12, 5, 6, 11, 0, 14, 9, 2],
[7, 11, 4, 1, 9, 12, 14, 2, 0, 6, 10, 13, 15, 3, 5, 8],
[2, 1, 14, 7, 4, 10, 8, 13, 15, 12, 9, 0, 3, 5, 6, 11]]
p_box :: Bits32 -> Bits32
p_box :: [Bool] -> [Bool]
p_box kb :: [Bool]
kb = (Int -> Bool) -> [Int] -> [Bool]
forall a b. (a -> b) -> [a] -> [b]
map ([Bool] -> Int -> Bool
forall a. [a] -> Int -> a
(!!) [Bool]
kb) [Int]
i
where i :: [Int]
i = [15, 6, 19, 20, 28, 11, 27, 16, 0, 14, 22, 25, 4, 17, 30, 9,
1, 7, 23, 13, 31, 26, 2, 8, 18, 12, 29, 5, 21, 10, 3, 24]
final_perm :: Bits64 -> Bits64
final_perm :: [Bool] -> [Bool]
final_perm kb :: [Bool]
kb = (Int -> Bool) -> [Int] -> [Bool]
forall a b. (a -> b) -> [a] -> [b]
map ([Bool] -> Int -> Bool
forall a. [a] -> Int -> a
(!!) [Bool]
kb) [Int]
i
where i :: [Int]
i = [39, 7, 47, 15, 55, 23, 63, 31, 38, 6, 46, 14, 54, 22, 62, 30,
37, 5, 45, 13, 53, 21, 61, 29, 36, 4, 44, 12, 52, 20, 60, 28,
35, 3, 43, 11, 51, 19, 59, 27, 34, 2, 42, 10, 50, 18, 58, 26,
33, 1, 41, 9, 49, 17, 57, 25, 32, 0, 40 , 8, 48, 16, 56, 24]
takeDrop :: Int -> [a] -> ([a], [a])
takeDrop :: Int -> [a] -> ([a], [a])
takeDrop _ [] = ([], [])
takeDrop 0 xs :: [a]
xs = ([], [a]
xs)
takeDrop n :: Int
n (x :: a
x:xs :: [a]
xs) = (a
xa -> [a] -> [a]
forall a. a -> [a] -> [a]
:[a]
ys, [a]
zs)
where (ys :: [a]
ys, zs :: [a]
zs) = Int -> [a] -> ([a], [a])
forall a. Int -> [a] -> ([a], [a])
takeDrop (Int
nInt -> Int -> Int
forall a. Num a => a -> a -> a
-1) [a]
xs
encrypt :: Word64 -> Block -> Block
encrypt :: Word64 -> Block -> Block
encrypt = (Block -> Word64 -> Block) -> Word64 -> Block -> Block
forall a b c. (a -> b -> c) -> b -> a -> c
flip Block -> Word64 -> Block
des_enc
decrypt :: Word64 -> Block -> Block
decrypt :: Word64 -> Block -> Block
decrypt = (Block -> Word64 -> Block) -> Word64 -> Block -> Block
forall a b c. (a -> b -> c) -> b -> a -> c
flip Block -> Word64 -> Block
des_dec