reverse-tunnel 0.0.1

data/lib/reverse-tunnel/version.rb

@@ -0,0 +1,3 @@
+module ReverseTunnel
+  VERSION = "0.0.1"
+end

data/reverse-tunnel.gemspec

@@ -0,0 +1,33 @@
+# -*- encoding: utf-8 -*-
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'reverse-tunnel/version'
+
+Gem::Specification.new do |gem|
+  gem.name          = "reverse-tunnel"
+  gem.version       = ReverseTunnel::VERSION
+  gem.authors       = ["Alban Peignier", "Florent Peyraud"]
+  gem.email         = ["alban@tryphon.eu", "florent@tryphon.eu"]
+  gem.description   = %q{Create easily a tunnel to forward connection (like a ssh) to the client host}
+  gem.summary       = %q{Forward a tcp connection to client host}
+  gem.homepage      = "http://projects.tryphon.eu/projects/reverse-tunnel"
+
+  gem.files         = `git ls-files`.split($/)
+  gem.executables   = gem.files.grep(%r{^bin/}).map{ |f| File.basename(f) }
+  gem.test_files    = gem.files.grep(%r{^(test|spec|features)/})
+  gem.require_paths = ["lib"]
+
+  gem.add_runtime_dependency "eventmachine"
+  gem.add_runtime_dependency "msgpack"
+  gem.add_runtime_dependency "eventmachine_httpserver"
+  gem.add_runtime_dependency "json"
+  gem.add_runtime_dependency "trollop"
+  gem.add_runtime_dependency "SyslogLogger"
+
+  gem.add_development_dependency "simplecov"
+  gem.add_development_dependency "rspec"
+  gem.add_development_dependency "guard"
+  gem.add_development_dependency "guard-rspec"
+  gem.add_development_dependency "rake"
+  gem.add_development_dependency "rest-client"
+end

data/spec/reverse-tunnel/cli_spec.rb

@@ -0,0 +1,118 @@
+require 'spec_helper'
+
+describe ReverseTunnel::CLI do
+
+  describe "#debug=" do
+
+    context "when true" do
+      it "should change Logger level to debug" do
+        ReverseTunnel.logger.level = Logger::INFO
+        lambda {
+          subject.debug = true
+        }.should change(ReverseTunnel.logger, :level).to(Logger::DEBUG)
+      end
+    end
+
+    context "when false" do
+      it "should change Logger level to info" do
+        ReverseTunnel.logger.level = Logger::DEBUG
+        lambda {
+          subject.debug = false
+        }.should change(ReverseTunnel.logger, :level).to(Logger::INFO)
+      end
+    end
+
+    after do
+      ReverseTunnel.reset_logger!
+    end
+
+  end
+
+  describe "#syslog=" do
+
+    context "when true" do
+      
+      it "should change ReverseTunnel.logger to Syslog::Logger" do
+        subject.syslog = true
+        ReverseTunnel.logger.should be_instance_of(Syslog::Logger)
+      end
+
+      after do
+        ReverseTunnel.reset_logger!
+      end
+                          
+    end
+    
+  end
+
+  
+
+
+
+end
+
+
+describe ReverseTunnel::CLI::Configurator do
+
+  describe "#server=" do
+    
+    it "should parse ip and port" do
+      subject.should_receive(:server_host=).with("host")
+      subject.should_receive(:server_port=).with(4893)
+      
+      subject.server = "host:4893"
+    end
+
+  end
+  
+end
+
+describe ReverseTunnel::CLI::Client do
+
+  describe "#api=" do
+    
+    it "should parse ip and port" do
+      subject.should_receive(:api_host=).with("host")
+      subject.should_receive(:api_port=).with(4895)
+      
+      subject.api = "host:4895"
+    end
+
+  end
+
+  describe "#token" do
+
+    it "should use first argument" do
+      subject.stub :arguments => [ "token"]
+      subject.token.should == "token"
+    end
+    
+  end
+
+end
+
+describe ReverseTunnel::CLI::Server do
+
+  describe "#api=" do
+    
+    it "should parse ip and port" do
+      subject.should_receive(:api_host=).with("host")
+      subject.should_receive(:api_port=).with(4894)
+      
+      subject.api = "host:4894"
+    end
+
+  end
+
+  describe "#local=" do
+    
+    it "should parse ip and port range" do
+      subject.should_receive(:local_host=).with("host")
+      subject.should_receive(:local_port_range=).with(123..456)
+      
+      subject.local = "host:123-456"
+    end
+
+  end
+
+end

