feat white process

2026-06-29 09:10:15 +08:00
parent 2719a2193d
commit 1a601fb2d3
2 changed files with 8 additions and 2 deletions
@@ -5,7 +5,6 @@ namespace comm_dispatch
 {
 	auto dispatch(CMD_COMM* data) -> NTSTATUS
 	{
-
 		auto status = STATUS_UNSUCCESSFUL;
 		switch (data->CommID)
 		{
@@ -17,7 +16,13 @@ namespace comm_dispatch
 		case CMD::DRIVER_PROTECT_PROCESS:
 		{
 			auto process_data = (IOCTL_PROCESS*)data->Buf;
-			status = protect_filter::add_list((ULONG)process_data->Pid, NULL);
+			status = protect_filter::add_protect_list((ULONG)process_data->Pid, NULL, FALSE);
+			break;
+		}
+		case CMD::ADD_WHITE_PROCESS:
+		{
+			auto process_data = (IOCTL_PROCESS*)data->Buf;
+			status = protect_filter::add_protect_list((ULONG)process_data->Pid, NULL, TRUE);
 			break;
 		}
 		case CMD::WINDOW_ANIT_SCREEN:
@@ -23,6 +23,7 @@ enum CMD
 	MSG_BASE = 0x10000,
 	DRIVER_COMM_TEST,
 	DRIVER_PROTECT_PROCESS,
+	ADD_WHITE_PROCESS,
 	WINDOW_ANIT_SCREEN,
 };