mirror of
https://github.com/rustfs/rustfs.git
synced 2026-08-03 20:07:42 +00:00
768 lines
27 KiB
Rust
768 lines
27 KiB
Rust
// Copyright 2024 RustFS Team
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
use rustls::RootCertStore;
|
|
use rustls::server::{
|
|
ClientHello, ResolvesServerCert, ResolvesServerCertUsingSni, WebPkiClientVerifier, danger::ClientCertVerifier,
|
|
};
|
|
use rustls::sign::CertifiedKey;
|
|
use rustls_pki_types::{CertificateDer, PrivateKeyDer, pem::PemObject};
|
|
use std::collections::HashMap;
|
|
use std::io::Error;
|
|
use std::path::PathBuf;
|
|
use std::sync::Arc;
|
|
use std::{fs, io};
|
|
use tracing::{debug, warn};
|
|
|
|
/// Options for loading certificate/key pairs from a directory tree.
|
|
#[derive(Debug, Clone)]
|
|
pub struct CertDirectoryLoadOptions {
|
|
dir_path: PathBuf,
|
|
cert_filename: String,
|
|
key_filename: String,
|
|
}
|
|
|
|
impl CertDirectoryLoadOptions {
|
|
/// Create a builder with explicit certificate and private key filenames.
|
|
pub fn builder(
|
|
dir_path: impl Into<PathBuf>,
|
|
cert_filename: impl Into<String>,
|
|
key_filename: impl Into<String>,
|
|
) -> CertDirectoryLoadOptionsBuilder {
|
|
CertDirectoryLoadOptionsBuilder {
|
|
dir_path: dir_path.into(),
|
|
cert_filename: cert_filename.into(),
|
|
key_filename: key_filename.into(),
|
|
}
|
|
}
|
|
|
|
fn validate(&self) -> io::Result<()> {
|
|
if self.cert_filename.is_empty() {
|
|
return Err(certs_error("certificate filename cannot be empty".to_string()));
|
|
}
|
|
if self.key_filename.is_empty() {
|
|
return Err(certs_error("private key filename cannot be empty".to_string()));
|
|
}
|
|
Ok(())
|
|
}
|
|
}
|
|
|
|
/// Builder for [`CertDirectoryLoadOptions`].
|
|
#[derive(Debug, Clone)]
|
|
pub struct CertDirectoryLoadOptionsBuilder {
|
|
dir_path: PathBuf,
|
|
cert_filename: String,
|
|
key_filename: String,
|
|
}
|
|
|
|
impl CertDirectoryLoadOptionsBuilder {
|
|
/// Override the certificate filename searched in the directory.
|
|
pub fn cert_filename(mut self, cert_filename: impl Into<String>) -> Self {
|
|
self.cert_filename = cert_filename.into();
|
|
self
|
|
}
|
|
|
|
/// Override the private key filename searched in the directory.
|
|
pub fn key_filename(mut self, key_filename: impl Into<String>) -> Self {
|
|
self.key_filename = key_filename.into();
|
|
self
|
|
}
|
|
|
|
/// Build the load options value.
|
|
pub fn build(self) -> CertDirectoryLoadOptions {
|
|
CertDirectoryLoadOptions {
|
|
dir_path: self.dir_path,
|
|
cert_filename: self.cert_filename,
|
|
key_filename: self.key_filename,
|
|
}
|
|
}
|
|
}
|
|
|
|
/// Options for building an mTLS WebPki client verifier.
|
|
#[derive(Debug, Clone)]
|
|
pub struct WebPkiClientVerifierOptions {
|
|
tls_path: PathBuf,
|
|
enabled: bool,
|
|
client_ca_cert_filename: String,
|
|
fallback_ca_cert_filename: String,
|
|
}
|
|
|
|
impl WebPkiClientVerifierOptions {
|
|
/// Create a builder with explicit CA bundle filenames.
|
|
pub fn builder(
|
|
tls_path: impl Into<PathBuf>,
|
|
client_ca_cert_filename: impl Into<String>,
|
|
fallback_ca_cert_filename: impl Into<String>,
|
|
) -> WebPkiClientVerifierOptionsBuilder {
|
|
WebPkiClientVerifierOptionsBuilder {
|
|
tls_path: tls_path.into(),
|
|
enabled: false,
|
|
client_ca_cert_filename: client_ca_cert_filename.into(),
|
|
fallback_ca_cert_filename: fallback_ca_cert_filename.into(),
|
|
}
|
|
}
|
|
}
|
|
|
|
/// Builder for [`WebPkiClientVerifierOptions`].
|
|
#[derive(Debug, Clone)]
|
|
pub struct WebPkiClientVerifierOptionsBuilder {
|
|
tls_path: PathBuf,
|
|
enabled: bool,
|
|
client_ca_cert_filename: String,
|
|
fallback_ca_cert_filename: String,
|
|
}
|
|
|
|
impl WebPkiClientVerifierOptionsBuilder {
|
|
/// Set whether mTLS verification should be enabled.
|
|
pub fn enabled(mut self, enabled: bool) -> Self {
|
|
self.enabled = enabled;
|
|
self
|
|
}
|
|
|
|
/// Override the preferred client CA bundle filename.
|
|
pub fn client_ca_cert_filename(mut self, client_ca_cert_filename: impl Into<String>) -> Self {
|
|
self.client_ca_cert_filename = client_ca_cert_filename.into();
|
|
self
|
|
}
|
|
|
|
/// Override the fallback CA bundle filename.
|
|
pub fn fallback_ca_cert_filename(mut self, fallback_ca_cert_filename: impl Into<String>) -> Self {
|
|
self.fallback_ca_cert_filename = fallback_ca_cert_filename.into();
|
|
self
|
|
}
|
|
|
|
/// Build the verifier options value.
|
|
pub fn build(self) -> WebPkiClientVerifierOptions {
|
|
WebPkiClientVerifierOptions {
|
|
tls_path: self.tls_path,
|
|
enabled: self.enabled,
|
|
client_ca_cert_filename: self.client_ca_cert_filename,
|
|
fallback_ca_cert_filename: self.fallback_ca_cert_filename,
|
|
}
|
|
}
|
|
}
|
|
|
|
/// Load public certificate from file.
|
|
/// This function loads a public certificate from the specified file.
|
|
///
|
|
/// # Arguments
|
|
/// * `filename` - A string slice that holds the name of the file containing the public certificate.
|
|
///
|
|
/// # Returns
|
|
/// * An io::Result containing a vector of CertificateDer if successful, or an io::Error if an error occurs during loading.
|
|
///
|
|
pub fn load_certs(filename: &str) -> io::Result<Vec<CertificateDer<'static>>> {
|
|
// Open certificate file.
|
|
let cert_file = fs::File::open(filename).map_err(|e| certs_error(format!("failed to open {filename}: {e}")))?;
|
|
let mut reader = io::BufReader::new(cert_file);
|
|
|
|
// Load and return certificate.
|
|
let certs = CertificateDer::pem_reader_iter(&mut reader)
|
|
.collect::<Result<Vec<_>, _>>()
|
|
.map_err(|e| certs_error(format!("certificate file {filename} format error:{e:?}")))?;
|
|
if certs.is_empty() {
|
|
return Err(certs_error(format!("No valid certificate was found in the certificate file {filename}")));
|
|
}
|
|
Ok(certs)
|
|
}
|
|
|
|
/// Load a PEM certificate bundle and return each certificate as DER bytes.
|
|
///
|
|
/// This is a low-level helper intended for TLS clients (reqwest/hyper-rustls) that
|
|
/// need to add root certificates one-by-one.
|
|
///
|
|
/// - Input: a PEM file that may contain multiple cert blocks.
|
|
/// - Output: Vec of DER-encoded cert bytes, one per cert.
|
|
///
|
|
/// NOTE: This intentionally returns raw bytes to avoid forcing downstream crates
|
|
/// to depend on rustls types.
|
|
pub fn load_cert_bundle_der_bytes(path: &str) -> io::Result<Vec<Vec<u8>>> {
|
|
let pem = fs::read(path)?;
|
|
let mut reader = io::BufReader::new(&pem[..]);
|
|
|
|
let certs = CertificateDer::pem_reader_iter(&mut reader)
|
|
.collect::<Result<Vec<_>, _>>()
|
|
.map_err(|e| certs_error(format!("Failed to parse PEM certs from {path}: {e}")))?;
|
|
|
|
Ok(certs.into_iter().map(|c| c.to_vec()).collect())
|
|
}
|
|
|
|
/// Builds a WebPkiClientVerifier for mTLS when enabled by the caller.
|
|
///
|
|
/// # Arguments
|
|
/// * `options` - mTLS verifier options, including the TLS directory and CA bundle filenames
|
|
///
|
|
/// # Returns
|
|
/// * `Ok(Some(verifier))` if mTLS is enabled and CA certs are found
|
|
/// * `Ok(None)` if mTLS is disabled
|
|
/// * `Err` if mTLS is enabled but configuration is invalid
|
|
pub fn build_webpki_client_verifier(options: WebPkiClientVerifierOptions) -> io::Result<Option<Arc<dyn ClientCertVerifier>>> {
|
|
if !options.enabled {
|
|
return Ok(None);
|
|
}
|
|
|
|
let tls_path = &options.tls_path;
|
|
let ca_path = mtls_ca_bundle_path(&options).ok_or_else(|| {
|
|
Error::other(format!(
|
|
"mTLS is enabled but missing {}/{} (or fallback {}/{})",
|
|
tls_path.display(),
|
|
options.client_ca_cert_filename,
|
|
tls_path.display(),
|
|
options.fallback_ca_cert_filename
|
|
))
|
|
})?;
|
|
|
|
let ca_path = ca_path
|
|
.to_str()
|
|
.ok_or_else(|| Error::other(format!("Invalid UTF-8 in mTLS CA path: {ca_path:?}")))?;
|
|
|
|
let der_list = load_cert_bundle_der_bytes(ca_path)?;
|
|
|
|
let mut store = RootCertStore::empty();
|
|
for der in der_list {
|
|
store
|
|
.add(der.into())
|
|
.map_err(|e| Error::other(format!("Invalid client CA cert: {e}")))?;
|
|
}
|
|
|
|
let verifier = WebPkiClientVerifier::builder(Arc::new(store))
|
|
.build()
|
|
.map_err(|e| Error::other(format!("Build client cert verifier failed: {e}")))?;
|
|
|
|
Ok(Some(verifier))
|
|
}
|
|
|
|
/// Locate the mTLS client CA bundle in the specified TLS path
|
|
fn mtls_ca_bundle_path(options: &WebPkiClientVerifierOptions) -> Option<PathBuf> {
|
|
let p1 = options.tls_path.join(&options.client_ca_cert_filename);
|
|
if p1.exists() {
|
|
return Some(p1);
|
|
}
|
|
let p2 = options.tls_path.join(&options.fallback_ca_cert_filename);
|
|
if p2.exists() {
|
|
return Some(p2);
|
|
}
|
|
None
|
|
}
|
|
|
|
/// Load private key from file.
|
|
/// This function loads a private key from the specified file.
|
|
///
|
|
/// # Arguments
|
|
/// * `filename` - A string slice that holds the name of the file containing the private key.
|
|
///
|
|
/// # Returns
|
|
/// * An io::Result containing the PrivateKeyDer if successful, or an io::Error if an error occurs during loading.
|
|
///
|
|
pub fn load_private_key(filename: &str) -> io::Result<PrivateKeyDer<'static>> {
|
|
// Open keyfile.
|
|
let keyfile = fs::File::open(filename).map_err(|e| certs_error(format!("failed to open {filename}: {e}")))?;
|
|
let mut reader = io::BufReader::new(keyfile);
|
|
|
|
// Load and return a single private key.
|
|
PrivateKeyDer::from_pem_reader(&mut reader)
|
|
.map_err(|e| certs_error(format!("failed to parse private key in {filename}: {e}")))
|
|
}
|
|
|
|
/// error function
|
|
/// This function creates a new io::Error with the provided error message.
|
|
///
|
|
/// # Arguments
|
|
/// * `err` - A string containing the error message.
|
|
///
|
|
/// # Returns
|
|
/// * An io::Error instance with the specified error message.
|
|
///
|
|
pub fn certs_error(err: String) -> Error {
|
|
Error::other(err)
|
|
}
|
|
|
|
fn is_discoverable_cert_domain_dir(domain_name: &str) -> bool {
|
|
!domain_name.starts_with('.')
|
|
}
|
|
|
|
/// Load all certificates and private keys in the directory
|
|
/// This function loads all certificate and private key pairs from the specified directory.
|
|
/// It looks for files named `options.cert_filename` and `options.key_filename` in each subdirectory.
|
|
/// The root directory can also contain a default certificate/private key pair.
|
|
///
|
|
/// # Arguments
|
|
/// * `options` - Directory and filename options for discovering certificates and private keys.
|
|
///
|
|
/// # Returns
|
|
/// * An io::Result containing a HashMap where the keys are domain names (or "default" for the root certificate) and the values are tuples of (Vec<CertificateDer>, PrivateKeyDer). If no valid certificate/private key pairs are found, an io::Error is returned.
|
|
///
|
|
pub fn load_all_certs_from_directory(
|
|
options: CertDirectoryLoadOptions,
|
|
) -> io::Result<HashMap<String, (Vec<CertificateDer<'static>>, PrivateKeyDer<'static>)>> {
|
|
options.validate()?;
|
|
|
|
let mut cert_key_pairs = HashMap::new();
|
|
let dir = options.dir_path.as_path();
|
|
|
|
if !dir.exists() || !dir.is_dir() {
|
|
return Err(certs_error(format!(
|
|
"The certificate directory does not exist or is not a directory: {}",
|
|
dir.display()
|
|
)));
|
|
}
|
|
|
|
// 1. First check whether there is a certificate/private key pair in the root directory
|
|
let root_cert_path = dir.join(&options.cert_filename);
|
|
let root_key_path = dir.join(&options.key_filename);
|
|
|
|
if root_cert_path.exists() && root_key_path.exists() {
|
|
debug!("find the root directory certificate: {:?}", root_cert_path);
|
|
let root_cert_str = root_cert_path
|
|
.to_str()
|
|
.ok_or_else(|| certs_error(format!("Invalid UTF-8 in root certificate path: {root_cert_path:?}")))?;
|
|
let root_key_str = root_key_path
|
|
.to_str()
|
|
.ok_or_else(|| certs_error(format!("Invalid UTF-8 in root key path: {root_key_path:?}")))?;
|
|
match load_cert_key_pair(root_cert_str, root_key_str) {
|
|
Ok((certs, key)) => {
|
|
// The root directory certificate is used as the default certificate and is stored using special keys.
|
|
cert_key_pairs.insert("default".to_string(), (certs, key));
|
|
}
|
|
Err(e) => {
|
|
warn!("unable to load root directory certificate: {}", e);
|
|
}
|
|
}
|
|
}
|
|
|
|
// 2.iterate through all folders in the directory
|
|
for entry in fs::read_dir(dir)? {
|
|
let entry = entry?;
|
|
let path = entry.path();
|
|
|
|
if path.is_dir() {
|
|
let domain_name: &str = path
|
|
.file_name()
|
|
.and_then(|name| name.to_str())
|
|
.ok_or_else(|| certs_error(format!("invalid domain name directory:{path:?}")))?;
|
|
if !is_discoverable_cert_domain_dir(domain_name) {
|
|
debug!("skip internal certificate directory: {:?}", path);
|
|
continue;
|
|
}
|
|
|
|
// find certificate and private key files
|
|
let cert_path = path.join(&options.cert_filename); // e.g., rustfs_cert.pem
|
|
let key_path = path.join(&options.key_filename); // e.g., rustfs_key.pem
|
|
|
|
if cert_path.exists() && key_path.exists() {
|
|
debug!("find the domain name certificate: {} in {:?}", domain_name, cert_path);
|
|
let cert_path = match cert_path.to_str() {
|
|
Some(path) => path,
|
|
None => {
|
|
warn!("skip domain certificate load, invalid UTF-8 path: {:?}", cert_path);
|
|
continue;
|
|
}
|
|
};
|
|
|
|
let key_path = match key_path.to_str() {
|
|
Some(path) => path,
|
|
None => {
|
|
warn!("skip domain key load, invalid UTF-8 path: {:?}", key_path);
|
|
continue;
|
|
}
|
|
};
|
|
|
|
match load_cert_key_pair(cert_path, key_path) {
|
|
Ok((certs, key)) => {
|
|
cert_key_pairs.insert(domain_name.to_string(), (certs, key));
|
|
}
|
|
Err(e) => {
|
|
warn!("unable to load the certificate for {} domain name: {}", domain_name, e);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
if cert_key_pairs.is_empty() {
|
|
return Err(certs_error(format!(
|
|
"No valid certificate/private key pair found in directory {}",
|
|
dir.display()
|
|
)));
|
|
}
|
|
|
|
Ok(cert_key_pairs)
|
|
}
|
|
|
|
/// loading a single certificate private key pair
|
|
/// This function loads a certificate and private key from the specified paths.
|
|
/// It returns a tuple containing the certificate and private key.
|
|
///
|
|
/// # Arguments
|
|
/// * `cert_path` - A string slice that holds the path to the certificate file.
|
|
/// * `key_path` - A string slice that holds the path to the private key file
|
|
///
|
|
/// # Returns
|
|
/// * An io::Result containing a tuple of (Vec<CertificateDer>, PrivateKeyDer) if successful, or an io::Error if an error occurs during loading.
|
|
///
|
|
fn load_cert_key_pair(cert_path: &str, key_path: &str) -> io::Result<(Vec<CertificateDer<'static>>, PrivateKeyDer<'static>)> {
|
|
let certs = load_certs(cert_path)?;
|
|
let key = load_private_key(key_path)?;
|
|
Ok((certs, key))
|
|
}
|
|
|
|
/// Create a multi-cert resolver
|
|
/// This function loads all certificates and private keys from the specified directory.
|
|
/// It uses the first certificate/private key pair found in the root directory as the default certificate.
|
|
/// The rest of the certificates/private keys are used for SNI resolution.
|
|
///
|
|
/// # Arguments
|
|
/// * `cert_key_pairs` - A HashMap where the keys are domain names (or "default" for the root certificate) and the values are tuples of (Vec<CertificateDer>, PrivateKeyDer).
|
|
///
|
|
/// # Returns
|
|
/// * An io::Result containing an implementation of ResolvesServerCert if successful, or an io::Error if an error occurs during loading.
|
|
///
|
|
pub fn create_multi_cert_resolver(
|
|
cert_key_pairs: HashMap<String, (Vec<CertificateDer<'static>>, PrivateKeyDer<'static>)>,
|
|
) -> io::Result<impl ResolvesServerCert> {
|
|
#[derive(Debug)]
|
|
struct MultiCertResolver {
|
|
cert_resolver: ResolvesServerCertUsingSni,
|
|
default_cert: Option<Arc<CertifiedKey>>,
|
|
}
|
|
impl ResolvesServerCert for MultiCertResolver {
|
|
fn resolve(&self, client_hello: ClientHello) -> Option<Arc<CertifiedKey>> {
|
|
// try matching certificates with sni
|
|
if let Some(cert) = self.cert_resolver.resolve(client_hello) {
|
|
return Some(cert);
|
|
}
|
|
|
|
// If there is no matching SNI certificate, use the default certificate
|
|
self.default_cert.clone()
|
|
}
|
|
}
|
|
|
|
let mut resolver = ResolvesServerCertUsingSni::new();
|
|
let mut default_cert = None;
|
|
|
|
for (domain, (certs, key)) in cert_key_pairs {
|
|
// create a signature
|
|
let signing_key = rustls::crypto::aws_lc_rs::sign::any_supported_type(&key)
|
|
.map_err(|e| certs_error(format!("unsupported private key types:{domain}, err:{e:?}")))?;
|
|
|
|
// create a CertifiedKey
|
|
let certified_key = CertifiedKey::new(certs, signing_key);
|
|
if domain == "default" {
|
|
default_cert = Some(Arc::new(certified_key.clone()));
|
|
} else {
|
|
// add certificate to resolver
|
|
resolver
|
|
.add(&domain, certified_key)
|
|
.map_err(|e| certs_error(format!("failed to add a domain name certificate:{domain},err: {e:?}")))?;
|
|
}
|
|
}
|
|
|
|
Ok(MultiCertResolver {
|
|
cert_resolver: resolver,
|
|
default_cert,
|
|
})
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::*;
|
|
use std::fs;
|
|
use std::io::ErrorKind;
|
|
use tempfile::TempDir;
|
|
|
|
fn default_load_options(path: impl Into<PathBuf>) -> CertDirectoryLoadOptions {
|
|
CertDirectoryLoadOptions::builder(path, "rustfs_cert.pem", "rustfs_key.pem").build()
|
|
}
|
|
|
|
fn write_test_cert_pair(dir: &std::path::Path) {
|
|
let rcgen::CertifiedKey { cert, signing_key } =
|
|
rcgen::generate_simple_self_signed(vec!["example.com".to_string()]).unwrap();
|
|
fs::write(dir.join("rustfs_cert.pem"), cert.pem()).unwrap();
|
|
fs::write(dir.join("rustfs_key.pem"), signing_key.serialize_pem()).unwrap();
|
|
}
|
|
|
|
#[test]
|
|
fn test_certs_error_function() {
|
|
let error_msg = "Test error message";
|
|
let error = certs_error(error_msg.to_string());
|
|
|
|
assert_eq!(error.kind(), ErrorKind::Other);
|
|
assert_eq!(error.to_string(), error_msg);
|
|
}
|
|
|
|
#[test]
|
|
fn test_load_certs_file_not_found() {
|
|
let result = load_certs("non_existent_file.pem");
|
|
assert!(result.is_err());
|
|
|
|
let error = result.unwrap_err();
|
|
assert_eq!(error.kind(), ErrorKind::Other);
|
|
assert!(error.to_string().contains("failed to open"));
|
|
}
|
|
|
|
#[test]
|
|
fn test_load_private_key_file_not_found() {
|
|
let result = load_private_key("non_existent_key.pem");
|
|
assert!(result.is_err());
|
|
|
|
let error = result.unwrap_err();
|
|
assert_eq!(error.kind(), ErrorKind::Other);
|
|
assert!(error.to_string().contains("failed to open"));
|
|
}
|
|
|
|
#[test]
|
|
fn test_load_certs_empty_file() {
|
|
let temp_dir = TempDir::new().unwrap();
|
|
let cert_path = temp_dir.path().join("empty.pem");
|
|
fs::write(&cert_path, "").unwrap();
|
|
|
|
let result = load_certs(cert_path.to_str().unwrap());
|
|
assert!(result.is_err());
|
|
|
|
let error = result.unwrap_err();
|
|
assert!(error.to_string().contains("No valid certificate was found"));
|
|
}
|
|
|
|
#[test]
|
|
fn test_load_certs_invalid_format() {
|
|
let temp_dir = TempDir::new().unwrap();
|
|
let cert_path = temp_dir.path().join("invalid.pem");
|
|
fs::write(&cert_path, "invalid certificate content").unwrap();
|
|
|
|
let result = load_certs(cert_path.to_str().unwrap());
|
|
assert!(result.is_err());
|
|
|
|
let error = result.unwrap_err();
|
|
assert!(error.to_string().contains("No valid certificate was found"));
|
|
}
|
|
|
|
#[test]
|
|
fn test_load_private_key_empty_file() {
|
|
let temp_dir = TempDir::new().unwrap();
|
|
let key_path = temp_dir.path().join("empty_key.pem");
|
|
fs::write(&key_path, "").unwrap();
|
|
|
|
let result = load_private_key(key_path.to_str().unwrap());
|
|
assert!(result.is_err());
|
|
|
|
let error = result.unwrap_err();
|
|
assert!(error.to_string().contains("failed to parse private key in"));
|
|
}
|
|
|
|
#[test]
|
|
fn test_load_private_key_invalid_format() {
|
|
let temp_dir = TempDir::new().unwrap();
|
|
let key_path = temp_dir.path().join("invalid_key.pem");
|
|
fs::write(&key_path, "invalid private key content").unwrap();
|
|
|
|
let result = load_private_key(key_path.to_str().unwrap());
|
|
assert!(result.is_err());
|
|
|
|
let error = result.unwrap_err();
|
|
assert!(error.to_string().contains("failed to parse private key in"));
|
|
}
|
|
|
|
#[test]
|
|
fn test_load_all_certs_from_directory_not_exists() {
|
|
let result = load_all_certs_from_directory(default_load_options("/non/existent/directory"));
|
|
assert!(result.is_err());
|
|
|
|
let error = result.unwrap_err();
|
|
assert!(error.to_string().contains("does not exist or is not a directory"));
|
|
}
|
|
|
|
#[test]
|
|
fn test_load_all_certs_from_directory_empty() {
|
|
let temp_dir = TempDir::new().unwrap();
|
|
|
|
let result = load_all_certs_from_directory(default_load_options(temp_dir.path()));
|
|
assert!(result.is_err());
|
|
|
|
let error = result.unwrap_err();
|
|
assert!(error.to_string().contains("No valid certificate/private key pair found"));
|
|
}
|
|
|
|
#[test]
|
|
fn test_load_all_certs_from_directory_file_instead_of_dir() {
|
|
let temp_dir = TempDir::new().unwrap();
|
|
let file_path = temp_dir.path().join("not_a_directory.txt");
|
|
fs::write(&file_path, "content").unwrap();
|
|
|
|
let result = load_all_certs_from_directory(default_load_options(&file_path));
|
|
assert!(result.is_err());
|
|
|
|
let error = result.unwrap_err();
|
|
assert!(error.to_string().contains("does not exist or is not a directory"));
|
|
}
|
|
|
|
#[test]
|
|
fn test_load_cert_key_pair_missing_cert() {
|
|
let temp_dir = TempDir::new().unwrap();
|
|
let key_path = temp_dir.path().join("test_key.pem");
|
|
fs::write(&key_path, "dummy key content").unwrap();
|
|
|
|
let result = load_cert_key_pair("non_existent_cert.pem", key_path.to_str().unwrap());
|
|
assert!(result.is_err());
|
|
}
|
|
|
|
#[test]
|
|
fn test_load_cert_key_pair_missing_key() {
|
|
let temp_dir = TempDir::new().unwrap();
|
|
let cert_path = temp_dir.path().join("test_cert.pem");
|
|
fs::write(&cert_path, "dummy cert content").unwrap();
|
|
|
|
let result = load_cert_key_pair(cert_path.to_str().unwrap(), "non_existent_key.pem");
|
|
assert!(result.is_err());
|
|
}
|
|
|
|
#[test]
|
|
fn test_create_multi_cert_resolver_empty_map() {
|
|
let empty_map = HashMap::new();
|
|
let result = create_multi_cert_resolver(empty_map);
|
|
|
|
// Should succeed even with empty map
|
|
assert!(result.is_ok());
|
|
}
|
|
|
|
#[test]
|
|
fn test_error_message_formatting() {
|
|
let test_cases = vec![
|
|
("file not found", "failed to open test.pem: file not found"),
|
|
("permission denied", "failed to open key.pem: permission denied"),
|
|
("invalid format", "certificate file cert.pem format error:invalid format"),
|
|
];
|
|
|
|
for (input, _expected_pattern) in test_cases {
|
|
let error1 = certs_error(format!("failed to open test.pem: {input}"));
|
|
assert!(error1.to_string().contains(input));
|
|
|
|
let error2 = certs_error(format!("failed to open key.pem: {input}"));
|
|
assert!(error2.to_string().contains(input));
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_path_handling_edge_cases() {
|
|
// Test with various path formats
|
|
let path_cases = vec![
|
|
"", // Empty path
|
|
".", // Current directory
|
|
"..", // Parent directory
|
|
"/", // Root directory (Unix)
|
|
"relative/path", // Relative path
|
|
"/absolute/path", // Absolute path
|
|
];
|
|
|
|
for path in path_cases {
|
|
let result = load_all_certs_from_directory(default_load_options(path));
|
|
// All should fail since these are not valid cert directories
|
|
assert!(result.is_err());
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_directory_structure_validation() {
|
|
let temp_dir = TempDir::new().unwrap();
|
|
|
|
// Create a subdirectory without certificates
|
|
let sub_dir = temp_dir.path().join("example.com");
|
|
fs::create_dir(&sub_dir).unwrap();
|
|
|
|
// Should fail because no certificates found
|
|
let result = load_all_certs_from_directory(default_load_options(temp_dir.path()));
|
|
assert!(result.is_err());
|
|
assert!(
|
|
result
|
|
.unwrap_err()
|
|
.to_string()
|
|
.contains("No valid certificate/private key pair found")
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn test_load_all_certs_skips_kubernetes_secret_projection_dirs() {
|
|
let temp_dir = TempDir::new().unwrap();
|
|
write_test_cert_pair(temp_dir.path());
|
|
|
|
let domain_dir = temp_dir.path().join("example.com");
|
|
fs::create_dir(&domain_dir).unwrap();
|
|
write_test_cert_pair(&domain_dir);
|
|
|
|
for internal_dir_name in ["..data", "..2026_04_28_18_33_53.4209048473"] {
|
|
let internal_dir = temp_dir.path().join(internal_dir_name);
|
|
fs::create_dir(&internal_dir).unwrap();
|
|
write_test_cert_pair(&internal_dir);
|
|
}
|
|
|
|
let certs = load_all_certs_from_directory(default_load_options(temp_dir.path())).unwrap();
|
|
|
|
assert!(certs.contains_key("default"));
|
|
assert!(certs.contains_key("example.com"));
|
|
assert!(!certs.contains_key("..data"));
|
|
assert!(!certs.contains_key("..2026_04_28_18_33_53.4209048473"));
|
|
assert_eq!(certs.len(), 2);
|
|
}
|
|
|
|
#[test]
|
|
fn test_unicode_path_handling() {
|
|
let temp_dir = TempDir::new().unwrap();
|
|
|
|
// Create directory with Unicode characters
|
|
let unicode_dir = temp_dir.path().join("test_directory");
|
|
fs::create_dir(&unicode_dir).unwrap();
|
|
|
|
let result = load_all_certs_from_directory(default_load_options(&unicode_dir));
|
|
assert!(result.is_err());
|
|
assert!(
|
|
result
|
|
.unwrap_err()
|
|
.to_string()
|
|
.contains("No valid certificate/private key pair found")
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn test_concurrent_access_safety() {
|
|
use std::sync::Arc;
|
|
use std::thread;
|
|
|
|
let temp_dir = TempDir::new().unwrap();
|
|
let dir_path = Arc::new(temp_dir.path().to_string_lossy().to_string());
|
|
|
|
let handles: Vec<_> = (0..5)
|
|
.map(|_| {
|
|
let path = Arc::clone(&dir_path);
|
|
thread::spawn(move || {
|
|
let result = load_all_certs_from_directory(default_load_options(path.as_str()));
|
|
// All should fail since directory is empty
|
|
assert!(result.is_err());
|
|
})
|
|
})
|
|
.collect();
|
|
|
|
for handle in handles {
|
|
handle.join().expect("Thread should complete successfully");
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_memory_efficiency() {
|
|
let error = certs_error("test".to_string());
|
|
let error_size = std::mem::size_of_val(&error);
|
|
|
|
// Error should not be excessively large
|
|
assert!(error_size < 1024, "Error size should be reasonable, got {error_size} bytes");
|
|
}
|
|
}
|