rustls_acme/
acme.rs

1use crate::any_ecdsa_type;
2use crate::crypto::error::{KeyRejected, Unspecified};
3use crate::crypto::rand::SystemRandom;
4use crate::crypto::signature::{EcdsaKeyPair, EcdsaSigningAlgorithm, ECDSA_P256_SHA256_FIXED_SIGNING};
5use crate::https_helper::{https, HttpsRequestError};
6use crate::jose::{key_authorization, key_authorization_sha256, sign, JoseError};
7use base64::prelude::*;
8use futures_rustls::pki_types::{PrivateKeyDer, PrivatePkcs8KeyDer};
9use futures_rustls::rustls::{sign::CertifiedKey, ClientConfig};
10use http::header::ToStrError;
11use http::{Method, Response};
12use rcgen::{CustomExtension, KeyPair, PKCS_ECDSA_P256_SHA256};
13use serde::{Deserialize, Serialize};
14use serde_json::json;
15use std::sync::Arc;
16use thiserror::Error;
17
18pub const LETS_ENCRYPT_STAGING_DIRECTORY: &str = "https://acme-staging-v02.api.letsencrypt.org/directory";
19pub const LETS_ENCRYPT_PRODUCTION_DIRECTORY: &str = "https://acme-v02.api.letsencrypt.org/directory";
20pub const ACME_TLS_ALPN_NAME: &[u8] = b"acme-tls/1";
21
22#[derive(Debug)]
23pub struct Account {
24    pub key_pair: EcdsaKeyPair,
25    pub directory: Directory,
26    pub kid: String,
27}
28
29static ALG: &EcdsaSigningAlgorithm = &ECDSA_P256_SHA256_FIXED_SIGNING;
30
31impl Account {
32    pub fn generate_key_pair() -> Vec<u8> {
33        let rng = SystemRandom::new();
34        let pkcs8 = EcdsaKeyPair::generate_pkcs8(ALG, &rng).unwrap();
35        pkcs8.as_ref().to_vec()
36    }
37    pub async fn create<'a, S, I>(client_config: &Arc<ClientConfig>, directory: Directory, contact: I) -> Result<Self, AcmeError>
38    where
39        S: AsRef<str> + 'a,
40        I: IntoIterator<Item = &'a S>,
41    {
42        let key_pair = Self::generate_key_pair();
43        Self::create_with_keypair(client_config, directory, contact, &key_pair).await
44    }
45    pub async fn create_with_keypair<'a, S, I>(
46        client_config: &Arc<ClientConfig>,
47        directory: Directory,
48        contact: I,
49        key_pair: &[u8],
50    ) -> Result<Self, AcmeError>
51    where
52        S: AsRef<str> + 'a,
53        I: IntoIterator<Item = &'a S>,
54    {
55        let key_pair = EcdsaKeyPair::from_pkcs8(
56            ALG,
57            key_pair,
58            // ring 0.17 has a third argument here; aws-lc-rs doesn't.
59            #[cfg(all(feature = "ring", not(feature = "aws-lc-rs")))]
60            &SystemRandom::new(),
61        )?;
62        let contact: Vec<&'a str> = contact.into_iter().map(AsRef::<str>::as_ref).collect();
63        let payload = json!({
64            "termsOfServiceAgreed": true,
65            "contact": contact,
66        })
67        .to_string();
68        let body = sign(&key_pair, None, directory.nonce(client_config).await?, &directory.new_account, &payload)?;
69        let response = https(client_config, &directory.new_account, Method::POST, Some(body)).await?;
70        let kid = get_header(&response, "Location")?;
71        Ok(Account { key_pair, kid, directory })
72    }
73    async fn request(&self, client_config: &Arc<ClientConfig>, url: impl AsRef<str>, payload: &str) -> Result<(Option<String>, String), AcmeError> {
74        let body = sign(
75            &self.key_pair,
76            Some(&self.kid),
77            self.directory.nonce(client_config).await?,
78            url.as_ref(),
79            payload,
80        )?;
81        let response = https(client_config, url.as_ref(), Method::POST, Some(body)).await?;
82        let location = get_header(&response, "Location").ok();
83        let body = response.into_body();
84        log::debug!("response: {:?}", body);
85        Ok((location, body))
86    }
87    pub async fn new_order(&self, client_config: &Arc<ClientConfig>, domains: Vec<String>) -> Result<(String, Order), AcmeError> {
88        let domains: Vec<Identifier> = domains.into_iter().map(Identifier::Dns).collect();
89        let payload = format!("{{\"identifiers\":{}}}", serde_json::to_string(&domains)?);
90        let response = self.request(client_config, &self.directory.new_order, &payload).await?;
91        let url = response.0.ok_or(AcmeError::MissingHeader("Location"))?;
92        let order = serde_json::from_str(&response.1)?;
93        Ok((url, order))
94    }
95    pub async fn auth(&self, client_config: &Arc<ClientConfig>, url: impl AsRef<str>) -> Result<Auth, AcmeError> {
96        let payload = "".to_string();
97        let response = self.request(client_config, url, &payload).await?;
98        Ok(serde_json::from_str(&response.1)?)
99    }
100    pub async fn challenge(&self, client_config: &Arc<ClientConfig>, url: impl AsRef<str>) -> Result<(), AcmeError> {
101        self.request(client_config, &url, "{}").await?;
102        Ok(())
103    }
104    pub async fn order(&self, client_config: &Arc<ClientConfig>, url: impl AsRef<str>) -> Result<Order, AcmeError> {
105        let response = self.request(client_config, &url, "").await?;
106        Ok(serde_json::from_str(&response.1)?)
107    }
108    pub async fn finalize(&self, client_config: &Arc<ClientConfig>, url: impl AsRef<str>, csr: &[u8]) -> Result<Order, AcmeError> {
109        let payload = format!("{{\"csr\":\"{}\"}}", BASE64_URL_SAFE_NO_PAD.encode(csr));
110        let response = self.request(client_config, &url, &payload).await?;
111        Ok(serde_json::from_str(&response.1)?)
112    }
113    pub async fn certificate(&self, client_config: &Arc<ClientConfig>, url: impl AsRef<str>) -> Result<String, AcmeError> {
114        Ok(self.request(client_config, &url, "").await?.1)
115    }
116    pub fn tls_alpn_01<'a>(&self, challenges: &'a [Challenge], domain: String) -> Result<(&'a Challenge, CertifiedKey), AcmeError> {
117        let challenge = challenges.iter().find(|c| c.typ == ChallengeType::TlsAlpn01);
118        let challenge = match challenge {
119            Some(challenge) => challenge,
120            None => return Err(AcmeError::NoTlsAlpn01Challenge),
121        };
122        let mut params = rcgen::CertificateParams::new(vec![domain])?;
123        let key_auth = key_authorization_sha256(&self.key_pair, &challenge.token)?;
124        params.custom_extensions = vec![CustomExtension::new_acme_identifier(key_auth.as_ref())];
125        let key_pair = KeyPair::generate_for(&PKCS_ECDSA_P256_SHA256)?;
126        let cert = params.self_signed(&key_pair)?;
127
128        let sk = any_ecdsa_type(&PrivateKeyDer::Pkcs8(PrivatePkcs8KeyDer::from(key_pair.serialize_der()))).unwrap();
129        let certified_key = CertifiedKey::new(vec![cert.der().clone()], sk);
130        Ok((challenge, certified_key))
131    }
132    pub fn http_01<'a>(&self, challenges: &'a [Challenge]) -> Result<(&'a Challenge, String), AcmeError> {
133        let challenge = challenges.iter().find(|c| c.typ == ChallengeType::Http01);
134        let challenge = match challenge {
135            Some(challenge) => challenge,
136            None => return Err(AcmeError::NoHttp01Challenge),
137        };
138        let key_auth = key_authorization(&self.key_pair, &challenge.token)?;
139        Ok((challenge, key_auth))
140    }
141}
142
143#[derive(Debug, Clone, Deserialize)]
144#[serde(rename_all = "camelCase")]
145pub struct Directory {
146    pub new_nonce: String,
147    pub new_account: String,
148    pub new_order: String,
149}
150
151impl Directory {
152    pub async fn discover(client_config: &Arc<ClientConfig>, url: impl AsRef<str>) -> Result<Self, AcmeError> {
153        let body = https(client_config, url, Method::GET, None).await?.into_body();
154        Ok(serde_json::from_str(&body)?)
155    }
156    pub async fn nonce(&self, client_config: &Arc<ClientConfig>) -> Result<String, AcmeError> {
157        let response = &https(client_config, &self.new_nonce.as_str(), Method::HEAD, None).await?;
158        get_header(response, "replay-nonce")
159    }
160}
161
162#[derive(Debug, Deserialize, Eq, PartialEq)]
163pub enum ChallengeType {
164    #[serde(rename = "http-01")]
165    Http01,
166    #[serde(rename = "dns-01")]
167    Dns01,
168    #[serde(rename = "tls-alpn-01")]
169    TlsAlpn01,
170}
171
172#[derive(Debug, Deserialize)]
173#[serde(rename_all = "camelCase")]
174pub struct Order {
175    #[serde(flatten)]
176    pub status: OrderStatus,
177    pub authorizations: Vec<String>,
178    pub finalize: String,
179    pub error: Option<Problem>,
180}
181
182#[derive(Debug, Deserialize, Clone, PartialEq, Eq)]
183#[serde(tag = "status", rename_all = "camelCase")]
184pub enum OrderStatus {
185    Pending,
186    Ready,
187    Valid { certificate: String },
188    Invalid,
189    Processing,
190}
191
192#[derive(Debug, Deserialize)]
193#[serde(rename_all = "camelCase")]
194pub struct Auth {
195    pub status: AuthStatus,
196    pub identifier: Identifier,
197    pub challenges: Vec<Challenge>,
198}
199
200#[derive(Debug, Deserialize)]
201#[serde(rename_all = "camelCase")]
202pub enum AuthStatus {
203    Pending,
204    Valid,
205    Invalid,
206    Revoked,
207    Expired,
208    Deactivated,
209}
210
211#[derive(Clone, Debug, Serialize, Deserialize)]
212#[serde(tag = "type", content = "value", rename_all = "camelCase")]
213pub enum Identifier {
214    Dns(String),
215}
216
217#[derive(Debug, Deserialize)]
218pub struct Challenge {
219    #[serde(rename = "type")]
220    pub typ: ChallengeType,
221    pub url: String,
222    pub token: String,
223    pub error: Option<Problem>,
224}
225
226#[derive(Clone, Debug, Serialize, Deserialize)]
227#[serde(rename_all = "camelCase")]
228pub struct Problem {
229    #[serde(rename = "type")]
230    pub typ: Option<String>,
231    pub detail: Option<String>,
232}
233
234#[derive(Error, Debug)]
235pub enum AcmeError {
236    #[error("io error: {0}")]
237    Io(#[from] std::io::Error),
238    #[error("certificate generation error: {0}")]
239    Rcgen(#[from] rcgen::Error),
240    #[error("JOSE error: {0}")]
241    Jose(#[from] JoseError),
242    #[error("JSON error: {0}")]
243    Json(#[from] serde_json::Error),
244    #[error("http request error: {0}")]
245    HttpRequest(#[from] HttpsRequestError),
246    #[error("non-string http response header: {0}")]
247    HttpResponseNonStringHeader(#[from] ToStrError),
248    #[error("invalid key pair: {0}")]
249    KeyRejected(#[from] KeyRejected),
250    #[error("crypto error: {0}")]
251    Crypto(#[from] Unspecified),
252    #[error("acme service response is missing {0} header")]
253    MissingHeader(&'static str),
254    #[error("no TLS-ALPN-01 challenge found")]
255    NoTlsAlpn01Challenge,
256    #[error("no HTTP-01 challenge found")]
257    NoHttp01Challenge,
258}
259
260impl From<http::Error> for AcmeError {
261    fn from(e: http::Error) -> Self {
262        Self::HttpRequest(HttpsRequestError::from(e))
263    }
264}
265
266fn get_header(response: &Response<String>, header: &'static str) -> Result<String, AcmeError> {
267    match response.headers().get_all(header).iter().last() {
268        None => Err(AcmeError::MissingHeader(header)),
269        Some(value) => Ok(value.to_str()?.to_string()),
270    }
271}