* Prevent set -u from falling over unset SSH_CONNECTION variable

[code/molly-guard.git] / shutdown

diff --git a/shutdown b/shutdown
index 46aa45f7e0a2cf3d3795a6f43d05c22e1166bee6..0c52d92803e65b8e7615b918e9910e7ea55102b6 100755 (executable)
--- a/shutdown
+++ b/shutdown
@@ -90,12 +90,13 @@ test -t 0                    || do_real_cmd
 # only run if we are being called over SSH, that is if the current terminal
 # was created by sshd.
 PTS=$(readlink /proc/$$/fd/0)
-if ! pgrep -f "^sshd.+${PTS#/dev/}[[:space:]]*$" >/dev/null; then
-  if [ $PRETEND_SSH -eq 1 ]; then
-    echo "I: this is not an SSH session, but --pretend-ssh was given..."
-  else
-    do_real_cmd
-  fi
+if ! pgrep -f "^sshd.+${PTS#/dev/}[[:space:]]*$" >/dev/null \
+  && [ -z "${SSH_CONNECTION:-}" ]; then
+    if [ $PRETEND_SSH -eq 1 ]; then
+      echo "I: this is not an SSH session, but --pretend-ssh was given..."
+    else
+      do_real_cmd
+    fi
 else
   echo "W: $ME: SSH session detected!"
 fi
@@ -108,7 +109,7 @@ case "$CMD $ARGS" in
     ;;
 esac
 
-HOSTNAME="$(hostname)"
+HOSTNAME="$(hostname --short)"
 
 sigh()
 {