network-facade 0.1

data/bin/network-facade-ssl

@@ -0,0 +1,117 @@
+#!/usr/bin/ruby
+require 'network-facade'
+
+def create_ca(opts = {})
+	[ :country, :hostname, :domainname ].each do |o|
+		raise "Missing option #{o}" if opts[o].nil?
+	end
+
+	name = [
+		[ 'C', opts[:country], OpenSSL::ASN1::PRINTABLESTRING ],
+		[ 'O', opts[:domainname], OpenSSL::ASN1::UTF8STRING ],
+		[ 'OU', opts[:hostname], OpenSSL::ASN1::UTF8STRING ],
+		[ 'CN', 'CA' ]
+	]
+
+	cert_opts = {
+		:is_ca => true,
+		:name => name,
+		:comment => 'NetworkFacade Certification Authority',
+		:period => 3650
+	}
+
+	create(cert_opts)
+end
+
+def create_cert(opts)
+	[ :country, :hostname, :domainname, :ca_cert, :ca_key ].each do |o|
+		raise "Missing option #{o}" if opts[o].nil?
+	end
+
+	name = [
+		[ 'C', opts[:country], OpenSSL::ASN1::PRINTABLESTRING ],
+		[ 'O', opts[:domainname], OpenSSL::ASN1::UTF8STRING ],
+		[ 'OU', opts[:hostname], OpenSSL::ASN1::UTF8STRING ]
+	]
+
+	opts[:ca_cert] = OpenSSL::X509::Certificate.new(File.read(opts[:ca_cert])) if opts[:ca_cert].is_a?(String)
+	opts[:ca_key] = OpenSSL::PKey::RSA.new(File.read(opts[:ca_key])) if opts[:ca_key].is_a?(String)
+
+	cert_opts = {
+		:name => name,
+		:ca_cert => opts[:ca_cert],
+		:ca_key => opts[:ca_key],
+		:comment => opts[:comment] || 'NetworkFacade Generated Certificate',
+		:period => 365
+	}
+
+	create(cert_opts)
+end
+
+def create(opts)
+	key = OpenSSL::PKey::RSA.new(1024)
+
+	cert = OpenSSL::X509::Certificate.new
+	cert.subject = ::OpenSSL::X509::Name.new(opts[:name])
+	cert.issuer = opts[:is_ca] ? cert.subject : opts[:ca_cert].subject
+	cert.not_before = Time.now
+	cert.not_after = Time.now + (opts[:period] * 24 * 60 * 60)
+	cert.public_key = key.public_key
+	cert.serial = opts[:is_ca] ? 0x0 : 0x2
+	cert.version = 2
+
+	ef = OpenSSL::X509::ExtensionFactory.new
+	ef.subject_certificate = cert
+	ef.issuer_certificate = opts[:is_ca] ? cert : opts[:ca_cert]
+
+	if opts[:is_ca]
+		cert.extensions = [
+			ef.create_extension('basicConstraints','CA:TRUE', true),
+			ef.create_extension('nsComment', opts[:comment]),
+			ef.create_extension('subjectKeyIdentifier', 'hash'),
+			ef.create_extension('keyUsage', 'cRLSign,keyCertSign', true)
+		]
+		cert.add_extension(ef.create_extension('authorityKeyIdentifier', 'keyid:always,issuer:always'))
+	else
+		cert.extensions = [
+			ef.create_extension('basicConstraints','CA:FALSE', true),
+			ef.create_extension('nsComment', opts[:comment]),
+			ef.create_extension('subjectKeyIdentifier', 'hash'),
+		]
+		cert.add_extension(ef.create_extension('authorityKeyIdentifier', 'keyid:always,issuer:always'))
+	end
+
+	cert.sign(opts[:is_ca] ? key : opts[:ca_key], OpenSSL::Digest::SHA1.new)
+
+	{ :key => key, :cert => cert }
+end
+
+# Authority
+
+CA = {
+	:country => 'FR',
+	:hostname => 'demo',
+	:domainname => 'example.com'
+}
+
+cert = create_ca(CA)
+File.open('ca.key', 'w') { |fd| fd.puts cert[:key] }
+File.open('ca.cert', 'w') { |fd| fd.puts cert[:cert].to_pem }
+
+# Client / Server
+
+OPTS = {
+	:country => CA[:country],
+	:hostname => CA[:hostname],
+	:domainname => CA[:domainname],
+	:ca_cert => 'ca.cert',
+	:ca_key => 'ca.key'
+}
+
+cert = create_cert(OPTS)
+File.open('client.key', 'w') { |fd| fd.puts cert[:key] }
+File.open('client.cert', 'w') { |fd| fd.puts cert[:cert].to_pem }
+
+cert = create_cert(OPTS)
+File.open('server.key', 'w') { |fd| fd.puts cert[:key] }
+File.open('server.cert', 'w') { |fd| fd.puts cert[:cert].to_pem }

