Allow specifying key length in add_by_key

stratis-storage · Sep 18, 2024 · ea45bd1 · ea45bd1
1 parent ed4db81
commit ea45bd1
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/keyslot.rs b/src/keyslot.rs
@@ -7,6 +7,7 @@ use std::{
     ptr,
 };
 
+use either::Either;
 use libc::{c_int, c_uint};
 
 use crate::{
@@ -111,12 +112,13 @@ impl<'a> CryptKeyslotHandle<'a> {
     pub fn add_by_key(
         &mut self,
         keyslot: Option<c_uint>,
-        volume_key: Option<&[u8]>,
+        volume_key: Option<Either<&[u8], usize>>,
         passphrase: &[u8],
         flags: CryptVolumeKey,
     ) -> Result<c_uint, LibcryptErr> {
         let (vk_ptr, vk_len) = match volume_key {
-            Some(vk) => (to_byte_ptr!(vk), vk.len()),
+            Some(Either::Left(vk)) => (to_byte_ptr!(vk), vk.len()),
+            Some(Either::Right(s)) => (std::ptr::null(), s),
             None => (std::ptr::null(), 0),
         };
         errno_int_success!(mutex!(libcryptsetup_rs_sys::crypt_keyslot_add_by_key(