Prepare hypervisor

src/driver/driver_main.cpp

@@ -3,6 +3,7 @@
 #include "sleep_callback.hpp"
 #include "irp.hpp"
 #include "exception.hpp"
+#include "hypervisor.hpp"
 
 #define DOS_DEV_NAME L"\\DosDevices\\HelloDev"
 #define DEV_NAME L"\\Device\\HelloDev"
@@ -38,17 +39,20 @@ public:
 private:
 	sleep_callback sleep_callback_{};
 	irp irp_{};
+	hypervisor hypervisor_{};
 
 	void sleep_notification(const sleep_callback::type type)
 	{
 		if (type == sleep_callback::type::sleep)
 		{
 			debug_log("Going to sleep!");
+			this->hypervisor_.on_sleep();
 		}
 
 		if (type == sleep_callback::type::wakeup)
 		{
 			debug_log("Waking up!");
+			this->hypervisor_.on_wakeup();
 		}
 	}
 };

src/driver/hypervisor.cpp

@@ -0,0 +1,54 @@
+#include "std_include.hpp"
+#include "hypervisor.hpp"
+#include "exception.hpp"
+#include "logging.hpp"
+
+#define IA32_FEATURE_CONTROL_MSR 0x3A
+#define IA32_FEATURE_CONTROL_MSR_LOCK 0x0001
+#define IA32_FEATURE_CONTROL_MSR_ENABLE_VMXON_OUTSIDE_SMX 0x0004
+
+namespace
+{
+	bool is_vmx_supported()
+	{
+		int cpuid_data[4] = {0};
+		__cpuid(cpuid_data, 1);
+		return cpuid_data[2] & 0x20;
+	}
+
+	bool is_vmx_available()
+	{
+		const auto feature_control = __readmsr(IA32_FEATURE_CONTROL_MSR);
+		const auto is_locked = (feature_control & IA32_FEATURE_CONTROL_MSR_LOCK) != 0;
+		const auto is_enabled_outside_smx = (feature_control & IA32_FEATURE_CONTROL_MSR_ENABLE_VMXON_OUTSIDE_SMX) != 0;
+
+		return is_locked && is_enabled_outside_smx;
+	}
+
+	bool is_virtualization_supported()
+	{
+		return is_vmx_supported() && is_vmx_available();
+	}
+}
+
+hypervisor::hypervisor()
+{
+	if(!is_virtualization_supported())
+	{
+		throw std::runtime_error("VMX not supported on this machine");
+	}
+
+	debug_log("VMX supported!\n");
+}
+
+hypervisor::~hypervisor()
+{
+}
+
+void hypervisor::on_sleep()
+{
+}
+
+void hypervisor::on_wakeup()
+{
+}

src/driver/hypervisor.hpp

@@ -0,0 +1,19 @@
+#pragma once
+
+class hypervisor
+{
+public:
+	hypervisor();
+	~hypervisor();
+
+	hypervisor(hypervisor&& obj) noexcept = delete;
+	hypervisor& operator=(hypervisor&& obj) noexcept = delete;
+
+	hypervisor(const hypervisor& obj) = delete;
+	hypervisor& operator=(const hypervisor& obj) = delete;
+
+	void on_sleep();
+	void on_wakeup();
+
+private:
+};

src/driver/logging.hpp

@@ -3,5 +3,5 @@
 #ifdef NDEBUG
 #define debug_log(...)
 #else
-#define debug_log(...) DbgPrintEx(DPFLTR_IHVDRIVER_ID, DPFLTR_ERROR_LEVEL, __VA_ARGS__)
+#define debug_log(msg, ...) DbgPrintEx(DPFLTR_IHVDRIVER_ID, DPFLTR_ERROR_LEVEL, "[MOMO] " msg, __VA_ARGS__)
 #endif

src/driver/std_include.hpp

@@ -1,5 +1,6 @@
 #pragma once
 
 #include <ntddk.h>
+#include <intrin.h>
 #include "nt_ext.hpp"
 #include "new.hpp"