git.oblomov.eu Git - wine/blob - dlls/crypt32/tests/oid.c

   1 /*
   2  * Unit test suite for crypt32.dll's OID support functions.
   3  *
   4  * Copyright 2005 Juan Lang
   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 #include <stdio.h>
  21 #include <stdarg.h>
  22 #include <windef.h>
  23 #include <winbase.h>
  24 #include <winerror.h>
  25 #include <wincrypt.h>
  26 #include <winreg.h>
  27
  28 #include "wine/test.h"
  29
  30
  31 static BOOL (WINAPI *pCryptEnumOIDInfo)(DWORD,DWORD,void*,PFN_CRYPT_ENUM_OID_INFO);
  32
  33
  34 struct OIDToAlgID
  35 {
  36     LPCSTR oid;
  37     DWORD algID;
  38 };
  39
  40 static const struct OIDToAlgID oidToAlgID[] = {
  41  { szOID_RSA_RSA, CALG_RSA_KEYX },
  42  { szOID_RSA_MD2RSA, CALG_MD2 },
  43  { szOID_RSA_MD4RSA, CALG_MD4 },
  44  { szOID_RSA_MD5RSA, CALG_MD5 },
  45  { szOID_RSA_SHA1RSA, CALG_SHA },
  46  { szOID_RSA_DH, CALG_DH_SF },
  47  { szOID_RSA_SMIMEalgESDH, CALG_DH_EPHEM },
  48  { szOID_RSA_SMIMEalgCMS3DESwrap, CALG_3DES },
  49  { szOID_RSA_SMIMEalgCMSRC2wrap, CALG_RC2 },
  50  { szOID_RSA_MD2, CALG_MD2 },
  51  { szOID_RSA_MD4, CALG_MD4 },
  52  { szOID_RSA_MD5, CALG_MD5 },
  53  { szOID_RSA_RC2CBC, CALG_RC2 },
  54  { szOID_RSA_RC4, CALG_RC4 },
  55  { szOID_RSA_DES_EDE3_CBC, CALG_3DES },
  56  { szOID_ANSI_X942_DH, CALG_DH_SF },
  57  { szOID_X957_DSA, CALG_DSS_SIGN },
  58  { szOID_X957_SHA1DSA, CALG_SHA },
  59  { szOID_OIWSEC_md4RSA, CALG_MD4 },
  60  { szOID_OIWSEC_md5RSA, CALG_MD5 },
  61  { szOID_OIWSEC_md4RSA2, CALG_MD4 },
  62  { szOID_OIWSEC_desCBC, CALG_DES },
  63  { szOID_OIWSEC_dsa, CALG_DSS_SIGN },
  64  { szOID_OIWSEC_shaDSA, CALG_SHA },
  65  { szOID_OIWSEC_shaRSA, CALG_SHA },
  66  { szOID_OIWSEC_sha, CALG_SHA },
  67  { szOID_OIWSEC_rsaXchg, CALG_RSA_KEYX },
  68  { szOID_OIWSEC_sha1, CALG_SHA },
  69  { szOID_OIWSEC_dsaSHA1, CALG_SHA },
  70  { szOID_OIWSEC_sha1RSASign, CALG_SHA },
  71  { szOID_OIWDIR_md2RSA, CALG_MD2 },
  72  { szOID_INFOSEC_mosaicUpdatedSig, CALG_SHA },
  73  { szOID_INFOSEC_mosaicKMandUpdSig, CALG_DSS_SIGN },
  74 };
  75
  76 static const struct OIDToAlgID algIDToOID[] = {
  77  { szOID_RSA_RSA, CALG_RSA_KEYX },
  78  { szOID_RSA_SMIMEalgESDH, CALG_DH_EPHEM },
  79  { szOID_RSA_MD2, CALG_MD2 },
  80  { szOID_RSA_MD4, CALG_MD4 },
  81  { szOID_RSA_MD5, CALG_MD5 },
  82  { szOID_RSA_RC2CBC, CALG_RC2 },
  83  { szOID_RSA_RC4, CALG_RC4 },
  84  { szOID_RSA_DES_EDE3_CBC, CALG_3DES },
  85  { szOID_ANSI_X942_DH, CALG_DH_SF },
  86  { szOID_X957_DSA, CALG_DSS_SIGN },
  87  { szOID_OIWSEC_desCBC, CALG_DES },
  88  { szOID_OIWSEC_sha1, CALG_SHA },
  89 };
  90
  91 static const WCHAR bogusDll[] = { 'b','o','g','u','s','.','d','l','l',0 };
  92 static const WCHAR bogus2Dll[] = { 'b','o','g','u','s','2','.','d','l','l',0 };
  93
  94 static void testOIDToAlgID(void)
  95 {
  96     int i;
  97     DWORD alg;
  98
  99     /* Test with a bogus one */
 100     SetLastError(0xdeadbeef);
 101     alg = CertOIDToAlgId("1.2.3");
 102     ok(!alg, "Expected failure, got %d\n", alg);
 103     ok(GetLastError() == 0xdeadbeef ||
 104        GetLastError() == ERROR_RESOURCE_NAME_NOT_FOUND ||
 105        GetLastError() == ERROR_INVALID_PARAMETER || /* Vista */
 106        GetLastError() == ERROR_SUCCESS || /* win2k */
 107        GetLastError() == ERROR_FILE_INVALID, /* another Vista */
 108        "Expected ERROR_RESOURCE_NAME_NOT_FOUND, ERROR_INVALID_PARAMETER, "
 109        "ERROR_SUCCESS or no error set, got %08x\n", GetLastError());
 110
 111     for (i = 0; i < sizeof(oidToAlgID) / sizeof(oidToAlgID[0]); i++)
 112     {
 113         alg = CertOIDToAlgId(oidToAlgID[i].oid);
 114         /* Not all Windows installations support all these, so make sure it's
 115          * at least not the wrong one.
 116          */
 117         ok(alg == 0 || alg == oidToAlgID[i].algID,
 118          "Expected %d, got %d\n", oidToAlgID[i].algID, alg);
 119     }
 120 }
 121
 122 static void testAlgIDToOID(void)
 123 {
 124     int i;
 125     LPCSTR oid;
 126
 127     /* Test with a bogus one */
 128     SetLastError(0xdeadbeef);
 129     oid = CertAlgIdToOID(ALG_CLASS_SIGNATURE | ALG_TYPE_ANY | 80);
 130     ok(!oid && GetLastError() == 0xdeadbeef,
 131      "Didn't expect last error (%08x) to be set\n", GetLastError());
 132     for (i = 0; i < sizeof(algIDToOID) / sizeof(algIDToOID[0]); i++)
 133     {
 134         oid = CertAlgIdToOID(algIDToOID[i].algID);
 135         /* Allow failure, not every version of Windows supports every algo */
 136         if (oid)
 137             ok(!strcmp(oid, algIDToOID[i].oid),
 138              "Expected %s, got %s\n", algIDToOID[i].oid, oid);
 139     }
 140 }
 141
 142 static void test_oidFunctionSet(void)
 143 {
 144     HCRYPTOIDFUNCSET set1, set2;
 145     BOOL ret;
 146     LPWSTR buf = NULL;
 147     DWORD size;
 148
 149     /* This crashes
 150     set = CryptInitOIDFunctionSet(NULL, 0);
 151      */
 152
 153     /* The name doesn't mean much */
 154     set1 = CryptInitOIDFunctionSet("funky", 0);
 155     ok(set1 != 0, "CryptInitOIDFunctionSet failed: %08x\n", GetLastError());
 156     if (set1)
 157     {
 158         /* These crash
 159         ret = CryptGetDefaultOIDDllList(NULL, 0, NULL, NULL);
 160         ret = CryptGetDefaultOIDDllList(NULL, 0, NULL, &size);
 161          */
 162         size = 0;
 163         ret = CryptGetDefaultOIDDllList(set1, 0, NULL, &size);
 164         ok(ret, "CryptGetDefaultOIDDllList failed: %08x\n", GetLastError());
 165         if (ret)
 166         {
 167             buf = HeapAlloc(GetProcessHeap(), 0, size * sizeof(WCHAR));
 168             if (buf)
 169             {
 170                 ret = CryptGetDefaultOIDDllList(set1, 0, buf, &size);
 171                 ok(ret, "CryptGetDefaultOIDDllList failed: %08x\n",
 172                  GetLastError());
 173                 ok(!*buf, "Expected empty DLL list\n");
 174                 HeapFree(GetProcessHeap(), 0, buf);
 175             }
 176         }
 177     }
 178
 179     /* MSDN says flags must be 0, but it's not checked */
 180     set1 = CryptInitOIDFunctionSet("", 1);
 181     ok(set1 != 0, "CryptInitOIDFunctionSet failed: %08x\n", GetLastError());
 182     set2 = CryptInitOIDFunctionSet("", 0);
 183     ok(set2 != 0, "CryptInitOIDFunctionSet failed: %08x\n", GetLastError());
 184     /* There isn't a free function, so there must be only one set per name to
 185      * limit leaks.  (I guess the sets are freed when crypt32 is unloaded.)
 186      */
 187     ok(set1 == set2, "Expected identical sets\n");
 188     if (set1)
 189     {
 190         /* The empty name function set used here seems to correspond to
 191          * DEFAULT.
 192          */
 193     }
 194
 195     /* There's no installed function for a built-in encoding. */
 196     set1 = CryptInitOIDFunctionSet("CryptDllEncodeObject", 0);
 197     ok(set1 != 0, "CryptInitOIDFunctionSet failed: %08x\n", GetLastError());
 198     if (set1)
 199     {
 200         void *funcAddr;
 201         HCRYPTOIDFUNCADDR hFuncAddr;
 202
 203         ret = CryptGetOIDFunctionAddress(set1, X509_ASN_ENCODING, X509_CERT, 0,
 204          &funcAddr, &hFuncAddr);
 205         ok(!ret && GetLastError() == ERROR_FILE_NOT_FOUND,
 206          "Expected ERROR_FILE_NOT_FOUND, got %08x\n", GetLastError());
 207     }
 208 }
 209
 210 typedef int (*funcY)(int);
 211
 212 static int funky(int x)
 213 {
 214     return x;
 215 }
 216
 217 static void test_installOIDFunctionAddress(void)
 218 {
 219     BOOL ret;
 220     CRYPT_OID_FUNC_ENTRY entry = { CRYPT_DEFAULT_OID, funky };
 221     HCRYPTOIDFUNCSET set;
 222
 223     /* This crashes
 224     ret = CryptInstallOIDFunctionAddress(NULL, 0, NULL, 0, NULL, 0);
 225      */
 226
 227     /* Installing zero functions should work */
 228     SetLastError(0xdeadbeef);
 229     ret = CryptInstallOIDFunctionAddress(NULL, 0, "CryptDllEncodeObject", 0,
 230      NULL, 0);
 231     ok(ret && GetLastError() == 0xdeadbeef, "Expected success, got %08x\n",
 232      GetLastError());
 233
 234     /* The function name doesn't much matter */
 235     SetLastError(0xdeadbeef);
 236     ret = CryptInstallOIDFunctionAddress(NULL, 0, "OhSoFunky", 0, NULL, 0);
 237     ok(ret && GetLastError() == 0xdeadbeef, "Expected success, got %08x\n",
 238      GetLastError());
 239     SetLastError(0xdeadbeef);
 240     entry.pszOID = X509_CERT;
 241     ret = CryptInstallOIDFunctionAddress(NULL, 0, "OhSoFunky", 1, &entry, 0);
 242     ok(ret && GetLastError() == 0xdeadbeef, "Expected success, got %08x\n",
 243      GetLastError());
 244     set = CryptInitOIDFunctionSet("OhSoFunky", 0);
 245     ok(set != 0, "CryptInitOIDFunctionSet failed: %08x\n", GetLastError());
 246     if (set)
 247     {
 248         funcY funcAddr = NULL;
 249         HCRYPTOIDFUNCADDR hFuncAddr = NULL;
 250
 251         /* This crashes
 252         ret = CryptGetOIDFunctionAddress(set, X509_ASN_ENCODING, 0, 0, NULL,
 253          NULL);
 254          */
 255         ret = CryptGetOIDFunctionAddress(set, X509_ASN_ENCODING, 0, 0,
 256          (void **)&funcAddr, &hFuncAddr);
 257         ok(!ret && GetLastError() == ERROR_FILE_NOT_FOUND,
 258          "Expected ERROR_FILE_NOT_FOUND, got %d\n", GetLastError());
 259         ret = CryptGetOIDFunctionAddress(set, X509_ASN_ENCODING, X509_CERT, 0,
 260          (void **)&funcAddr, &hFuncAddr);
 261         ok(!ret && GetLastError() == ERROR_FILE_NOT_FOUND,
 262          "Expected ERROR_FILE_NOT_FOUND, got %d\n", GetLastError());
 263         ret = CryptGetOIDFunctionAddress(set, 0, X509_CERT, 0,
 264          (void **)&funcAddr, &hFuncAddr);
 265         ok(ret, "CryptGetOIDFunctionAddress failed: %d\n", GetLastError());
 266         if (funcAddr)
 267         {
 268             int y = funcAddr(0xabadc0da);
 269
 270             ok(y == 0xabadc0da, "Unexpected return (%d) from function\n", y);
 271             CryptFreeOIDFunctionAddress(hFuncAddr, 0);
 272         }
 273     }
 274 }
 275
 276 static void test_registerOIDFunction(void)
 277 {
 278     BOOL ret;
 279
 280     /* oddly, this succeeds under WinXP; the function name key is merely
 281      * omitted.  This may be a side effect of the registry code, I don't know.
 282      * I don't check it because I doubt anyone would depend on it.
 283     ret = CryptRegisterOIDFunction(X509_ASN_ENCODING, NULL,
 284      "1.2.3.4.5.6.7.8.9.10", bogusDll, NULL);
 285      */
 286     /* On windows XP, GetLastError is incorrectly being set with an HRESULT,
 287      * E_INVALIDARG
 288      */
 289     ret = CryptRegisterOIDFunction(X509_ASN_ENCODING, "foo", NULL, bogusDll,
 290      NULL);
 291     ok(!ret && GetLastError() == E_INVALIDARG,
 292      "Expected E_INVALIDARG: %d\n", GetLastError());
 293     /* This has no effect, but "succeeds" on XP */
 294     ret = CryptRegisterOIDFunction(X509_ASN_ENCODING, "foo",
 295      "1.2.3.4.5.6.7.8.9.10", NULL, NULL);
 296     ok(ret, "Expected pseudo-success, got %d\n", GetLastError());
 297     SetLastError(0xdeadbeef);
 298     ret = CryptRegisterOIDFunction(X509_ASN_ENCODING, "CryptDllEncodeObject",
 299      "1.2.3.4.5.6.7.8.9.10", bogusDll, NULL);
 300     if (!ret && GetLastError() == ERROR_ACCESS_DENIED)
 301     {
 302         skip("Need admin rights\n");
 303         return;
 304     }
 305     ok(ret, "CryptRegisterOIDFunction failed: %d\n", GetLastError());
 306     ret = CryptUnregisterOIDFunction(X509_ASN_ENCODING, "CryptDllEncodeObject",
 307      "1.2.3.4.5.6.7.8.9.10");
 308     ok(ret, "CryptUnregisterOIDFunction failed: %d\n", GetLastError());
 309     ret = CryptRegisterOIDFunction(X509_ASN_ENCODING, "bogus",
 310      "1.2.3.4.5.6.7.8.9.10", bogusDll, NULL);
 311     ok(ret, "CryptRegisterOIDFunction failed: %d\n", GetLastError());
 312     ret = CryptUnregisterOIDFunction(X509_ASN_ENCODING, "bogus",
 313      "1.2.3.4.5.6.7.8.9.10");
 314     ok(ret, "CryptUnregisterOIDFunction failed: %d\n", GetLastError());
 315     /* Unwanted Cryptography\OID\EncodingType 1\bogus\ will still be there */
 316     ok(!RegDeleteKeyA(HKEY_LOCAL_MACHINE,
 317      "SOFTWARE\\Microsoft\\Cryptography\\OID\\EncodingType 1\\bogus"),
 318      "Could not delete bogus key\n");
 319     /* Shouldn't have effect but registry keys are created */
 320     ret = CryptRegisterOIDFunction(PKCS_7_ASN_ENCODING, "CryptDllEncodeObject",
 321      "1.2.3.4.5.6.7.8.9.10", bogusDll, NULL);
 322     ok(ret, "CryptRegisterOIDFunction failed: %d\n", GetLastError());
 323     ret = CryptUnregisterOIDFunction(PKCS_7_ASN_ENCODING, "CryptDllEncodeObject",
 324      "1.2.3.4.5.6.7.8.9.10");
 325     ok(ret, "CryptUnregisterOIDFunction failed: %d\n", GetLastError());
 326     /* Check with bogus encoding type. Registry keys are still created */
 327     ret = CryptRegisterOIDFunction(0, "CryptDllEncodeObject",
 328      "1.2.3.4.5.6.7.8.9.10", bogusDll, NULL);
 329     ok(ret, "CryptRegisterOIDFunction failed: %d\n", GetLastError());
 330     ret = CryptUnregisterOIDFunction(0, "CryptDllEncodeObject",
 331      "1.2.3.4.5.6.7.8.9.10");
 332     ok(ret, "CryptUnregisterOIDFunction failed: %d\n", GetLastError());
 333     /* Unwanted Cryptography\OID\EncodingType 0\CryptDllEncodeObject\
 334      * will still be there
 335      */
 336     ok(!RegDeleteKeyA(HKEY_LOCAL_MACHINE,
 337      "SOFTWARE\\Microsoft\\Cryptography\\OID\\EncodingType 0\\CryptDllEncodeObject"),
 338      "Could not delete CryptDllEncodeObject key\n");
 339     /* This is written with value 3 verbatim.  Thus, the encoding type isn't
 340      * (for now) treated as a mask. Registry keys are created.
 341      */
 342     ret = CryptRegisterOIDFunction(3, "CryptDllEncodeObject",
 343      "1.2.3.4.5.6.7.8.9.10", bogusDll, NULL);
 344     ok(ret, "CryptRegisterOIDFunction failed: %d\n", GetLastError());
 345     ret = CryptUnregisterOIDFunction(3, "CryptDllEncodeObject",
 346      "1.2.3.4.5.6.7.8.9.10");
 347     ok(ret, "CryptUnregisterOIDFunction failed: %d\n", GetLastError());
 348     /* Unwanted Cryptography\OID\EncodingType 3\CryptDllEncodeObject
 349      * will still be there.
 350      */
 351     ok(!RegDeleteKeyA(HKEY_LOCAL_MACHINE,
 352      "SOFTWARE\\Microsoft\\Cryptography\\OID\\EncodingType 3\\CryptDllEncodeObject"),
 353      "Could not delete CryptDllEncodeObject key\n");
 354     ok(!RegDeleteKeyA(HKEY_LOCAL_MACHINE,
 355      "SOFTWARE\\Microsoft\\Cryptography\\OID\\EncodingType 3"),
 356      "Could not delete 'EncodingType 3' key\n");
 357 }
 358
 359 static void test_registerDefaultOIDFunction(void)
 360 {
 361     static const char fmt[] =
 362      "Software\\Microsoft\\Cryptography\\OID\\EncodingType %d\\%s\\DEFAULT";
 363     static const char func[] = "CertDllOpenStoreProv";
 364     char buf[MAX_PATH];
 365     BOOL ret;
 366     long rc;
 367     HKEY key;
 368
 369     ret = CryptRegisterDefaultOIDFunction(0, NULL, 0, NULL);
 370     ok(!ret && GetLastError() == E_INVALIDARG,
 371      "Expected E_INVALIDARG, got %08x\n", GetLastError());
 372     /* This succeeds on WinXP, although the bogus entry is unusable.
 373     ret = CryptRegisterDefaultOIDFunction(0, NULL, 0, bogusDll);
 374      */
 375     /* Register one at index 0 */
 376     SetLastError(0xdeadbeef);
 377     ret = CryptRegisterDefaultOIDFunction(0, "CertDllOpenStoreProv", 0,
 378      bogusDll);
 379     if (!ret && GetLastError() == ERROR_ACCESS_DENIED)
 380     {
 381         skip("Need admin rights\n");
 382         return;
 383     }
 384     ok(ret, "CryptRegisterDefaultOIDFunction failed: %08x\n", GetLastError());
 385     /* Reregistering should fail */
 386     ret = CryptRegisterDefaultOIDFunction(0, "CertDllOpenStoreProv", 0,
 387      bogusDll);
 388     ok(!ret && GetLastError() == ERROR_FILE_EXISTS,
 389      "Expected ERROR_FILE_EXISTS, got %08x\n", GetLastError());
 390     /* Registering the same one at index 1 should also fail */
 391     ret = CryptRegisterDefaultOIDFunction(0, "CertDllOpenStoreProv", 1,
 392      bogusDll);
 393     ok(!ret && GetLastError() == ERROR_FILE_EXISTS,
 394      "Expected ERROR_FILE_EXISTS, got %08x\n", GetLastError());
 395     /* Registering a different one at index 1 succeeds */
 396     ret = CryptRegisterDefaultOIDFunction(0, "CertDllOpenStoreProv", 1,
 397      bogus2Dll);
 398     ok(ret, "CryptRegisterDefaultOIDFunction failed: %08x\n", GetLastError());
 399     sprintf(buf, fmt, 0, func);
 400     rc = RegOpenKeyA(HKEY_LOCAL_MACHINE, buf, &key);
 401     ok(rc == 0, "Expected key to exist, RegOpenKeyA failed: %ld\n", rc);
 402     if (rc == 0)
 403     {
 404         static const CHAR dllA[] = "Dll";
 405         static const CHAR bogusDll_A[] = "bogus.dll";
 406         static const CHAR bogus2Dll_A[] = "bogus2.dll";
 407         CHAR dllBuf[MAX_PATH];
 408         DWORD type, size;
 409         LPSTR ptr;
 410
 411         size = sizeof(dllBuf) / sizeof(dllBuf[0]);
 412         rc = RegQueryValueExA(key, dllA, NULL, &type, (LPBYTE)dllBuf, &size);
 413         ok(rc == 0,
 414          "Expected Dll value to exist, RegQueryValueExA failed: %ld\n", rc);
 415         ok(type == REG_MULTI_SZ, "Expected type REG_MULTI_SZ, got %d\n", type);
 416         /* bogusDll was registered first, so that should be first */
 417         ptr = dllBuf;
 418         ok(!lstrcmpiA(ptr, bogusDll_A), "Unexpected dll\n");
 419         ptr += lstrlenA(ptr) + 1;
 420         ok(!lstrcmpiA(ptr, bogus2Dll_A), "Unexpected dll\n");
 421         RegCloseKey(key);
 422     }
 423     /* Unregister both of them */
 424     ret = CryptUnregisterDefaultOIDFunction(0, "CertDllOpenStoreProv",
 425      bogusDll);
 426     ok(ret, "CryptUnregisterDefaultOIDFunction failed: %08x\n",
 427      GetLastError());
 428     ret = CryptUnregisterDefaultOIDFunction(0, "CertDllOpenStoreProv",
 429      bogus2Dll);
 430     ok(ret, "CryptUnregisterDefaultOIDFunction failed: %08x\n",
 431      GetLastError());
 432     /* Now that they're both unregistered, unregistering should fail */
 433     ret = CryptUnregisterDefaultOIDFunction(0, "CertDllOpenStoreProv",
 434      bogusDll);
 435     ok(!ret && GetLastError() == ERROR_FILE_NOT_FOUND,
 436      "Expected ERROR_FILE_NOT_FOUND, got %d\n", GetLastError());
 437
 438     /* Repeat a few tests on the normal encoding type */
 439     ret = CryptRegisterDefaultOIDFunction(X509_ASN_ENCODING,
 440      "CertDllOpenStoreProv", 0, bogusDll);
 441     ret = CryptUnregisterDefaultOIDFunction(X509_ASN_ENCODING,
 442      "CertDllOpenStoreProv", bogusDll);
 443     ok(ret, "CryptUnregisterDefaultOIDFunction failed\n");
 444     ret = CryptUnregisterDefaultOIDFunction(X509_ASN_ENCODING,
 445      "CertDllOpenStoreProv", bogusDll);
 446     ok(!ret && GetLastError() == ERROR_FILE_NOT_FOUND,
 447      "Expected ERROR_FILE_NOT_FOUND, got %08x\n", GetLastError());
 448 }
 449
 450 static void test_getDefaultOIDFunctionAddress(void)
 451 {
 452     BOOL ret;
 453     HCRYPTOIDFUNCSET set;
 454     void *funcAddr;
 455     HCRYPTOIDFUNCADDR hFuncAddr;
 456
 457     /* Crash
 458     ret = CryptGetDefaultOIDFunctionAddress(0, 0, NULL, 0, NULL, NULL);
 459     ret = CryptGetDefaultOIDFunctionAddress(0, 0, NULL, 0, &funcAddr, NULL);
 460     ret = CryptGetDefaultOIDFunctionAddress(0, 0, NULL, 0, NULL, &hFuncAddr);
 461     ret = CryptGetDefaultOIDFunctionAddress(0, 0, NULL, 0, &funcAddr,
 462      &hFuncAddr);
 463      */
 464     set = CryptInitOIDFunctionSet("CertDllOpenStoreProv", 0);
 465     ok(set != 0, "CryptInitOIDFunctionSet failed: %d\n", GetLastError());
 466     /* This crashes if hFuncAddr is not 0 to begin with */
 467     hFuncAddr = 0;
 468     ret = CryptGetDefaultOIDFunctionAddress(set, 0, NULL, 0, &funcAddr,
 469      &hFuncAddr);
 470     ok(!ret && GetLastError() == ERROR_FILE_NOT_FOUND,
 471      "Expected ERROR_FILE_NOT_FOUND, got %d\n", GetLastError());
 472     /* This fails with the normal encoding too, so built-in functions aren't
 473      * returned.
 474      */
 475     ret = CryptGetDefaultOIDFunctionAddress(set, X509_ASN_ENCODING, NULL, 0,
 476      &funcAddr, &hFuncAddr);
 477     ok(!ret && GetLastError() == ERROR_FILE_NOT_FOUND,
 478      "Expected ERROR_FILE_NOT_FOUND, got %d\n", GetLastError());
 479
 480     /* Even with a registered dll, this fails (since the dll doesn't exist) */
 481     SetLastError(0xdeadbeef);
 482     ret = CryptRegisterDefaultOIDFunction(0, "CertDllOpenStoreProv", 0,
 483      bogusDll);
 484     if (!ret && GetLastError() == ERROR_ACCESS_DENIED)
 485         skip("Need admin rights\n");
 486     else
 487         ok(ret, "CryptRegisterDefaultOIDFunction failed: %08x\n", GetLastError());
 488     ret = CryptGetDefaultOIDFunctionAddress(set, 0, NULL, 0, &funcAddr,
 489      &hFuncAddr);
 490     ok(!ret && GetLastError() == ERROR_FILE_NOT_FOUND,
 491      "Expected ERROR_FILE_NOT_FOUND, got %d\n", GetLastError());
 492     CryptUnregisterDefaultOIDFunction(0, "CertDllOpenStoreProv", bogusDll);
 493 }
 494
 495 static BOOL WINAPI countOidInfo(PCCRYPT_OID_INFO pInfo, void *pvArg)
 496 {
 497     (*(DWORD *)pvArg)++;
 498     return TRUE;
 499 }
 500
 501 static BOOL WINAPI noOidInfo(PCCRYPT_OID_INFO pInfo, void *pvArg)
 502 {
 503     return FALSE;
 504 }
 505
 506 static void test_enumOIDInfo(void)
 507 {
 508     BOOL ret;
 509     DWORD count = 0;
 510
 511     if (!pCryptEnumOIDInfo)
 512     {
 513         skip("CryptEnumOIDInfo() is not available\n");
 514         return;
 515     }
 516
 517     /* This crashes
 518     ret = pCryptEnumOIDInfo(7, 0, NULL, NULL);
 519      */
 520
 521     /* Silly tests, check that more than one thing is enumerated */
 522     ret = pCryptEnumOIDInfo(0, 0, &count, countOidInfo);
 523     ok(ret && count > 0, "Expected more than item enumerated\n");
 524     ret = pCryptEnumOIDInfo(0, 0, NULL, noOidInfo);
 525     ok(!ret, "Expected FALSE\n");
 526 }
 527
 528 static void test_findOIDInfo(void)
 529 {
 530     static WCHAR sha1[] = { 's','h','a','1',0 };
 531     static CHAR oid_rsa_md5[] = szOID_RSA_MD5;
 532     ALG_ID alg = CALG_SHA1;
 533     ALG_ID algs[2] = { CALG_MD5, CALG_RSA_SIGN };
 534     PCCRYPT_OID_INFO info;
 535
 536     info = CryptFindOIDInfo(0, NULL, 0);
 537     ok(info == NULL, "Expected NULL\n");
 538     info = CryptFindOIDInfo(CRYPT_OID_INFO_OID_KEY, oid_rsa_md5, 0);
 539     ok(info != NULL, "Expected to find szOID_RSA_MD5\n");
 540     if (info)
 541     {
 542         ok(!strcmp(info->pszOID, szOID_RSA_MD5), "Expected %s, got %s\n",
 543          szOID_RSA_MD5, info->pszOID);
 544         ok(U(*info).Algid == CALG_MD5, "Expected CALG_MD5, got %d\n",
 545            U(*info).Algid);
 546     }
 547     info = CryptFindOIDInfo(CRYPT_OID_INFO_NAME_KEY, sha1, 0);
 548     ok(info != NULL, "Expected to find sha1\n");
 549     if (info)
 550     {
 551         ok(!strcmp(info->pszOID, szOID_OIWSEC_sha1), "Expected %s, got %s\n",
 552          szOID_OIWSEC_sha1, info->pszOID);
 553         ok(U(*info).Algid == CALG_SHA1, "Expected CALG_SHA1, got %d\n",
 554            U(*info).Algid);
 555     }
 556     info = CryptFindOIDInfo(CRYPT_OID_INFO_ALGID_KEY, &alg, 0);
 557     ok(info != NULL, "Expected to find sha1\n");
 558     if (info)
 559     {
 560         ok(!strcmp(info->pszOID, szOID_OIWSEC_sha1), "Expected %s, got %s\n",
 561          szOID_OIWSEC_sha1, info->pszOID);
 562         ok(U(*info).Algid == CALG_SHA1, "Expected CALG_SHA1, got %d\n",
 563            U(*info).Algid);
 564     }
 565     info = CryptFindOIDInfo(CRYPT_OID_INFO_SIGN_KEY, algs, 0);
 566     ok(info != NULL, "Expected to find md5RSA\n");
 567     if (info)
 568     {
 569         ok(!strcmp(info->pszOID, szOID_RSA_MD5RSA), "Expected %s, got %s\n",
 570          szOID_RSA_MD5RSA, info->pszOID);
 571         ok(U(*info).Algid == CALG_MD5, "Expected CALG_MD5, got %d\n",
 572            U(*info).Algid);
 573     }
 574 }
 575
 576 START_TEST(oid)
 577 {
 578     HMODULE hCrypt32 = GetModuleHandleA("crypt32.dll");
 579     pCryptEnumOIDInfo = (void*)GetProcAddress(hCrypt32, "CryptEnumOIDInfo");
 580
 581     testOIDToAlgID();
 582     testAlgIDToOID();
 583     test_enumOIDInfo();
 584     test_findOIDInfo();
 585     test_oidFunctionSet();
 586     test_installOIDFunctionAddress();
 587     test_registerOIDFunction();
 588     test_registerDefaultOIDFunction();
 589     test_getDefaultOIDFunctionAddress();
 590 }