earthbucks_lib 0.8.5

EarthBucks library for data structures and algorithms
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
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

use crate::error::EbxError;
use crate::priv_key::PrivKey;
use crate::pub_key::PubKey;

#[derive(Clone, Debug)]
pub struct KeyPair {
    pub priv_key: PrivKey,
    pub pub_key: PubKey,
}

impl KeyPair {
    pub fn new(priv_key: [u8; 32]) -> Result<Self, EbxError> {
        let priv_key = PrivKey::new(priv_key);
        let pub_key = PubKey::from_priv_key(&priv_key);
        if pub_key.is_err() {
            return Err(pub_key.err().unwrap());
        }
        Ok(KeyPair {
            priv_key,
            pub_key: pub_key.unwrap(),
        })
    }

    pub fn from_priv_key(priv_key: &PrivKey) -> Result<Self, EbxError> {
        let pub_key_res = PubKey::from_priv_key(priv_key);
        if pub_key_res.is_err() {
            return Err(EbxError::InvalidKeyError { source: None });
        }
        Ok(KeyPair {
            priv_key: priv_key.clone(),
            pub_key: pub_key_res.unwrap(),
        })
    }

    pub fn from_random() -> Self {
        let priv_key = PrivKey::from_random();
        KeyPair::from_priv_key(&priv_key).unwrap()
    }

    pub fn is_valid(&self) -> bool {
        let pub_key = PubKey::from_priv_key(&self.priv_key);
        if pub_key.is_err() {
            return false;
        }
        self.pub_key.is_valid() && pub_key.unwrap().buf == self.pub_key.buf
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use serde::Deserialize;
    use std::fs;

    #[test]
    fn test_is_valid() {
        let key_pair = KeyPair::from_random();
        assert!(key_pair.is_valid());
    }

    // standard test vectors: key_pair.json
    #[derive(Deserialize)]
    struct JsonKeyPair {
        priv_key: String,
        pub_key: String,
    }

    #[derive(Deserialize)]
    struct JsonKeyPairs {
        key_pair: Vec<JsonKeyPair>,
    }

    #[test]
    fn test_key_pairs() {
        let data = fs::read_to_string("./test_vectors/key_pair.json").expect("Unable to read file");
        let key_pairs: JsonKeyPairs = serde_json::from_str(&data).expect("Unable to parse JSON");

        for pair in key_pairs.key_pair {
            let priv_key: PrivKey = PrivKey::from_strict_str(&pair.priv_key).unwrap();
            let key_pair = KeyPair::from_priv_key(&priv_key);
            let pub_key = key_pair.unwrap().pub_key;

            let expected_public_key = &pair.pub_key;
            let actual_public_key = pub_key.to_strict_str();

            assert_eq!(expected_public_key, &actual_public_key);
        }
    }
}