Add root check

jeevithakannan2 · Oct 13, 2024 · 4548922 · 4548922
1 parent 79eb752
commit 4548922
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 0 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/tui/Cargo.toml b/tui/Cargo.toml
@@ -30,6 +30,7 @@ tree-sitter-highlight = "0.24.2"
 tree-sitter-bash = "0.23.1"
 anstyle = "1.0.8"
 ansi-to-tui = "6.0.0"
+sudo = "0.6.0"
 zips = "0.1.7"
 
 [build-dependencies]

diff --git a/tui/src/main.rs b/tui/src/main.rs
@@ -36,6 +36,11 @@ struct Args {
 }
 
 fn main() -> io::Result<()> {
+    if sudo::check() != sudo::RunningAs::User {
+        eprintln!("This program should not be run as root");
+        std::process::exit(1);
+    }
+
     let args = Args::parse();
 
     let mut state = AppState::new(args.theme, args.override_validation);