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 run a number of checks
  41         to guard against accidental shutdowns/reboots.
  42
  43         Some of the checks available are:
  44          - Prompt the user for the machine's if the current shell is a child
  45            of an SSH connection (or --pretend-ssh) has been given on the
  46            command line, if the shell is connected to an interactive
  47            terminal, and the actual command to execute is does not involve
  48            --help or is \`shutdown -c'.
  49
  50         Only if the user satisfies all the checks will $ME take action.
  51         Specifying --molly-guard-do-nothing as argument to the command will
  52         make $ME echo the command it would execute rather than actually executing
  53         it.
  54
  55         The actual command's help output follows:
  56
  57         _eousage
  58 }
  59
  60 ARGS=
  61 DO_NOTHING=0
  62 CHECK_ARGS=
  63 for arg in "$@"; do
  64   case "$arg" in
  65     (*-molly-guard-do-nothing) DO_NOTHING=1;;
  66     (*-help)
  67       usage 2>&1
  68       eval $EXEC --help 2>&1
  69       exit 0
  70       ;;
  71     (*-pretend-ssh) CHECK_ARGS="${CHECK_ARGS:+$CHECK_ARGS }--arg --pretend-ssh";;
  72     *) ARGS="${ARGS:+$ARGS }$arg";;
  73   esac
  74 done
  75
  76 do_real_cmd()
  77 {
  78   if [ $DO_NOTHING -eq 1 ]; then
  79     echo "$ME: would run: $EXEC $ARGS"
  80     exit 0
  81   else
  82     eval exec $EXEC "$ARGS"
  83   fi
  84 }
  85
  86 if [ $DO_NOTHING -eq 1 ]; then
  87   echo "I: demo mode; $ME will not do anything due to --molly-guard-do-nothing."
  88 fi
  89
  90 # pass through certain commands
  91 case "$CMD $ARGS" in
  92   (*shutdown\ *-c*)
  93     echo "I: executing $CMD $ARGS regardless of check results."
  94     do_real_cmd
  95     ;;
  96 esac
  97
  98 run-parts --exit-on-error $CHECK_ARGS /usr/share/molly-guard/checks.d
  99
 100 # run-parts won't return to us if there are failures, but I'm paranoid.
 101 if [ $? == 0 ]; then
 102   do_real_cmd
 103   exit
 104 fi