summaryrefslogtreecommitdiff
path: root/lib/net/ssh/verifiers/accept_new.rb
blob: aa68ff2c98ae7f30e1c91a85b7be5bf867041a3a (plain)
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
require 'net/ssh/errors'
require 'net/ssh/known_hosts'
require 'net/ssh/verifiers/always'

module Net
  module SSH
    module Verifiers
      # Does a strict host verification, looking the server up in the known
      # host files to see if a key has already been seen for this server. If this
      # server does not appear in any host file, this will silently add the
      # server. If the server does appear at least once, but the key given does
      # not match any known for the server, an exception will be raised (HostKeyMismatch).
      # Otherwise, this returns true.
      class AcceptNew < Always
        def verify(arguments)
          begin
            super
          rescue HostKeyUnknown => err
            err.remember_host!
            return true
          end
        end

        def verify_signature(&block)
          yield
        rescue HostKeyUnknown => err
          err.remember_host!
          return true
        end
      end
    end
  end
end