ruby_ex 0.1.1

data/src/dlogger.rb

@@ -0,0 +1,62 @@
+# Copyright: Copyright (c) 2004  Nicolas Pouillard. All rights reserved.
+# Author: Nicolas Pouillard  <ertai@lrde.epita.fr>.
+# License: Gnu General Public License.
+
+# $LastChangedBy: ertai $
+# $Id: dlogger.rb 266 2005-06-01 14:27:18Z ertai $
+
+require 'ruby_ex'
+require 'logger'
+
+# This Logger subclass permit to use more debug levels
+#
+# log = DLogger.new(STDERR)
+# log.level = -1
+# log.debug1 "You see me"
+# log.debug2 "You see me"
+# log.debug3 "You don't see me"
+#
+class DLogger < Logger
+
+  def format_severity ( severity )
+    return super if severity >= 0
+    return "DEBUG#{severity.abs + 1}"
+  end
+  private :format_severity
+
+  def method_missing ( meth, progname=nil, *args, &block )
+    return super unless meth.to_s =~ /^debug(\d+)$/ or !args.empty
+    add(-$1.to_i + 1, nil, progname, &block)
+  end
+
+  Format = "%s, [%s#%d] %6s -- %s: %s\n"
+  def format_message(severity, timestamp, msg, progname)
+    Format % [severity[0..0], timestamp, $$, severity, progname, msg]
+  end
+  private :format_message
+
+end # class DLogger
+
+
+test_section __FILE__ do
+
+  require 'stringio'
+
+  class DLoggerTest < Test::Unit::TestCase
+
+    def test1
+      cout = StringIO.new
+      log = DLogger.new(cout)
+      log.level = -1
+      log.debug1 'You see me'
+      log.debug3 'You don\'t see me'
+      log.debug2 'You see me'
+      cout.rewind
+      ref = /D, \[\d+-\d+-\d+.\d+:\d+:\d+\.\d+ #\d+\]  DEBUG -- : You see me
+D, \[\d+-\d+-\d+.\d+:\d+:\d+\.\d+ #\d+\] DEBUG2 -- : You see me/
+      assert_match(ref, cout.readlines.join)
+    end
+
+  end # class DLoggerTest
+
+end

data/src/drb/drb_observable.rb

@@ -0,0 +1,95 @@
+# Copyright: Copyright (c) 2004  Nicolas Despres. All rights reserved.
+# Author: Nicolas Despres  <polrop@lrde.epita.fr>.
+# License: Gnu General Public License.
+
+# $LastChangedBy: ertai $
+# $Id: drb_observable.rb 266 2005-06-01 14:27:18Z ertai $
+
+
+require 'ruby_ex'
+require 'drb/observer'
+
+require 'observable'
+
+
+module DRb
+
+  module DRbObservable
+    include Observable
+
+    undef notify_observers
+    def notify_observers(*args, &block)
+      if defined? @observer_state and @observer_state
+	if defined? @observer_peers
+	  for i in @observer_peers.dup
+	    begin
+	      i.update(*args, &block)
+	    rescue
+	      delete_observer(i)
+	    end
+	  end
+	end
+	@observer_state = false
+      end
+    end
+
+    def notify_observer(observer, *arg, &block)
+      if defined? @observer_state and @observer_state
+        if defined? @observer_peers
+          if @observer_peers.include?(observer)
+            begin
+              observer.update(*arg, &block)
+            rescue
+              delete_observer(observer)
+            end
+          end
+        end
+        @observer_state = false
+      end
+    end
+
+  end # module DRbObservable
+
+end # module DRb
+
+
+test_section __FILE__ do
+
+
+module DRb
+
+  class ObservablePoolTest < ::Test::Unit::TestCase
+
+    class Obsv
+      include DRbObservable
+
+    end
+
+    class Foo
+
+      def update(*args, &block)
+        raise 'toto'
+      end
+
+    end
+
+    def test_simple
+      o = Obsv.new
+      assert_respond_to(o, :list_observers)
+      o.add_observer(Foo.new)
+      assert_equal(1, o.count_observers)
+      o.changed
+      o.notify_observers('msg')
+      assert_equal(0, o.count_observers)
+    end
+
+
+  end # class ObservablePoolTest
+
+end # module DRb
+
+
+end
+
+
+

data/src/drb/drb_observable_pool.rb

@@ -0,0 +1,27 @@
+# Copyright: Copyright (c) 2004  Nicolas Despres. All rights reserved.
+# Author: Nicolas Despres  <polrop@lrde.epita.fr>.
+# License: Gnu General Public License.
+
+# $LastChangedBy: polrop $
+# $Id: drb_observable_pool.rb 97 2005-01-13 18:20:49Z polrop $
+
+
+require 'observable_pool'
+
+require 'drb/drb_observable'
+
+
+module DRb
+
+  class DRbObservablePool
+
+    class SyncObserver
+      include DRbObservable
+
+    end # class SyncObserver
+
+  end # class DRbObservable
+
+end # module DRb
+
+

data/src/drb/drb_service.rb

@@ -0,0 +1,44 @@
+# Author:: Nicolas Despres  <nicolas.despres@epita.fr>.
+# Copyright:: Copyright (c) 2004 TTK Team. All rights reserved.
+# License:: Ruby license.
+
+# $LastChangedBy: ertai $
+# $Id: drb_service.rb 266 2005-06-01 14:27:18Z ertai $
+
+
+require 'ruby_ex'
+require 'drb/drb'
+
+
+module DRb
+
+  class DRbService
+    include DRbUndumped
+
+    def initialize(config={})
+      @config = config
+      @server = nil
+      @uri = nil
+      @front = nil
+    end
+
+    attr_reader :server, :uri, :front
+
+    def start_service(uri)
+      @server = DRb.start_service(uri.nil? ? uri : uri.to_s, @front, @config)
+      @uri = URI.parse(@server.uri)
+    end
+
+    def stop_service
+      @server.stop_service if @server
+    end
+
+    def thread
+      @server.thread if @server
+    end
+
+  end # class DRbService
+
+end # module DRb
+
+

data/src/drb/drb_undumped_attributes.rb

@@ -0,0 +1,56 @@
+# Copyright: Copyright (c) 2004  Nicolas Despres. All rights reserved.
+# Author: Nicolas Despres  <polrop@lrde.epita.fr>.
+# License: Gnu General Public License.
+
+# $LastChangedBy: ertai $
+# $Id: drb_undumped_attributes.rb 266 2005-06-01 14:27:18Z ertai $
+
+
+require 'ruby_ex'
+require 'drb/drb'
+
+
+module DRb
+
+  module DRbUndumpedAttributes
+
+    def DRbUndumpedAttributes.append_features(cl)
+      super
+      cl.module_eval do
+
+        def _dump(depth)
+          if self.class.include?(DRb::DRbUndumped)
+            raise(TypeError, 'can\'t dump')
+          end
+          attribs = {}
+          instance_variables.each do |attr|
+            obj = instance_variable_get(attr)
+            begin
+              attribs[attr] = Marshal.dump(obj)
+            rescue TypeError
+              if (defined? LOG) and (not obj.class.include? DRb::DRbUndumped)
+                LOG.warn("#{self.class}._dump") do
+                  "can't dump a #{attr}:#{obj.class}"
+                end
+              end
+              attribs[attr] = Marshal.dump(DRbObject.new(obj))
+            end
+          end
+          Marshal.dump([self.class, attribs])
+        end
+
+        def self._load(data)
+          klass, attribs = Marshal.load(data)
+          obj = klass.new
+          attribs.each do |attr, value|
+            obj.instance_variable_set(attr, Marshal.load(value))
+          end
+          obj
+        end
+      end
+
+    end
+
+  end # module DRbUndumpedAttributes
+
+end # module DRb

data/src/drb/drb_undumped_indexed_object.rb

@@ -0,0 +1,55 @@
+# Author:: Nicolas Despres  <nicolas.despres@epita.fr>.
+# Copyright:: Copyright (c) 2004 TTK Team. All rights reserved.
+# License:: Ruby license.
+
+# $LastChangedBy: ertai $
+# $Id: drb_undumped_indexed_object.rb 266 2005-06-01 14:27:18Z ertai $
+
+
+require 'ruby_ex'
+require 'drb'
+
+
+module DRb
+
+  module DRbUndumpedIndexedObject
+
+    def DRbUndumpedIndexedObject.append_features(cl)
+      super
+      cl.module_eval do
+
+        def _dump(depth)
+          if self.class.include?(DRb::DRbUndumped)
+            raise(TypeError, 'can\'t dump')
+          end
+          attribs = {}
+          each_pair do |k, v|
+            begin
+              attribs[k] = Marshal.dump(v)
+            rescue TypeError
+              if (defined? LOG) and (not v.class.include? DRb::DRbUndumped)
+                LOG.warn("#{self.class}._dump") do
+                  "can't dump a #{k}:#{v.class}"
+                end
+              end
+              attribs[k] = Marshal.dump(DRbObject.new(v))
+            end
+          end
+          Marshal.dump([self.class, attribs])
+        end
+
+        def self._load(data)
+          klass, attribs = Marshal.load(data)
+          obj = klass.new
+          attribs.each do |key, value|
+            obj[key] = Marshal.load(value)
+          end
+          obj
+        end
+      end
+
+    end
+
+  end # module DRbUndumpedIndexedObject
+
+end # module DRb

data/src/drb/insecure_protected_methods.rb

@@ -0,0 +1,101 @@
+# Copyright: Copyright (c) 2004  Nicolas Despres. All rights reserved.
+# Author: Nicolas Despres  <polrop@lrde.epita.fr>.
+# License: Gnu General Public License.
+
+# $LastChangedBy: ertai $
+# $Id: insecure_protected_methods.rb 268 2005-06-02 12:39:07Z ertai $
+
+
+require 'ruby_ex'
+require 'drb/drb'
+
+
+module DRb
+
+  class DRbServer
+
+    alias_method :ori_check_insecure_method, :check_insecure_method
+    def check_insecure_method(obj, msg_id)
+      ori_check_insecure_method(obj, msg_id)
+      if obj.protected_methods.include?(msg_id.to_s)
+        desc = any_to_s(obj)
+        if desc.nil? || desc[0] == '#'
+          desc << ":#{obj.class}"
+        end
+        raise NameError, "protected method `#{msg_id}' called for #{desc}"
+      end
+      true
+    end
+
+  end # class DRbServer
+
+end # module DRb
+
+
+#
+# Unit test suite
+#
+test_section __FILE__ do
+
+
+module DRb
+
+  class InsecureProtectedMethodsTest < Test::Unit::TestCase
+
+    class SrvFoo
+
+      def public_request
+      end
+
+      protected
+      def protected_request
+      end
+
+    end # class Foo
+
+    def setup
+      rd, wr = IO.pipe
+      unless @srv_pid = fork # son
+        rd.close
+        DRb.start_service("druby://localhost:42000", SrvFoo.new)
+        wr.puts('ready')
+        wr.close
+        DRb.thread.join
+      else # father
+        wr.close
+        rd.gets
+        @drb_srv = DRb.start_service
+        @drb_obj = DRbObject.new(nil, "druby://localhost:42000")
+        rd.close
+      end
+    end
+
+    def teardown
+      @drb_srv.stop_service unless @drb_srv.nil?
+      Process.kill('TERM', @srv_pid)
+      Process.waitpid(@srv_pid)
+    end
+
+    #
+    # Test
+    #
+    def test_simple
+      assert_raises(NameError) { @drb_obj.send(:protected_resquest) }
+      assert_nothing_raised { @drb_obj.public_request }
+    end
+
+    #
+    # Tools
+    #
+    def server_request(req)
+    end
+
+  end # class InsecureProtectedMethodsTest
+
+
+end # module DRb
+
+
+end
+
+