vcs 0.1 → 0.2.148

data/ruby_ex/queue.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: queue.rb 53 2004-12-02 22:24:03Z ertai $
+
+
+require 'thread'
+
+
+class Queue
+
+  alias put push
+  alias get pop
+
+end # class Queue
+
+
+#
+# Unit test suite
+#
+if defined? TEST_MODE or $0 == __FILE__
+
+
+require 'test/unit/ui/yaml/testrunner'
+
+
+class QueueTest < Test::Unit::TestCase
+
+  #
+  # tests
+  #
+  def test_simple
+    q = Queue.new
+    thr1 = Thread.new do
+      while q.empty? do end
+      assert_equal(2, q.size)
+      assert_equal(2, q.length)
+      assert_equal('hello', q.get)
+      assert_equal('world', q.get)
+      assert(q.empty?)
+    end
+    thr2 = Thread.new do
+      sleep(0.2)
+      q.put('hello')
+      q.put('world')
+    end
+    thr1.join
+    thr2.join
+  end
+
+end # class QueueTest
+
+
+end

data/ruby_ex/random_generators.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: random_generators.rb 218 2005-05-09 12:11:21Z ertai $
+
+require 'module/autoload_tree'
+require 'abstract'
+
+module RandomGenerators
+
+  autoloaded_module(__FILE__)
+
+  @@default = Ruby.new
+
+  module_function
+
+  def default
+    @@default
+  end
+
+  def default= ( aClass )
+    @@default = aClass
+  end
+
+end # module RandomGenerators

data/ruby_ex/random_generators/random_generator.rb

@@ -0,0 +1,31 @@
+# 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 218 2005-05-09 12:11:21Z ertai $
+
+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/ruby_ex/random_generators/ruby.rb

@@ -0,0 +1,23 @@
+# 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 218 2005-05-09 12:11:21Z ertai $
+
+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/ruby_ex/safe_eval.rb

@@ -0,0 +1,348 @@
+# Copyright: Copyright (c) 2004  Nicolas Despres. All rights reserved.
+# Author: Nicolas Despres  <polrop@lrde.epita.fr>.
+# License: Gnu General Public License.
+
+# $LastChangedBy: polrop $
+# $Id: safe_eval.rb 91 2005-01-09 22:24:40Z polrop $
+
+
+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
+#
+if defined? TEST_MODE or $0 == __FILE__
+
+
+require 'test/unit/ui/yaml/testrunner'
+
+
+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('/tmp', 0755)")
+    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