feat(xtask): add ci subcommand

Signed-off-by: Martin Kröning <[email protected]>

xtask/Cargo.toml

@@ -8,4 +8,7 @@ anyhow = "1.0"
 clap = { version = "4", features = ["derive"] }
 goblin = { version = "0.7", default-features = false, features = ["archive", "elf32", "elf64", "std"] }
 llvm-tools = "0.1"
+sysinfo = "0.29"
+ureq = "2"
+wait-timeout = "0.2"
 xshell = "0.2"

xtask/src/arch.rs

@@ -60,6 +60,19 @@ impl Arch {
 		}
 	}
 
+	pub fn ci_cargo_args(&self) -> &'static [&'static str] {
+		match self {
+			Arch::X86_64 => &[
+				"--target=x86_64-unknown-hermit",
+				"-Zbuild-std=std,panic_abort",
+			],
+			Arch::Aarch64 => &[
+				"--target=aarch64-unknown-hermit",
+				"-Zbuild-std=std,panic_abort",
+			],
+		}
+	}
+
 	pub fn rustflags(&self) -> &'static [&'static str] {
 		match self {
 			Self::X86_64 => &[],

xtask/src/artifact.rs

@@ -80,4 +80,16 @@ impl Artifact {
 		.collect::<PathBuf>()
 		.into()
 	}
+
+	pub fn ci_image(&self, package: &str) -> PathBuf {
+		[
+			"..".as_ref(),
+			self.target_dir(),
+			self.arch.hermit_triple().as_ref(),
+			self.profile_path_component().as_ref(),
+			package.as_ref(),
+		]
+		.iter()
+		.collect()
+	}
 }

xtask/src/ci/build.rs

@@ -0,0 +1,45 @@
+use std::path::PathBuf;
+
+use anyhow::Result;
+use clap::Args;
+use xshell::cmd;
+
+use crate::cargo_build::{CargoBuild, CmdExt};
+
+/// Build hermit-rs images.
+#[derive(Args)]
+#[command(next_help_heading = "Build options")]
+pub struct Build {
+	#[command(flatten)]
+	pub cargo_build: CargoBuild,
+
+	/// Package to build (see `cargo help pkgid`)
+	#[arg(short, long, id = "SPEC")]
+	pub package: String,
+}
+
+impl Build {
+	pub fn run(&self) -> Result<()> {
+		if super::in_ci() {
+			eprintln!("::group::cargo build")
+		}
+
+		let sh = crate::sh()?;
+
+		cmd!(sh, "cargo build --manifest-path ../Cargo.toml")
+			.args(self.cargo_build.artifact.arch.ci_cargo_args())
+			.cargo_build_args(&self.cargo_build)
+			.args(&["--package", self.package.as_str()])
+			.run()?;
+
+		if super::in_ci() {
+			eprintln!("::endgroup::")
+		}
+
+		Ok(())
+	}
+
+	pub fn image(&self) -> PathBuf {
+		self.cargo_build.artifact.ci_image(&self.package)
+	}
+}

xtask/src/ci/firecracker.rs

@@ -0,0 +1,42 @@
+use std::path::Path;
+
+use anyhow::Result;
+use clap::Args;
+use xshell::cmd;
+
+use super::build::Build;
+
+/// Run hermit-rs images on Firecracker.
+#[derive(Args)]
+pub struct Firecracker {
+	#[command(flatten)]
+	build: Build,
+}
+
+impl Firecracker {
+	pub fn run(self) -> Result<()> {
+		self.build.run()?;
+
+		let sh = crate::sh()?;
+
+		let config = format!(
+			include_str!("firecracker_vm_config.json"),
+			kernel_image_path = "../rusty-loader-x86_64-fc",
+			initrd_path = self.build.image().display()
+		);
+		eprintln!("firecracker config");
+		eprintln!("{config}");
+		let config_path = Path::new("firecracker_vm_config.json");
+		sh.write_file(config_path, config)?;
+
+		let log_path = Path::new("firecracker.log");
+		sh.write_file(log_path, "")?;
+		cmd!(sh, "firecracker --no-api --config-file {config_path} --log-path {log_path} --level Info --show-level --show-log-origin").run()?;
+		let log = sh.read_file(log_path)?;
+
+		eprintln!("firecracker log");
+		eprintln!("{log}");
+
+		Ok(())
+	}
+}

xtask/src/ci/firecracker_vm_config.json

@@ -0,0 +1,13 @@
+{{
+  "boot-source": {{
+    "kernel_image_path": "{kernel_image_path}",
+    "initrd_path": "{initrd_path}",
+    "boot_args": ""
+  }},
+  "drives": [],
+  "machine-config": {{
+    "vcpu_count": 1,
+    "mem_size_mib": 256,
+    "smt": false
+  }}
+}}

xtask/src/ci/mod.rs

@@ -0,0 +1,31 @@
+use anyhow::Result;
+use clap::Subcommand;
+
+mod build;
+mod firecracker;
+mod qemu;
+mod uhyve;
+
+/// Run CI tasks.
+#[derive(Subcommand)]
+pub enum Ci {
+	Build(build::Build),
+	Firecracker(firecracker::Firecracker),
+	Qemu(qemu::Qemu),
+	Uhyve(uhyve::Uhyve),
+}
+
+impl Ci {
+	pub fn run(self) -> Result<()> {
+		match self {
+			Self::Build(build) => build.run(),
+			Self::Firecracker(firecracker) => firecracker.run(),
+			Self::Qemu(qemu) => qemu.run(),
+			Self::Uhyve(uhyve) => uhyve.run(),
+		}
+	}
+}
+
+fn in_ci() -> bool {
+	std::env::var_os("CI") == Some("true".into())
+}