net-ssh 0.5.0

data/test/connection/tc_integration.rb

@@ -0,0 +1,85 @@
+#--
+# =============================================================================
+# 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/connection/services'
+  require 'net/ssh/transport/services'
+  require 'net/ssh/userauth/services'
+  require 'test/unit'
+
+  class TC_Connection_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 )
+      Net::SSH::Connection.register_services( @registry )
+
+      @registry.define do |b|
+        b.crypto_backend { :ossl }
+        b.transport_host { HOST }
+      end
+
+      @registry[:userauth][:driver].authenticate SERVICE, USER, PASSWORD
+
+      @connection = @registry[:connection][:driver]
+    end
+
+    def teardown
+      @registry[:transport][:session].close
+      @registry.logs.close
+    end
+
+    def test_exec
+      exec_data = ""
+      @connection.open_channel "session" do |chan|
+        chan.on_data { |ch,data| exec_data << data }
+        chan.exec "echo $HOME"
+      end
+      @connection.loop
+      assert_equal "/home/test\n", exec_data
+    end
+
+    def test_dialog
+      dialog = [ "2+2", "5*10+1", "quit" ]
+      results = []
+      @connection.open_channel "session" do |chan|
+        chan.on_data do |ch,data|
+          results << data
+          chan.send_data dialog.shift + "\n"
+        end
+        chan.exec "bc"
+        chan.send_data dialog.shift + "\n"
+      end
+      @connection.loop
+      assert_equal [ "4\n", "51\n" ], results
+    end
+
+  end
+
+end

data/test/proxy/tc_http.rb

