libr2fa 0.1.3

rust implementation for HTOP, TOTP and steam guard tow-factor-authentication
Documentation 
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

use std::{fmt::Display, rc::Rc};

use serde::{Deserialize, Serialize};

use crate::error;

/// HMACType is the type of the HMAC
/// SHA1 is the default
/// SHA256 is the recommended
/// SHA512 is the most secure
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash, Serialize, Deserialize, Default)]
pub enum HMACType {
    #[default]
    SHA1,
    SHA256,
    SHA512,
}

impl Display for HMACType {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        f.write_str(self.get_digest_name())
    }
}

impl From<String> for HMACType {
    fn from(s: String) -> Self {
        match s.to_ascii_lowercase().as_str() {
            "sha1" => HMACType::SHA1,
            "sha256" => HMACType::SHA256,
            "sha512" => HMACType::SHA512,
            _ => HMACType::default(),
        }
    }
}

impl HMACType {
    fn get_digest_name(&self) -> &'static str {
        match self {
            HMACType::SHA1 => "sha1",
            HMACType::SHA256 => "sha256",
            HMACType::SHA512 => "sha512",
        }
    }

    fn get_algorithm(&self) -> ring::hmac::Algorithm {
        match self {
            HMACType::SHA1 => ring::hmac::HMAC_SHA1_FOR_LEGACY_USE_ONLY,
            HMACType::SHA256 => ring::hmac::HMAC_SHA256,
            HMACType::SHA512 => ring::hmac::HMAC_SHA512,
        }
    }

    /// get_hash returns the hash of the key and the string
    pub fn get_hash(&self, key: &[u8], s: &[u8]) -> Result<Rc<[u8]>, error::Error> {
        let algorithm = self.get_algorithm();
        let signer = ring::hmac::Key::new(algorithm, key);
        let hmac = ring::hmac::sign(&signer, s);
        let block = hmac.as_ref();

        Ok(Rc::from(block))
    }
}