matz

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@858 b2dd03c8-39d4-4d8f-98ff-823fe69b080e

2 files changed, 8 insertions, 2 deletions

diff --git a/lib/delegate.rb b/lib/delegate.rb
index 29f2a5ded2..480e1ef6b8 100644
--- a/lib/delegate.rb
+++ b/lib/delegate.rb
@@ -39,7 +39,7 @@ class Delegator
 	    raise
 	  end
 	end
-	EOS
+      EOS
     end
   end
 
diff --git a/lib/weakref.rb b/lib/weakref.rb
index 9e510d6f1c..459f69f924 100644
--- a/lib/weakref.rb
+++ b/lib/weakref.rb
@@ -45,7 +45,13 @@ class WeakRef<Delegator
     @__id = orig.__id__
     ObjectSpace.define_finalizer orig, @@final
     ObjectSpace.define_finalizer self, @@final
-    ID_MAP[@__id] = [] unless ID_MAP[@__id]
+    __old_status = Thread.critical
+    begin
+      Thread.critical = true
+      ID_MAP[@__id] = [] unless ID_MAP[@__id]
+    ensure
+      Thread.critical = __old_status
+    end
     ID_MAP[@__id].push self.__id__
     ID_REV_MAP[self.id] = @__id
   end