?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
crypt32: Support the KeyId type of signer in CMSG_SIGNER_INFO_PARAM.
[wine.git] / dlls / crypt32 / msg.c
blob0ec75af20c433ea73dc6b81d882c952590a3ed86
1 /*
2 * Copyright 2007 Juan Lang
3 *
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.
8 *
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.
13 *
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
17 */
18
19 #include "config.h"
20 #include "wine/port.h"
21
22 #include <stdarg.h>
23 #include "windef.h"
24 #include "winbase.h"
25 #include "wincrypt.h"
26 #include "snmp.h"
27
28 #include "wine/debug.h"
29 #include "wine/exception.h"
30 #include "crypt32_private.h"
31
32 WINE_DEFAULT_DEBUG_CHANNEL(crypt);
33
34 /* Called when a message's ref count reaches zero. Free any message-specific
35 * data here.
36 */
37 typedef void (*CryptMsgCloseFunc)(HCRYPTMSG msg);
38
39 typedef BOOL (*CryptMsgGetParamFunc)(HCRYPTMSG hCryptMsg, DWORD dwParamType,
40 DWORD dwIndex, void *pvData, DWORD *pcbData);
41
42 typedef BOOL (*CryptMsgUpdateFunc)(HCRYPTMSG hCryptMsg, const BYTE *pbData,
43 DWORD cbData, BOOL fFinal);
44
45 typedef BOOL (*CryptMsgControlFunc)(HCRYPTMSG hCryptMsg, DWORD dwFlags,
46 DWORD dwCtrlType, const void *pvCtrlPara);
47
48 BOOL CRYPT_DefaultMsgControl(HCRYPTMSG hCryptMsg, DWORD dwFlags,
49 DWORD dwCtrlType, const void *pvCtrlPara)
50 {
51 TRACE("(%p, %08x, %d, %p)\n", hCryptMsg, dwFlags, dwCtrlType, pvCtrlPara);
52 SetLastError(E_INVALIDARG);
53 return FALSE;
54 }
55
56 typedef enum _CryptMsgState {
57 MsgStateInit,
58 MsgStateUpdated,
59 MsgStateDataFinalized,
60 MsgStateFinalized
61 } CryptMsgState;
62
63 typedef struct _CryptMsgBase
64 {
65 LONG ref;
66 DWORD open_flags;
67 BOOL streamed;
68 CMSG_STREAM_INFO stream_info;
69 CryptMsgState state;
70 CryptMsgCloseFunc close;
71 CryptMsgUpdateFunc update;
72 CryptMsgGetParamFunc get_param;
73 CryptMsgControlFunc control;
74 } CryptMsgBase;
75
76 static inline void CryptMsgBase_Init(CryptMsgBase *msg, DWORD dwFlags,
77 PCMSG_STREAM_INFO pStreamInfo, CryptMsgCloseFunc close,
78 CryptMsgGetParamFunc get_param, CryptMsgUpdateFunc update,
79 CryptMsgControlFunc control)
80 {
81 msg->ref = 1;
82 msg->open_flags = dwFlags;
83 if (pStreamInfo)
84 {
85 msg->streamed = TRUE;
86 msg->stream_info = *pStreamInfo;
87 }
88 else
89 {
90 msg->streamed = FALSE;
91 memset(&msg->stream_info, 0, sizeof(msg->stream_info));
92 }
93 msg->close = close;
94 msg->get_param = get_param;
95 msg->update = update;
96 msg->control = control;
97 msg->state = MsgStateInit;
98 }
99
100 typedef struct _CDataEncodeMsg
101 {
102 CryptMsgBase base;
103 DWORD bare_content_len;
104 LPBYTE bare_content;
105 } CDataEncodeMsg;
106
107 static const BYTE empty_data_content[] = { 0x04,0x00 };
108
109 static void CDataEncodeMsg_Close(HCRYPTMSG hCryptMsg)
110 {
111 CDataEncodeMsg *msg = (CDataEncodeMsg *)hCryptMsg;
112
113 if (msg->bare_content != empty_data_content)
114 LocalFree(msg->bare_content);
115 }
116
117 static BOOL WINAPI CRYPT_EncodeContentLength(DWORD dwCertEncodingType,
118 LPCSTR lpszStructType, const void *pvStructInfo, DWORD dwFlags,
119 PCRYPT_ENCODE_PARA pEncodePara, BYTE *pbEncoded, DWORD *pcbEncoded)
120 {
121 DWORD dataLen = *(DWORD *)pvStructInfo;
122 DWORD lenBytes;
123 BOOL ret = TRUE;
124
125 /* Trick: report bytes needed based on total message length, even though
126 * the message isn't available yet. The caller will use the length
127 * reported here to encode its length.
128 */
129 CRYPT_EncodeLen(dataLen, NULL, &lenBytes);
130 if (!pbEncoded)
131 *pcbEncoded = 1 + lenBytes + dataLen;
132 else
133 {
134 if ((ret = CRYPT_EncodeEnsureSpace(dwFlags, pEncodePara, pbEncoded,
135 pcbEncoded, 1 + lenBytes)))
136 {
137 if (dwFlags & CRYPT_ENCODE_ALLOC_FLAG)
138 pbEncoded = *(BYTE **)pbEncoded;
139 *pbEncoded++ = ASN_OCTETSTRING;
140 CRYPT_EncodeLen(dataLen, pbEncoded,
141 &lenBytes);
142 }
143 }
144 return ret;
145 }
146
147 static BOOL CRYPT_EncodeDataContentInfoHeader(CDataEncodeMsg *msg,
148 CRYPT_DATA_BLOB *header)
149 {
150 BOOL ret;
151
152 if (msg->base.streamed && msg->base.stream_info.cbContent == 0xffffffff)
153 {
154 static const BYTE headerValue[] = { 0x30,0x80,0x06,0x09,0x2a,0x86,0x48,
155 0x86,0xf7,0x0d,0x01,0x07,0x01,0xa0,0x80,0x24,0x80 };
156
157 header->pbData = LocalAlloc(0, sizeof(headerValue));
158 if (header->pbData)
159 {
160 header->cbData = sizeof(headerValue);
161 memcpy(header->pbData, headerValue, sizeof(headerValue));
162 ret = TRUE;
163 }
164 else
165 ret = FALSE;
166 }
167 else
168 {
169 struct AsnConstructedItem constructed = { 0,
170 &msg->base.stream_info.cbContent, CRYPT_EncodeContentLength };
171 struct AsnEncodeSequenceItem items[2] = {
172 { szOID_RSA_data, CRYPT_AsnEncodeOid, 0 },
173 { &constructed, CRYPT_AsnEncodeConstructed, 0 },
174 };
175
176 ret = CRYPT_AsnEncodeSequence(X509_ASN_ENCODING, items,
177 sizeof(items) / sizeof(items[0]), CRYPT_ENCODE_ALLOC_FLAG, NULL,
178 (LPBYTE)&header->pbData, &header->cbData);
179 if (ret)
180 {
181 /* Trick: subtract the content length from the reported length,
182 * as the actual content hasn't come yet.
183 */
184 header->cbData -= msg->base.stream_info.cbContent;
185 }
186 }
187 return ret;
188 }
189
190 static BOOL CDataEncodeMsg_Update(HCRYPTMSG hCryptMsg, const BYTE *pbData,
191 DWORD cbData, BOOL fFinal)
192 {
193 CDataEncodeMsg *msg = (CDataEncodeMsg *)hCryptMsg;
194 BOOL ret = FALSE;
195
196 if (msg->base.state == MsgStateFinalized)
197 SetLastError(CRYPT_E_MSG_ERROR);
198 else if (msg->base.streamed)
199 {
200 __TRY
201 {
202 if (msg->base.state != MsgStateUpdated)
203 {
204 CRYPT_DATA_BLOB header;
205
206 ret = CRYPT_EncodeDataContentInfoHeader(msg, &header);
207 if (ret)
208 {
209 ret = msg->base.stream_info.pfnStreamOutput(
210 msg->base.stream_info.pvArg, header.pbData, header.cbData,
211 FALSE);
212 LocalFree(header.pbData);
213 }
214 }
215 /* Curiously, every indefinite-length streamed update appears to
216 * get its own tag and length, regardless of fFinal.
217 */
218 if (msg->base.stream_info.cbContent == 0xffffffff)
219 {
220 BYTE *header;
221 DWORD headerLen;
222
223 ret = CRYPT_EncodeContentLength(X509_ASN_ENCODING, NULL,
224 &cbData, CRYPT_ENCODE_ALLOC_FLAG, NULL, (BYTE *)&header,
225 &headerLen);
226 if (ret)
227 {
228 ret = msg->base.stream_info.pfnStreamOutput(
229 msg->base.stream_info.pvArg, header, headerLen,
230 FALSE);
231 LocalFree(header);
232 }
233 }
234 if (!fFinal)
235 {
236 ret = msg->base.stream_info.pfnStreamOutput(
237 msg->base.stream_info.pvArg, (BYTE *)pbData, cbData,
238 FALSE);
239 msg->base.state = MsgStateUpdated;
240 }
241 else
242 {
243 msg->base.state = MsgStateFinalized;
244 if (msg->base.stream_info.cbContent == 0xffffffff)
245 {
246 BYTE indefinite_trailer[6] = { 0 };
247
248 ret = msg->base.stream_info.pfnStreamOutput(
249 msg->base.stream_info.pvArg, (BYTE *)pbData, cbData,
250 FALSE);
251 if (ret)
252 ret = msg->base.stream_info.pfnStreamOutput(
253 msg->base.stream_info.pvArg, indefinite_trailer,
254 sizeof(indefinite_trailer), TRUE);
255 }
256 else
257 ret = msg->base.stream_info.pfnStreamOutput(
258 msg->base.stream_info.pvArg, (BYTE *)pbData, cbData, TRUE);
259 }
260 }
261 __EXCEPT_PAGE_FAULT
262 {
263 SetLastError(STATUS_ACCESS_VIOLATION);
264 ret = FALSE;
265 }
266 __ENDTRY;
267 }
268 else
269 {
270 if (!fFinal)
271 {
272 if (msg->base.open_flags & CMSG_DETACHED_FLAG)
273 SetLastError(E_INVALIDARG);
274 else
275 SetLastError(CRYPT_E_MSG_ERROR);
276 }
277 else
278 {
279 msg->base.state = MsgStateFinalized;
280 if (!cbData)
281 SetLastError(E_INVALIDARG);
282 else
283 {
284 CRYPT_DATA_BLOB blob = { cbData, (LPBYTE)pbData };
285
286 /* non-streamed data messages don't allow non-final updates,
287 * don't bother checking whether data already exist, they can't.
288 */
289 ret = CryptEncodeObjectEx(X509_ASN_ENCODING, X509_OCTET_STRING,
290 &blob, CRYPT_ENCODE_ALLOC_FLAG, NULL, &msg->bare_content,
291 &msg->bare_content_len);
292 }
293 }
294 }
295 return ret;
296 }
297
298 static BOOL CRYPT_CopyParam(void *pvData, DWORD *pcbData, const void *src,
299 DWORD len)
300 {
301 BOOL ret = TRUE;
302
303 if (!pvData)
304 *pcbData = len;
305 else if (*pcbData < len)
306 {
307 *pcbData = len;
308 SetLastError(ERROR_MORE_DATA);
309 ret = FALSE;
310 }
311 else
312 {
313 *pcbData = len;
314 memcpy(pvData, src, len);
315 }
316 return ret;
317 }
318
319 static BOOL CDataEncodeMsg_GetParam(HCRYPTMSG hCryptMsg, DWORD dwParamType,
320 DWORD dwIndex, void *pvData, DWORD *pcbData)
321 {
322 CDataEncodeMsg *msg = (CDataEncodeMsg *)hCryptMsg;
323 BOOL ret = FALSE;
324
325 switch (dwParamType)
326 {
327 case CMSG_CONTENT_PARAM:
328 if (msg->base.streamed)
329 SetLastError(E_INVALIDARG);
330 else
331 {
332 CRYPT_CONTENT_INFO info;
333 char rsa_data[] = "1.2.840.113549.1.7.1";
334
335 info.pszObjId = rsa_data;
336 info.Content.cbData = msg->bare_content_len;
337 info.Content.pbData = msg->bare_content;
338 ret = CryptEncodeObject(X509_ASN_ENCODING, PKCS_CONTENT_INFO, &info,
339 pvData, pcbData);
340 }
341 break;
342 case CMSG_BARE_CONTENT_PARAM:
343 if (msg->base.streamed)
344 SetLastError(E_INVALIDARG);
345 else
346 ret = CRYPT_CopyParam(pvData, pcbData, msg->bare_content,
347 msg->bare_content_len);
348 break;
349 default:
350 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
351 }
352 return ret;
353 }
354
355 static HCRYPTMSG CDataEncodeMsg_Open(DWORD dwFlags, const void *pvMsgEncodeInfo,
356 LPSTR pszInnerContentObjID, PCMSG_STREAM_INFO pStreamInfo)
357 {
358 CDataEncodeMsg *msg;
359
360 if (pvMsgEncodeInfo)
361 {
362 SetLastError(E_INVALIDARG);
363 return NULL;
364 }
365 msg = CryptMemAlloc(sizeof(CDataEncodeMsg));
366 if (msg)
367 {
368 CryptMsgBase_Init((CryptMsgBase *)msg, dwFlags, pStreamInfo,
369 CDataEncodeMsg_Close, CDataEncodeMsg_GetParam, CDataEncodeMsg_Update,
370 CRYPT_DefaultMsgControl);
371 msg->bare_content_len = sizeof(empty_data_content);
372 msg->bare_content = (LPBYTE)empty_data_content;
373 }
374 return (HCRYPTMSG)msg;
375 }
376
377 typedef struct _CHashEncodeMsg
378 {
379 CryptMsgBase base;
380 HCRYPTPROV prov;
381 HCRYPTHASH hash;
382 CRYPT_DATA_BLOB data;
383 } CHashEncodeMsg;
384
385 static void CHashEncodeMsg_Close(HCRYPTMSG hCryptMsg)
386 {
387 CHashEncodeMsg *msg = (CHashEncodeMsg *)hCryptMsg;
388
389 CryptMemFree(msg->data.pbData);
390 CryptDestroyHash(msg->hash);
391 if (msg->base.open_flags & CMSG_CRYPT_RELEASE_CONTEXT_FLAG)
392 CryptReleaseContext(msg->prov, 0);
393 }
394
395 static BOOL CRYPT_EncodePKCSDigestedData(CHashEncodeMsg *msg, void *pvData,
396 DWORD *pcbData)
397 {
398 BOOL ret;
399 ALG_ID algID;
400 DWORD size = sizeof(algID);
401
402 ret = CryptGetHashParam(msg->hash, HP_ALGID, (BYTE *)&algID, &size, 0);
403 if (ret)
404 {
405 CRYPT_DIGESTED_DATA digestedData = { 0 };
406 char oid_rsa_data[] = szOID_RSA_data;
407
408 digestedData.version = CMSG_HASHED_DATA_PKCS_1_5_VERSION;
409 digestedData.DigestAlgorithm.pszObjId = (LPSTR)CertAlgIdToOID(algID);
410 /* FIXME: what about digestedData.DigestAlgorithm.Parameters? */
411 /* Quirk: OID is only encoded messages if an update has happened */
412 if (msg->base.state != MsgStateInit)
413 digestedData.ContentInfo.pszObjId = oid_rsa_data;
414 if (!(msg->base.open_flags & CMSG_DETACHED_FLAG) && msg->data.cbData)
415 {
416 ret = CRYPT_AsnEncodeOctets(0, NULL, &msg->data,
417 CRYPT_ENCODE_ALLOC_FLAG, NULL,
418 (LPBYTE)&digestedData.ContentInfo.Content.pbData,
419 &digestedData.ContentInfo.Content.cbData);
420 }
421 if (msg->base.state == MsgStateFinalized)
422 {
423 size = sizeof(DWORD);
424 ret = CryptGetHashParam(msg->hash, HP_HASHSIZE,
425 (LPBYTE)&digestedData.hash.cbData, &size, 0);
426 if (ret)
427 {
428 digestedData.hash.pbData = CryptMemAlloc(
429 digestedData.hash.cbData);
430 ret = CryptGetHashParam(msg->hash, HP_HASHVAL,
431 digestedData.hash.pbData, &digestedData.hash.cbData, 0);
432 }
433 }
434 if (ret)
435 ret = CRYPT_AsnEncodePKCSDigestedData(&digestedData, pvData,
436 pcbData);
437 CryptMemFree(digestedData.hash.pbData);
438 LocalFree(digestedData.ContentInfo.Content.pbData);
439 }
440 return ret;
441 }
442
443 static BOOL CHashEncodeMsg_GetParam(HCRYPTMSG hCryptMsg, DWORD dwParamType,
444 DWORD dwIndex, void *pvData, DWORD *pcbData)
445 {
446 CHashEncodeMsg *msg = (CHashEncodeMsg *)hCryptMsg;
447 BOOL ret = FALSE;
448
449 TRACE("(%p, %d, %d, %p, %p)\n", hCryptMsg, dwParamType, dwIndex,
450 pvData, pcbData);
451
452 switch (dwParamType)
453 {
454 case CMSG_BARE_CONTENT_PARAM:
455 if (msg->base.streamed)
456 SetLastError(E_INVALIDARG);
457 else
458 ret = CRYPT_EncodePKCSDigestedData(msg, pvData, pcbData);
459 break;
460 case CMSG_CONTENT_PARAM:
461 {
462 CRYPT_CONTENT_INFO info;
463
464 ret = CryptMsgGetParam(hCryptMsg, CMSG_BARE_CONTENT_PARAM, 0, NULL,
465 &info.Content.cbData);
466 if (ret)
467 {
468 info.Content.pbData = CryptMemAlloc(info.Content.cbData);
469 if (info.Content.pbData)
470 {
471 ret = CryptMsgGetParam(hCryptMsg, CMSG_BARE_CONTENT_PARAM, 0,
472 info.Content.pbData, &info.Content.cbData);
473 if (ret)
474 {
475 char oid_rsa_hashed[] = szOID_RSA_hashedData;
476
477 info.pszObjId = oid_rsa_hashed;
478 ret = CryptEncodeObjectEx(X509_ASN_ENCODING,
479 PKCS_CONTENT_INFO, &info, 0, NULL, pvData, pcbData);
480 }
481 CryptMemFree(info.Content.pbData);
482 }
483 else
484 ret = FALSE;
485 }
486 break;
487 }
488 case CMSG_COMPUTED_HASH_PARAM:
489 ret = CryptGetHashParam(msg->hash, HP_HASHVAL, (BYTE *)pvData, pcbData,
490 0);
491 break;
492 case CMSG_VERSION_PARAM:
493 if (msg->base.state != MsgStateFinalized)
494 SetLastError(CRYPT_E_MSG_ERROR);
495 else
496 {
497 DWORD version = CMSG_HASHED_DATA_PKCS_1_5_VERSION;
498
499 /* Since the data are always encoded as octets, the version is
500 * always 0 (see rfc3852, section 7)
501 */
502 ret = CRYPT_CopyParam(pvData, pcbData, &version, sizeof(version));
503 }
504 break;
505 default:
506 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
507 }
508 return ret;
509 }
510
511 static BOOL CHashEncodeMsg_Update(HCRYPTMSG hCryptMsg, const BYTE *pbData,
512 DWORD cbData, BOOL fFinal)
513 {
514 CHashEncodeMsg *msg = (CHashEncodeMsg *)hCryptMsg;
515 BOOL ret = FALSE;
516
517 TRACE("(%p, %p, %d, %d)\n", hCryptMsg, pbData, cbData, fFinal);
518
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))
522 {
523 /* Doesn't do much, as stream output is never called, and you
524 * can't get the content.
525 */
526 ret = CryptHashData(msg->hash, pbData, cbData, 0);
527 msg->base.state = fFinal ? MsgStateFinalized : MsgStateUpdated;
528 }
529 else
530 {
531 if (!fFinal)
532 SetLastError(CRYPT_E_MSG_ERROR);
533 else
534 {
535 ret = CryptHashData(msg->hash, pbData, cbData, 0);
536 if (ret)
537 {
538 msg->data.pbData = CryptMemAlloc(cbData);
539 if (msg->data.pbData)
540 {
541 memcpy(msg->data.pbData + msg->data.cbData, pbData, cbData);
542 msg->data.cbData += cbData;
543 }
544 else
545 ret = FALSE;
546 }
547 msg->base.state = MsgStateFinalized;
548 }
549 }
550 return ret;
551 }
552
553 static HCRYPTMSG CHashEncodeMsg_Open(DWORD dwFlags, const void *pvMsgEncodeInfo,
554 LPSTR pszInnerContentObjID, PCMSG_STREAM_INFO pStreamInfo)
555 {
556 CHashEncodeMsg *msg;
557 const CMSG_HASHED_ENCODE_INFO *info =
558 (const CMSG_HASHED_ENCODE_INFO *)pvMsgEncodeInfo;
559 HCRYPTPROV prov;
560 ALG_ID algID;
561
562 if (info->cbSize != sizeof(CMSG_HASHED_ENCODE_INFO))
563 {
564 SetLastError(E_INVALIDARG);
565 return NULL;
566 }
567 if (!(algID = CertOIDToAlgId(info->HashAlgorithm.pszObjId)))
568 {
569 SetLastError(CRYPT_E_UNKNOWN_ALGO);
570 return NULL;
571 }
572 if (info->hCryptProv)
573 prov = info->hCryptProv;
574 else
575 {
576 prov = CRYPT_GetDefaultProvider();
577 dwFlags &= ~CMSG_CRYPT_RELEASE_CONTEXT_FLAG;
578 }
579 msg = CryptMemAlloc(sizeof(CHashEncodeMsg));
580 if (msg)
581 {
582 CryptMsgBase_Init((CryptMsgBase *)msg, dwFlags, pStreamInfo,
583 CHashEncodeMsg_Close, CHashEncodeMsg_GetParam, CHashEncodeMsg_Update,
584 CRYPT_DefaultMsgControl);
585 msg->prov = prov;
586 msg->data.cbData = 0;
587 msg->data.pbData = NULL;
588 if (!CryptCreateHash(prov, algID, 0, 0, &msg->hash))
589 {
590 CryptMsgClose(msg);
591 msg = NULL;
592 }
593 }
594 return (HCRYPTMSG)msg;
595 }
596
597 typedef struct _CMSG_SIGNER_ENCODE_INFO_WITH_CMS
598 {
599 DWORD cbSize;
600 PCERT_INFO pCertInfo;
601 HCRYPTPROV hCryptProv;
602 DWORD dwKeySpec;
603 CRYPT_ALGORITHM_IDENTIFIER HashAlgorithm;
604 void *pvHashAuxInfo;
605 DWORD cAuthAttr;
606 PCRYPT_ATTRIBUTE rgAuthAttr;
607 DWORD cUnauthAttr;
608 PCRYPT_ATTRIBUTE rgUnauthAttr;
609 CERT_ID SignerId;
610 CRYPT_ALGORITHM_IDENTIFIER HashEncryptionAlgorithm;
611 void *pvHashEncryptionAuxInfo;
612 } CMSG_SIGNER_ENCODE_INFO_WITH_CMS, *PCMSG_SIGNER_ENCODE_INFO_WITH_CMS;
613
614 typedef struct _CMSG_SIGNED_ENCODE_INFO_WITH_CMS
615 {
616 DWORD cbSize;
617 DWORD cSigners;
618 PCMSG_SIGNER_ENCODE_INFO_WITH_CMS rgSigners;
619 DWORD cCertEncoded;
620 PCERT_BLOB rgCertEncoded;
621 DWORD cCrlEncoded;
622 PCRL_BLOB rgCrlEncoded;
623 DWORD cAttrCertEncoded;
624 PCERT_BLOB rgAttrCertEncoded;
625 } CMSG_SIGNED_ENCODE_INFO_WITH_CMS, *PCMSG_SIGNED_ENCODE_INFO_WITH_CMS;
626
627 static BOOL CRYPT_IsValidSigner(CMSG_SIGNER_ENCODE_INFO_WITH_CMS *signer)
628 {
629 if (signer->cbSize != sizeof(CMSG_SIGNER_ENCODE_INFO) &&
630 signer->cbSize != sizeof(CMSG_SIGNER_ENCODE_INFO_WITH_CMS))
631 {
632 SetLastError(E_INVALIDARG);
633 return FALSE;
634 }
635 if (signer->cbSize == sizeof(CMSG_SIGNER_ENCODE_INFO))
636 {
637 if (!signer->pCertInfo->SerialNumber.cbData)
638 {
639 SetLastError(E_INVALIDARG);
640 return FALSE;
641 }
642 if (!signer->pCertInfo->Issuer.cbData)
643 {
644 SetLastError(E_INVALIDARG);
645 return FALSE;
646 }
647 }
648 else if (signer->cbSize == sizeof(CMSG_SIGNER_ENCODE_INFO_WITH_CMS))
649 {
650 switch (signer->SignerId.dwIdChoice)
651 {
652 case 0:
653 if (!signer->pCertInfo->SerialNumber.cbData)
654 {
655 SetLastError(E_INVALIDARG);
656 return FALSE;
657 }
658 if (!signer->pCertInfo->Issuer.cbData)
659 {
660 SetLastError(E_INVALIDARG);
661 return FALSE;
662 }
663 break;
664 case CERT_ID_ISSUER_SERIAL_NUMBER:
665 if (!signer->SignerId.IssuerSerialNumber.SerialNumber.cbData)
666 {
667 SetLastError(E_INVALIDARG);
668 return FALSE;
669 }
670 if (!signer->SignerId.IssuerSerialNumber.Issuer.cbData)
671 {
672 SetLastError(E_INVALIDARG);
673 return FALSE;
674 }
675 break;
676 case CERT_ID_KEY_IDENTIFIER:
677 if (!signer->SignerId.KeyId.cbData)
678 {
679 SetLastError(E_INVALIDARG);
680 return FALSE;
681 }
682 break;
683 default:
684 SetLastError(E_INVALIDARG);
685 }
686 if (signer->HashEncryptionAlgorithm.pszObjId)
687 {
688 FIXME("CMSG_SIGNER_ENCODE_INFO with CMS fields unsupported\n");
689 return FALSE;
690 }
691 }
692 if (!signer->hCryptProv)
693 {
694 SetLastError(E_INVALIDARG);
695 return FALSE;
696 }
697 if (!CertOIDToAlgId(signer->HashAlgorithm.pszObjId))
698 {
699 SetLastError(CRYPT_E_UNKNOWN_ALGO);
700 return FALSE;
701 }
702 return TRUE;
703 }
704
705 static BOOL CRYPT_ConstructBlob(CRYPT_DATA_BLOB *out, const CRYPT_DATA_BLOB *in)
706 {
707 BOOL ret = TRUE;
708
709 out->cbData = in->cbData;
710 if (out->cbData)
711 {
712 out->pbData = CryptMemAlloc(out->cbData);
713 if (out->pbData)
714 memcpy(out->pbData, in->pbData, out->cbData);
715 else
716 ret = FALSE;
717 }
718 else
719 out->pbData = NULL;
720 return ret;
721 }
722
723 typedef struct _BlobArray
724 {
725 DWORD cBlobs;
726 PCRYPT_DATA_BLOB blobs;
727 } BlobArray;
728
729 static BOOL CRYPT_ConstructBlobArray(BlobArray *out, const BlobArray *in)
730 {
731 BOOL ret = TRUE;
732
733 out->cBlobs = in->cBlobs;
734 if (out->cBlobs)
735 {
736 out->blobs = CryptMemAlloc(out->cBlobs * sizeof(CRYPT_DATA_BLOB));
737 if (out->blobs)
738 {
739 DWORD i;
740
741 memset(out->blobs, 0, out->cBlobs * sizeof(CRYPT_DATA_BLOB));
742 for (i = 0; ret && i < out->cBlobs; i++)
743 ret = CRYPT_ConstructBlob(&out->blobs[i], &in->blobs[i]);
744 }
745 else
746 ret = FALSE;
747 }
748 return ret;
749 }
750
751 static void CRYPT_FreeBlobArray(BlobArray *array)
752 {
753 DWORD i;
754
755 for (i = 0; i < array->cBlobs; i++)
756 CryptMemFree(array->blobs[i].pbData);
757 CryptMemFree(array->blobs);
758 }
759
760 static BOOL CRYPT_ConstructAttribute(CRYPT_ATTRIBUTE *out,
761 const CRYPT_ATTRIBUTE *in)
762 {
763 BOOL ret;
764
765 out->pszObjId = CryptMemAlloc(strlen(in->pszObjId) + 1);
766 if (out->pszObjId)
767 {
768 strcpy(out->pszObjId, in->pszObjId);
769 ret = CRYPT_ConstructBlobArray((BlobArray *)&out->cValue,
770 (const BlobArray *)&in->cValue);
771 }
772 else
773 ret = FALSE;
774 return ret;
775 }
776
777 static BOOL CRYPT_ConstructAttributes(CRYPT_ATTRIBUTES *out,
778 const CRYPT_ATTRIBUTES *in)
779 {
780 BOOL ret = TRUE;
781
782 out->cAttr = in->cAttr;
783 if (out->cAttr)
784 {
785 out->rgAttr = CryptMemAlloc(out->cAttr * sizeof(CRYPT_ATTRIBUTE));
786 if (out->rgAttr)
787 {
788 DWORD i;
789
790 memset(out->rgAttr, 0, out->cAttr * sizeof(CRYPT_ATTRIBUTE));
791 for (i = 0; ret && i < out->cAttr; i++)
792 ret = CRYPT_ConstructAttribute(&out->rgAttr[i], &in->rgAttr[i]);
793 }
794 else
795 ret = FALSE;
796 }
797 else
798 out->rgAttr = NULL;
799 return ret;
800 }
801
802 /* Constructs a CMSG_CMS_SIGNER_INFO from a CMSG_SIGNER_ENCODE_INFO_WITH_CMS. */
803 static BOOL CSignerInfo_Construct(CMSG_CMS_SIGNER_INFO *info,
804 const CMSG_SIGNER_ENCODE_INFO_WITH_CMS *in)
805 {
806 BOOL ret;
807
808 if (in->cbSize == sizeof(CMSG_SIGNER_ENCODE_INFO))
809 {
810 info->dwVersion = CMSG_SIGNER_INFO_V1;
811 ret = CRYPT_ConstructBlob(&info->SignerId.IssuerSerialNumber.Issuer,
812 &in->pCertInfo->Issuer);
813 if (ret)
814 ret = CRYPT_ConstructBlob(
815 &info->SignerId.IssuerSerialNumber.SerialNumber,
816 &in->pCertInfo->SerialNumber);
817 info->SignerId.dwIdChoice = CERT_ID_ISSUER_SERIAL_NUMBER;
818 }
819 else
820 {
821 /* Implicitly in->cbSize == sizeof(CMSG_SIGNER_ENCODE_INFO_WITH_CMS).
822 * See CRYPT_IsValidSigner.
823 */
824 if (!in->SignerId.dwIdChoice)
825 {
826 info->dwVersion = CMSG_SIGNER_INFO_V1;
827 ret = CRYPT_ConstructBlob(&info->SignerId.IssuerSerialNumber.Issuer,
828 &in->pCertInfo->Issuer);
829 if (ret)
830 ret = CRYPT_ConstructBlob(
831 &info->SignerId.IssuerSerialNumber.SerialNumber,
832 &in->pCertInfo->SerialNumber);
833 info->SignerId.dwIdChoice = CERT_ID_ISSUER_SERIAL_NUMBER;
834 }
835 else if (in->SignerId.dwIdChoice == CERT_ID_ISSUER_SERIAL_NUMBER)
836 {
837 info->dwVersion = CMSG_SIGNER_INFO_V1;
838 info->SignerId.dwIdChoice = CERT_ID_ISSUER_SERIAL_NUMBER;
839 ret = CRYPT_ConstructBlob(&info->SignerId.IssuerSerialNumber.Issuer,
840 &in->SignerId.IssuerSerialNumber.Issuer);
841 if (ret)
842 ret = CRYPT_ConstructBlob(
843 &info->SignerId.IssuerSerialNumber.SerialNumber,
844 &in->SignerId.IssuerSerialNumber.SerialNumber);
845 }
846 else
847 {
848 /* Implicitly dwIdChoice == CERT_ID_KEY_IDENTIFIER */
849 info->dwVersion = CMSG_SIGNER_INFO_V3;
850 info->SignerId.dwIdChoice = CERT_ID_KEY_IDENTIFIER;
851 ret = CRYPT_ConstructBlob(&info->SignerId.KeyId,
852 &in->SignerId.KeyId);
853 }
854 }
855 /* Assumption: algorithm IDs will point to static strings, not
856 * stack-based ones, so copying the pointer values is safe.
857 */
858 info->HashAlgorithm.pszObjId = in->HashAlgorithm.pszObjId;
859 if (ret)
860 ret = CRYPT_ConstructBlob(&info->HashAlgorithm.Parameters,
861 &in->HashAlgorithm.Parameters);
862 memset(&info->HashEncryptionAlgorithm, 0,
863 sizeof(info->HashEncryptionAlgorithm));
864 if (ret)
865 ret = CRYPT_ConstructAttributes(&info->AuthAttrs,
866 (CRYPT_ATTRIBUTES *)&in->cAuthAttr);
867 if (ret)
868 ret = CRYPT_ConstructAttributes(&info->UnauthAttrs,
869 (CRYPT_ATTRIBUTES *)&in->cUnauthAttr);
870 return ret;
871 }
872
873 static void CSignerInfo_Free(CMSG_CMS_SIGNER_INFO *info)
874 {
875 DWORD i, j;
876
877 if (info->SignerId.dwIdChoice == CERT_ID_ISSUER_SERIAL_NUMBER)
878 {
879 CryptMemFree(info->SignerId.IssuerSerialNumber.Issuer.pbData);
880 CryptMemFree(info->SignerId.IssuerSerialNumber.SerialNumber.pbData);
881 }
882 else
883 CryptMemFree(info->SignerId.KeyId.pbData);
884 CryptMemFree(info->HashAlgorithm.Parameters.pbData);
885 CryptMemFree(info->EncryptedHash.pbData);
886 for (i = 0; i < info->AuthAttrs.cAttr; i++)
887 {
888 for (j = 0; j < info->AuthAttrs.rgAttr[i].cValue; j++)
889 CryptMemFree(info->AuthAttrs.rgAttr[i].rgValue[j].pbData);
890 CryptMemFree(info->AuthAttrs.rgAttr[i].rgValue);
891 CryptMemFree(info->AuthAttrs.rgAttr[i].pszObjId);
892 }
893 CryptMemFree(info->AuthAttrs.rgAttr);
894 for (i = 0; i < info->UnauthAttrs.cAttr; i++)
895 {
896 for (j = 0; j < info->UnauthAttrs.rgAttr[i].cValue; j++)
897 CryptMemFree(info->UnauthAttrs.rgAttr[i].rgValue[j].pbData);
898 CryptMemFree(info->UnauthAttrs.rgAttr[i].rgValue);
899 CryptMemFree(info->UnauthAttrs.rgAttr[i].pszObjId);
900 }
901 CryptMemFree(info->UnauthAttrs.rgAttr);
902 }
903
904 typedef struct _CSignerHandles
905 {
906 HCRYPTHASH contentHash;
907 HCRYPTHASH authAttrHash;
908 } CSignerHandles;
909
910 typedef struct _CSignedMsgData
911 {
912 CRYPT_SIGNED_INFO *info;
913 DWORD cSignerHandle;
914 CSignerHandles *signerHandles;
915 } CSignedMsgData;
916
917 /* Constructs the signer handles for the signerIndex'th signer of msg_data.
918 * Assumes signerIndex is a valid idnex, and that msg_data's info has already
919 * been constructed.
920 */
921 static BOOL CSignedMsgData_ConstructSignerHandles(CSignedMsgData *msg_data,
922 DWORD signerIndex, HCRYPTPROV crypt_prov)
923 {
924 ALG_ID algID;
925 BOOL ret;
926
927 algID = CertOIDToAlgId(
928 msg_data->info->rgSignerInfo[signerIndex].HashAlgorithm.pszObjId);
929 ret = CryptCreateHash(crypt_prov, algID, 0, 0,
930 &msg_data->signerHandles->contentHash);
931 if (ret && msg_data->info->rgSignerInfo[signerIndex].AuthAttrs.cAttr > 0)
932 ret = CryptCreateHash(crypt_prov, algID, 0, 0,
933 &msg_data->signerHandles->authAttrHash);
934 return ret;
935 }
936
937 /* Allocates a CSignedMsgData's handles. Assumes its info has already been
938 * constructed.
939 */
940 static BOOL CSignedMsgData_AllocateHandles(CSignedMsgData *msg_data)
941 {
942 BOOL ret = TRUE;
943
944 if (msg_data->info->cSignerInfo)
945 {
946 msg_data->signerHandles =
947 CryptMemAlloc(msg_data->info->cSignerInfo * sizeof(CSignerHandles));
948 if (msg_data->signerHandles)
949 {
950 msg_data->cSignerHandle = msg_data->info->cSignerInfo;
951 memset(msg_data->signerHandles, 0,
952 msg_data->info->cSignerInfo * sizeof(CSignerHandles));
953 }
954 else
955 {
956 msg_data->cSignerHandle = 0;
957 ret = FALSE;
958 }
959 }
960 else
961 {
962 msg_data->cSignerHandle = 0;
963 msg_data->signerHandles = NULL;
964 }
965 return ret;
966 }
967
968 static void CSignedMsgData_CloseHandles(CSignedMsgData *msg_data)
969 {
970 DWORD i;
971
972 for (i = 0; i < msg_data->cSignerHandle; i++)
973 {
974 if (msg_data->signerHandles[i].contentHash)
975 CryptDestroyHash(msg_data->signerHandles[i].contentHash);
976 if (msg_data->signerHandles[i].authAttrHash)
977 CryptDestroyHash(msg_data->signerHandles[i].authAttrHash);
978 }
979 CryptMemFree(msg_data->signerHandles);
980 msg_data->signerHandles = NULL;
981 msg_data->cSignerHandle = 0;
982 }
983
984 static BOOL CSignedMsgData_UpdateHash(CSignedMsgData *msg_data,
985 const BYTE *pbData, DWORD cbData)
986 {
987 DWORD i;
988 BOOL ret = TRUE;
989
990 for (i = 0; ret && i < msg_data->cSignerHandle; i++)
991 ret = CryptHashData(msg_data->signerHandles[i].contentHash, pbData,
992 cbData, 0);
993 return ret;
994 }
995
996 static BOOL CRYPT_AppendAttribute(CRYPT_ATTRIBUTES *out,
997 const CRYPT_ATTRIBUTE *in)
998 {
999 BOOL ret = FALSE;
1000
1001 out->rgAttr = CryptMemRealloc(out->rgAttr,
1002 (out->cAttr + 1) * sizeof(CRYPT_ATTRIBUTE));
1003 if (out->rgAttr)
1004 {
1005 ret = CRYPT_ConstructAttribute(&out->rgAttr[out->cAttr], in);
1006 if (ret)
1007 out->cAttr++;
1008 }
1009 return ret;
1010 }
1011
1012 static BOOL CSignedMsgData_AppendMessageDigestAttribute(
1013 CSignedMsgData *msg_data, DWORD signerIndex)
1014 {
1015 BOOL ret;
1016 DWORD size;
1017 CRYPT_HASH_BLOB hash = { 0, NULL }, encodedHash = { 0, NULL };
1018 char messageDigest[] = szOID_RSA_messageDigest;
1019 CRYPT_ATTRIBUTE messageDigestAttr = { messageDigest, 1, &encodedHash };
1020
1021 size = sizeof(DWORD);
1022 ret = CryptGetHashParam(
1023 msg_data->signerHandles[signerIndex].contentHash, HP_HASHSIZE,
1024 (LPBYTE)&hash.cbData, &size, 0);
1025 if (ret)
1026 {
1027 hash.pbData = CryptMemAlloc(hash.cbData);
1028 ret = CryptGetHashParam(
1029 msg_data->signerHandles[signerIndex].contentHash, HP_HASHVAL,
1030 hash.pbData, &hash.cbData, 0);
1031 if (ret)
1032 {
1033 ret = CRYPT_AsnEncodeOctets(0, NULL, &hash, CRYPT_ENCODE_ALLOC_FLAG,
1034 NULL, (LPBYTE)&encodedHash.pbData, &encodedHash.cbData);
1035 if (ret)
1036 {
1037 ret = CRYPT_AppendAttribute(
1038 &msg_data->info->rgSignerInfo[signerIndex].AuthAttrs,
1039 &messageDigestAttr);
1040 LocalFree(encodedHash.pbData);
1041 }
1042 }
1043 CryptMemFree(hash.pbData);
1044 }
1045 return ret;
1046 }
1047
1048 typedef enum {
1049 Sign,
1050 Verify
1051 } SignOrVerify;
1052
1053 static BOOL CSignedMsgData_UpdateAuthenticatedAttributes(
1054 CSignedMsgData *msg_data, SignOrVerify flag)
1055 {
1056 DWORD i;
1057 BOOL ret = TRUE;
1058
1059 TRACE("(%p)\n", msg_data);
1060
1061 for (i = 0; ret && i < msg_data->info->cSignerInfo; i++)
1062 {
1063 if (msg_data->info->rgSignerInfo[i].AuthAttrs.cAttr)
1064 {
1065 if (flag == Sign)
1066 {
1067 BYTE oid_rsa_data_encoded[] = { 0x06,0x09,0x2a,0x86,0x48,0x86,
1068 0xf7,0x0d,0x01,0x07,0x01 };
1069 CRYPT_DATA_BLOB content = { sizeof(oid_rsa_data_encoded),
1070 oid_rsa_data_encoded };
1071 char contentType[] = szOID_RSA_contentType;
1072 CRYPT_ATTRIBUTE contentTypeAttr = { contentType, 1, &content };
1073
1074 /* FIXME: does this depend on inner OID? */
1075 ret = CRYPT_AppendAttribute(
1076 &msg_data->info->rgSignerInfo[i].AuthAttrs, &contentTypeAttr);
1077 if (ret)
1078 ret = CSignedMsgData_AppendMessageDigestAttribute(msg_data,
1079 i);
1080 }
1081 if (ret)
1082 {
1083 LPBYTE encodedAttrs;
1084 DWORD size;
1085
1086 ret = CryptEncodeObjectEx(X509_ASN_ENCODING, PKCS_ATTRIBUTES,
1087 &msg_data->info->rgSignerInfo[i].AuthAttrs,
1088 CRYPT_ENCODE_ALLOC_FLAG, NULL, (LPBYTE)&encodedAttrs, &size);
1089 if (ret)
1090 {
1091 ret = CryptHashData(
1092 msg_data->signerHandles[i].authAttrHash, encodedAttrs,
1093 size, 0);
1094 LocalFree(encodedAttrs);
1095 }
1096 }
1097 }
1098 }
1099 TRACE("returning %d\n", ret);
1100 return ret;
1101 }
1102
1103 static void CRYPT_ReverseBytes(CRYPT_HASH_BLOB *hash)
1104 {
1105 DWORD i;
1106 BYTE tmp;
1107
1108 for (i = 0; i < hash->cbData / 2; i++)
1109 {
1110 tmp = hash->pbData[hash->cbData - i - 1];
1111 hash->pbData[hash->cbData - i - 1] = hash->pbData[i];
1112 hash->pbData[i] = tmp;
1113 }
1114 }
1115
1116 static BOOL CSignedMsgData_Sign(CSignedMsgData *msg_data)
1117 {
1118 DWORD i;
1119 BOOL ret = TRUE;
1120
1121 TRACE("(%p)\n", msg_data);
1122
1123 for (i = 0; ret && i < msg_data->info->cSignerInfo; i++)
1124 {
1125 HCRYPTHASH hash;
1126
1127 if (msg_data->info->rgSignerInfo[i].AuthAttrs.cAttr)
1128 hash = msg_data->signerHandles[i].authAttrHash;
1129 else
1130 hash = msg_data->signerHandles[i].contentHash;
1131 ret = CryptSignHashW(hash, AT_SIGNATURE, NULL, 0, NULL,
1132 &msg_data->info->rgSignerInfo[i].EncryptedHash.cbData);
1133 if (ret)
1134 {
1135 msg_data->info->rgSignerInfo[i].EncryptedHash.pbData =
1136 CryptMemAlloc(
1137 msg_data->info->rgSignerInfo[i].EncryptedHash.cbData);
1138 if (msg_data->info->rgSignerInfo[i].EncryptedHash.pbData)
1139 {
1140 ret = CryptSignHashW(hash, AT_SIGNATURE, NULL, 0,
1141 msg_data->info->rgSignerInfo[i].EncryptedHash.pbData,
1142 &msg_data->info->rgSignerInfo[i].EncryptedHash.cbData);
1143 if (ret)
1144 CRYPT_ReverseBytes(
1145 &msg_data->info->rgSignerInfo[i].EncryptedHash);
1146 }
1147 else
1148 ret = FALSE;
1149 }
1150 }
1151 return ret;
1152 }
1153
1154 static BOOL CSignedMsgData_Update(CSignedMsgData *msg_data,
1155 const BYTE *pbData, DWORD cbData, BOOL fFinal, SignOrVerify flag)
1156 {
1157 BOOL ret = CSignedMsgData_UpdateHash(msg_data, pbData, cbData);
1158
1159 if (ret && fFinal)
1160 {
1161 ret = CSignedMsgData_UpdateAuthenticatedAttributes(msg_data, flag);
1162 if (ret && flag == Sign)
1163 ret = CSignedMsgData_Sign(msg_data);
1164 }
1165 return ret;
1166 }
1167
1168 typedef struct _CSignedEncodeMsg
1169 {
1170 CryptMsgBase base;
1171 CRYPT_DATA_BLOB data;
1172 CSignedMsgData msg_data;
1173 } CSignedEncodeMsg;
1174
1175 static void CSignedEncodeMsg_Close(HCRYPTMSG hCryptMsg)
1176 {
1177 CSignedEncodeMsg *msg = (CSignedEncodeMsg *)hCryptMsg;
1178 DWORD i;
1179
1180 CryptMemFree(msg->data.pbData);
1181 CRYPT_FreeBlobArray((BlobArray *)&msg->msg_data.info->cCertEncoded);
1182 CRYPT_FreeBlobArray((BlobArray *)&msg->msg_data.info->cCrlEncoded);
1183 for (i = 0; i < msg->msg_data.info->cSignerInfo; i++)
1184 CSignerInfo_Free(&msg->msg_data.info->rgSignerInfo[i]);
1185 CSignedMsgData_CloseHandles(&msg->msg_data);
1186 CryptMemFree(msg->msg_data.info->rgSignerInfo);
1187 CryptMemFree(msg->msg_data.info);
1188 }
1189
1190 static BOOL CSignedEncodeMsg_GetParam(HCRYPTMSG hCryptMsg, DWORD dwParamType,
1191 DWORD dwIndex, void *pvData, DWORD *pcbData)
1192 {
1193 CSignedEncodeMsg *msg = (CSignedEncodeMsg *)hCryptMsg;
1194 BOOL ret = FALSE;
1195
1196 switch (dwParamType)
1197 {
1198 case CMSG_CONTENT_PARAM:
1199 {
1200 CRYPT_CONTENT_INFO info;
1201
1202 ret = CryptMsgGetParam(hCryptMsg, CMSG_BARE_CONTENT_PARAM, 0, NULL,
1203 &info.Content.cbData);
1204 if (ret)
1205 {
1206 info.Content.pbData = CryptMemAlloc(info.Content.cbData);
1207 if (info.Content.pbData)
1208 {
1209 ret = CryptMsgGetParam(hCryptMsg, CMSG_BARE_CONTENT_PARAM, 0,
1210 info.Content.pbData, &info.Content.cbData);
1211 if (ret)
1212 {
1213 char oid_rsa_signed[] = szOID_RSA_signedData;
1214
1215 info.pszObjId = oid_rsa_signed;
1216 ret = CryptEncodeObjectEx(X509_ASN_ENCODING,
1217 PKCS_CONTENT_INFO, &info, 0, NULL, pvData, pcbData);
1218 }
1219 CryptMemFree(info.Content.pbData);
1220 }
1221 else
1222 ret = FALSE;
1223 }
1224 break;
1225 }
1226 case CMSG_BARE_CONTENT_PARAM:
1227 {
1228 CRYPT_SIGNED_INFO info;
1229 char oid_rsa_data[] = szOID_RSA_data;
1230
1231 info = *msg->msg_data.info;
1232 /* Quirk: OID is only encoded messages if an update has happened */
1233 if (msg->base.state != MsgStateInit)
1234 info.content.pszObjId = oid_rsa_data;
1235 else
1236 info.content.pszObjId = NULL;
1237 if (msg->data.cbData)
1238 {
1239 CRYPT_DATA_BLOB blob = { msg->data.cbData, msg->data.pbData };
1240
1241 ret = CryptEncodeObjectEx(X509_ASN_ENCODING, X509_OCTET_STRING,
1242 &blob, CRYPT_ENCODE_ALLOC_FLAG, NULL,
1243 &info.content.Content.pbData, &info.content.Content.cbData);
1244 }
1245 else
1246 {
1247 info.content.Content.cbData = 0;
1248 info.content.Content.pbData = NULL;
1249 ret = TRUE;
1250 }
1251 if (ret)
1252 {
1253 ret = CRYPT_AsnEncodeCMSSignedInfo(&info, pvData, pcbData);
1254 LocalFree(info.content.Content.pbData);
1255 }
1256 break;
1257 }
1258 case CMSG_COMPUTED_HASH_PARAM:
1259 if (dwIndex >= msg->msg_data.cSignerHandle)
1260 SetLastError(CRYPT_E_INVALID_INDEX);
1261 else
1262 ret = CryptGetHashParam(
1263 msg->msg_data.signerHandles[dwIndex].contentHash, HP_HASHVAL,
1264 pvData, pcbData, 0);
1265 break;
1266 case CMSG_ENCODED_SIGNER:
1267 if (dwIndex >= msg->msg_data.info->cSignerInfo)
1268 SetLastError(CRYPT_E_INVALID_INDEX);
1269 else
1270 ret = CryptEncodeObjectEx(X509_ASN_ENCODING | PKCS_7_ASN_ENCODING,
1271 CMS_SIGNER_INFO, &msg->msg_data.info->rgSignerInfo[dwIndex], 0,
1272 NULL, pvData, pcbData);
1273 break;
1274 case CMSG_VERSION_PARAM:
1275 ret = CRYPT_CopyParam(pvData, pcbData, &msg->msg_data.info->version,
1276 sizeof(msg->msg_data.info->version));
1277 break;
1278 default:
1279 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
1280 }
1281 return ret;
1282 }
1283
1284 static BOOL CSignedEncodeMsg_Update(HCRYPTMSG hCryptMsg, const BYTE *pbData,
1285 DWORD cbData, BOOL fFinal)
1286 {
1287 CSignedEncodeMsg *msg = (CSignedEncodeMsg *)hCryptMsg;
1288 BOOL ret = FALSE;
1289
1290 if (msg->base.state == MsgStateFinalized)
1291 SetLastError(CRYPT_E_MSG_ERROR);
1292 else if (msg->base.streamed || (msg->base.open_flags & CMSG_DETACHED_FLAG))
1293 {
1294 ret = CSignedMsgData_Update(&msg->msg_data, pbData, cbData, fFinal,
1295 Sign);
1296 if (msg->base.streamed)
1297 FIXME("streamed partial stub\n");
1298 msg->base.state = fFinal ? MsgStateFinalized : MsgStateUpdated;
1299 }
1300 else
1301 {
1302 if (!fFinal)
1303 SetLastError(CRYPT_E_MSG_ERROR);
1304 else
1305 {
1306 if (cbData)
1307 {
1308 msg->data.pbData = CryptMemAlloc(cbData);
1309 if (msg->data.pbData)
1310 {
1311 memcpy(msg->data.pbData, pbData, cbData);
1312 msg->data.cbData = cbData;
1313 ret = TRUE;
1314 }
1315 }
1316 else
1317 ret = TRUE;
1318 if (ret)
1319 ret = CSignedMsgData_Update(&msg->msg_data, pbData, cbData,
1320 fFinal, Sign);
1321 msg->base.state = MsgStateFinalized;
1322 }
1323 }
1324 return ret;
1325 }
1326
1327 static HCRYPTMSG CSignedEncodeMsg_Open(DWORD dwFlags,
1328 const void *pvMsgEncodeInfo, LPSTR pszInnerContentObjID,
1329 PCMSG_STREAM_INFO pStreamInfo)
1330 {
1331 const CMSG_SIGNED_ENCODE_INFO_WITH_CMS *info =
1332 (const CMSG_SIGNED_ENCODE_INFO_WITH_CMS *)pvMsgEncodeInfo;
1333 DWORD i;
1334 CSignedEncodeMsg *msg;
1335
1336 if (info->cbSize != sizeof(CMSG_SIGNED_ENCODE_INFO) &&
1337 info->cbSize != sizeof(CMSG_SIGNED_ENCODE_INFO_WITH_CMS))
1338 {
1339 SetLastError(E_INVALIDARG);
1340 return NULL;
1341 }
1342 if (info->cbSize == sizeof(CMSG_SIGNED_ENCODE_INFO_WITH_CMS) &&
1343 info->rgAttrCertEncoded)
1344 {
1345 FIXME("CMSG_SIGNED_ENCODE_INFO with CMS fields unsupported\n");
1346 return NULL;
1347 }
1348 for (i = 0; i < info->cSigners; i++)
1349 if (!CRYPT_IsValidSigner(&info->rgSigners[i]))
1350 return NULL;
1351 msg = CryptMemAlloc(sizeof(CSignedEncodeMsg));
1352 if (msg)
1353 {
1354 BOOL ret = TRUE;
1355
1356 CryptMsgBase_Init((CryptMsgBase *)msg, dwFlags, pStreamInfo,
1357 CSignedEncodeMsg_Close, CSignedEncodeMsg_GetParam,
1358 CSignedEncodeMsg_Update, CRYPT_DefaultMsgControl);
1359 msg->data.cbData = 0;
1360 msg->data.pbData = NULL;
1361 msg->msg_data.info = CryptMemAlloc(sizeof(CRYPT_SIGNED_INFO));
1362 if (msg->msg_data.info)
1363 {
1364 memset(msg->msg_data.info, 0, sizeof(CRYPT_SIGNED_INFO));
1365 msg->msg_data.info->version = CMSG_SIGNED_DATA_V1;
1366 }
1367 else
1368 ret = FALSE;
1369 if (ret)
1370 {
1371 if (info->cSigners)
1372 {
1373 msg->msg_data.info->rgSignerInfo =
1374 CryptMemAlloc(info->cSigners * sizeof(CMSG_CMS_SIGNER_INFO));
1375 if (msg->msg_data.info->rgSignerInfo)
1376 {
1377 msg->msg_data.info->cSignerInfo = info->cSigners;
1378 memset(msg->msg_data.info->rgSignerInfo, 0,
1379 msg->msg_data.info->cSignerInfo *
1380 sizeof(CMSG_CMS_SIGNER_INFO));
1381 ret = CSignedMsgData_AllocateHandles(&msg->msg_data);
1382 for (i = 0; ret && i < msg->msg_data.info->cSignerInfo; i++)
1383 {
1384 if (info->rgSigners[i].SignerId.dwIdChoice ==
1385 CERT_ID_KEY_IDENTIFIER)
1386 msg->msg_data.info->version = CMSG_SIGNED_DATA_V3;
1387 ret = CSignerInfo_Construct(
1388 &msg->msg_data.info->rgSignerInfo[i],
1389 &info->rgSigners[i]);
1390 if (ret)
1391 {
1392 ret = CSignedMsgData_ConstructSignerHandles(
1393 &msg->msg_data, i, info->rgSigners[i].hCryptProv);
1394 if (dwFlags & CMSG_CRYPT_RELEASE_CONTEXT_FLAG)
1395 CryptReleaseContext(info->rgSigners[i].hCryptProv,
1396 0);
1397 }
1398 }
1399 }
1400 else
1401 ret = FALSE;
1402 }
1403 else
1404 {
1405 msg->msg_data.info->cSignerInfo = 0;
1406 msg->msg_data.signerHandles = NULL;
1407 msg->msg_data.cSignerHandle = 0;
1408 }
1409 }
1410 if (ret)
1411 ret = CRYPT_ConstructBlobArray(
1412 (BlobArray *)&msg->msg_data.info->cCertEncoded,
1413 (const BlobArray *)&info->cCertEncoded);
1414 if (ret)
1415 ret = CRYPT_ConstructBlobArray(
1416 (BlobArray *)&msg->msg_data.info->cCrlEncoded,
1417 (const BlobArray *)&info->cCrlEncoded);
1418 if (!ret)
1419 {
1420 CSignedEncodeMsg_Close(msg);
1421 msg = NULL;
1422 }
1423 }
1424 return msg;
1425 }
1426
1427 static inline const char *MSG_TYPE_STR(DWORD type)
1428 {
1429 switch (type)
1430 {
1431 #define _x(x) case (x): return #x
1432 _x(CMSG_DATA);
1433 _x(CMSG_SIGNED);
1434 _x(CMSG_ENVELOPED);
1435 _x(CMSG_SIGNED_AND_ENVELOPED);
1436 _x(CMSG_HASHED);
1437 _x(CMSG_ENCRYPTED);
1438 #undef _x
1439 default:
1440 return wine_dbg_sprintf("unknown (%d)", type);
1441 }
1442 }
1443
1444 HCRYPTMSG WINAPI CryptMsgOpenToEncode(DWORD dwMsgEncodingType, DWORD dwFlags,
1445 DWORD dwMsgType, const void *pvMsgEncodeInfo, LPSTR pszInnerContentObjID,
1446 PCMSG_STREAM_INFO pStreamInfo)
1447 {
1448 HCRYPTMSG msg = NULL;
1449
1450 TRACE("(%08x, %08x, %08x, %p, %s, %p)\n", dwMsgEncodingType, dwFlags,
1451 dwMsgType, pvMsgEncodeInfo, debugstr_a(pszInnerContentObjID), pStreamInfo);
1452
1453 if (GET_CMSG_ENCODING_TYPE(dwMsgEncodingType) != PKCS_7_ASN_ENCODING)
1454 {
1455 SetLastError(E_INVALIDARG);
1456 return NULL;
1457 }
1458 switch (dwMsgType)
1459 {
1460 case CMSG_DATA:
1461 msg = CDataEncodeMsg_Open(dwFlags, pvMsgEncodeInfo,
1462 pszInnerContentObjID, pStreamInfo);
1463 break;
1464 case CMSG_HASHED:
1465 msg = CHashEncodeMsg_Open(dwFlags, pvMsgEncodeInfo,
1466 pszInnerContentObjID, pStreamInfo);
1467 break;
1468 case CMSG_SIGNED:
1469 msg = CSignedEncodeMsg_Open(dwFlags, pvMsgEncodeInfo,
1470 pszInnerContentObjID, pStreamInfo);
1471 break;
1472 case CMSG_ENVELOPED:
1473 FIXME("unimplemented for type %s\n", MSG_TYPE_STR(dwMsgType));
1474 break;
1475 case CMSG_SIGNED_AND_ENVELOPED:
1476 case CMSG_ENCRYPTED:
1477 /* defined but invalid, fall through */
1478 default:
1479 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
1480 }
1481 return msg;
1482 }
1483
1484 typedef struct _CDecodeMsg
1485 {
1486 CryptMsgBase base;
1487 DWORD type;
1488 HCRYPTPROV crypt_prov;
1489 union {
1490 HCRYPTHASH hash;
1491 CSignedMsgData signed_data;
1492 } u;
1493 CRYPT_DATA_BLOB msg_data;
1494 PCONTEXT_PROPERTY_LIST properties;
1495 } CDecodeMsg;
1496
1497 static void CDecodeMsg_Close(HCRYPTMSG hCryptMsg)
1498 {
1499 CDecodeMsg *msg = (CDecodeMsg *)hCryptMsg;
1500
1501 if (msg->base.open_flags & CMSG_CRYPT_RELEASE_CONTEXT_FLAG)
1502 CryptReleaseContext(msg->crypt_prov, 0);
1503 switch (msg->type)
1504 {
1505 case CMSG_HASHED:
1506 if (msg->u.hash)
1507 CryptDestroyHash(msg->u.hash);
1508 break;
1509 case CMSG_SIGNED:
1510 if (msg->u.signed_data.info)
1511 {
1512 LocalFree(msg->u.signed_data.info);
1513 CSignedMsgData_CloseHandles(&msg->u.signed_data);
1514 }
1515 break;
1516 }
1517 CryptMemFree(msg->msg_data.pbData);
1518 ContextPropertyList_Free(msg->properties);
1519 }
1520
1521 static BOOL CDecodeMsg_CopyData(CDecodeMsg *msg, const BYTE *pbData,
1522 DWORD cbData)
1523 {
1524 BOOL ret = TRUE;
1525
1526 if (cbData)
1527 {
1528 if (msg->msg_data.cbData)
1529 msg->msg_data.pbData = CryptMemRealloc(msg->msg_data.pbData,
1530 msg->msg_data.cbData + cbData);
1531 else
1532 msg->msg_data.pbData = CryptMemAlloc(cbData);
1533 if (msg->msg_data.pbData)
1534 {
1535 memcpy(msg->msg_data.pbData + msg->msg_data.cbData, pbData, cbData);
1536 msg->msg_data.cbData += cbData;
1537 }
1538 else
1539 ret = FALSE;
1540 }
1541 return ret;
1542 }
1543
1544 static BOOL CDecodeMsg_DecodeDataContent(CDecodeMsg *msg, CRYPT_DER_BLOB *blob)
1545 {
1546 BOOL ret;
1547 CRYPT_DATA_BLOB *data;
1548 DWORD size;
1549
1550 ret = CryptDecodeObjectEx(X509_ASN_ENCODING, X509_OCTET_STRING,
1551 blob->pbData, blob->cbData, CRYPT_DECODE_ALLOC_FLAG, NULL, (LPBYTE)&data,
1552 &size);
1553 if (ret)
1554 {
1555 ret = ContextPropertyList_SetProperty(msg->properties,
1556 CMSG_CONTENT_PARAM, data->pbData, data->cbData);
1557 LocalFree(data);
1558 }
1559 return ret;
1560 }
1561
1562 static void CDecodeMsg_SaveAlgorithmID(CDecodeMsg *msg, DWORD param,
1563 const CRYPT_ALGORITHM_IDENTIFIER *id)
1564 {
1565 static const BYTE nullParams[] = { ASN_NULL, 0 };
1566 CRYPT_ALGORITHM_IDENTIFIER *copy;
1567 DWORD len = sizeof(CRYPT_ALGORITHM_IDENTIFIER);
1568
1569 /* Linearize algorithm id */
1570 len += strlen(id->pszObjId) + 1;
1571 len += id->Parameters.cbData;
1572 copy = CryptMemAlloc(len);
1573 if (copy)
1574 {
1575 copy->pszObjId =
1576 (LPSTR)((BYTE *)copy + sizeof(CRYPT_ALGORITHM_IDENTIFIER));
1577 strcpy(copy->pszObjId, id->pszObjId);
1578 copy->Parameters.pbData = (BYTE *)copy->pszObjId + strlen(id->pszObjId)
1579 + 1;
1580 /* Trick: omit NULL parameters */
1581 if (id->Parameters.cbData == sizeof(nullParams) &&
1582 !memcmp(id->Parameters.pbData, nullParams, sizeof(nullParams)))
1583 {
1584 copy->Parameters.cbData = 0;
1585 len -= sizeof(nullParams);
1586 }
1587 else
1588 copy->Parameters.cbData = id->Parameters.cbData;
1589 if (copy->Parameters.cbData)
1590 memcpy(copy->Parameters.pbData, id->Parameters.pbData,
1591 id->Parameters.cbData);
1592 ContextPropertyList_SetProperty(msg->properties, param, (BYTE *)copy,
1593 len);
1594 CryptMemFree(copy);
1595 }
1596 }
1597
1598 static inline void CRYPT_FixUpAlgorithmID(CRYPT_ALGORITHM_IDENTIFIER *id)
1599 {
1600 id->pszObjId = (LPSTR)((BYTE *)id + sizeof(CRYPT_ALGORITHM_IDENTIFIER));
1601 id->Parameters.pbData = (BYTE *)id->pszObjId + strlen(id->pszObjId) + 1;
1602 }
1603
1604 static BOOL CDecodeMsg_DecodeHashedContent(CDecodeMsg *msg,
1605 CRYPT_DER_BLOB *blob)
1606 {
1607 BOOL ret;
1608 CRYPT_DIGESTED_DATA *digestedData;
1609 DWORD size;
1610
1611 ret = CRYPT_AsnDecodePKCSDigestedData(blob->pbData, blob->cbData,
1612 CRYPT_DECODE_ALLOC_FLAG, NULL, (CRYPT_DIGESTED_DATA *)&digestedData,
1613 &size);
1614 if (ret)
1615 {
1616 ContextPropertyList_SetProperty(msg->properties, CMSG_VERSION_PARAM,
1617 (const BYTE *)&digestedData->version, sizeof(digestedData->version));
1618 CDecodeMsg_SaveAlgorithmID(msg, CMSG_HASH_ALGORITHM_PARAM,
1619 &digestedData->DigestAlgorithm);
1620 ContextPropertyList_SetProperty(msg->properties,
1621 CMSG_INNER_CONTENT_TYPE_PARAM,
1622 (const BYTE *)digestedData->ContentInfo.pszObjId,
1623 digestedData->ContentInfo.pszObjId ?
1624 strlen(digestedData->ContentInfo.pszObjId) + 1 : 0);
1625 if (digestedData->ContentInfo.Content.cbData)
1626 CDecodeMsg_DecodeDataContent(msg,
1627 &digestedData->ContentInfo.Content);
1628 else
1629 ContextPropertyList_SetProperty(msg->properties,
1630 CMSG_CONTENT_PARAM, NULL, 0);
1631 ContextPropertyList_SetProperty(msg->properties, CMSG_HASH_DATA_PARAM,
1632 digestedData->hash.pbData, digestedData->hash.cbData);
1633 LocalFree(digestedData);
1634 }
1635 return ret;
1636 }
1637
1638 static BOOL CDecodeMsg_DecodeSignedContent(CDecodeMsg *msg,
1639 CRYPT_DER_BLOB *blob)
1640 {
1641 BOOL ret;
1642 CRYPT_SIGNED_INFO *signedInfo;
1643 DWORD size;
1644
1645 ret = CRYPT_AsnDecodeCMSSignedInfo(blob->pbData, blob->cbData,
1646 CRYPT_DECODE_ALLOC_FLAG, NULL, (CRYPT_SIGNED_INFO *)&signedInfo,
1647 &size);
1648 if (ret)
1649 {
1650 DWORD i;
1651
1652 msg->u.signed_data.info = signedInfo;
1653 ret = CSignedMsgData_AllocateHandles(&msg->u.signed_data);
1654 for (i = 0; ret && i < msg->u.signed_data.info->cSignerInfo; i++)
1655 ret = CSignedMsgData_ConstructSignerHandles(&msg->u.signed_data, i,
1656 msg->crypt_prov);
1657 if (ret)
1658 {
1659 /* Now that we have all the content, update the hash handles with
1660 * it. Have to decode it if the type is szOID_RSA_data.
1661 */
1662 if (msg->u.signed_data.info->content.Content.cbData)
1663 {
1664 if (!strcmp(msg->u.signed_data.info->content.pszObjId,
1665 szOID_RSA_data))
1666 {
1667 CRYPT_DATA_BLOB *blob;
1668
1669 ret = CryptDecodeObjectEx(X509_ASN_ENCODING,
1670 X509_OCTET_STRING,
1671 msg->u.signed_data.info->content.Content.pbData,
1672 msg->u.signed_data.info->content.Content.cbData,
1673 CRYPT_DECODE_ALLOC_FLAG, NULL, (LPBYTE)&blob, &size);
1674 if (ret)
1675 {
1676 ret = CSignedMsgData_Update(&msg->u.signed_data,
1677 blob->pbData, blob->cbData, TRUE, Verify);
1678 LocalFree(blob);
1679 }
1680 }
1681 else
1682 ret = CSignedMsgData_Update(&msg->u.signed_data,
1683 msg->u.signed_data.info->content.Content.pbData,
1684 msg->u.signed_data.info->content.Content.cbData, TRUE,
1685 Verify);
1686 }
1687 }
1688 }
1689 return ret;
1690 }
1691 /* Decodes the content in blob as the type given, and updates the value
1692 * (type, parameters, etc.) of msg based on what blob contains.
1693 * It doesn't just use msg's type, to allow a recursive call from an implicitly
1694 * typed message once the outer content info has been decoded.
1695 */
1696 static BOOL CDecodeMsg_DecodeContent(CDecodeMsg *msg, CRYPT_DER_BLOB *blob,
1697 DWORD type)
1698 {
1699 BOOL ret;
1700
1701 switch (type)
1702 {
1703 case CMSG_DATA:
1704 if ((ret = CDecodeMsg_DecodeDataContent(msg, blob)))
1705 msg->type = CMSG_DATA;
1706 break;
1707 case CMSG_HASHED:
1708 if ((ret = CDecodeMsg_DecodeHashedContent(msg, blob)))
1709 msg->type = CMSG_HASHED;
1710 break;
1711 case CMSG_ENVELOPED:
1712 FIXME("unimplemented for type %s\n", MSG_TYPE_STR(type));
1713 ret = TRUE;
1714 break;
1715 case CMSG_SIGNED:
1716 if ((ret = CDecodeMsg_DecodeSignedContent(msg, blob)))
1717 msg->type = CMSG_SIGNED;
1718 break;
1719 default:
1720 {
1721 CRYPT_CONTENT_INFO *info;
1722 DWORD size;
1723
1724 ret = CryptDecodeObjectEx(X509_ASN_ENCODING, PKCS_CONTENT_INFO,
1725 msg->msg_data.pbData, msg->msg_data.cbData, CRYPT_DECODE_ALLOC_FLAG,
1726 NULL, (LPBYTE)&info, &size);
1727 if (ret)
1728 {
1729 if (!strcmp(info->pszObjId, szOID_RSA_data))
1730 ret = CDecodeMsg_DecodeContent(msg, &info->Content, CMSG_DATA);
1731 else if (!strcmp(info->pszObjId, szOID_RSA_digestedData))
1732 ret = CDecodeMsg_DecodeContent(msg, &info->Content,
1733 CMSG_HASHED);
1734 else if (!strcmp(info->pszObjId, szOID_RSA_envelopedData))
1735 ret = CDecodeMsg_DecodeContent(msg, &info->Content,
1736 CMSG_ENVELOPED);
1737 else if (!strcmp(info->pszObjId, szOID_RSA_signedData))
1738 ret = CDecodeMsg_DecodeContent(msg, &info->Content,
1739 CMSG_SIGNED);
1740 else
1741 {
1742 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
1743 ret = FALSE;
1744 }
1745 LocalFree(info);
1746 }
1747 }
1748 }
1749 return ret;
1750 }
1751
1752 static BOOL CDecodeMsg_Update(HCRYPTMSG hCryptMsg, const BYTE *pbData,
1753 DWORD cbData, BOOL fFinal)
1754 {
1755 CDecodeMsg *msg = (CDecodeMsg *)hCryptMsg;
1756 BOOL ret = FALSE;
1757
1758 TRACE("(%p, %p, %d, %d)\n", hCryptMsg, pbData, cbData, fFinal);
1759
1760 if (msg->base.state == MsgStateFinalized)
1761 SetLastError(CRYPT_E_MSG_ERROR);
1762 else if (msg->base.streamed)
1763 {
1764 FIXME("(%p, %p, %d, %d): streamed update stub\n", hCryptMsg, pbData,
1765 cbData, fFinal);
1766 if (fFinal)
1767 {
1768 if (msg->base.open_flags & CMSG_DETACHED_FLAG &&
1769 msg->base.state != MsgStateDataFinalized)
1770 {
1771 ret = CDecodeMsg_CopyData(msg, pbData, cbData);
1772 msg->base.state = MsgStateDataFinalized;
1773 if (ret)
1774 ret = CDecodeMsg_DecodeContent(msg, &msg->msg_data,
1775 msg->type);
1776 }
1777 else
1778 {
1779 FIXME("(%p, %p, %d, %d): detached update stub\n", hCryptMsg,
1780 pbData, cbData, fFinal);
1781 ret = TRUE;
1782 msg->base.state = MsgStateFinalized;
1783 }
1784 }
1785 else
1786 {
1787 ret = CDecodeMsg_CopyData(msg, pbData, cbData);
1788 if (msg->base.state == MsgStateInit)
1789 msg->base.state = MsgStateUpdated;
1790 }
1791 }
1792 else
1793 {
1794 if (!fFinal)
1795 SetLastError(CRYPT_E_MSG_ERROR);
1796 else
1797 {
1798 if (msg->base.state == MsgStateInit)
1799 {
1800 ret = CDecodeMsg_CopyData(msg, pbData, cbData);
1801 if (ret)
1802 ret = CDecodeMsg_DecodeContent(msg, &msg->msg_data,
1803 msg->type);
1804 if (msg->base.open_flags & CMSG_DETACHED_FLAG)
1805 msg->base.state = MsgStateDataFinalized;
1806 else
1807 msg->base.state = MsgStateFinalized;
1808 }
1809 else if (msg->base.state == MsgStateDataFinalized)
1810 {
1811 FIXME("(%p, %p, %d, %d): detached update stub\n", hCryptMsg,
1812 pbData, cbData, fFinal);
1813 ret = TRUE;
1814 msg->base.state = MsgStateFinalized;
1815 }
1816 }
1817 }
1818 return ret;
1819 }
1820
1821 static BOOL CDecodeHashMsg_GetParam(CDecodeMsg *msg, DWORD dwParamType,
1822 DWORD dwIndex, void *pvData, DWORD *pcbData)
1823 {
1824 BOOL ret = FALSE;
1825
1826 switch (dwParamType)
1827 {
1828 case CMSG_TYPE_PARAM:
1829 ret = CRYPT_CopyParam(pvData, pcbData, &msg->type, sizeof(msg->type));
1830 break;
1831 case CMSG_HASH_ALGORITHM_PARAM:
1832 {
1833 CRYPT_DATA_BLOB blob;
1834
1835 ret = ContextPropertyList_FindProperty(msg->properties, dwParamType,
1836 &blob);
1837 if (ret)
1838 {
1839 ret = CRYPT_CopyParam(pvData, pcbData, blob.pbData, blob.cbData);
1840 if (ret && pvData)
1841 CRYPT_FixUpAlgorithmID((CRYPT_ALGORITHM_IDENTIFIER *)pvData);
1842 }
1843 else
1844 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
1845 break;
1846 }
1847 case CMSG_COMPUTED_HASH_PARAM:
1848 if (!msg->u.hash)
1849 {
1850 CRYPT_ALGORITHM_IDENTIFIER *hashAlgoID = NULL;
1851 DWORD size = 0;
1852 ALG_ID algID = 0;
1853
1854 CryptMsgGetParam(msg, CMSG_HASH_ALGORITHM_PARAM, 0, NULL, &size);
1855 hashAlgoID = CryptMemAlloc(size);
1856 ret = CryptMsgGetParam(msg, CMSG_HASH_ALGORITHM_PARAM, 0,
1857 hashAlgoID, &size);
1858 if (ret)
1859 algID = CertOIDToAlgId(hashAlgoID->pszObjId);
1860 ret = CryptCreateHash(msg->crypt_prov, algID, 0, 0, &msg->u.hash);
1861 if (ret)
1862 {
1863 CRYPT_DATA_BLOB content;
1864
1865 ret = ContextPropertyList_FindProperty(msg->properties,
1866 CMSG_CONTENT_PARAM, &content);
1867 if (ret)
1868 ret = CryptHashData(msg->u.hash, content.pbData,
1869 content.cbData, 0);
1870 }
1871 CryptMemFree(hashAlgoID);
1872 }
1873 else
1874 ret = TRUE;
1875 if (ret)
1876 ret = CryptGetHashParam(msg->u.hash, HP_HASHVAL, pvData, pcbData,
1877 0);
1878 break;
1879 default:
1880 {
1881 CRYPT_DATA_BLOB blob;
1882
1883 ret = ContextPropertyList_FindProperty(msg->properties, dwParamType,
1884 &blob);
1885 if (ret)
1886 ret = CRYPT_CopyParam(pvData, pcbData, blob.pbData, blob.cbData);
1887 else
1888 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
1889 }
1890 }
1891 return ret;
1892 }
1893
1894 /* nextData is an in/out parameter - on input it's the memory location in
1895 * which a copy of in's data should be made, and on output it's the memory
1896 * location immediately after out's copy of in's data.
1897 */
1898 static inline void CRYPT_CopyBlob(CRYPT_DATA_BLOB *out,
1899 const CRYPT_DATA_BLOB *in, LPBYTE *nextData)
1900 {
1901 out->cbData = in->cbData;
1902 if (in->cbData)
1903 {
1904 out->pbData = *nextData;
1905 memcpy(out->pbData, in->pbData, in->cbData);
1906 *nextData += in->cbData;
1907 }
1908 }
1909
1910 static inline void CRYPT_CopyAlgorithmId(CRYPT_ALGORITHM_IDENTIFIER *out,
1911 const CRYPT_ALGORITHM_IDENTIFIER *in, LPBYTE *nextData)
1912 {
1913 if (in->pszObjId)
1914 {
1915 out->pszObjId = (LPSTR)*nextData;
1916 strcpy(out->pszObjId, in->pszObjId);
1917 *nextData += strlen(out->pszObjId) + 1;
1918 }
1919 CRYPT_CopyBlob(&out->Parameters, &in->Parameters, nextData);
1920 }
1921
1922 static inline void CRYPT_CopyAttributes(CRYPT_ATTRIBUTES *out,
1923 const CRYPT_ATTRIBUTES *in, LPBYTE *nextData)
1924 {
1925 out->cAttr = in->cAttr;
1926 if (in->cAttr)
1927 {
1928 DWORD i;
1929
1930 if ((*nextData - (LPBYTE)0) % sizeof(DWORD_PTR))
1931 *nextData += (*nextData - (LPBYTE)0) % sizeof(DWORD_PTR);
1932 out->rgAttr = (CRYPT_ATTRIBUTE *)*nextData;
1933 *nextData += in->cAttr * sizeof(CRYPT_ATTRIBUTE);
1934 for (i = 0; i < in->cAttr; i++)
1935 {
1936 if (in->rgAttr[i].pszObjId)
1937 {
1938 out->rgAttr[i].pszObjId = (LPSTR)*nextData;
1939 strcpy(out->rgAttr[i].pszObjId, in->rgAttr[i].pszObjId);
1940 *nextData += strlen(in->rgAttr[i].pszObjId) + 1;
1941 }
1942 if (in->rgAttr[i].cValue)
1943 {
1944 DWORD j;
1945
1946 out->rgAttr[i].cValue = in->rgAttr[i].cValue;
1947 if ((*nextData - (LPBYTE)0) % sizeof(DWORD_PTR))
1948 *nextData += (*nextData - (LPBYTE)0) % sizeof(DWORD_PTR);
1949 out->rgAttr[i].rgValue = (PCRYPT_DATA_BLOB)*nextData;
1950 *nextData += in->rgAttr[i].cValue * sizeof(CRYPT_DATA_BLOB);
1951 for (j = 0; j < in->rgAttr[i].cValue; j++)
1952 CRYPT_CopyBlob(&out->rgAttr[i].rgValue[j],
1953 &in->rgAttr[i].rgValue[j], nextData);
1954 }
1955 }
1956 }
1957 }
1958
1959 static DWORD CRYPT_SizeOfAttributes(const CRYPT_ATTRIBUTES *attr)
1960 {
1961 DWORD size = attr->cAttr * sizeof(CRYPT_ATTRIBUTE), i, j;
1962
1963 for (i = 0; i < attr->cAttr; i++)
1964 {
1965 if (attr->rgAttr[i].pszObjId)
1966 size += strlen(attr->rgAttr[i].pszObjId) + 1;
1967 /* align pointer */
1968 if (size % sizeof(DWORD_PTR))
1969 size += size % sizeof(DWORD_PTR);
1970 size += attr->rgAttr[i].cValue * sizeof(CRYPT_DATA_BLOB);
1971 for (j = 0; j < attr->rgAttr[i].cValue; j++)
1972 size += attr->rgAttr[i].rgValue[j].cbData;
1973 }
1974 /* align pointer again to be conservative */
1975 if (size % sizeof(DWORD_PTR))
1976 size += size % sizeof(DWORD_PTR);
1977 return size;
1978 }
1979
1980 static DWORD CRYPT_SizeOfKeyIdAsIssuerAndSerial(const CRYPT_DATA_BLOB *keyId)
1981 {
1982 static char oid_key_rdn[] = szOID_KEYID_RDN;
1983 DWORD size = 0;
1984 CERT_RDN_ATTR attr;
1985 CERT_RDN rdn = { 1, &attr };
1986 CERT_NAME_INFO name = { 1, &rdn };
1987
1988 attr.pszObjId = oid_key_rdn;
1989 attr.dwValueType = CERT_RDN_OCTET_STRING;
1990 attr.Value.cbData = keyId->cbData;
1991 attr.Value.pbData = keyId->pbData;
1992 if (CryptEncodeObject(X509_ASN_ENCODING, X509_NAME, &name, NULL, &size))
1993 size++; /* Only include size of special zero serial number on success */
1994 return size;
1995 }
1996
1997 static BOOL CRYPT_CopyKeyIdAsIssuerAndSerial(CERT_NAME_BLOB *issuer,
1998 CRYPT_INTEGER_BLOB *serialNumber, const CRYPT_DATA_BLOB *keyId, DWORD encodedLen,
1999 LPBYTE *nextData)
2000 {
2001 static char oid_key_rdn[] = szOID_KEYID_RDN;
2002 CERT_RDN_ATTR attr;
2003 CERT_RDN rdn = { 1, &attr };
2004 CERT_NAME_INFO name = { 1, &rdn };
2005 BOOL ret;
2006
2007 /* Encode special zero serial number */
2008 serialNumber->cbData = 1;
2009 serialNumber->pbData = *nextData;
2010 **nextData = 0;
2011 (*nextData)++;
2012 /* Encode issuer */
2013 issuer->pbData = *nextData;
2014 attr.pszObjId = oid_key_rdn;
2015 attr.dwValueType = CERT_RDN_OCTET_STRING;
2016 attr.Value.cbData = keyId->cbData;
2017 attr.Value.pbData = keyId->pbData;
2018 ret = CryptEncodeObject(X509_ASN_ENCODING, X509_NAME, &name, *nextData,
2019 &encodedLen);
2020 if (ret)
2021 {
2022 *nextData += encodedLen;
2023 issuer->cbData = encodedLen;
2024 }
2025 return ret;
2026 }
2027
2028 static BOOL CRYPT_CopySignerInfo(void *pvData, DWORD *pcbData,
2029 const CMSG_CMS_SIGNER_INFO *in)
2030 {
2031 DWORD size = sizeof(CMSG_SIGNER_INFO), rdnSize;
2032 BOOL ret;
2033
2034 TRACE("(%p, %d, %p)\n", pvData, pvData ? *pcbData : 0, in);
2035
2036 if (in->SignerId.dwIdChoice == CERT_ID_ISSUER_SERIAL_NUMBER)
2037 {
2038 size += in->SignerId.IssuerSerialNumber.Issuer.cbData;
2039 size += in->SignerId.IssuerSerialNumber.SerialNumber.cbData;
2040 }
2041 else
2042 {
2043 rdnSize = CRYPT_SizeOfKeyIdAsIssuerAndSerial(&in->SignerId.KeyId);
2044 size += rdnSize;
2045 }
2046 if (in->HashAlgorithm.pszObjId)
2047 size += strlen(in->HashAlgorithm.pszObjId) + 1;
2048 size += in->HashAlgorithm.Parameters.cbData;
2049 if (in->HashEncryptionAlgorithm.pszObjId)
2050 size += strlen(in->HashEncryptionAlgorithm.pszObjId) + 1;
2051 size += in->HashEncryptionAlgorithm.Parameters.cbData;
2052 size += in->EncryptedHash.cbData;
2053 /* align pointer */
2054 if (size % sizeof(DWORD_PTR))
2055 size += size % sizeof(DWORD_PTR);
2056 size += CRYPT_SizeOfAttributes(&in->AuthAttrs);
2057 size += CRYPT_SizeOfAttributes(&in->UnauthAttrs);
2058 if (!pvData)
2059 {
2060 *pcbData = size;
2061 ret = TRUE;
2062 }
2063 else if (*pcbData < size)
2064 {
2065 *pcbData = size;
2066 SetLastError(ERROR_MORE_DATA);
2067 ret = FALSE;
2068 }
2069 else
2070 {
2071 LPBYTE nextData = (BYTE *)pvData + sizeof(CMSG_SIGNER_INFO);
2072 CMSG_SIGNER_INFO *out = (CMSG_SIGNER_INFO *)pvData;
2073
2074 ret = TRUE;
2075 out->dwVersion = in->dwVersion;
2076 if (in->SignerId.dwIdChoice == CERT_ID_ISSUER_SERIAL_NUMBER)
2077 {
2078 CRYPT_CopyBlob(&out->Issuer,
2079 &in->SignerId.IssuerSerialNumber.Issuer, &nextData);
2080 CRYPT_CopyBlob(&out->SerialNumber,
2081 &in->SignerId.IssuerSerialNumber.SerialNumber, &nextData);
2082 }
2083 else
2084 ret = CRYPT_CopyKeyIdAsIssuerAndSerial(&out->Issuer, &out->SerialNumber,
2085 &in->SignerId.KeyId, rdnSize, &nextData);
2086 if (ret)
2087 {
2088 CRYPT_CopyAlgorithmId(&out->HashAlgorithm, &in->HashAlgorithm,
2089 &nextData);
2090 CRYPT_CopyAlgorithmId(&out->HashEncryptionAlgorithm,
2091 &in->HashEncryptionAlgorithm, &nextData);
2092 CRYPT_CopyBlob(&out->EncryptedHash, &in->EncryptedHash, &nextData);
2093 /* align pointer */
2094 if ((nextData - (LPBYTE)0) % sizeof(DWORD_PTR))
2095 nextData += (nextData - (LPBYTE)0) % sizeof(DWORD_PTR);
2096 CRYPT_CopyAttributes(&out->AuthAttrs, &in->AuthAttrs, &nextData);
2097 CRYPT_CopyAttributes(&out->UnauthAttrs, &in->UnauthAttrs, &nextData);
2098 }
2099 }
2100 TRACE("returning %d\n", ret);
2101 return ret;
2102 }
2103
2104 static BOOL CRYPT_CopyCMSSignerInfo(void *pvData, DWORD *pcbData,
2105 const CMSG_CMS_SIGNER_INFO *in)
2106 {
2107 DWORD size = sizeof(CMSG_CMS_SIGNER_INFO);
2108 BOOL ret;
2109
2110 TRACE("(%p, %d, %p)\n", pvData, pvData ? *pcbData : 0, in);
2111
2112 if (in->SignerId.dwIdChoice == CERT_ID_ISSUER_SERIAL_NUMBER)
2113 {
2114 size += in->SignerId.IssuerSerialNumber.Issuer.cbData;
2115 size += in->SignerId.IssuerSerialNumber.SerialNumber.cbData;
2116 }
2117 else
2118 size += in->SignerId.KeyId.cbData;
2119 if (in->HashAlgorithm.pszObjId)
2120 size += strlen(in->HashAlgorithm.pszObjId) + 1;
2121 size += in->HashAlgorithm.Parameters.cbData;
2122 if (in->HashEncryptionAlgorithm.pszObjId)
2123 size += strlen(in->HashEncryptionAlgorithm.pszObjId) + 1;
2124 size += in->HashEncryptionAlgorithm.Parameters.cbData;
2125 size += in->EncryptedHash.cbData;
2126 /* align pointer */
2127 if (size % sizeof(DWORD_PTR))
2128 size += size % sizeof(DWORD_PTR);
2129 size += CRYPT_SizeOfAttributes(&in->AuthAttrs);
2130 size += CRYPT_SizeOfAttributes(&in->UnauthAttrs);
2131 if (!pvData)
2132 {
2133 *pcbData = size;
2134 ret = TRUE;
2135 }
2136 else if (*pcbData < size)
2137 {
2138 *pcbData = size;
2139 SetLastError(ERROR_MORE_DATA);
2140 ret = FALSE;
2141 }
2142 else
2143 {
2144 LPBYTE nextData = (BYTE *)pvData + sizeof(CMSG_CMS_SIGNER_INFO);
2145 CMSG_CMS_SIGNER_INFO *out = (CMSG_CMS_SIGNER_INFO *)pvData;
2146
2147 out->dwVersion = in->dwVersion;
2148 out->SignerId.dwIdChoice = in->SignerId.dwIdChoice;
2149 if (in->SignerId.dwIdChoice == CERT_ID_ISSUER_SERIAL_NUMBER)
2150 {
2151 CRYPT_CopyBlob(&out->SignerId.IssuerSerialNumber.Issuer,
2152 &in->SignerId.IssuerSerialNumber.Issuer, &nextData);
2153 CRYPT_CopyBlob(&out->SignerId.IssuerSerialNumber.SerialNumber,
2154 &in->SignerId.IssuerSerialNumber.SerialNumber, &nextData);
2155 }
2156 else
2157 CRYPT_CopyBlob(&out->SignerId.KeyId, &in->SignerId.KeyId, &nextData);
2158 CRYPT_CopyAlgorithmId(&out->HashAlgorithm, &in->HashAlgorithm,
2159 &nextData);
2160 CRYPT_CopyAlgorithmId(&out->HashEncryptionAlgorithm,
2161 &in->HashEncryptionAlgorithm, &nextData);
2162 CRYPT_CopyBlob(&out->EncryptedHash, &in->EncryptedHash, &nextData);
2163 /* align pointer */
2164 if ((nextData - (LPBYTE)0) % sizeof(DWORD_PTR))
2165 nextData += (nextData - (LPBYTE)0) % sizeof(DWORD_PTR);
2166 CRYPT_CopyAttributes(&out->AuthAttrs, &in->AuthAttrs, &nextData);
2167 CRYPT_CopyAttributes(&out->UnauthAttrs, &in->UnauthAttrs, &nextData);
2168 ret = TRUE;
2169 }
2170 TRACE("returning %d\n", ret);
2171 return ret;
2172 }
2173
2174 static BOOL CRYPT_CopySignerCertInfo(void *pvData, DWORD *pcbData,
2175 const CMSG_CMS_SIGNER_INFO *in)
2176 {
2177 DWORD size = sizeof(CERT_INFO);
2178 BOOL ret;
2179
2180 TRACE("(%p, %d, %p)\n", pvData, pvData ? *pcbData : 0, in);
2181
2182 if (in->SignerId.dwIdChoice == CERT_ID_ISSUER_SERIAL_NUMBER)
2183 {
2184 size += in->SignerId.IssuerSerialNumber.Issuer.cbData;
2185 size += in->SignerId.IssuerSerialNumber.SerialNumber.cbData;
2186 }
2187 else
2188 {
2189 FIXME("unimplemented for key id\n");
2190 return FALSE;
2191 }
2192 if (!pvData)
2193 {
2194 *pcbData = size;
2195 ret = TRUE;
2196 }
2197 else if (*pcbData < size)
2198 {
2199 *pcbData = size;
2200 SetLastError(ERROR_MORE_DATA);
2201 ret = FALSE;
2202 }
2203 else
2204 {
2205 LPBYTE nextData = (BYTE *)pvData + sizeof(CERT_INFO);
2206 CERT_INFO *out = (CERT_INFO *)pvData;
2207
2208 memset(out, 0, sizeof(CERT_INFO));
2209 CRYPT_CopyBlob(&out->Issuer,
2210 &in->SignerId.IssuerSerialNumber.Issuer, &nextData);
2211 CRYPT_CopyBlob(&out->SerialNumber,
2212 &in->SignerId.IssuerSerialNumber.SerialNumber, &nextData);
2213 ret = TRUE;
2214 }
2215 TRACE("returning %d\n", ret);
2216 return ret;
2217 }
2218
2219 static BOOL CDecodeSignedMsg_GetParam(CDecodeMsg *msg, DWORD dwParamType,
2220 DWORD dwIndex, void *pvData, DWORD *pcbData)
2221 {
2222 BOOL ret = FALSE;
2223
2224 switch (dwParamType)
2225 {
2226 case CMSG_TYPE_PARAM:
2227 ret = CRYPT_CopyParam(pvData, pcbData, &msg->type, sizeof(msg->type));
2228 break;
2229 case CMSG_CONTENT_PARAM:
2230 if (msg->u.signed_data.info)
2231 {
2232 if (!strcmp(msg->u.signed_data.info->content.pszObjId,
2233 szOID_RSA_data))
2234 {
2235 CRYPT_DATA_BLOB *blob;
2236 DWORD size;
2237
2238 ret = CryptDecodeObjectEx(X509_ASN_ENCODING, X509_OCTET_STRING,
2239 msg->u.signed_data.info->content.Content.pbData,
2240 msg->u.signed_data.info->content.Content.cbData,
2241 CRYPT_DECODE_ALLOC_FLAG, NULL, (LPBYTE)&blob, &size);
2242 if (ret)
2243 {
2244 ret = CRYPT_CopyParam(pvData, pcbData, blob->pbData,
2245 blob->cbData);
2246 LocalFree(blob);
2247 }
2248 }
2249 else
2250 ret = CRYPT_CopyParam(pvData, pcbData,
2251 msg->u.signed_data.info->content.Content.pbData,
2252 msg->u.signed_data.info->content.Content.cbData);
2253 }
2254 else
2255 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2256 break;
2257 case CMSG_INNER_CONTENT_TYPE_PARAM:
2258 if (msg->u.signed_data.info)
2259 ret = CRYPT_CopyParam(pvData, pcbData,
2260 msg->u.signed_data.info->content.pszObjId,
2261 strlen(msg->u.signed_data.info->content.pszObjId) + 1);
2262 else
2263 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2264 break;
2265 case CMSG_SIGNER_COUNT_PARAM:
2266 if (msg->u.signed_data.info)
2267 ret = CRYPT_CopyParam(pvData, pcbData,
2268 &msg->u.signed_data.info->cSignerInfo, sizeof(DWORD));
2269 else
2270 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2271 break;
2272 case CMSG_SIGNER_INFO_PARAM:
2273 if (msg->u.signed_data.info)
2274 {
2275 if (dwIndex >= msg->u.signed_data.info->cSignerInfo)
2276 SetLastError(CRYPT_E_INVALID_INDEX);
2277 else
2278 ret = CRYPT_CopySignerInfo(pvData, pcbData,
2279 &msg->u.signed_data.info->rgSignerInfo[dwIndex]);
2280 }
2281 else
2282 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2283 break;
2284 case CMSG_SIGNER_CERT_INFO_PARAM:
2285 if (msg->u.signed_data.info)
2286 {
2287 if (dwIndex >= msg->u.signed_data.info->cSignerInfo)
2288 SetLastError(CRYPT_E_INVALID_INDEX);
2289 else
2290 ret = CRYPT_CopySignerCertInfo(pvData, pcbData,
2291 &msg->u.signed_data.info->rgSignerInfo[dwIndex]);
2292 }
2293 else
2294 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2295 break;
2296 case CMSG_CERT_COUNT_PARAM:
2297 if (msg->u.signed_data.info)
2298 ret = CRYPT_CopyParam(pvData, pcbData,
2299 &msg->u.signed_data.info->cCertEncoded, sizeof(DWORD));
2300 else
2301 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2302 break;
2303 case CMSG_CERT_PARAM:
2304 if (msg->u.signed_data.info)
2305 {
2306 if (dwIndex >= msg->u.signed_data.info->cCertEncoded)
2307 SetLastError(CRYPT_E_INVALID_INDEX);
2308 else
2309 ret = CRYPT_CopyParam(pvData, pcbData,
2310 msg->u.signed_data.info->rgCertEncoded[dwIndex].pbData,
2311 msg->u.signed_data.info->rgCertEncoded[dwIndex].cbData);
2312 }
2313 else
2314 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2315 break;
2316 case CMSG_CRL_COUNT_PARAM:
2317 if (msg->u.signed_data.info)
2318 ret = CRYPT_CopyParam(pvData, pcbData,
2319 &msg->u.signed_data.info->cCrlEncoded, sizeof(DWORD));
2320 else
2321 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2322 break;
2323 case CMSG_CRL_PARAM:
2324 if (msg->u.signed_data.info)
2325 {
2326 if (dwIndex >= msg->u.signed_data.info->cCrlEncoded)
2327 SetLastError(CRYPT_E_INVALID_INDEX);
2328 else
2329 ret = CRYPT_CopyParam(pvData, pcbData,
2330 msg->u.signed_data.info->rgCrlEncoded[dwIndex].pbData,
2331 msg->u.signed_data.info->rgCrlEncoded[dwIndex].cbData);
2332 }
2333 else
2334 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2335 break;
2336 case CMSG_COMPUTED_HASH_PARAM:
2337 if (msg->u.signed_data.info)
2338 {
2339 if (dwIndex >= msg->u.signed_data.cSignerHandle)
2340 SetLastError(CRYPT_E_INVALID_INDEX);
2341 else
2342 ret = CryptGetHashParam(
2343 msg->u.signed_data.signerHandles[dwIndex].contentHash,
2344 HP_HASHVAL, pvData, pcbData, 0);
2345 }
2346 else
2347 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2348 break;
2349 case CMSG_ATTR_CERT_COUNT_PARAM:
2350 if (msg->u.signed_data.info)
2351 {
2352 DWORD attrCertCount = 0;
2353
2354 ret = CRYPT_CopyParam(pvData, pcbData,
2355 &attrCertCount, sizeof(DWORD));
2356 }
2357 else
2358 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2359 break;
2360 case CMSG_ATTR_CERT_PARAM:
2361 if (msg->u.signed_data.info)
2362 SetLastError(CRYPT_E_INVALID_INDEX);
2363 else
2364 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2365 break;
2366 case CMSG_CMS_SIGNER_INFO_PARAM:
2367 if (msg->u.signed_data.info)
2368 {
2369 if (dwIndex >= msg->u.signed_data.info->cSignerInfo)
2370 SetLastError(CRYPT_E_INVALID_INDEX);
2371 else
2372 ret = CRYPT_CopyCMSSignerInfo(pvData, pcbData,
2373 &msg->u.signed_data.info->rgSignerInfo[dwIndex]);
2374 }
2375 else
2376 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2377 break;
2378 default:
2379 FIXME("unimplemented for %d\n", dwParamType);
2380 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2381 }
2382 return ret;
2383 }
2384
2385 static BOOL CDecodeMsg_GetParam(HCRYPTMSG hCryptMsg, DWORD dwParamType,
2386 DWORD dwIndex, void *pvData, DWORD *pcbData)
2387 {
2388 CDecodeMsg *msg = (CDecodeMsg *)hCryptMsg;
2389 BOOL ret = FALSE;
2390
2391 switch (msg->type)
2392 {
2393 case CMSG_HASHED:
2394 ret = CDecodeHashMsg_GetParam(msg, dwParamType, dwIndex, pvData,
2395 pcbData);
2396 break;
2397 case CMSG_SIGNED:
2398 ret = CDecodeSignedMsg_GetParam(msg, dwParamType, dwIndex, pvData,
2399 pcbData);
2400 break;
2401 default:
2402 switch (dwParamType)
2403 {
2404 case CMSG_TYPE_PARAM:
2405 ret = CRYPT_CopyParam(pvData, pcbData, &msg->type,
2406 sizeof(msg->type));
2407 break;
2408 default:
2409 {
2410 CRYPT_DATA_BLOB blob;
2411
2412 ret = ContextPropertyList_FindProperty(msg->properties, dwParamType,
2413 &blob);
2414 if (ret)
2415 ret = CRYPT_CopyParam(pvData, pcbData, blob.pbData,
2416 blob.cbData);
2417 else
2418 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2419 }
2420 }
2421 }
2422 return ret;
2423 }
2424
2425 static BOOL CDecodeHashMsg_VerifyHash(CDecodeMsg *msg)
2426 {
2427 BOOL ret;
2428 CRYPT_DATA_BLOB hashBlob;
2429
2430 ret = ContextPropertyList_FindProperty(msg->properties,
2431 CMSG_HASH_DATA_PARAM, &hashBlob);
2432 if (ret)
2433 {
2434 DWORD computedHashSize = 0;
2435
2436 ret = CDecodeHashMsg_GetParam(msg, CMSG_COMPUTED_HASH_PARAM, 0, NULL,
2437 &computedHashSize);
2438 if (hashBlob.cbData == computedHashSize)
2439 {
2440 LPBYTE computedHash = CryptMemAlloc(computedHashSize);
2441
2442 if (computedHash)
2443 {
2444 ret = CDecodeHashMsg_GetParam(msg, CMSG_COMPUTED_HASH_PARAM, 0,
2445 computedHash, &computedHashSize);
2446 if (ret)
2447 ret = !memcmp(hashBlob.pbData, computedHash,
2448 hashBlob.cbData);
2449 CryptMemFree(computedHash);
2450 }
2451 else
2452 ret = FALSE;
2453 }
2454 }
2455 return ret;
2456 }
2457
2458 static BOOL CDecodeSignedMsg_VerifySignatureWithKey(CDecodeMsg *msg,
2459 HCRYPTPROV prov, DWORD signerIndex, PCERT_PUBLIC_KEY_INFO keyInfo)
2460 {
2461 HCRYPTKEY key;
2462 BOOL ret;
2463
2464 if (!prov)
2465 prov = msg->crypt_prov;
2466 ret = CryptImportPublicKeyInfo(prov, X509_ASN_ENCODING, keyInfo, &key);
2467 if (ret)
2468 {
2469 HCRYPTHASH hash;
2470 CRYPT_HASH_BLOB reversedHash;
2471
2472 if (msg->u.signed_data.info->rgSignerInfo[signerIndex].AuthAttrs.cAttr)
2473 hash = msg->u.signed_data.signerHandles[signerIndex].authAttrHash;
2474 else
2475 hash = msg->u.signed_data.signerHandles[signerIndex].contentHash;
2476 ret = CRYPT_ConstructBlob(&reversedHash,
2477 &msg->u.signed_data.info->rgSignerInfo[signerIndex].EncryptedHash);
2478 if (ret)
2479 {
2480 CRYPT_ReverseBytes(&reversedHash);
2481 ret = CryptVerifySignatureW(hash, reversedHash.pbData,
2482 reversedHash.cbData, key, NULL, 0);
2483 CryptMemFree(reversedHash.pbData);
2484 }
2485 CryptDestroyKey(key);
2486 }
2487 return ret;
2488 }
2489
2490 static BOOL CDecodeSignedMsg_VerifySignature(CDecodeMsg *msg, PCERT_INFO info)
2491 {
2492 BOOL ret = FALSE;
2493 DWORD i;
2494
2495 for (i = 0; !ret && i < msg->u.signed_data.info->cSignerInfo; i++)
2496 {
2497 PCMSG_CMS_SIGNER_INFO signerInfo =
2498 &msg->u.signed_data.info->rgSignerInfo[i];
2499
2500 if (signerInfo->SignerId.dwIdChoice == CERT_ID_ISSUER_SERIAL_NUMBER)
2501 {
2502 ret = CertCompareCertificateName(X509_ASN_ENCODING,
2503 &signerInfo->SignerId.IssuerSerialNumber.Issuer,
2504 &info->Issuer);
2505 if (ret)
2506 {
2507 ret = CertCompareIntegerBlob(
2508 &signerInfo->SignerId.IssuerSerialNumber.SerialNumber,
2509 &info->SerialNumber);
2510 if (ret)
2511 break;
2512 }
2513 }
2514 else
2515 {
2516 FIXME("signer %d: unimplemented for key id\n", i);
2517 }
2518 }
2519 if (ret)
2520 ret = CDecodeSignedMsg_VerifySignatureWithKey(msg, 0, i,
2521 &info->SubjectPublicKeyInfo);
2522 else
2523 SetLastError(CRYPT_E_SIGNER_NOT_FOUND);
2524
2525 return ret;
2526 }
2527
2528 static BOOL CDecodeSignedMsg_VerifySignatureEx(CDecodeMsg *msg,
2529 PCMSG_CTRL_VERIFY_SIGNATURE_EX_PARA para)
2530 {
2531 BOOL ret = FALSE;
2532
2533 if (para->cbSize != sizeof(CMSG_CTRL_VERIFY_SIGNATURE_EX_PARA))
2534 SetLastError(ERROR_INVALID_PARAMETER);
2535 else if (para->dwSignerIndex >= msg->u.signed_data.info->cSignerInfo)
2536 SetLastError(CRYPT_E_SIGNER_NOT_FOUND);
2537 else
2538 {
2539 switch (para->dwSignerType)
2540 {
2541 case CMSG_VERIFY_SIGNER_PUBKEY:
2542 ret = CDecodeSignedMsg_VerifySignatureWithKey(msg,
2543 para->hCryptProv, para->dwSignerIndex,
2544 (PCERT_PUBLIC_KEY_INFO)para->pvSigner);
2545 break;
2546 case CMSG_VERIFY_SIGNER_CERT:
2547 {
2548 PCCERT_CONTEXT cert = (PCCERT_CONTEXT)para->pvSigner;
2549
2550 ret = CDecodeSignedMsg_VerifySignatureWithKey(msg, para->hCryptProv,
2551 para->dwSignerIndex, &cert->pCertInfo->SubjectPublicKeyInfo);
2552 break;
2553 }
2554 default:
2555 FIXME("unimplemented for signer type %d\n", para->dwSignerType);
2556 SetLastError(CRYPT_E_SIGNER_NOT_FOUND);
2557 }
2558 }
2559 return ret;
2560 }
2561
2562 static BOOL CDecodeMsg_Control(HCRYPTMSG hCryptMsg, DWORD dwFlags,
2563 DWORD dwCtrlType, const void *pvCtrlPara)
2564 {
2565 CDecodeMsg *msg = (CDecodeMsg *)hCryptMsg;
2566 BOOL ret = FALSE;
2567
2568 switch (dwCtrlType)
2569 {
2570 case CMSG_CTRL_VERIFY_SIGNATURE:
2571 switch (msg->type)
2572 {
2573 case CMSG_SIGNED:
2574 ret = CDecodeSignedMsg_VerifySignature(msg, (PCERT_INFO)pvCtrlPara);
2575 break;
2576 default:
2577 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2578 }
2579 break;
2580 case CMSG_CTRL_DECRYPT:
2581 switch (msg->type)
2582 {
2583 default:
2584 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2585 }
2586 break;
2587 case CMSG_CTRL_VERIFY_HASH:
2588 switch (msg->type)
2589 {
2590 case CMSG_HASHED:
2591 ret = CDecodeHashMsg_VerifyHash(msg);
2592 break;
2593 default:
2594 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2595 }
2596 break;
2597 case CMSG_CTRL_VERIFY_SIGNATURE_EX:
2598 switch (msg->type)
2599 {
2600 case CMSG_SIGNED:
2601 ret = CDecodeSignedMsg_VerifySignatureEx(msg,
2602 (PCMSG_CTRL_VERIFY_SIGNATURE_EX_PARA)pvCtrlPara);
2603 break;
2604 default:
2605 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2606 }
2607 break;
2608 default:
2609 SetLastError(CRYPT_E_CONTROL_TYPE);
2610 }
2611 return ret;
2612 }
2613
2614 HCRYPTMSG WINAPI CryptMsgOpenToDecode(DWORD dwMsgEncodingType, DWORD dwFlags,
2615 DWORD dwMsgType, HCRYPTPROV_LEGACY hCryptProv, PCERT_INFO pRecipientInfo,
2616 PCMSG_STREAM_INFO pStreamInfo)
2617 {
2618 CDecodeMsg *msg;
2619
2620 TRACE("(%08x, %08x, %08x, %08lx, %p, %p)\n", dwMsgEncodingType,
2621 dwFlags, dwMsgType, hCryptProv, pRecipientInfo, pStreamInfo);
2622
2623 if (GET_CMSG_ENCODING_TYPE(dwMsgEncodingType) != PKCS_7_ASN_ENCODING)
2624 {
2625 SetLastError(E_INVALIDARG);
2626 return NULL;
2627 }
2628 msg = CryptMemAlloc(sizeof(CDecodeMsg));
2629 if (msg)
2630 {
2631 CryptMsgBase_Init((CryptMsgBase *)msg, dwFlags, pStreamInfo,
2632 CDecodeMsg_Close, CDecodeMsg_GetParam, CDecodeMsg_Update,
2633 CDecodeMsg_Control);
2634 msg->type = dwMsgType;
2635 if (hCryptProv)
2636 msg->crypt_prov = hCryptProv;
2637 else
2638 {
2639 msg->crypt_prov = CRYPT_GetDefaultProvider();
2640 msg->base.open_flags &= ~CMSG_CRYPT_RELEASE_CONTEXT_FLAG;
2641 }
2642 memset(&msg->u, 0, sizeof(msg->u));
2643 msg->msg_data.cbData = 0;
2644 msg->msg_data.pbData = NULL;
2645 msg->properties = ContextPropertyList_Create();
2646 }
2647 return msg;
2648 }
2649
2650 HCRYPTMSG WINAPI CryptMsgDuplicate(HCRYPTMSG hCryptMsg)
2651 {
2652 TRACE("(%p)\n", hCryptMsg);
2653
2654 if (hCryptMsg)
2655 {
2656 CryptMsgBase *msg = (CryptMsgBase *)hCryptMsg;
2657
2658 InterlockedIncrement(&msg->ref);
2659 }
2660 return hCryptMsg;
2661 }
2662
2663 BOOL WINAPI CryptMsgClose(HCRYPTMSG hCryptMsg)
2664 {
2665 TRACE("(%p)\n", hCryptMsg);
2666
2667 if (hCryptMsg)
2668 {
2669 CryptMsgBase *msg = (CryptMsgBase *)hCryptMsg;
2670
2671 if (InterlockedDecrement(&msg->ref) == 0)
2672 {
2673 TRACE("freeing %p\n", msg);
2674 if (msg->close)
2675 msg->close(msg);
2676 CryptMemFree(msg);
2677 }
2678 }
2679 return TRUE;
2680 }
2681
2682 BOOL WINAPI CryptMsgUpdate(HCRYPTMSG hCryptMsg, const BYTE *pbData,
2683 DWORD cbData, BOOL fFinal)
2684 {
2685 CryptMsgBase *msg = (CryptMsgBase *)hCryptMsg;
2686
2687 TRACE("(%p, %p, %d, %d)\n", hCryptMsg, pbData, cbData, fFinal);
2688
2689 return msg->update(hCryptMsg, pbData, cbData, fFinal);
2690 }
2691
2692 BOOL WINAPI CryptMsgGetParam(HCRYPTMSG hCryptMsg, DWORD dwParamType,
2693 DWORD dwIndex, void *pvData, DWORD *pcbData)
2694 {
2695 CryptMsgBase *msg = (CryptMsgBase *)hCryptMsg;
2696
2697 TRACE("(%p, %d, %d, %p, %p)\n", hCryptMsg, dwParamType, dwIndex,
2698 pvData, pcbData);
2699 return msg->get_param(hCryptMsg, dwParamType, dwIndex, pvData, pcbData);
2700 }
2701
2702 BOOL WINAPI CryptMsgControl(HCRYPTMSG hCryptMsg, DWORD dwFlags,
2703 DWORD dwCtrlType, const void *pvCtrlPara)
2704 {
2705 CryptMsgBase *msg = (CryptMsgBase *)hCryptMsg;
2706
2707 TRACE("(%p, %08x, %d, %p)\n", hCryptMsg, dwFlags, dwCtrlType,
2708 pvCtrlPara);
2709 return msg->control(hCryptMsg, dwFlags, dwCtrlType, pvCtrlPara);
2710 }
2711
2712 static CERT_INFO *CRYPT_GetSignerCertInfoFromMsg(HCRYPTMSG msg,
2713 DWORD dwSignerIndex)
2714 {
2715 CERT_INFO *certInfo = NULL;
2716 DWORD size;
2717
2718 if (CryptMsgGetParam(msg, CMSG_SIGNER_CERT_INFO_PARAM, dwSignerIndex, NULL,
2719 &size))
2720 {
2721 certInfo = CryptMemAlloc(size);
2722 if (certInfo)
2723 {
2724 if (!CryptMsgGetParam(msg, CMSG_SIGNER_CERT_INFO_PARAM,
2725 dwSignerIndex, certInfo, &size))
2726 {
2727 CryptMemFree(certInfo);
2728 certInfo = NULL;
2729 }
2730 }
2731 }
2732 return certInfo;
2733 }
2734
2735 BOOL WINAPI CryptMsgGetAndVerifySigner(HCRYPTMSG hCryptMsg, DWORD cSignerStore,
2736 HCERTSTORE *rghSignerStore, DWORD dwFlags, PCCERT_CONTEXT *ppSigner,
2737 DWORD *pdwSignerIndex)
2738 {
2739 HCERTSTORE store;
2740 DWORD i, signerIndex;
2741 PCCERT_CONTEXT signerCert = NULL;
2742 BOOL ret = FALSE;
2743
2744 TRACE("(%p, %d, %p, %08x, %p, %p)\n", hCryptMsg, cSignerStore,
2745 rghSignerStore, dwFlags, ppSigner, pdwSignerIndex);
2746
2747 /* Clear output parameters */
2748 if (ppSigner)
2749 *ppSigner = NULL;
2750 if (pdwSignerIndex && !(dwFlags & CMSG_USE_SIGNER_INDEX_FLAG))
2751 *pdwSignerIndex = 0;
2752
2753 /* Create store to search for signer certificates */
2754 store = CertOpenStore(CERT_STORE_PROV_COLLECTION, 0, 0,
2755 CERT_STORE_CREATE_NEW_FLAG, NULL);
2756 if (!(dwFlags & CMSG_TRUSTED_SIGNER_FLAG))
2757 {
2758 HCERTSTORE msgStore = CertOpenStore(CERT_STORE_PROV_MSG, 0, 0, 0,
2759 hCryptMsg);
2760
2761 CertAddStoreToCollection(store, msgStore, 0, 0);
2762 CertCloseStore(msgStore, 0);
2763 }
2764 for (i = 0; i < cSignerStore; i++)
2765 CertAddStoreToCollection(store, rghSignerStore[i], 0, 0);
2766
2767 /* Find signer cert */
2768 if (dwFlags & CMSG_USE_SIGNER_INDEX_FLAG)
2769 {
2770 CERT_INFO *signer = CRYPT_GetSignerCertInfoFromMsg(hCryptMsg,
2771 *pdwSignerIndex);
2772
2773 if (signer)
2774 {
2775 signerIndex = *pdwSignerIndex;
2776 signerCert = CertFindCertificateInStore(store, X509_ASN_ENCODING,
2777 0, CERT_FIND_SUBJECT_CERT, signer, NULL);
2778 CryptMemFree(signer);
2779 }
2780 }
2781 else
2782 {
2783 DWORD count, size = sizeof(count);
2784
2785 if (CryptMsgGetParam(hCryptMsg, CMSG_SIGNER_COUNT_PARAM, 0, &count,
2786 &size))
2787 {
2788 for (i = 0; !signerCert && i < count; i++)
2789 {
2790 CERT_INFO *signer = CRYPT_GetSignerCertInfoFromMsg(hCryptMsg,
2791 i);
2792
2793 if (signer)
2794 {
2795 signerCert = CertFindCertificateInStore(store,
2796 X509_ASN_ENCODING, 0, CERT_FIND_SUBJECT_CERT, signer,
2797 NULL);
2798 if (signerCert)
2799 signerIndex = i;
2800 CryptMemFree(signer);
2801 }
2802 }
2803 }
2804 if (!signerCert)
2805 SetLastError(CRYPT_E_NO_TRUSTED_SIGNER);
2806 }
2807 if (signerCert)
2808 {
2809 if (!(dwFlags & CMSG_SIGNER_ONLY_FLAG))
2810 ret = CryptMsgControl(hCryptMsg, 0, CMSG_CTRL_VERIFY_SIGNATURE,
2811 signerCert->pCertInfo);
2812 else
2813 ret = TRUE;
2814 if (ret)
2815 {
2816 if (ppSigner)
2817 *ppSigner = CertDuplicateCertificateContext(signerCert);
2818 if (pdwSignerIndex)
2819 *pdwSignerIndex = signerIndex;
2820 }
2821 CertFreeCertificateContext(signerCert);
2822 }
2823
2824 CertCloseStore(store, 0);
2825 return ret;
2826 }
2827
2828 BOOL WINAPI CryptMsgVerifyCountersignatureEncodedEx(HCRYPTPROV_LEGACY hCryptProv,
2829 DWORD dwEncodingType, PBYTE pbSignerInfo, DWORD cbSignerInfo,
2830 PBYTE pbSignerInfoCountersignature, DWORD cbSignerInfoCountersignature,
2831 DWORD dwSignerType, void *pvSigner, DWORD dwFlags, void *pvReserved)
2832 {
2833 FIXME("(%08lx, %08x, %p, %d, %p, %d, %d, %p, %08x, %p): stub\n", hCryptProv,
2834 dwEncodingType, pbSignerInfo, cbSignerInfo, pbSignerInfoCountersignature,
2835 cbSignerInfoCountersignature, dwSignerType, pvSigner, dwFlags, pvReserved);
2836 return FALSE;
2837 }
Windows API implementation