1 /*
2  * Miscellaneous crypt32 tests
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 
21 #include <stdio.h>
22 #include <stdarg.h>
23 #include <windef.h>
24 #include <winbase.h>
25 #include <winerror.h>
26 #include <wincrypt.h>
27 #include <winreg.h>
28 
29 #include "wine/test.h"
30 
31 static HMODULE hCrypt;
32 
33 static void test_findAttribute(void)
34 {
35     PCRYPT_ATTRIBUTE ret;
36     BYTE blobbin[] = {0x02,0x01,0x01};
37     static CHAR oid[] = "1.2.3";
38     CRYPT_ATTR_BLOB blobs[] = { { sizeof blobbin, blobbin }, };
39     CRYPT_ATTRIBUTE attr = { oid, sizeof(blobs) / sizeof(blobs[0]), blobs };
40 
41     /* returns NULL, last error not set */
42     SetLastError(0xdeadbeef);
43     ret = CertFindAttribute(NULL, 0, NULL);
44     ok(ret == NULL, "Expected failure\n");
45     ok(GetLastError() == 0xdeadbeef, "Last error was set to %08x\n",
46      GetLastError());
47     if (0)
48     {
49         /* crashes */
50         CertFindAttribute(NULL, 1, NULL);
51         /* returns NULL, last error is ERROR_INVALID_PARAMETER
52          * crashes on Vista
53          */
54         SetLastError(0xdeadbeef);
55         ret = CertFindAttribute(NULL, 1, &attr);
56         ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
57          "Expected ERROR_INVALID_PARAMETER, got %d (%08x)\n", GetLastError(),
58          GetLastError());
59     }
60     /* returns NULL, last error not set */
61     SetLastError(0xdeadbeef);
62     ret = CertFindAttribute("bogus", 1, &attr);
63     ok(ret == NULL, "Expected failure\n");
64     ok(GetLastError() == 0xdeadbeef, "Last error was set to %08x\n",
65      GetLastError());
66     /* returns NULL, last error not set */
67     SetLastError(0xdeadbeef);
68     ret = CertFindAttribute("1.2.4", 1, &attr);
69     ok(ret == NULL, "Expected failure\n");
70     ok(GetLastError() == 0xdeadbeef, "Last error was set to %08x\n",
71      GetLastError());
72     /* succeeds, last error not set */
73     SetLastError(0xdeadbeef);
74     ret = CertFindAttribute("1.2.3", 1, &attr);
75     ok(ret != NULL, "CertFindAttribute failed: %08x\n", GetLastError());
76 }
77 
78 static void test_findExtension(void)
79 {
80     PCERT_EXTENSION ret;
81     static CHAR oid[] = "1.2.3";
82     BYTE blobbin[] = {0x02,0x01,0x01};
83     CERT_EXTENSION ext = { oid, TRUE, { sizeof blobbin, blobbin } };
84 
85     /* returns NULL, last error not set */
86     SetLastError(0xdeadbeef);
87     ret = CertFindExtension(NULL, 0, NULL);
88     ok(ret == NULL, "Expected failure\n");
89     ok(GetLastError() == 0xdeadbeef, "Last error was set to %08x\n",
90      GetLastError());
91     if (0)
92     {
93         /* crashes */
94         SetLastError(0xdeadbeef);
95         CertFindExtension(NULL, 1, NULL);
96         /* returns NULL, last error is ERROR_INVALID_PARAMETER
97          * crashes on Vista
98          */
99         SetLastError(0xdeadbeef);
100         ret = CertFindExtension(NULL, 1, &ext);
101         ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
102          "Expected ERROR_INVALID_PARAMETER, got %d (%08x)\n", GetLastError(),
103          GetLastError());
104     }
105     /* returns NULL, last error not set */
106     SetLastError(0xdeadbeef);
107     ret = CertFindExtension("bogus", 1, &ext);
108     ok(ret == NULL, "Expected failure\n");
109     ok(GetLastError() == 0xdeadbeef, "Last error was set to %08x\n",
110      GetLastError());
111     /* returns NULL, last error not set */
112     SetLastError(0xdeadbeef);
113     ret = CertFindExtension("1.2.4", 1, &ext);
114     ok(ret == NULL, "Expected failure\n");
115     ok(GetLastError() == 0xdeadbeef, "Last error was set to %08x\n",
116      GetLastError());
117     /* succeeds, last error not set */
118     SetLastError(0xdeadbeef);
119     ret = CertFindExtension("1.2.3", 1, &ext);
120     ok(ret != NULL, "CertFindExtension failed: %08x\n", GetLastError());
121 }
122 
123 static void test_findRDNAttr(void)
124 {
125     PCERT_RDN_ATTR ret;
126     static CHAR oid[] = "1.2.3";
127     BYTE bin[] = { 0x16,0x09,'J','u','a','n',' ','L','a','n','g' };
128     CERT_RDN_ATTR attrs[] = {
129      { oid, CERT_RDN_IA5_STRING, { sizeof bin, bin } },
130     };
131     CERT_RDN rdns[] = {
132      { sizeof(attrs) / sizeof(attrs[0]), attrs },
133     };
134     CERT_NAME_INFO nameInfo = { sizeof(rdns) / sizeof(rdns[0]), rdns };
135 
136     if (0)
137     {
138         /* crashes */
139         SetLastError(0xdeadbeef);
140         CertFindRDNAttr(NULL, NULL);
141         /* returns NULL, last error is ERROR_INVALID_PARAMETER
142          * crashes on Vista
143          */
144         SetLastError(0xdeadbeef);
145         ret = CertFindRDNAttr(NULL, &nameInfo);
146         ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
147          "Expected ERROR_INVALID_PARAMETER, got %d (%08x)\n", GetLastError(),
148          GetLastError());
149     }
150     /* returns NULL, last error not set */
151     SetLastError(0xdeadbeef);
152     ret = CertFindRDNAttr("bogus", &nameInfo);
153     ok(ret == NULL, "Expected failure\n");
154     ok(GetLastError() == 0xdeadbeef, "Last error was set to %08x\n",
155      GetLastError());
156     /* returns NULL, last error not set */
157     SetLastError(0xdeadbeef);
158     ret = CertFindRDNAttr("1.2.4", &nameInfo);
159     ok(ret == NULL, "Expected failure\n");
160     ok(GetLastError() == 0xdeadbeef, "Last error was set to %08x\n",
161      GetLastError());
162     /* succeeds, last error not set */
163     SetLastError(0xdeadbeef);
164     ret = CertFindRDNAttr("1.2.3", &nameInfo);
165     ok(ret != NULL, "CertFindRDNAttr failed: %08x\n", GetLastError());
166 }
167 
168 static void test_verifyTimeValidity(void)
169 {
170     SYSTEMTIME sysTime;
171     FILETIME fileTime;
172     CERT_INFO info = { 0 };
173     LONG ret;
174 
175     GetSystemTime(&sysTime);
176     SystemTimeToFileTime(&sysTime, &fileTime);
177     /* crashes
178     ret = CertVerifyTimeValidity(NULL, NULL);
179     ret = CertVerifyTimeValidity(&fileTime, NULL);
180      */
181     /* Check with 0 NotBefore and NotAfter */
182     ret = CertVerifyTimeValidity(&fileTime, &info);
183     ok(ret == 1, "Expected 1, got %d\n", ret);
184     info.NotAfter = fileTime;
185     /* Check with NotAfter equal to comparison time */
186     ret = CertVerifyTimeValidity(&fileTime, &info);
187     ok(ret == 0, "Expected 0, got %d\n", ret);
188     /* Check with NotBefore after comparison time */
189     info.NotBefore = fileTime;
190     info.NotBefore.dwLowDateTime += 5000;
191     ret = CertVerifyTimeValidity(&fileTime, &info);
192     ok(ret == -1, "Expected -1, got %d\n", ret);
193 }
194 
195 static void test_cryptAllocate(void)
196 {
197     LPVOID buf;
198 
199     buf = CryptMemAlloc(0);
200     ok(buf != NULL, "CryptMemAlloc failed: %08x\n", GetLastError());
201     CryptMemFree(buf);
202     /* CryptMemRealloc(NULL, 0) fails pre-Vista */
203     buf = CryptMemAlloc(0);
204     buf = CryptMemRealloc(buf, 1);
205     ok(buf != NULL, "CryptMemRealloc failed: %08x\n", GetLastError());
206     CryptMemFree(buf);
207 }
208 
209 
210 static void test_cryptTls(void)
211 {
212     DWORD  (WINAPI *pI_CryptAllocTls)(void);
213     LPVOID (WINAPI *pI_CryptDetachTls)(DWORD dwTlsIndex);
214     LPVOID (WINAPI *pI_CryptGetTls)(DWORD dwTlsIndex);
215     BOOL   (WINAPI *pI_CryptSetTls)(DWORD dwTlsIndex, LPVOID lpTlsValue);
216     BOOL   (WINAPI *pI_CryptFreeTls)(DWORD dwTlsIndex, DWORD unknown);
217     DWORD index;
218     BOOL ret;
219 
220     pI_CryptAllocTls = (void *)GetProcAddress(hCrypt, "I_CryptAllocTls");
221     pI_CryptDetachTls = (void *)GetProcAddress(hCrypt, "I_CryptDetachTls");
222     pI_CryptGetTls = (void *)GetProcAddress(hCrypt, "I_CryptGetTls");
223     pI_CryptSetTls = (void *)GetProcAddress(hCrypt, "I_CryptSetTls");
224     pI_CryptFreeTls = (void *)GetProcAddress(hCrypt, "I_CryptFreeTls");
225 
226     /* One normal pass */
227     index = pI_CryptAllocTls();
228     ok(index, "I_CryptAllocTls failed: %08x\n", GetLastError());
229     if (index)
230     {
231         LPVOID ptr;
232 
233         ptr = pI_CryptGetTls(index);
234         ok(!ptr, "Expected NULL\n");
235         ret = pI_CryptSetTls(index, (LPVOID)0xdeadbeef);
236         ok(ret, "I_CryptSetTls failed: %08x\n", GetLastError());
237         ptr = pI_CryptGetTls(index);
238         ok(ptr == (LPVOID)0xdeadbeef, "Expected 0xdeadbeef, got %p\n", ptr);
239         /* This crashes
240         ret = pI_CryptFreeTls(index, 1);
241          */
242         ret = pI_CryptFreeTls(index, 0);
243         ok(ret, "I_CryptFreeTls failed: %08x\n", GetLastError());
244         ret = pI_CryptFreeTls(index, 0);
245         ok(!ret, "I_CryptFreeTls succeeded\n");
246         ok(GetLastError() == E_INVALIDARG,
247          "Expected E_INVALIDARG, got %08x\n", GetLastError());
248     }
249     /* Similar pass, check I_CryptDetachTls */
250     index = pI_CryptAllocTls();
251     ok(index, "I_CryptAllocTls failed: %08x\n", GetLastError());
252     if (index)
253     {
254         LPVOID ptr;
255 
256         ptr = pI_CryptGetTls(index);
257         ok(!ptr, "Expected NULL\n");
258         ret = pI_CryptSetTls(index, (LPVOID)0xdeadbeef);
259         ok(ret, "I_CryptSetTls failed: %08x\n", GetLastError());
260         ptr = pI_CryptGetTls(index);
261         ok(ptr == (LPVOID)0xdeadbeef, "Expected 0xdeadbeef, got %p\n", ptr);
262         ptr = pI_CryptDetachTls(index);
263         ok(ptr == (LPVOID)0xdeadbeef, "Expected 0xdeadbeef, got %p\n", ptr);
264         ptr = pI_CryptGetTls(index);
265         ok(!ptr, "Expected NULL\n");
266     }
267 }
268 
269 static void test_readTrustedPublisherDWORD(void)
270 {
271 
272     BOOL (WINAPI *pReadDWORD)(LPCWSTR, DWORD *);
273 
274     pReadDWORD = (void *)GetProcAddress(hCrypt, "I_CryptReadTrustedPublisherDWORDValueFromRegistry");
275     if (pReadDWORD)
276     {
277         static const WCHAR safer[] = {
278          'S','o','f','t','w','a','r','e','\\',
279          'P','o','l','i','c','i','e','s','\\',
280          'M','i','c','r','o','s','o','f','t','\\','S','y','s','t','e','m',
281          'C','e','r','t','i','f','i','c','a','t','e','s','\\',
282          'T','r','u','s','t','e','d','P','u','b','l','i','s','h','e','r',
283          '\\','S','a','f','e','r',0 };
284         static const WCHAR authenticodeFlags[] = { 'A','u','t','h','e','n',
285          't','i','c','o','d','e','F','l','a','g','s',0 };
286         BOOL ret, exists = FALSE;
287         DWORD size, readFlags = 0, returnedFlags;
288         HKEY key;
289         LONG rc;
290 
291         rc = RegOpenKeyW(HKEY_LOCAL_MACHINE, safer, &key);
292         if (rc == ERROR_SUCCESS)
293         {
294             size = sizeof(readFlags);
295             rc = RegQueryValueExW(key, authenticodeFlags, NULL, NULL,
296              (LPBYTE)&readFlags, &size);
297             if (rc == ERROR_SUCCESS)
298                 exists = TRUE;
299         }
300         returnedFlags = 0xdeadbeef;
301         ret = pReadDWORD(authenticodeFlags, &returnedFlags);
302         ok(ret == exists, "Unexpected return value\n");
303         ok(readFlags == returnedFlags,
304          "Expected flags %08x, got %08x\n", readFlags, returnedFlags);
305     }
306 }
307 
308 static void test_getDefaultCryptProv(void)
309 {
310     HCRYPTPROV (WINAPI *pI_CryptGetDefaultCryptProv)(DWORD w);
311     HCRYPTPROV prov;
312 
313     pI_CryptGetDefaultCryptProv = (void *)GetProcAddress(hCrypt, "I_CryptGetDefaultCryptProv");
314     if (!pI_CryptGetDefaultCryptProv) return;
315 
316     prov = pI_CryptGetDefaultCryptProv(0xdeadbeef);
317     ok(prov == 0 && GetLastError() == E_INVALIDARG,
318      "Expected E_INVALIDARG, got %08x\n", GetLastError());
319     prov = pI_CryptGetDefaultCryptProv(PROV_RSA_FULL);
320     ok(prov == 0 && GetLastError() == E_INVALIDARG,
321      "Expected E_INVALIDARG, got %08x\n", GetLastError());
322     prov = pI_CryptGetDefaultCryptProv(1);
323     ok(prov == 0 && GetLastError() == E_INVALIDARG,
324      "Expected E_INVALIDARG, got %08x\n", GetLastError());
325     prov = pI_CryptGetDefaultCryptProv(0);
326     ok(prov != 0, "I_CryptGetDefaultCryptProv failed: %08x\n", GetLastError());
327     CryptReleaseContext(prov, 0);
328 }
329 
330 static void test_CryptInstallOssGlobal(void)
331 {
332     int (WINAPI *pI_CryptInstallOssGlobal)(DWORD,DWORD,DWORD);
333     int ret,i;
334 
335     pI_CryptInstallOssGlobal = (void *)GetProcAddress(hCrypt,"I_CryptInstallOssGlobal");
336     /* passing in some random values to I_CryptInstallOssGlobal, it always returns 9 the first time, then 10, 11 etc.*/
337     for(i=0;i<30;i++)
338     {
339       ret =  pI_CryptInstallOssGlobal(rand(),rand(),rand());
340       ok((9+i) == ret ||
341          ret == 0, /* Vista */
342          "Expected %d or 0, got %d\n",(9+i),ret);
343     }
344 }
345 
346 static const BYTE encodedInt[] = { 0x02,0x01,0x01 };
347 static const WCHAR encodedIntStr[] = { '0','2',' ','0','1',' ','0','1',0 };
348 static const BYTE encodedBigInt[] = { 0x02,0x1f,0x01,0x02,0x03,0x04,0x05,0x06,
349  0x07,0x08,0x09,0x0a,0x0b,0x0c,0x0d,0x0e,0x0f,0x10,0x11,0x12,0x13,0x14,0x15,
350  0x16,0x17,0x18,0x19,0x1a,0x1b,0x1c,0x1d,0x1e,0x1f };
351 static const WCHAR encodedBigIntStr[] = { '0','2',' ','1','f',' ','0','1',' ',
352  '0','2',' ','0','3',' ','0','4',' ','0','5',' ','0','6',' ','0','7',' ','0',
353  '8',' ','0','9',' ','0','a',' ','0','b',' ','0','c',' ','0','d',' ','0','e',
354  ' ','0','f',' ','1','0',' ','1','1',' ','1','2',' ','1','3',' ','1','4',' ',
355  '1','5',' ','1','6',' ','1','7',' ','1','8',' ','1','9',' ','1','a',' ','1',
356  'b',' ','1','c',' ','1','d',' ','1','e',' ','1','f',0 };
357 
358 static void test_format_object(void)
359 {
360     BOOL (WINAPI *pCryptFormatObject)(DWORD dwEncoding, DWORD dwFormatType,
361         DWORD dwFormatStrType, void *pFormatStruct, LPCSTR lpszStructType,
362         const BYTE *pbEncoded, DWORD dwEncoded, void *pbFormat,
363         DWORD *pcbFormat);
364     BOOL ret;
365     DWORD size;
366     LPWSTR str;
367 
368     pCryptFormatObject = (void *)GetProcAddress(hCrypt, "CryptFormatObject");
369     if (!pCryptFormatObject)
370     {
371         skip("No CryptFormatObject\n");
372         return;
373     }
374     /* Crash */
375     if (0)
376     {
377         pCryptFormatObject(0, 0, 0, NULL, NULL, NULL, 0, NULL, NULL);
378     }
379     /* When called with any but the default encoding, it fails to find a
380      * formatting function.
381      */
382     SetLastError(0xdeadbeef);
383     ret = pCryptFormatObject(0, 0, 0, NULL, NULL, NULL, 0, NULL, &size);
384     ok(!ret && GetLastError() == ERROR_FILE_NOT_FOUND,
385      "expected ERROR_FILE_NOT_FOUND, got %d\n", GetLastError());
386     /* When called with the default encoding type for any undefined struct type
387      * (including none), it succeeds:  the default encoding is a hex string
388      * encoding.
389      */
390     SetLastError(0xdeadbeef);
391     ret = pCryptFormatObject(X509_ASN_ENCODING, 0, 0, NULL, NULL, NULL, 0,
392      NULL, &size);
393     ok(ret, "CryptFormatObject failed: %d\n", GetLastError());
394     if (ret)
395     {
396         if (size == 0 && GetLastError() == ERROR_FILE_NOT_FOUND)
397         {
398             win_skip("CryptFormatObject has no default implementation\n");
399             return;
400         }
401         ok(size == sizeof(WCHAR), "unexpected size %d\n", size);
402         str = HeapAlloc(GetProcessHeap(), 0, size);
403         SetLastError(0xdeadbeef);
404         size = 0;
405         ret = pCryptFormatObject(X509_ASN_ENCODING, 0, 0, NULL, NULL, NULL, 0,
406          str, &size);
407         ok(!ret && GetLastError() == ERROR_MORE_DATA,
408          "expected ERROR_MORE_DATA, got %d\n", GetLastError());
409         size = sizeof(WCHAR);
410         ret = pCryptFormatObject(X509_ASN_ENCODING, 0, 0, NULL, NULL, NULL, 0,
411          str, &size);
412         ok(ret, "CryptFormatObject failed: %d\n", GetLastError());
413         ok(!str[0], "expected empty string\n");
414         HeapFree(GetProcessHeap(), 0, str);
415     }
416     ret = pCryptFormatObject(X509_ASN_ENCODING, 0, 0, NULL, NULL, encodedInt,
417      sizeof(encodedInt), NULL, &size);
418     ok(ret, "CryptFormatObject failed: %d\n", GetLastError());
419     if (ret)
420     {
421         str = HeapAlloc(GetProcessHeap(), 0, size);
422         ret = pCryptFormatObject(X509_ASN_ENCODING, 0, 0, NULL, NULL,
423          encodedInt, sizeof(encodedInt), str, &size);
424         ok(ret, "CryptFormatObject failed: %d\n", GetLastError());
425         ok(!lstrcmpW(str, encodedIntStr), "unexpected format string\n");
426         HeapFree(GetProcessHeap(), 0, str);
427     }
428     ret = pCryptFormatObject(X509_ASN_ENCODING, 0, 0, NULL, NULL,
429      encodedBigInt, sizeof(encodedBigInt), NULL, &size);
430     ok(ret, "CryptFormatObject failed: %d\n", GetLastError());
431     if (ret)
432     {
433         str = HeapAlloc(GetProcessHeap(), 0, size);
434         ret = pCryptFormatObject(X509_ASN_ENCODING, 0, 0, NULL, NULL,
435          encodedBigInt, sizeof(encodedBigInt), str, &size);
436         ok(ret, "CryptFormatObject failed: %d\n", GetLastError());
437         ok(!lstrcmpiW(str, encodedBigIntStr), "unexpected format string\n");
438         HeapFree(GetProcessHeap(), 0, str);
439     }
440     /* When called with the default encoding type for any undefined struct
441      * type but CRYPT_FORMAT_STR_NO_HEX specified, it fails to find a
442      * formatting function.
443      */
444     SetLastError(0xdeadbeef);
445     ret = pCryptFormatObject(X509_ASN_ENCODING, 0, CRYPT_FORMAT_STR_NO_HEX,
446      NULL, NULL, NULL, 0, NULL, &size);
447     ok(!ret, "CryptFormatObject succeeded\n");
448     ok(GetLastError() == ERROR_FILE_NOT_FOUND ||
449      GetLastError() == 0xdeadbeef, /* Vista, W2K8 */
450      "expected ERROR_FILE_NOT_FOUND or no change, got %d\n", GetLastError());
451     /* When called to format an AUTHORITY_KEY_ID2_INFO, it fails when no
452      * data are given.
453      */
454     SetLastError(0xdeadbeef);
455     ret = pCryptFormatObject(X509_ASN_ENCODING, 0, 0, NULL,
456      szOID_AUTHORITY_KEY_IDENTIFIER2, NULL, 0, NULL, &size);
457     ok(!ret && GetLastError() == E_INVALIDARG,
458      "expected E_INVALIDARG, got %d\n", GetLastError());
459 }
460 
461 START_TEST(main)
462 {
463     hCrypt = GetModuleHandleA("crypt32.dll");
464 
465     test_findAttribute();
466     test_findExtension();
467     test_findRDNAttr();
468     test_verifyTimeValidity();
469     test_cryptAllocate();
470     test_cryptTls();
471     test_readTrustedPublisherDWORD();
472     test_getDefaultCryptProv();
473     test_CryptInstallOssGlobal();
474     test_format_object();
475 }
476