data/lib/network-facade/base.rb

@@ -0,0 +1,150 @@
+module NetworkFacade
+	def self.log=(log)
+		@@log = log.is_a?(IO) ? Logger.new(log) : log
+	end
+
+	def self.log(level, msg, context = nil)
+		return unless defined? @@log
+		if msg.is_a? Array
+			msg.each { |l| @@log << "       #{l}\n"}
+		elsif msg.is_a? Exception
+			@@log << "       #{msg.message} (#{msg.class})\n"
+			msg.backtrace.each { |l| @@log << "       #{l}\n"}
+		else
+			if context.nil?
+				@@log.send(level, msg)
+			else
+				@@log.send(level, context, &Proc.new { msg })
+			end
+		end
+	end
+
+module Base
+
+	class Client
+		def self.uri=(uri)
+			@@uri = uri
+		end
+
+		def self.inherited(klass)
+			@@klass ||= {}
+			if defined? @@uri and not @@uri.nil?
+				@@klass[klass] = URI.parse(@@uri)
+				@@klass[klass].path = '/' + klass.name.downcase
+				NetworkFacade.log(:debug, "#{klass} is bound to #{@@klass[klass]}")
+			end
+			@@uri = nil
+		end
+
+		def initialize(options = {})
+			@options = options
+			@uri ||= @options[:uri] || @@klass[self.class] ||
+				URI::Generic.new('nf', nil, nil, nil, nil, '/' + self.class.name.downcase, nil, nil, nil)
+			@uri.host = @options[:host] if @options[:host]
+			@uri.port = @options[:port] if @options[:port]
+			@uri.path = @options[:path] if @options[:path]
+			@uri.query = @options[:query] if @options[:query]
+			@uri.userinfo = @options[:userinfo] if @options[:userinfo]
+			@mutex = Mutex.new
+		end
+
+		def method_missing(name, *args)
+			NetworkFacade.log(:debug, "Method called with #{args.inspect}", "#{self.class}##{name}")
+			begin
+				data = Marshal.dump([@uri.path[1..-1], name, args])
+				size = [data.size].pack('N')
+				@client.write(size)
+				@client.write(data)
+				size = @client.read(4).unpack('N').first
+				data = @client.read(size)
+				result = Marshal.load(data)
+				if result.is_a? Exception
+					NetworkFacade.log(:info, "Exception occured : #{result.inspect}", "#{self.class}##{name}")
+					result.backtrace.collect! do |line|
+						@uri.to_s + '/' + line
+					end
+					raise result
+				end
+				result
+			rescue Errno::EPIPE, EOFError, Errno::EINVAL, Errno::ECONNRESET
+				NetworkFacade.log(:warn, "#{$!.inspect} occured, re-connecting...", "#{self.class}##{name}")
+				connect
+				retry
+			end
+		end
+	end
+
+	class Server
+		def initialize(options = {})
+			@options = options
+			@fd = [@options[:server]]
+			@objs = {}
+		end
+
+		def accept
+			NetworkFacade.log(:info, "Accepting new client")
+			raise "Default Server#accept method"
+		end
+
+		def add(obj)
+			id = obj.class.name.downcase
+			NetworkFacade.log(:info, "Adding new object #{obj.inspect} at /#{id}")
+			@objs[id] = obj
+			self
+		end
+
+		def client_id(client)
+			"0x%08x" % client.object_id
+		end
+
+		def start
+			loop do
+				readable, writable, errors, timeout = IO.select(@fd)
+				begin
+					if readable.include?(@options[:server])
+						accept
+						readable.delete(@options[:server])
+					end
+				rescue Exception
+					NetworkFacade.log(:warn, "An error occured when accapting new client")
+					NetworkFacade.log(:warn, $!)
+					next
+				end
+
+				readable.each do |client|
+					size = nil
+					data = nil
+					begin
+						size = client.read(4)
+						raise EOFError if size.nil?
+						size = size.unpack('N').first
+						data = Marshal.load(client.read(size))
+						result = nil
+						begin
+							if @objs[data[0]].respond_to? data[1]
+								NetworkFacade.log(:info, "Call method #{data[1].inspect} with #{data[2].inspect}", client_id(client))
+								result = @objs[data[0]].send(data[1], *data[2])
+							else
+								NetworkFacade.log(:info, "Call unknown method #{data[1].inspect}", client_id(client))
+							end
+						rescue Exception
+							result = $!
+							NetworkFacade.log(:info, "Error occured when executing #{data[1].inspect} with #{data[2].inspect}", client_id(client))
+							NetworkFacade.log(:info, $!)
+						end
+						result = Marshal.dump(result)
+						client.write([result.size].pack('N'))
+						client.write(result)
+					rescue Exception
+						NetworkFacade.log(:info, "Close connection", client_id(client))
+						client.close
+						@fd.delete(client)
+						GC.start
+					end
+				end
+			end
+		end
+	end
+
+end
+end

