Como interromper a execução de um script se ele não for root (e ecoar "Não está executando como root! Saindo ...")

Aqui está a minha 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 "";

Ele limpa caches e outras coisas, e faz eco que ele precisa ser executado como root no terminal. Basicamente, só quero que o script pare de ser executado se detectar que não está sendo executado como root.

Exemplo:

"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 executado com privilégios elevados, é executado normalmente. Alguma ideia? Obrigado!

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

O usuário root possui o UID 0 (independentemente do nome da conta "root"). Se o UID efetivo retornado por id -unão for zero, o usuário não estará executando o script com privilégios de root. Use id -rupara testar o ID real (o UID do usuário que está chamando o script).

Não use $EUIDno script, pois isso pode ser modificado por um usuário não privilegiado:

$ bash -c 'echo $EUID'
1000

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

Se um usuário fizer isso, obviamente isso não levará à escalada de privilégios, mas poderá fazer com que os comandos no script não consigam fazer o que deveriam fazer e os arquivos sejam criados com o proprietário errado etc.

Acho que o que você deseja é verificar se você possui privilégios de superusuário, ou seja, se seu ID de usuário efetivo é 0.

zshe bashdisponibilize isso na $EUIDvariável, para que você possa:

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

Com qualquer shells do tipo POSIX, você pode usar o idcomando padrão:

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

Note-se que todos id -unou whoamiou a $USERNAMEvariável zshirá obter o primeiro nome de usuário para o uid. Em sistemas que possuem outros usuários com o ID 0, isso pode não ser o rootmesmo se o processo for descendente de um que foi autenticado como root.

$USERseria normalmente dar-lhe o usuário que autenticou, mas contando com ele é bastante frágil. Não é definido pelo shell, mas geralmente é definido pelo comando de autenticação (como login, su(nos sistemas GNU / Linux, não necessariamente em outros)) sudo, sshd(pelo menos o do openssh) ...). Nem sempre é assim (alterar o uid não define automaticamente essa variável, isso deve ser feito explicitamente pelo aplicativo que altera o uid) e também pode ter sido modificado por algum outro processo na ancestralidade do shell. $LOGNAME, com a mesma ressalva é mais confiável, conforme especificado pelo POSIX (originalmente do FIPS 151-2)

Você pode usar $USERou whoamipara verificar o usuário atual.

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

O que você realmente deseja é determinar se você tem acesso para executar essas operações. É uma prática recomendada verificar se você está ou não raiz em vez disso.

Se o sistema foi configurado para permitir que um usuário não root modifique o swap e descarte o cache da página, por que esse usuário não deve executar seu script?

Em vez disso, você pode simplesmente tentar a operação e sair com uma mensagem útil se ela falhar:

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