Skip to content

Commit b84a3f7

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 4e4ebe1 commit b84a3f7

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
@@ -519,8 +519,6 @@ impl KernelLoader for Image {
519519
#[cfg(test)]
520520
mod test {
521521
use super::*;
522-
#[cfg(any(feature = "elf", feature = "bzimage"))]
523-
#[cfg(any(target_arch = "x86", target_arch = "x86_64"))]
524522
use std::io::Cursor;
525523
use vm_memory::{Address, GuestAddress, GuestMemoryMmap};
526524

@@ -547,6 +545,15 @@ mod test {
547545
v
548546
}
549547

548+
// first 4096 bytes of a precompiled ARM64 kernel Image.
549+
#[cfg(feature = "image")]
550+
#[cfg(target_arch = "aarch64")]
551+
fn make_image_bin() -> Vec<u8> {
552+
let mut v = Vec::new();
553+
v.extend_from_slice(include_bytes!("test_image.bin"));
554+
v
555+
}
556+
550557
#[allow(safe_packed_borrows)]
551558
#[allow(non_snake_case)]
552559
#[test]
@@ -669,6 +676,24 @@ mod test {
669676
);
670677
}
671678

679+
#[test]
680+
#[cfg(feature = "image")]
681+
#[cfg(target_arch = "aarch64")]
682+
fn load_image() {
683+
let gm = create_guest_mem();
684+
let mut image = make_image_bin();
685+
let kernel_addr = GuestAddress(0x200000);
686+
687+
let loader_result =
688+
Image::load(&gm, Some(kernel_addr), &mut Cursor::new(&image), None).unwrap();
689+
assert_eq!(loader_result.kernel_load.raw_value(), 0x280000);
690+
assert_eq!(loader_result.kernel_end, 0x281000);
691+
692+
image[0x39] = 0x0;
693+
let loader_result = Image::load(&gm, Some(kernel_addr), &mut Cursor::new(&image), None);
694+
assert_eq!(loader_result, Err(Error::InvalidImageMagicNumber));
695+
}
696+
672697
#[test]
673698
fn cmdline_overflow() {
674699
let gm = create_guest_mem();

src/loader/test_image.bin

4 KB
Binary file not shown.

0 commit comments

Comments
 (0)