ruby-prof 1.5.0-x64-mingw-ucrt → 1.6.2-x64-mingw-ucrt

data/ext/ruby_prof/rp_call_tree.h

@@ -1,44 +1,47 @@
-/* Copyright (C) 2005-2019 Shugo Maeda <shugo@ruby-lang.org> and Charlie Savage <cfis@savagexi.com>
-   Please see the LICENSE file for copyright and distribution information */
-
-#ifndef __RP_CALL_TREE_H__
-#define __RP_CALL_TREE_H__
-
-#include "ruby_prof.h"
-#include "rp_measurement.h"
-#include "rp_method.h"
-
-extern VALUE cRpCallTree;
-
-/* Callers and callee information for a method. */
-typedef struct prof_call_tree_t
-{
-    prof_owner_t owner;
-    prof_method_t* method;
-    struct prof_call_tree_t* parent;
-    st_table* children;             /* Call infos that this call info calls */
-    prof_measurement_t* measurement;
-    VALUE object;
-
-    int visits;                             /* Current visits on the stack */
-
-    unsigned int source_line;
-    VALUE source_file;
-} prof_call_tree_t;
-
-prof_call_tree_t* prof_call_tree_create(prof_method_t* method, prof_call_tree_t* parent, VALUE source_file, int source_line);
-prof_call_tree_t* prof_call_tree_copy(prof_call_tree_t* other);
-void prof_call_tree_merge_internal(prof_call_tree_t* destination, prof_call_tree_t* other);
-void prof_call_tree_mark(void* data);
-prof_call_tree_t* call_tree_table_lookup(st_table* table, st_data_t key);
-
-void prof_call_tree_add_parent(prof_call_tree_t* self, prof_call_tree_t* parent);
-void prof_call_tree_add_child(prof_call_tree_t* self, prof_call_tree_t* child);
-
-uint32_t prof_call_figure_depth(prof_call_tree_t* call_tree_data);
-prof_call_tree_t* prof_get_call_tree(VALUE self);
-VALUE prof_call_tree_wrap(prof_call_tree_t* call_tree);
-void prof_call_tree_free(prof_call_tree_t* call_tree);
-void rp_init_call_tree();
-
-#endif //__RP_CALL_TREE_H__
+/* Copyright (C) 2005-2019 Shugo Maeda <shugo@ruby-lang.org> and Charlie Savage <cfis@savagexi.com>
+   Please see the LICENSE file for copyright and distribution information */
+
+#ifndef __RP_CALL_TREE_H__
+#define __RP_CALL_TREE_H__
+
+#include "ruby_prof.h"
+#include "rp_measurement.h"
+#include "rp_method.h"
+
+extern VALUE cRpCallTree;
+
+/* Callers and callee information for a method. */
+typedef struct prof_call_tree_t
+{
+    prof_owner_t owner;
+    prof_method_t* method;
+    struct prof_call_tree_t* parent;
+    st_table* children;             /* Call infos that this call info calls */
+    prof_measurement_t* measurement;
+    VALUE object;
+
+    int visits;                             /* Current visits on the stack */
+
+    unsigned int source_line;
+    VALUE source_file;
+} prof_call_tree_t;
+
+prof_call_tree_t* prof_call_tree_create(prof_method_t* method, prof_call_tree_t* parent, VALUE source_file, int source_line);
+prof_call_tree_t* prof_call_tree_copy(prof_call_tree_t* other);
+void prof_call_tree_merge_internal(prof_call_tree_t* destination, prof_call_tree_t* other, st_table* method_table);
+void prof_call_tree_mark(void* data);
+prof_call_tree_t* call_tree_table_lookup(st_table* table, st_data_t key);
+
+void prof_call_tree_add_parent(prof_call_tree_t* self, prof_call_tree_t* parent);
+void prof_call_tree_add_child(prof_call_tree_t* self, prof_call_tree_t* child);
+
+uint32_t prof_call_tree_figure_depth(prof_call_tree_t* call_tree);
+VALUE prof_call_tree_methods(prof_call_tree_t* call_tree);
+
+prof_call_tree_t* prof_get_call_tree(VALUE self);
+VALUE prof_call_tree_wrap(prof_call_tree_t* call_tree);
+void prof_call_tree_free(prof_call_tree_t* call_tree);
+
+void rp_init_call_tree();
+
+#endif //__RP_CALL_TREE_H__

