Как скрипт может проверить, запускается ли он от имени root?

Я пишу простой bash-скрипт, но мне нужно проверить, запущен ли он от имени пользователя root или нет. Я знаю, что, вероятно, есть очень простой способ сделать это, но я понятия не имею, как.

Просто чтобы быть ясно:
Что простой способ , чтобы написать сценарий foo.sh , так что команда ./foo.shвыводит 0, и команда sudo ./foo.shвыводит 1?

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

Пользователь root не обязательно должен называться «root». whoamiвозвращает первое имя пользователя с идентификатором пользователя 0. $USERсодержит имя вошедшего в систему пользователя, который может иметь идентификатор пользователя 0, но иметь другое имя.

Единственная надежная программа, которая проверяет, вошел ли аккаунт как root или нет:

id -u

Я использую -uдля эффективного идентификатора пользователя, а не -rдля реального идентификатора пользователя. Права доступа определяются с помощью эффективного идентификатора пользователя, а не реальные один.

тесты

/etc/passwdсодержит следующие имена пользователей с идентификатором пользователя 0в указанном порядке:

rootx
root2

Войдя в систему root2, дает следующие результаты:

• whoami: rootx
• echo $USER: root2(возвращает пустую строку, если программа была запущена в пустой среде, например env -i sh -c 'echo $USER')
• id -u: 0 Как видите, другие программы не id -uпрошли эту проверку, только пройдены.

Обновленный скрипт будет выглядеть так:

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

Как сказал @Lekensteyn, вы должны использовать эффективный идентификатор пользователя. Вам не нужно звонить id -uв Bash:

#!/bin/bash

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

Предложение @ geirha из комментариев использует арифметическую оценку:

#!/bin/bash

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

Вы можете сделать это с помощью whoamiкоманды, которая возвращает текущего пользователя:

#!/bin/bash

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

...

Выполнение вышеупомянутого напечатает, You must be root to do this.если текущий пользователь не root.

Примечание: альтернативой в некоторых случаях является просто проверка $USERпеременной:

if [ $USER != 'root' ]

Принимая во внимание эффективность, вы можете сначала протестировать EUIDпеременную окружения, а затем, если она не существует, вызвать стандартную idкоманду:

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

Таким образом, из-за сочетания клавиш OR вы не будете вызывать системную команду, отдавая приоритет запросу переменной в памяти.

Повторное использование кода:

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

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

Один простой способ сделать скрипт доступным только для пользователя root - запустить скрипт со строкой:
#!/bin/su root

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

Этот фрагмент будет:

• Проверьте на разных сессиях
• предложить использовать sudo !!
• и вернуть ошибку

if ["$ (whoami & 2> / dev / null)"! = "root"] && ["$ (id -un & 2> / dev / null)"! = "root"]
      тогда
      echo "Вы должны быть пользователем root, чтобы запустить этот скрипт!"
      echo "использовать 'sudo !!'"
      выход 1
фи

Этот ответ просто для того, чтобы сохранить идею, может быть полезным для некоторых из вас. Если вам нужен скрипт, который запускается из графического интерфейса рабочего стола и требует привилегий root, попробуйте этот способ:

#!/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

Таким образом, вы получите красивое диалоговое окно с просьбой ввести пароль пользователя root. Так же, как с командной строкой sudo.

gksudoМожет быть недоступен в вашей системе , то установите его sudo apt-get install gksu.

Этот код работает как для Bash, так и для стандартного Bourne sh (протестирован под FreeBSD), который не определяет EUID. Если EUID существует, его значение возвращается, в противном случае выполняется команда 'id'. Это немного короче, чем в приведенном выше примере «или», так как в нем используется встроенная оболочка «: -».

Корень в ш:

# echo $EUID

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

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