Fixed "meta unload" segfault.

Added fake responces to server queries. Added dynamic link ents. (except android).
2019-07-31 14:05:36 +03:00 · 2019-07-31 14:05:36 +03:00 · 535f298621
commit 535f298621
parent 4f664f0162
16 changed files with 763 additions and 246 deletions
--- a/include/crlib/cr-hook.h
+++ b/include/crlib/cr-hook.h
@ -0,0 +1,132 @@
+//
+// Yet Another POD-Bot, based on PODBot by Markus Klinge ("CountFloyd").
+// Copyright (c) YaPB Development Team.
+//
+// This software is licensed under the BSD-style license.
+// Additional exceptions apply. For full license details, see LICENSE.txt or visit:
+//     https://yapb.ru/license
+//
+
+#pragma once
+
+#include <crlib/cr-basic.h>
+
+#if !defined (CR_WINDOWS)
+#  include <sys/mman.h>
+#endif
+
+CR_NAMESPACE_BEGIN
+
+class SimpleHook : DenyCopying {
+private:
+   enum : uint32 {
+      CodeLength = 12
+   };
+
+private:
+   bool m_patched;
+
+   uint32 m_pageSize;
+   uint32 m_origFunc;
+   uint32 m_hookFunc;
+
+   uint8 m_origBytes[CodeLength];
+   uint8 m_hookBytes[CodeLength];
+
+private:
+   void setPageSize () {
+#if defined (CR_WINDOWS)
+      SYSTEM_INFO sysinfo;
+      GetSystemInfo (&sysinfo);
+
+      m_pageSize = sysinfo.dwPageSize;
+#else 
+      m_pageSize = sysconf (_SC_PAGESIZE);
+#endif
+   }
+
+   inline void *align (void *address) {
+      return reinterpret_cast <void *> ((reinterpret_cast <long> (address) & ~(m_pageSize - 1)));
+   }
+
+   bool unprotect () {
+      auto orig = reinterpret_cast <void *> (m_origFunc);
+
+#if defined (CR_WINDOWS)
+      DWORD oldProt;
+
+      FlushInstructionCache (GetCurrentProcess (), orig, m_pageSize);
+      return VirtualProtect (orig, m_pageSize, PAGE_EXECUTE_READWRITE, &oldProt);
+#else 
+      auto aligned = align (orig);
+      return !mprotect (aligned, m_pageSize, PROT_READ | PROT_WRITE | PROT_EXEC);
+#endif
+   }
+
+public:
+   SimpleHook () : m_patched (false), m_origFunc (0), m_hookFunc (0), m_pageSize (0) {
+      setPageSize ();
+   }
+
+   ~SimpleHook () {
+      disable ();
+   }
+
+public:
+   bool patch (void *address, void *replacement) {
+      uint8 *ptr = reinterpret_cast <uint8 *> (address);
+
+      while (*reinterpret_cast <uint16 *> (ptr) == 0x25ff) {
+         ptr = **reinterpret_cast <uint8 * **> ((ptr + 2));
+      }
+      m_origFunc = reinterpret_cast <uint32> (ptr);
+
+      if (!m_origFunc) {
+         return false;
+      }
+      m_hookFunc = reinterpret_cast <uint32> (replacement);
+
+      m_hookBytes[0] = 0x68;
+      m_hookBytes[5] = 0xc3;
+
+      *reinterpret_cast <uint32 *> (&m_hookBytes[1]) = m_hookFunc;
+
+      if (unprotect ()) {
+         memcpy (m_origBytes, reinterpret_cast <void *> (m_origFunc), CodeLength);
+         return enable ();
+      }
+      return false;
+   }
+
+   bool enable () {
+      if (m_patched) {
+         return false;
+      }
+      m_patched = true;
+
+      if (unprotect ()) {
+         memcpy (reinterpret_cast <void *> (m_origFunc), m_hookBytes, CodeLength);
+         return true;
+      }
+      return false;
+   }
+
+   bool disable () {
+      if (!m_patched) {
+         return false;
+      }
+      m_patched = false;
+
+      if (unprotect ()) {
+         memcpy (reinterpret_cast <void *> (m_origFunc), m_origBytes, CodeLength);
+         return true;
+      }
+      return false;
+   }
+
+   bool enabled () const {
+      return m_patched;
+   }
+};
+
+CR_NAMESPACE_END