mirror of
https://github.com/ruby/ruby.git
synced 2022-11-09 12:17:21 -05:00
44 lines
723 B
C
44 lines
723 B
C
#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);
|
|
}
|