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 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 SetLastError(0xdeadbeef);
51 ret = CertFindAttribute(NULL, 1, NULL);
52 /* returns NULL, last error is ERROR_INVALID_PARAMETER
53 * crashes on Vista
54 */
55 SetLastError(0xdeadbeef);
56 ret = CertFindAttribute(NULL, 1, &attr);
57 ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
58 "Expected ERROR_INVALID_PARAMETER, got %d (%08x)\n", GetLastError(),
59 GetLastError());
60 }
61 /* returns NULL, last error not set */
62 SetLastError(0xdeadbeef);
63 ret = CertFindAttribute("bogus", 1, &attr);
64 ok(ret == NULL, "Expected failure\n");
65 ok(GetLastError() == 0xdeadbeef, "Last error was set to %08x\n",
66 GetLastError());
67 /* returns NULL, last error not set */
68 SetLastError(0xdeadbeef);
69 ret = CertFindAttribute("1.2.4", 1, &attr);
70 ok(ret == NULL, "Expected failure\n");
71 ok(GetLastError() == 0xdeadbeef, "Last error was set to %08x\n",
72 GetLastError());
73 /* succeeds, last error not set */
74 SetLastError(0xdeadbeef);
75 ret = CertFindAttribute("1.2.3", 1, &attr);
76 ok(ret != NULL, "CertFindAttribute failed: %08x\n", GetLastError());
77 }
78
79 static void test_findExtension(void)
80 {
81 PCERT_EXTENSION ret;
82 static CHAR oid[] = "1.2.3";
83 BYTE blobbin[] = {0x02,0x01,0x01};
84 CERT_EXTENSION ext = { oid, TRUE, { sizeof blobbin, blobbin } };
85
86 /* returns NULL, last error not set */
87 SetLastError(0xdeadbeef);
88 ret = CertFindExtension(NULL, 0, NULL);
89 ok(ret == NULL, "Expected failure\n");
90 ok(GetLastError() == 0xdeadbeef, "Last error was set to %08x\n",
91 GetLastError());
92 if (0)
93 {
94 /* crashes */
95 SetLastError(0xdeadbeef);
96 ret = CertFindExtension(NULL, 1, NULL);
97 /* returns NULL, last error is ERROR_INVALID_PARAMETER
98 * crashes on Vista
99 */
100 SetLastError(0xdeadbeef);
101 ret = CertFindExtension(NULL, 1, &ext);
102 ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
103 "Expected ERROR_INVALID_PARAMETER, got %d (%08x)\n", GetLastError(),
104 GetLastError());
105 }
106 /* returns NULL, last error not set */
107 SetLastError(0xdeadbeef);
108 ret = CertFindExtension("bogus", 1, &ext);
109 ok(ret == NULL, "Expected failure\n");
110 ok(GetLastError() == 0xdeadbeef, "Last error was set to %08x\n",
111 GetLastError());
112 /* returns NULL, last error not set */
113 SetLastError(0xdeadbeef);
114 ret = CertFindExtension("1.2.4", 1, &ext);
115 ok(ret == NULL, "Expected failure\n");
116 ok(GetLastError() == 0xdeadbeef, "Last error was set to %08x\n",
117 GetLastError());
118 /* succeeds, last error not set */
119 SetLastError(0xdeadbeef);
120 ret = CertFindExtension("1.2.3", 1, &ext);
121 ok(ret != NULL, "CertFindExtension failed: %08x\n", GetLastError());
122 }
123
124 static void test_findRDNAttr(void)
125 {
126 PCERT_RDN_ATTR ret;
127 static CHAR oid[] = "1.2.3";
128 BYTE bin[] = { 0x16,0x09,'J','u','a','n',' ','L','a','n','g' };
129 CERT_RDN_ATTR attrs[] = {
130 { oid, CERT_RDN_IA5_STRING, { sizeof bin, bin } },
131 };
132 CERT_RDN rdns[] = {
133 { sizeof(attrs) / sizeof(attrs[0]), attrs },
134 };
135 CERT_NAME_INFO nameInfo = { sizeof(rdns) / sizeof(rdns[0]), rdns };
136
137 if (0)
138 {
139 /* crashes */
140 SetLastError(0xdeadbeef);
141 ret = CertFindRDNAttr(NULL, NULL);
142 /* returns NULL, last error is ERROR_INVALID_PARAMETER
143 * crashes on Vista
144 */
145 SetLastError(0xdeadbeef);
146 ret = CertFindRDNAttr(NULL, &nameInfo);
147 ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
148 "Expected ERROR_INVALID_PARAMETER, got %d (%08x)\n", GetLastError(),
149 GetLastError());
150 }
151 /* returns NULL, last error not set */
152 SetLastError(0xdeadbeef);
153 ret = CertFindRDNAttr("bogus", &nameInfo);
154 ok(ret == NULL, "Expected failure\n");
155 ok(GetLastError() == 0xdeadbeef, "Last error was set to %08x\n",
156 GetLastError());
157 /* returns NULL, last error not set */
158 SetLastError(0xdeadbeef);
159 ret = CertFindRDNAttr("1.2.4", &nameInfo);
160 ok(ret == NULL, "Expected failure\n");
161 ok(GetLastError() == 0xdeadbeef, "Last error was set to %08x\n",
162 GetLastError());
163 /* succeeds, last error not set */
164 SetLastError(0xdeadbeef);
165 ret = CertFindRDNAttr("1.2.3", &nameInfo);
166 ok(ret != NULL, "CertFindRDNAttr failed: %08x\n", GetLastError());
167 }
168
169 static void test_verifyTimeValidity(void)
170 {
171 SYSTEMTIME sysTime;
172 FILETIME fileTime;
173 CERT_INFO info = { 0 };
174 LONG ret;
175
176 GetSystemTime(&sysTime);
177 SystemTimeToFileTime(&sysTime, &fileTime);
178 /* crashes
179 ret = CertVerifyTimeValidity(NULL, NULL);
180 ret = CertVerifyTimeValidity(&fileTime, NULL);
181 */
182 /* Check with 0 NotBefore and NotAfter */
183 ret = CertVerifyTimeValidity(&fileTime, &info);
184 ok(ret == 1, "Expected 1, got %d\n", ret);
185 memcpy(&info.NotAfter, &fileTime, sizeof(info.NotAfter));
186 /* Check with NotAfter equal to comparison time */
187 ret = CertVerifyTimeValidity(&fileTime, &info);
188 ok(ret == 0, "Expected 0, got %d\n", ret);
189 /* Check with NotBefore after comparison time */
190 memcpy(&info.NotBefore, &fileTime, sizeof(info.NotBefore));
191 info.NotBefore.dwLowDateTime += 5000;
192 ret = CertVerifyTimeValidity(&fileTime, &info);
193 ok(ret == -1, "Expected -1, got %d\n", ret);
194 }
195
196 static void test_cryptAllocate(void)
197 {
198 LPVOID buf;
199
200 buf = CryptMemAlloc(0);
201 ok(buf != NULL, "CryptMemAlloc failed: %08x\n", GetLastError());
202 CryptMemFree(buf);
203 buf = CryptMemRealloc(NULL, 0);
204 ok(!buf, "Expected NULL\n");
205 buf = CryptMemAlloc(0);
206 buf = CryptMemRealloc(buf, 1);
207 ok(buf != NULL, "CryptMemRealloc failed: %08x\n", GetLastError());
208 CryptMemFree(buf);
209 }
210
211 typedef DWORD (WINAPI *I_CryptAllocTlsFunc)(void);
212 typedef LPVOID (WINAPI *I_CryptDetachTlsFunc)(DWORD dwTlsIndex);
213 typedef LPVOID (WINAPI *I_CryptGetTlsFunc)(DWORD dwTlsIndex);
214 typedef BOOL (WINAPI *I_CryptSetTlsFunc)(DWORD dwTlsIndex, LPVOID lpTlsValue);
215 typedef BOOL (WINAPI *I_CryptFreeTlsFunc)(DWORD dwTlsIndex, DWORD unknown);
216
217 static I_CryptAllocTlsFunc pI_CryptAllocTls;
218 static I_CryptDetachTlsFunc pI_CryptDetachTls;
219 static I_CryptGetTlsFunc pI_CryptGetTls;
220 static I_CryptSetTlsFunc pI_CryptSetTls;
221 static I_CryptFreeTlsFunc pI_CryptFreeTls;
222
223 static void test_cryptTls(void)
224 {
225 DWORD index;
226 BOOL ret;
227
228 pI_CryptAllocTls = (I_CryptAllocTlsFunc)GetProcAddress(hCrypt,
229 "I_CryptAllocTls");
230 pI_CryptDetachTls = (I_CryptDetachTlsFunc)GetProcAddress(hCrypt,
231 "I_CryptDetachTls");
232 pI_CryptGetTls = (I_CryptGetTlsFunc)GetProcAddress(hCrypt,
233 "I_CryptGetTls");
234 pI_CryptSetTls = (I_CryptSetTlsFunc)GetProcAddress(hCrypt,
235 "I_CryptSetTls");
236 pI_CryptFreeTls = (I_CryptFreeTlsFunc)GetProcAddress(hCrypt,
237 "I_CryptFreeTls");
238
239 /* One normal pass */
240 index = pI_CryptAllocTls();
241 ok(index, "I_CryptAllocTls failed: %08x\n", GetLastError());
242 if (index)
243 {
244 LPVOID ptr;
245
246 ptr = pI_CryptGetTls(index);
247 ok(!ptr, "Expected NULL\n");
248 ret = pI_CryptSetTls(index, (LPVOID)0xdeadbeef);
249 ok(ret, "I_CryptSetTls failed: %08x\n", GetLastError());
250 ptr = pI_CryptGetTls(index);
251 ok(ptr == (LPVOID)0xdeadbeef, "Expected 0xdeadbeef, got %p\n", ptr);
252 /* This crashes
253 ret = pI_CryptFreeTls(index, 1);
254 */
255 ret = pI_CryptFreeTls(index, 0);
256 ok(ret, "I_CryptFreeTls failed: %08x\n", GetLastError());
257 ret = pI_CryptFreeTls(index, 0);
258 /* Not sure if this fails because TlsFree should fail, so leave as
259 * todo for now.
260 */
261 todo_wine ok(!ret && GetLastError() == E_INVALIDARG,
262 "Expected E_INVALIDARG, got %08x\n", GetLastError());
263 }
264 /* Similar pass, check I_CryptDetachTls */
265 index = pI_CryptAllocTls();
266 ok(index, "I_CryptAllocTls failed: %08x\n", GetLastError());
267 if (index)
268 {
269 LPVOID ptr;
270
271 ptr = pI_CryptGetTls(index);
272 ok(!ptr, "Expected NULL\n");
273 ret = pI_CryptSetTls(index, (LPVOID)0xdeadbeef);
274 ok(ret, "I_CryptSetTls failed: %08x\n", GetLastError());
275 ptr = pI_CryptGetTls(index);
276 ok(ptr == (LPVOID)0xdeadbeef, "Expected 0xdeadbeef, got %p\n", ptr);
277 ptr = pI_CryptDetachTls(index);
278 ok(ptr == (LPVOID)0xdeadbeef, "Expected 0xdeadbeef, got %p\n", ptr);
279 ptr = pI_CryptGetTls(index);
280 ok(!ptr, "Expected NULL\n");
281 }
282 }
283
284 typedef BOOL (WINAPI *I_CryptReadTrustedPublisherDWORDValueFromRegistryFunc)
285 (LPCWSTR, DWORD *);
286
287 static void test_readTrustedPublisherDWORD(void)
288 {
289 I_CryptReadTrustedPublisherDWORDValueFromRegistryFunc pReadDWORD;
290
291 pReadDWORD =
292 (I_CryptReadTrustedPublisherDWORDValueFromRegistryFunc)GetProcAddress(
293 hCrypt, "I_CryptReadTrustedPublisherDWORDValueFromRegistry");
294 if (pReadDWORD)
295 {
296 static const WCHAR safer[] = {
297 'S','o','f','t','w','a','r','e','\\',
298 'P','o','l','i','c','i','e','s','\\',
299 'M','i','c','r','o','s','o','f','t','\\','S','y','s','t','e','m',
300 'C','e','r','t','i','f','i','c','a','t','e','s','\\',
301 'T','r','u','s','t','e','d','P','u','b','l','i','s','h','e','r',
302 '\\','S','a','f','e','r',0 };
303 static const WCHAR authenticodeFlags[] = { 'A','u','t','h','e','n',
304 't','i','c','o','d','e','F','l','a','g','s',0 };
305 BOOL ret, exists = FALSE;
306 DWORD size, readFlags = 0, returnedFlags;
307 HKEY key;
308 LONG rc;
309
310 rc = RegOpenKeyW(HKEY_LOCAL_MACHINE, safer, &key);
311 if (rc == ERROR_SUCCESS)
312 {
313 size = sizeof(readFlags);
314 rc = RegQueryValueExW(key, authenticodeFlags, NULL, NULL,
315 (LPBYTE)&readFlags, &size);
316 if (rc == ERROR_SUCCESS)
317 exists = TRUE;
318 }
319 returnedFlags = 0xdeadbeef;
320 ret = pReadDWORD(authenticodeFlags, &returnedFlags);
321 ok(ret == exists, "Unexpected return value\n");
322 ok(readFlags == returnedFlags,
323 "Expected flags %08x, got %08x\n", readFlags, returnedFlags);
324 }
325 }
326
327 typedef HCRYPTPROV (WINAPI *I_CryptGetDefaultCryptProvFunc)(DWORD w);
328
329 static void test_getDefaultCryptProv(void)
330 {
331 I_CryptGetDefaultCryptProvFunc pI_CryptGetDefaultCryptProv;
332 HCRYPTPROV prov;
333
334 pI_CryptGetDefaultCryptProv = (I_CryptGetDefaultCryptProvFunc)
335 GetProcAddress(hCrypt, "I_CryptGetDefaultCryptProv");
336 if (!pI_CryptGetDefaultCryptProv) return;
337
338 prov = pI_CryptGetDefaultCryptProv(0xdeadbeef);
339 ok(prov == 0 && GetLastError() == E_INVALIDARG,
340 "Expected E_INVALIDARG, got %08x\n", GetLastError());
341 prov = pI_CryptGetDefaultCryptProv(PROV_RSA_FULL);
342 ok(prov == 0 && GetLastError() == E_INVALIDARG,
343 "Expected E_INVALIDARG, got %08x\n", GetLastError());
344 prov = pI_CryptGetDefaultCryptProv(1);
345 ok(prov == 0 && GetLastError() == E_INVALIDARG,
346 "Expected E_INVALIDARG, got %08x\n", GetLastError());
347 prov = pI_CryptGetDefaultCryptProv(0);
348 ok(prov != 0, "I_CryptGetDefaultCryptProv failed: %08x\n", GetLastError());
349 CryptReleaseContext(prov, 0);
350 }
351
352 typedef int (WINAPI *I_CryptInstallOssGlobal)(DWORD,DWORD,DWORD);
353
354 static void test_CryptInstallOssGlobal(void)
355 {
356 int ret,i;
357 I_CryptInstallOssGlobal pI_CryptInstallOssGlobal;
358
359 pI_CryptInstallOssGlobal= (I_CryptInstallOssGlobal)GetProcAddress(hCrypt,"I_CryptInstallOssGlobal");
360 /* passing in some random values to I_CryptInstallOssGlobal, it always returns 9 the first time, then 10, 11 etc.*/
361 for(i=0;i<30;i++)
362 {
363 ret = pI_CryptInstallOssGlobal(rand(),rand(),rand());
364 ok((9+i) == ret,"Expected %d, got %d\n",(9+i),ret);
365 }
366 }
367
368 START_TEST(main)
369 {
370 hCrypt = GetModuleHandleA("crypt32.dll");
371
372 test_findAttribute();
373 test_findExtension();
374 test_findRDNAttr();
375 test_verifyTimeValidity();
376 test_cryptAllocate();
377 test_cryptTls();
378 test_readTrustedPublisherDWORD();
379 test_getDefaultCryptProv();
380 test_CryptInstallOssGlobal();
381 }
382
This page was automatically generated by the
LXR engine.
Visit the LXR main site for more
information.