How to check if my program was run with sudo?

在 Linux 和 Unix 系统中，sudo 命令允许普通用户以超级用户权限执行程序。然而，对于开发者而言，确保程序在正确的权限环境下运行至关重要——尤其是在处理系统配置、写入敏感文件或执行高危操作时。如果程序在非预期的高权限下运行，可能带来安全风险；反之，若需要 root 权限却未获得，则会导致功能异常。那么，如何在自己的程序中准确判断是否被 sudo 调用？本文将为你梳理几种主流的检测方法及背后的原理。

为何需要检测 sudo 运行状态？

许多命令行工具和脚本在启动时需要进行权限校验。例如： - 一个系统清理工具若以普通用户身份运行，可能无法删除日志文件，此时应提示用户使用 sudo。 - 一个日志分析程序若被误用 sudo 运行，可能意外修改只读文件，造成安全隐患。 - 安全敏感的守护进程（如防火墙配置脚本）需要拒绝非 root 执行，或仅在 root 下加载核心模块。

因此，程序内部主动检测 sudo 状态，可以提升用户体验、增强安全性，并避免因权限不足导致的错误中断。

主流检测方法详解

1. 检查 UID 与环境变量（适用于 C/C++/Shell）

在类 Unix 系统中，getuid() 函数返回当前进程的真实用户 ID。如果程序使用 sudo 运行，getuid() 通常返回 0（root）。但需要注意：sudo 默认会保留调用者的部分环境变量，比如 SUDO_USER、SUDO_UID 和 SUDO_GID。这些变量记录了原始用户的身份。

示例（Shell 脚本）：

if [ "$EUID" -eq 0 ]; then
    echo "以 root 身份运行"
    if [ -n "$SUDO_USER" ]; then
        echo "通过 sudo 调用，原始用户：$SUDO_USER"
    else
        echo "直接以 root 登录"
    fi
else
    echo "以普通用户身份运行"
fi

示例（C 语言）：

#include <stdio.h>
#include <unistd.h>
#include <stdlib.h>

int main() {
    uid_t uid = getuid();
    char *sudo_user = getenv("SUDO_USER");
    if (uid == 0) {
        if (sudo_user) {
            printf("程序通过 sudo 运行，原始用户: %s\n", sudo_user);
        } else {
            printf("程序由 root 直接执行\n");
        }
    } else {
        printf("程序以普通用户运行，UID: %d\n", uid);
    }
    return 0;
}

优点：简单直接，无需额外系统调用。
缺点：环境变量 SUDO_USER 可能被手动清除或伪造（但提升权限后无法伪造 root 身份），且某些 sudo 配置（如 env_reset）会清空这些变量。因此，更可靠的方法是通过进程关系判断。

2. 检查父进程信息（更严谨的方法）

sudo 在启动一个程序时，其父进程通常是 sudo 本身。通过读取 /proc 文件系统，可以获取父进程的命令行和名称。

Python 示例：

import os
import sys

def is_run_with_sudo():
    try:
        # 获取当前进程的父进程 PID
        ppid = os.getppid()
        # 读取父进程的命令行信息
        with open(f'/proc/{ppid}/comm', 'r') as f:
            parent_comm = f.read().strip()
        if parent_comm == 'sudo':
            return True
        # 进一步检查父进程的命令行参数（可选）
        with open(f'/proc/{ppid}/cmdline', 'rb') as f:
            cmdline = f.read().replace(b'\x00', b' ').decode()
        if 'sudo' in cmdline:
            return True
    except FileNotFoundError:
        # 兼容不支持 /proc 的系统
        pass
    return False

if __name__ == '__main__':
    if is_run_with_sudo():
        print("程序被 sudo 调用")
    else:
        print("未使用 sudo")

注意：部分系统上，sudo 可能以 /usr/bin/sudo 路径调用，或者使用 doas 等替代工具。此外，如果父进程已退出或程序被后台化，此方法可能失效。

3. 直接校验有效用户 ID（最简洁）

许多编程语言提供直接获取有效用户 ID（EUID）的接口。如果 EUID 为 0，通常意味着程序拥有 root 权限（无论是否通过 sudo）。这在大多数场景下已经足够，因为开发者关心的往往是“是否有权限做某事”，而非“如何获得权限”。

Python 示例：

import os
if os.geteuid() == 0:
    print("以 root 权限运行")
else:
    print("非 root 用户")

Go 示例：

package main

import (
    "fmt"
    "os/user"
)

func main() {
    u, err := user.Current()
    if err != nil {
        panic(err)
    }
    if u.Uid == "0" {
        fmt.Println("以 root 运行")
    } else {
        fmt.Println("普通用户")
    }
}

优点：跨平台性较好（部分语言在 Windows 下也有类似函数）。
缺点：无法区分 sudo 与直接 root 登录，也无法得知原始用户身份。

4. 对于图形界面或跨平台应用

若程序需要在多种操作系统上运行，建议使用封装好的权限检测库。例如，Rust 的 sudo-plugin crate，或 Python 的 pwd 模块结合 os 模块。对于 GUI 应用，还需考虑 pkexec（PolicyKit）或其他提权机制，此时检测方式需相应调整。

安全注意事项

不要单纯信任环境变量：攻击者可在低权限下设置 SUDO_USER 变量，虽然本程序以高权限运行时该变量会被覆盖，但应避免依赖此变量作出敏感决策，例如授权操作。
区分“真正需要 root”的情况：有时程序只需部分特权（如绑定低端口），而非全部 root 权限。考虑使用 capabilities 机制或设置 setuid 位，而非直接要求 sudo。
替代工具 doas：OpenBSD 的 doas 或 Linux 上的 opendoas 不设置 SUDO_USER 环境变量。如果你的系统可能使用 doas，需额外检测其相关进程名。

总结

检测程序是否通过 sudo 运行，核心思路是检查进程的权限状态和父进程信息。对于绝大多数脚本和工具，使用 geteuid() 或 os.geteuid() 判断是否为 root 即可满足需求；若需要精确区分是否由 sudo 唤起，推荐组合使用环境变量和 /proc 文件系统。开发者应根据目标用户群体的操作系统和安全性要求，选择最合适的方案。

无论采用哪种方法，都建议在程序启动时主动输出提示信息，让用户明确当前权限状态。毕竟，清晰透明的权限提示，往往比错误堆栈更能赢得用户的信任。

文末提示：本文提及的所有代码片段均可在 Linux 环境下测试。如果你使用的是 macOS，注意其部分系统机制有所差异，但 sudo 环境变量依然有效。欢迎在评论区分享你遇到的实际案例或更优雅的检测方法。