flexor/src/chromeos_install.rs

// Copyright 2023 The ChromiumOS Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

use std::{path::Path, process::Command};

use crate::util::execute_command;
use anyhow::Result;

/// Writes the GPT partition table to disk and writes the stateful partition.
pub fn write_partition_table_and_stateful(disk_path: &Path) -> Result<()> {
    let mut cmd = get_chromeos_install_cmd(disk_path);
    // We skip installing the rootfs and all other partitions beside stateful.
    cmd.arg("--skip_rootfs");
    // Give the path to the PMBR code.
    cmd.arg("--pmbr_code").arg("/usr/share/syslinux/gptmbr.bin");

    execute_command(cmd)
}

/// Installs an image onto a disk using the chromeos_install script.
pub fn install_image_to_disk(disk_path: &Path, image_path: &Path) -> Result<()> {
    let mut cmd = get_chromeos_install_cmd(disk_path);
    // Set the path to the image on disk.
    cmd.arg("--payload_image").arg(image_path);
    // Target BIOS is EFI.
    cmd.arg("--target_bios").arg("efi");
    // Skip writing the partition table since we already put this before.
    cmd.arg("--skip_gpt_creation");

    execute_command(cmd)
}

fn get_chromeos_install_cmd(disk_path: &Path) -> Command {
    let mut cmd = Command::new("/usr/sbin/chromeos-install");
    // Accept everything.
    cmd.arg("--yes");
    // Set the destination path.
    cmd.arg("--dst").arg(disk_path);
    // Skip sanity check for destination being removable.
    cmd.arg("--skip_dst_removable");

    cmd
}