Come interrompere l'esecuzione di uno script se non è root (ed echo "Non in esecuzione come root! Uscita ...")

Ecco la mia fonte:

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

Cancella cache e roba e fa eco che deve essere eseguito come root nel terminale. Fondamentalmente voglio solo che lo script cessi di funzionare se rileva che non viene eseguito come root.

Esempio:

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

Se eseguito con privilegi elevati, funziona normalmente. Qualche idea? Grazie!

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

L'utente root ha UID 0 (indipendentemente dal nome dell'account "root"). Se l'UID effettivo restituito id -unon è zero, l'utente non sta eseguendo lo script con i privilegi di root. Utilizzare id -ruper verificare l'ID reale (l'UID dell'utente che richiama lo script).

Non utilizzare $EUIDnello script poiché potrebbe essere modificato da un utente non privilegiato:

$ bash -c 'echo $EUID'
1000

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

Se un utente facesse questo, ovviamente non porterebbe a un'escalation privilegiata, ma potrebbe portare a comandi nello script che non sono in grado di fare ciò che dovrebbero fare e file creati con il proprietario sbagliato ecc.

Penso che ciò che desideri sia piuttosto verificare di disporre dei privilegi di superutente, ovvero che il tuo ID utente effettivo sia 0.

zshe bashrenderlo disponibile nella $EUIDvariabile, in modo da poter fare:

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

Con qualsiasi shell simile a POSIX, è possibile utilizzare il idcomando standard:

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

Si noti che tutti id -uno whoamio $USERNAMEvariabile zshsi arriva la prima nome utente per l'UID. Sui sistemi che hanno altri utenti con ID 0, ciò potrebbe non essere possibile rootanche se il processo è un discendente di uno che è stato autenticato come root.

$USERdi solito ti darebbe l'utente che è autenticato, ma fare affidamento su di esso è piuttosto fragile. Non è impostato dalla shell, ma di solito è impostato dal comando di autenticazione (come login, su(su sistemi GNU / Linux, non necessariamente altri) sudo, sshd(almeno quello di openssh) ...). Non sempre però (la modifica di uid non imposta automagicamente quella variabile, deve essere fatta esplicitamente dall'applicazione che modifica l'uid) e potrebbe anche essere stata modificata da qualche altro processo negli antenati della shell. $LOGNAME, con lo stesso avvertimento è più affidabile di quanto specificato da POSIX (originariamente da FIPS 151-2)

È possibile utilizzare $USERo whoamiper verificare l'utente corrente.

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

Quello che vuoi effettivamente è determinare se hai accesso a queste operazioni. È considerata una cattiva pratica verificare se si è root al posto di quello.

Se il sistema è stato configurato per consentire a un utente non root di modificare lo scambio e di eliminare la cache della pagina, perché l'utente non dovrebbe eseguire lo script?

Invece, puoi semplicemente provare l'operazione e uscire con un messaggio utile se fallisce:

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