data/lib/network-facade/config.rb

@@ -0,0 +1,10 @@
+module NetworkFacade
+	NAME = 'Network-Facade'
+	VERSION = '0.1'
+	COPYRIGHT = 'Copyright (C) 2007 Florent Solt'
+	DESC = 'Object-oriented netwotk facade'
+	AUTHOR = 'Florent Solt'
+	EMAIL = 'florent@solt.biz'
+	HOMEPAGE = 'http://network-facade.rubyforge.org'
+	LICENSE = 'BSD'
+end

data/lib/network-facade/defaults.rb

@@ -0,0 +1,14 @@
+module NetworkFacade
+
+	def self.Client(uri = nil)
+		TCP::Client.uri = uri
+		TCP::Client
+	end
+
+	class Client < TCP::Client
+	end
+
+	class Server < TCP::Server
+	end
+
+end

data/lib/network-facade/ssl.rb

@@ -0,0 +1,55 @@
+require 'openssl'
+
+module NetworkFacade
+class SSL
+
+	def self.Client(uri = nil)
+		Client.uri = uri
+		Client
+	end
+
+	PORT = 5043
+
+	class Client < Base::Client
+		def initialize(options = {})
+			[:key, :cert, :ca].each do |o|
+				raise "Missing option #{o}" if options[o].nil?
+				raise "File does not exists #{options[o]}" unless File.exists?(options[o])
+			end
+			super
+			@ctx = OpenSSL::SSL::SSLContext.new
+			@ctx.key = OpenSSL::PKey::RSA.new File.read(@options[:key])
+			@ctx.cert = OpenSSL::X509::Certificate.new File.read(@options[:cert])
+			@ctx.verify_mode = OpenSSL::SSL::VERIFY_PEER  | OpenSSL::SSL::VERIFY_FAIL_IF_NO_PEER_CERT
+			@ctx.ca_file = @options[:ca]
+			@client = TCPSocket.new(@uri.host || 'localhost', @uri.port || PORT)
+			@client = OpenSSL::SSL::SSLSocket.new(@client, @ctx)
+		end
+	end
+
+	class Server < TCP::Server
+		def initialize(options = {})
+			if options[:server].nil?
+				[:key, :cert, :ca].each do |o|
+					raise "Missing option #{o}" if options[o].nil?
+					raise "File does not exists #{options[o]}" unless File.exists?(options[o])
+				end
+				options[:port] ||= PORT
+				options[:host] ||= '0.0.0.0'
+				@ctx = OpenSSL::SSL::SSLContext.new
+				@ctx.key = OpenSSL::PKey::RSA.new File.read(options[:key])
+				@ctx.cert = OpenSSL::X509::Certificate.new File.read(options[:cert])
+				@ctx.verify_mode = OpenSSL::SSL::VERIFY_PEER | OpenSSL::SSL::VERIFY_FAIL_IF_NO_PEER_CERT
+				@ctx.ca_file = options[:ca]
+				options[:server] = TCPServer.new(options[:host], options[:port])
+				@ssl = OpenSSL::SSL::SSLSocket.new(options[:server], @ctx)
+				options[:server] = @ssl.to_io
+			end
+			super(options)
+		end
+	end
+
+end
+end
+
+

data/lib/network-facade/tcp.rb

@@ -0,0 +1,44 @@
+module NetworkFacade
+module TCP
+
+	def self.Client(uri = nil)
+		Client.uri = uri
+		Client
+	end
+
+	PORT = 5042
+
+	class Client < Base::Client
+		def initialize(options = {})
+			super
+			@options[:no_delay] ||= true
+			@client ||= TCPSocket.new(@uri.host || 'localhost', @uri.port || PORT)
+			@client.setsockopt(Socket::SOL_TCP, Socket::TCP_NODELAY, 1) if @options[:no_delay]
+		end
+	end
+
+	class Server < Base::Server
+		def initialize(options = {})
+			options[:port] ||= PORT
+			options[:host] ||= '0.0.0.0'
+			options[:no_delay] ||= true
+			options[:close_exec] ||= true
+			options[:server] ||= TCPServer.new(options[:host], options[:port])
+			super(options)
+		end
+
+		def accept
+			client = @options[:server].accept
+			NetworkFacade.log(:info, "Accept", client.peeraddr[2])
+			client.setsockopt(Socket::SOL_TCP, Socket::TCP_NODELAY, 1) if @options[:no_delay]
+			client.fcntl(Fcntl::F_SETFD, Fcntl::FD_CLOEXEC) if @options[:close_exec]
+			@fd << client
+		end
+
+		def client_id(client)
+			client.peeraddr[2]
+		end
+	end
+
+end
+end

