iseq_collector 0.2.0 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 65748299ec8c1bc18d1b2331297771eeecf8f3dd
-  data.tar.gz: d29f03e81016604d187f8353cf785e20564bcdde
+  metadata.gz: 361f51f935c26f922c5470d8043546c1f7101fb2
+  data.tar.gz: 567c243aabeb8c264d0bd4b108ca376798f62924
 SHA512:
-  metadata.gz: 42cc8cebb4387e286ba8aa17ea2b130b225e45405d913d28c698452b109e43c7030ef5e095e5e7c56078b68829177b95f13cb2975e518e4e48981953fffd0393
-  data.tar.gz: 12acb6034afd13b57f49059769a1f19eb31165bb4da0ae9af1c9bb967cc2ed69ec961a0af6c99a1ec7afea8366b7b661f55d32b0530816e6e4ceecd0377025c6
+  metadata.gz: fce8a1e216b67a9bc7a2e70c40709162615625864e048c0719b391b59bdc7c8af150ebaff6c3fe94dad1efe0cce39fb0a88a447afb574caffe26abd7e5daa8d1
+  data.tar.gz: 571af44fa45469d851950983412d8f4886d35a8c4cd181414a279746500b19032b12458948a49dc6ee3a42d965fbd60fbaf0c878e31048c017aed3dee150a2eb

data/ext/iseq_collector/iseq_collector.c

@@ -0,0 +1,101 @@
+#include <ruby/ruby.h>
+
+#ifdef T_IMEMO
+
+VALUE rb_iseqw_new(VALUE v);
+void rb_objspace_each_objects(
+    int (*callback)(void *start, void *end, size_t stride, void *data),
+    void *data);
+size_t rb_obj_memsize_of(VALUE);
+
+enum imemo_type {
+    imemo_none = 0,
+    imemo_cref = 1,
+    imemo_svar = 2,
+    imemo_throw_data = 3,
+    imemo_ifunc = 4,
+    imemo_memo = 5,
+    imemo_ment = 6,
+    imemo_iseq = 7,
+    imemo_mask = 0x07
+};
+
+static inline enum imemo_type
+imemo_type(VALUE imemo)
+{
+    return (RBASIC(imemo)->flags >> FL_USHIFT) & imemo_mask;
+}
+
+static inline int
+rb_obj_is_iseq(VALUE iseq)
+{
+    return RB_TYPE_P(iseq, T_IMEMO) && imemo_type(iseq) == imemo_iseq;
+}
+
+struct iseq_i_data {
+    void (*func)(VALUE v, void *data);
+    void *data;
+};
+
+int
+iseq_i(void *vstart, void *vend, size_t stride, void *ptr)
+{
+    VALUE v;
+    struct iseq_i_data *data = (struct iseq_i_data *)ptr;
+
+    for (v = (VALUE)vstart; v != (VALUE)vend; v += stride) {
+	if (RBASIC(v)->flags) {
+	    switch (BUILTIN_TYPE(v)) {
+	      case T_IMEMO:
+		if (rb_obj_is_iseq(v)) {
+		    data->func(v, data->data);
+		}
+		continue;
+	      default:
+		continue;
+	    }
+	}
+    }
+
+    return 0;
+}
+
+static void
+each_iseq_i(VALUE v, void *ptr)
+{
+    rb_yield(rb_iseqw_new(v));
+}
+
+static VALUE
+each_iseq(VALUE self)
+{
+    struct iseq_i_data data = {each_iseq_i, NULL};
+    rb_objspace_each_objects(iseq_i, &data);
+    return Qnil;
+}
+
+static void
+count_iseq_i(VALUE v, void *ptr)
+{
+    size_t *sizep = (size_t *)ptr;
+    *sizep += 1;
+}
+
+static VALUE
+count_iseq(VALUE self)
+{
+    size_t size = 0;
+    struct iseq_i_data data = {count_iseq_i, &size};
+    rb_objspace_each_objects(iseq_i, &data);
+    return SIZET2NUM(size);
+}
+
+void
+Init_iseq_collector(void)
+{
+    VALUE rb_mObjSpace = rb_const_get(rb_cObject, rb_intern("ObjectSpace"));
+    rb_define_singleton_method(rb_mObjSpace, "each_iseq", each_iseq, 0);
+    rb_define_singleton_method(rb_mObjSpace, "count_iseq", count_iseq, 0);
+}
+
+#endif

data/lib/iseq_collector/version.rb

@@ -1,3 +1,3 @@
 module IseqCollector
-  VERSION = "0.2.0"
+  VERSION = "0.3.0"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: iseq_collector
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.3.0
 platform: ruby
 authors:
 - Koichi Sasada
@@ -83,6 +83,7 @@ files:
 - bin/console
 - bin/setup
 - ext/iseq_collector/extconf.rb
+- ext/iseq_collector/iseq_collector.c
 - iseq_collector.gemspec
 - lib/iseq_collector.rb
 - lib/iseq_collector/version.rb