coderrr-rtunnel 0.3.9 → 0.4.0

data/test/test_socket_factory.rb

@@ -0,0 +1,42 @@
+require 'rtunnel'
+
+require 'test/unit'
+
+class SocketFactoryTest < Test::Unit::TestCase
+  SF = RTunnel::SocketFactory
+  
+  def setup
+    
+  end
+
+  def teardown
+    
+  end
+  
+  def test_host_from_address
+    assert_equal nil, SF.host_from_address(nil)
+    assert_equal '127.0.0.1', SF.host_from_address('127.0.0.1')
+    assert_equal '127.0.0.1', SF.host_from_address('127.0.0.1:1234')
+    assert_equal 'fe80::1%lo0', SF.host_from_address('fe80::1%lo0')
+    assert_equal 'fe80::1%lo0', SF.host_from_address('fe80::1%lo0:19020')
+  end
+  
+  def test_port_from_address
+    assert_equal nil, SF.port_from_address(nil)
+    assert_equal nil, SF.port_from_address('127.0.0.1')
+    assert_equal 1234, SF.port_from_address('127.0.0.1:1234')
+    assert_equal nil, SF.port_from_address('fe80::1%lo0')
+    assert_equal 19020, SF.port_from_address('fe80::1%lo0:19020')
+  end
+  
+  def test_inbound
+    assert SF.inbound?(:in_port => 1)
+    assert SF.inbound?(:in_host => '1')
+    assert SF.inbound?(:in_addr => '1')
+    assert SF.inbound?(:inbound => true)
+    assert !SF.inbound?(:out_port => 1)
+    assert !SF.inbound?(:out_host => '1')
+    assert !SF.inbound?(:out_addr => '1')
+  end
+end
+

data/test/test_tunnel.rb

