Permalink
Cannot retrieve contributors at this time
Join GitHub today
GitHub is home to over 50 million developers working together to host and review code, manage projects, and build software together.
Sign upGitHub is where the world builds software
Millions of developers and companies build, ship, and maintain their software on GitHub — the largest and most advanced development platform in the world.
| /* -*-c-*- */ | |
| /* | |
| * from eval.c | |
| */ | |
| #include "eval_intern.h" | |
| /* exit */ | |
| void | |
| rb_call_end_proc(VALUE data) | |
| { | |
| rb_proc_call(data, rb_ary_new()); | |
| } | |
| /* | |
| * call-seq: | |
| * at_exit { block } -> proc | |
| * | |
| * Converts _block_ to a +Proc+ object (and therefore | |
| * binds it at the point of call) and registers it for execution when | |
| * the program exits. If multiple handlers are registered, they are | |
| * executed in reverse order of registration. | |
| * | |
| * def do_at_exit(str1) | |
| * at_exit { print str1 } | |
| * end | |
| * at_exit { puts "cruel world" } | |
| * do_at_exit("goodbye ") | |
| * exit | |
| * | |
| * <em>produces:</em> | |
| * | |
| * goodbye cruel world | |
| */ | |
| static VALUE | |
| rb_f_at_exit(VALUE _) | |
| { | |
| VALUE proc; | |
| if (!rb_block_given_p()) { | |
| rb_raise(rb_eArgError, "called without a block"); | |
| } | |
| proc = rb_block_proc(); | |
| rb_set_end_proc(rb_call_end_proc, proc); | |
| return proc; | |
| } | |
| struct end_proc_data { | |
| void (*func) (); | |
| VALUE data; | |
| struct end_proc_data *next; | |
| }; | |
| static struct end_proc_data *end_procs, *ephemeral_end_procs; | |
| void | |
| rb_set_end_proc(void (*func)(VALUE), VALUE data) | |
| { | |
| struct end_proc_data *link = ALLOC(struct end_proc_data); | |
| struct end_proc_data **list; | |
| rb_thread_t *th = GET_THREAD(); | |
| if (th->top_wrapper) { | |
| list = &ephemeral_end_procs; | |
| } | |
| else { | |
| list = &end_procs; | |
| } | |
| link->next = *list; | |
| link->func = func; | |
| link->data = data; | |
| *list = link; | |
| } | |
| void | |
| rb_mark_end_proc(void) | |
| { | |
| struct end_proc_data *link; | |
| link = end_procs; | |
| while (link) { | |
| rb_gc_mark(link->data); | |
| link = link->next; | |
| } | |
| link = ephemeral_end_procs; | |
| while (link) { | |
| rb_gc_mark(link->data); | |
| link = link->next; | |
| } | |
| } | |
| static void | |
| exec_end_procs_chain(struct end_proc_data *volatile *procs, VALUE *errp) | |
| { | |
| struct end_proc_data volatile endproc; | |
| struct end_proc_data *link; | |
| VALUE errinfo = *errp; | |
| while ((link = *procs) != 0) { | |
| *procs = link->next; | |
| endproc = *link; | |
| xfree(link); | |
| (*endproc.func) (endproc.data); | |
| *errp = errinfo; | |
| } | |
| } | |
| static void | |
| rb_ec_exec_end_proc(rb_execution_context_t * ec) | |
| { | |
| enum ruby_tag_type state; | |
| volatile VALUE errinfo = ec->errinfo; | |
| EC_PUSH_TAG(ec); | |
| if ((state = EC_EXEC_TAG()) == TAG_NONE) { | |
| again: | |
| exec_end_procs_chain(&ephemeral_end_procs, &ec->errinfo); | |
| exec_end_procs_chain(&end_procs, &ec->errinfo); | |
| } | |
| else { | |
| EC_TMPPOP_TAG(); | |
| error_handle(ec, state); | |
| if (!NIL_P(ec->errinfo)) errinfo = ec->errinfo; | |
| EC_REPUSH_TAG(); | |
| goto again; | |
| } | |
| EC_POP_TAG(); | |
| ec->errinfo = errinfo; | |
| } | |
| void | |
| Init_jump(void) | |
| { | |
| rb_define_global_function("at_exit", rb_f_at_exit, 0); | |
| } |