data/spec/reverse-tunnel/client/api_server_spec.rb

@@ -0,0 +1,72 @@
+require 'spec_helper'
+
+require "rest_client"
+require "json"
+
+describe ReverseTunnel::Client::ApiServer do
+
+  let(:tunnel) { mock :to_json => '{"token":"1008B1A077343ED8B7AAAC9BC580","local_port":22,"server_host":"localhost","server_port":4893,"local_connections":[],"connection":{"created_at":"2012-12-27 12:13:50 +0100"}}' }
+
+  let(:api_host) { "127.0.0.1" }
+  let(:api_port) { 38589 }
+
+  def url(path)
+    "http://#{api_host}:#{api_port}#{path}"
+  end
+
+  def get(path)
+    JSON.parse RestClient.get url(path)
+  end
+
+  def wait_api
+    5.times do
+      begin
+        s = TCPSocket.new(api_host, api_port)
+        s.close
+        return true
+      rescue Errno::ECONNREFUSED
+        sleep 0.1
+      end
+    end
+  end
+
+  def wait_api_stopped
+    5.times do
+      begin
+        s = TCPSocket.new(api_host, api_port)
+        s.close
+        sleep 0.1
+      rescue Errno::ECONNREFUSED
+        return true
+      end
+    end
+  end
+
+  before do
+    Thread.new do
+      EventMachine.run do
+        EventMachine.start_server api_host, api_port, ReverseTunnel::Client::ApiServer, tunnel
+      end
+    end
+
+    wait_api
+  end
+
+  describe "GET '/status'" do
+    
+    it "should return tunnel description" do
+      get("/status").should == JSON.parse(tunnel.to_json)
+    end
+
+    it "should accept requests on /tunnels.json" do
+      get("/status.json").should == get("/status")
+    end
+
+  end
+
+  after do
+    EventMachine.stop
+    wait_api_stopped
+  end
+  
+end

data/spec/reverse-tunnel/client_spec.rb

@@ -0,0 +1,79 @@
+require 'spec_helper'
+
+describe ReverseTunnel::Client do
+  
+  describe "#server_host" do
+    
+    it "should be nil by default" do
+      subject.server_host.should be_nil
+    end
+
+  end
+
+  describe "#server_port" do
+    
+    it "should be 4893 by default" do
+      subject.server_port.should == 4893
+    end
+
+  end
+
+  describe "#local_port" do
+    
+    it "should be 22 by default" do
+      subject.local_port.should == 22
+    end
+
+  end
+
+  describe "#api_host" do
+    
+    it "should be nil by default" do
+      subject.api_host.should be_nil
+    end
+
+  end
+
+  describe "#api_port" do
+    
+    it "should be 4895 by default" do
+      subject.api_port.should == 4895
+    end
+
+  end
+
+  describe "#tunnel" do
+    it "should use server_host" do
+      subject.server_host = "localhost"
+      subject.tunnel.host.should == "localhost"
+    end
+
+    it "should use server_port" do
+      subject.server_port = 123
+      subject.tunnel.port.should == 123
+    end
+
+    it "should use local_port" do
+      subject.local_port = 123
+      subject.tunnel.local_port.should == 123
+    end
+  end
+
+  describe "#start_api" do
+    it "should use api_host and api_port" do
+      subject.api_host = "localhost"
+      subject.api_port = 123
+
+      EventMachine.should_receive(:start_server).with("localhost", 123, anything, anything)
+      subject.start_api
+    end
+
+    it "should not start api if api_host is not defined" do
+      subject.api_host = nil
+
+      EventMachine.should_not_receive(:start_server)
+      subject.start_api
+    end
+  end
+
+end

data/spec/reverse-tunnel/message_spec.rb

