Fix crash

Zepo · May 12, 2016 · 66b071f · 66b071f
1 parent e346665
commit 66b071f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/MLeaksFinder/NSObject+MemoryLeak.m b/MLeaksFinder/NSObject+MemoryLeak.m
@@ -20,8 +20,8 @@ - (BOOL)willDealloc {
     if ([[NSObject classNamesInWhiteList] containsObject:className])
         return NO;
 
-    id sender = objc_getAssociatedObject([UIApplication sharedApplication], kLatestSenderKey);
-    if (sender == self)
+    NSNumber *senderPtr = objc_getAssociatedObject([UIApplication sharedApplication], kLatestSenderKey);
+    if ([senderPtr isEqualToNumber:@((uintptr_t)self)])
         return NO;
 
     __weak id weakSelf = self;

diff --git a/MLeaksFinder/UIApplication+MemoryLeak.m b/MLeaksFinder/UIApplication+MemoryLeak.m
@@ -24,7 +24,7 @@ + (void)load {
 }
 
 - (BOOL)swizzled_sendAction:(SEL)action to:(id)target from:(id)sender forEvent:(UIEvent *)event {
-    objc_setAssociatedObject(self, kLatestSenderKey, sender, OBJC_ASSOCIATION_ASSIGN);
+    objc_setAssociatedObject(self, kLatestSenderKey, @((uintptr_t)sender), OBJC_ASSOCIATION_RETAIN);
 
     return [self swizzled_sendAction:action to:target from:sender forEvent:event];
 }