add IOCTL_CE_ENABLE_DRM

cheat-engine · Jan 11, 2017 · a98b7bc · a98b7bc
1 parent fe19861
commit a98b7bc
Show file tree

Hide file tree

Showing 13 changed files with 649 additions and 310 deletions.
diff --git a/DBKKernel/DBKDrvr.c b/DBKKernel/DBKDrvr.c
@@ -387,9 +387,8 @@ Return Value:
 
 	//Processlist init
 #ifndef CETC
-
 	ProcessEventCount=0;
-	KeInitializeSpinLock(&ProcesslistSL);
+	ExInitializeResourceLite(&ProcesslistR);	
 #endif
 
 	CreateProcessNotifyRoutineEnabled=FALSE;
@@ -673,7 +672,15 @@ void UnloadDriver(PDRIVER_OBJECT DriverObject)
 
 			if (CreateProcessNotifyRoutineEnabled)
 			{
+				DbgPrint("Removing process watch");
+#if (NTDDI_VERSION >= NTDDI_VISTASP1)
+				PsSetCreateProcessNotifyRoutineEx(CreateProcessNotifyRoutineEx,TRUE);
+#else
 				PsSetCreateProcessNotifyRoutine(CreateProcessNotifyRoutine,TRUE);
+#endif
+
+
+				DbgPrint("Removing thread watch");
 				PsRemoveCreateThreadNotifyRoutine2(CreateThreadNotifyRoutine);
 			}
 
@@ -700,4 +707,16 @@ void UnloadDriver(PDRIVER_OBJECT DriverObject)
 	ExFreePool(BufDeviceString);
 #endif
 
+	CleanProcessList();
+
+	ExDeleteResourceLite(&ProcesslistR);
+
+	RtlZeroMemory(&ProcesslistR, sizeof(ProcesslistR));
+
+	if (DRMHandle)
+	{
+		DbgPrint("Unregistering DRM handle");
+		ObUnRegisterCallbacks(DRMHandle);
+		DRMHandle = NULL;
+	}
 }
diff --git a/DBKKernel/DBKDrvr.h b/DBKKernel/DBKDrvr.h
@@ -3,7 +3,7 @@
 
 
 
-#define dbkversion 2000021
+#define dbkversion 2000022
 
 
 

diff --git a/DBKKernel/DBKFunc.h b/DBKKernel/DBKFunc.h
@@ -2,6 +2,9 @@
 #define DBKFUNC_H
 
 #include "ntifs.h"
+//#include <ntifs.h>
+#include <ntstrsafe.h>
+
 #include <windef.h>
 
 #include "interruptHook.h"