net-ssh 1.1.4 → 2.0.0

data/test/transport/tc_session.rb

@@ -1,296 +0,0 @@
-#--
-# =============================================================================
-# Copyright (c) 2004,2005 Jamis Buck (jamis@37signals.com)
-# 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 "#{File.dirname(__FILE__)}/../../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
-    attr_accessor :open_delay
-
-    def initialize( script )
-      @replies = []
-      @script = script
-      @open_delay = 0
-    end
-
-    def open( host, port )
-      @replies << "#{host}:#{port}"
-      sleep @open_delay
-      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
-
-  def test_timeout_expired
-    @socket.open_delay = 2
-    assert_raise(Timeout::Error) { do_setup( "the.host.com", :timeout => 1 ) }
-  end
-
-  def test_timeout_not_expired
-    @socket.open_delay = 1
-    assert_nothing_raised { do_setup( "the.host.com", :timeout => 2 ) }
-  end
-end

data/test/transport/tc_version_negotiator.rb

@@ -1,86 +0,0 @@
-#--
-# =============================================================================
-# Copyright (c) 2004,2005 Jamis Buck (jamis@37signals.com)
-# 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 "#{File.dirname(__FILE__)}/../../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

@@ -1,136 +0,0 @@
-#--
-# =============================================================================
-# Copyright (c) 2004,2005 Jamis Buck (jamis@37signals.com)
-# 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 "#{File.dirname(__FILE__)}/../../../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