diff --git a/Anti-Cheat_Driver/comm_dispatch.cpp b/Anti-Cheat_Driver/comm_dispatch.cpp index bb4d0db..bbf9123 100644 --- a/Anti-Cheat_Driver/comm_dispatch.cpp +++ b/Anti-Cheat_Driver/comm_dispatch.cpp @@ -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: diff --git a/Anti-Cheat_Driver/comm_dispatch.h b/Anti-Cheat_Driver/comm_dispatch.h index 58b254e..f6593b1 100644 --- a/Anti-Cheat_Driver/comm_dispatch.h +++ b/Anti-Cheat_Driver/comm_dispatch.h @@ -23,6 +23,7 @@ enum CMD MSG_BASE = 0x10000, DRIVER_COMM_TEST, DRIVER_PROTECT_PROCESS, + ADD_WHITE_PROCESS, WINDOW_ANIT_SCREEN, };