| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #import <Cocoa/Cocoa.h> | |
| 6 #include <dlfcn.h> | |
| 7 | |
| 8 #include "base/logging.h" | |
| 9 #import "base/memory/scoped_nsobject.h" | |
| 10 #import "chrome/browser/ui/cocoa/objc_zombie.h" | |
| 11 #include "testing/gtest/include/gtest/gtest.h" | |
| 12 #include "testing/platform_test.h" | |
| 13 | |
| 14 namespace { | |
| 15 | |
| 16 // Dynamically look up |objc_setAssociatedObject()|, which isn't | |
| 17 // available until the 10.6 SDK. | |
| 18 | |
| 19 typedef void objc_setAssociatedObjectFn(id object, void *key, id value, | |
| 20 int policy); | |
| 21 objc_setAssociatedObjectFn* LookupSetAssociatedObjectFn() { | |
| 22 return reinterpret_cast<objc_setAssociatedObjectFn*>( | |
| 23 dlsym(RTLD_DEFAULT, "objc_setAssociatedObject")); | |
| 24 } | |
| 25 | |
| 26 } // namespace | |
| 27 | |
| 28 @interface ZombieCxxDestructTest : NSObject | |
| 29 { | |
| 30 scoped_nsobject<id> aRef_; | |
| 31 } | |
| 32 - (id)initWith:(id)anObject; | |
| 33 @end | |
| 34 | |
| 35 @implementation ZombieCxxDestructTest | |
| 36 - (id)initWith:(id)anObject { | |
| 37 self = [super init]; | |
| 38 if (self) { | |
| 39 aRef_.reset([anObject retain]); | |
| 40 } | |
| 41 return self; | |
| 42 } | |
| 43 @end | |
| 44 | |
| 45 @interface ZombieAssociatedObjectTest : NSObject | |
| 46 + (BOOL)supportsAssociatedObjects; | |
| 47 - (id)initWithAssociatedObject:(id)anObject; | |
| 48 @end | |
| 49 | |
| 50 @implementation ZombieAssociatedObjectTest | |
| 51 | |
| 52 + (BOOL)supportsAssociatedObjects { | |
| 53 if (LookupSetAssociatedObjectFn()) | |
| 54 return YES; | |
| 55 return NO; | |
| 56 } | |
| 57 | |
| 58 - (id)initWithAssociatedObject:(id)anObject { | |
| 59 self = [super init]; | |
| 60 if (self) { | |
| 61 objc_setAssociatedObjectFn* fn = LookupSetAssociatedObjectFn(); | |
| 62 if (fn) { | |
| 63 // Cribbed from 10.6 <objc/runtime.h>. | |
| 64 static const int kObjcAssociationRetain = 01401; | |
| 65 | |
| 66 // The address of the variable itself is the unique key, the | |
| 67 // contents don't matter. | |
| 68 static char kAssociatedObjectKey = 'x'; | |
| 69 | |
| 70 (*fn)(self, &kAssociatedObjectKey, anObject, kObjcAssociationRetain); | |
| 71 } | |
| 72 } | |
| 73 return self; | |
| 74 } | |
| 75 | |
| 76 @end | |
| 77 | |
| 78 namespace { | |
| 79 | |
| 80 // Verify that the C++ destructors run when the last reference to the | |
| 81 // object is released. | |
| 82 // NOTE(shess): To test the negative, comment out the |g_objectDestruct()| | |
| 83 // call in |ZombieDealloc()|. | |
| 84 TEST(ObjcZombieTest, CxxDestructors) { | |
| 85 scoped_nsobject<id> anObject([[NSObject alloc] init]); | |
| 86 EXPECT_EQ(1u, [anObject retainCount]); | |
| 87 | |
| 88 ASSERT_TRUE(ObjcEvilDoers::ZombieEnable(YES, 100)); | |
| 89 | |
| 90 scoped_nsobject<ZombieCxxDestructTest> soonInfected( | |
| 91 [[ZombieCxxDestructTest alloc] initWith:anObject]); | |
| 92 EXPECT_EQ(2u, [anObject retainCount]); | |
| 93 | |
| 94 // When |soonInfected| becomes a zombie, the C++ destructors should | |
| 95 // run and release a reference to |anObject|. | |
| 96 soonInfected.reset(); | |
| 97 EXPECT_EQ(1u, [anObject retainCount]); | |
| 98 | |
| 99 // The local reference should remain (C++ destructors aren't re-run). | |
| 100 ObjcEvilDoers::ZombieDisable(); | |
| 101 EXPECT_EQ(1u, [anObject retainCount]); | |
| 102 } | |
| 103 | |
| 104 // Verify that the associated objects are released when the object is | |
| 105 // released. | |
| 106 // NOTE(shess): To test the negative, hardcode |g_objectDestruct| to | |
| 107 // the 10.5 version in |ZombieInit()|, and run this test on 10.6. | |
| 108 TEST(ObjcZombieTest, AssociatedObjectsReleased) { | |
| 109 if (![ZombieAssociatedObjectTest supportsAssociatedObjects]) { | |
| 110 LOG(ERROR) | |
| 111 << "ObjcZombieTest.AssociatedObjectsReleased not supported on 10.5"; | |
| 112 return; | |
| 113 } | |
| 114 | |
| 115 scoped_nsobject<id> anObject([[NSObject alloc] init]); | |
| 116 EXPECT_EQ(1u, [anObject retainCount]); | |
| 117 | |
| 118 ASSERT_TRUE(ObjcEvilDoers::ZombieEnable(YES, 100)); | |
| 119 | |
| 120 scoped_nsobject<ZombieAssociatedObjectTest> soonInfected( | |
| 121 [[ZombieAssociatedObjectTest alloc] initWithAssociatedObject:anObject]); | |
| 122 EXPECT_EQ(2u, [anObject retainCount]); | |
| 123 | |
| 124 // When |soonInfected| becomes a zombie, the associated object | |
| 125 // should be released. | |
| 126 soonInfected.reset(); | |
| 127 EXPECT_EQ(1u, [anObject retainCount]); | |
| 128 | |
| 129 // The local reference should remain (associated objects not re-released). | |
| 130 ObjcEvilDoers::ZombieDisable(); | |
| 131 EXPECT_EQ(1u, [anObject retainCount]); | |
| 132 } | |
| 133 | |
| 134 } // namespace | |
| OLD | NEW |