net-ssh 0.5.0

data/test/transport/ossl/hmac/tc_sha1_96.rb

@@ -0,0 +1,50 @@
+#--
+# =============================================================================
+# 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/ossl/hmac/sha1-96'
+require 'test/unit'
+
+class TC_HMAC_SHA1_96 < Test::Unit::TestCase
+
+  def setup
+    @hmac = Net::SSH::Transport::OSSL::HMAC::SHA1_96.new.new( "1234567890123456789012345" )
+  end
+
+  def test_key
+    assert_equal "12345678901234567890", @hmac.key
+  end
+
+  def test_key_length
+    assert_equal 20, @hmac.key_length
+  end
+
+  def test_mac_length
+    assert_equal 12, @hmac.mac_length
+  end
+
+  def test_digest_class
+    assert_equal OpenSSL::Digest::SHA1, @hmac.digest_class
+  end
+
+  def test_digest
+    expect = "\203\266\253\336\242\311zE\360~\325j"
+    assert_equal expect,
+      @hmac.digest( "To be, or not to be, that is the question" )
+  end
+
+end

data/test/transport/ossl/tc_buffer.rb

@@ -0,0 +1,97 @@
+#--
+# =============================================================================
+# 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/ossl/buffer'
+require 'test/unit'
+
+class TC_OSSLBuffer < Test::Unit::TestCase
+
+  def setup
+    @buffer = Net::SSH::Transport::OSSL::ReaderBuffer.new( "" )
+  end
+
+  def test_read_bignum
+    @buffer.append "\x00\x00\x00\x01\x01" +
+                   "\x00\x00\x00\x04\x12\x34\x56\x78" +
+                   "\x00\x00\x00\x07\x00\xAB\xCD\xEF\x12\x34\x56"
+    a = @buffer.read_bignum
+    b = @buffer.read_bignum
+    c = @buffer.read_bignum
+    d = @buffer.read_bignum
+    assert_equal 0x01, a
+    assert_equal 0x12345678, b
+    assert_equal 0xABCDEF123456, c
+    assert_nil d
+  end
+
+  def test_read_key
+    @buffer.append "\x00\x00\x00\x07ssh-dss" +
+                   "\x00\x00\x00\x01\x01" +
+                   "\x00\x00\x00\x01\x02" +
+                   "\x00\x00\x00\x01\x03" +
+                   "\x00\x00\x00\x01\x04" +
+                   "\x00\x00\x00\x07ssh-rsa" +
+                   "\x00\x00\x00\x01\x01" +
+                   "\x00\x00\x00\x01\x02"
+
+    k1 = @buffer.read_key
+    k2 = @buffer.read_key
+    k3 = @buffer.read_key
+
+    assert_instance_of OpenSSL::PKey::DSA, k1
+    assert_instance_of OpenSSL::PKey::RSA, k2
+    assert_equal 0x01, k1.p
+    assert_equal 0x02, k1.q
+    assert_equal 0x03, k1.g
+    assert_equal 0x04, k1.pub_key
+    assert_equal 0x01, k2.e
+    assert_equal 0x02, k2.n
+    assert_nil k3
+  end
+
+  def test_read_keyblob
+    blob1 = "\x00\x00\x00\x01\x01" +
+            "\x00\x00\x00\x01\x02" +
+            "\x00\x00\x00\x01\x03" +
+            "\x00\x00\x00\x01\x04" +
+
+    blob2 = "\x00\x00\x00\x01\x01" +
+            "\x00\x00\x00\x01\x02"
+
+    @buffer.append blob1
+    @buffer.append blob2
+
+    k1 = @buffer.read_keyblob( "ssh-dss" )
+    k2 = @buffer.read_keyblob( "ssh-rsa" )
+
+    assert_raises( NotImplementedError ) do
+      k3 = @buffer.read_keyblob( "bogus" )
+    end
+
+    assert_instance_of OpenSSL::PKey::DSA, k1
+    assert_instance_of OpenSSL::PKey::RSA, k2
+    assert_equal 0x01, k1.p
+    assert_equal 0x02, k1.q
+    assert_equal 0x03, k1.g
+    assert_equal 0x04, k1.pub_key
+    assert_equal 0x01, k2.e
+    assert_equal 0x02, k2.n
+  end
+
+
+end

