?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
push 0e883ac4a03c91e56787e1ec12e001b6558b4b62
[wine/hacks.git] / dlls / crypt32 / msg.c
blob8ac4b14d57b4087040188d3b7f1042929da0901b
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 HCRYPTMSG WINAPI CryptMsgOpenToEncode(DWORD dwMsgEncodingType, DWORD dwFlags,
1428 DWORD dwMsgType, const void *pvMsgEncodeInfo, LPSTR pszInnerContentObjID,
1429 PCMSG_STREAM_INFO pStreamInfo)
1430 {
1431 HCRYPTMSG msg = NULL;
1432
1433 TRACE("(%08x, %08x, %08x, %p, %s, %p)\n", dwMsgEncodingType, dwFlags,
1434 dwMsgType, pvMsgEncodeInfo, debugstr_a(pszInnerContentObjID), pStreamInfo);
1435
1436 if (GET_CMSG_ENCODING_TYPE(dwMsgEncodingType) != PKCS_7_ASN_ENCODING)
1437 {
1438 SetLastError(E_INVALIDARG);
1439 return NULL;
1440 }
1441 switch (dwMsgType)
1442 {
1443 case CMSG_DATA:
1444 msg = CDataEncodeMsg_Open(dwFlags, pvMsgEncodeInfo,
1445 pszInnerContentObjID, pStreamInfo);
1446 break;
1447 case CMSG_HASHED:
1448 msg = CHashEncodeMsg_Open(dwFlags, pvMsgEncodeInfo,
1449 pszInnerContentObjID, pStreamInfo);
1450 break;
1451 case CMSG_SIGNED:
1452 msg = CSignedEncodeMsg_Open(dwFlags, pvMsgEncodeInfo,
1453 pszInnerContentObjID, pStreamInfo);
1454 break;
1455 case CMSG_ENVELOPED:
1456 FIXME("unimplemented for type CMSG_ENVELOPED\n");
1457 break;
1458 case CMSG_SIGNED_AND_ENVELOPED:
1459 case CMSG_ENCRYPTED:
1460 /* defined but invalid, fall through */
1461 default:
1462 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
1463 }
1464 return msg;
1465 }
1466
1467 typedef struct _CDecodeMsg
1468 {
1469 CryptMsgBase base;
1470 DWORD type;
1471 HCRYPTPROV crypt_prov;
1472 union {
1473 HCRYPTHASH hash;
1474 CSignedMsgData signed_data;
1475 } u;
1476 CRYPT_DATA_BLOB msg_data;
1477 CRYPT_DATA_BLOB detached_data;
1478 PCONTEXT_PROPERTY_LIST properties;
1479 } CDecodeMsg;
1480
1481 static void CDecodeMsg_Close(HCRYPTMSG hCryptMsg)
1482 {
1483 CDecodeMsg *msg = (CDecodeMsg *)hCryptMsg;
1484
1485 if (msg->base.open_flags & CMSG_CRYPT_RELEASE_CONTEXT_FLAG)
1486 CryptReleaseContext(msg->crypt_prov, 0);
1487 switch (msg->type)
1488 {
1489 case CMSG_HASHED:
1490 if (msg->u.hash)
1491 CryptDestroyHash(msg->u.hash);
1492 break;
1493 case CMSG_SIGNED:
1494 if (msg->u.signed_data.info)
1495 {
1496 LocalFree(msg->u.signed_data.info);
1497 CSignedMsgData_CloseHandles(&msg->u.signed_data);
1498 }
1499 break;
1500 }
1501 CryptMemFree(msg->msg_data.pbData);
1502 CryptMemFree(msg->detached_data.pbData);
1503 ContextPropertyList_Free(msg->properties);
1504 }
1505
1506 static BOOL CDecodeMsg_CopyData(CRYPT_DATA_BLOB *blob, const BYTE *pbData,
1507 DWORD cbData)
1508 {
1509 BOOL ret = TRUE;
1510
1511 if (cbData)
1512 {
1513 if (blob->cbData)
1514 blob->pbData = CryptMemRealloc(blob->pbData,
1515 blob->cbData + cbData);
1516 else
1517 blob->pbData = CryptMemAlloc(cbData);
1518 if (blob->pbData)
1519 {
1520 memcpy(blob->pbData + blob->cbData, pbData, cbData);
1521 blob->cbData += cbData;
1522 }
1523 else
1524 ret = FALSE;
1525 }
1526 return ret;
1527 }
1528
1529 static BOOL CDecodeMsg_DecodeDataContent(CDecodeMsg *msg, CRYPT_DER_BLOB *blob)
1530 {
1531 BOOL ret;
1532 CRYPT_DATA_BLOB *data;
1533 DWORD size;
1534
1535 ret = CryptDecodeObjectEx(X509_ASN_ENCODING, X509_OCTET_STRING,
1536 blob->pbData, blob->cbData, CRYPT_DECODE_ALLOC_FLAG, NULL, (LPBYTE)&data,
1537 &size);
1538 if (ret)
1539 {
1540 ret = ContextPropertyList_SetProperty(msg->properties,
1541 CMSG_CONTENT_PARAM, data->pbData, data->cbData);
1542 LocalFree(data);
1543 }
1544 return ret;
1545 }
1546
1547 static void CDecodeMsg_SaveAlgorithmID(CDecodeMsg *msg, DWORD param,
1548 const CRYPT_ALGORITHM_IDENTIFIER *id)
1549 {
1550 static const BYTE nullParams[] = { ASN_NULL, 0 };
1551 CRYPT_ALGORITHM_IDENTIFIER *copy;
1552 DWORD len = sizeof(CRYPT_ALGORITHM_IDENTIFIER);
1553
1554 /* Linearize algorithm id */
1555 len += strlen(id->pszObjId) + 1;
1556 len += id->Parameters.cbData;
1557 copy = CryptMemAlloc(len);
1558 if (copy)
1559 {
1560 copy->pszObjId =
1561 (LPSTR)((BYTE *)copy + sizeof(CRYPT_ALGORITHM_IDENTIFIER));
1562 strcpy(copy->pszObjId, id->pszObjId);
1563 copy->Parameters.pbData = (BYTE *)copy->pszObjId + strlen(id->pszObjId)
1564 + 1;
1565 /* Trick: omit NULL parameters */
1566 if (id->Parameters.cbData == sizeof(nullParams) &&
1567 !memcmp(id->Parameters.pbData, nullParams, sizeof(nullParams)))
1568 {
1569 copy->Parameters.cbData = 0;
1570 len -= sizeof(nullParams);
1571 }
1572 else
1573 copy->Parameters.cbData = id->Parameters.cbData;
1574 if (copy->Parameters.cbData)
1575 memcpy(copy->Parameters.pbData, id->Parameters.pbData,
1576 id->Parameters.cbData);
1577 ContextPropertyList_SetProperty(msg->properties, param, (BYTE *)copy,
1578 len);
1579 CryptMemFree(copy);
1580 }
1581 }
1582
1583 static inline void CRYPT_FixUpAlgorithmID(CRYPT_ALGORITHM_IDENTIFIER *id)
1584 {
1585 id->pszObjId = (LPSTR)((BYTE *)id + sizeof(CRYPT_ALGORITHM_IDENTIFIER));
1586 id->Parameters.pbData = (BYTE *)id->pszObjId + strlen(id->pszObjId) + 1;
1587 }
1588
1589 static BOOL CDecodeMsg_DecodeHashedContent(CDecodeMsg *msg,
1590 CRYPT_DER_BLOB *blob)
1591 {
1592 BOOL ret;
1593 CRYPT_DIGESTED_DATA *digestedData;
1594 DWORD size;
1595
1596 ret = CRYPT_AsnDecodePKCSDigestedData(blob->pbData, blob->cbData,
1597 CRYPT_DECODE_ALLOC_FLAG, NULL, (CRYPT_DIGESTED_DATA *)&digestedData,
1598 &size);
1599 if (ret)
1600 {
1601 ContextPropertyList_SetProperty(msg->properties, CMSG_VERSION_PARAM,
1602 (const BYTE *)&digestedData->version, sizeof(digestedData->version));
1603 CDecodeMsg_SaveAlgorithmID(msg, CMSG_HASH_ALGORITHM_PARAM,
1604 &digestedData->DigestAlgorithm);
1605 ContextPropertyList_SetProperty(msg->properties,
1606 CMSG_INNER_CONTENT_TYPE_PARAM,
1607 (const BYTE *)digestedData->ContentInfo.pszObjId,
1608 digestedData->ContentInfo.pszObjId ?
1609 strlen(digestedData->ContentInfo.pszObjId) + 1 : 0);
1610 if (digestedData->ContentInfo.Content.cbData)
1611 CDecodeMsg_DecodeDataContent(msg,
1612 &digestedData->ContentInfo.Content);
1613 else
1614 ContextPropertyList_SetProperty(msg->properties,
1615 CMSG_CONTENT_PARAM, NULL, 0);
1616 ContextPropertyList_SetProperty(msg->properties, CMSG_HASH_DATA_PARAM,
1617 digestedData->hash.pbData, digestedData->hash.cbData);
1618 LocalFree(digestedData);
1619 }
1620 return ret;
1621 }
1622
1623 static BOOL CDecodeMsg_DecodeSignedContent(CDecodeMsg *msg,
1624 CRYPT_DER_BLOB *blob)
1625 {
1626 BOOL ret;
1627 CRYPT_SIGNED_INFO *signedInfo;
1628 DWORD size;
1629
1630 ret = CRYPT_AsnDecodeCMSSignedInfo(blob->pbData, blob->cbData,
1631 CRYPT_DECODE_ALLOC_FLAG, NULL, (CRYPT_SIGNED_INFO *)&signedInfo,
1632 &size);
1633 if (ret)
1634 {
1635 DWORD i;
1636
1637 msg->u.signed_data.info = signedInfo;
1638 ret = CSignedMsgData_AllocateHandles(&msg->u.signed_data);
1639 for (i = 0; ret && i < msg->u.signed_data.info->cSignerInfo; i++)
1640 ret = CSignedMsgData_ConstructSignerHandles(&msg->u.signed_data, i,
1641 msg->crypt_prov);
1642 if (ret)
1643 {
1644 CRYPT_DATA_BLOB *content;
1645
1646 /* Now that we have all the content, update the hash handles with
1647 * it. If the message is a detached message, the content is stored
1648 * in msg->detached_data rather than in the signed message's
1649 * content.
1650 */
1651 if (msg->base.open_flags & CMSG_DETACHED_FLAG)
1652 content = &msg->detached_data;
1653 else
1654 content = &msg->u.signed_data.info->content.Content;
1655 if (content->cbData)
1656 {
1657 /* If the message is not detached, have to decode the message's
1658 * content if the type is szOID_RSA_data.
1659 */
1660 if (!(msg->base.open_flags & CMSG_DETACHED_FLAG) &&
1661 !strcmp(msg->u.signed_data.info->content.pszObjId,
1662 szOID_RSA_data))
1663 {
1664 CRYPT_DATA_BLOB *blob;
1665
1666 ret = CryptDecodeObjectEx(X509_ASN_ENCODING,
1667 X509_OCTET_STRING, content->pbData, content->cbData,
1668 CRYPT_DECODE_ALLOC_FLAG, NULL, (LPBYTE)&blob, &size);
1669 if (ret)
1670 {
1671 ret = CSignedMsgData_Update(&msg->u.signed_data,
1672 blob->pbData, blob->cbData, TRUE, Verify);
1673 LocalFree(blob);
1674 }
1675 }
1676 else
1677 ret = CSignedMsgData_Update(&msg->u.signed_data,
1678 content->pbData, content->cbData, TRUE, Verify);
1679 }
1680 }
1681 }
1682 return ret;
1683 }
1684
1685 /* Decodes the content in blob as the type given, and updates the value
1686 * (type, parameters, etc.) of msg based on what blob contains.
1687 * It doesn't just use msg's type, to allow a recursive call from an implicitly
1688 * typed message once the outer content info has been decoded.
1689 */
1690 static BOOL CDecodeMsg_DecodeContent(CDecodeMsg *msg, CRYPT_DER_BLOB *blob,
1691 DWORD type)
1692 {
1693 BOOL ret;
1694
1695 switch (type)
1696 {
1697 case CMSG_DATA:
1698 if ((ret = CDecodeMsg_DecodeDataContent(msg, blob)))
1699 msg->type = CMSG_DATA;
1700 break;
1701 case CMSG_HASHED:
1702 if ((ret = CDecodeMsg_DecodeHashedContent(msg, blob)))
1703 msg->type = CMSG_HASHED;
1704 break;
1705 case CMSG_ENVELOPED:
1706 FIXME("unimplemented for type CMSG_ENVELOPED\n");
1707 ret = TRUE;
1708 break;
1709 case CMSG_SIGNED:
1710 if ((ret = CDecodeMsg_DecodeSignedContent(msg, blob)))
1711 msg->type = CMSG_SIGNED;
1712 break;
1713 default:
1714 {
1715 CRYPT_CONTENT_INFO *info;
1716 DWORD size;
1717
1718 ret = CryptDecodeObjectEx(X509_ASN_ENCODING, PKCS_CONTENT_INFO,
1719 msg->msg_data.pbData, msg->msg_data.cbData, CRYPT_DECODE_ALLOC_FLAG,
1720 NULL, (LPBYTE)&info, &size);
1721 if (ret)
1722 {
1723 if (!strcmp(info->pszObjId, szOID_RSA_data))
1724 ret = CDecodeMsg_DecodeContent(msg, &info->Content, CMSG_DATA);
1725 else if (!strcmp(info->pszObjId, szOID_RSA_digestedData))
1726 ret = CDecodeMsg_DecodeContent(msg, &info->Content,
1727 CMSG_HASHED);
1728 else if (!strcmp(info->pszObjId, szOID_RSA_envelopedData))
1729 ret = CDecodeMsg_DecodeContent(msg, &info->Content,
1730 CMSG_ENVELOPED);
1731 else if (!strcmp(info->pszObjId, szOID_RSA_signedData))
1732 ret = CDecodeMsg_DecodeContent(msg, &info->Content,
1733 CMSG_SIGNED);
1734 else
1735 {
1736 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
1737 ret = FALSE;
1738 }
1739 LocalFree(info);
1740 }
1741 }
1742 }
1743 return ret;
1744 }
1745
1746 static BOOL CDecodeMsg_Update(HCRYPTMSG hCryptMsg, const BYTE *pbData,
1747 DWORD cbData, BOOL fFinal)
1748 {
1749 CDecodeMsg *msg = (CDecodeMsg *)hCryptMsg;
1750 BOOL ret = FALSE;
1751
1752 TRACE("(%p, %p, %d, %d)\n", hCryptMsg, pbData, cbData, fFinal);
1753
1754 if (msg->base.state == MsgStateFinalized)
1755 SetLastError(CRYPT_E_MSG_ERROR);
1756 else if (msg->base.streamed)
1757 {
1758 FIXME("(%p, %p, %d, %d): streamed update stub\n", hCryptMsg, pbData,
1759 cbData, fFinal);
1760 switch (msg->base.state)
1761 {
1762 case MsgStateInit:
1763 ret = CDecodeMsg_CopyData(&msg->msg_data, pbData, cbData);
1764 if (fFinal)
1765 {
1766 if (msg->base.open_flags & CMSG_DETACHED_FLAG)
1767 msg->base.state = MsgStateDataFinalized;
1768 else
1769 msg->base.state = MsgStateFinalized;
1770 }
1771 else
1772 msg->base.state = MsgStateUpdated;
1773 break;
1774 case MsgStateUpdated:
1775 ret = CDecodeMsg_CopyData(&msg->msg_data, pbData, cbData);
1776 if (fFinal)
1777 {
1778 if (msg->base.open_flags & CMSG_DETACHED_FLAG)
1779 msg->base.state = MsgStateDataFinalized;
1780 else
1781 msg->base.state = MsgStateFinalized;
1782 }
1783 break;
1784 case MsgStateDataFinalized:
1785 ret = CDecodeMsg_CopyData(&msg->detached_data, pbData, cbData);
1786 if (fFinal)
1787 msg->base.state = MsgStateFinalized;
1788 break;
1789 default:
1790 SetLastError(CRYPT_E_MSG_ERROR);
1791 break;
1792 }
1793 }
1794 else
1795 {
1796 if (!fFinal)
1797 SetLastError(CRYPT_E_MSG_ERROR);
1798 else
1799 {
1800 switch (msg->base.state)
1801 {
1802 case MsgStateInit:
1803 ret = CDecodeMsg_CopyData(&msg->msg_data, pbData, cbData);
1804 if (msg->base.open_flags & CMSG_DETACHED_FLAG)
1805 msg->base.state = MsgStateDataFinalized;
1806 else
1807 msg->base.state = MsgStateFinalized;
1808 break;
1809 case MsgStateDataFinalized:
1810 ret = CDecodeMsg_CopyData(&msg->detached_data, pbData, cbData);
1811 msg->base.state = MsgStateFinalized;
1812 break;
1813 default:
1814 SetLastError(CRYPT_E_MSG_ERROR);
1815 }
1816 }
1817 }
1818 if (ret && msg->base.state == MsgStateFinalized)
1819 ret = CDecodeMsg_DecodeContent(msg, &msg->msg_data, msg->type);
1820 return ret;
1821 }
1822
1823 static BOOL CDecodeHashMsg_GetParam(CDecodeMsg *msg, DWORD dwParamType,
1824 DWORD dwIndex, void *pvData, DWORD *pcbData)
1825 {
1826 BOOL ret = FALSE;
1827
1828 switch (dwParamType)
1829 {
1830 case CMSG_TYPE_PARAM:
1831 ret = CRYPT_CopyParam(pvData, pcbData, &msg->type, sizeof(msg->type));
1832 break;
1833 case CMSG_HASH_ALGORITHM_PARAM:
1834 {
1835 CRYPT_DATA_BLOB blob;
1836
1837 ret = ContextPropertyList_FindProperty(msg->properties, dwParamType,
1838 &blob);
1839 if (ret)
1840 {
1841 ret = CRYPT_CopyParam(pvData, pcbData, blob.pbData, blob.cbData);
1842 if (ret && pvData)
1843 CRYPT_FixUpAlgorithmID((CRYPT_ALGORITHM_IDENTIFIER *)pvData);
1844 }
1845 else
1846 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
1847 break;
1848 }
1849 case CMSG_COMPUTED_HASH_PARAM:
1850 if (!msg->u.hash)
1851 {
1852 CRYPT_ALGORITHM_IDENTIFIER *hashAlgoID = NULL;
1853 DWORD size = 0;
1854 ALG_ID algID = 0;
1855
1856 CryptMsgGetParam(msg, CMSG_HASH_ALGORITHM_PARAM, 0, NULL, &size);
1857 hashAlgoID = CryptMemAlloc(size);
1858 ret = CryptMsgGetParam(msg, CMSG_HASH_ALGORITHM_PARAM, 0,
1859 hashAlgoID, &size);
1860 if (ret)
1861 algID = CertOIDToAlgId(hashAlgoID->pszObjId);
1862 ret = CryptCreateHash(msg->crypt_prov, algID, 0, 0, &msg->u.hash);
1863 if (ret)
1864 {
1865 CRYPT_DATA_BLOB content;
1866
1867 ret = ContextPropertyList_FindProperty(msg->properties,
1868 CMSG_CONTENT_PARAM, &content);
1869 if (ret)
1870 ret = CryptHashData(msg->u.hash, content.pbData,
1871 content.cbData, 0);
1872 }
1873 CryptMemFree(hashAlgoID);
1874 }
1875 else
1876 ret = TRUE;
1877 if (ret)
1878 ret = CryptGetHashParam(msg->u.hash, HP_HASHVAL, pvData, pcbData,
1879 0);
1880 break;
1881 default:
1882 {
1883 CRYPT_DATA_BLOB blob;
1884
1885 ret = ContextPropertyList_FindProperty(msg->properties, dwParamType,
1886 &blob);
1887 if (ret)
1888 ret = CRYPT_CopyParam(pvData, pcbData, blob.pbData, blob.cbData);
1889 else
1890 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
1891 }
1892 }
1893 return ret;
1894 }
1895
1896 /* nextData is an in/out parameter - on input it's the memory location in
1897 * which a copy of in's data should be made, and on output it's the memory
1898 * location immediately after out's copy of in's data.
1899 */
1900 static inline void CRYPT_CopyBlob(CRYPT_DATA_BLOB *out,
1901 const CRYPT_DATA_BLOB *in, LPBYTE *nextData)
1902 {
1903 out->cbData = in->cbData;
1904 if (in->cbData)
1905 {
1906 out->pbData = *nextData;
1907 memcpy(out->pbData, in->pbData, in->cbData);
1908 *nextData += in->cbData;
1909 }
1910 }
1911
1912 static inline void CRYPT_CopyAlgorithmId(CRYPT_ALGORITHM_IDENTIFIER *out,
1913 const CRYPT_ALGORITHM_IDENTIFIER *in, LPBYTE *nextData)
1914 {
1915 if (in->pszObjId)
1916 {
1917 out->pszObjId = (LPSTR)*nextData;
1918 strcpy(out->pszObjId, in->pszObjId);
1919 *nextData += strlen(out->pszObjId) + 1;
1920 }
1921 CRYPT_CopyBlob(&out->Parameters, &in->Parameters, nextData);
1922 }
1923
1924 static inline void CRYPT_CopyAttributes(CRYPT_ATTRIBUTES *out,
1925 const CRYPT_ATTRIBUTES *in, LPBYTE *nextData)
1926 {
1927 out->cAttr = in->cAttr;
1928 if (in->cAttr)
1929 {
1930 DWORD i;
1931
1932 if ((*nextData - (LPBYTE)0) % sizeof(DWORD_PTR))
1933 *nextData += (*nextData - (LPBYTE)0) % sizeof(DWORD_PTR);
1934 out->rgAttr = (CRYPT_ATTRIBUTE *)*nextData;
1935 *nextData += in->cAttr * sizeof(CRYPT_ATTRIBUTE);
1936 for (i = 0; i < in->cAttr; i++)
1937 {
1938 if (in->rgAttr[i].pszObjId)
1939 {
1940 out->rgAttr[i].pszObjId = (LPSTR)*nextData;
1941 strcpy(out->rgAttr[i].pszObjId, in->rgAttr[i].pszObjId);
1942 *nextData += strlen(in->rgAttr[i].pszObjId) + 1;
1943 }
1944 if (in->rgAttr[i].cValue)
1945 {
1946 DWORD j;
1947
1948 out->rgAttr[i].cValue = in->rgAttr[i].cValue;
1949 if ((*nextData - (LPBYTE)0) % sizeof(DWORD_PTR))
1950 *nextData += (*nextData - (LPBYTE)0) % sizeof(DWORD_PTR);
1951 out->rgAttr[i].rgValue = (PCRYPT_DATA_BLOB)*nextData;
1952 *nextData += in->rgAttr[i].cValue * sizeof(CRYPT_DATA_BLOB);
1953 for (j = 0; j < in->rgAttr[i].cValue; j++)
1954 CRYPT_CopyBlob(&out->rgAttr[i].rgValue[j],
1955 &in->rgAttr[i].rgValue[j], nextData);
1956 }
1957 }
1958 }
1959 }
1960
1961 static DWORD CRYPT_SizeOfAttributes(const CRYPT_ATTRIBUTES *attr)
1962 {
1963 DWORD size = attr->cAttr * sizeof(CRYPT_ATTRIBUTE), i, j;
1964
1965 for (i = 0; i < attr->cAttr; i++)
1966 {
1967 if (attr->rgAttr[i].pszObjId)
1968 size += strlen(attr->rgAttr[i].pszObjId) + 1;
1969 /* align pointer */
1970 if (size % sizeof(DWORD_PTR))
1971 size += size % sizeof(DWORD_PTR);
1972 size += attr->rgAttr[i].cValue * sizeof(CRYPT_DATA_BLOB);
1973 for (j = 0; j < attr->rgAttr[i].cValue; j++)
1974 size += attr->rgAttr[i].rgValue[j].cbData;
1975 }
1976 /* align pointer again to be conservative */
1977 if (size % sizeof(DWORD_PTR))
1978 size += size % sizeof(DWORD_PTR);
1979 return size;
1980 }
1981
1982 static DWORD CRYPT_SizeOfKeyIdAsIssuerAndSerial(const CRYPT_DATA_BLOB *keyId)
1983 {
1984 static char oid_key_rdn[] = szOID_KEYID_RDN;
1985 DWORD size = 0;
1986 CERT_RDN_ATTR attr;
1987 CERT_RDN rdn = { 1, &attr };
1988 CERT_NAME_INFO name = { 1, &rdn };
1989
1990 attr.pszObjId = oid_key_rdn;
1991 attr.dwValueType = CERT_RDN_OCTET_STRING;
1992 attr.Value.cbData = keyId->cbData;
1993 attr.Value.pbData = keyId->pbData;
1994 if (CryptEncodeObject(X509_ASN_ENCODING, X509_NAME, &name, NULL, &size))
1995 size++; /* Only include size of special zero serial number on success */
1996 return size;
1997 }
1998
1999 static BOOL CRYPT_CopyKeyIdAsIssuerAndSerial(CERT_NAME_BLOB *issuer,
2000 CRYPT_INTEGER_BLOB *serialNumber, const CRYPT_DATA_BLOB *keyId, DWORD encodedLen,
2001 LPBYTE *nextData)
2002 {
2003 static char oid_key_rdn[] = szOID_KEYID_RDN;
2004 CERT_RDN_ATTR attr;
2005 CERT_RDN rdn = { 1, &attr };
2006 CERT_NAME_INFO name = { 1, &rdn };
2007 BOOL ret;
2008
2009 /* Encode special zero serial number */
2010 serialNumber->cbData = 1;
2011 serialNumber->pbData = *nextData;
2012 **nextData = 0;
2013 (*nextData)++;
2014 /* Encode issuer */
2015 issuer->pbData = *nextData;
2016 attr.pszObjId = oid_key_rdn;
2017 attr.dwValueType = CERT_RDN_OCTET_STRING;
2018 attr.Value.cbData = keyId->cbData;
2019 attr.Value.pbData = keyId->pbData;
2020 ret = CryptEncodeObject(X509_ASN_ENCODING, X509_NAME, &name, *nextData,
2021 &encodedLen);
2022 if (ret)
2023 {
2024 *nextData += encodedLen;
2025 issuer->cbData = encodedLen;
2026 }
2027 return ret;
2028 }
2029
2030 static BOOL CRYPT_CopySignerInfo(void *pvData, DWORD *pcbData,
2031 const CMSG_CMS_SIGNER_INFO *in)
2032 {
2033 DWORD size = sizeof(CMSG_SIGNER_INFO), rdnSize;
2034 BOOL ret;
2035
2036 TRACE("(%p, %d, %p)\n", pvData, pvData ? *pcbData : 0, in);
2037
2038 if (in->SignerId.dwIdChoice == CERT_ID_ISSUER_SERIAL_NUMBER)
2039 {
2040 size += in->SignerId.IssuerSerialNumber.Issuer.cbData;
2041 size += in->SignerId.IssuerSerialNumber.SerialNumber.cbData;
2042 }
2043 else
2044 {
2045 rdnSize = CRYPT_SizeOfKeyIdAsIssuerAndSerial(&in->SignerId.KeyId);
2046 size += rdnSize;
2047 }
2048 if (in->HashAlgorithm.pszObjId)
2049 size += strlen(in->HashAlgorithm.pszObjId) + 1;
2050 size += in->HashAlgorithm.Parameters.cbData;
2051 if (in->HashEncryptionAlgorithm.pszObjId)
2052 size += strlen(in->HashEncryptionAlgorithm.pszObjId) + 1;
2053 size += in->HashEncryptionAlgorithm.Parameters.cbData;
2054 size += in->EncryptedHash.cbData;
2055 /* align pointer */
2056 if (size % sizeof(DWORD_PTR))
2057 size += size % sizeof(DWORD_PTR);
2058 size += CRYPT_SizeOfAttributes(&in->AuthAttrs);
2059 size += CRYPT_SizeOfAttributes(&in->UnauthAttrs);
2060 if (!pvData)
2061 {
2062 *pcbData = size;
2063 ret = TRUE;
2064 }
2065 else if (*pcbData < size)
2066 {
2067 *pcbData = size;
2068 SetLastError(ERROR_MORE_DATA);
2069 ret = FALSE;
2070 }
2071 else
2072 {
2073 LPBYTE nextData = (BYTE *)pvData + sizeof(CMSG_SIGNER_INFO);
2074 CMSG_SIGNER_INFO *out = (CMSG_SIGNER_INFO *)pvData;
2075
2076 ret = TRUE;
2077 out->dwVersion = in->dwVersion;
2078 if (in->SignerId.dwIdChoice == CERT_ID_ISSUER_SERIAL_NUMBER)
2079 {
2080 CRYPT_CopyBlob(&out->Issuer,
2081 &in->SignerId.IssuerSerialNumber.Issuer, &nextData);
2082 CRYPT_CopyBlob(&out->SerialNumber,
2083 &in->SignerId.IssuerSerialNumber.SerialNumber, &nextData);
2084 }
2085 else
2086 ret = CRYPT_CopyKeyIdAsIssuerAndSerial(&out->Issuer, &out->SerialNumber,
2087 &in->SignerId.KeyId, rdnSize, &nextData);
2088 if (ret)
2089 {
2090 CRYPT_CopyAlgorithmId(&out->HashAlgorithm, &in->HashAlgorithm,
2091 &nextData);
2092 CRYPT_CopyAlgorithmId(&out->HashEncryptionAlgorithm,
2093 &in->HashEncryptionAlgorithm, &nextData);
2094 CRYPT_CopyBlob(&out->EncryptedHash, &in->EncryptedHash, &nextData);
2095 /* align pointer */
2096 if ((nextData - (LPBYTE)0) % sizeof(DWORD_PTR))
2097 nextData += (nextData - (LPBYTE)0) % sizeof(DWORD_PTR);
2098 CRYPT_CopyAttributes(&out->AuthAttrs, &in->AuthAttrs, &nextData);
2099 CRYPT_CopyAttributes(&out->UnauthAttrs, &in->UnauthAttrs, &nextData);
2100 }
2101 }
2102 TRACE("returning %d\n", ret);
2103 return ret;
2104 }
2105
2106 static BOOL CRYPT_CopyCMSSignerInfo(void *pvData, DWORD *pcbData,
2107 const CMSG_CMS_SIGNER_INFO *in)
2108 {
2109 DWORD size = sizeof(CMSG_CMS_SIGNER_INFO);
2110 BOOL ret;
2111
2112 TRACE("(%p, %d, %p)\n", pvData, pvData ? *pcbData : 0, in);
2113
2114 if (in->SignerId.dwIdChoice == CERT_ID_ISSUER_SERIAL_NUMBER)
2115 {
2116 size += in->SignerId.IssuerSerialNumber.Issuer.cbData;
2117 size += in->SignerId.IssuerSerialNumber.SerialNumber.cbData;
2118 }
2119 else
2120 size += in->SignerId.KeyId.cbData;
2121 if (in->HashAlgorithm.pszObjId)
2122 size += strlen(in->HashAlgorithm.pszObjId) + 1;
2123 size += in->HashAlgorithm.Parameters.cbData;
2124 if (in->HashEncryptionAlgorithm.pszObjId)
2125 size += strlen(in->HashEncryptionAlgorithm.pszObjId) + 1;
2126 size += in->HashEncryptionAlgorithm.Parameters.cbData;
2127 size += in->EncryptedHash.cbData;
2128 /* align pointer */
2129 if (size % sizeof(DWORD_PTR))
2130 size += size % sizeof(DWORD_PTR);
2131 size += CRYPT_SizeOfAttributes(&in->AuthAttrs);
2132 size += CRYPT_SizeOfAttributes(&in->UnauthAttrs);
2133 if (!pvData)
2134 {
2135 *pcbData = size;
2136 ret = TRUE;
2137 }
2138 else if (*pcbData < size)
2139 {
2140 *pcbData = size;
2141 SetLastError(ERROR_MORE_DATA);
2142 ret = FALSE;
2143 }
2144 else
2145 {
2146 LPBYTE nextData = (BYTE *)pvData + sizeof(CMSG_CMS_SIGNER_INFO);
2147 CMSG_CMS_SIGNER_INFO *out = (CMSG_CMS_SIGNER_INFO *)pvData;
2148
2149 out->dwVersion = in->dwVersion;
2150 out->SignerId.dwIdChoice = in->SignerId.dwIdChoice;
2151 if (in->SignerId.dwIdChoice == CERT_ID_ISSUER_SERIAL_NUMBER)
2152 {
2153 CRYPT_CopyBlob(&out->SignerId.IssuerSerialNumber.Issuer,
2154 &in->SignerId.IssuerSerialNumber.Issuer, &nextData);
2155 CRYPT_CopyBlob(&out->SignerId.IssuerSerialNumber.SerialNumber,
2156 &in->SignerId.IssuerSerialNumber.SerialNumber, &nextData);
2157 }
2158 else
2159 CRYPT_CopyBlob(&out->SignerId.KeyId, &in->SignerId.KeyId, &nextData);
2160 CRYPT_CopyAlgorithmId(&out->HashAlgorithm, &in->HashAlgorithm,
2161 &nextData);
2162 CRYPT_CopyAlgorithmId(&out->HashEncryptionAlgorithm,
2163 &in->HashEncryptionAlgorithm, &nextData);
2164 CRYPT_CopyBlob(&out->EncryptedHash, &in->EncryptedHash, &nextData);
2165 /* align pointer */
2166 if ((nextData - (LPBYTE)0) % sizeof(DWORD_PTR))
2167 nextData += (nextData - (LPBYTE)0) % sizeof(DWORD_PTR);
2168 CRYPT_CopyAttributes(&out->AuthAttrs, &in->AuthAttrs, &nextData);
2169 CRYPT_CopyAttributes(&out->UnauthAttrs, &in->UnauthAttrs, &nextData);
2170 ret = TRUE;
2171 }
2172 TRACE("returning %d\n", ret);
2173 return ret;
2174 }
2175
2176 static BOOL CRYPT_CopySignerCertInfo(void *pvData, DWORD *pcbData,
2177 const CMSG_CMS_SIGNER_INFO *in)
2178 {
2179 DWORD size = sizeof(CERT_INFO), rdnSize;
2180 BOOL ret;
2181
2182 TRACE("(%p, %d, %p)\n", pvData, pvData ? *pcbData : 0, in);
2183
2184 if (in->SignerId.dwIdChoice == CERT_ID_ISSUER_SERIAL_NUMBER)
2185 {
2186 size += in->SignerId.IssuerSerialNumber.Issuer.cbData;
2187 size += in->SignerId.IssuerSerialNumber.SerialNumber.cbData;
2188 }
2189 else
2190 {
2191 rdnSize = CRYPT_SizeOfKeyIdAsIssuerAndSerial(&in->SignerId.KeyId);
2192 size += rdnSize;
2193 }
2194 if (!pvData)
2195 {
2196 *pcbData = size;
2197 ret = TRUE;
2198 }
2199 else if (*pcbData < size)
2200 {
2201 *pcbData = size;
2202 SetLastError(ERROR_MORE_DATA);
2203 ret = FALSE;
2204 }
2205 else
2206 {
2207 LPBYTE nextData = (BYTE *)pvData + sizeof(CERT_INFO);
2208 CERT_INFO *out = (CERT_INFO *)pvData;
2209
2210 memset(out, 0, sizeof(CERT_INFO));
2211 if (in->SignerId.dwIdChoice == CERT_ID_ISSUER_SERIAL_NUMBER)
2212 {
2213 CRYPT_CopyBlob(&out->Issuer,
2214 &in->SignerId.IssuerSerialNumber.Issuer, &nextData);
2215 CRYPT_CopyBlob(&out->SerialNumber,
2216 &in->SignerId.IssuerSerialNumber.SerialNumber, &nextData);
2217 ret = TRUE;
2218 }
2219 else
2220 ret = CRYPT_CopyKeyIdAsIssuerAndSerial(&out->Issuer, &out->SerialNumber,
2221 &in->SignerId.KeyId, rdnSize, &nextData);
2222 }
2223 TRACE("returning %d\n", ret);
2224 return ret;
2225 }
2226
2227 static BOOL CDecodeSignedMsg_GetParam(CDecodeMsg *msg, DWORD dwParamType,
2228 DWORD dwIndex, void *pvData, DWORD *pcbData)
2229 {
2230 BOOL ret = FALSE;
2231
2232 switch (dwParamType)
2233 {
2234 case CMSG_TYPE_PARAM:
2235 ret = CRYPT_CopyParam(pvData, pcbData, &msg->type, sizeof(msg->type));
2236 break;
2237 case CMSG_CONTENT_PARAM:
2238 if (msg->u.signed_data.info)
2239 {
2240 if (!strcmp(msg->u.signed_data.info->content.pszObjId,
2241 szOID_RSA_data))
2242 {
2243 CRYPT_DATA_BLOB *blob;
2244 DWORD size;
2245
2246 ret = CryptDecodeObjectEx(X509_ASN_ENCODING, X509_OCTET_STRING,
2247 msg->u.signed_data.info->content.Content.pbData,
2248 msg->u.signed_data.info->content.Content.cbData,
2249 CRYPT_DECODE_ALLOC_FLAG, NULL, (LPBYTE)&blob, &size);
2250 if (ret)
2251 {
2252 ret = CRYPT_CopyParam(pvData, pcbData, blob->pbData,
2253 blob->cbData);
2254 LocalFree(blob);
2255 }
2256 }
2257 else
2258 ret = CRYPT_CopyParam(pvData, pcbData,
2259 msg->u.signed_data.info->content.Content.pbData,
2260 msg->u.signed_data.info->content.Content.cbData);
2261 }
2262 else
2263 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2264 break;
2265 case CMSG_INNER_CONTENT_TYPE_PARAM:
2266 if (msg->u.signed_data.info)
2267 ret = CRYPT_CopyParam(pvData, pcbData,
2268 msg->u.signed_data.info->content.pszObjId,
2269 strlen(msg->u.signed_data.info->content.pszObjId) + 1);
2270 else
2271 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2272 break;
2273 case CMSG_SIGNER_COUNT_PARAM:
2274 if (msg->u.signed_data.info)
2275 ret = CRYPT_CopyParam(pvData, pcbData,
2276 &msg->u.signed_data.info->cSignerInfo, sizeof(DWORD));
2277 else
2278 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2279 break;
2280 case CMSG_SIGNER_INFO_PARAM:
2281 if (msg->u.signed_data.info)
2282 {
2283 if (dwIndex >= msg->u.signed_data.info->cSignerInfo)
2284 SetLastError(CRYPT_E_INVALID_INDEX);
2285 else
2286 ret = CRYPT_CopySignerInfo(pvData, pcbData,
2287 &msg->u.signed_data.info->rgSignerInfo[dwIndex]);
2288 }
2289 else
2290 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2291 break;
2292 case CMSG_SIGNER_CERT_INFO_PARAM:
2293 if (msg->u.signed_data.info)
2294 {
2295 if (dwIndex >= msg->u.signed_data.info->cSignerInfo)
2296 SetLastError(CRYPT_E_INVALID_INDEX);
2297 else
2298 ret = CRYPT_CopySignerCertInfo(pvData, pcbData,
2299 &msg->u.signed_data.info->rgSignerInfo[dwIndex]);
2300 }
2301 else
2302 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2303 break;
2304 case CMSG_CERT_COUNT_PARAM:
2305 if (msg->u.signed_data.info)
2306 ret = CRYPT_CopyParam(pvData, pcbData,
2307 &msg->u.signed_data.info->cCertEncoded, sizeof(DWORD));
2308 else
2309 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2310 break;
2311 case CMSG_CERT_PARAM:
2312 if (msg->u.signed_data.info)
2313 {
2314 if (dwIndex >= msg->u.signed_data.info->cCertEncoded)
2315 SetLastError(CRYPT_E_INVALID_INDEX);
2316 else
2317 ret = CRYPT_CopyParam(pvData, pcbData,
2318 msg->u.signed_data.info->rgCertEncoded[dwIndex].pbData,
2319 msg->u.signed_data.info->rgCertEncoded[dwIndex].cbData);
2320 }
2321 else
2322 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2323 break;
2324 case CMSG_CRL_COUNT_PARAM:
2325 if (msg->u.signed_data.info)
2326 ret = CRYPT_CopyParam(pvData, pcbData,
2327 &msg->u.signed_data.info->cCrlEncoded, sizeof(DWORD));
2328 else
2329 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2330 break;
2331 case CMSG_CRL_PARAM:
2332 if (msg->u.signed_data.info)
2333 {
2334 if (dwIndex >= msg->u.signed_data.info->cCrlEncoded)
2335 SetLastError(CRYPT_E_INVALID_INDEX);
2336 else
2337 ret = CRYPT_CopyParam(pvData, pcbData,
2338 msg->u.signed_data.info->rgCrlEncoded[dwIndex].pbData,
2339 msg->u.signed_data.info->rgCrlEncoded[dwIndex].cbData);
2340 }
2341 else
2342 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2343 break;
2344 case CMSG_COMPUTED_HASH_PARAM:
2345 if (msg->u.signed_data.info)
2346 {
2347 if (dwIndex >= msg->u.signed_data.cSignerHandle)
2348 SetLastError(CRYPT_E_INVALID_INDEX);
2349 else
2350 ret = CryptGetHashParam(
2351 msg->u.signed_data.signerHandles[dwIndex].contentHash,
2352 HP_HASHVAL, pvData, pcbData, 0);
2353 }
2354 else
2355 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2356 break;
2357 case CMSG_ATTR_CERT_COUNT_PARAM:
2358 if (msg->u.signed_data.info)
2359 {
2360 DWORD attrCertCount = 0;
2361
2362 ret = CRYPT_CopyParam(pvData, pcbData,
2363 &attrCertCount, sizeof(DWORD));
2364 }
2365 else
2366 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2367 break;
2368 case CMSG_ATTR_CERT_PARAM:
2369 if (msg->u.signed_data.info)
2370 SetLastError(CRYPT_E_INVALID_INDEX);
2371 else
2372 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2373 break;
2374 case CMSG_CMS_SIGNER_INFO_PARAM:
2375 if (msg->u.signed_data.info)
2376 {
2377 if (dwIndex >= msg->u.signed_data.info->cSignerInfo)
2378 SetLastError(CRYPT_E_INVALID_INDEX);
2379 else
2380 ret = CRYPT_CopyCMSSignerInfo(pvData, pcbData,
2381 &msg->u.signed_data.info->rgSignerInfo[dwIndex]);
2382 }
2383 else
2384 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2385 break;
2386 default:
2387 FIXME("unimplemented for %d\n", dwParamType);
2388 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2389 }
2390 return ret;
2391 }
2392
2393 static BOOL CDecodeMsg_GetParam(HCRYPTMSG hCryptMsg, DWORD dwParamType,
2394 DWORD dwIndex, void *pvData, DWORD *pcbData)
2395 {
2396 CDecodeMsg *msg = (CDecodeMsg *)hCryptMsg;
2397 BOOL ret = FALSE;
2398
2399 switch (msg->type)
2400 {
2401 case CMSG_HASHED:
2402 ret = CDecodeHashMsg_GetParam(msg, dwParamType, dwIndex, pvData,
2403 pcbData);
2404 break;
2405 case CMSG_SIGNED:
2406 ret = CDecodeSignedMsg_GetParam(msg, dwParamType, dwIndex, pvData,
2407 pcbData);
2408 break;
2409 default:
2410 switch (dwParamType)
2411 {
2412 case CMSG_TYPE_PARAM:
2413 ret = CRYPT_CopyParam(pvData, pcbData, &msg->type,
2414 sizeof(msg->type));
2415 break;
2416 default:
2417 {
2418 CRYPT_DATA_BLOB blob;
2419
2420 ret = ContextPropertyList_FindProperty(msg->properties, dwParamType,
2421 &blob);
2422 if (ret)
2423 ret = CRYPT_CopyParam(pvData, pcbData, blob.pbData,
2424 blob.cbData);
2425 else
2426 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2427 }
2428 }
2429 }
2430 return ret;
2431 }
2432
2433 static BOOL CDecodeHashMsg_VerifyHash(CDecodeMsg *msg)
2434 {
2435 BOOL ret;
2436 CRYPT_DATA_BLOB hashBlob;
2437
2438 ret = ContextPropertyList_FindProperty(msg->properties,
2439 CMSG_HASH_DATA_PARAM, &hashBlob);
2440 if (ret)
2441 {
2442 DWORD computedHashSize = 0;
2443
2444 ret = CDecodeHashMsg_GetParam(msg, CMSG_COMPUTED_HASH_PARAM, 0, NULL,
2445 &computedHashSize);
2446 if (hashBlob.cbData == computedHashSize)
2447 {
2448 LPBYTE computedHash = CryptMemAlloc(computedHashSize);
2449
2450 if (computedHash)
2451 {
2452 ret = CDecodeHashMsg_GetParam(msg, CMSG_COMPUTED_HASH_PARAM, 0,
2453 computedHash, &computedHashSize);
2454 if (ret)
2455 ret = !memcmp(hashBlob.pbData, computedHash,
2456 hashBlob.cbData);
2457 CryptMemFree(computedHash);
2458 }
2459 else
2460 ret = FALSE;
2461 }
2462 }
2463 return ret;
2464 }
2465
2466 static BOOL CDecodeSignedMsg_VerifySignatureWithKey(CDecodeMsg *msg,
2467 HCRYPTPROV prov, DWORD signerIndex, PCERT_PUBLIC_KEY_INFO keyInfo)
2468 {
2469 HCRYPTKEY key;
2470 BOOL ret;
2471
2472 if (!prov)
2473 prov = msg->crypt_prov;
2474 ret = CryptImportPublicKeyInfo(prov, X509_ASN_ENCODING, keyInfo, &key);
2475 if (ret)
2476 {
2477 HCRYPTHASH hash;
2478 CRYPT_HASH_BLOB reversedHash;
2479
2480 if (msg->u.signed_data.info->rgSignerInfo[signerIndex].AuthAttrs.cAttr)
2481 hash = msg->u.signed_data.signerHandles[signerIndex].authAttrHash;
2482 else
2483 hash = msg->u.signed_data.signerHandles[signerIndex].contentHash;
2484 ret = CRYPT_ConstructBlob(&reversedHash,
2485 &msg->u.signed_data.info->rgSignerInfo[signerIndex].EncryptedHash);
2486 if (ret)
2487 {
2488 CRYPT_ReverseBytes(&reversedHash);
2489 ret = CryptVerifySignatureW(hash, reversedHash.pbData,
2490 reversedHash.cbData, key, NULL, 0);
2491 CryptMemFree(reversedHash.pbData);
2492 }
2493 CryptDestroyKey(key);
2494 }
2495 return ret;
2496 }
2497
2498 static BOOL CDecodeSignedMsg_VerifySignature(CDecodeMsg *msg, PCERT_INFO info)
2499 {
2500 BOOL ret = FALSE;
2501 DWORD i;
2502
2503 for (i = 0; !ret && i < msg->u.signed_data.info->cSignerInfo; i++)
2504 {
2505 PCMSG_CMS_SIGNER_INFO signerInfo =
2506 &msg->u.signed_data.info->rgSignerInfo[i];
2507
2508 if (signerInfo->SignerId.dwIdChoice == CERT_ID_ISSUER_SERIAL_NUMBER)
2509 {
2510 ret = CertCompareCertificateName(X509_ASN_ENCODING,
2511 &signerInfo->SignerId.IssuerSerialNumber.Issuer,
2512 &info->Issuer);
2513 if (ret)
2514 {
2515 ret = CertCompareIntegerBlob(
2516 &signerInfo->SignerId.IssuerSerialNumber.SerialNumber,
2517 &info->SerialNumber);
2518 if (ret)
2519 break;
2520 }
2521 }
2522 else
2523 {
2524 FIXME("signer %d: unimplemented for key id\n", i);
2525 }
2526 }
2527 if (ret)
2528 ret = CDecodeSignedMsg_VerifySignatureWithKey(msg, 0, i,
2529 &info->SubjectPublicKeyInfo);
2530 else
2531 SetLastError(CRYPT_E_SIGNER_NOT_FOUND);
2532
2533 return ret;
2534 }
2535
2536 static BOOL CDecodeSignedMsg_VerifySignatureEx(CDecodeMsg *msg,
2537 PCMSG_CTRL_VERIFY_SIGNATURE_EX_PARA para)
2538 {
2539 BOOL ret = FALSE;
2540
2541 if (para->cbSize != sizeof(CMSG_CTRL_VERIFY_SIGNATURE_EX_PARA))
2542 SetLastError(ERROR_INVALID_PARAMETER);
2543 else if (para->dwSignerIndex >= msg->u.signed_data.info->cSignerInfo)
2544 SetLastError(CRYPT_E_SIGNER_NOT_FOUND);
2545 else
2546 {
2547 switch (para->dwSignerType)
2548 {
2549 case CMSG_VERIFY_SIGNER_PUBKEY:
2550 ret = CDecodeSignedMsg_VerifySignatureWithKey(msg,
2551 para->hCryptProv, para->dwSignerIndex,
2552 (PCERT_PUBLIC_KEY_INFO)para->pvSigner);
2553 break;
2554 case CMSG_VERIFY_SIGNER_CERT:
2555 {
2556 PCCERT_CONTEXT cert = (PCCERT_CONTEXT)para->pvSigner;
2557
2558 ret = CDecodeSignedMsg_VerifySignatureWithKey(msg, para->hCryptProv,
2559 para->dwSignerIndex, &cert->pCertInfo->SubjectPublicKeyInfo);
2560 break;
2561 }
2562 default:
2563 FIXME("unimplemented for signer type %d\n", para->dwSignerType);
2564 SetLastError(CRYPT_E_SIGNER_NOT_FOUND);
2565 }
2566 }
2567 return ret;
2568 }
2569
2570 static BOOL CDecodeMsg_Control(HCRYPTMSG hCryptMsg, DWORD dwFlags,
2571 DWORD dwCtrlType, const void *pvCtrlPara)
2572 {
2573 CDecodeMsg *msg = (CDecodeMsg *)hCryptMsg;
2574 BOOL ret = FALSE;
2575
2576 switch (dwCtrlType)
2577 {
2578 case CMSG_CTRL_VERIFY_SIGNATURE:
2579 switch (msg->type)
2580 {
2581 case CMSG_SIGNED:
2582 ret = CDecodeSignedMsg_VerifySignature(msg, (PCERT_INFO)pvCtrlPara);
2583 break;
2584 default:
2585 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2586 }
2587 break;
2588 case CMSG_CTRL_DECRYPT:
2589 switch (msg->type)
2590 {
2591 default:
2592 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2593 }
2594 break;
2595 case CMSG_CTRL_VERIFY_HASH:
2596 switch (msg->type)
2597 {
2598 case CMSG_HASHED:
2599 ret = CDecodeHashMsg_VerifyHash(msg);
2600 break;
2601 default:
2602 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2603 }
2604 break;
2605 case CMSG_CTRL_VERIFY_SIGNATURE_EX:
2606 switch (msg->type)
2607 {
2608 case CMSG_SIGNED:
2609 ret = CDecodeSignedMsg_VerifySignatureEx(msg,
2610 (PCMSG_CTRL_VERIFY_SIGNATURE_EX_PARA)pvCtrlPara);
2611 break;
2612 default:
2613 SetLastError(CRYPT_E_INVALID_MSG_TYPE);
2614 }
2615 break;
2616 default:
2617 SetLastError(CRYPT_E_CONTROL_TYPE);
2618 }
2619 return ret;
2620 }
2621
2622 HCRYPTMSG WINAPI CryptMsgOpenToDecode(DWORD dwMsgEncodingType, DWORD dwFlags,
2623 DWORD dwMsgType, HCRYPTPROV_LEGACY hCryptProv, PCERT_INFO pRecipientInfo,
2624 PCMSG_STREAM_INFO pStreamInfo)
2625 {
2626 CDecodeMsg *msg;
2627
2628 TRACE("(%08x, %08x, %08x, %08lx, %p, %p)\n", dwMsgEncodingType,
2629 dwFlags, dwMsgType, hCryptProv, pRecipientInfo, pStreamInfo);
2630
2631 if (GET_CMSG_ENCODING_TYPE(dwMsgEncodingType) != PKCS_7_ASN_ENCODING)
2632 {
2633 SetLastError(E_INVALIDARG);
2634 return NULL;
2635 }
2636 msg = CryptMemAlloc(sizeof(CDecodeMsg));
2637 if (msg)
2638 {
2639 CryptMsgBase_Init((CryptMsgBase *)msg, dwFlags, pStreamInfo,
2640 CDecodeMsg_Close, CDecodeMsg_GetParam, CDecodeMsg_Update,
2641 CDecodeMsg_Control);
2642 msg->type = dwMsgType;
2643 if (hCryptProv)
2644 msg->crypt_prov = hCryptProv;
2645 else
2646 {
2647 msg->crypt_prov = CRYPT_GetDefaultProvider();
2648 msg->base.open_flags &= ~CMSG_CRYPT_RELEASE_CONTEXT_FLAG;
2649 }
2650 memset(&msg->u, 0, sizeof(msg->u));
2651 msg->msg_data.cbData = 0;
2652 msg->msg_data.pbData = NULL;
2653 msg->detached_data.cbData = 0;
2654 msg->detached_data.pbData = NULL;
2655 msg->properties = ContextPropertyList_Create();
2656 }
2657 return msg;
2658 }
2659
2660 HCRYPTMSG WINAPI CryptMsgDuplicate(HCRYPTMSG hCryptMsg)
2661 {
2662 TRACE("(%p)\n", hCryptMsg);
2663
2664 if (hCryptMsg)
2665 {
2666 CryptMsgBase *msg = (CryptMsgBase *)hCryptMsg;
2667
2668 InterlockedIncrement(&msg->ref);
2669 }
2670 return hCryptMsg;
2671 }
2672
2673 BOOL WINAPI CryptMsgClose(HCRYPTMSG hCryptMsg)
2674 {
2675 TRACE("(%p)\n", hCryptMsg);
2676
2677 if (hCryptMsg)
2678 {
2679 CryptMsgBase *msg = (CryptMsgBase *)hCryptMsg;
2680
2681 if (InterlockedDecrement(&msg->ref) == 0)
2682 {
2683 TRACE("freeing %p\n", msg);
2684 if (msg->close)
2685 msg->close(msg);
2686 CryptMemFree(msg);
2687 }
2688 }
2689 return TRUE;
2690 }
2691
2692 BOOL WINAPI CryptMsgUpdate(HCRYPTMSG hCryptMsg, const BYTE *pbData,
2693 DWORD cbData, BOOL fFinal)
2694 {
2695 CryptMsgBase *msg = (CryptMsgBase *)hCryptMsg;
2696
2697 TRACE("(%p, %p, %d, %d)\n", hCryptMsg, pbData, cbData, fFinal);
2698
2699 return msg->update(hCryptMsg, pbData, cbData, fFinal);
2700 }
2701
2702 BOOL WINAPI CryptMsgGetParam(HCRYPTMSG hCryptMsg, DWORD dwParamType,
2703 DWORD dwIndex, void *pvData, DWORD *pcbData)
2704 {
2705 CryptMsgBase *msg = (CryptMsgBase *)hCryptMsg;
2706
2707 TRACE("(%p, %d, %d, %p, %p)\n", hCryptMsg, dwParamType, dwIndex,
2708 pvData, pcbData);
2709 return msg->get_param(hCryptMsg, dwParamType, dwIndex, pvData, pcbData);
2710 }
2711
2712 BOOL WINAPI CryptMsgControl(HCRYPTMSG hCryptMsg, DWORD dwFlags,
2713 DWORD dwCtrlType, const void *pvCtrlPara)
2714 {
2715 CryptMsgBase *msg = (CryptMsgBase *)hCryptMsg;
2716
2717 TRACE("(%p, %08x, %d, %p)\n", hCryptMsg, dwFlags, dwCtrlType,
2718 pvCtrlPara);
2719 return msg->control(hCryptMsg, dwFlags, dwCtrlType, pvCtrlPara);
2720 }
2721
2722 static CERT_INFO *CRYPT_GetSignerCertInfoFromMsg(HCRYPTMSG msg,
2723 DWORD dwSignerIndex)
2724 {
2725 CERT_INFO *certInfo = NULL;
2726 DWORD size;
2727
2728 if (CryptMsgGetParam(msg, CMSG_SIGNER_CERT_INFO_PARAM, dwSignerIndex, NULL,
2729 &size))
2730 {
2731 certInfo = CryptMemAlloc(size);
2732 if (certInfo)
2733 {
2734 if (!CryptMsgGetParam(msg, CMSG_SIGNER_CERT_INFO_PARAM,
2735 dwSignerIndex, certInfo, &size))
2736 {
2737 CryptMemFree(certInfo);
2738 certInfo = NULL;
2739 }
2740 }
2741 }
2742 return certInfo;
2743 }
2744
2745 BOOL WINAPI CryptMsgGetAndVerifySigner(HCRYPTMSG hCryptMsg, DWORD cSignerStore,
2746 HCERTSTORE *rghSignerStore, DWORD dwFlags, PCCERT_CONTEXT *ppSigner,
2747 DWORD *pdwSignerIndex)
2748 {
2749 HCERTSTORE store;
2750 DWORD i, signerIndex;
2751 PCCERT_CONTEXT signerCert = NULL;
2752 BOOL ret = FALSE;
2753
2754 TRACE("(%p, %d, %p, %08x, %p, %p)\n", hCryptMsg, cSignerStore,
2755 rghSignerStore, dwFlags, ppSigner, pdwSignerIndex);
2756
2757 /* Clear output parameters */
2758 if (ppSigner)
2759 *ppSigner = NULL;
2760 if (pdwSignerIndex && !(dwFlags & CMSG_USE_SIGNER_INDEX_FLAG))
2761 *pdwSignerIndex = 0;
2762
2763 /* Create store to search for signer certificates */
2764 store = CertOpenStore(CERT_STORE_PROV_COLLECTION, 0, 0,
2765 CERT_STORE_CREATE_NEW_FLAG, NULL);
2766 if (!(dwFlags & CMSG_TRUSTED_SIGNER_FLAG))
2767 {
2768 HCERTSTORE msgStore = CertOpenStore(CERT_STORE_PROV_MSG, 0, 0, 0,
2769 hCryptMsg);
2770
2771 CertAddStoreToCollection(store, msgStore, 0, 0);
2772 CertCloseStore(msgStore, 0);
2773 }
2774 for (i = 0; i < cSignerStore; i++)
2775 CertAddStoreToCollection(store, rghSignerStore[i], 0, 0);
2776
2777 /* Find signer cert */
2778 if (dwFlags & CMSG_USE_SIGNER_INDEX_FLAG)
2779 {
2780 CERT_INFO *signer = CRYPT_GetSignerCertInfoFromMsg(hCryptMsg,
2781 *pdwSignerIndex);
2782
2783 if (signer)
2784 {
2785 signerIndex = *pdwSignerIndex;
2786 signerCert = CertFindCertificateInStore(store, X509_ASN_ENCODING,
2787 0, CERT_FIND_SUBJECT_CERT, signer, NULL);
2788 CryptMemFree(signer);
2789 }
2790 }
2791 else
2792 {
2793 DWORD count, size = sizeof(count);
2794
2795 if (CryptMsgGetParam(hCryptMsg, CMSG_SIGNER_COUNT_PARAM, 0, &count,
2796 &size))
2797 {
2798 for (i = 0; !signerCert && i < count; i++)
2799 {
2800 CERT_INFO *signer = CRYPT_GetSignerCertInfoFromMsg(hCryptMsg,
2801 i);
2802
2803 if (signer)
2804 {
2805 signerCert = CertFindCertificateInStore(store,
2806 X509_ASN_ENCODING, 0, CERT_FIND_SUBJECT_CERT, signer,
2807 NULL);
2808 if (signerCert)
2809 signerIndex = i;
2810 CryptMemFree(signer);
2811 }
2812 }
2813 }
2814 if (!signerCert)
2815 SetLastError(CRYPT_E_NO_TRUSTED_SIGNER);
2816 }
2817 if (signerCert)
2818 {
2819 if (!(dwFlags & CMSG_SIGNER_ONLY_FLAG))
2820 ret = CryptMsgControl(hCryptMsg, 0, CMSG_CTRL_VERIFY_SIGNATURE,
2821 signerCert->pCertInfo);
2822 else
2823 ret = TRUE;
2824 if (ret)
2825 {
2826 if (ppSigner)
2827 *ppSigner = CertDuplicateCertificateContext(signerCert);
2828 if (pdwSignerIndex)
2829 *pdwSignerIndex = signerIndex;
2830 }
2831 CertFreeCertificateContext(signerCert);
2832 }
2833
2834 CertCloseStore(store, 0);
2835 return ret;
2836 }
2837
2838 BOOL WINAPI CryptMsgVerifyCountersignatureEncodedEx(HCRYPTPROV_LEGACY hCryptProv,
2839 DWORD dwEncodingType, PBYTE pbSignerInfo, DWORD cbSignerInfo,
2840 PBYTE pbSignerInfoCountersignature, DWORD cbSignerInfoCountersignature,
2841 DWORD dwSignerType, void *pvSigner, DWORD dwFlags, void *pvReserved)
2842 {
2843 FIXME("(%08lx, %08x, %p, %d, %p, %d, %d, %p, %08x, %p): stub\n", hCryptProv,
2844 dwEncodingType, pbSignerInfo, cbSignerInfo, pbSignerInfoCountersignature,
2845 cbSignerInfoCountersignature, dwSignerType, pvSigner, dwFlags, pvReserved);
2846 return FALSE;
2847 }
useful hacks and other patches not (yet) suitable for mainline