use std::path::Path;
use std::sync::Arc;
use git_lfs_git::HttpOptions;
use rustls::client::danger::{HandshakeSignatureValid, ServerCertVerified, ServerCertVerifier};
use rustls::{DigitallySignedStruct, SignatureScheme};
use rustls_pki_types::{CertificateDer, PrivateKeyDer, ServerName, UnixTime};
pub fn build(cwd: &Path, endpoint_url: &str) -> reqwest::Client {
let opts = HttpOptions::for_url(cwd, endpoint_url).unwrap_or_default();
let mut builder = reqwest::ClientBuilder::new();
if opts.ssl_verify == Some(false) {
builder = builder.danger_accept_invalid_certs(true);
} else if let Some(path) = opts.ssl_ca_info.as_deref()
&& let Some(config) =
pinned_cert_config(path, opts.ssl_cert.as_deref(), opts.ssl_key.as_deref())
{
builder = builder.use_preconfigured_tls(config);
}
builder.build().unwrap_or_else(|_| reqwest::Client::new())
}
fn pinned_cert_config(
path: &str,
cert_path: Option<&str>,
key_path: Option<&str>,
) -> Option<rustls::ClientConfig> {
let pem = std::fs::read(path).ok()?;
let mut cursor = std::io::Cursor::new(&pem);
let pinned: Vec<CertificateDer<'static>> = rustls_pemfile::certs(&mut cursor)
.filter_map(Result::ok)
.collect();
if pinned.is_empty() {
return None;
}
let provider = Arc::new(rustls::crypto::ring::default_provider());
let builder = rustls::ClientConfig::builder_with_provider(provider)
.with_safe_default_protocol_versions()
.ok()?
.dangerous()
.with_custom_certificate_verifier(Arc::new(PinnedCertVerifier { pinned }));
let config = match (cert_path, key_path) {
(Some(cp), Some(kp)) => {
let identity = load_client_identity(cp, kp)?;
builder.with_client_auth_cert(identity.0, identity.1).ok()?
}
_ => builder.with_no_client_auth(),
};
Some(config)
}
fn load_client_identity(
cert_path: &str,
key_path: &str,
) -> Option<(Vec<CertificateDer<'static>>, PrivateKeyDer<'static>)> {
let cert_pem = std::fs::read(cert_path).ok()?;
let mut cursor = std::io::Cursor::new(&cert_pem);
let certs: Vec<CertificateDer<'static>> = rustls_pemfile::certs(&mut cursor)
.filter_map(Result::ok)
.collect();
if certs.is_empty() {
return None;
}
let key_pem = std::fs::read(key_path).ok()?;
let mut cursor = std::io::Cursor::new(&key_pem);
let key = rustls_pemfile::private_key(&mut cursor).ok()??;
Some((certs, key))
}
#[derive(Debug)]
struct PinnedCertVerifier {
pinned: Vec<CertificateDer<'static>>,
}
impl ServerCertVerifier for PinnedCertVerifier {
fn verify_server_cert(
&self,
end_entity: &CertificateDer<'_>,
_intermediates: &[CertificateDer<'_>],
_server_name: &ServerName<'_>,
_ocsp_response: &[u8],
_now: UnixTime,
) -> Result<ServerCertVerified, rustls::Error> {
if self
.pinned
.iter()
.any(|p| p.as_ref() == end_entity.as_ref())
{
Ok(ServerCertVerified::assertion())
} else {
Err(rustls::Error::General(
"presented certificate does not match http.sslcainfo".into(),
))
}
}
fn verify_tls12_signature(
&self,
_message: &[u8],
_cert: &CertificateDer<'_>,
_dss: &DigitallySignedStruct,
) -> Result<HandshakeSignatureValid, rustls::Error> {
Ok(HandshakeSignatureValid::assertion())
}
fn verify_tls13_signature(
&self,
_message: &[u8],
_cert: &CertificateDer<'_>,
_dss: &DigitallySignedStruct,
) -> Result<HandshakeSignatureValid, rustls::Error> {
Ok(HandshakeSignatureValid::assertion())
}
fn supported_verify_schemes(&self) -> Vec<SignatureScheme> {
vec![
SignatureScheme::RSA_PKCS1_SHA256,
SignatureScheme::RSA_PKCS1_SHA384,
SignatureScheme::RSA_PKCS1_SHA512,
SignatureScheme::ECDSA_NISTP256_SHA256,
SignatureScheme::ECDSA_NISTP384_SHA384,
SignatureScheme::ECDSA_NISTP521_SHA512,
SignatureScheme::RSA_PSS_SHA256,
SignatureScheme::RSA_PSS_SHA384,
SignatureScheme::RSA_PSS_SHA512,
SignatureScheme::ED25519,
]
}
}