add: migrate source code of luahook

blcrack/cracker/Dobby/source/InterceptRouting/InterceptRouting.cpp

@@ -0,0 +1,97 @@
+#include "dobby/dobby_internal.h"
+
+#include "InterceptRouting/InterceptRouting.h"
+#include "InterceptRouting/RoutingPlugin/RoutingPlugin.h"
+
+using namespace zz;
+
+void log_hex_format(uint8_t *buffer, uint32_t buffer_size) {
+  char output[1024] = {0};
+  for (int i = 0; i < buffer_size && i < sizeof(output); i++) {
+    snprintf(output + strlen(output), 3, "%02x ", *((uint8_t *)buffer + i));
+  }
+  DEBUG_LOG("%s", output);
+};
+
+void InterceptRouting::Prepare() {
+}
+
+// generate relocated code
+bool InterceptRouting::GenerateRelocatedCode() {
+  uint32_t tramp_size = GetTrampolineBuffer()->GetBufferSize();
+  origin_ = new CodeMemBlock(entry_->patched_addr, tramp_size);
+  relocated_ = new CodeMemBlock();
+
+  auto buffer = (void *)entry_->patched_addr;
+#if defined(TARGET_ARCH_ARM)
+  if (entry_->thumb_mode) {
+    buffer = (void *)((addr_t)buffer + 1);
+  }
+#endif
+  GenRelocateCodeAndBranch(buffer, origin_, relocated_);
+  if (relocated_->size == 0) {
+    ERROR_LOG("[insn relocate]] failed");
+    return false;
+  }
+
+  // set the relocated instruction address
+  entry_->relocated_addr = relocated_->addr;
+
+  // save original prologue
+  memcpy((void *)entry_->origin_insns, (void *)origin_->addr, origin_->size);
+  entry_->origin_insn_size = origin_->size;
+
+  // log
+  DEBUG_LOG("[insn relocate] origin %p - %d", origin_->addr, origin_->size);
+  log_hex_format((uint8_t *)origin_->addr, origin_->size);
+
+  DEBUG_LOG("[insn relocate] relocated %p - %d", relocated_->addr, relocated_->size);
+  log_hex_format((uint8_t *)relocated_->addr, relocated_->size);
+
+  return true;
+}
+
+bool InterceptRouting::GenerateTrampolineBuffer(addr_t src, addr_t dst) {
+  // if near branch trampoline plugin enabled
+  if (RoutingPluginManager::near_branch_trampoline) {
+    auto plugin = static_cast<RoutingPluginInterface *>(RoutingPluginManager::near_branch_trampoline);
+    if (plugin->GenerateTrampolineBuffer(this, src, dst) == false) {
+      DEBUG_LOG("Failed enable near branch trampoline plugin");
+    }
+  }
+
+  if (GetTrampolineBuffer() == nullptr) {
+    auto tramp_buffer = GenerateNormalTrampolineBuffer(src, dst);
+    SetTrampolineBuffer(tramp_buffer);
+  }
+  return true;
+}
+
+// active routing, patch origin instructions as trampoline
+void InterceptRouting::Active() {
+  auto ret = DobbyCodePatch((void *)entry_->patched_addr, trampoline_buffer_->GetBuffer(),
+                            trampoline_buffer_->GetBufferSize());
+  if (ret == -1) {
+    ERROR_LOG("[intercept routing] active failed");
+    return;
+  }
+  DEBUG_LOG("[intercept routing] active");
+}
+
+void InterceptRouting::Commit() {
+  this->Active();
+}
+
+#if 0
+int InterceptRouting::PredefinedTrampolineSize() {
+#if __arm64__
+  return 12;
+#elif __arm__
+  return 8;
+#endif
+}
+#endif
+
+InterceptEntry *InterceptRouting::GetInterceptEntry() {
+  return entry_;
+};