#ifndef _OBJC_FILE_NEW_H
#define _OBJC_FILE_NEW_H
#if __OBJC2__
#include "objc-runtime-new.h"
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 protocol_t **_getObjc2ProtocolRefs(const header_info *hi, size_t *count);
struct UnsignedInitializer {
private:
uintptr_t storage;
public:
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 *_getObjc2NonlazyClassList(const headerType *mhdr, size_t *count);
extern category_t **_getObjc2NonlazyCategoryList(const headerType *mhdr, size_t *count);
extern UnsignedInitializer *getLibobjcInitializers(const headerType *mhdr, 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;
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);
}
seg = (const segmentType *) (mhdr + 1);
for (unsigned long i = 0; i < mhdr->ncmds; i++) {
if (seg->cmd == SEGMENT_CMD &&
segnameStartsWith(seg->segname, "__DATA"))
{
code(seg, slide);
}
seg = (const segmentType *)((char *)seg + seg->cmdsize);
}
}
#endif
#endif