data/test/transport/ossl/tc_buffer_factory.rb

@@ -0,0 +1,67 @@
+#--
+# =============================================================================
+# 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/ossl/buffer-factory'
+require 'test/unit'
+
+class TC_OSSLBufferFactory < Test::Unit::TestCase
+
+  def setup
+    @factory = Net::SSH::Transport::OSSL::BufferFactory.new
+  end
+
+  def test_reader
+    reader = @factory.reader( "test" )
+    assert_instance_of Net::SSH::Transport::OSSL::ReaderBuffer, reader
+  end
+
+  def test_reader_interface
+    reader = @factory.reader( "test" )
+    assert_respond_to reader, :read_key
+    assert_respond_to reader, :read_keyblob
+    assert_respond_to reader, :read_bignum
+  end
+
+  def test_reader_init
+    reader = @factory.reader( "test" )
+    assert_equal "test", reader.to_s
+  end
+
+  def test_buffer
+    buffer = @factory.buffer
+    assert_instance_of Net::SSH::Transport::OSSL::Buffer, buffer
+  end
+
+  def test_buffer_interface
+    buffer = @factory.buffer
+    assert_respond_to buffer, :read_key
+    assert_respond_to buffer, :read_keyblob
+    assert_respond_to buffer, :read_bignum
+  end
+
+  def test_buffer_init
+    buffer = @factory.reader( "test" )
+    assert_equal "test", buffer.to_s
+  end
+
+  def test_writer_buffer
+    buffer = @factory.writer
+    assert_instance_of Net::SSH::Util::WriterBuffer, buffer
+  end
+
+end

data/test/transport/ossl/tc_cipher_factory.rb

@@ -0,0 +1,84 @@
+#--
+# =============================================================================
+# 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/ossl/cipher-factory'
+require 'test/unit'
+
+class TC_CipherFactory < Test::Unit::TestCase
+
+  MockCipher = Object.new
+
+  def setup
+    map = { "bf" => "bf", "none" => "none" }
+    @factory = Net::SSH::Transport::OSSL::CipherFactory.new( map )
+    @factory.identity_cipher = MockCipher
+  end
+
+  def test_none
+    cipher = @factory.get( "none" )
+    assert_equal MockCipher, cipher
+  end
+
+  def test_bad_cipher
+    assert_raise( Net::SSH::Transport::CipherNotFound ) do
+      @factory.get "bogus"
+    end
+  end
+
+  def test_cipher_short
+    cipher = @factory.get( "bf", "1234", "1234", "IJKL", "MNOP",
+                           OpenSSL::Digest::MD5, true )
+
+    text = cipher.update( "12345678" )
+    text << cipher.final
+
+    assert_equal "\335\027\034\242?\243\0054", text
+  end
+
+  def test_cipher_exact
+    cipher = @factory.get( "bf", "12345678", "1234567890123456", "IJKL",
+                           "MNOP", OpenSSL::Digest::MD5, true )
+
+    text = cipher.update( "12345678" )
+    text << cipher.final
+
+    assert_equal "\030&\317\021\363\200|E", text
+  end
+
+  def test_cipher_long
+    cipher = @factory.get( "bf", "1234567890", "123456789012345678", "IJKL",
+                           "MNOP", OpenSSL::Digest::MD5, true )
+
+    text = cipher.update( "12345678" )
+    text << cipher.final
+
+    assert_equal "\030&\317\021\363\200|E", text
+  end
+
+  def test_cipher_lengths
+    result = @factory.get_lengths( "bogus" )
+    assert [0, 0], result
+
+    result = @factory.get_lengths( "none" )
+    assert [0, 0], result
+
+    result = @factory.get_lengths( "bf" )
+    assert [16, 8], result
+  end
+
+end

data/test/transport/ossl/tc_digest_factory.rb

