shutdown

   1 #!/bin/sh
   2 #
   3 # shutdown -- wrapper script to prevent erroneous shutdowns via SSH
   4 #
   5 # Copyright © martin f. krafft <madduck@madduck.net>
   6 # Released under the terms of the Artistic Licence 2.0
   7 #
   8 # $Id: shutdown 299 2006-10-16 14:40:47Z madduck $
   9 #
  10 set -eu
  11
  12 ME=molly-guard
  13
  14 CMD="${0##*/}"
  15 EXEC="/sbin/$CMD"
  16
  17 case "$CMD" in
  18   halt|reboot|shutdown|poweroff)
  19     if [ ! -f $EXEC ]; then
  20       echo "E: $ME: not a regular file: $EXEC" >&2
  21       exit 4
  22     fi
  23     if [ ! -x $EXEC ]; then
  24       echo "E: $ME: not an executable: $EXEC" >&2
  25       exit 3
  26     fi
  27     ;;
  28   *)
  29     echo "E: $ME: unsupported command: $CMD" >&2
  30     exit 1
  31     ;;
  32 esac
  33
  34 usage()
  35 {
  36   cat <<-_eousage
  37         Usage: $ME [options]
  38                (shielding $EXEC)
  39
  40         Instead of invoking $EXEC directly, $ME will prompt the user for the
  41         machine's hostname to guard against accidental shutdowns/reboots, if the
  42         current shell is a child of an SSH connection (or --pretend-ssh) has been
  43         given on the command line, if the shell is connected to an interactive
  44         terminal, and the actual command to execute is does not involve --help or is
  45         \`shutdown -c'.
  46
  47         Only if the user enters the machine's hostname correctly will $ME take
  48         action. Specifying --molly-guard-do-nothing as argument to the command will
  49         make $ME echo the command it would execute rather than actually executing
  50         it.
  51
  52         The actual command's help output follows:
  53
  54         _eousage
  55 }
  56
  57 ARGS=
  58 DO_NOTHING=0
  59 PRETEND_SSH=0
  60 for arg in "$@"; do
  61   case "$arg" in
  62     (*-molly-guard-do-nothing) DO_NOTHING=1;;
  63     (*-help)
  64       usage 2>&1
  65       eval $EXEC --help 2>&1
  66       exit 0
  67       ;;
  68     (*-pretend-ssh) PRETEND_SSH=1;;
  69     *) ARGS="${ARGS:+$ARGS }$arg";;
  70   esac
  71 done
  72
  73 do_real_cmd()
  74 {
  75   if [ $DO_NOTHING -eq 1 ]; then
  76     echo "$ME: would run: $EXEC $ARGS"
  77     exit 0
  78   else
  79     eval exec $EXEC "$ARGS"
  80   fi
  81 }
  82
  83 if [ $DO_NOTHING -eq 1 ]; then
  84   echo "I: demo mode; $ME will not do anything due to --molly-guard-do-nothing."
  85 fi
  86
  87 # require an interactive terminal connected to stdin
  88 test -t 0                    || do_real_cmd
  89
  90 # only run if we are being called over SSH, that is if the current terminal
  91 # was created by sshd.
  92 PTS=$(readlink /proc/$$/fd/0)
  93 if ! pgrep -f "^sshd.+${PTS#/dev/}[[:space:]]*$" >/dev/null \
  94   && [ -z "${SSH_CONNECTION:-}" ]; then
  95     if [ $PRETEND_SSH -eq 1 ]; then
  96       echo "I: this is not an SSH session, but --pretend-ssh was given..."
  97     else
  98       do_real_cmd
  99     fi
 100 else
 101   echo "W: $ME: SSH session detected!"
 102 fi
 103
 104 # pass through certain commands
 105 case "$CMD $ARGS" in
 106   (*shutdown\ *-c*)
 107     echo "I: executing $CMD $ARGS regardless of SSH session."
 108     do_real_cmd
 109     ;;
 110 esac
 111
 112 HOSTNAME="$(hostname --short)"
 113
 114 sigh()
 115 {
 116   echo "Good thing I asked; I won't $CMD $HOSTNAME ..."
 117   exit 2
 118 }
 119
 120 trap 'echo;sigh' 1 2 3 9 10 12 15
 121
 122 echo -n "Please type in hostname of the machine to $CMD: "
 123 read HOSTNAME_USER || :
 124
 125 [ "$HOSTNAME_USER" = "$HOSTNAME" ] || sigh
 126
 127 trap - 1 2 3 9 10 12 15
 128
 129 do_real_cmd