@@ -0,0 +1,209 @@
+#--
+# =============================================================================
+# 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 'test/unit'
+require 'net/ssh/proxy/http'
+require 'socket'
+
+class TC_Proxy_HTTP < Test::Unit::TestCase
+
+  HOST = "test.host"
+  PORT = 22117
+
+  class ScriptableHTTPServer
+    attr_reader :script
+    attr_reader :events
+
+    def initialize
+      @script = []
+      @events = []
+    end
+
+    def run
+      @socket = TCPServer.new( HOST, PORT )
+      @thread = Thread.new { run_server until @script.empty? }
+    end
+
+    def run_server
+      client = @socket.accept
+      loop do
+        @events << client.readline 
+        break if @events.last == "\n"
+      end
+      client.puts @script.shift
+    rescue Exception => e
+      puts "#{e.class}: #{e.message}"
+      puts e.backtrace.join("\n")
+    end
+
+    def wait
+      @thread.join
+    end
+
+    def shutdown
+      @socket.shutdown
+    end
+  end
+
+  def setup
+    ENV['HTTP_PROXY_USER'] = nil
+    ENV['HTTP_PROXY_PASSWORD'] = nil
+    ENV['CONNECT_USER'] = nil
+    ENV['CONNECT_PASSWORD'] = nil
+
+    @server = ScriptableHTTPServer.new
+    @options = Hash.new
+    @proxy = Net::SSH::Proxy::HTTP.new( HOST, PORT, @options )
+  end
+
+  def teardown
+    @server.shutdown
+  end
+
+  def test_no_auth_ok
+    @server.script << "HTTP/1.0 200 OK\n\n"
+    @server.run
+
+    @proxy.open( "foo.com", 1234 )
+    @server.wait
+
+    assert_equal [ "CONNECT foo.com:1234 HTTP/1.0\n", "\n" ], @server.events
+  end
+
+  def test_connect_error
+    @server.script << "HTTP/1.0 500 Error\n\n"
+    @server.run
+
+    assert_raise( Net::SSH::Proxy::ConnectError ) do
+      @proxy.open( "foo.com", 1234 )
+    end
+
+    @server.wait
+
+    assert_equal [ "CONNECT foo.com:1234 HTTP/1.0\n", "\n" ], @server.events
+  end
+
+  def test_unauthorized_error_no_user
+    @server.script << "HTTP/1.0 407 Error\n\n"
+    @server.run
+
+    assert_raise( Net::SSH::Proxy::UnauthorizedError ) do
+      @proxy.open( "foo.com", 1234 )
+    end
+
+    @server.wait
+
+    assert_equal [ "CONNECT foo.com:1234 HTTP/1.0\n", "\n" ], @server.events
+  end
+
+  def test_invalid_auth_scheme
+    @options[:user] = 'test'
+
+    @server.script << "HTTP/1.0 407 Error\nProxy-Authenticate: Foo 1 2 3\n\n"
+    @server.run
+
+    assert_raise( NotImplementedError ) do
+      @proxy.open( "foo.com", 1234 )
+    end
+
+    @server.wait
+
+    assert_equal [ "CONNECT foo.com:1234 HTTP/1.0\n", "\n" ], @server.events
+  end
+
+  def test_connect_error_bad_auth
+    @options[:user] = 'test'
+    @options[:password] = 'password'
+
+    @server.script << "HTTP/1.0 407 Error\nProxy-Authenticate: Basic\n\n"
+    @server.script << "HTTP/1.0 500 Error\n\n"
+    @server.run
+
+    assert_raise( Net::SSH::Proxy::ConnectError ) do
+      @proxy.open( "foo.com", 1234 )
+    end
+
+    @server.wait
+
+    assert_equal [ "CONNECT foo.com:1234 HTTP/1.0\n", "\n",
+                   "CONNECT foo.com:1234 HTTP/1.0\n",
+                   "Proxy-Authorization: Basic dGVzdDpwYXNzd29yZA==\n", "\n" ],
+                  @server.events
+  end
+
+  def test_connect_auth_success
+    @options[:user] = 'test'
+    @options[:password] = 'password'
+
+    @server.script << "HTTP/1.0 407 Error\nProxy-Authenticate: Basic\n\n"
+    @server.script << "HTTP/1.0 200 OK\n\n"
+    @server.run
+
+    assert_nothing_raised do
+      @proxy.open( "foo.com", 1234 )
+    end
+
+    @server.wait
+
+    assert_equal [ "CONNECT foo.com:1234 HTTP/1.0\n", "\n",
+                   "CONNECT foo.com:1234 HTTP/1.0\n",
+                   "Proxy-Authorization: Basic dGVzdDpwYXNzd29yZA==\n", "\n" ],
+                  @server.events
+  end
+
+  def test_connect_auth_success_with_HTTP_PROXY_vars
+    ENV['HTTP_PROXY_USER'] = 'test'
+    ENV['HTTP_PROXY_PASSWORD'] = 'password'
+
+    @server.script << "HTTP/1.0 407 Error\nProxy-Authenticate: Basic\n\n"
+    @server.script << "HTTP/1.0 200 OK\n\n"
+    @server.run
+
+    assert_nothing_raised do
+      @proxy.open( "foo.com", 1234 )
+    end
+
+    @server.wait
+
+    assert_equal [ "CONNECT foo.com:1234 HTTP/1.0\n", "\n",
+                   "CONNECT foo.com:1234 HTTP/1.0\n",
+                   "Proxy-Authorization: Basic dGVzdDpwYXNzd29yZA==\n", "\n" ],
+                  @server.events
+  end
+
+  def test_connect_auth_success_with_CONNECT_vars
+    ENV['CONNECT_USER'] = 'test'
+    ENV['CONNECT_PASSWORD'] = 'password'
+
+    @server.script << "HTTP/1.0 407 Error\nProxy-Authenticate: Basic\n\n"
+    @server.script << "HTTP/1.0 200 OK\n\n"
+    @server.run
+
+    assert_nothing_raised do
+      @proxy.open( "foo.com", 1234 )
+    end
+
+    @server.wait
+
+    assert_equal [ "CONNECT foo.com:1234 HTTP/1.0\n", "\n",
+                   "CONNECT foo.com:1234 HTTP/1.0\n",
+                   "Proxy-Authorization: Basic dGVzdDpwYXNzd29yZA==\n", "\n" ],
+                  @server.events
+  end
+
+end

data/test/proxy/tc_socks4.rb

