feat mdl memcopy and close MmverifyFlags

Anti-Cheat_Driver/kernel_function.cpp

@@ -1,6 +1,8 @@
 #include "kernel_function.h"
 #include "kernel_api.h"
-
+#include "utils.h"
+#pragma warning(disable : 4309)
+#pragma warning(disable : 4838)
 namespace kernel_function
 {
 	PEPROCESS process{ 0 };
@@ -66,5 +68,106 @@ namespace kernel_function
 		return base;
 	}
 
+	auto MmMDLPagesCopy(IN PVOID Address, PVOID Buff, SIZE_T Size, BYTE Type, CHAR AccessMode) -> NTSTATUS
+	{
+		PMDL pMDL = IoAllocateMdl(Address, (ULONG)Size, FALSE, FALSE, NULL);
+		if (!pMDL) return STATUS_INSUFFICIENT_RESOURCES;
+
+		__try
+		{
+			MmProbeAndLockPages(pMDL, AccessMode, IoReadAccess);
+		}
+		__except (EXCEPTION_EXECUTE_HANDLER)
+		{
+			IoFreeMdl(pMDL);
+			return STATUS_ACCESS_DENIED;
+		}
+
+		PVOID NewAddress = MmMapLockedPagesSpecifyCache(pMDL, KernelMode, MmNonCached, NULL, 0, NormalPagePriority);
+		if (!NewAddress)
+		{
+			MmUnlockPages(pMDL);
+			IoFreeMdl(pMDL);
+			return STATUS_INSUFFICIENT_RESOURCES;
+		}
+
+		if (Type == 0x0000001)
+			RtlCopyMemory(Buff, NewAddress, Size);
+		if (Type == 0x0000002)
+			RtlCopyMemory(NewAddress, Buff, Size);
+
+
+		MmUnmapLockedPages(NewAddress, pMDL);
+		MmUnlockPages(pMDL);
+		IoFreeMdl(pMDL);
+
+		return STATUS_SUCCESS;
+	}
+
+	auto GetMmverifyCallBackFlags() -> uintptr_t
+	{
+		uintptr_t pfunc_address = NULL;
+		auto version = utils::GetVersion();
+		if (version.dwBuildNumber == 7600 || version.dwBuildNumber == 7601)
+		{
+			auto pfunc_extport_address = reinterpret_cast<unsigned char*>(utils::kernel_get_expotr_func(L"ObRegisterCallbacks"));
+			for (size_t i = 0; i < 0x256; i++)
+			{
+				if (pfunc_extport_address[i] == 0x74 && pfunc_extport_address[i + 1] == 0x09)
+				{
+					pfunc_address = (uintptr_t)(*(long*)(&pfunc_extport_address[i + 3]) + &pfunc_extport_address[i + 7]);
+					break;
+				}
+			}
+		}
+		else
+		{
+			auto pfunc_extport_address = reinterpret_cast<unsigned char*>(utils::kernel_get_expotr_func(L"PsSetCreateThreadNotifyRoutineEx"));
+			for (size_t i = 0; i < 0x256; i++)
+			{
+				if (pfunc_extport_address[i] == 0xE8)
+				{
+					pfunc_address = (uintptr_t)(*(long*)(&pfunc_extport_address[i + 1]) + &pfunc_extport_address[i + 5]);
+					break;
+				}
+			}
+		}
+		return pfunc_address;
+	}
+
+	//ÆÁ±Î0xC0000022LL´íÎó
+	auto Mack_MmVerifyCallBackFlags(BYTE** orgin_byte) -> BOOL
+	{
+		uintptr_t pfunc_address = GetMmverifyCallBackFlags();
+		if (!pfunc_address)
+			return FALSE;
+
+		//xor,rax,rax
+		//mov rax,1
+		//ret
+		char shellcode[] = { 0x31, 0xC0, 0xB8, 0x01, 0x00 ,0x00, 0x00,0xC3 };
+
+		memcpy(orgin_byte, (CONST void*)pfunc_address, sizeof(shellcode));
+
+		auto status = MmMDLPagesCopy((PVOID)pfunc_address, &shellcode, sizeof(shellcode), 0x0000002, KernelMode);
+		if (!NT_SUCCESS(status))
+			return FALSE;
+
+		return TRUE;
+	}
+
+	auto Orgin_MmVerifyCallBackFlags(BYTE* orgin_byte) -> BOOL
+	{
+		uintptr_t pfunc_address = GetMmverifyCallBackFlags();
+		if (!pfunc_address)
+			return FALSE;
+
+		auto status = MmMDLPagesCopy((PVOID)pfunc_address, orgin_byte, sizeof(orgin_byte), 0x0000002, KernelMode);
+		if (!NT_SUCCESS(status))
+			return FALSE;
+
+		return TRUE;
+	}
+
 
 }

Anti-Cheat_Driver/kernel_function.h

@@ -3,7 +3,6 @@
 
 namespace kernel_function
 {
-
 	typedef enum _SYSTEM_INFORMATION_CLASS {
 		SystemBasicInformation = 0,
 		SystemPerformanceInformation = 2,
@@ -43,7 +42,11 @@ namespace kernel_function
 
 	auto GetKernelModule(CONST CHAR* Module)->uintptr_t;
 
+	auto MmMDLPagesCopy(IN PVOID Address, PVOID Buff, SIZE_T Size, BYTE Type, CHAR AccessMode)->NTSTATUS;
 
+	auto GetMmverifyCallBackFlags()->uintptr_t;
 
+	auto Mack_MmVerifyCallBackFlags(BYTE** orgin_byte)->BOOL;
 
+	auto Orgin_MmVerifyCallBackFlags(BYTE* orgin_byte)->BOOL;
 }