From mboxrd@z Thu Jan 1 00:00:00 1970 Return-Path: Received: from firstgate.proxmox.com (firstgate.proxmox.com [212.224.123.68]) (using TLSv1.3 with cipher TLS_AES_256_GCM_SHA384 (256/256 bits) key-exchange X25519 server-signature RSA-PSS (2048 bits)) (No client certificate requested) by lists.proxmox.com (Postfix) with ESMTPS id B9E26C26C4 for ; Tue, 23 Jan 2024 18:01:06 +0100 (CET) Received: from firstgate.proxmox.com (localhost [127.0.0.1]) by firstgate.proxmox.com (Proxmox) with ESMTP id 483F3AF4D for ; Tue, 23 Jan 2024 18:01:04 +0100 (CET) Received: from proxmox-new.maurer-it.com (proxmox-new.maurer-it.com [94.136.29.106]) (using TLSv1.3 with cipher TLS_AES_256_GCM_SHA384 (256/256 bits) key-exchange X25519 server-signature RSA-PSS (2048 bits)) (No client certificate requested) by firstgate.proxmox.com (Proxmox) with ESMTPS for ; Tue, 23 Jan 2024 18:00:58 +0100 (CET) Received: from proxmox-new.maurer-it.com (localhost.localdomain [127.0.0.1]) by proxmox-new.maurer-it.com (Proxmox) with ESMTP id A79284926B for ; Tue, 23 Jan 2024 18:00:58 +0100 (CET) From: Aaron Lauterer To: pve-devel@lists.proxmox.com Date: Tue, 23 Jan 2024 18:00:48 +0100 Message-Id: <20240123170053.490250-14-a.lauterer@proxmox.com> X-Mailer: git-send-email 2.39.2 In-Reply-To: <20240123170053.490250-1-a.lauterer@proxmox.com> References: <20240123170053.490250-1-a.lauterer@proxmox.com> MIME-Version: 1.0 Content-Transfer-Encoding: 8bit X-SPAM-LEVEL: Spam detection results: 0 AWL -0.068 Adjusted score from AWL reputation of From: address BAYES_00 -1.9 Bayes spam probability is 0 to 1% DMARC_MISSING 0.1 Missing DMARC policy KAM_DMARC_STATUS 0.01 Test Rule for DKIM or SPF Failure with Strict Alignment SPF_HELO_NONE 0.001 SPF: HELO does not publish an SPF Record SPF_PASS -0.001 SPF: sender matches SPF record T_SCC_BODY_TEXT_LINE -0.01 - Subject: [pve-devel] [PATCH v1 installer 13/18] auto-installer: add auto-installer binary X-BeenThere: pve-devel@lists.proxmox.com X-Mailman-Version: 2.1.29 Precedence: list List-Id: Proxmox VE development discussion List-Unsubscribe: , List-Archive: List-Post: List-Help: List-Subscribe: , X-List-Received-Date: Tue, 23 Jan 2024 17:01:06 -0000 It expects the contents of an answer file via stdin. It will then be parsed and the JSON for the low level installer is generated. It then calls the low level installer directly. The output of the installaton progress is kept rather simple for now. If configured in the answer file, commands will be run pre and post the low level installer. It also logs everything to the logfile, currently '/tmp/auto_installer.log'. Signed-off-by: Aaron Lauterer --- Makefile | 4 +- .../src/bin/proxmox-auto-installer.rs | 192 ++++++++++++++++++ 2 files changed, 195 insertions(+), 1 deletion(-) create mode 100644 proxmox-auto-installer/src/bin/proxmox-auto-installer.rs diff --git a/Makefile b/Makefile index 57bd7ae..9c933d9 100644 --- a/Makefile +++ b/Makefile @@ -18,7 +18,9 @@ INSTALLER_SOURCES=$(shell git ls-files) country.dat PREFIX = /usr BINDIR = $(PREFIX)/bin -USR_BIN := proxmox-tui-installer +USR_BIN := \ + proxmox-tui-installer\ + proxmox-auto-installer COMPILED_BINS := \ $(addprefix $(CARGO_COMPILEDIR)/,$(USR_BIN)) diff --git a/proxmox-auto-installer/src/bin/proxmox-auto-installer.rs b/proxmox-auto-installer/src/bin/proxmox-auto-installer.rs new file mode 100644 index 0000000..8ede670 --- /dev/null +++ b/proxmox-auto-installer/src/bin/proxmox-auto-installer.rs @@ -0,0 +1,192 @@ +use anyhow::{anyhow, bail, Result}; +use log::{error, info, LevelFilter}; +use std::{ + env, + io::{BufRead, BufReader, Write}, + path::PathBuf, + process::ExitCode, +}; + +use proxmox_installer_common::setup::{ + installer_setup, read_json, spawn_low_level_installer, LocaleInfo, RuntimeInfo, SetupInfo, +}; + +use proxmox_auto_installer::{ + answer::Answer, + log::AutoInstLogger, + udevinfo::UdevInfo, + utils, + utils::{parse_answer, LowLevelMessage}, +}; + +static LOGGER: AutoInstLogger = AutoInstLogger; + +pub fn init_log() -> Result<()> { + AutoInstLogger::init("/tmp/auto_installer.log")?; + log::set_logger(&LOGGER) + .map(|()| log::set_max_level(LevelFilter::Info)) + .map_err(|err| anyhow!(err)) +} + +fn auto_installer_setup(in_test_mode: bool) -> Result<(Answer, UdevInfo)> { + let base_path = if in_test_mode { "./testdir" } else { "/" }; + let mut path = PathBuf::from(base_path); + + path.push("run"); + path.push("proxmox-installer"); + + let udev_info: UdevInfo = { + let mut path = path.clone(); + path.push("run-env-udev.json"); + + read_json(&path).map_err(|err| anyhow!("Failed to retrieve udev info details: {err}"))? + }; + + let mut buffer = String::new(); + let lines = std::io::stdin().lock().lines(); + for line in lines { + buffer.push_str(&line.unwrap()); + buffer.push('\n'); + } + + let answer: Answer = + toml::from_str(&buffer).map_err(|err| anyhow!("Failed parsing answer file: {err}"))?; + + Ok((answer, udev_info)) +} + +fn main() -> ExitCode { + if let Err(err) = init_log() { + panic!("could not initilize logging: {}", err); + } + + let in_test_mode = match env::args().nth(1).as_deref() { + Some("-t") => true, + // Always force the test directory in debug builds + _ => cfg!(debug_assertions), + }; + info!("Starting auto installer"); + + let (setup_info, locales, runtime_info) = match installer_setup(in_test_mode) { + Ok(result) => result, + Err(err) => { + error!("Installer setup error: {err}"); + return ExitCode::FAILURE; + } + }; + + let (answer, udevadm_info) = match auto_installer_setup(in_test_mode) { + Ok(result) => result, + Err(err) => { + error!("Autoinstaller setup error: {err}"); + return ExitCode::FAILURE; + } + }; + + match utils::run_cmds("Pre", &answer.global.pre_command) { + Ok(_) => (), + Err(err) => { + error!("Error when running Pre-Commands: {}", err); + return exit_failure(answer.global.reboot_on_error); + } + }; + match run_installation(&answer, &locales, &runtime_info, &udevadm_info, &setup_info) { + Ok(_) => info!("Installation done."), + Err(err) => { + error!("Installation failed: {err}"); + return exit_failure(answer.global.reboot_on_error); + } + } + match utils::run_cmds("Post", &answer.global.post_command) { + Ok(_) => (), + Err(err) => { + error!("Error when running Post-Commands: {}", err); + return exit_failure(answer.global.reboot_on_error); + } + }; + ExitCode::SUCCESS +} + +/// When we exit with a failure, the installer will not automatically reboot. +/// Default value for reboot_on_error is false +fn exit_failure(reboot_on_error: Option) -> ExitCode { + if let Some(true) = reboot_on_error { + ExitCode::SUCCESS + } else { + ExitCode::FAILURE + } +} + +fn run_installation( + answer: &Answer, + locales: &LocaleInfo, + runtime_info: &RuntimeInfo, + udevadm_info: &UdevInfo, + setup_info: &SetupInfo, +) -> Result<()> { + let config = parse_answer(answer, udevadm_info, runtime_info, locales, setup_info)?; + info!("Calling low-level installer"); + + let mut child = match spawn_low_level_installer(false) { + Ok(child) => child, + Err(err) => { + bail!("Low level installer could not be started: {}", err); + } + }; + + let mut cur_counter = 111; + let mut inner = || -> Result<()> { + let reader = child + .stdout + .take() + .map(BufReader::new) + .ok_or(anyhow!("failed to get stdout reader"))?; + let mut writer = child + .stdin + .take() + .ok_or(anyhow!("failed to get stdin writer"))?; + + serde_json::to_writer(&mut writer, &config) + .map_err(|err| anyhow!("failed to serialize install config: {err}"))?; + writeln!(writer).map_err(|err| anyhow!("failed to write install config: {err}"))?; + + for line in reader.lines() { + let line = match line { + Ok(line) => line, + Err(_) => break, + }; + let msg = match serde_json::from_str::(&line) { + Ok(msg) => msg, + Err(_) => { + // Not a fatal error, so don't abort the installation by returning + continue; + } + }; + + match msg.clone() { + LowLevelMessage::Info { message } => info!("{message}"), + LowLevelMessage::Error { message } => error!("{message}"), + LowLevelMessage::Prompt { query } => info!("Got Query: {query}"), + LowLevelMessage::Progress { ratio, text: _ } => { + let counter = (ratio * 100.).floor() as usize; + if counter != cur_counter { + cur_counter = counter; + info!("Progress: {counter:>3}%"); + } + } + LowLevelMessage::Finished { state, message } => { + if state == "err" { + bail!("{message}"); + } else { + info!("Finished: '{state}' {message}"); + } + } + }; + } + Ok(()) + }; + match inner() { + Err(err) => bail!("low level installer returned early: {err}"), + _ => Ok(()), + } +} -- 2.39.2