You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
56 lines
1.7 KiB
Rust
56 lines
1.7 KiB
Rust
use std::process::Command;
|
|
|
|
use anyhow::Context as _;
|
|
use clap::Parser;
|
|
|
|
use crate::{build::{build, Options as BuildOptions}, build_ebpf::Architecture};
|
|
|
|
#[derive(Debug, Parser)]
|
|
pub struct Options {
|
|
/// Set the endianness of the BPF target
|
|
#[clap(default_value = "bpfel-unknown-none", long)]
|
|
pub bpf_target: Architecture,
|
|
/// Build and run the release target
|
|
#[clap(long)]
|
|
pub release: bool,
|
|
/// The command used to wrap your application
|
|
#[clap(short, long, default_value = "sudo -E")]
|
|
pub runner: String,
|
|
/// Arguments to pass to your application
|
|
#[clap(name = "args", last = true)]
|
|
pub run_args: Vec<String>,
|
|
}
|
|
|
|
|
|
/// Build and run the project
|
|
pub fn run(opts: Options) -> Result<(), anyhow::Error> {
|
|
// Build our ebpf program and the project
|
|
build(BuildOptions{
|
|
bpf_target: opts.bpf_target,
|
|
release: opts.release,
|
|
}).context("Error while building project")?;
|
|
|
|
// profile we are building (release or debug)
|
|
let profile = if opts.release { "release" } else { "debug" };
|
|
let bin_path = format!("target/{profile}/{{project-name}}");
|
|
|
|
// arguments to pass to the application
|
|
let mut run_args: Vec<_> = opts.run_args.iter().map(String::as_str).collect();
|
|
|
|
// configure args
|
|
let mut args: Vec<_> = opts.runner.trim().split_terminator(' ').collect();
|
|
args.push(bin_path.as_str());
|
|
args.append(&mut run_args);
|
|
|
|
// run the command
|
|
let status = Command::new(args.first().expect("No first argument"))
|
|
.args(args.iter().skip(1))
|
|
.status()
|
|
.expect("failed to run the command");
|
|
|
|
if !status.success() {
|
|
anyhow::bail!("Failed to run `{}`", args.join(" "));
|
|
}
|
|
Ok(())
|
|
}
|