mirror of https://github.com/vxunderground/VX-API
122 lines
3.6 KiB
C++
122 lines
3.6 KiB
C++
#include "Win32Helper.h"
|
|
|
|
BOOL WriteDataToPeSectionW(_In_ LPCWSTR Path, _In_ LPCWSTR SectionName, _In_ PBYTE DataToWrite, _In_ DWORD DataToWriteInBytes)
|
|
{
|
|
BOOL bFlag = FALSE;
|
|
HANDLE hHandle = INVALID_HANDLE_VALUE;
|
|
LONGLONG SizeOfTargetBinary = 0L;
|
|
|
|
PBYTE FileBuffer = NULL;
|
|
PIMAGE_DOS_HEADER Dos = NULL;
|
|
PIMAGE_NT_HEADERS Nt = NULL;
|
|
PIMAGE_FILE_HEADER File = NULL;
|
|
PIMAGE_OPTIONAL_HEADER Optional = NULL;
|
|
|
|
PIMAGE_SECTION_HEADER Section = NULL;
|
|
PIMAGE_SECTION_HEADER SectionHeaderArray = NULL;
|
|
|
|
CHAR DisposeableObject[32] = { 0 };
|
|
|
|
if (WCharStringToCharString(DisposeableObject, (PWCHAR)SectionName, StringLengthW(SectionName)) == 0)
|
|
goto EXIT_ROUTINE;
|
|
|
|
SizeOfTargetBinary = GetFileSizeFromPathW((PWCHAR)Path, FILE_ATTRIBUTE_NORMAL);
|
|
if (SizeOfTargetBinary == INVALID_FILE_SIZE)
|
|
goto EXIT_ROUTINE;
|
|
|
|
hHandle = CreateFileW(Path, GENERIC_READ | GENERIC_WRITE, 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
|
|
if (hHandle == INVALID_HANDLE_VALUE)
|
|
goto EXIT_ROUTINE;
|
|
|
|
FileBuffer = (PBYTE)HeapAlloc(GetProcessHeapFromTeb(), HEAP_ZERO_MEMORY, SizeOfTargetBinary);
|
|
if (FileBuffer == NULL)
|
|
goto EXIT_ROUTINE;
|
|
|
|
if (!ReadFile(hHandle, FileBuffer, (DWORD)SizeOfTargetBinary, NULL, NULL))
|
|
goto EXIT_ROUTINE;
|
|
|
|
RtlLoadPeHeaders(&Dos, &Nt, &File, &Optional, &FileBuffer);
|
|
|
|
SectionHeaderArray = (PIMAGE_SECTION_HEADER)(ULONGLONG(Nt) + sizeof(IMAGE_NT_HEADERS));
|
|
for (DWORD dwX = 0; dwX < File->NumberOfSections; dwX++)
|
|
{
|
|
if (StringCompareA((PCHAR)SectionHeaderArray[dwX].Name, DisposeableObject) == ERROR_SUCCESS)
|
|
{
|
|
if (SetFilePointer(hHandle, SectionHeaderArray[dwX].PointerToRawData, NULL, FILE_BEGIN) == INVALID_SET_FILE_POINTER)
|
|
goto EXIT_ROUTINE;
|
|
|
|
if (!WriteFile(hHandle, DataToWrite, DataToWriteInBytes, NULL, NULL))
|
|
goto EXIT_ROUTINE;
|
|
}
|
|
}
|
|
|
|
bFlag = TRUE;
|
|
|
|
EXIT_ROUTINE:
|
|
|
|
if (FileBuffer)
|
|
HeapFree(GetProcessHeapFromTeb(), HEAP_ZERO_MEMORY, FileBuffer);
|
|
|
|
if (hHandle)
|
|
CloseHandle(hHandle);
|
|
|
|
return bFlag;
|
|
}
|
|
|
|
BOOL WriteDataToPeSectionA(_In_ LPCSTR Path, _In_ LPCSTR SectionName, _In_ PBYTE DataToWrite, _In_ DWORD DataToWriteInBytes)
|
|
{
|
|
BOOL bFlag = FALSE;
|
|
HANDLE hHandle = INVALID_HANDLE_VALUE;
|
|
LONGLONG SizeOfTargetBinary = 0L;
|
|
|
|
PBYTE FileBuffer = NULL;
|
|
PIMAGE_DOS_HEADER Dos = NULL;
|
|
PIMAGE_NT_HEADERS Nt = NULL;
|
|
PIMAGE_FILE_HEADER File = NULL;
|
|
PIMAGE_OPTIONAL_HEADER Optional = NULL;
|
|
|
|
PIMAGE_SECTION_HEADER Section = NULL;
|
|
PIMAGE_SECTION_HEADER SectionHeaderArray = NULL;
|
|
|
|
SizeOfTargetBinary = GetFileSizeFromPathA((PCHAR)Path, FILE_ATTRIBUTE_NORMAL);
|
|
if (SizeOfTargetBinary == INVALID_FILE_SIZE)
|
|
goto EXIT_ROUTINE;
|
|
|
|
hHandle = CreateFileA(Path, GENERIC_READ | GENERIC_WRITE, 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
|
|
if (hHandle == INVALID_HANDLE_VALUE)
|
|
goto EXIT_ROUTINE;
|
|
|
|
FileBuffer = (PBYTE)HeapAlloc(GetProcessHeapFromTeb(), HEAP_ZERO_MEMORY, SizeOfTargetBinary);
|
|
if (FileBuffer == NULL)
|
|
goto EXIT_ROUTINE;
|
|
|
|
if (!ReadFile(hHandle, FileBuffer, (DWORD)SizeOfTargetBinary, NULL, NULL))
|
|
goto EXIT_ROUTINE;
|
|
|
|
RtlLoadPeHeaders(&Dos, &Nt, &File, &Optional, &FileBuffer);
|
|
|
|
SectionHeaderArray = (PIMAGE_SECTION_HEADER)(ULONGLONG(Nt) + sizeof(IMAGE_NT_HEADERS));
|
|
for (DWORD dwX = 0; dwX < File->NumberOfSections; dwX++)
|
|
{
|
|
if (StringCompareA((PCHAR)SectionHeaderArray[dwX].Name, SectionName) == ERROR_SUCCESS)
|
|
{
|
|
if (SetFilePointer(hHandle, SectionHeaderArray[dwX].PointerToRawData, NULL, FILE_BEGIN) == INVALID_SET_FILE_POINTER)
|
|
goto EXIT_ROUTINE;
|
|
|
|
if (!WriteFile(hHandle, DataToWrite, DataToWriteInBytes, NULL, NULL))
|
|
goto EXIT_ROUTINE;
|
|
}
|
|
}
|
|
|
|
bFlag = TRUE;
|
|
|
|
EXIT_ROUTINE:
|
|
|
|
if (FileBuffer)
|
|
HeapFree(GetProcessHeapFromTeb(), HEAP_ZERO_MEMORY, FileBuffer);
|
|
|
|
if (hHandle)
|
|
CloseHandle(hHandle);
|
|
|
|
return bFlag;
|
|
} |