#include <stddef.h>
#include <stdint.h>
#include <stdlib.h>
#include <string.h>
#include <mach-o/loader.h>
#include <vector>
#include "mach-o/dyld_gdb.h"
#include "mach-o/dyld_images.h"
#include "ImageLoader.h"
#include "dyld.h"
#if __IPHONE_OS_VERSION_MIN_REQUIRED
#define INITIAL_UUID_IMAGE_COUNT 4
#else
#define INITIAL_UUID_IMAGE_COUNT 32
#endif
VECTOR_NEVER_DESTRUCTED(dyld_image_info);
VECTOR_NEVER_DESTRUCTED(dyld_uuid_info);
static std::vector<dyld_image_info> sImageInfos;
static std::vector<dyld_uuid_info> sImageUUIDs;
void addImagesToAllImages(uint32_t infoCount, const dyld_image_info info[])
{
if ( sImageInfos.size() == 0 )
sImageInfos.reserve(INITIAL_IMAGE_COUNT);
if ( sImageUUIDs.capacity() == 0 )
sImageUUIDs.reserve(4);
dyld::gProcessInfo->infoArray = NULL;
for (uint32_t i=0; i < infoCount; ++i)
sImageInfos.push_back(info[i]);
dyld::gProcessInfo->infoArrayCount = sImageInfos.size();
dyld::gProcessInfo->infoArray = &sImageInfos[0];
}
#if TARGET_IPHONE_SIMULATOR
void syncProcessInfo()
{
if ( sImageInfos.size() == 0 ) {
sImageInfos.reserve(INITIAL_IMAGE_COUNT);
if ( dyld::gProcessInfo->infoArray != NULL ) {
for (uint32_t i=0; i < dyld::gProcessInfo->infoArrayCount; ++i) {
sImageInfos.push_back(dyld::gProcessInfo->infoArray[i]);
}
dyld::gProcessInfo->infoArray = &sImageInfos[0];
dyld::gProcessInfo->infoArrayCount = sImageInfos.size();
}
}
dyld::gProcessInfo->notification(dyld_image_info_change, 0, NULL);
}
#endif
const char* notifyGDB(enum dyld_image_states state, uint32_t infoCount, const dyld_image_info info[])
{
dyld::gProcessInfo->notification(dyld_image_adding, infoCount, info);
if ( dyld::gProcessInfo->initialImageCount == 0 )
dyld::gProcessInfo->initialImageCount = infoCount;
return NULL;
}
void addNonSharedCacheImageUUID(const dyld_uuid_info& info)
{
dyld::gProcessInfo->uuidArray = NULL;
sImageUUIDs.push_back(info);
dyld::gProcessInfo->uuidArrayCount = sImageUUIDs.size();
dyld::gProcessInfo->uuidArray = &sImageUUIDs[0];
}
void removeImageFromAllImages(const struct mach_header* loadAddress)
{
dyld_image_info goingAway;
dyld::gProcessInfo->infoArray = NULL;
for (std::vector<dyld_image_info>::iterator it=sImageInfos.begin(); it != sImageInfos.end(); it++) {
if ( it->imageLoadAddress == loadAddress ) {
goingAway = *it;
sImageInfos.erase(it);
break;
}
}
dyld::gProcessInfo->infoArrayCount = sImageInfos.size();
dyld::gProcessInfo->infoArray = &sImageInfos[0];
dyld::gProcessInfo->uuidArray = NULL;
for (std::vector<dyld_uuid_info>::iterator it=sImageUUIDs.begin(); it != sImageUUIDs.end(); it++) {
if ( it->imageLoadAddress == loadAddress ) {
sImageUUIDs.erase(it);
break;
}
}
dyld::gProcessInfo->uuidArrayCount = sImageUUIDs.size();
dyld::gProcessInfo->uuidArray = &sImageUUIDs[0];
dyld::gProcessInfo->notification(dyld_image_removing, 1, &goingAway);
}
void setAlImageInfosHalt(const char* message, uintptr_t flags)
{
dyld::gProcessInfo->errorMessage = message;
dyld::gProcessInfo->terminationFlags = flags;
}
#if TARGET_IPHONE_SIMULATOR
namespace dyld {
struct dyld_all_image_infos* gProcessInfo = NULL;
}
#else
#if __arm__
extern "C" void gdb_image_notifier(enum dyld_image_mode mode, uint32_t infoCount, const dyld_image_info info[]);
#else
static void gdb_image_notifier(enum dyld_image_mode mode, uint32_t infoCount, const dyld_image_info info[])
{
}
#endif
extern void* __dso_handle;
#define STR(s) # s
#define XSTR(s) STR(s)
struct dyld_all_image_infos dyld_all_image_infos __attribute__ ((section ("__DATA,__all_image_info")))
= {
14, 0, NULL, &gdb_image_notifier, false, false, (const mach_header*)&__dso_handle, NULL,
XSTR(DYLD_VERSION), NULL, 0, NULL, 0, 0, NULL, &dyld_all_image_infos,
0, dyld_error_kind_none, NULL, NULL, NULL, 0, {0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,}, 0
};
struct dyld_shared_cache_ranges dyld_shared_cache_ranges;
namespace dyld {
struct dyld_all_image_infos* gProcessInfo = &dyld_all_image_infos;
}
#endif