Как остановить запуск скрипта, если он не root (и вывести «Не запускается как root! Выход ...»)

Вот мой источник:

#!/bin/bash

echo "Running script to free general cached memory!"
echo "";
echo "Script must be run as root!";
echo "";
echo "Clearing swap!";
swapoff -a && swapon -a;
echo "";
echo "Clear inodes and page file!";
echo 1 > /proc/sys/vm/drop_caches;
echo "";

Он очищает кеши и прочее и повторяет, что его нужно запускать от имени пользователя root в терминале. Я просто хочу, чтобы скрипт прекратил работу, если обнаружит, что он не выполняется от имени пользователя root.

Пример:

"Running script to free general cached memory!"
"Warning: script must be run as root or with elevated privileges!"
"Error: script not running as root or with sudo! Exiting..."

Если запустить с повышенными привилегиями, он будет работать как обычно. Есть идеи? Благодарность!

#!/bin/sh

if [ "$(id -u)" -ne 0 ]; then
        echo 'This script must be run by root' >&2
        exit 1
fi

cat <<HEADER
Host:          $(hostname)
Time at start: $(date)

Running cache maintenance...
HEADER

swapoff -a && swapon -a
echo 1 >/proc/sys/vm/drop_caches

cat <<FOOTER
Cache maintenance done.
Time at end:   $(date)
FOOTER

Пользователь root имеет UID 0 (независимо от имени учетной записи «root»). Если возвращаемый эффективный UID id -uне равен нулю, пользователь не выполняет сценарий с привилегиями root. Используется id -ruдля проверки реального идентификатора (UID пользователя, вызывающего скрипт).

Не используйте $EUIDв скрипте, поскольку это может быть изменено непривилегированным пользователем:

$ bash -c 'echo $EUID'
1000

$ EUID=0 bash -c 'echo $EUID'
0

Если пользователь сделал это, это, очевидно, не приведет к повышению привилегий, но может привести к тому, что команды в сценарии не смогут выполнить то, что они должны делать, и файлы будут созданы с неправильным владельцем и т. Д.

Я думаю, что вы хотите скорее проверить, что у вас есть привилегии суперпользователя, то есть, что ваш эффективный идентификатор пользователя равен 0.

zshи bashсделайте это доступным в $EUIDпеременной, так что вы можете сделать:

if ((EUID != 0)); then
  echo >&2 "Error: script not running as root or with sudo! Exiting..."
  exit 1
fi

С любыми POSIX-подобными оболочками вы можете использовать idстандартную команду:

if [ "$(id -u)" -ne 0 ]; then
  echo >&2 "Error: script not running as root or with sudo! Exiting..."
  exit 1
fi

Обратите внимание, что все id -unили whoamiили $USERNAMEпеременная in zshполучат первое имя пользователя для uid. В системах, в которых есть другие пользователи с идентификатором 0, это может не произойти, rootдаже если процесс является потомком того, который был аутентифицирован как root.

$USERбудет , как правило , дают вам пользователю , что проверка подлинности, но полагаться на него довольно хрупкими. Это не задаются оболочкой, но, как правило , устанавливается командой проверки подлинности (например login, su(на системах GNU / Linux, не обязательно другие), sudo, sshd(одного из OpenSSH , по крайней мере) ...). Хотя это не всегда (изменение uid не устанавливает эту переменную автоматически, это должно быть сделано явно приложением, меняющим uid), и это также могло быть изменено каким-то другим процессом в происхождении оболочки. $LOGNAMEс тем же предупреждением более надежно, как указано в POSIX (первоначально из FIPS 151-2)

Вы можете использовать $USERили whoamiдля проверки текущего пользователя.

if [[ "$USER" != "root" ]]; then
    echo "Error: script not running as root or with sudo! Exiting..."
    exit 1
fi

if [[ $(whoami) != "root" ]]; then
    echo "Warning: script must be run as root or with elevated privileges!"
    exit 1
fi

if [[ $(id -u) != "0" ]]; then
    echo "Error: script not running as root or with sudo! Exiting..."
    exit 1
fi

На самом деле вы хотите определить, есть ли у вас доступ для выполнения этих операций. Проверять, действительно ли вы root вместо этого, считается плохой практикой.

Если система настроена так, чтобы пользователь без полномочий root мог изменять кеш страниц подкачки и удаления, почему этот пользователь не должен запускать ваш скрипт?

Вместо этого вы можете просто попробовать выполнить операцию и выйти с полезным сообщением, если оно завершится неудачно:

if ! ( swapoff -a && swapon -a )
then
  echo "Failed to clear swap (rerun as root?)" >&2
  exit 1
fi

if ! echo 1 > /proc/sys/vm/drop_caches
then 
  echo "Failed to free page cache (rerun as root?)" >&2
  exit 1
fi