@@ -0,0 +1,186 @@
+require 'rtunnel'
+
+require 'test/unit'
+
+require 'rubygems'
+require 'eventmachine'
+
+require 'test/scenario_connection.rb'
+
+# Integration tests ensuring that we can start a tunnel.
+class TunnelTest < Test::Unit::TestCase
+  def setup
+    super
+    
+    @connection_time = 0.001
+    @secure_connection_time = 0.5
+    @log_level = 'debug'
+    @local_host = '127.0.0.1'
+    @listen_port = 21335
+    @tunnel_port = 21336
+    @control_port = 21337
+    @key_file = 'test_data/ssh_host_rsa_key'
+    @hosts_file = 'test_data/known_hosts'
+    
+    @tunnel_server = new_server
+    @tunnel_client = new_client
+  end
+  
+  def new_server(extra_options = {})
+    RTunnel::Server.new({
+        :control_address => "#{@local_host}:#{@control_port}",
+        :log_level => @log_level
+        }.merge(extra_options))
+  end
+  
+  def new_client(extra_options = {})
+    RTunnel::Client.new({
+            :control_address => "#{@local_host}:#{@control_port}",
+            :remote_listen_address => "#{@local_host}:#{@listen_port}",
+            :tunnel_to_address => "#{@local_host}:#{@tunnel_port}",
+            :log_level => @log_level
+            }.merge(extra_options))
+  end
+  
+  def tunnel_test(connection_time = nil)
+    @stop_proc = proc do
+      @tunnel_client.stop
+      @tunnel_server.stop
+      @stop_proc = nil
+    end
+
+    EventMachine::run do
+      @tunnel_server.start
+      @tunnel_client.start
+
+      if connection_time
+        EventMachine.add_timer(connection_time) { yield }
+      else
+        EventMachine.next_tick { yield }
+      end
+    end
+  end
+  
+  def test_client_driven_tunnel
+    tunnel_test do
+      @tunnel_server.on_remote_listen do
+        EventMachine::start_server @local_host, @tunnel_port,
+            ScenarioConnection, self, [[:recv, 'Hello'], [:send, 'World'],
+                                       [:unbind], [:stop, @stop_proc]]
+
+        EventMachine::connect @local_host, @listen_port,
+            ScenarioConnection, self, [[:send, 'Hello'], [:recv, 'World'],
+                                       [:close]]
+      end
+    end
+  end
+  
+  def test_server_driven_tunnel
+    tunnel_test do
+      @tunnel_server.on_remote_listen do
+        EventMachine::start_server @local_host, @tunnel_port,
+            ScenarioConnection, self, [[:send, 'Hello'], [:recv, 'World'],
+                                       [:close]]
+
+        EventMachine::connect @local_host, @listen_port,
+            ScenarioConnection, self, [[:recv, 'Hello'], [:send, 'World'],
+                                       [:unbind], [:stop]]
+      end  
+    end    
+  end
+  
+  def test_two_tunnels
+    start_second = lambda do
+      @tunnel_client.stop
+      @tunnel_client.start
+      @tunnel_server.on_remote_listen do
+        EventMachine::connect @local_host, @listen_port,
+            ScenarioConnection, self, [[:send, 'Hello'], [:recv, 'World'],
+                                       [:unbind], [:stop, @stop_proc]]
+      end
+    end
+    
+    tunnel_test do
+      @tunnel_server.on_remote_listen do
+        EventMachine::start_server @local_host, @tunnel_port,
+            ScenarioConnection, self, [[:recv, 'Hello'], [:send, 'World'],
+                  [:close], [:recv, 'Hello'], [:send, 'World'], [:close]]
+
+        EventMachine::connect @local_host, @listen_port,
+            ScenarioConnection, self, [[:send, 'Hello'], [:recv, 'World'],
+                                       [:proc, start_second], [:unbind]]
+      end
+    end
+  end
+  
+  def test_secure_tunnel
+    @tunnel_server = new_server :authorized_keys => @hosts_file
+    @tunnel_client = new_client :private_key => @key_file
+    tunnel_test do
+      @tunnel_server.on_remote_listen do
+        EventMachine::start_server @local_host, @tunnel_port,
+            ScenarioConnection, self, [[:recv, 'Hello'], [:send, 'World'],
+                                       [:unbind], [:stop, @stop_proc]]
+
+        EventMachine::connect @local_host, @listen_port,
+            ScenarioConnection, self, [[:send, 'Hello'], [:recv, 'World'],
+                                       [:close]]
+      end
+    end
+  end
+
+  def test_secure_async_tunnel
+    @tunnel_server = new_server :authorized_keys => @hosts_file
+    @tunnel_client = new_client :private_key => @key_file
+    tunnel_test do
+      @tunnel_server.on_remote_listen do
+        EventMachine::start_server @local_host, @tunnel_port,
+            ScenarioConnection, self, [[:send, 'World'], [:recv, 'Hello'],
+                                       [:unbind], [:stop, @stop_proc]]
+
+        EventMachine::connect @local_host, @listen_port,
+            ScenarioConnection, self, [[:send, 'Hello'], [:recv, 'World'],
+                                       [:close]]
+      end
+    end
+  end
+  
+  def test_bad_listen_address
+    @tunnel_server = new_server
+    @tunnel_client = new_client :remote_listen_address =>
+                                "18.70.0.160:#{@listen_port}"
+                                
+    tunnel_test do
+      EventMachine::start_server @local_host, @tunnel_port,
+          ScenarioConnection, self, []
+                                     
+      EventMachine::connect @local_host, @listen_port,
+          ScenarioConnection, self, [[:unbind], [:stop, @stop_proc]]
+    end
+  end
+
+  # TODO: fix this
+  def test_secure_server_rejects_unsecure_client
+    @tunnel_server = new_server :authorized_keys => @hosts_file
+    tunnel_test(@secure_connection_time) do
+      EventMachine::start_server @local_host, @tunnel_port,
+          ScenarioConnection, self, []
+                                     
+      EventMachine::connect @local_host, @listen_port,
+          ScenarioConnection, self, [[:unbind], [:stop, @stop_proc]]
+    end
+  end
+
+  # hrmf... this is testing security and its not 100% reliable... but I can't think of a better way
+  def test_secure_server_rejects_unauthorized_key
+    @tunnel_server = new_server :authorized_keys => @hosts_file
+    @tunnel_client = new_client :private_key => 'test_data/random_rsa_key'
+    tunnel_test(@secure_connection_time) do
+      EventMachine::start_server @local_host, @tunnel_port,
+          ScenarioConnection, self, []
+
+      EventMachine::connect @local_host, @listen_port,
+          ScenarioConnection, self, [[:unbind], [:stop, @stop_proc]]
+    end
+  end
+end

