ruby/ext/-test-/vm/at_exit.c

45 строки
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);
}