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 268A07 mov al,[es:bx]
15 0003 0401 add al,0x1
16 0005 268807 mov [es:bx],al
17 0008 F4 hlt
18 */
19 let code = [0x26, 0x8a, 0x07, 0x04, 0x01, 0x26, 0x88, 0x07, 0xf4];
20 let mem_size = 0x2000;
21 let load_addr = GuestAddress(0x1000);
22 let guest_mem = GuestMemory::new(&[]).unwrap();
23 let mut mem = SharedMemory::new(None).expect("failed to create shared memory");
24 mem.set_size(mem_size)
25 .expect("failed to set shared memory size");
26 let mmap =
27 MemoryMapping::from_fd(&mem, mem_size as usize).expect("failed to create memory mapping");
28
29 mmap.write_slice(&code[..], load_addr.offset() as usize)
30 .expect("Writing code to memory failed.");
31
32 let kvm = Kvm::new().expect("new kvm failed");
33 let mut vm = Vm::new(&kvm, guest_mem).expect("new vm failed");
34 let vcpu = Vcpu::new(0, &kvm, &vm).expect("new vcpu failed");
35 let mut vcpu_sregs = vcpu.get_sregs().expect("get sregs failed");
36 vcpu_sregs.cs.base = 0;
37 vcpu_sregs.cs.selector = 0;
38 vcpu_sregs.es.base = 0x3000;
39 vcpu_sregs.es.selector = 0;
40 vcpu.set_sregs(&vcpu_sregs).expect("set sregs failed");
41
42 let mut vcpu_regs: kvm_regs = unsafe { std::mem::zeroed() };
43 vcpu_regs.rip = load_addr.offset() as u64;
44 vcpu_regs.rflags = 2;
45 vcpu_regs.rax = 0x66;
46 vcpu_regs.rbx = 0;
47 vcpu.set_regs(&vcpu_regs).expect("set regs failed");
48 vm.add_device_memory(
49 GuestAddress(0),
50 MemoryMapping::from_fd(&mem, mem_size as usize).expect("failed to create memory mapping"),
51 false,
52 false,
53 )
54 .expect("failed to register memory");
55
56 // Give some read only memory for the test code to read from and force a vcpu exit when it reads
57 // from it.
58 let mut mem_ro = SharedMemory::new(None).expect("failed to create shared memory");
59 mem_ro
60 .set_size(0x1000)
61 .expect("failed to set shared memory size");
62 let mmap_ro = MemoryMapping::from_fd(&mem_ro, 0x1000).expect("failed to create memory mapping");
63 mmap_ro
64 .write_obj(vcpu_regs.rax as u8, 0)
65 .expect("failed writing data to ro memory");
66 vm.add_device_memory(
67 GuestAddress(vcpu_sregs.es.base),
68 MemoryMapping::from_fd(&mem_ro, 0x1000).expect("failed to create memory mapping"),
69 true,
70 false,
71 )
72 .expect("failed to register memory");
73
74 // Ensure we get exactly 1 exit from attempting to write to read only memory.
75 let mut exits = 0;
76
77 loop {
78 match vcpu.run().expect("run failed") {
79 VcpuExit::Hlt => break,
80 VcpuExit::MmioWrite {
81 address,
82 size: 1,
83 data,
84 } => {
85 assert_eq!(address, vcpu_sregs.es.base);
86 assert_eq!(data[0] as u64, vcpu_regs.rax + 1);
87 exits += 1;
88 }
89 r => panic!("unexpected exit reason: {:?}", r),
90 }
91 }
92
93 // Check that exactly 1 attempt to write to read only memory was made, and that the memory is
94 // unchanged after that attempt.
95 assert_eq!(exits, 1);
96 assert_eq!(
97 mmap_ro
98 .read_obj::<u8>(0)
99 .expect("failed to read data from ro memory"),
100 vcpu_regs.rax as u8
101 );
102 }
103