#include "ob_reg_callback.h"
#include "create_thread_callback.h"
#include "process_notify_callback.h"
#include "load_Image_callback.h"
#include "protect_filter.h"
#include "comm.h"
#include "inline_hooks.h"
#include "mouse_keybord_hook.h"
#include "ssdt.h"


EXTERN_C NTSTATUS DriverEntry(PDRIVER_OBJECT drv_obj, PUNICODE_STRING)
{
	if (drv_obj)
	{
		drv_obj->DriverUnload = [](PDRIVER_OBJECT)
		{
			kernel_comm_create::remove();
			//ob_call_back::uninstall_ob_callback();
			process_notify_callback::remov_process_notify();
			thread_notify_routine::unload_thread_routine();

			mouse_keybord_hook::remove_mouse_keybord_hook();

			//load_image_notify_routine::remove_image_load_notify_routine();
		};
	}

	protect_filter::Init();
	kernel_api::kernel_api_init();
	kernel_comm_create::Init();

	protect_filter::add_protect_list(3632, 0, FALSE);


	mouse_keybord_hook::install_mouse_keybord_hook();
	//DbgPrintEx(77, 0, "[+]ob_reg_callback status:%x\n", ob_call_back::register_ob_reg_callback());
	DbgPrintEx(77, 0, "[+]process_notify_callback status:%x\n", process_notify_callback::install_process_notify());
	DbgPrintEx(77, 0, "[+]create_thread_routine status:%x\n", thread_notify_routine::create_thread_routine());

	//load_image_notify_routine::create_image_load_notify_routine();

	return STATUS_SUCCESS;
}