#include <stddef.h>
#include <stdint.h>
#include <stdlib.h>
#include <string.h>
#include <mach-o/loader.h>
#include <mach-o/dyld_images.h>
#include <mach-o/dyld_process_info.h>
#include <vector>
#include "Tracing.h"
#include "ImageLoader.h"
#include "dyld2.h"
extern "C" void _dyld_debugger_notification(enum dyld_notify_mode mode, unsigned long count, uint64_t machHeaders[]);
#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;
size_t allImagesCount()
{
return sImageInfos.size();
}
const mach_header* allImagesIndexedMachHeader(uint32_t index)
{
if ( index < sImageInfos.size() )
return sImageInfos[index].imageLoadAddress;
else
return NULL;
}
const char* allImagesIndexedPath(uint32_t index)
{
if ( index < sImageInfos.size() )
return sImageInfos[index].imageFilePath;
else
return NULL;
}
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 = (uint32_t)sImageInfos.size();
dyld::gProcessInfo->infoArrayChangeTimestamp = mach_absolute_time();
dyld::gProcessInfo->infoArray = &sImageInfos[0];
}
#if TARGET_OS_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 = (uint32_t)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[])
{
uint64_t t0 = mach_absolute_time();
dyld::gProcessInfo->notification(dyld_image_adding, infoCount, info);
uint64_t t1 = mach_absolute_time();
ImageLoader::fgTotalDebuggerPausedTime += (t1-t0);
if ( dyld::gProcessInfo->initialImageCount == 0 )
dyld::gProcessInfo->initialImageCount = dyld::gProcessInfo->infoArrayCount;
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 = (uint32_t)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->infoArrayChangeTimestamp = mach_absolute_time();
dyld::gProcessInfo->uuidArray = &sImageUUIDs[0];
dyld::gProcessInfo->notification(dyld_image_removing, 1, &goingAway);
}
#if TARGET_OS_SIMULATOR
namespace dyld {
struct dyld_all_image_infos* gProcessInfo = NULL;
}
#else
static void gdb_image_notifier(enum dyld_image_mode mode, uint32_t infoCount, const dyld_image_info info[])
{
dyld3::ScopedTimer(DBG_DYLD_GDB_IMAGE_NOTIFIER, 0, 0, 0);
uint64_t machHeaders[infoCount];
for (uint32_t i=0; i < infoCount; ++i) {
machHeaders[i] = (uintptr_t)(info[i].imageLoadAddress);
}
switch ( mode ) {
case dyld_image_adding:
_dyld_debugger_notification(dyld_notify_adding, infoCount, machHeaders);
break;
case dyld_image_removing:
_dyld_debugger_notification(dyld_notify_removing, infoCount, machHeaders);
break;
default:
break;
}
}
void resetAllImages()
{
sImageInfos.clear();
sImageUUIDs.clear();
_dyld_debugger_notification(dyld_notify_remove_all, 0, NULL);
}
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")))
= {
16, 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, 0, NULL, NULL, NULL, 0, {0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,},
0, {0}, "/usr/lib/dyld", {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