Paul Vriens : wintrust/tests:
Add tests for CryptCATAdminCalcHashFromFileHandle.
Alexandre Julliard
julliard at wine.codeweavers.com
Mon Jun 25 07:28:28 CDT 2007
Module: wine
Branch: master
Commit: 361280b22b5d74c9c4b17389d4930eb28c512f01
URL: http://source.winehq.org/git/wine.git/?a=commit;h=361280b22b5d74c9c4b17389d4930eb28c512f01
Author: Paul Vriens <paul.vriens.wine at gmail.com>
Date: Sun Jun 24 19:27:48 2007 +0200
wintrust/tests: Add tests for CryptCATAdminCalcHashFromFileHandle.
---
dlls/wintrust/tests/crypt.c | 135 +++++++++++++++++++++++++++++++++++++++++++
1 files changed, 135 insertions(+), 0 deletions(-)
diff --git a/dlls/wintrust/tests/crypt.c b/dlls/wintrust/tests/crypt.c
index a4f1b70..600299a 100644
--- a/dlls/wintrust/tests/crypt.c
+++ b/dlls/wintrust/tests/crypt.c
@@ -26,10 +26,15 @@
#include "wine/test.h"
+static char selfname[MAX_PATH];
+
+static CHAR CURR_DIR[MAX_PATH];
+
static HMODULE hWintrust = 0;
static BOOL (WINAPI * pCryptCATAdminAcquireContext)(HCATADMIN*, const GUID*, DWORD);
static BOOL (WINAPI * pCryptCATAdminReleaseContext)(HCATADMIN, DWORD);
+static BOOL (WINAPI * pCryptCATAdminCalcHashFromFileHandle)(HANDLE hFile, DWORD*, BYTE*, DWORD);
#define WINTRUST_GET_PROC(func) \
p ## func = (void*)GetProcAddress(hWintrust, #func); \
@@ -49,6 +54,7 @@ static BOOL InitFunctionPtrs(void)
WINTRUST_GET_PROC(CryptCATAdminAcquireContext)
WINTRUST_GET_PROC(CryptCATAdminReleaseContext)
+ WINTRUST_GET_PROC(CryptCATAdminCalcHashFromFileHandle)
return TRUE;
}
@@ -159,12 +165,141 @@ static void test_context(void)
ok(ret, "Expected success\n");
}
+/* TODO: Check whether SHA-1 is the algorithm that's always used */
+static void test_calchash(void)
+{
+ BOOL ret;
+ HANDLE file;
+ DWORD hashsize = 0;
+ BYTE* hash;
+ BYTE expectedhash[20] = {0x3a,0xa1,0x19,0x08,0xec,0xa6,0x0d,0x2e,0x7e,0xcc,0x7a,0xca,0xf5,0xb8,0x2e,0x62,0x6a,0xda,0xf0,0x19};
+ CHAR temp[MAX_PATH];
+ DWORD written;
+
+ if (!pCryptCATAdminCalcHashFromFileHandle)
+ {
+ skip("CryptCATAdminCalcHashFromFileHandle is not available\n");
+ return;
+ }
+
+ /* All NULL */
+ SetLastError(0xdeadbeef);
+ ret = pCryptCATAdminCalcHashFromFileHandle(NULL, NULL, NULL, 0);
+ todo_wine
+ {
+ ok(!ret, "Expected failure\n");
+ ok(GetLastError() == ERROR_INVALID_PARAMETER,
+ "Expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
+ }
+
+ /* NULL filehandle, rest is legal */
+ SetLastError(0xdeadbeef);
+ ret = pCryptCATAdminCalcHashFromFileHandle(NULL, &hashsize, NULL, 0);
+ todo_wine
+ {
+ ok(!ret, "Expected failure\n");
+ ok(GetLastError() == ERROR_INVALID_PARAMETER,
+ "Expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
+ }
+
+ /* Correct filehandle, rest is NULL */
+ file = CreateFileA(selfname, GENERIC_READ, 0, NULL, OPEN_EXISTING, 0, NULL);
+ SetLastError(0xdeadbeef);
+ ret = pCryptCATAdminCalcHashFromFileHandle(file, NULL, NULL, 0);
+ todo_wine
+ {
+ ok(!ret, "Expected failure\n");
+ ok(GetLastError() == ERROR_INVALID_PARAMETER,
+ "Expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
+ }
+ CloseHandle(file);
+
+ /* All OK, but dwFlags set to 1 */
+ file = CreateFileA(selfname, GENERIC_READ, 0, NULL, OPEN_EXISTING, 0, NULL);
+ SetLastError(0xdeadbeef);
+ ret = pCryptCATAdminCalcHashFromFileHandle(file, &hashsize, NULL, 1);
+ todo_wine
+ {
+ ok(!ret, "Expected failure\n");
+ ok(GetLastError() == ERROR_INVALID_PARAMETER,
+ "Expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
+ }
+ CloseHandle(file);
+
+ /* All OK, requesting the size of the hash */
+ file = CreateFileA(selfname, GENERIC_READ, 0, NULL, OPEN_EXISTING, 0, NULL);
+ SetLastError(0xdeadbeef);
+ ret = pCryptCATAdminCalcHashFromFileHandle(file, &hashsize, NULL, 0);
+ ok(ret, "Expected success\n");
+ todo_wine
+ {
+ ok(hashsize == 20," Expected a hash size of 20, got %d\n", hashsize);
+ ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER,
+ "Expected ERROR_INSUFFICIENT_BUFFER, got %d\n", GetLastError());
+ }
+ CloseHandle(file);
+
+ /* All OK, retrieve the hash
+ * Double the hash buffer to see what happens to the size parameter
+ */
+ file = CreateFileA(selfname, GENERIC_READ, 0, NULL, OPEN_EXISTING, 0, NULL);
+ hashsize *= 2;
+ hash = HeapAlloc(GetProcessHeap(), 0, hashsize);
+ SetLastError(0xdeadbeef);
+ ret = pCryptCATAdminCalcHashFromFileHandle(file, &hashsize, hash, 0);
+ ok(ret, "Expected success\n");
+ todo_wine
+ {
+ ok(hashsize == 20," Expected a hash size of 20, got %d\n", hashsize);
+ ok(GetLastError() == ERROR_SUCCESS,
+ "Expected ERROR_SUCCESS, got %d\n", GetLastError());
+ }
+ CloseHandle(file);
+ HeapFree(GetProcessHeap(), 0, hash);
+
+ /* Do the same test with a file created and filled by ourselves (and we thus
+ * have a known hash for).
+ */
+ GetTempFileNameA(CURR_DIR, "hsh", 0, temp);
+ file = CreateFileA(temp, GENERIC_WRITE, 0, NULL, OPEN_EXISTING, 0, NULL);
+ WriteFile(file, "Text in this file is needed to create a know hash", 49, &written, NULL);
+ CloseHandle(file);
+
+ /* All OK, first request the size and then retrieve the hash */
+ file = CreateFileA(temp, GENERIC_READ, 0, NULL, OPEN_EXISTING, 0, NULL);
+ hashsize = 0;
+ pCryptCATAdminCalcHashFromFileHandle(file, &hashsize, NULL, 0);
+ hash = HeapAlloc(GetProcessHeap(), 0, hashsize);
+ SetLastError(0xdeadbeef);
+ ret = pCryptCATAdminCalcHashFromFileHandle(file, &hashsize, hash, 0);
+ ok(ret, "Expected success\n");
+ todo_wine
+ {
+ ok(GetLastError() == ERROR_SUCCESS,
+ "Expected ERROR_SUCCESS, got %d\n", GetLastError());
+ ok(!memcmp(hash, expectedhash, sizeof(expectedhash)), "Hashes didn't match\n");
+ }
+ CloseHandle(file);
+
+ HeapFree(GetProcessHeap(), 0, hash);
+ DeleteFileA(temp);
+}
+
START_TEST(crypt)
{
+ int myARGC;
+ char** myARGV;
+
if(!InitFunctionPtrs())
return;
+ myARGC = winetest_get_mainargs(&myARGV);
+ strcpy(selfname, myARGV[0]);
+
+ GetCurrentDirectoryA(MAX_PATH, CURR_DIR);
+
test_context();
+ test_calchash();
FreeLibrary(hWintrust);
}
More information about the wine-cvs
mailing list