?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
push 8edcbf8579c1d48dd3fb4c679acf4b3012a9efac
[wine/hacks.git] / dlls / crypt32 / tests / store.c
blob450b754953b420830d4ed9163b004d3c773408dd
1 /*
2 * crypt32 cert store function tests
3 *
4 * Copyright 2005-2006 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 <assert.h>
22 #include <stdio.h>
23 #include <stdarg.h>
24 #include <windef.h>
25 #include <winbase.h>
26 #include <winreg.h>
27 #include <winerror.h>
28 #include <wincrypt.h>
29
30 #include "wine/test.h"
31
32 /* The following aren't defined in wincrypt.h, as they're "reserved" */
33 #define CERT_CERT_PROP_ID 32
34 #define CERT_CRL_PROP_ID 33
35 #define CERT_CTL_PROP_ID 34
36
37 struct CertPropIDHeader
38 {
39 DWORD propID;
40 DWORD unknown1;
41 DWORD cb;
42 };
43
44 static const BYTE emptyCert[] = { 0x30, 0x00 };
45 static const BYTE bigCert[] = { 0x30, 0x7a, 0x02, 0x01, 0x01, 0x30, 0x02, 0x06,
46 0x00, 0x30, 0x15, 0x31, 0x13, 0x30, 0x11, 0x06, 0x03, 0x55, 0x04, 0x03, 0x13,
47 0x0a, 0x4a, 0x75, 0x61, 0x6e, 0x20, 0x4c, 0x61, 0x6e, 0x67, 0x00, 0x30, 0x22,
48 0x18, 0x0f, 0x31, 0x36, 0x30, 0x31, 0x30, 0x31, 0x30, 0x31, 0x30, 0x30, 0x30,
49 0x30, 0x30, 0x30, 0x5a, 0x18, 0x0f, 0x31, 0x36, 0x30, 0x31, 0x30, 0x31, 0x30,
50 0x31, 0x30, 0x30, 0x30, 0x30, 0x30, 0x30, 0x5a, 0x30, 0x15, 0x31, 0x13, 0x30,
51 0x11, 0x06, 0x03, 0x55, 0x04, 0x03, 0x13, 0x0a, 0x4a, 0x75, 0x61, 0x6e, 0x20,
52 0x4c, 0x61, 0x6e, 0x67, 0x00, 0x30, 0x07, 0x30, 0x02, 0x06, 0x00, 0x03, 0x01,
53 0x00, 0xa3, 0x16, 0x30, 0x14, 0x30, 0x12, 0x06, 0x03, 0x55, 0x1d, 0x13, 0x01,
54 0x01, 0xff, 0x04, 0x08, 0x30, 0x06, 0x01, 0x01, 0xff, 0x02, 0x01, 0x01 };
55 static const BYTE signedBigCert[] = {
56 0x30, 0x81, 0x93, 0x30, 0x7a, 0x02, 0x01, 0x01, 0x30, 0x02, 0x06, 0x00, 0x30,
57 0x15, 0x31, 0x13, 0x30, 0x11, 0x06, 0x03, 0x55, 0x04, 0x03, 0x13, 0x0a, 0x4a,
58 0x75, 0x61, 0x6e, 0x20, 0x4c, 0x61, 0x6e, 0x67, 0x00, 0x30, 0x22, 0x18, 0x0f,
59 0x31, 0x36, 0x30, 0x31, 0x30, 0x31, 0x30, 0x31, 0x30, 0x30, 0x30, 0x30, 0x30,
60 0x30, 0x5a, 0x18, 0x0f, 0x31, 0x36, 0x30, 0x31, 0x30, 0x31, 0x30, 0x31, 0x30,
61 0x30, 0x30, 0x30, 0x30, 0x30, 0x5a, 0x30, 0x15, 0x31, 0x13, 0x30, 0x11, 0x06,
62 0x03, 0x55, 0x04, 0x03, 0x13, 0x0a, 0x4a, 0x75, 0x61, 0x6e, 0x20, 0x4c, 0x61,
63 0x6e, 0x67, 0x00, 0x30, 0x07, 0x30, 0x02, 0x06, 0x00, 0x03, 0x01, 0x00, 0xa3,
64 0x16, 0x30, 0x14, 0x30, 0x12, 0x06, 0x03, 0x55, 0x1d, 0x13, 0x01, 0x01, 0xff,
65 0x04, 0x08, 0x30, 0x06, 0x01, 0x01, 0xff, 0x02, 0x01, 0x01, 0x30, 0x02, 0x06,
66 0x00, 0x03, 0x11, 0x00, 0x0f, 0x0e, 0x0d, 0x0c, 0x0b, 0x0a, 0x09, 0x08, 0x07,
67 0x06, 0x05, 0x04, 0x03, 0x02, 0x01, 0x00 };
68 static const BYTE serializedCert[] = { 0x20, 0x00, 0x00, 0x00,
69 0x01, 0x00, 0x00, 0x00, 0x7c, 0x00, 0x00, 0x00, 0x30, 0x7a, 0x02, 0x01, 0x01,
70 0x30, 0x02, 0x06, 0x00, 0x30, 0x15, 0x31, 0x13, 0x30, 0x11, 0x06, 0x03, 0x55,
71 0x04, 0x03, 0x13, 0x0a, 0x4a, 0x75, 0x61, 0x6e, 0x20, 0x4c, 0x61, 0x6e, 0x67,
72 0x00, 0x30, 0x22, 0x18, 0x0f, 0x31, 0x36, 0x30, 0x31, 0x30, 0x31, 0x30, 0x31,
73 0x30, 0x30, 0x30, 0x30, 0x30, 0x30, 0x5a, 0x18, 0x0f, 0x31, 0x36, 0x30, 0x31,
74 0x30, 0x31, 0x30, 0x31, 0x30, 0x30, 0x30, 0x30, 0x30, 0x30, 0x5a, 0x30, 0x15,
75 0x31, 0x13, 0x30, 0x11, 0x06, 0x03, 0x55, 0x04, 0x03, 0x13, 0x0a, 0x4a, 0x75,
76 0x61, 0x6e, 0x20, 0x4c, 0x61, 0x6e, 0x67, 0x00, 0x30, 0x07, 0x30, 0x02, 0x06,
77 0x00, 0x03, 0x01, 0x00, 0xa3, 0x16, 0x30, 0x14, 0x30, 0x12, 0x06, 0x03, 0x55,
78 0x1d, 0x13, 0x01, 0x01, 0xff, 0x04, 0x08, 0x30, 0x06, 0x01, 0x01, 0xff, 0x02,
79 0x01, 0x01 };
80 static const BYTE signedCRL[] = { 0x30, 0x45, 0x30, 0x2c, 0x30, 0x02, 0x06,
81 0x00, 0x30, 0x15, 0x31, 0x13, 0x30, 0x11, 0x06, 0x03, 0x55, 0x04, 0x03, 0x13,
82 0x0a, 0x4a, 0x75, 0x61, 0x6e, 0x20, 0x4c, 0x61, 0x6e, 0x67, 0x00, 0x18, 0x0f,
83 0x31, 0x36, 0x30, 0x31, 0x30, 0x31, 0x30, 0x31, 0x30, 0x30, 0x30, 0x30, 0x30,
84 0x30, 0x5a, 0x30, 0x02, 0x06, 0x00, 0x03, 0x11, 0x00, 0x0f, 0x0e, 0x0d, 0x0c,
85 0x0b, 0x0a, 0x09, 0x08, 0x07, 0x06, 0x05, 0x04, 0x03, 0x02, 0x01, 0x00 };
86 static const BYTE bigCert2[] = { 0x30, 0x7a, 0x02, 0x01, 0x01, 0x30, 0x02, 0x06,
87 0x00, 0x30, 0x15, 0x31, 0x13, 0x30, 0x11, 0x06, 0x03, 0x55, 0x04, 0x03, 0x13,
88 0x0a, 0x41, 0x6c, 0x65, 0x78, 0x20, 0x4c, 0x61, 0x6e, 0x67, 0x00, 0x30, 0x22,
89 0x18, 0x0f, 0x31, 0x36, 0x30, 0x31, 0x30, 0x31, 0x30, 0x31, 0x30, 0x30, 0x30,
90 0x30, 0x30, 0x30, 0x5a, 0x18, 0x0f, 0x31, 0x36, 0x30, 0x31, 0x30, 0x31, 0x30,
91 0x31, 0x30, 0x30, 0x30, 0x30, 0x30, 0x30, 0x5a, 0x30, 0x15, 0x31, 0x13, 0x30,
92 0x11, 0x06, 0x03, 0x55, 0x04, 0x03, 0x13, 0x0a, 0x41, 0x6c, 0x65, 0x78, 0x20,
93 0x4c, 0x61, 0x6e, 0x67, 0x00, 0x30, 0x07, 0x30, 0x02, 0x06, 0x00, 0x03, 0x01,
94 0x00, 0xa3, 0x16, 0x30, 0x14, 0x30, 0x12, 0x06, 0x03, 0x55, 0x1d, 0x13, 0x01,
95 0x01, 0xff, 0x04, 0x08, 0x30, 0x06, 0x01, 0x01, 0xff, 0x02, 0x01, 0x01 };
96
97
98 static BOOL (WINAPI *pCertAddStoreToCollection)(HCERTSTORE,HCERTSTORE,DWORD,DWORD);
99 static BOOL (WINAPI *pCertControlStore)(HCERTSTORE,DWORD,DWORD,void const*);
100 static PCCRL_CONTEXT (WINAPI *pCertEnumCRLsInStore)(HCERTSTORE,PCCRL_CONTEXT);
101 static BOOL (WINAPI *pCertEnumSystemStore)(DWORD,void*,void*,PFN_CERT_ENUM_SYSTEM_STORE);
102 static BOOL (WINAPI *pCertGetStoreProperty)(HCERTSTORE,DWORD,void*,DWORD*);
103 static void (WINAPI *pCertRemoveStoreFromCollection)(HCERTSTORE,HCERTSTORE);
104 static BOOL (WINAPI *pCertSetStoreProperty)(HCERTSTORE,DWORD,DWORD,const void*);
105
106 static void testMemStore(void)
107 {
108 HCERTSTORE store1, store2;
109 PCCERT_CONTEXT context;
110 BOOL ret;
111 DWORD GLE;
112
113 /* NULL provider */
114 store1 = CertOpenStore(0, 0, 0, 0, NULL);
115 ok(!store1 && GetLastError() == ERROR_FILE_NOT_FOUND,
116 "Expected ERROR_FILE_NOT_FOUND, got %d\n", GetLastError());
117 /* weird flags */
118 store1 = CertOpenStore(CERT_STORE_PROV_MEMORY, 0, 0,
119 CERT_STORE_DELETE_FLAG, NULL);
120 ok(!store1 && GetLastError() == ERROR_CALL_NOT_IMPLEMENTED,
121 "Expected ERROR_CALL_NOT_IMPLEMENTED, got %d\n", GetLastError());
122
123 /* normal */
124 store1 = CertOpenStore(CERT_STORE_PROV_MEMORY, 0, 0,
125 CERT_STORE_CREATE_NEW_FLAG, NULL);
126 ok(store1 != NULL, "CertOpenStore failed: %d\n", GetLastError());
127 /* open existing doesn't */
128 store2 = CertOpenStore(CERT_STORE_PROV_MEMORY, 0, 0,
129 CERT_STORE_OPEN_EXISTING_FLAG, NULL);
130 ok(store2 != NULL, "CertOpenStore failed: %d\n", GetLastError());
131 ok(store1 != store2, "Expected different stores\n");
132
133 /* add a bogus (empty) cert */
134 context = NULL;
135 ret = CertAddEncodedCertificateToStore(store1, X509_ASN_ENCODING, emptyCert,
136 sizeof(emptyCert), CERT_STORE_ADD_ALWAYS, &context);
137 /* Windows returns CRYPT_E_ASN1_EOD or OSS_DATA_ERROR, but accept
138 * CRYPT_E_ASN1_CORRUPT as well (because matching errors is tough in this
139 * case)
140 */
141 GLE = GetLastError();
142 ok(!ret && (GLE == CRYPT_E_ASN1_EOD || GLE == CRYPT_E_ASN1_CORRUPT ||
143 GLE == OSS_DATA_ERROR),
144 "Expected CRYPT_E_ASN1_EOD or CRYPT_E_ASN1_CORRUPT or OSS_DATA_ERROR, got %08x\n",
145 GLE);
146 /* add a "signed" cert--the signature isn't a real signature, so this adds
147 * without any check of the signature's validity
148 */
149 ret = CertAddEncodedCertificateToStore(store1, X509_ASN_ENCODING,
150 signedBigCert, sizeof(signedBigCert), CERT_STORE_ADD_ALWAYS, &context);
151 ok(ret, "CertAddEncodedCertificateToStore failed: %08x\n", GetLastError());
152 ok(context != NULL, "Expected a valid cert context\n");
153 if (context)
154 {
155 ok(context->cbCertEncoded == sizeof(signedBigCert),
156 "Wrong cert size %d\n", context->cbCertEncoded);
157 ok(!memcmp(context->pbCertEncoded, signedBigCert,
158 sizeof(signedBigCert)), "Unexpected encoded cert in context\n");
159 /* remove it, the rest of the tests will work on an unsigned cert */
160 ret = CertDeleteCertificateFromStore(context);
161 ok(ret, "CertDeleteCertificateFromStore failed: %08x\n",
162 GetLastError());
163 }
164 /* try adding a "signed" CRL as a cert */
165 ret = CertAddEncodedCertificateToStore(store1, X509_ASN_ENCODING,
166 signedCRL, sizeof(signedCRL), CERT_STORE_ADD_ALWAYS, &context);
167 GLE = GetLastError();
168 ok(!ret && (GLE == CRYPT_E_ASN1_BADTAG || GLE == CRYPT_E_ASN1_CORRUPT ||
169 GLE == OSS_DATA_ERROR),
170 "Expected CRYPT_E_ASN1_BADTAG or CRYPT_E_ASN1_CORRUPT or OSS_DATA_ERROR, got %08x\n",
171 GLE);
172 /* add a cert to store1 */
173 ret = CertAddEncodedCertificateToStore(store1, X509_ASN_ENCODING, bigCert,
174 sizeof(bigCert), CERT_STORE_ADD_ALWAYS, &context);
175 ok(ret, "CertAddEncodedCertificateToStore failed: %08x\n", GetLastError());
176 ok(context != NULL, "Expected a valid cert context\n");
177 if (context)
178 {
179 DWORD size;
180 BYTE *buf;
181
182 ok(context->cbCertEncoded == sizeof(bigCert),
183 "Wrong cert size %d\n", context->cbCertEncoded);
184 ok(!memcmp(context->pbCertEncoded, bigCert, sizeof(bigCert)),
185 "Unexpected encoded cert in context\n");
186 ok(context->hCertStore == store1, "Unexpected store\n");
187
188 /* check serializing this element */
189 /* These crash
190 ret = CertSerializeCertificateStoreElement(NULL, 0, NULL, NULL);
191 ret = CertSerializeCertificateStoreElement(context, 0, NULL, NULL);
192 ret = CertSerializeCertificateStoreElement(NULL, 0, NULL, &size);
193 */
194 /* apparently flags are ignored */
195 ret = CertSerializeCertificateStoreElement(context, 1, NULL, &size);
196 ok(ret, "CertSerializeCertificateStoreElement failed: %08x\n",
197 GetLastError());
198 buf = HeapAlloc(GetProcessHeap(), 0, size);
199 if (buf)
200 {
201 ret = CertSerializeCertificateStoreElement(context, 0, buf, &size);
202 ok(size == sizeof(serializedCert), "Wrong size %d\n", size);
203 ok(!memcmp(serializedCert, buf, size),
204 "Unexpected serialized cert\n");
205 HeapFree(GetProcessHeap(), 0, buf);
206 }
207
208 ret = CertFreeCertificateContext(context);
209 ok(ret, "CertFreeCertificateContext failed: %08x\n", GetLastError());
210 }
211 /* verify the cert's in store1 */
212 context = CertEnumCertificatesInStore(store1, NULL);
213 ok(context != NULL, "Expected a valid context\n");
214 context = CertEnumCertificatesInStore(store1, context);
215 ok(!context && GetLastError() == CRYPT_E_NOT_FOUND,
216 "Expected CRYPT_E_NOT_FOUND, got %08x\n", GetLastError());
217 /* verify store2 (the "open existing" mem store) is still empty */
218 context = CertEnumCertificatesInStore(store2, NULL);
219 ok(!context, "Expected an empty store\n");
220 /* delete the cert from store1, and check it's empty */
221 context = CertEnumCertificatesInStore(store1, NULL);
222 if (context)
223 {
224 /* Deleting a bitwise copy crashes with an access to an uninitialized
225 * pointer, so a cert context has some special data out there in memory
226 * someplace
227 CERT_CONTEXT copy;
228 memcpy(&copy, context, sizeof(copy));
229 ret = CertDeleteCertificateFromStore(&copy);
230 */
231 PCCERT_CONTEXT copy = CertDuplicateCertificateContext(context);
232
233 ok(copy != NULL, "CertDuplicateCertificateContext failed: %08x\n",
234 GetLastError());
235 ret = CertDeleteCertificateFromStore(context);
236 ok(ret, "CertDeleteCertificateFromStore failed: %08x\n",
237 GetLastError());
238 /* try deleting a copy */
239 ret = CertDeleteCertificateFromStore(copy);
240 ok(ret, "CertDeleteCertificateFromStore failed: %08x\n",
241 GetLastError());
242 /* check that the store is empty */
243 context = CertEnumCertificatesInStore(store1, NULL);
244 ok(!context, "Expected an empty store\n");
245 }
246
247 /* close an empty store */
248 ret = CertCloseStore(NULL, 0);
249 ok(ret, "CertCloseStore failed: %d\n", GetLastError());
250 ret = CertCloseStore(store1, 0);
251 ok(ret, "CertCloseStore failed: %d\n", GetLastError());
252 ret = CertCloseStore(store2, 0);
253 ok(ret, "CertCloseStore failed: %d\n", GetLastError());
254
255 /* This seems nonsensical, but you can open a read-only mem store, only
256 * it isn't read-only
257 */
258 store1 = CertOpenStore(CERT_STORE_PROV_MEMORY, 0, 0,
259 CERT_STORE_READONLY_FLAG, NULL);
260 ok(store1 != NULL, "CertOpenStore failed: %d\n", GetLastError());
261 /* yep, this succeeds */
262 ret = CertAddEncodedCertificateToStore(store1, X509_ASN_ENCODING, bigCert,
263 sizeof(bigCert), CERT_STORE_ADD_ALWAYS, &context);
264 ok(ret, "CertAddEncodedCertificateToStore failed: %08x\n", GetLastError());
265 ok(context != NULL, "Expected a valid cert context\n");
266 if (context)
267 {
268 ok(context->cbCertEncoded == sizeof(bigCert),
269 "Wrong cert size %d\n", context->cbCertEncoded);
270 ok(!memcmp(context->pbCertEncoded, bigCert, sizeof(bigCert)),
271 "Unexpected encoded cert in context\n");
272 ok(context->hCertStore == store1, "Unexpected store\n");
273 ret = CertDeleteCertificateFromStore(context);
274 ok(ret, "CertDeleteCertificateFromStore failed: %08x\n",
275 GetLastError());
276 }
277 CertCloseStore(store1, 0);
278 }
279
280 static void testCollectionStore(void)
281 {
282 HCERTSTORE store1, store2, collection, collection2;
283 PCCERT_CONTEXT context;
284 BOOL ret;
285
286 if (!pCertAddStoreToCollection)
287 {
288 skip("CertAddStoreToCollection() is not available\n");
289 return;
290 }
291
292 collection = CertOpenStore(CERT_STORE_PROV_COLLECTION, 0, 0,
293 CERT_STORE_CREATE_NEW_FLAG, NULL);
294
295 /* Try adding a cert to any empty collection */
296 ret = CertAddEncodedCertificateToStore(collection, X509_ASN_ENCODING,
297 bigCert, sizeof(bigCert), CERT_STORE_ADD_ALWAYS, NULL);
298 ok(!ret && GetLastError() == E_ACCESSDENIED,
299 "Expected E_ACCESSDENIED, got %08x\n", GetLastError());
300
301 /* Create and add a cert to a memory store */
302 store1 = CertOpenStore(CERT_STORE_PROV_MEMORY, 0, 0,
303 CERT_STORE_CREATE_NEW_FLAG, NULL);
304 ret = CertAddEncodedCertificateToStore(store1, X509_ASN_ENCODING,
305 bigCert, sizeof(bigCert), CERT_STORE_ADD_ALWAYS, NULL);
306 ok(ret, "CertAddEncodedCertificateToStore failed: %08x\n", GetLastError());
307 /* Add the memory store to the collection, without allowing adding */
308 ret = pCertAddStoreToCollection(collection, store1, 0, 0);
309 ok(ret, "CertAddStoreToCollection failed: %08x\n", GetLastError());
310 /* Verify the cert is in the collection */
311 context = CertEnumCertificatesInStore(collection, NULL);
312 ok(context != NULL, "Expected a valid context\n");
313 if (context)
314 {
315 ok(context->hCertStore == collection, "Unexpected store\n");
316 CertFreeCertificateContext(context);
317 }
318 /* Check that adding to the collection isn't allowed */
319 ret = CertAddEncodedCertificateToStore(collection, X509_ASN_ENCODING,
320 bigCert2, sizeof(bigCert2), CERT_STORE_ADD_ALWAYS, NULL);
321 ok(!ret && GetLastError() == E_ACCESSDENIED,
322 "Expected E_ACCESSDENIED, got %08x\n", GetLastError());
323
324 /* Create a new memory store */
325 store2 = CertOpenStore(CERT_STORE_PROV_MEMORY, 0, 0,
326 CERT_STORE_CREATE_NEW_FLAG, NULL);
327 /* Try adding a store to a non-collection store */
328 ret = pCertAddStoreToCollection(store1, store2,
329 CERT_PHYSICAL_STORE_ADD_ENABLE_FLAG, 0);
330 ok(!ret && GetLastError() == E_INVALIDARG,
331 "Expected E_INVALIDARG, got %08x\n", GetLastError());
332 /* Try adding some bogus stores */
333 /* This crashes in Windows
334 ret = pCertAddStoreToCollection(0, store2,
335 CERT_PHYSICAL_STORE_ADD_ENABLE_FLAG, 0);
336 */
337 /* This "succeeds"... */
338 ret = pCertAddStoreToCollection(collection, 0,
339 CERT_PHYSICAL_STORE_ADD_ENABLE_FLAG, 0);
340 ok(ret, "CertAddStoreToCollection failed: %08x\n", GetLastError());
341 /* while this crashes.
342 ret = pCertAddStoreToCollection(collection, 1,
343 CERT_PHYSICAL_STORE_ADD_ENABLE_FLAG, 0);
344 */
345
346 /* Add it to the collection, this time allowing adding */
347 ret = pCertAddStoreToCollection(collection, store2,
348 CERT_PHYSICAL_STORE_ADD_ENABLE_FLAG, 0);
349 ok(ret, "CertAddStoreToCollection failed: %08x\n", GetLastError());
350 /* Check that adding to the collection is allowed */
351 ret = CertAddEncodedCertificateToStore(collection, X509_ASN_ENCODING,
352 bigCert2, sizeof(bigCert2), CERT_STORE_ADD_ALWAYS, NULL);
353 ok(ret, "CertAddEncodedCertificateToStore failed: %08x\n", GetLastError());
354 /* Now check that it was actually added to store2 */
355 context = CertEnumCertificatesInStore(store2, NULL);
356 ok(context != NULL, "Expected a valid context\n");
357 if (context)
358 {
359 ok(context->hCertStore == store2, "Unexpected store\n");
360 CertFreeCertificateContext(context);
361 }
362 /* Check that the collection has both bigCert and bigCert2. bigCert comes
363 * first because store1 was added first.
364 */
365 context = CertEnumCertificatesInStore(collection, NULL);
366 ok(context != NULL, "Expected a valid context\n");
367 if (context)
368 {
369 ok(context->hCertStore == collection, "Unexpected store\n");
370 ok(context->cbCertEncoded == sizeof(bigCert),
371 "Wrong size %d\n", context->cbCertEncoded);
372 ok(!memcmp(context->pbCertEncoded, bigCert, context->cbCertEncoded),
373 "Unexpected cert\n");
374 context = CertEnumCertificatesInStore(collection, context);
375 ok(context != NULL, "Expected a valid context\n");
376 if (context)
377 {
378 ok(context->hCertStore == collection, "Unexpected store\n");
379 ok(context->cbCertEncoded == sizeof(bigCert2),
380 "Wrong size %d\n", context->cbCertEncoded);
381 ok(!memcmp(context->pbCertEncoded, bigCert2,
382 context->cbCertEncoded), "Unexpected cert\n");
383 context = CertEnumCertificatesInStore(collection, context);
384 ok(!context, "Unexpected cert\n");
385 }
386 }
387 /* close store2, and check that the collection is unmodified */
388 CertCloseStore(store2, 0);
389 context = CertEnumCertificatesInStore(collection, NULL);
390 ok(context != NULL, "Expected a valid context\n");
391 if (context)
392 {
393 ok(context->hCertStore == collection, "Unexpected store\n");
394 ok(context->cbCertEncoded == sizeof(bigCert),
395 "Wrong size %d\n", context->cbCertEncoded);
396 ok(!memcmp(context->pbCertEncoded, bigCert, context->cbCertEncoded),
397 "Unexpected cert\n");
398 context = CertEnumCertificatesInStore(collection, context);
399 ok(context != NULL, "Expected a valid context\n");
400 if (context)
401 {
402 ok(context->hCertStore == collection, "Unexpected store\n");
403 ok(context->cbCertEncoded == sizeof(bigCert2),
404 "Wrong size %d\n", context->cbCertEncoded);
405 ok(!memcmp(context->pbCertEncoded, bigCert2,
406 context->cbCertEncoded), "Unexpected cert\n");
407 context = CertEnumCertificatesInStore(collection, context);
408 ok(!context, "Unexpected cert\n");
409 }
410 }
411
412 /* Adding a collection to a collection is legal */
413 collection2 = CertOpenStore(CERT_STORE_PROV_COLLECTION, 0, 0,
414 CERT_STORE_CREATE_NEW_FLAG, NULL);
415 ret = pCertAddStoreToCollection(collection2, collection,
416 CERT_PHYSICAL_STORE_ADD_ENABLE_FLAG, 0);
417 ok(ret, "CertAddStoreToCollection failed: %08x\n", GetLastError());
418 /* check the contents of collection2 */
419 context = CertEnumCertificatesInStore(collection2, NULL);
420 ok(context != NULL, "Expected a valid context\n");
421 if (context)
422 {
423 ok(context->hCertStore == collection2, "Unexpected store\n");
424 ok(context->cbCertEncoded == sizeof(bigCert),
425 "Wrong size %d\n", context->cbCertEncoded);
426 ok(!memcmp(context->pbCertEncoded, bigCert, context->cbCertEncoded),
427 "Unexpected cert\n");
428 context = CertEnumCertificatesInStore(collection2, context);
429 ok(context != NULL, "Expected a valid context\n");
430 if (context)
431 {
432 ok(context->hCertStore == collection2, "Unexpected store\n");
433 ok(context->cbCertEncoded == sizeof(bigCert2),
434 "Wrong size %d\n", context->cbCertEncoded);
435 ok(!memcmp(context->pbCertEncoded, bigCert2,
436 context->cbCertEncoded), "Unexpected cert\n");
437 context = CertEnumCertificatesInStore(collection2, context);
438 ok(!context, "Unexpected cert\n");
439 }
440 }
441
442 /* I'd like to test closing the collection in the middle of enumeration,
443 * but my tests have been inconsistent. The first time calling
444 * CertEnumCertificatesInStore on a closed collection succeeded, while the
445 * second crashed. So anything appears to be fair game.
446 * I'd also like to test removing a store from a collection in the middle
447 * of an enumeration, but my tests in Windows have been inconclusive.
448 * In one scenario it worked. In another scenario, about a third of the
449 * time this leads to "random" crashes elsewhere in the code. This
450 * probably means this is not allowed.
451 */
452
453 CertCloseStore(store1, 0);
454 CertCloseStore(collection, 0);
455 CertCloseStore(collection2, 0);
456
457 /* Add the same cert to two memory stores, then put them in a collection */
458 store1 = CertOpenStore(CERT_STORE_PROV_MEMORY, 0, 0,
459 CERT_STORE_CREATE_NEW_FLAG, NULL);
460 ok(store1 != 0, "CertOpenStore failed: %08x\n", GetLastError());
461 store2 = CertOpenStore(CERT_STORE_PROV_MEMORY, 0, 0,
462 CERT_STORE_CREATE_NEW_FLAG, NULL);
463 ok(store2 != 0, "CertOpenStore failed: %08x\n", GetLastError());
464
465 ret = CertAddEncodedCertificateToStore(store1, X509_ASN_ENCODING,
466 bigCert, sizeof(bigCert), CERT_STORE_ADD_ALWAYS, NULL);
467 ok(ret, "CertAddEncodedCertificateToStore failed: %08x\n", GetLastError());
468 ret = CertAddEncodedCertificateToStore(store2, X509_ASN_ENCODING,
469 bigCert, sizeof(bigCert), CERT_STORE_ADD_ALWAYS, NULL);
470 ok(ret, "CertAddEncodedCertificateToStore failed: %08x\n", GetLastError());
471 collection = CertOpenStore(CERT_STORE_PROV_COLLECTION, 0, 0,
472 CERT_STORE_CREATE_NEW_FLAG, NULL);
473 ok(collection != 0, "CertOpenStore failed: %08x\n", GetLastError());
474
475 ret = pCertAddStoreToCollection(collection, store1,
476 CERT_PHYSICAL_STORE_ADD_ENABLE_FLAG, 0);
477 ok(ret, "CertAddStoreToCollection failed: %08x\n", GetLastError());
478 ret = pCertAddStoreToCollection(collection, store2,
479 CERT_PHYSICAL_STORE_ADD_ENABLE_FLAG, 0);
480 ok(ret, "CertAddStoreToCollection failed: %08x\n", GetLastError());
481
482 /* Check that the collection has two copies of the same cert */
483 context = CertEnumCertificatesInStore(collection, NULL);
484 ok(context != NULL, "Expected a valid context\n");
485 if (context)
486 {
487 ok(context->hCertStore == collection, "Unexpected store\n");
488 ok(context->cbCertEncoded == sizeof(bigCert),
489 "Wrong size %d\n", context->cbCertEncoded);
490 ok(!memcmp(context->pbCertEncoded, bigCert, context->cbCertEncoded),
491 "Unexpected cert\n");
492 context = CertEnumCertificatesInStore(collection, context);
493 ok(context != NULL, "Expected a valid context\n");
494 if (context)
495 {
496 ok(context->hCertStore == collection, "Unexpected store\n");
497 ok(context->cbCertEncoded == sizeof(bigCert),
498 "Wrong size %d\n", context->cbCertEncoded);
499 ok(!memcmp(context->pbCertEncoded, bigCert, context->cbCertEncoded),
500 "Unexpected cert\n");
501 context = CertEnumCertificatesInStore(collection, context);
502 ok(context == NULL, "Unexpected cert\n");
503 }
504 }
505
506 /* The following would check whether I can delete an identical cert, rather
507 * than one enumerated from the store. It crashes, so that means I must
508 * only call CertDeleteCertificateFromStore with contexts enumerated from
509 * the store.
510 context = CertCreateCertificateContext(X509_ASN_ENCODING, bigCert,
511 sizeof(bigCert));
512 ok(context != NULL, "CertCreateCertificateContext failed: %08x\n",
513 GetLastError());
514 if (context)
515 {
516 ret = CertDeleteCertificateFromStore(collection, context);
517 printf("ret is %d, GetLastError is %08x\n", ret, GetLastError());
518 CertFreeCertificateContext(context);
519 }
520 */
521
522 /* Now check deleting from the collection. */
523 context = CertEnumCertificatesInStore(collection, NULL);
524 ok(context != NULL, "Expected a valid context\n");
525 if (context)
526 {
527 CertDeleteCertificateFromStore(context);
528 /* store1 should now be empty */
529 context = CertEnumCertificatesInStore(store1, NULL);
530 ok(!context, "Unexpected cert\n");
531 /* and there should be one certificate in the collection */
532 context = CertEnumCertificatesInStore(collection, NULL);
533 ok(context != NULL, "Expected a valid cert\n");
534 if (context)
535 {
536 ok(context->hCertStore == collection, "Unexpected store\n");
537 ok(context->cbCertEncoded == sizeof(bigCert),
538 "Wrong size %d\n", context->cbCertEncoded);
539 ok(!memcmp(context->pbCertEncoded, bigCert, context->cbCertEncoded),
540 "Unexpected cert\n");
541 }
542 context = CertEnumCertificatesInStore(collection, context);
543 ok(context == NULL, "Unexpected cert\n");
544 }
545
546 if (!pCertRemoveStoreFromCollection)
547 {
548 skip("CertRemoveStoreFromCollection() is not available\n");
549 }
550 else
551 {
552 /* Finally, test removing stores from the collection. No return
553 * value, so it's a bit funny to test.
554 */
555 /* This crashes
556 * pCertRemoveStoreFromCollection(NULL, NULL);
557 */
558 /* This "succeeds," no crash, no last error set */
559 SetLastError(0xdeadbeef);
560 pCertRemoveStoreFromCollection(store2, collection);
561 ok(GetLastError() == 0xdeadbeef,
562 "Didn't expect an error to be set: %08x\n", GetLastError());
563
564 /* After removing store2, the collection should be empty */
565 SetLastError(0xdeadbeef);
566 pCertRemoveStoreFromCollection(collection, store2);
567 ok(GetLastError() == 0xdeadbeef,
568 "Didn't expect an error to be set: %08x\n", GetLastError());
569 context = CertEnumCertificatesInStore(collection, NULL);
570 ok(!context, "Unexpected cert\n");
571 }
572
573 CertCloseStore(collection, 0);
574 CertCloseStore(store2, 0);
575 CertCloseStore(store1, 0);
576 }
577
578 /* Looks for the property with ID propID in the buffer buf. Returns a pointer
579 * to its header if found, NULL if not.
580 */
581 static const struct CertPropIDHeader *findPropID(const BYTE *buf, DWORD size,
582 DWORD propID)
583 {
584 const struct CertPropIDHeader *ret = NULL;
585 BOOL failed = FALSE;
586
587 while (size && !ret && !failed)
588 {
589 if (size < sizeof(struct CertPropIDHeader))
590 failed = TRUE;
591 else
592 {
593 const struct CertPropIDHeader *hdr =
594 (const struct CertPropIDHeader *)buf;
595
596 size -= sizeof(struct CertPropIDHeader);
597 buf += sizeof(struct CertPropIDHeader);
598 if (size < hdr->cb)
599 failed = TRUE;
600 else if (hdr->propID == propID)
601 ret = hdr;
602 else
603 {
604 buf += hdr->cb;
605 size -= hdr->cb;
606 }
607 }
608 }
609 return ret;
610 }
611
612 typedef DWORD (WINAPI *SHDeleteKeyAFunc)(HKEY, LPCSTR);
613
614 static void testRegStore(void)
615 {
616 static const char tempKey[] = "Software\\Wine\\CryptTemp";
617 HCERTSTORE store;
618 LONG rc;
619 HKEY key = NULL;
620 DWORD disp, GLE;
621
622 store = CertOpenStore(CERT_STORE_PROV_REG, 0, 0, 0, NULL);
623 GLE = GetLastError();
624 ok(!store && (GLE == ERROR_INVALID_HANDLE || GLE == ERROR_BADKEY),
625 "Expected ERROR_INVALID_HANDLE or ERROR_BADKEY, got %d\n", GLE);
626 store = CertOpenStore(CERT_STORE_PROV_REG, 0, 0, 0, key);
627 GLE = GetLastError();
628 ok(!store && (GLE == ERROR_INVALID_HANDLE || GLE == ERROR_BADKEY),
629 "Expected ERROR_INVALID_HANDLE or ERROR_BADKEY, got %d\n", GLE);
630
631 /* Opening up any old key works.. */
632 key = HKEY_CURRENT_USER;
633 store = CertOpenStore(CERT_STORE_PROV_REG, 0, 0, 0, key);
634 /* Not sure if this is a bug in DuplicateHandle, marking todo_wine for now
635 */
636 todo_wine ok(store != 0, "CertOpenStore failed: %08x\n", GetLastError());
637 CertCloseStore(store, 0);
638
639 /* It looks like the remainder pretty much needs CertControlStore() */
640 if (!pCertControlStore)
641 {
642 skip("CertControlStore() is not available\n");
643 return;
644 }
645
646 rc = RegCreateKeyExA(HKEY_CURRENT_USER, tempKey, 0, NULL, 0, KEY_ALL_ACCESS,
647 NULL, &key, NULL);
648 ok(!rc, "RegCreateKeyExA failed: %d\n", rc);
649 if (key)
650 {
651 BOOL ret;
652 BYTE hash[20];
653 DWORD size, i;
654 static const char certificates[] = "Certificates\\";
655 char subKeyName[sizeof(certificates) + 20 * 2 + 1], *ptr;
656 HKEY subKey;
657 PCCERT_CONTEXT context;
658
659 store = CertOpenStore(CERT_STORE_PROV_REG, 0, 0, 0, key);
660 ok(store != 0, "CertOpenStore failed: %08x\n", GetLastError());
661 /* Add a certificate. It isn't persisted right away, since it's only
662 * added to the cache..
663 */
664 ret = CertAddEncodedCertificateToStore(store, X509_ASN_ENCODING,
665 bigCert2, sizeof(bigCert2), CERT_STORE_ADD_ALWAYS, NULL);
666 ok(ret, "CertAddEncodedCertificateToStore failed: %08x\n",
667 GetLastError());
668 /* so flush the cache to force a commit.. */
669 ret = pCertControlStore(store, 0, CERT_STORE_CTRL_COMMIT, NULL);
670 ok(ret, "CertControlStore failed: %08x\n", GetLastError());
671 /* and check that the expected subkey was written. */
672 size = sizeof(hash);
673 ret = CryptHashCertificate(0, 0, 0, bigCert2, sizeof(bigCert2),
674 hash, &size);
675 ok(ret, "CryptHashCertificate failed: %d\n", GetLastError());
676 strcpy(subKeyName, certificates);
677 for (i = 0, ptr = subKeyName + sizeof(certificates) - 1; i < size;
678 i++, ptr += 2)
679 sprintf(ptr, "%02X", hash[i]);
680 rc = RegCreateKeyExA(key, subKeyName, 0, NULL, 0, KEY_ALL_ACCESS, NULL,
681 &subKey, NULL);
682 ok(!rc, "RegCreateKeyExA failed: %d\n", rc);
683 if (subKey)
684 {
685 LPBYTE buf;
686
687 size = 0;
688 RegQueryValueExA(subKey, "Blob", NULL, NULL, NULL, &size);
689 buf = HeapAlloc(GetProcessHeap(), 0, size);
690 if (buf)
691 {
692 rc = RegQueryValueExA(subKey, "Blob", NULL, NULL, buf, &size);
693 ok(!rc, "RegQueryValueExA failed: %d\n", rc);
694 if (!rc)
695 {
696 const struct CertPropIDHeader *hdr;
697
698 /* Both the hash and the cert should be present */
699 hdr = findPropID(buf, size, CERT_CERT_PROP_ID);
700 ok(hdr != NULL, "Expected to find a cert property\n");
701 if (hdr)
702 {
703 ok(hdr->cb == sizeof(bigCert2),
704 "Wrong size %d of cert property\n", hdr->cb);
705 ok(!memcmp((const BYTE *)hdr + sizeof(*hdr), bigCert2,
706 hdr->cb), "Unexpected cert in cert property\n");
707 }
708 hdr = findPropID(buf, size, CERT_HASH_PROP_ID);
709 ok(hdr != NULL, "Expected to find a hash property\n");
710 if (hdr)
711 {
712 ok(hdr->cb == sizeof(hash),
713 "Wrong size %d of hash property\n", hdr->cb);
714 ok(!memcmp((const BYTE *)hdr + sizeof(*hdr), hash,
715 hdr->cb), "Unexpected hash in cert property\n");
716 }
717 }
718 HeapFree(GetProcessHeap(), 0, buf);
719 }
720 RegCloseKey(subKey);
721 }
722
723 /* Remove the existing context */
724 context = CertEnumCertificatesInStore(store, NULL);
725 ok(context != NULL, "Expected a cert context\n");
726 if (context)
727 CertDeleteCertificateFromStore(context);
728 ret = pCertControlStore(store, 0, CERT_STORE_CTRL_COMMIT, NULL);
729 ok(ret, "CertControlStore failed: %08x\n", GetLastError());
730
731 /* Add a serialized cert with a bogus hash directly to the registry */
732 memset(hash, 0, sizeof(hash));
733 strcpy(subKeyName, certificates);
734 for (i = 0, ptr = subKeyName + sizeof(certificates) - 1;
735 i < sizeof(hash); i++, ptr += 2)
736 sprintf(ptr, "%02X", hash[i]);
737 rc = RegCreateKeyExA(key, subKeyName, 0, NULL, 0, KEY_ALL_ACCESS, NULL,
738 &subKey, NULL);
739 ok(!rc, "RegCreateKeyExA failed: %d\n", rc);
740 if (subKey)
741 {
742 BYTE buf[sizeof(struct CertPropIDHeader) * 2 + sizeof(hash) +
743 sizeof(bigCert)], *ptr;
744 DWORD certCount = 0;
745 struct CertPropIDHeader *hdr;
746
747 hdr = (struct CertPropIDHeader *)buf;
748 hdr->propID = CERT_HASH_PROP_ID;
749 hdr->unknown1 = 1;
750 hdr->cb = sizeof(hash);
751 ptr = buf + sizeof(*hdr);
752 memcpy(ptr, hash, sizeof(hash));
753 ptr += sizeof(hash);
754 hdr = (struct CertPropIDHeader *)ptr;
755 hdr->propID = CERT_CERT_PROP_ID;
756 hdr->unknown1 = 1;
757 hdr->cb = sizeof(bigCert);
758 ptr += sizeof(*hdr);
759 memcpy(ptr, bigCert, sizeof(bigCert));
760
761 rc = RegSetValueExA(subKey, "Blob", 0, REG_BINARY, buf,
762 sizeof(buf));
763 ok(!rc, "RegSetValueExA failed: %d\n", rc);
764
765 ret = pCertControlStore(store, 0, CERT_STORE_CTRL_RESYNC, NULL);
766 ok(ret, "CertControlStore failed: %08x\n", GetLastError());
767
768 /* Make sure the bogus hash cert gets loaded. */
769 certCount = 0;
770 context = NULL;
771 do {
772 context = CertEnumCertificatesInStore(store, context);
773 if (context)
774 certCount++;
775 } while (context != NULL);
776 ok(certCount == 1, "Expected 1 certificates, got %d\n", certCount);
777
778 RegCloseKey(subKey);
779 }
780
781 /* Add another serialized cert directly to the registry, this time
782 * under the correct key name (named with the correct hash value).
783 */
784 size = sizeof(hash);
785 ret = CryptHashCertificate(0, 0, 0, bigCert2,
786 sizeof(bigCert2), hash, &size);
787 ok(ret, "CryptHashCertificate failed: %d\n", GetLastError());
788 strcpy(subKeyName, certificates);
789 for (i = 0, ptr = subKeyName + sizeof(certificates) - 1;
790 i < sizeof(hash); i++, ptr += 2)
791 sprintf(ptr, "%02X", hash[i]);
792 rc = RegCreateKeyExA(key, subKeyName, 0, NULL, 0, KEY_ALL_ACCESS, NULL,
793 &subKey, NULL);
794 ok(!rc, "RegCreateKeyExA failed: %d\n", rc);
795 if (subKey)
796 {
797 BYTE buf[sizeof(struct CertPropIDHeader) * 2 + sizeof(hash) +
798 sizeof(bigCert2)], *ptr;
799 DWORD certCount = 0;
800 PCCERT_CONTEXT context;
801 struct CertPropIDHeader *hdr;
802
803 /* First try with a bogus hash... */
804 hdr = (struct CertPropIDHeader *)buf;
805 hdr->propID = CERT_HASH_PROP_ID;
806 hdr->unknown1 = 1;
807 hdr->cb = sizeof(hash);
808 ptr = buf + sizeof(*hdr);
809 memset(ptr, 0, sizeof(hash));
810 ptr += sizeof(hash);
811 hdr = (struct CertPropIDHeader *)ptr;
812 hdr->propID = CERT_CERT_PROP_ID;
813 hdr->unknown1 = 1;
814 hdr->cb = sizeof(bigCert2);
815 ptr += sizeof(*hdr);
816 memcpy(ptr, bigCert2, sizeof(bigCert2));
817
818 rc = RegSetValueExA(subKey, "Blob", 0, REG_BINARY, buf,
819 sizeof(buf));
820 ok(!rc, "RegSetValueExA failed: %d\n", rc);
821
822 ret = pCertControlStore(store, 0, CERT_STORE_CTRL_RESYNC, NULL);
823 ok(ret, "CertControlStore failed: %08x\n", GetLastError());
824
825 /* and make sure just one cert still gets loaded. */
826 certCount = 0;
827 context = NULL;
828 do {
829 context = CertEnumCertificatesInStore(store, context);
830 if (context)
831 certCount++;
832 } while (context != NULL);
833 ok(certCount == 1 ||
834 broken(certCount == 2), /* win9x */
835 "Expected 1 certificates, got %d\n", certCount);
836
837 /* Try again with the correct hash... */
838 ptr = buf + sizeof(*hdr);
839 memcpy(ptr, hash, sizeof(hash));
840
841 rc = RegSetValueExA(subKey, "Blob", 0, REG_BINARY, buf,
842 sizeof(buf));
843 ok(!rc, "RegSetValueExA failed: %d\n", rc);
844
845 ret = pCertControlStore(store, 0, CERT_STORE_CTRL_RESYNC, NULL);
846 ok(ret, "CertControlStore failed: %08x\n", GetLastError());
847
848 /* and make sure two certs get loaded. */
849 certCount = 0;
850 context = NULL;
851 do {
852 context = CertEnumCertificatesInStore(store, context);
853 if (context)
854 certCount++;
855 } while (context != NULL);
856 ok(certCount == 2, "Expected 2 certificates, got %d\n", certCount);
857
858 RegCloseKey(subKey);
859 }
860 CertCloseStore(store, 0);
861 /* Is delete allowed on a reg store? */
862 store = CertOpenStore(CERT_STORE_PROV_REG, 0, 0,
863 CERT_STORE_DELETE_FLAG, key);
864 ok(store == NULL, "Expected NULL return from CERT_STORE_DELETE_FLAG\n");
865 ok(GetLastError() == 0, "CertOpenStore failed: %08x\n",
866 GetLastError());
867
868 RegCloseKey(key);
869 }
870 /* The CertOpenStore with CERT_STORE_DELETE_FLAG above will delete the
871 * contents of the key, but not the key itself.
872 */
873 rc = RegCreateKeyExA(HKEY_CURRENT_USER, tempKey, 0, NULL, 0, KEY_ALL_ACCESS,
874 NULL, &key, &disp);
875 ok(!rc, "RegCreateKeyExA failed: %d\n", rc);
876 ok(disp == REG_OPENED_EXISTING_KEY,
877 "Expected REG_OPENED_EXISTING_KEY, got %d\n", disp);
878 if (!rc)
879 {
880 RegCloseKey(key);
881 rc = RegDeleteKeyA(HKEY_CURRENT_USER, tempKey);
882 if (rc)
883 {
884 HMODULE shlwapi = LoadLibraryA("shlwapi");
885
886 /* Use shlwapi's SHDeleteKeyA to _really_ blow away the key,
887 * otherwise subsequent tests will fail.
888 */
889 if (shlwapi)
890 {
891 SHDeleteKeyAFunc pSHDeleteKeyA =
892 (SHDeleteKeyAFunc)GetProcAddress(shlwapi, "SHDeleteKeyA");
893
894 if (pSHDeleteKeyA)
895 pSHDeleteKeyA(HKEY_CURRENT_USER, tempKey);
896 FreeLibrary(shlwapi);
897 }
898 }
899 }
900 }
901
902 static const char MyA[] = { 'M','y',0,0 };
903 static const WCHAR MyW[] = { 'M','y',0 };
904 static const WCHAR BogusW[] = { 'B','o','g','u','s',0 };
905 static const WCHAR BogusPathW[] = { 'S','o','f','t','w','a','r','e','\\',
906 'M','i','c','r','o','s','o','f','t','\\','S','y','s','t','e','m','C','e','r',
907 't','i','f','i','c','a','t','e','s','\\','B','o','g','u','s',0 };
908
909 static void testSystemRegStore(void)
910 {
911 HCERTSTORE store, memStore;
912
913 /* Check with a UNICODE name */
914 store = CertOpenStore(CERT_STORE_PROV_SYSTEM_REGISTRY, 0, 0,
915 CERT_SYSTEM_STORE_CURRENT_USER | CERT_STORE_OPEN_EXISTING_FLAG, MyW);
916 /* Not all OSes support CERT_STORE_PROV_SYSTEM_REGISTRY, so don't continue
917 * testing if they don't.
918 */
919 if (!store)
920 return;
921
922 /* Check that it isn't a collection store */
923 memStore = CertOpenStore(CERT_STORE_PROV_MEMORY, 0, 0,
924 CERT_STORE_CREATE_NEW_FLAG, NULL);
925 if (memStore)
926 {
927 if (pCertAddStoreToCollection)
928 {
929 BOOL ret = pCertAddStoreToCollection(store, memStore, 0, 0);
930 ok(!ret && GetLastError() == E_INVALIDARG,
931 "Expected E_INVALIDARG, got %08x\n", GetLastError());
932 }
933 CertCloseStore(memStore, 0);
934 }
935 CertCloseStore(store, 0);
936
937 /* Check opening a bogus store */
938 store = CertOpenStore(CERT_STORE_PROV_SYSTEM_REGISTRY, 0, 0,
939 CERT_SYSTEM_STORE_CURRENT_USER | CERT_STORE_OPEN_EXISTING_FLAG, BogusW);
940 ok(!store && GetLastError() == ERROR_FILE_NOT_FOUND,
941 "Expected ERROR_FILE_NOT_FOUND, got %08x\n", GetLastError());
942 store = CertOpenStore(CERT_STORE_PROV_SYSTEM_REGISTRY, 0, 0,
943 CERT_SYSTEM_STORE_CURRENT_USER, BogusW);
944 ok(store != 0, "CertOpenStore failed: %08x\n", GetLastError());
945 if (store)
946 CertCloseStore(store, 0);
947 /* Now check whether deleting is allowed */
948 store = CertOpenStore(CERT_STORE_PROV_SYSTEM_REGISTRY, 0, 0,
949 CERT_SYSTEM_STORE_CURRENT_USER | CERT_STORE_DELETE_FLAG, BogusW);
950 RegDeleteKeyW(HKEY_CURRENT_USER, BogusPathW);
951
952 store = CertOpenStore(CERT_STORE_PROV_SYSTEM_REGISTRY, 0, 0, 0, NULL);
953 ok(!store && GetLastError() == E_INVALIDARG,
954 "Expected E_INVALIDARG, got %08x\n", GetLastError());
955 store = CertOpenStore(CERT_STORE_PROV_SYSTEM_REGISTRY, 0, 0,
956 CERT_SYSTEM_STORE_LOCAL_MACHINE | CERT_SYSTEM_STORE_CURRENT_USER, MyA);
957 ok(!store && GetLastError() == E_INVALIDARG,
958 "Expected E_INVALIDARG, got %08x\n", GetLastError());
959 store = CertOpenStore(CERT_STORE_PROV_SYSTEM_REGISTRY, 0, 0,
960 CERT_SYSTEM_STORE_LOCAL_MACHINE | CERT_SYSTEM_STORE_CURRENT_USER, MyW);
961 ok(!store && GetLastError() == E_INVALIDARG,
962 "Expected E_INVALIDARG, got %08x\n", GetLastError());
963 /* The name is expected to be UNICODE, check with an ASCII name */
964 store = CertOpenStore(CERT_STORE_PROV_SYSTEM_REGISTRY, 0, 0,
965 CERT_SYSTEM_STORE_CURRENT_USER | CERT_STORE_OPEN_EXISTING_FLAG, MyA);
966 ok(!store && GetLastError() == ERROR_FILE_NOT_FOUND,
967 "Expected ERROR_FILE_NOT_FOUND, got %08x\n", GetLastError());
968 }
969
970 static void testSystemStore(void)
971 {
972 static const WCHAR baskslashW[] = { '\\',0 };
973 HCERTSTORE store;
974 WCHAR keyName[MAX_PATH];
975 HKEY key;
976 LONG rc;
977
978 store = CertOpenStore(CERT_STORE_PROV_SYSTEM, 0, 0, 0, NULL);
979 ok(!store && GetLastError() == ERROR_FILE_NOT_FOUND,
980 "Expected ERROR_FILE_NOT_FOUND, got %08x\n", GetLastError());
981 store = CertOpenStore(CERT_STORE_PROV_SYSTEM, 0, 0,
982 CERT_SYSTEM_STORE_LOCAL_MACHINE | CERT_SYSTEM_STORE_CURRENT_USER, MyA);
983 ok(!store && GetLastError() == ERROR_FILE_NOT_FOUND,
984 "Expected ERROR_FILE_NOT_FOUND, got %08x\n", GetLastError());
985 store = CertOpenStore(CERT_STORE_PROV_SYSTEM, 0, 0,
986 CERT_SYSTEM_STORE_LOCAL_MACHINE | CERT_SYSTEM_STORE_CURRENT_USER, MyW);
987 ok(!store && GetLastError() == ERROR_FILE_NOT_FOUND,
988 "Expected ERROR_FILE_NOT_FOUND, got %08x\n", GetLastError());
989 /* The name is expected to be UNICODE, first check with an ASCII name */
990 store = CertOpenStore(CERT_STORE_PROV_SYSTEM, 0, 0,
991 CERT_SYSTEM_STORE_CURRENT_USER | CERT_STORE_OPEN_EXISTING_FLAG, MyA);
992 ok(!store && GetLastError() == ERROR_FILE_NOT_FOUND,
993 "Expected ERROR_FILE_NOT_FOUND, got %08x\n", GetLastError());
994 /* Create the expected key */
995 lstrcpyW(keyName, CERT_LOCAL_MACHINE_SYSTEM_STORE_REGPATH);
996 lstrcatW(keyName, baskslashW);
997 lstrcatW(keyName, MyW);
998 rc = RegCreateKeyExW(HKEY_CURRENT_USER, keyName, 0, NULL, 0, KEY_READ,
999 NULL, &key, NULL);
1000 ok(!rc, "RegCreateKeyEx failed: %d\n", rc);
1001 if (!rc)
1002 RegCloseKey(key);
1003 /* Check opening with a UNICODE name, specifying the create new flag */
1004 store = CertOpenStore(CERT_STORE_PROV_SYSTEM, 0, 0,
1005 CERT_SYSTEM_STORE_CURRENT_USER | CERT_STORE_CREATE_NEW_FLAG, MyW);
1006 ok(!store && GetLastError() == ERROR_FILE_EXISTS,
1007 "Expected ERROR_FILE_EXISTS, got %08x\n", GetLastError());
1008 /* Now check opening with a UNICODE name, this time opening existing */
1009 store = CertOpenStore(CERT_STORE_PROV_SYSTEM, 0, 0,
1010 CERT_SYSTEM_STORE_CURRENT_USER | CERT_STORE_OPEN_EXISTING_FLAG, MyW);
1011 ok(store != 0, "CertOpenStore failed: %08x\n", GetLastError());
1012 if (store)
1013 {
1014 HCERTSTORE memStore = CertOpenStore(CERT_STORE_PROV_MEMORY, 0, 0,
1015 CERT_STORE_CREATE_NEW_FLAG, NULL);
1016
1017 /* Check that it's a collection store */
1018 if (memStore)
1019 {
1020 if (pCertAddStoreToCollection)
1021 {
1022 BOOL ret = pCertAddStoreToCollection(store, memStore, 0, 0);
1023 /* FIXME: this'll fail on NT4, but what error will it give? */
1024 ok(ret, "CertAddStoreToCollection failed: %08x\n", GetLastError());
1025 }
1026 CertCloseStore(memStore, 0);
1027 }
1028 CertCloseStore(store, 0);
1029 }
1030
1031 /* Check opening a bogus store */
1032 store = CertOpenStore(CERT_STORE_PROV_SYSTEM, 0, 0,
1033 CERT_SYSTEM_STORE_CURRENT_USER | CERT_STORE_OPEN_EXISTING_FLAG, BogusW);
1034 ok((!store ||
1035 broken(store != 0)) && /* win9x */
1036 GetLastError() == ERROR_FILE_NOT_FOUND,
1037 "Expected ERROR_FILE_NOT_FOUND, got %08x\n", GetLastError());
1038 store = CertOpenStore(CERT_STORE_PROV_SYSTEM, 0, 0,
1039 CERT_SYSTEM_STORE_CURRENT_USER, BogusW);
1040 ok(store != 0, "CertOpenStore failed: %08x\n", GetLastError());
1041 if (store)
1042 CertCloseStore(store, 0);
1043 /* Now check whether deleting is allowed */
1044 store = CertOpenStore(CERT_STORE_PROV_SYSTEM, 0, 0,
1045 CERT_SYSTEM_STORE_CURRENT_USER | CERT_STORE_DELETE_FLAG, BogusW);
1046 RegDeleteKeyW(HKEY_CURRENT_USER, BogusPathW);
1047 }
1048
1049 static const BYTE serializedStoreWithCert[] = {
1050 0x00,0x00,0x00,0x00,0x43,0x45,0x52,0x54,0x20,0x00,0x00,0x00,0x01,0x00,0x00,
1051 0x00,0x7c,0x00,0x00,0x00,0x30,0x7a,0x02,0x01,0x01,0x30,0x02,0x06,0x00,0x30,
1052 0x15,0x31,0x13,0x30,0x11,0x06,0x03,0x55,0x04,0x03,0x13,0x0a,0x4a,0x75,0x61,
1053 0x6e,0x20,0x4c,0x61,0x6e,0x67,0x00,0x30,0x22,0x18,0x0f,0x31,0x36,0x30,0x31,
1054 0x30,0x31,0x30,0x31,0x30,0x30,0x30,0x30,0x30,0x30,0x5a,0x18,0x0f,0x31,0x36,
1055 0x30,0x31,0x30,0x31,0x30,0x31,0x30,0x30,0x30,0x30,0x30,0x30,0x5a,0x30,0x15,
1056 0x31,0x13,0x30,0x11,0x06,0x03,0x55,0x04,0x03,0x13,0x0a,0x4a,0x75,0x61,0x6e,
1057 0x20,0x4c,0x61,0x6e,0x67,0x00,0x30,0x07,0x30,0x02,0x06,0x00,0x03,0x01,0x00,
1058 0xa3,0x16,0x30,0x14,0x30,0x12,0x06,0x03,0x55,0x1d,0x13,0x01,0x01,0xff,0x04,
1059 0x08,0x30,0x06,0x01,0x01,0xff,0x02,0x01,0x01,0x00,0x00,0x00,0x00,0x00,0x00,
1060 0x00,0x00,0x00,0x00,0x00,0x00 };
1061 static const BYTE serializedStoreWithCertAndCRL[] = {
1062 0x00,0x00,0x00,0x00,0x43,0x45,0x52,0x54,0x20,0x00,0x00,0x00,0x01,0x00,0x00,
1063 0x00,0x7c,0x00,0x00,0x00,0x30,0x7a,0x02,0x01,0x01,0x30,0x02,0x06,0x00,0x30,
1064 0x15,0x31,0x13,0x30,0x11,0x06,0x03,0x55,0x04,0x03,0x13,0x0a,0x4a,0x75,0x61,
1065 0x6e,0x20,0x4c,0x61,0x6e,0x67,0x00,0x30,0x22,0x18,0x0f,0x31,0x36,0x30,0x31,
1066 0x30,0x31,0x30,0x31,0x30,0x30,0x30,0x30,0x30,0x30,0x5a,0x18,0x0f,0x31,0x36,
1067 0x30,0x31,0x30,0x31,0x30,0x31,0x30,0x30,0x30,0x30,0x30,0x30,0x5a,0x30,0x15,
1068 0x31,0x13,0x30,0x11,0x06,0x03,0x55,0x04,0x03,0x13,0x0a,0x4a,0x75,0x61,0x6e,
1069 0x20,0x4c,0x61,0x6e,0x67,0x00,0x30,0x07,0x30,0x02,0x06,0x00,0x03,0x01,0x00,
1070 0xa3,0x16,0x30,0x14,0x30,0x12,0x06,0x03,0x55,0x1d,0x13,0x01,0x01,0xff,0x04,
1071 0x08,0x30,0x06,0x01,0x01,0xff,0x02,0x01,0x01,0x21,0x00,0x00,0x00,0x01,0x00,
1072 0x00,0x00,0x47,0x00,0x00,0x00,0x30,0x45,0x30,0x2c,0x30,0x02,0x06,0x00,0x30,
1073 0x15,0x31,0x13,0x30,0x11,0x06,0x03,0x55,0x04,0x03,0x13,0x0a,0x4a,0x75,0x61,
1074 0x6e,0x20,0x4c,0x61,0x6e,0x67,0x00,0x18,0x0f,0x31,0x36,0x30,0x31,0x30,0x31,
1075 0x30,0x31,0x30,0x30,0x30,0x30,0x30,0x30,0x5a,0x30,0x02,0x06,0x00,0x03,0x11,
1076 0x00,0x0f,0x0e,0x0d,0x0c,0x0b,0x0a,0x09,0x08,0x07,0x06,0x05,0x04,0x03,0x02,
1077 0x01,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00 };
1078
1079 static void compareFile(LPCWSTR filename, const BYTE *pb, DWORD cb)
1080 {
1081 HANDLE h;
1082 BYTE buf[200];
1083 BOOL ret;
1084 DWORD cbRead = 0, totalRead = 0;
1085
1086 h = CreateFileW(filename, GENERIC_READ, 0, NULL, OPEN_EXISTING,
1087 FILE_ATTRIBUTE_NORMAL, NULL);
1088 if (h == INVALID_HANDLE_VALUE)
1089 return;
1090 do {
1091 ret = ReadFile(h, buf, sizeof(buf), &cbRead, NULL);
1092 if (ret && cbRead)
1093 {
1094 ok(totalRead + cbRead <= cb, "Expected total count %d, see %d\n",
1095 cb, totalRead + cbRead);
1096 ok(!memcmp(pb + totalRead, buf, cbRead),
1097 "Unexpected data in file\n");
1098 totalRead += cbRead;
1099 }
1100 } while (ret && cbRead);
1101 CloseHandle(h);
1102 }
1103
1104 static void testFileStore(void)
1105 {
1106 static const WCHAR szPrefix[] = { 'c','e','r',0 };
1107 static const WCHAR szDot[] = { '.',0 };
1108 WCHAR filename[MAX_PATH];
1109 HCERTSTORE store;
1110 BOOL ret;
1111 PCCERT_CONTEXT cert;
1112 HANDLE file;
1113
1114 if (!pCertControlStore)
1115 {
1116 skip("CertControlStore() is not available\n");
1117 return;
1118 }
1119
1120 store = CertOpenStore(CERT_STORE_PROV_FILE, 0, 0, 0, NULL);
1121 ok(!store && GetLastError() == ERROR_INVALID_HANDLE,
1122 "Expected ERROR_INVALID_HANDLE, got %08x\n", GetLastError());
1123
1124 if (!GetTempFileNameW(szDot, szPrefix, 0, filename))
1125 return;
1126
1127 DeleteFileW(filename);
1128 file = CreateFileW(filename, GENERIC_READ | GENERIC_WRITE, 0, NULL,
1129 CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
1130 if (file == INVALID_HANDLE_VALUE)
1131 return;
1132
1133 store = CertOpenStore(CERT_STORE_PROV_FILE, 0, 0, CERT_STORE_DELETE_FLAG,
1134 file);
1135 ok(!store && GetLastError() == E_INVALIDARG,
1136 "Expected E_INVALIDARG, got %08x\n", GetLastError());
1137 store = CertOpenStore(CERT_STORE_PROV_FILE, 0, 0,
1138 CERT_FILE_STORE_COMMIT_ENABLE_FLAG | CERT_STORE_READONLY_FLAG, file);
1139 ok(!store && GetLastError() == E_INVALIDARG,
1140 "Expected E_INVALIDARG, got %08x\n", GetLastError());
1141
1142 /* A "read-only" file store.. */
1143 store = CertOpenStore(CERT_STORE_PROV_FILE, 0, 0,
1144 CERT_STORE_OPEN_EXISTING_FLAG | CERT_STORE_READONLY_FLAG, file);
1145 ok(store != NULL, "CertOpenStore failed: %08x\n", GetLastError());
1146 if (store)
1147 {
1148 DWORD size;
1149
1150 ret = CertAddEncodedCertificateToStore(store, X509_ASN_ENCODING,
1151 bigCert, sizeof(bigCert), CERT_STORE_ADD_ALWAYS, NULL);
1152 /* apparently allows adding certificates.. */
1153 ok(ret, "CertAddEncodedCertificateToStore failed: %d\n", ret);
1154 /* but not commits.. */
1155 ret = pCertControlStore(store, 0, CERT_STORE_CTRL_COMMIT, NULL);
1156 ok(!ret && GetLastError() == ERROR_CALL_NOT_IMPLEMENTED,
1157 "Expected ERROR_CALL_NOT_IMPLEMENTED, got %08x\n", GetLastError());
1158 /* It still has certs in memory.. */
1159 cert = CertEnumCertificatesInStore(store, NULL);
1160 ok(cert != NULL, "CertEnumCertificatesInStore failed: %08x\n",
1161 GetLastError());
1162 CertFreeCertificateContext(cert);
1163 /* but the file size is still 0. */
1164 size = GetFileSize(file, NULL);
1165 ok(size == 0, "Expected size 0, got %d\n", size);
1166 CertCloseStore(store, 0);
1167 }
1168
1169 /* The create new flag is allowed.. */
1170 store = CertOpenStore(CERT_STORE_PROV_FILE, 0, 0,
1171 CERT_STORE_CREATE_NEW_FLAG, file);
1172 ok(store != NULL, "CertOpenStore failed: %08x\n", GetLastError());
1173 if (store)
1174 {
1175 /* but without the commit enable flag, commits don't happen. */
1176 ret = CertAddEncodedCertificateToStore(store, X509_ASN_ENCODING,
1177 bigCert, sizeof(bigCert), CERT_STORE_ADD_ALWAYS, NULL);
1178 ok(ret, "CertAddEncodedCertificateToStore failed: %d\n", ret);
1179 ret = pCertControlStore(store, 0, CERT_STORE_CTRL_COMMIT, NULL);
1180 ok(!ret && GetLastError() == ERROR_CALL_NOT_IMPLEMENTED,
1181 "Expected ERROR_CALL_NOT_IMPLEMENTED, got %08x\n", GetLastError());
1182 CertCloseStore(store, 0);
1183 }
1184 /* as is the open existing flag. */
1185 store = CertOpenStore(CERT_STORE_PROV_FILE, 0, 0,
1186 CERT_STORE_OPEN_EXISTING_FLAG, file);
1187 ok(store != NULL, "CertOpenStore failed: %08x\n", GetLastError());
1188 if (store)
1189 {
1190 /* but without the commit enable flag, commits don't happen. */
1191 ret = CertAddEncodedCertificateToStore(store, X509_ASN_ENCODING,
1192 bigCert, sizeof(bigCert), CERT_STORE_ADD_ALWAYS, NULL);
1193 ok(ret, "CertAddEncodedCertificateToStore failed: %d\n", ret);
1194 ret = pCertControlStore(store, 0, CERT_STORE_CTRL_COMMIT, NULL);
1195 ok(!ret && GetLastError() == ERROR_CALL_NOT_IMPLEMENTED,
1196 "Expected ERROR_CALL_NOT_IMPLEMENTED, got %08x\n", GetLastError());
1197 CertCloseStore(store, 0);
1198 }
1199 store = CertOpenStore(CERT_STORE_PROV_FILE, 0, 0,
1200 CERT_FILE_STORE_COMMIT_ENABLE_FLAG, file);
1201 ok(store != NULL, "CertOpenStore failed: %08x\n", GetLastError());
1202 if (store)
1203 {
1204 CloseHandle(file);
1205 ret = CertAddEncodedCertificateToStore(store, X509_ASN_ENCODING,
1206 bigCert, sizeof(bigCert), CERT_STORE_ADD_ALWAYS, NULL);
1207 ok(ret, "CertAddEncodedCertificateToStore failed: %08x\n",
1208 GetLastError());
1209 /* with commits enabled, commit is allowed */
1210 ret = pCertControlStore(store, 0, CERT_STORE_CTRL_COMMIT, NULL);
1211 ok(ret, "CertControlStore failed: %d\n", ret);
1212 compareFile(filename, serializedStoreWithCert,
1213 sizeof(serializedStoreWithCert));
1214 CertCloseStore(store, 0);
1215 }
1216 file = CreateFileW(filename, GENERIC_READ | GENERIC_WRITE, 0, NULL,
1217 OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
1218 if (file == INVALID_HANDLE_VALUE)
1219 return;
1220 store = CertOpenStore(CERT_STORE_PROV_FILE, 0, 0,
1221 CERT_FILE_STORE_COMMIT_ENABLE_FLAG, file);
1222 ok(store != NULL, "CertOpenStore failed: %08x\n", GetLastError());
1223 if (store)
1224 {
1225 CloseHandle(file);
1226 ret = CertAddEncodedCRLToStore(store, X509_ASN_ENCODING, signedCRL,
1227 sizeof(signedCRL), CERT_STORE_ADD_ALWAYS, NULL);
1228 ok(ret, "CertAddEncodedCRLToStore failed: %08x\n", GetLastError());
1229 CertCloseStore(store, 0);
1230 compareFile(filename, serializedStoreWithCertAndCRL,
1231 sizeof(serializedStoreWithCertAndCRL));
1232 }
1233
1234 DeleteFileW(filename);
1235 }
1236
1237 static BOOL initFileFromData(LPCWSTR filename, const BYTE *pb, DWORD cb)
1238 {
1239 HANDLE file = CreateFileW(filename, GENERIC_READ | GENERIC_WRITE, 0, NULL,
1240 CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
1241 BOOL ret;
1242
1243 if (file != INVALID_HANDLE_VALUE)
1244 {
1245 DWORD written;
1246
1247 ret = WriteFile(file, pb, cb, &written, NULL);
1248 CloseHandle(file);
1249 }
1250 else
1251 ret = FALSE;
1252 return ret;
1253 }
1254 static void testFileNameStore(void)
1255 {
1256 static const WCHAR szPrefix[] = { 'c','e','r',0 };
1257 static const WCHAR szDot[] = { '.',0 };
1258 WCHAR filename[MAX_PATH];
1259 HCERTSTORE store;
1260 BOOL ret;
1261 DWORD GLE;
1262
1263 if (0)
1264 {
1265 /* Crashes on NT4 */
1266 store = CertOpenStore(CERT_STORE_PROV_FILENAME_W, 0, 0, 0, NULL);
1267 GLE = GetLastError();
1268 ok(!store && (GLE == ERROR_PATH_NOT_FOUND || GLE == ERROR_INVALID_PARAMETER),
1269 "Expected ERROR_PATH_NOT_FOUND or ERROR_INVALID_PARAMETER, got %08x\n",
1270 GLE);
1271 }
1272
1273 if (!GetTempFileNameW(szDot, szPrefix, 0, filename))
1274 return;
1275 DeleteFileW(filename);
1276
1277 /* The two flags are mutually exclusive */
1278 store = CertOpenStore(CERT_STORE_PROV_FILENAME_W, 0, 0,
1279 CERT_FILE_STORE_COMMIT_ENABLE_FLAG | CERT_STORE_READONLY_FLAG, filename);
1280 ok(!store && GetLastError() == E_INVALIDARG,
1281 "Expected E_INVALIDARG, got %08x\n", GetLastError());
1282
1283 /* In all of the following tests, the encoding type seems to be ignored */
1284 if (initFileFromData(filename, bigCert, sizeof(bigCert)))
1285 {
1286 PCCERT_CONTEXT cert;
1287 PCCRL_CONTEXT crl;
1288
1289 store = CertOpenStore(CERT_STORE_PROV_FILENAME_W, 0, 0,
1290 CERT_STORE_READONLY_FLAG, filename);
1291 ok(store != NULL, "CertOpenStore failed: %08x\n", GetLastError());
1292
1293 cert = CertEnumCertificatesInStore(store, NULL);
1294 ok(cert != NULL, "CertEnumCertificatesInStore failed: %08x\n",
1295 GetLastError());
1296 cert = CertEnumCertificatesInStore(store, cert);
1297 ok(!cert, "Expected only one cert\n");
1298 if (pCertEnumCRLsInStore)
1299 {
1300 crl = pCertEnumCRLsInStore(store, NULL);
1301 ok(!crl, "Expected no CRLs\n");
1302 }
1303
1304 CertCloseStore(store, 0);
1305 DeleteFileW(filename);
1306 }
1307 if (initFileFromData(filename, serializedStoreWithCert,
1308 sizeof(serializedStoreWithCert)))
1309 {
1310 PCCERT_CONTEXT cert;
1311 PCCRL_CONTEXT crl;
1312
1313 store = CertOpenStore(CERT_STORE_PROV_FILENAME_W, 0, 0,
1314 CERT_STORE_READONLY_FLAG, filename);
1315 ok(store != NULL, "CertOpenStore failed: %08x\n", GetLastError());
1316
1317 cert = CertEnumCertificatesInStore(store, NULL);
1318 ok(cert != NULL, "CertEnumCertificatesInStore failed: %08x\n",
1319 GetLastError());
1320 cert = CertEnumCertificatesInStore(store, cert);
1321 ok(!cert, "Expected only one cert\n");
1322 if (pCertEnumCRLsInStore)
1323 {
1324 crl = pCertEnumCRLsInStore(store, NULL);
1325 ok(!crl, "Expected no CRLs\n");
1326 }
1327
1328 CertCloseStore(store, 0);
1329 DeleteFileW(filename);
1330 }
1331 if (initFileFromData(filename, serializedStoreWithCertAndCRL,
1332 sizeof(serializedStoreWithCertAndCRL)))
1333 {
1334 PCCERT_CONTEXT cert;
1335 PCCRL_CONTEXT crl;
1336
1337 store = CertOpenStore(CERT_STORE_PROV_FILENAME_W, 0, 0,
1338 CERT_STORE_READONLY_FLAG, filename);
1339 ok(store != NULL, "CertOpenStore failed: %08x\n", GetLastError());
1340
1341 cert = CertEnumCertificatesInStore(store, NULL);
1342 ok(cert != NULL, "CertEnumCertificatesInStore failed: %08x\n",
1343 GetLastError());
1344 cert = CertEnumCertificatesInStore(store, cert);
1345 ok(!cert, "Expected only one cert\n");
1346 if (pCertEnumCRLsInStore)
1347 {
1348 crl = pCertEnumCRLsInStore(store, NULL);
1349 ok(crl != NULL, "CertEnumCRLsInStore failed: %08x\n", GetLastError());
1350 crl = pCertEnumCRLsInStore(store, crl);
1351 ok(!crl, "Expected only one CRL\n");
1352 }
1353
1354 CertCloseStore(store, 0);
1355 /* Don't delete it this time, the next test uses it */
1356 }
1357 /* Now that the file exists, we can open it read-only */
1358 store = CertOpenStore(CERT_STORE_PROV_FILENAME_W, 0, 0,
1359 CERT_STORE_READONLY_FLAG, filename);
1360 ok(store != NULL, "CertOpenStore failed: %08x\n", GetLastError());
1361 CertCloseStore(store, 0);
1362 DeleteFileW(filename);
1363
1364 store = CertOpenStore(CERT_STORE_PROV_FILENAME_W, 0, 0,
1365 CERT_FILE_STORE_COMMIT_ENABLE_FLAG | CERT_STORE_CREATE_NEW_FLAG, filename);
1366 ok(store != NULL, "CertOpenStore failed: %08x\n", GetLastError());
1367 if (store)
1368 {
1369 ret = CertAddEncodedCertificateToStore(store, X509_ASN_ENCODING,
1370 bigCert, sizeof(bigCert), CERT_STORE_ADD_ALWAYS, NULL);
1371 ok(ret, "CertAddEncodedCertificateToStore failed: %08x\n",
1372 GetLastError());
1373 CertCloseStore(store, 0);
1374 compareFile(filename, serializedStoreWithCert,
1375 sizeof(serializedStoreWithCert));
1376 }
1377 store = CertOpenStore(CERT_STORE_PROV_FILENAME_W, 0, 0,
1378 CERT_FILE_STORE_COMMIT_ENABLE_FLAG, filename);
1379 ok(store != NULL, "CertOpenStore failed: %08x\n", GetLastError());
1380 if (store)
1381 {
1382 ret = CertAddEncodedCRLToStore(store, X509_ASN_ENCODING,
1383 signedCRL, sizeof(signedCRL), CERT_STORE_ADD_ALWAYS, NULL);
1384 ok(ret, "CertAddEncodedCRLToStore failed: %08x\n", GetLastError());
1385 CertCloseStore(store, 0);
1386 compareFile(filename, serializedStoreWithCertAndCRL,
1387 sizeof(serializedStoreWithCertAndCRL));
1388 }
1389 DeleteFileW(filename);
1390 }
1391
1392 static const BYTE signedContent[] = {
1393 0x30,0x81,0xb2,0x06,0x09,0x2a,0x86,0x48,0x86,0xf7,0x0d,0x01,0x07,0x02,0xa0,
1394 0x81,0xa4,0x30,0x81,0xa1,0x02,0x01,0x01,0x31,0x0e,0x30,0x0c,0x06,0x08,0x2a,
1395 0x86,0x48,0x86,0xf7,0x0d,0x02,0x05,0x05,0x00,0x30,0x13,0x06,0x09,0x2a,0x86,
1396 0x48,0x86,0xf7,0x0d,0x01,0x07,0x01,0xa0,0x06,0x04,0x04,0x01,0x02,0x03,0x04,
1397 0x31,0x77,0x30,0x75,0x02,0x01,0x01,0x30,0x1a,0x30,0x15,0x31,0x13,0x30,0x11,
1398 0x06,0x03,0x55,0x04,0x03,0x13,0x0a,0x4a,0x75,0x61,0x6e,0x20,0x4c,0x61,0x6e,
1399 0x67,0x00,0x02,0x01,0x01,0x30,0x0c,0x06,0x08,0x2a,0x86,0x48,0x86,0xf7,0x0d,
1400 0x02,0x05,0x05,0x00,0x30,0x04,0x06,0x00,0x05,0x00,0x04,0x40,0x81,0xa6,0x70,
1401 0xb3,0xef,0x59,0xd1,0x66,0xd1,0x9b,0xc0,0x9a,0xb6,0x9a,0x5e,0x6d,0x6f,0x6d,
1402 0x0d,0x59,0xa9,0xaa,0x6e,0xe9,0x2c,0xa0,0x1e,0xee,0xc2,0x60,0xbc,0x59,0xbe,
1403 0x3f,0x63,0x06,0x8d,0xc9,0x11,0x1d,0x23,0x64,0x92,0xef,0x2e,0xfc,0x57,0x29,
1404 0xa4,0xaf,0xe0,0xee,0x93,0x19,0x39,0x51,0xe4,0x44,0xb8,0x0b,0x28,0xf4,0xa8,
1405 0x0d };
1406 static const BYTE signedWithCertAndCrlBareContent[] = {
1407 0x30,0x82,0x01,0x4f,0x02,0x01,0x01,0x31,0x0e,0x30,0x0c,0x06,0x08,0x2a,0x86,
1408 0x48,0x86,0xf7,0x0d,0x02,0x05,0x05,0x00,0x30,0x13,0x06,0x09,0x2a,0x86,0x48,
1409 0x86,0xf7,0x0d,0x01,0x07,0x01,0xa0,0x06,0x04,0x04,0x01,0x02,0x03,0x04,0xa0,
1410 0x7c,0x30,0x7a,0x02,0x01,0x01,0x30,0x02,0x06,0x00,0x30,0x15,0x31,0x13,0x30,
1411 0x11,0x06,0x03,0x55,0x04,0x03,0x13,0x0a,0x4a,0x75,0x61,0x6e,0x20,0x4c,0x61,
1412 0x6e,0x67,0x00,0x30,0x22,0x18,0x0f,0x31,0x36,0x30,0x31,0x30,0x31,0x30,0x31,
1413 0x30,0x30,0x30,0x30,0x30,0x30,0x5a,0x18,0x0f,0x31,0x36,0x30,0x31,0x30,0x31,
1414 0x30,0x31,0x30,0x30,0x30,0x30,0x30,0x30,0x5a,0x30,0x15,0x31,0x13,0x30,0x11,
1415 0x06,0x03,0x55,0x04,0x03,0x13,0x0a,0x4a,0x75,0x61,0x6e,0x20,0x4c,0x61,0x6e,
1416 0x67,0x00,0x30,0x07,0x30,0x02,0x06,0x00,0x03,0x01,0x00,0xa3,0x16,0x30,0x14,
1417 0x30,0x12,0x06,0x03,0x55,0x1d,0x13,0x01,0x01,0xff,0x04,0x08,0x30,0x06,0x01,
1418 0x01,0xff,0x02,0x01,0x01,0xa1,0x2e,0x30,0x2c,0x30,0x02,0x06,0x00,0x30,0x15,
1419 0x31,0x13,0x30,0x11,0x06,0x03,0x55,0x04,0x03,0x13,0x0a,0x4a,0x75,0x61,0x6e,
1420 0x20,0x4c,0x61,0x6e,0x67,0x00,0x18,0x0f,0x31,0x36,0x30,0x31,0x30,0x31,0x30,
1421 0x31,0x30,0x30,0x30,0x30,0x30,0x30,0x5a,0x31,0x77,0x30,0x75,0x02,0x01,0x01,
1422 0x30,0x1a,0x30,0x15,0x31,0x13,0x30,0x11,0x06,0x03,0x55,0x04,0x03,0x13,0x0a,
1423 0x4a,0x75,0x61,0x6e,0x20,0x4c,0x61,0x6e,0x67,0x00,0x02,0x01,0x01,0x30,0x0c,
1424 0x06,0x08,0x2a,0x86,0x48,0x86,0xf7,0x0d,0x02,0x05,0x05,0x00,0x30,0x04,0x06,
1425 0x00,0x05,0x00,0x04,0x40,0x81,0xa6,0x70,0xb3,0xef,0x59,0xd1,0x66,0xd1,0x9b,
1426 0xc0,0x9a,0xb6,0x9a,0x5e,0x6d,0x6f,0x6d,0x0d,0x59,0xa9,0xaa,0x6e,0xe9,0x2c,
1427 0xa0,0x1e,0xee,0xc2,0x60,0xbc,0x59,0xbe,0x3f,0x63,0x06,0x8d,0xc9,0x11,0x1d,
1428 0x23,0x64,0x92,0xef,0x2e,0xfc,0x57,0x29,0xa4,0xaf,0xe0,0xee,0x93,0x19,0x39,
1429 0x51,0xe4,0x44,0xb8,0x0b,0x28,0xf4,0xa8,0x0d };
1430 static const BYTE hashContent[] = {
1431 0x30,0x47,0x06,0x09,0x2a,0x86,0x48,0x86,0xf7,0x0d,0x01,0x07,0x05,0xa0,0x3a,
1432 0x30,0x38,0x02,0x01,0x00,0x30,0x0c,0x06,0x08,0x2a,0x86,0x48,0x86,0xf7,0x0d,
1433 0x02,0x05,0x05,0x00,0x30,0x13,0x06,0x09,0x2a,0x86,0x48,0x86,0xf7,0x0d,0x01,
1434 0x07,0x01,0xa0,0x06,0x04,0x04,0x01,0x02,0x03,0x04,0x04,0x10,0x08,0xd6,0xc0,
1435 0x5a,0x21,0x51,0x2a,0x79,0xa1,0xdf,0xeb,0x9d,0x2a,0x8f,0x26,0x2f };
1436 static const BYTE hashBareContent[] = {
1437 0x30,0x38,0x02,0x01,0x00,0x30,0x0c,0x06,0x08,0x2a,0x86,0x48,0x86,0xf7,0x0d,
1438 0x02,0x05,0x05,0x00,0x30,0x13,0x06,0x09,0x2a,0x86,0x48,0x86,0xf7,0x0d,0x01,
1439 0x07,0x01,0xa0,0x06,0x04,0x04,0x01,0x02,0x03,0x04,0x04,0x10,0x08,0xd6,0xc0,
1440 0x5a,0x21,0x51,0x2a,0x79,0xa1,0xdf,0xeb,0x9d,0x2a,0x8f,0x26,0x2f };
1441
1442 static void testMessageStore(void)
1443 {
1444 HCERTSTORE store;
1445 HCRYPTMSG msg = CryptMsgOpenToDecode(PKCS_7_ASN_ENCODING, 0, 0, 0, NULL,
1446 NULL);
1447 CRYPT_DATA_BLOB blob = { sizeof(signedWithCertAndCrlBareContent),
1448 (LPBYTE)signedWithCertAndCrlBareContent };
1449 DWORD count, size;
1450 BOOL ret;
1451
1452 /* Crashes
1453 store = CertOpenStore(CERT_STORE_PROV_MSG, 0, 0, 0, NULL);
1454 */
1455 SetLastError(0xdeadbeef);
1456 store = CertOpenStore(CERT_STORE_PROV_MSG, 0, 0, 0, msg);
1457 ok(!store && GetLastError() == CRYPT_E_INVALID_MSG_TYPE,
1458 "Expected CRYPT_E_INVALID_MSG_TYPE, got %08x\n", GetLastError());
1459 CryptMsgUpdate(msg, signedContent, sizeof(signedContent), TRUE);
1460 store = CertOpenStore(CERT_STORE_PROV_MSG, 0, 0, 0, msg);
1461 ok(store != NULL, "CertOpenStore failed: %08x\n", GetLastError());
1462 if (store)
1463 {
1464 PCCERT_CONTEXT cert = NULL;
1465 PCCRL_CONTEXT crl = NULL;
1466
1467 count = 0;
1468 do {
1469 cert = CertEnumCertificatesInStore(store, cert);
1470 if (cert)
1471 count++;
1472 } while (cert);
1473 ok(count == 0, "Expected 0 certificates, got %d\n", count);
1474
1475 if (pCertEnumCRLsInStore)
1476 {
1477 count = 0;
1478 do {
1479 crl = pCertEnumCRLsInStore(store, crl);
1480 if (crl)
1481 count++;
1482 } while (crl);
1483 ok(count == 0, "Expected 0 CRLs, got %d\n", count);
1484 }
1485
1486 /* Can add certs to a message store */
1487 ret = CertAddEncodedCertificateToStore(store, X509_ASN_ENCODING,
1488 bigCert, sizeof(bigCert), CERT_STORE_ADD_ALWAYS, NULL);
1489 ok(ret, "CertAddEncodedCertificateToStore failed: %08x\n",
1490 GetLastError());
1491 count = 0;
1492 do {
1493 cert = CertEnumCertificatesInStore(store, cert);
1494 if (cert)
1495 count++;
1496 } while (cert);
1497 ok(count == 1, "Expected 1 certificate, got %d\n", count);
1498
1499 CertCloseStore(store, 0);
1500 }
1501 /* but the added certs weren't actually added to the message */
1502 size = sizeof(count);
1503 ret = CryptMsgGetParam(msg, CMSG_CERT_COUNT_PARAM, 0, &count, &size);
1504 ok(ret, "CryptMsgGetParam failed: %08x\n", GetLastError());
1505 ok(count == 0, "Expected 0 certificates, got %d\n", count);
1506 CryptMsgClose(msg);
1507
1508 /* Crashes
1509 store = CertOpenStore(CERT_STORE_PROV_PKCS7, 0, 0, 0, NULL);
1510 */
1511 store = CertOpenStore(CERT_STORE_PROV_PKCS7, 0, 0, 0, &blob);
1512 ok(store != NULL, "CertOpenStore failed: %08x\n", GetLastError());
1513 if (store)
1514 {
1515 DWORD count = 0;
1516 PCCERT_CONTEXT cert = NULL;
1517 PCCRL_CONTEXT crl = NULL;
1518
1519 do {
1520 cert = CertEnumCertificatesInStore(store, cert);
1521 if (cert)
1522 count++;
1523 } while (cert);
1524 ok(count == 1, "Expected 1 certificate, got %d\n", count);
1525
1526 if (pCertEnumCRLsInStore)
1527 {
1528 count = 0;
1529 do {
1530 crl = pCertEnumCRLsInStore(store, crl);
1531 if (crl)
1532 count++;
1533 } while (crl);
1534 ok(count == 1, "Expected 1 CRL, got %d\n", count);
1535 }
1536 CertCloseStore(store, 0);
1537 }
1538 /* Encoding appears to be ignored */
1539 store = CertOpenStore(CERT_STORE_PROV_PKCS7, X509_ASN_ENCODING, 0, 0,
1540 &blob);
1541 ok(store != NULL, "CertOpenStore failed: %08x\n", GetLastError());
1542 if (store)
1543 CertCloseStore(store, 0);
1544 /* Messages other than signed messages aren't allowed */
1545 blob.cbData = sizeof(hashContent);
1546 blob.pbData = (LPBYTE)hashContent;
1547 SetLastError(0xdeadbeef);
1548 store = CertOpenStore(CERT_STORE_PROV_PKCS7, 0, 0, 0, &blob);
1549 ok(!store && GetLastError() == CRYPT_E_INVALID_MSG_TYPE,
1550 "Expected CRYPT_E_INVALID_MSG_TYPE, got %08x\n", GetLastError());
1551 blob.cbData = sizeof(hashBareContent);
1552 blob.pbData = (LPBYTE)hashBareContent;
1553 SetLastError(0xdeadbeef);
1554 store = CertOpenStore(CERT_STORE_PROV_PKCS7, 0, 0, 0, &blob);
1555 ok(!store &&
1556 (GetLastError() == CRYPT_E_ASN1_BADTAG ||
1557 GetLastError() == OSS_DATA_ERROR), /* win9x */
1558 "Expected CRYPT_E_ASN1_BADTAG, got %08x\n", GetLastError());
1559 }
1560
1561 static void testCertOpenSystemStore(void)
1562 {
1563 HCERTSTORE store;
1564
1565 store = CertOpenSystemStoreW(0, NULL);
1566 ok(!store && GetLastError() == E_INVALIDARG,
1567 "Expected E_INVALIDARG, got %08x\n", GetLastError());
1568 /* This succeeds, and on WinXP at least, the Bogus key is created under
1569 * HKCU (but not under HKLM, even when run as an administrator.)
1570 */
1571 store = CertOpenSystemStoreW(0, BogusW);
1572 ok(store != 0, "CertOpenSystemStore failed: %08x\n", GetLastError());
1573 if (store)
1574 CertCloseStore(store, 0);
1575 /* Delete it so other tests succeed next time around */
1576 store = CertOpenStore(CERT_STORE_PROV_SYSTEM, 0, 0,
1577 CERT_SYSTEM_STORE_CURRENT_USER | CERT_STORE_DELETE_FLAG, BogusW);
1578 RegDeleteKeyW(HKEY_CURRENT_USER, BogusPathW);
1579 }
1580
1581 struct EnumSystemStoreInfo
1582 {
1583 BOOL goOn;
1584 DWORD storeCount;
1585 };
1586
1587 static BOOL CALLBACK enumSystemStoreCB(const void *systemStore, DWORD dwFlags,
1588 PCERT_SYSTEM_STORE_INFO pStoreInfo, void *pvReserved, void *pvArg)
1589 {
1590 struct EnumSystemStoreInfo *info = (struct EnumSystemStoreInfo *)pvArg;
1591
1592 info->storeCount++;
1593 return info->goOn;
1594 }
1595
1596 static void testCertEnumSystemStore(void)
1597 {
1598 BOOL ret;
1599 struct EnumSystemStoreInfo info = { FALSE, 0 };
1600
1601 if (!pCertEnumSystemStore)
1602 {
1603 skip("CertEnumSystemStore() is not available\n");
1604 return;
1605 }
1606
1607 SetLastError(0xdeadbeef);
1608 ret = pCertEnumSystemStore(0, NULL, NULL, NULL);
1609 ok(!ret && GetLastError() == ERROR_FILE_NOT_FOUND,
1610 "Expected ERROR_FILE_NOT_FOUND, got %08x\n", GetLastError());
1611 /* Crashes
1612 ret = pCertEnumSystemStore(CERT_SYSTEM_STORE_LOCAL_MACHINE, NULL, NULL,
1613 NULL);
1614 */
1615
1616 SetLastError(0xdeadbeef);
1617 ret = pCertEnumSystemStore(CERT_SYSTEM_STORE_LOCAL_MACHINE, NULL, &info,
1618 enumSystemStoreCB);
1619 /* Callback returning FALSE stops enumeration */
1620 ok(!ret, "Expected CertEnumSystemStore to stop\n");
1621 ok(info.storeCount == 0 || info.storeCount == 1,
1622 "Expected 0 or 1 stores\n");
1623
1624 info.goOn = TRUE;
1625 info.storeCount = 0;
1626 ret = pCertEnumSystemStore(CERT_SYSTEM_STORE_LOCAL_MACHINE, NULL, &info,
1627 enumSystemStoreCB);
1628 ok(ret, "CertEnumSystemStore failed: %08x\n", GetLastError());
1629 /* There should always be at least My, Root, and CA stores */
1630 ok(info.storeCount == 0 || info.storeCount >= 3,
1631 "Expected at least 3 stores\n");
1632 }
1633
1634 static void testStoreProperty(void)
1635 {
1636 HCERTSTORE store;
1637 BOOL ret;
1638 DWORD propID, size = 0, state;
1639 CRYPT_DATA_BLOB blob;
1640
1641 if (!pCertGetStoreProperty || !pCertSetStoreProperty)
1642 {
1643 skip("CertGet/SetStoreProperty() is not available\n");
1644 return;
1645 }
1646
1647 /* Crash
1648 ret = pCertGetStoreProperty(NULL, 0, NULL, NULL);
1649 ret = pCertGetStoreProperty(NULL, 0, NULL, &size);
1650 ret = pCertGetStoreProperty(store, 0, NULL, NULL);
1651 */
1652
1653 store = CertOpenStore(CERT_STORE_PROV_MEMORY, 0, 0,
1654 CERT_STORE_CREATE_NEW_FLAG, NULL);
1655 /* Check a missing prop ID */
1656 SetLastError(0xdeadbeef);
1657 ret = pCertGetStoreProperty(store, 0, NULL, &size);
1658 ok(!ret && GetLastError() == CRYPT_E_NOT_FOUND,
1659 "Expected CRYPT_E_NOT_FOUND, got %08x\n", GetLastError());
1660 /* Contrary to MSDN, CERT_ACCESS_STATE_PROP_ID is supported for stores.. */
1661 size = sizeof(state);
1662 ret = pCertGetStoreProperty(store, CERT_ACCESS_STATE_PROP_ID, &state, &size);
1663 ok(ret, "CertGetStoreProperty failed for CERT_ACCESS_STATE_PROP_ID: %08x\n",
1664 GetLastError());
1665 ok(!state, "Expected a non-persisted store\n");
1666 /* and CERT_STORE_LOCALIZED_NAME_PROP_ID isn't supported by default. */
1667 size = 0;
1668 ret = pCertGetStoreProperty(store, CERT_STORE_LOCALIZED_NAME_PROP_ID, NULL,
1669 &size);
1670 ok(!ret && GetLastError() == CRYPT_E_NOT_FOUND,
1671 "Expected CRYPT_E_NOT_FOUND, got %08x\n", GetLastError());
1672 /* Delete an arbitrary property on a store */
1673 ret = pCertSetStoreProperty(store, CERT_FIRST_USER_PROP_ID, 0, NULL);
1674 ok(ret, "CertSetStoreProperty failed: %08x\n", GetLastError());
1675 /* Set an arbitrary property on a store */
1676 blob.pbData = (LPBYTE)&state;
1677 blob.cbData = sizeof(state);
1678 ret = pCertSetStoreProperty(store, CERT_FIRST_USER_PROP_ID, 0, &blob);
1679 ok(ret, "CertSetStoreProperty failed: %08x\n", GetLastError());
1680 /* Get an arbitrary property that's been set */
1681 ret = pCertGetStoreProperty(store, CERT_FIRST_USER_PROP_ID, NULL, &size);
1682 ok(ret, "CertGetStoreProperty failed: %08x\n", GetLastError());
1683 ok(size == sizeof(state), "Unexpected data size %d\n", size);
1684 ret = pCertGetStoreProperty(store, CERT_FIRST_USER_PROP_ID, &propID, &size);
1685 ok(ret, "CertGetStoreProperty failed: %08x\n", GetLastError());
1686 ok(propID == state, "CertGetStoreProperty got the wrong value\n");
1687 /* Delete it again */
1688 ret = pCertSetStoreProperty(store, CERT_FIRST_USER_PROP_ID, 0, NULL);
1689 ok(ret, "CertSetStoreProperty failed: %08x\n", GetLastError());
1690 /* And check that it's missing */
1691 SetLastError(0xdeadbeef);
1692 ret = pCertGetStoreProperty(store, CERT_FIRST_USER_PROP_ID, NULL, &size);
1693 ok(!ret && GetLastError() == CRYPT_E_NOT_FOUND,
1694 "Expected CRYPT_E_NOT_FOUND, got %08x\n", GetLastError());
1695 CertCloseStore(store, 0);
1696
1697 /* Recheck on the My store.. */
1698 store = CertOpenSystemStoreW(0, MyW);
1699 size = sizeof(state);
1700 ret = pCertGetStoreProperty(store, CERT_ACCESS_STATE_PROP_ID, &state, &size);
1701 ok(ret, "CertGetStoreProperty failed for CERT_ACCESS_STATE_PROP_ID: %08x\n",
1702 GetLastError());
1703 ok(state, "Expected a persisted store\n");
1704 SetLastError(0xdeadbeef);
1705 size = 0;
1706 ret = pCertGetStoreProperty(store, CERT_STORE_LOCALIZED_NAME_PROP_ID, NULL,
1707 &size);
1708 ok(!ret && GetLastError() == CRYPT_E_NOT_FOUND,
1709 "Expected CRYPT_E_NOT_FOUND, got %08x\n", GetLastError());
1710 CertCloseStore(store, 0);
1711 }
1712
1713 static void testAddSerialized(void)
1714 {
1715 BOOL ret;
1716 HCERTSTORE store;
1717 BYTE buf[sizeof(struct CertPropIDHeader) * 2 + 20 + sizeof(bigCert)] =
1718 { 0 };
1719 BYTE hash[20];
1720 struct CertPropIDHeader *hdr;
1721 PCCERT_CONTEXT context;
1722
1723 ret = CertAddSerializedElementToStore(0, NULL, 0, 0, 0, 0, NULL, NULL);
1724 ok(!ret && GetLastError() == ERROR_END_OF_MEDIA,
1725 "Expected ERROR_END_OF_MEDIA, got %08x\n", GetLastError());
1726
1727 store = CertOpenStore(CERT_STORE_PROV_MEMORY, 0, 0,
1728 CERT_STORE_CREATE_NEW_FLAG, NULL);
1729 ok(store != 0, "CertOpenStore failed: %08x\n", GetLastError());
1730
1731 ret = CertAddSerializedElementToStore(store, NULL, 0, 0, 0, 0, NULL, NULL);
1732 ok(!ret && GetLastError() == ERROR_END_OF_MEDIA,
1733 "Expected ERROR_END_OF_MEDIA, got %08x\n", GetLastError());
1734
1735 /* Test with an empty property */
1736 hdr = (struct CertPropIDHeader *)buf;
1737 hdr->propID = CERT_CERT_PROP_ID;
1738 hdr->unknown1 = 1;
1739 hdr->cb = 0;
1740 ret = CertAddSerializedElementToStore(store, buf, sizeof(buf), 0, 0, 0,
1741 NULL, NULL);
1742 ok(!ret && GetLastError() == E_INVALIDARG,
1743 "Expected E_INVALIDARG, got %08x\n", GetLastError());
1744 /* Test with a bad size in property header */
1745 hdr->cb = sizeof(bigCert) - 1;
1746 memcpy(buf + sizeof(struct CertPropIDHeader), bigCert, sizeof(bigCert));
1747 ret = CertAddSerializedElementToStore(store, buf, sizeof(buf), 0, 0, 0,
1748 NULL, NULL);
1749 ok(!ret && GetLastError() == E_INVALIDARG,
1750 "Expected E_INVALIDARG, got %08x\n", GetLastError());
1751 ret = CertAddSerializedElementToStore(store, buf,
1752 sizeof(struct CertPropIDHeader) + sizeof(bigCert), 0, 0, 0, NULL,
1753 NULL);
1754 ok(!ret && GetLastError() == E_INVALIDARG,
1755 "Expected E_INVALIDARG, got %08x\n", GetLastError());
1756 ret = CertAddSerializedElementToStore(store, buf,
1757 sizeof(struct CertPropIDHeader) + sizeof(bigCert), CERT_STORE_ADD_NEW,
1758 0, 0, NULL, NULL);
1759 ok(!ret && GetLastError() == E_INVALIDARG,
1760 "Expected E_INVALIDARG, got %08x\n", GetLastError());
1761 /* Kosher size in property header, but no context type */
1762 hdr->cb = sizeof(bigCert);
1763 ret = CertAddSerializedElementToStore(store, buf, sizeof(buf), 0, 0, 0,
1764 NULL, NULL);
1765 ok(!ret && GetLastError() == E_INVALIDARG,
1766 "Expected E_INVALIDARG, got %08x\n", GetLastError());
1767 ret = CertAddSerializedElementToStore(store, buf,
1768 sizeof(struct CertPropIDHeader) + sizeof(bigCert), 0, 0, 0, NULL,
1769 NULL);
1770 ok(!ret && GetLastError() == E_INVALIDARG,
1771 "Expected E_INVALIDARG, got %08x\n", GetLastError());
1772 ret = CertAddSerializedElementToStore(store, buf,
1773 sizeof(struct CertPropIDHeader) + sizeof(bigCert), CERT_STORE_ADD_NEW,
1774 0, 0, NULL, NULL);
1775 ok(!ret && GetLastError() == E_INVALIDARG,
1776 "Expected E_INVALIDARG, got %08x\n", GetLastError());
1777 /* With a bad context type */
1778 ret = CertAddSerializedElementToStore(store, buf, sizeof(buf), 0, 0,
1779 CERT_STORE_CRL_CONTEXT_FLAG, NULL, NULL);
1780 ok(!ret && GetLastError() == E_INVALIDARG,
1781 "Expected E_INVALIDARG, got %08x\n", GetLastError());
1782 ret = CertAddSerializedElementToStore(store, buf,
1783 sizeof(struct CertPropIDHeader) + sizeof(bigCert), 0, 0,
1784 CERT_STORE_CRL_CONTEXT_FLAG, NULL, NULL);
1785 ok(!ret && GetLastError() == E_INVALIDARG,
1786 "Expected E_INVALIDARG, got %08x\n", GetLastError());
1787 ret = CertAddSerializedElementToStore(store, buf,
1788 sizeof(struct CertPropIDHeader) + sizeof(bigCert), CERT_STORE_ADD_NEW,
1789 0, CERT_STORE_CRL_CONTEXT_FLAG, NULL, NULL);
1790 ok(!ret && GetLastError() == E_INVALIDARG,
1791 "Expected E_INVALIDARG, got %08x\n", GetLastError());
1792 /* Bad unknown field, good type */
1793 hdr->unknown1 = 2;
1794 ret = CertAddSerializedElementToStore(store, buf, sizeof(buf), 0, 0,
1795 CERT_STORE_CERTIFICATE_CONTEXT_FLAG, NULL, NULL);
1796 ok(!ret && GetLastError() == ERROR_FILE_NOT_FOUND,
1797 "Expected ERROR_FILE_NOT_FOUND got %08x\n", GetLastError());
1798 ret = CertAddSerializedElementToStore(store, buf,
1799 sizeof(struct CertPropIDHeader) + sizeof(bigCert), 0, 0,
1800 CERT_STORE_CERTIFICATE_CONTEXT_FLAG, NULL, NULL);
1801 ok(!ret && GetLastError() == ERROR_FILE_NOT_FOUND,
1802 "Expected ERROR_FILE_NOT_FOUND got %08x\n", GetLastError());
1803 ret = CertAddSerializedElementToStore(store, buf,
1804 sizeof(struct CertPropIDHeader) + sizeof(bigCert), CERT_STORE_ADD_NEW,
1805 0, CERT_STORE_CERTIFICATE_CONTEXT_FLAG, NULL, NULL);
1806 ok(!ret && GetLastError() == ERROR_FILE_NOT_FOUND,
1807 "Expected ERROR_FILE_NOT_FOUND got %08x\n", GetLastError());
1808 /* Most everything okay, but bad add disposition */
1809 hdr->unknown1 = 1;
1810 /* This crashes
1811 ret = CertAddSerializedElementToStore(store, buf, sizeof(buf), 0, 0,
1812 CERT_STORE_CERTIFICATE_CONTEXT_FLAG, NULL, NULL);
1813 * as does this
1814 ret = CertAddSerializedElementToStore(store, buf,
1815 sizeof(struct CertPropIDHeader) + sizeof(bigCert), 0, 0,
1816 CERT_STORE_CERTIFICATE_CONTEXT_FLAG, NULL, NULL);
1817 */
1818 /* Everything okay, but buffer's too big */
1819 ret = CertAddSerializedElementToStore(store, buf, sizeof(buf),
1820 CERT_STORE_ADD_NEW, 0, CERT_STORE_CERTIFICATE_CONTEXT_FLAG, NULL, NULL);
1821 ok(ret, "CertAddSerializedElementToStore failed: %08x\n", GetLastError());
1822 /* Everything okay, check it's not re-added */
1823 ret = CertAddSerializedElementToStore(store, buf,
1824 sizeof(struct CertPropIDHeader) + sizeof(bigCert), CERT_STORE_ADD_NEW,
1825 0, CERT_STORE_CERTIFICATE_CONTEXT_FLAG, NULL, NULL);
1826 ok(!ret && GetLastError() == CRYPT_E_EXISTS,
1827 "Expected CRYPT_E_EXISTS, got %08x\n", GetLastError());
1828
1829 context = CertEnumCertificatesInStore(store, NULL);
1830 ok(context != NULL, "Expected a cert\n");
1831 if (context)
1832 CertDeleteCertificateFromStore(context);
1833
1834 /* Try adding with a bogus hash. Oddly enough, it succeeds, and the hash,
1835 * when queried, is the real hash rather than the bogus hash.
1836 */
1837 hdr = (struct CertPropIDHeader *)(buf + sizeof(struct CertPropIDHeader) +
1838 sizeof(bigCert));
1839 hdr->propID = CERT_HASH_PROP_ID;
1840 hdr->unknown1 = 1;
1841 hdr->cb = sizeof(hash);
1842 memset(hash, 0xc, sizeof(hash));
1843 memcpy((LPBYTE)hdr + sizeof(struct CertPropIDHeader), hash, sizeof(hash));
1844 ret = CertAddSerializedElementToStore(store, buf, sizeof(buf),
1845 CERT_STORE_ADD_NEW, 0, CERT_STORE_CERTIFICATE_CONTEXT_FLAG, NULL,
1846 (const void **)&context);
1847 ok(ret, "CertAddSerializedElementToStore failed: %08x\n", GetLastError());
1848 if (context)
1849 {
1850 BYTE hashVal[20], realHash[20];
1851 DWORD size = sizeof(hashVal);
1852
1853 ret = CryptHashCertificate(0, 0, 0, bigCert, sizeof(bigCert),
1854 realHash, &size);
1855 ok(ret, "CryptHashCertificate failed: %08x\n", GetLastError());
1856 ret = CertGetCertificateContextProperty(context, CERT_HASH_PROP_ID,
1857 hashVal, &size);
1858 ok(ret, "CertGetCertificateContextProperty failed: %08x\n",
1859 GetLastError());
1860 ok(!memcmp(hashVal, realHash, size), "Unexpected hash\n");
1861 CertFreeCertificateContext(context);
1862 }
1863
1864 CertCloseStore(store, 0);
1865 }
1866
1867 static DWORD countCertsInStore(HCERTSTORE store)
1868 {
1869 PCCERT_CONTEXT cert = NULL;
1870 DWORD certs = 0;
1871
1872 do {
1873 cert = CertEnumCertificatesInStore(store, cert);
1874 if (cert)
1875 certs++;
1876 } while (cert);
1877 return certs;
1878 }
1879
1880 static DWORD countCRLsInStore(HCERTSTORE store)
1881 {
1882 PCCRL_CONTEXT crl = NULL;
1883 DWORD crls = 0;
1884
1885 do {
1886 crl = pCertEnumCRLsInStore(store, crl);
1887 if (crl)
1888 crls++;
1889 } while (crl);
1890 return crls;
1891 }
1892
1893 static void test_I_UpdateStore(void)
1894 {
1895 HMODULE lib = GetModuleHandleA("crypt32");
1896 BOOL (WINAPI *pI_CertUpdatestore)(HCERTSTORE, HCERTSTORE, DWORD, DWORD) =
1897 (void *)GetProcAddress(lib, "I_CertUpdateStore");
1898 BOOL ret;
1899 HCERTSTORE store1, store2;
1900 PCCERT_CONTEXT cert;
1901 DWORD certs;
1902
1903 if (!pI_CertUpdatestore)
1904 {
1905 skip("No I_CertUpdateStore\n");
1906 return;
1907 }
1908 store1 = CertOpenStore(CERT_STORE_PROV_MEMORY, 0, 0,
1909 CERT_STORE_CREATE_NEW_FLAG, NULL);
1910 store2 = CertOpenStore(CERT_STORE_PROV_MEMORY, 0, 0,
1911 CERT_STORE_CREATE_NEW_FLAG, NULL);
1912
1913 /* Crash
1914 ret = pI_CertUpdatestore(NULL, NULL, 0, 0);
1915 ret = pI_CertUpdatestore(store1, NULL, 0, 0);
1916 ret = pI_CertUpdatestore(NULL, store2, 0, 0);
1917 */
1918 ret = pI_CertUpdatestore(store1, store2, 0, 0);
1919 ok(ret, "I_CertUpdateStore failed: %08x\n", GetLastError());
1920
1921 CertAddEncodedCertificateToStore(store2, X509_ASN_ENCODING, bigCert,
1922 sizeof(bigCert), CERT_STORE_ADD_ALWAYS, &cert);
1923 /* I_CertUpdateStore adds the contexts from store2 to store1 */
1924 ret = pI_CertUpdatestore(store1, store2, 0, 0);
1925 ok(ret, "I_CertUpdateStore failed: %08x\n", GetLastError());
1926 certs = countCertsInStore(store1);
1927 ok(certs == 1, "Expected 1 cert, got %d\n", certs);
1928 /* Calling it a second time has no effect */
1929 ret = pI_CertUpdatestore(store1, store2, 0, 0);
1930 ok(ret, "I_CertUpdateStore failed: %08x\n", GetLastError());
1931 certs = countCertsInStore(store1);
1932 ok(certs == 1, "Expected 1 cert, got %d\n", certs);
1933
1934 /* The last parameters to I_CertUpdateStore appear to be ignored */
1935 ret = pI_CertUpdatestore(store1, store2, 1, 0);
1936 ok(ret, "I_CertUpdateStore failed: %08x\n", GetLastError());
1937 ret = pI_CertUpdatestore(store1, store2, 0, 1);
1938 ok(ret, "I_CertUpdateStore failed: %08x\n", GetLastError());
1939
1940 CertAddEncodedCRLToStore(store2, X509_ASN_ENCODING, signedCRL,
1941 sizeof(signedCRL), CERT_STORE_ADD_ALWAYS, NULL);
1942
1943 /* I_CertUpdateStore also adds the CRLs from store2 to store1 */
1944 ret = pI_CertUpdatestore(store1, store2, 0, 0);
1945 ok(ret, "I_CertUpdateStore failed: %08x\n", GetLastError());
1946 certs = countCertsInStore(store1);
1947 ok(certs == 1, "Expected 1 cert, got %d\n", certs);
1948 if (pCertEnumCRLsInStore)
1949 {
1950 certs = countCRLsInStore(store1);
1951 ok(certs == 1, "Expected 1 CRL, got %d\n", certs);
1952 }
1953
1954 CertDeleteCertificateFromStore(cert);
1955 /* If a context is deleted from store2, I_CertUpdateStore deletes it
1956 * from store1
1957 */
1958 ret = pI_CertUpdatestore(store1, store2, 0, 0);
1959 ok(ret, "I_CertUpdateStore failed: %08x\n", GetLastError());
1960 certs = countCertsInStore(store1);
1961 ok(certs == 0, "Expected 0 certs, got %d\n", certs);
1962
1963 CertFreeCertificateContext(cert);
1964 CertCloseStore(store1, 0);
1965 CertCloseStore(store2, 0);
1966 }
1967
1968 START_TEST(store)
1969 {
1970 HMODULE hdll;
1971
1972 hdll = GetModuleHandleA("Crypt32.dll");
1973 pCertAddStoreToCollection = (void*)GetProcAddress(hdll, "CertAddStoreToCollection");
1974 pCertControlStore = (void*)GetProcAddress(hdll, "CertControlStore");
1975 pCertEnumCRLsInStore = (void*)GetProcAddress(hdll, "CertEnumCRLsInStore");
1976 pCertEnumSystemStore = (void*)GetProcAddress(hdll, "CertEnumSystemStore");
1977 pCertGetStoreProperty = (void*)GetProcAddress(hdll, "CertGetStoreProperty");
1978 pCertRemoveStoreFromCollection = (void*)GetProcAddress(hdll, "CertRemoveStoreFromCollection");
1979 pCertSetStoreProperty = (void*)GetProcAddress(hdll, "CertSetStoreProperty");
1980
1981 /* various combinations of CertOpenStore */
1982 testMemStore();
1983 testCollectionStore();
1984 testRegStore();
1985 testSystemRegStore();
1986 testSystemStore();
1987 testFileStore();
1988 testFileNameStore();
1989 testMessageStore();
1990
1991 testCertOpenSystemStore();
1992 testCertEnumSystemStore();
1993 testStoreProperty();
1994
1995 testAddSerialized();
1996
1997 test_I_UpdateStore();
1998 }
useful hacks and other patches not (yet) suitable for mainline