@@ -0,0 +1,148 @@
+#--
+# =============================================================================
+# 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 'test/unit'
+require 'net/ssh/proxy/socks4'
+require 'socket'
+
+class TC_Proxy_SOCKS4 < Test::Unit::TestCase
+
+  HOST = "test.host"
+  PORT = 22117
+
+  class ScriptableSOCKS4Server
+    attr_reader :script
+    attr_reader :events
+
+    def initialize
+      @script = []
+      @events = []
+    end
+
+    def run
+      @socket = TCPServer.new( HOST, PORT )
+      @thread = Thread.new { run_server until @script.empty? }
+    end
+
+    def run_server
+      client = @socket.accept
+      packet = client.read(8)
+      c = nil
+      packet << c while ( c = client.read(1) ) != "\0"
+      packet << "\0"
+      @events << packet
+      client.send @script.shift, 0
+    rescue Exception => e
+      puts "#{e.class}: #{e.message}"
+      puts e.backtrace.join("\n")
+    end
+
+    def wait
+      @thread.join
+    end
+
+    def shutdown
+      @socket.shutdown
+    end
+  end
+
+  def setup
+    ENV['SOCKS_USER'] = nil
+    ENV['CONNECT_USER'] = nil
+
+    @server = ScriptableSOCKS4Server.new
+    @options = Hash.new
+    @proxy = Net::SSH::Proxy::SOCKS4.new( HOST, PORT, @options )
+  end
+
+  def teardown
+    @server.shutdown
+  end
+
+  def test_no_auth_ok
+    @server.script << "\4\132\0\0\0\0\0\0\0\0"
+    @server.run
+
+    assert_nothing_raised do
+      @proxy.open( "test.host", 1234 )
+    end
+
+    @server.wait
+
+    assert_equal [ "\4\1\4\xD2\x7F\0\0\1\0" ], @server.events
+  end
+
+  def test_no_auth_fail
+    @server.script << "\4\133\0\0\0\0\0\0\0\0"
+    @server.run
+
+    assert_raise( Net::SSH::Proxy::ConnectError ) do
+      @proxy.open( "test.host", 1234 )
+    end
+
+    @server.wait
+
+    assert_equal [ "\4\1\4\xD2\x7F\0\0\1\0" ], @server.events
+  end
+
+  def test_auth_options
+    @options[:user] = "test"
+
+    @server.script << "\4\132\0\0\0\0\0\0\0\0"
+    @server.run
+
+    assert_nothing_raised do
+      @proxy.open( "test.host", 1234 )
+    end
+
+    @server.wait
+
+    assert_equal [ "\4\1\4\xD2\x7F\0\0\1test\0" ], @server.events
+  end
+
+  def test_auth_SOCKS_var
+    ENV["SOCKS_USER"] = "test"
+
+    @server.script << "\4\132\0\0\0\0\0\0\0\0"
+    @server.run
+
+    assert_nothing_raised do
+      @proxy.open( "test.host", 1234 )
+    end
+
+    @server.wait
+
+    assert_equal [ "\4\1\4\xD2\x7F\0\0\1test\0" ], @server.events
+  end
+
+  def test_auth_CONNECT_var
+    ENV["CONNECT_USER"] = "test"
+
+    @server.script << "\4\132\0\0\0\0\0\0\0\0"
+    @server.run
+
+    assert_nothing_raised do
+      @proxy.open( "test.host", 1234 )
+    end
+
+    @server.wait
+
+    assert_equal [ "\4\1\4\xD2\x7F\0\0\1test\0" ], @server.events
+  end
+
+end

data/test/proxy/tc_socks5.rb

