ruby-prof 0.18.0 → 1.0.0

data/ext/ruby_prof/rp_method.h

@@ -4,23 +4,18 @@
 #ifndef __RP_METHOD_INFO__
 #define __RP_METHOD_INFO__
 
-#include <ruby.h>
+#include "ruby_prof.h"
+#include "rp_measurement.h"
 
-extern VALUE cMethodInfo;
-
-/* A key used to identify each method */
-typedef struct
-{
-    VALUE klass;                            /* The method's class. */
-    ID mid;                                 /* The method id. */
-    st_index_t key;                         /* Cache calculated key */
-} prof_method_key_t;
+extern VALUE cRpMethodInfo;
 
 /* Source relation bit offsets. */
 enum {
-    kModuleIncludee = 0,                    /* Included module */
-    kModuleSingleton,                       /* Singleton class of a module */
-    kObjectSingleton                        /* Singleton class of an object */
+    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 unkown object */
 };
 
 /* Forward declaration, see rp_call_info.h */
@@ -30,46 +25,46 @@ struct prof_call_infos_t;
 /* Excluded methods have no call_infos, source_klass, or source_file. */
 typedef struct
 {
-    /* Hot */
-
-    prof_method_key_t *key;                 /* Table key */
+    st_data_t key;                          /* Table key */
 
-    struct prof_call_infos_t *call_infos;   /* Call infos */
     int visits;                             /* Current visits on the stack */
+    bool excluded;                          /* Exclude from profile? */
 
-    unsigned int excluded : 1;              /* Exclude from profile? */
-    unsigned int recursive : 1;             /* Recursive (direct or mutual)? */
+    st_table* parent_call_infos;            /* Call infos that call this method */
+    st_table* child_call_infos;             /* Call infos that this method calls */
+    st_table* allocations_table;            /* Tracks object allocations */
 
-    /* Cold */
+    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 */
-    VALUE source_klass;                     /* Source class */
-    const char *source_file;                /* Source file */
-    int line;                               /* Line number */
 
-    unsigned int resolved : 1;              /* Source resolved? */
-    unsigned int relation : 3;              /* Source relation bits */
+    bool root;                               /* Is this a root method */
+    bool recursive;
+    VALUE source_file;                       /* Source file */
+    int source_line;                         /* Line number */
+
+    prof_measurement_t *measurement;
 } prof_method_t;
 
 void rp_init_method_info(void);
 
-void method_key(prof_method_key_t* key, VALUE klass, ID mid);
+st_data_t method_key(VALUE klass, VALUE msym);
 
-st_table * method_table_create();
-prof_method_t * method_table_lookup(st_table *table, const prof_method_key_t* key);
-size_t method_table_insert(st_table *table, const prof_method_key_t *key, prof_method_t *val);
+st_table *method_table_create(void);
+prof_method_t* prof_method_create_excluded(VALUE klass, VALUE msym);
+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 klass, ID mid, const char* source_file, int line);
-prof_method_t* prof_method_create_excluded(VALUE klass, ID mid);
+prof_method_t *prof_method_create(VALUE klass, VALUE msym, VALUE source_file, int source_line);
+prof_method_t *prof_method_get(VALUE self);
 
 VALUE prof_method_wrap(prof_method_t *result);
-void prof_method_mark(prof_method_t *method);
-
-/* Setup infrastructure to use method keys as hash comparisons */
-int method_table_cmp(prof_method_key_t *key1, prof_method_key_t *key2);
-st_index_t method_table_hash(prof_method_key_t *key);
+void prof_method_mark(void *data);
 
-extern struct st_hash_type type_method_hash;
+VALUE resolve_klass(VALUE klass, unsigned int *klass_flags);
+VALUE resolve_klass_name(VALUE klass, unsigned int* klass_flags);
 
-#endif
+#endif //__RP_METHOD_INFO__

data/ext/ruby_prof/rp_profile.c

