src/bin/rdcore/cmdline.rs - main - githubmirror/coreos-installer

src/bin/rdcore/cmdline.rs
161 строка5 KB
Nikita Dubrovskii
s390x: Support multiple hostkeys in rdcore zipl
26 июн 2026, 16:49
26 июн 2026, 16:49bee4500
// Copyright 2020 CoreOS, Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

// For consistency, have all parse_*() functions return Result.
#![allow(clippy::unnecessary_wraps)]

use clap::Parser;

#[cfg(target_arch = "s390x")]
use libcoreinst::s390x;

// Args are listed in --help in the order declared in these structs/enums.

#[derive(Debug, Parser)]
#[command(name = "rdcore", version)]
#[command(args_conflicts_with_subcommands = true)]
#[command(disable_help_subcommand = true)]
#[command(help_expected = true)]
pub enum Cmd {
    /// Generate rootmap kargs and optionally inject into BLS configs
    Rootmap(RootmapConfig),
    /// Generate bootmap kargs and binds bootfs to rootfs and GRUB
    BindBoot(BindBootConfig),
    /// Modify kargs in BLS configs
    Kargs(KargsConfig),
    /// Copy data from stdin to stdout, checking piecewise hashes
    StreamHash(StreamHashConfig),
    /// Checks there is only one filesystem with given label
    VerifyUniqueFsLabel(VerifyUniqueFsLabelConfig),
    #[cfg(target_arch = "s390x")]
    /// Runs zipl
    Zipl(ZiplConfig),
}

#[derive(Debug, Parser)]
pub struct RootmapConfig {
    // we allow mounting /boot ourselves (--boot-device) or letting our
    // caller do the mount and point us to it (--boot-mount); lots of
    // backstory on /boot mounting in the initrd, so leave some flexibility
    // for changing implementation details on the OS side without having to
    // respin rdcore
    /// Boot device containing BLS entries to modify
    #[arg(long, value_name = "DEVPATH", conflicts_with = "boot_mount")]
    pub boot_device: Option<String>,
    /// Boot mount containing BLS entries to modify
    #[arg(long, value_name = "BOOT_MOUNT", conflicts_with = "boot_device")]
    pub boot_mount: Option<String>,
    /// Path to rootfs mount
    #[arg(value_name = "ROOT_MOUNT")]
    pub root_mount: String,
}

#[derive(Debug, Parser)]
pub struct BindBootConfig {
    /// Path to rootfs mount
    #[arg(value_name = "ROOT_MOUNT")]
    pub root_mount: String,
    /// Path to bootfs mount
    #[arg(value_name = "BOOT_MOUNT")]
    pub boot_mount: String,
}

#[derive(Debug, Parser)]
pub struct KargsConfig {
    // see comment block in rootmap command above
    /// Boot device containing BLS entries to modify
    #[arg(long, value_name = "DEVPATH")]
    #[arg(conflicts_with_all = ["boot_mount", "current"])]
    pub boot_device: Option<String>,
    /// Boot mount containing BLS entries to modify
    #[arg(long, value_name = "BOOT_MOUNT")]
    #[arg(conflicts_with_all = ["boot_device", "current"])]
    pub boot_mount: Option<String>,
    /// Dry run using kargs from this boot
    #[arg(long)]
    #[arg(conflicts_with_all = ["boot_device", "boot_mount"])]
    pub current: bool,
    /// Modify this option string instead of fetching from BLS entry
    // this is purely for dev testing
    #[arg(long, value_name = "OPTIONS", hide = true)]
    pub override_options: Option<String>,
    /// File to create if BLS entry was modified
    #[arg(long, value_name = "PATH")]
    pub create_if_changed: Option<String>,
    /// Append kernel arg
    #[arg(long, value_name = "ARG")]
    pub append: Vec<String>,
    /// Append kernel arg if missing
    #[arg(long, value_name = "ARG")]
    #[arg(alias = "should-exist")]
    pub append_if_missing: Vec<String>,
    /// Delete kernel arg
    #[arg(long, value_name = "ARG")]
    #[arg(alias = "should-not-exist")]
    pub delete: Vec<String>,
}

#[derive(Debug, Parser)]
pub struct StreamHashConfig {
    /// Path to the piecewise hash file
    #[arg(value_name = "hash-file")]
    pub hash_file: String,
}

#[derive(Debug, Parser)]
pub struct VerifyUniqueFsLabelConfig {
    /// Filesystem's label
    #[arg(value_name = "LABEL")]
    pub label: String,

    /// Force rereading of partition table
    #[arg(long)]
    pub rereadpt: bool,
}

#[cfg(target_arch = "s390x")]
#[derive(Debug, Parser)]
pub struct ZiplConfig {
    /// Boot device containing BLS entries to use
    #[arg(long, value_name = "BOOT_MOUNT")]
    pub boot_mount: String,

    /// Zipl mode for Secure Execution
    #[arg(long, value_enum, default_value = "auto")]
    pub secex_mode: s390x::ZiplSecexMode,

    /// Path to hostkey
    #[arg(short = 'k', long = "hostkey", value_name = "HOSTKEY")]
    pub hostkeys: Option<Vec<String>>,

    /// Append kernel argument
    #[arg(long, value_name = "KARG")]
    #[arg(alias = "kargs")]
    pub append_karg: Option<Vec<String>>,

    /// Append file to sdboot image
    #[arg(long, value_name = "FILE")]
    pub append_file: Option<Vec<String>>,
}

#[cfg(test)]
mod test {
    use super::*;
    use clap::CommandFactory;

    #[test]
    fn clap_app() {
        Cmd::command().debug_assert()
    }
}