1 // Copyright 2017 The Chromium OS Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #![cfg(any(target_arch = "x86", target_arch = "x86_64"))]
6
7 use kvm::*;
8 use kvm_sys::kvm_regs;
9 use sys_util::{GuestAddress, GuestMemory, MemoryMapping, SharedMemory};
10
11 #[test]
test_run()12 fn test_run() {
13 /*
14 0000 881C mov [si],bl
15 0002 F4 hlt
16 */
17 let code = [0x88, 0x1c, 0xf4];
18 let mem_size = 0x10000;
19 let load_addr = GuestAddress(0x1000);
20 let guest_mem = GuestMemory::new(&[]).unwrap();
21 let mut mem = SharedMemory::new(None).expect("failed to create shared memory");
22 mem.set_size(mem_size)
23 .expect("failed to set shared memory size");
24 let mmap =
25 MemoryMapping::from_fd(&mem, mem_size as usize).expect("failed to create memory mapping");
26
27 mmap.write_slice(&code[..], load_addr.offset() as usize)
28 .expect("Writing code to memory failed.");
29
30 let kvm = Kvm::new().expect("new kvm failed");
31 let mut vm = Vm::new(&kvm, guest_mem).expect("new vm failed");
32 let vcpu = Vcpu::new(0, &kvm, &vm).expect("new vcpu failed");
33 let mut vcpu_sregs = vcpu.get_sregs().expect("get sregs failed");
34 vcpu_sregs.cs.base = 0;
35 vcpu_sregs.cs.selector = 0;
36 vcpu.set_sregs(&vcpu_sregs).expect("set sregs failed");
37
38 let mut vcpu_regs: kvm_regs = unsafe { std::mem::zeroed() };
39 vcpu_regs.rip = load_addr.offset() as u64;
40 vcpu_regs.rflags = 2;
41 // Write 0x12 to the beginning of the 9th page.
42 vcpu_regs.rsi = 0x8000;
43 vcpu_regs.rbx = 0x12;
44 vcpu.set_regs(&vcpu_regs).expect("set regs failed");
45 let slot = vm
46 .add_device_memory(
47 GuestAddress(0),
48 MemoryMapping::from_fd(&mem, mem_size as usize)
49 .expect("failed to create memory mapping"),
50 false,
51 true,
52 )
53 .expect("failed to register memory");
54
55 loop {
56 match vcpu.run().expect("run failed") {
57 VcpuExit::Hlt => break,
58 r => panic!("unexpected exit reason: {:?}", r),
59 }
60 }
61
62 let mut dirty_log = [0x0, 0x0];
63 vm.get_dirty_log(slot, &mut dirty_log[..])
64 .expect("failed to get dirty log");
65 // Tests the 9th page was written to.
66 assert_eq!(dirty_log[1], 0x1);
67 assert_eq!(
68 mmap.read_obj::<u64>(vcpu_regs.rsi as usize).unwrap(),
69 vcpu_regs.rbx
70 );
71 }
72