oc ~ iOS内存泄漏检测

废话篇

最近忙于生活～～～很忙，很盲，很茻
如果一个全新的项目需要去检测项目内存泄漏怎么做？
请见Swift之内存泄漏检测。
NSTableMap也可以做主题色切换，可能还有其他更多的技巧...

实战篇

swizzling无感知入侵
请见代码：

SEPLeakDetector.h

//
//  SEPLeakDetector.h
//  account-module-ios
//
//  Created by Rudy.Li on 2018/9/3.
//

#import <Foundation/Foundation.h>

@interface SEPLeakDetector : NSObject


#pragma mark - 单例方法.
+(SEPLeakDetector *) sharedInstance;
@end

@interface UINavigationController (SEPHook)
+ (void)hookUINavigationController_push;
+ (void)hookUINavigationController_pop;
@end

@interface UIViewController (SEPHook)
+ (void)hookUIViewController_push;
+ (void) hookUIViewController_present;
+ (void) hookUIViewController_dismiss;
@end
@interface UIView (SEPHook)
+ (void)hookUIView_moveToSuperView;
@end

SEPLeakDetector.m

//
//  SEPLeakDetector.m
//  account-module-ios
//
//  Created by Rudy.Li on 2018/9/3.
//

#import "SEPLeakDetector.h"
#import <objc/objc.h>
#import <objc/runtime.h>
#import "SEPToast.h"

@interface  SEPLeakDetector()
@property (nonatomic) NSMapTable *trackingObjects;
@property (nonatomic) NSMapTable *globalTrackingObjects;//cached current be leaked UIViewControllers
@property (nonatomic,copy) NSArray *currentTrackingKeys; //cached old be leaked UIViewControllers

@property (nonatomic) NSMapTable *globalViewTrackingObjects;//cached current be leaked UIViews
@property (nonatomic,copy) NSArray *currentViewTrackingKeys; //cached old be leaked UIViews
@property (nonatomic,assign) BOOL isLoading;
@end

@implementation SEPLeakDetector
static SEPLeakDetector * sharedLocalSession = nil;

+ (void)load{
#ifdef DEBUG
    [[SEPLeakDetector sharedInstance] setup];
#endif
}

+(SEPLeakDetector *) sharedInstance{
    @synchronized(self){
        if (sharedLocalSession == nil) {
            sharedLocalSession = [[self alloc] init];
        }
    }
    return  sharedLocalSession;
}

- (void)setup{
    _trackingObjects = [NSMapTable mapTableWithKeyOptions:NSMapTableWeakMemory valueOptions:NSMapTableWeakMemory];
    _globalTrackingObjects = [NSMapTable mapTableWithKeyOptions:NSMapTableStrongMemory valueOptions:NSMapTableWeakMemory];
    _globalViewTrackingObjects = [NSMapTable mapTableWithKeyOptions:NSMapTableStrongMemory valueOptions:NSMapTableWeakMemory];
    [UIView hookUIView_moveToSuperView];
    [UIViewController hookUIViewController_push];
    [UIViewController hookUIViewController_present];
    [UIViewController hookUIViewController_dismiss];
    [UINavigationController hookUINavigationController_push];
    [UINavigationController hookUINavigationController_pop];
}


