So stoppen Sie die Ausführung eines Skripts, wenn es nicht als Root ausgeführt wird

Hier ist meine Quelle:

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

Es löscht Caches und andere Daten und gibt an, dass es als Root im Terminal ausgeführt werden muss. Ich möchte im Grunde nur, dass das Skript nicht mehr ausgeführt wird, wenn es feststellt, dass es nicht als root ausgeführt wird.

Beispiel:

"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..."

Wenn es mit erhöhten Rechten ausgeführt wird, wird es wie gewohnt ausgeführt. Irgendwelche Ideen? Vielen Dank!

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

Der Root-Benutzer hat die UID 0 (unabhängig vom Namen des "Root" -Kontos). Wenn die von zurückgegebene effektive UID id -unicht Null ist, führt der Benutzer das Skript nicht mit Root-Rechten aus. Verwenden Sie id -ruzum Test gegen die reale ID (die UID des Benutzers Aufruf des Skripts).

Nicht $EUIDim Skript verwenden, da dies von einem nicht privilegierten Benutzer geändert werden kann:

$ bash -c 'echo $EUID'
1000

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

Wenn ein Benutzer dies tun würde, würde dies offensichtlich nicht zu einer Eskalation der Rechte führen, sondern könnte dazu führen, dass Befehle im Skript nicht in der Lage sind, das zu tun, was sie tun sollen, und dass Dateien mit dem falschen Eigentümer erstellt werden usw.

Ich denke, was Sie wollen, ist eher zu überprüfen, ob Sie Super-User-Berechtigungen haben, das heißt, dass Ihre effektive Benutzer-ID 0 ist.

zshund bashstellen $EUIDSie dies in der Variablen zur Verfügung, damit Sie Folgendes tun können:

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

Mit allen POSIX-ähnlichen Shells können Sie den idStandardbefehl verwenden:

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

Beachten Sie, dass alle id -unoder whoamioder die $USERNAMEVariable in zshIhnen den ersten Benutzernamen für die UID erhalten. Auf Systemen mit anderen Benutzern mit der ID 0 ist dies möglicherweise nicht der Fall, rootauch wenn der Prozess ein Nachkomme eines Prozesses ist, der als authentifiziert wurde root.

$USERwürde Ihnen normalerweise den Benutzer geben, der sich authentifiziert, aber sich darauf zu verlassen, ist ziemlich spröde. Es ist nicht von dem Shell gesetzt, sondern wird in der Regel durch den authentifizierten Befehl (wie festgelegt login, su(auf GNU / Linux - Systeme, die nicht unbedingt andere) sudo, sshd(der einer von OpenSSH zumindest) ...). Dies ist jedoch nicht immer der Fall (das Ändern der UID setzt diese Variable nicht automatisch, sondern muss explizit von der Anwendung vorgenommen werden, die die UID ändert) und könnte auch durch einen anderen Prozess in der Vorfahren der Shell geändert worden sein. $LOGNAME, mit dem gleichen Vorbehalt ist zuverlässiger als es von POSIX angegeben wird (ursprünglich von FIPS 151-2)

Mit $USERoder whoamikönnen Sie den aktuellen Benutzer überprüfen.

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

Was Sie tatsächlich wollen, ist festzustellen, ob Sie Zugriff auf diese Vorgänge haben. Es wird als schlechte Praxis angesehen, stattdessen zu prüfen, ob Sie Root sind oder nicht.

Wenn das System so konfiguriert wurde, dass ein Benutzer ohne Rootberechtigung den Swap- und Drop-Page-Cache ändern kann, warum sollte dieser Benutzer dann nicht das Skript ausführen können?

Stattdessen können Sie die Operation einfach ausprobieren und mit einer hilfreichen Meldung beenden, falls dies fehlschlägt:

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