data/test_data/authorized_keys2

@@ -0,0 +1,4 @@
+no-port-forwarding,no-agent-forwarding,command="/usr/NX/bin/nxnode" ssh-rsa AAAAB3NzaC1yc2EAAAABIwAAAQEAq2A7hRGmdnm9tUDbO9IDSwBK6TbQa+PXYPCPy6rbTrTtw7PHkccKrpp0yVhp5HdEIcKr6pLlVDBfOLX9QUsyCOV0wzfjIJNlGEYsdlLJizHhbn2mUjvSAHQqZETYP81eFzLQNnPHt4EVVUh7VfDESU84KezmD5QlWpXLmvU31/yMf+Se8xhHTvKSCZIFImWwoG6mbUoWf9nzpIoaSjB+weqqUUmpaaasXVal72J+UX2B+2RPW3RcT0eOzQgqlJL3RKrTJvdsjE3JEAvGq3lGHSZXy28G3skua2SmVi/w4yCE6gbODqnTWlg7+wC604ydGXA8VJiS5ap43JXiUFFAaQ== stub@github.org
+no-port-forwarding,no-agent-forwarding,command="/usr/NX/bin/nxnode" ssh-rsa AAAAB3NzaC1yc2EAAAABIwAAAQEAq7VciLo2wB+/2ThHXejC/kgkN205zlyS5cvN40Mqi3qvRVS75X1RawDoLot8eJ9KCYqZFr2Dr73d/EQNltN7dKJoPIj1IKxoraWkyNFbhhzpYuOltg9oO5UBSTNLupqla7zcdj4IwCCkBYk4+TS0dwmi20buOJ0FPY5PgbzmMnUiV9ipBqeJSdZB+TePH1gqlt7AP/6ti/0gxb2K7F69dZl/BSxMEzRCfBlTFC3f/4n8IdCuSJvNxxY+TtRnLL5CKUhj9QaIBan6JCkdRvVOBY7wmsNT8nGDzfDFSDD3KKn93g4LRkyMeaYlSDLxKy8PnNhjWgBNH1YNYyicsGfBKQ== costan@obsidian.local
+no-port-forwarding,no-agent-forwarding,command="/usr/NX/bin/nxnode" ssh-rsa AAAAB3NzaC1yc2EAAAABIwAAAQEA3tPhdPUFGAYyrT1quRSOevLbAdKAJ6Ovwqw0m99R0QkqUwMUh09pgedWZeij7HtAHtoWPrNFev8FrwcwnL14NgA/gwNnXxbqd4twC1HyFShUf7POry8bz3Qk+84STHeMY8++hhn8LgNyfuVQswHoW661aqieLM6pF8q8xIUtkXA7daNpJAL4nTN1TxgUoCDpCa0EbUPkGpwFPNtGPuokRXNOCR9g8T6LmPQbzGUTc4CzFfQ9rrHaimqkEmRWJbBOaik1bdQNqOh6MUDDuUSpkJV7fwu3bl4fF5/1kw2HCREEjJmESOYnZhOS+MCp1qAUcuXqpMqXD8ATNsuXqrIhTQ== stub@rubyforge.org
+no-port-forwarding,no-agent-forwarding,command="/usr/NX/bin/nxnode" ssh-dss AAAAB3NzaC1kc3MAAACBAPdWp4HhWUoNawosyEBvrhPSbjhCJiKnVWiUS6bo0BCGzTHhugrkv2HgtlKhWo8nqTw5E4YxzFVyZ0YQt4m7NYDLTZVjrqbIpL/3F5qNXco127O/im0cG27AKC8Jf7knmUTjd8EBhtK65tNDmxPtzKQtemlNTVPX1VccOn6eLtn1AAAAFQDtQC21TJrf/p5WNsU9UIJzO9/hIwAAAIEAgwTrIfleQMEAK9N3xeMVZpAGfSAoX6owLtk3z+iQ3rM9FRvM/CgezOgezLowJghkw/bcQDmMuudBUuijrM3zZWdr6eqoNbFTR/KKiUx3cYf0LAHNPbXfVz+P7BXqjcEj75qnwuHQMp7vNMg+dmV40UA2TiC5/8QlaZVwkOSPN6gAAACAPMGJEFkoR0ayfkd0S/tnY9ilO17T6rdoDuF25ATtNUd6Zji6tslxBkQFWtTeinO3rGkqJRPndq0wp3E33AOHhJE/FOIlWl4Tf6aeU95Y4enYujKQDiImSTXmdiw5wq/LFdc3a2waOUvuI+647wxgHhqTmD7xI2biGZLYN9Oasy0= costan@obsidian.local

