quartz: Fix EC_COMPLETE handling on video renderer.
[wine/multimedia.git] / dlls / crypt32 / msg.c
blob6b2d8b6391d81d1895eaf9f12f81bea076685d2e
1 /*
2 * Copyright 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 #define NONAMELESSUNION
24 #include "windef.h"
25 #include "winbase.h"
26 #include "wincrypt.h"
27 #include "snmp.h"
29 #include "wine/debug.h"
30 #include "wine/exception.h"
31 #include "crypt32_private.h"
33 WINE_DEFAULT_DEBUG_CHANNEL(crypt);
35 /* Called when a message's ref count reaches zero. Free any message-specific
36 * data here.
38 typedef void (*CryptMsgCloseFunc)(HCRYPTMSG msg);
40 typedef BOOL (*CryptMsgGetParamFunc)(HCRYPTMSG hCryptMsg, DWORD dwParamType,
41 DWORD dwIndex, void *pvData, DWORD *pcbData);
43 typedef BOOL (*CryptMsgUpdateFunc)(HCRYPTMSG hCryptMsg, const BYTE *pbData,
44 DWORD cbData, BOOL fFinal);
46 typedef BOOL (*CryptMsgControlFunc)(HCRYPTMSG hCryptMsg, DWORD dwFlags,
47 DWORD dwCtrlType, const void *pvCtrlPara);
49 static BOOL CRYPT_DefaultMsgControl(HCRYPTMSG hCryptMsg, DWORD dwFlags,
50 DWORD dwCtrlType, const void *pvCtrlPara)
52 TRACE("(%p, %08x, %d, %p)\n", hCryptMsg, dwFlags, dwCtrlType, pvCtrlPara);
53 SetLastError(E_INVALIDARG);
54 return FALSE;
57 typedef enum _CryptMsgState {
58 MsgStateInit,
59 MsgStateUpdated,
60 MsgStateDataFinalized,
61 MsgStateFinalized
62 } CryptMsgState;
64 typedef struct _CryptMsgBase
66 LONG ref;
67 DWORD open_flags;
68 BOOL streamed;
69 CMSG_STREAM_INFO stream_info;
70 CryptMsgState state;
71 CryptMsgCloseFunc close;
72 CryptMsgUpdateFunc update;
73 CryptMsgGetParamFunc get_param;
74 CryptMsgControlFunc control;
75 } CryptMsgBase;
77 static inline void CryptMsgBase_Init(CryptMsgBase *msg, DWORD dwFlags,
78 PCMSG_STREAM_INFO pStreamInfo, CryptMsgCloseFunc close,
79 CryptMsgGetParamFunc get_param, CryptMsgUpdateFunc update,
80 CryptMsgControlFunc control)
82 msg->ref = 1;
83 msg->open_flags = dwFlags;
84 if (pStreamInfo)
86 msg->streamed = TRUE;
87 msg->stream_info = *pStreamInfo;
89 else
91 msg->streamed = FALSE;
92 memset(&msg->stream_info, 0, sizeof(msg->stream_info));
94 msg->close = close;
95 msg->get_param = get_param;
96 msg->update = update;
97 msg->control = control;
98 msg->state = MsgStateInit;
101 typedef struct _CDataEncodeMsg
103 CryptMsgBase base;
104 DWORD bare_content_len;
105 LPBYTE bare_content;
106 } CDataEncodeMsg;
108 static const BYTE empty_data_content[] = { 0x04,0x00 };
110 static void CDataEncodeMsg_Close(HCRYPTMSG hCryptMsg)
112 CDataEncodeMsg *msg = hCryptMsg;
114 if (msg->bare_content != empty_data_content)
115 LocalFree(msg->bare_content);
118 static BOOL WINAPI CRYPT_EncodeContentLength(DWORD dwCertEncodingType,
119 LPCSTR lpszStructType, const void *pvStructInfo, DWORD dwFlags,
120 PCRYPT_ENCODE_PARA pEncodePara, BYTE *pbEncoded, DWORD *pcbEncoded)
122 DWORD dataLen = *(DWORD *)pvStructInfo;
123 DWORD lenBytes;
124 BOOL ret = TRUE;
126 /* Trick: report bytes needed based on total message length, even though
127 * the message isn't available yet. The caller will use the length
128 * reported here to encode its length.
130 CRYPT_EncodeLen(dataLen, NULL, &lenBytes);
131 if (!pbEncoded)
132 *pcbEncoded = 1 + lenBytes + dataLen;
133 else
135 if ((ret = CRYPT_EncodeEnsureSpace(dwFlags, pEncodePara, pbEncoded,
136 pcbEncoded, 1 + lenBytes)))
138 if (dwFlags & CRYPT_ENCODE_ALLOC_FLAG)
139 pbEncoded = *(BYTE **)pbEncoded;
140 *pbEncoded++ = ASN_OCTETSTRING;
141 CRYPT_EncodeLen(dataLen, pbEncoded,
142 &lenBytes);
145 return ret;
148 static BOOL CRYPT_EncodeDataContentInfoHeader(const CDataEncodeMsg *msg,
149 CRYPT_DATA_BLOB *header)
151 BOOL ret;
153 if (msg->base.streamed && msg->base.stream_info.cbContent == 0xffffffff)
155 static const BYTE headerValue[] = { 0x30,0x80,0x06,0x09,0x2a,0x86,0x48,
156 0x86,0xf7,0x0d,0x01,0x07,0x01,0xa0,0x80,0x24,0x80 };
158 header->pbData = LocalAlloc(0, sizeof(headerValue));
159 if (header->pbData)
161 header->cbData = sizeof(headerValue);
162 memcpy(header->pbData, headerValue, sizeof(headerValue));
163 ret = TRUE;
165 else
166 ret = FALSE;
168 else
170 struct AsnConstructedItem constructed = { 0,
171 &msg->base.stream_info.cbContent, CRYPT_EncodeContentLength };
172 struct AsnEncodeSequenceItem items[2] = {
173 { szOID_RSA_data, CRYPT_AsnEncodeOid, 0 },
174 { &constructed, CRYPT_AsnEncodeConstructed, 0 },
177 ret = CRYPT_AsnEncodeSequence(X509_ASN_ENCODING, items,
178 sizeof(items) / sizeof(items[0]), CRYPT_ENCODE_ALLOC_FLAG, NULL,
179 (LPBYTE)&header->pbData, &header->cbData);
180 if (ret)
182 /* Trick: subtract the content length from the reported length,
183 * as the actual content hasn't come yet.
185 header->cbData -= msg->base.stream_info.cbContent;
188 return ret;
191 static BOOL CDataEncodeMsg_Update(HCRYPTMSG hCryptMsg, const BYTE *pbData,
192 DWORD cbData, BOOL fFinal)
194 CDataEncodeMsg *msg = hCryptMsg;
195 BOOL ret = FALSE;
197 if (msg->base.state == MsgStateFinalized)
198 SetLastError(CRYPT_E_MSG_ERROR);
199 else if (msg->base.streamed)
201 __TRY
203 if (msg->base.state != MsgStateUpdated)
205 CRYPT_DATA_BLOB header;
207 ret = CRYPT_EncodeDataContentInfoHeader(msg, &header);
208 if (ret)
210 ret = msg->base.stream_info.pfnStreamOutput(
211 msg->base.stream_info.pvArg, header.pbData, header.cbData,
212 FALSE);
213 LocalFree(header.pbData);
216 /* Curiously, every indefinite-length streamed update appears to
217 * get its own tag and length, regardless of fFinal.
219 if (msg->base.stream_info.cbContent == 0xffffffff)
221 BYTE *header;
222 DWORD headerLen;
224 ret = CRYPT_EncodeContentLength(X509_ASN_ENCODING, NULL,
225 &cbData, CRYPT_ENCODE_ALLOC_FLAG, NULL, (BYTE *)&header,
226 &headerLen);
227 if (ret)
229 ret = msg->base.stream_info.pfnStreamOutput(
230 msg->base.stream_info.pvArg, header, headerLen,
231 FALSE);
232 LocalFree(header);
235 if (!fFinal)
237 ret = msg->base.stream_info.pfnStreamOutput(
238 msg->base.stream_info.pvArg, (BYTE *)pbData, cbData,
239 FALSE);
240 msg->base.state = MsgStateUpdated;
242 else
244 msg->base.state = MsgStateFinalized;
245 if (msg->base.stream_info.cbContent == 0xffffffff)
247 BYTE indefinite_trailer[6] = { 0 };
249 ret = msg->base.stream_info.pfnStreamOutput(
250 msg->base.stream_info.pvArg, (BYTE *)pbData, cbData,
251 FALSE);
252 if (ret)
253 ret = msg->base.stream_info.pfnStreamOutput(
254 msg->base.stream_info.pvArg, indefinite_trailer,
255 sizeof(indefinite_trailer), TRUE);
257 else
258 ret = msg->base.stream_info.pfnStreamOutput(
259 msg->base.stream_info.pvArg, (BYTE *)pbData, cbData, TRUE);
262 __EXCEPT_PAGE_FAULT
264 SetLastError(STATUS_ACCESS_VIOLATION);
265 ret = FALSE;
267 __ENDTRY;
269 else
271 if (!fFinal)
273 if (msg->base.open_flags & CMSG_DETACHED_FLAG)
274 SetLastError(E_INVALIDARG);
275 else
276 SetLastError(CRYPT_E_MSG_ERROR);
278 else
280 msg->base.state = MsgStateFinalized;
281 if (!cbData)
282 SetLastError(E_INVALIDARG);
283 else
285 CRYPT_DATA_BLOB blob = { cbData, (LPBYTE)pbData };
287 /* non-streamed data messages don't allow non-final updates,
288 * don't bother checking whether data already exist, they can't.
290 ret = CryptEncodeObjectEx(X509_ASN_ENCODING, X509_OCTET_STRING,
291 &blob, CRYPT_ENCODE_ALLOC_FLAG, NULL, &msg->bare_content,
292 &msg->bare_content_len);
296 return ret;
299 static BOOL CRYPT_CopyParam(void *pvData, DWORD *pcbData, const void *src,
300 DWORD len)
302 BOOL ret = TRUE;
304 if (!pvData)
305 *pcbData = len;
306 else if (*pcbData < len)
308 *pcbData = len;
309 SetLastError(ERROR_MORE_DATA);
310 ret = FALSE;
312 else
314 *pcbData = len;
315 memcpy(pvData, src, len);
317 return ret;
320 static BOOL CDataEncodeMsg_GetParam(HCRYPTMSG hCryptMsg, DWORD dwParamType,
321 DWORD dwIndex, void *pvData, DWORD *pcbData)
323 CDataEncodeMsg *msg = hCryptMsg;
324 BOOL ret = FALSE;
326 switch (dwParamType)
328 case CMSG_CONTENT_PARAM:
329 if (msg->base.streamed)
330 SetLastError(E_INVALIDARG);
331 else
333 CRYPT_CONTENT_INFO info;
334 char rsa_data[] = "1.2.840.113549.1.7.1";
336 info.pszObjId = rsa_data;
337 info.Content.cbData = msg->bare_content_len;
338 info.Content.pbData = msg->bare_content;
339 ret = CryptEncodeObject(X509_ASN_ENCODING, PKCS_CONTENT_INFO, &info,
340 pvData, pcbData);
342 break;
343 case CMSG_BARE_CONTENT_PARAM:
344 if (msg->base.streamed)
345 SetLastError(E_INVALIDARG);
346 else
347 ret = CRYPT_CopyParam(pvData, pcbData, msg->bare_content,
348 msg->bare_content_len);
349 break;
350 default:
351 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
353 return ret;
356 static HCRYPTMSG CDataEncodeMsg_Open(DWORD dwFlags, const void *pvMsgEncodeInfo,
357 LPSTR pszInnerContentObjID, PCMSG_STREAM_INFO pStreamInfo)
359 CDataEncodeMsg *msg;
361 if (pvMsgEncodeInfo)
363 SetLastError(E_INVALIDARG);
364 return NULL;
366 msg = CryptMemAlloc(sizeof(CDataEncodeMsg));
367 if (msg)
369 CryptMsgBase_Init((CryptMsgBase *)msg, dwFlags, pStreamInfo,
370 CDataEncodeMsg_Close, CDataEncodeMsg_GetParam, CDataEncodeMsg_Update,
371 CRYPT_DefaultMsgControl);
372 msg->bare_content_len = sizeof(empty_data_content);
373 msg->bare_content = (LPBYTE)empty_data_content;
375 return msg;
378 typedef struct _CHashEncodeMsg
380 CryptMsgBase base;
381 HCRYPTPROV prov;
382 HCRYPTHASH hash;
383 CRYPT_DATA_BLOB data;
384 } CHashEncodeMsg;
386 static void CHashEncodeMsg_Close(HCRYPTMSG hCryptMsg)
388 CHashEncodeMsg *msg = hCryptMsg;
390 CryptMemFree(msg->data.pbData);
391 CryptDestroyHash(msg->hash);
392 if (msg->base.open_flags & CMSG_CRYPT_RELEASE_CONTEXT_FLAG)
393 CryptReleaseContext(msg->prov, 0);
396 static BOOL CRYPT_EncodePKCSDigestedData(CHashEncodeMsg *msg, void *pvData,
397 DWORD *pcbData)
399 BOOL ret;
400 ALG_ID algID;
401 DWORD size = sizeof(algID);
403 ret = CryptGetHashParam(msg->hash, HP_ALGID, (BYTE *)&algID, &size, 0);
404 if (ret)
406 CRYPT_DIGESTED_DATA digestedData = { 0 };
407 char oid_rsa_data[] = szOID_RSA_data;
409 digestedData.version = CMSG_HASHED_DATA_PKCS_1_5_VERSION;
410 digestedData.DigestAlgorithm.pszObjId = (LPSTR)CertAlgIdToOID(algID);
411 /* FIXME: what about digestedData.DigestAlgorithm.Parameters? */
412 /* Quirk: OID is only encoded messages if an update has happened */
413 if (msg->base.state != MsgStateInit)
414 digestedData.ContentInfo.pszObjId = oid_rsa_data;
415 if (!(msg->base.open_flags & CMSG_DETACHED_FLAG) && msg->data.cbData)
417 ret = CRYPT_AsnEncodeOctets(0, NULL, &msg->data,
418 CRYPT_ENCODE_ALLOC_FLAG, NULL,
419 (LPBYTE)&digestedData.ContentInfo.Content.pbData,
420 &digestedData.ContentInfo.Content.cbData);
422 if (msg->base.state == MsgStateFinalized)
424 size = sizeof(DWORD);
425 ret = CryptGetHashParam(msg->hash, HP_HASHSIZE,
426 (LPBYTE)&digestedData.hash.cbData, &size, 0);
427 if (ret)
429 digestedData.hash.pbData = CryptMemAlloc(
430 digestedData.hash.cbData);
431 ret = CryptGetHashParam(msg->hash, HP_HASHVAL,
432 digestedData.hash.pbData, &digestedData.hash.cbData, 0);
435 if (ret)
436 ret = CRYPT_AsnEncodePKCSDigestedData(&digestedData, pvData,
437 pcbData);
438 CryptMemFree(digestedData.hash.pbData);
439 LocalFree(digestedData.ContentInfo.Content.pbData);
441 return ret;
444 static BOOL CHashEncodeMsg_GetParam(HCRYPTMSG hCryptMsg, DWORD dwParamType,
445 DWORD dwIndex, void *pvData, DWORD *pcbData)
447 CHashEncodeMsg *msg = hCryptMsg;
448 BOOL ret = FALSE;
450 TRACE("(%p, %d, %d, %p, %p)\n", hCryptMsg, dwParamType, dwIndex,
451 pvData, pcbData);
453 switch (dwParamType)
455 case CMSG_BARE_CONTENT_PARAM:
456 if (msg->base.streamed)
457 SetLastError(E_INVALIDARG);
458 else
459 ret = CRYPT_EncodePKCSDigestedData(msg, pvData, pcbData);
460 break;
461 case CMSG_CONTENT_PARAM:
463 CRYPT_CONTENT_INFO info;
465 ret = CryptMsgGetParam(hCryptMsg, CMSG_BARE_CONTENT_PARAM, 0, NULL,
466 &info.Content.cbData);
467 if (ret)
469 info.Content.pbData = CryptMemAlloc(info.Content.cbData);
470 if (info.Content.pbData)
472 ret = CryptMsgGetParam(hCryptMsg, CMSG_BARE_CONTENT_PARAM, 0,
473 info.Content.pbData, &info.Content.cbData);
474 if (ret)
476 char oid_rsa_hashed[] = szOID_RSA_hashedData;
478 info.pszObjId = oid_rsa_hashed;
479 ret = CryptEncodeObjectEx(X509_ASN_ENCODING,
480 PKCS_CONTENT_INFO, &info, 0, NULL, pvData, pcbData);
482 CryptMemFree(info.Content.pbData);
484 else
485 ret = FALSE;
487 break;
489 case CMSG_COMPUTED_HASH_PARAM:
490 ret = CryptGetHashParam(msg->hash, HP_HASHVAL, pvData, pcbData, 0);
491 break;
492 case CMSG_VERSION_PARAM:
493 if (msg->base.state != MsgStateFinalized)
494 SetLastError(CRYPT_E_MSG_ERROR);
495 else
497 DWORD version = CMSG_HASHED_DATA_PKCS_1_5_VERSION;
499 /* Since the data are always encoded as octets, the version is
500 * always 0 (see rfc3852, section 7)
502 ret = CRYPT_CopyParam(pvData, pcbData, &version, sizeof(version));
504 break;
505 default:
506 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
508 return ret;
511 static BOOL CHashEncodeMsg_Update(HCRYPTMSG hCryptMsg, const BYTE *pbData,
512 DWORD cbData, BOOL fFinal)
514 CHashEncodeMsg *msg = hCryptMsg;
515 BOOL ret = FALSE;
517 TRACE("(%p, %p, %d, %d)\n", hCryptMsg, pbData, cbData, fFinal);
519 if (msg->base.state == MsgStateFinalized)
520 SetLastError(CRYPT_E_MSG_ERROR);
521 else if (msg->base.streamed || (msg->base.open_flags & CMSG_DETACHED_FLAG))
523 /* Doesn't do much, as stream output is never called, and you
524 * can't get the content.
526 ret = CryptHashData(msg->hash, pbData, cbData, 0);
527 msg->base.state = fFinal ? MsgStateFinalized : MsgStateUpdated;
529 else
531 if (!fFinal)
532 SetLastError(CRYPT_E_MSG_ERROR);
533 else
535 ret = CryptHashData(msg->hash, pbData, cbData, 0);
536 if (ret)
538 msg->data.pbData = CryptMemAlloc(cbData);
539 if (msg->data.pbData)
541 memcpy(msg->data.pbData + msg->data.cbData, pbData, cbData);
542 msg->data.cbData += cbData;
544 else
545 ret = FALSE;
547 msg->base.state = MsgStateFinalized;
550 return ret;
553 static HCRYPTMSG CHashEncodeMsg_Open(DWORD dwFlags, const void *pvMsgEncodeInfo,
554 LPSTR pszInnerContentObjID, PCMSG_STREAM_INFO pStreamInfo)
556 CHashEncodeMsg *msg;
557 const CMSG_HASHED_ENCODE_INFO *info = pvMsgEncodeInfo;
558 HCRYPTPROV prov;
559 ALG_ID algID;
561 if (info->cbSize != sizeof(CMSG_HASHED_ENCODE_INFO))
563 SetLastError(E_INVALIDARG);
564 return NULL;
566 if (!(algID = CertOIDToAlgId(info->HashAlgorithm.pszObjId)))
568 SetLastError(CRYPT_E_UNKNOWN_ALGO);
569 return NULL;
571 if (info->hCryptProv)
572 prov = info->hCryptProv;
573 else
575 prov = CRYPT_GetDefaultProvider();
576 dwFlags &= ~CMSG_CRYPT_RELEASE_CONTEXT_FLAG;
578 msg = CryptMemAlloc(sizeof(CHashEncodeMsg));
579 if (msg)
581 CryptMsgBase_Init((CryptMsgBase *)msg, dwFlags, pStreamInfo,
582 CHashEncodeMsg_Close, CHashEncodeMsg_GetParam, CHashEncodeMsg_Update,
583 CRYPT_DefaultMsgControl);
584 msg->prov = prov;
585 msg->data.cbData = 0;
586 msg->data.pbData = NULL;
587 if (!CryptCreateHash(prov, algID, 0, 0, &msg->hash))
589 CryptMsgClose(msg);
590 msg = NULL;
593 return msg;
596 typedef struct _CMSG_SIGNER_ENCODE_INFO_WITH_CMS
598 DWORD cbSize;
599 PCERT_INFO pCertInfo;
600 HCRYPTPROV hCryptProv;
601 DWORD dwKeySpec;
602 CRYPT_ALGORITHM_IDENTIFIER HashAlgorithm;
603 void *pvHashAuxInfo;
604 DWORD cAuthAttr;
605 PCRYPT_ATTRIBUTE rgAuthAttr;
606 DWORD cUnauthAttr;
607 PCRYPT_ATTRIBUTE rgUnauthAttr;
608 CERT_ID SignerId;
609 CRYPT_ALGORITHM_IDENTIFIER HashEncryptionAlgorithm;
610 void *pvHashEncryptionAuxInfo;
611 } CMSG_SIGNER_ENCODE_INFO_WITH_CMS, *PCMSG_SIGNER_ENCODE_INFO_WITH_CMS;
613 typedef struct _CMSG_SIGNED_ENCODE_INFO_WITH_CMS
615 DWORD cbSize;
616 DWORD cSigners;
617 PCMSG_SIGNER_ENCODE_INFO_WITH_CMS rgSigners;
618 DWORD cCertEncoded;
619 PCERT_BLOB rgCertEncoded;
620 DWORD cCrlEncoded;
621 PCRL_BLOB rgCrlEncoded;
622 DWORD cAttrCertEncoded;
623 PCERT_BLOB rgAttrCertEncoded;
624 } CMSG_SIGNED_ENCODE_INFO_WITH_CMS, *PCMSG_SIGNED_ENCODE_INFO_WITH_CMS;
626 static BOOL CRYPT_IsValidSigner(const CMSG_SIGNER_ENCODE_INFO_WITH_CMS *signer)
628 if (signer->cbSize != sizeof(CMSG_SIGNER_ENCODE_INFO) &&
629 signer->cbSize != sizeof(CMSG_SIGNER_ENCODE_INFO_WITH_CMS))
631 SetLastError(E_INVALIDARG);
632 return FALSE;
634 if (signer->cbSize == sizeof(CMSG_SIGNER_ENCODE_INFO))
636 if (!signer->pCertInfo->SerialNumber.cbData)
638 SetLastError(E_INVALIDARG);
639 return FALSE;
641 if (!signer->pCertInfo->Issuer.cbData)
643 SetLastError(E_INVALIDARG);
644 return FALSE;
647 else if (signer->cbSize == sizeof(CMSG_SIGNER_ENCODE_INFO_WITH_CMS))
649 switch (signer->SignerId.dwIdChoice)
651 case 0:
652 if (!signer->pCertInfo->SerialNumber.cbData)
654 SetLastError(E_INVALIDARG);
655 return FALSE;
657 if (!signer->pCertInfo->Issuer.cbData)
659 SetLastError(E_INVALIDARG);
660 return FALSE;
662 break;
663 case CERT_ID_ISSUER_SERIAL_NUMBER:
664 if (!signer->SignerId.u.IssuerSerialNumber.SerialNumber.cbData)
666 SetLastError(E_INVALIDARG);
667 return FALSE;
669 if (!signer->SignerId.u.IssuerSerialNumber.Issuer.cbData)
671 SetLastError(E_INVALIDARG);
672 return FALSE;
674 break;
675 case CERT_ID_KEY_IDENTIFIER:
676 if (!signer->SignerId.u.KeyId.cbData)
678 SetLastError(E_INVALIDARG);
679 return FALSE;
681 break;
682 default:
683 SetLastError(E_INVALIDARG);
685 if (signer->HashEncryptionAlgorithm.pszObjId)
687 FIXME("CMSG_SIGNER_ENCODE_INFO with CMS fields unsupported\n");
688 return FALSE;
691 if (!signer->hCryptProv)
693 SetLastError(E_INVALIDARG);
694 return FALSE;
696 if (!CertOIDToAlgId(signer->HashAlgorithm.pszObjId))
698 SetLastError(CRYPT_E_UNKNOWN_ALGO);
699 return FALSE;
701 return TRUE;
704 static BOOL CRYPT_ConstructBlob(CRYPT_DATA_BLOB *out, const CRYPT_DATA_BLOB *in)
706 BOOL ret = TRUE;
708 out->cbData = in->cbData;
709 if (out->cbData)
711 out->pbData = CryptMemAlloc(out->cbData);
712 if (out->pbData)
713 memcpy(out->pbData, in->pbData, out->cbData);
714 else
715 ret = FALSE;
717 else
718 out->pbData = NULL;
719 return ret;
722 static BOOL CRYPT_ConstructBlobArray(DWORD *outCBlobs,
723 PCRYPT_DATA_BLOB *outPBlobs, DWORD cBlobs, const CRYPT_DATA_BLOB *pBlobs)
725 BOOL ret = TRUE;
727 *outCBlobs = cBlobs;
728 if (cBlobs)
730 *outPBlobs = CryptMemAlloc(cBlobs * sizeof(CRYPT_DATA_BLOB));
731 if (*outPBlobs)
733 DWORD i;
735 memset(*outPBlobs, 0, cBlobs * sizeof(CRYPT_DATA_BLOB));
736 for (i = 0; ret && i < cBlobs; i++)
737 ret = CRYPT_ConstructBlob(&(*outPBlobs)[i], &pBlobs[i]);
739 else
740 ret = FALSE;
742 return ret;
745 static void CRYPT_FreeBlobArray(DWORD cBlobs, PCRYPT_DATA_BLOB blobs)
747 DWORD i;
749 for (i = 0; i < cBlobs; i++)
750 CryptMemFree(blobs[i].pbData);
751 CryptMemFree(blobs);
754 static BOOL CRYPT_ConstructAttribute(CRYPT_ATTRIBUTE *out,
755 const CRYPT_ATTRIBUTE *in)
757 BOOL ret;
759 out->pszObjId = CryptMemAlloc(strlen(in->pszObjId) + 1);
760 if (out->pszObjId)
762 strcpy(out->pszObjId, in->pszObjId);
763 ret = CRYPT_ConstructBlobArray(&out->cValue, &out->rgValue,
764 in->cValue, in->rgValue);
766 else
767 ret = FALSE;
768 return ret;
771 static BOOL CRYPT_ConstructAttributes(CRYPT_ATTRIBUTES *out,
772 const CRYPT_ATTRIBUTES *in)
774 BOOL ret = TRUE;
776 out->cAttr = in->cAttr;
777 if (out->cAttr)
779 out->rgAttr = CryptMemAlloc(out->cAttr * sizeof(CRYPT_ATTRIBUTE));
780 if (out->rgAttr)
782 DWORD i;
784 memset(out->rgAttr, 0, out->cAttr * sizeof(CRYPT_ATTRIBUTE));
785 for (i = 0; ret && i < out->cAttr; i++)
786 ret = CRYPT_ConstructAttribute(&out->rgAttr[i], &in->rgAttr[i]);
788 else
789 ret = FALSE;
791 else
792 out->rgAttr = NULL;
793 return ret;
796 /* Constructs a CMSG_CMS_SIGNER_INFO from a CMSG_SIGNER_ENCODE_INFO_WITH_CMS. */
797 static BOOL CSignerInfo_Construct(CMSG_CMS_SIGNER_INFO *info,
798 const CMSG_SIGNER_ENCODE_INFO_WITH_CMS *in)
800 BOOL ret;
802 if (in->cbSize == sizeof(CMSG_SIGNER_ENCODE_INFO))
804 info->dwVersion = CMSG_SIGNER_INFO_V1;
805 ret = CRYPT_ConstructBlob(&info->SignerId.u.IssuerSerialNumber.Issuer,
806 &in->pCertInfo->Issuer);
807 if (ret)
808 ret = CRYPT_ConstructBlob(
809 &info->SignerId.u.IssuerSerialNumber.SerialNumber,
810 &in->pCertInfo->SerialNumber);
811 info->SignerId.dwIdChoice = CERT_ID_ISSUER_SERIAL_NUMBER;
812 info->HashEncryptionAlgorithm.pszObjId =
813 in->pCertInfo->SubjectPublicKeyInfo.Algorithm.pszObjId;
814 if (ret)
815 ret = CRYPT_ConstructBlob(&info->HashEncryptionAlgorithm.Parameters,
816 &in->pCertInfo->SubjectPublicKeyInfo.Algorithm.Parameters);
818 else
820 const CRYPT_ALGORITHM_IDENTIFIER *pEncrAlg;
822 /* Implicitly in->cbSize == sizeof(CMSG_SIGNER_ENCODE_INFO_WITH_CMS).
823 * See CRYPT_IsValidSigner.
825 if (!in->SignerId.dwIdChoice)
827 info->dwVersion = CMSG_SIGNER_INFO_V1;
828 ret = CRYPT_ConstructBlob(&info->SignerId.u.IssuerSerialNumber.Issuer,
829 &in->pCertInfo->Issuer);
830 if (ret)
831 ret = CRYPT_ConstructBlob(
832 &info->SignerId.u.IssuerSerialNumber.SerialNumber,
833 &in->pCertInfo->SerialNumber);
834 info->SignerId.dwIdChoice = CERT_ID_ISSUER_SERIAL_NUMBER;
836 else if (in->SignerId.dwIdChoice == CERT_ID_ISSUER_SERIAL_NUMBER)
838 info->dwVersion = CMSG_SIGNER_INFO_V1;
839 info->SignerId.dwIdChoice = CERT_ID_ISSUER_SERIAL_NUMBER;
840 ret = CRYPT_ConstructBlob(&info->SignerId.u.IssuerSerialNumber.Issuer,
841 &in->SignerId.u.IssuerSerialNumber.Issuer);
842 if (ret)
843 ret = CRYPT_ConstructBlob(
844 &info->SignerId.u.IssuerSerialNumber.SerialNumber,
845 &in->SignerId.u.IssuerSerialNumber.SerialNumber);
847 else
849 /* Implicitly dwIdChoice == CERT_ID_KEY_IDENTIFIER */
850 info->dwVersion = CMSG_SIGNER_INFO_V3;
851 info->SignerId.dwIdChoice = CERT_ID_KEY_IDENTIFIER;
852 ret = CRYPT_ConstructBlob(&info->SignerId.u.KeyId,
853 &in->SignerId.u.KeyId);
855 pEncrAlg = in->HashEncryptionAlgorithm.pszObjId ?
856 &in->HashEncryptionAlgorithm :
857 &in->pCertInfo->SubjectPublicKeyInfo.Algorithm;
858 info->HashEncryptionAlgorithm.pszObjId = pEncrAlg->pszObjId;
859 if (ret)
860 ret = CRYPT_ConstructBlob(&info->HashEncryptionAlgorithm.Parameters,
861 &pEncrAlg->Parameters);
863 /* Assumption: algorithm IDs will point to static strings, not
864 * stack-based ones, so copying the pointer values is safe.
866 info->HashAlgorithm.pszObjId = in->HashAlgorithm.pszObjId;
867 if (ret)
868 ret = CRYPT_ConstructBlob(&info->HashAlgorithm.Parameters,
869 &in->HashAlgorithm.Parameters);
870 if (ret)
871 ret = CRYPT_ConstructAttributes(&info->AuthAttrs,
872 (CRYPT_ATTRIBUTES *)&in->cAuthAttr);
873 if (ret)
874 ret = CRYPT_ConstructAttributes(&info->UnauthAttrs,
875 (CRYPT_ATTRIBUTES *)&in->cUnauthAttr);
876 return ret;
879 static void CSignerInfo_Free(CMSG_CMS_SIGNER_INFO *info)
881 DWORD i, j;
883 if (info->SignerId.dwIdChoice == CERT_ID_ISSUER_SERIAL_NUMBER)
885 CryptMemFree(info->SignerId.u.IssuerSerialNumber.Issuer.pbData);
886 CryptMemFree(info->SignerId.u.IssuerSerialNumber.SerialNumber.pbData);
888 else
889 CryptMemFree(info->SignerId.u.KeyId.pbData);
890 CryptMemFree(info->HashAlgorithm.Parameters.pbData);
891 CryptMemFree(info->HashEncryptionAlgorithm.Parameters.pbData);
892 CryptMemFree(info->EncryptedHash.pbData);
893 for (i = 0; i < info->AuthAttrs.cAttr; i++)
895 for (j = 0; j < info->AuthAttrs.rgAttr[i].cValue; j++)
896 CryptMemFree(info->AuthAttrs.rgAttr[i].rgValue[j].pbData);
897 CryptMemFree(info->AuthAttrs.rgAttr[i].rgValue);
898 CryptMemFree(info->AuthAttrs.rgAttr[i].pszObjId);
900 CryptMemFree(info->AuthAttrs.rgAttr);
901 for (i = 0; i < info->UnauthAttrs.cAttr; i++)
903 for (j = 0; j < info->UnauthAttrs.rgAttr[i].cValue; j++)
904 CryptMemFree(info->UnauthAttrs.rgAttr[i].rgValue[j].pbData);
905 CryptMemFree(info->UnauthAttrs.rgAttr[i].rgValue);
906 CryptMemFree(info->UnauthAttrs.rgAttr[i].pszObjId);
908 CryptMemFree(info->UnauthAttrs.rgAttr);
911 typedef struct _CSignerHandles
913 HCRYPTHASH contentHash;
914 HCRYPTHASH authAttrHash;
915 } CSignerHandles;
917 typedef struct _CSignedMsgData
919 CRYPT_SIGNED_INFO *info;
920 DWORD cSignerHandle;
921 CSignerHandles *signerHandles;
922 } CSignedMsgData;
924 /* Constructs the signer handles for the signerIndex'th signer of msg_data.
925 * Assumes signerIndex is a valid idnex, and that msg_data's info has already
926 * been constructed.
928 static BOOL CSignedMsgData_ConstructSignerHandles(CSignedMsgData *msg_data,
929 DWORD signerIndex, HCRYPTPROV crypt_prov)
931 ALG_ID algID;
932 BOOL ret;
934 algID = CertOIDToAlgId(
935 msg_data->info->rgSignerInfo[signerIndex].HashAlgorithm.pszObjId);
936 ret = CryptCreateHash(crypt_prov, algID, 0, 0,
937 &msg_data->signerHandles->contentHash);
938 if (ret && msg_data->info->rgSignerInfo[signerIndex].AuthAttrs.cAttr > 0)
939 ret = CryptCreateHash(crypt_prov, algID, 0, 0,
940 &msg_data->signerHandles->authAttrHash);
941 return ret;
944 /* Allocates a CSignedMsgData's handles. Assumes its info has already been
945 * constructed.
947 static BOOL CSignedMsgData_AllocateHandles(CSignedMsgData *msg_data)
949 BOOL ret = TRUE;
951 if (msg_data->info->cSignerInfo)
953 msg_data->signerHandles =
954 CryptMemAlloc(msg_data->info->cSignerInfo * sizeof(CSignerHandles));
955 if (msg_data->signerHandles)
957 msg_data->cSignerHandle = msg_data->info->cSignerInfo;
958 memset(msg_data->signerHandles, 0,
959 msg_data->info->cSignerInfo * sizeof(CSignerHandles));
961 else
963 msg_data->cSignerHandle = 0;
964 ret = FALSE;
967 else
969 msg_data->cSignerHandle = 0;
970 msg_data->signerHandles = NULL;
972 return ret;
975 static void CSignedMsgData_CloseHandles(CSignedMsgData *msg_data)
977 DWORD i;
979 for (i = 0; i < msg_data->cSignerHandle; i++)
981 if (msg_data->signerHandles[i].contentHash)
982 CryptDestroyHash(msg_data->signerHandles[i].contentHash);
983 if (msg_data->signerHandles[i].authAttrHash)
984 CryptDestroyHash(msg_data->signerHandles[i].authAttrHash);
986 CryptMemFree(msg_data->signerHandles);
987 msg_data->signerHandles = NULL;
988 msg_data->cSignerHandle = 0;
991 static BOOL CSignedMsgData_UpdateHash(CSignedMsgData *msg_data,
992 const BYTE *pbData, DWORD cbData)
994 DWORD i;
995 BOOL ret = TRUE;
997 for (i = 0; ret && i < msg_data->cSignerHandle; i++)
998 ret = CryptHashData(msg_data->signerHandles[i].contentHash, pbData,
999 cbData, 0);
1000 return ret;
1003 static BOOL CRYPT_AppendAttribute(CRYPT_ATTRIBUTES *out,
1004 const CRYPT_ATTRIBUTE *in)
1006 BOOL ret = FALSE;
1008 out->rgAttr = CryptMemRealloc(out->rgAttr,
1009 (out->cAttr + 1) * sizeof(CRYPT_ATTRIBUTE));
1010 if (out->rgAttr)
1012 ret = CRYPT_ConstructAttribute(&out->rgAttr[out->cAttr], in);
1013 if (ret)
1014 out->cAttr++;
1016 return ret;
1019 static BOOL CSignedMsgData_AppendMessageDigestAttribute(
1020 CSignedMsgData *msg_data, DWORD signerIndex)
1022 BOOL ret;
1023 DWORD size;
1024 CRYPT_HASH_BLOB hash = { 0, NULL }, encodedHash = { 0, NULL };
1025 char messageDigest[] = szOID_RSA_messageDigest;
1026 CRYPT_ATTRIBUTE messageDigestAttr = { messageDigest, 1, &encodedHash };
1028 size = sizeof(DWORD);
1029 ret = CryptGetHashParam(
1030 msg_data->signerHandles[signerIndex].contentHash, HP_HASHSIZE,
1031 (LPBYTE)&hash.cbData, &size, 0);
1032 if (ret)
1034 hash.pbData = CryptMemAlloc(hash.cbData);
1035 ret = CryptGetHashParam(
1036 msg_data->signerHandles[signerIndex].contentHash, HP_HASHVAL,
1037 hash.pbData, &hash.cbData, 0);
1038 if (ret)
1040 ret = CRYPT_AsnEncodeOctets(0, NULL, &hash, CRYPT_ENCODE_ALLOC_FLAG,
1041 NULL, (LPBYTE)&encodedHash.pbData, &encodedHash.cbData);
1042 if (ret)
1044 ret = CRYPT_AppendAttribute(
1045 &msg_data->info->rgSignerInfo[signerIndex].AuthAttrs,
1046 &messageDigestAttr);
1047 LocalFree(encodedHash.pbData);
1050 CryptMemFree(hash.pbData);
1052 return ret;
1055 typedef enum {
1056 Sign,
1057 Verify
1058 } SignOrVerify;
1060 static BOOL CSignedMsgData_UpdateAuthenticatedAttributes(
1061 CSignedMsgData *msg_data, SignOrVerify flag)
1063 DWORD i;
1064 BOOL ret = TRUE;
1066 TRACE("(%p)\n", msg_data);
1068 for (i = 0; ret && i < msg_data->info->cSignerInfo; i++)
1070 if (msg_data->info->rgSignerInfo[i].AuthAttrs.cAttr)
1072 if (flag == Sign)
1074 BYTE oid_rsa_data_encoded[] = { 0x06,0x09,0x2a,0x86,0x48,0x86,
1075 0xf7,0x0d,0x01,0x07,0x01 };
1076 CRYPT_DATA_BLOB content = { sizeof(oid_rsa_data_encoded),
1077 oid_rsa_data_encoded };
1078 char contentType[] = szOID_RSA_contentType;
1079 CRYPT_ATTRIBUTE contentTypeAttr = { contentType, 1, &content };
1081 /* FIXME: does this depend on inner OID? */
1082 ret = CRYPT_AppendAttribute(
1083 &msg_data->info->rgSignerInfo[i].AuthAttrs, &contentTypeAttr);
1084 if (ret)
1085 ret = CSignedMsgData_AppendMessageDigestAttribute(msg_data,
1088 if (ret)
1090 LPBYTE encodedAttrs;
1091 DWORD size;
1093 ret = CryptEncodeObjectEx(X509_ASN_ENCODING, PKCS_ATTRIBUTES,
1094 &msg_data->info->rgSignerInfo[i].AuthAttrs,
1095 CRYPT_ENCODE_ALLOC_FLAG, NULL, &encodedAttrs, &size);
1096 if (ret)
1098 ret = CryptHashData(
1099 msg_data->signerHandles[i].authAttrHash, encodedAttrs,
1100 size, 0);
1101 LocalFree(encodedAttrs);
1106 TRACE("returning %d\n", ret);
1107 return ret;
1110 static void CRYPT_ReverseBytes(CRYPT_HASH_BLOB *hash)
1112 DWORD i;
1113 BYTE tmp;
1115 for (i = 0; i < hash->cbData / 2; i++)
1117 tmp = hash->pbData[hash->cbData - i - 1];
1118 hash->pbData[hash->cbData - i - 1] = hash->pbData[i];
1119 hash->pbData[i] = tmp;
1123 static BOOL CSignedMsgData_Sign(CSignedMsgData *msg_data)
1125 DWORD i;
1126 BOOL ret = TRUE;
1128 TRACE("(%p)\n", msg_data);
1130 for (i = 0; ret && i < msg_data->info->cSignerInfo; i++)
1132 HCRYPTHASH hash;
1134 if (msg_data->info->rgSignerInfo[i].AuthAttrs.cAttr)
1135 hash = msg_data->signerHandles[i].authAttrHash;
1136 else
1137 hash = msg_data->signerHandles[i].contentHash;
1138 ret = CryptSignHashW(hash, AT_SIGNATURE, NULL, 0, NULL,
1139 &msg_data->info->rgSignerInfo[i].EncryptedHash.cbData);
1140 if (ret)
1142 msg_data->info->rgSignerInfo[i].EncryptedHash.pbData =
1143 CryptMemAlloc(
1144 msg_data->info->rgSignerInfo[i].EncryptedHash.cbData);
1145 if (msg_data->info->rgSignerInfo[i].EncryptedHash.pbData)
1147 ret = CryptSignHashW(hash, AT_SIGNATURE, NULL, 0,
1148 msg_data->info->rgSignerInfo[i].EncryptedHash.pbData,
1149 &msg_data->info->rgSignerInfo[i].EncryptedHash.cbData);
1150 if (ret)
1151 CRYPT_ReverseBytes(
1152 &msg_data->info->rgSignerInfo[i].EncryptedHash);
1154 else
1155 ret = FALSE;
1158 return ret;
1161 static BOOL CSignedMsgData_Update(CSignedMsgData *msg_data,
1162 const BYTE *pbData, DWORD cbData, BOOL fFinal, SignOrVerify flag)
1164 BOOL ret = CSignedMsgData_UpdateHash(msg_data, pbData, cbData);
1166 if (ret && fFinal)
1168 ret = CSignedMsgData_UpdateAuthenticatedAttributes(msg_data, flag);
1169 if (ret && flag == Sign)
1170 ret = CSignedMsgData_Sign(msg_data);
1172 return ret;
1175 typedef struct _CSignedEncodeMsg
1177 CryptMsgBase base;
1178 LPSTR innerOID;
1179 CRYPT_DATA_BLOB data;
1180 CSignedMsgData msg_data;
1181 } CSignedEncodeMsg;
1183 static void CSignedEncodeMsg_Close(HCRYPTMSG hCryptMsg)
1185 CSignedEncodeMsg *msg = hCryptMsg;
1186 DWORD i;
1188 CryptMemFree(msg->innerOID);
1189 CryptMemFree(msg->data.pbData);
1190 CRYPT_FreeBlobArray(msg->msg_data.info->cCertEncoded,
1191 msg->msg_data.info->rgCertEncoded);
1192 CRYPT_FreeBlobArray(msg->msg_data.info->cCrlEncoded,
1193 msg->msg_data.info->rgCrlEncoded);
1194 for (i = 0; i < msg->msg_data.info->cSignerInfo; i++)
1195 CSignerInfo_Free(&msg->msg_data.info->rgSignerInfo[i]);
1196 CSignedMsgData_CloseHandles(&msg->msg_data);
1197 CryptMemFree(msg->msg_data.info->rgSignerInfo);
1198 CryptMemFree(msg->msg_data.info);
1201 static BOOL CSignedEncodeMsg_GetParam(HCRYPTMSG hCryptMsg, DWORD dwParamType,
1202 DWORD dwIndex, void *pvData, DWORD *pcbData)
1204 CSignedEncodeMsg *msg = hCryptMsg;
1205 BOOL ret = FALSE;
1207 switch (dwParamType)
1209 case CMSG_CONTENT_PARAM:
1211 CRYPT_CONTENT_INFO info;
1213 ret = CryptMsgGetParam(hCryptMsg, CMSG_BARE_CONTENT_PARAM, 0, NULL,
1214 &info.Content.cbData);
1215 if (ret)
1217 info.Content.pbData = CryptMemAlloc(info.Content.cbData);
1218 if (info.Content.pbData)
1220 ret = CryptMsgGetParam(hCryptMsg, CMSG_BARE_CONTENT_PARAM, 0,
1221 info.Content.pbData, &info.Content.cbData);
1222 if (ret)
1224 char oid_rsa_signed[] = szOID_RSA_signedData;
1226 info.pszObjId = oid_rsa_signed;
1227 ret = CryptEncodeObjectEx(X509_ASN_ENCODING,
1228 PKCS_CONTENT_INFO, &info, 0, NULL, pvData, pcbData);
1230 CryptMemFree(info.Content.pbData);
1232 else
1233 ret = FALSE;
1235 break;
1237 case CMSG_BARE_CONTENT_PARAM:
1239 CRYPT_SIGNED_INFO info;
1240 BOOL freeContent = FALSE;
1242 info = *msg->msg_data.info;
1243 if (!msg->innerOID || !strcmp(msg->innerOID, szOID_RSA_data))
1245 char oid_rsa_data[] = szOID_RSA_data;
1247 /* Quirk: OID is only encoded messages if an update has happened */
1248 if (msg->base.state != MsgStateInit)
1249 info.content.pszObjId = oid_rsa_data;
1250 else
1251 info.content.pszObjId = NULL;
1252 if (msg->data.cbData)
1254 CRYPT_DATA_BLOB blob = { msg->data.cbData, msg->data.pbData };
1256 ret = CryptEncodeObjectEx(X509_ASN_ENCODING, X509_OCTET_STRING,
1257 &blob, CRYPT_ENCODE_ALLOC_FLAG, NULL,
1258 &info.content.Content.pbData, &info.content.Content.cbData);
1259 freeContent = TRUE;
1261 else
1263 info.content.Content.cbData = 0;
1264 info.content.Content.pbData = NULL;
1265 ret = TRUE;
1268 else
1270 info.content.pszObjId = msg->innerOID;
1271 info.content.Content.cbData = msg->data.cbData;
1272 info.content.Content.pbData = msg->data.pbData;
1273 ret = TRUE;
1275 if (ret)
1277 ret = CRYPT_AsnEncodeCMSSignedInfo(&info, pvData, pcbData);
1278 if (freeContent)
1279 LocalFree(info.content.Content.pbData);
1281 break;
1283 case CMSG_COMPUTED_HASH_PARAM:
1284 if (dwIndex >= msg->msg_data.cSignerHandle)
1285 SetLastError(CRYPT_E_INVALID_INDEX);
1286 else
1287 ret = CryptGetHashParam(
1288 msg->msg_data.signerHandles[dwIndex].contentHash, HP_HASHVAL,
1289 pvData, pcbData, 0);
1290 break;
1291 case CMSG_ENCODED_SIGNER:
1292 if (dwIndex >= msg->msg_data.info->cSignerInfo)
1293 SetLastError(CRYPT_E_INVALID_INDEX);
1294 else
1295 ret = CryptEncodeObjectEx(X509_ASN_ENCODING | PKCS_7_ASN_ENCODING,
1296 CMS_SIGNER_INFO, &msg->msg_data.info->rgSignerInfo[dwIndex], 0,
1297 NULL, pvData, pcbData);
1298 break;
1299 case CMSG_VERSION_PARAM:
1300 ret = CRYPT_CopyParam(pvData, pcbData, &msg->msg_data.info->version,
1301 sizeof(msg->msg_data.info->version));
1302 break;
1303 default:
1304 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
1306 return ret;
1309 static BOOL CSignedEncodeMsg_Update(HCRYPTMSG hCryptMsg, const BYTE *pbData,
1310 DWORD cbData, BOOL fFinal)
1312 CSignedEncodeMsg *msg = hCryptMsg;
1313 BOOL ret = FALSE;
1315 if (msg->base.state == MsgStateFinalized)
1316 SetLastError(CRYPT_E_MSG_ERROR);
1317 else if (msg->base.streamed || (msg->base.open_flags & CMSG_DETACHED_FLAG))
1319 ret = CSignedMsgData_Update(&msg->msg_data, pbData, cbData, fFinal,
1320 Sign);
1321 if (msg->base.streamed)
1322 FIXME("streamed partial stub\n");
1323 msg->base.state = fFinal ? MsgStateFinalized : MsgStateUpdated;
1325 else
1327 if (!fFinal)
1328 SetLastError(CRYPT_E_MSG_ERROR);
1329 else
1331 if (cbData)
1333 msg->data.pbData = CryptMemAlloc(cbData);
1334 if (msg->data.pbData)
1336 memcpy(msg->data.pbData, pbData, cbData);
1337 msg->data.cbData = cbData;
1338 ret = TRUE;
1341 else
1342 ret = TRUE;
1343 if (ret)
1344 ret = CSignedMsgData_Update(&msg->msg_data, pbData, cbData,
1345 fFinal, Sign);
1346 msg->base.state = MsgStateFinalized;
1349 return ret;
1352 static HCRYPTMSG CSignedEncodeMsg_Open(DWORD dwFlags,
1353 const void *pvMsgEncodeInfo, LPCSTR pszInnerContentObjID,
1354 PCMSG_STREAM_INFO pStreamInfo)
1356 const CMSG_SIGNED_ENCODE_INFO_WITH_CMS *info = pvMsgEncodeInfo;
1357 DWORD i;
1358 CSignedEncodeMsg *msg;
1360 if (info->cbSize != sizeof(CMSG_SIGNED_ENCODE_INFO) &&
1361 info->cbSize != sizeof(CMSG_SIGNED_ENCODE_INFO_WITH_CMS))
1363 SetLastError(E_INVALIDARG);
1364 return NULL;
1366 if (info->cbSize == sizeof(CMSG_SIGNED_ENCODE_INFO_WITH_CMS) &&
1367 info->cAttrCertEncoded)
1369 FIXME("CMSG_SIGNED_ENCODE_INFO with CMS fields unsupported\n");
1370 return NULL;
1372 for (i = 0; i < info->cSigners; i++)
1373 if (!CRYPT_IsValidSigner(&info->rgSigners[i]))
1374 return NULL;
1375 msg = CryptMemAlloc(sizeof(CSignedEncodeMsg));
1376 if (msg)
1378 BOOL ret = TRUE;
1380 CryptMsgBase_Init((CryptMsgBase *)msg, dwFlags, pStreamInfo,
1381 CSignedEncodeMsg_Close, CSignedEncodeMsg_GetParam,
1382 CSignedEncodeMsg_Update, CRYPT_DefaultMsgControl);
1383 if (pszInnerContentObjID)
1385 msg->innerOID = CryptMemAlloc(strlen(pszInnerContentObjID) + 1);
1386 if (msg->innerOID)
1387 strcpy(msg->innerOID, pszInnerContentObjID);
1388 else
1389 ret = FALSE;
1391 else
1392 msg->innerOID = NULL;
1393 msg->data.cbData = 0;
1394 msg->data.pbData = NULL;
1395 if (ret)
1396 msg->msg_data.info = CryptMemAlloc(sizeof(CRYPT_SIGNED_INFO));
1397 else
1398 msg->msg_data.info = NULL;
1399 if (msg->msg_data.info)
1401 memset(msg->msg_data.info, 0, sizeof(CRYPT_SIGNED_INFO));
1402 msg->msg_data.info->version = CMSG_SIGNED_DATA_V1;
1404 else
1405 ret = FALSE;
1406 if (ret)
1408 if (info->cSigners)
1410 msg->msg_data.info->rgSignerInfo =
1411 CryptMemAlloc(info->cSigners * sizeof(CMSG_CMS_SIGNER_INFO));
1412 if (msg->msg_data.info->rgSignerInfo)
1414 msg->msg_data.info->cSignerInfo = info->cSigners;
1415 memset(msg->msg_data.info->rgSignerInfo, 0,
1416 msg->msg_data.info->cSignerInfo *
1417 sizeof(CMSG_CMS_SIGNER_INFO));
1418 ret = CSignedMsgData_AllocateHandles(&msg->msg_data);
1419 for (i = 0; ret && i < msg->msg_data.info->cSignerInfo; i++)
1421 if (info->rgSigners[i].SignerId.dwIdChoice ==
1422 CERT_ID_KEY_IDENTIFIER)
1423 msg->msg_data.info->version = CMSG_SIGNED_DATA_V3;
1424 ret = CSignerInfo_Construct(
1425 &msg->msg_data.info->rgSignerInfo[i],
1426 &info->rgSigners[i]);
1427 if (ret)
1429 ret = CSignedMsgData_ConstructSignerHandles(
1430 &msg->msg_data, i, info->rgSigners[i].hCryptProv);
1431 if (dwFlags & CMSG_CRYPT_RELEASE_CONTEXT_FLAG)
1432 CryptReleaseContext(info->rgSigners[i].hCryptProv,
1437 else
1438 ret = FALSE;
1440 else
1442 msg->msg_data.info->cSignerInfo = 0;
1443 msg->msg_data.signerHandles = NULL;
1444 msg->msg_data.cSignerHandle = 0;
1447 if (ret)
1448 ret = CRYPT_ConstructBlobArray(&msg->msg_data.info->cCertEncoded,
1449 &msg->msg_data.info->rgCertEncoded, info->cCertEncoded,
1450 info->rgCertEncoded);
1451 if (ret)
1452 ret = CRYPT_ConstructBlobArray(&msg->msg_data.info->cCrlEncoded,
1453 &msg->msg_data.info->rgCrlEncoded, info->cCrlEncoded,
1454 info->rgCrlEncoded);
1455 if (!ret)
1457 CSignedEncodeMsg_Close(msg);
1458 msg = NULL;
1461 return msg;
1464 HCRYPTMSG WINAPI CryptMsgOpenToEncode(DWORD dwMsgEncodingType, DWORD dwFlags,
1465 DWORD dwMsgType, const void *pvMsgEncodeInfo, LPSTR pszInnerContentObjID,
1466 PCMSG_STREAM_INFO pStreamInfo)
1468 HCRYPTMSG msg = NULL;
1470 TRACE("(%08x, %08x, %08x, %p, %s, %p)\n", dwMsgEncodingType, dwFlags,
1471 dwMsgType, pvMsgEncodeInfo, debugstr_a(pszInnerContentObjID), pStreamInfo);
1473 if (GET_CMSG_ENCODING_TYPE(dwMsgEncodingType) != PKCS_7_ASN_ENCODING)
1475 SetLastError(E_INVALIDARG);
1476 return NULL;
1478 switch (dwMsgType)
1480 case CMSG_DATA:
1481 msg = CDataEncodeMsg_Open(dwFlags, pvMsgEncodeInfo,
1482 pszInnerContentObjID, pStreamInfo);
1483 break;
1484 case CMSG_HASHED:
1485 msg = CHashEncodeMsg_Open(dwFlags, pvMsgEncodeInfo,
1486 pszInnerContentObjID, pStreamInfo);
1487 break;
1488 case CMSG_SIGNED:
1489 msg = CSignedEncodeMsg_Open(dwFlags, pvMsgEncodeInfo,
1490 pszInnerContentObjID, pStreamInfo);
1491 break;
1492 case CMSG_ENVELOPED:
1493 FIXME("unimplemented for type CMSG_ENVELOPED\n");
1494 break;
1495 case CMSG_SIGNED_AND_ENVELOPED:
1496 case CMSG_ENCRYPTED:
1497 /* defined but invalid, fall through */
1498 default:
1499 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
1501 return msg;
1504 typedef struct _CDecodeMsg
1506 CryptMsgBase base;
1507 DWORD type;
1508 HCRYPTPROV crypt_prov;
1509 union {
1510 HCRYPTHASH hash;
1511 CSignedMsgData signed_data;
1512 } u;
1513 CRYPT_DATA_BLOB msg_data;
1514 CRYPT_DATA_BLOB detached_data;
1515 PCONTEXT_PROPERTY_LIST properties;
1516 } CDecodeMsg;
1518 static void CDecodeMsg_Close(HCRYPTMSG hCryptMsg)
1520 CDecodeMsg *msg = hCryptMsg;
1522 if (msg->base.open_flags & CMSG_CRYPT_RELEASE_CONTEXT_FLAG)
1523 CryptReleaseContext(msg->crypt_prov, 0);
1524 switch (msg->type)
1526 case CMSG_HASHED:
1527 if (msg->u.hash)
1528 CryptDestroyHash(msg->u.hash);
1529 break;
1530 case CMSG_SIGNED:
1531 if (msg->u.signed_data.info)
1533 LocalFree(msg->u.signed_data.info);
1534 CSignedMsgData_CloseHandles(&msg->u.signed_data);
1536 break;
1538 CryptMemFree(msg->msg_data.pbData);
1539 CryptMemFree(msg->detached_data.pbData);
1540 ContextPropertyList_Free(msg->properties);
1543 static BOOL CDecodeMsg_CopyData(CRYPT_DATA_BLOB *blob, const BYTE *pbData,
1544 DWORD cbData)
1546 BOOL ret = TRUE;
1548 if (cbData)
1550 if (blob->cbData)
1551 blob->pbData = CryptMemRealloc(blob->pbData,
1552 blob->cbData + cbData);
1553 else
1554 blob->pbData = CryptMemAlloc(cbData);
1555 if (blob->pbData)
1557 memcpy(blob->pbData + blob->cbData, pbData, cbData);
1558 blob->cbData += cbData;
1560 else
1561 ret = FALSE;
1563 return ret;
1566 static BOOL CDecodeMsg_DecodeDataContent(CDecodeMsg *msg, const CRYPT_DER_BLOB *blob)
1568 BOOL ret;
1569 CRYPT_DATA_BLOB *data;
1570 DWORD size;
1572 ret = CryptDecodeObjectEx(X509_ASN_ENCODING, X509_OCTET_STRING,
1573 blob->pbData, blob->cbData, CRYPT_DECODE_ALLOC_FLAG, NULL, &data, &size);
1574 if (ret)
1576 ret = ContextPropertyList_SetProperty(msg->properties,
1577 CMSG_CONTENT_PARAM, data->pbData, data->cbData);
1578 LocalFree(data);
1580 return ret;
1583 static void CDecodeMsg_SaveAlgorithmID(CDecodeMsg *msg, DWORD param,
1584 const CRYPT_ALGORITHM_IDENTIFIER *id)
1586 static const BYTE nullParams[] = { ASN_NULL, 0 };
1587 CRYPT_ALGORITHM_IDENTIFIER *copy;
1588 DWORD len = sizeof(CRYPT_ALGORITHM_IDENTIFIER);
1590 /* Linearize algorithm id */
1591 len += strlen(id->pszObjId) + 1;
1592 len += id->Parameters.cbData;
1593 copy = CryptMemAlloc(len);
1594 if (copy)
1596 copy->pszObjId =
1597 (LPSTR)((BYTE *)copy + sizeof(CRYPT_ALGORITHM_IDENTIFIER));
1598 strcpy(copy->pszObjId, id->pszObjId);
1599 copy->Parameters.pbData = (BYTE *)copy->pszObjId + strlen(id->pszObjId)
1600 + 1;
1601 /* Trick: omit NULL parameters */
1602 if (id->Parameters.cbData == sizeof(nullParams) &&
1603 !memcmp(id->Parameters.pbData, nullParams, sizeof(nullParams)))
1605 copy->Parameters.cbData = 0;
1606 len -= sizeof(nullParams);
1608 else
1609 copy->Parameters.cbData = id->Parameters.cbData;
1610 if (copy->Parameters.cbData)
1611 memcpy(copy->Parameters.pbData, id->Parameters.pbData,
1612 id->Parameters.cbData);
1613 ContextPropertyList_SetProperty(msg->properties, param, (BYTE *)copy,
1614 len);
1615 CryptMemFree(copy);
1619 static inline void CRYPT_FixUpAlgorithmID(CRYPT_ALGORITHM_IDENTIFIER *id)
1621 id->pszObjId = (LPSTR)((BYTE *)id + sizeof(CRYPT_ALGORITHM_IDENTIFIER));
1622 id->Parameters.pbData = (BYTE *)id->pszObjId + strlen(id->pszObjId) + 1;
1625 static BOOL CDecodeMsg_DecodeHashedContent(CDecodeMsg *msg,
1626 const CRYPT_DER_BLOB *blob)
1628 BOOL ret;
1629 CRYPT_DIGESTED_DATA *digestedData;
1630 DWORD size;
1632 ret = CRYPT_AsnDecodePKCSDigestedData(blob->pbData, blob->cbData,
1633 CRYPT_DECODE_ALLOC_FLAG, NULL, (CRYPT_DIGESTED_DATA *)&digestedData,
1634 &size);
1635 if (ret)
1637 ContextPropertyList_SetProperty(msg->properties, CMSG_VERSION_PARAM,
1638 (const BYTE *)&digestedData->version, sizeof(digestedData->version));
1639 CDecodeMsg_SaveAlgorithmID(msg, CMSG_HASH_ALGORITHM_PARAM,
1640 &digestedData->DigestAlgorithm);
1641 ContextPropertyList_SetProperty(msg->properties,
1642 CMSG_INNER_CONTENT_TYPE_PARAM,
1643 (const BYTE *)digestedData->ContentInfo.pszObjId,
1644 digestedData->ContentInfo.pszObjId ?
1645 strlen(digestedData->ContentInfo.pszObjId) + 1 : 0);
1646 if (!(msg->base.open_flags & CMSG_DETACHED_FLAG))
1648 if (digestedData->ContentInfo.Content.cbData)
1649 CDecodeMsg_DecodeDataContent(msg,
1650 &digestedData->ContentInfo.Content);
1651 else
1652 ContextPropertyList_SetProperty(msg->properties,
1653 CMSG_CONTENT_PARAM, NULL, 0);
1655 ContextPropertyList_SetProperty(msg->properties, CMSG_HASH_DATA_PARAM,
1656 digestedData->hash.pbData, digestedData->hash.cbData);
1657 LocalFree(digestedData);
1659 return ret;
1662 static BOOL CDecodeMsg_DecodeSignedContent(CDecodeMsg *msg,
1663 const CRYPT_DER_BLOB *blob)
1665 BOOL ret;
1666 CRYPT_SIGNED_INFO *signedInfo;
1667 DWORD size;
1669 ret = CRYPT_AsnDecodeCMSSignedInfo(blob->pbData, blob->cbData,
1670 CRYPT_DECODE_ALLOC_FLAG, NULL, (CRYPT_SIGNED_INFO *)&signedInfo,
1671 &size);
1672 if (ret)
1673 msg->u.signed_data.info = signedInfo;
1674 return ret;
1677 /* Decodes the content in blob as the type given, and updates the value
1678 * (type, parameters, etc.) of msg based on what blob contains.
1679 * It doesn't just use msg's type, to allow a recursive call from an implicitly
1680 * typed message once the outer content info has been decoded.
1682 static BOOL CDecodeMsg_DecodeContent(CDecodeMsg *msg, const CRYPT_DER_BLOB *blob,
1683 DWORD type)
1685 BOOL ret;
1687 switch (type)
1689 case CMSG_DATA:
1690 if ((ret = CDecodeMsg_DecodeDataContent(msg, blob)))
1691 msg->type = CMSG_DATA;
1692 break;
1693 case CMSG_HASHED:
1694 if ((ret = CDecodeMsg_DecodeHashedContent(msg, blob)))
1695 msg->type = CMSG_HASHED;
1696 break;
1697 case CMSG_ENVELOPED:
1698 FIXME("unimplemented for type CMSG_ENVELOPED\n");
1699 ret = TRUE;
1700 break;
1701 case CMSG_SIGNED:
1702 if ((ret = CDecodeMsg_DecodeSignedContent(msg, blob)))
1703 msg->type = CMSG_SIGNED;
1704 break;
1705 default:
1707 CRYPT_CONTENT_INFO *info;
1708 DWORD size;
1710 ret = CryptDecodeObjectEx(X509_ASN_ENCODING, PKCS_CONTENT_INFO,
1711 msg->msg_data.pbData, msg->msg_data.cbData, CRYPT_DECODE_ALLOC_FLAG,
1712 NULL, &info, &size);
1713 if (ret)
1715 if (!strcmp(info->pszObjId, szOID_RSA_data))
1716 ret = CDecodeMsg_DecodeContent(msg, &info->Content, CMSG_DATA);
1717 else if (!strcmp(info->pszObjId, szOID_RSA_digestedData))
1718 ret = CDecodeMsg_DecodeContent(msg, &info->Content,
1719 CMSG_HASHED);
1720 else if (!strcmp(info->pszObjId, szOID_RSA_envelopedData))
1721 ret = CDecodeMsg_DecodeContent(msg, &info->Content,
1722 CMSG_ENVELOPED);
1723 else if (!strcmp(info->pszObjId, szOID_RSA_signedData))
1724 ret = CDecodeMsg_DecodeContent(msg, &info->Content,
1725 CMSG_SIGNED);
1726 else
1728 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
1729 ret = FALSE;
1731 LocalFree(info);
1735 return ret;
1738 static BOOL CDecodeMsg_FinalizeHashedContent(CDecodeMsg *msg,
1739 CRYPT_DER_BLOB *blob)
1741 CRYPT_ALGORITHM_IDENTIFIER *hashAlgoID = NULL;
1742 DWORD size = 0;
1743 ALG_ID algID = 0;
1744 BOOL ret;
1746 CryptMsgGetParam(msg, CMSG_HASH_ALGORITHM_PARAM, 0, NULL, &size);
1747 hashAlgoID = CryptMemAlloc(size);
1748 ret = CryptMsgGetParam(msg, CMSG_HASH_ALGORITHM_PARAM, 0, hashAlgoID,
1749 &size);
1750 if (ret)
1751 algID = CertOIDToAlgId(hashAlgoID->pszObjId);
1752 ret = CryptCreateHash(msg->crypt_prov, algID, 0, 0, &msg->u.hash);
1753 if (ret)
1755 CRYPT_DATA_BLOB content;
1757 if (msg->base.open_flags & CMSG_DETACHED_FLAG)
1759 /* Unlike for non-detached messages, the data were never stored as
1760 * the content param, but were saved in msg->detached_data instead.
1762 content.pbData = msg->detached_data.pbData;
1763 content.cbData = msg->detached_data.cbData;
1765 else
1766 ret = ContextPropertyList_FindProperty(msg->properties,
1767 CMSG_CONTENT_PARAM, &content);
1768 if (ret)
1769 ret = CryptHashData(msg->u.hash, content.pbData, content.cbData, 0);
1771 CryptMemFree(hashAlgoID);
1772 return ret;
1775 static BOOL CDecodeMsg_FinalizeSignedContent(CDecodeMsg *msg,
1776 CRYPT_DER_BLOB *blob)
1778 BOOL ret;
1779 DWORD i, size;
1781 ret = CSignedMsgData_AllocateHandles(&msg->u.signed_data);
1782 for (i = 0; ret && i < msg->u.signed_data.info->cSignerInfo; i++)
1783 ret = CSignedMsgData_ConstructSignerHandles(&msg->u.signed_data, i,
1784 msg->crypt_prov);
1785 if (ret)
1787 CRYPT_DATA_BLOB *content;
1789 /* Now that we have all the content, update the hash handles with
1790 * it. If the message is a detached message, the content is stored
1791 * in msg->detached_data rather than in the signed message's
1792 * content.
1794 if (msg->base.open_flags & CMSG_DETACHED_FLAG)
1795 content = &msg->detached_data;
1796 else
1797 content = &msg->u.signed_data.info->content.Content;
1798 if (content->cbData)
1800 /* If the message is not detached, have to decode the message's
1801 * content if the type is szOID_RSA_data.
1803 if (!(msg->base.open_flags & CMSG_DETACHED_FLAG) &&
1804 !strcmp(msg->u.signed_data.info->content.pszObjId,
1805 szOID_RSA_data))
1807 CRYPT_DATA_BLOB *blob;
1809 ret = CryptDecodeObjectEx(X509_ASN_ENCODING,
1810 X509_OCTET_STRING, content->pbData, content->cbData,
1811 CRYPT_DECODE_ALLOC_FLAG, NULL, &blob, &size);
1812 if (ret)
1814 ret = CSignedMsgData_Update(&msg->u.signed_data,
1815 blob->pbData, blob->cbData, TRUE, Verify);
1816 LocalFree(blob);
1819 else
1820 ret = CSignedMsgData_Update(&msg->u.signed_data,
1821 content->pbData, content->cbData, TRUE, Verify);
1824 return ret;
1827 static BOOL CDecodeMsg_FinalizeContent(CDecodeMsg *msg, CRYPT_DER_BLOB *blob)
1829 BOOL ret = FALSE;
1831 switch (msg->type)
1833 case CMSG_HASHED:
1834 ret = CDecodeMsg_FinalizeHashedContent(msg, blob);
1835 break;
1836 case CMSG_SIGNED:
1837 ret = CDecodeMsg_FinalizeSignedContent(msg, blob);
1838 break;
1839 default:
1840 ret = TRUE;
1842 return ret;
1845 static BOOL CDecodeMsg_Update(HCRYPTMSG hCryptMsg, const BYTE *pbData,
1846 DWORD cbData, BOOL fFinal)
1848 CDecodeMsg *msg = hCryptMsg;
1849 BOOL ret = FALSE;
1851 TRACE("(%p, %p, %d, %d)\n", hCryptMsg, pbData, cbData, fFinal);
1853 if (msg->base.state == MsgStateFinalized)
1854 SetLastError(CRYPT_E_MSG_ERROR);
1855 else if (msg->base.streamed)
1857 FIXME("(%p, %p, %d, %d): streamed update stub\n", hCryptMsg, pbData,
1858 cbData, fFinal);
1859 switch (msg->base.state)
1861 case MsgStateInit:
1862 ret = CDecodeMsg_CopyData(&msg->msg_data, pbData, cbData);
1863 if (fFinal)
1865 if (msg->base.open_flags & CMSG_DETACHED_FLAG)
1866 msg->base.state = MsgStateDataFinalized;
1867 else
1868 msg->base.state = MsgStateFinalized;
1870 else
1871 msg->base.state = MsgStateUpdated;
1872 break;
1873 case MsgStateUpdated:
1874 ret = CDecodeMsg_CopyData(&msg->msg_data, pbData, cbData);
1875 if (fFinal)
1877 if (msg->base.open_flags & CMSG_DETACHED_FLAG)
1878 msg->base.state = MsgStateDataFinalized;
1879 else
1880 msg->base.state = MsgStateFinalized;
1882 break;
1883 case MsgStateDataFinalized:
1884 ret = CDecodeMsg_CopyData(&msg->detached_data, pbData, cbData);
1885 if (fFinal)
1886 msg->base.state = MsgStateFinalized;
1887 break;
1888 default:
1889 SetLastError(CRYPT_E_MSG_ERROR);
1890 break;
1893 else
1895 if (!fFinal)
1896 SetLastError(CRYPT_E_MSG_ERROR);
1897 else
1899 switch (msg->base.state)
1901 case MsgStateInit:
1902 ret = CDecodeMsg_CopyData(&msg->msg_data, pbData, cbData);
1903 if (msg->base.open_flags & CMSG_DETACHED_FLAG)
1904 msg->base.state = MsgStateDataFinalized;
1905 else
1906 msg->base.state = MsgStateFinalized;
1907 break;
1908 case MsgStateDataFinalized:
1909 ret = CDecodeMsg_CopyData(&msg->detached_data, pbData, cbData);
1910 msg->base.state = MsgStateFinalized;
1911 break;
1912 default:
1913 SetLastError(CRYPT_E_MSG_ERROR);
1917 if (ret && fFinal &&
1918 ((msg->base.open_flags & CMSG_DETACHED_FLAG && msg->base.state ==
1919 MsgStateDataFinalized) ||
1920 (!(msg->base.open_flags & CMSG_DETACHED_FLAG) && msg->base.state ==
1921 MsgStateFinalized)))
1922 ret = CDecodeMsg_DecodeContent(msg, &msg->msg_data, msg->type);
1923 if (ret && msg->base.state == MsgStateFinalized)
1924 ret = CDecodeMsg_FinalizeContent(msg, &msg->msg_data);
1925 return ret;
1928 static BOOL CDecodeHashMsg_GetParam(CDecodeMsg *msg, DWORD dwParamType,
1929 DWORD dwIndex, void *pvData, DWORD *pcbData)
1931 BOOL ret = FALSE;
1933 switch (dwParamType)
1935 case CMSG_TYPE_PARAM:
1936 ret = CRYPT_CopyParam(pvData, pcbData, &msg->type, sizeof(msg->type));
1937 break;
1938 case CMSG_HASH_ALGORITHM_PARAM:
1940 CRYPT_DATA_BLOB blob;
1942 ret = ContextPropertyList_FindProperty(msg->properties, dwParamType,
1943 &blob);
1944 if (ret)
1946 ret = CRYPT_CopyParam(pvData, pcbData, blob.pbData, blob.cbData);
1947 if (ret && pvData)
1948 CRYPT_FixUpAlgorithmID(pvData);
1950 else
1951 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
1952 break;
1954 case CMSG_COMPUTED_HASH_PARAM:
1955 ret = CryptGetHashParam(msg->u.hash, HP_HASHVAL, pvData, pcbData, 0);
1956 break;
1957 default:
1959 CRYPT_DATA_BLOB blob;
1961 ret = ContextPropertyList_FindProperty(msg->properties, dwParamType,
1962 &blob);
1963 if (ret)
1964 ret = CRYPT_CopyParam(pvData, pcbData, blob.pbData, blob.cbData);
1965 else
1966 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
1969 return ret;
1972 /* nextData is an in/out parameter - on input it's the memory location in
1973 * which a copy of in's data should be made, and on output it's the memory
1974 * location immediately after out's copy of in's data.
1976 static inline void CRYPT_CopyBlob(CRYPT_DATA_BLOB *out,
1977 const CRYPT_DATA_BLOB *in, LPBYTE *nextData)
1979 out->cbData = in->cbData;
1980 if (in->cbData)
1982 out->pbData = *nextData;
1983 memcpy(out->pbData, in->pbData, in->cbData);
1984 *nextData += in->cbData;
1988 static inline void CRYPT_CopyAlgorithmId(CRYPT_ALGORITHM_IDENTIFIER *out,
1989 const CRYPT_ALGORITHM_IDENTIFIER *in, LPBYTE *nextData)
1991 if (in->pszObjId)
1993 out->pszObjId = (LPSTR)*nextData;
1994 strcpy(out->pszObjId, in->pszObjId);
1995 *nextData += strlen(out->pszObjId) + 1;
1997 CRYPT_CopyBlob(&out->Parameters, &in->Parameters, nextData);
2000 static inline void CRYPT_CopyAttributes(CRYPT_ATTRIBUTES *out,
2001 const CRYPT_ATTRIBUTES *in, LPBYTE *nextData)
2003 out->cAttr = in->cAttr;
2004 if (in->cAttr)
2006 DWORD i;
2008 *nextData = POINTER_ALIGN_DWORD_PTR(*nextData);
2009 out->rgAttr = (CRYPT_ATTRIBUTE *)*nextData;
2010 *nextData += in->cAttr * sizeof(CRYPT_ATTRIBUTE);
2011 for (i = 0; i < in->cAttr; i++)
2013 if (in->rgAttr[i].pszObjId)
2015 out->rgAttr[i].pszObjId = (LPSTR)*nextData;
2016 strcpy(out->rgAttr[i].pszObjId, in->rgAttr[i].pszObjId);
2017 *nextData += strlen(in->rgAttr[i].pszObjId) + 1;
2019 if (in->rgAttr[i].cValue)
2021 DWORD j;
2023 out->rgAttr[i].cValue = in->rgAttr[i].cValue;
2024 *nextData = POINTER_ALIGN_DWORD_PTR(*nextData);
2025 out->rgAttr[i].rgValue = (PCRYPT_DATA_BLOB)*nextData;
2026 *nextData += in->rgAttr[i].cValue * sizeof(CRYPT_DATA_BLOB);
2027 for (j = 0; j < in->rgAttr[i].cValue; j++)
2028 CRYPT_CopyBlob(&out->rgAttr[i].rgValue[j],
2029 &in->rgAttr[i].rgValue[j], nextData);
2035 static DWORD CRYPT_SizeOfAttributes(const CRYPT_ATTRIBUTES *attr)
2037 DWORD size = attr->cAttr * sizeof(CRYPT_ATTRIBUTE), i, j;
2039 for (i = 0; i < attr->cAttr; i++)
2041 if (attr->rgAttr[i].pszObjId)
2042 size += strlen(attr->rgAttr[i].pszObjId) + 1;
2043 /* align pointer */
2044 size = ALIGN_DWORD_PTR(size);
2045 size += attr->rgAttr[i].cValue * sizeof(CRYPT_DATA_BLOB);
2046 for (j = 0; j < attr->rgAttr[i].cValue; j++)
2047 size += attr->rgAttr[i].rgValue[j].cbData;
2049 /* align pointer again to be conservative */
2050 size = ALIGN_DWORD_PTR(size);
2051 return size;
2054 static DWORD CRYPT_SizeOfKeyIdAsIssuerAndSerial(const CRYPT_DATA_BLOB *keyId)
2056 static char oid_key_rdn[] = szOID_KEYID_RDN;
2057 DWORD size = 0;
2058 CERT_RDN_ATTR attr;
2059 CERT_RDN rdn = { 1, &attr };
2060 CERT_NAME_INFO name = { 1, &rdn };
2062 attr.pszObjId = oid_key_rdn;
2063 attr.dwValueType = CERT_RDN_OCTET_STRING;
2064 attr.Value.cbData = keyId->cbData;
2065 attr.Value.pbData = keyId->pbData;
2066 if (CryptEncodeObject(X509_ASN_ENCODING, X509_NAME, &name, NULL, &size))
2067 size++; /* Only include size of special zero serial number on success */
2068 return size;
2071 static BOOL CRYPT_CopyKeyIdAsIssuerAndSerial(CERT_NAME_BLOB *issuer,
2072 CRYPT_INTEGER_BLOB *serialNumber, const CRYPT_DATA_BLOB *keyId, DWORD encodedLen,
2073 LPBYTE *nextData)
2075 static char oid_key_rdn[] = szOID_KEYID_RDN;
2076 CERT_RDN_ATTR attr;
2077 CERT_RDN rdn = { 1, &attr };
2078 CERT_NAME_INFO name = { 1, &rdn };
2079 BOOL ret;
2081 /* Encode special zero serial number */
2082 serialNumber->cbData = 1;
2083 serialNumber->pbData = *nextData;
2084 **nextData = 0;
2085 (*nextData)++;
2086 /* Encode issuer */
2087 issuer->pbData = *nextData;
2088 attr.pszObjId = oid_key_rdn;
2089 attr.dwValueType = CERT_RDN_OCTET_STRING;
2090 attr.Value.cbData = keyId->cbData;
2091 attr.Value.pbData = keyId->pbData;
2092 ret = CryptEncodeObject(X509_ASN_ENCODING, X509_NAME, &name, *nextData,
2093 &encodedLen);
2094 if (ret)
2096 *nextData += encodedLen;
2097 issuer->cbData = encodedLen;
2099 return ret;
2102 static BOOL CRYPT_CopySignerInfo(void *pvData, DWORD *pcbData,
2103 const CMSG_CMS_SIGNER_INFO *in)
2105 DWORD size = sizeof(CMSG_SIGNER_INFO), rdnSize = 0;
2106 BOOL ret;
2108 TRACE("(%p, %d, %p)\n", pvData, pvData ? *pcbData : 0, in);
2110 if (in->SignerId.dwIdChoice == CERT_ID_ISSUER_SERIAL_NUMBER)
2112 size += in->SignerId.u.IssuerSerialNumber.Issuer.cbData;
2113 size += in->SignerId.u.IssuerSerialNumber.SerialNumber.cbData;
2115 else
2117 rdnSize = CRYPT_SizeOfKeyIdAsIssuerAndSerial(&in->SignerId.u.KeyId);
2118 size += rdnSize;
2120 if (in->HashAlgorithm.pszObjId)
2121 size += strlen(in->HashAlgorithm.pszObjId) + 1;
2122 size += in->HashAlgorithm.Parameters.cbData;
2123 if (in->HashEncryptionAlgorithm.pszObjId)
2124 size += strlen(in->HashEncryptionAlgorithm.pszObjId) + 1;
2125 size += in->HashEncryptionAlgorithm.Parameters.cbData;
2126 size += in->EncryptedHash.cbData;
2127 /* align pointer */
2128 size = ALIGN_DWORD_PTR(size);
2129 size += CRYPT_SizeOfAttributes(&in->AuthAttrs);
2130 size += CRYPT_SizeOfAttributes(&in->UnauthAttrs);
2131 if (!pvData)
2133 *pcbData = size;
2134 ret = TRUE;
2136 else if (*pcbData < size)
2138 *pcbData = size;
2139 SetLastError(ERROR_MORE_DATA);
2140 ret = FALSE;
2142 else
2144 LPBYTE nextData = (BYTE *)pvData + sizeof(CMSG_SIGNER_INFO);
2145 CMSG_SIGNER_INFO *out = pvData;
2147 ret = TRUE;
2148 out->dwVersion = in->dwVersion;
2149 if (in->SignerId.dwIdChoice == CERT_ID_ISSUER_SERIAL_NUMBER)
2151 CRYPT_CopyBlob(&out->Issuer,
2152 &in->SignerId.u.IssuerSerialNumber.Issuer, &nextData);
2153 CRYPT_CopyBlob(&out->SerialNumber,
2154 &in->SignerId.u.IssuerSerialNumber.SerialNumber, &nextData);
2156 else
2157 ret = CRYPT_CopyKeyIdAsIssuerAndSerial(&out->Issuer, &out->SerialNumber,
2158 &in->SignerId.u.KeyId, rdnSize, &nextData);
2159 if (ret)
2161 CRYPT_CopyAlgorithmId(&out->HashAlgorithm, &in->HashAlgorithm,
2162 &nextData);
2163 CRYPT_CopyAlgorithmId(&out->HashEncryptionAlgorithm,
2164 &in->HashEncryptionAlgorithm, &nextData);
2165 CRYPT_CopyBlob(&out->EncryptedHash, &in->EncryptedHash, &nextData);
2166 nextData = POINTER_ALIGN_DWORD_PTR(nextData);
2167 CRYPT_CopyAttributes(&out->AuthAttrs, &in->AuthAttrs, &nextData);
2168 CRYPT_CopyAttributes(&out->UnauthAttrs, &in->UnauthAttrs, &nextData);
2171 TRACE("returning %d\n", ret);
2172 return ret;
2175 static BOOL CRYPT_CopyCMSSignerInfo(void *pvData, DWORD *pcbData,
2176 const CMSG_CMS_SIGNER_INFO *in)
2178 DWORD size = sizeof(CMSG_CMS_SIGNER_INFO);
2179 BOOL ret;
2181 TRACE("(%p, %d, %p)\n", pvData, pvData ? *pcbData : 0, in);
2183 if (in->SignerId.dwIdChoice == CERT_ID_ISSUER_SERIAL_NUMBER)
2185 size += in->SignerId.u.IssuerSerialNumber.Issuer.cbData;
2186 size += in->SignerId.u.IssuerSerialNumber.SerialNumber.cbData;
2188 else
2189 size += in->SignerId.u.KeyId.cbData;
2190 if (in->HashAlgorithm.pszObjId)
2191 size += strlen(in->HashAlgorithm.pszObjId) + 1;
2192 size += in->HashAlgorithm.Parameters.cbData;
2193 if (in->HashEncryptionAlgorithm.pszObjId)
2194 size += strlen(in->HashEncryptionAlgorithm.pszObjId) + 1;
2195 size += in->HashEncryptionAlgorithm.Parameters.cbData;
2196 size += in->EncryptedHash.cbData;
2197 /* align pointer */
2198 size = ALIGN_DWORD_PTR(size);
2199 size += CRYPT_SizeOfAttributes(&in->AuthAttrs);
2200 size += CRYPT_SizeOfAttributes(&in->UnauthAttrs);
2201 if (!pvData)
2203 *pcbData = size;
2204 ret = TRUE;
2206 else if (*pcbData < size)
2208 *pcbData = size;
2209 SetLastError(ERROR_MORE_DATA);
2210 ret = FALSE;
2212 else
2214 LPBYTE nextData = (BYTE *)pvData + sizeof(CMSG_CMS_SIGNER_INFO);
2215 CMSG_CMS_SIGNER_INFO *out = pvData;
2217 out->dwVersion = in->dwVersion;
2218 out->SignerId.dwIdChoice = in->SignerId.dwIdChoice;
2219 if (in->SignerId.dwIdChoice == CERT_ID_ISSUER_SERIAL_NUMBER)
2221 CRYPT_CopyBlob(&out->SignerId.u.IssuerSerialNumber.Issuer,
2222 &in->SignerId.u.IssuerSerialNumber.Issuer, &nextData);
2223 CRYPT_CopyBlob(&out->SignerId.u.IssuerSerialNumber.SerialNumber,
2224 &in->SignerId.u.IssuerSerialNumber.SerialNumber, &nextData);
2226 else
2227 CRYPT_CopyBlob(&out->SignerId.u.KeyId, &in->SignerId.u.KeyId, &nextData);
2228 CRYPT_CopyAlgorithmId(&out->HashAlgorithm, &in->HashAlgorithm,
2229 &nextData);
2230 CRYPT_CopyAlgorithmId(&out->HashEncryptionAlgorithm,
2231 &in->HashEncryptionAlgorithm, &nextData);
2232 CRYPT_CopyBlob(&out->EncryptedHash, &in->EncryptedHash, &nextData);
2233 nextData = POINTER_ALIGN_DWORD_PTR(nextData);
2234 CRYPT_CopyAttributes(&out->AuthAttrs, &in->AuthAttrs, &nextData);
2235 CRYPT_CopyAttributes(&out->UnauthAttrs, &in->UnauthAttrs, &nextData);
2236 ret = TRUE;
2238 TRACE("returning %d\n", ret);
2239 return ret;
2242 static BOOL CRYPT_CopySignerCertInfo(void *pvData, DWORD *pcbData,
2243 const CMSG_CMS_SIGNER_INFO *in)
2245 DWORD size = sizeof(CERT_INFO), rdnSize = 0;
2246 BOOL ret;
2248 TRACE("(%p, %d, %p)\n", pvData, pvData ? *pcbData : 0, in);
2250 if (in->SignerId.dwIdChoice == CERT_ID_ISSUER_SERIAL_NUMBER)
2252 size += in->SignerId.u.IssuerSerialNumber.Issuer.cbData;
2253 size += in->SignerId.u.IssuerSerialNumber.SerialNumber.cbData;
2255 else
2257 rdnSize = CRYPT_SizeOfKeyIdAsIssuerAndSerial(&in->SignerId.u.KeyId);
2258 size += rdnSize;
2260 if (!pvData)
2262 *pcbData = size;
2263 ret = TRUE;
2265 else if (*pcbData < size)
2267 *pcbData = size;
2268 SetLastError(ERROR_MORE_DATA);
2269 ret = FALSE;
2271 else
2273 LPBYTE nextData = (BYTE *)pvData + sizeof(CERT_INFO);
2274 CERT_INFO *out = pvData;
2276 memset(out, 0, sizeof(CERT_INFO));
2277 if (in->SignerId.dwIdChoice == CERT_ID_ISSUER_SERIAL_NUMBER)
2279 CRYPT_CopyBlob(&out->Issuer,
2280 &in->SignerId.u.IssuerSerialNumber.Issuer, &nextData);
2281 CRYPT_CopyBlob(&out->SerialNumber,
2282 &in->SignerId.u.IssuerSerialNumber.SerialNumber, &nextData);
2283 ret = TRUE;
2285 else
2286 ret = CRYPT_CopyKeyIdAsIssuerAndSerial(&out->Issuer, &out->SerialNumber,
2287 &in->SignerId.u.KeyId, rdnSize, &nextData);
2289 TRACE("returning %d\n", ret);
2290 return ret;
2293 static BOOL CDecodeSignedMsg_GetParam(CDecodeMsg *msg, DWORD dwParamType,
2294 DWORD dwIndex, void *pvData, DWORD *pcbData)
2296 BOOL ret = FALSE;
2298 switch (dwParamType)
2300 case CMSG_TYPE_PARAM:
2301 ret = CRYPT_CopyParam(pvData, pcbData, &msg->type, sizeof(msg->type));
2302 break;
2303 case CMSG_CONTENT_PARAM:
2304 if (msg->u.signed_data.info)
2306 if (!strcmp(msg->u.signed_data.info->content.pszObjId,
2307 szOID_RSA_data))
2309 CRYPT_DATA_BLOB *blob;
2310 DWORD size;
2312 ret = CryptDecodeObjectEx(X509_ASN_ENCODING, X509_OCTET_STRING,
2313 msg->u.signed_data.info->content.Content.pbData,
2314 msg->u.signed_data.info->content.Content.cbData,
2315 CRYPT_DECODE_ALLOC_FLAG, NULL, &blob, &size);
2316 if (ret)
2318 ret = CRYPT_CopyParam(pvData, pcbData, blob->pbData,
2319 blob->cbData);
2320 LocalFree(blob);
2323 else
2324 ret = CRYPT_CopyParam(pvData, pcbData,
2325 msg->u.signed_data.info->content.Content.pbData,
2326 msg->u.signed_data.info->content.Content.cbData);
2328 else
2329 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2330 break;
2331 case CMSG_INNER_CONTENT_TYPE_PARAM:
2332 if (msg->u.signed_data.info)
2333 ret = CRYPT_CopyParam(pvData, pcbData,
2334 msg->u.signed_data.info->content.pszObjId,
2335 strlen(msg->u.signed_data.info->content.pszObjId) + 1);
2336 else
2337 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2338 break;
2339 case CMSG_SIGNER_COUNT_PARAM:
2340 if (msg->u.signed_data.info)
2341 ret = CRYPT_CopyParam(pvData, pcbData,
2342 &msg->u.signed_data.info->cSignerInfo, sizeof(DWORD));
2343 else
2344 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2345 break;
2346 case CMSG_SIGNER_INFO_PARAM:
2347 if (msg->u.signed_data.info)
2349 if (dwIndex >= msg->u.signed_data.info->cSignerInfo)
2350 SetLastError(CRYPT_E_INVALID_INDEX);
2351 else
2352 ret = CRYPT_CopySignerInfo(pvData, pcbData,
2353 &msg->u.signed_data.info->rgSignerInfo[dwIndex]);
2355 else
2356 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2357 break;
2358 case CMSG_SIGNER_CERT_INFO_PARAM:
2359 if (msg->u.signed_data.info)
2361 if (dwIndex >= msg->u.signed_data.info->cSignerInfo)
2362 SetLastError(CRYPT_E_INVALID_INDEX);
2363 else
2364 ret = CRYPT_CopySignerCertInfo(pvData, pcbData,
2365 &msg->u.signed_data.info->rgSignerInfo[dwIndex]);
2367 else
2368 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2369 break;
2370 case CMSG_CERT_COUNT_PARAM:
2371 if (msg->u.signed_data.info)
2372 ret = CRYPT_CopyParam(pvData, pcbData,
2373 &msg->u.signed_data.info->cCertEncoded, sizeof(DWORD));
2374 else
2375 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2376 break;
2377 case CMSG_CERT_PARAM:
2378 if (msg->u.signed_data.info)
2380 if (dwIndex >= msg->u.signed_data.info->cCertEncoded)
2381 SetLastError(CRYPT_E_INVALID_INDEX);
2382 else
2383 ret = CRYPT_CopyParam(pvData, pcbData,
2384 msg->u.signed_data.info->rgCertEncoded[dwIndex].pbData,
2385 msg->u.signed_data.info->rgCertEncoded[dwIndex].cbData);
2387 else
2388 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2389 break;
2390 case CMSG_CRL_COUNT_PARAM:
2391 if (msg->u.signed_data.info)
2392 ret = CRYPT_CopyParam(pvData, pcbData,
2393 &msg->u.signed_data.info->cCrlEncoded, sizeof(DWORD));
2394 else
2395 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2396 break;
2397 case CMSG_CRL_PARAM:
2398 if (msg->u.signed_data.info)
2400 if (dwIndex >= msg->u.signed_data.info->cCrlEncoded)
2401 SetLastError(CRYPT_E_INVALID_INDEX);
2402 else
2403 ret = CRYPT_CopyParam(pvData, pcbData,
2404 msg->u.signed_data.info->rgCrlEncoded[dwIndex].pbData,
2405 msg->u.signed_data.info->rgCrlEncoded[dwIndex].cbData);
2407 else
2408 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2409 break;
2410 case CMSG_COMPUTED_HASH_PARAM:
2411 if (msg->u.signed_data.info)
2413 if (dwIndex >= msg->u.signed_data.cSignerHandle)
2414 SetLastError(CRYPT_E_INVALID_INDEX);
2415 else
2416 ret = CryptGetHashParam(
2417 msg->u.signed_data.signerHandles[dwIndex].contentHash,
2418 HP_HASHVAL, pvData, pcbData, 0);
2420 else
2421 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2422 break;
2423 case CMSG_ENCODED_SIGNER:
2424 if (msg->u.signed_data.info)
2426 if (dwIndex >= msg->u.signed_data.info->cSignerInfo)
2427 SetLastError(CRYPT_E_INVALID_INDEX);
2428 else
2429 ret = CryptEncodeObjectEx(
2430 X509_ASN_ENCODING | PKCS_7_ASN_ENCODING, CMS_SIGNER_INFO,
2431 &msg->u.signed_data.info->rgSignerInfo[dwIndex], 0, NULL,
2432 pvData, pcbData);
2434 else
2435 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2436 break;
2437 case CMSG_ATTR_CERT_COUNT_PARAM:
2438 if (msg->u.signed_data.info)
2440 DWORD attrCertCount = 0;
2442 ret = CRYPT_CopyParam(pvData, pcbData,
2443 &attrCertCount, sizeof(DWORD));
2445 else
2446 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2447 break;
2448 case CMSG_ATTR_CERT_PARAM:
2449 if (msg->u.signed_data.info)
2450 SetLastError(CRYPT_E_INVALID_INDEX);
2451 else
2452 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2453 break;
2454 case CMSG_CMS_SIGNER_INFO_PARAM:
2455 if (msg->u.signed_data.info)
2457 if (dwIndex >= msg->u.signed_data.info->cSignerInfo)
2458 SetLastError(CRYPT_E_INVALID_INDEX);
2459 else
2460 ret = CRYPT_CopyCMSSignerInfo(pvData, pcbData,
2461 &msg->u.signed_data.info->rgSignerInfo[dwIndex]);
2463 else
2464 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2465 break;
2466 default:
2467 FIXME("unimplemented for %d\n", dwParamType);
2468 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2470 return ret;
2473 static BOOL CDecodeMsg_GetParam(HCRYPTMSG hCryptMsg, DWORD dwParamType,
2474 DWORD dwIndex, void *pvData, DWORD *pcbData)
2476 CDecodeMsg *msg = hCryptMsg;
2477 BOOL ret = FALSE;
2479 switch (msg->type)
2481 case CMSG_HASHED:
2482 ret = CDecodeHashMsg_GetParam(msg, dwParamType, dwIndex, pvData,
2483 pcbData);
2484 break;
2485 case CMSG_SIGNED:
2486 ret = CDecodeSignedMsg_GetParam(msg, dwParamType, dwIndex, pvData,
2487 pcbData);
2488 break;
2489 default:
2490 switch (dwParamType)
2492 case CMSG_TYPE_PARAM:
2493 ret = CRYPT_CopyParam(pvData, pcbData, &msg->type,
2494 sizeof(msg->type));
2495 break;
2496 default:
2498 CRYPT_DATA_BLOB blob;
2500 ret = ContextPropertyList_FindProperty(msg->properties, dwParamType,
2501 &blob);
2502 if (ret)
2503 ret = CRYPT_CopyParam(pvData, pcbData, blob.pbData,
2504 blob.cbData);
2505 else
2506 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2510 return ret;
2513 static BOOL CDecodeHashMsg_VerifyHash(CDecodeMsg *msg)
2515 BOOL ret;
2516 CRYPT_DATA_BLOB hashBlob;
2518 ret = ContextPropertyList_FindProperty(msg->properties,
2519 CMSG_HASH_DATA_PARAM, &hashBlob);
2520 if (ret)
2522 DWORD computedHashSize = 0;
2524 ret = CDecodeHashMsg_GetParam(msg, CMSG_COMPUTED_HASH_PARAM, 0, NULL,
2525 &computedHashSize);
2526 if (hashBlob.cbData == computedHashSize)
2528 LPBYTE computedHash = CryptMemAlloc(computedHashSize);
2530 if (computedHash)
2532 ret = CDecodeHashMsg_GetParam(msg, CMSG_COMPUTED_HASH_PARAM, 0,
2533 computedHash, &computedHashSize);
2534 if (ret)
2536 if (memcmp(hashBlob.pbData, computedHash, hashBlob.cbData))
2538 SetLastError(CRYPT_E_HASH_VALUE);
2539 ret = FALSE;
2542 CryptMemFree(computedHash);
2544 else
2546 SetLastError(ERROR_OUTOFMEMORY);
2547 ret = FALSE;
2550 else
2552 SetLastError(CRYPT_E_HASH_VALUE);
2553 ret = FALSE;
2556 return ret;
2559 static BOOL CDecodeSignedMsg_VerifySignatureWithKey(CDecodeMsg *msg,
2560 HCRYPTPROV prov, DWORD signerIndex, PCERT_PUBLIC_KEY_INFO keyInfo)
2562 HCRYPTKEY key;
2563 BOOL ret;
2565 if (!prov)
2566 prov = msg->crypt_prov;
2567 ret = CryptImportPublicKeyInfo(prov, X509_ASN_ENCODING, keyInfo, &key);
2568 if (ret)
2570 HCRYPTHASH hash;
2571 CRYPT_HASH_BLOB reversedHash;
2573 if (msg->u.signed_data.info->rgSignerInfo[signerIndex].AuthAttrs.cAttr)
2574 hash = msg->u.signed_data.signerHandles[signerIndex].authAttrHash;
2575 else
2576 hash = msg->u.signed_data.signerHandles[signerIndex].contentHash;
2577 ret = CRYPT_ConstructBlob(&reversedHash,
2578 &msg->u.signed_data.info->rgSignerInfo[signerIndex].EncryptedHash);
2579 if (ret)
2581 CRYPT_ReverseBytes(&reversedHash);
2582 ret = CryptVerifySignatureW(hash, reversedHash.pbData,
2583 reversedHash.cbData, key, NULL, 0);
2584 CryptMemFree(reversedHash.pbData);
2586 CryptDestroyKey(key);
2588 return ret;
2591 static BOOL CDecodeSignedMsg_VerifySignature(CDecodeMsg *msg, PCERT_INFO info)
2593 BOOL ret = FALSE;
2594 DWORD i;
2596 if (!msg->u.signed_data.signerHandles)
2598 SetLastError(NTE_BAD_SIGNATURE);
2599 return FALSE;
2601 for (i = 0; !ret && i < msg->u.signed_data.info->cSignerInfo; i++)
2603 PCMSG_CMS_SIGNER_INFO signerInfo =
2604 &msg->u.signed_data.info->rgSignerInfo[i];
2606 if (signerInfo->SignerId.dwIdChoice == CERT_ID_ISSUER_SERIAL_NUMBER)
2608 ret = CertCompareCertificateName(X509_ASN_ENCODING,
2609 &signerInfo->SignerId.u.IssuerSerialNumber.Issuer,
2610 &info->Issuer);
2611 if (ret)
2613 ret = CertCompareIntegerBlob(
2614 &signerInfo->SignerId.u.IssuerSerialNumber.SerialNumber,
2615 &info->SerialNumber);
2616 if (ret)
2617 break;
2620 else
2622 FIXME("signer %d: unimplemented for key id\n", i);
2625 if (ret)
2626 ret = CDecodeSignedMsg_VerifySignatureWithKey(msg, 0, i,
2627 &info->SubjectPublicKeyInfo);
2628 else
2629 SetLastError(CRYPT_E_SIGNER_NOT_FOUND);
2631 return ret;
2634 static BOOL CDecodeSignedMsg_VerifySignatureEx(CDecodeMsg *msg,
2635 PCMSG_CTRL_VERIFY_SIGNATURE_EX_PARA para)
2637 BOOL ret = FALSE;
2639 if (para->cbSize != sizeof(CMSG_CTRL_VERIFY_SIGNATURE_EX_PARA))
2640 SetLastError(ERROR_INVALID_PARAMETER);
2641 else if (para->dwSignerIndex >= msg->u.signed_data.info->cSignerInfo)
2642 SetLastError(CRYPT_E_SIGNER_NOT_FOUND);
2643 else if (!msg->u.signed_data.signerHandles)
2644 SetLastError(NTE_BAD_SIGNATURE);
2645 else
2647 switch (para->dwSignerType)
2649 case CMSG_VERIFY_SIGNER_PUBKEY:
2650 ret = CDecodeSignedMsg_VerifySignatureWithKey(msg,
2651 para->hCryptProv, para->dwSignerIndex, para->pvSigner);
2652 break;
2653 case CMSG_VERIFY_SIGNER_CERT:
2655 PCCERT_CONTEXT cert = para->pvSigner;
2657 ret = CDecodeSignedMsg_VerifySignatureWithKey(msg, para->hCryptProv,
2658 para->dwSignerIndex, &cert->pCertInfo->SubjectPublicKeyInfo);
2659 break;
2661 default:
2662 FIXME("unimplemented for signer type %d\n", para->dwSignerType);
2663 SetLastError(CRYPT_E_SIGNER_NOT_FOUND);
2666 return ret;
2669 static BOOL CDecodeMsg_Control(HCRYPTMSG hCryptMsg, DWORD dwFlags,
2670 DWORD dwCtrlType, const void *pvCtrlPara)
2672 CDecodeMsg *msg = hCryptMsg;
2673 BOOL ret = FALSE;
2675 switch (dwCtrlType)
2677 case CMSG_CTRL_VERIFY_SIGNATURE:
2678 switch (msg->type)
2680 case CMSG_SIGNED:
2681 ret = CDecodeSignedMsg_VerifySignature(msg, (PCERT_INFO)pvCtrlPara);
2682 break;
2683 default:
2684 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2686 break;
2687 case CMSG_CTRL_DECRYPT:
2688 switch (msg->type)
2690 default:
2691 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2693 break;
2694 case CMSG_CTRL_VERIFY_HASH:
2695 switch (msg->type)
2697 case CMSG_HASHED:
2698 ret = CDecodeHashMsg_VerifyHash(msg);
2699 break;
2700 default:
2701 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2703 break;
2704 case CMSG_CTRL_VERIFY_SIGNATURE_EX:
2705 switch (msg->type)
2707 case CMSG_SIGNED:
2708 ret = CDecodeSignedMsg_VerifySignatureEx(msg,
2709 (PCMSG_CTRL_VERIFY_SIGNATURE_EX_PARA)pvCtrlPara);
2710 break;
2711 default:
2712 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2714 break;
2715 default:
2716 SetLastError(CRYPT_E_CONTROL_TYPE);
2718 return ret;
2721 HCRYPTMSG WINAPI CryptMsgOpenToDecode(DWORD dwMsgEncodingType, DWORD dwFlags,
2722 DWORD dwMsgType, HCRYPTPROV_LEGACY hCryptProv, PCERT_INFO pRecipientInfo,
2723 PCMSG_STREAM_INFO pStreamInfo)
2725 CDecodeMsg *msg;
2727 TRACE("(%08x, %08x, %08x, %08lx, %p, %p)\n", dwMsgEncodingType,
2728 dwFlags, dwMsgType, hCryptProv, pRecipientInfo, pStreamInfo);
2730 if (GET_CMSG_ENCODING_TYPE(dwMsgEncodingType) != PKCS_7_ASN_ENCODING)
2732 SetLastError(E_INVALIDARG);
2733 return NULL;
2735 msg = CryptMemAlloc(sizeof(CDecodeMsg));
2736 if (msg)
2738 CryptMsgBase_Init((CryptMsgBase *)msg, dwFlags, pStreamInfo,
2739 CDecodeMsg_Close, CDecodeMsg_GetParam, CDecodeMsg_Update,
2740 CDecodeMsg_Control);
2741 msg->type = dwMsgType;
2742 if (hCryptProv)
2743 msg->crypt_prov = hCryptProv;
2744 else
2746 msg->crypt_prov = CRYPT_GetDefaultProvider();
2747 msg->base.open_flags &= ~CMSG_CRYPT_RELEASE_CONTEXT_FLAG;
2749 memset(&msg->u, 0, sizeof(msg->u));
2750 msg->msg_data.cbData = 0;
2751 msg->msg_data.pbData = NULL;
2752 msg->detached_data.cbData = 0;
2753 msg->detached_data.pbData = NULL;
2754 msg->properties = ContextPropertyList_Create();
2756 return msg;
2759 HCRYPTMSG WINAPI CryptMsgDuplicate(HCRYPTMSG hCryptMsg)
2761 TRACE("(%p)\n", hCryptMsg);
2763 if (hCryptMsg)
2765 CryptMsgBase *msg = hCryptMsg;
2767 InterlockedIncrement(&msg->ref);
2769 return hCryptMsg;
2772 BOOL WINAPI CryptMsgClose(HCRYPTMSG hCryptMsg)
2774 TRACE("(%p)\n", hCryptMsg);
2776 if (hCryptMsg)
2778 CryptMsgBase *msg = hCryptMsg;
2780 if (InterlockedDecrement(&msg->ref) == 0)
2782 TRACE("freeing %p\n", msg);
2783 if (msg->close)
2784 msg->close(msg);
2785 CryptMemFree(msg);
2788 return TRUE;
2791 BOOL WINAPI CryptMsgUpdate(HCRYPTMSG hCryptMsg, const BYTE *pbData,
2792 DWORD cbData, BOOL fFinal)
2794 CryptMsgBase *msg = hCryptMsg;
2796 TRACE("(%p, %p, %d, %d)\n", hCryptMsg, pbData, cbData, fFinal);
2798 return msg->update(hCryptMsg, pbData, cbData, fFinal);
2801 BOOL WINAPI CryptMsgGetParam(HCRYPTMSG hCryptMsg, DWORD dwParamType,
2802 DWORD dwIndex, void *pvData, DWORD *pcbData)
2804 CryptMsgBase *msg = hCryptMsg;
2806 TRACE("(%p, %d, %d, %p, %p)\n", hCryptMsg, dwParamType, dwIndex,
2807 pvData, pcbData);
2808 return msg->get_param(hCryptMsg, dwParamType, dwIndex, pvData, pcbData);
2811 BOOL WINAPI CryptMsgControl(HCRYPTMSG hCryptMsg, DWORD dwFlags,
2812 DWORD dwCtrlType, const void *pvCtrlPara)
2814 CryptMsgBase *msg = hCryptMsg;
2816 TRACE("(%p, %08x, %d, %p)\n", hCryptMsg, dwFlags, dwCtrlType,
2817 pvCtrlPara);
2818 return msg->control(hCryptMsg, dwFlags, dwCtrlType, pvCtrlPara);
2821 static CERT_INFO *CRYPT_GetSignerCertInfoFromMsg(HCRYPTMSG msg,
2822 DWORD dwSignerIndex)
2824 CERT_INFO *certInfo = NULL;
2825 DWORD size;
2827 if (CryptMsgGetParam(msg, CMSG_SIGNER_CERT_INFO_PARAM, dwSignerIndex, NULL,
2828 &size))
2830 certInfo = CryptMemAlloc(size);
2831 if (certInfo)
2833 if (!CryptMsgGetParam(msg, CMSG_SIGNER_CERT_INFO_PARAM,
2834 dwSignerIndex, certInfo, &size))
2836 CryptMemFree(certInfo);
2837 certInfo = NULL;
2841 return certInfo;
2844 BOOL WINAPI CryptMsgGetAndVerifySigner(HCRYPTMSG hCryptMsg, DWORD cSignerStore,
2845 HCERTSTORE *rghSignerStore, DWORD dwFlags, PCCERT_CONTEXT *ppSigner,
2846 DWORD *pdwSignerIndex)
2848 HCERTSTORE store;
2849 DWORD i, signerIndex = 0;
2850 PCCERT_CONTEXT signerCert = NULL;
2851 BOOL ret = FALSE;
2853 TRACE("(%p, %d, %p, %08x, %p, %p)\n", hCryptMsg, cSignerStore,
2854 rghSignerStore, dwFlags, ppSigner, pdwSignerIndex);
2856 /* Clear output parameters */
2857 if (ppSigner)
2858 *ppSigner = NULL;
2859 if (pdwSignerIndex && !(dwFlags & CMSG_USE_SIGNER_INDEX_FLAG))
2860 *pdwSignerIndex = 0;
2862 /* Create store to search for signer certificates */
2863 store = CertOpenStore(CERT_STORE_PROV_COLLECTION, 0, 0,
2864 CERT_STORE_CREATE_NEW_FLAG, NULL);
2865 if (!(dwFlags & CMSG_TRUSTED_SIGNER_FLAG))
2867 HCERTSTORE msgStore = CertOpenStore(CERT_STORE_PROV_MSG, 0, 0, 0,
2868 hCryptMsg);
2870 CertAddStoreToCollection(store, msgStore, 0, 0);
2871 CertCloseStore(msgStore, 0);
2873 for (i = 0; i < cSignerStore; i++)
2874 CertAddStoreToCollection(store, rghSignerStore[i], 0, 0);
2876 /* Find signer cert */
2877 if (dwFlags & CMSG_USE_SIGNER_INDEX_FLAG)
2879 CERT_INFO *signer = CRYPT_GetSignerCertInfoFromMsg(hCryptMsg,
2880 *pdwSignerIndex);
2882 if (signer)
2884 signerIndex = *pdwSignerIndex;
2885 signerCert = CertFindCertificateInStore(store, X509_ASN_ENCODING,
2886 0, CERT_FIND_SUBJECT_CERT, signer, NULL);
2887 CryptMemFree(signer);
2890 else
2892 DWORD count, size = sizeof(count);
2894 if (CryptMsgGetParam(hCryptMsg, CMSG_SIGNER_COUNT_PARAM, 0, &count,
2895 &size))
2897 for (i = 0; !signerCert && i < count; i++)
2899 CERT_INFO *signer = CRYPT_GetSignerCertInfoFromMsg(hCryptMsg,
2902 if (signer)
2904 signerCert = CertFindCertificateInStore(store,
2905 X509_ASN_ENCODING, 0, CERT_FIND_SUBJECT_CERT, signer,
2906 NULL);
2907 if (signerCert)
2908 signerIndex = i;
2909 CryptMemFree(signer);
2913 if (!signerCert)
2914 SetLastError(CRYPT_E_NO_TRUSTED_SIGNER);
2916 if (signerCert)
2918 if (!(dwFlags & CMSG_SIGNER_ONLY_FLAG))
2919 ret = CryptMsgControl(hCryptMsg, 0, CMSG_CTRL_VERIFY_SIGNATURE,
2920 signerCert->pCertInfo);
2921 else
2922 ret = TRUE;
2923 if (ret)
2925 if (ppSigner)
2926 *ppSigner = CertDuplicateCertificateContext(signerCert);
2927 if (pdwSignerIndex)
2928 *pdwSignerIndex = signerIndex;
2930 CertFreeCertificateContext(signerCert);
2933 CertCloseStore(store, 0);
2934 return ret;
2937 BOOL WINAPI CryptMsgVerifyCountersignatureEncodedEx(HCRYPTPROV_LEGACY hCryptProv,
2938 DWORD dwEncodingType, PBYTE pbSignerInfo, DWORD cbSignerInfo,
2939 PBYTE pbSignerInfoCountersignature, DWORD cbSignerInfoCountersignature,
2940 DWORD dwSignerType, void *pvSigner, DWORD dwFlags, void *pvReserved)
2942 FIXME("(%08lx, %08x, %p, %d, %p, %d, %d, %p, %08x, %p): stub\n", hCryptProv,
2943 dwEncodingType, pbSignerInfo, cbSignerInfo, pbSignerInfoCountersignature,
2944 cbSignerInfoCountersignature, dwSignerType, pvSigner, dwFlags, pvReserved);
2945 return FALSE;
2948 BOOL WINAPI CryptMsgEncodeAndSignCTL(DWORD dwMsgEncodingType,
2949 PCTL_INFO pCtlInfo, PCMSG_SIGNED_ENCODE_INFO pSignInfo, DWORD dwFlags,
2950 BYTE *pbEncoded, DWORD *pcbEncoded)
2952 BOOL ret;
2953 BYTE *pbCtlContent;
2954 DWORD cbCtlContent;
2956 TRACE("(%08x, %p, %p, %08x, %p, %p)\n", dwMsgEncodingType, pCtlInfo,
2957 pSignInfo, dwFlags, pbEncoded, pcbEncoded);
2959 if (dwFlags)
2961 FIXME("unimplemented for flags %08x\n", dwFlags);
2962 return FALSE;
2964 if ((ret = CryptEncodeObjectEx(dwMsgEncodingType, PKCS_CTL, pCtlInfo,
2965 CRYPT_ENCODE_ALLOC_FLAG, NULL, &pbCtlContent, &cbCtlContent)))
2967 ret = CryptMsgSignCTL(dwMsgEncodingType, pbCtlContent, cbCtlContent,
2968 pSignInfo, dwFlags, pbEncoded, pcbEncoded);
2969 LocalFree(pbCtlContent);
2971 return ret;
2974 BOOL WINAPI CryptMsgSignCTL(DWORD dwMsgEncodingType, BYTE *pbCtlContent,
2975 DWORD cbCtlContent, PCMSG_SIGNED_ENCODE_INFO pSignInfo, DWORD dwFlags,
2976 BYTE *pbEncoded, DWORD *pcbEncoded)
2978 static char oid_ctl[] = szOID_CTL;
2979 BOOL ret;
2980 HCRYPTMSG msg;
2982 TRACE("(%08x, %p, %d, %p, %08x, %p, %p)\n", dwMsgEncodingType,
2983 pbCtlContent, cbCtlContent, pSignInfo, dwFlags, pbEncoded, pcbEncoded);
2985 if (dwFlags)
2987 FIXME("unimplemented for flags %08x\n", dwFlags);
2988 return FALSE;
2990 msg = CryptMsgOpenToEncode(dwMsgEncodingType, 0, CMSG_SIGNED, pSignInfo,
2991 oid_ctl, NULL);
2992 if (msg)
2994 ret = CryptMsgUpdate(msg, pbCtlContent, cbCtlContent, TRUE);
2995 if (ret)
2996 ret = CryptMsgGetParam(msg, CMSG_CONTENT_PARAM, 0, pbEncoded,
2997 pcbEncoded);
2998 CryptMsgClose(msg);
3000 else
3001 ret = FALSE;
3002 return ret;