Detect and work on ruby 1.9.1

rapportive-oss · Jan 24, 2011 · 1021c9d · 1021c9d
1 parent ad1035b
commit 1021c9d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ext/perftools.c b/ext/perftools.c
@@ -339,7 +339,7 @@ uc_get_ip(ucontext_t *uc) {
   return (char**)&uc->program_counter;
 }
 
-#ifdef RB_EVENT_HOOKS_HAVE_CALLBACK_DATA
+#if defined(RB_EVENT_HOOKS_HAVE_CALLBACK_DATA) || defined(RUBY_EVENT_VM)
 static void
 event_handler(rb_event_flag_t event, VALUE data, VALUE self, ID id, VALUE klass) {
   ProfilerRecord(0, NULL, NULL);
@@ -357,7 +357,7 @@ methprofiler_setup()
   if (bMethProfilerRunning)
     return Qtrue;
 
-#ifdef RB_EVENT_HOOKS_HAVE_CALLBACK_DATA
+#if defined(RB_EVENT_HOOKS_HAVE_CALLBACK_DATA) || defined(RUBY_EVENT_VM)
   rb_add_event_hook(event_handler, RUBY_EVENT_CALL|RUBY_EVENT_C_CALL, 0);
 #else
   rb_add_event_hook(event_handler, RUBY_EVENT_CALL|RUBY_EVENT_C_CALL);