mirror of
https://chromium.googlesource.com/crosvm/crosvm
synced 2024-10-23 20:59:45 +00:00
fc88a6091d
Handling page faults using userfaultfd for all the time after vmm-swap is enabled originally has overhead. By disabling swapping, we can use crosvm without the overhead when the crosvm become used heavily again. This also refactor PageHandler and extract register logic to lib.rs. BUG=b:215093219 TEST=cargo test -p swap Change-Id: Id81d84ebe40067b4f19b212f6b81cbaf249c0c3c Reviewed-on: https://chromium-review.googlesource.com/c/crosvm/crosvm/+/4005474 Reviewed-by: David Stevens <stevensd@chromium.org> Commit-Queue: Shin Kawamura <kawasin@google.com>
83 lines
2.6 KiB
Rust
83 lines
2.6 KiB
Rust
// Copyright 2022 The ChromiumOS Authors
|
|
// Use of this source code is governed by a BSD-style license that can be
|
|
// found in the LICENSE file.
|
|
|
|
//! Integration tests for vmm-swap feature
|
|
|
|
mod common;
|
|
|
|
use base::pagesize;
|
|
use base::sys::wait_for_pid;
|
|
use base::AsRawDescriptor;
|
|
use base::FromRawDescriptor;
|
|
use base::IntoRawDescriptor;
|
|
use base::SafeDescriptor;
|
|
use base::Tube;
|
|
use common::*;
|
|
use swap::register_regions;
|
|
use swap::unregister_regions;
|
|
use swap::userfaultfd::Userfaultfd;
|
|
|
|
#[test]
|
|
fn register_region_skip_obsolete_process() {
|
|
let shm = create_shared_memory("test", 3 * pagesize());
|
|
let uffd = create_uffd_for_test();
|
|
let base_addr = shm.base_addr();
|
|
let regions = [base_addr..(base_addr + 3 * pagesize())];
|
|
let (tube_main, tube_child) = Tube::pair().unwrap();
|
|
let pid = unsafe { libc::fork() };
|
|
if pid == 0 {
|
|
// child process
|
|
let uffd = create_uffd_for_test();
|
|
tube_child
|
|
.send(&unsafe { SafeDescriptor::from_raw_descriptor(uffd.as_raw_descriptor()) })
|
|
.unwrap();
|
|
std::process::exit(0);
|
|
}
|
|
let uffd_descriptor = tube_main
|
|
.recv::<SafeDescriptor>()
|
|
.unwrap()
|
|
.into_raw_descriptor();
|
|
wait_for_pid(pid, 0).unwrap();
|
|
let uffd_child = unsafe { Userfaultfd::from_raw_descriptor(uffd_descriptor) };
|
|
|
|
let result = unsafe { register_regions(®ions, &[uffd, uffd_child]) };
|
|
|
|
// no error from ENOMEM
|
|
assert_eq!(result.is_ok(), true);
|
|
}
|
|
|
|
#[test]
|
|
fn unregister_region_skip_obsolete_process() {
|
|
let shm = create_shared_memory("test", 3 * pagesize());
|
|
let uffd = create_uffd_for_test();
|
|
let base_addr = shm.base_addr();
|
|
let regions = [base_addr..(base_addr + 3 * pagesize())];
|
|
let (tube_main, tube_child) = Tube::pair().unwrap();
|
|
let pid = unsafe { libc::fork() };
|
|
if pid == 0 {
|
|
// child process
|
|
let uffd = create_uffd_for_test();
|
|
tube_child
|
|
.send(&unsafe { SafeDescriptor::from_raw_descriptor(uffd.as_raw_descriptor()) })
|
|
.unwrap();
|
|
tube_child.recv::<u8>().unwrap();
|
|
std::process::exit(0);
|
|
}
|
|
let uffd_descriptor = tube_main
|
|
.recv::<SafeDescriptor>()
|
|
.unwrap()
|
|
.into_raw_descriptor();
|
|
let uffd_child = unsafe { Userfaultfd::from_raw_descriptor(uffd_descriptor) };
|
|
let uffds = [uffd, uffd_child];
|
|
|
|
unsafe { register_regions(®ions, &uffds) }.unwrap();
|
|
tube_main.send(&0_u8).unwrap();
|
|
// wait until the child process die and the uffd_child become obsolete.
|
|
wait_for_pid(pid, 0).unwrap();
|
|
let result = unregister_regions(®ions, &uffds);
|
|
|
|
// no error from ENOMEM
|
|
assert_eq!(result.is_ok(), true);
|
|
}
|