data/test_data/known_hosts

@@ -0,0 +1,4 @@
+github.com,65.74.177.129 ssh-rsa AAAAB3NzaC1yc2EAAAABIwAAAQEAq2A7hRGmdnm9tUDbO9IDSwBK6TbQa+PXYPCPy6rbTrTtw7PHkccKrpp0yVhp5HdEIcKr6pLlVDBfOLX9QUsyCOV0wzfjIJNlGEYsdlLJizHhbn2mUjvSAHQqZETYP81eFzLQNnPHt4EVVUh7VfDESU84KezmD5QlWpXLmvU31/yMf+Se8xhHTvKSCZIFImWwoG6mbUoWf9nzpIoaSjB+weqqUUmpaaasXVal72J+UX2B+2RPW3RcT0eOzQgqlJL3RKrTJvdsjE3JEAvGq3lGHSZXy28G3skua2SmVi/w4yCE6gbODqnTWlg7+wC604ydGXA8VJiS5ap43JXiUFFAaQ==
+obsidian.local ssh-rsa AAAAB3NzaC1yc2EAAAABIwAAAQEAq7VciLo2wB+/2ThHXejC/kgkN205zlyS5cvN40Mqi3qvRVS75X1RawDoLot8eJ9KCYqZFr2Dr73d/EQNltN7dKJoPIj1IKxoraWkyNFbhhzpYuOltg9oO5UBSTNLupqla7zcdj4IwCCkBYk4+TS0dwmi20buOJ0FPY5PgbzmMnUiV9ipBqeJSdZB+TePH1gqlt7AP/6ti/0gxb2K7F69dZl/BSxMEzRCfBlTFC3f/4n8IdCuSJvNxxY+TtRnLL5CKUhj9QaIBan6JCkdRvVOBY7wmsNT8nGDzfDFSDD3KKn93g4LRkyMeaYlSDLxKy8PnNhjWgBNH1YNYyicsGfBKQ==
+rubyforge.org,205.234.109.19 ssh-rsa AAAAB3NzaC1yc2EAAAABIwAAAQEA3tPhdPUFGAYyrT1quRSOevLbAdKAJ6Ovwqw0m99R0QkqUwMUh09pgedWZeij7HtAHtoWPrNFev8FrwcwnL14NgA/gwNnXxbqd4twC1HyFShUf7POry8bz3Qk+84STHeMY8++hhn8LgNyfuVQswHoW661aqieLM6pF8q8xIUtkXA7daNpJAL4nTN1TxgUoCDpCa0EbUPkGpwFPNtGPuokRXNOCR9g8T6LmPQbzGUTc4CzFfQ9rrHaimqkEmRWJbBOaik1bdQNqOh6MUDDuUSpkJV7fwu3bl4fF5/1kw2HCREEjJmESOYnZhOS+MCp1qAUcuXqpMqXD8ATNsuXqrIhTQ==
+obsidian.local ssh-dss AAAAB3NzaC1kc3MAAACBAPdWp4HhWUoNawosyEBvrhPSbjhCJiKnVWiUS6bo0BCGzTHhugrkv2HgtlKhWo8nqTw5E4YxzFVyZ0YQt4m7NYDLTZVjrqbIpL/3F5qNXco127O/im0cG27AKC8Jf7knmUTjd8EBhtK65tNDmxPtzKQtemlNTVPX1VccOn6eLtn1AAAAFQDtQC21TJrf/p5WNsU9UIJzO9/hIwAAAIEAgwTrIfleQMEAK9N3xeMVZpAGfSAoX6owLtk3z+iQ3rM9FRvM/CgezOgezLowJghkw/bcQDmMuudBUuijrM3zZWdr6eqoNbFTR/KKiUx3cYf0LAHNPbXfVz+P7BXqjcEj75qnwuHQMp7vNMg+dmV40UA2TiC5/8QlaZVwkOSPN6gAAACAPMGJEFkoR0ayfkd0S/tnY9ilO17T6rdoDuF25ATtNUd6Zji6tslxBkQFWtTeinO3rGkqJRPndq0wp3E33AOHhJE/FOIlWl4Tf6aeU95Y4enYujKQDiImSTXmdiw5wq/LFdc3a2waOUvuI+647wxgHhqTmD7xI2biGZLYN9Oasy0=