data/ext/ruby_prof/rp_call_trees.c

@@ -170,7 +170,7 @@ VALUE prof_call_trees_min_depth(VALUE self)
     prof_call_trees_t* call_trees = prof_get_call_trees(self);
     for (prof_call_tree_t** p_call_tree = call_trees->start; p_call_tree < call_trees->ptr; p_call_tree++)
     {
-        unsigned int call_tree_depth = prof_call_figure_depth(*p_call_tree);
+        unsigned int call_tree_depth = prof_call_tree_figure_depth(*p_call_tree);
         if (call_tree_depth < depth)
             depth = call_tree_depth;
     }

data/ext/ruby_prof/rp_measurement.c

@@ -102,10 +102,16 @@ void prof_measurement_mark(void* data)
 {
     if (!data) return;
 
-    prof_measurement_t* measurement_data = (prof_measurement_t*)data;
+    prof_measurement_t* measurement = (prof_measurement_t*)data;
 
-    if (measurement_data->object != Qnil)
-        rb_gc_mark(measurement_data->object);
+    if (measurement->object != Qnil)
+        rb_gc_mark_movable(measurement->object);
+}
+
+void prof_measurement_compact(void* data)
+{
+    prof_measurement_t* measurement = (prof_measurement_t*)data;
+    measurement->object = rb_gc_location(measurement->object);
 }
 
 void prof_measurement_free(prof_measurement_t* measurement)
@@ -155,6 +161,7 @@ static const rb_data_type_t measurement_type =
         .dmark = prof_measurement_mark,
         .dfree = prof_measurement_ruby_gc_free,
         .dsize = prof_measurement_size,
+        .dcompact = prof_measurement_compact
     },
     .data = NULL,
     .flags = RUBY_TYPED_FREE_IMMEDIATELY

data/ext/ruby_prof/rp_method.c

@@ -22,31 +22,38 @@ VALUE resolve_klass(VALUE klass, unsigned int* klass_flags)
            figure out what it is attached to.*/
         VALUE attached = rb_iv_get(klass, "__attached__");
 
