rack-tunnel 0.1.0

data/lib/rack/tunnel.rb

@@ -0,0 +1,63 @@
+require 'socket'
+require 'uri'
+require 'open4'
+
+module Rack
+  class Tunnel
+    VERSION = '0.1.0'
+    HEADER  = 'X-Tunnel-Uri'
+
+    def initialize(app, uri)
+      @app, @uri = app, URI.parse(uri)
+
+      if @uri.port <= 1024 && @uri.user != 'root'
+        raise Error, "root user required for port #{@uri.port}"
+      end
+    end
+
+    def call(env)
+      establish_tunnel(env['SERVER_PORT']) unless @tunnel_established
+      @app.call(env.merge(HEADER => @uri.to_s))
+    end
+
+    def establish_tunnel(local_port)
+      pid, _ = Open4.popen4(command(local_port))
+
+      at_exit do
+        Process.kill(9, pid)
+        Process.wait
+      end
+
+			wait_for_tunnel
+      @tunnel_established = true
+    end
+
+    def command(local_port)
+      cmd = %w"ssh"
+      cmd << "-R 0.0.0.0:#{@uri.port}:localhost:#{local_port}"
+      cmd << @uri.host
+      cmd << "-l #{@uri.user}"
+      cmd << "-o TCPKeepAlive=yes"
+      cmd << "-o ServerAliveInterval=30"
+      cmd << "-o ServerAliveCountMax=10"
+      cmd << "-o GatewayPorts=yes"
+      cmd << "-N"
+
+      cmd.join(' ')
+    end
+
+		def wait_for_tunnel
+			Timeout::timeout(30) do
+				begin
+					return TCPSocket.new(@uri.host, @uri.port)
+				rescue Errno::ECONNREFUSED
+					retry
+				rescue => e
+					raise Error, e.message
+				end
+			end
+		end
+
+    class Error < StandardError ; end
+  end
+end

data/spec/rack_tunnel_spec.rb

@@ -0,0 +1,24 @@
+require File.join(File.dirname(__FILE__), 'spec_helper')
+
+describe Rack::Tunnel do
+  it 'adds the "X-Tunnel-Uri" header' do
+    env = JSON.parse(@client.get('/').body)
+    env['X-Tunnel-Uri'].should =~ %r{^http://}
+  end
+
+  it 'tunnels over ssh' do
+    env = JSON.parse(@client.get('/').body)
+    uri = env['X-Tunnel-Uri']
+
+    Rack::Client.new(uri).get('/').status.should == 200
+  end
+
+  it 'raise an exception if the port is restricted and the user is not root' do
+    lambda {
+      Rack::Client.new do
+        use Rack::Tunnel, "http://#{ENV['USER']}@localhost"
+        run lambda { [200, {}, []] }
+      end
+    }.should raise_error(Rack::Tunnel::Error)
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,16 @@
+$:.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+
+require 'rack/tunnel'
+require 'realweb'
+require 'rack/client'
+require 'json'
+
+RSpec.configure do |config|
+  config.color_enabled = true
+
+  config.before(:all) do
+    @server = RealWeb.start_server_in_thread(File.join(File.dirname(__FILE__), 'config.ru'))
+    @client = Rack::Client.new("http://127.0.0.1:#{@server.port}")
+  end
+end
+

metadata

@@ -0,0 +1,172 @@
+--- !ruby/object:Gem::Specification 
+name: rack-tunnel
+version: !ruby/object:Gem::Version 
+  hash: 27
+  prerelease: false
+  segments: 
+  - 0
+  - 1
+  - 0
+  version: 0.1.0
+platform: ruby
+authors: 
+- Ben Burkert
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-10-17 00:00:00 -07:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: rack
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: open4
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id002
+- !ruby/object:Gem::Dependency 
+  name: rack-client
+  prerelease: false
+  requirement: &id003 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 1923832053
+        segments: 
+        - 0
+        - 3
+        - 1
+        - pre
+        - b
+        version: 0.3.1.pre.b
+  type: :development
+  version_requirements: *id003
+- !ruby/object:Gem::Dependency 
+  name: json
+  prerelease: false
+  requirement: &id004 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id004
+- !ruby/object:Gem::Dependency 
+  name: rspec
+  prerelease: false
+  requirement: &id005 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id005
+- !ruby/object:Gem::Dependency 
+  name: realweb
+  prerelease: false
+  requirement: &id006 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id006
+- !ruby/object:Gem::Dependency 
+  name: rake
+  prerelease: false
+  requirement: &id007 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id007
+description: Automatic port forwading via SSH tunneling.
+email: 
+- ben@benburkert.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- ./lib/rack/tunnel.rb
+- ./spec/rack_tunnel_spec.rb
+- ./spec/spec_helper.rb
+has_rdoc: true
+homepage: http://github.com/benburkert/rack-tunnel
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.7
+signing_key: 
+specification_version: 3
+summary: Automatic port forwading via SSH tunneling.
+test_files: 
+- ./spec/rack_tunnel_spec.rb
+- ./spec/spec_helper.rb