net-ssh 0.5.0

data/test/transport/tc_session.rb

@@ -0,0 +1,283 @@
+#--
+# =============================================================================
+# 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/transport/constants'
+require 'net/ssh/transport/session'
+require 'net/ssh/util/buffer'
+require 'test/unit'
+require 'ostruct'
+
+class TC_Session < Test::Unit::TestCase
+  include Net::SSH::Transport::Constants
+
+  class Logger
+    attr_reader :msgs
+    def initialize
+      @msgs = []
+    end
+    def debug?; true; end
+    def debug(msg)
+      @msgs << "[D] #{msg}"
+    end
+    def info?; true; end
+    def info(msg)
+      @msgs << "[I] #{msg}"
+    end
+    def warn?; true; end
+    def warn(msg)
+      @msgs << "[W] #{msg}"
+    end
+  end
+
+  class VersionNegotiator
+    def negotiate( socket, version ); "A"; end
+  end
+
+  class AlgorithmNegotiator
+    def negotiate( session, options )
+      OpenStruct.new(
+        :server_packet => "A",
+        :client_packet => "B",
+        :kex => "C",
+        :host_key => "D",
+        :encryption_c2s => "E",
+        :encryption_s2c => "E",
+        :mac_c2s => "F",
+        :mac_s2c => "F",
+        :compression_c2s => "G",
+        :compression_s2c => "G",
+        :language_c2s => "",
+        :language_s2c => ""
+      )
+    end
+  end
+
+  class ScriptedSocket
+    attr_reader :replies
+
+    def initialize( script )
+      @replies = []
+      @script = script
+    end
+
+    def open( host, port )
+      @replies << "#{host}:#{port}"
+      self
+    end
+
+    def write( msg )
+      @replies << msg
+    end
+
+    def read
+      @script.shift
+    end
+  end
+
+  class PacketHandler
+    attr_writer :socket
+    def on_new_algos( &block )
+      @on_new_algos = block
+    end
+    def set_algorithms( *args )
+      @on_new_algos.call *args if @on_new_algos
+    end
+  end
+
+  class PacketSender < PacketHandler
+    def send( msg )
+      @socket.write msg
+    end
+  end
+
+  class PacketReceiver < PacketHandler
+    def get
+      @socket.read
+    end
+  end
+
+  class Ciphers
+    def get( *args )
+      args.first
+    end
+    def get_lengths( name )
+      [ 24, 8 ]
+    end
+  end
+
+  class HMACs
+    def get( *args )
+      args.first
+    end
+    def get_key_length( name )
+      24
+    end
+  end
+
+  class Compressor
+    def initialize( *args )
+    end
+  end
+
+  class SSHAble; def to_ssh; ""; end; end
+
+  class Digester
+    def digest( text )
+      text
+    end
+  end
+
+  class Kex
+    def exchange_keys( session, info )
+      {
+        :shared_secret => SSHAble.new,
+        :session_id => "",
+        :server_key => "",
+        :hashing_algorithm => Digester.new
+      }
+    end
+  end
+
+  def self.method_added( name )
+    super
+
+  end
+
+  def reader(text)
+    Net::SSH::Util::ReaderBuffer.new( text )
+  end
+
+  def setup
+    @script = []
+    @logger = Logger.new
+    @socket = ScriptedSocket.new( @script )
+    @sender = PacketSender.new
+    @getter = PacketReceiver.new
+  end
+
+  def do_setup( host, opts={} )
+    @session = Net::SSH::Transport::Session.new( host, opts ) do |s|
+      s.logger = @logger
+      s.default_port = 22
+      s.version_negotiator = VersionNegotiator.new
+      s.algorithm_negotiator = AlgorithmNegotiator.new
+      s.socket_factory = @socket
+      s.packet_sender = @sender
+      s.packet_receiver = @getter
+      s.ciphers = Ciphers.new
+      s.hmacs = HMACs.new
+      s.kexs = { "C" => Kex.new }
+      s.compressors = { "G" => Compressor }
+      s.decompressors = { "G" => Compressor }
+    end
+  end
+
+  def test_bad_option
+    assert_raise( ArgumentError ) do
+      do_setup( "the-host", :bogus => "thing" )
+    end
+  end
+
+  def test_open
+    @sender.on_new_algos do |a,b,c|
+      assert_equal "E", a
+      assert_equal "F", b
+      assert_instance_of Compressor, c
+    end
+
+    @getter.on_new_algos do |a,b,c|
+      assert_equal "E", a
+      assert_equal "F", b
+      assert_instance_of Compressor, c
+    end
+
+    do_setup "the.host.com"
+
+    assert_equal [ "the.host.com:22" ], @socket.replies
+  end
+
+  def test_send_message
+    do_setup "the.host.com"
+    @session.send_message "sending"
+    assert_equal [ "the.host.com:22", "sending" ], @socket.replies
+  end
+
+  def test_wait_for_message
+    @script << reader( "\xFFhello" )
+    do_setup "the.host.com"
+    type, buffer = @session.wait_for_message
+    assert_equal 255, type
+    assert_equal "hello", buffer.remainder_as_buffer.content
+  end
+
+  def test_wait_for_disconnect
+    @script << reader( "#{DISCONNECT.chr}\0\0\0\1\0\0\0\1A\0\0\0\1B" )
+    do_setup "the.host.com"
+    assert_raise( Net::SSH::Transport::Disconnect ) do
+      @session.wait_for_message
+    end
+  end
+
+  def test_wait_for_ignore
+    @script << reader( "#{IGNORE.chr}\0\0\0\1A" )
+    @script << reader( "\xFFhello" )
+    do_setup "the.host.com"
+    type, buffer = @session.wait_for_message
+
+    assert_equal 255, type
+    assert_equal "hello", buffer.remainder_as_buffer.content
+
+    assert @logger.msgs.include?("[I] received IGNORE message (\"A\")")
+  end
+
+  def test_wait_for_debug_quiet
+    @script << reader( "#{DEBUG.chr}\0\0\0\0\1A\0\0\0\1B" )
+    @script << reader( "\xFFhello" )
+    do_setup "the.host.com"
+    type, buffer = @session.wait_for_message
+
+    assert_equal 255, type
+    assert_equal "hello", buffer.remainder_as_buffer.content
+
+    assert @logger.msgs.include?("[D] A (B)")
+  end
+
+  def test_wait_for_debug_verbose
+    @script << reader( "#{DEBUG.chr}\1\0\0\0\1A\0\0\0\1B" )
+    @script << reader( "\xFFhello" )
+    do_setup "the.host.com"
+    type, buffer = @session.wait_for_message
+
+    assert_equal 255, type
+    assert_equal "hello", buffer.remainder_as_buffer.content
+
+    assert @logger.msgs.include?("[W] A (B)")
+  end
+
+  def test_wait_for_kexinit
+    @script << reader( "#{KEXINIT.chr}\1\0\0\0\1A\0\0\0\1B" )
+    @script << reader( "\xFFhello" )
+    do_setup "the.host.com"
+    type, buffer = @session.wait_for_message
+
+    assert_equal 255, type
+    assert_equal "hello", buffer.remainder_as_buffer.content
+
+    assert @logger.msgs.include?("[I] re-key requested")
+  end
+end

