dssenh/tests: Added block cipher mode tests for the DSSENH cryptographic service provider.

This commit is contained in:
Marek Chmiel 2012-08-16 14:28:08 -05:00 committed by Alexandre Julliard
parent b1a5291e31
commit cb082b9079
1 changed files with 104 additions and 0 deletions

View File

@ -640,10 +640,114 @@ static void test_data_encryption(const struct encrypt_test *tests, int testLen)
ok(result, "Expected release of the provider\n");
}
struct ciphermode_test {
DWORD cipherMode;
BOOL expectedResult;
DWORD expectedError;
const BYTE *encrypted;
};
static const BYTE encryptedCFB[] = {
0x51,0x15,0x77,0xab,0x62,0x1f,0x7d,0xcb, 0x35,0x1e,0xd8,0xd3,0x2a,0x00,0xf0,0x94,
0x7c,0xa5,0x28,0xda,0xb8,0x81,0x15,0x99, 0xd1,0xd5,0x06,0x18,0xb8,0x85,0xfc,0xc9
};
static const BYTE encryptedCBC[] = {
0x8f,0x7b,0x56,0xeb,0xad,0x4d,0x76,0xc2, 0xd5,0x1d,0xf0,0x60,0x9d,0xde,0x96,0xe8,
0xb7,0x7b,0xeb,0x4b,0xee,0x3f,0xae,0x05, 0xdd,0x3b,0x62,0x47,0x7f,0x6f,0x79,0x6c
};
static const BYTE encryptedECB[] = {
0x8f,0x7b,0x56,0xeb,0xad,0x4d,0x76,0xc2, 0x8b,0xe0,0x4e,0xe4,0x98,0x4f,0xb8,0x3b,
0xf3,0xeb,0x6f,0x0a,0x57,0x91,0xdd,0xc7, 0x64,0x8b,0xb9,0x50,0xe6,0x5e,0x76,0x72
};
static const struct ciphermode_test ciphermode_data[] = {
{CRYPT_MODE_CFB, TRUE, 0xdeadbeef, encryptedCFB}, /* Testing cipher block chaining */
{CRYPT_MODE_CBC, TRUE, 0xdeadbeef, encryptedCBC}, /* Testing cipher feedback */
{CRYPT_MODE_ECB, TRUE, 0xdeadbeef, encryptedECB}, /* Testing electronic codebook */
{CRYPT_MODE_OFB, FALSE, NTE_BAD_DATA}/* DSSENH does not support Output Feedback cipher mode */
};
static void test_cipher_modes(const struct ciphermode_test *tests, int testLen)
{
HCRYPTPROV hProv = 0;
HCRYPTKEY pKey = 0;
HCRYPTHASH hHash;
const char plainText[] = "Testing block cipher modes.";
const char dataToHash[] = "GSOC is awesome!";
unsigned char pbData[36];
int plainLen = sizeof(plainText), i;
DWORD mode, dataLen;
BOOL result;
/* acquire dss enhanced provider */
SetLastError(0xdeadbeef);
result = CryptAcquireContextA(
&hProv, NULL, MS_ENH_DSS_DH_PROV, PROV_DSS_DH, CRYPT_VERIFYCONTEXT);
if(!result && GetLastError() == NTE_KEYSET_NOT_DEF)
{
skip("DSSENH is currently not available, skipping block cipher mode tests.\n");
return;
}
ok(result, "Expected no errors.\n");
SetLastError(0xdeadbeef);
result = CryptCreateHash(hProv, CALG_MD5, 0, 0, &hHash);
ok(result, "Expected creation of a MD5 hash for key derivation.\n");
result = CryptHashData(hHash, (BYTE *)dataToHash, sizeof(dataToHash), 0);
ok(result, "Expected data to be added to hash for key derivation.\n");
/* Derive a CALG_RC2 key, but could be any other encryption cipher */
result = CryptDeriveKey(hProv, CALG_RC2, hHash, 40 << 16, &pKey);
ok(result, "Expected a derived key.\n");
result = CryptDestroyHash(hHash);
ok(result, "Expected destruction of hash after deriving key.\n");
/* test block cipher modes */
for(i = 0; i < testLen; i++)
{
SetLastError(0xdeadbeef);
dataLen = plainLen;
mode = tests[i].cipherMode;
memcpy(pbData, plainText, plainLen);
result = CryptSetKeyParam(pKey, KP_MODE, (BYTE*)&mode, 0);
if(tests[i].expectedResult)
{
ok(result, "Expected setting of KP_MODE, got %x.\n", GetLastError());
result = CryptEncrypt(pKey, 0, TRUE, 0, pbData, &dataLen, 36);
ok(result, "Expected data encryption, got %x.\n", GetLastError());
/* Verify we have the correct encrypted data */
ok(!memcmp(pbData, tests[i].encrypted, dataLen), "Incorrect encrypted data.\n");
result = CryptDecrypt(pKey, 0, TRUE, 0, pbData, &dataLen);
ok(result, "Expected data decryption, got %x.\n", GetLastError());
/* Verify we have the correct decrypted data */
ok(!memcmp(pbData, (BYTE *)plainText, dataLen), "Incorrect decrypted data.\n");
}
else
{ /* Expected error */
ok(!result && GetLastError() == tests[i].expectedError, "Expected %d, got %x.\n",
tests[i].expectedError, GetLastError());
}
}
result = CryptDestroyKey(pKey);
ok(result, "Expected no DestroyKey errors.\n");
result = CryptReleaseContext(hProv, 0);
ok(result, "Expected release of the provider.\n");
}
START_TEST(dssenh)
{
test_acquire_context();
test_keylength();
test_hash(hash_data, TESTLEN(hash_data));
test_data_encryption(encrypt_data, TESTLEN(encrypt_data));
test_cipher_modes(ciphermode_data, TESTLEN(ciphermode_data));
}