docs: refine

assets/MockClassInfo.h

@@ -0,0 +1,143 @@
+//
+//  MockClassInfo.h
+//  TestClass
+//
+//  Created by MJ Lee on 2018/3/8.
+//  Copyright © 2018年 MJ Lee. All rights reserved.
+//
+
+#import <Foundation/Foundation.h>
+
+#ifndef MockClassInfo_h
+#define MockClassInfo_h
+
+# if __arm64__
+#   define ISA_MASK        0x0000000ffffffff8ULL
+# elif __x86_64__
+#   define ISA_MASK        0x00007ffffffffff8ULL
+# endif
+
+#if __LP64__
+typedef uint32_t mask_t;
+#else
+typedef uint16_t mask_t;
+#endif
+typedef uintptr_t cache_key_t;
+
+struct bucket_t {
+    cache_key_t _key;
+    IMP _imp;
+};
+
+struct cache_t {
+    bucket_t *_buckets;
+    mask_t _mask;
+    mask_t _occupied;
+};
+
+struct entsize_list_tt {
+    uint32_t entsizeAndFlags;
+    uint32_t count;
+};
+
+struct method_t {
+    SEL name;
+    const char *types;
+    IMP imp;
+};
+
+struct method_list_t : entsize_list_tt {
+    method_t first;
+};
+
+struct ivar_t {
+    int32_t *offset;
+    const char *name;
+    const char *type;
+    uint32_t alignment_raw;
+    uint32_t size;
+};
+
+struct ivar_list_t : entsize_list_tt {
+    ivar_t first;
+};
+
+struct property_t {
+    const char *name;
+    const char *attributes;
+};
+
+struct property_list_t : entsize_list_tt {
+    property_t first;
+};
+
+struct chained_property_list {
+    chained_property_list *next;
+    uint32_t count;
+    property_t list[0];
+};
+
+typedef uintptr_t protocol_ref_t;
+struct protocol_list_t {
+    uintptr_t count;
+    protocol_ref_t list[0];
+};
+
+struct class_ro_t {
+    uint32_t flags;
+    uint32_t instanceStart;
+    uint32_t instanceSize;  // instance对象占用的内存空间
+#ifdef __LP64__
+    uint32_t reserved;
+#endif
+    const uint8_t * ivarLayout;
+    const char * name;  // 类名
+    method_list_t * baseMethodList;
+    protocol_list_t * baseProtocols;
+    const ivar_list_t * ivars;  // 成员变量列表
+    const uint8_t * weakIvarLayout;
+    property_list_t *baseProperties;
+};
+
+struct class_rw_t {
+    uint32_t flags;
+    uint32_t version;
+    const class_ro_t *ro;
+    method_list_t * methods;    // 方法列表
+    property_list_t *properties;    // 属性列表
+    const protocol_list_t * protocols;  // 协议列表
+    Class firstSubclass;
+    Class nextSiblingClass;
+    char *demangledName;
+};
+
+#define FAST_DATA_MASK          0x00007ffffffffff8UL
+struct class_data_bits_t {
+    uintptr_t bits;
+public:
+    class_rw_t* data() {
+        return (class_rw_t *)(bits & FAST_DATA_MASK);
+    }
+};
+
+/* OC对象 */
+struct mock_objc_object {
+    void *isa;
+};
+
+/* 类对象 */
+struct mock_objc_class : mock_objc_object {
+    Class superclass;
+    cache_t cache;
+    class_data_bits_t bits;
+public:
+    class_rw_t* data() {
+        return bits.data();
+    }
+    
+    mock_objc_class* metaClass() {
+        return (mock_objc_class *)((long long)isa & ISA_MASK);
+    }
+};
+
+#endif /* MockClassInfo_h */

assets/__forwarding__clean.c

@@ -0,0 +1,35 @@
+int __forwarding__(void *frameStackPointer, int isStret) {
+    id receiver = *(id *)frameStackPointer;
+    SEL sel = *(SEL *)(frameStackPointer + 8);
+    const char *selName = sel_getName(sel);
+    Class receiverClass = object_getClass(receiver);
+
+    // 调用 forwardingTargetForSelector:
+    if (class_respondsToSelector(receiverClass, @selector(forwardingTargetForSelector:))) {
+        id forwardingTarget = [receiver forwardingTargetForSelector:sel];
+        if (forwardingTarget && forwardingTarget != receiver) {
+            return objc_msgSend(forwardingTarget, sel, ...);
+        }
+    }
+
+    // 调用 methodSignatureForSelector 获取方法签名后再调用 forwardInvocation
+    if (class_respondsToSelector(receiverClass, @selector(methodSignatureForSelector:))) {
+        NSMethodSignature *methodSignature = [receiver methodSignatureForSelector:sel];
+        if (methodSignature && class_respondsToSelector(receiverClass, @selector(forwardInvocation:))) {
+            NSInvocation *invocation = [NSInvocation _invocationWithMethodSignature:methodSignature frame:frameStackPointer];
+
+            [receiver forwardInvocation:invocation];
+
+            void *returnValue = NULL;
+            [invocation getReturnValue:&value];
+            return returnValue;
+        }
+    }
+
+    if (class_respondsToSelector(receiverClass,@selector(doesNotRecognizeSelector:))) {
+        [receiver doesNotRecognizeSelector:sel];
+    }
+
+    // The point of no return.
+    kill(getpid(), 9);
+}