shell32: Add printers CLSID to test, clean up a bit.
[wine/testsucceed.git] / dlls / crypt32 / serialize.c
blob0c366bf41252fba68dcedd7ad7e7d50e0318b902
1 /*
2 * Copyright 2004-2007 Juan Lang
4 * This library is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU Lesser General Public
6 * License as published by the Free Software Foundation; either
7 * version 2.1 of the License, or (at your option) any later version.
9 * This library is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 * Lesser General Public License for more details.
14 * You should have received a copy of the GNU Lesser General Public
15 * License along with this library; if not, write to the Free Software
16 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
19 #include "config.h"
20 #include "wine/port.h"
22 #include <stdarg.h>
23 #include "windef.h"
24 #include "winbase.h"
25 #include "wincrypt.h"
26 #include "wine/debug.h"
27 #include "wine/exception.h"
28 #include "crypt32_private.h"
30 WINE_DEFAULT_DEBUG_CHANNEL(crypt);
32 /* An extended certificate property in serialized form is prefixed by this
33 * header.
35 typedef struct _WINE_CERT_PROP_HEADER
37 DWORD propID;
38 DWORD unknown; /* always 1 */
39 DWORD cb;
40 } WINE_CERT_PROP_HEADER, *PWINE_CERT_PROP_HEADER;
42 static BOOL CRYPT_SerializeStoreElement(const void *context,
43 const BYTE *encodedContext, DWORD cbEncodedContext, DWORD contextPropID,
44 PCWINE_CONTEXT_INTERFACE contextInterface, DWORD dwFlags, BOOL omitHashes,
45 BYTE *pbElement, DWORD *pcbElement)
47 BOOL ret;
49 TRACE("(%p, %p, %08x, %d, %p, %p)\n", context, contextInterface, dwFlags,
50 omitHashes, pbElement, pcbElement);
52 if (context)
54 DWORD bytesNeeded = sizeof(WINE_CERT_PROP_HEADER) + cbEncodedContext;
55 DWORD prop = 0;
57 ret = TRUE;
58 do {
59 prop = contextInterface->enumProps(context, prop);
60 if (prop && (!omitHashes || !IS_CERT_HASH_PROP_ID(prop)))
62 DWORD propSize = 0;
64 ret = contextInterface->getProp(context, prop, NULL, &propSize);
65 if (ret)
66 bytesNeeded += sizeof(WINE_CERT_PROP_HEADER) + propSize;
68 } while (ret && prop != 0);
70 if (!pbElement)
72 *pcbElement = bytesNeeded;
73 ret = TRUE;
75 else if (*pcbElement < bytesNeeded)
77 *pcbElement = bytesNeeded;
78 SetLastError(ERROR_MORE_DATA);
79 ret = FALSE;
81 else
83 PWINE_CERT_PROP_HEADER hdr;
84 DWORD bufSize = 0;
85 LPBYTE buf = NULL;
87 prop = 0;
88 do {
89 prop = contextInterface->enumProps(context, prop);
90 if (prop && (!omitHashes || !IS_CERT_HASH_PROP_ID(prop)))
92 DWORD propSize = 0;
94 ret = contextInterface->getProp(context, prop, NULL,
95 &propSize);
96 if (ret)
98 if (bufSize < propSize)
100 if (buf)
101 buf = CryptMemRealloc(buf, propSize);
102 else
103 buf = CryptMemAlloc(propSize);
104 bufSize = propSize;
106 if (buf)
108 ret = contextInterface->getProp(context, prop, buf,
109 &propSize);
110 if (ret)
112 hdr = (PWINE_CERT_PROP_HEADER)pbElement;
113 hdr->propID = prop;
114 hdr->unknown = 1;
115 hdr->cb = propSize;
116 pbElement += sizeof(WINE_CERT_PROP_HEADER);
117 if (propSize)
119 memcpy(pbElement, buf, propSize);
120 pbElement += propSize;
124 else
125 ret = FALSE;
128 } while (ret && prop != 0);
129 CryptMemFree(buf);
131 hdr = (PWINE_CERT_PROP_HEADER)pbElement;
132 hdr->propID = contextPropID;
133 hdr->unknown = 1;
134 hdr->cb = cbEncodedContext;
135 memcpy(pbElement + sizeof(WINE_CERT_PROP_HEADER),
136 encodedContext, cbEncodedContext);
139 else
140 ret = FALSE;
141 return ret;
144 BOOL WINAPI CertSerializeCertificateStoreElement(PCCERT_CONTEXT pCertContext,
145 DWORD dwFlags, BYTE *pbElement, DWORD *pcbElement)
147 return CRYPT_SerializeStoreElement(pCertContext,
148 pCertContext->pbCertEncoded, pCertContext->cbCertEncoded,
149 CERT_CERT_PROP_ID, pCertInterface, dwFlags, FALSE, pbElement, pcbElement);
152 BOOL WINAPI CertSerializeCRLStoreElement(PCCRL_CONTEXT pCrlContext,
153 DWORD dwFlags, BYTE *pbElement, DWORD *pcbElement)
155 return CRYPT_SerializeStoreElement(pCrlContext,
156 pCrlContext->pbCrlEncoded, pCrlContext->cbCrlEncoded,
157 CERT_CRL_PROP_ID, pCRLInterface, dwFlags, FALSE, pbElement, pcbElement);
160 BOOL WINAPI CertSerializeCTLStoreElement(PCCTL_CONTEXT pCtlContext,
161 DWORD dwFlags, BYTE *pbElement, DWORD *pcbElement)
163 return CRYPT_SerializeStoreElement(pCtlContext,
164 pCtlContext->pbCtlEncoded, pCtlContext->cbCtlEncoded,
165 CERT_CTL_PROP_ID, pCTLInterface, dwFlags, FALSE, pbElement, pcbElement);
168 /* Looks for the property with ID propID in the buffer buf. Returns a pointer
169 * to its header if a valid header is found, NULL if not. Valid means the
170 * length of thte property won't overrun buf, and the unknown field is 1.
172 static const WINE_CERT_PROP_HEADER *CRYPT_findPropID(const BYTE *buf,
173 DWORD size, DWORD propID)
175 const WINE_CERT_PROP_HEADER *ret = NULL;
176 BOOL done = FALSE;
178 while (size && !ret && !done)
180 if (size < sizeof(WINE_CERT_PROP_HEADER))
182 SetLastError(CRYPT_E_FILE_ERROR);
183 done = TRUE;
185 else
187 const WINE_CERT_PROP_HEADER *hdr =
188 (const WINE_CERT_PROP_HEADER *)buf;
190 size -= sizeof(WINE_CERT_PROP_HEADER);
191 buf += sizeof(WINE_CERT_PROP_HEADER);
192 if (size < hdr->cb)
194 SetLastError(E_INVALIDARG);
195 done = TRUE;
197 else if (!hdr->propID)
199 /* assume a zero prop ID means the data are uninitialized, so
200 * stop looking.
202 done = TRUE;
204 else if (hdr->unknown != 1)
206 SetLastError(ERROR_FILE_NOT_FOUND);
207 done = TRUE;
209 else if (hdr->propID == propID)
210 ret = hdr;
211 else
213 buf += hdr->cb;
214 size -= hdr->cb;
218 return ret;
221 static BOOL CRYPT_ReadContextProp(
222 const WINE_CONTEXT_INTERFACE *contextInterface, const void *context,
223 const WINE_CERT_PROP_HEADER *hdr, const BYTE *pbElement, DWORD cbElement)
225 BOOL ret;
227 if (cbElement < hdr->cb)
229 SetLastError(E_INVALIDARG);
230 ret = FALSE;
232 else if (hdr->unknown != 1)
234 SetLastError(ERROR_FILE_NOT_FOUND);
235 ret = FALSE;
237 else if (hdr->propID != CERT_CERT_PROP_ID &&
238 hdr->propID != CERT_CRL_PROP_ID && hdr->propID != CERT_CTL_PROP_ID)
240 /* Have to create a blob for most types, but not
241 * for all.. arghh.
243 switch (hdr->propID)
245 case CERT_AUTO_ENROLL_PROP_ID:
246 case CERT_CTL_USAGE_PROP_ID:
247 case CERT_DESCRIPTION_PROP_ID:
248 case CERT_FRIENDLY_NAME_PROP_ID:
249 case CERT_HASH_PROP_ID:
250 case CERT_KEY_IDENTIFIER_PROP_ID:
251 case CERT_MD5_HASH_PROP_ID:
252 case CERT_NEXT_UPDATE_LOCATION_PROP_ID:
253 case CERT_PUBKEY_ALG_PARA_PROP_ID:
254 case CERT_PVK_FILE_PROP_ID:
255 case CERT_SIGNATURE_HASH_PROP_ID:
256 case CERT_ISSUER_PUBLIC_KEY_MD5_HASH_PROP_ID:
257 case CERT_SUBJECT_PUBLIC_KEY_MD5_HASH_PROP_ID:
258 case CERT_ENROLLMENT_PROP_ID:
259 case CERT_CROSS_CERT_DIST_POINTS_PROP_ID:
260 case CERT_RENEWAL_PROP_ID:
262 CRYPT_DATA_BLOB blob = { hdr->cb,
263 (LPBYTE)pbElement };
265 ret = contextInterface->setProp(context,
266 hdr->propID, 0, &blob);
267 break;
269 case CERT_DATE_STAMP_PROP_ID:
270 ret = contextInterface->setProp(context,
271 hdr->propID, 0, pbElement);
272 break;
273 case CERT_KEY_PROV_INFO_PROP_ID:
275 PCRYPT_KEY_PROV_INFO info =
276 (PCRYPT_KEY_PROV_INFO)pbElement;
278 CRYPT_FixKeyProvInfoPointers(info);
279 ret = contextInterface->setProp(context,
280 hdr->propID, 0, pbElement);
281 break;
283 default:
284 ret = FALSE;
287 else
289 /* ignore the context itself */
290 ret = TRUE;
292 return ret;
295 const void *CRYPT_ReadSerializedElement(const BYTE *pbElement, DWORD cbElement,
296 DWORD dwContextTypeFlags, DWORD *pdwContentType)
298 const void *context;
300 TRACE("(%p, %d, %08x, %p)\n", pbElement, cbElement, dwContextTypeFlags,
301 pdwContentType);
303 if (!cbElement)
305 SetLastError(ERROR_END_OF_MEDIA);
306 return NULL;
309 __TRY
311 const WINE_CONTEXT_INTERFACE *contextInterface = NULL;
312 const WINE_CERT_PROP_HEADER *hdr = NULL;
313 DWORD type = 0;
314 BOOL ret;
316 ret = TRUE;
317 context = NULL;
318 if (dwContextTypeFlags == CERT_STORE_ALL_CONTEXT_FLAG)
320 hdr = CRYPT_findPropID(pbElement, cbElement, CERT_CERT_PROP_ID);
321 if (hdr)
322 type = CERT_STORE_CERTIFICATE_CONTEXT;
323 else
325 hdr = CRYPT_findPropID(pbElement, cbElement, CERT_CRL_PROP_ID);
326 if (hdr)
327 type = CERT_STORE_CRL_CONTEXT;
328 else
330 hdr = CRYPT_findPropID(pbElement, cbElement,
331 CERT_CTL_PROP_ID);
332 if (hdr)
333 type = CERT_STORE_CTL_CONTEXT;
337 else if (dwContextTypeFlags & CERT_STORE_CERTIFICATE_CONTEXT_FLAG)
339 hdr = CRYPT_findPropID(pbElement, cbElement, CERT_CERT_PROP_ID);
340 type = CERT_STORE_CERTIFICATE_CONTEXT;
342 else if (dwContextTypeFlags & CERT_STORE_CRL_CONTEXT_FLAG)
344 hdr = CRYPT_findPropID(pbElement, cbElement, CERT_CRL_PROP_ID);
345 type = CERT_STORE_CRL_CONTEXT;
347 else if (dwContextTypeFlags & CERT_STORE_CTL_CONTEXT_FLAG)
349 hdr = CRYPT_findPropID(pbElement, cbElement, CERT_CTL_PROP_ID);
350 type = CERT_STORE_CTL_CONTEXT;
353 switch (type)
355 case CERT_STORE_CERTIFICATE_CONTEXT:
356 contextInterface = pCertInterface;
357 break;
358 case CERT_STORE_CRL_CONTEXT:
359 contextInterface = pCRLInterface;
360 break;
361 case CERT_STORE_CTL_CONTEXT:
362 contextInterface = pCTLInterface;
363 break;
364 default:
365 SetLastError(E_INVALIDARG);
366 ret = FALSE;
368 if (!hdr)
369 ret = FALSE;
371 if (ret)
372 context = contextInterface->create(X509_ASN_ENCODING,
373 (BYTE *)hdr + sizeof(WINE_CERT_PROP_HEADER), hdr->cb);
374 if (ret && context)
376 BOOL noMoreProps = FALSE;
378 while (!noMoreProps && ret)
380 if (cbElement < sizeof(WINE_CERT_PROP_HEADER))
381 ret = FALSE;
382 else
384 const WINE_CERT_PROP_HEADER *hdr =
385 (const WINE_CERT_PROP_HEADER *)pbElement;
387 TRACE("prop is %d\n", hdr->propID);
388 cbElement -= sizeof(WINE_CERT_PROP_HEADER);
389 pbElement += sizeof(WINE_CERT_PROP_HEADER);
390 if (!hdr->propID)
392 /* Like in CRYPT_findPropID, stop if the propID is zero
394 noMoreProps = TRUE;
396 else
397 ret = CRYPT_ReadContextProp(contextInterface, context,
398 hdr, pbElement, cbElement);
399 pbElement += hdr->cb;
400 cbElement -= hdr->cb;
401 if (!cbElement)
402 noMoreProps = TRUE;
405 if (ret)
407 if (pdwContentType)
408 *pdwContentType = type;
410 else
412 contextInterface->free(context);
413 context = NULL;
417 __EXCEPT_PAGE_FAULT
419 SetLastError(STATUS_ACCESS_VIOLATION);
420 context = NULL;
422 __ENDTRY
423 return context;
426 static const BYTE fileHeader[] = { 0, 0, 0, 0, 'C','E','R','T' };
428 typedef BOOL (*read_serialized_func)(void *handle, void *buffer,
429 DWORD bytesToRead, DWORD *bytesRead);
431 static BOOL CRYPT_ReadSerializedStore(void *handle,
432 read_serialized_func read_func, HCERTSTORE store)
434 BYTE fileHeaderBuf[sizeof(fileHeader)];
435 DWORD read;
436 BOOL ret;
438 /* Failure reading is non-critical, we'll leave the store empty */
439 ret = read_func(handle, fileHeaderBuf, sizeof(fileHeaderBuf), &read);
440 if (ret)
442 if (!read)
443 ; /* an empty file is okay */
444 else if (read != sizeof(fileHeaderBuf))
445 ret = FALSE;
446 else if (!memcmp(fileHeaderBuf, fileHeader, read))
448 WINE_CERT_PROP_HEADER propHdr;
449 const void *context = NULL;
450 const WINE_CONTEXT_INTERFACE *contextInterface = NULL;
451 LPBYTE buf = NULL;
452 DWORD bufSize = 0;
454 do {
455 ret = read_func(handle, &propHdr, sizeof(propHdr), &read);
456 if (ret && read == sizeof(propHdr))
458 if (contextInterface && context &&
459 (propHdr.propID == CERT_CERT_PROP_ID ||
460 propHdr.propID == CERT_CRL_PROP_ID ||
461 propHdr.propID == CERT_CTL_PROP_ID))
463 /* We have a new context, so free the existing one */
464 contextInterface->free(context);
466 if (propHdr.cb > bufSize)
468 /* Not reusing realloc, because the old data aren't
469 * needed any longer.
471 CryptMemFree(buf);
472 buf = CryptMemAlloc(propHdr.cb);
473 bufSize = propHdr.cb;
475 if (buf)
477 ret = read_func(handle, buf, propHdr.cb, &read);
478 if (ret && read == propHdr.cb)
480 if (propHdr.propID == CERT_CERT_PROP_ID)
482 contextInterface = pCertInterface;
483 ret = contextInterface->addEncodedToStore(store,
484 X509_ASN_ENCODING, buf, read,
485 CERT_STORE_ADD_NEW, &context);
487 else if (propHdr.propID == CERT_CRL_PROP_ID)
489 contextInterface = pCRLInterface;
490 ret = contextInterface->addEncodedToStore(store,
491 X509_ASN_ENCODING, buf, read,
492 CERT_STORE_ADD_NEW, &context);
494 else if (propHdr.propID == CERT_CTL_PROP_ID)
496 contextInterface = pCTLInterface;
497 ret = contextInterface->addEncodedToStore(store,
498 X509_ASN_ENCODING, buf, read,
499 CERT_STORE_ADD_NEW, &context);
501 else
503 if (!contextInterface)
505 WARN("prop id %d before a context id\n",
506 propHdr.propID);
507 ret = FALSE;
509 else
510 ret = CRYPT_ReadContextProp(
511 contextInterface, context, &propHdr, buf,
512 read);
516 else
517 ret = FALSE;
519 } while (ret && read > 0);
520 if (contextInterface && context)
522 /* Free the last context added */
523 contextInterface->free(context);
525 CryptMemFree(buf);
526 ret = TRUE;
528 else
529 ret = FALSE;
531 else
532 ret = TRUE;
533 return ret;
536 static BOOL read_file_wrapper(void *handle, void *buffer, DWORD bytesToRead,
537 DWORD *bytesRead)
539 return ReadFile(handle, buffer, bytesToRead, bytesRead, NULL);
542 BOOL CRYPT_ReadSerializedStoreFromFile(HANDLE file, HCERTSTORE store)
544 return CRYPT_ReadSerializedStore(file, read_file_wrapper, store);
547 struct BlobReader
549 const CRYPT_DATA_BLOB *blob;
550 DWORD current;
553 static BOOL read_blob_wrapper(void *handle, void *buffer, DWORD bytesToRead,
554 DWORD *bytesRead)
556 struct BlobReader *reader = handle;
557 BOOL ret;
559 if (reader->current < reader->blob->cbData)
561 *bytesRead = min(bytesToRead, reader->blob->cbData - reader->current);
562 memcpy(buffer, reader->blob->pbData + reader->current, *bytesRead);
563 ret = TRUE;
565 else
566 ret = FALSE;
567 return ret;
570 BOOL CRYPT_ReadSerializedStoreFromBlob(const CRYPT_DATA_BLOB *blob,
571 HCERTSTORE store)
573 struct BlobReader reader = { blob, 0 };
575 return CRYPT_ReadSerializedStore(&reader, read_blob_wrapper, store);
578 static BOOL WINAPI CRYPT_SerializeCertNoHash(PCCERT_CONTEXT pCertContext,
579 DWORD dwFlags, BYTE *pbElement, DWORD *pcbElement)
581 return CRYPT_SerializeStoreElement(pCertContext,
582 pCertContext->pbCertEncoded, pCertContext->cbCertEncoded,
583 CERT_CERT_PROP_ID, pCertInterface, dwFlags, TRUE, pbElement, pcbElement);
586 static BOOL WINAPI CRYPT_SerializeCRLNoHash(PCCRL_CONTEXT pCrlContext,
587 DWORD dwFlags, BYTE *pbElement, DWORD *pcbElement)
589 return CRYPT_SerializeStoreElement(pCrlContext,
590 pCrlContext->pbCrlEncoded, pCrlContext->cbCrlEncoded,
591 CERT_CRL_PROP_ID, pCRLInterface, dwFlags, TRUE, pbElement, pcbElement);
594 static BOOL WINAPI CRYPT_SerializeCTLNoHash(PCCTL_CONTEXT pCtlContext,
595 DWORD dwFlags, BYTE *pbElement, DWORD *pcbElement)
597 return CRYPT_SerializeStoreElement(pCtlContext,
598 pCtlContext->pbCtlEncoded, pCtlContext->cbCtlEncoded,
599 CERT_CTL_PROP_ID, pCTLInterface, dwFlags, TRUE, pbElement, pcbElement);
602 typedef BOOL (*SerializedOutputFunc)(void *handle, const void *buffer,
603 DWORD size);
605 static BOOL CRYPT_SerializeContextsToStream(SerializedOutputFunc output,
606 void *handle, const WINE_CONTEXT_INTERFACE *contextInterface, HCERTSTORE store)
608 const void *context = NULL;
609 BOOL ret;
611 do {
612 context = contextInterface->enumContextsInStore(store, context);
613 if (context)
615 DWORD size = 0;
616 LPBYTE buf = NULL;
618 ret = contextInterface->serialize(context, 0, NULL, &size);
619 if (size)
620 buf = CryptMemAlloc(size);
621 if (buf)
623 ret = contextInterface->serialize(context, 0, buf, &size);
624 if (ret)
625 ret = output(handle, buf, size);
627 CryptMemFree(buf);
629 else
630 ret = TRUE;
631 } while (ret && context != NULL);
632 if (context)
633 contextInterface->free(context);
634 return ret;
637 static BOOL CRYPT_WriteSerializedStoreToStream(HCERTSTORE store,
638 SerializedOutputFunc output, void *handle)
640 static const BYTE fileTrailer[12] = { 0 };
641 WINE_CONTEXT_INTERFACE interface;
642 BOOL ret;
644 ret = output(handle, fileHeader, sizeof(fileHeader));
645 if (ret)
647 memcpy(&interface, pCertInterface, sizeof(interface));
648 interface.serialize = (SerializeElementFunc)CRYPT_SerializeCertNoHash;
649 ret = CRYPT_SerializeContextsToStream(output, handle, &interface,
650 store);
652 if (ret)
654 memcpy(&interface, pCRLInterface, sizeof(interface));
655 interface.serialize = (SerializeElementFunc)CRYPT_SerializeCRLNoHash;
656 ret = CRYPT_SerializeContextsToStream(output, handle, &interface,
657 store);
659 if (ret)
661 memcpy(&interface, pCTLInterface, sizeof(interface));
662 interface.serialize = (SerializeElementFunc)CRYPT_SerializeCTLNoHash;
663 ret = CRYPT_SerializeContextsToStream(output, handle, &interface,
664 store);
666 if (ret)
667 ret = output(handle, fileTrailer, sizeof(fileTrailer));
668 return ret;
671 static BOOL CRYPT_FileOutputFunc(void *handle, const void *buffer, DWORD size)
673 return WriteFile(handle, buffer, size, &size, NULL);
676 static BOOL CRYPT_WriteSerializedStoreToFile(HANDLE file, HCERTSTORE store)
678 SetFilePointer(file, 0, NULL, FILE_BEGIN);
679 return CRYPT_WriteSerializedStoreToStream(store, CRYPT_FileOutputFunc,
680 file);
683 static BOOL CRYPT_SavePKCSToMem(HCERTSTORE store,
684 DWORD dwMsgAndCertEncodingType, void *handle)
686 CERT_BLOB *blob = handle;
687 CRYPT_SIGNED_INFO signedInfo = { 0 };
688 PCCERT_CONTEXT cert = NULL;
689 PCCRL_CONTEXT crl = NULL;
690 DWORD size;
691 BOOL ret = TRUE;
693 TRACE("(%d, %p)\n", blob->pbData ? blob->cbData : 0, blob->pbData);
695 do {
696 cert = CertEnumCertificatesInStore(store, cert);
697 if (cert)
698 signedInfo.cCertEncoded++;
699 } while (cert);
700 if (signedInfo.cCertEncoded)
702 signedInfo.rgCertEncoded = CryptMemAlloc(
703 signedInfo.cCertEncoded * sizeof(CERT_BLOB));
704 if (!signedInfo.rgCertEncoded)
706 SetLastError(ERROR_OUTOFMEMORY);
707 ret = FALSE;
709 else
711 DWORD i = 0;
713 do {
714 cert = CertEnumCertificatesInStore(store, cert);
715 if (cert)
717 signedInfo.rgCertEncoded[i].cbData = cert->cbCertEncoded;
718 signedInfo.rgCertEncoded[i].pbData = cert->pbCertEncoded;
719 i++;
721 } while (cert);
725 do {
726 crl = CertEnumCRLsInStore(store, crl);
727 if (crl)
728 signedInfo.cCrlEncoded++;
729 } while (crl);
730 if (signedInfo.cCrlEncoded)
732 signedInfo.rgCrlEncoded = CryptMemAlloc(
733 signedInfo.cCrlEncoded * sizeof(CERT_BLOB));
734 if (!signedInfo.rgCrlEncoded)
736 SetLastError(ERROR_OUTOFMEMORY);
737 ret = FALSE;
739 else
741 DWORD i = 0;
743 do {
744 crl = CertEnumCRLsInStore(store, crl);
745 if (crl)
747 signedInfo.rgCrlEncoded[i].cbData = crl->cbCrlEncoded;
748 signedInfo.rgCrlEncoded[i].pbData = crl->pbCrlEncoded;
749 i++;
751 } while (crl);
754 if (ret)
756 ret = CRYPT_AsnEncodeCMSSignedInfo(&signedInfo, NULL, &size);
757 if (ret)
759 if (!blob->pbData)
760 blob->cbData = size;
761 else if (blob->cbData < size)
763 blob->cbData = size;
764 SetLastError(ERROR_MORE_DATA);
765 ret = FALSE;
767 else
769 blob->cbData = size;
770 ret = CRYPT_AsnEncodeCMSSignedInfo(&signedInfo, blob->pbData,
771 &blob->cbData);
775 CryptMemFree(signedInfo.rgCertEncoded);
776 CryptMemFree(signedInfo.rgCrlEncoded);
777 TRACE("returning %d\n", ret);
778 return ret;
781 static BOOL CRYPT_SavePKCSToFile(HCERTSTORE store,
782 DWORD dwMsgAndCertEncodingType, void *handle)
784 CERT_BLOB blob = { 0, NULL };
785 BOOL ret;
787 TRACE("(%p)\n", handle);
789 ret = CRYPT_SavePKCSToMem(store, dwMsgAndCertEncodingType, &blob);
790 if (ret)
792 blob.pbData = CryptMemAlloc(blob.cbData);
793 if (blob.pbData)
795 ret = CRYPT_SavePKCSToMem(store, dwMsgAndCertEncodingType, &blob);
796 if (ret)
797 ret = WriteFile(handle, blob.pbData, blob.cbData,
798 &blob.cbData, NULL);
800 else
802 SetLastError(ERROR_OUTOFMEMORY);
803 ret = FALSE;
806 TRACE("returning %d\n", ret);
807 return ret;
810 static BOOL CRYPT_SaveSerializedToFile(HCERTSTORE store,
811 DWORD dwMsgAndCertEncodingType, void *handle)
813 return CRYPT_WriteSerializedStoreToFile(handle, store);
816 struct MemWrittenTracker
818 DWORD cbData;
819 BYTE *pbData;
820 DWORD written;
823 /* handle is a pointer to a MemWrittenTracker. Assumes its pointer is valid. */
824 static BOOL CRYPT_MemOutputFunc(void *handle, const void *buffer, DWORD size)
826 struct MemWrittenTracker *tracker = handle;
827 BOOL ret;
829 if (tracker->written + size > tracker->cbData)
831 SetLastError(ERROR_MORE_DATA);
832 /* Update written so caller can notify its caller of the required size
834 tracker->written += size;
835 ret = FALSE;
837 else
839 memcpy(tracker->pbData + tracker->written, buffer, size);
840 tracker->written += size;
841 ret = TRUE;
843 return ret;
846 static BOOL CRYPT_CountSerializedBytes(void *handle, const void *buffer,
847 DWORD size)
849 *(DWORD *)handle += size;
850 return TRUE;
853 static BOOL CRYPT_SaveSerializedToMem(HCERTSTORE store,
854 DWORD dwMsgAndCertEncodingType, void *handle)
856 CERT_BLOB *blob = handle;
857 DWORD size = 0;
858 BOOL ret;
860 ret = CRYPT_WriteSerializedStoreToStream(store, CRYPT_CountSerializedBytes,
861 &size);
862 if (ret)
864 if (!blob->pbData)
865 blob->cbData = size;
866 else if (blob->cbData < size)
868 SetLastError(ERROR_MORE_DATA);
869 blob->cbData = size;
870 ret = FALSE;
872 else
874 struct MemWrittenTracker tracker = { blob->cbData, blob->pbData,
875 0 };
877 ret = CRYPT_WriteSerializedStoreToStream(store, CRYPT_MemOutputFunc,
878 &tracker);
879 if (!ret && GetLastError() == ERROR_MORE_DATA)
880 blob->cbData = tracker.written;
883 TRACE("returning %d\n", ret);
884 return ret;
887 BOOL WINAPI CertSaveStore(HCERTSTORE hCertStore, DWORD dwMsgAndCertEncodingType,
888 DWORD dwSaveAs, DWORD dwSaveTo, void *pvSaveToPara, DWORD dwFlags)
890 BOOL (*saveFunc)(HCERTSTORE, DWORD, void *);
891 void *handle;
892 BOOL ret, closeFile = TRUE;
894 TRACE("(%p, %08x, %d, %d, %p, %08x)\n", hCertStore,
895 dwMsgAndCertEncodingType, dwSaveAs, dwSaveTo, pvSaveToPara, dwFlags);
897 switch (dwSaveAs)
899 case CERT_STORE_SAVE_AS_STORE:
900 if (dwSaveTo == CERT_STORE_SAVE_TO_MEMORY)
901 saveFunc = CRYPT_SaveSerializedToMem;
902 else
903 saveFunc = CRYPT_SaveSerializedToFile;
904 break;
905 case CERT_STORE_SAVE_AS_PKCS7:
906 if (dwSaveTo == CERT_STORE_SAVE_TO_MEMORY)
907 saveFunc = CRYPT_SavePKCSToMem;
908 else
909 saveFunc = CRYPT_SavePKCSToFile;
910 break;
911 default:
912 WARN("unimplemented for %d\n", dwSaveAs);
913 SetLastError(ERROR_INVALID_PARAMETER);
914 return FALSE;
916 switch (dwSaveTo)
918 case CERT_STORE_SAVE_TO_FILE:
919 handle = pvSaveToPara;
920 closeFile = FALSE;
921 break;
922 case CERT_STORE_SAVE_TO_FILENAME_A:
923 handle = CreateFileA(pvSaveToPara, GENERIC_WRITE, 0, NULL,
924 CREATE_ALWAYS, 0, NULL);
925 break;
926 case CERT_STORE_SAVE_TO_FILENAME_W:
927 handle = CreateFileW(pvSaveToPara, GENERIC_WRITE, 0, NULL,
928 CREATE_ALWAYS, 0, NULL);
929 break;
930 case CERT_STORE_SAVE_TO_MEMORY:
931 handle = pvSaveToPara;
932 break;
933 default:
934 WARN("unimplemented for %d\n", dwSaveTo);
935 SetLastError(ERROR_INVALID_PARAMETER);
936 return FALSE;
938 ret = saveFunc(hCertStore, dwMsgAndCertEncodingType, handle);
939 if (closeFile)
940 CloseHandle(handle);
941 TRACE("returning %d\n", ret);
942 return ret;
945 BOOL WINAPI CertAddSerializedElementToStore(HCERTSTORE hCertStore,
946 const BYTE *pbElement, DWORD cbElement, DWORD dwAddDisposition, DWORD dwFlags,
947 DWORD dwContextTypeFlags, DWORD *pdwContentType, const void **ppvContext)
949 const void *context;
950 DWORD type;
951 BOOL ret;
953 TRACE("(%p, %p, %d, %08x, %08x, %08x, %p, %p)\n", hCertStore,
954 pbElement, cbElement, dwAddDisposition, dwFlags, dwContextTypeFlags,
955 pdwContentType, ppvContext);
957 /* Call the internal function, then delete the hashes. Tests show this
958 * function uses real hash values, not whatever's stored in the hash
959 * property.
961 context = CRYPT_ReadSerializedElement(pbElement, cbElement,
962 dwContextTypeFlags, &type);
963 if (context)
965 const WINE_CONTEXT_INTERFACE *contextInterface = NULL;
967 switch (type)
969 case CERT_STORE_CERTIFICATE_CONTEXT:
970 contextInterface = pCertInterface;
971 break;
972 case CERT_STORE_CRL_CONTEXT:
973 contextInterface = pCRLInterface;
974 break;
975 case CERT_STORE_CTL_CONTEXT:
976 contextInterface = pCTLInterface;
977 break;
978 default:
979 SetLastError(E_INVALIDARG);
981 if (contextInterface)
983 contextInterface->setProp(context, CERT_HASH_PROP_ID, 0, NULL);
984 contextInterface->setProp(context, CERT_MD5_HASH_PROP_ID, 0, NULL);
985 contextInterface->setProp(context, CERT_SIGNATURE_HASH_PROP_ID, 0,
986 NULL);
987 if (pdwContentType)
988 *pdwContentType = type;
989 ret = contextInterface->addContextToStore(hCertStore, context,
990 dwAddDisposition, ppvContext);
991 contextInterface->free(context);
993 else
994 ret = FALSE;
996 else
997 ret = FALSE;
998 return ret;