- (void) expectDeallocate:(NSObject*)obj isDismiss:(BOOL)dismiss{
    
    NSString *objectDescription = obj.description;
    // 键：对象地址+方法名 值：对象
    NSString *pointObjString = [NSString stringWithFormat:@"%@", objectDescription];
    
    BOOL isExistGlobalTracking = [[SEPLeakDetector sharedInstance].currentTrackingKeys containsObject:pointObjString];
    if (isExistGlobalTracking == NO) {
        [_trackingObjects setObject:obj forKey:pointObjString];
    }

    dispatch_after(dispatch_time(DISPATCH_TIME_NOW, (int64_t)(1.0 * NSEC_PER_SEC)), dispatch_get_main_queue(), ^{
        if (dismiss) {
            NSArray *_currentArray = [SEPLeakDetector sharedInstance].currentTrackingKeys;
            NSArray *_globalArray = [[SEPLeakDetector sharedInstance] globalTrackingObjects].keyEnumerator.allObjects;
            NSMutableArray *_leakArray = [NSMutableArray array];
            for (NSString *_globalString in _globalArray) {
                @autoreleasepool {
                    BOOL result = NO;
                    for (NSString *_currentString in _currentArray) {
                        if ([_globalString isEqualToString:_currentString]) {
                            result = YES;
                            break;
                        }
                    }
                    if (result == NO) {
                        [_leakArray addObject:_globalString];
                    }
                }
            }
            //NSLog(@"%@",_leakArray);
            if ([_leakArray count]>0) {
                NSString *str = [NSString stringWithFormat:@"present内存泄漏警告\r\n%@",_leakArray];
                [[UIApplication sharedApplication] sep_showToastWithText:[NSString stringWithFormat:@"内存泄漏警告\r\n%@",str]];
                return ;
            }
        }
        
        BOOL didDeallocate = [self.trackingObjects objectForKey:pointObjString] == nil;
        if (!didDeallocate) {
            // 弹出警告框，提示异常信息
            [[UIApplication sharedApplication] sep_showToastWithText:[NSString stringWithFormat:@"内存泄漏警告\r\n%@",pointObjString]];
            return ;
        }
        
        //check subview was leaked
        NSArray *_currentArray = [SEPLeakDetector sharedInstance].currentViewTrackingKeys;
        NSArray *_globalArray = [[SEPLeakDetector sharedInstance] globalViewTrackingObjects].keyEnumerator.allObjects;
        NSMutableArray *_leakArray = [NSMutableArray array];
        for (NSString *_globalString in _globalArray) {
            @autoreleasepool {
                BOOL result = NO;
                for (NSString *_currentString in _currentArray) {
                    if ([_globalString isEqualToString:_currentString]) {
                        result = YES;
                        break;
                    }
                }
                if (result == NO) {
                    [_leakArray addObject:_globalString];
                }
            }
        }
        
        if ([_leakArray count]>0) {
            // 弹出警告框，提示异常信息
            [[UIApplication sharedApplication] sep_showToastWithText:[NSString stringWithFormat:@"subViews内存泄漏警告\r\n%@",_leakArray]];
        }
    });
}



@end

@implementation UINavigationController (SEPHook)
+ (void)hookUINavigationController_push{
    Method pushMethod = class_getInstanceMethod([self class], @selector(pushViewController:animated:));
    Method hookMethod = class_getInstanceMethod([self class], @selector(hook_pushViewController:animated:));
    method_exchangeImplementations(pushMethod, hookMethod);
}


- (void)hook_pushViewController:(UIViewController *)viewController animated:(BOOL)animated{
    NSString *popDetailInfo = [NSString stringWithFormat: @"%@ - %@ - %@", NSStringFromClass([self class]), @"push", NSStringFromClass([viewController class])];
    NSLog(@"%@", popDetailInfo);
    [SEPLeakDetector sharedInstance].currentViewTrackingKeys = [[SEPLeakDetector sharedInstance] globalViewTrackingObjects].keyEnumerator.allObjects;
    [self hook_pushViewController:viewController animated:animated];
}


+ (void)hookUINavigationController_pop{
    Method popMethod = class_getInstanceMethod([self class], @selector(popViewControllerAnimated:));
    Method hookMethod = class_getInstanceMethod([self class], @selector(hook_popViewControllerAnimated:));
    method_exchangeImplementations(popMethod, hookMethod);
}

- (nullable UIViewController *)hook_popViewControllerAnimated:(BOOL)animated{
    [[SEPLeakDetector sharedInstance] expectDeallocate:[self.viewControllers lastObject] isDismiss:NO];
    return [self hook_popViewControllerAnimated:animated];
}

@end

@implementation UIViewController (SEPHook)

