secure 0.0.5 → 0.1.0

data/lib/secure.rb

@@ -2,6 +2,8 @@ require "secure/version"
 require "secure/response"
 require "secure/errors"
 require "secure/guard_thread"
+require "secure/parent_process"
+require "secure/child_process"
 require "secure/runner"
 
 module Secure

data/lib/secure/child_process.rb

@@ -0,0 +1,31 @@
+module Secure
+  class ChildProcess
+    def initialize(opts, read_file, write_file)
+      read_file.close
+      @pipe = write_file
+      @timeout = opts[:timeout] || 1
+    end
+
+    def guard_threads
+      @guard_threads || []
+    end
+
+    def safely_run_block
+      thread = Thread.start do
+        $SAFE=3
+        yield
+      end
+
+      guard_threads << GuardThread.kill_thread_on_timeout(@timeout, thread)
+
+      Response.success(thread.value)
+    rescue Exception => e
+      Response.error(e)
+    end
+
+    def execute
+      ret = safely_run_block { yield }
+      @pipe.write(Marshal.dump(ret))
+    end
+  end
+end

data/lib/secure/parent_process.rb

@@ -0,0 +1,12 @@
+module Secure
+  class ParentProcess
+    def initialize(read_file, write_file)
+      @pipe = read_file
+      write_file.close
+    end
+
+    def execute
+      Marshal.load(@pipe.read)
+    end
+  end
+end

data/lib/secure/runner.rb

@@ -1,26 +1,25 @@
 module Secure
   class Runner
     def initialize(opts)
-      @timeout = opts[:timeout] || 1
-    end
-
-    def guard_threads
-      @guard_threads || []
+      @opts = opts
     end
 
     def run
-      thread = Thread.start do
-        $SAFE=3
-        Response.success(yield)
-      end
+      read_file, write_file = IO.pipe
 
-      guard_threads << GuardThread.kill_thread_on_timeout(@timeout, thread)
+      child = fork do
+        begin
+          ChildProcess.new(@opts, read_file, write_file).execute { yield }
+        ensure
+          exit!
+        end
+      end
 
-      thread.value
-    rescue SecurityError, TimeoutError => e
-      Response.error(e)
+      Process.wait(child)
+      ParentProcess.new(read_file, write_file).execute
     ensure
-      #guard_threads.each(&:exit!)
+      read_file.close unless read_file.closed?
+      write_file.close unless write_file.closed?
     end
   end
 end

data/lib/secure/version.rb

@@ -1,3 +1,3 @@
 module Secure
-  VERSION = "0.0.5"
+  VERSION = "0.1.0"
 end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: secure
 version: !ruby/object:Gem::Version 
-  hash: 21
+  hash: 27
   prerelease: 
   segments: 
   - 0
+  - 1
   - 0
-  - 5
-  version: 0.0.5
+  version: 0.1.0
 platform: ruby
 authors: 
 - Tejas Dinkar
@@ -15,7 +15,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2011-08-29 00:00:00 +05:30
+date: 2011-08-30 00:00:00 +05:30
 default_executable: 
 dependencies: []
 
@@ -33,8 +33,10 @@ files:
 - Gemfile
 - Rakefile
 - lib/secure.rb
+- lib/secure/child_process.rb
 - lib/secure/errors.rb
 - lib/secure/guard_thread.rb
+- lib/secure/parent_process.rb
 - lib/secure/response.rb
 - lib/secure/runner.rb
 - lib/secure/version.rb