Як скрипт може перевірити, чи він запускається як 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

Таким чином, завдяки ярлику АБО ви уникаєте виклику системної команди, надаючи пріоритет на запит змінної в пам'яті.

Повторне використання коду:

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

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

Один з простих способів зробити сценарій тільки виконуваним під корінь - це запустити сценарій з рядка:
#!/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 для запуску цього сценарію!"
      ехо "використовувати" судо !! "
      вихід 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

Таким чином ви отримаєте гарне діалогове вікно із запитом пароля користувача користувача. Як і в командному рядку sudo.

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

Цей код працює як в Bash, так і в запасі Bourne sh (тестується в FreeBSD), що не визначає EUID. Якщо існує EUID, його значення повертається, інакше виконується команда 'id'. Це tad коротше, ніж приклад "або" вище, оскільки він використовує вбудовану оболонку ": -".

Корінь в ш:

# echo $EUID

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

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