1 /*
2  * Unit test suite for crypt32.dll's OID support functions.
3  *
4  * Copyright 2005 Juan Lang
5  * Copyright 2018 Dmitry Timoshkov
6  *
7  * This library is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Lesser General Public
9  * License as published by the Free Software Foundation; either
10  * version 2.1 of the License, or (at your option) any later version.
11  *
12  * This library is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15  * Lesser General Public License for more details.
16  *
17  * You should have received a copy of the GNU Lesser General Public
18  * License along with this library; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
20  */
21 #include <stdio.h>
22 #include <stdarg.h>
23 #include <windef.h>
24 #include <winbase.h>
25 #include <winerror.h>
26 #define CRYPT_OID_INFO_HAS_EXTRA_FIELDS
27 #include <wincrypt.h>
28 #include <winreg.h>
29 
30 #include "wine/test.h"
31 
32 
33 static BOOL (WINAPI *pCryptEnumOIDInfo)(DWORD,DWORD,void*,PFN_CRYPT_ENUM_OID_INFO);
34 
35 
36 struct OIDToAlgID
37 {
38     LPCSTR oid;
39     LPCSTR altOid;
40     DWORD algID;
41     DWORD altAlgID;
42 };
43 
44 static const struct OIDToAlgID oidToAlgID[] = {
45  { szOID_RSA_RSA, NULL, CALG_RSA_KEYX },
46  { szOID_RSA_MD2RSA, NULL, CALG_MD2 },
47  { szOID_RSA_MD4RSA, NULL, CALG_MD4 },
48  { szOID_RSA_MD5RSA, NULL, CALG_MD5 },
49  { szOID_RSA_SHA1RSA, NULL, CALG_SHA },
50  { szOID_RSA_DH, NULL, CALG_DH_SF },
51  { szOID_RSA_SMIMEalgESDH, NULL, CALG_DH_EPHEM },
52  { szOID_RSA_SMIMEalgCMS3DESwrap, NULL, CALG_3DES },
53  { szOID_RSA_SMIMEalgCMSRC2wrap, NULL, CALG_RC2 },
54  { szOID_RSA_MD2, NULL, CALG_MD2 },
55  { szOID_RSA_MD4, NULL, CALG_MD4 },
56  { szOID_RSA_MD5, NULL, CALG_MD5 },
57  { szOID_RSA_RC2CBC, NULL, CALG_RC2 },
58  { szOID_RSA_RC4, NULL, CALG_RC4 },
59  { szOID_RSA_DES_EDE3_CBC, NULL, CALG_3DES },
60  { szOID_ANSI_X942_DH, NULL, CALG_DH_SF },
61  { szOID_X957_DSA, NULL, CALG_DSS_SIGN },
62  { szOID_X957_SHA1DSA, NULL, CALG_SHA },
63  { szOID_OIWSEC_md4RSA, NULL, CALG_MD4 },
64  { szOID_OIWSEC_md5RSA, NULL, CALG_MD5 },
65  { szOID_OIWSEC_md4RSA2, NULL, CALG_MD4 },
66  { szOID_OIWSEC_desCBC, NULL, CALG_DES },
67  { szOID_OIWSEC_dsa, NULL, CALG_DSS_SIGN },
68  { szOID_OIWSEC_shaDSA, NULL, CALG_SHA },
69  { szOID_OIWSEC_shaRSA, NULL, CALG_SHA },
70  { szOID_OIWSEC_sha, NULL, CALG_SHA },
71  { szOID_OIWSEC_rsaXchg, NULL, CALG_RSA_KEYX },
72  { szOID_OIWSEC_sha1, NULL, CALG_SHA },
73  { szOID_OIWSEC_dsaSHA1, NULL, CALG_SHA },
74  { szOID_OIWSEC_sha1RSASign, NULL, CALG_SHA },
75  { szOID_OIWDIR_md2RSA, NULL, CALG_MD2 },
76  { szOID_INFOSEC_mosaicUpdatedSig, NULL, CALG_SHA },
77  { szOID_INFOSEC_mosaicKMandUpdSig, NULL, CALG_DSS_SIGN },
78  { szOID_NIST_sha256, NULL, CALG_SHA_256, -1 },
79  { szOID_NIST_sha384, NULL, CALG_SHA_384, -1 },
80  { szOID_NIST_sha512, NULL, CALG_SHA_512, -1 }
81 };
82 
83 static const struct OIDToAlgID algIDToOID[] = {
84  { szOID_RSA_RSA, NULL, CALG_RSA_KEYX },
85  { szOID_RSA_SMIMEalgESDH, NULL, CALG_DH_EPHEM },
86  { szOID_RSA_MD2, NULL, CALG_MD2 },
87  { szOID_RSA_MD4, NULL, CALG_MD4 },
88  { szOID_RSA_MD5, NULL, CALG_MD5 },
89  { szOID_RSA_RC2CBC, NULL, CALG_RC2 },
90  { szOID_RSA_RC4, NULL, CALG_RC4 },
91  { szOID_RSA_DES_EDE3_CBC, NULL, CALG_3DES },
92  { szOID_ANSI_X942_DH, NULL, CALG_DH_SF },
93  { szOID_X957_DSA, szOID_OIWSEC_dsa /* some Win98 */, CALG_DSS_SIGN },
94  { szOID_OIWSEC_desCBC, NULL, CALG_DES },
95  { szOID_OIWSEC_sha1, NULL, CALG_SHA },
96 };
97 
98 static const WCHAR bogusDll[] = { 'b','o','g','u','s','.','d','l','l',0 };
99 static const WCHAR bogus2Dll[] = { 'b','o','g','u','s','2','.','d','l','l',0 };
100 
testOIDToAlgID(void)101 static void testOIDToAlgID(void)
102 {
103     int i;
104     DWORD alg;
105 
106     /* Test with a bogus one */
107     alg = CertOIDToAlgId("1.2.3");
108     ok(!alg, "Expected failure, got %d\n", alg);
109 
110     for (i = 0; i < ARRAY_SIZE(oidToAlgID); i++)
111     {
112         alg = CertOIDToAlgId(oidToAlgID[i].oid);
113         ok(alg == oidToAlgID[i].algID || (oidToAlgID[i].altAlgID && alg == oidToAlgID[i].altAlgID),
114          "Expected %d, got %d\n", oidToAlgID[i].algID, alg);
115     }
116 }
117 
testAlgIDToOID(void)118 static void testAlgIDToOID(void)
119 {
120     int i;
121     LPCSTR oid;
122 
123     /* Test with a bogus one */
124     SetLastError(0xdeadbeef);
125     oid = CertAlgIdToOID(ALG_CLASS_SIGNATURE | ALG_TYPE_ANY | 80);
126     ok(!oid && GetLastError() == 0xdeadbeef,
127      "Didn't expect last error (%08x) to be set\n", GetLastError());
128     for (i = 0; i < ARRAY_SIZE(algIDToOID); i++)
129     {
130         oid = CertAlgIdToOID(algIDToOID[i].algID);
131         /* Allow failure, not every version of Windows supports every algo */
132         ok(oid != NULL || broken(!oid), "CertAlgIdToOID failed, expected %s\n", algIDToOID[i].oid);
133         if (oid)
134         {
135             if (strcmp(oid, algIDToOID[i].oid))
136             {
137                 if (algIDToOID[i].altOid)
138                     ok(!strcmp(oid, algIDToOID[i].altOid),
139                      "Expected %s or %s, got %s\n", algIDToOID[i].oid,
140                      algIDToOID[i].altOid, oid);
141                 else
142                 {
143                     /* No need to rerun the test, we already know it failed. */
144                     ok(0, "Expected %s, got %s\n", algIDToOID[i].oid, oid);
145                 }
146             }
147             else
148             {
149                 /* No need to rerun the test, we already know it succeeded. */
150                 ok(1, "Expected %s, got %s\n", algIDToOID[i].oid, oid);
151             }
152         }
153     }
154 }
155 
test_oidFunctionSet(void)156 static void test_oidFunctionSet(void)
157 {
158     HCRYPTOIDFUNCSET set1, set2;
159     BOOL ret;
160     LPWSTR buf = NULL;
161     DWORD size;
162 
163     /* This crashes
164     set = CryptInitOIDFunctionSet(NULL, 0);
165      */
166 
167     /* The name doesn't mean much */
168     set1 = CryptInitOIDFunctionSet("funky", 0);
169     ok(set1 != 0, "CryptInitOIDFunctionSet failed: %08x\n", GetLastError());
170     if (set1)
171     {
172         /* These crash
173         ret = CryptGetDefaultOIDDllList(NULL, 0, NULL, NULL);
174         ret = CryptGetDefaultOIDDllList(NULL, 0, NULL, &size);
175          */
176         size = 0;
177         ret = CryptGetDefaultOIDDllList(set1, 0, NULL, &size);
178         ok(ret, "CryptGetDefaultOIDDllList failed: %08x\n", GetLastError());
179         if (ret)
180         {
181             buf = HeapAlloc(GetProcessHeap(), 0, size * sizeof(WCHAR));
182             if (buf)
183             {
184                 ret = CryptGetDefaultOIDDllList(set1, 0, buf, &size);
185                 ok(ret, "CryptGetDefaultOIDDllList failed: %08x\n",
186                  GetLastError());
187                 ok(!*buf, "Expected empty DLL list\n");
188                 HeapFree(GetProcessHeap(), 0, buf);
189             }
190         }
191     }
192 
193     /* MSDN says flags must be 0, but it's not checked */
194     set1 = CryptInitOIDFunctionSet("", 1);
195     ok(set1 != 0, "CryptInitOIDFunctionSet failed: %08x\n", GetLastError());
196     set2 = CryptInitOIDFunctionSet("", 0);
197     ok(set2 != 0, "CryptInitOIDFunctionSet failed: %08x\n", GetLastError());
198     /* There isn't a free function, so there must be only one set per name to
199      * limit leaks.  (I guess the sets are freed when crypt32 is unloaded.)
200      */
201     ok(set1 == set2, "Expected identical sets\n");
202     if (set1)
203     {
204         /* The empty name function set used here seems to correspond to
205          * DEFAULT.
206          */
207     }
208 
209     /* There's no installed function for a built-in encoding. */
210     set1 = CryptInitOIDFunctionSet("CryptDllEncodeObject", 0);
211     ok(set1 != 0, "CryptInitOIDFunctionSet failed: %08x\n", GetLastError());
212     if (set1)
213     {
214         void *funcAddr;
215         HCRYPTOIDFUNCADDR hFuncAddr;
216 
217         ret = CryptGetOIDFunctionAddress(set1, X509_ASN_ENCODING, X509_CERT, 0,
218          &funcAddr, &hFuncAddr);
219         ok((!ret && GetLastError() == ERROR_FILE_NOT_FOUND) ||
220          broken(ret) /* some Win98 */,
221          "Expected ERROR_FILE_NOT_FOUND, got %08x\n", GetLastError());
222     }
223 }
224 
225 typedef int (*funcY)(int);
226 
funky(int x)227 static int funky(int x)
228 {
229     return x;
230 }
231 
test_installOIDFunctionAddress(void)232 static void test_installOIDFunctionAddress(void)
233 {
234     BOOL ret;
235     CRYPT_OID_FUNC_ENTRY entry = { CRYPT_DEFAULT_OID, funky };
236     HCRYPTOIDFUNCSET set;
237 
238     /* This crashes
239     ret = CryptInstallOIDFunctionAddress(NULL, 0, NULL, 0, NULL, 0);
240      */
241 
242     /* Installing zero functions should work */
243     SetLastError(0xdeadbeef);
244     ret = CryptInstallOIDFunctionAddress(NULL, 0, "CryptDllEncodeObject", 0,
245      NULL, 0);
246     ok(ret && GetLastError() == 0xdeadbeef, "Expected success, got %08x\n",
247      GetLastError());
248 
249     /* The function name doesn't much matter */
250     SetLastError(0xdeadbeef);
251     ret = CryptInstallOIDFunctionAddress(NULL, 0, "OhSoFunky", 0, NULL, 0);
252     ok(ret && GetLastError() == 0xdeadbeef, "Expected success, got %08x\n",
253      GetLastError());
254     SetLastError(0xdeadbeef);
255     entry.pszOID = X509_CERT;
256     ret = CryptInstallOIDFunctionAddress(NULL, 0, "OhSoFunky", 1, &entry, 0);
257     ok(ret && GetLastError() == 0xdeadbeef, "Expected success, got %08x\n",
258      GetLastError());
259     set = CryptInitOIDFunctionSet("OhSoFunky", 0);
260     ok(set != 0, "CryptInitOIDFunctionSet failed: %08x\n", GetLastError());
261     if (set)
262     {
263         funcY funcAddr = NULL;
264         HCRYPTOIDFUNCADDR hFuncAddr = NULL;
265 
266         /* This crashes
267         ret = CryptGetOIDFunctionAddress(set, X509_ASN_ENCODING, 0, 0, NULL,
268          NULL);
269          */
270         ret = CryptGetOIDFunctionAddress(set, X509_ASN_ENCODING, 0, 0,
271          (void **)&funcAddr, &hFuncAddr);
272         ok(!ret && (GetLastError() == ERROR_FILE_NOT_FOUND ||
273          GetLastError() == E_INVALIDARG /* some Win98 */),
274          "Expected ERROR_FILE_NOT_FOUND or E_INVALIDARG, got %d\n",
275          GetLastError());
276         ret = CryptGetOIDFunctionAddress(set, X509_ASN_ENCODING, X509_CERT, 0,
277          (void **)&funcAddr, &hFuncAddr);
278         ok(!ret && GetLastError() == ERROR_FILE_NOT_FOUND,
279          "Expected ERROR_FILE_NOT_FOUND, got %d\n", GetLastError());
280         ret = CryptGetOIDFunctionAddress(set, 0, X509_CERT, 0,
281          (void **)&funcAddr, &hFuncAddr);
282         ok(ret, "CryptGetOIDFunctionAddress failed: %d\n", GetLastError());
283         if (funcAddr)
284         {
285             int y = funcAddr(0xabadc0da);
286 
287             ok(y == 0xabadc0da, "Unexpected return (%d) from function\n", y);
288             CryptFreeOIDFunctionAddress(hFuncAddr, 0);
289         }
290     }
291 }
292 
test_registerOIDFunction(void)293 static void test_registerOIDFunction(void)
294 {
295     BOOL ret;
296 
297     /* oddly, this succeeds under WinXP; the function name key is merely
298      * omitted.  This may be a side effect of the registry code, I don't know.
299      * I don't check it because I doubt anyone would depend on it.
300     ret = CryptRegisterOIDFunction(X509_ASN_ENCODING, NULL,
301      "1.2.3.4.5.6.7.8.9.10", bogusDll, NULL);
302      */
303     /* On windows XP, GetLastError is incorrectly being set with an HRESULT,
304      * E_INVALIDARG
305      */
306     ret = CryptRegisterOIDFunction(X509_ASN_ENCODING, "foo", NULL, bogusDll,
307      NULL);
308     ok(!ret && GetLastError() == E_INVALIDARG,
309      "Expected E_INVALIDARG: %d\n", GetLastError());
310     /* This has no effect, but "succeeds" on XP */
311     ret = CryptRegisterOIDFunction(X509_ASN_ENCODING, "foo",
312      "1.2.3.4.5.6.7.8.9.10", NULL, NULL);
313     ok(ret, "Expected pseudo-success, got %d\n", GetLastError());
314     SetLastError(0xdeadbeef);
315     ret = CryptRegisterOIDFunction(X509_ASN_ENCODING, "CryptDllEncodeObject",
316      "1.2.3.4.5.6.7.8.9.10", bogusDll, NULL);
317     if (!ret && GetLastError() == ERROR_ACCESS_DENIED)
318     {
319         skip("Need admin rights\n");
320         return;
321     }
322     ok(ret, "CryptRegisterOIDFunction failed: %d\n", GetLastError());
323     ret = CryptUnregisterOIDFunction(X509_ASN_ENCODING, "CryptDllEncodeObject",
324      "1.2.3.4.5.6.7.8.9.10");
325     ok(ret, "CryptUnregisterOIDFunction failed: %d\n", GetLastError());
326     ret = CryptRegisterOIDFunction(X509_ASN_ENCODING, "bogus",
327      "1.2.3.4.5.6.7.8.9.10", bogusDll, NULL);
328     ok(ret, "CryptRegisterOIDFunction failed: %d\n", GetLastError());
329     ret = CryptUnregisterOIDFunction(X509_ASN_ENCODING, "bogus",
330      "1.2.3.4.5.6.7.8.9.10");
331     ok(ret, "CryptUnregisterOIDFunction failed: %d\n", GetLastError());
332     /* Unwanted Cryptography\OID\EncodingType 1\bogus\ will still be there */
333     ok(!RegDeleteKeyA(HKEY_LOCAL_MACHINE,
334      "SOFTWARE\\Microsoft\\Cryptography\\OID\\EncodingType 1\\bogus"),
335      "Could not delete bogus key\n");
336     /* Shouldn't have effect but registry keys are created */
337     ret = CryptRegisterOIDFunction(PKCS_7_ASN_ENCODING, "CryptDllEncodeObject",
338      "1.2.3.4.5.6.7.8.9.10", bogusDll, NULL);
339     ok(ret, "CryptRegisterOIDFunction failed: %d\n", GetLastError());
340     ret = CryptUnregisterOIDFunction(PKCS_7_ASN_ENCODING, "CryptDllEncodeObject",
341      "1.2.3.4.5.6.7.8.9.10");
342     ok(ret, "CryptUnregisterOIDFunction failed: %d\n", GetLastError());
343     /* Check with bogus encoding type. Registry keys are still created */
344     ret = CryptRegisterOIDFunction(0, "CryptDllEncodeObject",
345      "1.2.3.4.5.6.7.8.9.10", bogusDll, NULL);
346     ok(ret, "CryptRegisterOIDFunction failed: %d\n", GetLastError());
347     ret = CryptUnregisterOIDFunction(0, "CryptDllEncodeObject",
348      "1.2.3.4.5.6.7.8.9.10");
349     ok(ret, "CryptUnregisterOIDFunction failed: %d\n", GetLastError());
350     /* Unwanted Cryptography\OID\EncodingType 0\CryptDllEncodeObject\
351      * will still be there
352      */
353     ok(!RegDeleteKeyA(HKEY_LOCAL_MACHINE,
354      "SOFTWARE\\Microsoft\\Cryptography\\OID\\EncodingType 0\\CryptDllEncodeObject"),
355      "Could not delete CryptDllEncodeObject key\n");
356     /* This is written with value 3 verbatim.  Thus, the encoding type isn't
357      * (for now) treated as a mask. Registry keys are created.
358      */
359     ret = CryptRegisterOIDFunction(3, "CryptDllEncodeObject",
360      "1.2.3.4.5.6.7.8.9.10", bogusDll, NULL);
361     ok(ret, "CryptRegisterOIDFunction failed: %d\n", GetLastError());
362     ret = CryptUnregisterOIDFunction(3, "CryptDllEncodeObject",
363      "1.2.3.4.5.6.7.8.9.10");
364     ok(ret, "CryptUnregisterOIDFunction failed: %d\n", GetLastError());
365     /* Unwanted Cryptography\OID\EncodingType 3\CryptDllEncodeObject
366      * will still be there.
367      */
368     ok(!RegDeleteKeyA(HKEY_LOCAL_MACHINE,
369      "SOFTWARE\\Microsoft\\Cryptography\\OID\\EncodingType 3\\CryptDllEncodeObject"),
370      "Could not delete CryptDllEncodeObject key\n");
371     ok(!RegDeleteKeyA(HKEY_LOCAL_MACHINE,
372      "SOFTWARE\\Microsoft\\Cryptography\\OID\\EncodingType 3"),
373      "Could not delete 'EncodingType 3' key\n");
374 }
375 
test_registerDefaultOIDFunction(void)376 static void test_registerDefaultOIDFunction(void)
377 {
378     static const char fmt[] =
379      "Software\\Microsoft\\Cryptography\\OID\\EncodingType %d\\%s\\DEFAULT";
380     static const char func[] = "CertDllOpenStoreProv";
381     char buf[MAX_PATH];
382     BOOL ret;
383     LSTATUS rc;
384     HKEY key;
385 
386     ret = CryptRegisterDefaultOIDFunction(0, NULL, 0, NULL);
387     ok(!ret && GetLastError() == E_INVALIDARG,
388      "Expected E_INVALIDARG, got %08x\n", GetLastError());
389     /* This succeeds on WinXP, although the bogus entry is unusable.
390     ret = CryptRegisterDefaultOIDFunction(0, NULL, 0, bogusDll);
391      */
392     /* Register one at index 0 */
393     SetLastError(0xdeadbeef);
394     ret = CryptRegisterDefaultOIDFunction(0, "CertDllOpenStoreProv", 0,
395      bogusDll);
396     if (!ret && GetLastError() == ERROR_ACCESS_DENIED)
397     {
398         skip("Need admin rights\n");
399         return;
400     }
401     ok(ret, "CryptRegisterDefaultOIDFunction failed: %08x\n", GetLastError());
402     /* Reregistering should fail */
403     ret = CryptRegisterDefaultOIDFunction(0, "CertDllOpenStoreProv", 0,
404      bogusDll);
405     ok(!ret && GetLastError() == ERROR_FILE_EXISTS,
406      "Expected ERROR_FILE_EXISTS, got %08x\n", GetLastError());
407     /* Registering the same one at index 1 should also fail */
408     ret = CryptRegisterDefaultOIDFunction(0, "CertDllOpenStoreProv", 1,
409      bogusDll);
410     ok(!ret && GetLastError() == ERROR_FILE_EXISTS,
411      "Expected ERROR_FILE_EXISTS, got %08x\n", GetLastError());
412     /* Registering a different one at index 1 succeeds */
413     ret = CryptRegisterDefaultOIDFunction(0, "CertDllOpenStoreProv", 1,
414      bogus2Dll);
415     ok(ret, "CryptRegisterDefaultOIDFunction failed: %08x\n", GetLastError());
416     sprintf(buf, fmt, 0, func);
417     rc = RegOpenKeyA(HKEY_LOCAL_MACHINE, buf, &key);
418     ok(rc == 0, "Expected key to exist, RegOpenKeyA failed: %d\n", rc);
419     if (rc == 0)
420     {
421         static const CHAR dllA[] = "Dll";
422         static const CHAR bogusDll_A[] = "bogus.dll";
423         static const CHAR bogus2Dll_A[] = "bogus2.dll";
424         CHAR dllBuf[MAX_PATH];
425         DWORD type, size;
426         LPSTR ptr;
427 
428         size = ARRAY_SIZE(dllBuf);
429         rc = RegQueryValueExA(key, dllA, NULL, &type, (LPBYTE)dllBuf, &size);
430         ok(rc == 0,
431          "Expected Dll value to exist, RegQueryValueExA failed: %d\n", rc);
432         ok(type == REG_MULTI_SZ, "Expected type REG_MULTI_SZ, got %d\n", type);
433         /* bogusDll was registered first, so that should be first */
434         ptr = dllBuf;
435         ok(!lstrcmpiA(ptr, bogusDll_A), "Unexpected dll\n");
436         ptr += lstrlenA(ptr) + 1;
437         ok(!lstrcmpiA(ptr, bogus2Dll_A), "Unexpected dll\n");
438         RegCloseKey(key);
439     }
440     /* Unregister both of them */
441     ret = CryptUnregisterDefaultOIDFunction(0, "CertDllOpenStoreProv",
442      bogusDll);
443     ok(ret, "CryptUnregisterDefaultOIDFunction failed: %08x\n",
444      GetLastError());
445     ret = CryptUnregisterDefaultOIDFunction(0, "CertDllOpenStoreProv",
446      bogus2Dll);
447     ok(ret, "CryptUnregisterDefaultOIDFunction failed: %08x\n",
448      GetLastError());
449     /* Now that they're both unregistered, unregistering should fail */
450     ret = CryptUnregisterDefaultOIDFunction(0, "CertDllOpenStoreProv",
451      bogusDll);
452     ok(!ret && GetLastError() == ERROR_FILE_NOT_FOUND,
453      "Expected ERROR_FILE_NOT_FOUND, got %d\n", GetLastError());
454 
455     /* Repeat a few tests on the normal encoding type */
456     ret = CryptRegisterDefaultOIDFunction(X509_ASN_ENCODING,
457      "CertDllOpenStoreProv", 0, bogusDll);
458     ok(ret, "CryptRegisterDefaultOIDFunction failed\n");
459     ret = CryptUnregisterDefaultOIDFunction(X509_ASN_ENCODING,
460      "CertDllOpenStoreProv", bogusDll);
461     ok(ret, "CryptUnregisterDefaultOIDFunction failed\n");
462     ret = CryptUnregisterDefaultOIDFunction(X509_ASN_ENCODING,
463      "CertDllOpenStoreProv", bogusDll);
464     ok(!ret && GetLastError() == ERROR_FILE_NOT_FOUND,
465      "Expected ERROR_FILE_NOT_FOUND, got %08x\n", GetLastError());
466 }
467 
test_getDefaultOIDFunctionAddress(void)468 static void test_getDefaultOIDFunctionAddress(void)
469 {
470     BOOL ret;
471     HCRYPTOIDFUNCSET set;
472     void *funcAddr;
473     HCRYPTOIDFUNCADDR hFuncAddr;
474 
475     /* Crash
476     ret = CryptGetDefaultOIDFunctionAddress(0, 0, NULL, 0, NULL, NULL);
477     ret = CryptGetDefaultOIDFunctionAddress(0, 0, NULL, 0, &funcAddr, NULL);
478     ret = CryptGetDefaultOIDFunctionAddress(0, 0, NULL, 0, NULL, &hFuncAddr);
479     ret = CryptGetDefaultOIDFunctionAddress(0, 0, NULL, 0, &funcAddr,
480      &hFuncAddr);
481      */
482     set = CryptInitOIDFunctionSet("CertDllOpenStoreProv", 0);
483     ok(set != 0, "CryptInitOIDFunctionSet failed: %d\n", GetLastError());
484     /* This crashes if hFuncAddr is not 0 to begin with */
485     hFuncAddr = 0;
486     ret = CryptGetDefaultOIDFunctionAddress(set, 0, NULL, 0, &funcAddr,
487      &hFuncAddr);
488     ok(!ret && GetLastError() == ERROR_FILE_NOT_FOUND,
489      "Expected ERROR_FILE_NOT_FOUND, got %d\n", GetLastError());
490     /* This fails with the normal encoding too, so built-in functions aren't
491      * returned.
492      */
493     ret = CryptGetDefaultOIDFunctionAddress(set, X509_ASN_ENCODING, NULL, 0,
494      &funcAddr, &hFuncAddr);
495     ok(!ret && GetLastError() == ERROR_FILE_NOT_FOUND,
496      "Expected ERROR_FILE_NOT_FOUND, got %d\n", GetLastError());
497 
498     /* Even with a registered dll, this fails (since the dll doesn't exist) */
499     SetLastError(0xdeadbeef);
500     ret = CryptRegisterDefaultOIDFunction(0, "CertDllOpenStoreProv", 0,
501      bogusDll);
502     if (!ret && GetLastError() == ERROR_ACCESS_DENIED)
503         skip("Need admin rights\n");
504     else
505         ok(ret, "CryptRegisterDefaultOIDFunction failed: %08x\n", GetLastError());
506     ret = CryptGetDefaultOIDFunctionAddress(set, 0, NULL, 0, &funcAddr,
507      &hFuncAddr);
508     ok(!ret && GetLastError() == ERROR_FILE_NOT_FOUND,
509      "Expected ERROR_FILE_NOT_FOUND, got %d\n", GetLastError());
510     CryptUnregisterDefaultOIDFunction(0, "CertDllOpenStoreProv", bogusDll);
511 }
512 
countOidInfo(PCCRYPT_OID_INFO pInfo,void * pvArg)513 static BOOL WINAPI countOidInfo(PCCRYPT_OID_INFO pInfo, void *pvArg)
514 {
515     (*(DWORD *)pvArg)++;
516     return TRUE;
517 }
518 
noOidInfo(PCCRYPT_OID_INFO pInfo,void * pvArg)519 static BOOL WINAPI noOidInfo(PCCRYPT_OID_INFO pInfo, void *pvArg)
520 {
521     return FALSE;
522 }
523 
test_enumOIDInfo(void)524 static void test_enumOIDInfo(void)
525 {
526     BOOL ret;
527     DWORD count = 0;
528 
529     if (!pCryptEnumOIDInfo)
530     {
531         win_skip("CryptEnumOIDInfo() is not available\n");
532         return;
533     }
534 
535     /* This crashes
536     ret = pCryptEnumOIDInfo(7, 0, NULL, NULL);
537      */
538 
539     /* Silly tests, check that more than one thing is enumerated */
540     ret = pCryptEnumOIDInfo(0, 0, &count, countOidInfo);
541     ok(ret && count > 0, "Expected more than item enumerated\n");
542     ret = pCryptEnumOIDInfo(0, 0, NULL, noOidInfo);
543     ok(!ret, "Expected FALSE\n");
544 }
545 
test_findOIDInfo(void)546 static void test_findOIDInfo(void)
547 {
548     static WCHAR sha256ECDSA[] = { 's','h','a','2','5','6','E','C','D','S','A',0 };
549     static WCHAR sha1[] = { 's','h','a','1',0 };
550     static CHAR oid_rsa_md5[] = szOID_RSA_MD5, oid_sha256[] = szOID_NIST_sha256;
551     static CHAR oid_ecdsa_sha256[] = szOID_ECDSA_SHA256;
552     ALG_ID alg = CALG_SHA1;
553     ALG_ID algs[2] = { CALG_MD5, CALG_RSA_SIGN };
554     const struct oid_info
555     {
556         DWORD key_type;
557         void *key;
558         const char *oid;
559         ALG_ID algid;
560         ALG_ID broken_algid;
561     } oid_test_info [] =
562     {
563         { CRYPT_OID_INFO_OID_KEY, oid_rsa_md5, szOID_RSA_MD5, CALG_MD5 },
564         { CRYPT_OID_INFO_NAME_KEY, sha1, szOID_OIWSEC_sha1, CALG_SHA1 },
565         { CRYPT_OID_INFO_ALGID_KEY, &alg, szOID_OIWSEC_sha1, CALG_SHA1 },
566         { CRYPT_OID_INFO_SIGN_KEY, algs, szOID_RSA_MD5RSA, CALG_MD5 },
567         { CRYPT_OID_INFO_OID_KEY, oid_sha256, szOID_NIST_sha256, CALG_SHA_256, -1 },
568     };
569     PCCRYPT_OID_INFO info;
570     int i;
571 
572     info = CryptFindOIDInfo(0, NULL, 0);
573     ok(info == NULL, "Expected NULL\n");
574 
575     for (i = 0; i < ARRAY_SIZE(oid_test_info); i++)
576     {
577         const struct oid_info *test = &oid_test_info[i];
578 
579         info = CryptFindOIDInfo(test->key_type, test->key, 0);
580         ok(info != NULL, "Failed to find %s.\n", test->oid);
581         if (info)
582         {
583             ok(!strcmp(info->pszOID, test->oid), "Unexpected OID %s, expected %s\n", info->pszOID, test->oid);
584             ok(U(*info).Algid == test->algid || broken(U(*info).Algid == test->broken_algid),
585                 "Unexpected Algid %d, expected %d\n", U(*info).Algid, test->algid);
586         }
587     }
588 
589     info = CryptFindOIDInfo(CRYPT_OID_INFO_OID_KEY, oid_ecdsa_sha256, 0);
590     if (info)
591     {
592         DWORD *data;
593 
594         ok(info->cbSize == sizeof(*info), "Unexpected structure size %d.\n", info->cbSize);
595         ok(!strcmp(info->pszOID, oid_ecdsa_sha256), "Expected %s, got %s\n", oid_ecdsa_sha256, info->pszOID);
596         ok(!lstrcmpW(info->pwszName, sha256ECDSA), "Expected %s, got %s\n",
597             wine_dbgstr_w(sha256ECDSA), wine_dbgstr_w(info->pwszName));
598         ok(info->dwGroupId == CRYPT_SIGN_ALG_OID_GROUP_ID,
599            "Expected CRYPT_SIGN_ALG_OID_GROUP_ID, got %u\n", info->dwGroupId);
600         ok(U(*info).Algid == CALG_OID_INFO_CNG_ONLY,
601            "Expected CALG_OID_INFO_CNG_ONLY, got %d\n", U(*info).Algid);
602 
603         data = (DWORD *)info->ExtraInfo.pbData;
604         ok(info->ExtraInfo.cbData == 8, "Expected 8, got %d\n", info->ExtraInfo.cbData);
605         ok(data[0] == CALG_OID_INFO_PARAMETERS, "Expected CALG_OID_INFO_PARAMETERS, got %x\n", data[0]);
606         ok(data[1] == CRYPT_OID_NO_NULL_ALGORITHM_PARA_FLAG,
607             "Expected CRYPT_OID_NO_NULL_ALGORITHM_PARA_FLAG, got %x\n", data[1]);
608 
609         ok(!lstrcmpW(info->pwszCNGAlgid, BCRYPT_SHA256_ALGORITHM), "Expected %s, got %s\n",
610            wine_dbgstr_w(BCRYPT_SHA256_ALGORITHM), wine_dbgstr_w(info->pwszCNGAlgid));
611         ok(!lstrcmpW(info->pwszCNGExtraAlgid, CRYPT_OID_INFO_ECC_PARAMETERS_ALGORITHM), "Expected %s, got %s\n",
612            wine_dbgstr_w(CRYPT_OID_INFO_ECC_PARAMETERS_ALGORITHM), wine_dbgstr_w(info->pwszCNGExtraAlgid));
613     }
614     else
615         win_skip("Host does not support ECDSA_SHA256, skipping test\n");
616 }
617 
test_registerOIDInfo(void)618 static void test_registerOIDInfo(void)
619 {
620     static const WCHAR winetestW[] = { 'w','i','n','e','t','e','s','t',0 };
621     static char test_oid[] = "1.2.3.4.5.6.7.8.9.10";
622     CRYPT_OID_INFO info1;
623     const CRYPT_OID_INFO *info2;
624     HKEY key;
625     DWORD ret, size, type, value;
626     char buf[256];
627 
628     SetLastError(0xdeadbeef);
629     ret = CryptUnregisterOIDInfo(NULL);
630     ok(!ret, "should fail\n");
631     ok(GetLastError() == E_INVALIDARG, "got %#x\n", GetLastError());
632 
633     memset(&info1, 0, sizeof(info1));
634     SetLastError(0xdeadbeef);
635     ret = CryptUnregisterOIDInfo(&info1);
636     ok(!ret, "should fail\n");
637     ok(GetLastError() == E_INVALIDARG, "got %#x\n", GetLastError());
638 
639     info1.cbSize = sizeof(info1);
640     SetLastError(0xdeadbeef);
641     ret = CryptUnregisterOIDInfo(&info1);
642     ok(!ret, "should fail\n");
643     ok(GetLastError() == E_INVALIDARG, "got %#x\n", GetLastError());
644 
645     info1.pszOID = test_oid;
646     SetLastError(0xdeadbeef);
647     ret = CryptUnregisterOIDInfo(&info1);
648     ok(!ret, "should fail\n");
649     ok(GetLastError() == ERROR_FILE_NOT_FOUND, "got %u\n", GetLastError());
650 
651     info2 = CryptFindOIDInfo(CRYPT_OID_INFO_OID_KEY, (void *)test_oid, 0);
652     ok(!info2, "should fail\n");
653 
654     SetLastError(0xdeadbeef);
655     /* While it succeeds, the next call does not write anything to the
656      * registry on Windows because dwGroupId == 0.
657      */
658     ret = CryptRegisterOIDInfo(&info1, 0);
659     ok(ret, "got %u\n", GetLastError());
660 
661     ret = RegOpenKeyA(HKEY_LOCAL_MACHINE, "Software\\Microsoft\\Cryptography\\OID\\EncodingType 0\\CryptDllFindOIDInfo\\1.2.3.4.5.6.7.8.9.10!1", &key);
662     ok(ret == ERROR_FILE_NOT_FOUND, "got %u\n", ret);
663 
664     info2 = CryptFindOIDInfo(CRYPT_OID_INFO_OID_KEY, (void *)test_oid, 0);
665     ok(!info2, "should fail\n");
666 
667     info1.pwszName = winetestW;
668     info1.dwGroupId = CRYPT_HASH_ALG_OID_GROUP_ID;
669     SetLastError(0xdeadbeef);
670     ret = CryptRegisterOIDInfo(&info1, CRYPT_INSTALL_OID_INFO_BEFORE_FLAG);
671     if (!ret && GetLastError() == ERROR_ACCESS_DENIED)
672     {
673         skip("Need admin rights\n");
674         return;
675     }
676     ok(ret, "got %u\n", GetLastError());
677 
678     /* It looks like crypt32 reads the OID info from registry only on load,
679      * and CryptFindOIDInfo will find the registered OID on next run
680      */
681     info2 = CryptFindOIDInfo(CRYPT_OID_INFO_OID_KEY, (void *)test_oid, 0);
682     ok(!info2, "should fail\n");
683 
684     ret = RegCreateKeyA(HKEY_LOCAL_MACHINE, "Software\\Microsoft\\Cryptography\\OID\\EncodingType 0\\CryptDllFindOIDInfo\\1.2.3.4.5.6.7.8.9.10!1", &key);
685     ok(!ret, "got %u\n", ret);
686 
687     memset(buf, 0, sizeof(buf));
688     size = sizeof(buf);
689     ret = RegQueryValueExA(key, "Name", NULL, &type, (BYTE *)buf, &size);
690     ok(!ret, "got %u\n", ret);
691     ok(type == REG_SZ, "got %u\n", type);
692     ok(!strcmp(buf, "winetest"), "got %s\n", buf);
693 
694     value = 0xdeadbeef;
695     size = sizeof(value);
696     ret = RegQueryValueExA(key, "Flags", NULL, &type, (BYTE *)&value, &size);
697     ok(!ret, "got %u\n", ret);
698     ok(type == REG_DWORD, "got %u\n", type);
699     ok(value == 1, "got %u\n", value);
700 
701     RegCloseKey(key);
702 
703     CryptUnregisterOIDInfo(&info1);
704 
705     ret = RegOpenKeyA(HKEY_LOCAL_MACHINE, "Software\\Microsoft\\Cryptography\\OID\\EncodingType 0\\CryptDllFindOIDInfo\\1.2.3.4.5.6.7.8.9.10!1", &key);
706     ok(ret == ERROR_FILE_NOT_FOUND, "got %u\n", ret);
707 }
708 
START_TEST(oid)709 START_TEST(oid)
710 {
711     HMODULE hCrypt32 = GetModuleHandleA("crypt32.dll");
712     pCryptEnumOIDInfo = (void*)GetProcAddress(hCrypt32, "CryptEnumOIDInfo");
713 
714     testOIDToAlgID();
715     testAlgIDToOID();
716     test_enumOIDInfo();
717     test_findOIDInfo();
718     test_registerOIDInfo();
719     test_oidFunctionSet();
720     test_installOIDFunctionAddress();
721     test_registerOIDFunction();
722     test_registerDefaultOIDFunction();
723     test_getDefaultOIDFunctionAddress();
724 }
725