net-ssh 0.5.0

data/test/userauth/tc_driver.rb

@@ -0,0 +1,190 @@
+#--
+# =============================================================================
+# Copyright (c) 2004, Jamis Buck (jgb3@email.byu.edu)
+# All rights reserved.
+#
+# This source file is distributed as part of the Net::SSH Secure Shell Client
+# library for Ruby. This file (and the library as a whole) may be used only as
+# allowed by either the BSD license, or the Ruby license (or, by association
+# with the Ruby license, the GPL). See the "doc" subdirectory of the Net::SSH
+# distribution for the texts of these licenses.
+# -----------------------------------------------------------------------------
+# net-ssh website : http://net-ssh.rubyforge.org
+# project website: http://rubyforge.org/projects/net-ssh
+# =============================================================================
+#++
+
+$:.unshift "../../lib"
+
+require 'net/ssh/userauth/driver'
+require 'net/ssh/util/buffer'
+require 'test/unit'
+
+class TC_UserAuth_Driver < Test::Unit::TestCase
+
+  class Log
+    def debug?
+      false
+    end
+  end
+
+  class Buffers
+    def writer
+      Net::SSH::Util::WriterBuffer.new
+    end
+  end
+
+  class KeyManager
+    attr_reader :events
+
+    def initialize
+      @events = []
+    end
+
+    def clear!
+      @events << :clear!
+    end
+
+    def clear_host!
+      @events << :clear_host!
+    end
+
+    def finish
+      @events << :finish
+    end
+
+    def add( file )
+      @events << { :name => :add, :file => file }
+    end
+    alias :<< :add
+
+    def add_host_key( file )
+      @events << { :name => :add_host_key, :file => file }
+    end
+  end
+
+  class Session
+    attr_reader :script
+    attr_reader :messages
+
+    def initialize
+      @script = []
+      @messages = []
+    end
+
+    def send_message( message )
+      @messages << message
+    end
+
+    def wait_for_message
+      type, data = @script.shift
+      [ type, Net::SSH::Util::ReaderBuffer.new( data ) ]
+    end
+  end
+
+  class AuthMethod
+    attr_reader :next_service
+    attr_reader :username
+    attr_reader :data
+
+    def initialize( good )
+      @good = good
+    end
+
+    def authenticate( next_service, username, data={} )
+      @next_service = next_service
+      @username = username
+      @data = data
+      @good
+    end
+  end
+
+  def setup
+    @methods = { :test_fail => AuthMethod.new( false ),
+                :test_succeed => AuthMethod.new( true ) }
+    order = [ "test-fail", "test-succeed" ]
+    @driver = Net::SSH::UserAuth::Driver.new( Log.new,
+      Buffers.new, @methods, order )
+    @driver.session = @session = Session.new
+    @driver.key_manager = @manager = KeyManager.new
+  end
+
+  def test_set_key_files
+    @driver.set_key_files [ :one, :two ]
+    assert_equal [ :clear!,
+                   { :name => :add, :file => :one },
+                   { :name => :add, :file => :two } ], @manager.events
+  end
+
+  def test_set_host_key_files
+    @driver.set_host_key_files [ :one, :two ]
+    assert_equal [ :clear_host!,
+                   { :name => :add_host_key, :file => :one },
+                   { :name => :add_host_key, :file => :two } ], @manager.events
+  end
+
+  def test_order
+    assert_equal [ "test-fail", "test-succeed" ], @driver.order
+    original = [ "one", "two", "three" ]
+    @driver.set_auth_method_order( original )
+    @driver.order << "four"
+    assert_equal [ "one", "two", "three" ], original
+    assert_equal [ "one", "two", "three", "four" ], @driver.order
+    @driver.set_auth_method_order *original
+    assert_equal [ "one", "two", "three" ], @driver.order
+  end
+
+  def test_on_banner
+    msg = lang = nil
+    @driver.on_banner { |m,l| msg, lang = m, l }
+    @session.script << [ 53, "\0\0\0\15hello, world!\0\0\0\2en" ]
+    @session.script << [ 51, "\0\0\0\15howdy, earth!\0" ]
+    result = @driver.wait_for_message
+
+    assert_equal "hello, world!", msg
+    assert_equal "en", lang
+    assert_equal 51, result.message_type
+  end
+
+  def test_send_message
+    @driver.send_message "konnichi wa"
+    assert_equal [ "konnichi wa" ], @session.messages
+  end
+
+  def test_authenticate_bad_reply
+    @session.script << [ 0, "\0\0\0\0" ]
+    assert_raise( Net::SSH::Exception ) do
+      @driver.authenticate( "service", "test_user", "passwd" )
+    end
+  end
+
+  def test_authenticate_unexpected_reply
+    @session.script << [ 60, "\0\0\0\0" ]
+    assert_raise( Net::SSH::Exception ) do
+      @driver.authenticate( "service", "test_user", "passwd" )
+    end
+  end
+
+  def test_authenticate
+    @session.script << [ 6, "\0\0\0\14ssh-userauth" ]
+    assert @driver.authenticate( "service", "test_user", "passwd" )
+
+    @methods.each_value do |method|
+      assert_equal "service", method.next_service
+      assert_equal "test_user", method.username
+      assert_equal( { :password => "passwd", :key_manager => @manager }, method.data )
+    end
+
+    assert_equal :finish, @manager.events.last
+  end
+
+  def test_authenticate_not_implemented
+    @session.script << [ 6, "\0\0\0\14ssh-userauth" ]
+    @driver.set_auth_method_order "bogus"
+    assert_raise( NotImplementedError ) do
+      @driver.authenticate( "service", "test_user", "passwd" )
+    end
+    assert_equal :finish, @manager.events.last
+  end
+
+end

