86 lines
1.8 KiB
Rust
86 lines
1.8 KiB
Rust
use std::path::{Path, PathBuf};
|
|
|
|
use anyhow::Result;
|
|
use clap::{Parser, Subcommand};
|
|
|
|
use crate::install;
|
|
|
|
#[derive(Parser)]
|
|
pub struct Cli {
|
|
#[clap(subcommand)]
|
|
pub commands: Commands,
|
|
}
|
|
|
|
#[derive(Subcommand)]
|
|
pub enum Commands {
|
|
Install {
|
|
// Secure Boot Public Key
|
|
#[clap(long)]
|
|
public_key: PathBuf,
|
|
|
|
// Secure Boot Private Key
|
|
#[clap(long)]
|
|
private_key: PathBuf,
|
|
|
|
// Secure Boot PKI Bundle for auto enrolling key
|
|
#[clap(long)]
|
|
pki_bundle: Option<PathBuf>,
|
|
|
|
// Enable auto enrolling your keys in UEFI
|
|
// Be aware that this might irrevocably brick your device
|
|
#[clap(long, default_value = "false")]
|
|
auto_enroll: bool,
|
|
|
|
bootspec: PathBuf,
|
|
},
|
|
}
|
|
|
|
impl Cli {
|
|
pub fn call(self) -> Result<()> {
|
|
self.commands.call()
|
|
}
|
|
}
|
|
|
|
impl Commands {
|
|
pub fn call(self) -> Result<()> {
|
|
match self {
|
|
Commands::Install {
|
|
public_key,
|
|
private_key,
|
|
pki_bundle,
|
|
auto_enroll,
|
|
bootspec,
|
|
} => install(
|
|
&public_key,
|
|
&private_key,
|
|
pki_bundle,
|
|
auto_enroll,
|
|
&bootspec,
|
|
),
|
|
}
|
|
}
|
|
}
|
|
|
|
fn install(
|
|
public_key: &Path,
|
|
private_key: &Path,
|
|
pki_bundle: Option<PathBuf>,
|
|
auto_enroll: bool,
|
|
bootspec: &Path,
|
|
) -> Result<()> {
|
|
let lanzaboote_stub = std::env::var("LANZABOOTE_STUB")?;
|
|
let initrd_stub = std::env::var("LANZABOOTE_INITRD_STUB")?;
|
|
let sbsigntool = std::env::var("SBSIGNTOOL")?;
|
|
|
|
install::install(
|
|
public_key,
|
|
private_key,
|
|
pki_bundle,
|
|
auto_enroll,
|
|
bootspec,
|
|
Path::new(&sbsigntool),
|
|
Path::new(&lanzaboote_stub),
|
|
Path::new(&initrd_stub)
|
|
)
|
|
}
|