[ruby-dev:25027] Re: IO#flush dumps core again

なかだです。

At Wed, 1 Dec 2004 17:02:39 +0900,
Yukihiro Matsumoto wrote in [ruby-dev:25015]:
> |まぁ、それはそれとして上記の微妙な問題とは別ではありますが、なかださん
> |の継続を制限するパッチは適切な方向性を示していると思います。以前ちょっ
> |と可能性に触れた (でもいまだに実験はしていない) GUI の callback で
> |callcc というのもどうにかできる気がしますし。
> 
> GUIのコールバックの件はよく理解できてないのですが、じゃあ、
> 中田さんの変更を取り込みましょうか。[ruby-dev:25004]をコミッ
> トしていただけませんか ＞ なかださん。

[ruby-dev:25004]ではprot_tagを使ってsignal handlerを判断してい
ますが、struct tagはスタック上に取られる(環境がほとんど)なので、
ループ内で割込みを受けた場合などは同じアドレスになってしまって
区別できなくなる可能性があります。

frame_uniqueのような序数を使う方法も考えられますが、ダミーのオ
ブジェクトを使う方法を試してみました。

また、[ruby-dev:25022]のような例を見ると、callback呼び出し用の
関数を用意したほうがいいかもという気がします。


Index: eval.c
===================================================================
RCS file: /cvs/ruby/src/ruby/eval.c,v
retrieving revision 1.739
diff -U2 -p -d -r1.739 eval.c
--- eval.c	2 Dec 2004 00:31:17 -0000	1.739
+++ eval.c	2 Dec 2004 04:02:45 -0000
@@ -9670,4 +9670,34 @@ thread_reset_raised()
 static void rb_thread_ready _((rb_thread_t));
 
+static VALUE trapping_top;
+
+VALUE
+rb_run_callback(func, arg, status)
+    VALUE (*func)_((VALUE));
+    VALUE arg;
+    int *status;
+{
+    VALUE val = Qnil;		/* OK */
+
+    PUSH_TAG(PROT_NONE);
+    PUSH_ITER(ITER_NOT);
+    trapping_top = (VALUE)rb_node_newnode(NODE_MEMO, trapping_top, 0, 0);
+    if ((*status = EXEC_TAG()) == 0) {
+	val = (*func)(arg);
+    }
+    trapping_top = ((NODE *)trapping_top)->u1.value;
+    POP_ITER();
+    POP_TAG();
+    return val;
+}
+
+static VALUE
+run_trap_eval(arg)
+    VALUE arg;
+{
+    VALUE *p = (VALUE *)arg;
+    return rb_eval_cmd(p[0], p[1], (int)p[2]);
+}
+
 static VALUE
 rb_trap_eval(cmd, sig, safe)
@@ -9678,14 +9708,12 @@ rb_trap_eval(cmd, sig, safe)
     VALUE val = Qnil;		/* OK */
     volatile struct thread_status_t save;
+    VALUE arg[3];
 
+    arg[0] = cmd;
+    arg[1] = rb_ary_new3(1, INT2FIX(sig));
+    arg[2] = (VALUE)safe;
     THREAD_COPY_STATUS(curr_thread, &save);
     rb_thread_ready(curr_thread);
-    PUSH_TAG(PROT_NONE);
-    PUSH_ITER(ITER_NOT);
-    if ((state = EXEC_TAG()) == 0) {
-	val = rb_eval_cmd(cmd, rb_ary_new3(1, INT2FIX(sig)), safe);
-    }
-    POP_ITER();
-    POP_TAG();
+    val = rb_run_callback(run_trap_eval, (VALUE)&arg, &state);
     THREAD_COPY_STATUS(&save, curr_thread);
 
@@ -11918,4 +11946,8 @@ rb_thread_trap_eval(cmd, sig, safe)
 {
     rb_thread_critical = 0;
+    if (curr_thread == main_thread) {
+	rb_trap_eval(cmd, sig, safe);
+	return;
+    }
     if (!rb_thread_dead(curr_thread)) {
 	if (THREAD_SAVE_CONTEXT(curr_thread)) {
@@ -12273,4 +12305,5 @@ rb_callcc(self)
     }
     th->thread = curr_thread->thread;
+    th->thgroup = trapping_top;
 
     for (vars = ruby_dyna_vars; vars; vars = vars->next) {
@@ -12314,4 +12347,7 @@ rb_cont_call(argc, argv, cont)
 	rb_raise(rb_eRuntimeError, "continuation called across threads");
     }
+    if (th->thgroup != trapping_top) {
+	rb_raise(rb_eRuntimeError, "continuation called across trap");
+    }
     switch (argc) {
       case 0:
@@ -12581,4 +12617,5 @@ Init_Thread()
     rb_define_method(rb_cCont, "[]", rb_cont_call, -1);
     rb_define_global_function("callcc", rb_callcc, 0);
+    rb_global_variable(&trapping_top);
 
     cThGroup = rb_define_class("ThreadGroup", rb_cObject);


-- 
--- 僕の前にBugはない。
--- 僕の後ろにBugはできる。
    中田 伸悦