nginx-secure-link 0.1.1

Library to generate or verify checksums for use with the Nginx secure_link_module
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

use base64;
use md5;

pub struct SecureLink<'a> {
    key: &'a str,
}

impl SecureLink<'_> {
    pub fn new(key: &str) -> SecureLink {
        return SecureLink { key };
    }

    pub fn generate(self, url: &str) -> String {
        let digest = md5::compute(format!("{} {}", url, self.key));
        return base64::encode_config(digest.as_slice(), base64::URL_SAFE_NO_PAD);
    }

    pub fn verify(self, url: &str, key: &str) -> bool {
        return key == self.generate(url);
    }
}

#[cfg(test)]
mod tests {
    use super::SecureLink;

    #[test]
    fn test_key_generation() {
        let slink = SecureLink::new("testkey");
        let key = slink.generate("https://www.google.com/");
        assert_eq!(key, "HY4On4h_ZipVaOa0uOHtOg");
    }

    #[test]
    fn test_key_verification() {
        let slink = SecureLink::new("testkey");
        assert_eq!(
            slink.verify("https://www.google.com/", "HY4On4h_ZipVaOa0uOHtOg"),
            true
        );
    }

    #[test]
    fn test_bad_key_verification() {
        let slink = SecureLink::new("testkey");
        assert_eq!(slink.verify("https://www.google.com/", "invalidKey"), false);
    }
}