@ -61,6 +61,11 @@ use std::{
thread , thread ,
} ; } ;
use tokio ::{
fs ::File ,
io ::{ AsyncBufReadExt , BufReader } ,
} ;
#[ macro_use ] #[ macro_use ]
mod error ; mod error ;
mod api ; mod api ;
@ -89,7 +94,7 @@ async fn main() -> Result<(), Error> {
let extra_debug = matches! ( level , LF ::Trace | LF ::Debug ) ; let extra_debug = matches! ( level , LF ::Trace | LF ::Debug ) ;
check_data_folder ( ) check_data_folder ( ) .await ;
check_rsa_keys ( ) . unwrap_or_else ( | _ | { check_rsa_keys ( ) . unwrap_or_else ( | _ | {
error ! ( "Error creating keys, exiting..." ) ; error ! ( "Error creating keys, exiting..." ) ;
exit ( 1 ) ; exit ( 1 ) ;
@ -286,7 +291,7 @@ fn create_dir(path: &str, description: &str) {
create_dir_all ( path ) . expect ( & err_msg ) ; create_dir_all ( path ) . expect ( & err_msg ) ;
} }
fn check_data_folder ( ) { async fn check_data_folder ( ) {
let data_folder = & CONFIG . data_folder ( ) ; let data_folder = & CONFIG . data_folder ( ) ;
let path = Path ::new ( data_folder ) ; let path = Path ::new ( data_folder ) ;
if ! path . exists ( ) { if ! path . exists ( ) {
@ -299,9 +304,10 @@ fn check_data_folder() {
exit ( 1 ) ; exit ( 1 ) ;
} }
let persistent_volume_check_file = format! ( "{data_folder}/vaultwarden_docker_persistent_volume_check" ) ; if is_running_in_docker ( )
let check_file = Path ::new ( & persistent_volume_check_file ) ; & & std ::env ::var ( "I_REALLY_WANT_VOLATILE_STORAGE" ) . is_err ( )
if check_file . exists ( ) & & std ::env ::var ( "I_REALLY_WANT_VOLATILE_STORAGE" ) . is_err ( ) { & & ! docker_data_folder_is_persistent ( data_folder ) . await
{
error ! ( error ! (
" No persistent volume ! \ n \ " No persistent volume ! \ n \
########################################################################################\ n \ ########################################################################################\ n \
@ -314,6 +320,38 @@ fn check_data_folder() {
} }
} }
/// Detect when using Docker or Podman the DATA_FOLDER is either a bind-mount or a volume created manually.
/// If not created manually, then the data will not be persistent.
/// A none persistent volume in either Docker or Podman is represented by a 64 alphanumerical string.
/// If we detect this string, we will alert about not having a persistent self defined volume.
/// This probably means that someone forgot to add `-v /path/to/vaultwarden_data/:/data`
async fn docker_data_folder_is_persistent ( data_folder : & str ) -> bool {
if let Ok ( mountinfo ) = File ::open ( "/proc/self/mountinfo" ) . await {
// Since there can only be one mountpoint to the DATA_FOLDER
// We do a basic check for this mountpoint surrounded by a space.
let data_folder_match = if data_folder . starts_with ( '/' ) {
format! ( " {data_folder} " )
} else {
format! ( " /{data_folder} " )
} ;
let mut lines = BufReader ::new ( mountinfo ) . lines ( ) ;
while let Some ( line ) = lines . next_line ( ) . await . unwrap_or_default ( ) {
// Only execute a regex check if we find the base match
if line . contains ( & data_folder_match ) {
let re = regex ::Regex ::new ( r"/volumes/[a-z0-9]{64}/_data /" ) . unwrap ( ) ;
if re . is_match ( & line ) {
return false ;
}
// If we did found a match for the mountpoint, but not the regex, then still stop searching.
break ;
}
}
}
// In all other cases, just assume a true.
// This is just an informative check to try and prevent data loss.
true
}
fn check_rsa_keys ( ) -> Result < ( ) , crate ::error ::Error > { fn check_rsa_keys ( ) -> Result < ( ) , crate ::error ::Error > {
// If the RSA keys don't exist, try to create them
// If the RSA keys don't exist, try to create them
let priv_path = CONFIG . private_rsa_key ( ) ; let priv_path = CONFIG . private_rsa_key ( ) ;
BlackDex
2 years ago