data/test/transport/tc_version_negotiator.rb

@@ -0,0 +1,86 @@
+#--
+# =============================================================================
+# 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/transport/version-negotiator'
+require 'test/unit'
+
+class TC_VersionNegotiator < Test::Unit::TestCase
+
+  class MockLogger; def debug?; false; end; end
+
+  class ScriptedSocket
+    attr_reader :lines
+
+    def initialize( *script )
+      @lines = []
+      @script = script
+    end
+
+    def readline
+      @script.shift
+    end
+
+    def print( msg )
+      @lines << msg
+    end
+  end
+
+  def setup
+    logger = MockLogger.new
+    @negotiator = Net::SSH::Transport::VersionNegotiator.new( logger )
+  end
+
+  def test_negotiate_bad_version
+    socket = ScriptedSocket.new( "SSH-1.5-Bogus/Thing\n" )
+    assert_raise( Net::SSH::Exception ) do
+      @negotiator.negotiate( socket, "SSH-2.0-My/Version" )
+    end
+  end
+
+  def test_negotiate_compat_version
+    socket = ScriptedSocket.new( "SSH-1.99-Bogus/Thing\n" )
+    version = nil
+    assert_nothing_raised do
+      version = @negotiator.negotiate( socket, "SSH-2.0-My/Version" )
+    end
+    assert_equal "SSH-1.99-Bogus/Thing", version
+    assert_equal [ "SSH-2.0-My/Version\r\n" ], socket.lines
+  end
+
+  def test_negotiate_good_version
+    socket = ScriptedSocket.new( "SSH-2.0-Bogus/Thing\n" )
+    version = nil
+    assert_nothing_raised do
+      version = @negotiator.negotiate( socket, "SSH-2.0-My/Version" )
+    end
+    assert_equal "SSH-2.0-Bogus/Thing", version
+    assert_equal [ "SSH-2.0-My/Version\r\n" ], socket.lines
+  end
+
+  def test_header_lines
+    socket = ScriptedSocket.new( "First Line", "Second Line", "SSH-2.0-Bogus/Thing\n" )
+    version = nil
+    assert_nothing_raised do
+      version = @negotiator.negotiate( socket, "SSH-2.0-My/Version" )
+    end
+    assert_equal "SSH-2.0-Bogus/Thing", version
+    assert_equal [ "SSH-2.0-My/Version\r\n" ], socket.lines
+    assert_equal [ "First Line", "Second Line" ], @negotiator.header_lines
+  end
+
+end

