lichen_cli: Require root to run

Signed-off-by: Ikey Doherty <[email protected]>
AerynOS · Aug 1, 2024 · ee8f814 · ee8f814
1 parent 0cf636c
commit ee8f814
Show file tree

Hide file tree

Showing 3 changed files with 29 additions and 2 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/lichen_cli/Cargo.toml b/lichen_cli/Cargo.toml
@@ -13,3 +13,4 @@ futures.workspace = true
 tokio.workspace = true
 installer = { path = "../crates/installer" }
 console = "0.15.8"
+nix = "0.29"
diff --git a/lichen_cli/src/main.rs b/lichen_cli/src/main.rs
@@ -11,11 +11,13 @@ use std::{
     time::Duration,
 };
 
+use color_eyre::eyre::bail;
 use console::{set_colors_enabled, style};
 use crossterm::style::Stylize;
 use dialoguer::theme::ColorfulTheme;
 use indicatif::ProgressStyle;
 use installer::{selections, steps::Context, systemd, Account, BootPartition, Installer, Locale, SystemPartition};
+use nix::libc::geteuid;
 use tokio::io::AsyncWriteExt;
 use tokio::process::Command;
 
@@ -151,6 +153,11 @@ async fn main() -> color_eyre::Result<()> {
     color_eyre::install().unwrap();
     set_colors_enabled(true);
 
+    let euid = unsafe { geteuid() };
+    if euid != 0 {
+        bail!("lichen must be run as root. Re-run with sudo")
+    }
+
     // Test selection management, force GNOME
     let selections = selections::Manager::new().with_groups([
         selections::Group::from_str(include_str!("../../selections/base.json"))?,