Làm thế nào một kịch bản có thể kiểm tra nếu nó đang được chạy như root?

Tôi đang viết một tập lệnh bash đơn giản, nhưng tôi cần nó để kiểm tra xem nó có được chạy dưới quyền root hay không. Tôi biết có lẽ có một cách rất đơn giản để làm điều đó, nhưng tôi không biết làm thế nào.

Nói rõ hơn: Cách đơn giản để viết tập lệnh foo.sh , sao cho lệnh ./foo.shxuất ra 0và lệnh sudo ./foo.shxuất ra 1?

#!/bin/bash
if [[ $EUID -ne 0 ]]; then
   echo "This script must be run as root" 
   exit 1
fi

Người dùng root không phải đặt tên là "root". whoamitrả về tên người dùng đầu tiên với ID người dùng 0. $USERchứa tên của người dùng đã đăng nhập, có thể có ID người dùng 0, nhưng có một tên khác.

Chương trình đáng tin cậy duy nhất để kiểm tra xem tài khoản đã đăng nhập bằng root hay chưa:

id -u

Tôi sử dụng -ucho ID người dùng hiệu quả , không phải -rcho ID người dùng thực . Quyền được xác định bởi ID người dùng hiệu quả , không phải ID thực .

Xét nghiệm

/etc/passwdchứa các tên người dùng sau với ID người dùng 0theo thứ tự đã cho:

rootx
root2

Đăng nhập như là root2, cho kết quả tiếp theo:

• whoami: rootx
• echo $USER: root2(điều này trả về một chuỗi rỗng nếu chương trình được bắt đầu trong một môi trường trống, vd env -i sh -c 'echo $USER')
• id -u: 0 Như bạn có thể thấy, các chương trình khác đã thất bại trong kiểm tra này, chỉ id -uđược thông qua.

Kịch bản cập nhật sẽ trông như thế này:

#!/bin/bash
if ! [ $(id -u) = 0 ]; then
   echo "I am not root!"
   exit 1
fi

Như @Lekensteyn nói rằng bạn nên sử dụng ID người dùng hiệu quả. Bạn không cần phải gọi id -utrong bash:

#!/bin/bash

if [[ $EUID -ne 0 ]]; then
   echo "You must be root to do this." 1>&2
   exit 100
fi

Đề xuất của @ geirha từ các ý kiến ​​sử dụng đánh giá số học:

#!/bin/bash

if (( EUID != 0 )); then
   echo "You must be root to do this." 1>&2
   exit 100
fi

Bạn có thể thực hiện điều này bằng cách sử dụng whoamilệnh, trả về người dùng hiện tại:

#!/bin/bash

if [ `whoami` != 'root' ]
  then
    echo "You must be root to do this."
    exit
fi

...

Chạy ở trên sẽ in You must be root to do this.nếu người dùng hiện tại không root.

Lưu ý: một sự thay thế trong một số trường hợp là chỉ cần kiểm tra $USERbiến:

if [ $USER != 'root' ]

Cân nhắc hiệu quả, trước tiên, bạn có thể kiểm tra EUIDbiến môi trường và sau đó, nếu nó không tồn tại, hãy gọi idlệnh tiêu chuẩn :

if ((${EUID:-0} || "$(id -u)")); then
  echo You are not root.
else
  echo Hello, root.
fi

Bằng cách này, do phím tắt OR , bạn tránh gọi lệnh hệ thống, ưu tiên truy vấn của biến trong bộ nhớ.

Tái sử dụng mã:

function amIRoot() {
  ! ((${EUID:-0} || "$(id -u)"))
}

#!/bin/bash
[[ $(id -u) != 0 ]] 
echo $?

Một cách đơn giản để làm cho tập lệnh chỉ có thể chạy được bằng root là bắt đầu tập lệnh với dòng:
#!/bin/su root

#!/bin/bash
uid=`id -u`
if [ "$uid" == "0" ]
then
    echo 1
else
    echo 0
fi

Đoạn trích này sẽ:

• Kiểm tra các phiên khác nhau
• đề nghị sử dụng sudo !!
• và trả lại lỗi

if ["$ (whoami & 2> / dev / null)"! = "root"] && ["$ (id -un & 2> / dev / null)"! = "root"]
      sau đó
      echo "Bạn phải root để chạy tập lệnh này!"
      tiếng vang "sử dụng 'sudo !!'"
      thoát 1
fi

Câu trả lời này chỉ là để lưu một ý tưởng có thể hữu ích cho một số bạn. Nếu bạn cần tập lệnh chạy từ GUI trên máy tính để bàn và yêu cầu quyền root, hãy thử theo cách này:

#!/bin/bash
if ! [ $(id -u) = 0 ]; then
   gksudo -w $0 $@
   exit
fi

#here go superuser commands, e.g. (switch intel cpu into powersave mode):
echo 1 > /sys/devices/system/cpu/intel_pstate/no_turbo
cpupower frequency-set -g powersave

Bằng cách này, bạn sẽ nhận được cửa sổ hộp thoại đẹp yêu cầu bạn nhập mật khẩu người dùng root. Giống như với sudo dòng lệnh.

Có gksudothể không có sẵn trong hệ thống của bạn sau đó cài đặt nó với sudo apt-get install gksu.

Mã này hoạt động theo cả Bash và stock Bourne sh (được thử nghiệm theo FreeBSD) không xác định EUID. Nếu EUID tồn tại, giá trị của nó được trả về, nếu không lệnh 'id' sẽ được chạy. Nó ngắn hơn một chút so với ví dụ "hoặc" ở trên, vì nó sử dụng lớp vỏ tích hợp ": -".

Root trong sh:

# echo $EUID

# euid=${EUID:-`id -u`}
# echo $euid
0

In bash:
[chaapala@tenfwd2 ese]$ euid=${EUID:-`id -u`}
[chaapala@tenfwd2 ese]$ echo $euid
10260