data/lib/network-facade/unix.rb

@@ -0,0 +1,31 @@
+module NetworkFacade
+module Unix
+
+	PATH = '/tmp/nf-socket'
+
+	class Client < Base::Client
+		def initialize(options = {})
+			super
+			options[:path] ||= PATH
+			@client = UNIXSocket.open(options[:path])
+		end
+	end
+
+	class Server < Base::Server
+		def initialize(options = {})
+			options[:path] ||= PATH
+			options[:close_exec] ||= true
+			File.unlink(PATH)
+			options[:server] ||= UNIXServer.open(options[:path])
+			super(options)
+		end
+
+		def accept
+			client = @options[:server].accept
+			client.fcntl(Fcntl::F_SETFD, Fcntl::FD_CLOEXEC) if @options[:close_exec]
+			@fd << client
+		end
+	end
+
+end
+end

data/lib/network-facade.rb

@@ -0,0 +1,66 @@
+require 'socket'
+require 'uri'
+require 'fcntl'
+require 'thread'
+require 'logger'
+require 'network-facade/config'
+require 'network-facade/base'
+require 'network-facade/tcp'
+require 'network-facade/unix'
+require 'network-facade/ssl'
+require 'network-facade/defaults'
+
+if __FILE__ == $0
+	NetworkFacade.log = STDERR
+
+	case ARGV.first
+		#=======================<client>==============================#
+	when 'client-tcp-uri'
+		class Foo < NetworkFacade::Client 'nf://backoffice:5042/?hu=dtc'
+		end
+		f = Foo.new
+		p f.bar
+	when 'client-tcp-class'
+		class Foo < NetworkFacade::Client
+		end
+		f = Foo.new(:host => 'dev', :port => 5042)
+		p f.bar
+	when 'client-unix'
+		class Foo < NetworkFacade::Unix::Client
+		end
+		f = Foo.new
+		p f.bar
+	when 'client-ssl'
+		class Foo < NetworkFacade::SSL::Client 'nf://installclick:5044'
+		end
+		f = Foo.new(:ca => '../cert/ca.cert',
+					:cert => '../cert/client.cert',
+					:key => '../cert/client.key')
+		p f.bar
+		#=======================</client>=============================#
+	else
+		class Foo
+			def bar
+				42
+			end
+			def error
+				raise "Error"
+			end
+		end
+		case ARGV.first
+			#=======================<server>==============================#
+		when 'server-tcp'
+			NetworkFacade::Server.new.add(Foo.new).start
+		when 'server-unix'
+			NetworkFacade::Unix::Server.new.add(Foo.new).start
+		when 'server-ssl'
+			s = NetworkFacade::SSL::Server.new(
+				:port => 5044,
+				:ca => '../cert/ca.cert',
+				:cert => '../cert/server.cert',
+				:key => '../cert/server.key')
+			s.add(Foo.new).start
+			#=======================</server>=============================#
+		end
+	end
+end

metadata

@@ -0,0 +1,60 @@
+--- !ruby/object:Gem::Specification 
+rubygems_version: 0.9.2
+specification_version: 1
+name: network-facade
+version: !ruby/object:Gem::Version 
+  version: "0.1"
+date: 2007-05-02 00:00:00 +02:00
+summary: Object-oriented netwotk facade
+require_paths: 
+- lib
+email: florent@solt.biz
+homepage: http://network-facade.rubyforge.org
+rubyforge_project: 
+description: 
+autorequire: 
+default_executable: 
+bindir: bin
+has_rdoc: false
+required_ruby_version: !ruby/object:Gem::Version::Requirement 
+  requirements: 
+  - - ">"
+    - !ruby/object:Gem::Version 
+      version: 0.0.0
+  version: 
+platform: ruby
+signing_key: 
+cert_chain: 
+post_install_message: 
+authors: 
+- Florent Solt
+files: 
+- lib/network-facade/unix.rb
+- lib/network-facade/tcp.rb
+- lib/network-facade/config.rb
+- lib/network-facade/base.rb
+- lib/network-facade/defaults.rb
+- lib/network-facade/ssl.rb
+- lib/network-facade.rb
+test_files: []
+
+rdoc_options: []
+
+extra_rdoc_files: []
+
+executables: 
+- network-facade-ssl
+extensions: []
+
+requirements: []
+
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: xmlobject
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Version::Requirement 
+    requirements: 
+    - - ">"
+      - !ruby/object:Gem::Version 
+        version: 0.0.0
+    version: