mirror of https://github.com/vxunderground/VX-API
135 lines
4.0 KiB
C++
135 lines
4.0 KiB
C++
#include "Win32Helper.h"
|
|
|
|
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(_Inout_ PPROCESS_INFORMATION Pi, _In_ 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 = {0};// 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(GetProcessHeapFromTeb(), 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(GetProcessHeapFromTeb(), HEAP_ZERO_MEMORY, (PPROC_THREAD_ATTRIBUTE_LIST)ThreadAttributes);
|
|
|
|
return bFlag;
|
|
}
|
|
|
|
BOOL CreateProcessWithCfGuardA(_Inout_ PPROCESS_INFORMATION Pi, _In_ 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 = {0}; //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(GetProcessHeapFromTeb(), 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(GetProcessHeapFromTeb(), HEAP_ZERO_MEMORY, (PPROC_THREAD_ATTRIBUTE_LIST)ThreadAttributes);
|
|
|
|
return bFlag;
|
|
} |