looksee 3.1.1 → 4.0.0

data/ext/mri/2.3.0/method.h

@@ -0,0 +1,213 @@
+/**********************************************************************
+
+  method.h -
+
+  $Author: ko1 $
+  created at: Wed Jul 15 20:02:33 2009
+
+  Copyright (C) 2009 Koichi Sasada
+
+**********************************************************************/
+#ifndef RUBY_METHOD_H
+#define RUBY_METHOD_H 1
+
+#include "internal.h"
+
+#ifndef END_OF_ENUMERATION
+# if defined(__GNUC__) &&! defined(__STRICT_ANSI__)
+#   define END_OF_ENUMERATION(key)
+# else
+#   define END_OF_ENUMERATION(key) END_OF_##key##_PLACEHOLDER = 0
+# endif
+#endif
+
+/* cref */
+
+typedef enum {
+    METHOD_VISI_UNDEF     = 0x00,
+    METHOD_VISI_PUBLIC    = 0x01,
+    METHOD_VISI_PRIVATE   = 0x02,
+    METHOD_VISI_PROTECTED = 0x03,
+
+    METHOD_VISI_MASK = 0x03
+} rb_method_visibility_t;
+
+typedef struct rb_scope_visi_struct {
+    rb_method_visibility_t method_visi : 3;
+    unsigned int module_func : 1;
+} rb_scope_visibility_t;
+
+typedef struct rb_cref_struct {
+    VALUE flags;
+    const VALUE refinements;
+    const VALUE klass;
+    struct rb_cref_struct * const next;
+    const rb_scope_visibility_t scope_visi;
+} rb_cref_t;
+
+/* method data type */
+
+typedef struct rb_method_entry_struct {
+    VALUE flags;
+    const VALUE defined_class;
+    struct rb_method_definition_struct * const def;
+    ID called_id;
+    const VALUE owner;
+} rb_method_entry_t;
+
+typedef struct rb_callable_method_entry_struct { /* same fields with rb_method_entry_t */
+    VALUE flags;
+    const VALUE defined_class;
+    struct rb_method_definition_struct * const def;
+    ID called_id;
+    const VALUE owner;
+} rb_callable_method_entry_t;
+
+#define METHOD_ENTRY_VISI(me)  (rb_method_visibility_t)(((me)->flags & (IMEMO_FL_USER0 | IMEMO_FL_USER1)) >> (IMEMO_FL_USHIFT+0))
+#define METHOD_ENTRY_BASIC(me) (int)                   (((me)->flags & (IMEMO_FL_USER2                 )) >> (IMEMO_FL_USHIFT+2))
+#define METHOD_ENTRY_COMPLEMENTED(me)     ((me)->flags & IMEMO_FL_USER3)
+#define METHOD_ENTRY_COMPLEMENTED_SET(me) ((me)->flags = (me)->flags | IMEMO_FL_USER3)
+
+static inline void
+METHOD_ENTRY_VISI_SET(rb_method_entry_t *me, rb_method_visibility_t visi)
+{
+    VM_ASSERT((int)visi >= 0 && visi <= 3);
+    me->flags = (me->flags & ~(IMEMO_FL_USER0 | IMEMO_FL_USER1)) | (visi << (IMEMO_FL_USHIFT+0));
+}
+static inline void
+METHOD_ENTRY_BASIC_SET(rb_method_entry_t *me, unsigned int basic)
+{
+    VM_ASSERT(basic <= 1);
+    me->flags = (me->flags & ~(IMEMO_FL_USER2                 )) | (basic << (IMEMO_FL_USHIFT+2));
+}
+static inline void
+METHOD_ENTRY_FLAGS_SET(rb_method_entry_t *me, rb_method_visibility_t visi, unsigned int basic)
+{
+    VM_ASSERT((int)visi >= 0 && visi <= 3);
+    VM_ASSERT(basic <= 1);
+    me->flags =
+      (me->flags & ~(IMEMO_FL_USER0|IMEMO_FL_USER1|IMEMO_FL_USER2)) |
+	((visi << (IMEMO_FL_USHIFT+0)) | (basic << (IMEMO_FL_USHIFT+2)));
+}
+static inline void
+METHOD_ENTRY_FLAGS_COPY(rb_method_entry_t *dst, const rb_method_entry_t *src)
+{
+    dst->flags =
+      (dst->flags & ~(IMEMO_FL_USER0|IMEMO_FL_USER1|IMEMO_FL_USER2)) |
+	(src->flags & (IMEMO_FL_USER0|IMEMO_FL_USER1|IMEMO_FL_USER2));
+}
+
+typedef enum {
+    VM_METHOD_TYPE_ISEQ,
+    VM_METHOD_TYPE_CFUNC,
+    VM_METHOD_TYPE_ATTRSET,
+    VM_METHOD_TYPE_IVAR,
+    VM_METHOD_TYPE_BMETHOD,
+    VM_METHOD_TYPE_ZSUPER,
+    VM_METHOD_TYPE_ALIAS,
+    VM_METHOD_TYPE_UNDEF,
+    VM_METHOD_TYPE_NOTIMPLEMENTED,
+    VM_METHOD_TYPE_OPTIMIZED, /* Kernel#send, Proc#call, etc */
+    VM_METHOD_TYPE_MISSING,   /* wrapper for method_missing(id) */
+    VM_METHOD_TYPE_REFINED,
+
+    END_OF_ENUMERATION(VM_METHOD_TYPE)
+} rb_method_type_t;
+
+#ifndef rb_iseq_t
+typedef struct rb_iseq_struct rb_iseq_t;
+#define rb_iseq_t rb_iseq_t
+#endif
+
+typedef struct rb_method_iseq_struct {
+    const rb_iseq_t * const iseqptr;              /* should be separated from iseqval */
+    rb_cref_t * const cref;                       /* should be marked */
+} rb_method_iseq_t; /* check rb_add_method_iseq() when modify the fields */
+
+typedef struct rb_method_cfunc_struct {
+    VALUE (*func)(ANYARGS);
+    VALUE (*invoker)(VALUE (*func)(ANYARGS), VALUE recv, int argc, const VALUE *argv);
+    int argc;
+} rb_method_cfunc_t;
+
+typedef struct rb_method_attr_struct {
+    ID id;
+    const VALUE location; /* should be marked */
+} rb_method_attr_t;
+
+typedef struct rb_method_alias_struct {
+    const struct rb_method_entry_struct * const original_me; /* original_me->klass is original owner */
+} rb_method_alias_t;
+
+typedef struct rb_method_refined_struct {
+    const struct rb_method_entry_struct * const orig_me;
+    const VALUE owner;
+} rb_method_refined_t;
+
+typedef struct rb_method_definition_struct {
+    rb_method_type_t type :  8; /* method type */
+    int alias_count       : 28;
+    int complemented_count: 28;
+
+    union {
+	rb_method_iseq_t iseq;
+	rb_method_cfunc_t cfunc;
+	rb_method_attr_t attr;
+	rb_method_alias_t alias;
+	rb_method_refined_t refined;
+
+	const VALUE proc;                 /* should be marked */
+	enum method_optimized_type {
+	    OPTIMIZED_METHOD_TYPE_SEND,
+	    OPTIMIZED_METHOD_TYPE_CALL,
+
+	    OPTIMIZED_METHOD_TYPE__MAX
+	} optimize_type;
+    } body;
+
+    ID original_id;
+} rb_method_definition_t;
+
+#define UNDEFINED_METHOD_ENTRY_P(me) (!(me) || !(me)->def || (me)->def->type == VM_METHOD_TYPE_UNDEF)
+#define UNDEFINED_REFINED_METHOD_P(def) \
+    ((def)->type == VM_METHOD_TYPE_REFINED && \
+     UNDEFINED_METHOD_ENTRY_P((def)->body.refined.orig_me))
+
+void rb_add_method_cfunc(VALUE klass, ID mid, VALUE (*func)(ANYARGS), int argc, rb_method_visibility_t visi);
+void rb_add_method_iseq(VALUE klass, ID mid, const rb_iseq_t *iseq, rb_cref_t *cref, rb_method_visibility_t visi);
+void rb_add_refined_method_entry(VALUE refined_class, ID mid);
+
+rb_method_entry_t *rb_add_method(VALUE klass, ID mid, rb_method_type_t type, void *option, rb_method_visibility_t visi);
+rb_method_entry_t *rb_method_entry_set(VALUE klass, ID mid, const rb_method_entry_t *, rb_method_visibility_t noex);
+rb_method_entry_t *rb_method_entry_create(ID called_id, VALUE klass, rb_method_visibility_t visi, const rb_method_definition_t *def);
+
+const rb_method_entry_t *rb_method_entry_at(VALUE obj, ID id);
+
+const rb_method_entry_t *rb_method_entry(VALUE klass, ID id);
+const rb_method_entry_t *rb_method_entry_with_refinements(VALUE klass, ID id);
+const rb_method_entry_t *rb_method_entry_without_refinements(VALUE klass, ID id);
+const rb_method_entry_t *rb_resolve_refined_method(VALUE refinements, const rb_method_entry_t *me);
+
+const rb_callable_method_entry_t *rb_callable_method_entry(VALUE klass, ID id);
+const rb_callable_method_entry_t *rb_callable_method_entry_with_refinements(VALUE klass, ID id);
+const rb_callable_method_entry_t *rb_callable_method_entry_without_refinements(VALUE klass, ID id);
+const rb_callable_method_entry_t *rb_resolve_refined_method_callable(VALUE refinements, const rb_callable_method_entry_t *me);
+
+int rb_method_entry_arity(const rb_method_entry_t *me);
+int rb_method_entry_eq(const rb_method_entry_t *m1, const rb_method_entry_t *m2);
+st_index_t rb_hash_method_entry(st_index_t hash, const rb_method_entry_t *me);
+
+VALUE rb_method_entry_location(const rb_method_entry_t *me);
+VALUE rb_mod_method_location(VALUE mod, ID id);
+VALUE rb_obj_method_location(VALUE obj, ID id);
+
+void rb_free_method_entry(const rb_method_entry_t *me);
+void rb_sweep_method_entry(void *vm);
+
+const rb_method_entry_t *rb_method_entry_clone(const rb_method_entry_t *me);
+const rb_callable_method_entry_t *rb_method_entry_complement_defined_class(const rb_method_entry_t *src_me, VALUE defined_class);
+void rb_method_entry_copy(rb_method_entry_t *dst, const rb_method_entry_t *src);
+
+void rb_scope_visibility_set(rb_method_visibility_t);
+
+#endif /* RUBY_METHOD_H */

