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     void operator () () const {
  58         using Initializer = void(*)();
  59         Initializer init =
  60             ptrauth_sign_unauthenticated((Initializer)storage,
  61                                          ptrauth_key_function_pointer, 0);
  62         init();
  63     }
  64 };
  65
  66 extern UnsignedInitializer *getLibobjcInitializers(const header_info *hi, size_t *count);
  67
  68 extern classref_t const *_getObjc2NonlazyClassList(const headerType *mhdr, size_t *count);
  69 extern category_t * const *_getObjc2CategoryList(const headerType *mhdr, size_t *count);
  70 extern category_t * const *_getObjc2CategoryList2(const headerType *mhdr, size_t *count);
  71 extern category_t * const *_getObjc2NonlazyCategoryList(const headerType *mhdr, size_t *count);
  72 extern UnsignedInitializer *getLibobjcInitializers(const headerType *mhdr, size_t *count);
  73
  74 static inline void
  75 foreach_data_segment(const headerType *mhdr,
  76                      std::function<void(const segmentType *, intptr_t slide)> code)
  77 {
  78     intptr_t slide = 0;
  79
  80     // compute VM slide
  81     const segmentType *seg = (const segmentType *) (mhdr + 1);
  82     for (unsigned long i = 0; i < mhdr->ncmds; i++) {
  83         if (seg->cmd == SEGMENT_CMD  &&
  84             segnameEquals(seg->segname, "__TEXT"))
  85         {
  86             slide = (char *)mhdr - (char *)seg->vmaddr;
  87             break;
  88         }
  89         seg = (const segmentType *)((char *)seg + seg->cmdsize);
  90     }
  91
  92     // enumerate __DATA* segments
  93     seg = (const segmentType *) (mhdr + 1);
  94     for (unsigned long i = 0; i < mhdr->ncmds; i++) {
  95         if (seg->cmd == SEGMENT_CMD  &&
  96             segnameStartsWith(seg->segname, "__DATA"))
  97         {
  98             code(seg, slide);
  99         }
 100         seg = (const segmentType *)((char *)seg + seg->cmdsize);
 101     }
 102 }
 103
 104 #endif
 105
 106 #endif