-        /* Is this a singleton class acting as a metaclass? */
-        if (BUILTIN_TYPE(attached) == T_CLASS)
+        switch (BUILTIN_TYPE(attached))
         {
-            *klass_flags |= kClassSingleton;
-            result = attached;
-        }
-        /* Is this for singleton methods on a module? */
-        else if (BUILTIN_TYPE(attached) == T_MODULE)
-        {
-            *klass_flags |= kModuleSingleton;
-            result = attached;
-        }
-        /* Is this for singleton methods on an object? */
-        else if (BUILTIN_TYPE(attached) == T_OBJECT)
-        {
-            *klass_flags |= kObjectSingleton;
-            result = rb_class_superclass(klass);
-        }
-        /* Ok, this could be other things like an array put onto
-           a singleton object (yeah, it happens, see the singleton
-           objects test case). */
-        else
-        {
-            *klass_flags |= kOtherSingleton;
-            result = klass;
+            /* Is this a singleton class acting as a metaclass? */
+            case T_CLASS:
+            {
+                *klass_flags |= kClassSingleton;
+                result = attached;
+                break;
+            }
+            /* Is this for singleton methods on a module? */
+            case T_MODULE:
+            {
+                *klass_flags |= kModuleSingleton;
+                result = attached;
+                break;
+            }
+            /* Is this for singleton methods on an object? */
+            case T_OBJECT:
+            {
+                *klass_flags |= kObjectSingleton;
+                result = rb_class_superclass(klass);
+                break;
+            }
+            /* Ok, this could be other things like an array put onto
+               a singleton object (yeah, it happens, see the singleton
+               objects test case). */
+            default:
+            {
+                *klass_flags |= kOtherSingleton;
+                result = klass;
+                break;
+            }
         }
     }
     /* Is this an include for a module?  If so get the actual
@@ -56,7 +63,7 @@ VALUE resolve_klass(VALUE klass, unsigned int* klass_flags)
     {
         unsigned int dummy;
         *klass_flags |= kModuleIncludee;
-        result = resolve_klass(RBASIC(klass)->klass, &dummy);
+        result = resolve_klass(RBASIC_CLASS(klass), &dummy);
     }
     return result;
 }
@@ -94,7 +101,7 @@ st_data_t method_key(VALUE klass, VALUE msym)
     }
     else if (BUILTIN_TYPE(klass) == T_ICLASS)
     {
-        resolved_klass = RBASIC(klass)->klass;
+        resolved_klass = RBASIC_CLASS(klass);
     }
 
     st_data_t hash = rb_hash_start(0);
@@ -105,39 +112,6 @@ st_data_t method_key(VALUE klass, VALUE msym)
     return hash;
 }
 
-/* ======   Allocation Table  ====== */
-st_table* allocations_table_create()
-{
-    return rb_st_init_numtable();
-}
-
-static int allocations_table_free_iterator(st_data_t key, st_data_t value, st_data_t dummy)
-{
-    prof_allocation_free((prof_allocation_t*)value);
-    return ST_CONTINUE;
-}
-
-static int prof_method_collect_allocations(st_data_t key, st_data_t value, st_data_t result)
-{
-    prof_allocation_t* allocation = (prof_allocation_t*)value;
-    VALUE arr = (VALUE)result;
-    rb_ary_push(arr, prof_allocation_wrap(allocation));
-    return ST_CONTINUE;
-}
-
-static int prof_method_mark_allocations(st_data_t key, st_data_t value, st_data_t data)
-{
-    prof_allocation_t* allocation = (prof_allocation_t*)value;
-    prof_allocation_mark(allocation);
-    return ST_CONTINUE;
-}
-
-void allocations_table_free(st_table* table)
-{
-    rb_st_foreach(table, allocations_table_free_iterator, 0);
-    rb_st_free_table(table);
-}
-
 /* ================  prof_method_t   =================*/
 prof_method_t* prof_get_method(VALUE self)
 {
@@ -167,7 +141,7 @@ prof_method_t* prof_method_create(VALUE profile, VALUE klass, VALUE msym, VALUE
     result->measurement = prof_measurement_create();
 
     result->call_trees = prof_call_trees_create();
-    result->allocations_table = allocations_table_create();
+    result->allocations_table = prof_allocations_create();
 
     result->visits = 0;
     result->recursive = false;
@@ -180,6 +154,14 @@ prof_method_t* prof_method_create(VALUE profile, VALUE klass, VALUE msym, VALUE
     return result;
 }
 
+prof_method_t* prof_method_copy(prof_method_t* other)
+{
+    prof_method_t* result = prof_method_create(other->profile, other->klass, other->method_name, other->source_file, other->source_line);
+    result->measurement = prof_measurement_copy(other->measurement);
+
+    return result;
+}
+
 /* The underlying c structures are freed when the parent profile is freed.
    However, on shutdown the Ruby GC frees objects in any will-nilly order.
    That means the ruby thread object wrapping the c thread struct may
@@ -205,7 +187,7 @@ static void prof_method_free(prof_method_t* method)
         method->object = Qnil;
     }
 
-    allocations_table_free(method->allocations_table);
+    prof_allocations_free(method->allocations_table);
     prof_call_trees_free(method->call_trees);
     prof_measurement_free(method->measurement);
     xfree(method);
@@ -223,21 +205,29 @@ void prof_method_mark(void* data)
     prof_method_t* method = (prof_method_t*)data;
 
     if (method->profile != Qnil)
-        rb_gc_mark(method->profile);
+        rb_gc_mark_movable(method->profile);
 
     if (method->object != Qnil)
-        rb_gc_mark(method->object);
+        rb_gc_mark_movable(method->object);
 
-    rb_gc_mark(method->klass_name);
-    rb_gc_mark(method->method_name);
+    rb_gc_mark_movable(method->klass_name);
+    rb_gc_mark_movable(method->method_name);
     rb_gc_mark(method->source_file);
 
     if (method->klass != Qnil)
         rb_gc_mark(method->klass);
 
     prof_measurement_mark(method->measurement);
+    prof_allocations_mark(method->allocations_table);
+}
 
-    rb_st_foreach(method->allocations_table, prof_method_mark_allocations, 0);
+void prof_method_compact(void* data)
+{
+    prof_method_t* method = (prof_method_t*)data;
+    method->object = rb_gc_location(method->object);
+    method->profile = rb_gc_location(method->profile);
+    method->klass_name = rb_gc_location(method->klass_name);
+    method->method_name = rb_gc_location(method->method_name);
 }
 
 static VALUE prof_method_allocate(VALUE klass)
@@ -255,6 +245,7 @@ static const rb_data_type_t method_info_type =
         .dmark = prof_method_mark,
         .dfree = prof_method_ruby_gc_free,
         .dsize = prof_method_size,
+        .dcompact = prof_method_compact
     },
     .data = NULL,
     .flags = RUBY_TYPED_FREE_IMMEDIATELY
@@ -291,6 +282,30 @@ size_t method_table_insert(st_table* table, st_data_t key, prof_method_t* val)
     return rb_st_insert(table, (st_data_t)key, (st_data_t)val);
 }
 
+static int prof_method_table_merge_internal(st_data_t key, st_data_t value, st_data_t data)
+{
+    st_table* self_table = (st_table*)data;
+    prof_method_t* other_child = (prof_method_t*)value;
+
+    prof_method_t* self_child = method_table_lookup(self_table, other_child->key);
+    if (self_child)
+    {
+        prof_measurement_merge_internal(self_child->measurement, other_child->measurement);
+    }
+    else
+    {
+        prof_method_t* copy = prof_method_copy(other_child);
+        method_table_insert(self_table, copy->key, copy); 
+    }
+  
+    return ST_CONTINUE;
+}
+
+void prof_method_table_merge(st_table* self, st_table* other)
+{   
+  rb_st_foreach(other, prof_method_table_merge_internal, (st_data_t)self);
+}
+
 prof_method_t* method_table_lookup(st_table* table, st_data_t key)
 {
     st_data_t val;
@@ -360,9 +375,7 @@ Returns an array of allocation information.*/
 static VALUE prof_method_allocations(VALUE self)
 {
     prof_method_t* method = prof_get_method(self);
-    VALUE result = rb_ary_new();
-    rb_st_foreach(method->allocations_table, prof_method_collect_allocations, result);
-    return result;
+    return prof_allocations_wrap(method->allocations_table);
 }
 
 /* call-seq:
@@ -397,7 +410,7 @@ static VALUE prof_method_line(VALUE self)
 }
 
 /* call-seq:
-   klass_name -> string
+   klass_name -> String
 
 Returns the name of this method's class.  Singleton classes
 will have the form <Object::Object>. */
@@ -423,9 +436,9 @@ static VALUE prof_method_klass_flags(VALUE self)
 }
 
 /* call-seq:
-   method_name -> string
+   method_name -> Symbol
 
-Returns the name of this method in the format Object#method.  Singletons
+Returns the name of this method in the format Object#method. Singletons
 methods will be returned in the format <Object::Object>#method.*/
 
 static VALUE prof_method_name(VALUE self)
@@ -499,13 +512,7 @@ static VALUE prof_method_load(VALUE self, VALUE data)
     method_data->measurement = prof_get_measurement(measurement);
 
     VALUE allocations = rb_hash_aref(data, ID2SYM(rb_intern("allocations")));
-    for (int i = 0; i < rb_array_len(allocations); i++)
-    {
-        VALUE allocation = rb_ary_entry(allocations, i);
-        prof_allocation_t* allocation_data = prof_allocation_get(allocation);
-
-        rb_st_insert(method_data->allocations_table, allocation_data->key, (st_data_t)allocation_data);
-    }
+    prof_allocations_unwrap(method_data->allocations_table, allocations);
     return data;
 }
 

