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

#pragma once

#include "CoreTypes.h"
#include "LC_Executable.h"
#include "LC_Process.h"


class ExecutablePatcher
{
public:
#if LC_64_BIT
	static const size_t INJECTED_CODE_SIZE = 3u;
#else
	static const size_t INJECTED_CODE_SIZE = 5u;
#endif

	// reads the original entry point code from the executable
	ExecutablePatcher(executable::Image* image, executable::ImageSectionDB* imageSections);

	// uses the given entry point code 
	explicit ExecutablePatcher(const uint8_t* entryPointCode);


	// disables the entry point directly in the image file, and returns the RVA of the entry point
	uint32_t DisableEntryPointInImage(executable::Image* image, executable::ImageSectionDB* imageSections);

	// disables the entry point in memory
	void DisableEntryPoint(process::Handle processHandle, void* moduleBase, uint32_t entryPointRva);

	// restores the entry point of a loaded image that previously had its entry point disabled
	void RestoreEntryPoint(process::Handle processHandle, void* moduleBase, uint32_t entryPointRva);


	inline const uint8_t* GetEntryPointCode(void) const
	{
		return m_originalCode;
	}

private:
	uint8_t m_originalCode[INJECTED_CODE_SIZE] = {};
};