russh/mac/
crypto_etm.rs

1use std::marker::PhantomData;
2
3use digest::KeyInit;
4use generic_array::{ArrayLength, GenericArray};
5
6use super::crypto::{CryptoMac, CryptoMacAlgorithm};
7use super::{Mac, MacAlgorithm};
8
9pub struct CryptoEtmMacAlgorithm<
10    M: digest::Mac + KeyInit + Send + 'static,
11    KL: ArrayLength<u8> + 'static,
12>(pub PhantomData<M>, pub PhantomData<KL>);
13
14impl<M: digest::Mac + KeyInit + Send + 'static, KL: ArrayLength<u8> + 'static> MacAlgorithm
15    for CryptoEtmMacAlgorithm<M, KL>
16{
17    fn key_len(&self) -> usize {
18        CryptoMacAlgorithm::<M, KL>(self.0, self.1).key_len()
19    }
20
21    fn make_mac(&self, mac_key: &[u8]) -> Box<dyn Mac + Send> {
22        let mut key = GenericArray::<u8, KL>::default();
23        key.clone_from_slice(mac_key);
24        Box::new(CryptoEtmMac::<M, KL>(CryptoMac::<M, KL> {
25            key,
26            p: PhantomData,
27        })) as Box<dyn Mac + Send>
28    }
29}
30
31pub struct CryptoEtmMac<M: digest::Mac + KeyInit + Send + 'static, KL: ArrayLength<u8> + 'static>(
32    CryptoMac<M, KL>,
33);
34
35impl<M: digest::Mac + KeyInit + Send + 'static, KL: ArrayLength<u8> + 'static> Mac
36    for CryptoEtmMac<M, KL>
37{
38    fn is_etm(&self) -> bool {
39        true
40    }
41
42    fn mac_len(&self) -> usize {
43        self.0.mac_len()
44    }
45
46    fn compute(&self, sequence_number: u32, payload: &[u8], output: &mut [u8]) {
47        self.0.compute(sequence_number, payload, output)
48    }
49
50    fn verify(&self, sequence_number: u32, payload: &[u8], mac: &[u8]) -> bool {
51        self.0.verify(sequence_number, payload, mac)
52    }
53}