data/ext/ruby_prof/rp_method.h

@@ -1,62 +1,63 @@
-/* Copyright (C) 2005-2019 Shugo Maeda <shugo@ruby-lang.org> and Charlie Savage <cfis@savagexi.com>
-   Please see the LICENSE file for copyright and distribution information */
-
-#ifndef __RP_METHOD_INFO__
-#define __RP_METHOD_INFO__
-
-#include "ruby_prof.h"
-#include "rp_measurement.h"
-
-extern VALUE cRpMethodInfo;
-
-// Source relation bit offsets.
-enum {
-    kModuleIncludee = 0x1,                    // Included in module
-    kClassSingleton = 0x2,                    // Singleton of a class
-    kModuleSingleton = 0x4,                   // Singleton of a module
-    kObjectSingleton = 0x8,                   // Singleton of an object
-    kOtherSingleton = 0x10                    // Singleton of unknown object
-};
-
-// Profiling information for each method.
-// Excluded methods have no call_trees, source_klass, or source_file.
-typedef struct prof_method_t
-{
-    VALUE profile;                          // Profile this method is associated with - needed for mark phase
-    struct prof_call_trees_t* call_trees;   // Call infos that call this method
-    st_table* allocations_table;            // Tracks object allocations
-
-    st_data_t key;                          // Table key
-    unsigned int klass_flags;               // Information about the type of class
-    VALUE klass;                            // Resolved klass
-    VALUE klass_name;                       // Resolved klass name for this method
-    VALUE method_name;                      // Resolved method name for this method
-
-    VALUE object;                           // Cached ruby object
-
-    bool recursive;
-    int visits;                             // Current visits on the stack
-    VALUE source_file;                      // Source file
-    int source_line;                        // Line number
-
-    prof_measurement_t* measurement;        // Stores measurement data for this method
-} prof_method_t;
-
-void rp_init_method_info(void);
-
-st_data_t method_key(VALUE klass, VALUE msym);
-
-st_table* method_table_create(void);
-prof_method_t* method_table_lookup(st_table* table, st_data_t key);
-size_t method_table_insert(st_table* table, st_data_t key, prof_method_t* val);
-void method_table_free(st_table* table);
-prof_method_t* prof_method_create(VALUE profile, VALUE klass, VALUE msym, VALUE source_file, int source_line);
-prof_method_t* prof_get_method(VALUE self);
-
-VALUE prof_method_wrap(prof_method_t* result);
-void prof_method_mark(void* data);
-
-VALUE resolve_klass(VALUE klass, unsigned int* klass_flags);
-VALUE resolve_klass_name(VALUE klass, unsigned int* klass_flags);
-
-#endif //__RP_METHOD_INFO__
+/* Copyright (C) 2005-2019 Shugo Maeda <shugo@ruby-lang.org> and Charlie Savage <cfis@savagexi.com>
+   Please see the LICENSE file for copyright and distribution information */
+
+#ifndef __RP_METHOD_INFO__
+#define __RP_METHOD_INFO__
+
+#include "ruby_prof.h"
+#include "rp_measurement.h"
+
+extern VALUE cRpMethodInfo;
+
+// Source relation bit offsets.
+enum {
+    kModuleIncludee = 0x1,                    // Included in module
+    kClassSingleton = 0x2,                    // Singleton of a class
+    kModuleSingleton = 0x4,                   // Singleton of a module
+    kObjectSingleton = 0x8,                   // Singleton of an object
+    kOtherSingleton = 0x10                    // Singleton of unknown object
+};
+
+// Profiling information for each method.
+// Excluded methods have no call_trees, source_klass, or source_file.
+typedef struct prof_method_t
+{
+    VALUE profile;                          // Profile this method is associated with - needed for mark phase
+    struct prof_call_trees_t* call_trees;   // Call infos that call this method
+    st_table* allocations_table;            // Tracks object allocations
+
+    st_data_t key;                          // Table key
+    unsigned int klass_flags;               // Information about the type of class
+    VALUE klass;                            // Resolved klass
+    VALUE klass_name;                       // Resolved klass name for this method
+    VALUE method_name;                      // Resolved method name for this method
+
+    VALUE object;                           // Cached ruby object
+
+    bool recursive;
+    int visits;                             // Current visits on the stack
+    VALUE source_file;                      // Source file
+    int source_line;                        // Line number
+
+    prof_measurement_t* measurement;        // Stores measurement data for this method
+} prof_method_t;
+
+void rp_init_method_info(void);
+
+st_data_t method_key(VALUE klass, VALUE msym);
+
+st_table* method_table_create(void);
+prof_method_t* method_table_lookup(st_table* table, st_data_t key);
+size_t method_table_insert(st_table* table, st_data_t key, prof_method_t* val);
+void method_table_free(st_table* table);
+void prof_method_table_merge(st_table* self, st_table* other);
+prof_method_t* prof_method_create(VALUE profile, VALUE klass, VALUE msym, VALUE source_file, int source_line);
+prof_method_t* prof_get_method(VALUE self);
+
+VALUE prof_method_wrap(prof_method_t* result);
+void prof_method_mark(void* data);
+
+VALUE resolve_klass(VALUE klass, unsigned int* klass_flags);
+VALUE resolve_klass_name(VALUE klass, unsigned int* klass_flags);
+
+#endif //__RP_METHOD_INFO__