data/test/userauth/tc_integration.rb

@@ -0,0 +1,81 @@
+#--
+# =============================================================================
+# Copyright (c) 2004, Jamis Buck (jgb3@email.byu.edu)
+# All rights reserved.
+#
+# This source file is distributed as part of the Net::SSH Secure Shell Client
+# library for Ruby. This file (and the library as a whole) may be used only as
+# allowed by either the BSD license, or the Ruby license (or, by association
+# with the Ruby license, the GPL). See the "doc" subdirectory of the Net::SSH
+# distribution for the texts of these licenses.
+# -----------------------------------------------------------------------------
+# net-ssh website : http://net-ssh.rubyforge.org
+# project website: http://rubyforge.org/projects/net-ssh
+# =============================================================================
+#++
+
+$:.unshift "../../lib"
+
+if $run_integration_tests || __FILE__ == $0
+
+  require 'needle'
+  require 'net/ssh/transport/services'
+  require 'net/ssh/userauth/services'
+  require 'test/unit'
+
+  class TC_UserAuth_Integration < Test::Unit::TestCase
+
+    HOST = "test.host"
+    USER = "test"
+    PASSWORD = "test/unit"
+    SERVICE = "ssh-connection"
+
+    def setup
+      @registry = Needle::Registry.new(
+        :logs => { :device=>STDOUT, :default_level => :WARN }
+      )
+
+      Net::SSH::Transport.register_services( @registry )
+      Net::SSH::UserAuth.register_services( @registry )
+
+      @registry.define do |b|
+        b.crypto_backend { :ossl }
+        b.transport_host { HOST }
+      end
+
+      @userauth = @registry[:userauth][:driver]
+    end
+
+    def teardown
+      @registry[:transport][:session].close
+      @registry.logs.close
+    end
+
+    def test_keyboard_interactive
+      @userauth.set_auth_method_order "keyboard-interactive"
+      assert @userauth.authenticate( SERVICE, USER, PASSWORD )
+    end
+
+    def test_password
+      @userauth.set_auth_method_order "password"
+      assert @userauth.authenticate( SERVICE, USER, PASSWORD )
+    end
+
+    def test_password_bad
+      @userauth.set_auth_method_order "password"
+      assert !@userauth.authenticate( SERVICE, USER, PASSWORD + 'K' )
+    end
+
+    def test_publickey_bad
+      @userauth.set_auth_method_order "publickey"
+      assert !@userauth.authenticate( SERVICE, USER )
+    end
+
+    def test_hostbased_bad
+      @userauth.set_auth_method_order "hostbased"
+      assert !@userauth.authenticate( SERVICE, USER )
+    end
+
+  end
+
+end

data/test/userauth/tc_userkeys.rb

