#include "DLLBaseAddress.h"

// Global data

vector<DLLBaseAddress::DLLBase> DLLBaseAddress::DLLOffsets;

size_t DLLBaseAddress::CurrentDLLOffset = 4;

// Create kernel32 DLL base index

void DLLBaseAddress::InitDLLBaseAddress()

{

DLLBase kernel;

kernel.Name = "kernel32.dll";

kernel.Offset = 1;

DLLOffsets.push_back(kernel);

}

// Check if a DLL base index already exists

bool DLLBaseAddress::DLLBaseExists(string p_sDLLName)

{

for (size_t i = 0; i < DLLOffsets.size(); i++)

if (DLLOffsets[i].Name.compare(p_sDLLName) == 0) return true;

return false;

}

// Add a DLL base index

void DLLBaseAddress::AddDLLBase(string p_sDLLName)

{

DLLBase dll;

// Check if DLL already exists

if (DLLBaseExists(p_sDLLName)) return;

dll.Name = p_sDLLName;

dll.Offset = CurrentDLLOffset;

DLLOffsets.push_back(dll);

CurrentDLLOffset++;

}

// Get DLL base index

size_t DLLBaseAddress::GetDLLBase(string p_sDLLName)

{

for (size_t i = 0; i < DLLOffsets.size(); i++)

if (DLLOffsets[i].Name.compare(p_sDLLName) == 0)

{

// For Windows x64, we do not have other things on the stack: kernel32 base, GetProcAddress and Loadlibrary so we substract 3

if(Platform::GetPlatform() == PLATFORM_TYPE_WINDOWS_X64) return DLLOffsets[i].Offset - 3;

else return DLLOffsets[i].Offset;

}

cout << "Error: Cannot find DLL base index for " << p_sDLLName << endl;

return 0;

}