blob: ed45a5110f5a21191e798bc54a74e92048105fc5 [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;
Jiyong Park08eee7b2022-11-11 15:01:26 +090032use std::fmt;
David Brazdil4b4c5102022-12-19 22:56:20 +000033use std::fs::{read_to_string, File};
Andrew Walbranb27681f2022-02-23 15:11:52 +000034use std::io::{self, Read};
Andrew Walbranf8d94112021-09-07 11:45:36 +000035use std::mem;
Nikita Ioffe5776f082023-02-10 21:38:26 +000036use std::num::{NonZeroU16, NonZeroU32};
Jiyong Park69001a22024-08-16 20:45:34 +090037use std::os::unix::io::{AsRawFd, OwnedFd};
Seungjae Yoo93430e82022-12-05 16:37:42 +090038use std::os::unix::process::ExitStatusExt;
Jiyong Park1612b902022-08-22 14:47:39 +090039use std::path::{Path, PathBuf};
Andrew Walbranc92d35f2022-01-12 12:45:19 +000040use std::process::{Command, ExitStatus};
Andrew Walbran9c03a3a2024-09-03 12:12:59 +010041use std::sync::{Arc, Condvar, Mutex, LazyLock};
Seungjae Yoo2e7beea2022-08-24 16:09:12 +090042use std::time::{Duration, SystemTime};
David Brazdilf18e1162022-12-18 18:27:23 +000043use std::thread::{self, JoinHandle};
David Brazdil49f96f52022-12-16 21:29:13 +000044use android_system_virtualizationcommon::aidl::android::system::virtualizationcommon::DeathReason::DeathReason;
Jaewan Kim84b91212023-02-28 00:11:57 +090045use android_system_virtualizationservice::aidl::android::system::virtualizationservice::{
Jeongik Cha1df1c032024-04-03 16:03:12 +090046 VirtualMachineAppConfig::DebugLevel::DebugLevel,
Mu-Le Leeb2d5a312024-06-05 10:52:47 +080047 AudioConfig::AudioConfig as AudioConfigParcelable,
Jeongik Cha1df1c032024-04-03 16:03:12 +090048 DisplayConfig::DisplayConfig as DisplayConfigParcelable,
Jason Macnakea7b0fa2024-05-08 17:02:44 -070049 GpuConfig::GpuConfig as GpuConfigParcelable,
Elie Kheirallah29d72912024-08-07 20:17:26 +000050 UsbConfig::UsbConfig as UsbConfigParcelable,
Jaewan Kim84b91212023-02-28 00:11:57 +090051};
David Brazdil528e0472022-10-10 15:06:02 +010052use android_system_virtualizationservice_internal::aidl::android::system::virtualizationservice_internal::IGlobalVmContext::IGlobalVmContext;
Jakob Vukalovicd42aa2c2023-11-09 16:04:00 +000053use android_system_virtualizationservice_internal::aidl::android::system::virtualizationservice_internal::IBoundDevice::IBoundDevice;
Alan Stokes0e82b502022-08-08 14:44:48 +010054use binder::Strong;
Inseob Kimc7d28c72021-10-25 14:28:10 +000055use android_system_virtualmachineservice::aidl::android::system::virtualmachineservice::IVirtualMachineService::IVirtualMachineService;
Jiyong Park1612b902022-08-22 14:47:39 +090056use tombstoned_client::{TombstonedConnection, DebuggerdDumpType};
David Brazdil73988ea2022-11-11 15:10:32 +000057use rpcbinder::RpcServer;
Andrew Walbrand6dce6f2021-03-05 16:39:08 +000058
Keir Fraser13a956a2022-07-14 14:20:46 +000059/// external/crosvm
Keir Frasercdd4b112022-11-24 14:02:25 +000060use vm_control::{BalloonControlCommand, VmRequest, VmResponse};
Keir Fraser13a956a2022-07-14 14:20:46 +000061
Andrew Walbrand6dce6f2021-03-05 16:39:08 +000062const CROSVM_PATH: &str = "/apex/com.android.virt/bin/crosvm";
63
Jiyong Parkdcf17412022-02-08 15:07:23 +090064/// Version of the platform that crosvm currently implements. The format follows SemVer. This
65/// should be updated when there is a platform change in the crosvm side. Having this value here is
66/// fine because virtualizationservice and crosvm are supposed to be updated together in the virt
67/// APEX.
68const CROSVM_PLATFORM_VERSION: &str = "1.0.0";
69
Andrew Walbrand15c5632022-02-03 13:38:31 +000070/// The exit status which crosvm returns when it has an error starting a VM.
Alan Stokes7c459e82022-12-06 12:21:49 +000071const CROSVM_START_ERROR_STATUS: i32 = 1;
Andrew Walbranc92d35f2022-01-12 12:45:19 +000072/// The exit status which crosvm returns when a VM requests a reboot.
73const CROSVM_REBOOT_STATUS: i32 = 32;
Andrew Walbrand15c5632022-02-03 13:38:31 +000074/// The exit status which crosvm returns when it crashes due to an error.
75const CROSVM_CRASH_STATUS: i32 = 33;
Sebastian Ene23167d82022-10-07 14:09:53 +000076/// The exit status which crosvm returns when vcpu is stalled.
77const CROSVM_WATCHDOG_REBOOT_STATUS: i32 = 36;
Jiyong Park3115cdb2023-02-24 18:04:44 +090078/// The size of memory (in MiB) reserved for ramdump
79const RAMDUMP_RESERVED_MIB: u32 = 17;
Andrew Walbranc92d35f2022-01-12 12:45:19 +000080
Seungjae Yoo6d265d92022-11-15 10:51:33 +090081const MILLIS_PER_SEC: i64 = 1000;
82
Jaewan Kim46422812022-11-25 10:59:39 +090083const SYSPROP_CUSTOM_PVMFW_PATH: &str = "hypervisor.pvmfw.path";
84
Yi-Yo Chiang8dd32552024-05-22 19:38:16 +080085/// Serial device for VM console input.
86/// Hypervisor (virtio-console)
87const CONSOLE_HVC0: &str = "hvc0";
88/// Serial (emulated uart)
89const CONSOLE_TTYS0: &str = "ttyS0";
90
Andrew Walbran9c03a3a2024-09-03 12:12:59 +010091/// If the VM doesn't move to the Started state within this amount time, a hang-up error is
92/// triggered.
93static BOOT_HANGUP_TIMEOUT: LazyLock<Duration> = LazyLock::new(|| {
94 if nested_virt::is_nested_virtualization().unwrap() {
Jiyong Parke6ed0f92022-06-22 00:13:00 +090095 // Nested virtualization is slow, so we need a longer timeout.
Inseob Kime557cec2023-01-26 22:49:58 +090096 Duration::from_secs(300)
Jiyong Parke6ed0f92022-06-22 00:13:00 +090097 } else {
Inseob Kime557cec2023-01-26 22:49:58 +090098 Duration::from_secs(30)
Andrew Walbran9c03a3a2024-09-03 12:12:59 +010099 }
100});
Jiyong Parke6ed0f92022-06-22 00:13:00 +0900101
Andrew Walbranf5fbb7d2021-05-12 17:15:48 +0000102/// Configuration for a VM to run with crosvm.
103#[derive(Debug)]
Andrew Walbrand3a84182021-09-07 14:48:52 +0000104pub struct CrosvmConfig {
Andrew Walbranf5fbb7d2021-05-12 17:15:48 +0000105 pub cid: Cid,
Seungjae Yoo62085c02022-08-12 04:44:52 +0000106 pub name: String,
Andrew Walbrand3a84182021-09-07 14:48:52 +0000107 pub bootloader: Option<File>,
108 pub kernel: Option<File>,
109 pub initrd: Option<File>,
Andrew Walbranf5fbb7d2021-05-12 17:15:48 +0000110 pub disks: Vec<DiskFile>,
111 pub params: Option<String>,
Andrew Walbranf8650422021-06-09 15:54:09 +0000112 pub protected: bool,
Jaewan Kim61f86142023-03-28 15:12:52 +0900113 pub debug_config: DebugConfig,
Frederick Mayle68d06e72024-07-12 17:18:11 -0700114 pub memory_mib: NonZeroU32,
Jiyong Park032615f2022-01-10 13:55:34 +0900115 pub cpus: Option<NonZeroU32>,
David Brazdil7d1e5ec2023-02-06 17:56:29 +0000116 pub host_cpu_topology: bool,
Jiyong Parke6fb1672023-06-26 16:45:55 +0900117 pub console_out_fd: Option<File>,
118 pub console_in_fd: Option<File>,
Andrew Walbrand3a84182021-09-07 14:48:52 +0000119 pub log_fd: Option<File>,
Jiyong Parke558ab12022-07-07 20:18:55 +0900120 pub ramdump: Option<File>,
Andrew Walbrand3a84182021-09-07 14:48:52 +0000121 pub indirect_files: Vec<File>,
Jiyong Parkdcf17412022-02-08 15:07:23 +0900122 pub platform_version: VersionReq,
Jiyong Parke6ed0f92022-06-22 00:13:00 +0900123 pub detect_hangup: bool,
Nikita Ioffe5776f082023-02-10 21:38:26 +0000124 pub gdb_port: Option<NonZeroU16>,
Inseob Kim7307a892023-09-14 13:37:58 +0900125 pub vfio_devices: Vec<VfioDevice>,
David Brazdil2dfefd12023-11-17 14:07:36 +0000126 pub dtbo: Option<File>,
Shikha Panwar55e10ec2024-02-13 12:53:49 +0000127 pub device_tree_overlay: Option<File>,
Jeongik Cha1df1c032024-04-03 16:03:12 +0900128 pub display_config: Option<DisplayConfig>,
Jeongik Chac181be72024-03-27 00:18:30 +0900129 pub input_device_options: Vec<InputDeviceOption>,
Vincent Donnefort538a2c62024-03-20 16:01:10 +0000130 pub hugepages: bool,
Seungjae Yoo986d7a42024-05-28 14:06:58 +0900131 pub tap: Option<File>,
Yi-Yo Chiang8dd32552024-05-22 19:38:16 +0800132 pub console_input_device: Option<String>,
David Dai23cff712024-06-13 19:23:45 +0000133 pub boost_uclamp: bool,
Jason Macnakea7b0fa2024-05-08 17:02:44 -0700134 pub gpu_config: Option<GpuConfig>,
Mu-Le Leeb2d5a312024-06-05 10:52:47 +0800135 pub audio_config: Option<AudioConfig>,
Jeongik Cha75057452024-09-03 11:26:04 +0900136 pub no_balloon: bool,
Elie Kheirallah29d72912024-08-07 20:17:26 +0000137 pub usb_config: UsbConfig,
Elie Kheirallah5c807a22024-09-23 20:40:42 +0000138 pub dump_dt_fd: Option<File>,
Mu-Le Leeb2d5a312024-06-05 10:52:47 +0800139}
140
141#[derive(Debug)]
142pub struct AudioConfig {
143 pub use_microphone: bool,
144 pub use_speaker: bool,
145}
146
147impl AudioConfig {
148 pub fn new(raw_config: &AudioConfigParcelable) -> Self {
149 AudioConfig { use_microphone: raw_config.useMicrophone, use_speaker: raw_config.useSpeaker }
150 }
Jeongik Cha1df1c032024-04-03 16:03:12 +0900151}
152
153#[derive(Debug)]
Elie Kheirallah29d72912024-08-07 20:17:26 +0000154pub struct UsbConfig {
155 pub controller: bool,
156}
157
158impl UsbConfig {
159 pub fn new(raw_config: &UsbConfigParcelable) -> Result<UsbConfig> {
160 Ok(UsbConfig { controller: raw_config.controller })
161 }
162}
163
164#[derive(Debug)]
Jeongik Cha1df1c032024-04-03 16:03:12 +0900165pub struct DisplayConfig {
166 pub width: NonZeroU32,
167 pub height: NonZeroU32,
168 pub horizontal_dpi: NonZeroU32,
169 pub vertical_dpi: NonZeroU32,
170 pub refresh_rate: NonZeroU32,
171}
172
173impl DisplayConfig {
174 pub fn new(raw_config: &DisplayConfigParcelable) -> Result<DisplayConfig> {
175 let width = try_into_non_zero_u32(raw_config.width)?;
176 let height = try_into_non_zero_u32(raw_config.height)?;
177 let horizontal_dpi = try_into_non_zero_u32(raw_config.horizontalDpi)?;
178 let vertical_dpi = try_into_non_zero_u32(raw_config.verticalDpi)?;
179 let refresh_rate = try_into_non_zero_u32(raw_config.refreshRate)?;
180 Ok(DisplayConfig { width, height, horizontal_dpi, vertical_dpi, refresh_rate })
181 }
182}
183
Jason Macnakea7b0fa2024-05-08 17:02:44 -0700184#[derive(Debug)]
185pub struct GpuConfig {
186 pub backend: Option<String>,
187 pub context_types: Option<Vec<String>>,
188 pub pci_address: Option<String>,
189 pub renderer_features: Option<String>,
190 pub renderer_use_egl: Option<bool>,
191 pub renderer_use_gles: Option<bool>,
192 pub renderer_use_glx: Option<bool>,
193 pub renderer_use_surfaceless: Option<bool>,
194 pub renderer_use_vulkan: Option<bool>,
195}
196
197impl GpuConfig {
198 pub fn new(raw_config: &GpuConfigParcelable) -> Result<GpuConfig> {
199 Ok(GpuConfig {
200 backend: raw_config.backend.clone(),
201 context_types: raw_config.contextTypes.clone().map(|context_types| {
202 context_types.iter().filter_map(|context_type| context_type.clone()).collect()
203 }),
204 pci_address: raw_config.pciAddress.clone(),
205 renderer_features: raw_config.rendererFeatures.clone(),
206 renderer_use_egl: Some(raw_config.rendererUseEgl),
207 renderer_use_gles: Some(raw_config.rendererUseGles),
208 renderer_use_glx: Some(raw_config.rendererUseGlx),
209 renderer_use_surfaceless: Some(raw_config.rendererUseSurfaceless),
210 renderer_use_vulkan: Some(raw_config.rendererUseVulkan),
211 })
212 }
213}
214
Jeongik Cha1df1c032024-04-03 16:03:12 +0900215fn try_into_non_zero_u32(value: i32) -> Result<NonZeroU32> {
216 let u32_value = value.try_into()?;
217 NonZeroU32::new(u32_value).ok_or(anyhow!("value should be greater than 0"))
Andrew Walbranf5fbb7d2021-05-12 17:15:48 +0000218}
219
220/// A disk image to pass to crosvm for a VM.
221#[derive(Debug)]
222pub struct DiskFile {
223 pub image: File,
224 pub writable: bool,
225}
226
Jeongik Chac181be72024-03-27 00:18:30 +0900227/// virtio-input device configuration from `external/crosvm/src/crosvm/config.rs`
228#[derive(Debug)]
229#[allow(dead_code)]
230pub enum InputDeviceOption {
231 EvDev(File),
232 SingleTouch { file: File, width: u32, height: u32, name: Option<String> },
Jeongik Cha985b6402024-04-15 18:13:00 +0900233 Keyboard(File),
Jeongik Cha343894e2024-05-17 20:39:31 +0900234 Mouse(File),
Jiyong Park6c1b9f02024-07-05 16:21:10 +0900235 Switches(File),
Jeongik Cha112a2682024-07-09 12:28:45 +0900236 MultiTouchTrackpad { file: File, width: u32, height: u32, name: Option<String> },
Jeongik Cha10494912024-07-16 11:19:50 +0900237 MultiTouch { file: File, width: u32, height: u32, name: Option<String> },
Jeongik Chac181be72024-03-27 00:18:30 +0900238}
239
Jakob Vukalovicd42aa2c2023-11-09 16:04:00 +0000240type VfioDevice = Strong<dyn IBoundDevice>;
Inseob Kim7307a892023-09-14 13:37:58 +0900241
Andrew Walbran6b650662021-09-07 13:13:23 +0000242/// The lifecycle state which the payload in the VM has reported itself to be in.
243///
244/// Note that the order of enum variants is significant; only forward transitions are allowed by
245/// [`VmInstance::update_payload_state`].
246#[derive(Copy, Clone, Debug, Eq, Ord, PartialEq, PartialOrd)]
247pub enum PayloadState {
248 Starting,
249 Started,
250 Ready,
251 Finished,
Jiyong Parka4eebde2022-07-12 18:01:12 +0900252 Hangup, // Hasn't reached to Ready before timeout expires
Andrew Walbran6b650662021-09-07 13:13:23 +0000253}
254
Andrew Walbranf8d94112021-09-07 11:45:36 +0000255/// The current state of the VM itself.
256#[derive(Debug)]
257pub enum VmState {
258 /// The VM has not yet tried to start.
259 NotStarted {
260 ///The configuration needed to start the VM, if it has not yet been started.
Inseob Kim7307a892023-09-14 13:37:58 +0900261 config: Box<CrosvmConfig>,
Andrew Walbranf8d94112021-09-07 11:45:36 +0000262 },
263 /// The VM has been started.
264 Running {
265 /// The crosvm child process.
266 child: Arc<SharedChild>,
David Brazdilf18e1162022-12-18 18:27:23 +0000267 /// The thread waiting for crosvm to finish.
268 monitor_vm_exit_thread: Option<JoinHandle<()>>,
Andrew Walbranf8d94112021-09-07 11:45:36 +0000269 },
270 /// The VM died or was killed.
271 Dead,
272 /// The VM failed to start.
273 Failed,
274}
275
Seungjae Yoo6d265d92022-11-15 10:51:33 +0900276/// RSS values of VM and CrosVM process itself.
277#[derive(Copy, Clone, Debug, Default)]
278pub struct Rss {
279 pub vm: i64,
280 pub crosvm: i64,
281}
282
283/// Metrics regarding the VM.
284#[derive(Debug, Default)]
285pub struct VmMetric {
286 /// Recorded timestamp when the VM is started.
287 pub start_timestamp: Option<SystemTime>,
Pierre-Clément Tosi44aba642023-09-15 11:44:34 +0100288 /// Update most recent guest_time periodically from /proc/[crosvm pid]/stat while VM is
289 /// running.
Seungjae Yoo6d265d92022-11-15 10:51:33 +0900290 pub cpu_guest_time: Option<i64>,
291 /// Update maximum RSS values periodically from /proc/[crosvm pid]/smaps while VM is running.
292 pub rss: Option<Rss>,
293}
294
Andrew Walbranf8d94112021-09-07 11:45:36 +0000295impl VmState {
296 /// Tries to start the VM, if it is in the `NotStarted` state.
297 ///
298 /// Returns an error if the VM is in the wrong state, or fails to start.
299 fn start(&mut self, instance: Arc<VmInstance>) -> Result<(), Error> {
300 let state = mem::replace(self, VmState::Failed);
301 if let VmState::NotStarted { config } = state {
Inseob Kim7307a892023-09-14 13:37:58 +0900302 let config = *config;
Jiyong Parke6ed0f92022-06-22 00:13:00 +0900303 let detect_hangup = config.detect_hangup;
Andrew Walbranb27681f2022-02-23 15:11:52 +0000304 let (failure_pipe_read, failure_pipe_write) = create_pipe()?;
Inseob Kim6ef80972023-07-20 17:23:36 +0900305 let vfio_devices = config.vfio_devices.clone();
Seungjae Yoo3271f172024-06-11 10:14:07 +0900306 let tap =
307 if let Some(tap_file) = &config.tap { Some(tap_file.try_clone()?) } else { None };
Andrew Walbranb27681f2022-02-23 15:11:52 +0000308
Andrew Walbranf8d94112021-09-07 11:45:36 +0000309 // If this fails and returns an error, `self` will be left in the `Failed` state.
Keir Fraser13a956a2022-07-14 14:20:46 +0000310 let child =
Keir Frasercdd4b112022-11-24 14:02:25 +0000311 Arc::new(run_vm(config, &instance.crosvm_control_socket_path, failure_pipe_write)?);
Andrew Walbranf8d94112021-09-07 11:45:36 +0000312
Seungjae Yoo6d265d92022-11-15 10:51:33 +0900313 let instance_monitor_status = instance.clone();
314 let child_monitor_status = child.clone();
315 thread::spawn(move || {
316 instance_monitor_status.clone().monitor_vm_status(child_monitor_status);
317 });
318
Andrew Walbranf8d94112021-09-07 11:45:36 +0000319 let child_clone = child.clone();
Jiyong Parka4eebde2022-07-12 18:01:12 +0900320 let instance_clone = instance.clone();
David Brazdilf18e1162022-12-18 18:27:23 +0000321 let monitor_vm_exit_thread = Some(thread::spawn(move || {
Seungjae Yoo3271f172024-06-11 10:14:07 +0900322 instance_clone.monitor_vm_exit(child_clone, failure_pipe_read, vfio_devices, tap);
David Brazdilf18e1162022-12-18 18:27:23 +0000323 }));
Andrew Walbranf8d94112021-09-07 11:45:36 +0000324
Jiyong Parka4eebde2022-07-12 18:01:12 +0900325 if detect_hangup {
326 let child_clone = child.clone();
327 thread::spawn(move || {
328 instance.monitor_payload_hangup(child_clone);
329 });
330 }
331
Andrew Walbranf8d94112021-09-07 11:45:36 +0000332 // If it started correctly, update the state.
David Brazdilf18e1162022-12-18 18:27:23 +0000333 *self = VmState::Running { child, monitor_vm_exit_thread };
Andrew Walbranf8d94112021-09-07 11:45:36 +0000334 Ok(())
335 } else {
336 *self = state;
337 bail!("VM already started or failed")
338 }
339 }
340}
341
David Brazdil8cf8f482022-11-23 14:21:26 +0000342/// Internal struct that holds the handles to globally unique resources of a VM.
343#[derive(Debug)]
344pub struct VmContext {
345 #[allow(dead_code)] // Keeps the global context alive
Yi-Yo Chiang2fbf0da2024-06-14 22:56:56 +0800346 pub(crate) global_context: Strong<dyn IGlobalVmContext>,
David Brazdil8cf8f482022-11-23 14:21:26 +0000347 #[allow(dead_code)] // Keeps the server alive
348 vm_server: RpcServer,
349}
350
351impl VmContext {
352 /// Construct new VmContext.
353 pub fn new(global_context: Strong<dyn IGlobalVmContext>, vm_server: RpcServer) -> VmContext {
354 VmContext { global_context, vm_server }
355 }
356}
357
Andrew Walbranf8d94112021-09-07 11:45:36 +0000358/// Information about a particular instance of a VM which may be running.
Andrew Walbrand6dce6f2021-03-05 16:39:08 +0000359#[derive(Debug)]
360pub struct VmInstance {
Andrew Walbranf8d94112021-09-07 11:45:36 +0000361 /// The current state of the VM.
362 pub vm_state: Mutex<VmState>,
David Brazdil8cf8f482022-11-23 14:21:26 +0000363 /// Global resources allocated for this VM.
364 #[allow(dead_code)] // Keeps the context alive
Yi-Yo Chiang2fbf0da2024-06-14 22:56:56 +0800365 pub(crate) vm_context: VmContext,
Andrew Walbrand6dce6f2021-03-05 16:39:08 +0000366 /// The CID assigned to the VM for vsock communication.
367 pub cid: Cid,
Keir Frasercdd4b112022-11-24 14:02:25 +0000368 /// Path to crosvm control socket
369 crosvm_control_socket_path: PathBuf,
Seungjae Yoo62085c02022-08-12 04:44:52 +0000370 /// The name of the VM.
371 pub name: String,
Andrew Walbranf8650422021-06-09 15:54:09 +0000372 /// Whether the VM is a protected VM.
373 pub protected: bool,
Andrew Walbranf5fbb7d2021-05-12 17:15:48 +0000374 /// Directory of temporary files used by the VM while it is running.
375 pub temporary_directory: PathBuf,
Andrew Walbranf6a1eb92021-04-01 11:16:02 +0000376 /// The UID of the process which requested the VM.
377 pub requester_uid: u32,
Andrew Walbranf6a1eb92021-04-01 11:16:02 +0000378 /// The PID of the process which requested the VM. Note that this process may no longer exist
379 /// and the PID may have been reused for a different process, so this should not be trusted.
Andrew Walbran02034492021-04-13 15:05:07 +0000380 pub requester_debug_pid: i32,
Andrew Walbrandae07162021-03-12 17:05:20 +0000381 /// Callbacks to clients of the VM.
382 pub callbacks: VirtualMachineCallbacks,
Inseob Kimc7d28c72021-10-25 14:28:10 +0000383 /// VirtualMachineService binder object for the VM.
Stephen Hinesaf21e512024-07-30 15:43:39 -0700384 #[allow(dead_code)]
Inseob Kimc7d28c72021-10-25 14:28:10 +0000385 pub vm_service: Mutex<Option<Strong<dyn IVirtualMachineService>>>,
Seungjae Yoo6d265d92022-11-15 10:51:33 +0900386 /// Recorded metrics of VM such as timestamp or cpu / memory usage.
387 pub vm_metric: Mutex<VmMetric>,
Andrew Walbran6b650662021-09-07 13:13:23 +0000388 /// The latest lifecycle state which the payload reported itself to be in.
389 payload_state: Mutex<PayloadState>,
Jiyong Parka4eebde2022-07-12 18:01:12 +0900390 /// Represents the condition that payload_state was updated
391 payload_state_updated: Condvar,
Jiyong Park08eee7b2022-11-11 15:01:26 +0900392 /// The human readable name of requester_uid
393 requester_uid_name: String,
394}
395
396impl fmt::Display for VmInstance {
397 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
398 let adj = if self.protected { "Protected" } else { "Non-protected" };
399 write!(
400 f,
401 "{} virtual machine \"{}\" (owner: {}, cid: {})",
402 adj, self.name, self.requester_uid_name, self.cid
403 )
404 }
Andrew Walbrand6dce6f2021-03-05 16:39:08 +0000405}
406
407impl VmInstance {
Andrew Walbranf8d94112021-09-07 11:45:36 +0000408 /// Validates the given config and creates a new `VmInstance` but doesn't start running it.
409 pub fn new(
Andrew Walbrand3a84182021-09-07 14:48:52 +0000410 config: CrosvmConfig,
Andrew Walbranf5fbb7d2021-05-12 17:15:48 +0000411 temporary_directory: PathBuf,
Andrew Walbranf6a1eb92021-04-01 11:16:02 +0000412 requester_uid: u32,
Andrew Walbran02034492021-04-13 15:05:07 +0000413 requester_debug_pid: i32,
David Brazdil8cf8f482022-11-23 14:21:26 +0000414 vm_context: VmContext,
Andrew Walbranf8d94112021-09-07 11:45:36 +0000415 ) -> Result<VmInstance, Error> {
416 validate_config(&config)?;
Andrew Walbrand3a84182021-09-07 14:48:52 +0000417 let cid = config.cid;
Seungjae Yoo62085c02022-08-12 04:44:52 +0000418 let name = config.name.clone();
Andrew Walbrand3a84182021-09-07 14:48:52 +0000419 let protected = config.protected;
Jiyong Park08eee7b2022-11-11 15:01:26 +0900420 let requester_uid_name = User::from_uid(Uid::from_raw(requester_uid))
421 .ok()
422 .flatten()
423 .map_or_else(|| format!("{}", requester_uid), |u| u.name);
424 let instance = VmInstance {
Inseob Kim7307a892023-09-14 13:37:58 +0900425 vm_state: Mutex::new(VmState::NotStarted { config: Box::new(config) }),
David Brazdil528e0472022-10-10 15:06:02 +0100426 vm_context,
Andrew Walbrand3a84182021-09-07 14:48:52 +0000427 cid,
Keir Frasercdd4b112022-11-24 14:02:25 +0000428 crosvm_control_socket_path: temporary_directory.join("crosvm.sock"),
Seungjae Yoo62085c02022-08-12 04:44:52 +0000429 name,
Andrew Walbrand3a84182021-09-07 14:48:52 +0000430 protected,
Andrew Walbranf5fbb7d2021-05-12 17:15:48 +0000431 temporary_directory,
Andrew Walbran02034492021-04-13 15:05:07 +0000432 requester_uid,
Andrew Walbran02034492021-04-13 15:05:07 +0000433 requester_debug_pid,
Andrew Walbranf8d94112021-09-07 11:45:36 +0000434 callbacks: Default::default(),
Inseob Kimc7d28c72021-10-25 14:28:10 +0000435 vm_service: Mutex::new(None),
Seungjae Yoo6d265d92022-11-15 10:51:33 +0900436 vm_metric: Mutex::new(Default::default()),
Andrew Walbranf8d94112021-09-07 11:45:36 +0000437 payload_state: Mutex::new(PayloadState::Starting),
Jiyong Parka4eebde2022-07-12 18:01:12 +0900438 payload_state_updated: Condvar::new(),
Jiyong Park08eee7b2022-11-11 15:01:26 +0900439 requester_uid_name,
440 };
441 info!("{} created", &instance);
442 Ok(instance)
Andrew Walbrandae07162021-03-12 17:05:20 +0000443 }
444
Andrew Walbranf8d94112021-09-07 11:45:36 +0000445 /// Starts an instance of `crosvm` to manage the VM. The `crosvm` instance will be killed when
446 /// the `VmInstance` is dropped.
447 pub fn start(self: &Arc<Self>) -> Result<(), Error> {
Seungjae Yoo6d265d92022-11-15 10:51:33 +0900448 let mut vm_metric = self.vm_metric.lock().unwrap();
449 vm_metric.start_timestamp = Some(SystemTime::now());
Jiyong Park08eee7b2022-11-11 15:01:26 +0900450 let ret = self.vm_state.lock().unwrap().start(self.clone());
451 if ret.is_ok() {
452 info!("{} started", &self);
453 }
454 ret.with_context(|| format!("{} failed to start", &self))
Andrew Walbranf8d94112021-09-07 11:45:36 +0000455 }
456
Jiyong Parka4eebde2022-07-12 18:01:12 +0900457 /// Monitors the exit of the VM (i.e. termination of the `child` process). When that happens,
458 /// handles the event by updating the state, noityfing the event to clients by calling
459 /// callbacks, and removing temporary files for the VM.
Inseob Kim6ef80972023-07-20 17:23:36 +0900460 fn monitor_vm_exit(
461 &self,
462 child: Arc<SharedChild>,
463 mut failure_pipe_read: File,
Inseob Kim7307a892023-09-14 13:37:58 +0900464 vfio_devices: Vec<VfioDevice>,
Seungjae Yoo3271f172024-06-11 10:14:07 +0900465 tap: Option<File>,
Inseob Kim6ef80972023-07-20 17:23:36 +0900466 ) {
Andrew Walbranc92d35f2022-01-12 12:45:19 +0000467 let result = child.wait();
468 match &result {
Jooyung Hanbfe086f2021-10-28 10:15:45 +0900469 Err(e) => error!("Error waiting for crosvm({}) instance to die: {}", child.id(), e),
Sebastian Ene23167d82022-10-07 14:09:53 +0000470 Ok(status) => {
471 info!("crosvm({}) exited with status {}", child.id(), status);
472 if let Some(exit_status_code) = status.code() {
473 if exit_status_code == CROSVM_WATCHDOG_REBOOT_STATUS {
474 info!("detected vcpu stall on crosvm");
475 }
476 }
477 }
Andrew Walbrandae07162021-03-12 17:05:20 +0000478 }
Andrew Walbranf8d94112021-09-07 11:45:36 +0000479
480 let mut vm_state = self.vm_state.lock().unwrap();
481 *vm_state = VmState::Dead;
482 // Ensure that the mutex is released before calling the callbacks.
483 drop(vm_state);
Jiyong Park08eee7b2022-11-11 15:01:26 +0900484 info!("{} exited", &self);
Andrew Walbranf8d94112021-09-07 11:45:36 +0000485
Jiyong Parka4eebde2022-07-12 18:01:12 +0900486 // Read the pipe to see if any failure reason is written
487 let mut failure_reason = String::new();
488 match failure_pipe_read.read_to_string(&mut failure_reason) {
489 Err(e) => error!("Error reading VM failure reason from pipe: {}", e),
490 Ok(len) if len > 0 => info!("VM returned failure reason '{}'", &failure_reason),
491 _ => (),
Jiyong Parke6ed0f92022-06-22 00:13:00 +0900492 };
Andrew Walbranb27681f2022-02-23 15:11:52 +0000493
Jiyong Parka4eebde2022-07-12 18:01:12 +0900494 // In case of hangup, the pipe doesn't give us any information because the hangup can't be
495 // detected on the VM side (otherwise, it isn't a hangup), but in the
496 // monitor_payload_hangup function below which updates the payload state to Hangup.
497 let failure_reason =
498 if failure_reason.is_empty() && self.payload_state() == PayloadState::Hangup {
499 Cow::from("HANGUP")
500 } else {
501 Cow::from(failure_reason)
502 };
503
Jiyong Parke558ab12022-07-07 20:18:55 +0900504 self.handle_ramdump().unwrap_or_else(|e| error!("Error handling ramdump: {}", e));
Seungjae Yoob4c07ba2022-08-12 04:44:52 +0000505
506 let death_reason = death_reason(&result, &failure_reason);
Seungjae Yoo93430e82022-12-05 16:37:42 +0900507 let exit_signal = exit_signal(&result);
508
Seungjae Yoob4c07ba2022-08-12 04:44:52 +0000509 self.callbacks.callback_on_died(self.cid, death_reason);
Seungjae Yoo2e7beea2022-08-24 16:09:12 +0900510
Seungjae Yoo6d265d92022-11-15 10:51:33 +0900511 let vm_metric = self.vm_metric.lock().unwrap();
Seungjae Yoob2db3d22023-04-06 18:05:43 +0900512 write_vm_exited_stats_sync(
Seungjae Yoo93430e82022-12-05 16:37:42 +0900513 self.requester_uid as i32,
514 &self.name,
515 death_reason,
516 exit_signal,
Chris Wailes75269622022-12-05 23:01:44 -0800517 &vm_metric,
Seungjae Yoo93430e82022-12-05 16:37:42 +0900518 );
Andrew Walbranf5fbb7d2021-05-12 17:15:48 +0000519
David Brazdil4b4c5102022-12-19 22:56:20 +0000520 // Delete temporary files. The folder itself is removed by VirtualizationServiceInternal.
521 remove_temporary_files(&self.temporary_directory).unwrap_or_else(|e| {
522 error!("Error removing temporary files from {:?}: {}", self.temporary_directory, e);
523 });
Inseob Kim6ef80972023-07-20 17:23:36 +0900524
Seungjae Yoo3271f172024-06-11 10:14:07 +0900525 if let Some(tap_file) = tap {
526 GLOBAL_SERVICE
527 .deleteTapInterface(&ParcelFileDescriptor::new(OwnedFd::from(tap_file)))
528 .unwrap_or_else(|e| {
529 error!("Error deleting TAP interface: {e:?}");
530 });
531 }
532
Jakob Vukalovicd42aa2c2023-11-09 16:04:00 +0000533 drop(vfio_devices); // Cleanup devices.
Andrew Walbrandae07162021-03-12 17:05:20 +0000534 }
535
Jiyong Parka4eebde2022-07-12 18:01:12 +0900536 /// Waits until payload is started, or timeout expires. When timeout occurs, kill
537 /// the VM to prevent indefinite hangup and update the payload_state accordingly.
538 fn monitor_payload_hangup(&self, child: Arc<SharedChild>) {
539 debug!("Starting to monitor hangup for Microdroid({})", child.id());
Jiyong Parkcf975612024-04-07 13:27:10 +0900540 let (state, result) = self
Jiyong Parka4eebde2022-07-12 18:01:12 +0900541 .payload_state_updated
542 .wait_timeout_while(self.payload_state.lock().unwrap(), *BOOT_HANGUP_TIMEOUT, |s| {
543 *s < PayloadState::Started
544 })
545 .unwrap();
Jiyong Parkcf975612024-04-07 13:27:10 +0900546 drop(state); // we are not interested in state
Jiyong Parka4eebde2022-07-12 18:01:12 +0900547 let child_still_running = child.try_wait().ok() == Some(None);
548 if result.timed_out() && child_still_running {
549 error!(
550 "Microdroid({}) failed to start payload within {} secs timeout. Shutting down.",
551 child.id(),
552 BOOT_HANGUP_TIMEOUT.as_secs()
553 );
554 self.update_payload_state(PayloadState::Hangup).unwrap();
555 if let Err(e) = self.kill() {
556 error!("Error stopping timed-out VM with CID {}: {:?}", child.id(), e);
557 }
558 }
559 }
560
Seungjae Yoo6d265d92022-11-15 10:51:33 +0900561 fn monitor_vm_status(&self, child: Arc<SharedChild>) {
562 let pid = child.id();
563
564 loop {
565 {
566 // Check VM state
567 let vm_state = &*self.vm_state.lock().unwrap();
568 if let VmState::Dead = vm_state {
569 break;
570 }
571
572 let mut vm_metric = self.vm_metric.lock().unwrap();
573
574 // Get CPU Information
Jiyong Park85ea5382024-02-19 15:04:23 +0900575 match get_guest_time(pid) {
576 Ok(guest_time) => vm_metric.cpu_guest_time = Some(guest_time),
577 Err(e) => error!("Failed to get guest CPU time: {e:?}"),
Seungjae Yoo6d265d92022-11-15 10:51:33 +0900578 }
579
580 // Get Memory Information
Jiyong Park85ea5382024-02-19 15:04:23 +0900581 match get_rss(pid) {
582 Ok(rss) => {
583 vm_metric.rss = match &vm_metric.rss {
584 Some(x) => Some(Rss::extract_max(x, &rss)),
585 None => Some(rss),
586 }
Seungjae Yoo6d265d92022-11-15 10:51:33 +0900587 }
Jiyong Park85ea5382024-02-19 15:04:23 +0900588 Err(e) => error!("Failed to get guest RSS: {}", e),
Seungjae Yoo6d265d92022-11-15 10:51:33 +0900589 }
590 }
591
592 thread::sleep(Duration::from_secs(1));
593 }
594 }
595
Andrew Walbran6b650662021-09-07 13:13:23 +0000596 /// Returns the last reported state of the VM payload.
597 pub fn payload_state(&self) -> PayloadState {
598 *self.payload_state.lock().unwrap()
599 }
600
601 /// Updates the payload state to the given value, if it is a valid state transition.
602 pub fn update_payload_state(&self, new_state: PayloadState) -> Result<(), Error> {
603 let mut state_locked = self.payload_state.lock().unwrap();
604 // Only allow forward transitions, e.g. from starting to started or finished, not back in
605 // the other direction.
606 if new_state > *state_locked {
607 *state_locked = new_state;
Jiyong Parka4eebde2022-07-12 18:01:12 +0900608 self.payload_state_updated.notify_all();
Andrew Walbran6b650662021-09-07 13:13:23 +0000609 Ok(())
610 } else {
611 bail!("Invalid payload state transition from {:?} to {:?}", *state_locked, new_state)
612 }
613 }
614
Andrew Walbranf8d94112021-09-07 11:45:36 +0000615 /// Kills the crosvm instance, if it is running.
Inseob Kima446f802022-07-11 19:46:37 +0900616 pub fn kill(&self) -> Result<(), Error> {
David Brazdilf18e1162022-12-18 18:27:23 +0000617 let monitor_vm_exit_thread = {
618 let vm_state = &mut *self.vm_state.lock().unwrap();
619 if let VmState::Running { child, monitor_vm_exit_thread } = vm_state {
620 let id = child.id();
621 debug!("Killing crosvm({})", id);
622 // TODO: Talk to crosvm to shutdown cleanly.
623 child.kill().with_context(|| format!("Error killing crosvm({id}) instance"))?;
624 monitor_vm_exit_thread.take()
Inseob Kima446f802022-07-11 19:46:37 +0900625 } else {
David Brazdilf18e1162022-12-18 18:27:23 +0000626 bail!("VM is not running")
Andrew Walbranf8d94112021-09-07 11:45:36 +0000627 }
David Brazdilf18e1162022-12-18 18:27:23 +0000628 };
629
630 // Wait for monitor_vm_exit() to finish. Must release vm_state lock
631 // first, as monitor_vm_exit() takes it as well.
632 monitor_vm_exit_thread.map(JoinHandle::join);
633
David Brazdil5af55832023-03-08 13:08:56 +0000634 // Now that the VM has been killed, shut down the VirtualMachineService
635 // server to eagerly free up the server threads.
636 self.vm_context.vm_server.shutdown()?;
637
David Brazdilf18e1162022-12-18 18:27:23 +0000638 Ok(())
Andrew Walbrand6dce6f2021-03-05 16:39:08 +0000639 }
Jiyong Parke558ab12022-07-07 20:18:55 +0900640
Keir Frasercdd4b112022-11-24 14:02:25 +0000641 /// Responds to memory-trimming notifications by inflating the virtio
642 /// balloon to reclaim guest memory.
Keir Fraser48221ba2024-07-12 14:05:02 +0000643 pub fn get_memory_balloon(&self) -> Result<u64, Error> {
Keir Frasercdd4b112022-11-24 14:02:25 +0000644 let request = VmRequest::BalloonCommand(BalloonControlCommand::Stats {});
Keir Fraser48221ba2024-07-12 14:05:02 +0000645 let result =
646 match vm_control::client::handle_request(&request, &self.crosvm_control_socket_path) {
647 Ok(VmResponse::BalloonStats { stats: _, balloon_actual }) => balloon_actual,
648 Ok(VmResponse::Err(e)) => {
649 // ENOTSUP is returned when the balloon protocol is not initialized. This
650 // can occur for numerous reasons: Guest is still booting, guest doesn't
651 // support ballooning, host doesn't support ballooning. We don't log or
652 // raise an error in this case: trim is just a hint and we can ignore it.
653 if e.errno() != libc::ENOTSUP {
654 bail!("Errno return when requesting balloon stats: {}", e.errno())
Keir Frasercdd4b112022-11-24 14:02:25 +0000655 }
Keir Fraser48221ba2024-07-12 14:05:02 +0000656 0
Keir Frasercdd4b112022-11-24 14:02:25 +0000657 }
Keir Fraser48221ba2024-07-12 14:05:02 +0000658 e => bail!("Error requesting balloon stats: {:?}", e),
659 };
660 Ok(result)
661 }
662
663 /// Responds to memory-trimming notifications by inflating the virtio
664 /// balloon to reclaim guest memory.
665 pub fn set_memory_balloon(&self, num_bytes: u64) -> Result<(), Error> {
666 let command = BalloonControlCommand::Adjust { num_bytes, wait_for_success: false };
667 if let Err(e) = vm_control::client::handle_request(
668 &VmRequest::BalloonCommand(command),
669 &self.crosvm_control_socket_path,
670 ) {
671 bail!("Error sending balloon adjustment: {:?}", e);
Keir Frasercdd4b112022-11-24 14:02:25 +0000672 }
673 Ok(())
674 }
675
Alan Stokes3e98d292022-12-14 15:18:22 +0000676 /// Checks if ramdump has been created. If so, send it to tombstoned.
Jiyong Parke558ab12022-07-07 20:18:55 +0900677 fn handle_ramdump(&self) -> Result<(), Error> {
678 let ramdump_path = self.temporary_directory.join("ramdump");
Jiyong Park933d2b22023-02-28 11:41:19 +0900679 if !ramdump_path.as_path().try_exists()? {
680 return Ok(());
681 }
Jiyong Parke558ab12022-07-07 20:18:55 +0900682 if std::fs::metadata(&ramdump_path)?.len() > 0 {
Jiyong Park1612b902022-08-22 14:47:39 +0900683 Self::send_ramdump_to_tombstoned(&ramdump_path)?;
Jiyong Parke558ab12022-07-07 20:18:55 +0900684 }
685 Ok(())
686 }
Jiyong Park1612b902022-08-22 14:47:39 +0900687
688 fn send_ramdump_to_tombstoned(ramdump_path: &Path) -> Result<(), Error> {
689 let mut input = File::open(ramdump_path)
Alan Stokes3e98d292022-12-14 15:18:22 +0000690 .context(format!("Failed to open ramdump {:?} for reading", ramdump_path))?;
Jiyong Park1612b902022-08-22 14:47:39 +0900691
692 let pid = std::process::id() as i32;
693 let conn = TombstonedConnection::connect(pid, DebuggerdDumpType::Tombstone)
694 .context("Failed to connect to tombstoned")?;
695 let mut output = conn
696 .text_output
697 .as_ref()
698 .ok_or_else(|| anyhow!("Could not get file to write the tombstones on"))?;
699
700 std::io::copy(&mut input, &mut output).context("Failed to send ramdump to tombstoned")?;
701 info!("Ramdump {:?} sent to tombstoned", ramdump_path);
702
703 conn.notify_completion()?;
704 Ok(())
705 }
Jiyong Park23b67392024-07-04 13:51:42 +0900706
707 /// Suspends the VM
708 pub fn suspend(&self) -> Result<(), Error> {
709 match vm_control::client::handle_request(
710 &VmRequest::SuspendVcpus,
711 &self.crosvm_control_socket_path,
712 ) {
713 Ok(VmResponse::Ok) => Ok(()),
714 e => bail!("Failed to suspend VM: {e:?}"),
715 }
716 }
717
718 /// Resumes the suspended VM
719 pub fn resume(&self) -> Result<(), Error> {
720 match vm_control::client::handle_request(
721 &VmRequest::ResumeVcpus,
722 &self.crosvm_control_socket_path,
723 ) {
724 Ok(VmResponse::Ok) => Ok(()),
725 e => bail!("Failed to resume: {e:?}"),
726 }
727 }
Andrew Walbrand6dce6f2021-03-05 16:39:08 +0000728}
729
Seungjae Yoo6d265d92022-11-15 10:51:33 +0900730impl Rss {
731 fn extract_max(x: &Rss, y: &Rss) -> Rss {
732 Rss { vm: max(x.vm, y.vm), crosvm: max(x.crosvm, y.crosvm) }
733 }
734}
735
David Daiaaecf652024-02-22 14:41:39 -0800736// Get Cpus_allowed mask
737fn check_if_all_cpus_allowed() -> Result<bool> {
738 let file = read_to_string("/proc/self/status")?;
739 let lines: Vec<_> = file.split('\n').collect();
740
741 for line in lines {
742 if line.contains("Cpus_allowed_list") {
743 let prop: Vec<_> = line.split_whitespace().collect();
744 if prop.len() != 2 {
745 return Ok(false);
746 }
747 let cpu_list: Vec<_> = prop[1].split('-').collect();
748 //Only contiguous Cpu list allowed
749 if cpu_list.len() != 2 {
750 return Ok(false);
751 }
752 if let Some(cpus) = get_num_cpus() {
753 let max_cpu = cpu_list[1].parse::<usize>()?;
754 if max_cpu == cpus - 1 {
755 return Ok(true);
756 } else {
757 return Ok(false);
758 }
759 }
760 }
761 }
762 Ok(false)
763}
764
Seungjae Yoo6d265d92022-11-15 10:51:33 +0900765// Get guest time from /proc/[crosvm pid]/stat
766fn get_guest_time(pid: u32) -> Result<i64> {
767 let file = read_to_string(format!("/proc/{}/stat", pid))?;
768 let data_list: Vec<_> = file.split_whitespace().collect();
769
770 // Information about guest_time is at 43th place of the file split with the whitespace.
771 // Example of /proc/[pid]/stat :
772 // 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
773 // 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
774 if data_list.len() < 43 {
775 bail!("Failed to parse command result for getting guest time : {}", file);
776 }
777
778 let guest_time_ticks = data_list[42].parse::<i64>()?;
Andrew Walbranb58d1b42023-07-07 13:54:49 +0100779 // SAFETY: It just returns an integer about CPU tick information.
Charisee96113f32023-01-26 09:00:42 +0000780 let ticks_per_sec = unsafe { sysconf(_SC_CLK_TCK) };
Seungjae Yoo6d265d92022-11-15 10:51:33 +0900781 Ok(guest_time_ticks * MILLIS_PER_SEC / ticks_per_sec)
782}
783
784// Get rss from /proc/[crosvm pid]/smaps
785fn get_rss(pid: u32) -> Result<Rss> {
786 let file = read_to_string(format!("/proc/{}/smaps", pid))?;
787 let lines: Vec<_> = file.split('\n').collect();
788
789 let mut rss_vm_total = 0i64;
790 let mut rss_crosvm_total = 0i64;
791 let mut is_vm = false;
792 for line in lines {
793 if line.contains("crosvm_guest") {
794 is_vm = true;
795 } else if line.contains("Rss:") {
796 let data_list: Vec<_> = line.split_whitespace().collect();
797 if data_list.len() < 2 {
798 bail!("Failed to parse command result for getting rss :\n{}", line);
799 }
800 let rss = data_list[1].parse::<i64>()?;
801
802 if is_vm {
803 rss_vm_total += rss;
804 is_vm = false;
805 }
806 rss_crosvm_total += rss;
807 }
808 }
809
810 Ok(Rss { vm: rss_vm_total, crosvm: rss_crosvm_total })
811}
812
Alan Stokes3ba10fd2022-10-06 15:46:51 +0100813fn death_reason(result: &Result<ExitStatus, io::Error>, mut failure_reason: &str) -> DeathReason {
Pierre-Clément Tosic0611e72023-09-15 11:41:07 +0100814 if let Some((reason, info)) = failure_reason.split_once('|') {
Alan Stokes3ba10fd2022-10-06 15:46:51 +0100815 // Separator indicates extra context information is present after the failure name.
Pierre-Clément Tosic0611e72023-09-15 11:41:07 +0100816 error!("Failure info: {info}");
817 failure_reason = reason;
Alan Stokes3ba10fd2022-10-06 15:46:51 +0100818 }
Andrew Walbranc92d35f2022-01-12 12:45:19 +0000819 if let Ok(status) = result {
Andrew Walbranb27681f2022-02-23 15:11:52 +0000820 match failure_reason {
821 "PVM_FIRMWARE_PUBLIC_KEY_MISMATCH" => {
822 return DeathReason::PVM_FIRMWARE_PUBLIC_KEY_MISMATCH
823 }
824 "PVM_FIRMWARE_INSTANCE_IMAGE_CHANGED" => {
825 return DeathReason::PVM_FIRMWARE_INSTANCE_IMAGE_CHANGED
826 }
Inseob Kim272f5722022-06-13 17:14:51 +0900827 "MICRODROID_FAILED_TO_CONNECT_TO_VIRTUALIZATION_SERVICE" => {
828 return DeathReason::MICRODROID_FAILED_TO_CONNECT_TO_VIRTUALIZATION_SERVICE
829 }
830 "MICRODROID_PAYLOAD_HAS_CHANGED" => return DeathReason::MICRODROID_PAYLOAD_HAS_CHANGED,
831 "MICRODROID_PAYLOAD_VERIFICATION_FAILED" => {
832 return DeathReason::MICRODROID_PAYLOAD_VERIFICATION_FAILED
833 }
834 "MICRODROID_INVALID_PAYLOAD_CONFIG" => {
835 return DeathReason::MICRODROID_INVALID_PAYLOAD_CONFIG
836 }
837 "MICRODROID_UNKNOWN_RUNTIME_ERROR" => {
838 return DeathReason::MICRODROID_UNKNOWN_RUNTIME_ERROR
839 }
Jiyong Parke6ed0f92022-06-22 00:13:00 +0900840 "HANGUP" => return DeathReason::HANGUP,
Andrew Walbranb27681f2022-02-23 15:11:52 +0000841 _ => {}
842 }
Andrew Walbranc92d35f2022-01-12 12:45:19 +0000843 match status.code() {
844 None => DeathReason::KILLED,
845 Some(0) => DeathReason::SHUTDOWN,
Alan Stokes7c459e82022-12-06 12:21:49 +0000846 Some(CROSVM_START_ERROR_STATUS) => DeathReason::START_FAILED,
Andrew Walbranc92d35f2022-01-12 12:45:19 +0000847 Some(CROSVM_REBOOT_STATUS) => DeathReason::REBOOT,
Andrew Walbrand15c5632022-02-03 13:38:31 +0000848 Some(CROSVM_CRASH_STATUS) => DeathReason::CRASH,
Sebastian Ene23167d82022-10-07 14:09:53 +0000849 Some(CROSVM_WATCHDOG_REBOOT_STATUS) => DeathReason::WATCHDOG_REBOOT,
Andrew Walbranc92d35f2022-01-12 12:45:19 +0000850 Some(_) => DeathReason::UNKNOWN,
851 }
852 } else {
853 DeathReason::INFRASTRUCTURE_ERROR
854 }
855}
856
Seungjae Yoo93430e82022-12-05 16:37:42 +0900857fn exit_signal(result: &Result<ExitStatus, io::Error>) -> Option<i32> {
858 match result {
859 Ok(status) => status.signal(),
860 Err(_) => None,
861 }
862}
863
Inseob Kim6ef80972023-07-20 17:23:36 +0900864const SYSFS_PLATFORM_DEVICES_PATH: &str = "/sys/devices/platform/";
865const VFIO_PLATFORM_DRIVER_PATH: &str = "/sys/bus/platform/drivers/vfio-platform";
866
Inseob Kim7307a892023-09-14 13:37:58 +0900867fn vfio_argument_for_platform_device(device: &VfioDevice) -> Result<String, Error> {
Inseob Kim6ef80972023-07-20 17:23:36 +0900868 // Check platform device exists
Jakob Vukalovicd42aa2c2023-11-09 16:04:00 +0000869 let path = Path::new(&device.getSysfsPath()?).canonicalize()?;
Inseob Kim6ef80972023-07-20 17:23:36 +0900870 if !path.starts_with(SYSFS_PLATFORM_DEVICES_PATH) {
871 bail!("{path:?} is not a platform device");
872 }
873
874 // Check platform device is bound to VFIO driver
875 let dev_driver_path = path.join("driver").canonicalize()?;
876 if dev_driver_path != Path::new(VFIO_PLATFORM_DRIVER_PATH) {
877 bail!("{path:?} is not bound to VFIO-platform driver");
878 }
879
880 if let Some(p) = path.to_str() {
Jakob Vukalovicd42aa2c2023-11-09 16:04:00 +0000881 Ok(format!("--vfio={p},iommu=pkvm-iommu,dt-symbol={0}", device.getDtboLabel()?))
Inseob Kim6ef80972023-07-20 17:23:36 +0900882 } else {
883 bail!("invalid path {path:?}");
884 }
885}
886
Andrew Walbrand3a84182021-09-07 14:48:52 +0000887/// Starts an instance of `crosvm` to manage a new VM.
Keir Fraser13a956a2022-07-14 14:20:46 +0000888fn run_vm(
889 config: CrosvmConfig,
Keir Frasercdd4b112022-11-24 14:02:25 +0000890 crosvm_control_socket_path: &Path,
Keir Fraser13a956a2022-07-14 14:20:46 +0000891 failure_pipe_write: File,
892) -> Result<SharedChild, Error> {
Andrew Walbrand3a84182021-09-07 14:48:52 +0000893 validate_config(&config)?;
Andrew Walbrand6dce6f2021-03-05 16:39:08 +0000894
895 let mut command = Command::new(CROSVM_PATH);
896 // TODO(qwandor): Remove --disable-sandbox.
Andrew Walbranc92d35f2022-01-12 12:45:19 +0000897 command
898 .arg("--extended-status")
Jiyong Park6c60fea2022-10-24 16:10:01 +0900899 // Configure the logger for the crosvm process to silence logs from the disk crate which
900 // don't provide much information to us (but do spamming us).
901 .arg("--log-level")
Frederick Mayle138f53c2023-01-24 18:55:26 -0800902 .arg("info,disk=warn")
Andrew Walbranc92d35f2022-01-12 12:45:19 +0000903 .arg("run")
904 .arg("--disable-sandbox")
905 .arg("--cid")
906 .arg(config.cid.to_string());
Andrew Walbran3a5a9212021-05-04 17:09:08 +0000907
Jeongik Cha75057452024-09-03 11:26:04 +0900908 if system_properties::read_bool("hypervisor.memory_reclaim.supported", false)?
909 && !config.no_balloon
910 {
Keir Fraserf25cb922022-11-23 14:26:00 +0000911 command.arg("--balloon-page-reporting");
912 } else {
913 command.arg("--no-balloon");
914 }
915
Elie Kheirallah29d72912024-08-07 20:17:26 +0000916 if !config.usb_config.controller {
917 command.arg("--no-usb");
918 }
919
Jiyong Parkae625b62024-06-13 21:47:52 +0900920 let mut memory_mib = config.memory_mib;
921
Andrew Walbranf8650422021-06-09 15:54:09 +0000922 if config.protected {
Jaewan Kim46422812022-11-25 10:59:39 +0900923 match system_properties::read(SYSPROP_CUSTOM_PVMFW_PATH)? {
924 Some(pvmfw_path) if !pvmfw_path.is_empty() => {
925 command.arg("--protected-vm-with-firmware").arg(pvmfw_path)
926 }
927 _ => command.arg("--protected-vm"),
928 };
Andrew Walbran0b5789f2022-02-04 13:57:57 +0000929
930 // 3 virtio-console devices + vsock = 4.
931 let virtio_pci_device_count = 4 + config.disks.len();
932 // crosvm virtio queue has 256 entries, so 2 MiB per device (2 pages per entry) should be
933 // enough.
Jiyong Park3115cdb2023-02-24 18:04:44 +0900934 let swiotlb_size_mib = 2 * virtio_pci_device_count as u32;
Andrew Walbran0b5789f2022-02-04 13:57:57 +0000935 command.arg("--swiotlb").arg(swiotlb_size_mib.to_string());
Frederick Mayle58baa302023-02-14 19:06:37 -0800936
Frederick Maylea24bbe02024-07-12 16:50:48 -0700937 // b/346770542 for consistent "usable" memory across protected and non-protected VMs.
Frederick Mayle68d06e72024-07-12 17:18:11 -0700938 memory_mib = memory_mib.saturating_add(swiotlb_size_mib);
Jiyong Parkae625b62024-06-13 21:47:52 +0900939
Frederick Mayle58baa302023-02-14 19:06:37 -0800940 // Workaround to keep crash_dump from trying to read protected guest memory.
941 // Context in b/238324526.
942 command.arg("--unmap-guest-memory-on-fork");
Jiyong Park3115cdb2023-02-24 18:04:44 +0900943
Jiyong Parked180932023-02-24 19:55:41 +0900944 if config.ramdump.is_some() {
Jaewan Kim3124ef02023-03-23 19:25:20 +0900945 // Protected VM needs to reserve memory for ramdump here. Note that we reserve more
Jiyong Parked180932023-02-24 19:55:41 +0900946 // memory for the restricted dma pool.
947 let ramdump_reserve = RAMDUMP_RESERVED_MIB + swiotlb_size_mib;
948 command.arg("--params").arg(format!("crashkernel={ramdump_reserve}M"));
949 }
Jaewan Kimd4dae652023-03-28 13:45:43 +0900950 } else if config.ramdump.is_some() {
951 command.arg("--params").arg(format!("crashkernel={RAMDUMP_RESERVED_MIB}M"));
952 }
Jaewan Kim61f86142023-03-28 15:12:52 +0900953 if config.debug_config.debug_level == DebugLevel::NONE
954 && config.debug_config.should_prepare_console_output()
955 {
Jaewan Kimd4dae652023-03-28 13:45:43 +0900956 // bootconfig.normal will be used, but we need log.
957 command.arg("--params").arg("printk.devkmsg=on");
958 command.arg("--params").arg("console=hvc0");
Andrew Walbranf8650422021-06-09 15:54:09 +0000959 }
960
Frederick Mayle68d06e72024-07-12 17:18:11 -0700961 command.arg("--mem").arg(memory_mib.to_string());
Andrew Walbranb15cd6e2021-07-05 16:38:07 +0000962
Jiyong Park032615f2022-01-10 13:55:34 +0900963 if let Some(cpus) = config.cpus {
964 command.arg("--cpus").arg(cpus.to_string());
965 }
966
David Brazdil7d1e5ec2023-02-06 17:56:29 +0000967 if config.host_cpu_topology {
David Daiaaecf652024-02-22 14:41:39 -0800968 if cfg!(virt_cpufreq) && check_if_all_cpus_allowed()? {
David Daie6f5bb02024-01-25 19:17:02 -0800969 command.arg("--host-cpu-topology");
David Daic97e82d2024-02-17 12:10:31 -0800970 cfg_if::cfg_if! {
971 if #[cfg(any(target_arch = "aarch64"))] {
972 command.arg("--virt-cpufreq");
973 }
974 }
David Daie6f5bb02024-01-25 19:17:02 -0800975 } else if let Some(cpus) = get_num_cpus() {
David Brazdil25e8c052023-02-17 12:53:01 +0000976 command.arg("--cpus").arg(cpus.to_string());
977 } else {
978 bail!("Could not determine the number of CPUs in the system");
979 }
David Brazdil7d1e5ec2023-02-06 17:56:29 +0000980 }
981
Nikita Ioffe5776f082023-02-10 21:38:26 +0000982 if let Some(gdb_port) = config.gdb_port {
983 command.arg("--gdb").arg(gdb_port.to_string());
984 }
985
Jiyong Parkfa91d702021-10-18 23:51:39 +0900986 // Keep track of what file descriptors should be mapped to the crosvm process.
Jiyong Park69001a22024-08-16 20:45:34 +0900987 let mut preserved_fds = config.indirect_files.into_iter().map(|f| f.into()).collect();
Jiyong Parkfa91d702021-10-18 23:51:39 +0900988
Elie Kheirallah5c807a22024-09-23 20:40:42 +0000989 if let Some(dump_dt_fd) = config.dump_dt_fd {
990 let dump_dt_fd = add_preserved_fd(&mut preserved_fds, dump_dt_fd);
991 command.arg("--dump-device-tree-blob").arg(dump_dt_fd);
992 }
993
Jiyong Park747d6362021-10-19 17:12:52 +0900994 // Setup the serial devices.
995 // 1. uart device: used as the output device by bootloaders and as early console by linux
Andrew Walbranb27681f2022-02-23 15:11:52 +0000996 // 2. uart device: used to report the reason for the VM failing.
997 // 3. virtio-console device: used as the console device where kmsg is redirected to
Jiyong Park4afe2012022-07-08 05:38:49 +0900998 // 4. virtio-console device: used as the ramdump output
Andrew Walbranb27681f2022-02-23 15:11:52 +0000999 // 5. virtio-console device: used as the logcat output
Jiyong Park747d6362021-10-19 17:12:52 +09001000 //
Jiyong Parkb8182bb2021-10-26 22:53:08 +09001001 // When [console|log]_fd is not specified, the devices are attached to sink, which means what's
1002 // written there is discarded.
Jiyong Park69001a22024-08-16 20:45:34 +09001003 let console_out_arg = format_serial_out_arg(&mut preserved_fds, config.console_out_fd);
Jiyong Parke6fb1672023-06-26 16:45:55 +09001004 let console_in_arg = config
1005 .console_in_fd
Jiyong Parke6fb1672023-06-26 16:45:55 +09001006 .map(|fd| format!(",input={}", add_preserved_fd(&mut preserved_fds, fd)))
1007 .unwrap_or_default();
Jiyong Park69001a22024-08-16 20:45:34 +09001008 let log_arg = format_serial_out_arg(&mut preserved_fds, config.log_fd);
1009 let failure_serial_path = add_preserved_fd(&mut preserved_fds, failure_pipe_write);
1010 let ramdump_arg = format_serial_out_arg(&mut preserved_fds, config.ramdump);
Yi-Yo Chiang8dd32552024-05-22 19:38:16 +08001011 let console_input_device = config.console_input_device.as_deref().unwrap_or(CONSOLE_HVC0);
1012 match console_input_device {
1013 CONSOLE_HVC0 | CONSOLE_TTYS0 => {}
1014 _ => bail!("Unsupported serial device {console_input_device}"),
1015 };
Jiyong Parkfa91d702021-10-18 23:51:39 +09001016
Jiyong Park747d6362021-10-19 17:12:52 +09001017 // Warning: Adding more serial devices requires you to shift the PCI device ID of the boot
1018 // disks in bootconfig.x86_64. This is because x86 crosvm puts serial devices and the block
1019 // devices in the same PCI bus and serial devices comes before the block devices. Arm crosvm
1020 // doesn't have the issue.
Jiyong Parkfa91d702021-10-18 23:51:39 +09001021 // /dev/ttyS0
Yi-Yo Chiang8dd32552024-05-22 19:38:16 +08001022 command.arg(format!(
1023 "--serial={}{},hardware=serial,num=1",
1024 &console_out_arg,
1025 if console_input_device == CONSOLE_TTYS0 { &console_in_arg } else { "" }
1026 ));
Andrew Walbranb27681f2022-02-23 15:11:52 +00001027 // /dev/ttyS1
1028 command.arg(format!("--serial=type=file,path={},hardware=serial,num=2", &failure_serial_path));
Jiyong Parkfa91d702021-10-18 23:51:39 +09001029 // /dev/hvc0
Jiyong Parke6fb1672023-06-26 16:45:55 +09001030 command.arg(format!(
1031 "--serial={}{},hardware=virtio-console,num=1",
Yi-Yo Chiang8dd32552024-05-22 19:38:16 +08001032 &console_out_arg,
1033 if console_input_device == CONSOLE_HVC0 { &console_in_arg } else { "" }
Jiyong Parke6fb1672023-06-26 16:45:55 +09001034 ));
Jiyong Park4afe2012022-07-08 05:38:49 +09001035 // /dev/hvc1
1036 command.arg(format!("--serial={},hardware=virtio-console,num=2", &ramdump_arg));
Jiyong Parkae5a4ed2021-11-01 18:36:28 +09001037 // /dev/hvc2
1038 command.arg(format!("--serial={},hardware=virtio-console,num=3", &log_arg));
Andrew Walbran3a5a9212021-05-04 17:09:08 +00001039
Jiyong Park69001a22024-08-16 20:45:34 +09001040 if let Some(bootloader) = config.bootloader {
Andrew Walbran02b8ec02021-06-22 13:07:02 +00001041 command.arg("--bios").arg(add_preserved_fd(&mut preserved_fds, bootloader));
Andrew Walbrand6dce6f2021-03-05 16:39:08 +00001042 }
Andrew Walbran3a5a9212021-05-04 17:09:08 +00001043
Jiyong Park69001a22024-08-16 20:45:34 +09001044 if let Some(initrd) = config.initrd {
Andrew Walbran02b8ec02021-06-22 13:07:02 +00001045 command.arg("--initrd").arg(add_preserved_fd(&mut preserved_fds, initrd));
Andrew Walbrand6dce6f2021-03-05 16:39:08 +00001046 }
Andrew Walbran3a5a9212021-05-04 17:09:08 +00001047
Andrew Walbrand6dce6f2021-03-05 16:39:08 +00001048 if let Some(params) = &config.params {
1049 command.arg("--params").arg(params);
1050 }
Andrew Walbran3a5a9212021-05-04 17:09:08 +00001051
Jiyong Park69001a22024-08-16 20:45:34 +09001052 for disk in config.disks {
Frederick Mayleeae6d2d2024-09-12 11:05:36 -07001053 // Disk file locking is disabled because of missing SELinux policies.
Frederick Mayle1ff7b842024-07-12 16:20:09 -07001054 command.arg("--block").arg(format!(
Frederick Mayleeae6d2d2024-09-12 11:05:36 -07001055 "path={},ro={},lock=false",
Jiyong Park69001a22024-08-16 20:45:34 +09001056 add_preserved_fd(&mut preserved_fds, disk.image),
Frederick Mayle1ff7b842024-07-12 16:20:09 -07001057 !disk.writable,
1058 ));
Andrew Walbrand6dce6f2021-03-05 16:39:08 +00001059 }
Andrew Walbran3a5a9212021-05-04 17:09:08 +00001060
Jiyong Park69001a22024-08-16 20:45:34 +09001061 if let Some(kernel) = config.kernel {
Andrew Walbran02b8ec02021-06-22 13:07:02 +00001062 command.arg(add_preserved_fd(&mut preserved_fds, kernel));
Andrew Walbrand6dce6f2021-03-05 16:39:08 +00001063 }
Andrew Walbran3a5a9212021-05-04 17:09:08 +00001064
Elie Kheirallah35dbb8e2024-10-23 00:15:14 +00001065 #[cfg(target_arch = "aarch64")]
1066 command.arg("--no-pmu");
1067
Frederick Maylee0ed2352024-09-12 16:16:00 -07001068 let control_sock = create_crosvm_control_listener(crosvm_control_socket_path)
1069 .context("failed to create control listener")?;
Jiyong Parkcd0f74b2024-08-20 11:06:52 +09001070 command.arg("--socket").arg(add_preserved_fd(&mut preserved_fds, control_sock));
Keir Fraser13a956a2022-07-14 14:20:46 +00001071
Jiyong Park69001a22024-08-16 20:45:34 +09001072 if let Some(dt_overlay) = config.device_tree_overlay {
Shikha Panwar55e10ec2024-02-13 12:53:49 +00001073 command.arg("--device-tree-overlay").arg(add_preserved_fd(&mut preserved_fds, dt_overlay));
Seungjae Yoo4dd38182023-11-09 15:05:45 +09001074 }
Jeongik Cha798401c2024-04-11 21:54:49 +09001075
1076 if cfg!(paravirtualized_devices) {
Jason Macnakea7b0fa2024-05-08 17:02:44 -07001077 if let Some(gpu_config) = &config.gpu_config {
1078 let mut gpu_args = Vec::new();
1079 if let Some(backend) = &gpu_config.backend {
1080 gpu_args.push(format!("backend={}", backend));
1081 }
1082 if let Some(context_types) = &gpu_config.context_types {
1083 gpu_args.push(format!("context-types={}", context_types.join(":")));
1084 }
1085 if let Some(pci_address) = &gpu_config.pci_address {
1086 gpu_args.push(format!("pci-address={}", pci_address));
1087 }
1088 if let Some(renderer_features) = &gpu_config.renderer_features {
1089 gpu_args.push(format!("renderer-features={}", renderer_features));
1090 }
1091 if gpu_config.renderer_use_egl.unwrap_or(false) {
1092 gpu_args.push("egl=true".to_string());
1093 }
1094 if gpu_config.renderer_use_gles.unwrap_or(false) {
1095 gpu_args.push("gles=true".to_string());
1096 }
1097 if gpu_config.renderer_use_glx.unwrap_or(false) {
1098 gpu_args.push("glx=true".to_string());
1099 }
1100 if gpu_config.renderer_use_surfaceless.unwrap_or(false) {
1101 gpu_args.push("surfaceless=true".to_string());
1102 }
1103 if gpu_config.renderer_use_vulkan.unwrap_or(false) {
1104 gpu_args.push("vulkan=true".to_string());
1105 }
1106 command.arg(format!("--gpu={}", gpu_args.join(",")));
1107 }
Jeongik Cha798401c2024-04-11 21:54:49 +09001108 if let Some(display_config) = &config.display_config {
Jason Macnakea7b0fa2024-05-08 17:02:44 -07001109 command
1110 .arg(format!(
1111 "--gpu-display=mode=windowed[{},{}],dpi=[{},{}],refresh-rate={}",
1112 display_config.width,
1113 display_config.height,
1114 display_config.horizontal_dpi,
1115 display_config.vertical_dpi,
1116 display_config.refresh_rate
1117 ))
1118 .arg(format!("--android-display-service={}", config.name));
Jeongik Cha798401c2024-04-11 21:54:49 +09001119 }
Jeongik Cha1df1c032024-04-03 16:03:12 +09001120 }
Seungjae Yooec3bc522023-11-09 10:14:30 +09001121
Seungjae Yoo986d7a42024-05-28 14:06:58 +09001122 if cfg!(network) {
Jiyong Park69001a22024-08-16 20:45:34 +09001123 if let Some(tap) = config.tap {
Andreas Huber487810d2024-08-22 07:43:01 -07001124 add_preserved_fd(&mut preserved_fds, tap);
1125 let tap_fd = preserved_fds.last().unwrap().as_raw_fd();
1126 command.arg("--net").arg(format!("tap-fd={tap_fd}"));
Seungjae Yoo986d7a42024-05-28 14:06:58 +09001127 }
1128 }
1129
Jeongik Cha6be4f8b2024-04-16 01:13:13 +09001130 if cfg!(paravirtualized_devices) {
Jiyong Park69001a22024-08-16 20:45:34 +09001131 for input_device_option in config.input_device_options.into_iter() {
Jeongik Chac181be72024-03-27 00:18:30 +09001132 command.arg("--input");
1133 command.arg(match input_device_option {
1134 InputDeviceOption::EvDev(file) => {
1135 format!("evdev[path={}]", add_preserved_fd(&mut preserved_fds, file))
1136 }
Jeongik Cha985b6402024-04-15 18:13:00 +09001137 InputDeviceOption::Keyboard(file) => {
1138 format!("keyboard[path={}]", add_preserved_fd(&mut preserved_fds, file))
1139 }
Jeongik Cha343894e2024-05-17 20:39:31 +09001140 InputDeviceOption::Mouse(file) => {
1141 format!("mouse[path={}]", add_preserved_fd(&mut preserved_fds, file))
1142 }
Jeongik Chac181be72024-03-27 00:18:30 +09001143 InputDeviceOption::SingleTouch { file, width, height, name } => format!(
1144 "single-touch[path={},width={},height={}{}]",
1145 add_preserved_fd(&mut preserved_fds, file),
1146 width,
1147 height,
1148 name.as_ref().map_or("".into(), |n| format!(",name={}", n))
1149 ),
Jiyong Park6c1b9f02024-07-05 16:21:10 +09001150 InputDeviceOption::Switches(file) => {
1151 format!("switches[path={}]", add_preserved_fd(&mut preserved_fds, file))
1152 }
Jeongik Cha112a2682024-07-09 12:28:45 +09001153 InputDeviceOption::MultiTouchTrackpad { file, width, height, name } => format!(
1154 "multi-touch-trackpad[path={},width={},height={}{}]",
1155 add_preserved_fd(&mut preserved_fds, file),
1156 width,
1157 height,
1158 name.as_ref().map_or("".into(), |n| format!(",name={}", n))
1159 ),
Jeongik Cha10494912024-07-16 11:19:50 +09001160 InputDeviceOption::MultiTouch { file, width, height, name } => format!(
1161 "multi-touch[path={},width={},height={}{}]",
1162 add_preserved_fd(&mut preserved_fds, file),
1163 width,
1164 height,
1165 name.as_ref().map_or("".into(), |n| format!(",name={}", n))
1166 ),
Jeongik Chac181be72024-03-27 00:18:30 +09001167 });
1168 }
1169 }
Vincent Donnefort538a2c62024-03-20 16:01:10 +00001170
1171 if config.hugepages {
1172 command.arg("--hugepages");
1173 }
1174
David Dai23cff712024-06-13 19:23:45 +00001175 if config.boost_uclamp {
1176 command.arg("--boost-uclamp");
1177 }
1178
Jiyong Park69001a22024-08-16 20:45:34 +09001179 if !config.vfio_devices.is_empty() {
1180 if let Some(dtbo) = config.dtbo {
1181 command.arg(format!(
1182 "--device-tree-overlay={},filter",
1183 add_preserved_fd(&mut preserved_fds, dtbo)
1184 ));
1185 } else {
1186 bail!("VFIO devices assigned but no DTBO available");
1187 }
1188 };
1189 for device in config.vfio_devices {
1190 command.arg(vfio_argument_for_platform_device(&device)?);
1191 }
Inseob Kim6ef80972023-07-20 17:23:36 +09001192
Andrew Walbran02b8ec02021-06-22 13:07:02 +00001193 debug!("Preserving FDs {:?}", preserved_fds);
1194 command.preserved_fds(preserved_fds);
Andrew Walbran3a5a9212021-05-04 17:09:08 +00001195
Mu-Le Lee134180a2024-04-16 15:13:07 +08001196 if cfg!(paravirtualized_devices) {
Mu-Le Leeb2d5a312024-06-05 10:52:47 +08001197 if let Some(audio_config) = &config.audio_config {
1198 command.arg("--virtio-snd").arg(format!(
1199 "backend=aaudio,num_input_devices={},num_output_devices={}",
1200 if audio_config.use_microphone { 1 } else { 0 },
1201 if audio_config.use_speaker { 1 } else { 0 }
1202 ));
Mu-Le Lee134180a2024-04-16 15:13:07 +08001203 }
1204 }
1205
Jiyong Park2d736562022-10-24 22:40:12 +09001206 print_crosvm_args(&command);
1207
Andrew Walbran3a5a9212021-05-04 17:09:08 +00001208 let result = SharedChild::spawn(&mut command)?;
Jooyung Hanbfe086f2021-10-28 10:15:45 +09001209 debug!("Spawned crosvm({}).", result.id());
Andrew Walbran3a5a9212021-05-04 17:09:08 +00001210 Ok(result)
1211}
1212
1213/// Ensure that the configuration has a valid combination of fields set, or return an error if not.
Andrew Walbranf5fbb7d2021-05-12 17:15:48 +00001214fn validate_config(config: &CrosvmConfig) -> Result<(), Error> {
Andrew Walbran3a5a9212021-05-04 17:09:08 +00001215 if config.bootloader.is_none() && config.kernel.is_none() {
1216 bail!("VM must have either a bootloader or a kernel image.");
1217 }
1218 if config.bootloader.is_some() && (config.kernel.is_some() || config.initrd.is_some()) {
1219 bail!("Can't have both bootloader and kernel/initrd image.");
1220 }
Jiyong Parkdcf17412022-02-08 15:07:23 +09001221 let version = Version::parse(CROSVM_PLATFORM_VERSION).unwrap();
1222 if !config.platform_version.matches(&version) {
1223 bail!(
1224 "Incompatible platform version. The config is compatible with platform version(s) \
1225 {}, but the actual platform version is {}",
1226 config.platform_version,
1227 version
1228 );
1229 }
1230
Andrew Walbran3a5a9212021-05-04 17:09:08 +00001231 Ok(())
1232}
1233
Jiyong Park2d736562022-10-24 22:40:12 +09001234/// Print arguments of the crosvm command. In doing so, /proc/self/fd/XX is annotated with the
1235/// actual file path if the FD is backed by a regular file. If not, the /proc path is printed
1236/// unmodified.
1237fn print_crosvm_args(command: &Command) {
1238 let re = Regex::new(r"/proc/self/fd/[\d]+").unwrap();
1239 info!(
1240 "Running crosvm with args: {:?}",
1241 command
1242 .get_args()
1243 .map(|s| s.to_string_lossy())
1244 .map(|s| {
1245 re.replace_all(&s, |caps: &Captures| {
1246 let path = &caps[0];
1247 if let Ok(realpath) = std::fs::canonicalize(path) {
1248 format!("{} ({})", path, realpath.to_string_lossy())
1249 } else {
1250 path.to_owned()
1251 }
1252 })
1253 .into_owned()
1254 })
1255 .collect::<Vec<_>>()
1256 );
1257}
1258
Andrew Walbran02b8ec02021-06-22 13:07:02 +00001259/// Adds the file descriptor for `file` to `preserved_fds`, and returns a string of the form
1260/// "/proc/self/fd/N" where N is the file descriptor.
Jiyong Park69001a22024-08-16 20:45:34 +09001261fn add_preserved_fd<F: Into<OwnedFd>>(preserved_fds: &mut Vec<OwnedFd>, file: F) -> String {
1262 let fd = file.into();
1263 let raw_fd = fd.as_raw_fd();
Andrew Walbran02b8ec02021-06-22 13:07:02 +00001264 preserved_fds.push(fd);
Jiyong Park69001a22024-08-16 20:45:34 +09001265 format!("/proc/self/fd/{}", raw_fd)
Andrew Walbrand6dce6f2021-03-05 16:39:08 +00001266}
Andrew Walbranb27681f2022-02-23 15:11:52 +00001267
1268/// Adds the file descriptor for `file` (if any) to `preserved_fds`, and returns the appropriate
1269/// string for a crosvm `--serial` flag. If `file` is none, creates a dummy sink device.
Jiyong Park69001a22024-08-16 20:45:34 +09001270fn format_serial_out_arg(preserved_fds: &mut Vec<OwnedFd>, file: Option<File>) -> String {
Andrew Walbranb27681f2022-02-23 15:11:52 +00001271 if let Some(file) = file {
1272 format!("type=file,path={}", add_preserved_fd(preserved_fds, file))
1273 } else {
1274 "type=sink".to_string()
1275 }
1276}
1277
1278/// Creates a new pipe with the `O_CLOEXEC` flag set, and returns the read side and write side.
1279fn create_pipe() -> Result<(File, File), Error> {
Frederick Maylefbbcfcd2024-04-08 16:31:54 -07001280 let (read_fd, write_fd) = pipe2(OFlag::O_CLOEXEC)?;
1281 Ok((read_fd.into(), write_fd.into()))
Andrew Walbranb27681f2022-02-23 15:11:52 +00001282}
Frederick Maylee0ed2352024-09-12 16:16:00 -07001283
1284/// Creates and binds a unix seqpacket listening socket to be passed as crosvm's `--socket`
1285/// argument. See `UnixSeqpacketListener::bind` in crosvm's code for reference.
1286fn create_crosvm_control_listener(crosvm_control_socket_path: &Path) -> Result<OwnedFd> {
1287 use nix::sys::socket;
1288 let fd = socket::socket(
1289 socket::AddressFamily::Unix,
1290 socket::SockType::SeqPacket,
1291 socket::SockFlag::empty(),
1292 None,
1293 )
1294 .context("socket failed")?;
1295 socket::bind(fd.as_raw_fd(), &socket::UnixAddr::new(crosvm_control_socket_path)?)
1296 .context("bind failed")?;
1297 // The exact backlog size isn't imporant. crosvm uses 128 internally. We use 127 here
1298 // because of a `nix` bug.
1299 socket::listen(&fd, socket::Backlog::new(127).unwrap()).context("listen failed")?;
1300 Ok(fd)
1301}