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
82
83
84
85
86
87
88
89


extern crate crypto;
extern crate hex;
use crypto::sha1::Sha1;
use crypto::digest::Digest;
use std::convert::TryInto;
use std::convert::TryFrom;


mod adverbs;
mod adjectives;
mod nouns;

#[cfg(feature = "default-implementations")]
pub mod default_implementations;

#[cfg(test)]
mod tests;

/// Obfuscates the input string into a human readable hash.
///
/// # Arguments
/// * `input` - The string to obfuscate.
///
/// # Examples
///
/// ```
/// let result: String = logged_fu_skater::obf("hello");
/// assert_eq!(&result, "TerriblyHolyThrill");
/// ```
///
pub fn obf(input: &str) -> String {
    obfp(input, 0)
}

/// Obfuscates the input string into a human readable hash with 0-8 padding_bytes at the end.
///
/// # Arguments
/// * `input` - The string to obfuscate.
/// * `padding_bytes` - Number of bytes to add as padding (0-8).
///
/// # Examples
///
/// ```
/// let result: String = logged_fu_skater::obfp("hello", 2);
/// assert_eq!(&result, "TerriblyHolyThrill3B48");
/// ```
///
/// # Panics
///
/// Panics if padding_bytes is greater than 8.
///
/// ```rust,should_panic
/// logged_fu_skater::obfp("hello", 99);
/// ```
///
pub fn obfp(input: &str, padding_bytes: u8) -> String {
    if padding_bytes > 8 {
        panic!("padding_bytes cannot exceed 8.");
    }
    let mut hasher = Sha1::new();
    hasher.input_str(input);
    let mut result: [u8;20] = [0;20];
    hasher.result(&mut result);

    let adverbs_index = usize::try_from(u32::from_le_bytes(result[0..4].try_into()
        .expect("Could not convert slice to array")))
        .expect("Could not convert u32 to usize");
    let adjectives_index = usize::try_from(u32::from_le_bytes(result[4..8].try_into()
        .expect("Could not convert slice to array")))
        .expect("Could not convert u32 to usize");
    let nouns_index = usize::try_from(u32::from_le_bytes(result[8..12].try_into()
        .expect("Could not convert slice to array")))
        .expect("Could not convert u32 to usize");

    let adverb = adverbs::WORDS[adverbs_index % adverbs::WORDS.len()];
    let adjective = adjectives::WORDS[adjectives_index % adjectives::WORDS.len()];
    let noun = nouns::WORDS[nouns_index % nouns::WORDS.len()];

    let padding: Vec<u8> = result[12..(12 + padding_bytes as usize)].try_into().expect("Could not convert padding to Vec");

    format!("{}{}{}{}", adverb, adjective, noun, hex::encode_upper(padding))
}

/// Trait to implement obfuscate functionality on relevant data types.
pub trait Obfuscateable {
    fn obf(&self) -> String;
    fn obfp(&self, padding_bytes: u8) -> String;
}