make subspec

Author: gsdios

SDHook/SDMRCTool.m

@@ -10,22 +10,6 @@
 #include <objc/runtime.h>
 #import "SDMagicHookUtils.h"
 
-extern Class _Nullable (*sd_original_setclass) (id _Nullable obj, Class _Nonnull cls);
-
-Class _Nullable sd_magichook_set_calss(id _Nullable obj, Class _Nonnull cls) {
-
-    if (sd_ifClassNameHasPrefix(cls, "NSKVONotifying_")) {
-        Class originalClass = object_getClass(obj);
-        if (sd_ifClassIsSDMagicClass(originalClass)) {
-            return originalClass;
-        } else {
-            return sd_original_setclass(obj, cls);
-        }
-    } else {
-        return sd_original_setclass(obj, cls);
-    }
-}
-
 @implementation SDMRCTool
 
 + (void)object_copyIndexedIvars:(id)obj toCopy:(id)toCopy {

SDHook/SDMagicHook.m

@@ -20,8 +20,6 @@
 #import "SDMagicHookUtils.h"
 
 BOOL SDMagicHookDebugFlag = true;
-Class _Nullable (*sd_original_setclass) (id _Nullable obj, Class _Nonnull cls);
-extern Class _Nullable sd_magichook_set_calss(id _Nullable obj, Class _Nonnull cls);
 
 static NSString *const currentCallIndexDictKey = @"SDMagicHook-currentCallIndexDictKey";
 static NSString *const originalCallFlagDictKey = @"SDMagicHook-originalCallFlagDictKey";
@@ -143,10 +141,7 @@ - (BOOL)isSDMagicHookClass:(Class)cls {
 - (NSString *)hookMethod:(SEL)sel strId:(NSString *)strId imp:(IMP)imp {
     pthread_rwlock_wrlock(&[self getManagerLock]->_rw_lock3);
 
-    static dispatch_once_t onceToken;
-    dispatch_once(&onceToken, ^{
-        rebind_symbols((struct rebinding[1]){{"object_setClass", sd_magichook_set_calss, (void *)&sd_original_setclass}}, 1);
-    });
+    [SDMagicHookUtils hookSetClassFuncJustOnce];
 
     SDNewClassManager *mgr = [self getClassManager];
     if (!mgr.hasSetupKVO) {

SDHook/SDMagicHookUtils.h

@@ -29,6 +29,8 @@ NSMutableDictionary *threadStoredDict(const NSString *const key, NSString *class
 
 @interface SDMagicHookUtils : NSObject
 
++ (void)hookSetClassFuncJustOnce;
+
 @end
 
 NS_ASSUME_NONNULL_END

SDHook/SDMagicHookUtils.m

@@ -8,6 +8,23 @@
 #import "SDMagicHookUtils.h"
 #import <pthread.h>
 #import <objc/runtime.h>
+#import <fishhook/fishhook.h>
+
+Class _Nullable (*sd_original_setclass) (id _Nullable obj, Class _Nonnull cls);
+
+Class _Nullable sd_magichook_set_calss(id _Nullable obj, Class _Nonnull cls) {
+
+    if (sd_ifClassNameHasPrefix(cls, "NSKVONotifying_")) {
+        Class originalClass = object_getClass(obj);
+        if (sd_ifClassIsSDMagicClass(originalClass)) {
+            return originalClass;
+        } else {
+            return sd_original_setclass(obj, cls);
+        }
+    } else {
+        return sd_original_setclass(obj, cls);
+    }
+}
 
 bool sd_ifClassNameHasPrefix(Class cls, const char *prefix) {
     const char *clsCStrName = class_getName(cls);
@@ -63,4 +80,11 @@ SEL createSelB(SEL sel, int index, NSString *flag) {
 
 @implementation SDMagicHookUtils
 
++ (void)hookSetClassFuncJustOnce {
+    static dispatch_once_t onceToken;
+    dispatch_once(&onceToken, ^{
+        rebind_symbols((struct rebinding[1]){{"object_setClass", sd_magichook_set_calss, (void *)&sd_original_setclass}}, 1);
+    });
+}
+
 @end