@@ -0,0 +1,55 @@
+require 'spec_helper'
+
+describe Message do
+  
+end
+
+describe Message::Data do
+  subject { Message::Data.new(123, "dummy") }
+
+  its(:type) { should == :data }
+
+  it { should be_data }
+
+  it "should have the same session_id after pack/unpack" do
+    Message.unpack(subject.pack).session_id.should == subject.session_id
+  end
+
+  it "should have the same data after pack/unpack" do
+    Message.unpack(subject.pack).data.should == subject.data
+  end
+end
+
+describe Message::OpenSession do
+  its(:type) { should == :open_session }
+
+  it { should be_open_session }
+
+  it "should have the same session_id after pack/unpack" do
+    subject.session_id = 123
+    Message.unpack(subject.pack).session_id.should == subject.session_id
+  end
+end
+
+describe Message::OpenTunnel do
+  its(:type) { should == :open_tunnel }
+
+  it { should be_open_tunnel }
+
+  it "should have the same token after pack/unpack" do
+    subject.token = 123
+    Message.unpack(subject.pack).token.should == subject.token
+  end
+end
+
+describe Message::Ping do
+  its(:type) { should == :ping }
+
+  it { should be_ping }
+
+  it "should have the same sequence_number after pack/unpack" do
+    subject.sequence_number = 123
+    Message.unpack(subject.pack).sequence_number.should == subject.sequence_number
+  end
+end
+

data/spec/reverse-tunnel/server/api_server_spec.rb

@@ -0,0 +1,125 @@
+require 'spec_helper'
+
+require "rest_client"
+require "json"
+
+describe ReverseTunnel::Server::ApiServer do
+
+  let(:server) { ReverseTunnel::Server.new }
+
+  let(:api_host) { "127.0.0.1" }
+  let(:api_port) { 38589 }
+
+  def url(path)
+    "http://#{api_host}:#{api_port}#{path}"
+  end
+
+  def get(path)
+    JSON.parse RestClient.get url(path)
+  end
+
+  def post(path, params)
+    JSON.parse RestClient.post url(path), params.to_json
+  end
+
+  def delete(path)
+    JSON.parse RestClient.delete url(path)
+  end
+
+  def wait_api
+    5.times do
+      begin
+        s = TCPSocket.new(api_host, api_port)
+        s.close
+        return true
+      rescue Errno::ECONNREFUSED
+        sleep 0.1
+      end
+    end
+  end
+
+  def wait_api_stopped
+    5.times do
+      begin
+        s = TCPSocket.new(api_host, api_port)
+        s.close
+        sleep 0.1
+      rescue Errno::ECONNREFUSED
+        return true
+      end
+    end
+  end
+
+  before do
+    Thread.new do
+      EventMachine.run do
+        EventMachine.start_server api_host, api_port, ReverseTunnel::Server::ApiServer, server
+      end
+    end
+
+    wait_api
+  end
+
+  describe "GET '/tunnels'" do
+    
+    it "should return an empty array without tunnels" do
+      get("/tunnels").should be_empty
+    end
+
+    it "should return tunnels description" do
+      tunnel = server.tunnels.create
+      get("/tunnels").should include("token" => tunnel.token, "local_port" => tunnel.local_port)
+    end
+
+    it "should accept requests on /tunnels.json" do
+      get("/tunnels.json").should == get("/tunnels")
+    end
+
+  end
+
+  describe "POST '/tunnels'" do
+
+    it "should create a new tunnel" do
+      post("/tunnels", :token => "dummy")
+      server.tunnels.find("dummy").should_not be_nil
+    end
+
+    it "should return new tunnel description" do
+      post("/tunnels", :token => "dummy")["token"].should == "dummy"
+    end
+    
+  end
+
+  describe "GET '/tunnels/:id'" do
+    
+    it "should return tunnel description" do
+      tunnel = server.tunnels.create
+      get("/tunnels/#{tunnel.token}")["local_port"].should == tunnel.local_port
+    end
+
+    it "should return a 404 when tunnel doesn't exist" do
+      lambda { get("/tunnels/123") }.should raise_error(RestClient::ResourceNotFound)
+    end
+
+  end
+
+  describe "DELETE '/tunnels/:id'" do
+    
+    it "should delete the specified tunnel" do
+      tunnel = server.tunnels.create
+      delete("/tunnels/#{tunnel.token}")
+      server.tunnels.find(tunnel.token).should be_nil
+    end
+
+    it "should return a 404 when tunnel doesn't exist" do
+      lambda { delete("/tunnels/123") }.should raise_error(RestClient::ResourceNotFound)
+    end
+
+  end
+
+  after do
+    EventMachine.stop
+    wait_api_stopped
+  end
+  
+end