Implemented CPGetProvParam's PP_ENUMCONTAINERS parameter type.

dlls/rsaenh/rsaenh.c

@@ -119,6 +119,7 @@ typedef struct tagKEYCONTAINER
     DWORD        dwFlags;
     DWORD        dwPersonality;
     DWORD        dwEnumAlgsCtr;
+    DWORD        dwEnumContainersCtr;
     CHAR         szName[MAX_PATH];
     CHAR         szProvName[MAX_PATH];
     HCRYPTKEY    hKeyExchangeKeyPair;
@@ -977,6 +978,24 @@ static HCRYPTPROV new_key_container(PCHAR pszContainerName, DWORD dwFlags, PVTab
                 pKeyContainer->dwPersonality = RSAENH_PERSONALITY_STRONG;
             }
         }
+
+        /* The new key container has to be inserted into the CSP immediately 
+         * after creation to be available for CPGetProvParam's PP_ENUMCONTAINERS. */
+        if (!(dwFlags & CRYPT_VERIFYCONTEXT)) {
+            BYTE szRSABase[MAX_PATH];
+            HKEY hRootKey, hKey;
+
+            sprintf(szRSABase, RSAENH_REGKEY, pKeyContainer->szName);
+
+            if (pKeyContainer->dwFlags & CRYPT_MACHINE_KEYSET) {
+                hRootKey = HKEY_LOCAL_MACHINE;
+            } else {
+                hRootKey = HKEY_CURRENT_USER;
+            }
+
+            RegCreateKeyA(hRootKey, szRSABase, &hKey);
+            RegCloseKey(hKey);
+        }
     }
 
     return hKeyContainer;
@@ -2717,6 +2736,8 @@ BOOL WINAPI RSAENH_CPGetProvParam(HCRYPTPROV hProv, DWORD dwParam, BYTE *pbData,
     KEYCONTAINER *pKeyContainer;
     PROV_ENUMALGS provEnumalgs;
     DWORD dwTemp;
+    BYTE szRSABase[MAX_PATH];
+    HKEY hKey, hRootKey;
    
     /* This is for dwParam 41, which does not seem to be documented
      * on MSDN. IE6 SP1 asks for it in the 'About' dialog, however.
@@ -2764,6 +2785,47 @@ BOOL WINAPI RSAENH_CPGetProvParam(HCRYPTPROV hProv, DWORD dwParam, BYTE *pbData,
             dwTemp = 8;
             return copy_param(pbData, pdwDataLen, (CONST BYTE*)&dwTemp, sizeof(dwTemp));
  
+        case PP_ENUMCONTAINERS:
+            if ((dwFlags & CRYPT_FIRST) == CRYPT_FIRST) pKeyContainer->dwEnumContainersCtr = 0;
+
+            if (!pbData) {
+                *pdwDataLen = (DWORD)MAX_PATH + 1;
+                return TRUE;
+            }
+ 
+            sprintf(szRSABase, RSAENH_REGKEY, "");
+
+            if (dwFlags & CRYPT_MACHINE_KEYSET) {
+                hRootKey = HKEY_LOCAL_MACHINE;
+            } else {
+                hRootKey = HKEY_CURRENT_USER;
+            }
+
+            if (RegOpenKeyExA(hRootKey, szRSABase, 0, KEY_READ, &hKey) != ERROR_SUCCESS)
+            {
+                SetLastError(ERROR_NO_MORE_ITEMS);
+                return FALSE;
+            }
+
+            dwTemp = *pdwDataLen;
+            switch (RegEnumKeyExA(hKey, pKeyContainer->dwEnumContainersCtr, pbData, &dwTemp,
+                    NULL, NULL, NULL, NULL))
+            {
+                case ERROR_MORE_DATA:
+                    *pdwDataLen = (DWORD)MAX_PATH + 1;
+ 
+                case ERROR_SUCCESS:
+                    pKeyContainer->dwEnumContainersCtr++;
+                    RegCloseKey(hKey);
+                    return TRUE;
+
+                case ERROR_NO_MORE_ITEMS:
+                default:
+                    SetLastError(ERROR_NO_MORE_ITEMS);
+                    RegCloseKey(hKey);
+                    return FALSE;
+            }
+ 
         case PP_ENUMALGS:
         case PP_ENUMALGS_EX:
             if (((pKeyContainer->dwEnumAlgsCtr >= RSAENH_MAX_ENUMALGS-1) ||

dlls/rsaenh/tests/rsaenh.c

@@ -1366,6 +1366,31 @@ void test_schannel_provider()
     CryptReleaseContext(hProv, 0);
 }
 
+void test_enum_container()
+{
+    BYTE abContainerName[256];
+    DWORD dwBufferLen;
+    BOOL result, fFound = FALSE;
+
+    /* If PP_ENUMCONTAINERS is queried with CRYPT_FIRST and abData == NULL, it returns
+     * the maximum legal length of container names (which is MAX_PATH + 1 == 261) */
+    result = CryptGetProvParam(hProv, PP_ENUMCONTAINERS, NULL, &dwBufferLen, CRYPT_FIRST);
+    ok (result && dwBufferLen == MAX_PATH + 1, "%08lx\n", GetLastError());
+
+    /* If the result fits into abContainerName dwBufferLen is left untouched */
+    dwBufferLen = (DWORD)sizeof(abContainerName);
+    result = CryptGetProvParam(hProv, PP_ENUMCONTAINERS, abContainerName, &dwBufferLen, CRYPT_FIRST);
+    ok (result && dwBufferLen == (DWORD)sizeof(abContainerName), "%08lx\n", GetLastError());
+    
+    /* We only check, if the currently open 'winetest' container is among the enumerated. */
+    do {
+        if (!strcmp(abContainerName, "winetest")) fFound = TRUE;
+        dwBufferLen = (DWORD)sizeof(abContainerName);
+    } while (CryptGetProvParam(hProv, PP_ENUMCONTAINERS, abContainerName, &dwBufferLen, 0));
+        
+    ok (fFound && GetLastError() == ERROR_NO_MORE_ITEMS, "%d, %08lx\n", fFound, GetLastError());
+}
+
 START_TEST(rsaenh)
 {
     if (!init_environment()) 
@@ -1384,6 +1409,7 @@ START_TEST(rsaenh)
     test_import_private();
     test_verify_signature();
     test_rsa_encrypt();
+    test_enum_container();
     clean_up_environment();
     test_schannel_provider();
 }