UNLOZE/sourcemod
10
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Print OS errors if mprotect fails

Browse Source 
This helped catch a bug in DHooks before, so it's worth adding it in here as well.
This commit is contained in:
Peace-Maker 2021-11-02 16:46:12 +01:00 committed by peace-maker
parent 8f7ecf66a4
commit f32b210ec9
1 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
public/CDetour/detourhelpers.h
View File

@ -39,6 +39,8 @@
#include <amtl/am-bits.h>
#include <jit/x86/x86_macros.h>
#include <amtl/os/am-system-errors.h>
#include <cstdio>
struct patch_t
{
@ -53,14 +55,21 @@ struct patch_t
inline void ProtectMemory(void *addr, int length, int prot)
{
char error[256];
#if defined PLATFORM_POSIX
long pageSize = sysconf(_SC_PAGESIZE);
void *startPage = ke::AlignedBase(addr, pageSize);
void *endPage = ke::AlignedBase((void *)((intptr_t)addr + length), pageSize);
mprotect(startPage, ((intptr_t)endPage - (intptr_t)startPage) + pageSize, prot);
if (mprotect(startPage, ((intptr_t)endPage - (intptr_t)startPage) + pageSize, prot) == -1) {
ke::FormatSystemError(error, sizeof(error));
fprintf(stderr, "mprotect: %s\n", error);
}
#elif defined PLATFORM_WINDOWS
DWORD old_prot;
VirtualProtect(addr, length, prot, &old_prot);
if (!VirtualProtect(addr, length, prot, &old_prot)) {
ke::FormatSystemError(error, sizeof(error));
fprintf(stderr, "VirtualProtect: %s\n", error);
}
#endif
}