objc4-818.2.tar.gz

[apple/objc4.git] / runtime / objc-file.h

diff --git a/runtime/objc-file.h b/runtime/objc-file.h
index 09a8293f76fcae173fec6e5525369d2aebe0170d..597fd3ba124c9138b3efc26a8b3ec99d17b51c96 100644 (file)
--- a/runtime/objc-file.h
+++ b/runtime/objc-file.h
@@ -28,23 +28,84 @@
 
 #include "objc-runtime-new.h"
 
-
-__BEGIN_DECLS
-
 // classref_t is not fixed up at launch; use remapClass() to convert
 
+// classlist and catlist and protolist sections are marked const here
+// because those sections may be in read-only __DATA_CONST segments.
+
 extern SEL *_getObjc2SelectorRefs(const header_info *hi, size_t *count);
 extern message_ref_t *_getObjc2MessageRefs(const header_info *hi, size_t *count);
 extern Class*_getObjc2ClassRefs(const header_info *hi, size_t *count);
 extern Class*_getObjc2SuperRefs(const header_info *hi, size_t *count);
-extern classref_t *_getObjc2ClassList(const header_info *hi, size_t *count);
-extern classref_t *_getObjc2NonlazyClassList(const header_info *hi, size_t *count);
-extern category_t **_getObjc2CategoryList(const header_info *hi, size_t *count);
-extern category_t **_getObjc2NonlazyCategoryList(const header_info *hi, size_t *count);
-extern protocol_t **_getObjc2ProtocolList(const header_info *hi, size_t *count);
+extern classref_t const *_getObjc2ClassList(const header_info *hi, size_t *count);
+// Use hi->nlclslist() instead
+// extern classref_t const *_getObjc2NonlazyClassList(const header_info *hi, size_t *count);
+// Use hi->catlist() instead
+// extern category_t * const *_getObjc2CategoryList(const header_info *hi, size_t *count);
+// Use hi->catlist2() instead
+// extern category_t * const *_getObjc2CategoryList2(const header_info *hi, size_t *count);
+// Use hi->nlcatlist() instead
+// extern category_t * const *_getObjc2NonlazyCategoryList(const header_info *hi, size_t *count);
+extern protocol_t * const *_getObjc2ProtocolList(const header_info *hi, size_t *count);
 extern protocol_t **_getObjc2ProtocolRefs(const header_info *hi, size_t *count);
 
-__END_DECLS
+// FIXME: rdar://29241917&33734254 clang doesn't sign static initializers.
+struct UnsignedInitializer {
+private:
+    uintptr_t storage;
+public:
+    UnsignedInitializer(uint32_t offset) {
+        storage = (uintptr_t)&_mh_dylib_header + offset;
+    }
+
+    void operator () () const {
+        using Initializer = void(*)();
+        Initializer init =
+            ptrauth_sign_unauthenticated((Initializer)storage,
+                                         ptrauth_key_function_pointer, 0);
+        init();
+    }
+};
+
+extern UnsignedInitializer *getLibobjcInitializers(const header_info *hi, size_t *count);
+
+extern classref_t const *_getObjc2NonlazyClassList(const headerType *mhdr, size_t *count);
+extern category_t * const *_getObjc2CategoryList(const headerType *mhdr, size_t *count);
+extern category_t * const *_getObjc2CategoryList2(const headerType *mhdr, size_t *count);
+extern category_t * const *_getObjc2NonlazyCategoryList(const headerType *mhdr, size_t *count);
+extern UnsignedInitializer *getLibobjcInitializers(const headerType *mhdr, size_t *count);
+extern uint32_t *getLibobjcInitializerOffsets(const headerType *hi, size_t *count);
+
+static inline void
+foreach_data_segment(const headerType *mhdr,
+                     std::function<void(const segmentType *, intptr_t slide)> code)
+{
+    intptr_t slide = 0;
+
+    // compute VM slide
+    const segmentType *seg = (const segmentType *) (mhdr + 1);
+    for (unsigned long i = 0; i < mhdr->ncmds; i++) {
+        if (seg->cmd == SEGMENT_CMD  &&
+            segnameEquals(seg->segname, "__TEXT"))
+        {
+            slide = (char *)mhdr - (char *)seg->vmaddr;
+            break;
+        }
+        seg = (const segmentType *)((char *)seg + seg->cmdsize);
+    }
+
+    // enumerate __DATA* and __AUTH* segments
+    seg = (const segmentType *) (mhdr + 1);
+    for (unsigned long i = 0; i < mhdr->ncmds; i++) {
+        if (seg->cmd == SEGMENT_CMD  &&
+            (segnameStartsWith(seg->segname, "__DATA") ||
+             segnameStartsWith(seg->segname, "__AUTH")))
+        {
+            code(seg, slide);
+        }
+        seg = (const segmentType *)((char *)seg + seg->cmdsize);
+    }
+}
 
 #endif