ShellcodeCompiler/ShellcodeCompiler/DLLBaseAddress.cpp at master · kissthink/ShellcodeCompiler · GitHub

Repository navigation

History

56 lines (38 loc) · 1.11 KB

Raw

#include "DLLBaseAddress.h"

// Global data

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

size_t DLLBaseAddress::CurrentDLLIOffset = 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 = CurrentDLLIOffset;

DLLOffsets.push_back(dll);

CurrentDLLIOffset++;

}

// 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) return DLLOffsets[i].Offset;

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

return 0;

}

ShellcodeCompiler/ShellcodeCompiler/DLLBaseAddress.cpp at master · kissthink/ShellcodeCompiler · GitHub

Repository navigation

Expand file tree

Breadcrumbs

DLLBaseAddress.cpp

Latest commit

History

Breadcrumbs

DLLBaseAddress.cpp

File metadata and controls