data/ext/mri/mri.c

@@ -1,309 +1,66 @@
 #include "ruby.h"
 
-#if RUBY_VERSION >= 200
-#  include "method.h"
-#  include "internal.h"
-#elif RUBY_VERSION >= 193
-#  include "ruby/st.h"
-#  ifdef SA_EMPTY
-#    include "internal_falcon.h"
-#    define Looksee_method_table_foreach sa_foreach
-#    define Looksee_method_table_lookup sa_lookup
-#  else
-#    include "internal.h"
-#  endif
-#  include "vm_core.h"
-#  include "method.h"
-#elif RUBY_VERSION >= 192
-#  include "vm_core.h"
-#  include "method.h"
-#  include "ruby/st.h"
-#elif RUBY_VERSION >= 190
-#  include "node-1.9.h"
-#  include "ruby/st.h"
-#else
-#  include "node.h"
-#  include "st.h"
+#if RUBY_VERSION >= 230
+#  define VM_ASSERT(expr) ((void)0)
 #endif
 
+#include "method.h"
+#include "internal.h"
+
 #ifndef Looksee_method_table_foreach
 #  define Looksee_method_table_foreach st_foreach
 #  define Looksee_method_table_lookup st_lookup
 #endif
 
-#if RUBY_VERSION < 187
-#  define RCLASS_IV_TBL(c) (RCLASS(c)->iv_tbl)
-#  define RCLASS_M_TBL(c) (RCLASS(c)->m_tbl)
-#  define RCLASS_SUPER(c) (RCLASS(c)->super)
-#endif
-
-/*
- * Return the internal superclass of this class.
- *
- * This is either a Class or "IClass."  IClasses represent Modules
- * included in the ancestry, and should be treated as opaque objects
- * in ruby space. Convert the IClass to a Module using #iclass_to_module
- * before using it in ruby.
- */
-VALUE Looksee_internal_superclass(VALUE self, VALUE internal_class) {
-  VALUE super = RCLASS_SUPER(internal_class);
-  if (!super)
-    return Qnil;
-  return super;
-}
-
-/*
- * Return the internal class of the given object.
- *
- * This is either the object's singleton class, if it exists, or the
- * object's birth class.
- */
-VALUE Looksee_internal_class(VALUE self, VALUE object) {
-  return CLASS_OF(object);
-}
-
-#if RUBY_VERSION >= 192
-
-#  define VISIBILITY_TYPE rb_method_flag_t
-
-typedef struct add_method_if_matching_arg {
-  VALUE names;
-  VISIBILITY_TYPE visibility;
-} add_method_if_matching_arg_t;
-
-static int add_method_if_matching(ID method_name, rb_method_entry_t *me, add_method_if_matching_arg_t *arg) {
-#  ifdef ID_ALLOCATOR
-  if (method_name == ID_ALLOCATOR)
-    return ST_CONTINUE;
-#  endif
-
-  if (UNDEFINED_METHOD_ENTRY_P(me))
-    return ST_CONTINUE;
-
-  if ((me->flag & NOEX_MASK) == arg->visibility)
-    rb_ary_push(arg->names, ID2SYM(method_name));
-
-  return ST_CONTINUE;
-}
-
 static int add_method_if_undefined(ID method_name, rb_method_entry_t *me, VALUE *names) {
-#  ifdef ID_ALLOCATOR
+#ifdef ID_ALLOCATOR
   /* The allocator can be undefined with rb_undef_alloc_func, e.g. Struct. */
   if (method_name == ID_ALLOCATOR)
     return ST_CONTINUE;
-#  endif
+#endif
 
   if (UNDEFINED_METHOD_ENTRY_P(me))
     rb_ary_push(*names, ID2SYM(method_name));
   return ST_CONTINUE;
 }
 
