rust - 如何确保在Linux上以root权限运行Rust可执行文件？

Question

我想确保我的可执行文件在Linux下以root权限运行。我想出了这个:

use std::env;
use std::process::exit;

fn main() {
    match env::var("USER") {
        Err(e) => {
            println!("Something went wrong: {:?}", e);
            exit(2);
        }
        Ok(name) => {
            if name != "root" {
                println!("Must be root...");
                exit(1);
            }
        }
    }

    println!("ok!");
}

有没有更简洁/惯用的方式来做到这一点？如何获得有效的用户ID(EUID)？

Answer 1

我发现的东西(以及 libcryptsetup-rs 中使用的东西)是这样的:

use nix::unistd::Uid;

fn main() {
    if !Uid::effective().is_root() {
        panic!("You must run this executable with root permissions");
    }
}

是否 panic!， println!或 exit由程序员决定。