+ (void)hookUIViewController_push{
    Method pushMethod = class_getInstanceMethod([self class], @selector(viewDidLoad));
    Method hookMethod = class_getInstanceMethod([self class], @selector(hook_ViewDidLoad));
    method_exchangeImplementations(pushMethod, hookMethod);
}

- (void)hook_ViewDidLoad{
    @synchronized (self) {
        [SEPLeakDetector sharedInstance].isLoading = YES;
        NSString *objectDescription = self.description;
        // 键：对象地址+方法名 值：对象
        NSString *pointObjString = [NSString stringWithFormat:@"%@", objectDescription];
        [[[SEPLeakDetector sharedInstance] globalTrackingObjects] setObject:self forKey:pointObjString];
    }
    dispatch_after(dispatch_time(DISPATCH_TIME_NOW, (int64_t)(0.5 * NSEC_PER_SEC)), dispatch_get_main_queue(), ^{
        @synchronized (self) {
            [SEPLeakDetector sharedInstance].isLoading = NO;
        }
    });
    [self hook_ViewDidLoad];
}

+ (void) hookUIViewController_present{
    Method popMethod = class_getInstanceMethod([self class], @selector(presentViewController:animated:completion:));
    Method hookMethod = class_getInstanceMethod([self class], @selector(hook_presentViewController:animated:completion:));
    method_exchangeImplementations(popMethod, hookMethod);
}

- (void)hook_presentViewController:(UIViewController*)controller animated: (BOOL)flag completion: (void (^ __nullable)(void))completion{
    if ([@"UIAlertController" isEqualToString:NSStringFromClass([self class])]) {
        
    }
    else{
        [SEPLeakDetector sharedInstance].currentTrackingKeys = [[SEPLeakDetector sharedInstance] globalTrackingObjects].keyEnumerator.allObjects;
        [SEPLeakDetector sharedInstance].currentViewTrackingKeys = [[SEPLeakDetector sharedInstance] globalViewTrackingObjects].keyEnumerator.allObjects;
    }
    [self hook_presentViewController:controller animated:flag completion:completion];
}


+ (void) hookUIViewController_dismiss{
    Method dismissMethod = class_getInstanceMethod([self class], @selector(dismissViewControllerAnimated:completion:));
    Method hookMethod = class_getInstanceMethod([self class], @selector(hook_dismissViewControllerAnimated:completion:));
    method_exchangeImplementations(dismissMethod, hookMethod);
}

- (void)hook_dismissViewControllerAnimated: (BOOL)flag completion: (void (^ __nullable)(void))completion{
    
    if ([@"UIApplicationRotationFollowingController" isEqualToString:NSStringFromClass([self class])]) {
        
    }
    else{
        [[SEPLeakDetector sharedInstance] expectDeallocate:self isDismiss:YES];
    }
    [self hook_dismissViewControllerAnimated:flag completion:completion];
}

@end

@implementation UIView (SEPHook)
+ (void)hookUIView_moveToSuperView{
    Method pushMethod = class_getInstanceMethod([self class], @selector(addSubview:));
    Method hookMethod = class_getInstanceMethod([self class], @selector(hook_addSubview:));
    method_exchangeImplementations(pushMethod, hookMethod);
}


- (void)hook_addSubview:(UIView*)view{
    @synchronized (self) {
        if ([[SEPLeakDetector sharedInstance] isLoading]) {
            NSString *objectDescription = self.description;
            if ([objectDescription rangeOfString:@"_"].location == NSNotFound) {
                NSString *pointObjString = [NSString stringWithFormat:@"%@", objectDescription];
                [[[SEPLeakDetector sharedInstance] globalViewTrackingObjects] setObject:self forKey:pointObjString];
            }
        }
    }
    [self hook_addSubview:view];
}
@end

拷贝代码到项目无缝集成,即可检测内存泄漏了...

总结篇

1.本文仅基于UIViewController检测
2.泄漏的原因大部分是block循环引用
3.看到弹框见喜蛋😄

