mirror of
https://github.com/ultimatepp/ultimatepp.git
synced 2026-07-11 22:03:01 -06:00
Bazaar/Protect : updated Encryptor and test packages
git-svn-id: svn://ultimatepp.org/upp/trunk@2718 f0d560ea-af0d-0410-9eb7-867de7ffcac7
This commit is contained in:
micio
parent
8a462f4233
commit
91a08fa1cd
2 changed files with 224 additions and 155 deletions
|
|
@ -1,144 +1,195 @@
|
|||
#include <Core/Core.h>
|
||||
|
||||
#include <Protect/Protect.h>
|
||||
|
||||
using namespace Upp;
|
||||
|
||||
// search a buffer for a matching pattern
|
||||
// returns position of pattern
|
||||
byte *ProtectSearchBuf(byte *buf, byte *bufEnd, const byte *pattern, size_t patternLen)
|
||||
{
|
||||
while(buf < bufEnd - patternLen)
|
||||
{
|
||||
if(*buf != *pattern)
|
||||
{
|
||||
buf++;
|
||||
continue;
|
||||
}
|
||||
if(!memcmp(buf, pattern, patternLen))
|
||||
return buf;
|
||||
buf++;
|
||||
}
|
||||
return NULL;
|
||||
}
|
||||
|
||||
// encrypts code inside buffer
|
||||
int CryptBuf(byte *buf, byte *bufEnd, String const &key)
|
||||
{
|
||||
int patches = 0;
|
||||
|
||||
byte *bStart = buf;
|
||||
while( (bStart = ProtectSearchBuf(bStart, bufEnd, (const byte *)PROTECT_START_MARKER, strlen(PROTECT_START_MARKER))) != NULL)
|
||||
{
|
||||
// overwrite start pattern, just to fool a bit
|
||||
// symple pattern search
|
||||
for(unsigned i = 0; i < strlen(PROTECT_START_MARKER); i++)
|
||||
*bStart++ = (byte)(Random() & 0xff);
|
||||
|
||||
// locate end pattern
|
||||
byte *bEnd = ProtectSearchBuf(bStart, bufEnd, (const byte *)PROTECT_END_MARKER, strlen(PROTECT_END_MARKER));
|
||||
if(!bEnd)
|
||||
return 0;
|
||||
|
||||
// get size of chunk to patch
|
||||
size_t size = bEnd - bStart;
|
||||
|
||||
// overwrite end pattern, just to fool a bit
|
||||
// symple pattern search
|
||||
for(unsigned i = 0; i < strlen(PROTECT_END_MARKER); i++)
|
||||
*bEnd++ = (byte)(Random() & 0xff);
|
||||
|
||||
// crypt buffer
|
||||
RC4 rc4(key);
|
||||
rc4.Crypt(bStart, bStart, size);
|
||||
patches++;
|
||||
}
|
||||
|
||||
return patches;
|
||||
}
|
||||
|
||||
CONSOLE_APP_MAIN
|
||||
{
|
||||
// setup exit code for errors
|
||||
SetExitCode(1);
|
||||
|
||||
// command line parameters are :
|
||||
// 1) file to encrypt (full path)
|
||||
// 2) key, in hex ascii format (AABB001122CCDD....)
|
||||
if(CommandLine().GetCount() != 2)
|
||||
{
|
||||
Cerr() << "USAGE : ProtectEncrypt filename key\n";
|
||||
return;
|
||||
}
|
||||
|
||||
// gets key, check it and convert to HEX String
|
||||
|
||||
// string must contain an even number of digits
|
||||
// and they must be hax ones (0-9, A-F)
|
||||
String key0 = ToUpper(CommandLine()[1]);
|
||||
if(!key0.GetCount())
|
||||
{
|
||||
Cerr() << "Error: null KEY\n";
|
||||
return;
|
||||
}
|
||||
if(key0.GetCount() % 2)
|
||||
{
|
||||
Cerr() << "Error: KEY must contain an even number of digits\n";
|
||||
return;
|
||||
}
|
||||
String key;
|
||||
for(int i = 0; i < key0.GetCount()-1;)
|
||||
{
|
||||
byte b;
|
||||
char c = key0[i++];
|
||||
if(c >= '0' && c <= '9')
|
||||
b = c - '0';
|
||||
else if(c >= 'A' && c <= 'F')
|
||||
b = c - 'A' + 10;
|
||||
else
|
||||
{
|
||||
Cerr() << "Invalid hex digit '" << c << "' in key\n";
|
||||
return;
|
||||
}
|
||||
b <<= 4;
|
||||
c = key0[i++];
|
||||
if(c >= '0' && c <= '9')
|
||||
b += c - '0';
|
||||
else if(c >= 'A' && c <= 'F')
|
||||
b += c - 'A' + 10;
|
||||
else
|
||||
{
|
||||
Cerr() << "Invalid hex digit '" << c << "' in key\n";
|
||||
return;
|
||||
}
|
||||
key.Cat(b);
|
||||
}
|
||||
|
||||
// loads file into buffer
|
||||
String fName = CommandLine()[0];
|
||||
if(!FileExists(fName))
|
||||
{
|
||||
Cerr() << "File '" << fName << "' not found\n";
|
||||
return;
|
||||
}
|
||||
FileIn f(fName);
|
||||
dword size = (dword)f.GetSize();
|
||||
Buffer<byte>buf(size);
|
||||
f.GetAll(buf, size);
|
||||
|
||||
int patches = CryptBuf(buf, buf + size, key);
|
||||
if(patches)
|
||||
{
|
||||
Cerr() << "Successfully patched " << patches << " functions\n";
|
||||
FileOut f(fName);
|
||||
f.Put(buf, size);
|
||||
}
|
||||
else
|
||||
Cerr() << "No patchpoints found\n";
|
||||
|
||||
// sets up exit code
|
||||
SetExitCode(0);
|
||||
|
||||
|
||||
}
|
||||
|
||||
#include <Core/Core.h>
|
||||
|
||||
#include <Protect/Protect.h>
|
||||
|
||||
using namespace Upp;
|
||||
|
||||
// search a buffer for a matching pattern
|
||||
// returns position of pattern
|
||||
byte *ProtectSearchBuf(byte *buf, byte *bufEnd, const byte *pattern, size_t patternLen)
|
||||
{
|
||||
while(buf < bufEnd - patternLen)
|
||||
{
|
||||
if(*buf != *pattern)
|
||||
{
|
||||
buf++;
|
||||
continue;
|
||||
}
|
||||
if(!memcmp(buf, pattern, patternLen))
|
||||
return buf;
|
||||
buf++;
|
||||
}
|
||||
return NULL;
|
||||
}
|
||||
|
||||
// encrypts code inside buffer
|
||||
int CryptBuf(byte *buf, byte *bufEnd, String const &key)
|
||||
{
|
||||
int patches = 0;
|
||||
|
||||
byte *bStart = buf;
|
||||
while( (bStart = ProtectSearchBuf(bStart, bufEnd, (const byte *)PROTECT_START_MARKER, strlen(PROTECT_START_MARKER))) != NULL)
|
||||
{
|
||||
// overwrite start pattern, just to fool a bit
|
||||
// symple pattern search
|
||||
for(unsigned i = 0; i < strlen(PROTECT_START_MARKER); i++)
|
||||
*bStart++ = (byte)(Random() & 0xff);
|
||||
|
||||
// locate end pattern
|
||||
byte *bEnd = ProtectSearchBuf(bStart, bufEnd, (const byte *)PROTECT_END_MARKER, strlen(PROTECT_END_MARKER));
|
||||
if(!bEnd)
|
||||
return 0;
|
||||
|
||||
// get size of chunk to patch
|
||||
size_t size = bEnd - bStart;
|
||||
|
||||
// overwrite end pattern, just to fool a bit
|
||||
// symple pattern search
|
||||
for(unsigned i = 0; i < strlen(PROTECT_END_MARKER); i++)
|
||||
*bEnd++ = (byte)(Random() & 0xff);
|
||||
|
||||
// crypt buffer
|
||||
RC4 rc4(key);
|
||||
rc4.Crypt(bStart, bStart, size);
|
||||
patches++;
|
||||
}
|
||||
|
||||
return patches;
|
||||
}
|
||||
|
||||
// obfuscates code inside buffer
|
||||
int ObfuscateBuf(byte *buf, byte *bufEnd)
|
||||
{
|
||||
int patches = 0;
|
||||
|
||||
byte *bStart = buf;
|
||||
while( (bStart = ProtectSearchBuf(bStart, bufEnd, (const byte *)OBFUSCATE_START_MARKER, strlen(OBFUSCATE_START_MARKER))) != NULL)
|
||||
{
|
||||
// builds a random key, overwriting start pattern and following 10 bytes
|
||||
String key;
|
||||
for(unsigned i = 0; i < strlen(OBFUSCATE_START_MARKER) + 10; i++)
|
||||
{
|
||||
byte k = (byte)(Random() & 0xff);
|
||||
*bStart++ = k;
|
||||
key += (char)k;
|
||||
}
|
||||
|
||||
// locate end pattern
|
||||
byte *bEnd = ProtectSearchBuf(bStart, bufEnd, (const byte *)OBFUSCATE_END_MARKER, strlen(OBFUSCATE_END_MARKER));
|
||||
if(!bEnd)
|
||||
return 0;
|
||||
|
||||
// get size of chunk to patch
|
||||
size_t size = bEnd - bStart;
|
||||
|
||||
// overwrite end pattern, just to fool a bit
|
||||
// symple pattern search
|
||||
for(unsigned i = 0; i < strlen(OBFUSCATE_END_MARKER); i++)
|
||||
*bEnd++ = (byte)(Random() & 0xff);
|
||||
|
||||
// obfuscate buffer
|
||||
RC4 rc4(key);
|
||||
rc4.Crypt(bStart, bStart, size);
|
||||
patches++;
|
||||
}
|
||||
|
||||
return patches;
|
||||
}
|
||||
|
||||
CONSOLE_APP_MAIN
|
||||
{
|
||||
// setup exit code for errors
|
||||
SetExitCode(1);
|
||||
|
||||
// command line parameters are :
|
||||
// 1) file to encrypt (full path)
|
||||
// 2) key, in hex ascii format (AABB001122CCDD....)
|
||||
if(CommandLine().GetCount() != 2)
|
||||
{
|
||||
Cerr() << "USAGE : ProtectEncrypt filename key\n";
|
||||
return;
|
||||
}
|
||||
|
||||
// gets key, check it and convert to HEX String
|
||||
|
||||
// string must contain an even number of digits
|
||||
// and they must be hax ones (0-9, A-F)
|
||||
String key0 = ToUpper(CommandLine()[1]);
|
||||
if(!key0.GetCount())
|
||||
{
|
||||
Cerr() << "Error: null KEY\n";
|
||||
return;
|
||||
}
|
||||
if(key0.GetCount() % 2)
|
||||
{
|
||||
Cerr() << "Error: KEY must contain an even number of digits\n";
|
||||
return;
|
||||
}
|
||||
String key;
|
||||
for(int i = 0; i < key0.GetCount()-1;)
|
||||
{
|
||||
byte b;
|
||||
char c = key0[i++];
|
||||
if(c >= '0' && c <= '9')
|
||||
b = c - '0';
|
||||
else if(c >= 'A' && c <= 'F')
|
||||
b = c - 'A' + 10;
|
||||
else
|
||||
{
|
||||
Cerr() << "Invalid hex digit '" << c << "' in key\n";
|
||||
return;
|
||||
}
|
||||
b <<= 4;
|
||||
c = key0[i++];
|
||||
if(c >= '0' && c <= '9')
|
||||
b += c - '0';
|
||||
else if(c >= 'A' && c <= 'F')
|
||||
b += c - 'A' + 10;
|
||||
else
|
||||
{
|
||||
Cerr() << "Invalid hex digit '" << c << "' in key\n";
|
||||
return;
|
||||
}
|
||||
key.Cat(b);
|
||||
}
|
||||
|
||||
// loads file into buffer
|
||||
String fName = CommandLine()[0];
|
||||
if(!FileExists(fName))
|
||||
{
|
||||
Cerr() << "File '" << fName << "' not found\n";
|
||||
return;
|
||||
}
|
||||
FileIn f(fName);
|
||||
dword size = (dword)f.GetSize();
|
||||
Buffer<byte>buf(size);
|
||||
f.GetAll(buf, size);
|
||||
|
||||
// crypt part
|
||||
int patches = CryptBuf(buf, buf + size, key);
|
||||
if(patches)
|
||||
{
|
||||
Cerr() << "Successfully encrypted " << patches << " functions\n";
|
||||
FileOut f(fName);
|
||||
f.Put(buf, size);
|
||||
}
|
||||
else
|
||||
Cerr() << "No encrypt points found\n";
|
||||
|
||||
// obfuscation part
|
||||
patches = ObfuscateBuf(buf, buf + size);
|
||||
if(patches)
|
||||
{
|
||||
Cerr() << "Successfully obfuscated " << patches << " functions\n";
|
||||
FileOut f(fName);
|
||||
f.Put(buf, size);
|
||||
}
|
||||
else
|
||||
Cerr() << "No obfuscate points found\n";
|
||||
|
||||
// sets up exit code
|
||||
SetExitCode(0);
|
||||
|
||||
|
||||
}
|
||||
|
||||
|
|
|
|||
|
|
@ -10,37 +10,55 @@ String GetKey(void)
|
|||
|
||||
}
|
||||
|
||||
bool Decrypt(byte *start, size_t len)
|
||||
void Decrypt(byte *start, size_t len)
|
||||
{
|
||||
return PROTECT_DECRYPT ( start, len, GetKey() );
|
||||
PROTECT_DECRYPT ( start, len, GetKey() );
|
||||
}
|
||||
|
||||
double testfn1(double d, double e)
|
||||
double CryptedTest(double d, double e)
|
||||
{
|
||||
PROTECT_START_FUNC(Decrypt);
|
||||
|
||||
double f;
|
||||
f = d * e;
|
||||
|
||||
PromptOK("testfn1 DECRYPTED SUCCESFULLY!!!");
|
||||
PromptOK("CryptedTest DECRYPTED SUCCESFULLY!!!");
|
||||
return 2 * f + e;
|
||||
|
||||
PROTECT_END_FUNC;
|
||||
}
|
||||
|
||||
void testfn2(void)
|
||||
double ObfuscatedTest(double d, double e)
|
||||
{
|
||||
PROTECT_START_FUNC(Decrypt);
|
||||
|
||||
PromptOK("testfn2 DECRYPTED SUCCESFULLY!!!");
|
||||
return;
|
||||
// WARNING -- DON'T PUT ANY return STATEMENT BETWEEN
|
||||
// OBFUSCATE_START and OBFUSCATE_END
|
||||
OBFUSCATE_START_FUNC;
|
||||
|
||||
PROTECT_END_FUNC;
|
||||
double f;
|
||||
f = d * e;
|
||||
|
||||
PromptOK("ObfuscatedTest DEOBFUSCATED SUCCESFULLY!!!");
|
||||
|
||||
OBFUSCATE_END_FUNC;
|
||||
|
||||
return 2 * f + e;
|
||||
}
|
||||
|
||||
GUI_APP_MAIN
|
||||
{
|
||||
double d = testfn1(3, 4);
|
||||
testfn2();
|
||||
ON_PROTECT_BAD_KEY(Decrypt)
|
||||
{
|
||||
bool res = PromptYesNo("Bad key !!&Do you want to continue anyways ?");
|
||||
if(!res)
|
||||
exit(0);
|
||||
}
|
||||
|
||||
double d = CryptedTest(3, 4);
|
||||
d = ObfuscatedTest(3, 4);
|
||||
d = ObfuscatedTest(3, 4);
|
||||
|
||||
PromptOK("FINISHED OK !!");
|
||||
|
||||
}
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue