feat dispatch protect and anit screen

2026-06-23 13:17:33 +08:00
parent ec66fbbf8a
commit 351ffe7413
2 changed files with 26 additions and 4 deletions
@@ -1,10 +1,11 @@
 #include "comm_dispatch.h"
 #include "protect_filter.h"
-
+#include "anit_screen_grap.h"
 namespace comm_dispatch
 {
 	auto dispatch(CMD_COMM* data) -> NTSTATUS
 	{
+
 		auto status = STATUS_UNSUCCESSFUL;
 		switch (data->CommID)
 		{
@@ -15,7 +16,17 @@ namespace comm_dispatch
 		}
 		case CMD::DRIVER_PROTECT_PROCESS:
 		{
-			status = protect_filter::add_list((ULONG)data->Pid, NULL);
+			auto process_data = (IOCTL_PROCESS*)data->Buf;
+			status = protect_filter::add_list((ULONG)process_data->Pid, NULL);
+			break;
+		}
+		case CMD::WINDOW_ANIT_SCREEN:
+		{
+			auto window_data = (IOCTL_WINDOW*)data->Buf;
+
+			status = anit_screen_grap::GreProtectSpriteContent(
+				(HWND)window_data->HWND, (UINT)window_data->Flags);
+
 			break;
 		}
 		default: