utils

0vercl0k · May 18, 2024 · 3fe3992 · 3fe3992
1 parent 9ea3e43
commit 3fe3992
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 7 deletions.
diff --git a/src/wtf/utils.cc b/src/wtf/utils.cc
@@ -485,7 +485,9 @@ ExceptionCodeToStr(const uint32_t ExceptionCode) {
   return "UNKNOWN";
 }
 
-[[nodiscard]] bool BreakOnIDTEntries(Backend_t &Backend, const CpuState_t &CpuState) {
+[[nodiscard]] std::optional<Gva_t>
+ReadIDTEntryHandler(const Backend_t &Backend, const CpuState_t &CpuState,
+                    const size_t Vector) {
   struct IdtEntry {
     uint16_t Low;
     uint16_t Selector;
@@ -503,16 +505,26 @@ ExceptionCodeToStr(const uint32_t ExceptionCode) {
     }
   };
 
+  const auto Address = Gva_t(CpuState.Idtr.Base + (Vector * sizeof(IdtEntry)));
+  IdtEntry Entry;
+  if (!Backend.VirtReadStruct<IdtEntry>(Address, &Entry)) {
+    return {};
+  }
+
+  return Entry.Handler();
+}
+
+[[nodiscard]] bool BreakOnIDTEntries(Backend_t &Backend,
+                                     const CpuState_t &CpuState) {
   for (size_t Idx = 0; Idx < 256; Idx++) {
-    const auto Address = Gva_t(CpuState.Idtr.Base + (Idx * sizeof(IdtEntry)));
-    IdtEntry Entry;
-    if (!Backend.VirtReadStruct<IdtEntry>(Address, &Entry)) {
+    const auto Handler = ReadIDTEntryHandler(Backend, CpuState, Idx);
+    if (!Handler) {
+      fmt::print("ReadIDTEntryHandler failed\n");
       return false;
     }
 
-    if (!Backend.SetBreakpoint(Entry.Handler(), [](Backend_t *Backend) {
-          Backend->TrapFlag(true);
-        })) {
+    if (!Backend.SetBreakpoint(
+            *Handler, [](Backend_t *Backend) { Backend->TrapFlag(true); })) {
       return false;
     }
   }

diff --git a/src/wtf/utils.h b/src/wtf/utils.h
@@ -261,5 +261,17 @@ SaveFile(const fs::path &Path, const uint8_t *Buffer, const size_t BufferSize);
 
 [[nodiscard]] std::string_view ExceptionCodeToStr(const uint32_t ExceptionCode);
 
+//
+// Read the IDT[Vector] handler.
+//
+
+[[nodiscard]] std::optional<Gva_t>
+ReadIDTEntryHandler(const Backend_t &Backend, const CpuState_t &CpuState,
+                    const size_t Vector);
+
+//
+// Set a breakpoint on every IDT handlers that turn on TF.
+//
+
 [[nodiscard]] bool BreakOnIDTEntries(Backend_t &Backend,
                                      const CpuState_t &CpuState);