blob: 6cfbfafa9e2690f51066674efa4969a734c4cb8c (
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
|
#include <ruby/ruby.h>
#include <ruby/vm.h>
static void
do_nothing(ruby_vm_t *vm)
{
}
static void
print_begin(ruby_vm_t *vm)
{
printf("begin\n");
}
static void
print_end(ruby_vm_t *vm)
{
printf("end\n");
}
static VALUE
register_at_exit(VALUE self, VALUE t)
{
switch (t) {
case Qtrue:
ruby_vm_at_exit(print_begin);
break;
case Qfalse:
ruby_vm_at_exit(print_end);
break;
default:
ruby_vm_at_exit(do_nothing);
break;
}
return self;
}
void
Init_at_exit(void)
{
VALUE m = rb_define_module("Bug");
VALUE c = rb_define_class_under(m, "VM", rb_cObject);
rb_define_singleton_method(c, "register_at_exit", register_at_exit, 1);
}
|