mirror of
https://github.com/dashr9230/SA-MP.git
synced 2024-12-22 22:47:29 +08:00
70 lines
1.5 KiB
C++
70 lines
1.5 KiB
C++
|
|
#include "CryptoContext.h"
|
|
#include "CryptoFns.h"
|
|
|
|
//------------------------------------
|
|
|
|
DWORD CCryptoContext::ms_dwRefCount = 0;
|
|
DWORD CCryptoContext::ms_dwProviderType = PROV_RSA_FULL;
|
|
LPTSTR CCryptoContext::ms_szProviderName = NULL;
|
|
LPTSTR CCryptoContext::ms_szContainerName = (LPTSTR)"SAMP";
|
|
HMODULE CCryptoContext::ms_hAdvApi32 = NULL;
|
|
|
|
//------------------------------------
|
|
|
|
CCryptoContext::CCryptoContext(void)
|
|
{
|
|
/*
|
|
if (!ms_hAdvApi32)
|
|
{
|
|
ms_hAdvApi32 = LoadLibrary("advapi32.dll");
|
|
}
|
|
*/
|
|
|
|
// Open existing context, if not found, create one!
|
|
if ( !CRYPT(AcquireContext)(&m_hCryptProv, ms_szContainerName, ms_szProviderName, ms_dwProviderType, 0) ) {
|
|
if( !CRYPT(AcquireContext)(&m_hCryptProv, ms_szContainerName, ms_szProviderName, ms_dwProviderType, CRYPT_NEWKEYSET) ) {
|
|
throw(1);
|
|
}
|
|
}
|
|
|
|
ms_dwRefCount++;
|
|
|
|
}
|
|
|
|
//------------------------------------
|
|
|
|
CCryptoContext::~CCryptoContext(void)
|
|
{
|
|
// Release the context
|
|
CRYPT(ReleaseContext)(m_hCryptProv, 0);
|
|
ms_dwRefCount--;
|
|
|
|
if (ms_dwRefCount == 0) {
|
|
/*
|
|
// Free the library
|
|
if (ms_hAdvApi32)
|
|
FreeLibrary(ms_hAdvApi32);
|
|
*/
|
|
|
|
// Delete the context
|
|
CRYPT(AcquireContext)(&m_hCryptProv, ms_szContainerName, ms_szProviderName, ms_dwProviderType, CRYPT_DELETEKEYSET);
|
|
}
|
|
}
|
|
|
|
//------------------------------------
|
|
|
|
HCRYPTPROV CCryptoContext::GetProvider()
|
|
{
|
|
return m_hCryptProv;
|
|
}
|
|
|
|
//------------------------------------
|
|
|
|
#ifdef ARCTOOL
|
|
void CCryptoContext::GenerateRandom(DWORD dwLength, BYTE* pbBuffer)
|
|
{
|
|
CryptGenRandom(m_hCryptProv, dwLength, pbBuffer);
|
|
}
|
|
#endif
|