git.oblomov.eu Git - wine/blob - dlls/advapi32/tests/crypt.c

   1 /*
   2  * Unit tests for crypt functions
   3  *
   4  * Copyright (c) 2004 Michael Jung
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  */
  20
  21 #include <stdarg.h>
  22
  23 #include "windef.h"
  24 #include "winbase.h"
  25 #include "wincrypt.h"
  26 #include "winerror.h"
  27 #include "winreg.h"
  28
  29 #include "wine/test.h"
  30
  31 static const char szRsaBaseProv[] = MS_DEF_PROV_A;
  32 static const char szNonExistentProv[] = "Wine Nonexistent Cryptographic Provider v11.2";
  33 static const char szKeySet[] = "wine_test_keyset";
  34 static const char szBadKeySet[] = "wine_test_bad_keyset";
  35 #define NON_DEF_PROV_TYPE 999
  36
  37 static HMODULE hadvapi32;
  38 static BOOL (WINAPI *pCryptAcquireContextA)(HCRYPTPROV*,LPCSTR,LPCSTR,DWORD,DWORD);
  39 static BOOL (WINAPI *pCryptEnumProviderTypesA)(DWORD, DWORD*, DWORD, DWORD*, LPSTR, DWORD*);
  40 static BOOL (WINAPI *pCryptEnumProvidersA)(DWORD, DWORD*, DWORD, DWORD*, LPSTR, DWORD*);
  41 static BOOL (WINAPI *pCryptGetDefaultProviderA)(DWORD, DWORD*, DWORD, LPSTR, DWORD*);
  42 static BOOL (WINAPI *pCryptReleaseContext)(HCRYPTPROV, DWORD);
  43 static BOOL (WINAPI *pCryptSetProviderExA)(LPCSTR, DWORD, DWORD*, DWORD);
  44 static BOOL (WINAPI *pCryptCreateHash)(HCRYPTPROV, ALG_ID, HCRYPTKEY, DWORD, HCRYPTHASH*);
  45 static BOOL (WINAPI *pCryptDestroyHash)(HCRYPTHASH);
  46 static BOOL (WINAPI *pCryptGenRandom)(HCRYPTPROV, DWORD, BYTE*);
  47 static BOOL (WINAPI *pCryptContextAddRef)(HCRYPTPROV, DWORD*, DWORD dwFlags);
  48 static BOOL (WINAPI *pCryptGenKey)(HCRYPTPROV, ALG_ID, DWORD, HCRYPTKEY*);
  49 static BOOL (WINAPI *pCryptDestroyKey)(HCRYPTKEY);
  50 static BOOL (WINAPI *pCryptDecrypt)(HCRYPTKEY, HCRYPTHASH, BOOL, DWORD, BYTE*, DWORD*);
  51 static BOOL (WINAPI *pCryptDeriveKey)(HCRYPTPROV, ALG_ID, HCRYPTHASH, DWORD, HCRYPTKEY*);
  52 static BOOL (WINAPI *pCryptDuplicateHash)(HCRYPTHASH, DWORD*, DWORD, HCRYPTHASH*);
  53 static BOOL (WINAPI *pCryptDuplicateKey)(HCRYPTKEY, DWORD*, DWORD, HCRYPTKEY*);
  54 static BOOL (WINAPI *pCryptEncrypt)(HCRYPTKEY, HCRYPTHASH, BOOL, DWORD, BYTE*, DWORD*, DWORD);
  55 static BOOL (WINAPI *pCryptExportKey)(HCRYPTKEY, HCRYPTKEY, DWORD, DWORD, BYTE*, DWORD*);
  56 static BOOL (WINAPI *pCryptGetHashParam)(HCRYPTHASH, DWORD, BYTE*, DWORD*, DWORD);
  57 static BOOL (WINAPI *pCryptGetKeyParam)(HCRYPTKEY, DWORD, BYTE*, DWORD*, DWORD);
  58 static BOOL (WINAPI *pCryptGetProvParam)(HCRYPTPROV, DWORD, BYTE*, DWORD*, DWORD);
  59 static BOOL (WINAPI *pCryptGetUserKey)(HCRYPTPROV, DWORD, HCRYPTKEY*);
  60 static BOOL (WINAPI *pCryptHashData)(HCRYPTHASH, BYTE*, DWORD, DWORD);
  61 static BOOL (WINAPI *pCryptHashSessionKey)(HCRYPTHASH, HCRYPTKEY, DWORD);
  62 static BOOL (WINAPI *pCryptImportKey)(HCRYPTPROV, BYTE*, DWORD, HCRYPTKEY, DWORD, HCRYPTKEY*);
  63 static BOOL (WINAPI *pCryptSignHashW)(HCRYPTHASH, DWORD, LPCWSTR, DWORD, BYTE*, DWORD*);
  64 static BOOL (WINAPI *pCryptSetHashParam)(HCRYPTKEY, DWORD, BYTE*, DWORD);
  65 static BOOL (WINAPI *pCryptSetKeyParam)(HCRYPTKEY, DWORD, BYTE*, DWORD);
  66 static BOOL (WINAPI *pCryptSetProvParam)(HCRYPTPROV, DWORD, BYTE*, DWORD);
  67 static BOOL (WINAPI *pCryptVerifySignatureW)(HCRYPTHASH, BYTE*, DWORD, HCRYPTKEY, LPCWSTR, DWORD);
  68
  69 static void init_function_pointers(void)
  70 {
  71     hadvapi32 = GetModuleHandleA("advapi32.dll");
  72
  73     pCryptAcquireContextA = (void*)GetProcAddress(hadvapi32, "CryptAcquireContextA");
  74     pCryptEnumProviderTypesA = (void*)GetProcAddress(hadvapi32, "CryptEnumProviderTypesA");
  75     pCryptEnumProvidersA = (void*)GetProcAddress(hadvapi32, "CryptEnumProvidersA");
  76     pCryptGetDefaultProviderA = (void*)GetProcAddress(hadvapi32, "CryptGetDefaultProviderA");
  77     pCryptReleaseContext = (void*)GetProcAddress(hadvapi32, "CryptReleaseContext");
  78     pCryptSetProviderExA = (void*)GetProcAddress(hadvapi32, "CryptSetProviderExA");
  79     pCryptCreateHash = (void*)GetProcAddress(hadvapi32, "CryptCreateHash");
  80     pCryptDestroyHash = (void*)GetProcAddress(hadvapi32, "CryptDestroyHash");
  81     pCryptGenRandom = (void*)GetProcAddress(hadvapi32, "CryptGenRandom");
  82     pCryptContextAddRef = (void*)GetProcAddress(hadvapi32, "CryptContextAddRef");
  83     pCryptGenKey = (void*)GetProcAddress(hadvapi32, "CryptGenKey");
  84     pCryptDestroyKey = (void*)GetProcAddress(hadvapi32, "CryptDestroyKey");
  85     pCryptDecrypt = (void*)GetProcAddress(hadvapi32, "CryptDecrypt");
  86     pCryptDeriveKey = (void*)GetProcAddress(hadvapi32, "CryptDeriveKey");
  87     pCryptDuplicateHash = (void*)GetProcAddress(hadvapi32, "CryptDuplicateHash");
  88     pCryptDuplicateKey = (void*)GetProcAddress(hadvapi32, "CryptDuplicateKey");
  89     pCryptEncrypt = (void*)GetProcAddress(hadvapi32, "CryptEncrypt");
  90     pCryptExportKey = (void*)GetProcAddress(hadvapi32, "CryptExportKey");
  91     pCryptGetHashParam = (void*)GetProcAddress(hadvapi32, "CryptGetHashParam");
  92     pCryptGetKeyParam = (void*)GetProcAddress(hadvapi32, "CryptGetKeyParam");
  93     pCryptGetProvParam = (void*)GetProcAddress(hadvapi32, "CryptGetProvParam");
  94     pCryptGetUserKey = (void*)GetProcAddress(hadvapi32, "CryptGetUserKey");
  95     pCryptHashData = (void*)GetProcAddress(hadvapi32, "CryptHashData");
  96     pCryptHashSessionKey = (void*)GetProcAddress(hadvapi32, "CryptHashSessionKey");
  97     pCryptImportKey = (void*)GetProcAddress(hadvapi32, "CryptImportKey");
  98     pCryptSignHashW = (void*)GetProcAddress(hadvapi32, "CryptSignHashW");
  99     pCryptSetHashParam = (void*)GetProcAddress(hadvapi32, "CryptSetHashParam");
 100     pCryptSetKeyParam = (void*)GetProcAddress(hadvapi32, "CryptSetKeyParam");
 101     pCryptSetProvParam = (void*)GetProcAddress(hadvapi32, "CryptSetProvParam");
 102     pCryptVerifySignatureW = (void*)GetProcAddress(hadvapi32, "CryptVerifySignatureW");
 103 }
 104
 105 static void init_environment(void)
 106 {
 107         HCRYPTPROV hProv;
 108
 109         /* Ensure that container "wine_test_keyset" does exist */
 110         if (!pCryptAcquireContextA(&hProv, szKeySet, szRsaBaseProv, PROV_RSA_FULL, 0))
 111         {
 112                 pCryptAcquireContextA(&hProv, szKeySet, szRsaBaseProv, PROV_RSA_FULL, CRYPT_NEWKEYSET);
 113         }
 114         pCryptReleaseContext(hProv, 0);
 115
 116         /* Ensure that container "wine_test_keyset" does exist in default PROV_RSA_FULL type provider */
 117         if (!pCryptAcquireContextA(&hProv, szKeySet, NULL, PROV_RSA_FULL, 0))
 118         {
 119                 pCryptAcquireContextA(&hProv, szKeySet, NULL, PROV_RSA_FULL, CRYPT_NEWKEYSET);
 120         }
 121         pCryptReleaseContext(hProv, 0);
 122
 123         /* Ensure that container "wine_test_bad_keyset" does not exist. */
 124         if (pCryptAcquireContextA(&hProv, szBadKeySet, szRsaBaseProv, PROV_RSA_FULL, 0))
 125         {
 126                 pCryptReleaseContext(hProv, 0);
 127                 pCryptAcquireContextA(&hProv, szBadKeySet, szRsaBaseProv, PROV_RSA_FULL, CRYPT_DELETEKEYSET);
 128         }
 129 }
 130
 131 static void clean_up_environment(void)
 132 {
 133         HCRYPTPROV hProv;
 134
 135         /* Remove container "wine_test_keyset" */
 136         if (pCryptAcquireContextA(&hProv, szKeySet, szRsaBaseProv, PROV_RSA_FULL, 0))
 137         {
 138                 pCryptReleaseContext(hProv, 0);
 139                 pCryptAcquireContextA(&hProv, szKeySet, szRsaBaseProv, PROV_RSA_FULL, CRYPT_DELETEKEYSET);
 140         }
 141
 142         /* Remove container "wine_test_keyset" from default PROV_RSA_FULL type provider */
 143         if (pCryptAcquireContextA(&hProv, szKeySet, NULL, PROV_RSA_FULL, 0))
 144         {
 145                 pCryptReleaseContext(hProv, 0);
 146                 pCryptAcquireContextA(&hProv, szKeySet, NULL, PROV_RSA_FULL, CRYPT_DELETEKEYSET);
 147         }
 148 }
 149
 150 static void test_acquire_context(void)
 151 {
 152         BOOL result;
 153         HCRYPTPROV hProv;
 154         DWORD GLE;
 155
 156         /* Provoke all kinds of error conditions (which are easy to provoke).
 157          * The order of the error tests seems to match Windows XP's rsaenh.dll CSP,
 158          * but since this is likely to change between CSP versions, we don't check
 159          * this. Please don't change the order of tests. */
 160         result = pCryptAcquireContextA(&hProv, NULL, NULL, 0, 0);
 161         ok(!result && GetLastError()==NTE_BAD_PROV_TYPE, "%d\n", GetLastError());
 162
 163         result = pCryptAcquireContextA(&hProv, NULL, NULL, 1000, 0);
 164         ok(!result && GetLastError()==NTE_BAD_PROV_TYPE, "%d\n", GetLastError());
 165
 166         result = pCryptAcquireContextA(&hProv, NULL, NULL, NON_DEF_PROV_TYPE, 0);
 167         ok(!result && GetLastError()==NTE_PROV_TYPE_NOT_DEF, "%d\n", GetLastError());
 168
 169         result = pCryptAcquireContextA(&hProv, szKeySet, szNonExistentProv, PROV_RSA_FULL, 0);
 170         ok(!result && GetLastError()==NTE_KEYSET_NOT_DEF, "%d\n", GetLastError());
 171
 172         result = pCryptAcquireContextA(&hProv, szKeySet, szRsaBaseProv, NON_DEF_PROV_TYPE, 0);
 173         ok(!result && GetLastError()==NTE_PROV_TYPE_NO_MATCH, "%d\n", GetLastError());
 174
 175         /* This test fails under Win2k SP4:
 176            result = TRUE, GetLastError() == ERROR_INVALID_PARAMETER
 177         SetLastError(0xdeadbeef);
 178         result = pCryptAcquireContextA(NULL, szKeySet, szRsaBaseProv, PROV_RSA_FULL, 0);
 179         ok(!result && GetLastError()==ERROR_INVALID_PARAMETER, "%d/%d\n", result, GetLastError());
 180         */
 181
 182         /* Last not least, try to really acquire a context. */
 183         hProv = 0;
 184         SetLastError(0xdeadbeef);
 185         result = pCryptAcquireContextA(&hProv, szKeySet, szRsaBaseProv, PROV_RSA_FULL, 0);
 186         GLE = GetLastError();
 187         ok(result && (GLE == ERROR_ENVVAR_NOT_FOUND   ||
 188                       GLE == ERROR_SUCCESS            ||
 189                       GLE == ERROR_RING2_STACK_IN_USE ||
 190                       GLE == NTE_FAIL                 ||
 191                       GLE == ERROR_NOT_LOGGED_ON), "%d/%d\n", result, GLE);
 192
 193         if (hProv)
 194                 pCryptReleaseContext(hProv, 0);
 195
 196         /* Try again, witch an empty ("\0") szProvider parameter */
 197         hProv = 0;
 198         SetLastError(0xdeadbeef);
 199         result = pCryptAcquireContextA(&hProv, szKeySet, "", PROV_RSA_FULL, 0);
 200         GLE = GetLastError();
 201         ok(result && (GLE == ERROR_ENVVAR_NOT_FOUND   ||
 202                       GLE == ERROR_SUCCESS            ||
 203                       GLE == ERROR_RING2_STACK_IN_USE ||
 204                       GLE == NTE_FAIL                 ||
 205                       GLE == ERROR_NOT_LOGGED_ON), "%d/%d\n", result, GetLastError());
 206
 207         if (hProv)
 208                 pCryptReleaseContext(hProv, 0);
 209 }
 210
 211 static void test_incorrect_api_usage(void)
 212 {
 213     BOOL result;
 214     HCRYPTPROV hProv, hProv2;
 215     HCRYPTHASH hHash, hHash2;
 216     HCRYPTKEY hKey, hKey2;
 217     BYTE temp;
 218     DWORD dwLen, dwTemp;
 219
 220     /* This is to document incorrect api usage in the
 221      * "Uru - Ages beyond Myst Demo" installer as reported by Paul Vriens.
 222      *
 223      * The installer destroys a hash object after having released the context
 224      * with which the hash was created. This is not allowed according to MSDN,
 225      * since CryptReleaseContext destroys all hash and key objects belonging to
 226      * the respective context. However, while wine used to crash, Windows is more
 227      * robust here and returns an ERROR_INVALID_PARAMETER code.
 228      */
 229
 230     result = pCryptAcquireContextA(&hProv, szBadKeySet, szRsaBaseProv,
 231                                    PROV_RSA_FULL, CRYPT_NEWKEYSET);
 232     ok (result, "%08x\n", GetLastError());
 233     if (!result) return;
 234
 235     result = pCryptCreateHash(hProv, CALG_SHA, 0, 0, &hHash);
 236     ok (result, "%d\n", GetLastError());
 237     if (!result) return;
 238
 239     result = pCryptGenKey(hProv, CALG_RC4, 0, &hKey);
 240     ok (result, "%d\n", GetLastError());
 241     if (!result) return;
 242
 243     result = pCryptGenKey(hProv, CALG_RC4, 0, &hKey2);
 244     ok (result, "%d\n", GetLastError());
 245     if (!result) return;
 246
 247     result = pCryptDestroyKey(hKey2);
 248     ok (result, "%d\n", GetLastError());
 249
 250     dwTemp = CRYPT_MODE_ECB;
 251     result = pCryptSetKeyParam(hKey2, KP_MODE, (BYTE*)&dwTemp, sizeof(DWORD));
 252     ok (!result && GetLastError() == ERROR_INVALID_PARAMETER, "%d\n", GetLastError());
 253
 254     result = pCryptAcquireContextA(&hProv2, szBadKeySet, NULL, PROV_RSA_FULL,
 255                                    CRYPT_DELETEKEYSET);
 256     ok (result, "%d\n", GetLastError());
 257     if (!result) return;
 258
 259     result = pCryptReleaseContext(hProv, 0);
 260     ok (result, "%d\n", GetLastError());
 261     if (!result) return;
 262
 263     result = pCryptReleaseContext(hProv, 0);
 264     ok (!result && GetLastError() == ERROR_INVALID_PARAMETER, "%d\n", GetLastError());
 265
 266     result = pCryptGenRandom(hProv, 1, &temp);
 267     ok (!result && GetLastError() == ERROR_INVALID_PARAMETER, "%d\n", GetLastError());
 268
 269 #ifdef CRASHES_ON_NT40
 270     result = pCryptContextAddRef(hProv, NULL, 0);
 271     ok (!result && GetLastError() == ERROR_INVALID_PARAMETER, "%d\n", GetLastError());
 272 #endif
 273
 274     result = pCryptCreateHash(hProv, CALG_SHA, 0, 0, &hHash2);
 275     ok (!result && GetLastError() == ERROR_INVALID_PARAMETER, "%d\n", GetLastError());
 276
 277     dwLen = 1;
 278     result = pCryptDecrypt(hKey, (HCRYPTHASH)NULL, TRUE, 0, &temp, &dwLen);
 279     ok (!result && GetLastError() == ERROR_INVALID_PARAMETER, "%d\n", GetLastError());
 280
 281     dwLen = 1;
 282     result = pCryptEncrypt(hKey, (HCRYPTHASH)NULL, TRUE, 0, &temp, &dwLen, 1);
 283     ok (!result && GetLastError() == ERROR_INVALID_PARAMETER, "%d\n", GetLastError());
 284
 285     result = pCryptDeriveKey(hProv, CALG_RC4, hHash, 0, &hKey2);
 286     ok (!result && GetLastError() == ERROR_INVALID_PARAMETER, "%d\n", GetLastError());
 287
 288 #ifdef CRASHES_ON_NT40
 289     result = pCryptDuplicateHash(hHash, NULL, 0, &hHash2);
 290     ok (!result && GetLastError() == ERROR_INVALID_PARAMETER, "%d\n", GetLastError());
 291
 292     result = pCryptDuplicateKey(hKey, NULL, 0, &hKey2);
 293     ok (!result && GetLastError() == ERROR_INVALID_PARAMETER, "%d\n", GetLastError());
 294 #endif
 295
 296     dwLen = 1;
 297     result = pCryptExportKey(hKey, (HCRYPTPROV)NULL, 0, 0, &temp, &dwLen);
 298     ok (!result && GetLastError() == ERROR_INVALID_PARAMETER, "%d\n", GetLastError());
 299
 300     result = pCryptGenKey(hProv, CALG_RC4, 0, &hKey2);
 301     ok (!result && GetLastError() == ERROR_INVALID_PARAMETER, "%d\n", GetLastError());
 302
 303     dwLen = 1;
 304     result = pCryptGetHashParam(hHash, 0, &temp, &dwLen, 0);
 305     ok (!result && GetLastError() == ERROR_INVALID_PARAMETER, "%d\n", GetLastError());
 306
 307     dwLen = 1;
 308     result = pCryptGetKeyParam(hKey, 0, &temp, &dwLen, 0);
 309     ok (!result && GetLastError() == ERROR_INVALID_PARAMETER, "%d\n", GetLastError());
 310
 311     dwLen = 1;
 312     result = pCryptGetProvParam(hProv, 0, &temp, &dwLen, 0);
 313     ok (!result && GetLastError() == ERROR_INVALID_PARAMETER, "%d\n", GetLastError());
 314
 315     result = pCryptGetUserKey(hProv, 0, &hKey2);
 316     ok (!result && GetLastError() == ERROR_INVALID_PARAMETER, "%d\n", GetLastError());
 317
 318     result = pCryptHashData(hHash, &temp, 1, 0);
 319     ok (!result && GetLastError() == ERROR_INVALID_PARAMETER, "%d\n", GetLastError());
 320
 321     result = pCryptHashSessionKey(hHash, hKey, 0);
 322     ok (!result && GetLastError() == ERROR_INVALID_PARAMETER, "%d\n", GetLastError());
 323
 324     result = pCryptImportKey(hProv, &temp, 1, (HCRYPTKEY)NULL, 0, &hKey2);
 325     ok (!result && GetLastError() == ERROR_INVALID_PARAMETER, "%d\n", GetLastError());
 326
 327     dwLen = 1;
 328     result = pCryptSignHashW(hHash, 0, NULL, 0, &temp, &dwLen);
 329     ok (!result && (GetLastError() == ERROR_INVALID_PARAMETER ||
 330         GetLastError() == ERROR_CALL_NOT_IMPLEMENTED), "%d\n", GetLastError());
 331
 332     result = pCryptSetKeyParam(hKey, 0, &temp, 1);
 333     ok (!result && GetLastError() == ERROR_INVALID_PARAMETER, "%d\n", GetLastError());
 334
 335     result = pCryptSetHashParam(hHash, 0, &temp, 1);
 336     ok (!result && GetLastError() == ERROR_INVALID_PARAMETER, "%d\n", GetLastError());
 337
 338     result = pCryptSetProvParam(hProv, 0, &temp, 1);
 339     ok (!result && GetLastError() == ERROR_INVALID_PARAMETER, "%d\n", GetLastError());
 340
 341     result = pCryptVerifySignatureW(hHash, &temp, 1, hKey, NULL, 0);
 342     ok (!result && (GetLastError() == ERROR_INVALID_PARAMETER ||
 343         GetLastError() == ERROR_CALL_NOT_IMPLEMENTED), "%d\n", GetLastError());
 344
 345     result = pCryptDestroyHash(hHash);
 346     ok (!result && GetLastError() == ERROR_INVALID_PARAMETER, "%d\n", GetLastError());
 347
 348     result = pCryptDestroyKey(hKey);
 349     ok (!result && GetLastError() == ERROR_INVALID_PARAMETER, "%d\n", GetLastError());
 350 }
 351
 352 static BOOL FindProvRegVals(DWORD dwIndex, DWORD *pdwProvType, LPSTR *pszProvName,
 353                             DWORD *pcbProvName, DWORD *pdwProvCount)
 354 {
 355         HKEY hKey;
 356         HKEY subkey;
 357         DWORD size = sizeof(DWORD);
 358
 359         if (RegOpenKey(HKEY_LOCAL_MACHINE, "Software\\Microsoft\\Cryptography\\Defaults\\Provider", &hKey))
 360                 return FALSE;
 361
 362         RegQueryInfoKey(hKey, NULL, NULL, NULL, pdwProvCount, pcbProvName,
 363                                  NULL, NULL, NULL, NULL, NULL, NULL);
 364         (*pcbProvName)++;
 365
 366         if (!(*pszProvName = ((LPSTR)LocalAlloc(LMEM_ZEROINIT, *pcbProvName))))
 367                 return FALSE;
 368
 369         RegEnumKeyEx(hKey, dwIndex, *pszProvName, pcbProvName, NULL, NULL, NULL, NULL);
 370         (*pcbProvName)++;
 371
 372         RegOpenKey(hKey, *pszProvName, &subkey);
 373         RegQueryValueEx(subkey, "Type", NULL, NULL, (LPBYTE)pdwProvType, &size);
 374
 375         RegCloseKey(subkey);
 376         RegCloseKey(hKey);
 377
 378         return TRUE;
 379 }
 380
 381 static void test_enum_providers(void)
 382 {
 383         /* expected results */
 384         CHAR *pszProvName = NULL;
 385         DWORD cbName;
 386         DWORD dwType;
 387         DWORD provCount;
 388         DWORD dwIndex = 0;
 389
 390         /* actual results */
 391         CHAR *provider = NULL;
 392         DWORD providerLen;
 393         DWORD type;
 394         DWORD count;
 395         DWORD result;
 396         DWORD notNull = 5;
 397         DWORD notZeroFlags = 5;
 398
 399         if(!pCryptEnumProvidersA)
 400         {
 401             skip("CryptEnumProvidersA is not available\n");
 402             return;
 403         }
 404
 405         if (!FindProvRegVals(dwIndex, &dwType, &pszProvName, &cbName, &provCount))
 406         {
 407             skip("Could not find providers in registry\n");
 408             return;
 409         }
 410
 411         /* check pdwReserved flag for NULL */
 412         result = pCryptEnumProvidersA(dwIndex, &notNull, 0, &type, NULL, &providerLen);
 413         ok(!result && GetLastError()==ERROR_INVALID_PARAMETER, "%d\n", GetLastError());
 414
 415         /* check dwFlags == 0 */
 416         result = pCryptEnumProvidersA(dwIndex, NULL, notZeroFlags, &type, NULL, &providerLen);
 417         ok(!result && GetLastError()==NTE_BAD_FLAGS, "%d\n", GetLastError());
 418
 419         /* alloc provider to half the size required
 420          * cbName holds the size required */
 421         providerLen = cbName / 2;
 422         if (!(provider = ((LPSTR)LocalAlloc(LMEM_ZEROINIT, providerLen))))
 423                 return;
 424
 425         result = pCryptEnumProvidersA(dwIndex, NULL, 0, &type, provider, &providerLen);
 426         ok(!result && GetLastError()==ERROR_MORE_DATA, "expected %i, got %d\n",
 427                 ERROR_MORE_DATA, GetLastError());
 428
 429         LocalFree(provider);
 430
 431         /* loop through the providers to get the number of providers
 432          * after loop ends, count should be provCount + 1 so subtract 1
 433          * to get actual number of providers */
 434         count = 0;
 435         while(pCryptEnumProvidersA(count++, NULL, 0, &type, NULL, &providerLen))
 436                 ;
 437         count--;
 438         ok(count==provCount, "expected %i, got %i\n", (int)provCount, (int)count);
 439
 440         /* loop past the actual number of providers to get the error
 441          * ERROR_NO_MORE_ITEMS */
 442         for (count = 0; count < provCount + 1; count++)
 443                 result = pCryptEnumProvidersA(count, NULL, 0, &type, NULL, &providerLen);
 444         ok(!result && GetLastError()==ERROR_NO_MORE_ITEMS, "expected %i, got %d\n",
 445                         ERROR_NO_MORE_ITEMS, GetLastError());
 446
 447         /* check expected versus actual values returned */
 448         result = pCryptEnumProvidersA(dwIndex, NULL, 0, &type, NULL, &providerLen);
 449         ok(result && providerLen==cbName, "expected %i, got %i\n", (int)cbName, (int)providerLen);
 450         if (!(provider = ((LPSTR)LocalAlloc(LMEM_ZEROINIT, providerLen))))
 451                 return;
 452
 453         providerLen = 0xdeadbeef;
 454         result = pCryptEnumProvidersA(dwIndex, NULL, 0, &type, provider, &providerLen);
 455         ok(result, "expected TRUE, got %d\n", result);
 456         ok(type==dwType, "expected %d, got %d\n", dwType, type);
 457         if (pszProvName)
 458             ok(!strcmp(pszProvName, provider), "expected %s, got %s\n", pszProvName, provider);
 459         ok(cbName==providerLen, "expected %d, got %d\n", cbName, providerLen);
 460
 461         LocalFree(provider);
 462 }
 463
 464 static BOOL FindProvTypesRegVals(DWORD dwIndex, DWORD *pdwProvType, LPSTR *pszTypeName,
 465                                  DWORD *pcbTypeName, DWORD *pdwTypeCount)
 466 {
 467         HKEY hKey;
 468         HKEY hSubKey;
 469         PSTR ch;
 470
 471         if (RegOpenKey(HKEY_LOCAL_MACHINE, "Software\\Microsoft\\Cryptography\\Defaults\\Provider Types", &hKey))
 472                 return FALSE;
 473
 474         if (RegQueryInfoKey(hKey, NULL, NULL, NULL, pdwTypeCount, pcbTypeName, NULL,
 475                         NULL, NULL, NULL, NULL, NULL))
 476             return FALSE;
 477         (*pcbTypeName)++;
 478
 479         if (!(*pszTypeName = ((LPSTR)LocalAlloc(LMEM_ZEROINIT, *pcbTypeName))))
 480                 return FALSE;
 481
 482         if (RegEnumKeyEx(hKey, dwIndex, *pszTypeName, pcbTypeName, NULL, NULL, NULL, NULL))
 483             return FALSE;
 484         (*pcbTypeName)++;
 485         ch = *pszTypeName + strlen(*pszTypeName);
 486         /* Convert "Type 000" to 0, etc/ */
 487         *pdwProvType = *(--ch) - '0';
 488         *pdwProvType += (*(--ch) - '0') * 10;
 489         *pdwProvType += (*(--ch) - '0') * 100;
 490
 491         if (RegOpenKey(hKey, *pszTypeName, &hSubKey))
 492             return FALSE;
 493
 494         if (RegQueryValueEx(hSubKey, "TypeName", NULL, NULL, NULL, pcbTypeName))
 495             return FALSE;
 496
 497         if (!(*pszTypeName = ((LPSTR)LocalAlloc(LMEM_ZEROINIT, *pcbTypeName))))
 498                 return FALSE;
 499
 500         if (RegQueryValueEx(hSubKey, "TypeName", NULL, NULL, (LPBYTE)*pszTypeName, pcbTypeName))
 501             return FALSE;
 502
 503         RegCloseKey(hSubKey);
 504         RegCloseKey(hKey);
 505
 506         return TRUE;
 507 }
 508
 509 static void test_enum_provider_types(void)
 510 {
 511         /* expected values */
 512         DWORD dwProvType;
 513         LPSTR pszTypeName = NULL;
 514         DWORD cbTypeName;
 515         DWORD dwTypeCount;
 516
 517         /* actual values */
 518         DWORD index = 0;
 519         DWORD provType;
 520         LPSTR typeName = NULL;
 521         DWORD typeNameSize;
 522         DWORD typeCount;
 523         DWORD result;
 524         DWORD notNull = 5;
 525         DWORD notZeroFlags = 5;
 526
 527         if(!pCryptEnumProviderTypesA)
 528         {
 529             skip("CryptEnumProviderTypesA is not available\n");
 530             return;
 531         }
 532
 533         if (!FindProvTypesRegVals(index, &dwProvType, &pszTypeName, &cbTypeName, &dwTypeCount))
 534         {
 535             skip("Could not find provider types in registry\n");
 536             return;
 537         }
 538
 539         /* check pdwReserved for NULL */
 540         result = pCryptEnumProviderTypesA(index, &notNull, 0, &provType, typeName, &typeNameSize);
 541         ok(!result && GetLastError()==ERROR_INVALID_PARAMETER, "expected %i, got %d\n",
 542                 ERROR_INVALID_PARAMETER, GetLastError());
 543
 544         /* check dwFlags == zero */
 545         result = pCryptEnumProviderTypesA(index, NULL, notZeroFlags, &provType, typeName, &typeNameSize);
 546         ok(!result && GetLastError()==NTE_BAD_FLAGS, "expected %i, got %d\n",
 547                 ERROR_INVALID_PARAMETER, GetLastError());
 548
 549         /* alloc provider type to half the size required
 550          * cbTypeName holds the size required */
 551         typeNameSize = cbTypeName / 2;
 552         if (!(typeName = ((LPSTR)LocalAlloc(LMEM_ZEROINIT, typeNameSize))))
 553                 return;
 554
 555         /* This test fails under Win2k SP4:
 556            result = TRUE, GetLastError() == 0xdeadbeef
 557         SetLastError(0xdeadbeef);
 558         result = pCryptEnumProviderTypesA(index, NULL, 0, &provType, typeName, &typeNameSize);
 559         ok(!result && GetLastError()==ERROR_MORE_DATA, "expected 0/ERROR_MORE_DATA, got %d/%08lx\n",
 560                 result, GetLastError());
 561         */
 562
 563         LocalFree(typeName);
 564
 565         /* loop through the provider types to get the number of provider types
 566          * after loop ends, count should be dwTypeCount + 1 so subtract 1
 567          * to get actual number of provider types */
 568         typeCount = 0;
 569         while(pCryptEnumProviderTypesA(typeCount++, NULL, 0, &provType, NULL, &typeNameSize))
 570                 ;
 571         typeCount--;
 572         ok(typeCount==dwTypeCount, "expected %d, got %d\n", dwTypeCount, typeCount);
 573
 574         /* loop past the actual number of provider types to get the error
 575          * ERROR_NO_MORE_ITEMS */
 576         for (typeCount = 0; typeCount < dwTypeCount + 1; typeCount++)
 577                 result = pCryptEnumProviderTypesA(typeCount, NULL, 0, &provType, NULL, &typeNameSize);
 578         ok(!result && GetLastError()==ERROR_NO_MORE_ITEMS, "expected %i, got %d\n",
 579                         ERROR_NO_MORE_ITEMS, GetLastError());
 580
 581
 582         /* check expected versus actual values returned */
 583         result = pCryptEnumProviderTypesA(index, NULL, 0, &provType, NULL, &typeNameSize);
 584         ok(result && typeNameSize==cbTypeName, "expected %d, got %d\n", cbTypeName, typeNameSize);
 585         if (!(typeName = ((LPSTR)LocalAlloc(LMEM_ZEROINIT, typeNameSize))))
 586                 return;
 587
 588         typeNameSize = 0xdeadbeef;
 589         result = pCryptEnumProviderTypesA(index, NULL, 0, &provType, typeName, &typeNameSize);
 590         ok(result, "expected TRUE, got %d\n", result);
 591         ok(provType==dwProvType, "expected %d, got %d\n", dwProvType, provType);
 592         if (pszTypeName)
 593             ok(!strcmp(pszTypeName, typeName), "expected %s, got %s\n", pszTypeName, typeName);
 594         ok(typeNameSize==cbTypeName, "expected %d, got %d\n", cbTypeName, typeNameSize);
 595
 596         LocalFree(typeName);
 597 }
 598
 599 static BOOL FindDfltProvRegVals(DWORD dwProvType, DWORD dwFlags, LPSTR *pszProvName, DWORD *pcbProvName)
 600 {
 601         HKEY hKey;
 602         PSTR keyname;
 603         PSTR ptr;
 604         DWORD user = dwFlags & CRYPT_USER_DEFAULT;
 605
 606         LPCSTR machinestr = "Software\\Microsoft\\Cryptography\\Defaults\\Provider Types\\Type XXX";
 607         LPCSTR userstr = "Software\\Microsoft\\Cryptography\\Provider Type XXX";
 608
 609         keyname = LocalAlloc(LMEM_ZEROINIT, (user ? strlen(userstr) : strlen(machinestr)) + 1);
 610         if (keyname)
 611         {
 612                 user ? strcpy(keyname, userstr) : strcpy(keyname, machinestr);
 613                 ptr = keyname + strlen(keyname);
 614                 *(--ptr) = (dwProvType % 10) + '0';
 615                 *(--ptr) = ((dwProvType / 10) % 10) + '0';
 616                 *(--ptr) = (dwProvType / 100) + '0';
 617         } else
 618                 return FALSE;
 619
 620         if (RegOpenKey((dwFlags & CRYPT_USER_DEFAULT) ?  HKEY_CURRENT_USER : HKEY_LOCAL_MACHINE ,keyname, &hKey))
 621         {
 622                 LocalFree(keyname);
 623                 return FALSE;
 624         }
 625         LocalFree(keyname);
 626
 627         if (RegQueryValueEx(hKey, "Name", NULL, NULL, (LPBYTE)*pszProvName, pcbProvName))
 628         {
 629                 if (GetLastError() != ERROR_MORE_DATA)
 630                         SetLastError(NTE_PROV_TYPE_ENTRY_BAD);
 631                 return FALSE;
 632         }
 633
 634         if (!(*pszProvName = LocalAlloc(LMEM_ZEROINIT, *pcbProvName)))
 635                 return FALSE;
 636
 637         if (RegQueryValueEx(hKey, "Name", NULL, NULL, (LPBYTE)*pszProvName, pcbProvName))
 638         {
 639                 if (GetLastError() != ERROR_MORE_DATA)
 640                         SetLastError(NTE_PROV_TYPE_ENTRY_BAD);
 641                 return FALSE;
 642         }
 643
 644         RegCloseKey(hKey);
 645
 646         return TRUE;
 647 }
 648
 649 static void test_get_default_provider(void)
 650 {
 651         /* expected results */
 652         DWORD dwProvType = PROV_RSA_FULL;
 653         DWORD dwFlags = CRYPT_MACHINE_DEFAULT;
 654         LPSTR pszProvName = NULL;
 655         DWORD cbProvName;
 656
 657         /* actual results */
 658         DWORD provType = PROV_RSA_FULL;
 659         DWORD flags = CRYPT_MACHINE_DEFAULT;
 660         LPSTR provName = NULL;
 661         DWORD provNameSize;
 662         DWORD result;
 663         DWORD notNull = 5;
 664
 665         if(!pCryptGetDefaultProviderA)
 666         {
 667             skip("CryptGetDefaultProviderA is not available\n");
 668             return;
 669         }
 670
 671         if(!FindDfltProvRegVals(dwProvType, dwFlags, &pszProvName, &cbProvName))
 672         {
 673             skip("Could not find default provider in registry\n");
 674             return;
 675         }
 676
 677         /* check pdwReserved for NULL */
 678         result = pCryptGetDefaultProviderA(provType, &notNull, flags, provName, &provNameSize);
 679         ok(!result && GetLastError()==ERROR_INVALID_PARAMETER, "expected %i, got %d\n",
 680                 ERROR_INVALID_PARAMETER, GetLastError());
 681
 682         /* check for invalid flag */
 683         flags = 0xdeadbeef;
 684         result = pCryptGetDefaultProviderA(provType, NULL, flags, provName, &provNameSize);
 685         ok(!result && GetLastError()==NTE_BAD_FLAGS, "expected %d, got %d\n",
 686                 NTE_BAD_FLAGS, GetLastError());
 687         flags = CRYPT_MACHINE_DEFAULT;
 688
 689         /* check for invalid prov type */
 690         provType = 0xdeadbeef;
 691         result = pCryptGetDefaultProviderA(provType, NULL, flags, provName, &provNameSize);
 692         ok(!result && (GetLastError() == NTE_BAD_PROV_TYPE ||
 693                        GetLastError() == ERROR_INVALID_PARAMETER),
 694                 "expected NTE_BAD_PROV_TYPE or ERROR_INVALID_PARAMETER, got %d/%d\n",
 695                 result, GetLastError());
 696         provType = PROV_RSA_FULL;
 697
 698         SetLastError(0);
 699
 700         /* alloc provName to half the size required
 701          * cbProvName holds the size required */
 702         provNameSize = cbProvName / 2;
 703         if (!(provName = LocalAlloc(LMEM_ZEROINIT, provNameSize)))
 704                 return;
 705
 706         result = pCryptGetDefaultProviderA(provType, NULL, flags, provName, &provNameSize);
 707         ok(!result && GetLastError()==ERROR_MORE_DATA, "expected %i, got %d\n",
 708                 ERROR_MORE_DATA, GetLastError());
 709
 710         LocalFree(provName);
 711
 712         /* check expected versus actual values returned */
 713         result = pCryptGetDefaultProviderA(provType, NULL, flags, NULL, &provNameSize);
 714         ok(result && provNameSize==cbProvName, "expected %d, got %d\n", cbProvName, provNameSize);
 715         provNameSize = cbProvName;
 716
 717         if (!(provName = LocalAlloc(LMEM_ZEROINIT, provNameSize)))
 718                 return;
 719
 720         provNameSize = 0xdeadbeef;
 721         result = pCryptGetDefaultProviderA(provType, NULL, flags, provName, &provNameSize);
 722         ok(result, "expected TRUE, got %d\n", result);
 723         if(pszProvName)
 724             ok(!strcmp(pszProvName, provName), "expected %s, got %s\n", pszProvName, provName);
 725         ok(provNameSize==cbProvName, "expected %d, got %d\n", cbProvName, provNameSize);
 726
 727         LocalFree(provName);
 728 }
 729
 730 static void test_set_provider_ex(void)
 731 {
 732         DWORD result;
 733         DWORD notNull = 5;
 734
 735         /* results */
 736         LPSTR pszProvName = NULL;
 737         DWORD cbProvName;
 738
 739         if(!pCryptGetDefaultProviderA || !pCryptSetProviderExA)
 740         {
 741             skip("CryptGetDefaultProviderA and/or CryptSetProviderExA are not available\n");
 742             return;
 743         }
 744
 745         /* check pdwReserved for NULL */
 746         result = pCryptSetProviderExA(MS_DEF_PROV, PROV_RSA_FULL, &notNull, CRYPT_MACHINE_DEFAULT);
 747         ok(!result && GetLastError()==ERROR_INVALID_PARAMETER, "expected %i, got %d\n",
 748                 ERROR_INVALID_PARAMETER, GetLastError());
 749
 750         /* remove the default provider and then set it to MS_DEF_PROV/PROV_RSA_FULL */
 751         SetLastError(0xdeadbeef);
 752         result = pCryptSetProviderExA(MS_DEF_PROV, PROV_RSA_FULL, NULL, CRYPT_MACHINE_DEFAULT | CRYPT_DELETE_DEFAULT);
 753         if (!result && (GetLastError() == ERROR_ACCESS_DENIED))
 754         {
 755                 skip("Not enough rights to remove the default provider\n");
 756                 return;
 757         }
 758         ok(result, "%d\n", GetLastError());
 759
 760         result = pCryptSetProviderExA(MS_DEF_PROV, PROV_RSA_FULL, NULL, CRYPT_MACHINE_DEFAULT);
 761         ok(result, "%d\n", GetLastError());
 762
 763         /* call CryptGetDefaultProvider to see if they match */
 764         result = pCryptGetDefaultProviderA(PROV_RSA_FULL, NULL, CRYPT_MACHINE_DEFAULT, NULL, &cbProvName);
 765         if (!(pszProvName = LocalAlloc(LMEM_ZEROINIT, cbProvName)))
 766                 return;
 767
 768         result = pCryptGetDefaultProviderA(PROV_RSA_FULL, NULL, CRYPT_MACHINE_DEFAULT, pszProvName, &cbProvName);
 769         ok(result && !strcmp(MS_DEF_PROV, pszProvName), "expected %s, got %s\n", MS_DEF_PROV, pszProvName);
 770         ok(result && cbProvName==(strlen(MS_DEF_PROV) + 1), "expected %i, got %d\n", (lstrlenA(MS_DEF_PROV) + 1), cbProvName);
 771
 772         LocalFree(pszProvName);
 773 }
 774
 775 START_TEST(crypt)
 776 {
 777         init_function_pointers();
 778         if(pCryptAcquireContextA && pCryptReleaseContext) {
 779         init_environment();
 780         test_acquire_context();
 781         test_incorrect_api_usage();
 782         clean_up_environment();
 783         }
 784
 785         test_enum_providers();
 786         test_enum_provider_types();
 787         test_get_default_provider();
 788         test_set_provider_ex();
 789 }