Module: wine Branch: master Commit: b1a5291e314a7dbe3eca6f65baf5a967bea7fa00 URL: http://source.winehq.org/git/wine.git/?a=commit;h=b1a5291e314a7dbe3eca6f65ba...
Author: Marek Chmiel kcmark@gmail.com Date: Mon Aug 13 18:46:55 2012 -0500
dssenh/tests: Added encryption tests for the DSSENH cryptographic service provider.
---
dlls/dssenh/tests/dssenh.c | 162 ++++++++++++++++++++++++++++++++++++++++++++ 1 files changed, 162 insertions(+), 0 deletions(-)
diff --git a/dlls/dssenh/tests/dssenh.c b/dlls/dssenh/tests/dssenh.c index 3a5bdef..b5faef8 100644 --- a/dlls/dssenh/tests/dssenh.c +++ b/dlls/dssenh/tests/dssenh.c @@ -479,9 +479,171 @@ static void test_hash(const struct hash_test *tests, int testLen) ok(result, "Expected release of the DSS Enhanced provider.\n"); }
+struct encrypt_test { + ALG_ID algid; + DWORD keyLength; + const char *plain; + DWORD plainLen; + const BYTE *decrypted; + const BYTE *encrypted; +}; + +static const char dataToEncrypt1[] = "Great performance with Wine."; +static const char dataToEncrypt2[] = "Wine implements Windows API"; +static const char dataToEncrypt3[] = ""; + +static const BYTE encrypted3DES_1[] = { +0x6c,0x60,0x19,0x41,0x27,0xc1,0x16,0x69, 0x6f,0x96,0x0c,0x2e,0xa4,0x5f,0xf5,0x6a, +0xed,0x4b,0xec,0xd4,0x92,0x0c,0xe2,0x34, 0xe1,0x4a,0xb5,0xe2,0x05,0x43,0xfe,0x17 +}; +static const BYTE encrypted3DES_2[] = { +0x17,0xeb,0x80,0xde,0xac,0x4d,0x9e,0xd0, 0xa9,0xae,0x74,0xb5,0x86,0x1a,0xea,0xb4, +0x96,0x27,0x5d,0x75,0x4f,0xdd,0x87,0x60, 0xfc,0xaf,0xa1,0x82,0x83,0x09,0xf1,0xca +}; +static const BYTE encrypted3DES_3[] = {0xaf, 0x36, 0xc0, 0x3d, 0x78, 0x64, 0xc4, 0x4a}; + +static const BYTE encrypted3DES112_1[] = { +0xb3,0xf8,0x4b,0x08,0xd6,0x23,0xcb,0xca, 0x43,0x26,0xd9,0x9f,0x6b,0x99,0x09,0xe9, +0x8c,0x4c,0x7d,0xef,0x49,0xda,0x0b,0x44, 0xcc,0x8d,0x06,0x6b,0xed,0xb7,0xf1,0x67 +}; +static const BYTE encrypted3DES112_2[] = { +0xdc,0xcf,0x93,0x11,0x7a,0xe4,0xcd,0x3f, 0x11,0xd8,0xe0,0x1e,0xe0,0x8d,0x9c,0xba, +0x97,0x5d,0x74,0x4d,0x83,0x03,0x5c,0xf2, 0x01,0xaf,0xed,0x7a,0x87,0x8f,0x88,0x8b +}; +static const BYTE encrypted3DES112_3[] = {0x04, 0xb3, 0x9c, 0x59, 0x48, 0xc7, 0x2f, 0xd1}; + +static const BYTE encryptedDES_1[] = { +0x3d,0xdc,0x54,0xaf,0x66,0x72,0x4e,0xef, 0x9d,0x35,0x02,0xc2,0x1a,0xf4,0x1f,0x01, +0xb1,0xaf,0x13,0xd9,0xbe,0x7b,0xd4,0xf3, 0xf5,0x9d,0x2a,0xd8,0x32,0x90,0xe9,0x0b +}; +static const BYTE encryptedDES_2[] = { +0xa8,0x05,0xd7,0xe9,0x61,0xf4,0x6c,0xce, 0x95,0x2b,0x52,0x08,0x25,0x03,0x30,0xac, +0xd7,0xe7,0xd3,0x07,0xb2,0x68,0x63,0x7b, 0xe3,0xab,0x26,0x1e,0x5c,0xec,0x42,0x4f +}; +static const BYTE encryptedDES_3[] = {0x35, 0x02, 0xbb, 0x7c, 0x43, 0x5b, 0xf5, 0x59}; + +static const BYTE encryptedRC2_1[] = { +0x9e,0xcb,0xa2,0x27,0xc2,0xec,0x10,0xe0, 0x94,0xb3,0xc3,0x9d,0x7d,0xe2,0x12,0xe4, +0xb0,0xde,0xd9,0x46,0xca,0x1f,0xa6,0xfa, 0xa4,0x79,0x08,0x59,0xa6,0x00,0x62,0x16 +}; +static const BYTE encryptedRC2_2[] = { +0x29,0x06,0xfd,0xa1,0xe0,0x88,0x89,0xb0, 0x4d,0x7f,0x96,0x9d,0x2c,0x44,0xa1,0xd2, +0xbe,0xc6,0xaf,0x10,0xb8,0x86,0x68,0x1b, 0x1d,0x9f,0x3c,0xc4,0x12,0x02,0xbc,0x73 +}; +static const BYTE encryptedRC2_3[] = {0x26,0x40,0x73,0xfe,0x13,0xbb,0x32,0xa8}; + +static const BYTE encryptedRC4_1[] = { +0x5a,0x48,0xeb,0x16,0x96,0x23,0x16,0xb7, 0xbb,0x36,0xe8,0x43,0x88,0x74,0xb1,0x9d, +0x96,0xf0,0x84,0x0f,0x5a,0x56,0xf9,0x62, 0xae,0xb5,0x4a,0xce,0x52 +}; +static const BYTE encryptedRC4_2[] = { +0x4a,0x53,0xe0,0x12,0xc2,0x6a,0x0b,0xa2, 0xa5,0x35,0xea,0x54,0x8b,0x61,0xac,0xde, +0xa4,0xb9,0x9d,0x02,0x41,0x49,0xaa,0x15, 0x86,0x8b,0x66,0xe0 +}; +static const BYTE encryptedRC4_3[] = {0x1d}; + +static const struct encrypt_test encrypt_data[] = { + {CALG_3DES, 168 << 16, dataToEncrypt1, sizeof(dataToEncrypt1), (BYTE *)dataToEncrypt1, + encrypted3DES_1}, + {CALG_3DES, 168 << 16, dataToEncrypt2, sizeof(dataToEncrypt2), (BYTE *)dataToEncrypt2, + encrypted3DES_2}, + {CALG_3DES, 168 << 16, dataToEncrypt3, sizeof(dataToEncrypt3), (BYTE *)dataToEncrypt3, + encrypted3DES_3}, + {CALG_3DES_112, 112 << 16, dataToEncrypt1, sizeof(dataToEncrypt1), (BYTE *)dataToEncrypt1, + encrypted3DES112_1}, + {CALG_3DES_112, 112 << 16, dataToEncrypt2, sizeof(dataToEncrypt2), (BYTE *)dataToEncrypt2, + encrypted3DES112_2}, + {CALG_3DES_112, 112 << 16, dataToEncrypt3, sizeof(dataToEncrypt3), (BYTE *)dataToEncrypt3, + encrypted3DES112_3}, + {CALG_DES, 56 << 16, dataToEncrypt1, sizeof(dataToEncrypt1), (BYTE *)dataToEncrypt1, + encryptedDES_1}, + {CALG_DES, 56 << 16, dataToEncrypt2, sizeof(dataToEncrypt2), (BYTE *)dataToEncrypt2, + encryptedDES_2}, + {CALG_DES, 56 << 16, dataToEncrypt3, sizeof(dataToEncrypt3), (BYTE *)dataToEncrypt3, + encryptedDES_3}, + /* CALG_RC2 key unexpected results under Win2K when default key length is used, here we use + minimum length because Win2K's DSSENH provider has a differnt default key length compared + to the younger operating systems, though there is no default key len issue with CALG_RC4 */ + {CALG_RC2, 40 << 16, dataToEncrypt1, sizeof(dataToEncrypt1), (BYTE *)dataToEncrypt1, + encryptedRC2_1}, + {CALG_RC2, 40 << 16, dataToEncrypt2, sizeof(dataToEncrypt2), (BYTE *)dataToEncrypt2, + encryptedRC2_2}, + {CALG_RC2, 40 << 16, dataToEncrypt3, sizeof(dataToEncrypt3), (BYTE *)dataToEncrypt3, + encryptedRC2_3}, + {CALG_RC4, 40 << 16, dataToEncrypt1, sizeof(dataToEncrypt1), (BYTE *)dataToEncrypt1, + encryptedRC4_1}, + {CALG_RC4, 40 << 16, dataToEncrypt2, sizeof(dataToEncrypt2), (BYTE *)dataToEncrypt2, + encryptedRC4_2}, + {CALG_RC4, 40 << 16, dataToEncrypt3, sizeof(dataToEncrypt3), (BYTE *)dataToEncrypt3, + encryptedRC4_3} +}; + +static void test_data_encryption(const struct encrypt_test *tests, int testLen) +{ /* Here we test the same encryption ciphers as the RSAENH cryptographic service provider */ + HCRYPTPROV hProv = 0; + HCRYPTKEY pKey = 0; + HCRYPTHASH hHash; + const char dataToHash[] = "I love working with Wine"; + unsigned char pbData[36]; + DWORD dataLen; + BOOL result; + int i; + + /* 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 encryption tests.\n"); + return; + } + ok(result, "Expected no errors.\n"); + + /* testing various encryption algorithms */ + for(i = 0; i < testLen; i++) + { + memcpy(pbData, tests[i].plain, tests[i].plainLen); + dataLen = tests[i].plainLen; + + 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 key */ + result = CryptDeriveKey(hProv, tests[i].algid, hHash, tests[i].keyLength, &pKey); + ok(result, "Expected a derived key.\n"); + + result = CryptDestroyHash(hHash); + ok(result, "Expected destruction of hash after deriving key.\n"); + + /* testing CryptEncrypt with ALGID from array */ + result = CryptEncrypt(pKey, 0, TRUE, 0, pbData, &dataLen, 36); + ok(result, "Expected data encryption.\n"); + + /* Verify we have received expected 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.\n"); + + /* Verify we have received expected decrypted data */ + ok(!memcmp(pbData, tests[i].decrypted, dataLen), "Incorrect decrypted data.\n"); + + 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)); }