diff --git a/.editorconfig b/.editorconfig index 2d9a31096..9270938df 100644 --- a/.editorconfig +++ b/.editorconfig @@ -28,3 +28,7 @@ indent_size = 4 [*.rs] indent_style = space indent_size = 4 + +[*.sh] +indent_size = 4 +indent_style = space diff --git a/src/commands/logger.rs b/src/commands/logger.rs index 2cd7b3638..a24bcbda7 100644 --- a/src/commands/logger.rs +++ b/src/commands/logger.rs @@ -16,7 +16,7 @@ impl log::Log for SimpleLogger { Level::Debug => "DEBUG".bright_blue(), Level::Trace => "TRACE".into(), }; - println!( + eprintln!( "{} - {}: {}", debug_prefix, record.module_path().unwrap_or_default().bright_black(), diff --git a/src/key.rs b/src/key.rs index 54959d6e6..96eb491d2 100644 --- a/src/key.rs +++ b/src/key.rs @@ -110,25 +110,32 @@ impl KeyHandle { } } - pub fn new_from_search(uuid: &Uuid) -> Result { - let key_name = Self::format_key_name(uuid); - let key_name = CStr::as_ptr(&key_name); + fn search_keyring(keyring: i32, key_name: &CStr) -> Result { + let key_name = CStr::as_ptr(key_name); let key_type = c_str!("user"); - let key_id = - unsafe { keyctl_search(keyutils::KEY_SPEC_USER_KEYRING, key_type, key_name, 0) }; + let key_id = unsafe { keyctl_search(keyring, key_type, key_name, 0) }; if key_id > 0 { info!("Found key in keyring"); - Ok(Self { - _uuid: *uuid, - _id: key_id, - }) + Ok(key_id) } else { Err(ErrnoError(errno::errno()).into()) } } + pub fn new_from_search(uuid: &Uuid) -> Result { + let key_name = Self::format_key_name(uuid); + + Self::search_keyring(keyutils::KEY_SPEC_SESSION_KEYRING, &key_name) + .or_else(|_| Self::search_keyring(keyutils::KEY_SPEC_USER_KEYRING, &key_name)) + .or_else(|_| Self::search_keyring(keyutils::KEY_SPEC_USER_SESSION_KEYRING, &key_name)) + .map(|id| Self { + _uuid: *uuid, + _id: id, + }) + } + fn wait_for_unlock(uuid: &Uuid) -> Result { loop { match Self::new_from_search(uuid) { @@ -158,7 +165,7 @@ impl Passphrase { line }; - Ok(Self(CString::new(passphrase.as_str())?)) + Ok(Self(CString::new(passphrase.trim_end_matches('\n'))?)) } pub fn new_from_file(sb: &bch_sb_handle, passphrase_file: impl AsRef) -> Result { @@ -172,6 +179,6 @@ impl Passphrase { let passphrase = Zeroizing::new(fs::read_to_string(passphrase_file)?); - Ok(Self(CString::new(passphrase.as_str())?)) + Ok(Self(CString::new(passphrase.trim_end_matches('\n'))?)) } }