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

use std::fmt;

use puffin::codec::{Codec, Reader};

/// This type contains a private key by value.
///
/// The private key must be DER-encoded ASN.1 in either
/// PKCS#8 or PKCS#1 format.
///
/// The `rustls-pemfile` crate can be used to extract
/// private keys from a PEM file in these formats.
#[derive(Debug, Clone, Eq, PartialEq)]
pub struct PrivateKey(pub Vec<u8>);

/// This type contains a single certificate by value.
///
/// The certificate must be DER-encoded X.509.
///
/// The `rustls-pemfile` crate can be used to parse a PEM file.
#[derive(Clone, Eq, Hash, Ord, PartialEq, PartialOrd)]
pub struct Certificate(pub Vec<u8>);

impl Codec for PrivateKey {
    fn encode(&self, bytes: &mut Vec<u8>) {
        bytes.append(&mut self.0.clone())
    }

    fn read(r: &mut Reader) -> Option<Self> {
        <Vec<u8> as Codec>::read(r).map(PrivateKey)
    }
}

impl AsRef<[u8]> for Certificate {
    fn as_ref(&self) -> &[u8] {
        &self.0
    }
}

impl fmt::Debug for Certificate {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        f.debug_tuple("Certificate").field(&self.0).finish()
    }
}

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

    #[test_log::test]
    fn certificate_debug() {
        assert_eq!(
            "Certificate([97, 98])",
            format!("{:?}", Certificate(b"ab".to_vec()))
        );
    }
}