mirror of
https://github.com/ultimatepp/ultimatepp.git
synced 2026-06-03 14:22:33 -06:00
47 lines
1 KiB
C++
47 lines
1 KiB
C++
#include "Protect.h"
|
|
|
|
#ifdef PLATFORM_POSIX
|
|
#include <sys/mman.h>
|
|
#include <unistd.h>
|
|
|
|
bool PROTECT_WRITE_ACCESS(byte *start, size_t size, bool access)
|
|
{
|
|
// round start and size to page size -- needed for mprotect
|
|
long pSize = sysconf(_SC_PAGESIZE );
|
|
long delta = (long)start % pSize;
|
|
if(delta)
|
|
{
|
|
start -= delta;
|
|
size += delta;
|
|
}
|
|
return mprotect(start, size, access ? PROT_READ | PROT_WRITE | PROT_EXEC : PROT_READ | PROT_EXEC) == 0;
|
|
}
|
|
#else
|
|
|
|
bool PROTECT_WRITE_ACCESS(byte *start, size_t size, bool access)
|
|
{
|
|
dword oldProt;
|
|
bool res = VirtualProtect(start, size, access ? PAGE_EXECUTE_READWRITE : PAGE_EXECUTE_READ, &oldProt);
|
|
|
|
return res;
|
|
}
|
|
#endif
|
|
|
|
void PROTECT_DECRYPT(byte *start, size_t size, String const &key)
|
|
{
|
|
RC4 rc4(key);
|
|
|
|
rc4.Crypt(start, start, size);
|
|
}
|
|
|
|
void PROTECT_OBFUSCATE(byte *start, size_t len, byte *key, size_t keyLen)
|
|
{
|
|
String k;
|
|
for(unsigned i = 0; i < keyLen; i++)
|
|
k += *key++;
|
|
if(!PROTECT_WRITE_ACCESS(start, len, true))
|
|
return;
|
|
RC4 rc4(k);
|
|
rc4.Crypt(start, start, len);
|
|
PROTECT_WRITE_ACCESS(start, len, false);
|
|
}
|