Paul Vriens : wintrust/tests: Start of some crypt tests for wintrust.

[Alexandre Julliard]

Module: wine
Branch: master
Commit: 49a20d1af7fb9d3c02db8a51499dbf78a58230b8
URL:    http://source.winehq.org/git/wine.git/?a=commit;h=49a20d1af7fb9d3c02db8a51499dbf78a58230b8

Author: Paul Vriens <paul.vriens.wine at 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);
+}