-#else
-
-#  if RUBY_VERSION >= 190
-#    define VISIBILITY(node) ((node)->nd_body->nd_noex & NOEX_MASK)
-#  else
-#    define VISIBILITY(node) ((node)->nd_noex & NOEX_MASK)
-#  endif
-
-#  define VISIBILITY_TYPE unsigned long
-
-typedef struct add_method_if_matching_arg {
-  VALUE names;
-  VISIBILITY_TYPE visibility;
-} add_method_if_matching_arg_t;
-
-static int add_method_if_matching(ID method_name, NODE *body, add_method_if_matching_arg_t *arg) {
-#  ifdef ID_ALLOCATOR
-  /* This entry is for the internal allocator function. */
-  if (method_name == ID_ALLOCATOR)
-    return ST_CONTINUE;
-#  endif
-
-  /* Module#undef_method:
-   *   * sets body->nd_body to NULL in ruby <= 1.8
-   *   * sets body to NULL in ruby >= 1.9
-   */
-  if (!body || !body->nd_body)
-    return ST_CONTINUE;
-
-  if (VISIBILITY(body) == arg->visibility)
-    rb_ary_push(arg->names, ID2SYM(method_name));
-  return ST_CONTINUE;
-}
-
-static int add_method_if_undefined(ID method_name, NODE *body, VALUE *names) {
-#  ifdef ID_ALLOCATOR
-  /* The allocator can be undefined with rb_undef_alloc_func, e.g. Struct. */
-  if (method_name == ID_ALLOCATOR)
-    return ST_CONTINUE;
-#  endif
-
-  if (!body || !body->nd_body)
-    rb_ary_push(*names, ID2SYM(method_name));
-  return ST_CONTINUE;
-}
-
-#endif
-
-static VALUE internal_instance_methods(VALUE klass, VISIBILITY_TYPE visibility) {
-  add_method_if_matching_arg_t arg;
-  arg.names = rb_ary_new();
-  arg.visibility = visibility;
-
-  Looksee_method_table_foreach(RCLASS_M_TBL(klass), add_method_if_matching, (st_data_t)&arg);
-  return arg.names;
-}
-
-/*
- * Return the list of public instance methods (as Symbols) of the
- * given internal class.
- */
-VALUE Looksee_internal_public_instance_methods(VALUE self, VALUE klass) {
-  return internal_instance_methods(klass, NOEX_PUBLIC);
-}
-
-/*
- * Return the list of protected instance methods (as Symbols) of the
- * given internal class.
- */
-VALUE Looksee_internal_protected_instance_methods(VALUE self, VALUE klass) {
-  return internal_instance_methods(klass, NOEX_PROTECTED);
-}
-
-/*
- * Return the list of private instance methods (as Symbols) of the
- * given internal class.
- */
-VALUE Looksee_internal_private_instance_methods(VALUE self, VALUE klass) {
-  return internal_instance_methods(klass, NOEX_PRIVATE);
-}
-
 /*
  * Return the list of undefined instance methods (as Symbols) of the
  * given internal class.
  */
 VALUE Looksee_internal_undefined_instance_methods(VALUE self, VALUE klass) {
+#if RUBY_VERSION >= 230
+  static int warned = 0;
+  if (!warned) {
+    rb_warn("Looksee cannot display undef'd methods on MRI 2.3");
+    warned = 1;
+  }
+  return rb_ary_new();
+#else
   VALUE names = rb_ary_new();
+  if (RCLASS_ORIGIN(klass) != klass)
+    klass = RCLASS_ORIGIN(klass);
   Looksee_method_table_foreach(RCLASS_M_TBL(klass), add_method_if_undefined, (st_data_t)&names);
   return names;
+#endif
 }
 
