Paul Vriens : advapi32/tests: Restore default cryptographic provider after the tests.

[Alexandre Julliard]

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

Author: Paul Vriens 
Date:   Mon Mar  8 11:16:15 2010 +0100

advapi32/tests: Restore default cryptographic provider after the tests.

---

dlls/advapi32/tests/crypt.c |   18 +++++++++++++++++-
 1 files changed, 17 insertions(+), 1 deletions(-)

diff --git a/dlls/advapi32/tests/crypt.c b/dlls/advapi32/tests/crypt.c
index 2a5993c..12288fd 100644
--- a/dlls/advapi32/tests/crypt.c
+++ b/dlls/advapi32/tests/crypt.c
@@ -856,6 +856,8 @@ static void test_set_provider_ex(void)
 {
    DWORD result;
    DWORD notNull = 5;
+        LPSTR curProvName = NULL;
+        DWORD curlen;
    
    /* results */
    LPSTR pszProvName = NULL;
@@ -867,6 +869,13 @@ static void test_set_provider_ex(void)
        return;
    }

+        /* store the current one */
+        pCryptGetDefaultProviderA(PROV_RSA_FULL, NULL, CRYPT_MACHINE_DEFAULT, NULL, &curlen);
+        if (!(curProvName = LocalAlloc(LMEM_ZEROINIT, curlen)))
+            return;
+        result = pCryptGetDefaultProviderA(PROV_RSA_FULL, NULL, CRYPT_MACHINE_DEFAULT, curProvName, &curlen);
+        ok(result, "%d\n", GetLastError());
+
    /* check pdwReserved for NULL */
    result = pCryptSetProviderExA(MS_DEF_PROV, PROV_RSA_FULL, &notNull, CRYPT_MACHINE_DEFAULT);
    ok(!result && GetLastError()==ERROR_INVALID_PARAMETER, "expected %i, got %d\n",
@@ -880,6 +889,7 @@ static void test_set_provider_ex(void)
                 ok( GetLastError() == ERROR_ACCESS_DENIED || broken(GetLastError() == ERROR_INVALID_PARAMETER),
                     "wrong error %u\n", GetLastError() );
    	skip("Not enough rights to remove the default provider\n");
+                LocalFree(curProvName);
    	return;
    }

@@ -889,13 +899,19 @@ static void test_set_provider_ex(void)
    /* call CryptGetDefaultProvider to see if they match */
    result = pCryptGetDefaultProviderA(PROV_RSA_FULL, NULL, CRYPT_MACHINE_DEFAULT, NULL, &cbProvName);
    if (!(pszProvName = LocalAlloc(LMEM_ZEROINIT, cbProvName)))
-		return;
+		goto reset;

result = pCryptGetDefaultProviderA(PROV_RSA_FULL, NULL, CRYPT_MACHINE_DEFAULT, pszProvName, &cbProvName);
    ok(result && !strcmp(MS_DEF_PROV, pszProvName), "expected %s, got %s\n", MS_DEF_PROV, pszProvName);
    ok(result && cbProvName==(strlen(MS_DEF_PROV) + 1), "expected %i, got %d\n", (lstrlenA(MS_DEF_PROV) + 1), cbProvName);

LocalFree(pszProvName);
+
+reset:
+        /* Set the provider back to it's original */
+        result = pCryptSetProviderExA(curProvName, PROV_RSA_FULL, NULL, CRYPT_MACHINE_DEFAULT);
+        ok(result, "%d\n", GetLastError());
+        LocalFree(curProvName);
 }

static void test_machine_guid(void)