@@ -0,0 +1,39 @@
+#--
+# =============================================================================
+# 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/ossl/digest-factory'
+require 'test/unit'
+
+class TC_OSSLDigestFactory < Test::Unit::TestCase
+
+  def setup
+    map = { "test" => "hello" }
+    @factory = Net::SSH::Transport::OSSL::DigestFactory.new( map )
+  end
+
+  def test_get_not_found
+    assert_raise( Net::SSH::Transport::DigestTypeNotFound ) do
+      @factory.get( "bogus" )
+    end
+  end
+
+  def test_get_found
+    assert_equal "hello", @factory.get( "test" )
+  end
+
+end

data/test/transport/ossl/tc_hmac_factory.rb

@@ -0,0 +1,72 @@
+#--
+# =============================================================================
+# 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/ossl/hmac-factory'
+require 'test/unit'
+
+class TC_HMACFactory < Test::Unit::TestCase
+
+  class MockHMAC
+    attr_reader :key_length
+    attr_reader :key
+
+    def initialize
+      @key_length = 8
+    end
+
+    def new( key )
+      @key = key
+      self
+    end
+  end
+
+  def setup
+    map = [ { "test" => MockHMAC.new } ]
+    @factory = Net::SSH::Transport::OSSL::HMACFactory.new( map )
+  end
+
+  def test_get_not_found
+    assert_raise( Net::SSH::Transport::HMACAlgorithmNotFound ) do
+      @factory.get( "bogus" )
+    end
+  end
+
+  def test_get_found
+    hmac = nil
+    assert_nothing_raised do
+      hmac = @factory.get( "test", "12345678" )
+    end
+    assert_equal 8, hmac.key_length
+    assert_equal "12345678", hmac.key
+  end
+
+  def test_get_key_length_not_found
+    assert_raise( Net::SSH::Transport::HMACAlgorithmNotFound ) do
+      @factory.get_key_length( "bogus" )
+    end
+  end
+
+  def test_get_key_length_found
+    length = nil
+    assert_nothing_raised do
+      length = @factory.get_key_length( "test" )
+    end
+    assert_equal 8, length
+  end
+
+end

data/test/transport/ossl/tc_key_factory.rb

