rex-core 0.1.1 → 0.1.2

data/lib/rex/sync/ref.rb

@@ -0,0 +1,58 @@
+# -*- coding: binary -*-
+require 'thread'
+
+module Rex
+
+###
+#
+# This module provides a uniform reference counted interface for classes to
+# use.
+#
+###
+module Ref
+
+  #
+  # Initializes the reference count to one.
+  #
+  def refinit
+    @_references       = 1
+    @_references_mutex = Mutex.new
+
+    self
+  end
+
+  #
+  # Increments the total number of references.
+  #
+  def ref
+    @_references_mutex.synchronize {
+      @_references += 1
+    }
+
+    self
+  end
+
+  #
+  # Decrements the total number of references.  If the reference count
+  # reaches zero, true is returned.  Otherwise, false is returned.
+  #
+  def deref
+    @_references_mutex.synchronize {
+      if ((@_references -= 1) == 0)
+        cleanup
+
+        true
+      else
+        false
+      end
+    }
+  end
+
+  #
+  # Called to clean up resources once the ref count drops to zero.
+  #
+  def cleanup
+  end
+
+end
+end

data/lib/rex/sync/thread_safe.rb

@@ -0,0 +1,83 @@
+# -*- coding: binary -*-
+require 'timeout'
+
+module Rex
+
+###
+#
+# This module provides a set of methods for performing various blocking
+# operations in a manner that is compatible with ruby style threads.
+#
+###
+module ThreadSafe
+
+  DefaultCycle = 0.2
+
+  #
+  # Wraps calls to select with a lower timeout period and does the
+  # calculations to walk down to zero timeout.  This has a little room for
+  # improvement in that it should probably check how much time actually
+  # elapsed during the select call considering ruby threading wont be exactly
+  # accurate perhaps.
+  #
+  def self.select(rfd = nil, wfd = nil, efd = nil, t = nil)
+    left = t
+
+    # Immediately raise a StreamClosedError if the socket was closed. This
+    # prevents a bad fd from being passed downstream and solves an issue
+    # with Ruby on Windows.
+    rfd.each { |fd| raise StreamClosedError.new(fd) if (fd.closed?)	} if rfd
+
+    begin
+      orig_size = rfd.length if (rfd)
+
+      # Poll the set supplied to us at least once.
+      begin
+        rv = ::IO.select(rfd, wfd, efd, DefaultCycle)
+      rescue ::IOError, ::Errno::EBADF, ::Errno::ENOTSOCK
+        # If a stream was detected as being closed, re-raise the error as
+        # a StreamClosedError with the specific file descriptor that was
+        # detected as being closed.  This is to better handle the case of
+        # a closed socket being detected so that it can be cleaned up and
+        # removed.
+        rfd.each { |fd| raise StreamClosedError.new(fd) if (fd.closed?)	} if rfd
+
+        # If the original rfd length is not the same as the current
+        # length, then the list may have been altered and as such may not
+        # contain the socket that caused the IOError.  This is a bad way
+        # to do this since it's possible that the array length could be
+        # back to the size that it was originally and yet have had the
+        # socket that caused the IOError to be removed.
+        return nil if (rfd and rfd.length != orig_size)
+
+        # Re-raise the exception since we didn't handle it here.
+        raise $!
+#			rescue ::Exception => e
+#			$stderr.puts "SELECT(#{t}) #{[rfd,wfd,efd].inspect} #{e.class} #{e} #{e.backtrace}"
+      end
+
+      return rv if (rv)
+
+      # Decrement the amount of time left by the polling cycle
+      left -= DefaultCycle if (left)
+
+      # Keep chugging until we run out of time, if time was supplied.
+    end while ((left == nil) or (left > 0))
+
+    # Nothin.
+    nil
+  end
+
+  #
+  # Simulates a sleep operation by selecting on nil until a timeout period
+  # expires.
+  #
+  def self.sleep(seconds=nil)
+    self.select(nil, nil, nil, seconds)
+
+    seconds
+  end
+
+end
+
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: rex-core
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors:
 - David Maloney
@@ -85,7 +85,7 @@ cert_chain:
   2SpuQH+SWteq3NXkAmFEEqvLJQ4sbptZt8OP8ghL3pVAvZNFmww/YVszSkShSzcg
   QdihYCSEL2drS2cFd50jBeq71sxUtxbv82DUa2b+
   -----END CERTIFICATE-----
-date: 2016-08-15 00:00:00.000000000 Z
+date: 2016-08-26 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -148,7 +148,19 @@ files:
 - lib/rex/compat.rb
 - lib/rex/core.rb
 - lib/rex/core/version.rb
+- lib/rex/exceptions.rb
 - lib/rex/file.rb
+- lib/rex/io/datagram_abstraction.rb
+- lib/rex/io/ring_buffer.rb
+- lib/rex/io/socket_abstraction.rb
+- lib/rex/io/stream.rb
+- lib/rex/io/stream_abstraction.rb
+- lib/rex/io/stream_server.rb
+- lib/rex/sync.rb
+- lib/rex/sync/event.rb
+- lib/rex/sync/read_write_lock.rb
+- lib/rex/sync/ref.rb
+- lib/rex/sync/thread_safe.rb
 - rex-core.gemspec
 homepage: https://github.com/rapid7/rex-core
 licenses: []