Skip to content

Commit 69fef7e

Browse files
committed
tests: add simple test case for ARM64 Image loader
The test_image.bin is a cut of the first 4096 bytes of a pre-compiled linux kernel ARM64 Image. You can generate by: make Image head -c 4096 arch/arm64/boot/Image > test_image.bin Signed-off-by: Qiu Wenbo <qiuwenbo@phytium.com.cn>
1 parent 5d92cb9 commit 69fef7e

File tree

2 files changed

+27
-2
lines changed

2 files changed

+27
-2
lines changed

src/loader/mod.rs

Lines changed: 27 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -635,8 +635,6 @@ impl KernelLoader for Image {
635635
#[cfg(test)]
636636
mod test {
637637
use super::*;
638-
#[cfg(any(feature = "elf", feature = "bzimage"))]
639-
#[cfg(any(target_arch = "x86", target_arch = "x86_64"))]
640638
use std::io::Cursor;
641639
use vm_memory::{Address, GuestAddress, GuestMemoryMmap};
642640

@@ -681,6 +679,15 @@ mod test {
681679
include_bytes!("test_badnote.bin").to_vec()
682680
}
683681

682+
// first 4096 bytes of a precompiled ARM64 kernel Image.
683+
#[cfg(feature = "image")]
684+
#[cfg(target_arch = "aarch64")]
685+
fn make_image_bin() -> Vec<u8> {
686+
let mut v = Vec::new();
687+
v.extend_from_slice(include_bytes!("test_image.bin"));
688+
v
689+
}
690+
684691
#[allow(safe_packed_borrows)]
685692
#[allow(non_snake_case)]
686693
#[test]
@@ -839,6 +846,24 @@ mod test {
839846
);
840847
}
841848

849+
#[test]
850+
#[cfg(feature = "image")]
851+
#[cfg(target_arch = "aarch64")]
852+
fn load_image() {
853+
let gm = create_guest_mem();
854+
let mut image = make_image_bin();
855+
let kernel_addr = GuestAddress(0x200000);
856+
857+
let loader_result =
858+
Image::load(&gm, Some(kernel_addr), &mut Cursor::new(&image), None).unwrap();
859+
assert_eq!(loader_result.kernel_load.raw_value(), 0x280000);
860+
assert_eq!(loader_result.kernel_end, 0x281000);
861+
862+
image[0x39] = 0x0;
863+
let loader_result = Image::load(&gm, Some(kernel_addr), &mut Cursor::new(&image), None);
864+
assert_eq!(loader_result, Err(Error::InvalidImageMagicNumber));
865+
}
866+
842867
#[test]
843868
fn cmdline_overflow() {
844869
let gm = create_guest_mem();

src/loader/test_image.bin

4 KB
Binary file not shown.

0 commit comments

Comments
 (0)