ext/-test-/vm/at_exit.c


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);
}