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 extern classref_t const *_getObjc2NonlazyClassList(const header_info *hi, size_t *count);
  42 extern category_t * const *_getObjc2CategoryList(const header_info *hi, size_t *count);
  43 extern category_t * const *_getObjc2CategoryList2(const header_info *hi, size_t *count);
  44 extern category_t * const *_getObjc2NonlazyCategoryList(const header_info *hi, size_t *count);
  45 extern protocol_t * const *_getObjc2ProtocolList(const header_info *hi, size_t *count);
  46 extern protocol_t **_getObjc2ProtocolRefs(const header_info *hi, size_t *count);
  47
  48 // FIXME: rdar://29241917&33734254 clang doesn't sign static initializers.
  49 struct UnsignedInitializer {
  50 private:
  51     uintptr_t storage;
  52 public:
  53     void operator () () const {
  54         using Initializer = void(*)();
  55         Initializer init =
  56             ptrauth_sign_unauthenticated((Initializer)storage,
  57                                          ptrauth_key_function_pointer, 0);
  58         init();
  59     }
  60 };
  61
  62 extern UnsignedInitializer *getLibobjcInitializers(const header_info *hi, size_t *count);
  63
  64 extern classref_t const *_getObjc2NonlazyClassList(const headerType *mhdr, size_t *count);
  65 extern category_t * const *_getObjc2NonlazyCategoryList(const headerType *mhdr, size_t *count);
  66 extern UnsignedInitializer *getLibobjcInitializers(const headerType *mhdr, size_t *count);
  67
  68 static inline void
  69 foreach_data_segment(const headerType *mhdr,
  70                      std::function<void(const segmentType *, intptr_t slide)> code)
  71 {
  72     intptr_t slide = 0;
  73
  74     // compute VM slide
  75     const segmentType *seg = (const segmentType *) (mhdr + 1);
  76     for (unsigned long i = 0; i < mhdr->ncmds; i++) {
  77         if (seg->cmd == SEGMENT_CMD  &&
  78             segnameEquals(seg->segname, "__TEXT"))
  79         {
  80             slide = (char *)mhdr - (char *)seg->vmaddr;
  81             break;
  82         }
  83         seg = (const segmentType *)((char *)seg + seg->cmdsize);
  84     }
  85
  86     // enumerate __DATA* segments
  87     seg = (const segmentType *) (mhdr + 1);
  88     for (unsigned long i = 0; i < mhdr->ncmds; i++) {
  89         if (seg->cmd == SEGMENT_CMD  &&
  90             segnameStartsWith(seg->segname, "__DATA"))
  91         {
  92             code(seg, slide);
  93         }
  94         seg = (const segmentType *)((char *)seg + seg->cmdsize);
  95     }
  96 }
  97
  98 #endif
  99
 100 #endif