From 647250f1af64515e3481bb3fa99e1bffd38c868a Mon Sep 17 00:00:00 2001
From: _xeroxz <xerox@back.engineer>
Date: Sun, 7 Feb 2021 22:30:47 -0800
Subject: [PATCH] event injection is wrong

---
 bluepill.vcxproj         |  3 +++
 bluepill.vcxproj.filters |  9 +++++++++
 hv_types.hpp             | 29 +++++++++++++++++++++++++++++
 idt.cpp                  | 20 ++++++++++++++++++++
 idt.hpp                  | 10 ++++++++++
 idt_handler.asm          |  0
 mm.hpp                   |  2 --
 7 files changed, 71 insertions(+), 2 deletions(-)
 create mode 100644 idt.cpp
 create mode 100644 idt.hpp
 create mode 100644 idt_handler.asm

diff --git a/bluepill.vcxproj b/bluepill.vcxproj
index e716e20..9dc78fe 100644
--- a/bluepill.vcxproj
+++ b/bluepill.vcxproj
@@ -85,6 +85,7 @@
   <ItemGroup>
     <ClCompile Include="entry.cpp" />
     <ClCompile Include="exit_handler.cpp" />
+    <ClCompile Include="idt.cpp" />
     <ClCompile Include="mm.cpp" />
     <ClCompile Include="segment.cpp" />
     <ClCompile Include="vmcs.cpp" />
@@ -93,6 +94,7 @@
   </ItemGroup>
   <ItemGroup>
     <ClInclude Include="ia32.hpp" />
+    <ClInclude Include="idt.hpp" />
     <ClInclude Include="mm.hpp" />
     <ClInclude Include="segment_intrin.h" />
     <ClInclude Include="hv_types.hpp" />
@@ -102,6 +104,7 @@
     <ClInclude Include="vmxon.hpp" />
   </ItemGroup>
   <ItemGroup>
+    <MASM Include="idt_handler.asm" />
     <MASM Include="segment_intrin.asm" />
     <MASM Include="vmxexit_handler.asm" />
   </ItemGroup>
diff --git a/bluepill.vcxproj.filters b/bluepill.vcxproj.filters
index cab2d58..f4b14c8 100644
--- a/bluepill.vcxproj.filters
+++ b/bluepill.vcxproj.filters
@@ -32,6 +32,9 @@
     <ClCompile Include="mm.cpp">
       <Filter>Source Files</Filter>
     </ClCompile>
+    <ClCompile Include="idt.cpp">
+      <Filter>Source Files</Filter>
+    </ClCompile>
   </ItemGroup>
   <ItemGroup>
     <ClInclude Include="vmxon.hpp">
@@ -58,6 +61,9 @@
     <ClInclude Include="mm.hpp">
       <Filter>Header Files</Filter>
     </ClInclude>
+    <ClInclude Include="idt.hpp">
+      <Filter>Header Files</Filter>
+    </ClInclude>
   </ItemGroup>
   <ItemGroup>
     <MASM Include="segment_intrin.asm">
@@ -66,5 +72,8 @@
     <MASM Include="vmxexit_handler.asm">
       <Filter>Source Files</Filter>
     </MASM>
+    <MASM Include="idt_handler.asm">
+      <Filter>Source Files</Filter>
+    </MASM>
   </ItemGroup>
 </Project>
\ No newline at end of file
diff --git a/hv_types.hpp b/hv_types.hpp
index c3e1edb..6baf6dd 100644
--- a/hv_types.hpp
+++ b/hv_types.hpp
@@ -79,6 +79,35 @@ namespace hv
 		};
 	};
 
+	typedef union _idt_entry_t
+	{
+		u128 flags;
+		struct
+		{
+			u64 offset_low : 16;
+			u64 segment_selector : 16;
+			u64 reserved_0 : 8;
+			u64 gate_type : 4;
+			u64 storage_segment : 1;
+			u64 dpl : 2;
+			u64 present : 1;
+			u64 offset_middle : 16;
+			u64 offset_high : 32;
+			u64 reserved_1 : 32;
+		};
+	} idt_entry_t, *pidt_entry_t;
+
+	union idt_addr_t
+	{
+		u64 addr;
+		struct
+		{
+			u64 offset_low : 16;
+			u64 offset_middle : 16;
+			u64 offset_high : 32;
+		};
+	};
+
 	union ia32_efer_t
 	{
 		unsigned __int64 control;
diff --git a/idt.cpp b/idt.cpp
new file mode 100644
index 0000000..a54bf22
--- /dev/null
+++ b/idt.cpp
@@ -0,0 +1,20 @@
+#include "idt.hpp"
+
+namespace idt
+{
+	auto create_entry(void* address) -> hv::idt_entry_t
+	{
+		hv::idt_addr_t idt_addr{ (u64) address };
+		hv::idt_entry_t result{};
+
+		result.dpl = 0;
+		result.storage_segment = 0;
+		result.segment_selector = readcs();
+		result.gate_type = SEGMENT_DESCRIPTOR_TYPE_INTERRUPT_GATE;
+		result.present = 1;
+		result.offset_high = idt_addr.offset_high;
+		result.offset_middle = idt_addr.offset_middle;
+		result.offset_low = idt_addr.offset_low;
+		return result;
+	}
+}
\ No newline at end of file
diff --git a/idt.hpp b/idt.hpp
new file mode 100644
index 0000000..b4a1575
--- /dev/null
+++ b/idt.hpp
@@ -0,0 +1,10 @@
+#pragma once
+#include "hv_types.hpp"
+#include "segment_intrin.h"
+#pragma section(".idt", read, write)
+
+namespace idt
+{
+	__declspec(allocate(".idt")) inline hv::idt_entry_t table[256];
+	auto create_entry(void* address) -> hv::idt_entry_t;
+}
\ No newline at end of file
diff --git a/idt_handler.asm b/idt_handler.asm
new file mode 100644
index 0000000..e69de29
diff --git a/mm.hpp b/mm.hpp
index c12bb4c..b117685 100644
--- a/mm.hpp
+++ b/mm.hpp
@@ -1,7 +1,5 @@
 #pragma once
 #include "hv_types.hpp"
-
-// the pml4 itself is inside of the HV...
 #pragma section(".pml4", read, write)
 
 namespace mm