52 lines
1.5 KiB
C
52 lines
1.5 KiB
C
|
|
||
|
|
#include <winpr/crt.h>
|
||
|
|
#include <winpr/print.h>
|
||
|
|
#include <winpr/crypto.h>
|
||
|
|
#include <winpr/ssl.h>
|
||
|
|
#include <winpr/wlog.h>
|
||
|
|
|
||
|
|
static const char* SECRET_PASSWORD_TEST = "MySecretPassword123!";
|
||
|
|
|
||
|
|
int TestCryptoProtectMemory(int argc, char* argv[])
|
||
|
|
{
|
||
|
|
int cbPlainText;
|
||
|
|
int cbCipherText;
|
||
|
|
const char* pPlainText;
|
||
|
|
BYTE* pCipherText;
|
||
|
|
pPlainText = SECRET_PASSWORD_TEST;
|
||
|
|
cbPlainText = strlen(pPlainText) + 1;
|
||
|
|
cbCipherText = cbPlainText +
|
||
|
|
(CRYPTPROTECTMEMORY_BLOCK_SIZE - (cbPlainText % CRYPTPROTECTMEMORY_BLOCK_SIZE));
|
||
|
|
printf("cbPlainText: %d cbCipherText: %d\n", cbPlainText, cbCipherText);
|
||
|
|
pCipherText = (BYTE*)malloc(cbCipherText);
|
||
|
|
if (!pCipherText)
|
||
|
|
{
|
||
|
|
printf("Unable to allocate memory\n");
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
CopyMemory(pCipherText, pPlainText, cbPlainText);
|
||
|
|
ZeroMemory(&pCipherText[cbPlainText], (cbCipherText - cbPlainText));
|
||
|
|
winpr_InitializeSSL(WINPR_SSL_INIT_DEFAULT);
|
||
|
|
|
||
|
|
if (!CryptProtectMemory(pCipherText, cbCipherText, CRYPTPROTECTMEMORY_SAME_PROCESS))
|
||
|
|
{
|
||
|
|
printf("CryptProtectMemory failure\n");
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
|
||
|
|
printf("PlainText: %s (cbPlainText = %d, cbCipherText = %d)\n", pPlainText, cbPlainText,
|
||
|
|
cbCipherText);
|
||
|
|
winpr_HexDump("crypto.test", WLOG_DEBUG, pCipherText, cbCipherText);
|
||
|
|
|
||
|
|
if (!CryptUnprotectMemory(pCipherText, cbCipherText, CRYPTPROTECTMEMORY_SAME_PROCESS))
|
||
|
|
{
|
||
|
|
printf("CryptUnprotectMemory failure\n");
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
|
||
|
|
printf("Decrypted CipherText: %s\n", pCipherText);
|
||
|
|
SecureZeroMemory(pCipherText, cbCipherText);
|
||
|
|
free(pCipherText);
|
||
|
|
return 0;
|
||
|
|
}
|