mirrors/crosvm
1
0
Fork 0
mirror of https://chromium.googlesource.com/crosvm/crosvm synced 2025-01-12 16:45:31 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

kvm: fix get_msrs() to handle case where KVM does not fetch all MSRs

Browse source 
KVM may not return all MSRs that were requested in KVM_GET_MSRS call and
instead stop early. We should handle this case.

BUG=None
TEST=cargo test -p kvm

Change-Id: I18402c0a07b1d0c7657c171873d521fd2f223611
Signed-off-by: Dmitry Torokhov <dtor@chromium.org>
Reviewed-on: https://chromium-review.googlesource.com/1192231
Reviewed-by: Zach Reizner <zachr@chromium.org>
Reviewed-by: Dylan Reid <dgreid@chromium.org>
This commit is contained in:
Dmitry Torokhov 2018-08-27 12:35:49 -07:00 committed by chrome-bot
parent 45e6c843bb
commit 9dec40e242
1 changed files with 13 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
kvm/src/lib.rs
View file

@ -1076,7 +1076,7 @@ impl Vcpu {
///
/// See the documentation for KVM_SET_MSRS.
#[cfg(any(target_arch = "x86", target_arch = "x86_64"))]
pub fn get_msrs(&self, msr_entries: &mut [kvm_msr_entry]) -> Result<()> {
pub fn get_msrs(&self, msr_entries: &mut Vec<kvm_msr_entry>) -> Result<()> {
let vec_size_bytes = size_of::<kvm_msrs>() +
(msr_entries.len() * size_of::<kvm_msr_entry>());
let vec: Vec<u8> = vec![0; vec_size_bytes];
@ -1101,7 +1101,10 @@ impl Vcpu {
return errno_result();
}
unsafe {
let entries: &mut [kvm_msr_entry] = msrs.entries.as_mut_slice(msr_entries.len());
let count = ret as usize;
assert!(count <= msr_entries.len());
let entries: &mut [kvm_msr_entry] = msrs.entries.as_mut_slice(count);
msr_entries.truncate(count);
msr_entries.copy_from_slice(&entries);
}
Ok(())
@ -1691,15 +1694,14 @@ mod tests {
let gm = GuestMemory::new(&vec![(GuestAddress(0), 0x10000)]).unwrap();
let vm = Vm::new(&kvm, gm).unwrap();
let vcpu = Vcpu::new(0, &kvm, &vm).unwrap();
vcpu.get_msrs(&mut [kvm_msr_entry {
index: 0x0000011e,
..Default::default()
},
kvm_msr_entry {
index: 0x000003f1,
..Default::default()
}])
.unwrap();
let mut msrs = vec![
// This one should succeed
kvm_msr_entry { index: 0x0000011e, ..Default::default() },
// This one will fail to fetch
kvm_msr_entry { index: 0x000003f1, ..Default::default() },
];
vcpu.get_msrs(&mut msrs).unwrap();
assert_eq!(msrs.len(), 1);
}
#[test]