support the UnrecognizedSelector for the class method

Author: 李杰

JJException/Source/MRC/NSObject+UnrecognizedSelectorHook.m

@@ -14,18 +14,45 @@
 @implementation NSObject (UnrecognizedSelectorHook)
 
 + (void)jj_swizzleUnrecognizedSelector{
+    
+    //Class Method
+    swizzleClassMethod([self class], @selector(methodSignatureForSelector:), @selector(classMethodSignatureForSelectorSwizzled:));
+    swizzleClassMethod([self class], @selector(forwardInvocation:), @selector(forwardClassInvocationSwizzled:));
+    
+    //Instance Method
     swizzleInstanceMethod([self class], @selector(methodSignatureForSelector:), @selector(methodSignatureForSelectorSwizzled:));
     swizzleInstanceMethod([self class], @selector(forwardInvocation:), @selector(forwardInvocationSwizzled:));
 }
 
++ (NSMethodSignature*)classMethodSignatureForSelectorSwizzled:(SEL)aSelector {
+    NSMethodSignature* methodSignature = [self classMethodSignatureForSelectorSwizzled:aSelector];
+    if (methodSignature) {
+        return methodSignature;
+    }
+    
+    return [self.class checkObjectSignatureAndCurrentClass:self.class];
+}
+
 - (NSMethodSignature*)methodSignatureForSelectorSwizzled:(SEL)aSelector {
     NSMethodSignature* methodSignature = [self methodSignatureForSelectorSwizzled:aSelector];
     if (methodSignature) {
         return methodSignature;
     }
 
+    return [self.class checkObjectSignatureAndCurrentClass:self.class];
+}
+
+/**
+ * Check the class method signature to the [NSObject class]
+ * If not equals,return nil
+ * If equals,return the v@:@ method
+
+ @param currentClass Class
+ @return NSMethodSignature
+ */
++ (NSMethodSignature *)checkObjectSignatureAndCurrentClass:(Class)currentClass{
     IMP originIMP = class_getMethodImplementation([NSObject class], @selector(methodSignatureForSelector:));
-    IMP currentClassIMP = class_getMethodImplementation(self.class, @selector(methodSignatureForSelector:));
+    IMP currentClassIMP = class_getMethodImplementation(currentClass, @selector(methodSignatureForSelector:));
 
     // If current class override methodSignatureForSelector return nil
     if (originIMP != currentClassIMP){
@@ -37,8 +64,23 @@ - (NSMethodSignature*)methodSignatureForSelectorSwizzled:(SEL)aSelector {
     return [NSMethodSignature signatureWithObjCTypes:"v@:@"];
 }
 
+/**
+ Forward instance object
+
+ @param invocation NSInvocation
+ */
 - (void)forwardInvocationSwizzled:(NSInvocation*)invocation{
-    NSString* message = [NSString stringWithFormat:@"Unrecognized selector class:%@ and selector:%@",NSStringFromClass(self.class),NSStringFromSelector(invocation.selector)];
+    NSString* message = [NSString stringWithFormat:@"Unrecognized instance class:%@ and selector:%@",NSStringFromClass(self.class),NSStringFromSelector(invocation.selector)];
+    handleCrashException(JJExceptionGuardUnrecognizedSelector,message);
+}
+
+/**
+ Forward class object
+
+ @param invocation NSInvocation
+ */
++ (void)forwardClassInvocationSwizzled:(NSInvocation*)invocation{
+    NSString* message = [NSString stringWithFormat:@"Unrecognized static class:%@ and selector:%@",NSStringFromClass(self.class),NSStringFromSelector(invocation.selector)];
     handleCrashException(JJExceptionGuardUnrecognizedSelector,message);
 }