You've already forked UnrealEngineUWP
mirror of
https://github.com/izzy2lost/UnrealEngineUWP.git
synced 2026-03-26 18:15:20 -07:00
44 lines
1.2 KiB
C++
44 lines
1.2 KiB
C++
// 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] = {};
|
|
};
|