@@ -0,0 +1,881 @@
+/* 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 */
+
+/* Document-class: RubyProf::Profile
+
+The Profile class represents a single profiling run and provides the main API for using ruby-prof.
+After creating a Profile instance, start profiling code by calling the Profile#start method. To finish profiling,
+call Profile#stop. Once profiling is completed, the Profile instance contains the results.
+
+  profile = RubyProf::Profile.new
+  profile.start
+  ...
+  result = profile.stop
+
+Alternatively, you can use the block syntax:
+
+  profile = RubyProf::Profile.profile do
+    ...
+  end
+*/
+
+#include <assert.h>
+
+#include "rp_allocation.h"
+#include "rp_call_info.h"
+#include "rp_profile.h"
+#include "rp_method.h"
+
+VALUE cProfile;
+
+/* support tracing ruby events from ruby-prof. useful for getting at
+ what actually happens inside the ruby interpreter (and ruby-prof).
+ set environment variable RUBY_PROF_TRACE to filename you want to
+ find the trace in.
+ */
+FILE* trace_file = NULL;
+
+static int
+excludes_method(st_data_t key, prof_profile_t* profile)
+{
+    return (profile->exclude_methods_tbl &&
+            method_table_lookup(profile->exclude_methods_tbl, key) != NULL);
+}
+
+static prof_method_t*
+create_method(prof_profile_t* profile, st_data_t key, VALUE klass, VALUE msym, VALUE source_file, int source_line)
+{
+    prof_method_t* result = NULL;
+    
+    if (excludes_method(key, profile))
+    {
+        /* We found a exclusion sentinel so propagate it into the thread's local hash table. */
+        /* TODO(nelgau): Is there a way to avoid this allocation completely so that all these
+         tables share the same exclusion method struct? The first attempt failed due to my
+         ignorance of the whims of the GC. */
+        result = prof_method_create_excluded(klass, msym);
+    }
+    else
+    {
+        result = prof_method_create(klass, msym, source_file, source_line);
+    }
+    
+    /* Insert the newly created method, or the exlcusion sentinel. */
+    method_table_insert(profile->last_thread_data->method_table, result->key, result);
+    
+    return result;
+}
+
+static const char *
+get_event_name(rb_event_flag_t event)
+{
+    switch (event) {
+        case RUBY_EVENT_LINE:
+            return "line";
+        case RUBY_EVENT_CLASS:
+            return "class";
+        case RUBY_EVENT_END:
+            return "end";
+        case RUBY_EVENT_CALL:
+            return "call";
+        case RUBY_EVENT_RETURN:
+            return "return";
+        case RUBY_EVENT_B_CALL:
+            return "b-call";
+        case RUBY_EVENT_B_RETURN:
+            return "b-return";
+        case RUBY_EVENT_C_CALL:
+            return "c-call";
+        case RUBY_EVENT_C_RETURN:
+            return "c-return";
+        case RUBY_EVENT_THREAD_BEGIN:
+            return "thread-begin";
+        case RUBY_EVENT_THREAD_END:
+            return "thread-end";
+        case RUBY_EVENT_FIBER_SWITCH:
+            return "fiber-switch";
+        case RUBY_EVENT_RAISE:
+            return "raise";
+        default:
+            return "unknown";
+    }
+}
+
+/* ===========  Profiling ================= */
+thread_data_t* check_fiber(prof_profile_t *profile, double measurement)
+{
+    thread_data_t* result = NULL;
+    
+    /* Get the current thread and fiber information. */
+    VALUE fiber = rb_fiber_current();
+    
+    /* We need to switch the profiling context if we either had none before,
+     we don't merge fibers and the fiber ids differ, or the thread ids differ. */
+    if (profile->last_thread_data->fiber != fiber)
+    {
+        result = threads_table_lookup(profile, fiber);
+        if (!result)
+        {
+            result = threads_table_insert(profile, fiber);
+        }
+        switch_thread(profile, result, measurement);
+    }
+    else
+    {
+        result = profile->last_thread_data;
+    }
+    return result;
+}
+
+static void
+prof_trace(prof_profile_t* profile, rb_trace_arg_t *trace_arg, double measurement)
+{
+    static VALUE last_fiber = Qnil;
+    VALUE fiber = rb_fiber_current();
+    
+    rb_event_flag_t event = rb_tracearg_event_flag(trace_arg);
+    const char* event_name = get_event_name(event);
+    
+    VALUE source_file = rb_tracearg_path(trace_arg);
+    int source_line = FIX2INT(rb_tracearg_lineno(trace_arg));
+    
+    #ifdef HAVE_RB_TRACEARG_CALLEE_ID
+    VALUE msym = rb_tracearg_callee_id(trace_arg);
+    #else
+    VALUE msym = rb_tracearg_method_id(trace_arg);
+    #endif
+    
+    unsigned int klass_flags;
+    VALUE klass = rb_tracearg_defined_class(trace_arg);
+    VALUE resolved_klass = resolve_klass(klass, &klass_flags);
+    const char* class_name = "";
+    
+    if (resolved_klass != Qnil)
+        class_name = rb_class2name(resolved_klass);
+    
+    if (last_fiber != fiber)
+    {
+        fprintf(trace_file, "\n");
+    }
+    
+    const char* method_name_char = (msym != Qnil ? rb_id2name(SYM2ID(msym)) : "");
+    const char* source_file_char = (source_file != Qnil ? StringValuePtr(source_file) : "");
+    
+    fprintf(trace_file, "%2lu:%2f %-8s %s#%s    %s:%2d\n",
+            FIX2ULONG(fiber), (double) measurement,
+            event_name, class_name, method_name_char, source_file_char, source_line);
+    fflush(trace_file);
+    last_fiber = fiber;
+}
+
+static void
+prof_event_hook(VALUE trace_point, void* data)
+{
+    prof_profile_t* profile = (prof_profile_t*)data;
+    thread_data_t* thread_data = NULL;
+    prof_frame_t *frame = NULL;
+    rb_trace_arg_t* trace_arg = rb_tracearg_from_tracepoint(trace_point);
+    double measurement = prof_measure(profile->measurer, trace_arg);
+    rb_event_flag_t event = rb_tracearg_event_flag(trace_arg);
+    VALUE self = rb_tracearg_self(trace_arg);
+    
+    if (trace_file != NULL)
+    {
+        prof_trace(profile, trace_arg, measurement);
+    }
+    
+    /* Special case - skip any methods from the mProf
+     module since they clutter the results but aren't important to them results. */
+    if (self == mProf)
+        return;
+    
+    thread_data = check_fiber(profile, measurement);
+    
+    if (!thread_data->trace)
+        return;
+    
+    /* Get the current frame for the current thread. */
+    frame = thread_data->stack->ptr;
+    
+    switch (event)
+    {
+        case RUBY_EVENT_LINE:
+        {
+            /* Keep track of the current line number in this method.  When
+             a new method is called, we know what line number it was
+             called from. */
+            if (frame->call_info)
+            {
+                if (prof_frame_is_real(frame))
+                {
+                    frame->source_file = rb_tracearg_path(trace_arg);
+                    frame->source_line = FIX2INT(rb_tracearg_lineno(trace_arg));
+                }
+                break;
+            }
+            
+            /* If we get here there was no frame, which means this is
+             the first method seen for this thread, so fall through
+             to below to create it. */
+        }
+        case RUBY_EVENT_CALL:
+        case RUBY_EVENT_C_CALL:
+        {
+            prof_frame_t* next_frame;
+            prof_call_info_t* call_info;
+            prof_method_t* method;
+
+            /* Get current measurement */
+            measurement = prof_measure(profile->measurer, trace_arg);
+
+            VALUE klass = rb_tracearg_defined_class(trace_arg);
+            
+            /* Special case - skip any methods from the mProf
+             module or cProfile class since they clutter
+             the results but aren't important to them results. */
+            if (klass == cProfile)
+                return;
+            
+            #ifdef HAVE_RB_TRACEARG_CALLEE_ID
+            VALUE msym = rb_tracearg_callee_id(trace_arg);
+            #else
+            VALUE msym = rb_tracearg_method_id(trace_arg);
+            #endif
+
+            st_data_t key = method_key(klass, msym);
+            
+            method = method_table_lookup(thread_data->method_table, key);
+            
+            if (!method)
+            {
+                VALUE source_file = (event != RUBY_EVENT_C_CALL ? rb_tracearg_path(trace_arg) : Qnil);
+                int source_line = (event != RUBY_EVENT_C_CALL ? FIX2INT(rb_tracearg_lineno(trace_arg)) : 0);
+                method = create_method(profile, key, klass, msym, source_file, source_line);
+            }
+            
+            if (method->excluded)
+            {
+                prof_stack_pass(thread_data->stack);
+                break;
+            }
+            
+            if (!frame->call_info)
+            {
+                method->root = true;
+                call_info = prof_call_info_create(method, NULL, method->source_file, method->source_line);
+                st_insert(method->parent_call_infos, (st_data_t)&key, (st_data_t)call_info);
+            }
+            else
+            {
+                call_info = call_info_table_lookup(method->parent_call_infos, frame->call_info->method->key);
+                
+                if (!call_info)
+                {
+                    /* This call info does not yet exist.  So create it, then add
+                     it to previous callinfo's children and to the current method .*/
+                    call_info = prof_call_info_create(method, frame->call_info->method, frame->source_file, frame->source_line);
+                    call_info_table_insert(method->parent_call_infos, frame->call_info->method->key, call_info);
+                    call_info_table_insert(frame->call_info->method->child_call_infos, method->key, call_info);
+                }
+            }
+
+            /* Push a new frame onto the stack for a new c-call or ruby call (into a method) */
+            next_frame = prof_stack_push(thread_data->stack, call_info, measurement, RTEST(profile->paused));
+            next_frame->source_file = method->source_file;
+            next_frame->source_line = method->source_line;
+            break;
+        }
+        case RUBY_EVENT_RETURN:
+        case RUBY_EVENT_C_RETURN:
+        {
+            /* Get current measurement */
+            prof_stack_pop(thread_data->stack, measurement);
+            break;
+        }
+        case RUBY_INTERNAL_EVENT_NEWOBJ:
+        {
+            /* We want to assign the allocations lexically, not the execution context (otherwise all allocations will
+             show up under Class#new */
+            int source_line = FIX2INT(rb_tracearg_lineno(trace_arg));
+            VALUE source_file = rb_tracearg_path(trace_arg);
+
+            prof_method_t* method = prof_find_method(thread_data->stack, source_file, source_line);
+            if (method)
+                prof_allocate_increment(method, trace_arg);
+
+            break;
+        }
+    }
+}
+
+void
+prof_install_hook(VALUE self)
+{
+    prof_profile_t* profile = prof_get_profile(self);
+    
+    VALUE event_tracepoint = rb_tracepoint_new(Qnil,
+                                               RUBY_EVENT_CALL | RUBY_EVENT_RETURN |
+                                               RUBY_EVENT_C_CALL | RUBY_EVENT_C_RETURN |
+                                               RUBY_EVENT_LINE,
+                                               prof_event_hook, profile);
+    rb_ary_push(profile->tracepoints, event_tracepoint);
+    
+    if (profile->measurer->track_allocations)
+    {
+        VALUE allocation_tracepoint = rb_tracepoint_new(Qnil, RUBY_INTERNAL_EVENT_NEWOBJ, prof_event_hook, profile);
+        rb_ary_push(profile->tracepoints, allocation_tracepoint);
+    }
+    
+    for (int i = 0; i < RARRAY_LEN(profile->tracepoints); i++)
+    {
+        rb_tracepoint_enable(rb_ary_entry(profile->tracepoints, i));
+    }
+}
+
+void
+prof_remove_hook(VALUE self)
+{
+    prof_profile_t* profile = prof_get_profile(self);
+    
+    for (int i = 0; i < RARRAY_LEN(profile->tracepoints); i++)
+    {
+        rb_tracepoint_disable(rb_ary_entry(profile->tracepoints, i));
+    }
+    rb_ary_clear(profile->tracepoints);
+}
+
+prof_profile_t*
+prof_get_profile(VALUE self)
+{
+    /* Can't use Data_Get_Struct because that triggers the event hook
+       ending up in endless recursion. */
+    return DATA_PTR(self);
+}
+
+static int
+collect_threads(st_data_t key, st_data_t value, st_data_t result)
+{
+    thread_data_t* thread_data = (thread_data_t*) value;
+    if (thread_data->trace)
+    {
+        VALUE threads_array = (VALUE)result;
+        rb_ary_push(threads_array, prof_thread_wrap(thread_data));
+    }
+    return ST_CONTINUE;
+}
+
+/* ========  Profile Class ====== */
+static int
+mark_threads(st_data_t key, st_data_t value, st_data_t result)
+{
+    thread_data_t *thread = (thread_data_t *) value;
+    prof_thread_mark(thread);
+    return ST_CONTINUE;
+}
+
+static int
+mark_methods(st_data_t key, st_data_t value, st_data_t result)
+{
+    prof_method_t *method = (prof_method_t *) value;
+    prof_method_mark(method);
+    return ST_CONTINUE;
+}
+
+static void
+prof_mark(prof_profile_t *profile)
+{
+    rb_gc_mark(profile->tracepoints);
+    st_foreach(profile->threads_tbl, mark_threads, 0);
+    st_foreach(profile->exclude_methods_tbl, mark_methods, 0);
+}
+
+/* Freeing the profile creates a cascade of freeing.
+   It fress the thread table, which frees its methods,
+   which frees its call infos. */
+static void
+prof_free(prof_profile_t *profile)
+{
+    profile->last_thread_data = NULL;
+
+    threads_table_free(profile->threads_tbl);
+    profile->threads_tbl = NULL;
+
+    if (profile->exclude_threads_tbl)
+    {
+        st_free_table(profile->exclude_threads_tbl);
+        profile->exclude_threads_tbl = NULL;
+    }
+
+    if (profile->include_threads_tbl)
+    {
+        st_free_table(profile->include_threads_tbl);
+        profile->include_threads_tbl = NULL;
+    }
+
+    /* This table owns the excluded sentinels for now. */
+    method_table_free(profile->exclude_methods_tbl);
+    profile->exclude_methods_tbl = NULL;
+
+    xfree(profile->measurer);
+    profile->measurer = NULL;
+
+    xfree(profile);
+}
+
+static VALUE
+prof_allocate(VALUE klass)
+{
+    VALUE result;
+    prof_profile_t* profile;
+    result = Data_Make_Struct(klass, prof_profile_t, prof_mark, prof_free, profile);
+    profile->threads_tbl = threads_table_create();
+    profile->exclude_threads_tbl = NULL;
+    profile->include_threads_tbl = NULL;
+    profile->running = Qfalse;
+    profile->allow_exceptions = 0;
+    profile->exclude_methods_tbl = method_table_create();
+    profile->running = Qfalse;
+    profile->tracepoints = rb_ary_new();
+    return result;
+}
+
+static void
+prof_exclude_common_methods(VALUE profile)
+{
+    rb_funcall(profile, rb_intern("exclude_common_methods!"), 0);
+}
+
+static int
+pop_frames(VALUE key, st_data_t value, st_data_t data)
+{
+    thread_data_t* thread_data = (thread_data_t*)value;
+    prof_profile_t* profile = (prof_profile_t*)data;
+    double measurement = prof_measure(profile->measurer, NULL);
+
+    if (profile->last_thread_data->fiber != thread_data->fiber)
+        switch_thread(profile, thread_data, measurement);
+
+    while (prof_stack_pop(thread_data->stack, measurement));
+
+    return ST_CONTINUE;
+}
+
+static void
+prof_stop_threads(prof_profile_t* profile)
+{
+    st_foreach(profile->threads_tbl, pop_frames, (st_data_t)profile);
+}
+
+/* call-seq:
+   new()
+   new(options)
+
+   Returns a new profiler. Possible options for the options hash are:
+
+   measure_mode::     Measure mode. Specifies the profile measure mode.
+                      If not specified, defaults to RubyProf::WALL_TIME.
+   exclude_threads::  Threads to exclude from the profiling results.
+   include_threads::  Focus profiling on only the given threads. This will ignore
+                      all other threads.
+   allow_exceptions:: Whether to raise exceptions encountered during profiling,
+                      or to suppress all exceptions during profiling
+*/
+static VALUE
+prof_initialize(int argc,  VALUE *argv, VALUE self)
+{
+    prof_profile_t* profile = prof_get_profile(self);
+    VALUE mode_or_options;
+    VALUE mode = Qnil;
+    VALUE exclude_threads = Qnil;
+    VALUE include_threads = Qnil;
+    VALUE exclude_common = Qnil;
+    VALUE allow_exceptions = Qfalse;
+    VALUE track_allocations = Qfalse;
+
+    int i;
+
+    switch (rb_scan_args(argc, argv, "02", &mode_or_options, &exclude_threads))
+    {
+    case 0:
+        break;
+    case 1:
+        if (FIXNUM_P(mode_or_options))
+        {
+            mode = mode_or_options;
+        }
+        else
+        {
+            Check_Type(mode_or_options, T_HASH);
+            mode = rb_hash_aref(mode_or_options, ID2SYM(rb_intern("measure_mode")));
+            track_allocations = rb_hash_aref(mode_or_options, ID2SYM(rb_intern("track_allocations")));
+            allow_exceptions = rb_hash_aref(mode_or_options, ID2SYM(rb_intern("allow_exceptions")));
+            exclude_common = rb_hash_aref(mode_or_options, ID2SYM(rb_intern("exclude_common")));
+            exclude_threads = rb_hash_aref(mode_or_options, ID2SYM(rb_intern("exclude_threads")));
+            include_threads = rb_hash_aref(mode_or_options, ID2SYM(rb_intern("include_threads")));
+        }
+        break;
+    case 2:
+        Check_Type(exclude_threads, T_ARRAY);
+        break;
+    }
+
+    if (mode == Qnil)
+    {
+        mode = INT2NUM(MEASURE_WALL_TIME);
+    }
+    else
+    {
+        Check_Type(mode, T_FIXNUM);
+    }
+    profile->measurer = prof_get_measurer(NUM2INT(mode), track_allocations == Qtrue);
+    profile->allow_exceptions = (allow_exceptions == Qtrue);
+
+    if (exclude_threads != Qnil)
+    {
+        Check_Type(exclude_threads, T_ARRAY);
+        assert(profile->exclude_threads_tbl == NULL);
+        profile->exclude_threads_tbl = threads_table_create();
+        for (i = 0; i < RARRAY_LEN(exclude_threads); i++)
+        {
+            VALUE thread = rb_ary_entry(exclude_threads, i);
+            st_insert(profile->exclude_threads_tbl, thread, Qtrue);
+        }
+    }
+
+    if (include_threads != Qnil)
+    {
+        Check_Type(include_threads, T_ARRAY);
+        assert(profile->include_threads_tbl == NULL);
+        profile->include_threads_tbl = threads_table_create();
+        for (i = 0; i < RARRAY_LEN(include_threads); i++)
+        {
+            VALUE thread = rb_ary_entry(include_threads, i);
+            st_insert(profile->include_threads_tbl, thread, Qtrue);
+        }
+    }
+
+    if (RTEST(exclude_common))
+    {
+        prof_exclude_common_methods(self);
+    }
+
+    return self;
+}
+
+/* call-seq:
+   paused? -> boolean
+
+   Returns whether a profile is currently paused.*/
+static VALUE
+prof_paused(VALUE self)
+{
+    prof_profile_t* profile = prof_get_profile(self);
+    return profile->paused;
+}
+
+/* call-seq:
+   running? -> boolean
+
+   Returns whether a profile is currently running.*/
+static VALUE
+prof_running(VALUE self)
+{
+    prof_profile_t* profile = prof_get_profile(self);
+    return profile->running;
+}
+
+/* call-seq:
+   mode -> measure_mode
+
+   Returns the measure mode used in this profile.*/
+static VALUE
+prof_profile_measure_mode(VALUE self)
+{
+    prof_profile_t* profile = prof_get_profile(self);
+    return INT2NUM(profile->measurer->mode);
+}
+
+/* call-seq:
+   track_allocations -> boolean
+
+   Returns if object allocations were tracked in this profile.*/
+static VALUE
+prof_profile_track_allocations(VALUE self)
+{
+    prof_profile_t* profile = prof_get_profile(self);
+    return profile->measurer->track_allocations ? Qtrue : Qfalse;
+}
+
+/* call-seq:
+   start -> self
+
+   Starts recording profile data.*/
+static VALUE
+prof_start(VALUE self)
+{
+    char* trace_file_name;
+
+    prof_profile_t* profile = prof_get_profile(self);
+
+    if (profile->running == Qtrue)
+    {
+        rb_raise(rb_eRuntimeError, "RubyProf.start was already called");
+    }
+
+    profile->running = Qtrue;
+    profile->paused = Qfalse;
+    profile->last_thread_data = threads_table_insert(profile, rb_fiber_current());
+
+    /* open trace file if environment wants it */
+    trace_file_name = getenv("RUBY_PROF_TRACE");
+    if (trace_file_name != NULL) 
+    {
+      if (strcmp(trace_file_name, "stdout") == 0) 
+      {
+        trace_file = stdout;
+      } 
+      else if (strcmp(trace_file_name, "stderr") == 0)
+      {
+        trace_file = stderr;
+      }
+      else 
+      {
+        trace_file = fopen(trace_file_name, "w");
+      }
+    }
+
+    prof_install_hook(self);
+    return self;
+}
+
+/* call-seq:
+   pause -> self
+
+   Pauses collecting profile data. */
+static VALUE
+prof_pause(VALUE self)
+{
+    prof_profile_t* profile = prof_get_profile(self);
+    if (profile->running == Qfalse)
+    {
+        rb_raise(rb_eRuntimeError, "RubyProf is not running.");
+    }
+
+    if (profile->paused == Qfalse)
+    {
+        profile->paused = Qtrue;
+        profile->measurement_at_pause_resume = prof_measure(profile->measurer, NULL);
+        st_foreach(profile->threads_tbl, pause_thread, (st_data_t) profile);
+    }
+
+    return self;
+}
+
+/* call-seq:
+   resume -> self
+   resume(&block) -> self
+
+   Resumes recording profile data.*/
+static VALUE
+prof_resume(VALUE self)
+{
+    prof_profile_t* profile = prof_get_profile(self);
+    if (profile->running == Qfalse)
+    {
+        rb_raise(rb_eRuntimeError, "RubyProf is not running.");
+    }
+
+    if (profile->paused == Qtrue)
+    {
+        profile->paused = Qfalse;
+        profile->measurement_at_pause_resume = prof_measure(profile->measurer, NULL);
+        st_foreach(profile->threads_tbl, unpause_thread, (st_data_t) profile);
+    }
+
+    return rb_block_given_p() ? rb_ensure(rb_yield, self, prof_pause, self) : self;
+}
+
+/* call-seq:
+   stop -> self
+
+   Stops collecting profile data.*/
+static VALUE
+prof_stop(VALUE self)
+{
+    prof_profile_t* profile = prof_get_profile(self);
+
+    if (profile->running == Qfalse)
+    {
+        rb_raise(rb_eRuntimeError, "RubyProf.start was not yet called");
+    }
+
+    prof_remove_hook(self);
+
+    /* close trace file if open */
+    if (trace_file != NULL)
+    {
+      if (trace_file !=stderr && trace_file != stdout)
+      {
+#ifdef _MSC_VER
+          _fcloseall();
+#else
+        fclose(trace_file);
+#endif
+      }
+      trace_file = NULL;
+    }
+
+    prof_stop_threads(profile);
+
+    /* Unset the last_thread_data (very important!)
+       and the threads table */
+    profile->running = profile->paused = Qfalse;
+    profile->last_thread_data = NULL;
+
+    return self;
+}
+
+/* call-seq:
+   threads -> Array of RubyProf::Thread
+
+Returns an array of RubyProf::Thread instances that were profiled. */
+static VALUE
+prof_threads(VALUE self)
+{
+    VALUE result = rb_ary_new();
+    prof_profile_t* profile = prof_get_profile(self);
+    st_foreach(profile->threads_tbl, collect_threads, result);
+    return result;
+}
+
+/* Document-method: RubyProf::Profile#Profile
+   call-seq:
+   profile(&block) -> self
+
+   Profiles the specified block.
+
+     profile = RubyProf::Profile.new
+     profile.profile do
+       ..
+     end
+*/
+static VALUE
+prof_profile_object(VALUE self)
+{
+    int result;
+    prof_profile_t* profile = prof_get_profile(self);
+
+    if (!rb_block_given_p())
+    {
+        rb_raise(rb_eArgError, "A block must be provided to the profile method.");
+    }
+
+    prof_start(self);
+    rb_protect(rb_yield, self, &result);
+    self = prof_stop(self);
+
+    if (profile->allow_exceptions && result != 0)
+    {
+        rb_jump_tag(result);
+    }
+
+    return self;
+
+}
+
+/* Document-method: RubyProf::Profile::Profile
+   call-seq:
+   profile(&block) -> RubyProf::Profile
+   profile(options, &block) -> RubyProf::Profile
+
+   Profiles the specified block and returns a RubyProf::Profile
+   object. Arguments are passed to Profile initialize method.
+
+     profile = RubyProf::Profile.profile do
+       ..
+     end
+*/
+static VALUE
+prof_profile_class(int argc,  VALUE *argv, VALUE klass)
+{
+    return prof_profile_object(rb_class_new_instance(argc, argv, cProfile));
+}
+
+/* call-seq:
+   exclude_method!(module, method_name) -> self
+
+   Excludes the method from profiling results.
+*/
+static VALUE
+prof_exclude_method(VALUE self, VALUE klass, VALUE msym)
+{
+    prof_profile_t* profile = prof_get_profile(self);
+
+    st_data_t key = method_key(klass, msym);
+    prof_method_t *method;
+
+    if (profile->running == Qtrue)
+    {
+        rb_raise(rb_eRuntimeError, "RubyProf.start was already called");
+    }
+
+    method = method_table_lookup(profile->exclude_methods_tbl, key);
+
+    if (!method)
+    {
+      method = prof_method_create_excluded(klass, msym);
+      method_table_insert(profile->exclude_methods_tbl, method->key, method);
+    }
+
+    return self;
+}
+
+/* :nodoc: */
+VALUE prof_profile_dump(VALUE self)
+{
+    VALUE result = rb_hash_new();
+    rb_hash_aset(result, ID2SYM(rb_intern("threads")), prof_threads(self));
+    return result;
+}
+
+/* :nodoc: */
+VALUE prof_profile_load(VALUE self, VALUE data)
+{
+    prof_profile_t* profile = prof_get_profile(self);
+
+    VALUE threads = rb_hash_aref(data, ID2SYM(rb_intern("threads")));
+    for (int i = 0; i < rb_array_len(threads); i++)
+    {
+        VALUE thread = rb_ary_entry(threads, i);
+        thread_data_t* thread_data = DATA_PTR(thread);
+        st_insert(profile->threads_tbl, (st_data_t)thread_data->fiber_id, (st_data_t)thread_data);
+    }
+
+    return data;
+}
+
+void rp_init_profile(void)
+{
+    mProf = rb_define_module("RubyProf");
+
+    cProfile = rb_define_class_under(mProf, "Profile", rb_cObject);
+    rb_define_alloc_func(cProfile, prof_allocate);
+
+    rb_define_singleton_method(cProfile, "profile", prof_profile_class, -1);
+    rb_define_method(cProfile, "initialize", prof_initialize, -1);
+    rb_define_method(cProfile, "start", prof_start, 0);
+    rb_define_method(cProfile, "stop", prof_stop, 0);
+    rb_define_method(cProfile, "resume", prof_resume, 0);
+    rb_define_method(cProfile, "pause", prof_pause, 0);
+    rb_define_method(cProfile, "running?", prof_running, 0);
+    rb_define_method(cProfile, "paused?", prof_paused, 0);
+    rb_define_method(cProfile, "threads", prof_threads, 0);
+    rb_define_method(cProfile, "exclude_method!", prof_exclude_method, 2);
+    rb_define_method(cProfile, "profile", prof_profile_object, 0);
+
+    rb_define_method(cProfile, "measure_mode", prof_profile_measure_mode, 0);
+    rb_define_method(cProfile, "track_allocations?", prof_profile_track_allocations, 0);
+
+    rb_define_method(cProfile, "_dump_data", prof_profile_dump, 0);
+    rb_define_method(cProfile, "_load_data", prof_profile_load, 1);
+}