@@ -0,0 +1,199 @@
+#--
+# =============================================================================
+# 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/ossl/buffer'
+require 'net/ssh/transport/ossl/key-factory'
+require 'test/unit'
+
+class TC_OSSLKeyFactory < Test::Unit::TestCase
+
+  class Prompter
+    def initialize( passwd )
+      @passwd = passwd
+    end
+
+    def password( prompt )
+      if @passwd.is_a? Array
+        @passwd.shift
+      else
+        @passwd
+      end
+    end
+  end
+
+  class Buffers
+    def reader( text )
+      Net::SSH::Transport::OSSL::ReaderBuffer.new( text )
+    end
+  end
+
+  def fixture( name )
+    "#{File.dirname __FILE__}/fixtures/#{name}"
+  end
+
+  def setup
+    algos = { "test" => String }
+    @factory = Net::SSH::Transport::OSSL::KeyFactory.new( algos )
+    @factory.buffers = Buffers.new
+  end
+
+  def test_get_found
+    key = nil
+    assert_nothing_raised { key = @factory.get( "test" ) }
+    assert_equal "", key
+  end
+
+  def test_get_not_found
+    assert_raise( Net::SSH::Transport::KeyTypeNotFound ) do
+      @factory.get( "bogus" )
+    end
+  end
+
+  def test_load_private_key_not_found
+    assert_raise( Errno::ENOENT ) do
+      @factory.load_private_key "bogus"
+    end
+  end
+
+  def test_load_private_key_not_supported
+    assert_raise( OpenSSL::PKey::PKeyError ) do
+      @factory.load_private_key fixture( "not-supported" )
+    end
+  end
+
+  def test_load_private_key_not_private_key
+    assert_raise( OpenSSL::PKey::PKeyError ) do
+      @factory.load_private_key fixture( "not-a-private-key" )
+    end
+  end
+
+  def test_load_private_key_dsa_unencrypted_fail
+    assert_raise( OpenSSL::PKey::DSAError ) do
+      @factory.load_private_key fixture( "dsa-unencrypted-bad" )
+    end
+  end
+
+  def test_load_private_key_dsa_unencrypted_success
+    result = nil
+    assert_nothing_raised do
+      result = @factory.load_private_key fixture( "dsa-unencrypted" )
+    end
+    assert_instance_of OpenSSL::PKey::DSA, result
+  end
+
+  def test_load_private_key_rsa_unencrypted_fail
+    assert_raise( OpenSSL::PKey::RSAError ) do
+      @factory.load_private_key fixture( "rsa-unencrypted-bad" )
+    end
+  end
+
+  def test_load_private_key_rsa_unencrypted_success
+    result = nil
+    assert_nothing_raised do
+      result = @factory.load_private_key fixture( "rsa-unencrypted" )
+    end
+    assert_instance_of OpenSSL::PKey::RSA, result
+  end
+
+  def test_load_private_key_dsa_encrypted_fail_bad
+    @factory.prompter = Prompter.new( "key-test" )
+    assert_raise( OpenSSL::PKey::DSAError ) do
+      @factory.load_private_key fixture( "dsa-encrypted-bad" )
+    end
+  end
+
+  def test_load_private_key_dsa_encrypted_fail_wrong_password
+    @factory.prompter = Prompter.new( "key-test-bad" )
+    assert_raise( OpenSSL::PKey::DSAError ) do
+      @factory.load_private_key fixture( "dsa-encrypted" )
+    end
+  end
+
+  def test_load_private_key_dsa_encrypted_success
+    @factory.prompter = Prompter.new( "key-test" )
+    result = nil
+    assert_nothing_raised do
+      result = @factory.load_private_key fixture( "dsa-encrypted" )
+    end
+    assert_instance_of OpenSSL::PKey::DSA, result
+  end
+
+  def test_load_private_key_rsa_encrypted_fail_bad
+    @factory.prompter = Prompter.new( "key-test" )
+    assert_raise( OpenSSL::PKey::RSAError ) do
+      @factory.load_private_key fixture( "rsa-encrypted-bad" )
+    end
+  end
+
+  def test_load_private_key_rsa_encrypted_fail_wrong_password
+    @factory.prompter = Prompter.new( "key-test-bad" )
+    assert_raise( OpenSSL::PKey::RSAError ) do
+      @factory.load_private_key fixture( "rsa-encrypted" )
+    end
+  end
+
+  def test_load_private_key_rsa_encrypted_success
+    @factory.prompter = Prompter.new( "key-test" )
+    result = nil
+    assert_nothing_raised do
+      result = @factory.load_private_key fixture( "rsa-encrypted" )
+    end
+    assert_instance_of OpenSSL::PKey::RSA, result
+  end
+
+  def test_load_public_key_not_found
+    assert_raise( Errno::ENOENT ) do
+      @factory.load_public_key "bogus"
+    end
+  end
+
+  def test_load_public_key_success_rsa
+    result = nil
+    assert_nothing_raised do
+      result = @factory.load_public_key fixture( "rsa-unencrypted.pub" )
+    end
+    assert_instance_of OpenSSL::PKey::RSA, result
+  end
+
+  def test_load_public_key_success_dsa
+    result = nil
+    assert_nothing_raised do
+      result = @factory.load_public_key fixture( "dsa-unencrypted.pub" )
+    end
+    assert_instance_of OpenSSL::PKey::DSA, result
+  end
+
+  def test_load_private_key_encrypted_bad_passwd_thrice
+    passwords = [ "one", "two", "three", "four" ]
+    @factory.prompter = Prompter.new( passwords )
+    assert_raise( OpenSSL::PKey::DSAError ) do
+      @factory.load_private_key fixture( "dsa-encrypted" )
+    end
+    assert_equal [ "four" ], passwords
+  end
+
+  def test_load_private_key_encrypted_good_passwd_eventually
+    passwords = [ "one", "two", "key-test" ]
+    @factory.prompter = Prompter.new( passwords )
+    assert_nothing_raised do
+      @factory.load_private_key fixture( "dsa-encrypted" )
+    end
+    assert passwords.empty?
+  end
+
+end