mirror of
https://github.com/vxunderground/VX-API
synced 2026-06-06 16:54:55 +00:00
vxunderground
f0b5d2bd5d
Some function names renamed to avoid confusion, some function annotations added.
150 lines
4.4 KiB
C++
150 lines
4.4 KiB
C++
#include "Win32Helper.h"
|
|
|
|
typedef struct _PROC_THREAD_ATTRIBUTE {
|
|
ULONG64 Attribute;
|
|
ULONG64 Size;
|
|
ULONG64 Value;
|
|
}PROC_THREAD_ATTRIBUTE, * PPROC_THREAD_ATTRIBUTE;
|
|
|
|
typedef struct _PROC_THREAD_ATTRIBUTE_LIST {
|
|
ULONG PresentFlags;
|
|
ULONG AttributeCount;
|
|
ULONG LastAttribute;
|
|
ULONG SpareUlong0;
|
|
struct _PROC_THREAD_ATTRIBUTE* ExtendedFlagsAttribute;
|
|
struct _PROC_THREAD_ATTRIBUTE Attributes[1];
|
|
}PROC_THREAD_ATTRIBUTE_LIST, * PPROC_THREAD_ATTRIBUTE_LIST;
|
|
|
|
BOOL UnusedSubroutineInitializeProcThreadAttributeList(LPPROC_THREAD_ATTRIBUTE_LIST lpAttributeList, DWORD dwAttributeCount, DWORD dwFlags, PSIZE_T lpSize)
|
|
{
|
|
BOOL bFlag = FALSE;
|
|
DWORD dwSize = ERROR_SUCCESS;
|
|
|
|
if (dwFlags || (dwAttributeCount > 0x1B))
|
|
{
|
|
SetLastErrorInTeb(ERROR_INVALID_PARAMETER);
|
|
return bFlag;
|
|
}
|
|
|
|
dwSize = (24 * (dwAttributeCount + 1));
|
|
|
|
if (lpAttributeList && *lpSize >= dwSize)
|
|
{
|
|
lpAttributeList->PresentFlags = 0;
|
|
lpAttributeList->ExtendedFlagsAttribute = 0;
|
|
lpAttributeList->AttributeCount = dwAttributeCount;
|
|
lpAttributeList->LastAttribute = 0;
|
|
bFlag = TRUE;
|
|
}
|
|
else
|
|
SetLastErrorInTeb(ERROR_INSUFFICIENT_BUFFER);
|
|
|
|
*lpSize = dwSize;
|
|
return bFlag;
|
|
}
|
|
|
|
DWORD UnusedSubroutineGetProcThreadAttributeListSize(VOID)
|
|
{
|
|
SIZE_T dwSize = 0;
|
|
|
|
UnusedSubroutineInitializeProcThreadAttributeList(NULL, 1, 0, &dwSize);
|
|
|
|
return (DWORD)dwSize;
|
|
}
|
|
|
|
VOID UnusedSubroutineUpdateProcThreadAttribute(LPPROC_THREAD_ATTRIBUTE_LIST AttributeList, DWORD_PTR Attribute, PVOID Policy, SIZE_T Size)
|
|
{
|
|
PPROC_THREAD_ATTRIBUTE ExtendedAttributes;
|
|
|
|
AttributeList->PresentFlags |= (1 << (Attribute & 0x0000FFFF));
|
|
|
|
ExtendedAttributes = AttributeList->Attributes;
|
|
ExtendedAttributes->Attribute = Attribute;
|
|
ExtendedAttributes->Size = Size;
|
|
ExtendedAttributes->Value = (ULONG64)Policy;
|
|
AttributeList->LastAttribute++;
|
|
|
|
return;
|
|
}
|
|
|
|
BOOL CreateProcessWithCfGuardW(PPROCESS_INFORMATION Pi, PWCHAR Path)
|
|
{
|
|
BOOL bFlag = FALSE;
|
|
PPROC_THREAD_ATTRIBUTE_LIST ThreadAttributes = NULL;
|
|
SIZE_T dwAttributeSize = 0;
|
|
DWORD64 Policy = PROCESS_CREATION_MITIGATION_POLICY_BLOCK_NON_MICROSOFT_BINARIES_ALWAYS_ON;
|
|
|
|
STARTUPINFOEXW Si; ZeroMemoryEx(&Si, sizeof(STARTUPINFOEXW));
|
|
Si.StartupInfo.cb = sizeof(STARTUPINFOEXW);
|
|
ZeroMemoryEx(Pi, sizeof(PROCESS_INFORMATION));
|
|
|
|
dwAttributeSize = UnusedSubroutineGetProcThreadAttributeListSize();
|
|
if (dwAttributeSize == 0)
|
|
goto EXIT_ROUTINE;
|
|
|
|
ThreadAttributes = (PPROC_THREAD_ATTRIBUTE_LIST)HeapAlloc(GetProcessHeapEx(), HEAP_ZERO_MEMORY, dwAttributeSize);
|
|
if (ThreadAttributes == NULL)
|
|
goto EXIT_ROUTINE;
|
|
|
|
if (!UnusedSubroutineInitializeProcThreadAttributeList(ThreadAttributes, 1, 0, &dwAttributeSize))
|
|
goto EXIT_ROUTINE;
|
|
|
|
UnusedSubroutineUpdateProcThreadAttribute(ThreadAttributes, PROC_THREAD_ATTRIBUTE_MITIGATION_POLICY, &Policy, sizeof(DWORD64));
|
|
|
|
Si.lpAttributeList = ThreadAttributes;
|
|
|
|
if (!CreateProcessW(Path, NULL, NULL, NULL, TRUE, EXTENDED_STARTUPINFO_PRESENT, NULL, NULL, &Si.StartupInfo, Pi))
|
|
goto EXIT_ROUTINE;
|
|
|
|
Si.lpAttributeList = ThreadAttributes;
|
|
|
|
bFlag = TRUE;
|
|
|
|
EXIT_ROUTINE:
|
|
|
|
if (ThreadAttributes)
|
|
HeapFree(GetProcessHeapEx(), HEAP_ZERO_MEMORY, (PPROC_THREAD_ATTRIBUTE_LIST)ThreadAttributes);
|
|
|
|
return bFlag;
|
|
}
|
|
|
|
BOOL CreateProcessWithCfGuardA(PPROCESS_INFORMATION Pi, PCHAR Path)
|
|
{
|
|
BOOL bFlag = FALSE;
|
|
PPROC_THREAD_ATTRIBUTE_LIST ThreadAttributes = NULL;
|
|
SIZE_T dwAttributeSize = 0;
|
|
DWORD64 Policy = PROCESS_CREATION_MITIGATION_POLICY_BLOCK_NON_MICROSOFT_BINARIES_ALWAYS_ON;
|
|
|
|
STARTUPINFOEXA Si; ZeroMemoryEx(&Si, sizeof(STARTUPINFOEXA));
|
|
Si.StartupInfo.cb = sizeof(STARTUPINFOEXW);
|
|
ZeroMemoryEx(Pi, sizeof(PROCESS_INFORMATION));
|
|
|
|
dwAttributeSize = UnusedSubroutineGetProcThreadAttributeListSize();
|
|
if (dwAttributeSize == 0)
|
|
goto EXIT_ROUTINE;
|
|
|
|
ThreadAttributes = (PPROC_THREAD_ATTRIBUTE_LIST)HeapAlloc(GetProcessHeapEx(), HEAP_ZERO_MEMORY, dwAttributeSize);
|
|
if (ThreadAttributes == NULL)
|
|
goto EXIT_ROUTINE;
|
|
|
|
if (!UnusedSubroutineInitializeProcThreadAttributeList(ThreadAttributes, 1, 0, &dwAttributeSize))
|
|
goto EXIT_ROUTINE;
|
|
|
|
UnusedSubroutineUpdateProcThreadAttribute(ThreadAttributes, PROC_THREAD_ATTRIBUTE_MITIGATION_POLICY, &Policy, sizeof(DWORD64));
|
|
|
|
Si.lpAttributeList = ThreadAttributes;
|
|
|
|
if (!CreateProcessA(Path, NULL, NULL, NULL, TRUE, EXTENDED_STARTUPINFO_PRESENT, NULL, NULL, &Si.StartupInfo, Pi))
|
|
goto EXIT_ROUTINE;
|
|
|
|
Si.lpAttributeList = ThreadAttributes;
|
|
|
|
bFlag = TRUE;
|
|
|
|
EXIT_ROUTINE:
|
|
|
|
if (ThreadAttributes)
|
|
HeapFree(GetProcessHeapEx(), HEAP_ZERO_MEMORY, (PPROC_THREAD_ATTRIBUTE_LIST)ThreadAttributes);
|
|
|
|
return bFlag;
|
|
} |