nobu	2008-07-11 20:51:39 +0900 (Fri, 11 Jul 2008)

  New Revision: 18027

  http://svn.ruby-lang.org/cgi-bin/viewvc.cgi?view=rev&revision=18027

  Log:
    * io.c (rb_io_wait_readable, rb_io_wait_writable): check if the file
      descriptor is closed.

    * thread.c (rb_thread_wait_fd_rw): ditto.

  Modified files:
    trunk/ChangeLog
    trunk/bootstraptest/runner.rb
    trunk/bootstraptest/test_io.rb
    trunk/io.c
    trunk/thread.c

Index: ChangeLog
===================================================================
--- ChangeLog	(revision 18026)
+++ ChangeLog	(revision 18027)
@@ -1,3 +1,10 @@
+Fri Jul 11 20:51:36 2008  Nobuyoshi Nakada  <nobu@r...>
+
+	* io.c (rb_io_wait_readable, rb_io_wait_writable): check if the file
+	  descriptor is closed.
+
+	* thread.c (rb_thread_wait_fd_rw): ditto.
+
 Fri Jul 11 16:16:43 2008  NAKAMURA Usaku  <usa@r...>
 
 	* win32/win32.c (rb_w32_accept, rb_w32_socket, rb_w32_socketpair):
Index: bootstraptest/test_io.rb
===================================================================
--- bootstraptest/test_io.rb	(revision 18026)
+++ bootstraptest/test_io.rb	(revision 18027)
@@ -73,3 +73,22 @@
 assert_normal_exit %q{
   ARGF.set_encoding "foo"
 }
+
+50.times do
+  assert_normal_exit %q{
+    at_exit { p :foo }
+
+    megacontent = "abc" * 12345678
+    File.open("megasrc", "w") {|f| f << megacontent }
+
+    Thread.new { sleep rand*0.2; Process.kill(:INT, $$) }
+
+    r1, w1 = IO.pipe
+    r2, w2 = IO.pipe
+    t1 = Thread.new { w1 << megacontent; w1.close }
+    t2 = Thread.new { r2.read }
+    IO.copy_stream(r1, w2) rescue nil
+    r2.close; w2.close
+    r1.close; w1.close
+  }, '', ["INT"] or break
+end
Index: bootstraptest/runner.rb
===================================================================
--- bootstraptest/runner.rb	(revision 18026)
+++ bootstraptest/runner.rb	(revision 18027)
@@ -188,7 +188,7 @@
   }
 end
 
-def assert_normal_exit(testsrc, message = '')
+def assert_normal_exit(testsrc, message = '', ignore_signals = nil)
   newtest
   $stderr.puts "\##{@count} #{@location}" if @verbose
   faildesc = nil
@@ -205,28 +205,33 @@
   if status.signaled?
     signo = status.termsig
     signame = Signal.list.invert[signo]
-    sigdesc = "signal #{signo}"
-    if signame
-      sigdesc = "SIG#{signame} (#{sigdesc})"
+    unless ignore_signals and ignore_signals.include?(signame)
+      sigdesc = "signal #{signo}"
+      if signame
+        sigdesc = "SIG#{signame} (#{sigdesc})"
+      end
+      faildesc = pretty(testsrc, "killed by #{sigdesc}", nil)
+      stderr_log = File.read("assert_normal_exit_stderr.log")
+      if !stderr_log.empty?
+        faildesc << "\n" if /\n\z/ !~ faildesc
+        stderr_log << "\n" if /\n\z/ !~ stderr_log
+        stderr_log.gsub!(/^.*\n/) { '| ' + $& }
+        faildesc << stderr_log
+      end
     end
-    faildesc = pretty(testsrc, "killed by #{sigdesc}", nil)
-    stderr_log = File.read("assert_normal_exit_stderr.log")
-    if !stderr_log.empty?
-      faildesc << "\n" if /\n\z/ !~ faildesc
-      stderr_log << "\n" if /\n\z/ !~ stderr_log
-      stderr_log.gsub!(/^.*\n/) { '| ' + $& }
-      faildesc << stderr_log
-    end
   end
   if !faildesc
     $stderr.print '.'
+    true
   else
     $stderr.print 'F'
     error faildesc, message
+    false
   end
 rescue Exception => err
   $stderr.print 'E'
   error err.message, message
+  false
 end
 
 def assert_finish(timeout_seconds, testsrc, message = '')
Index: io.c
===================================================================
--- io.c	(revision 18026)
+++ io.c	(revision 18027)
@@ -608,6 +608,9 @@
 {
     rb_fdset_t rfds;
 
+    if (f < 0) {
+	rb_raise(rb_eIOError, "closed stream");
+    }
     switch (errno) {
       case EINTR:
 #if defined(ERESTART)
@@ -650,6 +653,9 @@
 {
     rb_fdset_t wfds;
 
+    if (f < 0) {
+	rb_raise(rb_eIOError, "closed stream");
+    }
     switch (errno) {
       case EINTR:
 #if defined(ERESTART)
@@ -1484,11 +1490,8 @@
 	if (RSTRING_LEN(str) != len) goto modified;
         if (nonblock) {
             rb_io_set_nonblock(fptr);
-	    n = rb_read_internal(fptr->fd, RSTRING_PTR(str), len);
         }
-        else {
-            n = rb_read_internal(fptr->fd, RSTRING_PTR(str), len);
-	}
+	n = rb_read_internal(fptr->fd, RSTRING_PTR(str), len);
         if (n < 0) {
             if (!nonblock && rb_io_wait_readable(fptr->fd))
                 goto again;
Index: thread.c
===================================================================
--- thread.c	(revision 18026)
+++ thread.c	(revision 18027)
@@ -2018,6 +2018,9 @@
     int result = 0;
     thread_debug("rb_thread_wait_fd_rw(%d, %s)\n", fd, read ? "read" : "write");
 
+    if (fd < 0) {
+	rb_raise(rb_eIOError, "closed stream");
+    }
     while (result <= 0) {
 	rb_fdset_t set;
 	rb_fd_init(&set);

--
ML: ruby-changes@q...
Info: http://www.atdot.net/~ko1/quickml/