xtask: Avoid lossy conversions, spurious vectors

reviewable/pr640/r1
Tamir Duberstein 1 year ago
parent e93e3c4a55
commit de1fe1f5e1
No known key found for this signature in database

@ -1,4 +1,8 @@
use std::{os::unix::process::CommandExt, path::PathBuf, process::Command}; use std::{
os::unix::process::CommandExt,
path::{Path, PathBuf},
process::Command,
};
use anyhow::Context as _; use anyhow::Context as _;
use clap::Parser; use clap::Parser;
@ -18,28 +22,26 @@ pub struct Options {
pub runner: String, pub runner: String,
/// libbpf directory /// libbpf directory
#[clap(long, action)] #[clap(long, action)]
pub libbpf_dir: String, pub libbpf_dir: PathBuf,
/// Arguments to pass to your application /// Arguments to pass to your application
#[clap(name = "args", last = true)] #[clap(name = "args", last = true)]
pub run_args: Vec<String>, pub run_args: Vec<String>,
} }
/// Build the project /// Build the project
fn build(opts: &Options) -> Result<(), anyhow::Error> { fn build(release: bool) -> Result<(), anyhow::Error> {
let mut args = vec!["build"]; let mut cmd = Command::new("cargo");
if opts.release { cmd.arg("build").arg("-p").arg("integration-test");
args.push("--release") if release {
cmd.arg("--release");
} }
args.push("-p");
args.push("integration-test"); let status = cmd.status().expect("failed to build userspace");
let status = Command::new("cargo")
.args(&args)
.status()
.expect("failed to build userspace");
match status.code() { match status.code() {
Some(code) => match code { Some(code) => match code {
0 => Ok(()), 0 => Ok(()),
code => Err(anyhow::anyhow!("exited with status code: {code}")), code => Err(anyhow::anyhow!("{cmd:?} exited with status code: {code}")),
}, },
None => Err(anyhow::anyhow!("process terminated by signal")), None => Err(anyhow::anyhow!("process terminated by signal")),
} }
@ -47,30 +49,33 @@ fn build(opts: &Options) -> Result<(), anyhow::Error> {
/// Build and run the project /// Build and run the project
pub fn run(opts: Options) -> Result<(), anyhow::Error> { pub fn run(opts: Options) -> Result<(), anyhow::Error> {
let Options {
bpf_target,
release,
runner,
libbpf_dir,
run_args,
} = opts;
// build our ebpf program followed by our application // build our ebpf program followed by our application
build_ebpf(BuildOptions { build_ebpf(BuildOptions {
target: opts.bpf_target, target: bpf_target,
libbpf_dir: PathBuf::from(&opts.libbpf_dir), libbpf_dir,
}) })
.context("Error while building eBPF program")?; .context("Error while building eBPF program")?;
build(&opts).context("Error while building userspace application")?; build(release).context("Error while building userspace application")?;
// profile we are building (release or debug) // profile we are building (release or debug)
let profile = if opts.release { "release" } else { "debug" }; let profile = if release { "release" } else { "debug" };
let bin_path = format!("target/{profile}/integration-test"); let bin_path = Path::new("target").join(profile).join("integration-test");
// arguments to pass to the application let mut args = runner.trim().split_terminator(' ');
let mut run_args: Vec<_> = opts.run_args.iter().map(String::as_str).collect();
// configure args let mut cmd = Command::new(args.next().expect("No first argument"));
let mut args: Vec<_> = opts.runner.trim().split_terminator(' ').collect(); cmd.args(args).arg(bin_path).args(run_args);
args.push(bin_path.as_str());
args.append(&mut run_args);
// spawn the command // spawn the command
let err = Command::new(args.first().expect("No first argument")) let err = cmd.exec();
.args(args.iter().skip(1))
.exec();
// we shouldn't get here unless the command failed to spawn // we shouldn't get here unless the command failed to spawn
Err(anyhow::Error::from(err).context(format!("Failed to run `{}`", args.join(" ")))) Err(anyhow::Error::from(err).context(format!("Failed to run `{cmd:?}`")))
} }

Loading…
Cancel
Save