ruby_ex 0.1.1

data/src/pp_hierarchy.rb

@@ -0,0 +1,30 @@
+# Copyright: Copyright (c) 2004  Nicolas Despres. All rights reserved.
+# Author: Nicolas Despres  <polrop@lrde.epita.fr>.
+# License: Gnu General Public License.
+
+# $LastChangedBy: polrop $
+# $Id: hierarchy.rb 145 2005-01-30 14:16:14Z polrop $
+
+require 'ruby_ex'
+
+class Hash
+
+  # Pretty print a hash symbolizing a class hierarchy.
+  def pp_hierarchy(prefix=//, abstract=false)
+    pp_hierarchy_rec('', prefix, abstract)
+  end
+
+  protected
+  def pp_hierarchy_rec(indent, prefix, abstract)
+    self.each do |k, v|
+      k_str = k.to_s.sub(prefix, '')
+      if abstract and k.abstract?
+        puts "#{indent}/#{k_str}/"
+      else
+        puts "#{indent}#{k_str}"
+      end
+      v.pp_hierarchy_rec(indent + '  ', prefix, abstract)
+    end
+  end
+
+end # class Hash

data/src/random_generators.rb

@@ -0,0 +1,29 @@
+# Author::    Nicolas Pouillard  <ertai@lrde.epita.fr>.
+# Copyright:: Copyright (c) 2005 Nicolas Pouillard. All rights reserved.
+# License::   GNU General Public License (GPL).
+# Revision::  $Id: random_generators.rb 266 2005-06-01 14:27:18Z ertai $
+
+require 'ruby_ex'
+require 'module/autoload_tree'
+require 'abstract'
+
+module RandomGenerators
+
+  autoloaded_module(__FILE__)
+
+  module_function
+
+  def default
+    class << self
+      def default
+        @@default
+      end
+    end
+    @@default = Ruby.new
+  end
+
+  def default= ( aClass )
+    @@default = aClass
+  end
+
+end # module RandomGenerators

data/src/random_generators/random_generator.rb

@@ -0,0 +1,33 @@
+# Author::    Nicolas Pouillard  <ertai@lrde.epita.fr>.
+# Copyright:: Copyright (c) 2005 Nicolas Pouillard. All rights reserved.
+# License::   GNU General Public License (GPL).
+# Revision::  $Id: random_generator.rb 258 2005-06-01 00:22:51Z ertai $
+
+require 'random_generators'
+
+module RandomGenerators
+
+  class RandomGenerator
+    abstract
+
+    def check_choose_integer ( n )
+      unless n.is_a? Integer
+        raise ArgumentError, "Need an integer (not #{n})"
+      end
+      if n <= 0
+        raise ArgumentError, "Need an positive value (not #{n})"
+      end
+    end
+
+    def check_choose_float ( f )
+      unless f.is_a? Float
+        raise ArgumentError, "Need a float (not #{f})"
+      end
+      if f <= 0
+        raise ArgumentError, "Need an positive value (not #{f})"
+      end
+    end
+
+  end # class RandomGenerator
+
+end # module RandomGenerators

data/src/random_generators/ruby.rb

@@ -0,0 +1,25 @@
+# Author::    Nicolas Pouillard  <ertai@lrde.epita.fr>.
+# Copyright:: Copyright (c) 2005 Nicolas Pouillard. All rights reserved.
+# License::   GNU General Public License (GPL).
+# Revision::  $Id: ruby.rb 258 2005-06-01 00:22:51Z ertai $
+
+require 'random_generators'
+
+module RandomGenerators
+
+  class Ruby < RandomGenerator
+    concrete
+
+    def choose_integer ( n )
+      check_choose_integer(n)
+      Kernel.rand(n)
+    end
+
+    def choose_float ( f=1.0 )
+      check_choose_float(f)
+      Kernel.rand() * f
+    end
+
+  end # class Ruby
+
+end # module RandomGenerators

data/src/ruby_ex.rb

@@ -0,0 +1,124 @@
+# Copyright:: Copyright (c) 2005  Nicolas Pouillard. All rights reserved.
+# Author::    Nicolas Pouillard  <ertai@lrde.epita.fr>.
+# License::   Gnu General Public License.
+# Revision::  $Id: ruby_ex.rb 266 2005-06-01 14:27:18Z ertai $
+
+begin
+  require 'rubygems'
+  require_gem 'core_ex' # FIXME version
+  require 'core_ex'
+rescue LoadError => ex
+  core_ex = @@dir.parent + 'core_ex/lib/core_ex'
+  if core_ex.exists? and not $".include? 'core_ex.rb'
+    require core_ex.to_s
+  else
+    raise ex
+  end
+end
+  
+
+module DRb
+end
+
+module Algorithms
+end
+
+module RubyEx
+
+  @@dir = Pathname.new(__FILE__).dirname.expand_path
+  $: << @@dir.to_s
+
+  module_function
+
+  def sources
+    @@contents ||= Hash.new([].freeze)
+    %[
+      abstract.rb                        Abstract Concrete Class
+      abstract_node.rb                   AbstractNode
+      algorithms.rb                      Algorithms
+      attributed_class.rb                AttributedClass
+      cache.rb                           Cache
+      config_file.rb                     ConfigFile
+      const_regexp.rb                    ConstRegexp
+      daemon.rb                          Daemon
+      diff.rb                            Diff
+      dlogger.rb                         DLogger
+      drb.rb                             DRb
+      dumpable_proc.rb                   DumpableProc
+      filetype.rb                        FileType
+      histogram.rb                       Histogram
+      hookable.rb                        Hookable
+      hooker.rb                          Hooker
+      inactive_timeout.rb                InactiveTimeout
+      indexed_node.rb                    IndexedNode
+      ioo.rb                             IOO IOOTCPSocket
+      killall.rb                         KillAll
+      labeled_node.rb                    LabeledNode
+      node.rb                            Node
+      object_monitor.rb                  ObjectMonitor
+      object_monitor_activity.rb         ObjectMonitorActivity
+      observable.rb                      Observable
+      observable_pool.rb                 ObservablePool
+      orderedhash.rb                     OrderedHash_ext OHash
+      pseudo_cache.rb                    PseudoCache
+      random_generators.rb               RandomGenerators
+      timeout_ex.rb                      TimeoutEx
+      verbose_object.rb                  VerboseObject
+      spring.rb                          Spring
+      spring_set.rb                      SpringSet
+      symtbl.rb                          SymTbl
+      synflow.rb SynFlowFactory          SynFlow
+      safe_eval.rb                       SafeEval
+      sendmail.rb                        Sendmail
+      service_manager.rb                 ServiceManager
+      test/unit/ui/yaml/testrunner.rb    Test::Unit::UI::Yaml::TestRunner
+      # drb/session_manager.rb             DRb::SessionManager
+      # drb/dispatcher.rb                  DRb::Dispatcher
+    ].split(/\n/).each do |line|
+      line.sub!(/#.*$/, '')
+      line.strip!
+      next if line =~ /^\s*$/
+      file, *names = line.split(/\s+/)
+      @@contents[@@dir + file] = names
+    end
+    @@contents.freeze
+
+    @@sources = FileList[@@dir + '**/*.rb'].exclude(/\bsession\/error.rb\b/)
+
+    class << self
+      def sources
+        @@sources
+      end
+    end
+    @@sources
+  end
+
+  def dir
+    @@dir
+  end
+
+  def each ( &block )
+    sources.each do |file|
+      block[file, @@contents[file]]
+    end
+    self
+  end
+
+  def autoload
+    each do |file, names|
+      names.each do |name|
+        name = name.split(/::/).first
+        Kernel.autoload(name, file.to_s)
+      end
+    end
+  end
+
+  def require
+    each { |file, names| file.require }
+  end
+
+end # module RubyEx
+
+# RubyEx.each { |*a| p a }
+# RubyEx.require
+# RubyEx.autoload

data/src/safe_eval.rb

@@ -0,0 +1,346 @@
+# Copyright: Copyright (c) 2004  Nicolas Despres. All rights reserved.
+# Author: Nicolas Despres  <polrop@lrde.epita.fr>.
+# License: Gnu General Public License.
+
+# $LastChangedBy: ertai $
+# $Id: safe_eval.rb 266 2005-06-01 14:27:18Z ertai $
+
+require 'ruby_ex'
+
+class SafeEval
+
+  attr_accessor :collect, :safe
+  attr_reader   :output
+
+  def initialize(safe=3, collect=true)
+    @safe = safe
+    @collect = collect
+  end
+
+  def run(code_str='', binding=nil, filename='(SafeEval)', lineno=1)
+    binding = yield if block_given?
+    lineno -= 1
+    @output = nil
+    @output = [] if @collect
+    result = nil
+    code_str.strip!
+    return result if code_str.empty?
+    rd_result, wr_result = IO.pipe
+    open('|-') do |io|
+      if io.nil? then # son
+        status = 0
+        rd_result.close
+        set_environment
+        begin
+          code = "$SAFE = #@safe\n#{code_str}"
+          result = eval(code, binding, filename, lineno)
+          status = 0 # needed if the user change the status
+        rescue Exception => exception
+          status = 1
+          result = exception
+        ensure
+          Marshal.dump(result, wr_result)
+          wr_result.close
+          exit!(status)
+        end
+      else # father
+        wr_result.close
+        io.each_line { |l| @output << l.chomp } if @collect
+        result = Marshal.load(rd_result)
+        rd_result.close
+        Process.waitpid(io.pid)
+      end
+    end
+    raise result if $?.exitstatus == 1
+    @output = nil unless @collect
+    return result
+  end
+  public :run
+
+  # Can be overloaded to specify environment restriction.
+  def set_environment
+  end
+  protected :set_environment
+
+end # class SafeEval
+
+
+#
+# Unit test suite
+#
+test_section __FILE__ do
+
+
+class SafeEvalTest < Test::Unit::TestCase
+
+  #
+  # Tests
+  #
+  def test_interface
+    se = SafeEval.new
+    assert_respond_to(se, :run)
+    assert_respond_to(se, :output)
+    assert_respond_to(se, :safe)
+    assert_respond_to(se, :collect)
+    assert(se.protected_methods.include?('set_environment'),
+           "no protected method 'set_environment'")
+  end
+
+  def test_nil
+    se = SafeEval.new
+    assert_equal(nil, se.run)
+    assert_equal([], se.output)
+  end
+
+  def test_true
+    se = SafeEval.new
+    assert_equal(true, se.run('true'))
+    assert_equal([], se.output)
+  end
+
+  def test_false
+    se = SafeEval.new
+    assert_equal(false, se.run('false'))
+    assert_equal([], se.output)
+  end
+
+  def test_nil_no_collect
+    se = SafeEval.new(3, false)
+    assert_equal(nil, se.run)
+    assert_equal(nil, se.output)
+  end
+
+  def test_almost_empty_code
+    se = SafeEval.new
+    assert_equal(nil, se.run('            '))
+    assert_equal([], se.output)
+  end
+
+  def test_true_no_collect
+    se = SafeEval.new(3, false)
+    assert_equal(true, se.run('true'))
+    assert_equal(nil, se.output)
+  end
+
+  def test_binding_argument
+    def getBinding; a = 42; return binding; end
+    se = SafeEval.new
+    assert_equal(nil, se.run('puts a', getBinding))
+    assert_equal(['42'], se.output)
+  end
+
+  def test_binding_block
+    se = SafeEval.new
+    assert_equal(true, se.run('puts a; true') { a = 42; return binding })
+    assert_equal(['42'], se.output)
+  end
+
+  def test_binding_both
+    def getBinding2; a = 51; return binding; end
+    se = SafeEval.new
+    ret = se.run('puts a; true', getBinding2) { a = 42; return binding }
+    assert_equal(true, ret)
+    assert_equal(['42'], se.output)
+  end
+
+  def test_exception
+    mypid = $$
+    se = SafeEval.new
+    $tmp = 42
+    begin
+      $tmp = se.run('raise NoMethodError')
+    rescue NoMethodError
+      assert_equal(mypid, $$, "bad pid")
+      assert_equal(42, $tmp, "result must be nil")
+      assert_equal([], se.output)
+    ensure
+      assert_equal(mypid, $$, "bad pid")
+    end
+  end
+
+  def test_exception_collect
+    se = SafeEval.new
+    assert_raises(Exception) do
+      se.run('puts "toto"; raise Exception')
+    end
+    assert_equal(['toto'], se.output)
+  end
+
+# FIXME: this test block if DRb::SessionManager test suite is executed before
+#        this test suite in check-ruby_ex.
+#   def test_exception_no_collect
+#     se = SafeEval.new(3, false)
+#     assert_raises(Exception) do
+#       se.run('puts "toto"; raise Exception')
+#     end
+#     assert_equal(nil, se.output)
+#   end
+
+  def test_require_forbidden
+    se = SafeEval.new
+    assert_raises(SecurityError) do
+      se.run("require 'foo'")
+    end
+    assert_equal([], se.output)
+  end
+
+  def test_File_chmod_forbidden
+    se = SafeEval.new
+    assert_raises(SecurityError) do
+      se.run("File.chmod(0755, '/tmp')")
+    end
+    assert_equal([], se.output)
+  end
+
+  def test_system_forbidden
+    se = SafeEval.new
+    assert_raises(SecurityError) do
+      se.run("system('rm /tmp/x')")
+    end
+    assert_equal([], se.output)
+  end
+
+  def test_system_allowed
+    se = SafeEval.new(2)
+    se.run("system(\"echo 'toto'\")")
+    assert_equal(['toto'], se.output)
+  end
+
+  def test_print
+    se = SafeEval.new
+    assert_equal(nil, se.run('print "hello"'))
+    assert_equal(['hello'], se.output)
+  end
+
+  def test_puts
+    se = SafeEval.new
+    assert_equal(nil, se.run('puts "hello"'))
+    assert_equal(['hello'], se.output)
+  end
+
+  def test_print_no_collect
+    se = SafeEval.new(3, false)
+    assert_equal(nil, se.run('print "hello"'))
+    assert_equal(nil, se.output)
+  end
+
+  def test_puts_several_lines
+    se = SafeEval.new
+    assert_equal(nil, se.run('puts "hello\nworld"'))
+    assert_equal(['hello', 'world'], se.output)
+  end
+
+  def test_syntax_error
+    se = SafeEval.new
+    assert_raises(SyntaxError) do
+      se.run('4syntaxerror')
+    end
+    assert_equal([], se.output)
+  end
+
+  def test_bad_identifier
+    se = SafeEval.new
+    assert_raises(NameError) do
+      se.run('bad_identifier.directory?')
+    end
+    assert_equal([], se.output)
+  end
+
+  def test_syntax_error_location
+    se = SafeEval.new
+    exc = assert_raises(SyntaxError) do
+      se.run('4syntaxerror', nil, 'toto.yml', 1)
+    end
+    expected_message = <<EOF
+compile error
+toto.yml:1: syntax error
+4syntaxerror
+            ^
+EOF
+    expected_message.chomp!
+    assert_equal(expected_message, exc.message)
+  end
+
+  def test_syntax_error_location_several_line
+    code = <<EOF
+puts "hello"
+puts "world"
+4syntaxerror;
+EOF
+    se = SafeEval.new
+    exc = assert_raises(SyntaxError) do
+      se.run(code, nil, 'toto.yml', 1)
+    end
+    expected_message = <<EOF
+compile error
+toto.yml:3: syntax error
+4syntaxerror;
+            ^
+EOF
+    expected_message.chomp!
+    assert_equal(expected_message, exc.message)
+  end
+
+  def test_exit_forbidden
+    se = SafeEval.new(3, false)
+    assert_raises(SystemExit) do
+      se.run('exit(42)')
+    end
+    assert_equal(nil, se.output)
+  end
+
+  def test_exit_strong_forbidden
+    se = SafeEval.new
+    assert_raises(EOFError) do
+      se.run('exit!(42)')
+    end
+    assert_equal([], se.output)
+  end
+
+  def test_undef_me
+    se = SafeEval.new
+    assert_raises(NoMethodError) do
+      se.run('undef_method :run')
+    end
+  end
+
+  def test_read_my_var
+    se = SafeEval.new
+    str = 'puts code'
+    se.run(str)
+    assert_equal(str, se.output[-1])
+  end
+
+  def test_write_status
+    se = SafeEval.new
+    assert_equal(1, se.run('status = 1'))
+  end
+
+  def test_write_safe
+    se = SafeEval.new
+    assert_equal(1, se.run('@safe = 1'))
+    assert_equal(true, se.run('@safe == 3'))
+  end
+
+  def test_write_output
+    se = SafeEval.new
+    assert_equal(['foo'], se.run('@output << "foo"'))
+    assert_equal([], se.output)
+  end
+
+  def test_write_output_and_print
+    se = SafeEval.new
+    assert_equal(nil, se.run('@output << "foo"; p @output'))
+    assert_equal(["[\"foo\"]"], se.output)
+  end
+
+  def test_sleep
+    se = SafeEval.new
+    assert(se.run('sleep(1)') >= 1)
+    assert_equal([], se.output)
+  end
+
+end # class SafeEvalTest
+
+
+end