data/test_data/random_rsa_key

@@ -0,0 +1,27 @@
+-----BEGIN RSA PRIVATE KEY-----
+MIIEowIBAAKCAQEAoSWzcqViB/JUz6Zd80c3km114/OWfxbGo/xeZqJm4s9+kR8o
+xtFjfbj9oDiuckSkAPNQW5VUWCeeNdpNjLmBe4Jm40R+FfibLOR5lw5I/7jVozBZ
+gQKuF395ODgbOwGXMsmRc1xViDYsOzo4A/f+5WhbpG20AtBz6+mg3yDLWXDffhat
+ecRf7PvWvlkj0QtL6538a7OE4MY75pCiTLWWbHJnBoQgNmYFZZdBaarHsenlKSQ9
+P6xL6ENyDOom3/Xk9bvqHQNNvgHyN4e4PRAtZUqsjyoRMm8sfDIJhK+cvpvqqWWm
+QK1ef1Fb1yXnmjkmUfx15UXxGgB79v3/vYWmJwIBIwKCAQBugECJIPLDntsSC64j
+KYUxNR2jn7enmpbW2PBGYLuUNoKsp6buu3dsJxRQm+VVq2klOSEo220JP7zxukPO
+Ng+lJjfsTDkzoyiTz95Y89pI88WxusEIAdZ2g0vOx2MhJamB6U3LcoPLHdUv7Wg9
+PFcDtUYnm66i2BT5ikJtHc1EpBTte71pZVRQsHYEsgcnNfsIc9hpXZacQ8CxF3Cc
+gM35GRSC7z/UVouyMbhQCjeKrlR/aZIE1whpnWpdOa+AeYtdndwxn6gNeplV8lA1
+8J0mpQlsy9i8cl2Nx/euN+dAW87zI3kwm1znj4pHF5e12v+SziZqnjuI4PhxNJO1
+CSRLAoGBAM9WfBRkVuWv+RdVm5VBpSwvBDytJpqcSbTxkypmV7sN6I4A2rBcNws9
+IaJ7Bf+ZxG4R8THXg5jCt1yfjIwMYrQ1Jg7aVALwF+JzeZgGTWFYJwCcC5VXCvy7
+gsJhnboBPVuM3Vxhz98nHKUngbLDbV8Sl+DVp1NDJ+tDJqY6tVZNAoGBAMb38M+G
+ZF9HVr33Oe7LjLxQRiQk5G6gmbidZfOszK797/BxYd1vMMPEl8d56zcYkeaIDmvL
+xHGLhoT6vtA4N0WCuk96USBk7JrRYhXoGTw0yFOJyopzPWdR66iaVbF/g1aknzki
+pZfb2LeyocztlxKJTUe0IocEkHu5aqzFeFBDAoGBAI4srNrIdhnXwL+LKNtgNr82
+LsqFXEzC8LaXBdPuaAs8vLklKD5rHm4bSkOHjxWrRN3DKQw8Andg8sMrk5M7sWWg
+yaPI6SaWAb/aJ34wNQ+M54tjsCvj6kbm+pPrKlOpFCGFKhN2RWXRrT4MdjFwEHu0
++m5JXMthP/HHpXlM1B3rAoGBAMFIoMmYfsL01+u89nLxnqhN+v6KPP3AlVRBITXb
+D/pzBldJkkTSaeK++de4Q5SbhmrqkaqLbl+sHEaqcDf8GG9pDMI8Ts8C9HkjLBVH
+3f/4wppLVwomze9W5OztslSnwWocQucLts4Iw+WmNscSsANgzrqu/PgwUddGZ6CF
+UE33AoGBAJ3VCQx0hwDSIQuuakjUb5ouPup+8B1A7wspGirkQNlTZ1F2q4Zug/Fp
+dovyzPZuV3/0dVDNbQzz21gRuiE2I/l9uF4Bz2LqY1Jg1R9WuQH5sdRG0ncf6Ikn
+1LRLgE2tCGGfhLUi2vDMwORx1o46p4hboF8ZaOBK1+WIwNnfRGNl
+-----END RSA PRIVATE KEY-----

