Module: wine Branch: master Commit: 49a20d1af7fb9d3c02db8a51499dbf78a58230b8 URL: http://source.winehq.org/git/wine.git/?a=commit;h=49a20d1af7fb9d3c02db8a5149...
Author: Paul Vriens paul.vriens.wine@gmail.com Date: Tue Jun 5 14:58:04 2007 +0200
wintrust/tests: Start of some crypt tests for wintrust.
---
dlls/wintrust/tests/Makefile.in | 1 + dlls/wintrust/tests/crypt.c | 170 +++++++++++++++++++++++++++++++++++++++ 2 files changed, 171 insertions(+), 0 deletions(-)
diff --git a/dlls/wintrust/tests/Makefile.in b/dlls/wintrust/tests/Makefile.in index 790b1af..50fadcc 100644 --- a/dlls/wintrust/tests/Makefile.in +++ b/dlls/wintrust/tests/Makefile.in @@ -6,6 +6,7 @@ TESTDLL = wintrust.dll IMPORTS = user32 advapi32 kernel32
CTESTS = \ + crypt.c \ register.c
@MAKE_TEST_RULES@ diff --git a/dlls/wintrust/tests/crypt.c b/dlls/wintrust/tests/crypt.c new file mode 100644 index 0000000..a4f1b70 --- /dev/null +++ b/dlls/wintrust/tests/crypt.c @@ -0,0 +1,170 @@ +/* Unit test suite for wintrust crypt functions + * + * Copyright 2007 Paul Vriens + * + * This library is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License as published by the Free Software Foundation; either + * version 2.1 of the License, or (at your option) any later version. + * + * This library is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * Lesser General Public License for more details. + * + * You should have received a copy of the GNU Lesser General Public + * License along with this library; if not, write to the Free Software + * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA + * + */ + +#include <stdarg.h> +#include <stdio.h> + +#include "windows.h" +#include "mscat.h" + +#include "wine/test.h" + +static HMODULE hWintrust = 0; + +static BOOL (WINAPI * pCryptCATAdminAcquireContext)(HCATADMIN*, const GUID*, DWORD); +static BOOL (WINAPI * pCryptCATAdminReleaseContext)(HCATADMIN, DWORD); + +#define WINTRUST_GET_PROC(func) \ + p ## func = (void*)GetProcAddress(hWintrust, #func); \ + if(!p ## func) { \ + trace("GetProcAddress(%s) failed\n", #func); \ + } + +static BOOL InitFunctionPtrs(void) +{ + hWintrust = LoadLibraryA("wintrust.dll"); + + if(!hWintrust) + { + skip("Could not load wintrust.dll\n"); + return FALSE; + } + + WINTRUST_GET_PROC(CryptCATAdminAcquireContext) + WINTRUST_GET_PROC(CryptCATAdminReleaseContext) + + return TRUE; +} + +static void test_context(void) +{ + BOOL ret; + HCATADMIN hca; + static GUID dummy = { 0xdeadbeef, 0xdead, 0xbeef, { 0xde,0xad,0xbe,0xef,0xde,0xad,0xbe,0xef }}; + static GUID unknown = { 0xC689AABA, 0x8E78, 0x11D0, { 0x8C,0x47,0x00,0xC0,0x4F,0xC2,0x95,0xEE }}; /* WINTRUST.DLL */ + + if (!pCryptCATAdminAcquireContext || !pCryptCATAdminReleaseContext) + { + skip("CryptCATAdminAcquireContext and/or CryptCATAdminReleaseContext are not available\n"); + return; + } + + /* All NULL */ + SetLastError(0xdeadbeef); + ret = pCryptCATAdminAcquireContext(NULL, NULL, 0); + todo_wine + { + ok(!ret, "Expected failure\n"); + ok(GetLastError() == ERROR_INVALID_PARAMETER, + "Expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError()); + } + + /* NULL GUID */ + SetLastError(0xdeadbeef); + ret = pCryptCATAdminAcquireContext(&hca, NULL, 0); + ok(ret, "Expected success\n"); + ok(GetLastError() == ERROR_SUCCESS || + GetLastError() == 0xdeadbeef /* Vista */, + "Expected ERROR_SUCCESS or 0xdeadbeef, got %d\n", GetLastError()); + ok(hca != NULL, "Expected a context handle, got NULL\n"); + + /* All NULL */ + SetLastError(0xdeadbeef); + ret = pCryptCATAdminReleaseContext(NULL, 0); + todo_wine + { + ok(!ret, "Expected failure\n"); + ok(GetLastError() == ERROR_INVALID_PARAMETER, + "Expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError()); + } + + /* Proper release */ + SetLastError(0xdeadbeef); + ret = pCryptCATAdminReleaseContext(hca, 0); + ok(ret, "Expected success\n"); + ok(GetLastError() == 0xdeadbeef, + "Expected no change in last error, got %d\n", GetLastError()); + + /* Try to release a second time */ + SetLastError(0xdeadbeef); + ret = pCryptCATAdminReleaseContext(hca, 0); + todo_wine + { + ok(!ret, "Expected failure\n"); + ok(GetLastError() == ERROR_INVALID_PARAMETER, + "Expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError()); + } + + /* NULL context handle and dummy GUID */ + SetLastError(0xdeadbeef); + ret = pCryptCATAdminAcquireContext(NULL, &dummy, 0); + todo_wine + { + ok(!ret, "Expected failure\n"); + ok(GetLastError() == ERROR_INVALID_PARAMETER, + "Expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError()); + } + + /* Correct context handle and dummy GUID */ + SetLastError(0xdeadbeef); + ret = pCryptCATAdminAcquireContext(&hca, &dummy, 0); + ok(ret, "Expected success\n"); + ok(GetLastError() == ERROR_SUCCESS || + GetLastError() == 0xdeadbeef /* Vista */, + "Expected ERROR_SUCCESS or 0xdeadbeef, got %d\n", GetLastError()); + ok(hca != NULL, "Expected a context handle, got NULL\n"); + + ret = pCryptCATAdminReleaseContext(hca, 0); + ok(ret, "Expected success\n"); + + /* Correct context handle and GUID */ + SetLastError(0xdeadbeef); + ret = pCryptCATAdminAcquireContext(&hca, &unknown, 0); + ok(ret, "Expected success\n"); + ok(GetLastError() == ERROR_SUCCESS || + GetLastError() == 0xdeadbeef /* Vista */, + "Expected ERROR_SUCCESS or 0xdeadbeef, got %d\n", GetLastError()); + ok(hca != NULL, "Expected a context handle, got NULL\n"); + + ret = pCryptCATAdminReleaseContext(hca, 0); + ok(ret, "Expected success\n"); + + /* Flags not equal to 0 */ + SetLastError(0xdeadbeef); + ret = pCryptCATAdminAcquireContext(&hca, &unknown, 1); + ok(ret, "Expected success\n"); + ok(GetLastError() == ERROR_SUCCESS || + GetLastError() == 0xdeadbeef /* Vista */, + "Expected ERROR_SUCCESS or 0xdeadbeef, got %d\n", GetLastError()); + ok(hca != NULL, "Expected a context handle, got NULL\n"); + + ret = pCryptCATAdminReleaseContext(hca, 0); + ok(ret, "Expected success\n"); +} + +START_TEST(crypt) +{ + if(!InitFunctionPtrs()) + return; + + test_context(); + + FreeLibrary(hWintrust); +}