runtime/objc-file.h

   1 /*
   2  * Copyright (c) 2009 Apple Inc.  All Rights Reserved.
   3  *
   4  * @APPLE_LICENSE_HEADER_START@
   5  *
   6  * This file contains Original Code and/or Modifications of Original Code
   7  * as defined in and that are subject to the Apple Public Source License
   8  * Version 2.0 (the 'License'). You may not use this file except in
   9  * compliance with the License. Please obtain a copy of the License at
  10  * http://www.opensource.apple.com/apsl/ and read it before using this
  11  * file.
  12  *
  13  * The Original Code and all software distributed under the License are
  14  * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
  15  * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
  16  * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
  17  * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
  18  * Please see the License for the specific language governing rights and
  19  * limitations under the License.
  20  *
  21  * @APPLE_LICENSE_HEADER_END@
  22  */
  23
  24 #ifndef _OBJC_FILE_NEW_H
  25 #define _OBJC_FILE_NEW_H
  26
  27 #if __OBJC2__
  28
  29 #include "objc-runtime-new.h"
  30
  31 // classref_t is not fixed up at launch; use remapClass() to convert
  32
  33 // classlist and catlist and protolist sections are marked const here
  34 // because those sections may be in read-only __DATA_CONST segments.
  35
  36 extern SEL *_getObjc2SelectorRefs(const header_info *hi, size_t *count);
  37 extern message_ref_t *_getObjc2MessageRefs(const header_info *hi, size_t *count);
  38 extern Class*_getObjc2ClassRefs(const header_info *hi, size_t *count);
  39 extern Class*_getObjc2SuperRefs(const header_info *hi, size_t *count);
  40 extern classref_t const *_getObjc2ClassList(const header_info *hi, size_t *count);
  41 // Use hi->nlclslist() instead
  42 // extern classref_t const *_getObjc2NonlazyClassList(const header_info *hi, size_t *count);
  43 // Use hi->catlist() instead
  44 // extern category_t * const *_getObjc2CategoryList(const header_info *hi, size_t *count);
  45 // Use hi->catlist2() instead
  46 // extern category_t * const *_getObjc2CategoryList2(const header_info *hi, size_t *count);
  47 // Use hi->nlcatlist() instead
  48 // extern category_t * const *_getObjc2NonlazyCategoryList(const header_info *hi, size_t *count);
  49 extern protocol_t * const *_getObjc2ProtocolList(const header_info *hi, size_t *count);
  50 extern protocol_t **_getObjc2ProtocolRefs(const header_info *hi, size_t *count);
  51
  52 // FIXME: rdar://29241917&33734254 clang doesn't sign static initializers.
  53 struct UnsignedInitializer {
  54 private:
  55     uintptr_t storage;
  56 public:
  57     UnsignedInitializer(uint32_t offset) {
  58         storage = (uintptr_t)&_mh_dylib_header + offset;
  59     }
  60
  61     void operator () () const {
  62         using Initializer = void(*)();
  63         Initializer init =
  64             ptrauth_sign_unauthenticated((Initializer)storage,
  65                                          ptrauth_key_function_pointer, 0);
  66         init();
  67     }
  68 };
  69
  70 extern UnsignedInitializer *getLibobjcInitializers(const header_info *hi, size_t *count);
  71
  72 extern classref_t const *_getObjc2NonlazyClassList(const headerType *mhdr, size_t *count);
  73 extern category_t * const *_getObjc2CategoryList(const headerType *mhdr, size_t *count);
  74 extern category_t * const *_getObjc2CategoryList2(const headerType *mhdr, size_t *count);
  75 extern category_t * const *_getObjc2NonlazyCategoryList(const headerType *mhdr, size_t *count);
  76 extern UnsignedInitializer *getLibobjcInitializers(const headerType *mhdr, size_t *count);
  77 extern uint32_t *getLibobjcInitializerOffsets(const headerType *hi, size_t *count);
  78
  79 static inline void
  80 foreach_data_segment(const headerType *mhdr,
  81                      std::function<void(const segmentType *, intptr_t slide)> code)
  82 {
  83     intptr_t slide = 0;
  84
  85     // compute VM slide
  86     const segmentType *seg = (const segmentType *) (mhdr + 1);
  87     for (unsigned long i = 0; i < mhdr->ncmds; i++) {
  88         if (seg->cmd == SEGMENT_CMD  &&
  89             segnameEquals(seg->segname, "__TEXT"))
  90         {
  91             slide = (char *)mhdr - (char *)seg->vmaddr;
  92             break;
  93         }
  94         seg = (const segmentType *)((char *)seg + seg->cmdsize);
  95     }
  96
  97     // enumerate __DATA* and __AUTH* segments
  98     seg = (const segmentType *) (mhdr + 1);
  99     for (unsigned long i = 0; i < mhdr->ncmds; i++) {
 100         if (seg->cmd == SEGMENT_CMD  &&
 101             (segnameStartsWith(seg->segname, "__DATA") ||
 102              segnameStartsWith(seg->segname, "__AUTH")))
 103         {
 104             code(seg, slide);
 105         }
 106         seg = (const segmentType *)((char *)seg + seg->cmdsize);
 107     }
 108 }
 109
 110 #endif
 111
 112 #endif