123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110 |
- #if __OBJC2__
- #include "objc-private.h"
- #include "objc-file.h"
- template <typename T>
- T* getDataSection(const headerType *mhdr, const char *sectname,
- size_t *outBytes, size_t *outCount)
- {
- unsigned long byteCount = 0;
- T* data = (T*)getsectiondata(mhdr, "__DATA", sectname, &byteCount);
- if (!data) {
- data = (T*)getsectiondata(mhdr, "__DATA_CONST", sectname, &byteCount);
- }
- if (!data) {
- data = (T*)getsectiondata(mhdr, "__DATA_DIRTY", sectname, &byteCount);
- }
- if (outBytes) *outBytes = byteCount;
- if (outCount) *outCount = byteCount / sizeof(T);
- return data;
- }
- #define GETSECT(name, type, sectname) \
- type *name(const headerType *mhdr, size_t *outCount) { \
- return getDataSection<type>(mhdr, sectname, nil, outCount); \
- } \
- type *name(const header_info *hi, size_t *outCount) { \
- return getDataSection<type>(hi->mhdr(), sectname, nil, outCount); \
- }
- GETSECT(_getObjc2SelectorRefs, SEL, "__objc_selrefs");
- GETSECT(_getObjc2MessageRefs, message_ref_t, "__objc_msgrefs");
- GETSECT(_getObjc2ClassRefs, Class, "__objc_classrefs");
- GETSECT(_getObjc2SuperRefs, Class, "__objc_superrefs");
- GETSECT(_getObjc2ClassList, classref_t, "__objc_classlist");
- GETSECT(_getObjc2NonlazyClassList, classref_t, "__objc_nlclslist");
- GETSECT(_getObjc2CategoryList, category_t *, "__objc_catlist");
- GETSECT(_getObjc2NonlazyCategoryList, category_t *, "__objc_nlcatlist");
- GETSECT(_getObjc2ProtocolList, protocol_t *, "__objc_protolist");
- GETSECT(_getObjc2ProtocolRefs, protocol_t *, "__objc_protorefs");
- GETSECT(getLibobjcInitializers, UnsignedInitializer, "__objc_init_func");
- objc_image_info *
- _getObjcImageInfo(const headerType *mhdr, size_t *outBytes)
- {
- return getDataSection<objc_image_info>(mhdr, "__objc_imageinfo",
- outBytes, nil);
- }
- static bool segmentHasObjcContents(const segmentType *seg)
- {
- for (uint32_t i = 0; i < seg->nsects; i++) {
- const sectionType *sect = ((const sectionType *)(seg+1))+i;
- if (sectnameStartsWith(sect->sectname, "__objc_") &&
- !sectnameEquals(sect->sectname, "__objc_imageinfo"))
- {
- return true;
- }
- }
- return false;
- }
- bool
- _hasObjcContents(const header_info *hi)
- {
- bool foundObjC = false;
- foreach_data_segment(hi->mhdr(), [&](const segmentType *seg, intptr_t slide)
- {
- if (segmentHasObjcContents(seg)) foundObjC = true;
- });
- return foundObjC;
-
- }
- #endif
|