summaryrefslogtreecommitdiff
path: root/ext/objspace/gc_hook.c
blob: 1613227a2af9fd2f048653745b2b524fbe22c5c3 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
/**********************************************************************

  gc_hook.c - GC hook mechanism/ObjectSpace extender for MRI.

  $Author: sorah $
  created at: Tue May 28 01:34:25 2013

  NOTE: This extension library is not expected to exist except C Ruby.
  NOTE: This feature is an example usage of internal event tracing APIs.

  All the files in this distribution are covered under the Ruby's
  license (see the file COPYING).

**********************************************************************/

#include "ruby/ruby.h"
#include "ruby/debug.h"

static void
invoke_proc(void *data)
{
    VALUE proc = (VALUE)data;
    rb_proc_call(proc, rb_ary_new());
}

static void
gc_start_end_i(VALUE tpval, void *data)
{
    rb_trace_arg_t *tparg = rb_tracearg_from_tracepoint(tpval);
    if (0) fprintf(stderr, "trace: %s\n", rb_tracearg_event_flag(tparg) == RUBY_INTERNAL_EVENT_GC_START ? "gc_start" : "gc_end");
    rb_postponed_job_register(0, invoke_proc, data);
}

static VALUE
set_gc_hook(VALUE rb_mObjSpace, VALUE proc, rb_event_flag_t event, const char *tp_str, const char *proc_str)
{
    VALUE tpval;
    ID tp_key = rb_intern(tp_str);
    ID proc_key = rb_intern(proc_str);

    if (RTEST(tpval = rb_ivar_get(rb_mObjSpace, tp_key))) {
	rb_tracepoint_disable(tpval);
	rb_ivar_set(rb_mObjSpace, tp_key, Qnil);
	rb_ivar_set(rb_mObjSpace, proc_key, Qnil);
    }

    if (RTEST(proc)) {
	if (!rb_obj_is_proc(proc)) {
	    rb_raise(rb_eTypeError, "trace_func needs to be Proc");
	}

	tpval = rb_tracepoint_new(0, event, gc_start_end_i, (void *)proc);
	rb_ivar_set(rb_mObjSpace, tp_key, tpval);
	rb_ivar_set(rb_mObjSpace, proc_key, proc); /* GC guard */
	rb_tracepoint_enable(tpval);
    }

    return proc;
}

static VALUE
set_after_gc_start(VALUE rb_mObjSpace, VALUE proc)
{
    return set_gc_hook(rb_mObjSpace, proc, RUBY_INTERNAL_EVENT_GC_START,
		       "__set_after_gc_start_tpval__", "__set_after_gc_start_proc__");
}

static VALUE
set_after_gc_end(VALUE rb_mObjSpace, VALUE proc)
{
    return set_gc_hook(rb_mObjSpace, proc, RUBY_INTERNAL_EVENT_GC_END,
		       "__set_after_gc_end_tpval__", "__set_after_gc_end_proc__");
}

void
Init_gc_hook(VALUE rb_mObjSpace)
{
    rb_define_module_function(rb_mObjSpace, "after_gc_start_hook=", set_after_gc_start, 1);
    rb_define_module_function(rb_mObjSpace, "after_gc_end_hook=", set_after_gc_end, 1);
}