blob: fcc5d3b558c016295158e49f7b22d0b810e36688 [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>,
Akilesh Kailash05515dc2024-10-22 21:42:52 -0700111 pub shared_paths: Vec<SharedPathConfig>,
Andrew Walbranf5fbb7d2021-05-12 17:15:48 +0000112 pub params: Option<String>,
Andrew Walbranf8650422021-06-09 15:54:09 +0000113 pub protected: bool,
Jaewan Kim61f86142023-03-28 15:12:52 +0900114 pub debug_config: DebugConfig,
Frederick Mayle68d06e72024-07-12 17:18:11 -0700115 pub memory_mib: NonZeroU32,
Jiyong Park032615f2022-01-10 13:55:34 +0900116 pub cpus: Option<NonZeroU32>,
David Brazdil7d1e5ec2023-02-06 17:56:29 +0000117 pub host_cpu_topology: bool,
Jiyong Parke6fb1672023-06-26 16:45:55 +0900118 pub console_out_fd: Option<File>,
119 pub console_in_fd: Option<File>,
Andrew Walbrand3a84182021-09-07 14:48:52 +0000120 pub log_fd: Option<File>,
Jiyong Parke558ab12022-07-07 20:18:55 +0900121 pub ramdump: Option<File>,
Andrew Walbrand3a84182021-09-07 14:48:52 +0000122 pub indirect_files: Vec<File>,
Jiyong Parkdcf17412022-02-08 15:07:23 +0900123 pub platform_version: VersionReq,
Jiyong Parke6ed0f92022-06-22 00:13:00 +0900124 pub detect_hangup: bool,
Nikita Ioffe5776f082023-02-10 21:38:26 +0000125 pub gdb_port: Option<NonZeroU16>,
Inseob Kim7307a892023-09-14 13:37:58 +0900126 pub vfio_devices: Vec<VfioDevice>,
David Brazdil2dfefd12023-11-17 14:07:36 +0000127 pub dtbo: Option<File>,
Shikha Panwar55e10ec2024-02-13 12:53:49 +0000128 pub device_tree_overlay: Option<File>,
Jeongik Cha1df1c032024-04-03 16:03:12 +0900129 pub display_config: Option<DisplayConfig>,
Jeongik Chac181be72024-03-27 00:18:30 +0900130 pub input_device_options: Vec<InputDeviceOption>,
Vincent Donnefort538a2c62024-03-20 16:01:10 +0000131 pub hugepages: bool,
Seungjae Yoo986d7a42024-05-28 14:06:58 +0900132 pub tap: Option<File>,
Yi-Yo Chiang8dd32552024-05-22 19:38:16 +0800133 pub console_input_device: Option<String>,
David Dai23cff712024-06-13 19:23:45 +0000134 pub boost_uclamp: bool,
Jason Macnakea7b0fa2024-05-08 17:02:44 -0700135 pub gpu_config: Option<GpuConfig>,
Mu-Le Leeb2d5a312024-06-05 10:52:47 +0800136 pub audio_config: Option<AudioConfig>,
Jeongik Cha75057452024-09-03 11:26:04 +0900137 pub no_balloon: bool,
Elie Kheirallah29d72912024-08-07 20:17:26 +0000138 pub usb_config: UsbConfig,
Elie Kheirallah5c807a22024-09-23 20:40:42 +0000139 pub dump_dt_fd: Option<File>,
Mu-Le Leeb2d5a312024-06-05 10:52:47 +0800140}
141
142#[derive(Debug)]
143pub struct AudioConfig {
144 pub use_microphone: bool,
145 pub use_speaker: bool,
146}
147
148impl AudioConfig {
149 pub fn new(raw_config: &AudioConfigParcelable) -> Self {
150 AudioConfig { use_microphone: raw_config.useMicrophone, use_speaker: raw_config.useSpeaker }
151 }
Jeongik Cha1df1c032024-04-03 16:03:12 +0900152}
153
154#[derive(Debug)]
Elie Kheirallah29d72912024-08-07 20:17:26 +0000155pub struct UsbConfig {
156 pub controller: bool,
157}
158
159impl UsbConfig {
160 pub fn new(raw_config: &UsbConfigParcelable) -> Result<UsbConfig> {
161 Ok(UsbConfig { controller: raw_config.controller })
162 }
163}
164
165#[derive(Debug)]
Jeongik Cha1df1c032024-04-03 16:03:12 +0900166pub struct DisplayConfig {
167 pub width: NonZeroU32,
168 pub height: NonZeroU32,
169 pub horizontal_dpi: NonZeroU32,
170 pub vertical_dpi: NonZeroU32,
171 pub refresh_rate: NonZeroU32,
172}
173
174impl DisplayConfig {
175 pub fn new(raw_config: &DisplayConfigParcelable) -> Result<DisplayConfig> {
176 let width = try_into_non_zero_u32(raw_config.width)?;
177 let height = try_into_non_zero_u32(raw_config.height)?;
178 let horizontal_dpi = try_into_non_zero_u32(raw_config.horizontalDpi)?;
179 let vertical_dpi = try_into_non_zero_u32(raw_config.verticalDpi)?;
180 let refresh_rate = try_into_non_zero_u32(raw_config.refreshRate)?;
181 Ok(DisplayConfig { width, height, horizontal_dpi, vertical_dpi, refresh_rate })
182 }
183}
184
Jason Macnakea7b0fa2024-05-08 17:02:44 -0700185#[derive(Debug)]
186pub struct GpuConfig {
187 pub backend: Option<String>,
188 pub context_types: Option<Vec<String>>,
189 pub pci_address: Option<String>,
190 pub renderer_features: Option<String>,
191 pub renderer_use_egl: Option<bool>,
192 pub renderer_use_gles: Option<bool>,
193 pub renderer_use_glx: Option<bool>,
194 pub renderer_use_surfaceless: Option<bool>,
195 pub renderer_use_vulkan: Option<bool>,
196}
197
198impl GpuConfig {
199 pub fn new(raw_config: &GpuConfigParcelable) -> Result<GpuConfig> {
200 Ok(GpuConfig {
201 backend: raw_config.backend.clone(),
202 context_types: raw_config.contextTypes.clone().map(|context_types| {
203 context_types.iter().filter_map(|context_type| context_type.clone()).collect()
204 }),
205 pci_address: raw_config.pciAddress.clone(),
206 renderer_features: raw_config.rendererFeatures.clone(),
207 renderer_use_egl: Some(raw_config.rendererUseEgl),
208 renderer_use_gles: Some(raw_config.rendererUseGles),
209 renderer_use_glx: Some(raw_config.rendererUseGlx),
210 renderer_use_surfaceless: Some(raw_config.rendererUseSurfaceless),
211 renderer_use_vulkan: Some(raw_config.rendererUseVulkan),
212 })
213 }
214}
215
Jeongik Cha1df1c032024-04-03 16:03:12 +0900216fn try_into_non_zero_u32(value: i32) -> Result<NonZeroU32> {
217 let u32_value = value.try_into()?;
218 NonZeroU32::new(u32_value).ok_or(anyhow!("value should be greater than 0"))
Andrew Walbranf5fbb7d2021-05-12 17:15:48 +0000219}
220
221/// A disk image to pass to crosvm for a VM.
222#[derive(Debug)]
223pub struct DiskFile {
224 pub image: File,
225 pub writable: bool,
226}
227
Akilesh Kailash05515dc2024-10-22 21:42:52 -0700228/// Shared path between host and guest VM.
229#[derive(Debug)]
230pub struct SharedPathConfig {
231 pub path: String,
232 pub host_uid: i32,
233 pub host_gid: i32,
234 pub guest_uid: i32,
235 pub guest_gid: i32,
236 pub mask: i32,
237 pub tag: String,
238 pub socket_path: String,
239}
240
Jeongik Chac181be72024-03-27 00:18:30 +0900241/// virtio-input device configuration from `external/crosvm/src/crosvm/config.rs`
242#[derive(Debug)]
243#[allow(dead_code)]
244pub enum InputDeviceOption {
245 EvDev(File),
246 SingleTouch { file: File, width: u32, height: u32, name: Option<String> },
Jeongik Cha985b6402024-04-15 18:13:00 +0900247 Keyboard(File),
Jeongik Cha343894e2024-05-17 20:39:31 +0900248 Mouse(File),
Jiyong Park6c1b9f02024-07-05 16:21:10 +0900249 Switches(File),
Jeongik Cha112a2682024-07-09 12:28:45 +0900250 MultiTouchTrackpad { file: File, width: u32, height: u32, name: Option<String> },
Jeongik Cha10494912024-07-16 11:19:50 +0900251 MultiTouch { file: File, width: u32, height: u32, name: Option<String> },
Jeongik Chac181be72024-03-27 00:18:30 +0900252}
253
Jakob Vukalovicd42aa2c2023-11-09 16:04:00 +0000254type VfioDevice = Strong<dyn IBoundDevice>;
Inseob Kim7307a892023-09-14 13:37:58 +0900255
Andrew Walbran6b650662021-09-07 13:13:23 +0000256/// The lifecycle state which the payload in the VM has reported itself to be in.
257///
258/// Note that the order of enum variants is significant; only forward transitions are allowed by
259/// [`VmInstance::update_payload_state`].
260#[derive(Copy, Clone, Debug, Eq, Ord, PartialEq, PartialOrd)]
261pub enum PayloadState {
262 Starting,
263 Started,
264 Ready,
265 Finished,
Jiyong Parka4eebde2022-07-12 18:01:12 +0900266 Hangup, // Hasn't reached to Ready before timeout expires
Andrew Walbran6b650662021-09-07 13:13:23 +0000267}
268
Andrew Walbranf8d94112021-09-07 11:45:36 +0000269/// The current state of the VM itself.
270#[derive(Debug)]
271pub enum VmState {
272 /// The VM has not yet tried to start.
273 NotStarted {
274 ///The configuration needed to start the VM, if it has not yet been started.
Inseob Kim7307a892023-09-14 13:37:58 +0900275 config: Box<CrosvmConfig>,
Andrew Walbranf8d94112021-09-07 11:45:36 +0000276 },
277 /// The VM has been started.
278 Running {
279 /// The crosvm child process.
280 child: Arc<SharedChild>,
David Brazdilf18e1162022-12-18 18:27:23 +0000281 /// The thread waiting for crosvm to finish.
282 monitor_vm_exit_thread: Option<JoinHandle<()>>,
Andrew Walbranf8d94112021-09-07 11:45:36 +0000283 },
284 /// The VM died or was killed.
285 Dead,
286 /// The VM failed to start.
287 Failed,
288}
289
Seungjae Yoo6d265d92022-11-15 10:51:33 +0900290/// RSS values of VM and CrosVM process itself.
291#[derive(Copy, Clone, Debug, Default)]
292pub struct Rss {
293 pub vm: i64,
294 pub crosvm: i64,
295}
296
297/// Metrics regarding the VM.
298#[derive(Debug, Default)]
299pub struct VmMetric {
300 /// Recorded timestamp when the VM is started.
301 pub start_timestamp: Option<SystemTime>,
Pierre-Clément Tosi44aba642023-09-15 11:44:34 +0100302 /// Update most recent guest_time periodically from /proc/[crosvm pid]/stat while VM is
303 /// running.
Seungjae Yoo6d265d92022-11-15 10:51:33 +0900304 pub cpu_guest_time: Option<i64>,
305 /// Update maximum RSS values periodically from /proc/[crosvm pid]/smaps while VM is running.
306 pub rss: Option<Rss>,
307}
308
Andrew Walbranf8d94112021-09-07 11:45:36 +0000309impl VmState {
310 /// Tries to start the VM, if it is in the `NotStarted` state.
311 ///
312 /// Returns an error if the VM is in the wrong state, or fails to start.
313 fn start(&mut self, instance: Arc<VmInstance>) -> Result<(), Error> {
314 let state = mem::replace(self, VmState::Failed);
315 if let VmState::NotStarted { config } = state {
Inseob Kim7307a892023-09-14 13:37:58 +0900316 let config = *config;
Jiyong Parke6ed0f92022-06-22 00:13:00 +0900317 let detect_hangup = config.detect_hangup;
Andrew Walbranb27681f2022-02-23 15:11:52 +0000318 let (failure_pipe_read, failure_pipe_write) = create_pipe()?;
Inseob Kim6ef80972023-07-20 17:23:36 +0900319 let vfio_devices = config.vfio_devices.clone();
Seungjae Yoo3271f172024-06-11 10:14:07 +0900320 let tap =
321 if let Some(tap_file) = &config.tap { Some(tap_file.try_clone()?) } else { None };
Andrew Walbranb27681f2022-02-23 15:11:52 +0000322
Akilesh Kailash05515dc2024-10-22 21:42:52 -0700323 run_virtiofs(&config)?;
324
Andrew Walbranf8d94112021-09-07 11:45:36 +0000325 // If this fails and returns an error, `self` will be left in the `Failed` state.
Keir Fraser13a956a2022-07-14 14:20:46 +0000326 let child =
Keir Frasercdd4b112022-11-24 14:02:25 +0000327 Arc::new(run_vm(config, &instance.crosvm_control_socket_path, failure_pipe_write)?);
Andrew Walbranf8d94112021-09-07 11:45:36 +0000328
Seungjae Yoo6d265d92022-11-15 10:51:33 +0900329 let instance_monitor_status = instance.clone();
330 let child_monitor_status = child.clone();
331 thread::spawn(move || {
332 instance_monitor_status.clone().monitor_vm_status(child_monitor_status);
333 });
334
Andrew Walbranf8d94112021-09-07 11:45:36 +0000335 let child_clone = child.clone();
Jiyong Parka4eebde2022-07-12 18:01:12 +0900336 let instance_clone = instance.clone();
David Brazdilf18e1162022-12-18 18:27:23 +0000337 let monitor_vm_exit_thread = Some(thread::spawn(move || {
Seungjae Yoo3271f172024-06-11 10:14:07 +0900338 instance_clone.monitor_vm_exit(child_clone, failure_pipe_read, vfio_devices, tap);
David Brazdilf18e1162022-12-18 18:27:23 +0000339 }));
Andrew Walbranf8d94112021-09-07 11:45:36 +0000340
Jiyong Parka4eebde2022-07-12 18:01:12 +0900341 if detect_hangup {
342 let child_clone = child.clone();
343 thread::spawn(move || {
344 instance.monitor_payload_hangup(child_clone);
345 });
346 }
347
Andrew Walbranf8d94112021-09-07 11:45:36 +0000348 // If it started correctly, update the state.
David Brazdilf18e1162022-12-18 18:27:23 +0000349 *self = VmState::Running { child, monitor_vm_exit_thread };
Andrew Walbranf8d94112021-09-07 11:45:36 +0000350 Ok(())
351 } else {
352 *self = state;
353 bail!("VM already started or failed")
354 }
355 }
356}
357
David Brazdil8cf8f482022-11-23 14:21:26 +0000358/// Internal struct that holds the handles to globally unique resources of a VM.
359#[derive(Debug)]
360pub struct VmContext {
361 #[allow(dead_code)] // Keeps the global context alive
Yi-Yo Chiang2fbf0da2024-06-14 22:56:56 +0800362 pub(crate) global_context: Strong<dyn IGlobalVmContext>,
David Brazdil8cf8f482022-11-23 14:21:26 +0000363 #[allow(dead_code)] // Keeps the server alive
364 vm_server: RpcServer,
365}
366
367impl VmContext {
368 /// Construct new VmContext.
369 pub fn new(global_context: Strong<dyn IGlobalVmContext>, vm_server: RpcServer) -> VmContext {
370 VmContext { global_context, vm_server }
371 }
372}
373
Andrew Walbranf8d94112021-09-07 11:45:36 +0000374/// Information about a particular instance of a VM which may be running.
Andrew Walbrand6dce6f2021-03-05 16:39:08 +0000375#[derive(Debug)]
376pub struct VmInstance {
Andrew Walbranf8d94112021-09-07 11:45:36 +0000377 /// The current state of the VM.
378 pub vm_state: Mutex<VmState>,
David Brazdil8cf8f482022-11-23 14:21:26 +0000379 /// Global resources allocated for this VM.
380 #[allow(dead_code)] // Keeps the context alive
Yi-Yo Chiang2fbf0da2024-06-14 22:56:56 +0800381 pub(crate) vm_context: VmContext,
Andrew Walbrand6dce6f2021-03-05 16:39:08 +0000382 /// The CID assigned to the VM for vsock communication.
383 pub cid: Cid,
Keir Frasercdd4b112022-11-24 14:02:25 +0000384 /// Path to crosvm control socket
385 crosvm_control_socket_path: PathBuf,
Seungjae Yoo62085c02022-08-12 04:44:52 +0000386 /// The name of the VM.
387 pub name: String,
Andrew Walbranf8650422021-06-09 15:54:09 +0000388 /// Whether the VM is a protected VM.
389 pub protected: bool,
Andrew Walbranf5fbb7d2021-05-12 17:15:48 +0000390 /// Directory of temporary files used by the VM while it is running.
391 pub temporary_directory: PathBuf,
Andrew Walbranf6a1eb92021-04-01 11:16:02 +0000392 /// The UID of the process which requested the VM.
393 pub requester_uid: u32,
Andrew Walbranf6a1eb92021-04-01 11:16:02 +0000394 /// The PID of the process which requested the VM. Note that this process may no longer exist
395 /// and the PID may have been reused for a different process, so this should not be trusted.
Andrew Walbran02034492021-04-13 15:05:07 +0000396 pub requester_debug_pid: i32,
Andrew Walbrandae07162021-03-12 17:05:20 +0000397 /// Callbacks to clients of the VM.
398 pub callbacks: VirtualMachineCallbacks,
Inseob Kimc7d28c72021-10-25 14:28:10 +0000399 /// VirtualMachineService binder object for the VM.
Stephen Hinesaf21e512024-07-30 15:43:39 -0700400 #[allow(dead_code)]
Inseob Kimc7d28c72021-10-25 14:28:10 +0000401 pub vm_service: Mutex<Option<Strong<dyn IVirtualMachineService>>>,
Seungjae Yoo6d265d92022-11-15 10:51:33 +0900402 /// Recorded metrics of VM such as timestamp or cpu / memory usage.
403 pub vm_metric: Mutex<VmMetric>,
Andrew Walbran6b650662021-09-07 13:13:23 +0000404 /// The latest lifecycle state which the payload reported itself to be in.
405 payload_state: Mutex<PayloadState>,
Jiyong Parka4eebde2022-07-12 18:01:12 +0900406 /// Represents the condition that payload_state was updated
407 payload_state_updated: Condvar,
Jiyong Park08eee7b2022-11-11 15:01:26 +0900408 /// The human readable name of requester_uid
409 requester_uid_name: String,
410}
411
412impl fmt::Display for VmInstance {
413 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
414 let adj = if self.protected { "Protected" } else { "Non-protected" };
415 write!(
416 f,
417 "{} virtual machine \"{}\" (owner: {}, cid: {})",
418 adj, self.name, self.requester_uid_name, self.cid
419 )
420 }
Andrew Walbrand6dce6f2021-03-05 16:39:08 +0000421}
422
423impl VmInstance {
Andrew Walbranf8d94112021-09-07 11:45:36 +0000424 /// Validates the given config and creates a new `VmInstance` but doesn't start running it.
425 pub fn new(
Andrew Walbrand3a84182021-09-07 14:48:52 +0000426 config: CrosvmConfig,
Andrew Walbranf5fbb7d2021-05-12 17:15:48 +0000427 temporary_directory: PathBuf,
Andrew Walbranf6a1eb92021-04-01 11:16:02 +0000428 requester_uid: u32,
Andrew Walbran02034492021-04-13 15:05:07 +0000429 requester_debug_pid: i32,
David Brazdil8cf8f482022-11-23 14:21:26 +0000430 vm_context: VmContext,
Andrew Walbranf8d94112021-09-07 11:45:36 +0000431 ) -> Result<VmInstance, Error> {
432 validate_config(&config)?;
Andrew Walbrand3a84182021-09-07 14:48:52 +0000433 let cid = config.cid;
Seungjae Yoo62085c02022-08-12 04:44:52 +0000434 let name = config.name.clone();
Andrew Walbrand3a84182021-09-07 14:48:52 +0000435 let protected = config.protected;
Jiyong Park08eee7b2022-11-11 15:01:26 +0900436 let requester_uid_name = User::from_uid(Uid::from_raw(requester_uid))
437 .ok()
438 .flatten()
439 .map_or_else(|| format!("{}", requester_uid), |u| u.name);
440 let instance = VmInstance {
Inseob Kim7307a892023-09-14 13:37:58 +0900441 vm_state: Mutex::new(VmState::NotStarted { config: Box::new(config) }),
David Brazdil528e0472022-10-10 15:06:02 +0100442 vm_context,
Andrew Walbrand3a84182021-09-07 14:48:52 +0000443 cid,
Keir Frasercdd4b112022-11-24 14:02:25 +0000444 crosvm_control_socket_path: temporary_directory.join("crosvm.sock"),
Seungjae Yoo62085c02022-08-12 04:44:52 +0000445 name,
Andrew Walbrand3a84182021-09-07 14:48:52 +0000446 protected,
Andrew Walbranf5fbb7d2021-05-12 17:15:48 +0000447 temporary_directory,
Andrew Walbran02034492021-04-13 15:05:07 +0000448 requester_uid,
Andrew Walbran02034492021-04-13 15:05:07 +0000449 requester_debug_pid,
Andrew Walbranf8d94112021-09-07 11:45:36 +0000450 callbacks: Default::default(),
Inseob Kimc7d28c72021-10-25 14:28:10 +0000451 vm_service: Mutex::new(None),
Seungjae Yoo6d265d92022-11-15 10:51:33 +0900452 vm_metric: Mutex::new(Default::default()),
Andrew Walbranf8d94112021-09-07 11:45:36 +0000453 payload_state: Mutex::new(PayloadState::Starting),
Jiyong Parka4eebde2022-07-12 18:01:12 +0900454 payload_state_updated: Condvar::new(),
Jiyong Park08eee7b2022-11-11 15:01:26 +0900455 requester_uid_name,
456 };
457 info!("{} created", &instance);
458 Ok(instance)
Andrew Walbrandae07162021-03-12 17:05:20 +0000459 }
460
Andrew Walbranf8d94112021-09-07 11:45:36 +0000461 /// Starts an instance of `crosvm` to manage the VM. The `crosvm` instance will be killed when
462 /// the `VmInstance` is dropped.
463 pub fn start(self: &Arc<Self>) -> Result<(), Error> {
Seungjae Yoo6d265d92022-11-15 10:51:33 +0900464 let mut vm_metric = self.vm_metric.lock().unwrap();
465 vm_metric.start_timestamp = Some(SystemTime::now());
Jiyong Park08eee7b2022-11-11 15:01:26 +0900466 let ret = self.vm_state.lock().unwrap().start(self.clone());
467 if ret.is_ok() {
468 info!("{} started", &self);
469 }
470 ret.with_context(|| format!("{} failed to start", &self))
Andrew Walbranf8d94112021-09-07 11:45:36 +0000471 }
472
Jiyong Parka4eebde2022-07-12 18:01:12 +0900473 /// Monitors the exit of the VM (i.e. termination of the `child` process). When that happens,
474 /// handles the event by updating the state, noityfing the event to clients by calling
475 /// callbacks, and removing temporary files for the VM.
Inseob Kim6ef80972023-07-20 17:23:36 +0900476 fn monitor_vm_exit(
477 &self,
478 child: Arc<SharedChild>,
479 mut failure_pipe_read: File,
Inseob Kim7307a892023-09-14 13:37:58 +0900480 vfio_devices: Vec<VfioDevice>,
Seungjae Yoo3271f172024-06-11 10:14:07 +0900481 tap: Option<File>,
Inseob Kim6ef80972023-07-20 17:23:36 +0900482 ) {
Andrew Walbranc92d35f2022-01-12 12:45:19 +0000483 let result = child.wait();
484 match &result {
Jooyung Hanbfe086f2021-10-28 10:15:45 +0900485 Err(e) => error!("Error waiting for crosvm({}) instance to die: {}", child.id(), e),
Sebastian Ene23167d82022-10-07 14:09:53 +0000486 Ok(status) => {
487 info!("crosvm({}) exited with status {}", child.id(), status);
488 if let Some(exit_status_code) = status.code() {
489 if exit_status_code == CROSVM_WATCHDOG_REBOOT_STATUS {
490 info!("detected vcpu stall on crosvm");
491 }
492 }
493 }
Andrew Walbrandae07162021-03-12 17:05:20 +0000494 }
Andrew Walbranf8d94112021-09-07 11:45:36 +0000495
496 let mut vm_state = self.vm_state.lock().unwrap();
497 *vm_state = VmState::Dead;
498 // Ensure that the mutex is released before calling the callbacks.
499 drop(vm_state);
Jiyong Park08eee7b2022-11-11 15:01:26 +0900500 info!("{} exited", &self);
Andrew Walbranf8d94112021-09-07 11:45:36 +0000501
Jiyong Parka4eebde2022-07-12 18:01:12 +0900502 // Read the pipe to see if any failure reason is written
503 let mut failure_reason = String::new();
504 match failure_pipe_read.read_to_string(&mut failure_reason) {
505 Err(e) => error!("Error reading VM failure reason from pipe: {}", e),
506 Ok(len) if len > 0 => info!("VM returned failure reason '{}'", &failure_reason),
507 _ => (),
Jiyong Parke6ed0f92022-06-22 00:13:00 +0900508 };
Andrew Walbranb27681f2022-02-23 15:11:52 +0000509
Jiyong Parka4eebde2022-07-12 18:01:12 +0900510 // In case of hangup, the pipe doesn't give us any information because the hangup can't be
511 // detected on the VM side (otherwise, it isn't a hangup), but in the
512 // monitor_payload_hangup function below which updates the payload state to Hangup.
513 let failure_reason =
514 if failure_reason.is_empty() && self.payload_state() == PayloadState::Hangup {
515 Cow::from("HANGUP")
516 } else {
517 Cow::from(failure_reason)
518 };
519
Jiyong Parke558ab12022-07-07 20:18:55 +0900520 self.handle_ramdump().unwrap_or_else(|e| error!("Error handling ramdump: {}", e));
Seungjae Yoob4c07ba2022-08-12 04:44:52 +0000521
522 let death_reason = death_reason(&result, &failure_reason);
Seungjae Yoo93430e82022-12-05 16:37:42 +0900523 let exit_signal = exit_signal(&result);
524
Seungjae Yoob4c07ba2022-08-12 04:44:52 +0000525 self.callbacks.callback_on_died(self.cid, death_reason);
Seungjae Yoo2e7beea2022-08-24 16:09:12 +0900526
Seungjae Yoo6d265d92022-11-15 10:51:33 +0900527 let vm_metric = self.vm_metric.lock().unwrap();
Seungjae Yoob2db3d22023-04-06 18:05:43 +0900528 write_vm_exited_stats_sync(
Seungjae Yoo93430e82022-12-05 16:37:42 +0900529 self.requester_uid as i32,
530 &self.name,
531 death_reason,
532 exit_signal,
Chris Wailes75269622022-12-05 23:01:44 -0800533 &vm_metric,
Seungjae Yoo93430e82022-12-05 16:37:42 +0900534 );
Andrew Walbranf5fbb7d2021-05-12 17:15:48 +0000535
David Brazdil4b4c5102022-12-19 22:56:20 +0000536 // Delete temporary files. The folder itself is removed by VirtualizationServiceInternal.
537 remove_temporary_files(&self.temporary_directory).unwrap_or_else(|e| {
538 error!("Error removing temporary files from {:?}: {}", self.temporary_directory, e);
539 });
Inseob Kim6ef80972023-07-20 17:23:36 +0900540
Seungjae Yoo3271f172024-06-11 10:14:07 +0900541 if let Some(tap_file) = tap {
542 GLOBAL_SERVICE
543 .deleteTapInterface(&ParcelFileDescriptor::new(OwnedFd::from(tap_file)))
544 .unwrap_or_else(|e| {
545 error!("Error deleting TAP interface: {e:?}");
546 });
547 }
548
Jakob Vukalovicd42aa2c2023-11-09 16:04:00 +0000549 drop(vfio_devices); // Cleanup devices.
Andrew Walbrandae07162021-03-12 17:05:20 +0000550 }
551
Jiyong Parka4eebde2022-07-12 18:01:12 +0900552 /// Waits until payload is started, or timeout expires. When timeout occurs, kill
553 /// the VM to prevent indefinite hangup and update the payload_state accordingly.
554 fn monitor_payload_hangup(&self, child: Arc<SharedChild>) {
555 debug!("Starting to monitor hangup for Microdroid({})", child.id());
Jiyong Parkcf975612024-04-07 13:27:10 +0900556 let (state, result) = self
Jiyong Parka4eebde2022-07-12 18:01:12 +0900557 .payload_state_updated
558 .wait_timeout_while(self.payload_state.lock().unwrap(), *BOOT_HANGUP_TIMEOUT, |s| {
559 *s < PayloadState::Started
560 })
561 .unwrap();
Jiyong Parkcf975612024-04-07 13:27:10 +0900562 drop(state); // we are not interested in state
Jiyong Parka4eebde2022-07-12 18:01:12 +0900563 let child_still_running = child.try_wait().ok() == Some(None);
564 if result.timed_out() && child_still_running {
565 error!(
566 "Microdroid({}) failed to start payload within {} secs timeout. Shutting down.",
567 child.id(),
568 BOOT_HANGUP_TIMEOUT.as_secs()
569 );
570 self.update_payload_state(PayloadState::Hangup).unwrap();
571 if let Err(e) = self.kill() {
572 error!("Error stopping timed-out VM with CID {}: {:?}", child.id(), e);
573 }
574 }
575 }
576
Seungjae Yoo6d265d92022-11-15 10:51:33 +0900577 fn monitor_vm_status(&self, child: Arc<SharedChild>) {
578 let pid = child.id();
579
580 loop {
581 {
582 // Check VM state
583 let vm_state = &*self.vm_state.lock().unwrap();
584 if let VmState::Dead = vm_state {
585 break;
586 }
587
588 let mut vm_metric = self.vm_metric.lock().unwrap();
589
590 // Get CPU Information
Jiyong Park85ea5382024-02-19 15:04:23 +0900591 match get_guest_time(pid) {
592 Ok(guest_time) => vm_metric.cpu_guest_time = Some(guest_time),
593 Err(e) => error!("Failed to get guest CPU time: {e:?}"),
Seungjae Yoo6d265d92022-11-15 10:51:33 +0900594 }
595
596 // Get Memory Information
Jiyong Park85ea5382024-02-19 15:04:23 +0900597 match get_rss(pid) {
598 Ok(rss) => {
599 vm_metric.rss = match &vm_metric.rss {
600 Some(x) => Some(Rss::extract_max(x, &rss)),
601 None => Some(rss),
602 }
Seungjae Yoo6d265d92022-11-15 10:51:33 +0900603 }
Jiyong Park85ea5382024-02-19 15:04:23 +0900604 Err(e) => error!("Failed to get guest RSS: {}", e),
Seungjae Yoo6d265d92022-11-15 10:51:33 +0900605 }
606 }
607
608 thread::sleep(Duration::from_secs(1));
609 }
610 }
611
Andrew Walbran6b650662021-09-07 13:13:23 +0000612 /// Returns the last reported state of the VM payload.
613 pub fn payload_state(&self) -> PayloadState {
614 *self.payload_state.lock().unwrap()
615 }
616
617 /// Updates the payload state to the given value, if it is a valid state transition.
618 pub fn update_payload_state(&self, new_state: PayloadState) -> Result<(), Error> {
619 let mut state_locked = self.payload_state.lock().unwrap();
620 // Only allow forward transitions, e.g. from starting to started or finished, not back in
621 // the other direction.
622 if new_state > *state_locked {
623 *state_locked = new_state;
Jiyong Parka4eebde2022-07-12 18:01:12 +0900624 self.payload_state_updated.notify_all();
Andrew Walbran6b650662021-09-07 13:13:23 +0000625 Ok(())
626 } else {
627 bail!("Invalid payload state transition from {:?} to {:?}", *state_locked, new_state)
628 }
629 }
630
Andrew Walbranf8d94112021-09-07 11:45:36 +0000631 /// Kills the crosvm instance, if it is running.
Inseob Kima446f802022-07-11 19:46:37 +0900632 pub fn kill(&self) -> Result<(), Error> {
David Brazdilf18e1162022-12-18 18:27:23 +0000633 let monitor_vm_exit_thread = {
634 let vm_state = &mut *self.vm_state.lock().unwrap();
635 if let VmState::Running { child, monitor_vm_exit_thread } = vm_state {
636 let id = child.id();
637 debug!("Killing crosvm({})", id);
638 // TODO: Talk to crosvm to shutdown cleanly.
639 child.kill().with_context(|| format!("Error killing crosvm({id}) instance"))?;
640 monitor_vm_exit_thread.take()
Inseob Kima446f802022-07-11 19:46:37 +0900641 } else {
David Brazdilf18e1162022-12-18 18:27:23 +0000642 bail!("VM is not running")
Andrew Walbranf8d94112021-09-07 11:45:36 +0000643 }
David Brazdilf18e1162022-12-18 18:27:23 +0000644 };
645
646 // Wait for monitor_vm_exit() to finish. Must release vm_state lock
647 // first, as monitor_vm_exit() takes it as well.
648 monitor_vm_exit_thread.map(JoinHandle::join);
649
David Brazdil5af55832023-03-08 13:08:56 +0000650 // Now that the VM has been killed, shut down the VirtualMachineService
651 // server to eagerly free up the server threads.
652 self.vm_context.vm_server.shutdown()?;
653
David Brazdilf18e1162022-12-18 18:27:23 +0000654 Ok(())
Andrew Walbrand6dce6f2021-03-05 16:39:08 +0000655 }
Jiyong Parke558ab12022-07-07 20:18:55 +0900656
Keir Frasercdd4b112022-11-24 14:02:25 +0000657 /// Responds to memory-trimming notifications by inflating the virtio
658 /// balloon to reclaim guest memory.
Keir Fraser48221ba2024-07-12 14:05:02 +0000659 pub fn get_memory_balloon(&self) -> Result<u64, Error> {
Keir Frasercdd4b112022-11-24 14:02:25 +0000660 let request = VmRequest::BalloonCommand(BalloonControlCommand::Stats {});
Keir Fraser48221ba2024-07-12 14:05:02 +0000661 let result =
662 match vm_control::client::handle_request(&request, &self.crosvm_control_socket_path) {
663 Ok(VmResponse::BalloonStats { stats: _, balloon_actual }) => balloon_actual,
664 Ok(VmResponse::Err(e)) => {
665 // ENOTSUP is returned when the balloon protocol is not initialized. This
666 // can occur for numerous reasons: Guest is still booting, guest doesn't
667 // support ballooning, host doesn't support ballooning. We don't log or
668 // raise an error in this case: trim is just a hint and we can ignore it.
669 if e.errno() != libc::ENOTSUP {
670 bail!("Errno return when requesting balloon stats: {}", e.errno())
Keir Frasercdd4b112022-11-24 14:02:25 +0000671 }
Keir Fraser48221ba2024-07-12 14:05:02 +0000672 0
Keir Frasercdd4b112022-11-24 14:02:25 +0000673 }
Keir Fraser48221ba2024-07-12 14:05:02 +0000674 e => bail!("Error requesting balloon stats: {:?}", e),
675 };
676 Ok(result)
677 }
678
679 /// Responds to memory-trimming notifications by inflating the virtio
680 /// balloon to reclaim guest memory.
681 pub fn set_memory_balloon(&self, num_bytes: u64) -> Result<(), Error> {
682 let command = BalloonControlCommand::Adjust { num_bytes, wait_for_success: false };
683 if let Err(e) = vm_control::client::handle_request(
684 &VmRequest::BalloonCommand(command),
685 &self.crosvm_control_socket_path,
686 ) {
687 bail!("Error sending balloon adjustment: {:?}", e);
Keir Frasercdd4b112022-11-24 14:02:25 +0000688 }
689 Ok(())
690 }
691
Alan Stokes3e98d292022-12-14 15:18:22 +0000692 /// Checks if ramdump has been created. If so, send it to tombstoned.
Jiyong Parke558ab12022-07-07 20:18:55 +0900693 fn handle_ramdump(&self) -> Result<(), Error> {
694 let ramdump_path = self.temporary_directory.join("ramdump");
Jiyong Park933d2b22023-02-28 11:41:19 +0900695 if !ramdump_path.as_path().try_exists()? {
696 return Ok(());
697 }
Jiyong Parke558ab12022-07-07 20:18:55 +0900698 if std::fs::metadata(&ramdump_path)?.len() > 0 {
Jiyong Park1612b902022-08-22 14:47:39 +0900699 Self::send_ramdump_to_tombstoned(&ramdump_path)?;
Jiyong Parke558ab12022-07-07 20:18:55 +0900700 }
701 Ok(())
702 }
Jiyong Park1612b902022-08-22 14:47:39 +0900703
704 fn send_ramdump_to_tombstoned(ramdump_path: &Path) -> Result<(), Error> {
705 let mut input = File::open(ramdump_path)
Alan Stokes3e98d292022-12-14 15:18:22 +0000706 .context(format!("Failed to open ramdump {:?} for reading", ramdump_path))?;
Jiyong Park1612b902022-08-22 14:47:39 +0900707
708 let pid = std::process::id() as i32;
709 let conn = TombstonedConnection::connect(pid, DebuggerdDumpType::Tombstone)
710 .context("Failed to connect to tombstoned")?;
711 let mut output = conn
712 .text_output
713 .as_ref()
714 .ok_or_else(|| anyhow!("Could not get file to write the tombstones on"))?;
715
716 std::io::copy(&mut input, &mut output).context("Failed to send ramdump to tombstoned")?;
717 info!("Ramdump {:?} sent to tombstoned", ramdump_path);
718
719 conn.notify_completion()?;
720 Ok(())
721 }
Jiyong Park23b67392024-07-04 13:51:42 +0900722
723 /// Suspends the VM
724 pub fn suspend(&self) -> Result<(), Error> {
725 match vm_control::client::handle_request(
726 &VmRequest::SuspendVcpus,
727 &self.crosvm_control_socket_path,
728 ) {
729 Ok(VmResponse::Ok) => Ok(()),
730 e => bail!("Failed to suspend VM: {e:?}"),
731 }
732 }
733
734 /// Resumes the suspended VM
735 pub fn resume(&self) -> Result<(), Error> {
736 match vm_control::client::handle_request(
737 &VmRequest::ResumeVcpus,
738 &self.crosvm_control_socket_path,
739 ) {
740 Ok(VmResponse::Ok) => Ok(()),
741 e => bail!("Failed to resume: {e:?}"),
742 }
743 }
Andrew Walbrand6dce6f2021-03-05 16:39:08 +0000744}
745
Seungjae Yoo6d265d92022-11-15 10:51:33 +0900746impl Rss {
747 fn extract_max(x: &Rss, y: &Rss) -> Rss {
748 Rss { vm: max(x.vm, y.vm), crosvm: max(x.crosvm, y.crosvm) }
749 }
750}
751
David Daiaaecf652024-02-22 14:41:39 -0800752// Get Cpus_allowed mask
753fn check_if_all_cpus_allowed() -> Result<bool> {
754 let file = read_to_string("/proc/self/status")?;
755 let lines: Vec<_> = file.split('\n').collect();
756
757 for line in lines {
758 if line.contains("Cpus_allowed_list") {
759 let prop: Vec<_> = line.split_whitespace().collect();
760 if prop.len() != 2 {
761 return Ok(false);
762 }
763 let cpu_list: Vec<_> = prop[1].split('-').collect();
764 //Only contiguous Cpu list allowed
765 if cpu_list.len() != 2 {
766 return Ok(false);
767 }
768 if let Some(cpus) = get_num_cpus() {
769 let max_cpu = cpu_list[1].parse::<usize>()?;
770 if max_cpu == cpus - 1 {
771 return Ok(true);
772 } else {
773 return Ok(false);
774 }
775 }
776 }
777 }
778 Ok(false)
779}
780
Seungjae Yoo6d265d92022-11-15 10:51:33 +0900781// Get guest time from /proc/[crosvm pid]/stat
782fn get_guest_time(pid: u32) -> Result<i64> {
783 let file = read_to_string(format!("/proc/{}/stat", pid))?;
784 let data_list: Vec<_> = file.split_whitespace().collect();
785
786 // Information about guest_time is at 43th place of the file split with the whitespace.
787 // Example of /proc/[pid]/stat :
788 // 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
789 // 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
790 if data_list.len() < 43 {
791 bail!("Failed to parse command result for getting guest time : {}", file);
792 }
793
794 let guest_time_ticks = data_list[42].parse::<i64>()?;
Andrew Walbranb58d1b42023-07-07 13:54:49 +0100795 // SAFETY: It just returns an integer about CPU tick information.
Charisee96113f32023-01-26 09:00:42 +0000796 let ticks_per_sec = unsafe { sysconf(_SC_CLK_TCK) };
Seungjae Yoo6d265d92022-11-15 10:51:33 +0900797 Ok(guest_time_ticks * MILLIS_PER_SEC / ticks_per_sec)
798}
799
800// Get rss from /proc/[crosvm pid]/smaps
801fn get_rss(pid: u32) -> Result<Rss> {
802 let file = read_to_string(format!("/proc/{}/smaps", pid))?;
803 let lines: Vec<_> = file.split('\n').collect();
804
805 let mut rss_vm_total = 0i64;
806 let mut rss_crosvm_total = 0i64;
807 let mut is_vm = false;
808 for line in lines {
809 if line.contains("crosvm_guest") {
810 is_vm = true;
811 } else if line.contains("Rss:") {
812 let data_list: Vec<_> = line.split_whitespace().collect();
813 if data_list.len() < 2 {
814 bail!("Failed to parse command result for getting rss :\n{}", line);
815 }
816 let rss = data_list[1].parse::<i64>()?;
817
818 if is_vm {
819 rss_vm_total += rss;
820 is_vm = false;
821 }
822 rss_crosvm_total += rss;
823 }
824 }
825
826 Ok(Rss { vm: rss_vm_total, crosvm: rss_crosvm_total })
827}
828
Alan Stokes3ba10fd2022-10-06 15:46:51 +0100829fn death_reason(result: &Result<ExitStatus, io::Error>, mut failure_reason: &str) -> DeathReason {
Pierre-Clément Tosic0611e72023-09-15 11:41:07 +0100830 if let Some((reason, info)) = failure_reason.split_once('|') {
Alan Stokes3ba10fd2022-10-06 15:46:51 +0100831 // Separator indicates extra context information is present after the failure name.
Pierre-Clément Tosic0611e72023-09-15 11:41:07 +0100832 error!("Failure info: {info}");
833 failure_reason = reason;
Alan Stokes3ba10fd2022-10-06 15:46:51 +0100834 }
Andrew Walbranc92d35f2022-01-12 12:45:19 +0000835 if let Ok(status) = result {
Andrew Walbranb27681f2022-02-23 15:11:52 +0000836 match failure_reason {
837 "PVM_FIRMWARE_PUBLIC_KEY_MISMATCH" => {
838 return DeathReason::PVM_FIRMWARE_PUBLIC_KEY_MISMATCH
839 }
840 "PVM_FIRMWARE_INSTANCE_IMAGE_CHANGED" => {
841 return DeathReason::PVM_FIRMWARE_INSTANCE_IMAGE_CHANGED
842 }
Inseob Kim272f5722022-06-13 17:14:51 +0900843 "MICRODROID_FAILED_TO_CONNECT_TO_VIRTUALIZATION_SERVICE" => {
844 return DeathReason::MICRODROID_FAILED_TO_CONNECT_TO_VIRTUALIZATION_SERVICE
845 }
846 "MICRODROID_PAYLOAD_HAS_CHANGED" => return DeathReason::MICRODROID_PAYLOAD_HAS_CHANGED,
847 "MICRODROID_PAYLOAD_VERIFICATION_FAILED" => {
848 return DeathReason::MICRODROID_PAYLOAD_VERIFICATION_FAILED
849 }
850 "MICRODROID_INVALID_PAYLOAD_CONFIG" => {
851 return DeathReason::MICRODROID_INVALID_PAYLOAD_CONFIG
852 }
853 "MICRODROID_UNKNOWN_RUNTIME_ERROR" => {
854 return DeathReason::MICRODROID_UNKNOWN_RUNTIME_ERROR
855 }
Jiyong Parke6ed0f92022-06-22 00:13:00 +0900856 "HANGUP" => return DeathReason::HANGUP,
Andrew Walbranb27681f2022-02-23 15:11:52 +0000857 _ => {}
858 }
Andrew Walbranc92d35f2022-01-12 12:45:19 +0000859 match status.code() {
860 None => DeathReason::KILLED,
861 Some(0) => DeathReason::SHUTDOWN,
Alan Stokes7c459e82022-12-06 12:21:49 +0000862 Some(CROSVM_START_ERROR_STATUS) => DeathReason::START_FAILED,
Andrew Walbranc92d35f2022-01-12 12:45:19 +0000863 Some(CROSVM_REBOOT_STATUS) => DeathReason::REBOOT,
Andrew Walbrand15c5632022-02-03 13:38:31 +0000864 Some(CROSVM_CRASH_STATUS) => DeathReason::CRASH,
Sebastian Ene23167d82022-10-07 14:09:53 +0000865 Some(CROSVM_WATCHDOG_REBOOT_STATUS) => DeathReason::WATCHDOG_REBOOT,
Andrew Walbranc92d35f2022-01-12 12:45:19 +0000866 Some(_) => DeathReason::UNKNOWN,
867 }
868 } else {
869 DeathReason::INFRASTRUCTURE_ERROR
870 }
871}
872
Seungjae Yoo93430e82022-12-05 16:37:42 +0900873fn exit_signal(result: &Result<ExitStatus, io::Error>) -> Option<i32> {
874 match result {
875 Ok(status) => status.signal(),
876 Err(_) => None,
877 }
878}
879
Inseob Kim6ef80972023-07-20 17:23:36 +0900880const SYSFS_PLATFORM_DEVICES_PATH: &str = "/sys/devices/platform/";
881const VFIO_PLATFORM_DRIVER_PATH: &str = "/sys/bus/platform/drivers/vfio-platform";
882
Inseob Kim7307a892023-09-14 13:37:58 +0900883fn vfio_argument_for_platform_device(device: &VfioDevice) -> Result<String, Error> {
Inseob Kim6ef80972023-07-20 17:23:36 +0900884 // Check platform device exists
Jakob Vukalovicd42aa2c2023-11-09 16:04:00 +0000885 let path = Path::new(&device.getSysfsPath()?).canonicalize()?;
Inseob Kim6ef80972023-07-20 17:23:36 +0900886 if !path.starts_with(SYSFS_PLATFORM_DEVICES_PATH) {
887 bail!("{path:?} is not a platform device");
888 }
889
890 // Check platform device is bound to VFIO driver
891 let dev_driver_path = path.join("driver").canonicalize()?;
892 if dev_driver_path != Path::new(VFIO_PLATFORM_DRIVER_PATH) {
893 bail!("{path:?} is not bound to VFIO-platform driver");
894 }
895
896 if let Some(p) = path.to_str() {
Jakob Vukalovicd42aa2c2023-11-09 16:04:00 +0000897 Ok(format!("--vfio={p},iommu=pkvm-iommu,dt-symbol={0}", device.getDtboLabel()?))
Inseob Kim6ef80972023-07-20 17:23:36 +0900898 } else {
899 bail!("invalid path {path:?}");
900 }
901}
902
Akilesh Kailash05515dc2024-10-22 21:42:52 -0700903fn run_virtiofs(config: &CrosvmConfig) -> io::Result<()> {
904 for shared_path in &config.shared_paths {
905 let ugid_map_value = format!(
906 "{} {} {} {} {} /",
907 shared_path.guest_uid,
908 shared_path.guest_gid,
909 shared_path.host_uid,
910 shared_path.host_gid,
911 shared_path.mask,
912 );
913
914 let cfg_arg = format!("writeback=true,cache_policy=always,ugid_map='{}'", ugid_map_value);
915
916 let mut command = Command::new(CROSVM_PATH);
917 command
918 .arg("device")
919 .arg("fs")
920 .arg(format!("--socket={}", &shared_path.socket_path))
921 .arg(format!("--tag={}", &shared_path.tag))
922 .arg(format!("--shared-dir={}", &shared_path.path))
923 .arg("--cfg")
924 .arg(cfg_arg.as_str())
925 .arg("--disable-sandbox");
926
927 print_crosvm_args(&command);
928
929 let result = SharedChild::spawn(&mut command)?;
930 info!("Spawned virtiofs crosvm({})", result.id());
931 }
932
933 Ok(())
934}
935
Andrew Walbrand3a84182021-09-07 14:48:52 +0000936/// Starts an instance of `crosvm` to manage a new VM.
Keir Fraser13a956a2022-07-14 14:20:46 +0000937fn run_vm(
938 config: CrosvmConfig,
Keir Frasercdd4b112022-11-24 14:02:25 +0000939 crosvm_control_socket_path: &Path,
Keir Fraser13a956a2022-07-14 14:20:46 +0000940 failure_pipe_write: File,
941) -> Result<SharedChild, Error> {
Andrew Walbrand3a84182021-09-07 14:48:52 +0000942 validate_config(&config)?;
Andrew Walbrand6dce6f2021-03-05 16:39:08 +0000943
944 let mut command = Command::new(CROSVM_PATH);
945 // TODO(qwandor): Remove --disable-sandbox.
Andrew Walbranc92d35f2022-01-12 12:45:19 +0000946 command
947 .arg("--extended-status")
Jiyong Park6c60fea2022-10-24 16:10:01 +0900948 // Configure the logger for the crosvm process to silence logs from the disk crate which
949 // don't provide much information to us (but do spamming us).
950 .arg("--log-level")
Frederick Mayle138f53c2023-01-24 18:55:26 -0800951 .arg("info,disk=warn")
Andrew Walbranc92d35f2022-01-12 12:45:19 +0000952 .arg("run")
953 .arg("--disable-sandbox")
954 .arg("--cid")
955 .arg(config.cid.to_string());
Andrew Walbran3a5a9212021-05-04 17:09:08 +0000956
Jeongik Cha75057452024-09-03 11:26:04 +0900957 if system_properties::read_bool("hypervisor.memory_reclaim.supported", false)?
958 && !config.no_balloon
959 {
Keir Fraserf25cb922022-11-23 14:26:00 +0000960 command.arg("--balloon-page-reporting");
961 } else {
962 command.arg("--no-balloon");
963 }
964
Elie Kheirallah29d72912024-08-07 20:17:26 +0000965 if !config.usb_config.controller {
966 command.arg("--no-usb");
967 }
968
Jiyong Parkae625b62024-06-13 21:47:52 +0900969 let mut memory_mib = config.memory_mib;
970
Andrew Walbranf8650422021-06-09 15:54:09 +0000971 if config.protected {
Jaewan Kim46422812022-11-25 10:59:39 +0900972 match system_properties::read(SYSPROP_CUSTOM_PVMFW_PATH)? {
973 Some(pvmfw_path) if !pvmfw_path.is_empty() => {
974 command.arg("--protected-vm-with-firmware").arg(pvmfw_path)
975 }
976 _ => command.arg("--protected-vm"),
977 };
Andrew Walbran0b5789f2022-02-04 13:57:57 +0000978
979 // 3 virtio-console devices + vsock = 4.
980 let virtio_pci_device_count = 4 + config.disks.len();
981 // crosvm virtio queue has 256 entries, so 2 MiB per device (2 pages per entry) should be
982 // enough.
Jiyong Park3115cdb2023-02-24 18:04:44 +0900983 let swiotlb_size_mib = 2 * virtio_pci_device_count as u32;
Andrew Walbran0b5789f2022-02-04 13:57:57 +0000984 command.arg("--swiotlb").arg(swiotlb_size_mib.to_string());
Frederick Mayle58baa302023-02-14 19:06:37 -0800985
Frederick Maylea24bbe02024-07-12 16:50:48 -0700986 // b/346770542 for consistent "usable" memory across protected and non-protected VMs.
Frederick Mayle68d06e72024-07-12 17:18:11 -0700987 memory_mib = memory_mib.saturating_add(swiotlb_size_mib);
Jiyong Parkae625b62024-06-13 21:47:52 +0900988
Frederick Mayle58baa302023-02-14 19:06:37 -0800989 // Workaround to keep crash_dump from trying to read protected guest memory.
990 // Context in b/238324526.
991 command.arg("--unmap-guest-memory-on-fork");
Jiyong Park3115cdb2023-02-24 18:04:44 +0900992
Jiyong Parked180932023-02-24 19:55:41 +0900993 if config.ramdump.is_some() {
Jaewan Kim3124ef02023-03-23 19:25:20 +0900994 // Protected VM needs to reserve memory for ramdump here. Note that we reserve more
Jiyong Parked180932023-02-24 19:55:41 +0900995 // memory for the restricted dma pool.
996 let ramdump_reserve = RAMDUMP_RESERVED_MIB + swiotlb_size_mib;
997 command.arg("--params").arg(format!("crashkernel={ramdump_reserve}M"));
998 }
Jaewan Kimd4dae652023-03-28 13:45:43 +0900999 } else if config.ramdump.is_some() {
1000 command.arg("--params").arg(format!("crashkernel={RAMDUMP_RESERVED_MIB}M"));
1001 }
Jaewan Kim61f86142023-03-28 15:12:52 +09001002 if config.debug_config.debug_level == DebugLevel::NONE
1003 && config.debug_config.should_prepare_console_output()
1004 {
Jaewan Kimd4dae652023-03-28 13:45:43 +09001005 // bootconfig.normal will be used, but we need log.
1006 command.arg("--params").arg("printk.devkmsg=on");
1007 command.arg("--params").arg("console=hvc0");
Andrew Walbranf8650422021-06-09 15:54:09 +00001008 }
1009
Frederick Mayle68d06e72024-07-12 17:18:11 -07001010 command.arg("--mem").arg(memory_mib.to_string());
Andrew Walbranb15cd6e2021-07-05 16:38:07 +00001011
Jiyong Park032615f2022-01-10 13:55:34 +09001012 if let Some(cpus) = config.cpus {
1013 command.arg("--cpus").arg(cpus.to_string());
1014 }
1015
David Brazdil7d1e5ec2023-02-06 17:56:29 +00001016 if config.host_cpu_topology {
David Daiaaecf652024-02-22 14:41:39 -08001017 if cfg!(virt_cpufreq) && check_if_all_cpus_allowed()? {
David Daie6f5bb02024-01-25 19:17:02 -08001018 command.arg("--host-cpu-topology");
David Daic97e82d2024-02-17 12:10:31 -08001019 cfg_if::cfg_if! {
1020 if #[cfg(any(target_arch = "aarch64"))] {
1021 command.arg("--virt-cpufreq");
1022 }
1023 }
David Daie6f5bb02024-01-25 19:17:02 -08001024 } else if let Some(cpus) = get_num_cpus() {
David Brazdil25e8c052023-02-17 12:53:01 +00001025 command.arg("--cpus").arg(cpus.to_string());
1026 } else {
1027 bail!("Could not determine the number of CPUs in the system");
1028 }
David Brazdil7d1e5ec2023-02-06 17:56:29 +00001029 }
1030
Nikita Ioffe5776f082023-02-10 21:38:26 +00001031 if let Some(gdb_port) = config.gdb_port {
1032 command.arg("--gdb").arg(gdb_port.to_string());
1033 }
1034
Jiyong Parkfa91d702021-10-18 23:51:39 +09001035 // Keep track of what file descriptors should be mapped to the crosvm process.
Jiyong Park69001a22024-08-16 20:45:34 +09001036 let mut preserved_fds = config.indirect_files.into_iter().map(|f| f.into()).collect();
Jiyong Parkfa91d702021-10-18 23:51:39 +09001037
Elie Kheirallah5c807a22024-09-23 20:40:42 +00001038 if let Some(dump_dt_fd) = config.dump_dt_fd {
1039 let dump_dt_fd = add_preserved_fd(&mut preserved_fds, dump_dt_fd);
1040 command.arg("--dump-device-tree-blob").arg(dump_dt_fd);
1041 }
1042
Jiyong Park747d6362021-10-19 17:12:52 +09001043 // Setup the serial devices.
1044 // 1. uart device: used as the output device by bootloaders and as early console by linux
Andrew Walbranb27681f2022-02-23 15:11:52 +00001045 // 2. uart device: used to report the reason for the VM failing.
1046 // 3. virtio-console device: used as the console device where kmsg is redirected to
Jiyong Park4afe2012022-07-08 05:38:49 +09001047 // 4. virtio-console device: used as the ramdump output
Andrew Walbranb27681f2022-02-23 15:11:52 +00001048 // 5. virtio-console device: used as the logcat output
Jiyong Park747d6362021-10-19 17:12:52 +09001049 //
Jiyong Parkb8182bb2021-10-26 22:53:08 +09001050 // When [console|log]_fd is not specified, the devices are attached to sink, which means what's
1051 // written there is discarded.
Jiyong Park69001a22024-08-16 20:45:34 +09001052 let console_out_arg = format_serial_out_arg(&mut preserved_fds, config.console_out_fd);
Jiyong Parke6fb1672023-06-26 16:45:55 +09001053 let console_in_arg = config
1054 .console_in_fd
Jiyong Parke6fb1672023-06-26 16:45:55 +09001055 .map(|fd| format!(",input={}", add_preserved_fd(&mut preserved_fds, fd)))
1056 .unwrap_or_default();
Jiyong Park69001a22024-08-16 20:45:34 +09001057 let log_arg = format_serial_out_arg(&mut preserved_fds, config.log_fd);
1058 let failure_serial_path = add_preserved_fd(&mut preserved_fds, failure_pipe_write);
1059 let ramdump_arg = format_serial_out_arg(&mut preserved_fds, config.ramdump);
Yi-Yo Chiang8dd32552024-05-22 19:38:16 +08001060 let console_input_device = config.console_input_device.as_deref().unwrap_or(CONSOLE_HVC0);
1061 match console_input_device {
1062 CONSOLE_HVC0 | CONSOLE_TTYS0 => {}
1063 _ => bail!("Unsupported serial device {console_input_device}"),
1064 };
Jiyong Parkfa91d702021-10-18 23:51:39 +09001065
Jiyong Park747d6362021-10-19 17:12:52 +09001066 // Warning: Adding more serial devices requires you to shift the PCI device ID of the boot
1067 // disks in bootconfig.x86_64. This is because x86 crosvm puts serial devices and the block
1068 // devices in the same PCI bus and serial devices comes before the block devices. Arm crosvm
1069 // doesn't have the issue.
Jiyong Parkfa91d702021-10-18 23:51:39 +09001070 // /dev/ttyS0
Yi-Yo Chiang8dd32552024-05-22 19:38:16 +08001071 command.arg(format!(
1072 "--serial={}{},hardware=serial,num=1",
1073 &console_out_arg,
1074 if console_input_device == CONSOLE_TTYS0 { &console_in_arg } else { "" }
1075 ));
Andrew Walbranb27681f2022-02-23 15:11:52 +00001076 // /dev/ttyS1
1077 command.arg(format!("--serial=type=file,path={},hardware=serial,num=2", &failure_serial_path));
Jiyong Parkfa91d702021-10-18 23:51:39 +09001078 // /dev/hvc0
Jiyong Parke6fb1672023-06-26 16:45:55 +09001079 command.arg(format!(
1080 "--serial={}{},hardware=virtio-console,num=1",
Yi-Yo Chiang8dd32552024-05-22 19:38:16 +08001081 &console_out_arg,
1082 if console_input_device == CONSOLE_HVC0 { &console_in_arg } else { "" }
Jiyong Parke6fb1672023-06-26 16:45:55 +09001083 ));
Jiyong Park4afe2012022-07-08 05:38:49 +09001084 // /dev/hvc1
1085 command.arg(format!("--serial={},hardware=virtio-console,num=2", &ramdump_arg));
Jiyong Parkae5a4ed2021-11-01 18:36:28 +09001086 // /dev/hvc2
1087 command.arg(format!("--serial={},hardware=virtio-console,num=3", &log_arg));
Andrew Walbran3a5a9212021-05-04 17:09:08 +00001088
Jiyong Park69001a22024-08-16 20:45:34 +09001089 if let Some(bootloader) = config.bootloader {
Andrew Walbran02b8ec02021-06-22 13:07:02 +00001090 command.arg("--bios").arg(add_preserved_fd(&mut preserved_fds, bootloader));
Andrew Walbrand6dce6f2021-03-05 16:39:08 +00001091 }
Andrew Walbran3a5a9212021-05-04 17:09:08 +00001092
Jiyong Park69001a22024-08-16 20:45:34 +09001093 if let Some(initrd) = config.initrd {
Andrew Walbran02b8ec02021-06-22 13:07:02 +00001094 command.arg("--initrd").arg(add_preserved_fd(&mut preserved_fds, initrd));
Andrew Walbrand6dce6f2021-03-05 16:39:08 +00001095 }
Andrew Walbran3a5a9212021-05-04 17:09:08 +00001096
Andrew Walbrand6dce6f2021-03-05 16:39:08 +00001097 if let Some(params) = &config.params {
1098 command.arg("--params").arg(params);
1099 }
Andrew Walbran3a5a9212021-05-04 17:09:08 +00001100
Jiyong Park69001a22024-08-16 20:45:34 +09001101 for disk in config.disks {
Frederick Mayleeae6d2d2024-09-12 11:05:36 -07001102 // Disk file locking is disabled because of missing SELinux policies.
Frederick Mayle1ff7b842024-07-12 16:20:09 -07001103 command.arg("--block").arg(format!(
Frederick Mayleeae6d2d2024-09-12 11:05:36 -07001104 "path={},ro={},lock=false",
Jiyong Park69001a22024-08-16 20:45:34 +09001105 add_preserved_fd(&mut preserved_fds, disk.image),
Frederick Mayle1ff7b842024-07-12 16:20:09 -07001106 !disk.writable,
1107 ));
Andrew Walbrand6dce6f2021-03-05 16:39:08 +00001108 }
Andrew Walbran3a5a9212021-05-04 17:09:08 +00001109
Jiyong Park69001a22024-08-16 20:45:34 +09001110 if let Some(kernel) = config.kernel {
Andrew Walbran02b8ec02021-06-22 13:07:02 +00001111 command.arg(add_preserved_fd(&mut preserved_fds, kernel));
Andrew Walbrand6dce6f2021-03-05 16:39:08 +00001112 }
Andrew Walbran3a5a9212021-05-04 17:09:08 +00001113
Frederick Maylee0ed2352024-09-12 16:16:00 -07001114 let control_sock = create_crosvm_control_listener(crosvm_control_socket_path)
1115 .context("failed to create control listener")?;
Jiyong Parkcd0f74b2024-08-20 11:06:52 +09001116 command.arg("--socket").arg(add_preserved_fd(&mut preserved_fds, control_sock));
Keir Fraser13a956a2022-07-14 14:20:46 +00001117
Jiyong Park69001a22024-08-16 20:45:34 +09001118 if let Some(dt_overlay) = config.device_tree_overlay {
Shikha Panwar55e10ec2024-02-13 12:53:49 +00001119 command.arg("--device-tree-overlay").arg(add_preserved_fd(&mut preserved_fds, dt_overlay));
Seungjae Yoo4dd38182023-11-09 15:05:45 +09001120 }
Jeongik Cha798401c2024-04-11 21:54:49 +09001121
1122 if cfg!(paravirtualized_devices) {
Jason Macnakea7b0fa2024-05-08 17:02:44 -07001123 if let Some(gpu_config) = &config.gpu_config {
1124 let mut gpu_args = Vec::new();
1125 if let Some(backend) = &gpu_config.backend {
1126 gpu_args.push(format!("backend={}", backend));
1127 }
1128 if let Some(context_types) = &gpu_config.context_types {
1129 gpu_args.push(format!("context-types={}", context_types.join(":")));
1130 }
1131 if let Some(pci_address) = &gpu_config.pci_address {
1132 gpu_args.push(format!("pci-address={}", pci_address));
1133 }
1134 if let Some(renderer_features) = &gpu_config.renderer_features {
1135 gpu_args.push(format!("renderer-features={}", renderer_features));
1136 }
1137 if gpu_config.renderer_use_egl.unwrap_or(false) {
1138 gpu_args.push("egl=true".to_string());
1139 }
1140 if gpu_config.renderer_use_gles.unwrap_or(false) {
1141 gpu_args.push("gles=true".to_string());
1142 }
1143 if gpu_config.renderer_use_glx.unwrap_or(false) {
1144 gpu_args.push("glx=true".to_string());
1145 }
1146 if gpu_config.renderer_use_surfaceless.unwrap_or(false) {
1147 gpu_args.push("surfaceless=true".to_string());
1148 }
1149 if gpu_config.renderer_use_vulkan.unwrap_or(false) {
1150 gpu_args.push("vulkan=true".to_string());
1151 }
1152 command.arg(format!("--gpu={}", gpu_args.join(",")));
1153 }
Jeongik Cha798401c2024-04-11 21:54:49 +09001154 if let Some(display_config) = &config.display_config {
Jason Macnakea7b0fa2024-05-08 17:02:44 -07001155 command
1156 .arg(format!(
1157 "--gpu-display=mode=windowed[{},{}],dpi=[{},{}],refresh-rate={}",
1158 display_config.width,
1159 display_config.height,
1160 display_config.horizontal_dpi,
1161 display_config.vertical_dpi,
1162 display_config.refresh_rate
1163 ))
1164 .arg(format!("--android-display-service={}", config.name));
Jeongik Cha798401c2024-04-11 21:54:49 +09001165 }
Jeongik Cha1df1c032024-04-03 16:03:12 +09001166 }
Seungjae Yooec3bc522023-11-09 10:14:30 +09001167
Seungjae Yoo986d7a42024-05-28 14:06:58 +09001168 if cfg!(network) {
Jiyong Park69001a22024-08-16 20:45:34 +09001169 if let Some(tap) = config.tap {
Andreas Huber487810d2024-08-22 07:43:01 -07001170 add_preserved_fd(&mut preserved_fds, tap);
1171 let tap_fd = preserved_fds.last().unwrap().as_raw_fd();
1172 command.arg("--net").arg(format!("tap-fd={tap_fd}"));
Seungjae Yoo986d7a42024-05-28 14:06:58 +09001173 }
1174 }
1175
Jeongik Cha6be4f8b2024-04-16 01:13:13 +09001176 if cfg!(paravirtualized_devices) {
Jiyong Park69001a22024-08-16 20:45:34 +09001177 for input_device_option in config.input_device_options.into_iter() {
Jeongik Chac181be72024-03-27 00:18:30 +09001178 command.arg("--input");
1179 command.arg(match input_device_option {
1180 InputDeviceOption::EvDev(file) => {
1181 format!("evdev[path={}]", add_preserved_fd(&mut preserved_fds, file))
1182 }
Jeongik Cha985b6402024-04-15 18:13:00 +09001183 InputDeviceOption::Keyboard(file) => {
1184 format!("keyboard[path={}]", add_preserved_fd(&mut preserved_fds, file))
1185 }
Jeongik Cha343894e2024-05-17 20:39:31 +09001186 InputDeviceOption::Mouse(file) => {
1187 format!("mouse[path={}]", add_preserved_fd(&mut preserved_fds, file))
1188 }
Jeongik Chac181be72024-03-27 00:18:30 +09001189 InputDeviceOption::SingleTouch { file, width, height, name } => format!(
1190 "single-touch[path={},width={},height={}{}]",
1191 add_preserved_fd(&mut preserved_fds, file),
1192 width,
1193 height,
1194 name.as_ref().map_or("".into(), |n| format!(",name={}", n))
1195 ),
Jiyong Park6c1b9f02024-07-05 16:21:10 +09001196 InputDeviceOption::Switches(file) => {
1197 format!("switches[path={}]", add_preserved_fd(&mut preserved_fds, file))
1198 }
Jeongik Cha112a2682024-07-09 12:28:45 +09001199 InputDeviceOption::MultiTouchTrackpad { file, width, height, name } => format!(
1200 "multi-touch-trackpad[path={},width={},height={}{}]",
1201 add_preserved_fd(&mut preserved_fds, file),
1202 width,
1203 height,
1204 name.as_ref().map_or("".into(), |n| format!(",name={}", n))
1205 ),
Jeongik Cha10494912024-07-16 11:19:50 +09001206 InputDeviceOption::MultiTouch { file, width, height, name } => format!(
1207 "multi-touch[path={},width={},height={}{}]",
1208 add_preserved_fd(&mut preserved_fds, file),
1209 width,
1210 height,
1211 name.as_ref().map_or("".into(), |n| format!(",name={}", n))
1212 ),
Jeongik Chac181be72024-03-27 00:18:30 +09001213 });
1214 }
1215 }
Vincent Donnefort538a2c62024-03-20 16:01:10 +00001216
1217 if config.hugepages {
1218 command.arg("--hugepages");
1219 }
1220
David Dai23cff712024-06-13 19:23:45 +00001221 if config.boost_uclamp {
1222 command.arg("--boost-uclamp");
1223 }
1224
Jiyong Park69001a22024-08-16 20:45:34 +09001225 if !config.vfio_devices.is_empty() {
1226 if let Some(dtbo) = config.dtbo {
1227 command.arg(format!(
1228 "--device-tree-overlay={},filter",
1229 add_preserved_fd(&mut preserved_fds, dtbo)
1230 ));
1231 } else {
1232 bail!("VFIO devices assigned but no DTBO available");
1233 }
1234 };
1235 for device in config.vfio_devices {
1236 command.arg(vfio_argument_for_platform_device(&device)?);
1237 }
Inseob Kim6ef80972023-07-20 17:23:36 +09001238
Akilesh Kailash05515dc2024-10-22 21:42:52 -07001239 for shared_path in &config.shared_paths {
1240 command
1241 .arg("--vhost-user-fs")
1242 .arg(format!("{},tag={}", &shared_path.socket_path, &shared_path.tag));
1243 }
1244
Andrew Walbran02b8ec02021-06-22 13:07:02 +00001245 debug!("Preserving FDs {:?}", preserved_fds);
1246 command.preserved_fds(preserved_fds);
Andrew Walbran3a5a9212021-05-04 17:09:08 +00001247
Mu-Le Lee134180a2024-04-16 15:13:07 +08001248 if cfg!(paravirtualized_devices) {
Mu-Le Leeb2d5a312024-06-05 10:52:47 +08001249 if let Some(audio_config) = &config.audio_config {
1250 command.arg("--virtio-snd").arg(format!(
1251 "backend=aaudio,num_input_devices={},num_output_devices={}",
1252 if audio_config.use_microphone { 1 } else { 0 },
1253 if audio_config.use_speaker { 1 } else { 0 }
1254 ));
Mu-Le Lee134180a2024-04-16 15:13:07 +08001255 }
1256 }
1257
Jiyong Park2d736562022-10-24 22:40:12 +09001258 print_crosvm_args(&command);
1259
Andrew Walbran3a5a9212021-05-04 17:09:08 +00001260 let result = SharedChild::spawn(&mut command)?;
Jooyung Hanbfe086f2021-10-28 10:15:45 +09001261 debug!("Spawned crosvm({}).", result.id());
Andrew Walbran3a5a9212021-05-04 17:09:08 +00001262 Ok(result)
1263}
1264
1265/// Ensure that the configuration has a valid combination of fields set, or return an error if not.
Andrew Walbranf5fbb7d2021-05-12 17:15:48 +00001266fn validate_config(config: &CrosvmConfig) -> Result<(), Error> {
Andrew Walbran3a5a9212021-05-04 17:09:08 +00001267 if config.bootloader.is_none() && config.kernel.is_none() {
1268 bail!("VM must have either a bootloader or a kernel image.");
1269 }
1270 if config.bootloader.is_some() && (config.kernel.is_some() || config.initrd.is_some()) {
1271 bail!("Can't have both bootloader and kernel/initrd image.");
1272 }
Jiyong Parkdcf17412022-02-08 15:07:23 +09001273 let version = Version::parse(CROSVM_PLATFORM_VERSION).unwrap();
1274 if !config.platform_version.matches(&version) {
1275 bail!(
1276 "Incompatible platform version. The config is compatible with platform version(s) \
1277 {}, but the actual platform version is {}",
1278 config.platform_version,
1279 version
1280 );
1281 }
1282
Andrew Walbran3a5a9212021-05-04 17:09:08 +00001283 Ok(())
1284}
1285
Jiyong Park2d736562022-10-24 22:40:12 +09001286/// Print arguments of the crosvm command. In doing so, /proc/self/fd/XX is annotated with the
1287/// actual file path if the FD is backed by a regular file. If not, the /proc path is printed
1288/// unmodified.
1289fn print_crosvm_args(command: &Command) {
1290 let re = Regex::new(r"/proc/self/fd/[\d]+").unwrap();
1291 info!(
1292 "Running crosvm with args: {:?}",
1293 command
1294 .get_args()
1295 .map(|s| s.to_string_lossy())
1296 .map(|s| {
1297 re.replace_all(&s, |caps: &Captures| {
1298 let path = &caps[0];
1299 if let Ok(realpath) = std::fs::canonicalize(path) {
1300 format!("{} ({})", path, realpath.to_string_lossy())
1301 } else {
1302 path.to_owned()
1303 }
1304 })
1305 .into_owned()
1306 })
1307 .collect::<Vec<_>>()
1308 );
1309}
1310
Andrew Walbran02b8ec02021-06-22 13:07:02 +00001311/// Adds the file descriptor for `file` to `preserved_fds`, and returns a string of the form
1312/// "/proc/self/fd/N" where N is the file descriptor.
Jiyong Park69001a22024-08-16 20:45:34 +09001313fn add_preserved_fd<F: Into<OwnedFd>>(preserved_fds: &mut Vec<OwnedFd>, file: F) -> String {
1314 let fd = file.into();
1315 let raw_fd = fd.as_raw_fd();
Andrew Walbran02b8ec02021-06-22 13:07:02 +00001316 preserved_fds.push(fd);
Jiyong Park69001a22024-08-16 20:45:34 +09001317 format!("/proc/self/fd/{}", raw_fd)
Andrew Walbrand6dce6f2021-03-05 16:39:08 +00001318}
Andrew Walbranb27681f2022-02-23 15:11:52 +00001319
1320/// Adds the file descriptor for `file` (if any) to `preserved_fds`, and returns the appropriate
1321/// string for a crosvm `--serial` flag. If `file` is none, creates a dummy sink device.
Jiyong Park69001a22024-08-16 20:45:34 +09001322fn format_serial_out_arg(preserved_fds: &mut Vec<OwnedFd>, file: Option<File>) -> String {
Andrew Walbranb27681f2022-02-23 15:11:52 +00001323 if let Some(file) = file {
1324 format!("type=file,path={}", add_preserved_fd(preserved_fds, file))
1325 } else {
1326 "type=sink".to_string()
1327 }
1328}
1329
1330/// Creates a new pipe with the `O_CLOEXEC` flag set, and returns the read side and write side.
1331fn create_pipe() -> Result<(File, File), Error> {
Frederick Maylefbbcfcd2024-04-08 16:31:54 -07001332 let (read_fd, write_fd) = pipe2(OFlag::O_CLOEXEC)?;
1333 Ok((read_fd.into(), write_fd.into()))
Andrew Walbranb27681f2022-02-23 15:11:52 +00001334}
Frederick Maylee0ed2352024-09-12 16:16:00 -07001335
1336/// Creates and binds a unix seqpacket listening socket to be passed as crosvm's `--socket`
1337/// argument. See `UnixSeqpacketListener::bind` in crosvm's code for reference.
1338fn create_crosvm_control_listener(crosvm_control_socket_path: &Path) -> Result<OwnedFd> {
1339 use nix::sys::socket;
1340 let fd = socket::socket(
1341 socket::AddressFamily::Unix,
1342 socket::SockType::SeqPacket,
1343 socket::SockFlag::empty(),
1344 None,
1345 )
1346 .context("socket failed")?;
1347 socket::bind(fd.as_raw_fd(), &socket::UnixAddr::new(crosvm_control_socket_path)?)
1348 .context("bind failed")?;
1349 // The exact backlog size isn't imporant. crosvm uses 128 internally. We use 127 here
1350 // because of a `nix` bug.
1351 socket::listen(&fd, socket::Backlog::new(127).unwrap()).context("listen failed")?;
1352 Ok(fd)
1353}