admin/bats3c-DarkLoadLibrary
1
0
Fork 0
mirror of https://github.com/bats3c/DarkLoadLibrary synced 2026-06-06 15:14:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add some extra checks into darkloadlibrary.c

Browse Source
This commit is contained in:
Hypervisor
2021-08-07 14:38:49 -07:00
parent da43012aab
commit 53211bef19
1 changed files with 20 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
DarkLoadLibrary/src/darkloadlibrary.c
+20
View File
@@ -151,6 +151,8 @@ PDARKMODULE DarkLoadLibrary(
)
{
PDARKMODULE dModule = (DARKMODULE*)HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(DARKMODULE));
if (!dModule)
return NULL;
dModule->bSuccess = FALSE;
dModule->bLinkedToPeb = TRUE;
@@ -175,6 +177,9 @@ PDARKMODULE DarkLoadLibrary(
dModule->CrackedDLLName = lpwName;
dModule->LocalDLLName = lpwName;
if (lpwName == NULL)
goto Cleanup;
break;
default:
@@ -206,6 +211,9 @@ PDARKMODULE DarkLoadLibrary(
if (!IsValidPE(dModule->pbDllData))
{
dModule->ErrorMsg = (wchar_t*)HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, 500);
if (!dModule->ErrorMsg)
goto Cleanup;
wcscat(dModule->ErrorMsg, L"Data is an invalid PE: ");
wcscat(dModule->ErrorMsg, lpwName);
goto Cleanup;
@@ -215,6 +223,9 @@ PDARKMODULE DarkLoadLibrary(
if (!MapSections(dModule))
{
dModule->ErrorMsg = (wchar_t*)HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, 500);
if (!dModule->ErrorMsg)
goto Cleanup;
wcscat(dModule->ErrorMsg, L"Failed to map sections: ");
wcscat(dModule->ErrorMsg, lpwName);
goto Cleanup;
@@ -224,6 +235,9 @@ PDARKMODULE DarkLoadLibrary(
if (!ResolveImports(dModule))
{
dModule->ErrorMsg = (wchar_t*)HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, 500);
if (!dModule->ErrorMsg)
goto Cleanup;
wcscat(dModule->ErrorMsg, L"Failed to resolve imports: ");
wcscat(dModule->ErrorMsg, lpwName);
goto Cleanup;
@@ -235,6 +249,9 @@ PDARKMODULE DarkLoadLibrary(
if (!LinkModuleToPEB(dModule))
{
dModule->ErrorMsg = (wchar_t*)HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, 500);
if (!dModule->ErrorMsg)
goto Cleanup;
wcscat(dModule->ErrorMsg, L"Failed to link module to PEB: ");
wcscat(dModule->ErrorMsg, lpwName);
goto Cleanup;
@@ -245,6 +262,9 @@ PDARKMODULE DarkLoadLibrary(
if (!BeginExecution(dModule))
{
dModule->ErrorMsg = (wchar_t*)HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, 500);
if (!dModule->ErrorMsg)
goto Cleanup;
wcscat(dModule->ErrorMsg, L"Failed to execute: ");
wcscat(dModule->ErrorMsg, lpwName);
goto Cleanup;