最后编辑于 ：
©著作权归作者所有,转载或内容合作请联系作者
  • 人面猴
    序言：七十年代末，一起剥皮案震惊了整个滨河市，随后出现的几起案子，更是在滨河造成了极大的恐慌，老刑警刘岩，带你破解...
    沈念sama阅读 157,012评论 4赞 359
  • 死咒
    序言：滨河连续发生了三起死亡事件，死亡现场离奇诡异，居然都是意外死亡，警方通过查阅死者的电脑和手机，发现死者居然都...
    沈念sama阅读 66,589评论 1赞 290
  • 救了他两次的神仙让他今天三更去死
    文/潘晓璐 我一进店门，熙熙楼的掌柜王于贵愁眉苦脸地迎上来，“玉大人，你说我怎么就摊上这事。” “怎么了？”我有些...
    开封第一讲书人阅读 106,819评论 0赞 237
  • 道士缉凶录：失踪的卖姜人
    文/不坏的土叔 我叫张陵，是天一观的道长。 经常有香客问我，道长，这世上最难降的妖魔是什么？ 我笑而不...
    开封第一讲书人阅读 43,652评论 0赞 202
  • 港岛之恋（遗憾婚礼）
    正文 为了忘掉前任，我火速办了婚礼，结果婚礼上，老公的妹妹穿的比我还像新娘。我一直安慰自己，他们只是感情好，可当我...
    茶点故事阅读 51,954评论 3赞 285
  • 恶毒庶女顶嫁案：这布局不是一般人想出来的
    文/花漫 我一把揭开白布。 她就那样静静地躺着，像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上，一...
    开封第一讲书人阅读 40,381评论 1赞 210
  • 城市分裂传说
    那天，我揣着相机与录音，去河边找鬼。 笑死，一个胖子当着我的面吹牛，可吹牛的内容都是我干的。 我是一名探鬼主播，决...
    沈念sama阅读 31,687评论 2赞 310
  • 双鸳鸯连环套：你想象不到人心有多黑
    文/苍兰香墨 我猛地睁开眼，长吁一口气：“原来是场噩梦啊……” “哼！你这毒妇竟也来了？” 一声冷哼从身侧响起，我...
    开封第一讲书人阅读 30,404评论 0赞 194
  • 万荣杀人案实录
    序言：老挝万荣一对情侣失踪，失踪者是张志新（化名）和其女友刘颖，没想到半个月后，有当地人在树林里发现了一具尸体，经...
    沈念sama阅读 34,082评论 1赞 238
  • 护林员之死
    正文 独居荒郊野岭守林人离奇死亡，尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,355评论 2赞 241
  • 白月光启示录
    正文 我和宋清朗相恋三年，在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 31,880评论 1赞 255
  • 活死人
    序言：一个原本活蹦乱跳的男人离奇死亡，死状恐怖，灵堂内的尸体忽然破棺而出，到底是诈尸还是另有隐情，我是刑警宁泽，带...
    沈念sama阅读 28,249评论 2赞 250
  • 日本核电站爆炸内幕
    正文 年R本政府宣布，位于F岛的核电站，受9级特大地震影响，放射性物质发生泄漏。R本人自食恶果不足惜，却给世界环境...
    茶点故事阅读 32,864评论 3赞 232
  • 男人毒药：我在死后第九天来索命
    文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹，春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,007评论 0赞 8
  • 一桩弑父案，背后竟有这般阴谋
    文/苍兰香墨 我抬头看了看天上的太阳。三九已至，却和暖如春，着一层夹袄步出监牢的瞬间，已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,760评论 0赞 192
  • 情欲美人皮
    我被黑心中介骗来泰国打工， 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留，地道东北人。 一个月前我还...
    沈念sama阅读 35,394评论 2赞 269
  • 代替公主和亲
    正文 我出身青楼，却偏偏与公主长得像，于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子，可洞房花烛夜当晚...
    茶点故事阅读 35,281评论 2赞 259

推荐阅读更多精彩内容