mirror of
https://github.com/alliedmodders/metamod-source.git
synced 2025-01-31 20:52:18 +01:00
Merge.
This commit is contained in:
commit
f54b5f402e
@ -12,7 +12,7 @@
|
|||||||
#define _INCLUDE_SMM_LIST_H
|
#define _INCLUDE_SMM_LIST_H
|
||||||
|
|
||||||
#include <new>
|
#include <new>
|
||||||
#include <malloc.h>
|
#include <stdlib.h>
|
||||||
|
|
||||||
namespace SourceHook
|
namespace SourceHook
|
||||||
{
|
{
|
||||||
|
@ -16,12 +16,12 @@
|
|||||||
// Unprotect now sets to readwrite
|
// Unprotect now sets to readwrite
|
||||||
// The vtable doesn't need to be executable anyway
|
// The vtable doesn't need to be executable anyway
|
||||||
|
|
||||||
# if /********/ defined _WIN32
|
# if SH_XP == SH_XP_WINAPI
|
||||||
# include <windows.h>
|
# include <windows.h>
|
||||||
# define SH_MEM_READ 1
|
# define SH_MEM_READ 1
|
||||||
# define SH_MEM_WRITE 2
|
# define SH_MEM_WRITE 2
|
||||||
# define SH_MEM_EXEC 4
|
# define SH_MEM_EXEC 4
|
||||||
# elif /******/ defined __linux__
|
# elif SH_XP == SH_XP_POSIX
|
||||||
# include <sys/mman.h>
|
# include <sys/mman.h>
|
||||||
# include <stdio.h>
|
# include <stdio.h>
|
||||||
# include <signal.h>
|
# include <signal.h>
|
||||||
@ -48,7 +48,7 @@ namespace SourceHook
|
|||||||
{
|
{
|
||||||
inline bool SetMemAccess(void *addr, size_t len, int access)
|
inline bool SetMemAccess(void *addr, size_t len, int access)
|
||||||
{
|
{
|
||||||
# ifdef __linux__
|
# if SH_XP == SH_XP_POSIX
|
||||||
return mprotect(SH_LALIGN(addr), len + SH_LALDIF(addr), access)==0 ? true : false;
|
return mprotect(SH_LALIGN(addr), len + SH_LALDIF(addr), access)==0 ? true : false;
|
||||||
# else
|
# else
|
||||||
DWORD tmp;
|
DWORD tmp;
|
||||||
@ -69,17 +69,25 @@ namespace SourceHook
|
|||||||
# endif
|
# endif
|
||||||
}
|
}
|
||||||
|
|
||||||
#ifdef __linux__
|
#if SH_XP == SH_XP_POSIX
|
||||||
namespace
|
namespace
|
||||||
{
|
{
|
||||||
bool g_BadReadCalled;
|
bool g_BadReadCalled;
|
||||||
jmp_buf g_BadReadJmpBuf;
|
jmp_buf g_BadReadJmpBuf;
|
||||||
|
|
||||||
|
# if SH_SYS == SH_SYS_LINUX
|
||||||
static void BadReadHandler(int sig)
|
static void BadReadHandler(int sig)
|
||||||
{
|
{
|
||||||
if (g_BadReadCalled)
|
if (g_BadReadCalled)
|
||||||
longjmp(g_BadReadJmpBuf, 1);
|
longjmp(g_BadReadJmpBuf, 1);
|
||||||
}
|
}
|
||||||
|
# elif SH_SYS == SH_SYS_APPLE
|
||||||
|
static void BadReadHandler(int signal, siginfo_t* my_siginfo, void* my_context)
|
||||||
|
{
|
||||||
|
if (g_BadReadCalled)
|
||||||
|
longjmp(g_BadReadJmpBuf, 1);
|
||||||
|
}
|
||||||
|
# endif
|
||||||
}
|
}
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
@ -93,7 +101,7 @@ namespace SourceHook
|
|||||||
{
|
{
|
||||||
bool ModuleInMemory(char *addr, size_t len)
|
bool ModuleInMemory(char *addr, size_t len)
|
||||||
{
|
{
|
||||||
#ifdef __linux__
|
#if SH_SYS == SH_SYS_LINUX
|
||||||
// On linux, first check /proc/self/maps
|
// On linux, first check /proc/self/maps
|
||||||
long lower = reinterpret_cast<long>(addr);
|
long lower = reinterpret_cast<long>(addr);
|
||||||
long upper = lower + len;
|
long upper = lower + len;
|
||||||
@ -175,7 +183,29 @@ namespace SourceHook
|
|||||||
signal(SIGSEGV, prevHandler);
|
signal(SIGSEGV, prevHandler);
|
||||||
|
|
||||||
return false;
|
return false;
|
||||||
#else
|
#elif SH_SYS == SH_SYS_APPLE
|
||||||
|
struct sigaction sa, osa;
|
||||||
|
sa.sa_sigaction = BadReadHandler;
|
||||||
|
sa.sa_flags = SA_SIGINFO | SA_RESTART;
|
||||||
|
|
||||||
|
g_BadReadCalled = true;
|
||||||
|
|
||||||
|
if (setjmp(g_BadReadJmpBuf))
|
||||||
|
return false;
|
||||||
|
|
||||||
|
if (sigaction(SIGBUS, &sa, &osa) == -1)
|
||||||
|
return false;
|
||||||
|
|
||||||
|
volatile const char *p = reinterpret_cast<const char *>(addr);
|
||||||
|
char dummy;
|
||||||
|
|
||||||
|
for (size_t i = 0; i < len; i++)
|
||||||
|
dummy = p[i];
|
||||||
|
|
||||||
|
g_BadReadCalled = false;
|
||||||
|
|
||||||
|
return true;
|
||||||
|
#elif SH_SYS == SH_SYS_WINAPI
|
||||||
// On Win32, simply use IsBadReadPtr
|
// On Win32, simply use IsBadReadPtr
|
||||||
return !IsBadReadPtr(addr, len);
|
return !IsBadReadPtr(addr, len);
|
||||||
#endif
|
#endif
|
||||||
|
@ -61,11 +61,21 @@
|
|||||||
// System
|
// System
|
||||||
#define SH_SYS_WIN32 1
|
#define SH_SYS_WIN32 1
|
||||||
#define SH_SYS_LINUX 2
|
#define SH_SYS_LINUX 2
|
||||||
|
#define SH_SYS_APPLE 3
|
||||||
|
|
||||||
|
// OS
|
||||||
|
#define SH_XP_POSIX 10
|
||||||
|
#define SH_XP_WINAPI 20
|
||||||
|
|
||||||
#ifdef _WIN32
|
#ifdef _WIN32
|
||||||
# define SH_SYS SH_SYS_WIN32
|
# define SH_SYS SH_SYS_WIN32
|
||||||
|
# define SH_XP SH_XP_WINAPI
|
||||||
#elif defined __linux__
|
#elif defined __linux__
|
||||||
# define SH_SYS SH_SYS_LINUX
|
# define SH_SYS SH_SYS_LINUX
|
||||||
|
# define SH_XP SH_XP_POSIX
|
||||||
|
#elif defined __APPLE__
|
||||||
|
# define SH_SYS SH_SYS_APPLE
|
||||||
|
# define SH_XP SH_XP_POSIX
|
||||||
#else
|
#else
|
||||||
# error Unsupported system
|
# error Unsupported system
|
||||||
#endif
|
#endif
|
||||||
|
Loading…
x
Reference in New Issue
Block a user