objspace_helpers 0.0.2 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: d0175d9d52d2d5fdc414e9b5e9cda0475dcf33bd
-  data.tar.gz: 5c92f3b27147ecef0e63a55332a3c72262d78f68
+  metadata.gz: cbb176043e7fcf26f81faec74c26725353c914aa
+  data.tar.gz: 3dd9ba529a9b4fbc949cecf0f085c542858d46c6
 SHA512:
-  metadata.gz: 4a31190183b4d4d8b1d348720dd4a974c3fed2410184dcaeac3ffdef6263785cde179398581cb60b2b518e723470a82cf5926193270c5cdbd25363bcbd893c60
-  data.tar.gz: ec1a6ff89ac9d196d12eb791ab8abf0f70b993e7d5b552b6f51691a25b663e63999173d0930daa2b6a2f12ce622f542186670fb2b9cc6fe3ed80093945acd9bf
+  metadata.gz: 14b5bb3c4aed3a80f68c7873cf681cb6e0df30452ccdc2bf4908649aed509338ef06b7be47aef212798e4e4874e478114ea2b7af01f461d601409313c5f38312
+  data.tar.gz: 60219c051f2596767b4aa28b9a5178a98255a6ec91e9e7862e87014926658b724e07332853a0ab0c3038e5a78c29c7237ffdfa86000501b0a2248607e44dc15e

data/ext/objspace_helpers/id2ref.c

@@ -0,0 +1,46 @@
+/* Note (LukasFittl): Copy of ObjectSpace._id2ref from Ruby 2.1 source,
+ * directly callable without the VM overhead, and more foregiving for internal objects.
+ */
+
+#include <id2ref.h>
+
+#if SIZEOF_LONG == SIZEOF_VOIDP
+# define nonspecial_obj_id(obj) (VALUE)((SIGNED_VALUE)(obj)|FIXNUM_FLAG)
+# define obj_id_to_ref(objid) ((objid) ^ FIXNUM_FLAG) /* unset FIXNUM_FLAG */
+#elif SIZEOF_LONG_LONG == SIZEOF_VOIDP
+# define nonspecial_obj_id(obj) LL2NUM((SIGNED_VALUE)(obj) / 2)
+# define obj_id_to_ref(objid) (FIXNUM_P(objid) ? \
+   ((objid) ^ FIXNUM_FLAG) : (NUM2PTR(objid) << 1))
+#else
+# error not supported
+#endif
+
+VALUE
+oh_id2ref(VALUE self, VALUE objid)
+{
+#if SIZEOF_LONG == SIZEOF_VOIDP
+#define NUM2PTR(x) NUM2ULONG(x)
+#elif SIZEOF_LONG_LONG == SIZEOF_VOIDP
+#define NUM2PTR(x) NUM2ULL(x)
+#endif
+  VALUE ptr;
+  void *p0;
+
+  ptr = NUM2PTR(objid);
+  p0 = (void *)ptr;
+
+  if (ptr == Qtrue) return Qtrue;
+  if (ptr == Qfalse) return Qfalse;
+  if (ptr == Qnil) return Qnil;
+  if (FIXNUM_P(ptr)) return (VALUE)ptr;
+  if (FLONUM_P(ptr)) return (VALUE)ptr;
+  ptr = obj_id_to_ref(objid);
+
+  if ((ptr % rb_intern("RVALUE_SIZE")) == (4 << 2)) {
+    ID symid = ptr / rb_intern("RVALUE_SIZE");
+    if (rb_id2name(symid) == 0) return Qundef;
+    return ID2SYM(symid);
+  }
+
+  return (VALUE)ptr;
+}

data/ext/objspace_helpers/id2ref.h

@@ -0,0 +1,9 @@
+#ifndef RUBY_ID2REF
+#define RUBY_ID2REF
+
+#include <ruby.h>
+#include <ruby_private/gc.h>
+
+VALUE oh_id2ref(VALUE self, VALUE objid);
+
+#endif

data/lib/objspace_helpers/leaks.rb

@@ -0,0 +1,55 @@
+class ObjspaceHelpers
+  class << self
+    def find_leaks(trace: false, &block)
+      ObjectSpace::trace_object_allocations_start if trace
+
+      objs_before = _dump_addresses
+
+      block.call
+
+      # Run GC twice to try to get rid of as much false positives as possible
+      ObjectSpace.garbage_collect
+      ObjectSpace.garbage_collect
+
+      objs_after = _dump_addresses
+
+      ObjectSpace::trace_object_allocations_stop if trace
+
+      leaked_addresses = objs_after - objs_before - [objs_after.object_id]
+
+      GC.disable
+
+      # Note (LukasFittl): 2x speedup if we move this to native code
+      referenced_by = {}
+      _addresses_to_references(objs_after).each do |addr, references|
+        (references & leaked_addresses).each do |ref|
+          referenced_by[ref] ||= []
+          referenced_by[ref] << addr
+        end
+      end
+
+      GC.enable
+
+      TrackedObject.wrap(leaked_addresses, referenced_by)
+    end
+
+    def find_leak_sources(trace: false, &block)
+      leaks = find_leaks(trace: trace, &block)
+      leaks_by_source = {}
+      top_level_leaks = []
+
+      leaks.each do |leak|
+        if leak.referenced_by.empty?
+          top_level_leaks << leak
+        else
+          leak.referenced_by.each do |source|
+            leaks_by_source[source] ||= []
+            leaks_by_source[source] << leak
+          end
+        end
+      end
+
+      [top_level_leaks, leaks_by_source]
+    end
+  end
+end

data/lib/objspace_helpers/tracked_object.rb

@@ -0,0 +1,31 @@
+class ObjspaceHelpers::TrackedObject
+  def self.wrap(addresses, referenced_by = [])
+    Array.new(addresses || []).map do |address|
+      new(address, wrap(referenced_by[address]))
+    end
+  end
+
+  def self.for_object(obj)
+    new ObjspaceHelpers._address_of_obj(obj)
+  end
+
+  attr_reader :address
+  attr_reader :referenced_by
+
+  def initialize(address, referenced_by = [])
+    @address = address
+    @referenced_by = referenced_by
+  end
+
+  def info
+    ObjspaceHelpers._addresses_to_info([@address]).values.first
+  end
+
+  def klass
+    self.class.new(info['class'])
+  end
+
+  def dereference
+    ObjspaceHelpers._id2ref(@address)
+  end
+end

data/lib/objspace_helpers/version.rb

@@ -1,3 +1,3 @@
 class ObjspaceHelpers
-  VERSION = '0.0.2'
+  VERSION = '0.0.3'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: objspace_helpers
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors:
 - Lukas Fittl
@@ -62,6 +62,8 @@ files:
 - LICENSE
 - Rakefile
 - ext/objspace_helpers/extconf.rb
+- ext/objspace_helpers/id2ref.c
+- ext/objspace_helpers/id2ref.h
 - ext/objspace_helpers/objspace_helpers.c
 - ext/objspace_helpers/objspace_helpers.h
 - ext/objspace_helpers/objspace_info.c
@@ -71,6 +73,8 @@ files:
 - ext/objspace_helpers/ruby_private/objspace.h
 - lib/objspace_helpers.rb
 - lib/objspace_helpers/helpers.rb
+- lib/objspace_helpers/leaks.rb
+- lib/objspace_helpers/tracked_object.rb
 - lib/objspace_helpers/version.rb
 homepage: http://github.com/lfittl/objspace_helpers
 licenses: