images.cpp

#include <vector>
#include <iostream>

#include "images.h"
#include "outfile.h"
#include "bashformats.h"


// List of images.
std::vector<ImageData *> _images;

// Invalid image for not resolvable addresses.
ImageData* _invalidImage = new ImageData(0, 0, static_cast<UINT64>(-1), "invalid");

// Records the given Pin image in the internal list and returns the newly created image object.
ImageData* HandleImageLoad(IMG& img)
{
	auto imageData = new ImageData(IMG_Id(img), IMG_LowAddress(img), IMG_HighAddress(img), IMG_Name(img));

	_images.push_back(imageData);

	// Log image load
	std::cerr
		<< _colorCyan << "load "
		<< _colorDefault << std::dec << imageData->Id << " "
		<< _colorRed << std::hex << imageData->LowAddress << " "	<< imageData->HighAddress << " "
		<< _colorYellow << imageData->Name
		<< _colorDefault
		<< std::endl;
	outputFile << "L"
		<< " " << std::dec << imageData->Id
		<< " " << std::setw(16) << std::setfill('0') << std::hex << imageData->LowAddress
		<< " " << std::setw(16) << std::setfill('0') << std::hex << imageData->HighAddress
		<< " " << imageData->Name
		<< std::endl;

	return imageData;
}

// Returns the image object containing the given address, or an invalid dummy image.
ImageData* GetImageContainingAddress(UINT64 addr)
{
	for(auto imageData : _images)
	{
		if (imageData->LowAddress <= addr && addr < imageData->HighAddress)
			return imageData;
	}

	return _invalidImage;
}