Skip to content

Commit

Permalink
chore(deps): Update rustls and axum-server (#2382)
Browse files Browse the repository at this point in the history
  • Loading branch information
tobiasge authored Sep 9, 2024
1 parent 5ed36b7 commit 51650ff
Show file tree
Hide file tree
Showing 4 changed files with 28 additions and 65 deletions.
15 changes: 8 additions & 7 deletions Cargo.lock

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

8 changes: 4 additions & 4 deletions crates/atuin-server/Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -25,13 +25,13 @@ base64 = { workspace = true }
rand = { workspace = true }
tokio = { workspace = true }
async-trait = { workspace = true }
axum = "0.7.4"
axum-server = { version = "0.6.0", features = ["tls-rustls"] }
axum = "0.7"
axum-server = { version = "0.7", features = ["tls-rustls-no-provider"] }
fs-err = { workspace = true }
tower = { workspace = true }
tower-http = { version = "0.5.1", features = ["trace"] }
tower-http = { version = "0.5", features = ["trace"] }
reqwest = { workspace = true }
rustls = "0.21"
rustls = { version = "0.23", features = ["ring"], default-features = false }
rustls-pemfile = "2.1"
argon2 = "0.5"
semver = { workspace = true }
Expand Down
28 changes: 15 additions & 13 deletions crates/atuin-server/src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -2,19 +2,18 @@

use std::future::Future;
use std::net::SocketAddr;
use std::sync::Arc;

use atuin_server_database::Database;
use axum::{serve, Router};
use axum_server::tls_rustls::RustlsConfig;
use axum_server::Handle;
use eyre::{Context, Result};
use eyre::{eyre, Context, Result};

mod handlers;
mod metrics;
mod router;
mod utils;

use rustls::ServerConfig;
pub use settings::example_config;
pub use settings::Settings;

Expand Down Expand Up @@ -83,16 +82,19 @@ async fn launch_with_tls<Db: Database>(
addr: SocketAddr,
shutdown: impl Future<Output = ()>,
) -> Result<()> {
let certificates = settings.tls.certificates()?;
let pkey = settings.tls.private_key()?;

let server_config = ServerConfig::builder()
.with_safe_defaults()
.with_no_client_auth()
.with_single_cert(certificates, pkey)?;

let server_config = Arc::new(server_config);
let rustls_config = axum_server::tls_rustls::RustlsConfig::from_config(server_config);
let crypto_provider = rustls::crypto::ring::default_provider().install_default();
if crypto_provider.is_err() {
return Err(eyre!("Failed to install default crypto provider"));
}
let rustls_config = RustlsConfig::from_pem_file(
settings.tls.cert_path.clone(),
settings.tls.pkey_path.clone(),
)
.await;
if rustls_config.is_err() {
return Err(eyre!("Failed to load TLS key and/or certificate"));
}
let rustls_config = rustls_config.unwrap();

let r = make_router::<Db>(settings).await?;

Expand Down
42 changes: 1 addition & 41 deletions crates/atuin-server/src/settings.rs
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
use std::{io::prelude::*, path::PathBuf};

use config::{Config, Environment, File as ConfigFile, FileFormat};
use eyre::{bail, eyre, Context, Result};
use eyre::{eyre, Result};
use fs_err::{create_dir_all, File};
use serde::{de::DeserializeOwned, Deserialize, Serialize};

Expand Down Expand Up @@ -146,43 +146,3 @@ pub struct Tls {
pub cert_path: PathBuf,
pub pkey_path: PathBuf,
}

impl Tls {
pub fn certificates(&self) -> Result<Vec<rustls::Certificate>> {
let cert_file = std::fs::File::open(&self.cert_path)
.with_context(|| format!("tls.cert_path {:?} is missing", self.cert_path))?;
let mut reader = std::io::BufReader::new(cert_file);
let certs: Vec<_> = rustls_pemfile::certs(&mut reader)
.map(|c| c.map(|c| rustls::Certificate(c.to_vec())))
.collect::<Result<Vec<_>, _>>()
.with_context(|| format!("tls.cert_path {:?} is invalid", self.cert_path))?;

if certs.is_empty() {
bail!(
"tls.cert_path {:?} must have at least one certificate",
self.cert_path
);
}

Ok(certs)
}

pub fn private_key(&self) -> Result<rustls::PrivateKey> {
let pkey_file = std::fs::File::open(&self.pkey_path)
.with_context(|| format!("tls.pkey_path {:?} is missing", self.pkey_path))?;
let mut reader = std::io::BufReader::new(pkey_file);
let keys = rustls_pemfile::pkcs8_private_keys(&mut reader)
.map(|c| c.map(|c| rustls::PrivateKey(c.secret_pkcs8_der().to_vec())))
.collect::<Result<Vec<_>, _>>()
.with_context(|| format!("tls.pkey_path {:?} is not PKCS8-encoded", self.pkey_path))?;

if keys.is_empty() {
bail!(
"tls.pkey_path {:?} must have at least one private key",
self.pkey_path
);
}

Ok(keys[0].clone())
}
}

0 comments on commit 51650ff

Please sign in to comment.