#include "comm_dispatch.h"
#include "protect_filter.h"

namespace comm_dispatch
{
	auto dispatch(CMD_COMM* data) -> NTSTATUS
	{
		auto status = STATUS_UNSUCCESSFUL;
		switch (data->CommID)
		{
		case CMD::DRIVER_COMM_TEST:
		{
			status = STATUS_SUCCESS;
			break;
		}
		case CMD::DRIVER_PROTECT_PROCESS:
		{
			status = protect_filter::add_list((ULONG)data->Pid, NULL);
			break;
		}
		default:
			break;
		}

		*(NTSTATUS*)data->status = status;
		return status;
	}
}