data/test_data/ssh_host_dsa_key

@@ -0,0 +1,12 @@
+-----BEGIN DSA PRIVATE KEY-----
+MIIBvAIBAAKBgQD3VqeB4VlKDWsKLMhAb64T0m44QiYip1VolEum6NAQhs0x4boK
+5L9h4LZSoVqPJ6k8OROGMcxVcmdGELeJuzWAy02VY66myKS/9xeajV3KNduzv4pt
+HBtuwCgvCX+5J5lE43fBAYbSuubTQ5sT7cykLXppTU1T19VXHDp+ni7Z9QIVAO1A
+LbVMmt/+nlY2xT1QgnM73+EjAoGBAIME6yH5XkDBACvTd8XjFWaQBn0gKF+qMC7Z
+N8/okN6zPRUbzPwoHszoHsy6MCYIZMP23EA5jLrnQVLoo6zN82Vna+nqqDWxU0fy
+iolMd3GH9CwBzT2131c/j+wV6o3BI++ap8Lh0DKe7zTIPnZleNFANk4guf/EJWmV
+cJDkjzeoAoGAPMGJEFkoR0ayfkd0S/tnY9ilO17T6rdoDuF25ATtNUd6Zji6tslx
+BkQFWtTeinO3rGkqJRPndq0wp3E33AOHhJE/FOIlWl4Tf6aeU95Y4enYujKQDiIm
+STXmdiw5wq/LFdc3a2waOUvuI+647wxgHhqTmD7xI2biGZLYN9Oasy0CFQDL/9Z1
+fSLn88m1sUeWAZ4Ys2IIxw==
+-----END DSA PRIVATE KEY-----

data/test_data/ssh_host_rsa_key

