1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
use crate::{utils::xor, GostPadding};
use block_modes::{block_padding::Padding, BlockMode};
use cipher::{Block, BlockCipher, BlockDecrypt, BlockEncrypt, NewBlockCipher};
use core::{marker::PhantomData, ops::Mul};
use generic_array::typenum::{
type_operators::{IsGreater, IsLessOrEqual},
Prod, Unsigned, U0, U1, U255,
};
use generic_array::{ArrayLength, GenericArray};
#[derive(Clone)]
pub struct GostCbc<C, P = GostPadding, Z = U1>
where
C: BlockCipher + NewBlockCipher,
C::BlockSize: IsLessOrEqual<U255>,
Z: ArrayLength<Block<C>> + Unsigned + Mul<C::BlockSize> + IsGreater<U0> + IsLessOrEqual<U255>,
Prod<Z, C::BlockSize>: ArrayLength<u8>,
P: Padding,
{
cipher: C,
state: GenericArray<Block<C>, Z>,
pos: u8,
_p: PhantomData<P>,
}
impl<C, P, Z> BlockMode<C, P> for GostCbc<C, P, Z>
where
C: BlockCipher + BlockEncrypt + BlockDecrypt + NewBlockCipher,
C::BlockSize: IsLessOrEqual<U255>,
Z: ArrayLength<Block<C>> + Unsigned + Mul<C::BlockSize> + IsGreater<U0> + IsLessOrEqual<U255>,
Prod<Z, C::BlockSize>: ArrayLength<u8>,
P: Padding,
{
type IvSize = Prod<Z, C::BlockSize>;
fn new(cipher: C, iv: &GenericArray<u8, Self::IvSize>) -> Self {
let bs = C::BlockSize::USIZE;
let mut state = GenericArray::<Block<C>, Z>::default();
for (block, chunk) in state.iter_mut().zip(iv.chunks_exact(bs)) {
*block = GenericArray::clone_from_slice(chunk);
}
Self {
cipher,
state,
pos: 0,
_p: Default::default(),
}
}
fn encrypt_blocks(&mut self, blocks: &mut [Block<C>]) {
for block in blocks {
let sb = &mut self.state[self.pos as usize];
xor(block, sb);
self.cipher.encrypt_block(block);
*sb = block.clone();
self.pos += 1;
self.pos %= Z::U8;
}
}
fn decrypt_blocks(&mut self, blocks: &mut [Block<C>]) {
for block in blocks {
let pos = self.pos as usize;
let b = self.state[pos].clone();
self.state[pos] = block.clone();
self.cipher.decrypt_block(block);
xor(block, &b);
self.pos += 1;
self.pos %= Z::U8;
}
}
}