@@ -0,0 +1,265 @@
+#--
+# =============================================================================
+# Copyright (c) 2004, Jamis Buck (jgb3@email.byu.edu)
+# All rights reserved.
+#
+# This source file is distributed as part of the Net::SSH Secure Shell Client
+# library for Ruby. This file (and the library as a whole) may be used only as
+# allowed by either the BSD license, or the Ruby license (or, by association
+# with the Ruby license, the GPL). See the "doc" subdirectory of the Net::SSH
+# distribution for the texts of these licenses.
+# -----------------------------------------------------------------------------
+# net-ssh website : http://net-ssh.rubyforge.org
+# project website: http://rubyforge.org/projects/net-ssh
+# =============================================================================
+#++
+
+$:.unshift "../../lib"
+
+require 'net/ssh/userauth/userkeys'
+require 'net/ssh/util/buffer'
+require 'test/unit'
+require 'logger'
+require 'stringio'
+
+class TC_UserKeyManager < Test::Unit::TestCase
+
+  class Buffers
+    def writer
+      Net::SSH::Util::WriterBuffer.new
+    end
+  end
+
+  class Keys
+    def load_public_key( file )
+      o = Object.new
+      singleton = class << o; self; end
+      singleton.send( :define_method, :ssh_type ) { "test" }
+      o
+    end
+
+    def load_private_key( file )
+      count = 0
+      o = Object.new
+      singleton = class << o; self; end
+      singleton.send( :define_method, :ssh_do_sign ) { |data| count += 1; "<#{data}:#{count}>" }
+      o
+    end
+  end
+
+  class AgentFactory
+    attr_reader :state
+
+    def initialize
+      @state = :new
+    end
+
+    def open
+      @state = :opened
+      self
+    end
+
+    def sign( identity, data )
+      "from the agent (#{identity.inspect}, #{data.inspect})"
+    end
+
+    def identities
+      [ :one, :two, :three ]
+    end
+
+    def close
+      @state = :closed
+    end
+  end
+
+  class ExistenceTester
+    def readable?( file )
+      true
+    end
+  end
+
+  def setup
+    @userkeys = Net::SSH::UserAuth::UserKeyManager.new
+    @userkeys.agent_factory = @agent_factory = AgentFactory.new
+    @userkeys.keys = Keys.new
+    @userkeys.log = @log = Logger.new( StringIO.new )
+    @userkeys.buffers = Buffers.new
+    @userkeys.key_existence_tester = ExistenceTester.new
+  end
+
+  def test_initialize
+    assert_equal 0, @userkeys.key_files.length
+    assert_equal 0, @userkeys.host_key_files.length
+    assert @userkeys.use_agent?
+  end
+
+  def test_add
+    assert_equal 0, @userkeys.key_files.length
+    @userkeys.add "hello"
+    assert_equal 1, @userkeys.key_files.length
+    assert_equal "hello", @userkeys.key_files.first
+    @userkeys.add "world"
+    assert_equal 2, @userkeys.key_files.length
+    @userkeys.add "hello"
+    assert_equal 2, @userkeys.key_files.length
+  end
+
+  def test_add_host_key
+    assert_equal 0, @userkeys.host_key_files.length
+    @userkeys.add_host_key "hello"
+    assert_equal 1, @userkeys.host_key_files.length
+    assert_equal "hello", @userkeys.host_key_files.first
+    @userkeys.add_host_key "world"
+    assert_equal 2, @userkeys.host_key_files.length
+    @userkeys.add_host_key "hello"
+    assert_equal 2, @userkeys.host_key_files.length
+  end
+
+  def test_clear!
+    @userkeys.add "hello"
+    @userkeys.add "howdy"
+    assert_equal 2, @userkeys.key_files.length
+    @userkeys.clear!
+    assert_equal 0, @userkeys.key_files.length
+  end
+
+  def test_clear_host!
+    @userkeys.add_host_key "hello"
+    @userkeys.add_host_key "howdy"
+    assert_equal 2, @userkeys.host_key_files.length
+    @userkeys.clear_host!
+    assert_equal 0, @userkeys.host_key_files.length
+  end
+
+  def test_clear_and_clear_host!
+    @userkeys.add "hello"
+    @userkeys.add "howdy"
+    @userkeys.add_host_key "hello"
+    @userkeys.add_host_key "howdy"
+    assert_equal 2, @userkeys.key_files.length
+    assert_equal 2, @userkeys.host_key_files.length
+    @userkeys.clear!
+    assert_equal 0, @userkeys.key_files.length
+    assert_equal 2, @userkeys.host_key_files.length
+    @userkeys.add "hello"
+    @userkeys.add "howdy"
+    assert_equal 2, @userkeys.key_files.length
+    assert_equal 2, @userkeys.host_key_files.length
+    @userkeys.clear_host!
+    assert_equal 2, @userkeys.key_files.length
+    assert_equal 0, @userkeys.host_key_files.length
+  end
+
+  def test_finish_use_agent_unopened
+    @userkeys.use_agent = true
+    assert_equal :new, @agent_factory.state
+    @userkeys.finish
+    assert_equal :new, @agent_factory.state
+  end
+
+  def test_finish_unopened_no_use_agent
+    @userkeys.use_agent = false
+    assert_equal :new, @agent_factory.state
+    @userkeys.finish
+    assert_equal :new, @agent_factory.state
+  end
+
+  def test_finish_use_agent_opened
+    @userkeys.use_agent = true
+    @userkeys.identities
+    assert_equal :opened, @agent_factory.state
+    @userkeys.finish
+    assert_equal :closed, @agent_factory.state
+  end
+
+  def test_finish_no_use_agent_opened
+    @userkeys.use_agent = false
+    @userkeys.identities
+    assert_equal :new, @agent_factory.state
+    @userkeys.finish
+    assert_equal :new, @agent_factory.state
+  end
+
+  def test_identities_no_use_agent_no_files
+    @userkeys.use_agent = false
+    ids = @userkeys.identities
+    assert_equal 0, ids.length
+  end
+
+  def test_identities_use_agent_no_files
+    @userkeys.use_agent = true
+    ids = @userkeys.identities
+    assert_equal 3, ids.length
+  end
+
+  def test_identities_no_use_agent_files
+    @userkeys.use_agent = false
+    @userkeys.add "one"
+    @userkeys.add "two"
+    @userkeys.add "three"
+    ids = @userkeys.identities
+    assert_equal 3, ids.length
+  end
+
+  def test_identities_use_agent_files
+    @userkeys.use_agent = true
+    @userkeys.add "one"
+    @userkeys.add "two"
+    @userkeys.add "three"
+    ids = @userkeys.identities
+    assert_equal 6, ids.length
+  end
+
+  def test_host_identities_no_files
+    ids = @userkeys.host_identities
+    assert_equal 0, ids.length
+  end
+
+  def test_host_identities_files
+    @userkeys.add_host_key "one"
+    @userkeys.add_host_key "two"
+    @userkeys.add_host_key "three"
+    ids = @userkeys.host_identities
+    assert_equal 3, ids.length
+  end
+
+  def test_sign_from_file
+    @userkeys.use_agent = false
+    @userkeys.add "one"
+    ids = @userkeys.identities
+    assert_equal 1, ids.length
+    data = @userkeys.sign( ids.first, "hello" )
+    assert_equal "\0\0\0\4test\0\0\0\11<hello:1>", data
+  end
+
+  def test_sign_from_agent
+    @userkeys.use_agent = true
+    ids = @userkeys.identities
+    assert_equal 3, ids.length
+    data = @userkeys.sign( ids.first, "hello" )
+    assert_equal %q{from the agent (:one, "hello")}, data
+  end
+
+  def test_sign_from_key
+    @userkeys.use_agent = false
+    @userkeys.add "one"
+    ids = @userkeys.identities
+    assert_equal 1, ids.length
+    @userkeys.sign( ids.first, "hello" )
+    data = @userkeys.sign( ids.first, "hello" )
+    assert_equal "\0\0\0\4test\0\0\0\11<hello:2>", data
+  end
+
+  def test_use_agent
+    assert @userkeys.use_agent?
+    @userkeys.identities
+    assert_equal :opened, @agent_factory.state
+    @userkeys.use_agent = false
+    assert_equal :closed, @agent_factory.state
+    @userkeys.use_agent = true
+    assert_equal :closed, @agent_factory.state
+    @userkeys.identities
+    assert_equal :opened, @agent_factory.state
+  end
+
+end