data/test/userauth/methods/tc_hostbased.rb

@@ -0,0 +1,136 @@
+#--
+# =============================================================================
+# 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/methods/hostbased'
+require 'net/ssh/util/buffer'
+require 'test/unit'
+require 'ostruct'
+
+class TC_Methods_HostBased < Test::Unit::TestCase
+
+  class Buffers
+    def writer
+      Net::SSH::Util::WriterBuffer.new
+    end
+  end
+
+  class Key < OpenStruct
+    def initialize( e, n )
+      super( :ssh_type => "ssh-rsa",
+             :e => OpenStruct.new( :to_ssh => [ e ].pack("N") ),
+             :n => OpenStruct.new( :to_ssh => [ n ].pack("N") ) )
+    end
+  end
+
+  class Messenger
+    attr_reader :data
+    attr_reader :messages
+
+    def initialize
+      @data = []
+      @messages = []
+    end
+
+    def send_message( msg )
+      @messages << msg.to_s
+    end
+
+    def wait_for_message
+      @data.shift
+    end
+  end
+
+  class KeyManager
+    attr_reader :host_identities
+    attr_reader :state
+    attr_reader :sigdata
+
+    def initialize( *identities )
+      @host_identities = identities.flatten
+      @state = :open
+    end
+
+    def sign( identity, data )
+      @sigdata = [ identity, data.to_s ]
+      "<signature>"
+    end
+
+    def finish
+      @state = :finished
+    end
+  end
+
+  def setup
+    ENV["USER"] = "test_client_user"
+    buffers = Buffers.new
+    @messenger = Messenger.new
+    @method = Net::SSH::UserAuth::Methods::HostBased.new( buffers )
+    @method.messenger = @messenger
+    @method.session_id = "test"
+    @method.hostname = "test.host"
+  end
+
+  def test_authenticate_no_key_manager
+    assert !@method.authenticate( "test", "test_user" )
+    assert @messenger.messages.empty?
+  end
+
+  def test_authenticate_no_identities
+    manager = KeyManager.new
+    assert !@method.authenticate( "test", "test_user", :key_manager => manager )
+    assert_equal :finished, manager.state
+  end
+
+  def test_authenticate_success
+    manager = KeyManager.new( Key.new( 0x01010101, 0x02020202 ) )
+    @messenger.data.concat [ OpenStruct.new( :message_type => 52 ) ]
+    assert @method.authenticate( "test", "test_user", :key_manager => manager )
+    assert_equal :finished, manager.state
+    assert_equal 1, @messenger.messages.length
+
+    sig = "#{50.chr}\0\0\0\11test_user\0\0\0\4test\0\0\0\11hostbased\0\0\0\7ssh-rsa" +
+          "\0\0\0\23\0\0\0\7ssh-rsa\1\1\1\1\2\2\2\2" +
+          "\0\0\0\12test.host.\0\0\0\20test_client_user"
+    packet1 = sig + "\0\0\0\13<signature>"
+    assert_equal packet1, @messenger.messages[0]
+
+    sigdata = manager.sigdata
+    assert_equal "\0\0\0\4test" + sig, sigdata[1]
+  end
+
+  def test_authenticate_fail
+    manager = KeyManager.new( Key.new( 0x01010101, 0x02020202 ),
+                              Key.new( 0x03030303, 0x04040404 ) )
+    @messenger.data.concat [ OpenStruct.new( :message_type => 51 ),
+                             OpenStruct.new( :message_type => 51 ) ]
+    assert !@method.authenticate( "test", "test_user", :key_manager => manager )
+    assert_equal :finished, manager.state
+    assert_equal 2, @messenger.messages.length
+  end
+
+  def test_authenticate_acceptible_identities_error
+    manager = KeyManager.new( Key.new( 0x01010101, 0x02020202 ) )
+    @messenger.data.concat [ OpenStruct.new( :message_type => 60 ),
+                             OpenStruct.new( :message_type => 0 ) ]
+    assert_raise( Net::SSH::Exception ) do
+      @method.authenticate( "test", "test_user", :key_manager => manager )
+    end
+    assert_equal 1, @messenger.messages.length
+  end
+
+end