#include "stdafx.h"
#include <windows.h>
#include <wincrypt.h>
#include <WincryptEx.h>
#include <stdio.h>

HCERTSTORE CertificateStore;
PCCERT_CONTEXT CertificateContext=NULL;

int _tmain(int argc, _TCHAR* argv[])
{
	HCRYPTPROV CryptoProvider;
	BOOL isGetCryptoProvider=FALSE;
	DWORD TypeKey;
	BOOL b;

	CertificateStore = CertOpenStore(CERT_STORE_PROV_SYSTEM, 0,	NULL, CERT_SYSTEM_STORE_CURRENT_USER, L"MY");
	CertificateContext = CertFindCertificateInStore(CertificateStore, X509_ASN_ENCODING | PKCS_7_ASN_ENCODING, 0, CERT_FIND_SUBJECT_STR, L"testcert", NULL );
	isGetCryptoProvider = CryptAcquireCertificatePrivateKey(CertificateContext, 0, NULL, &CryptoProvider, &TypeKey, &b);
	if(!isGetCryptoProvider)
	{
		printf( "CryptAcquireCertificatePrivateKey\n" );
		exit(1);
	}
	HCRYPTKEY SessionKey = NULL;
	BOOL isCreateSessionKey = CryptGenKey(CryptoProvider, CALG_G28147, CRYPT_EXPORTABLE, &SessionKey);
	if(!isCreateSessionKey)
	{
		printf( "CryptGenKey\n" );
		exit(1);
	}

	BYTE *pbData;
	DWORD dwDataLen;
	BYTE data[1024];
	for( int i=0; i < 50000; i++ )
	{
		memset( data, ' ', 1024 );
		BOOL isGetIV = CryptGetKeyParam(SessionKey, KP_IV, NULL, &dwDataLen, 0);
		if(!isGetIV)
		{
			printf( "CryptGetKeyParam - size\n" );
			exit(1);
		}
		pbData = (BYTE*)malloc(dwDataLen);
		isGetIV = CryptGetKeyParam(SessionKey, KP_IV, pbData, &dwDataLen, 0);
		if(!isGetIV)
		{
			printf( "CryptGetKeyParam - data\n" );
			exit(1);
		}

		BOOL isEncrypt = CryptEncrypt(SessionKey, 0, TRUE, 0, data, &dwDataLen, 1024);
		if( !isEncrypt )
		{

			printf( "CryptEncrypt i=%d GetLastError()=%u\n", i, GetLastError() );
			exit(1);
		}
		free( pbData );
	}

	CryptReleaseContext( CryptoProvider, 0 );
	CertFreeCertificateContext( CertificateContext );
	CertCloseStore(CertificateStore,CERT_CLOSE_STORE_FORCE_FLAG);
	printf( "Success!\n" );
	return 0;
}