From 72b8120bf9d339e7051abec7a1c2896db8681dc0 Mon Sep 17 00:00:00 2001 From: Andrew Gunnerson Date: Sun, 17 Dec 2023 15:00:50 -0500 Subject: [PATCH] avb: hashtree: Fix incorrect I/O read size for input smaller than one block Signed-off-by: Andrew Gunnerson --- avbroot/src/format/avb.rs | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/avbroot/src/format/avb.rs b/avbroot/src/format/avb.rs index 390e76e..ae1c025 100644 --- a/avbroot/src/format/avb.rs +++ b/avbroot/src/format/avb.rs @@ -475,7 +475,7 @@ impl HashTreeDescriptor { // Small files are hashed directly, exactly like a hash descriptor. if image_size <= u64::from(block_size) { let mut reader = input.reopen_boxed()?; - let mut buf = vec![0u8; block_size as usize]; + let mut buf = vec![0u8; image_size as usize]; reader.read_exact(&mut buf)?; let mut context = Context::new(algorithm);