proksi 0.10.0

A batteries-included reverse proxy with automatic HTTPS using Cloudflare Pingora and Let's Encrypt.
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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111

use anyhow::anyhow;
use openssl::{pkey::PKey, x509::X509};

use crate::{
    config::{host::Host, RouteSslPath},
    stores::{self, certificates::Certificate},
};

/// Loads the certificate of a host that has one in files (`tls.certificate`) into the store
pub async fn load_host_certificate(host: &Host) -> Result<(), anyhow::Error> {
    match &host.tls.certificate {
        Some(files) => load_certificate(&host.name, files).await,
        None => Ok(()),
    }
}

/// Loads a certificate from files into the store, under the lowercase `name`
pub(super) async fn load_certificate(
    name: &str,
    files: &RouteSslPath,
) -> Result<(), anyhow::Error> {
    let key = std::fs::read(&files.key)
        .map_err(anyhow::Error::from)
        .and_then(|key| Ok(PKey::private_key_from_pem(&key)?))
        .map_err(|err| anyhow!("failed to load private key from {:?}: {err}", files.key))?;
    // The leaf, followed by its chain
    let mut certificates = std::fs::read(&files.pem)
        .map_err(anyhow::Error::from)
        .and_then(|pem| Ok(X509::stack_from_pem(&pem)?))
        .map_err(|err| anyhow!("failed to load certificate from {:?}: {err}", files.pem))?
        .into_iter();
    let leaf = certificates
        .next()
        .ok_or_else(|| anyhow!("no certificate in {:?}", files.pem))?;
    // Files being replaced may hold a certificate and the previous key for a moment
    if !leaf.public_key().is_ok_and(|public| public.public_eq(&key)) {
        return Err(anyhow!(
            "the key in {:?} doesn't match the certificate in {:?}",
            files.key,
            files.pem
        ));
    }

    stores::global::get_store()
        .set_certificate(
            &name.to_lowercase(),
            Certificate {
                key,
                leaf,
                chain: certificates.collect(),
            },
        )
        .await
        .map_err(|err| anyhow!("failed to store certificate for {name}: {err}"))
}

/// The contents of a host's certificate files, to know when they change
pub fn certificate_files_fingerprint(host: &Host) -> Option<u64> {
    let files = host.tls.certificate.as_ref()?;
    let mut hasher = std::hash::DefaultHasher::new();
    std::hash::Hash::hash(&std::fs::read(&files.pem).ok()?, &mut hasher);
    std::hash::Hash::hash(&std::fs::read(&files.key).ok()?, &mut hasher);
    Some(std::hash::Hasher::finish(&hasher))
}

#[cfg(test)]
mod tests {
    use crate::config::{host::Host, RouteSslPath};

    #[tokio::test]
    async fn test_load_host_certificate() {
        let dir = tempfile::tempdir().unwrap();
        let key =
            openssl::pkey::PKey::from_rsa(openssl::rsa::Rsa::generate(2048).unwrap()).unwrap();
        let mut cert = openssl::x509::X509::builder().unwrap();
        cert.set_pubkey(&key).unwrap();
        cert.set_not_before(&openssl::asn1::Asn1Time::days_from_now(0).unwrap())
            .unwrap();
        cert.set_not_after(&openssl::asn1::Asn1Time::days_from_now(1).unwrap())
            .unwrap();
        cert.sign(&key, openssl::hash::MessageDigest::sha256())
            .unwrap();
        let cert = cert.build();
        std::fs::write(
            dir.path().join("key.pem"),
            key.private_key_to_pem_pkcs8().unwrap(),
        )
        .unwrap();
        std::fs::write(dir.path().join("cert.pem"), cert.to_pem().unwrap()).unwrap();

        let mut host = crate::routing::test_utils::host("Host-Certificate.Test", "10.0.0.1");
        host.tls.certificate = Some(RouteSslPath {
            key: dir.path().join("key.pem"),
            pem: dir.path().join("cert.pem"),
        });

        super::load_host_certificate(&host).await.unwrap();

        let stored = crate::stores::global::get_store()
            .get_certificate("host-certificate.test")
            .await
            .unwrap();
        assert_eq!(stored.leaf.to_pem().unwrap(), cert.to_pem().unwrap());
    }

    #[tokio::test]
    async fn test_host_without_certificate_files() {
        let host: Host = crate::routing::test_utils::host("no-certificate.test", "10.0.0.1");
        assert!(super::load_host_certificate(&host).await.is_ok());
    }
}