redcap 0.1.0

data/.gitignore

@@ -0,0 +1,4 @@
+*.gem
+.bundle
+Gemfile.lock
+pkg/*

data/Gemfile

@@ -0,0 +1,3 @@
+source :rubygems
+
+gemspec

data/LICENCE

@@ -0,0 +1,20 @@
+Copyright (c) 2011 Pat Allan
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/Rakefile

@@ -0,0 +1,9 @@
+require 'rubygems'
+require 'bundler'
+
+Bundler.require :default, :development
+
+Bundler::GemHelper.install_tasks
+
+require 'rspec/core/rake_task'
+RSpec::Core::RakeTask.new

data/bin/redcap

@@ -0,0 +1,5 @@
+#!/usr/bin/env ruby
+require 'redcap'
+
+Redcap::App.daemonise!
+Redcap::App.run

data/lib/redcap/app.rb

@@ -0,0 +1,95 @@
+class Redcap::App
+  include Singleton
+  
+  attr_reader :server
+  
+  def self.run
+    instance.run
+  end
+  
+  def self.daemonise!
+    # Daemon won't need these
+    $stdin.reopen  '/dev/null'
+    $stdout.reopen '/dev/null', 'w'
+    $stderr.reopen '/dev/null', 'w'
+    
+    # Send to the background
+    pid = fork
+    if pid
+      # This is in the context of the parent - save the pid file, then exit.
+      File.open(instance.pid_file, 'w') { |file| file.puts pid }
+      exit!
+    end
+  end
+  
+  def run
+    update_procline
+    
+    EventMachine.epoll
+    EventMachine.run do
+      @server = EventMachine.start_server('127.0.0.1', 11000, Redcap::Server) do |connection|
+        connection.logger = logger
+      end
+      
+      setup_signal_handlers
+    end
+  end
+  
+  def stop!(signal)
+    logger.info "Received #{signal} signal, stopping server."
+    stop_listening
+    forget_server
+    stop
+  end
+  
+  def die(message)
+    $stderr.puts message
+    logger.fatal message
+    
+    exit 1
+  end
+  
+  private
+  
+  def update_procline
+    $0 = 'redcap'
+  end
+  
+  def log_file
+    @log_file ||= '/var/log/redcap.log'
+  end
+  
+  def pid_file
+    @log_file ||= '/var/tmp/redcap.pid'
+  end
+  
+  def logger
+    @logger ||= begin
+      logger                 = Logger.new(log_file)
+      logger.formatter       = Logger::Formatter.new
+      logger.datetime_format = '%Y-%m-%d %H:%M:%S'
+      logger.progname        = 'redcap'
+      logger.level           = Logger::INFO
+      
+      logger
+    end
+  end
+  
+  def setup_signal_handlers
+    %w{ QUIT TERM INT }.each do |signal|
+      trap(signal) { stop!(signal) and exit! }
+    end
+  end
+  
+  def stop_listening
+    EventMachine.stop_server(server) unless server.nil?
+  end
+  
+  def forget_server
+    @server = nil
+  end
+  
+  def stop
+    EventMachine.stop
+  end
+end

data/lib/redcap/server.rb

@@ -0,0 +1,19 @@
+class Redcap::Server < EventMachine::Protocols::LineAndTextProtocol
+  attr_accessor :logger
+  
+  def receive_line(line)
+    port = line.to_i
+    pid  = pid_for_port(port)
+    
+    logger.info "port: #{port} -> pid: #{pid}"
+    
+    send_data pid
+    close_connection_after_writing
+  end
+  
+  private
+  
+  def pid_for_port(port)
+    `lsof -i :#{port} | grep ssh | awk '{print $2}'`[/\d+/] || 'Unknown'
+  end
+end

data/lib/redcap/version.rb

@@ -0,0 +1 @@
+Redcap::Version = '0.1.0'

data/lib/redcap.rb

@@ -0,0 +1,22 @@
+require 'socket'
+require 'singleton'
+require 'logger'
+require 'eventmachine'
+
+module Redcap
+  def self.pid_for_port(port, options = {})
+    options[:server] ||= '127.0.0.1'
+    options[:port]   ||= 11000
+    
+    socket = TCPSocket.new options[:server], options[:port]
+    socket.write "#{port}\n"
+    pid    = socket.read.to_i
+    socket.close
+    
+    pid
+  end
+end
+
+require 'redcap/app'
+require 'redcap/server'
+require 'redcap/version'

data/redcap.gemspec

@@ -0,0 +1,26 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path('../lib', __FILE__)
+require 'redcap'
+
+Gem::Specification.new do |s|
+  s.name        = 'redcap'
+  s.version     = Redcap::Version
+  s.platform    = Gem::Platform::RUBY
+  s.authors     = ['Pat Allan']
+  s.email       = ['pat@freelancing-gods.com']
+  s.homepage    = 'http://github.com/flying-sphinx/redcap'
+  s.summary     = %q{Translates SSH forward ports to process IDs}
+  s.description = %q{A service that translates SSH forward ports to process IDs. Built for Flying Sphinx.}
+
+  s.rubyforge_project = 'redcap'
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map { |f|
+    File.basename(f)
+  }
+  s.require_paths = ['lib']
+  
+  s.add_runtime_dependency     'eventmachine', ['>= 0.12.10']
+  s.add_development_dependency 'rspec',        ['2.5.0']
+end

data/spec/redcap/server_spec.rb

@@ -0,0 +1,35 @@
+require 'spec_helper'
+
+describe Redcap::Server do
+  describe '#receive_line' do
+    let(:server) { Redcap::Server.new :spec }
+    let(:logger) { stub('logger', :info => nil) }
+    
+    before :each do
+      server.logger = logger
+      server.stub!(
+        :`                              => '24',
+        :close_connection_after_writing => nil,
+        :send_data                      => nil
+      )
+    end
+    
+    it "sends the process id for a given port" do
+      server.should_receive(:send_data).with('24')
+      
+      server.receive_line('101')
+    end
+    
+    it "closes the connection once writing is finished" do
+      server.should_receive(:close_connection_after_writing)
+      
+      server.receive_line('101')
+    end
+    
+    it "logs the request" do
+      logger.should_receive(:info).with('port: 101 -> pid: 24')
+      
+      server.receive_line('101')
+    end
+  end
+end

data/spec/redcap_spec.rb

@@ -0,0 +1,21 @@
+require 'spec_helper'
+
+describe Redcap do
+  describe '.pid_for_port' do
+    let(:socket) { stub('socket', :write => nil, :read => '42', :close => nil) }
+    
+    before :each do
+      TCPSocket.stub! :new => socket
+    end
+    
+    it "sends the given port to the socket" do
+      socket.should_receive(:write).with("101\n")
+      
+      Redcap.pid_for_port(101)
+    end
+    
+    it "returns the socket output as an integer" do
+      Redcap.pid_for_port(101).should == 42
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,13 @@
+require 'rubygems'
+require 'bundler'
+
+$:.unshift File.dirname(__FILE__) + '/../lib'
+$:.unshift File.dirname(__FILE__) + '/..'
+
+Bundler.require :default, :development
+
+require 'redcap'
+
+RSpec.configure do |config|
+  
+end

metadata

@@ -0,0 +1,99 @@
+--- !ruby/object:Gem::Specification 
+name: redcap
+version: !ruby/object:Gem::Version 
+  prerelease: 
+  version: 0.1.0
+platform: ruby
+authors: 
+- Pat Allan
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-05-09 00:00:00 +10:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: eventmachine
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 0.12.10
+  type: :runtime
+  prerelease: false
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: rspec
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - "="
+      - !ruby/object:Gem::Version 
+        version: 2.5.0
+  type: :development
+  prerelease: false
+  version_requirements: *id002
+description: A service that translates SSH forward ports to process IDs. Built for Flying Sphinx.
+email: 
+- pat@freelancing-gods.com
+executables: 
+- redcap
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- .gitignore
+- Gemfile
+- LICENCE
+- README.textile
+- Rakefile
+- bin/redcap
+- lib/redcap.rb
+- lib/redcap/app.rb
+- lib/redcap/server.rb
+- lib/redcap/version.rb
+- redcap.gemspec
+- spec/redcap/server_spec.rb
+- spec/redcap_spec.rb
+- spec/spec_helper.rb
+has_rdoc: true
+homepage: http://github.com/flying-sphinx/redcap
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 2640545869079275918
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 2640545869079275918
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: redcap
+rubygems_version: 1.6.2
+signing_key: 
+specification_version: 3
+summary: Translates SSH forward ports to process IDs
+test_files: 
+- spec/redcap/server_spec.rb
+- spec/redcap_spec.rb
+- spec/spec_helper.rb