ssh-agent.plugin.zsh 2.3 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980
  1. typeset _agent_forwarding _ssh_env_cache
  2. function _start_agent() {
  3. local lifetime
  4. zstyle -s :omz:plugins:ssh-agent lifetime lifetime
  5. # start ssh-agent and setup environment
  6. echo starting ssh-agent...
  7. ssh-agent -s ${lifetime:+-t} ${lifetime} | sed 's/^echo/#echo/' >! $_ssh_env_cache
  8. chmod 600 $_ssh_env_cache
  9. . $_ssh_env_cache > /dev/null
  10. }
  11. function _add_identities() {
  12. local id line sig
  13. local -a identities loaded_sigs loaded_ids not_loaded
  14. zstyle -a :omz:plugins:ssh-agent identities identities
  15. # check for .ssh folder presence
  16. if [[ ! -d $HOME/.ssh ]]; then
  17. return
  18. fi
  19. # add default keys if no identities were set up via zstyle
  20. # this is to mimic the call to ssh-add with no identities
  21. if [[ ${#identities} -eq 0 ]]; then
  22. # key list found on `ssh-add` man page's DESCRIPTION section
  23. for id in id_rsa id_dsa id_ecdsa id_ed25519 identity; do
  24. # check if file exists
  25. [[ -f "$HOME/.ssh/$id" ]] && identities+=$id
  26. done
  27. fi
  28. # get list of loaded identities' signatures and filenames
  29. for line in ${(f)"$(ssh-add -l)"}; do
  30. loaded_sigs+=${${(z)line}[2]}
  31. loaded_ids+=${${(z)line}[3]}
  32. done
  33. # add identities if not already loaded
  34. for id in $identities; do
  35. # check for filename match, otherwise try for signature match
  36. if [[ ${loaded_ids[(I)$HOME/.ssh/$id]} -le 0 ]]; then
  37. sig="$(ssh-keygen -lf "$HOME/.ssh/$id" | awk '{print $2}')"
  38. [[ ${loaded_sigs[(I)$sig]} -le 0 ]] && not_loaded+="$HOME/.ssh/$id"
  39. fi
  40. done
  41. [[ -n "$not_loaded" ]] && ssh-add ${^not_loaded}
  42. }
  43. # Get the filename to store/lookup the environment from
  44. _ssh_env_cache="$HOME/.ssh/environment-$SHORT_HOST"
  45. # test if agent-forwarding is enabled
  46. zstyle -b :omz:plugins:ssh-agent agent-forwarding _agent_forwarding
  47. if [[ $_agent_forwarding == "yes" && -n "$SSH_AUTH_SOCK" ]]; then
  48. # Add a nifty symlink for screen/tmux if agent forwarding
  49. [[ -L $SSH_AUTH_SOCK ]] || ln -sf "$SSH_AUTH_SOCK" /tmp/ssh-agent-$USER-screen
  50. elif [[ -f "$_ssh_env_cache" ]]; then
  51. # Source SSH settings, if applicable
  52. . $_ssh_env_cache > /dev/null
  53. if [[ $USER == "root" ]]; then
  54. FILTER="ax"
  55. else
  56. FILTER="x"
  57. fi
  58. ps $FILTER | grep ssh-agent | grep -q $SSH_AGENT_PID || {
  59. _start_agent
  60. }
  61. else
  62. _start_agent
  63. fi
  64. _add_identities
  65. # tidy up after ourselves
  66. unset _agent_forwarding _ssh_env_cache
  67. unfunction _start_agent _add_identities