1 /* -*-c-*- */
2 /*
3 * from eval.c
4 */
5
6 #include "eval_intern.h"
7
8 /* exit */
9
10 void
rb_call_end_proc(VALUE data)11 rb_call_end_proc(VALUE data)
12 {
13 rb_proc_call(data, rb_ary_new());
14 }
15
16 /*
17 * call-seq:
18 * at_exit { block } -> proc
19 *
20 * Converts _block_ to a +Proc+ object (and therefore
21 * binds it at the point of call) and registers it for execution when
22 * the program exits. If multiple handlers are registered, they are
23 * executed in reverse order of registration.
24 *
25 * def do_at_exit(str1)
26 * at_exit { print str1 }
27 * end
28 * at_exit { puts "cruel world" }
29 * do_at_exit("goodbye ")
30 * exit
31 *
32 * <em>produces:</em>
33 *
34 * goodbye cruel world
35 */
36
37 static VALUE
rb_f_at_exit(void)38 rb_f_at_exit(void)
39 {
40 VALUE proc;
41
42 if (!rb_block_given_p()) {
43 rb_raise(rb_eArgError, "called without a block");
44 }
45 proc = rb_block_proc();
46 rb_set_end_proc(rb_call_end_proc, proc);
47 return proc;
48 }
49
50 struct end_proc_data {
51 void (*func) ();
52 VALUE data;
53 struct end_proc_data *next;
54 };
55
56 static struct end_proc_data *end_procs, *ephemeral_end_procs;
57
58 void
rb_set_end_proc(void (* func)(VALUE),VALUE data)59 rb_set_end_proc(void (*func)(VALUE), VALUE data)
60 {
61 struct end_proc_data *link = ALLOC(struct end_proc_data);
62 struct end_proc_data **list;
63 rb_thread_t *th = GET_THREAD();
64
65 if (th->top_wrapper) {
66 list = &ephemeral_end_procs;
67 }
68 else {
69 list = &end_procs;
70 }
71 link->next = *list;
72 link->func = func;
73 link->data = data;
74 *list = link;
75 }
76
77 void
rb_mark_end_proc(void)78 rb_mark_end_proc(void)
79 {
80 struct end_proc_data *link;
81
82 link = end_procs;
83 while (link) {
84 rb_gc_mark(link->data);
85 link = link->next;
86 }
87 link = ephemeral_end_procs;
88 while (link) {
89 rb_gc_mark(link->data);
90 link = link->next;
91 }
92 }
93
94 static void
exec_end_procs_chain(struct end_proc_data * volatile * procs,VALUE * errp)95 exec_end_procs_chain(struct end_proc_data *volatile *procs, VALUE *errp)
96 {
97 struct end_proc_data volatile endproc;
98 struct end_proc_data *link;
99 VALUE errinfo = *errp;
100
101 while ((link = *procs) != 0) {
102 *procs = link->next;
103 endproc = *link;
104 xfree(link);
105 (*endproc.func) (endproc.data);
106 *errp = errinfo;
107 }
108 }
109
110 void
rb_exec_end_proc(void)111 rb_exec_end_proc(void)
112 {
113 enum ruby_tag_type state;
114 rb_execution_context_t * volatile ec = GET_EC();
115 volatile VALUE errinfo = ec->errinfo;
116
117 EC_PUSH_TAG(ec);
118 if ((state = EC_EXEC_TAG()) == TAG_NONE) {
119 again:
120 exec_end_procs_chain(&ephemeral_end_procs, &ec->errinfo);
121 exec_end_procs_chain(&end_procs, &ec->errinfo);
122 }
123 else {
124 EC_TMPPOP_TAG();
125 error_handle(state);
126 if (!NIL_P(ec->errinfo)) errinfo = ec->errinfo;
127 EC_REPUSH_TAG();
128 goto again;
129 }
130 EC_POP_TAG();
131
132 ec->errinfo = errinfo;
133 }
134
135 void
Init_jump(void)136 Init_jump(void)
137 {
138 rb_define_global_function("at_exit", rb_f_at_exit, 0);
139 }
140