欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

如何判断当前的进程是否以root权限运行?

程序员文章站 2022-03-03 07:51:54
...

当前登录系统的用户叫做实际用户,而实际运行的进程是以有效用户ID来进行访问权限管理的,正常情况下有效用户就是实际用户,但是如果我们使用了sudo或者使能了设置用户ID位,
按照前面文章的介绍,这个有效用户是可能改变的,比如使用sudo执行,进程的有效用户ID就会变为root,那么此进程就相当于具有了root权限。如果我们的程序只允许使用root运行,那么该如何设计呢?

C语言程序设计

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

int main(int argc, char *argv[])
{
    if (geteuid() == 0)
        printf("Run as root, euid:%ld\n", (long) geteuid());
    else
        printf("Not Run as root, euid:%ld\n", (long) geteuid());

    return 0;
}

euid为0说明当前的有效用户ID就是root。运行结果如下:

$ ./euid 
Not Run as root, euid:1000
$ sudo ./euid 
Run as root, euid:0

shell脚本设计

#!/bin/bash

echo ${USER},${UID}

if [ ! x"${USER}" = x"root" ];then
    echo "Please rerun `basename $0` as root" 
    exit 1
else
    echo "Run as root"
fi

shell脚本中可以通过查看环境变量USERUSER和UID来判断是否是使用root权限来运行该脚本。
运行结果如下:

$ ./check_root.sh 
xiehaocheng,1000
Please rerun check_root.sh as root
$ sudo ./check_root.sh 
root,0
Run as root