1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
|
//! Virtual machine configuration.
use std::env;
use std::path::{Path, PathBuf};
use serde::{Deserialize, Serialize};
#[derive(clap::Args, Debug, Serialize, Deserialize, Clone)]
#[serde(rename_all = "snake_case")]
pub struct VmConfig {
/// Type of virtual machine to use
pub vm_type: VmType,
/// Path to a VM disk image
pub image_path: String,
/// Type of operating system.
pub os_type: OsType,
/// Package type to use, e.g. deb or rpm
#[arg(long, required_if_eq("os_type", "linux"))]
pub package_type: Option<PackageType>,
/// CPU architecture
#[arg(long)]
pub architecture: Architecture,
/// Tool to use for provisioning
#[arg(long, default_value = "noop")]
pub provisioner: Provisioner,
/// Username to use for SSH
#[arg(long, required_if_eq("provisioner", "ssh"))]
pub ssh_user: Option<String>,
/// Password to use for SSH
#[arg(long, required_if_eq("provisioner", "ssh"))]
pub ssh_password: Option<String>,
/// Additional disk images to mount/include
#[arg(long)]
pub disks: Vec<String>,
/// Where artifacts, such as app packages, are stored.
/// Usually /opt/testing on Linux.
#[arg(long)]
pub artifacts_dir: Option<String>,
/// Emulate a TPM. This also enables UEFI implicitly
#[serde(default)]
#[arg(long)]
pub tpm: bool,
/// Override the path to `OVMF_VARS.secboot.fd`. Requires `tpm`.
#[serde(default)]
#[arg(long, requires("tpm"))]
pub ovmf_vars_path: Option<String>,
/// Override the path to `OVMF_CODE.secboot.fd`. Requires `tpm`.
#[serde(default)]
#[arg(long, requires("tpm"))]
pub ovmf_code_path: Option<String>,
/// Number of vCPUs
#[serde(default)]
#[arg(long)]
pub vcpus: Option<usize>,
/// Amount of memory, in MBs
#[serde(default)]
#[arg(long)]
pub memory: Option<usize>,
}
impl VmConfig {
/// Combine authentication details, if all are present
pub fn get_ssh_options(&self) -> Option<(&str, &str)> {
Some((self.ssh_user.as_ref()?, self.ssh_password.as_ref()?))
}
pub fn get_default_runner_dir(&self) -> PathBuf {
let target_dir = self.get_target_dir();
let subdir = match self.architecture {
Architecture::X64 => self.get_x64_runner_subdir(),
Architecture::Aarch64 => self.get_aarch64_runner_subdir(),
};
target_dir.join(subdir)
}
fn get_x64_runner_subdir(&self) -> &Path {
pub const X64_LINUX_TARGET_DIR: &str = "x86_64-unknown-linux-gnu/release";
pub const X64_WINDOWS_TARGET_DIR: &str = "x86_64-pc-windows-gnu/release";
pub const X64_MACOS_TARGET_DIR: &str = "x86_64-apple-darwin/release";
match self.os_type {
OsType::Linux => Path::new(X64_LINUX_TARGET_DIR),
OsType::Windows => Path::new(X64_WINDOWS_TARGET_DIR),
OsType::Macos => Path::new(X64_MACOS_TARGET_DIR),
}
}
fn get_aarch64_runner_subdir(&self) -> &Path {
pub const AARCH64_LINUX_TARGET_DIR: &str = "aarch64-unknown-linux-gnu/release";
pub const AARCH64_MACOS_TARGET_DIR: &str = "aarch64-apple-darwin/release";
match self.os_type {
OsType::Linux => Path::new(AARCH64_LINUX_TARGET_DIR),
OsType::Macos => Path::new(AARCH64_MACOS_TARGET_DIR),
_ => unimplemented!(),
}
}
fn get_target_dir(&self) -> PathBuf {
env::var("CARGO_TARGET_DIR")
.unwrap_or_else(|_| "./target".into())
.into()
}
}
#[derive(clap::ValueEnum, Debug, Serialize, Deserialize, Clone, PartialEq, Eq)]
#[serde(rename_all = "snake_case")]
pub enum VmType {
/// QEMU VM
Qemu,
/// Tart VM
Tart,
}
#[derive(clap::ValueEnum, Debug, Serialize, Deserialize, Clone, Copy, PartialEq, Eq)]
#[serde(rename_all = "snake_case")]
pub enum OsType {
Windows,
Linux,
Macos,
}
impl From<OsType> for test_rpc::meta::Os {
fn from(ostype: OsType) -> Self {
match ostype {
OsType::Windows => Self::Windows,
OsType::Linux => Self::Linux,
OsType::Macos => Self::Macos,
}
}
}
#[derive(clap::ValueEnum, Debug, Serialize, Deserialize, Clone, Copy, PartialEq, Eq)]
#[serde(rename_all = "snake_case")]
pub enum PackageType {
Deb,
Rpm,
}
#[derive(clap::ValueEnum, Debug, Serialize, Deserialize, Clone, Copy, PartialEq, Eq)]
#[serde(rename_all = "snake_case")]
pub enum Architecture {
X64,
Aarch64,
}
impl Architecture {
pub fn get_identifiers(self) -> Vec<&'static str> {
match self {
Architecture::X64 => vec!["x86_64", "amd64"],
Architecture::Aarch64 => vec!["arm64", "aarch64"],
}
}
}
#[derive(clap::ValueEnum, Default, Debug, Serialize, Deserialize, Clone, Copy, PartialEq, Eq)]
#[serde(rename_all = "snake_case")]
pub enum Provisioner {
/// Do nothing: The image already includes a test runner service
#[default]
Noop,
/// Set up test runner over SSH.
Ssh,
}
|
