blob: 4532ef9aa5ac3d168697be6fb7e9956bddb25a5d [file] [log] [blame]
Andrew Walbrand6dce6f2021-03-05 16:39:08 +00001// Copyright 2021, The Android Open Source Project
2//
3// Licensed under the Apache License, Version 2.0 (the "License");
4// you may not use this file except in compliance with the License.
5// You may obtain a copy of the License at
6//
7// http://www.apache.org/licenses/LICENSE-2.0
8//
9// Unless required by applicable law or agreed to in writing, software
10// distributed under the License is distributed on an "AS IS" BASIS,
11// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12// See the License for the specific language governing permissions and
13// limitations under the License.
14
15//! Functions for running instances of `crosvm`.
16
Seungjae Yoo3271f172024-06-11 10:14:07 +090017use crate::aidl::{remove_temporary_files, Cid, GLOBAL_SERVICE, VirtualMachineCallbacks};
Seungjae Yoob2db3d22023-04-06 18:05:43 +090018use crate::atom::{get_num_cpus, write_vm_exited_stats_sync};
Jaewan Kim61f86142023-03-28 15:12:52 +090019use crate::debug_config::DebugConfig;
Seungjae Yoo6d265d92022-11-15 10:51:33 +090020use anyhow::{anyhow, bail, Context, Error, Result};
Seungjae Yoo3271f172024-06-11 10:14:07 +090021use binder::ParcelFileDescriptor;
Andrew Walbran02b8ec02021-06-22 13:07:02 +000022use command_fds::CommandFdExt;
Seungjae Yoo6d265d92022-11-15 10:51:33 +090023use libc::{sysconf, _SC_CLK_TCK};
Andrew Walbran3a5a9212021-05-04 17:09:08 +000024use log::{debug, error, info};
Jiyong Parkdcf17412022-02-08 15:07:23 +090025use semver::{Version, VersionReq};
Jiyong Park08eee7b2022-11-11 15:01:26 +090026use nix::{fcntl::OFlag, unistd::pipe2, unistd::Uid, unistd::User};
Jiyong Park2d736562022-10-24 22:40:12 +090027use regex::{Captures, Regex};
Keir Fraserf25cb922022-11-23 14:26:00 +000028use rustutils::system_properties;
Andrew Walbrandae07162021-03-12 17:05:20 +000029use shared_child::SharedChild;
Jiyong Parke6ed0f92022-06-22 00:13:00 +090030use std::borrow::Cow;
Seungjae Yoo6d265d92022-11-15 10:51:33 +090031use std::cmp::max;
Frederick Maylef375ef12024-10-21 16:07:21 -070032use std::ffi::CString;
Jiyong Park08eee7b2022-11-11 15:01:26 +090033use std::fmt;
David Brazdil4b4c5102022-12-19 22:56:20 +000034use std::fs::{read_to_string, File};
Andrew Walbranb27681f2022-02-23 15:11:52 +000035use std::io::{self, Read};
Andrew Walbranf8d94112021-09-07 11:45:36 +000036use std::mem;
Nikita Ioffe5776f082023-02-10 21:38:26 +000037use std::num::{NonZeroU16, NonZeroU32};
Jiyong Park69001a22024-08-16 20:45:34 +090038use std::os::unix::io::{AsRawFd, OwnedFd};
Seungjae Yoo93430e82022-12-05 16:37:42 +090039use std::os::unix::process::ExitStatusExt;
Jiyong Park1612b902022-08-22 14:47:39 +090040use std::path::{Path, PathBuf};
Andrew Walbranc92d35f2022-01-12 12:45:19 +000041use std::process::{Command, ExitStatus};
Andrew Walbran9c03a3a2024-09-03 12:12:59 +010042use std::sync::{Arc, Condvar, Mutex, LazyLock};
Seungjae Yoo2e7beea2022-08-24 16:09:12 +090043use std::time::{Duration, SystemTime};
David Brazdilf18e1162022-12-18 18:27:23 +000044use std::thread::{self, JoinHandle};
David Brazdil49f96f52022-12-16 21:29:13 +000045use android_system_virtualizationcommon::aidl::android::system::virtualizationcommon::DeathReason::DeathReason;
Jaewan Kim84b91212023-02-28 00:11:57 +090046use android_system_virtualizationservice::aidl::android::system::virtualizationservice::{
Jeongik Cha1df1c032024-04-03 16:03:12 +090047 VirtualMachineAppConfig::DebugLevel::DebugLevel,
Mu-Le Leeb2d5a312024-06-05 10:52:47 +080048 AudioConfig::AudioConfig as AudioConfigParcelable,
Jeongik Cha1df1c032024-04-03 16:03:12 +090049 DisplayConfig::DisplayConfig as DisplayConfigParcelable,
Jason Macnakea7b0fa2024-05-08 17:02:44 -070050 GpuConfig::GpuConfig as GpuConfigParcelable,
Elie Kheirallah29d72912024-08-07 20:17:26 +000051 UsbConfig::UsbConfig as UsbConfigParcelable,
Jaewan Kim84b91212023-02-28 00:11:57 +090052};
David Brazdil528e0472022-10-10 15:06:02 +010053use android_system_virtualizationservice_internal::aidl::android::system::virtualizationservice_internal::IGlobalVmContext::IGlobalVmContext;
Jakob Vukalovicd42aa2c2023-11-09 16:04:00 +000054use android_system_virtualizationservice_internal::aidl::android::system::virtualizationservice_internal::IBoundDevice::IBoundDevice;
Alan Stokes0e82b502022-08-08 14:44:48 +010055use binder::Strong;
Inseob Kimc7d28c72021-10-25 14:28:10 +000056use android_system_virtualmachineservice::aidl::android::system::virtualmachineservice::IVirtualMachineService::IVirtualMachineService;
Jiyong Park1612b902022-08-22 14:47:39 +090057use tombstoned_client::{TombstonedConnection, DebuggerdDumpType};
David Brazdil73988ea2022-11-11 15:10:32 +000058use rpcbinder::RpcServer;
Andrew Walbrand6dce6f2021-03-05 16:39:08 +000059
60const CROSVM_PATH: &str = "/apex/com.android.virt/bin/crosvm";
61
Jiyong Parkdcf17412022-02-08 15:07:23 +090062/// Version of the platform that crosvm currently implements. The format follows SemVer. This
63/// should be updated when there is a platform change in the crosvm side. Having this value here is
64/// fine because virtualizationservice and crosvm are supposed to be updated together in the virt
65/// APEX.
66const CROSVM_PLATFORM_VERSION: &str = "1.0.0";
67
Andrew Walbrand15c5632022-02-03 13:38:31 +000068/// The exit status which crosvm returns when it has an error starting a VM.
Alan Stokes7c459e82022-12-06 12:21:49 +000069const CROSVM_START_ERROR_STATUS: i32 = 1;
Andrew Walbranc92d35f2022-01-12 12:45:19 +000070/// The exit status which crosvm returns when a VM requests a reboot.
71const CROSVM_REBOOT_STATUS: i32 = 32;
Andrew Walbrand15c5632022-02-03 13:38:31 +000072/// The exit status which crosvm returns when it crashes due to an error.
73const CROSVM_CRASH_STATUS: i32 = 33;
Sebastian Ene23167d82022-10-07 14:09:53 +000074/// The exit status which crosvm returns when vcpu is stalled.
75const CROSVM_WATCHDOG_REBOOT_STATUS: i32 = 36;
Jiyong Park3115cdb2023-02-24 18:04:44 +090076/// The size of memory (in MiB) reserved for ramdump
77const RAMDUMP_RESERVED_MIB: u32 = 17;
Andrew Walbranc92d35f2022-01-12 12:45:19 +000078
Seungjae Yoo6d265d92022-11-15 10:51:33 +090079const MILLIS_PER_SEC: i64 = 1000;
80
Jaewan Kim46422812022-11-25 10:59:39 +090081const SYSPROP_CUSTOM_PVMFW_PATH: &str = "hypervisor.pvmfw.path";
82
Yi-Yo Chiang8dd32552024-05-22 19:38:16 +080083/// Serial device for VM console input.
84/// Hypervisor (virtio-console)
85const CONSOLE_HVC0: &str = "hvc0";
86/// Serial (emulated uart)
87const CONSOLE_TTYS0: &str = "ttyS0";
88
Andrew Walbran9c03a3a2024-09-03 12:12:59 +010089/// If the VM doesn't move to the Started state within this amount time, a hang-up error is
90/// triggered.
91static BOOT_HANGUP_TIMEOUT: LazyLock<Duration> = LazyLock::new(|| {
92 if nested_virt::is_nested_virtualization().unwrap() {
Jiyong Parke6ed0f92022-06-22 00:13:00 +090093 // Nested virtualization is slow, so we need a longer timeout.
Inseob Kime557cec2023-01-26 22:49:58 +090094 Duration::from_secs(300)
Jiyong Parke6ed0f92022-06-22 00:13:00 +090095 } else {
Inseob Kime557cec2023-01-26 22:49:58 +090096 Duration::from_secs(30)
Andrew Walbran9c03a3a2024-09-03 12:12:59 +010097 }
98});
Jiyong Parke6ed0f92022-06-22 00:13:00 +090099
Andrew Walbranf5fbb7d2021-05-12 17:15:48 +0000100/// Configuration for a VM to run with crosvm.
101#[derive(Debug)]
Andrew Walbrand3a84182021-09-07 14:48:52 +0000102pub struct CrosvmConfig {
Andrew Walbranf5fbb7d2021-05-12 17:15:48 +0000103 pub cid: Cid,
Seungjae Yoo62085c02022-08-12 04:44:52 +0000104 pub name: String,
Andrew Walbrand3a84182021-09-07 14:48:52 +0000105 pub bootloader: Option<File>,
106 pub kernel: Option<File>,
107 pub initrd: Option<File>,
Andrew Walbranf5fbb7d2021-05-12 17:15:48 +0000108 pub disks: Vec<DiskFile>,
109 pub params: Option<String>,
Andrew Walbranf8650422021-06-09 15:54:09 +0000110 pub protected: bool,
Jaewan Kim61f86142023-03-28 15:12:52 +0900111 pub debug_config: DebugConfig,
Frederick Mayle68d06e72024-07-12 17:18:11 -0700112 pub memory_mib: NonZeroU32,
Jiyong Park032615f2022-01-10 13:55:34 +0900113 pub cpus: Option<NonZeroU32>,
David Brazdil7d1e5ec2023-02-06 17:56:29 +0000114 pub host_cpu_topology: bool,
Jiyong Parke6fb1672023-06-26 16:45:55 +0900115 pub console_out_fd: Option<File>,
116 pub console_in_fd: Option<File>,
Andrew Walbrand3a84182021-09-07 14:48:52 +0000117 pub log_fd: Option<File>,
Jiyong Parke558ab12022-07-07 20:18:55 +0900118 pub ramdump: Option<File>,
Andrew Walbrand3a84182021-09-07 14:48:52 +0000119 pub indirect_files: Vec<File>,
Jiyong Parkdcf17412022-02-08 15:07:23 +0900120 pub platform_version: VersionReq,
Jiyong Parke6ed0f92022-06-22 00:13:00 +0900121 pub detect_hangup: bool,
Nikita Ioffe5776f082023-02-10 21:38:26 +0000122 pub gdb_port: Option<NonZeroU16>,
Inseob Kim7307a892023-09-14 13:37:58 +0900123 pub vfio_devices: Vec<VfioDevice>,
David Brazdil2dfefd12023-11-17 14:07:36 +0000124 pub dtbo: Option<File>,
Shikha Panwar55e10ec2024-02-13 12:53:49 +0000125 pub device_tree_overlay: Option<File>,
Jeongik Cha1df1c032024-04-03 16:03:12 +0900126 pub display_config: Option<DisplayConfig>,
Jeongik Chac181be72024-03-27 00:18:30 +0900127 pub input_device_options: Vec<InputDeviceOption>,
Vincent Donnefort538a2c62024-03-20 16:01:10 +0000128 pub hugepages: bool,
Seungjae Yoo986d7a42024-05-28 14:06:58 +0900129 pub tap: Option<File>,
Yi-Yo Chiang8dd32552024-05-22 19:38:16 +0800130 pub console_input_device: Option<String>,
David Dai23cff712024-06-13 19:23:45 +0000131 pub boost_uclamp: bool,
Jason Macnakea7b0fa2024-05-08 17:02:44 -0700132 pub gpu_config: Option<GpuConfig>,
Mu-Le Leeb2d5a312024-06-05 10:52:47 +0800133 pub audio_config: Option<AudioConfig>,
Jeongik Cha75057452024-09-03 11:26:04 +0900134 pub no_balloon: bool,
Elie Kheirallah29d72912024-08-07 20:17:26 +0000135 pub usb_config: UsbConfig,
Elie Kheirallah5c807a22024-09-23 20:40:42 +0000136 pub dump_dt_fd: Option<File>,
Mu-Le Leeb2d5a312024-06-05 10:52:47 +0800137}
138
139#[derive(Debug)]
140pub struct AudioConfig {
141 pub use_microphone: bool,
142 pub use_speaker: bool,
143}
144
145impl AudioConfig {
146 pub fn new(raw_config: &AudioConfigParcelable) -> Self {
147 AudioConfig { use_microphone: raw_config.useMicrophone, use_speaker: raw_config.useSpeaker }
148 }
Jeongik Cha1df1c032024-04-03 16:03:12 +0900149}
150
151#[derive(Debug)]
Elie Kheirallah29d72912024-08-07 20:17:26 +0000152pub struct UsbConfig {
153 pub controller: bool,
154}
155
156impl UsbConfig {
157 pub fn new(raw_config: &UsbConfigParcelable) -> Result<UsbConfig> {
158 Ok(UsbConfig { controller: raw_config.controller })
159 }
160}
161
162#[derive(Debug)]
Jeongik Cha1df1c032024-04-03 16:03:12 +0900163pub struct DisplayConfig {
164 pub width: NonZeroU32,
165 pub height: NonZeroU32,
166 pub horizontal_dpi: NonZeroU32,
167 pub vertical_dpi: NonZeroU32,
168 pub refresh_rate: NonZeroU32,
169}
170
171impl DisplayConfig {
172 pub fn new(raw_config: &DisplayConfigParcelable) -> Result<DisplayConfig> {
173 let width = try_into_non_zero_u32(raw_config.width)?;
174 let height = try_into_non_zero_u32(raw_config.height)?;
175 let horizontal_dpi = try_into_non_zero_u32(raw_config.horizontalDpi)?;
176 let vertical_dpi = try_into_non_zero_u32(raw_config.verticalDpi)?;
177 let refresh_rate = try_into_non_zero_u32(raw_config.refreshRate)?;
178 Ok(DisplayConfig { width, height, horizontal_dpi, vertical_dpi, refresh_rate })
179 }
180}
181
Jason Macnakea7b0fa2024-05-08 17:02:44 -0700182#[derive(Debug)]
183pub struct GpuConfig {
184 pub backend: Option<String>,
185 pub context_types: Option<Vec<String>>,
186 pub pci_address: Option<String>,
187 pub renderer_features: Option<String>,
188 pub renderer_use_egl: Option<bool>,
189 pub renderer_use_gles: Option<bool>,
190 pub renderer_use_glx: Option<bool>,
191 pub renderer_use_surfaceless: Option<bool>,
192 pub renderer_use_vulkan: Option<bool>,
193}
194
195impl GpuConfig {
196 pub fn new(raw_config: &GpuConfigParcelable) -> Result<GpuConfig> {
197 Ok(GpuConfig {
198 backend: raw_config.backend.clone(),
199 context_types: raw_config.contextTypes.clone().map(|context_types| {
200 context_types.iter().filter_map(|context_type| context_type.clone()).collect()
201 }),
202 pci_address: raw_config.pciAddress.clone(),
203 renderer_features: raw_config.rendererFeatures.clone(),
204 renderer_use_egl: Some(raw_config.rendererUseEgl),
205 renderer_use_gles: Some(raw_config.rendererUseGles),
206 renderer_use_glx: Some(raw_config.rendererUseGlx),
207 renderer_use_surfaceless: Some(raw_config.rendererUseSurfaceless),
208 renderer_use_vulkan: Some(raw_config.rendererUseVulkan),
209 })
210 }
211}
212
Jeongik Cha1df1c032024-04-03 16:03:12 +0900213fn try_into_non_zero_u32(value: i32) -> Result<NonZeroU32> {
214 let u32_value = value.try_into()?;
215 NonZeroU32::new(u32_value).ok_or(anyhow!("value should be greater than 0"))
Andrew Walbranf5fbb7d2021-05-12 17:15:48 +0000216}
217
218/// A disk image to pass to crosvm for a VM.
219#[derive(Debug)]
220pub struct DiskFile {
221 pub image: File,
222 pub writable: bool,
223}
224
Jeongik Chac181be72024-03-27 00:18:30 +0900225/// virtio-input device configuration from `external/crosvm/src/crosvm/config.rs`
226#[derive(Debug)]
227#[allow(dead_code)]
228pub enum InputDeviceOption {
229 EvDev(File),
230 SingleTouch { file: File, width: u32, height: u32, name: Option<String> },
Jeongik Cha985b6402024-04-15 18:13:00 +0900231 Keyboard(File),
Jeongik Cha343894e2024-05-17 20:39:31 +0900232 Mouse(File),
Jiyong Park6c1b9f02024-07-05 16:21:10 +0900233 Switches(File),
Jeongik Cha112a2682024-07-09 12:28:45 +0900234 MultiTouchTrackpad { file: File, width: u32, height: u32, name: Option<String> },
Jeongik Cha10494912024-07-16 11:19:50 +0900235 MultiTouch { file: File, width: u32, height: u32, name: Option<String> },
Jeongik Chac181be72024-03-27 00:18:30 +0900236}
237
Jakob Vukalovicd42aa2c2023-11-09 16:04:00 +0000238type VfioDevice = Strong<dyn IBoundDevice>;
Inseob Kim7307a892023-09-14 13:37:58 +0900239
Andrew Walbran6b650662021-09-07 13:13:23 +0000240/// The lifecycle state which the payload in the VM has reported itself to be in.
241///
242/// Note that the order of enum variants is significant; only forward transitions are allowed by
243/// [`VmInstance::update_payload_state`].
244#[derive(Copy, Clone, Debug, Eq, Ord, PartialEq, PartialOrd)]
245pub enum PayloadState {
246 Starting,
247 Started,
248 Ready,
249 Finished,
Jiyong Parka4eebde2022-07-12 18:01:12 +0900250 Hangup, // Hasn't reached to Ready before timeout expires
Andrew Walbran6b650662021-09-07 13:13:23 +0000251}
252
Andrew Walbranf8d94112021-09-07 11:45:36 +0000253/// The current state of the VM itself.
254#[derive(Debug)]
255pub enum VmState {
256 /// The VM has not yet tried to start.
257 NotStarted {
258 ///The configuration needed to start the VM, if it has not yet been started.
Inseob Kim7307a892023-09-14 13:37:58 +0900259 config: Box<CrosvmConfig>,
Andrew Walbranf8d94112021-09-07 11:45:36 +0000260 },
261 /// The VM has been started.
262 Running {
263 /// The crosvm child process.
264 child: Arc<SharedChild>,
David Brazdilf18e1162022-12-18 18:27:23 +0000265 /// The thread waiting for crosvm to finish.
266 monitor_vm_exit_thread: Option<JoinHandle<()>>,
Andrew Walbranf8d94112021-09-07 11:45:36 +0000267 },
268 /// The VM died or was killed.
269 Dead,
270 /// The VM failed to start.
271 Failed,
272}
273
Seungjae Yoo6d265d92022-11-15 10:51:33 +0900274/// RSS values of VM and CrosVM process itself.
275#[derive(Copy, Clone, Debug, Default)]
276pub struct Rss {
277 pub vm: i64,
278 pub crosvm: i64,
279}
280
281/// Metrics regarding the VM.
282#[derive(Debug, Default)]
283pub struct VmMetric {
284 /// Recorded timestamp when the VM is started.
285 pub start_timestamp: Option<SystemTime>,
Pierre-Clément Tosi44aba642023-09-15 11:44:34 +0100286 /// Update most recent guest_time periodically from /proc/[crosvm pid]/stat while VM is
287 /// running.
Seungjae Yoo6d265d92022-11-15 10:51:33 +0900288 pub cpu_guest_time: Option<i64>,
289 /// Update maximum RSS values periodically from /proc/[crosvm pid]/smaps while VM is running.
290 pub rss: Option<Rss>,
291}
292
Andrew Walbranf8d94112021-09-07 11:45:36 +0000293impl VmState {
294 /// Tries to start the VM, if it is in the `NotStarted` state.
295 ///
296 /// Returns an error if the VM is in the wrong state, or fails to start.
297 fn start(&mut self, instance: Arc<VmInstance>) -> Result<(), Error> {
298 let state = mem::replace(self, VmState::Failed);
299 if let VmState::NotStarted { config } = state {
Inseob Kim7307a892023-09-14 13:37:58 +0900300 let config = *config;
Jiyong Parke6ed0f92022-06-22 00:13:00 +0900301 let detect_hangup = config.detect_hangup;
Andrew Walbranb27681f2022-02-23 15:11:52 +0000302 let (failure_pipe_read, failure_pipe_write) = create_pipe()?;
Inseob Kim6ef80972023-07-20 17:23:36 +0900303 let vfio_devices = config.vfio_devices.clone();
Seungjae Yoo3271f172024-06-11 10:14:07 +0900304 let tap =
305 if let Some(tap_file) = &config.tap { Some(tap_file.try_clone()?) } else { None };
Andrew Walbranb27681f2022-02-23 15:11:52 +0000306
Andrew Walbranf8d94112021-09-07 11:45:36 +0000307 // If this fails and returns an error, `self` will be left in the `Failed` state.
Keir Fraser13a956a2022-07-14 14:20:46 +0000308 let child =
Keir Frasercdd4b112022-11-24 14:02:25 +0000309 Arc::new(run_vm(config, &instance.crosvm_control_socket_path, failure_pipe_write)?);
Andrew Walbranf8d94112021-09-07 11:45:36 +0000310
Seungjae Yoo6d265d92022-11-15 10:51:33 +0900311 let instance_monitor_status = instance.clone();
312 let child_monitor_status = child.clone();
313 thread::spawn(move || {
314 instance_monitor_status.clone().monitor_vm_status(child_monitor_status);
315 });
316
Andrew Walbranf8d94112021-09-07 11:45:36 +0000317 let child_clone = child.clone();
Jiyong Parka4eebde2022-07-12 18:01:12 +0900318 let instance_clone = instance.clone();
David Brazdilf18e1162022-12-18 18:27:23 +0000319 let monitor_vm_exit_thread = Some(thread::spawn(move || {
Seungjae Yoo3271f172024-06-11 10:14:07 +0900320 instance_clone.monitor_vm_exit(child_clone, failure_pipe_read, vfio_devices, tap);
David Brazdilf18e1162022-12-18 18:27:23 +0000321 }));
Andrew Walbranf8d94112021-09-07 11:45:36 +0000322
Jiyong Parka4eebde2022-07-12 18:01:12 +0900323 if detect_hangup {
324 let child_clone = child.clone();
325 thread::spawn(move || {
326 instance.monitor_payload_hangup(child_clone);
327 });
328 }
329
Andrew Walbranf8d94112021-09-07 11:45:36 +0000330 // If it started correctly, update the state.
David Brazdilf18e1162022-12-18 18:27:23 +0000331 *self = VmState::Running { child, monitor_vm_exit_thread };
Andrew Walbranf8d94112021-09-07 11:45:36 +0000332 Ok(())
333 } else {
334 *self = state;
335 bail!("VM already started or failed")
336 }
337 }
338}
339
David Brazdil8cf8f482022-11-23 14:21:26 +0000340/// Internal struct that holds the handles to globally unique resources of a VM.
341#[derive(Debug)]
342pub struct VmContext {
343 #[allow(dead_code)] // Keeps the global context alive
Yi-Yo Chiang2fbf0da2024-06-14 22:56:56 +0800344 pub(crate) global_context: Strong<dyn IGlobalVmContext>,
David Brazdil8cf8f482022-11-23 14:21:26 +0000345 #[allow(dead_code)] // Keeps the server alive
346 vm_server: RpcServer,
347}
348
349impl VmContext {
350 /// Construct new VmContext.
351 pub fn new(global_context: Strong<dyn IGlobalVmContext>, vm_server: RpcServer) -> VmContext {
352 VmContext { global_context, vm_server }
353 }
354}
355
Andrew Walbranf8d94112021-09-07 11:45:36 +0000356/// Information about a particular instance of a VM which may be running.
Andrew Walbrand6dce6f2021-03-05 16:39:08 +0000357#[derive(Debug)]
358pub struct VmInstance {
Andrew Walbranf8d94112021-09-07 11:45:36 +0000359 /// The current state of the VM.
360 pub vm_state: Mutex<VmState>,
David Brazdil8cf8f482022-11-23 14:21:26 +0000361 /// Global resources allocated for this VM.
362 #[allow(dead_code)] // Keeps the context alive
Yi-Yo Chiang2fbf0da2024-06-14 22:56:56 +0800363 pub(crate) vm_context: VmContext,
Andrew Walbrand6dce6f2021-03-05 16:39:08 +0000364 /// The CID assigned to the VM for vsock communication.
365 pub cid: Cid,
Keir Frasercdd4b112022-11-24 14:02:25 +0000366 /// Path to crosvm control socket
367 crosvm_control_socket_path: PathBuf,
Seungjae Yoo62085c02022-08-12 04:44:52 +0000368 /// The name of the VM.
369 pub name: String,
Andrew Walbranf8650422021-06-09 15:54:09 +0000370 /// Whether the VM is a protected VM.
371 pub protected: bool,
Andrew Walbranf5fbb7d2021-05-12 17:15:48 +0000372 /// Directory of temporary files used by the VM while it is running.
373 pub temporary_directory: PathBuf,
Andrew Walbranf6a1eb92021-04-01 11:16:02 +0000374 /// The UID of the process which requested the VM.
375 pub requester_uid: u32,
Andrew Walbranf6a1eb92021-04-01 11:16:02 +0000376 /// The PID of the process which requested the VM. Note that this process may no longer exist
377 /// and the PID may have been reused for a different process, so this should not be trusted.
Andrew Walbran02034492021-04-13 15:05:07 +0000378 pub requester_debug_pid: i32,
Andrew Walbrandae07162021-03-12 17:05:20 +0000379 /// Callbacks to clients of the VM.
380 pub callbacks: VirtualMachineCallbacks,
Inseob Kimc7d28c72021-10-25 14:28:10 +0000381 /// VirtualMachineService binder object for the VM.
Stephen Hinesaf21e512024-07-30 15:43:39 -0700382 #[allow(dead_code)]
Inseob Kimc7d28c72021-10-25 14:28:10 +0000383 pub vm_service: Mutex<Option<Strong<dyn IVirtualMachineService>>>,
Seungjae Yoo6d265d92022-11-15 10:51:33 +0900384 /// Recorded metrics of VM such as timestamp or cpu / memory usage.
385 pub vm_metric: Mutex<VmMetric>,
Andrew Walbran6b650662021-09-07 13:13:23 +0000386 /// The latest lifecycle state which the payload reported itself to be in.
387 payload_state: Mutex<PayloadState>,
Jiyong Parka4eebde2022-07-12 18:01:12 +0900388 /// Represents the condition that payload_state was updated
389 payload_state_updated: Condvar,
Jiyong Park08eee7b2022-11-11 15:01:26 +0900390 /// The human readable name of requester_uid
391 requester_uid_name: String,
392}
393
394impl fmt::Display for VmInstance {
395 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
396 let adj = if self.protected { "Protected" } else { "Non-protected" };
397 write!(
398 f,
399 "{} virtual machine \"{}\" (owner: {}, cid: {})",
400 adj, self.name, self.requester_uid_name, self.cid
401 )
402 }
Andrew Walbrand6dce6f2021-03-05 16:39:08 +0000403}
404
405impl VmInstance {
Andrew Walbranf8d94112021-09-07 11:45:36 +0000406 /// Validates the given config and creates a new `VmInstance` but doesn't start running it.
407 pub fn new(
Andrew Walbrand3a84182021-09-07 14:48:52 +0000408 config: CrosvmConfig,
Andrew Walbranf5fbb7d2021-05-12 17:15:48 +0000409 temporary_directory: PathBuf,
Andrew Walbranf6a1eb92021-04-01 11:16:02 +0000410 requester_uid: u32,
Andrew Walbran02034492021-04-13 15:05:07 +0000411 requester_debug_pid: i32,
David Brazdil8cf8f482022-11-23 14:21:26 +0000412 vm_context: VmContext,
Andrew Walbranf8d94112021-09-07 11:45:36 +0000413 ) -> Result<VmInstance, Error> {
414 validate_config(&config)?;
Andrew Walbrand3a84182021-09-07 14:48:52 +0000415 let cid = config.cid;
Seungjae Yoo62085c02022-08-12 04:44:52 +0000416 let name = config.name.clone();
Andrew Walbrand3a84182021-09-07 14:48:52 +0000417 let protected = config.protected;
Jiyong Park08eee7b2022-11-11 15:01:26 +0900418 let requester_uid_name = User::from_uid(Uid::from_raw(requester_uid))
419 .ok()
420 .flatten()
421 .map_or_else(|| format!("{}", requester_uid), |u| u.name);
422 let instance = VmInstance {
Inseob Kim7307a892023-09-14 13:37:58 +0900423 vm_state: Mutex::new(VmState::NotStarted { config: Box::new(config) }),
David Brazdil528e0472022-10-10 15:06:02 +0100424 vm_context,
Andrew Walbrand3a84182021-09-07 14:48:52 +0000425 cid,
Keir Frasercdd4b112022-11-24 14:02:25 +0000426 crosvm_control_socket_path: temporary_directory.join("crosvm.sock"),
Seungjae Yoo62085c02022-08-12 04:44:52 +0000427 name,
Andrew Walbrand3a84182021-09-07 14:48:52 +0000428 protected,
Andrew Walbranf5fbb7d2021-05-12 17:15:48 +0000429 temporary_directory,
Andrew Walbran02034492021-04-13 15:05:07 +0000430 requester_uid,
Andrew Walbran02034492021-04-13 15:05:07 +0000431 requester_debug_pid,
Andrew Walbranf8d94112021-09-07 11:45:36 +0000432 callbacks: Default::default(),
Inseob Kimc7d28c72021-10-25 14:28:10 +0000433 vm_service: Mutex::new(None),
Seungjae Yoo6d265d92022-11-15 10:51:33 +0900434 vm_metric: Mutex::new(Default::default()),
Andrew Walbranf8d94112021-09-07 11:45:36 +0000435 payload_state: Mutex::new(PayloadState::Starting),
Jiyong Parka4eebde2022-07-12 18:01:12 +0900436 payload_state_updated: Condvar::new(),
Jiyong Park08eee7b2022-11-11 15:01:26 +0900437 requester_uid_name,
438 };
439 info!("{} created", &instance);
440 Ok(instance)
Andrew Walbrandae07162021-03-12 17:05:20 +0000441 }
442
Andrew Walbranf8d94112021-09-07 11:45:36 +0000443 /// Starts an instance of `crosvm` to manage the VM. The `crosvm` instance will be killed when
444 /// the `VmInstance` is dropped.
445 pub fn start(self: &Arc<Self>) -> Result<(), Error> {
Seungjae Yoo6d265d92022-11-15 10:51:33 +0900446 let mut vm_metric = self.vm_metric.lock().unwrap();
447 vm_metric.start_timestamp = Some(SystemTime::now());
Jiyong Park08eee7b2022-11-11 15:01:26 +0900448 let ret = self.vm_state.lock().unwrap().start(self.clone());
449 if ret.is_ok() {
450 info!("{} started", &self);
451 }
452 ret.with_context(|| format!("{} failed to start", &self))
Andrew Walbranf8d94112021-09-07 11:45:36 +0000453 }
454
Jiyong Parka4eebde2022-07-12 18:01:12 +0900455 /// Monitors the exit of the VM (i.e. termination of the `child` process). When that happens,
456 /// handles the event by updating the state, noityfing the event to clients by calling
457 /// callbacks, and removing temporary files for the VM.
Inseob Kim6ef80972023-07-20 17:23:36 +0900458 fn monitor_vm_exit(
459 &self,
460 child: Arc<SharedChild>,
461 mut failure_pipe_read: File,
Inseob Kim7307a892023-09-14 13:37:58 +0900462 vfio_devices: Vec<VfioDevice>,
Seungjae Yoo3271f172024-06-11 10:14:07 +0900463 tap: Option<File>,
Inseob Kim6ef80972023-07-20 17:23:36 +0900464 ) {
Andrew Walbranc92d35f2022-01-12 12:45:19 +0000465 let result = child.wait();
466 match &result {
Jooyung Hanbfe086f2021-10-28 10:15:45 +0900467 Err(e) => error!("Error waiting for crosvm({}) instance to die: {}", child.id(), e),
Sebastian Ene23167d82022-10-07 14:09:53 +0000468 Ok(status) => {
469 info!("crosvm({}) exited with status {}", child.id(), status);
470 if let Some(exit_status_code) = status.code() {
471 if exit_status_code == CROSVM_WATCHDOG_REBOOT_STATUS {
472 info!("detected vcpu stall on crosvm");
473 }
474 }
475 }
Andrew Walbrandae07162021-03-12 17:05:20 +0000476 }
Andrew Walbranf8d94112021-09-07 11:45:36 +0000477
478 let mut vm_state = self.vm_state.lock().unwrap();
479 *vm_state = VmState::Dead;
480 // Ensure that the mutex is released before calling the callbacks.
481 drop(vm_state);
Jiyong Park08eee7b2022-11-11 15:01:26 +0900482 info!("{} exited", &self);
Andrew Walbranf8d94112021-09-07 11:45:36 +0000483
Jiyong Parka4eebde2022-07-12 18:01:12 +0900484 // Read the pipe to see if any failure reason is written
485 let mut failure_reason = String::new();
486 match failure_pipe_read.read_to_string(&mut failure_reason) {
487 Err(e) => error!("Error reading VM failure reason from pipe: {}", e),
488 Ok(len) if len > 0 => info!("VM returned failure reason '{}'", &failure_reason),
489 _ => (),
Jiyong Parke6ed0f92022-06-22 00:13:00 +0900490 };
Andrew Walbranb27681f2022-02-23 15:11:52 +0000491
Jiyong Parka4eebde2022-07-12 18:01:12 +0900492 // In case of hangup, the pipe doesn't give us any information because the hangup can't be
493 // detected on the VM side (otherwise, it isn't a hangup), but in the
494 // monitor_payload_hangup function below which updates the payload state to Hangup.
495 let failure_reason =
496 if failure_reason.is_empty() && self.payload_state() == PayloadState::Hangup {
497 Cow::from("HANGUP")
498 } else {
499 Cow::from(failure_reason)
500 };
501
Jiyong Parke558ab12022-07-07 20:18:55 +0900502 self.handle_ramdump().unwrap_or_else(|e| error!("Error handling ramdump: {}", e));
Seungjae Yoob4c07ba2022-08-12 04:44:52 +0000503
504 let death_reason = death_reason(&result, &failure_reason);
Seungjae Yoo93430e82022-12-05 16:37:42 +0900505 let exit_signal = exit_signal(&result);
506
Seungjae Yoob4c07ba2022-08-12 04:44:52 +0000507 self.callbacks.callback_on_died(self.cid, death_reason);
Seungjae Yoo2e7beea2022-08-24 16:09:12 +0900508
Seungjae Yoo6d265d92022-11-15 10:51:33 +0900509 let vm_metric = self.vm_metric.lock().unwrap();
Seungjae Yoob2db3d22023-04-06 18:05:43 +0900510 write_vm_exited_stats_sync(
Seungjae Yoo93430e82022-12-05 16:37:42 +0900511 self.requester_uid as i32,
512 &self.name,
513 death_reason,
514 exit_signal,
Chris Wailes75269622022-12-05 23:01:44 -0800515 &vm_metric,
Seungjae Yoo93430e82022-12-05 16:37:42 +0900516 );
Andrew Walbranf5fbb7d2021-05-12 17:15:48 +0000517
David Brazdil4b4c5102022-12-19 22:56:20 +0000518 // Delete temporary files. The folder itself is removed by VirtualizationServiceInternal.
519 remove_temporary_files(&self.temporary_directory).unwrap_or_else(|e| {
520 error!("Error removing temporary files from {:?}: {}", self.temporary_directory, e);
521 });
Inseob Kim6ef80972023-07-20 17:23:36 +0900522
Seungjae Yoo3271f172024-06-11 10:14:07 +0900523 if let Some(tap_file) = tap {
524 GLOBAL_SERVICE
525 .deleteTapInterface(&ParcelFileDescriptor::new(OwnedFd::from(tap_file)))
526 .unwrap_or_else(|e| {
527 error!("Error deleting TAP interface: {e:?}");
528 });
529 }
530
Jakob Vukalovicd42aa2c2023-11-09 16:04:00 +0000531 drop(vfio_devices); // Cleanup devices.
Andrew Walbrandae07162021-03-12 17:05:20 +0000532 }
533
Jiyong Parka4eebde2022-07-12 18:01:12 +0900534 /// Waits until payload is started, or timeout expires. When timeout occurs, kill
535 /// the VM to prevent indefinite hangup and update the payload_state accordingly.
536 fn monitor_payload_hangup(&self, child: Arc<SharedChild>) {
537 debug!("Starting to monitor hangup for Microdroid({})", child.id());
Jiyong Parkcf975612024-04-07 13:27:10 +0900538 let (state, result) = self
Jiyong Parka4eebde2022-07-12 18:01:12 +0900539 .payload_state_updated
540 .wait_timeout_while(self.payload_state.lock().unwrap(), *BOOT_HANGUP_TIMEOUT, |s| {
541 *s < PayloadState::Started
542 })
543 .unwrap();
Jiyong Parkcf975612024-04-07 13:27:10 +0900544 drop(state); // we are not interested in state
Jiyong Parka4eebde2022-07-12 18:01:12 +0900545 let child_still_running = child.try_wait().ok() == Some(None);
546 if result.timed_out() && child_still_running {
547 error!(
548 "Microdroid({}) failed to start payload within {} secs timeout. Shutting down.",
549 child.id(),
550 BOOT_HANGUP_TIMEOUT.as_secs()
551 );
552 self.update_payload_state(PayloadState::Hangup).unwrap();
553 if let Err(e) = self.kill() {
554 error!("Error stopping timed-out VM with CID {}: {:?}", child.id(), e);
555 }
556 }
557 }
558
Seungjae Yoo6d265d92022-11-15 10:51:33 +0900559 fn monitor_vm_status(&self, child: Arc<SharedChild>) {
560 let pid = child.id();
561
562 loop {
563 {
564 // Check VM state
565 let vm_state = &*self.vm_state.lock().unwrap();
566 if let VmState::Dead = vm_state {
567 break;
568 }
569
570 let mut vm_metric = self.vm_metric.lock().unwrap();
571
572 // Get CPU Information
Jiyong Park85ea5382024-02-19 15:04:23 +0900573 match get_guest_time(pid) {
574 Ok(guest_time) => vm_metric.cpu_guest_time = Some(guest_time),
575 Err(e) => error!("Failed to get guest CPU time: {e:?}"),
Seungjae Yoo6d265d92022-11-15 10:51:33 +0900576 }
577
578 // Get Memory Information
Jiyong Park85ea5382024-02-19 15:04:23 +0900579 match get_rss(pid) {
580 Ok(rss) => {
581 vm_metric.rss = match &vm_metric.rss {
582 Some(x) => Some(Rss::extract_max(x, &rss)),
583 None => Some(rss),
584 }
Seungjae Yoo6d265d92022-11-15 10:51:33 +0900585 }
Jiyong Park85ea5382024-02-19 15:04:23 +0900586 Err(e) => error!("Failed to get guest RSS: {}", e),
Seungjae Yoo6d265d92022-11-15 10:51:33 +0900587 }
588 }
589
590 thread::sleep(Duration::from_secs(1));
591 }
592 }
593
Andrew Walbran6b650662021-09-07 13:13:23 +0000594 /// Returns the last reported state of the VM payload.
595 pub fn payload_state(&self) -> PayloadState {
596 *self.payload_state.lock().unwrap()
597 }
598
599 /// Updates the payload state to the given value, if it is a valid state transition.
600 pub fn update_payload_state(&self, new_state: PayloadState) -> Result<(), Error> {
601 let mut state_locked = self.payload_state.lock().unwrap();
602 // Only allow forward transitions, e.g. from starting to started or finished, not back in
603 // the other direction.
604 if new_state > *state_locked {
605 *state_locked = new_state;
Jiyong Parka4eebde2022-07-12 18:01:12 +0900606 self.payload_state_updated.notify_all();
Andrew Walbran6b650662021-09-07 13:13:23 +0000607 Ok(())
608 } else {
609 bail!("Invalid payload state transition from {:?} to {:?}", *state_locked, new_state)
610 }
611 }
612
Andrew Walbranf8d94112021-09-07 11:45:36 +0000613 /// Kills the crosvm instance, if it is running.
Inseob Kima446f802022-07-11 19:46:37 +0900614 pub fn kill(&self) -> Result<(), Error> {
David Brazdilf18e1162022-12-18 18:27:23 +0000615 let monitor_vm_exit_thread = {
616 let vm_state = &mut *self.vm_state.lock().unwrap();
617 if let VmState::Running { child, monitor_vm_exit_thread } = vm_state {
618 let id = child.id();
619 debug!("Killing crosvm({})", id);
620 // TODO: Talk to crosvm to shutdown cleanly.
621 child.kill().with_context(|| format!("Error killing crosvm({id}) instance"))?;
622 monitor_vm_exit_thread.take()
Inseob Kima446f802022-07-11 19:46:37 +0900623 } else {
David Brazdilf18e1162022-12-18 18:27:23 +0000624 bail!("VM is not running")
Andrew Walbranf8d94112021-09-07 11:45:36 +0000625 }
David Brazdilf18e1162022-12-18 18:27:23 +0000626 };
627
628 // Wait for monitor_vm_exit() to finish. Must release vm_state lock
629 // first, as monitor_vm_exit() takes it as well.
630 monitor_vm_exit_thread.map(JoinHandle::join);
631
David Brazdil5af55832023-03-08 13:08:56 +0000632 // Now that the VM has been killed, shut down the VirtualMachineService
633 // server to eagerly free up the server threads.
634 self.vm_context.vm_server.shutdown()?;
635
David Brazdilf18e1162022-12-18 18:27:23 +0000636 Ok(())
Andrew Walbrand6dce6f2021-03-05 16:39:08 +0000637 }
Jiyong Parke558ab12022-07-07 20:18:55 +0900638
Frederick Maylef375ef12024-10-21 16:07:21 -0700639 /// Returns current virtio-balloon size.
Keir Fraser48221ba2024-07-12 14:05:02 +0000640 pub fn get_memory_balloon(&self) -> Result<u64, Error> {
Frederick Maylef375ef12024-10-21 16:07:21 -0700641 let socket_path_cstring = path_to_cstring(&self.crosvm_control_socket_path);
642 let mut balloon_actual = 0u64;
643 // SAFETY: Pointers are valid for the lifetime of the call. Null `stats` is valid.
644 let success = unsafe {
645 crosvm_control::crosvm_client_balloon_stats(
646 socket_path_cstring.as_ptr(),
647 /* stats= */ std::ptr::null_mut(),
648 &mut balloon_actual,
649 )
650 };
651 if !success {
652 bail!("Error requesting balloon stats");
653 }
654 Ok(balloon_actual)
Keir Fraser48221ba2024-07-12 14:05:02 +0000655 }
656
Frederick Maylef375ef12024-10-21 16:07:21 -0700657 /// Inflates the virtio-balloon by `num_bytes` to reclaim guest memory. Called in response to
658 /// memory-trimming notifications.
Keir Fraser48221ba2024-07-12 14:05:02 +0000659 pub fn set_memory_balloon(&self, num_bytes: u64) -> Result<(), Error> {
Frederick Maylef375ef12024-10-21 16:07:21 -0700660 let socket_path_cstring = path_to_cstring(&self.crosvm_control_socket_path);
661 // SAFETY: Pointer is valid for the lifetime of the call.
662 let success = unsafe {
663 crosvm_control::crosvm_client_balloon_vms(socket_path_cstring.as_ptr(), num_bytes)
664 };
665 if !success {
666 bail!("Error sending balloon adjustment");
Keir Frasercdd4b112022-11-24 14:02:25 +0000667 }
668 Ok(())
669 }
670
Alan Stokes3e98d292022-12-14 15:18:22 +0000671 /// Checks if ramdump has been created. If so, send it to tombstoned.
Jiyong Parke558ab12022-07-07 20:18:55 +0900672 fn handle_ramdump(&self) -> Result<(), Error> {
673 let ramdump_path = self.temporary_directory.join("ramdump");
Jiyong Park933d2b22023-02-28 11:41:19 +0900674 if !ramdump_path.as_path().try_exists()? {
675 return Ok(());
676 }
Jiyong Parke558ab12022-07-07 20:18:55 +0900677 if std::fs::metadata(&ramdump_path)?.len() > 0 {
Jiyong Park1612b902022-08-22 14:47:39 +0900678 Self::send_ramdump_to_tombstoned(&ramdump_path)?;
Jiyong Parke558ab12022-07-07 20:18:55 +0900679 }
680 Ok(())
681 }
Jiyong Park1612b902022-08-22 14:47:39 +0900682
683 fn send_ramdump_to_tombstoned(ramdump_path: &Path) -> Result<(), Error> {
684 let mut input = File::open(ramdump_path)
Alan Stokes3e98d292022-12-14 15:18:22 +0000685 .context(format!("Failed to open ramdump {:?} for reading", ramdump_path))?;
Jiyong Park1612b902022-08-22 14:47:39 +0900686
687 let pid = std::process::id() as i32;
688 let conn = TombstonedConnection::connect(pid, DebuggerdDumpType::Tombstone)
689 .context("Failed to connect to tombstoned")?;
690 let mut output = conn
691 .text_output
692 .as_ref()
693 .ok_or_else(|| anyhow!("Could not get file to write the tombstones on"))?;
694
695 std::io::copy(&mut input, &mut output).context("Failed to send ramdump to tombstoned")?;
696 info!("Ramdump {:?} sent to tombstoned", ramdump_path);
697
698 conn.notify_completion()?;
699 Ok(())
700 }
Jiyong Park23b67392024-07-04 13:51:42 +0900701
Frederick Maylef375ef12024-10-21 16:07:21 -0700702 /// Suspends the VM's vCPUs.
Jiyong Park23b67392024-07-04 13:51:42 +0900703 pub fn suspend(&self) -> Result<(), Error> {
Frederick Maylef375ef12024-10-21 16:07:21 -0700704 let socket_path_cstring = path_to_cstring(&self.crosvm_control_socket_path);
705 // SAFETY: Pointer is valid for the lifetime of the call.
706 let success =
707 unsafe { crosvm_control::crosvm_client_suspend_vm(socket_path_cstring.as_ptr()) };
708 if !success {
709 bail!("Failed to suspend VM");
Jiyong Park23b67392024-07-04 13:51:42 +0900710 }
Frederick Maylef375ef12024-10-21 16:07:21 -0700711 Ok(())
Jiyong Park23b67392024-07-04 13:51:42 +0900712 }
713
Frederick Maylef375ef12024-10-21 16:07:21 -0700714 /// Resumes the VM's vCPUs.
Jiyong Park23b67392024-07-04 13:51:42 +0900715 pub fn resume(&self) -> Result<(), Error> {
Frederick Maylef375ef12024-10-21 16:07:21 -0700716 let socket_path_cstring = path_to_cstring(&self.crosvm_control_socket_path);
717 // SAFETY: Pointer is valid for the lifetime of the call.
718 let success =
719 unsafe { crosvm_control::crosvm_client_resume_vm(socket_path_cstring.as_ptr()) };
720 if !success {
721 bail!("Failed to resume VM");
Jiyong Park23b67392024-07-04 13:51:42 +0900722 }
Frederick Maylef375ef12024-10-21 16:07:21 -0700723 Ok(())
Jiyong Park23b67392024-07-04 13:51:42 +0900724 }
Andrew Walbrand6dce6f2021-03-05 16:39:08 +0000725}
726
Seungjae Yoo6d265d92022-11-15 10:51:33 +0900727impl Rss {
728 fn extract_max(x: &Rss, y: &Rss) -> Rss {
729 Rss { vm: max(x.vm, y.vm), crosvm: max(x.crosvm, y.crosvm) }
730 }
731}
732
David Daiaaecf652024-02-22 14:41:39 -0800733// Get Cpus_allowed mask
734fn check_if_all_cpus_allowed() -> Result<bool> {
735 let file = read_to_string("/proc/self/status")?;
736 let lines: Vec<_> = file.split('\n').collect();
737
738 for line in lines {
739 if line.contains("Cpus_allowed_list") {
740 let prop: Vec<_> = line.split_whitespace().collect();
741 if prop.len() != 2 {
742 return Ok(false);
743 }
744 let cpu_list: Vec<_> = prop[1].split('-').collect();
745 //Only contiguous Cpu list allowed
746 if cpu_list.len() != 2 {
747 return Ok(false);
748 }
749 if let Some(cpus) = get_num_cpus() {
750 let max_cpu = cpu_list[1].parse::<usize>()?;
751 if max_cpu == cpus - 1 {
752 return Ok(true);
753 } else {
754 return Ok(false);
755 }
756 }
757 }
758 }
759 Ok(false)
760}
761
Seungjae Yoo6d265d92022-11-15 10:51:33 +0900762// Get guest time from /proc/[crosvm pid]/stat
763fn get_guest_time(pid: u32) -> Result<i64> {
764 let file = read_to_string(format!("/proc/{}/stat", pid))?;
765 let data_list: Vec<_> = file.split_whitespace().collect();
766
767 // Information about guest_time is at 43th place of the file split with the whitespace.
768 // Example of /proc/[pid]/stat :
769 // 6603 (kworker/104:1H-kblockd) I 2 0 0 0 -1 69238880 0 0 0 0 0 88 0 0 0 -20 1 0 1845 0 0
770 // 18446744073709551615 0 0 0 0 0 0 0 2147483647 0 0 0 0 17 104 0 0 0 0 0 0 0 0 0 0 0 0 0
771 if data_list.len() < 43 {
772 bail!("Failed to parse command result for getting guest time : {}", file);
773 }
774
775 let guest_time_ticks = data_list[42].parse::<i64>()?;
Andrew Walbranb58d1b42023-07-07 13:54:49 +0100776 // SAFETY: It just returns an integer about CPU tick information.
Charisee96113f32023-01-26 09:00:42 +0000777 let ticks_per_sec = unsafe { sysconf(_SC_CLK_TCK) };
Seungjae Yoo6d265d92022-11-15 10:51:33 +0900778 Ok(guest_time_ticks * MILLIS_PER_SEC / ticks_per_sec)
779}
780
781// Get rss from /proc/[crosvm pid]/smaps
782fn get_rss(pid: u32) -> Result<Rss> {
783 let file = read_to_string(format!("/proc/{}/smaps", pid))?;
784 let lines: Vec<_> = file.split('\n').collect();
785
786 let mut rss_vm_total = 0i64;
787 let mut rss_crosvm_total = 0i64;
788 let mut is_vm = false;
789 for line in lines {
790 if line.contains("crosvm_guest") {
791 is_vm = true;
792 } else if line.contains("Rss:") {
793 let data_list: Vec<_> = line.split_whitespace().collect();
794 if data_list.len() < 2 {
795 bail!("Failed to parse command result for getting rss :\n{}", line);
796 }
797 let rss = data_list[1].parse::<i64>()?;
798
799 if is_vm {
800 rss_vm_total += rss;
801 is_vm = false;
802 }
803 rss_crosvm_total += rss;
804 }
805 }
806
807 Ok(Rss { vm: rss_vm_total, crosvm: rss_crosvm_total })
808}
809
Alan Stokes3ba10fd2022-10-06 15:46:51 +0100810fn death_reason(result: &Result<ExitStatus, io::Error>, mut failure_reason: &str) -> DeathReason {
Pierre-Clément Tosic0611e72023-09-15 11:41:07 +0100811 if let Some((reason, info)) = failure_reason.split_once('|') {
Alan Stokes3ba10fd2022-10-06 15:46:51 +0100812 // Separator indicates extra context information is present after the failure name.
Pierre-Clément Tosic0611e72023-09-15 11:41:07 +0100813 error!("Failure info: {info}");
814 failure_reason = reason;
Alan Stokes3ba10fd2022-10-06 15:46:51 +0100815 }
Andrew Walbranc92d35f2022-01-12 12:45:19 +0000816 if let Ok(status) = result {
Andrew Walbranb27681f2022-02-23 15:11:52 +0000817 match failure_reason {
818 "PVM_FIRMWARE_PUBLIC_KEY_MISMATCH" => {
819 return DeathReason::PVM_FIRMWARE_PUBLIC_KEY_MISMATCH
820 }
821 "PVM_FIRMWARE_INSTANCE_IMAGE_CHANGED" => {
822 return DeathReason::PVM_FIRMWARE_INSTANCE_IMAGE_CHANGED
823 }
Inseob Kim272f5722022-06-13 17:14:51 +0900824 "MICRODROID_FAILED_TO_CONNECT_TO_VIRTUALIZATION_SERVICE" => {
825 return DeathReason::MICRODROID_FAILED_TO_CONNECT_TO_VIRTUALIZATION_SERVICE
826 }
827 "MICRODROID_PAYLOAD_HAS_CHANGED" => return DeathReason::MICRODROID_PAYLOAD_HAS_CHANGED,
828 "MICRODROID_PAYLOAD_VERIFICATION_FAILED" => {
829 return DeathReason::MICRODROID_PAYLOAD_VERIFICATION_FAILED
830 }
831 "MICRODROID_INVALID_PAYLOAD_CONFIG" => {
832 return DeathReason::MICRODROID_INVALID_PAYLOAD_CONFIG
833 }
834 "MICRODROID_UNKNOWN_RUNTIME_ERROR" => {
835 return DeathReason::MICRODROID_UNKNOWN_RUNTIME_ERROR
836 }
Jiyong Parke6ed0f92022-06-22 00:13:00 +0900837 "HANGUP" => return DeathReason::HANGUP,
Andrew Walbranb27681f2022-02-23 15:11:52 +0000838 _ => {}
839 }
Andrew Walbranc92d35f2022-01-12 12:45:19 +0000840 match status.code() {
841 None => DeathReason::KILLED,
842 Some(0) => DeathReason::SHUTDOWN,
Alan Stokes7c459e82022-12-06 12:21:49 +0000843 Some(CROSVM_START_ERROR_STATUS) => DeathReason::START_FAILED,
Andrew Walbranc92d35f2022-01-12 12:45:19 +0000844 Some(CROSVM_REBOOT_STATUS) => DeathReason::REBOOT,
Andrew Walbrand15c5632022-02-03 13:38:31 +0000845 Some(CROSVM_CRASH_STATUS) => DeathReason::CRASH,
Sebastian Ene23167d82022-10-07 14:09:53 +0000846 Some(CROSVM_WATCHDOG_REBOOT_STATUS) => DeathReason::WATCHDOG_REBOOT,
Andrew Walbranc92d35f2022-01-12 12:45:19 +0000847 Some(_) => DeathReason::UNKNOWN,
848 }
849 } else {
850 DeathReason::INFRASTRUCTURE_ERROR
851 }
852}
853
Seungjae Yoo93430e82022-12-05 16:37:42 +0900854fn exit_signal(result: &Result<ExitStatus, io::Error>) -> Option<i32> {
855 match result {
856 Ok(status) => status.signal(),
857 Err(_) => None,
858 }
859}
860
Inseob Kim6ef80972023-07-20 17:23:36 +0900861const SYSFS_PLATFORM_DEVICES_PATH: &str = "/sys/devices/platform/";
862const VFIO_PLATFORM_DRIVER_PATH: &str = "/sys/bus/platform/drivers/vfio-platform";
863
Inseob Kim7307a892023-09-14 13:37:58 +0900864fn vfio_argument_for_platform_device(device: &VfioDevice) -> Result<String, Error> {
Inseob Kim6ef80972023-07-20 17:23:36 +0900865 // Check platform device exists
Jakob Vukalovicd42aa2c2023-11-09 16:04:00 +0000866 let path = Path::new(&device.getSysfsPath()?).canonicalize()?;
Inseob Kim6ef80972023-07-20 17:23:36 +0900867 if !path.starts_with(SYSFS_PLATFORM_DEVICES_PATH) {
868 bail!("{path:?} is not a platform device");
869 }
870
871 // Check platform device is bound to VFIO driver
872 let dev_driver_path = path.join("driver").canonicalize()?;
873 if dev_driver_path != Path::new(VFIO_PLATFORM_DRIVER_PATH) {
874 bail!("{path:?} is not bound to VFIO-platform driver");
875 }
876
877 if let Some(p) = path.to_str() {
Jakob Vukalovicd42aa2c2023-11-09 16:04:00 +0000878 Ok(format!("--vfio={p},iommu=pkvm-iommu,dt-symbol={0}", device.getDtboLabel()?))
Inseob Kim6ef80972023-07-20 17:23:36 +0900879 } else {
880 bail!("invalid path {path:?}");
881 }
882}
883
Andrew Walbrand3a84182021-09-07 14:48:52 +0000884/// Starts an instance of `crosvm` to manage a new VM.
Keir Fraser13a956a2022-07-14 14:20:46 +0000885fn run_vm(
886 config: CrosvmConfig,
Keir Frasercdd4b112022-11-24 14:02:25 +0000887 crosvm_control_socket_path: &Path,
Keir Fraser13a956a2022-07-14 14:20:46 +0000888 failure_pipe_write: File,
889) -> Result<SharedChild, Error> {
Andrew Walbrand3a84182021-09-07 14:48:52 +0000890 validate_config(&config)?;
Andrew Walbrand6dce6f2021-03-05 16:39:08 +0000891
892 let mut command = Command::new(CROSVM_PATH);
893 // TODO(qwandor): Remove --disable-sandbox.
Andrew Walbranc92d35f2022-01-12 12:45:19 +0000894 command
895 .arg("--extended-status")
Jiyong Park6c60fea2022-10-24 16:10:01 +0900896 // Configure the logger for the crosvm process to silence logs from the disk crate which
897 // don't provide much information to us (but do spamming us).
898 .arg("--log-level")
Frederick Mayle138f53c2023-01-24 18:55:26 -0800899 .arg("info,disk=warn")
Andrew Walbranc92d35f2022-01-12 12:45:19 +0000900 .arg("run")
901 .arg("--disable-sandbox")
902 .arg("--cid")
903 .arg(config.cid.to_string());
Andrew Walbran3a5a9212021-05-04 17:09:08 +0000904
Jeongik Cha75057452024-09-03 11:26:04 +0900905 if system_properties::read_bool("hypervisor.memory_reclaim.supported", false)?
906 && !config.no_balloon
907 {
Keir Fraserf25cb922022-11-23 14:26:00 +0000908 command.arg("--balloon-page-reporting");
909 } else {
910 command.arg("--no-balloon");
911 }
912
Elie Kheirallah29d72912024-08-07 20:17:26 +0000913 if !config.usb_config.controller {
914 command.arg("--no-usb");
915 }
916
Jiyong Parkae625b62024-06-13 21:47:52 +0900917 let mut memory_mib = config.memory_mib;
918
Andrew Walbranf8650422021-06-09 15:54:09 +0000919 if config.protected {
Jaewan Kim46422812022-11-25 10:59:39 +0900920 match system_properties::read(SYSPROP_CUSTOM_PVMFW_PATH)? {
921 Some(pvmfw_path) if !pvmfw_path.is_empty() => {
922 command.arg("--protected-vm-with-firmware").arg(pvmfw_path)
923 }
924 _ => command.arg("--protected-vm"),
925 };
Andrew Walbran0b5789f2022-02-04 13:57:57 +0000926
927 // 3 virtio-console devices + vsock = 4.
928 let virtio_pci_device_count = 4 + config.disks.len();
929 // crosvm virtio queue has 256 entries, so 2 MiB per device (2 pages per entry) should be
930 // enough.
Jiyong Park3115cdb2023-02-24 18:04:44 +0900931 let swiotlb_size_mib = 2 * virtio_pci_device_count as u32;
Andrew Walbran0b5789f2022-02-04 13:57:57 +0000932 command.arg("--swiotlb").arg(swiotlb_size_mib.to_string());
Frederick Mayle58baa302023-02-14 19:06:37 -0800933
Frederick Maylea24bbe02024-07-12 16:50:48 -0700934 // b/346770542 for consistent "usable" memory across protected and non-protected VMs.
Frederick Mayle68d06e72024-07-12 17:18:11 -0700935 memory_mib = memory_mib.saturating_add(swiotlb_size_mib);
Jiyong Parkae625b62024-06-13 21:47:52 +0900936
Frederick Mayle58baa302023-02-14 19:06:37 -0800937 // Workaround to keep crash_dump from trying to read protected guest memory.
938 // Context in b/238324526.
939 command.arg("--unmap-guest-memory-on-fork");
Jiyong Park3115cdb2023-02-24 18:04:44 +0900940
Jiyong Parked180932023-02-24 19:55:41 +0900941 if config.ramdump.is_some() {
Jaewan Kim3124ef02023-03-23 19:25:20 +0900942 // Protected VM needs to reserve memory for ramdump here. Note that we reserve more
Jiyong Parked180932023-02-24 19:55:41 +0900943 // memory for the restricted dma pool.
944 let ramdump_reserve = RAMDUMP_RESERVED_MIB + swiotlb_size_mib;
945 command.arg("--params").arg(format!("crashkernel={ramdump_reserve}M"));
946 }
Jaewan Kimd4dae652023-03-28 13:45:43 +0900947 } else if config.ramdump.is_some() {
948 command.arg("--params").arg(format!("crashkernel={RAMDUMP_RESERVED_MIB}M"));
949 }
Jaewan Kim61f86142023-03-28 15:12:52 +0900950 if config.debug_config.debug_level == DebugLevel::NONE
951 && config.debug_config.should_prepare_console_output()
952 {
Jaewan Kimd4dae652023-03-28 13:45:43 +0900953 // bootconfig.normal will be used, but we need log.
954 command.arg("--params").arg("printk.devkmsg=on");
955 command.arg("--params").arg("console=hvc0");
Andrew Walbranf8650422021-06-09 15:54:09 +0000956 }
957
Frederick Mayle68d06e72024-07-12 17:18:11 -0700958 command.arg("--mem").arg(memory_mib.to_string());
Andrew Walbranb15cd6e2021-07-05 16:38:07 +0000959
Jiyong Park032615f2022-01-10 13:55:34 +0900960 if let Some(cpus) = config.cpus {
961 command.arg("--cpus").arg(cpus.to_string());
962 }
963
David Brazdil7d1e5ec2023-02-06 17:56:29 +0000964 if config.host_cpu_topology {
David Daiaaecf652024-02-22 14:41:39 -0800965 if cfg!(virt_cpufreq) && check_if_all_cpus_allowed()? {
David Daie6f5bb02024-01-25 19:17:02 -0800966 command.arg("--host-cpu-topology");
David Daic97e82d2024-02-17 12:10:31 -0800967 cfg_if::cfg_if! {
968 if #[cfg(any(target_arch = "aarch64"))] {
969 command.arg("--virt-cpufreq");
970 }
971 }
David Daie6f5bb02024-01-25 19:17:02 -0800972 } else if let Some(cpus) = get_num_cpus() {
David Brazdil25e8c052023-02-17 12:53:01 +0000973 command.arg("--cpus").arg(cpus.to_string());
974 } else {
975 bail!("Could not determine the number of CPUs in the system");
976 }
David Brazdil7d1e5ec2023-02-06 17:56:29 +0000977 }
978
Nikita Ioffe5776f082023-02-10 21:38:26 +0000979 if let Some(gdb_port) = config.gdb_port {
980 command.arg("--gdb").arg(gdb_port.to_string());
981 }
982
Jiyong Parkfa91d702021-10-18 23:51:39 +0900983 // Keep track of what file descriptors should be mapped to the crosvm process.
Jiyong Park69001a22024-08-16 20:45:34 +0900984 let mut preserved_fds = config.indirect_files.into_iter().map(|f| f.into()).collect();
Jiyong Parkfa91d702021-10-18 23:51:39 +0900985
Elie Kheirallah5c807a22024-09-23 20:40:42 +0000986 if let Some(dump_dt_fd) = config.dump_dt_fd {
987 let dump_dt_fd = add_preserved_fd(&mut preserved_fds, dump_dt_fd);
988 command.arg("--dump-device-tree-blob").arg(dump_dt_fd);
989 }
990
Jiyong Park747d6362021-10-19 17:12:52 +0900991 // Setup the serial devices.
992 // 1. uart device: used as the output device by bootloaders and as early console by linux
Andrew Walbranb27681f2022-02-23 15:11:52 +0000993 // 2. uart device: used to report the reason for the VM failing.
994 // 3. virtio-console device: used as the console device where kmsg is redirected to
Jiyong Park4afe2012022-07-08 05:38:49 +0900995 // 4. virtio-console device: used as the ramdump output
Andrew Walbranb27681f2022-02-23 15:11:52 +0000996 // 5. virtio-console device: used as the logcat output
Jiyong Park747d6362021-10-19 17:12:52 +0900997 //
Jiyong Parkb8182bb2021-10-26 22:53:08 +0900998 // When [console|log]_fd is not specified, the devices are attached to sink, which means what's
999 // written there is discarded.
Jiyong Park69001a22024-08-16 20:45:34 +09001000 let console_out_arg = format_serial_out_arg(&mut preserved_fds, config.console_out_fd);
Jiyong Parke6fb1672023-06-26 16:45:55 +09001001 let console_in_arg = config
1002 .console_in_fd
Jiyong Parke6fb1672023-06-26 16:45:55 +09001003 .map(|fd| format!(",input={}", add_preserved_fd(&mut preserved_fds, fd)))
1004 .unwrap_or_default();
Jiyong Park69001a22024-08-16 20:45:34 +09001005 let log_arg = format_serial_out_arg(&mut preserved_fds, config.log_fd);
1006 let failure_serial_path = add_preserved_fd(&mut preserved_fds, failure_pipe_write);
1007 let ramdump_arg = format_serial_out_arg(&mut preserved_fds, config.ramdump);
Yi-Yo Chiang8dd32552024-05-22 19:38:16 +08001008 let console_input_device = config.console_input_device.as_deref().unwrap_or(CONSOLE_HVC0);
1009 match console_input_device {
1010 CONSOLE_HVC0 | CONSOLE_TTYS0 => {}
1011 _ => bail!("Unsupported serial device {console_input_device}"),
1012 };
Jiyong Parkfa91d702021-10-18 23:51:39 +09001013
Jiyong Park747d6362021-10-19 17:12:52 +09001014 // Warning: Adding more serial devices requires you to shift the PCI device ID of the boot
1015 // disks in bootconfig.x86_64. This is because x86 crosvm puts serial devices and the block
1016 // devices in the same PCI bus and serial devices comes before the block devices. Arm crosvm
1017 // doesn't have the issue.
Jiyong Parkfa91d702021-10-18 23:51:39 +09001018 // /dev/ttyS0
Yi-Yo Chiang8dd32552024-05-22 19:38:16 +08001019 command.arg(format!(
1020 "--serial={}{},hardware=serial,num=1",
1021 &console_out_arg,
1022 if console_input_device == CONSOLE_TTYS0 { &console_in_arg } else { "" }
1023 ));
Andrew Walbranb27681f2022-02-23 15:11:52 +00001024 // /dev/ttyS1
1025 command.arg(format!("--serial=type=file,path={},hardware=serial,num=2", &failure_serial_path));
Jiyong Parkfa91d702021-10-18 23:51:39 +09001026 // /dev/hvc0
Jiyong Parke6fb1672023-06-26 16:45:55 +09001027 command.arg(format!(
1028 "--serial={}{},hardware=virtio-console,num=1",
Yi-Yo Chiang8dd32552024-05-22 19:38:16 +08001029 &console_out_arg,
1030 if console_input_device == CONSOLE_HVC0 { &console_in_arg } else { "" }
Jiyong Parke6fb1672023-06-26 16:45:55 +09001031 ));
Jiyong Park4afe2012022-07-08 05:38:49 +09001032 // /dev/hvc1
1033 command.arg(format!("--serial={},hardware=virtio-console,num=2", &ramdump_arg));
Jiyong Parkae5a4ed2021-11-01 18:36:28 +09001034 // /dev/hvc2
1035 command.arg(format!("--serial={},hardware=virtio-console,num=3", &log_arg));
Andrew Walbran3a5a9212021-05-04 17:09:08 +00001036
Jiyong Park69001a22024-08-16 20:45:34 +09001037 if let Some(bootloader) = config.bootloader {
Andrew Walbran02b8ec02021-06-22 13:07:02 +00001038 command.arg("--bios").arg(add_preserved_fd(&mut preserved_fds, bootloader));
Andrew Walbrand6dce6f2021-03-05 16:39:08 +00001039 }
Andrew Walbran3a5a9212021-05-04 17:09:08 +00001040
Jiyong Park69001a22024-08-16 20:45:34 +09001041 if let Some(initrd) = config.initrd {
Andrew Walbran02b8ec02021-06-22 13:07:02 +00001042 command.arg("--initrd").arg(add_preserved_fd(&mut preserved_fds, initrd));
Andrew Walbrand6dce6f2021-03-05 16:39:08 +00001043 }
Andrew Walbran3a5a9212021-05-04 17:09:08 +00001044
Andrew Walbrand6dce6f2021-03-05 16:39:08 +00001045 if let Some(params) = &config.params {
1046 command.arg("--params").arg(params);
1047 }
Andrew Walbran3a5a9212021-05-04 17:09:08 +00001048
Jiyong Park69001a22024-08-16 20:45:34 +09001049 for disk in config.disks {
Frederick Mayleeae6d2d2024-09-12 11:05:36 -07001050 // Disk file locking is disabled because of missing SELinux policies.
Frederick Mayle1ff7b842024-07-12 16:20:09 -07001051 command.arg("--block").arg(format!(
Frederick Mayleeae6d2d2024-09-12 11:05:36 -07001052 "path={},ro={},lock=false",
Jiyong Park69001a22024-08-16 20:45:34 +09001053 add_preserved_fd(&mut preserved_fds, disk.image),
Frederick Mayle1ff7b842024-07-12 16:20:09 -07001054 !disk.writable,
1055 ));
Andrew Walbrand6dce6f2021-03-05 16:39:08 +00001056 }
Andrew Walbran3a5a9212021-05-04 17:09:08 +00001057
Jiyong Park69001a22024-08-16 20:45:34 +09001058 if let Some(kernel) = config.kernel {
Andrew Walbran02b8ec02021-06-22 13:07:02 +00001059 command.arg(add_preserved_fd(&mut preserved_fds, kernel));
Andrew Walbrand6dce6f2021-03-05 16:39:08 +00001060 }
Andrew Walbran3a5a9212021-05-04 17:09:08 +00001061
Frederick Maylee0ed2352024-09-12 16:16:00 -07001062 let control_sock = create_crosvm_control_listener(crosvm_control_socket_path)
1063 .context("failed to create control listener")?;
Jiyong Parkcd0f74b2024-08-20 11:06:52 +09001064 command.arg("--socket").arg(add_preserved_fd(&mut preserved_fds, control_sock));
Keir Fraser13a956a2022-07-14 14:20:46 +00001065
Jiyong Park69001a22024-08-16 20:45:34 +09001066 if let Some(dt_overlay) = config.device_tree_overlay {
Shikha Panwar55e10ec2024-02-13 12:53:49 +00001067 command.arg("--device-tree-overlay").arg(add_preserved_fd(&mut preserved_fds, dt_overlay));
Seungjae Yoo4dd38182023-11-09 15:05:45 +09001068 }
Jeongik Cha798401c2024-04-11 21:54:49 +09001069
1070 if cfg!(paravirtualized_devices) {
Jason Macnakea7b0fa2024-05-08 17:02:44 -07001071 if let Some(gpu_config) = &config.gpu_config {
1072 let mut gpu_args = Vec::new();
1073 if let Some(backend) = &gpu_config.backend {
1074 gpu_args.push(format!("backend={}", backend));
1075 }
1076 if let Some(context_types) = &gpu_config.context_types {
1077 gpu_args.push(format!("context-types={}", context_types.join(":")));
1078 }
1079 if let Some(pci_address) = &gpu_config.pci_address {
1080 gpu_args.push(format!("pci-address={}", pci_address));
1081 }
1082 if let Some(renderer_features) = &gpu_config.renderer_features {
1083 gpu_args.push(format!("renderer-features={}", renderer_features));
1084 }
1085 if gpu_config.renderer_use_egl.unwrap_or(false) {
1086 gpu_args.push("egl=true".to_string());
1087 }
1088 if gpu_config.renderer_use_gles.unwrap_or(false) {
1089 gpu_args.push("gles=true".to_string());
1090 }
1091 if gpu_config.renderer_use_glx.unwrap_or(false) {
1092 gpu_args.push("glx=true".to_string());
1093 }
1094 if gpu_config.renderer_use_surfaceless.unwrap_or(false) {
1095 gpu_args.push("surfaceless=true".to_string());
1096 }
1097 if gpu_config.renderer_use_vulkan.unwrap_or(false) {
1098 gpu_args.push("vulkan=true".to_string());
1099 }
1100 command.arg(format!("--gpu={}", gpu_args.join(",")));
1101 }
Jeongik Cha798401c2024-04-11 21:54:49 +09001102 if let Some(display_config) = &config.display_config {
Jason Macnakea7b0fa2024-05-08 17:02:44 -07001103 command
1104 .arg(format!(
1105 "--gpu-display=mode=windowed[{},{}],dpi=[{},{}],refresh-rate={}",
1106 display_config.width,
1107 display_config.height,
1108 display_config.horizontal_dpi,
1109 display_config.vertical_dpi,
1110 display_config.refresh_rate
1111 ))
1112 .arg(format!("--android-display-service={}", config.name));
Jeongik Cha798401c2024-04-11 21:54:49 +09001113 }
Jeongik Cha1df1c032024-04-03 16:03:12 +09001114 }
Seungjae Yooec3bc522023-11-09 10:14:30 +09001115
Seungjae Yoo986d7a42024-05-28 14:06:58 +09001116 if cfg!(network) {
Jiyong Park69001a22024-08-16 20:45:34 +09001117 if let Some(tap) = config.tap {
Andreas Huber487810d2024-08-22 07:43:01 -07001118 add_preserved_fd(&mut preserved_fds, tap);
1119 let tap_fd = preserved_fds.last().unwrap().as_raw_fd();
1120 command.arg("--net").arg(format!("tap-fd={tap_fd}"));
Seungjae Yoo986d7a42024-05-28 14:06:58 +09001121 }
1122 }
1123
Jeongik Cha6be4f8b2024-04-16 01:13:13 +09001124 if cfg!(paravirtualized_devices) {
Jiyong Park69001a22024-08-16 20:45:34 +09001125 for input_device_option in config.input_device_options.into_iter() {
Jeongik Chac181be72024-03-27 00:18:30 +09001126 command.arg("--input");
1127 command.arg(match input_device_option {
1128 InputDeviceOption::EvDev(file) => {
1129 format!("evdev[path={}]", add_preserved_fd(&mut preserved_fds, file))
1130 }
Jeongik Cha985b6402024-04-15 18:13:00 +09001131 InputDeviceOption::Keyboard(file) => {
1132 format!("keyboard[path={}]", add_preserved_fd(&mut preserved_fds, file))
1133 }
Jeongik Cha343894e2024-05-17 20:39:31 +09001134 InputDeviceOption::Mouse(file) => {
1135 format!("mouse[path={}]", add_preserved_fd(&mut preserved_fds, file))
1136 }
Jeongik Chac181be72024-03-27 00:18:30 +09001137 InputDeviceOption::SingleTouch { file, width, height, name } => format!(
1138 "single-touch[path={},width={},height={}{}]",
1139 add_preserved_fd(&mut preserved_fds, file),
1140 width,
1141 height,
1142 name.as_ref().map_or("".into(), |n| format!(",name={}", n))
1143 ),
Jiyong Park6c1b9f02024-07-05 16:21:10 +09001144 InputDeviceOption::Switches(file) => {
1145 format!("switches[path={}]", add_preserved_fd(&mut preserved_fds, file))
1146 }
Jeongik Cha112a2682024-07-09 12:28:45 +09001147 InputDeviceOption::MultiTouchTrackpad { file, width, height, name } => format!(
1148 "multi-touch-trackpad[path={},width={},height={}{}]",
1149 add_preserved_fd(&mut preserved_fds, file),
1150 width,
1151 height,
1152 name.as_ref().map_or("".into(), |n| format!(",name={}", n))
1153 ),
Jeongik Cha10494912024-07-16 11:19:50 +09001154 InputDeviceOption::MultiTouch { file, width, height, name } => format!(
1155 "multi-touch[path={},width={},height={}{}]",
1156 add_preserved_fd(&mut preserved_fds, file),
1157 width,
1158 height,
1159 name.as_ref().map_or("".into(), |n| format!(",name={}", n))
1160 ),
Jeongik Chac181be72024-03-27 00:18:30 +09001161 });
1162 }
1163 }
Vincent Donnefort538a2c62024-03-20 16:01:10 +00001164
1165 if config.hugepages {
1166 command.arg("--hugepages");
1167 }
1168
David Dai23cff712024-06-13 19:23:45 +00001169 if config.boost_uclamp {
1170 command.arg("--boost-uclamp");
1171 }
1172
Jiyong Park69001a22024-08-16 20:45:34 +09001173 if !config.vfio_devices.is_empty() {
1174 if let Some(dtbo) = config.dtbo {
1175 command.arg(format!(
1176 "--device-tree-overlay={},filter",
1177 add_preserved_fd(&mut preserved_fds, dtbo)
1178 ));
1179 } else {
1180 bail!("VFIO devices assigned but no DTBO available");
1181 }
1182 };
1183 for device in config.vfio_devices {
1184 command.arg(vfio_argument_for_platform_device(&device)?);
1185 }
Inseob Kim6ef80972023-07-20 17:23:36 +09001186
Andrew Walbran02b8ec02021-06-22 13:07:02 +00001187 debug!("Preserving FDs {:?}", preserved_fds);
1188 command.preserved_fds(preserved_fds);
Andrew Walbran3a5a9212021-05-04 17:09:08 +00001189
Mu-Le Lee134180a2024-04-16 15:13:07 +08001190 if cfg!(paravirtualized_devices) {
Mu-Le Leeb2d5a312024-06-05 10:52:47 +08001191 if let Some(audio_config) = &config.audio_config {
1192 command.arg("--virtio-snd").arg(format!(
1193 "backend=aaudio,num_input_devices={},num_output_devices={}",
1194 if audio_config.use_microphone { 1 } else { 0 },
1195 if audio_config.use_speaker { 1 } else { 0 }
1196 ));
Mu-Le Lee134180a2024-04-16 15:13:07 +08001197 }
1198 }
1199
Jiyong Park2d736562022-10-24 22:40:12 +09001200 print_crosvm_args(&command);
1201
Andrew Walbran3a5a9212021-05-04 17:09:08 +00001202 let result = SharedChild::spawn(&mut command)?;
Jooyung Hanbfe086f2021-10-28 10:15:45 +09001203 debug!("Spawned crosvm({}).", result.id());
Andrew Walbran3a5a9212021-05-04 17:09:08 +00001204 Ok(result)
1205}
1206
1207/// Ensure that the configuration has a valid combination of fields set, or return an error if not.
Andrew Walbranf5fbb7d2021-05-12 17:15:48 +00001208fn validate_config(config: &CrosvmConfig) -> Result<(), Error> {
Andrew Walbran3a5a9212021-05-04 17:09:08 +00001209 if config.bootloader.is_none() && config.kernel.is_none() {
1210 bail!("VM must have either a bootloader or a kernel image.");
1211 }
1212 if config.bootloader.is_some() && (config.kernel.is_some() || config.initrd.is_some()) {
1213 bail!("Can't have both bootloader and kernel/initrd image.");
1214 }
Jiyong Parkdcf17412022-02-08 15:07:23 +09001215 let version = Version::parse(CROSVM_PLATFORM_VERSION).unwrap();
1216 if !config.platform_version.matches(&version) {
1217 bail!(
1218 "Incompatible platform version. The config is compatible with platform version(s) \
1219 {}, but the actual platform version is {}",
1220 config.platform_version,
1221 version
1222 );
1223 }
1224
Andrew Walbran3a5a9212021-05-04 17:09:08 +00001225 Ok(())
1226}
1227
Jiyong Park2d736562022-10-24 22:40:12 +09001228/// Print arguments of the crosvm command. In doing so, /proc/self/fd/XX is annotated with the
1229/// actual file path if the FD is backed by a regular file. If not, the /proc path is printed
1230/// unmodified.
1231fn print_crosvm_args(command: &Command) {
1232 let re = Regex::new(r"/proc/self/fd/[\d]+").unwrap();
1233 info!(
1234 "Running crosvm with args: {:?}",
1235 command
1236 .get_args()
1237 .map(|s| s.to_string_lossy())
1238 .map(|s| {
1239 re.replace_all(&s, |caps: &Captures| {
1240 let path = &caps[0];
1241 if let Ok(realpath) = std::fs::canonicalize(path) {
1242 format!("{} ({})", path, realpath.to_string_lossy())
1243 } else {
1244 path.to_owned()
1245 }
1246 })
1247 .into_owned()
1248 })
1249 .collect::<Vec<_>>()
1250 );
1251}
1252
Andrew Walbran02b8ec02021-06-22 13:07:02 +00001253/// Adds the file descriptor for `file` to `preserved_fds`, and returns a string of the form
1254/// "/proc/self/fd/N" where N is the file descriptor.
Jiyong Park69001a22024-08-16 20:45:34 +09001255fn add_preserved_fd<F: Into<OwnedFd>>(preserved_fds: &mut Vec<OwnedFd>, file: F) -> String {
1256 let fd = file.into();
1257 let raw_fd = fd.as_raw_fd();
Andrew Walbran02b8ec02021-06-22 13:07:02 +00001258 preserved_fds.push(fd);
Jiyong Park69001a22024-08-16 20:45:34 +09001259 format!("/proc/self/fd/{}", raw_fd)
Andrew Walbrand6dce6f2021-03-05 16:39:08 +00001260}
Andrew Walbranb27681f2022-02-23 15:11:52 +00001261
1262/// Adds the file descriptor for `file` (if any) to `preserved_fds`, and returns the appropriate
1263/// string for a crosvm `--serial` flag. If `file` is none, creates a dummy sink device.
Jiyong Park69001a22024-08-16 20:45:34 +09001264fn format_serial_out_arg(preserved_fds: &mut Vec<OwnedFd>, file: Option<File>) -> String {
Andrew Walbranb27681f2022-02-23 15:11:52 +00001265 if let Some(file) = file {
1266 format!("type=file,path={}", add_preserved_fd(preserved_fds, file))
1267 } else {
1268 "type=sink".to_string()
1269 }
1270}
1271
1272/// Creates a new pipe with the `O_CLOEXEC` flag set, and returns the read side and write side.
1273fn create_pipe() -> Result<(File, File), Error> {
Frederick Maylefbbcfcd2024-04-08 16:31:54 -07001274 let (read_fd, write_fd) = pipe2(OFlag::O_CLOEXEC)?;
1275 Ok((read_fd.into(), write_fd.into()))
Andrew Walbranb27681f2022-02-23 15:11:52 +00001276}
Frederick Maylee0ed2352024-09-12 16:16:00 -07001277
1278/// Creates and binds a unix seqpacket listening socket to be passed as crosvm's `--socket`
1279/// argument. See `UnixSeqpacketListener::bind` in crosvm's code for reference.
1280fn create_crosvm_control_listener(crosvm_control_socket_path: &Path) -> Result<OwnedFd> {
1281 use nix::sys::socket;
1282 let fd = socket::socket(
1283 socket::AddressFamily::Unix,
1284 socket::SockType::SeqPacket,
1285 socket::SockFlag::empty(),
1286 None,
1287 )
1288 .context("socket failed")?;
1289 socket::bind(fd.as_raw_fd(), &socket::UnixAddr::new(crosvm_control_socket_path)?)
1290 .context("bind failed")?;
1291 // The exact backlog size isn't imporant. crosvm uses 128 internally. We use 127 here
1292 // because of a `nix` bug.
1293 socket::listen(&fd, socket::Backlog::new(127).unwrap()).context("listen failed")?;
1294 Ok(fd)
1295}
Frederick Maylef375ef12024-10-21 16:07:21 -07001296
1297fn path_to_cstring(path: &Path) -> CString {
1298 if let Some(s) = path.to_str() {
1299 if let Ok(s) = CString::new(s) {
1300 return s;
1301 }
1302 }
1303 // The path contains invalid utf8 or a null, which should never happen.
1304 panic!("bad path: {path:?}");
1305}