ttk 0.1.576

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/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

data/ruby_ex/service_manager.rb

@@ -0,0 +1,121 @@
+# Copyright: Copyright (c) 2004  Nicolas Despres. All rights reserved.
+# Author: Nicolas Despres  <polrop@lrde.epita.fr>.
+# License: Gnu General Public License.
+
+# $LastChangedBy: ertai $
+# $Id: service_manager.rb 186 2005-04-03 00:07:45Z ertai $
+
+
+# A service must implement at least:
+#   - start_service(uri, *args, &block)
+#   - stop_service
+#   - server
+#   - thread
+#   - uri
+module ServiceManager
+
+  def start_service(service, uri=nil, *args, &block)
+    @services = {} unless defined? @services
+    service.start_service(uri, *args, &block)
+    @services[service.uri] = service
+  end
+
+  def stop_service(uri)
+    if defined? @services
+      if service = @services.delete(uri)
+        service.stop_service
+      end
+    end
+  end
+
+  def stop_services
+    if defined? @services
+      @services.keys.each do |uri|
+        stop_service(uri)
+      end
+    end
+  end
+
+  def count_services
+    if defined? @services
+      @services.size
+    else
+      0
+    end
+  end
+
+  def list_services
+    if defined? @services
+      @services.values
+    else
+      []
+    end
+  end
+
+  def join_services
+    if defined? @services
+      @services.each { |uri, service| service.thread.join }
+    end
+  end
+
+  def each_services(&block)
+    if defined? @services
+      @services.each { |uri, service| block[uri, service] }
+    end
+  end
+
+  def service_server(uri)
+    @services[uri].server if defined? @services
+  end
+
+end # module ServiceManager
+
+
+if defined? TEST_MODE or __FILE__ == $0
+
+
+require 'test/unit/ui/yaml/testrunner'
+require 'drb'
+
+
+class ServiceManagerTest < Test::Unit::TestCase
+
+  class Foo
+    include ServiceManager
+
+  end
+
+  module MyService
+
+    module_function
+    def start_service(uri); end
+    def stop_service; end
+    def uri; 'toto'; end
+
+  end
+
+  def test_simple
+    f = Foo.new
+    f.start_service(DRb)
+    assert_equal(1, f.count_services)
+    f.start_service(DRb)
+    assert_equal(2, f.count_services)
+    f.start_service(MyService)
+    [ DRb, DRb, MyService ].each do |s|
+      assert(f.list_services.include?(s))
+    end
+    assert_equal(3, f.count_services)
+    f.stop_service(MyService.uri)
+    assert_equal(2, f.count_services)
+    f.start_service(MyService)
+    [ DRb, DRb, MyService ].each do |s|
+      assert(f.list_services.include?(s))
+    end
+    f.stop_services
+    assert_equal([], f.list_services)
+  end
+
+end # class ServiceManagerTest
+
+end
+