@@ -0,0 +1,214 @@
+#--
+# =============================================================================
+# 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 'test/unit'
+require 'net/ssh/proxy/socks5'
+require 'socket'
+
+class TC_Proxy_SOCKS5 < Test::Unit::TestCase
+
+  HOST = "test.host"
+  PORT = 22117
+
+  class ScriptableSOCKS5Server
+    attr_reader :script
+    attr_reader :events
+
+    def initialize
+      @script = []
+      @events = []
+    end
+
+    def run
+      @socket = TCPServer.new( HOST, PORT )
+      @thread = Thread.new { run_server }
+    end
+
+    def run_server
+      client = @socket.accept
+
+      data = client.read(2)
+      count = data[1]
+      data << client.read(count)
+      @events << data
+      client.send @script.shift, 0
+      return if @script.empty?
+
+      if count > 1
+        data = client.read(2)
+        data << client.read(data[data.length-1]+1)
+        data << client.read(data[data.length-1])
+        @events << data
+        client.send @script.shift, 0
+        return if @script.empty?
+      end
+
+      data = client.read(4)
+      t = data[3]
+      if t == 1
+        data << client.read(4)
+      elsif t == 3
+        data << client.read(1)
+        length = data[4]
+        data << client.read(length)
+      end
+      data << client.read(2)
+      @events << data
+      client.send @script.shift, 0
+
+    rescue Exception => e
+      puts "#{e.class}: #{e.message}"
+      puts e.backtrace.join("\n")
+    end
+
+    def wait
+      @thread.join
+    end
+
+    def shutdown
+      @socket.shutdown
+    end
+  end
+
+  def setup
+    ENV['SOCKS_USER'] = nil
+    ENV['SOCKS_PASSWORD'] = nil
+    ENV['CONNECT_USER'] = nil
+    ENV['CONNECT_PASSWORD'] = nil
+
+    @server = ScriptableSOCKS5Server.new
+    @options = Hash.new
+    @proxy = Net::SSH::Proxy::SOCKS5.new( HOST, PORT, @options )
+  end
+
+  def teardown
+    @server.shutdown
+  end
+
+  def test_bad_version
+    @server.script << "\4\0"
+    @server.run
+
+    assert_raise( Net::SSH::Proxy::Error ) do
+      @proxy.open( "test.host", 1234 )
+    end
+
+    @server.wait
+
+    assert_equal [ "\5\1\0" ], @server.events
+  end
+
+  def test_no_supported_methods
+    @server.script << "\5\xff"
+    @server.run
+
+    assert_raise( Net::SSH::Proxy::Error ) do
+      @proxy.open( "test.host", 1234 )
+    end
+
+    @server.wait
+
+    assert_equal [ "\5\1\0" ], @server.events
+  end
+
+  def test_no_auth_fail
+    @server.script << "\5\0"
+    @server.script << "\5\1\0\0\4\0\0\0\0\0\0"
+    @server.run
+
+    assert_raise( Net::SSH::Proxy::ConnectError ) do
+      @proxy.open( "test.host", 1234 )
+    end
+
+    @server.wait
+
+    assert_equal [ "\5\1\0", "\5\1\0\3\11test.host\4\322" ], @server.events
+  end
+
+  def test_no_auth_succeed_atyp_ipv4
+    @server.script << "\5\0"
+    @server.script << "\5\0\0\0\4\0\0\0\0\0\0"
+    @server.run
+
+    assert_nothing_raised do
+      @proxy.open( "1.2.3.4", 1234 )
+    end
+
+    @server.wait
+
+    assert_equal [ "\5\1\0", "\5\1\0\1\1\2\3\4\4\322" ], @server.events
+  end
+
+  def test_no_auth_succeed_atyp_domain
+    @server.script << "\5\0"
+    @server.script << "\5\0\0\0\4\0\0\0\0\0\0"
+    @server.run
+
+    assert_nothing_raised do
+      @proxy.open( "test.host", 1234 )
+    end
+
+    @server.wait
+
+    assert_equal [ "\5\1\0", "\5\1\0\3\11test.host\4\322" ], @server.events
+  end
+
+  def test_authorize_fail
+    @options[:user], @options[:password] = "foo", "bar"
+
+    @server.script << "\5\2"
+    @server.script << "\5\1"
+    @server.run
+
+    assert_raise( Net::SSH::Proxy::UnauthorizedError ) do
+      @proxy.open( "test.host", 1234 )
+    end
+
+    @server.wait
+
+    assert_equal [ "\5\2\0\2", "\5\3foo\3bar" ], @server.events
+  end
+
+  [ 
+    [:options,
+      lambda { |o| o[:user], o[:password] = "foo", "bar" } ],
+    [:socks,
+      lambda { ENV['SOCKS_USER'], ENV['SOCKS_PASSWORD'] = "foo", "bar" } ],
+    [:connect,
+      lambda { ENV['CONNECT_USER'], ENV['CONNECT_PASSWORD'] = "foo", "bar" } ]
+  ].each do |name, cb|
+    define_method "test_authorize_via_#{name}".to_sym do
+      cb.call( @options )
+      
+      @server.script << "\5\2"
+      @server.script << "\5\0"
+      @server.script << "\5\0\0\0\4\0\0\0\0\0\0"
+      @server.run
+
+      assert_nothing_raised do
+        @proxy.open( "test.host", 1234 )
+      end
+
+      @server.wait
+
+      assert_equal [ "\5\2\0\2", "\5\3foo\3bar",
+        "\5\1\0\3\11test.host\4\322" ], @server.events
+    end
+  end
+
+end