Added some unit tests for the CryptAcquireContext API function.

dlls/advapi32/tests/.cvsignore

@@ -1,3 +1,4 @@
 Makefile
+crypt.ok
 registry.ok
 testlist.c

dlls/advapi32/tests/Makefile.in

@@ -6,6 +6,7 @@ TESTDLL   = advapi32.dll
 IMPORTS   = advapi32 kernel32
 
 CTESTS = \
+	crypt.c \
 	registry.c
 
 @MAKE_TEST_RULES@

dlls/advapi32/tests/crypt.c

@@ -0,0 +1,133 @@
+/*
+ * Unit tests for crypt functions
+ *
+ * Copyright (c) 2004 Michael Jung
+ *
+ * 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+ */
+
+#include <stdio.h>
+
+#include "wine/test.h"
+#include "windef.h"
+#include "winbase.h"
+#include "wincrypt.h"
+#include "winerror.h"
+
+static const char szRsaBaseProv[] = MS_DEF_PROV_A;
+static const char szNonExistentProv[] = "Wine Non Existent Cryptographic Provider v11.2";
+static const char szKeySet[] = "wine_test_keyset";
+static const char szBadKeySet[] = "wine_test_bad_keyset";
+#define NON_DEF_PROV_TYPE 999
+
+static void init_environment(void)
+{
+	HCRYPTPROV hProv;
+	
+	/* Ensure that container "wine_test_keyset" does exist */
+	if (!CryptAcquireContext(&hProv, szKeySet, szRsaBaseProv, PROV_RSA_FULL, 0))
+	{
+		CryptAcquireContext(&hProv, szKeySet, szRsaBaseProv, PROV_RSA_FULL, CRYPT_NEWKEYSET);
+	}
+	CryptReleaseContext(hProv, 0);
+
+	/* Ensure that container "wine_test_keyset" does exist in default PROV_RSA_FULL type provider */
+	if (!CryptAcquireContext(&hProv, szKeySet, NULL, PROV_RSA_FULL, 0))
+	{
+		CryptAcquireContext(&hProv, szKeySet, NULL, PROV_RSA_FULL, CRYPT_NEWKEYSET);
+	}
+	CryptReleaseContext(hProv, 0);
+
+	/* Ensure that container "wine_test_bad_keyset" does not exist. */
+	if (CryptAcquireContext(&hProv, szBadKeySet, szRsaBaseProv, PROV_RSA_FULL, 0))
+	{
+		CryptReleaseContext(hProv, 0);
+		CryptAcquireContext(&hProv, szBadKeySet, szRsaBaseProv, PROV_RSA_FULL, CRYPT_DELETEKEYSET);
+	}
+}
+
+static void clean_up_environment(void)
+{
+	HCRYPTPROV hProv;
+
+	/* Remove container "wine_test_keyset" */
+	if (CryptAcquireContext(&hProv, szKeySet, szRsaBaseProv, PROV_RSA_FULL, 0))
+	{
+		CryptReleaseContext(hProv, 0);
+		CryptAcquireContext(&hProv, szKeySet, szRsaBaseProv, PROV_RSA_FULL, CRYPT_DELETEKEYSET);
+	}
+
+	/* Remove container "wine_test_keyset" from default PROV_RSA_FULL type provider */
+	if (CryptAcquireContext(&hProv, szKeySet, NULL, PROV_RSA_FULL, 0))
+	{
+		CryptReleaseContext(hProv, 0);
+		CryptAcquireContext(&hProv, szKeySet, NULL, PROV_RSA_FULL, CRYPT_DELETEKEYSET);
+	}
+}
+
+static void test_acquire_context(void)
+{
+	BOOL result;
+	HCRYPTPROV hProv;
+
+	/* Provoke all kinds of error conditions (which are easy to provoke). 
+	 * The order of the error tests seems to match Windows XP's rsaenh.dll CSP,
+	 * but since this is likely to change between CSP versions, we don't check
+	 * this. Please don't change the order of tests. */
+	todo_wine {
+		result = CryptAcquireContext(&hProv, NULL, NULL, 0, 0);
+		ok(!result && GetLastError()==NTE_BAD_PROV_TYPE, "%08x\n", (unsigned int)GetLastError());
+	}
+	
+	result = CryptAcquireContext(&hProv, NULL, NULL, 1000, 0);
+	ok(!result && GetLastError()==NTE_BAD_PROV_TYPE, "%08x\n", (unsigned int)GetLastError());
+
+	result = CryptAcquireContext(&hProv, NULL, NULL, NON_DEF_PROV_TYPE, 0);
+	ok(!result && GetLastError()==NTE_PROV_TYPE_NOT_DEF, "%08x\n", (unsigned int)GetLastError());
+	
+	todo_wine {
+		result = CryptAcquireContext(&hProv, szKeySet, szNonExistentProv, PROV_RSA_FULL, 0);
+		ok(!result && GetLastError()==NTE_KEYSET_NOT_DEF, "%08x\n", (unsigned int)GetLastError());
+
+		result = CryptAcquireContext(&hProv, szKeySet, szRsaBaseProv, NON_DEF_PROV_TYPE, 0);
+		ok(!result && GetLastError()==NTE_PROV_TYPE_NO_MATCH, "%08x\n", (unsigned int)GetLastError());
+	}
+	
+	result = CryptAcquireContext(NULL, szKeySet, szRsaBaseProv, PROV_RSA_FULL, 0);
+	ok(!result && GetLastError()==ERROR_INVALID_PARAMETER, "%08x\n", (unsigned int)GetLastError());
+	
+	/* Last not least, try to really acquire a context. */
+	result = CryptAcquireContext(&hProv, szKeySet, szRsaBaseProv, PROV_RSA_FULL, 0);
+	ok(result, "%08x\n", (unsigned int)GetLastError());
+
+	if (GetLastError() == ERROR_SUCCESS) 
+		CryptReleaseContext(hProv, 0);
+
+	todo_wine {
+		/* Try again, witch an empty ("\0") szProvider parameter */
+		result = CryptAcquireContext(&hProv, szKeySet, "", PROV_RSA_FULL, 0);
+		ok(result, "%08x\n", (unsigned int)GetLastError());
+
+		if (GetLastError() == ERROR_SUCCESS)
+			CryptReleaseContext(hProv, 0);
+	}
+}
+
+START_TEST(crypt)
+{
+	init_environment();
+	test_acquire_context();
+	clean_up_environment();
+}