-/*
- * Return true if the given object is an included class or origin class, false
- * otherwise.
- */
-VALUE Looksee_included_class_p(VALUE self, VALUE object) {
-  return !SPECIAL_CONST_P(object) && BUILTIN_TYPE(object) == T_ICLASS ?
-    Qtrue : Qfalse;
-}
-
-VALUE Looksee_singleton_class_p(VALUE self, VALUE object) {
-  return BUILTIN_TYPE(object) == T_CLASS && FL_TEST(object, FL_SINGLETON) ? Qtrue : Qfalse;
-}
-
-VALUE Looksee_singleton_instance(VALUE self, VALUE singleton_class) {
-  if (BUILTIN_TYPE(singleton_class) == T_CLASS && FL_TEST(singleton_class, FL_SINGLETON)) {
+VALUE Looksee_singleton_instance(VALUE self, VALUE klass) {
+  if (!SPECIAL_CONST_P(klass) && BUILTIN_TYPE(klass) == T_CLASS && FL_TEST(klass, FL_SINGLETON)) {
     VALUE object;
-    if (!Looksee_method_table_lookup(RCLASS_IV_TBL(singleton_class), rb_intern("__attached__"), (st_data_t *)&object))
+    if (!Looksee_method_table_lookup(RCLASS_IV_TBL(klass), rb_intern("__attached__"), (st_data_t *)&object))
       rb_raise(rb_eRuntimeError, "[looksee bug] can't find singleton object");
     return object;
   } else {
-    rb_raise(rb_eTypeError, "expected singleton class, got %s", rb_obj_classname(singleton_class));
-  }
-}
-
-VALUE Looksee_real_module(VALUE self, VALUE module_or_included_class) {
-  if (BUILTIN_TYPE(module_or_included_class) == T_ICLASS)
-    return RBASIC(module_or_included_class)->klass;
-  else
-    return module_or_included_class;
-}
-
-VALUE Looksee_module_name(VALUE self, VALUE module) {
-  if (BUILTIN_TYPE(module) == T_CLASS || BUILTIN_TYPE(module) == T_MODULE) {
-    VALUE name = rb_mod_name(module);
-    return name == Qnil ? rb_str_new2("") : name;
-  } else if (BUILTIN_TYPE(module) == T_ICLASS) {
-    VALUE wrapped = RBASIC(module)->klass;
-    VALUE name = Looksee_module_name(self, wrapped);
-    if (BUILTIN_TYPE(wrapped) == T_CLASS)
-      name = rb_str_cat2(name, " (origin)");
-    return name;
-  } else {
-    rb_raise(rb_eTypeError, "expected module, got %s", rb_obj_classname(module));
-  }
-}
-
-#if RUBY_VERSION < 190
-
-#include "env-1.8.h"
-#include "eval_c-1.8.h"
-
-/*
- * Return the source file and line number of the given object and method.
- */
-VALUE Looksee_source_location(VALUE self, VALUE unbound_method) {
-  if (!rb_obj_is_kind_of(unbound_method, rb_cUnboundMethod))
-    rb_raise(rb_eTypeError, "expected UnboundMethod, got %s", rb_obj_classname(unbound_method));
-
-  struct METHOD *method;
-  Data_Get_Struct(unbound_method, struct METHOD, method);
-
-  NODE *node;
-  switch (nd_type(method->body)) {
-    // Can't be a FBODY or ZSUPER.
-  case NODE_SCOPE:
-    node = method->body->nd_defn;
-    break;
-  case NODE_BMETHOD:
-    {
-      struct BLOCK *block;
-      Data_Get_Struct(method->body->nd_orig, struct BLOCK, block);
-      (node = block->frame.node) || (node = block->body);
-      // Proc#to_s suggests this may be NULL sometimes.
-      if (!node)
-        return Qnil;
-    }
-    break;
-  case NODE_DMETHOD:
-    {
-      struct METHOD *original_method;
-      NODE *body = method->body;
-      Data_Get_Struct(body->nd_orig, struct METHOD, original_method);
-      node = original_method->body->nd_defn;
-    }
-    break;
-  default:
     return Qnil;
   }
-  VALUE file = rb_str_new2(node->nd_file);
-  VALUE line = INT2NUM(nd_line(node));
-  VALUE location = rb_ary_new2(2);
-  rb_ary_store(location, 0, file);
-  rb_ary_store(location, 1, line);
-  return location;
 }
 
-#endif
-
 void Init_mri(void) {
   VALUE mLooksee = rb_const_get(rb_cObject, rb_intern("Looksee"));
   VALUE mAdapter = rb_const_get(mLooksee, rb_intern("Adapter"));
   VALUE mBase = rb_const_get(mAdapter, rb_intern("Base"));
   VALUE mMRI = rb_define_class_under(mAdapter, "MRI", mBase);
-  rb_define_method(mMRI, "internal_superclass", Looksee_internal_superclass, 1);
-  rb_define_method(mMRI, "internal_class", Looksee_internal_class, 1);
-  rb_define_method(mMRI, "internal_public_instance_methods", Looksee_internal_public_instance_methods, 1);
-  rb_define_method(mMRI, "internal_protected_instance_methods", Looksee_internal_protected_instance_methods, 1);
-  rb_define_method(mMRI, "internal_private_instance_methods", Looksee_internal_private_instance_methods, 1);
   rb_define_method(mMRI, "internal_undefined_instance_methods", Looksee_internal_undefined_instance_methods, 1);
-  rb_define_method(mMRI, "included_class?", Looksee_included_class_p, 1);
-  rb_define_method(mMRI, "singleton_class?", Looksee_singleton_class_p, 1);
   rb_define_method(mMRI, "singleton_instance", Looksee_singleton_instance, 1);
-  rb_define_method(mMRI, "real_module", Looksee_real_module, 1);
-  rb_define_method(mMRI, "module_name", Looksee_module_name, 1);
-#if RUBY_VERSION < 190
-  rb_define_method(mMRI, "source_location", Looksee_source_location, 1);
-#endif
 }