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