X-Git-Url: http://git.iain.cx/?a=blobdiff_plain;f=.profile.d%2Fssh.bashrc;h=5f3c1a5e27be10cd1885024326b163815197344c;hb=7e973fb6a4ec8da564b4621658e2c4d583560d33;hp=b82a9874cd07e96b4ae10e9855b68654d85f0182;hpb=8c75606a52da5a4df6018134e0806e7041ca69e1;p=profile.git

diff --git a/.profile.d/ssh.bashrc b/.profile.d/ssh.bashrc
index b82a987..5f3c1a5 100644
--- a/.profile.d/ssh.bashrc
+++ b/.profile.d/ssh.bashrc
@@ -1,4 +1,11 @@
 # $Id$
+
+SOCKET=~/.ssh/agent
+
+function get_agent_pid() {
+  ps waux | grep -- ^$USER\ .\*ssh-agent\ -a\ .\*\ -s | grep -v grep | awk '{ print $2 }' | head -n 1
+}
+
 if [ ! -z "$SSH_CLIENT" ]; then
   # Set display if we aren't already forwarding X11
   if [ "$DISPLAY" = "" ]; then
@@ -8,19 +15,34 @@ fi
 
 # Try to hook up with already running ssh-agent.
 if [ $UID -gt 0 ]; then
-  RUNNING_AGENT="$(ps waux | grep -- ^$USER\ .\*ssh-agent\ -s | grep -v grep | awk '{ print $2 }' | head -n 1)"
+  # Not all OSes support ``ps -wu''.
+  RUNNING_AGENT="$(get_agent_pid)"
+
+  # Has the socket gone away?
+  if [ ! -z "$RUNNING_AGENT" -a ! -S "$SOCKET" ]; then
+    if kill $RUNNING_AGENT; then
+      RUNNING_AGENT="$(get_agent_pid)"
+    else
+      echo >&2 "$SOCKET has gone away but agent is running as PID $RUNNING_AGENT."
+    fi
+  fi
+
   if [ -z "$SSH_AGENT_PID" -o ! "$SSH_AGENT_PID" = "$RUNNING_AGENT" ]; then
     export SSH_AGENT_PID="$RUNNING_AGENT"
     if [ "$SSH_AGENT_PID" = "" ]; then
       unset SSH_AGENT_PID
       unset SSH_AUTH_SOCK
       # Start ssh-agent up then.
-      eval $(/usr/bin/ssh-agent -s)
+      rm -f "$SOCKET"
+      eval $(/usr/bin/ssh-agent -a "$SOCKET" -s)
       trap "kill $SSH_AGENT_PID" 0
       ssh-add
     else
-      export SSH_AUTH_SOCK="/$(lsof -p $SSH_AGENT_PID | grep 'agent.[0-9]*$' | cut -s -d / -f 2-)"
+      export SSH_AUTH_SOCK="$SOCKET"
     fi
   fi
   unset RUNNING_AGENT
 fi
+
+unset SOCKET
+unset get_agent_pid