UnrealEngineUWP/Engine/Source/Developer/Windows/LiveCodingServer/Private/External/LC_RelocationPatcher.h

// Copyright 2011-2019 Molecular Matters GmbH, all rights reserved.

#pragma once

#include "CoreTypes.h"
#include "LC_Coff.h"
#include "LC_Symbols.h"
#include "LC_ModuleCache.h"
#include "LC_Process.h"


namespace relocations
{
	struct Record
	{
		coff::Relocation::Type::Enum relocationType;
		uint16_t patchIndex;
		uint32_t newModuleRva;

		union Data
		{
			struct RelativeRelocation
			{
				uint32_t originalModuleRva;
			} relativeRelocation;

			struct SectionRelativeRelocation
			{
				uint32_t sectionRelativeRva;
			} sectionRelativeRelocation;

			struct VA32Relocation
			{
				uint32_t originalModuleRva;
			} va32Relocation;

			struct RVA32Relocation
			{
				uint32_t originalModuleRva;
			} rva32Relocation;

			struct VA64Relocation
			{
				uint32_t originalModuleRva;
			} va64Relocation;
		} data;
	};

	bool WouldPatchRelocation(const ImmutableString& dstSymbolName);

	bool WouldPatchRelocation
	(
		const coff::Relocation* relocation,
		const coff::CoffDB* coffDb,
		const ImmutableString& srcSymbolName,
		const ModuleCache::FindSymbolData& originalData
	);

	Record PatchRelocation
	(
		const coff::Relocation* relocation,
		const coff::CoffDB* coffDb,
		const types::StringSet& forceRelocationSymbols,
		const ModuleCache* moduleCache,
		const ImmutableString& srcSymbolName,
		const symbols::Symbol* srcSymbol,
		size_t newModuleIndex,
		void* newModuleBases[]
	);

	void PatchRelocation
	(
		const Record& record,
		process::Handle processHandle,
		void* processModuleBases[],
		void* newModuleBase
	);

	bool IsValidRecord(const Record& record);
};