温馨提示×

温馨提示×

您好,登录后才能下订单哦!

密码登录×
登录注册×
其他方式登录
点击 登录注册 即表示同意《亿速云用户服务条款》

iOS中如何实现检测Zoombie对象功能

发布时间:2021-06-28 13:49:15 来源:亿速云 阅读:133 作者:小新 栏目:移动开发

这篇文章将为大家详细讲解有关iOS中如何实现检测Zoombie对象功能,小编觉得挺实用的,因此分享给大家做个参考,希望大家阅读完这篇文章后可以有所收获。

前言

我们大家都知道,如果在XCode中开启了Zoombie Objects。如图。

iOS中如何实现检测Zoombie对象功能

那么在一个对象释放后,再次给该对象发送消息,在Xcode控制台中,可看到如下打印信息。这些信息可以帮助我们定位问题。

ZoombieDemo[12275:2841478] *** -[Test test]: message sent to deallocated instance 0x60800000b000

那么究竟XCode是如何实现僵尸对象的检查的,我们将来一一揭晓。

实现原理

在《Effective Objective-C 》一书中有提到过僵尸指针的实现方式。

通过hook NSObject的dealloc的方法,在一个对象要释放的时候,通过objcduplicateClass复制NSZombie类,生成NSZombieOriginaClass,并且将当前对象的isa指向新生成的类。这块内存不会释放。

因为在给该对象发消息时,NSZombieOriginaClass并未实现原有类的方法,所以会走完整的消息转发。所以我们能取出具体的OriginaClass(去掉NS_Zombie),当前sel,打印出来。

[class seletor]:message sent to deallocated instance 0x22909"

简单来说,就是将对象指向一个新的类,因为新类里面并没有原有类方法的实现,所以必定会走到消息转发中。

以上说的是动态生成新的类,类名是通过固定前缀拼接而成,将isa指向该类。其实还有一种方式,就是指向固定的类,原有类名通过关联对象的方式来存储。

既然知道了原理,可以动手实现一下。

动手实现

首先是hook dealloc方法。在NSObject+HookDealloc中实现。

+ (void)load {
 static dispatch_once_t onceToken;
 dispatch_once(&onceToken, ^{
  Class class = [self class];
  SEL originalSelector = NSSelectorFromString(@"dealloc");
  SEL swizzledSelector = @selector(swizzledDealloc);
  Method originalMethod = class_getInstanceMethod(class, originalSelector);
  Method swizzledMethod = class_getInstanceMethod(class, swizzledSelector);  
  BOOL success = class_addMethod(class, originalSelector, method_getImplementation(swizzledMethod), method_getTypeEncoding(swizzledMethod));
  if (success) {
   class_replaceMethod(class, swizzledSelector, method_getImplementation(originalMethod), method_getTypeEncoding(originalMethod));
  } else {
   method_exchangeImplementations(originalMethod, swizzledMethod);
  }
 });
}

动态生成新的类

在swizzledDealloc中,我们通过"Zoombie_"拼接原始类名,得到一个新的类名。然后生成该类,添加 forwardingTargetForSelector的实现。便于在消息转发的时候得到调用信息。

NSString *Zoombie_Class_Prefix = @"Zoombie_";
// 指向动态生成的类,用Zoombie拼接原有类名
NSString *className = NSStringFromClass([self class]);
NSString *zombieClassName = [Zoombie_Class_Prefix stringByAppendingString: className]; 
Class zombieClass = NSClassFromString(zombieClassName);
if(zombieClass) return; 
zombieClass = objc_allocateClassPair([NSObject class], [zombieClassName UTF8String], 0); 
objc_registerClassPair(zombieClass);
class_addMethod([zombieClass class], @selector(forwardingTargetForSelector:), (IMP)forwardingTargetForSelector, "@@:@");
object_setClass(self, zombieClass);

forwardingTargetForSelector的方法实现,原始类名,去掉前缀即可得到。因为这里已经是调用到已释放对象的方法,我们直接abort掉,程序将崩溃。

id forwardingTargetForSelector(id self, SEL _cmd, SEL aSelector) {
 NSString *className = NSStringFromClass([self class]);
 NSString *realClass = [className stringByReplacingOccurrencesOfString:Zoombie_Class_Prefix withString:@""];
 NSLog(@"[%@ %@] message sent to deallocated instance %@", realClass, NSStringFromSelector(aSelector), self);
 abort();
}

指向固定类

指向已有的ZoombieObject类,类名存在关联对象中。

 // 指向固定的类,原有类名存储在关联对象中
NSString *originClassName = NSStringFromClass([self class]);
objc_setAssociatedObject(self, "OrigClassNameKey", originClassName, OBJC_ASSOCIATION_COPY_NONATOMIC);
object_setClass(self, [ZoombieObject class]);

同上,在ZoombieObject中实现forwardingTargetForSelector方法,可以得到调用信息。原始类名通过关联对象获取。

- (id)forwardingTargetForSelector:(SEL)aSelector {
 NSLog(@"[%@ %@] message sent to deallocated instance %@", objc_getAssociatedObject(self, "OrigClassNameKey"), NSStringFromSelector(aSelector), self);
 abort();
}

forwardingTargetForSelector是消息转发的第二步,我们也可以不在这里处理,等到最后一步forwardInvocation,不过要生成方法签名,要略微复杂些。

要想走到forwardInvocation,methodSignatureForSelector返回不能是空。这里我们返回了StubProxy类中stub的方法签名(已经定义好的类和方法),最后就回走到forwardInvocation,通过invocation.selector可得到当前调用方法名。通过关联对象获取到原始类名。

- (NSMethodSignature *)methodSignatureForSelector:(SEL)aSelector {
 NSMethodSignature *sig = [super methodSignatureForSelector:aSelector];
 if (!sig) {
  sig = [StubProxy instanceMethodSignatureForSelector:@selector(stub)];
 } 
 return sig;
}
- (void)forwardInvocation:(NSInvocation *)anInvocation {
 NSLog(@"[%@ %@] message sent to deallocated instance %@", objc_getAssociatedObject(self, "OrigClassNameKey"), NSStringFromSelector(anInvocation.selector), self);
}

这样,一个简单的检测僵尸指针的方案就实现了。

demo在此。

两种方式都实现了,可通过调整NSObject+HookDealloc中,swizzledSelector的值来切换。my_dealloc是指向动态类,swizzledDealloc是指向固定类。

SEL swizzledSelector = @selector(my_dealloc);

在App运行起来后,点击button,即可触发。

关于“iOS中如何实现检测Zoombie对象功能”这篇文章就分享到这里了,希望以上内容可以对大家有一定的帮助,使各位可以学到更多知识,如果觉得文章不错,请把它分享出去让更多的人看到。

向AI问一下细节

免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。

ios
AI