about summary refs log tree commit diff
path: root/Completion/User/_ssh
blob: 2c71b49a040b19986af4a507db8587277183d8cc (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
#compdef ssh slogin scp ssh-add ssh-agent ssh-keygen

_ssh () {
  local context state lstate line ret=1 expl args tmp
  typeset -A opt_args

  local accounts_users_hosts

  local varname="$words[1]_accounts"

  if (( ${(P)+varname} )); then
    accounts_users_hosts=( ${(P)varname} )
  else
    accounts_users_hosts=( $my_accounts )
  fi

  args=()

  # ssh-opt is a pseudo-command used to complete ssh options for `scp -o'.

  case "$words[1]" in
  ssh|slogin)
    args=(
      ':remote host name:->userhost'
      ':command: _command_names -e'
      '*::args:->command'
    )
    ;&
  ssh-opt)
    _arguments -s \
      '-a[disable forwarding of authentication agent connection]' \
      '-c[select encryption cipher]:encryption cipher:(idea des 3des blowfish arcfour tss none)' \
      '-e[set escape character]:escape character (or `none'"'"'):' \
      '(-n)-f[go to background]' \
      '-i[select identity file]:SSH identity file:_files' \
      '-k[disable forwarding of kerberos tickets]' \
      '-l[specify login name]:login name:_ssh_users' \
      '-n[redirect stdin from /dev/null]' \
      '*-o[specify extra options]:option string:->option' \
      '-p[specify port on remote host]:port number on remote host:' \
      '-q[quiet operation]' \
      '-P[use non priviledged port]' \
      '-t[force pseudo-tty allocation]' \
      '-v[verbose mode]' \
      '-V[show version number]' \
      '-x[disable X11 forwarding]' \
      '-C[compress all data]' \
      '-L[specify local port forwarding]:local port forwarding:->forward' \
      '-R[specify remote port forwarding]:remote port forwarding:->forward' \
      "$args[@]" && ret=0

    while [[ -n "$state" ]]; do
      lstate="$state"
      state=''

      case "$lstate" in
      option)
        if compset -P '*[= ]'; then
          case "$IPREFIX" in
          *(#i)(batchmode|compression|fallbacktorsh|forward(agent|x11)|keepalive|passwordauthentication|rhosts(|rsa)authentication|rsaauthentication|usersh|kerberos(authetication|tgtparsing)|usepriviledgedport)*)
	    _tags "$context" values && compadd yes no && ret=0
            ;;
          *(#i)cipher*)
	    if _tags "$context" values; then
              _description expl 'encryption cipher'
              compadd "$expl[@]" idea des 3des blowfish arcfour tss none && ret=0
	    fi
            ;;
          *(#i)globalknownhostsfile*)
	    if _tags "$context" files; then
              _description expl 'global file with known hosts'
              _files "$expl[@]" && ret=0
	    fi
            ;;
          *(#i)hostname*)
	    if _tags "$context" hosts; then
              _description expl 'real host name to log into'
              _ssh_hosts "$expl[@]" && ret=0
	    fi
            ;;
          *(#i)identityfile*)
	    if _tags "$context" files; then
              _description expl 'SSH identity file'
              _files "$expl[@]" && ret=0
	    fi
            ;;
          *(#i)(local|remote)forward*)
            state=forward
            ;;
          *(#i)proxycommand*)
            compset -q
            shift 1 words
            (( CURRENT-- ))
            _normal && ret=0
            ;;
          *(#i)stricthostkeychecking*)
            _tags "$context" values && compadd yes no ask
            ;;
          *(#i)userknownhostsfile*)
	    if _tags "$context" files; then
              _description expl 'user file with known hosts'
              _files "$expl[@]" && ret=0
	    fi
            ;;
          *(#i)user*)
	    if _tags "$context" files; then
              _description expl 'user to log in as'
              _ssh_users "$expl[@]" && ret=0
	    fi
            ;;
          *(#i)xauthlocation*)
	    if _tags "$context" files; then
              _description expl 'xauth program'
              _files "$expl[@]" -g '*(*)' && ret=0
	    fi
            ;;
          esac
        else
          if _tags "$context" values; then
            _description expl 'configure file option'
            compadd "$expl[@]" -M 'm:{a-z}={A-Z}' -S '=' - \
                BatchMode ClearAllForwardings Cipher Compression \
                CompressionLevel Host ConnectionAttempts EscapeChar \
                FallBackToRsh ForwardAgent ForwardX11 \
                GlobalKnownHostsFile HostName IdentityFile KeepAlive \
                KerberosAuthentication KerberosTgtPassing LocalForward \
                NumberOfPasswordPrompts PasswordAuthentication Port \
                ProxyCommand RemoteForward RhostsAuthentication \
                RhostsRSAAuthentication RSAAuthentication \
                StrictHostKeyChecking TISAuthentication \
                UsePriviledgedPort User UserKnownHostsFile UseRsh \
                XAuthLocation \
              && ret=0
	  fi
        fi
        ;;
      forward)
        if compset -P 1 '*:'; then
          if compset -P '*:'; then
            _message 'port number'
          else
	    _tags "$context" hosts && _ssh_hosts -S: -q
          fi
        else
          _message 'listen-port number'
        fi
        return
        ;;
      command)
        shift 1 words
        (( CURRENT-- ))
        _normal
        return
        ;;
      userhost)
        if compset -P '*@'; then
	  if _tags "$context" hosts; then
            _description expl 'remote host name'
            _ssh_hosts "$expl[@]" && ret=0
	  fi
        else
          if (( $+opt_args[-l] )); then
	    tmp=()
	  else
	    tmp=( 'users:login name:_ssh_users -qS@' )
	  fi
	  _alternative "$context" \
	      'hosts:remote host name:_ssh_hosts' \
	      "$tmp[@]" && ret=0
        fi
        ;;
      esac

      break;
    done

    return ret
    ;;
  scp)
    _arguments -s \
      '-c[select encryption cipher]:encryption cipher:(idea des 3des blowfish arcfour tss none)' \
      '-P[specify port on remote host]:port number on remote host:' \
      '-i[select identity file]:SSH identity file:_files' \
      '-S[specify path to ssh]:path to ssh:_files -g \*\(\*\)' \
      '-o[give SSH options]:options:->options' \
      '-p[preserve modification times]' \
      '-r[recursively copy directories]' \
      '-v[verbose mode]' \
      '-B[batch mode]' \
      '-C[compress data]' \
      '*:file:->file' && ret=0

    if [[ "$state" = options ]]; then
      compset -q
      words=(ssh-opt "$words[@]" )
      (( CURRENT++ ))
      _ssh
      return
    elif [[ -n "$state" ]]; then
      if compset -P '*:'; then
        _tags "$context" files && _files && ret=0
      elif compset -P '*@'; then
        _tags "$context" hosts && _ssh_hosts -S: && ret=0
      else
        _alternative "$context" \
	    'files:: _files' \
	    'hosts:: _ssh_hosts -S:' \
	    'users:: _ssh_users -S@' && ret=0
      fi
    fi
    return ret
    ;;
  ssh-add)
    _arguments -s \
      '-p[read passphrase from stdin]' \
      '-l[list all identities]' \
      '-d[remove identity]' \
      '-D[delete all identities]' \
      '*:SSH identity file:_files'
    return
    ;;
  ssh-agent)
    _arguments -s \
      '(:)-k[kill agent automatically]' \
      '(:)-s[force sh-style shell]' \
      '(:)-c[force csh-style shell]' \
      '*::command: _normal'
    return
    ;;
  ssh-keygen)
    _arguments \
      '(-p -c -u -P)-b[specify number of bits in key]:bits in key:' \
      '(-p -c)-f[key file]:key file:_files' \
      '(-c -u)-N[specify new passphrase]:new passphrase:' \
      '(-b -u -f)-P[specify old passphrase]:old passphrase:' \
      '(-p -u)-C[specify new comment]:new comment:' \
      '(-b -f -C -u -c)-p[change passphrase of private key file]' \
      '(-b -f -N -p -u)-c[change comment in private and public key files]' \
      '(-b -N -C -p -c)-u[change key cipher to default]'
    return
    ;;
  esac
}

_ssh_users () {
  _tags any users && _combination accounts_users_hosts users "$@"
}

_ssh_hosts () {
  _tags any hosts || return 1

  if [[ "$IPREFIX" == *@ ]]; then
    _combination accounts_users_hosts "users=${IPREFIX/@}" hosts "$@"
  else
    _combination accounts_users_hosts \
      ${opt_args[-l]:+"users=${opt_args[-l]:q}"} hosts "$@"
  fi
}

_ssh "$@"