@@ -0,0 +1,27 @@
+-----BEGIN RSA PRIVATE KEY-----
+MIIEoQIBAAKCAQEAq7VciLo2wB+/2ThHXejC/kgkN205zlyS5cvN40Mqi3qvRVS7
+5X1RawDoLot8eJ9KCYqZFr2Dr73d/EQNltN7dKJoPIj1IKxoraWkyNFbhhzpYuOl
+tg9oO5UBSTNLupqla7zcdj4IwCCkBYk4+TS0dwmi20buOJ0FPY5PgbzmMnUiV9ip
+BqeJSdZB+TePH1gqlt7AP/6ti/0gxb2K7F69dZl/BSxMEzRCfBlTFC3f/4n8IdCu
+SJvNxxY+TtRnLL5CKUhj9QaIBan6JCkdRvVOBY7wmsNT8nGDzfDFSDD3KKn93g4L
+RkyMeaYlSDLxKy8PnNhjWgBNH1YNYyicsGfBKQIBIwKCAQA/xwUc13Nr7okWKtit
+2hyKVU9HyXve7y8/aPSzf1jx+l5blIBN7LfXSXrPdaNCvtJbULyEygxXN+S8yNHY
+73b/b4XNV3D9gd281x/y0WsjL09fPpyitUP435LDatL8Ks+6TXZ1D7obedaFtqAi
+DEMHpH5Rch33xUsW3RY3gK1F8GHzYh+zMd0llH8w6IVBv0hrWttyp+4By1dhXA3L
+alYYOlGbu5cNuJfz50v4T/uB9F63JqHtAxEjqD7SvjCUbogGPxEH1k4Or+3xGuvR
+yTPaiuR9kehBstjbqc91rTLnGOr5a1VUIq6++0tCVOXQa+x5J5zCu/baGTdfkgRU
+loILAoGBAOHQcGIEgD1xHwFX9AafarF+1j4pPkM1p2u+9DNTIl1f2Rk6yOHlHZGF
+mfozMyh6l4HpvPTT7nflM4YR0JWJNbQIpCY8CdYKMXBVWcH4PXuqwyGKJBZMXWkn
+hKvelJ0UZUwx5+n8fTmSxLcIIlz5uKEE5k23/3cKkP7P6+Z6YuIZAoGBAMKpYBkM
+A4JMJ3QJ5TVsdFOoO0b1wThRgQTZR6ic9rNeXnZiL0mmr44DX2T1rTnEm6KZv2f2
+kWLQp5jx8BPFI+TUv16j4xrnnuHPhuBordCYQizSPcxXqL8b8a/KU+S5xofRjwB6
++8szd/JmGFDY8WgWrtIL32TdRhcIFg0Dg1mRAoGAE1sCUYtbcvsRSUINmirr41QD
+vC9rvJ4y6/pss/Eu1M2zhdHWtEbWphLEDiGlTJzLKGR96Rl61xOlVKJw9t/gCB3/
+cP3U9RbRCaDqb7YxJ9tvz6zBQ71nF6RNM02XtbFKgt+0yunBlzh3QuNwqOI0ZZK0
+p5Ntqx4pr3DoVZV2MKMCgYBprGdeDdYE54MhvDqZWCHkRWIB8x+/fLPAzbkvpap+
+oPFzdyD8GKhxqg82zoKbs991hqm8GCMJwboRMuFp0WtBtVHxjCrUF1ZAEZJckJje
+85GjTY9DCwPVdZHUdSY6VjiS33mD6v23c7YkgJDbbnRrtIrJy+5MsqJkRjfbLcr2
+GwKBgQCWDKVM8mZJcmM3KJwzBWh+b5T7tIbOBFaCvOBq4G0UUZGytWVUaddSbdh5
+YJ+1CoLDB+Wp3U95B1USQuSYwwr0wOYy42HBII7HnkaphT9HgDIhgvttFhX162oC
+bRizr31NgiGtgJbHJ9QPhaLvn3mZtZAFfRaEEdZslUzqRSACww==
+-----END RSA PRIVATE KEY-----

data/tests/_ab_test.rb

@@ -0,0 +1,16 @@
+base_dir = File.dirname(__FILE__)
+lo_server_file = File.join(base_dir, '../tests/lo_http_server.rb')
+lo_port = 4000
+Kernel.system "ruby #{lo_server_file} start #{lo_port}"
+
+fork{ exec "ruby #{base_dir}/../lib/rtunnel/rtunnel_server.rb > /dev/null 2>&1" }
+fork{ exec "ruby #{base_dir}/../lib/rtunnel/rtunnel_client.rb -c localhost -f 5000 -t 4000 > /dev/null" }
+
+sleep 2
+
+(puts "you need ab (apache bench) to run this stress test" ; exit) if %x{which ab}.empty?
+
+Process.wait fork { exec("ab -c 500 -n 10000 http://localhost:5000/") }
+
+puts "done, hit ^C"
+sleep 999999

data/tests/_stress_test.rb

