135 lines
2.8 KiB
C++
135 lines
2.8 KiB
C++
#include "rar.hpp"
|
|
|
|
#ifndef SFX_MODULE
|
|
#include "crypt1.cpp"
|
|
#include "crypt2.cpp"
|
|
#endif
|
|
#include "crypt3.cpp"
|
|
#include "crypt5.cpp"
|
|
|
|
|
|
CryptData::CryptData()
|
|
{
|
|
Method=CRYPT_NONE;
|
|
memset(KDF3Cache,0,sizeof(KDF3Cache));
|
|
memset(KDF5Cache,0,sizeof(KDF5Cache));
|
|
KDF3CachePos=0;
|
|
KDF5CachePos=0;
|
|
memset(CRCTab,0,sizeof(CRCTab));
|
|
}
|
|
|
|
|
|
CryptData::~CryptData()
|
|
{
|
|
cleandata(KDF3Cache,sizeof(KDF3Cache));
|
|
cleandata(KDF5Cache,sizeof(KDF5Cache));
|
|
}
|
|
|
|
|
|
|
|
|
|
void CryptData::DecryptBlock(byte *Buf,size_t Size)
|
|
{
|
|
switch(Method)
|
|
{
|
|
#ifndef SFX_MODULE
|
|
case CRYPT_RAR13:
|
|
Decrypt13(Buf,Size);
|
|
break;
|
|
case CRYPT_RAR15:
|
|
Crypt15(Buf,Size);
|
|
break;
|
|
case CRYPT_RAR20:
|
|
for (size_t I=0;I<Size;I+=CRYPT_BLOCK_SIZE)
|
|
DecryptBlock20(Buf+I);
|
|
break;
|
|
#endif
|
|
case CRYPT_RAR30:
|
|
case CRYPT_RAR50:
|
|
rin.blockDecrypt(Buf,Size,Buf);
|
|
break;
|
|
}
|
|
}
|
|
|
|
|
|
bool CryptData::SetCryptKeys(bool Encrypt,CRYPT_METHOD Method,
|
|
SecPassword *Password,const byte *Salt,
|
|
const byte *InitV,uint Lg2Cnt,byte *HashKey,byte *PswCheck)
|
|
{
|
|
if (!Password->IsSet() || Method==CRYPT_NONE)
|
|
return false;
|
|
|
|
CryptData::Method=Method;
|
|
|
|
wchar PwdW[MAXPASSWORD];
|
|
Password->Get(PwdW,ASIZE(PwdW));
|
|
char PwdA[MAXPASSWORD];
|
|
WideToChar(PwdW,PwdA,ASIZE(PwdA));
|
|
|
|
switch(Method)
|
|
{
|
|
#ifndef SFX_MODULE
|
|
case CRYPT_RAR13:
|
|
SetKey13(PwdA);
|
|
break;
|
|
case CRYPT_RAR15:
|
|
SetKey15(PwdA);
|
|
break;
|
|
case CRYPT_RAR20:
|
|
SetKey20(PwdA);
|
|
break;
|
|
#endif
|
|
case CRYPT_RAR30:
|
|
SetKey30(Encrypt,Password,PwdW,Salt);
|
|
break;
|
|
case CRYPT_RAR50:
|
|
SetKey50(Encrypt,Password,PwdW,Salt,InitV,Lg2Cnt,HashKey,PswCheck);
|
|
break;
|
|
}
|
|
cleandata(PwdA,sizeof(PwdA));
|
|
cleandata(PwdW,sizeof(PwdW));
|
|
return true;
|
|
}
|
|
|
|
|
|
// Use the current system time to additionally randomize data.
|
|
static void TimeRandomize(byte *RndBuf,size_t BufSize)
|
|
{
|
|
static uint Count=0;
|
|
RarTime CurTime;
|
|
CurTime.SetCurrentTime();
|
|
uint64 Random=CurTime.GetWin()+clock();
|
|
for (size_t I=0;I<BufSize;I++)
|
|
{
|
|
byte RndByte = byte (Random >> ( (I & 7) * 8 ));
|
|
RndBuf[I]=byte( (RndByte ^ I) + Count++);
|
|
}
|
|
}
|
|
|
|
|
|
|
|
|
|
// Fill buffer with random data.
|
|
void GetRnd(byte *RndBuf,size_t BufSize)
|
|
{
|
|
bool Success=false;
|
|
#if defined(_WIN_ALL)
|
|
HCRYPTPROV hProvider = 0;
|
|
if (CryptAcquireContext(&hProvider, 0, 0, PROV_RSA_FULL, CRYPT_VERIFYCONTEXT | CRYPT_SILENT))
|
|
{
|
|
Success=CryptGenRandom(hProvider, (DWORD)BufSize, RndBuf) == TRUE;
|
|
CryptReleaseContext(hProvider, 0);
|
|
}
|
|
#elif defined(_UNIX)
|
|
FILE *rndf = fopen("/dev/urandom", "r");
|
|
if (rndf!=NULL)
|
|
{
|
|
Success=fread(RndBuf, BufSize, 1, rndf) == BufSize;
|
|
fclose(rndf);
|
|
}
|
|
#endif
|
|
// We use this code only as the last resort if code above failed.
|
|
if (!Success)
|
|
TimeRandomize(RndBuf,BufSize);
|
|
}
|