em-udp-proxy 0.1.0

data/.gitignore

@@ -0,0 +1 @@
+pkg/

data/Gemfile

@@ -0,0 +1,3 @@
+source 'http://rubygems.org'
+
+gem 'eventmachine'

data/README

@@ -0,0 +1,12 @@
+EventMachine based UDP proxy.
+
+Example usage:
+
+> em-udp-proxy -a 127.0.0.1 -p 5001 -A 192.168.1.1 -P 5001
+
+Starts the proxy binded to 127.0.0.1 with port 5001 and forwards traffic to/from 192.168.1.1 on port 5001.
+
+TODO:
+1. More documentation :-)
+2. Remove Client objects for inactive clients (after certain threshold).
+3. Add some way to get stats from it. 

data/Rakefile

@@ -0,0 +1,2 @@
+require 'bundler'
+Bundler::GemHelper.install_tasks

data/bin/em-udp-proxy

@@ -0,0 +1,23 @@
+#!/usr/bin/env ruby
+require 'rubygems'
+require File.expand_path('../lib/udp_proxy.rb', File.dirname(__FILE__))
+require 'optparse'
+
+options = {}
+
+OptionParser.new do |opts|
+  opts.banner = "Usage: em-udp-proxy [options]"
+
+  opts.on('-a', '--address ADDRESS', String, "Address to use") { |value| options[:address] = value }
+	opts.on('-p', '--port PORT', Numeric, "Listening port") { |value| options[:port] = value }
+  opts.on('-A','--remote-address ADDRESS', String, "Remote address to connect to") { |value| options[:remote_address] = value}
+  opts.on('-P','--remote-port PORT', Numeric, "Remote port to connect to") { |value| options[:remote_port] = value}
+
+  opts.on_tail("-h", "--help", "Show this message") do
+    puts opts 
+    exit
+  end
+end.parse!
+
+Kernel.abort "Missing arguments, see --help for details." if options.length < 4
+UDPProxy.start options[:address], options[:port], options[:remote_address], options[:remote_port]

data/em-udp-proxy.gemspec

@@ -0,0 +1,18 @@
+$:.push File.expand_path("../lib", __FILE__)
+
+Gem::Specification.new do |s|
+  s.name        = "em-udp-proxy"
+  s.version     = "0.1.0"
+  s.platform    = Gem::Platform::RUBY
+  s.authors     = ["Arik Fraimovich"]
+  s.email       = ["arik@arikfr.com"]
+  s.homepage    = "http://github.com/arikfr/em-udp-proxy"
+  s.summary     = %q{EventMachine UDP Proxy}
+  s.description = s.summary
+
+  s.add_dependency "eventmachine"
+
+  s.files         = `git ls-files`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+end

data/lib/udp_proxy.rb

@@ -0,0 +1,30 @@
+require 'eventmachine'
+require 'syslog'
+require 'socket'
+require File.expand_path('./udp_proxy/server.rb', File.dirname(__FILE__))
+require File.expand_path('./udp_proxy/client.rb', File.dirname(__FILE__))
+
+module UDPProxy
+  def self.start(ip, port, relay_ip, relay_port)
+  	start_logger
+
+    EM::run do
+      Syslog.notice "Starting proxy on: #{ip}:#{port}"
+      EM::open_datagram_socket ip, port, UDPProxy::Server, ip, relay_ip, relay_port
+      trap("TERM") { stop }
+      trap("INT") { stop }
+    end
+  end
+
+  def self.stop
+    Syslog.warning "Stopping proxy..."
+    EventMachine.stop
+  end
+
+  private
+  def self.start_logger
+  	Syslog.open('UDP-Proxy')
+  end
+end
+
+# UDPProxy.start "127.0.0.1", 50001, "127.0.0.1", 50000

data/lib/udp_proxy/client.rb

@@ -0,0 +1,39 @@
+# TODO: periodically kill all dead connections
+module UDPProxy
+  class Client < EM::Connection
+    attr_accessor :client_ip, :client_port
+
+    def initialize(client_ip, client_port, server)
+      @client_ip, @client_port = client_ip, client_port
+      @server = server
+    end
+
+    def receive_data(data)
+      port, ip = Socket.unpack_sockaddr_in(get_peername)
+      @server.send_datagram(data, client_ip, client_port)
+    end
+  end
+
+  class Clients
+    def initialize(server, address)
+      @address = address
+      @clients = {}
+      @server = server
+    end
+
+    def client(ip, port)
+      @clients[key(ip, port)] || @clients[key(ip,port)] = create_client(ip, port)
+    end
+
+  private
+    def key(ip, port)
+      "#{ip}:#{port}"
+    end
+
+    def create_client(ip, port)
+      Syslog.notice "Creating new client for #{ip}:#{port}"
+      client = EM::open_datagram_socket @address, 0, Client, ip, port, @server
+    end
+  end
+
+end

data/lib/udp_proxy/server.rb

@@ -0,0 +1,14 @@
+module UDPProxy
+  class Server < EM::Connection
+    def initialize(address, relay_ip, relay_port)
+      @relay_ip, @relay_port = relay_ip, relay_port
+      @clients = Clients.new(self, address)
+    end
+
+    def receive_data(data)
+      port, ip = Socket.unpack_sockaddr_in(get_peername)
+      client = @clients.client(ip, port)
+      client.send_datagram(data, @relay_ip, @relay_port)
+    end
+  end
+end

metadata

@@ -0,0 +1,73 @@
+--- !ruby/object:Gem::Specification 
+name: em-udp-proxy
+version: !ruby/object:Gem::Version 
+  prerelease: 
+  version: 0.1.0
+platform: ruby
+authors: 
+- Arik Fraimovich
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2012-03-06 00:00:00 Z
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: eventmachine
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "0"
+  type: :runtime
+  version_requirements: *id001
+description: EventMachine UDP Proxy
+email: 
+- arik@arikfr.com
+executables: 
+- em-udp-proxy
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- .gitignore
+- Gemfile
+- README
+- Rakefile
+- bin/em-udp-proxy
+- em-udp-proxy.gemspec
+- lib/udp_proxy.rb
+- lib/udp_proxy/client.rb
+- lib/udp_proxy/server.rb
+homepage: http://github.com/arikfr/em-udp-proxy
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.8.10
+signing_key: 
+specification_version: 3
+summary: EventMachine UDP Proxy
+test_files: []
+