@@ -0,0 +1,96 @@
+require 'rubygems'
+require 'eventmachine'
+require 'logger'
+require 'stringio'
+
+CONCURRENT_CONNECTIONS = 50
+DISPLAY_RTUNNEL_OUTPUT = true
+TUNNEL_PORT = 5000
+HTTP_PORT = 4444
+
+#################
+
+TUNNEL_URI = "http://localhost:#{TUNNEL_PORT}"
+srand(0)  # deterministic random data
+EXPECTED_DATA = Array.new(16*1024) { rand(?z).chr }*''
+
+$pids = [$$]
+
+def cleanup
+  puts $!, $@  if $!
+
+  # move the current process to the end of the kill list
+  $pids << $pids.delete($$)
+  $pids.each {|pid| Process.kill 9, pid  rescue nil }
+
+  exit!
+end
+
+at_exit { cleanup }
+
+ENV['RTUNNEL_DEBUG'] = '1'  if DISPLAY_RTUNNEL_OUTPUT
+
+base_dir = File.join(File.dirname(__FILE__), '..', 'bin')
+$pids << fork{ exec "ruby", "-Ilib", "#{base_dir}/rtunnel_server" }
+$pids << fork{ exec "ruby", "-Ilib", "#{base_dir}/rtunnel_client", '-c', '127.0.0.1', '-f', TUNNEL_PORT.to_s, '-t', "127.0.0.1:#{HTTP_PORT}" }
+
+$pids << fork do
+  require 'thin'
+
+  app = lambda do |env|
+    body = env['rack.input'].string 
+    if body != EXPECTED_DATA
+      p body, EXPECTED_DATA
+      puts "server received BAD DATA!"
+
+      cleanup
+    end
+
+    [200, {}, EXPECTED_DATA]
+  end
+
+  Thin::Server.new('localhost', HTTP_PORT, app).start
+end
+
+puts 'wait 2 secs...'
+sleep 2
+
+module Stresser
+  @@open_connections = 0
+
+  def post_init
+    @@open_connections += 1
+    send_data "POST / HTTP/1.0\r\nContent-Length: #{EXPECTED_DATA.size}\r\n\r\n#{EXPECTED_DATA}"
+    @data = ''
+    print '('; $stdout.flush
+  end
+
+  def receive_data(data)
+    @data << data
+    print '.'; $stdout.flush
+  end
+
+  def unbind
+    if @data.gsub(/\A.+\r\n\r\n/m,'') != EXPECTED_DATA
+      p EXPECTED_DATA, @data
+      puts "client received BAD DATA!"
+
+      cleanup
+    end
+
+    print ')'; $stdout.flush
+
+    @@open_connections -= 1
+    EventMachine.stop_event_loop  if @@open_connections == 0
+  end
+end
+
+loop do
+  EventMachine.run do
+    CONCURRENT_CONNECTIONS.times do
+      EventMachine.connect 'localhost', TUNNEL_PORT, Stresser
+    end
+  end
+
+  puts
+end

data/tests/lo_http_server.rb

@@ -0,0 +1,55 @@
+require 'net/http'
+require 'socket'
+require 'time'
+
+require 'rubygems'
+require 'simple-daemon'
+  
+class LoHttpServer < SimpleDaemon::Base
+  def send_headers(socket, status, length = 0, mime_type = 'text/html', headers = {})
+    reason = headers[:reason] || {200 => 'OK', 404 => 'File not found'}[status]
+    
+    response_line = "HTTP/1.1 #{status} #{reason}\r\n"    
+
+    response_headers = { 'Connection' => 'close', 'Server' => 'Rubylicious/1.0', 'Date' => Time.now.rfc2822,
+      'Content-Type' => mime_type, 'Content-Length' => length }.merge headers
+
+    socket.write response_line
+    socket.write response_headers.to_a.map { |k,v| k.to_s + ': ' + v.to_s }.join("\r\n") + "\r\n\r\n"
+  end
+  
+  def send_html(socket, html_data = "Loopback page\n")
+    send_headers(socket, 200, html_data.length)
+    socket.write html_data
+    socket.close
+  end
+
+  def run(listen_port)
+    listen_socket = TCPServer.new listen_port
+    
+    loop do
+      client_socket = listen_socket.accept
+      Thread.new do
+        begin
+          print client_socket.readpartial(16384), "\n"          
+          send_html client_socket
+        rescue
+          print "#{$!.class.name}: #{$!}\n"
+          print $!.backtrace.join("\n"), "\n"
+        end
+      end
+    end    
+  end
+  
+  SimpleDaemon::WORKING_DIRECTORY = File.dirname(__FILE__)
+
+  def self.start
+    port = ARGV[1] || 3000
+    self.new.run(port)    
+  end
+
+  def self.stop
+  end
+end
+
+LoHttpServer.daemonize