mirror of
https://github.com/Mauler125/r5sdk.git
synced 2025-02-09 19:15:03 +01:00
3ea7cc1cd4
* Added 'GetImportedFunction'. * Remove extraneous std::string copy constructors during construction of 'm_vModuleSections'. * Added extra constructor using base address.
61 lines
2.2 KiB
C++
61 lines
2.2 KiB
C++
#ifndef MODULE_H
|
|
#define MODULE_H
|
|
|
|
class CModule
|
|
{
|
|
public:
|
|
struct ModuleSections_t
|
|
{
|
|
ModuleSections_t(void) = default;
|
|
ModuleSections_t(const string& svSectionName, uintptr_t pSectionBase, size_t nSectionSize) :
|
|
m_svSectionName(svSectionName), m_pSectionBase(pSectionBase), m_nSectionSize(nSectionSize) {}
|
|
|
|
bool IsSectionValid(void) const
|
|
{
|
|
return m_nSectionSize != 0;
|
|
}
|
|
|
|
string m_svSectionName; // Name of section.
|
|
uintptr_t m_pSectionBase{}; // Start address of section.
|
|
size_t m_nSectionSize{}; // Size of section.
|
|
};
|
|
|
|
CModule(void) = default;
|
|
CModule(const string& moduleName);
|
|
CModule(const uintptr_t nModuleBase, const string& svModuleName);
|
|
|
|
void Init();
|
|
void LoadSections();
|
|
#ifndef PLUGINSDK
|
|
CMemory FindPatternSIMD(const string& svPattern, const ModuleSections_t* moduleSection = nullptr) const;
|
|
CMemory FindString(const string& string, const ptrdiff_t occurrence = 1, bool nullTerminator = false) const;
|
|
CMemory FindStringReadOnly(const string& svString, bool nullTerminator) const;
|
|
|
|
CMemory GetVirtualMethodTable(const string& svTableName, const uint32_t nRefIndex = 0);
|
|
#endif // !PLUGINSDK
|
|
CMemory GetImportedFunction(const string& svModuleName, const string& svFunctionName, const bool bGetFunctionReference) const;
|
|
CMemory GetExportedFunction(const string& svFunctionName) const;
|
|
ModuleSections_t GetSectionByName(const string& svSectionName) const;
|
|
uintptr_t GetModuleBase(void) const;
|
|
DWORD GetModuleSize(void) const;
|
|
string GetModuleName(void) const;
|
|
uintptr_t GetRVA(const uintptr_t nAddress) const;
|
|
|
|
IMAGE_NT_HEADERS64* m_pNTHeaders = nullptr;
|
|
IMAGE_DOS_HEADER* m_pDOSHeader = nullptr;
|
|
|
|
ModuleSections_t m_ExecutableCode;
|
|
ModuleSections_t m_ExceptionTable;
|
|
ModuleSections_t m_RunTimeData;
|
|
ModuleSections_t m_ReadOnlyData;
|
|
|
|
private:
|
|
CMemory FindPatternSIMD(const uint8_t* szPattern, const char* szMask, const ModuleSections_t* moduleSection = nullptr, const uint32_t nOccurrence = 0) const;
|
|
|
|
string m_svModuleName;
|
|
uintptr_t m_pModuleBase{};
|
|
DWORD m_nModuleSize{};
|
|
vector<ModuleSections_t> m_vModuleSections;
|
|
};
|
|
|
|
#endif // MODULE_H
|