ARC fix for blocks.

14 years ago · e7ab10611f
parent 8cf5e0412f
commit e7ab10611f
1 changed files with 3 additions and 1 deletions
--- a/arc.m
+++ b/arc.m
@ -17,6 +17,7 @@ pthread_key_t ARCThreadKey;
 #endif

 extern void _NSConcreteMallocBlock;
+extern void _NSConcreteStackBlock;
 extern void _NSConcreteGlobalBlock;

@interface NSAutoreleasePool
@ -167,7 +168,8 @@ static inline id retain(id obj)
 {
 	if (isSmallObject(obj)) { return obj; }
 	Class cls = obj->isa;
-	if ((Class)&_NSConcreteMallocBlock == cls)
+	if ((Class)&_NSConcreteMallocBlock == cls ||
+	    (Class)&_NSConcreteStackBlock == cls)
 	{
 		return Block_copy(obj);
 	}