wintrust: Use helper function for setting confidence in SoftpubCheckCert.
[wine] / dlls / wintrust / softpub.c
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 #include <stdarg.h>
19
20 #define NONAMELESSUNION
21
22 #include "windef.h"
23 #include "winbase.h"
24 #include "wintrust.h"
25 #include "mssip.h"
26 #include "softpub.h"
27 #include "wine/debug.h"
28
29 WINE_DEFAULT_DEBUG_CHANNEL(wintrust);
30
31 HRESULT WINAPI SoftpubDefCertInit(CRYPT_PROVIDER_DATA *data)
32 {
33     HRESULT ret = S_FALSE;
34
35     TRACE("(%p)\n", data);
36
37     if (data->padwTrustStepErrors &&
38      !data->padwTrustStepErrors[TRUSTERROR_STEP_FINAL_WVTINIT])
39         ret = S_OK;
40     TRACE("returning %08x\n", ret);
41     return ret;
42 }
43
44 HRESULT WINAPI SoftpubInitialize(CRYPT_PROVIDER_DATA *data)
45 {
46     HRESULT ret = S_FALSE;
47
48     TRACE("(%p)\n", data);
49
50     if (data->padwTrustStepErrors &&
51      !data->padwTrustStepErrors[TRUSTERROR_STEP_FINAL_WVTINIT])
52         ret = S_OK;
53     TRACE("returning %08x\n", ret);
54     return ret;
55 }
56
57 /* Assumes data->pWintrustData->u.pFile exists.  Makes sure a file handle is
58  * open for the file.
59  */
60 static BOOL SOFTPUB_OpenFile(CRYPT_PROVIDER_DATA *data)
61 {
62     BOOL ret = TRUE;
63
64     /* PSDK implies that all values should be initialized to NULL, so callers
65      * typically have hFile as NULL rather than INVALID_HANDLE_VALUE.  Check
66      * for both.
67      */
68     if (!data->pWintrustData->u.pFile->hFile ||
69      data->pWintrustData->u.pFile->hFile == INVALID_HANDLE_VALUE)
70     {
71         data->pWintrustData->u.pFile->hFile =
72             CreateFileW(data->pWintrustData->u.pFile->pcwszFilePath, GENERIC_READ,
73           FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
74         if (data->pWintrustData->u.pFile->hFile != INVALID_HANDLE_VALUE)
75             data->fOpenedFile = TRUE;
76         else
77             ret = FALSE;
78     }
79     if (ret)
80         GetFileTime(data->pWintrustData->u.pFile->hFile, &data->sftSystemTime,
81          NULL, NULL);
82     TRACE("returning %d\n", ret);
83     return ret;
84 }
85
86 /* Assumes data->pWintrustData->u.pFile exists.  Sets data->pPDSip->gSubject to
87  * the file's subject GUID.
88  */
89 static BOOL SOFTPUB_GetFileSubject(CRYPT_PROVIDER_DATA *data)
90 {
91     BOOL ret;
92
93     if (!data->pWintrustData->u.pFile->pgKnownSubject)
94     {
95         ret = CryptSIPRetrieveSubjectGuid(
96          data->pWintrustData->u.pFile->pcwszFilePath,
97          data->pWintrustData->u.pFile->hFile,
98          &data->u.pPDSip->gSubject);
99     }
100     else
101     {
102         data->u.pPDSip->gSubject = *data->pWintrustData->u.pFile->pgKnownSubject;
103         ret = TRUE;
104     }
105     TRACE("returning %d\n", ret);
106     return ret;
107 }
108
109 /* Assumes data->u.pPDSip exists, and its gSubject member set.
110  * Allocates data->u.pPDSip->pSip and loads it, if possible.
111  */
112 static BOOL SOFTPUB_GetSIP(CRYPT_PROVIDER_DATA *data)
113 {
114     BOOL ret;
115
116     data->u.pPDSip->pSip = data->psPfns->pfnAlloc(sizeof(SIP_DISPATCH_INFO));
117     if (data->u.pPDSip->pSip)
118         ret = CryptSIPLoad(&data->u.pPDSip->gSubject, 0, data->u.pPDSip->pSip);
119     else
120     {
121         SetLastError(ERROR_OUTOFMEMORY);
122         ret = FALSE;
123     }
124     TRACE("returning %d\n", ret);
125     return ret;
126 }
127
128 /* Assumes data->u.pPDSip has been loaded, and data->u.pPDSip->pSip allocated.
129  * Calls data->u.pPDSip->pSip->pfGet to construct data->hMsg.
130  */
131 static BOOL SOFTPUB_GetMessageFromFile(CRYPT_PROVIDER_DATA *data)
132 {
133     BOOL ret;
134     LPBYTE buf = NULL;
135     DWORD size = 0;
136
137     data->u.pPDSip->psSipSubjectInfo =
138      data->psPfns->pfnAlloc(sizeof(SIP_SUBJECTINFO));
139     if (!data->u.pPDSip->psSipSubjectInfo)
140     {
141         SetLastError(ERROR_OUTOFMEMORY);
142         return FALSE;
143     }
144
145     data->u.pPDSip->psSipSubjectInfo->cbSize = sizeof(SIP_SUBJECTINFO);
146     data->u.pPDSip->psSipSubjectInfo->pgSubjectType = &data->u.pPDSip->gSubject;
147     data->u.pPDSip->psSipSubjectInfo->hFile = data->pWintrustData->u.pFile->hFile;
148     data->u.pPDSip->psSipSubjectInfo->pwsFileName =
149      data->pWintrustData->u.pFile->pcwszFilePath;
150     data->u.pPDSip->psSipSubjectInfo->hProv = data->hProv;
151     ret = data->u.pPDSip->pSip->pfGet(data->u.pPDSip->psSipSubjectInfo,
152      &data->dwEncoding, 0, &size, 0);
153     if (!ret)
154     {
155         SetLastError(TRUST_E_NOSIGNATURE);
156         return FALSE;
157     }
158
159     buf = data->psPfns->pfnAlloc(size);
160     if (!buf)
161     {
162         SetLastError(ERROR_OUTOFMEMORY);
163         return FALSE;
164     }
165
166     ret = data->u.pPDSip->pSip->pfGet(data->u.pPDSip->psSipSubjectInfo,
167      &data->dwEncoding, 0, &size, buf);
168     if (ret)
169     {
170         data->hMsg = CryptMsgOpenToDecode(data->dwEncoding, 0, 0, data->hProv,
171          NULL, NULL);
172         if (data->hMsg)
173             ret = CryptMsgUpdate(data->hMsg, buf, size, TRUE);
174     }
175
176     data->psPfns->pfnFree(buf);
177     TRACE("returning %d\n", ret);
178     return ret;
179 }
180
181 static BOOL SOFTPUB_CreateStoreFromMessage(CRYPT_PROVIDER_DATA *data)
182 {
183     BOOL ret = FALSE;
184     HCERTSTORE store;
185
186     store = CertOpenStore(CERT_STORE_PROV_MSG, data->dwEncoding,
187      data->hProv, CERT_STORE_NO_CRYPT_RELEASE_FLAG, data->hMsg);
188     if (store)
189     {
190         ret = data->psPfns->pfnAddStore2Chain(data, store);
191         CertCloseStore(store, 0);
192     }
193     TRACE("returning %d\n", ret);
194     return ret;
195 }
196
197 static DWORD SOFTPUB_DecodeInnerContent(CRYPT_PROVIDER_DATA *data)
198 {
199     BOOL ret;
200     DWORD size;
201     LPBYTE buf = NULL;
202
203     ret = CryptMsgGetParam(data->hMsg, CMSG_INNER_CONTENT_TYPE_PARAM, 0, NULL,
204      &size);
205     if (!ret)
206         goto error;
207     buf = data->psPfns->pfnAlloc(size);
208     if (!buf)
209     {
210         SetLastError(ERROR_OUTOFMEMORY);
211         ret = FALSE;
212         goto error;
213     }
214     ret = CryptMsgGetParam(data->hMsg, CMSG_INNER_CONTENT_TYPE_PARAM, 0, buf,
215      &size);
216     if (!ret)
217         goto error;
218     if (!strcmp((LPCSTR)buf, SPC_INDIRECT_DATA_OBJID))
219     {
220         data->psPfns->pfnFree(buf);
221         buf = NULL;
222         ret = CryptMsgGetParam(data->hMsg, CMSG_CONTENT_PARAM, 0, NULL, &size);
223         if (!ret)
224             goto error;
225         buf = data->psPfns->pfnAlloc(size);
226         if (!buf)
227         {
228             SetLastError(ERROR_OUTOFMEMORY);
229             ret = FALSE;
230             goto error;
231         }
232         ret = CryptMsgGetParam(data->hMsg, CMSG_CONTENT_PARAM, 0, buf, &size);
233         if (!ret)
234             goto error;
235         ret = CryptDecodeObject(data->dwEncoding,
236          SPC_INDIRECT_DATA_CONTENT_STRUCT, buf, size, 0, NULL, &size);
237         if (!ret)
238             goto error;
239         data->u.pPDSip->psIndirectData = data->psPfns->pfnAlloc(size);
240         if (!data->u.pPDSip->psIndirectData)
241         {
242             SetLastError(ERROR_OUTOFMEMORY);
243             ret = FALSE;
244             goto error;
245         }
246         ret = CryptDecodeObject(data->dwEncoding,
247          SPC_INDIRECT_DATA_CONTENT_STRUCT, buf, size, 0,
248          data->u.pPDSip->psIndirectData, &size);
249     }
250     else
251     {
252         FIXME("unimplemented for OID %s\n", (LPCSTR)buf);
253         SetLastError(TRUST_E_SUBJECT_FORM_UNKNOWN);
254         ret = FALSE;
255     }
256
257 error:
258     TRACE("returning %d\n", ret);
259     return ret;
260 }
261
262 HRESULT WINAPI SoftpubLoadMessage(CRYPT_PROVIDER_DATA *data)
263 {
264     BOOL ret;
265
266     TRACE("(%p)\n", data);
267
268     if (!data->padwTrustStepErrors)
269         return S_FALSE;
270
271     switch (data->pWintrustData->dwUnionChoice)
272     {
273     case WTD_CHOICE_CERT:
274         if (data->pWintrustData->u.pCert &&
275          data->pWintrustData->u.pCert->cbStruct == sizeof(WINTRUST_CERT_INFO))
276         {
277             if (data->psPfns)
278             {
279                 CRYPT_PROVIDER_SGNR signer = { sizeof(signer), { 0 } };
280                 DWORD i;
281
282                 /* Add a signer with nothing but the time to verify, so we can
283                  * add a cert to it
284                  */
285                 if (data->pWintrustData->u.pCert->psftVerifyAsOf)
286                     data->sftSystemTime = signer.sftVerifyAsOf;
287                 else
288                 {
289                     SYSTEMTIME sysTime;
290
291                     GetSystemTime(&sysTime);
292                     SystemTimeToFileTime(&sysTime, &signer.sftVerifyAsOf);
293                 }
294                 ret = data->psPfns->pfnAddSgnr2Chain(data, FALSE, 0, &signer);
295                 if (!ret)
296                     goto error;
297                 ret = data->psPfns->pfnAddCert2Chain(data, 0, FALSE, 0,
298                  data->pWintrustData->u.pCert->psCertContext);
299                 if (!ret)
300                     goto error;
301                 for (i = 0; ret && i < data->pWintrustData->u.pCert->chStores;
302                  i++)
303                     ret = data->psPfns->pfnAddStore2Chain(data,
304                      data->pWintrustData->u.pCert->pahStores[i]);
305             }
306             else
307             {
308                 /* Do nothing!?  See the tests */
309                 ret = TRUE;
310             }
311         }
312         else
313         {
314             SetLastError(ERROR_INVALID_PARAMETER);
315             ret = FALSE;
316         }
317         break;
318     case WTD_CHOICE_FILE:
319         if (!data->pWintrustData->u.pFile)
320         {
321             SetLastError(ERROR_INVALID_PARAMETER);
322             ret = FALSE;
323             goto error;
324         }
325         ret = SOFTPUB_OpenFile(data);
326         if (!ret)
327             goto error;
328         ret = SOFTPUB_GetFileSubject(data);
329         if (!ret)
330             goto error;
331         ret = SOFTPUB_GetSIP(data);
332         if (!ret)
333             goto error;
334         ret = SOFTPUB_GetMessageFromFile(data);
335         if (!ret)
336             goto error;
337         ret = SOFTPUB_CreateStoreFromMessage(data);
338         if (!ret)
339             goto error;
340         ret = SOFTPUB_DecodeInnerContent(data);
341         break;
342     default:
343         FIXME("unimplemented for %d\n", data->pWintrustData->dwUnionChoice);
344         SetLastError(ERROR_INVALID_PARAMETER);
345         ret = FALSE;
346     }
347
348 error:
349     if (!ret)
350         data->padwTrustStepErrors[TRUSTERROR_STEP_FINAL_OBJPROV] =
351          GetLastError();
352     TRACE("returning %d (%08x)\n", ret ? S_OK : S_FALSE,
353      data->padwTrustStepErrors[TRUSTERROR_STEP_FINAL_OBJPROV]);
354     return ret ? S_OK : S_FALSE;
355 }
356
357 static CMSG_SIGNER_INFO *WINTRUST_GetSigner(CRYPT_PROVIDER_DATA *data,
358  DWORD signerIdx)
359 {
360     BOOL ret;
361     CMSG_SIGNER_INFO *signerInfo = NULL;
362     DWORD size;
363
364     ret = CryptMsgGetParam(data->hMsg, CMSG_SIGNER_INFO_PARAM, signerIdx,
365      NULL, &size);
366     if (ret)
367     {
368         signerInfo = data->psPfns->pfnAlloc(size);
369         if (signerInfo)
370         {
371             ret = CryptMsgGetParam(data->hMsg, CMSG_SIGNER_INFO_PARAM,
372              signerIdx, signerInfo, &size);
373             if (!ret)
374             {
375                 data->psPfns->pfnFree(signerInfo);
376                 signerInfo = NULL;
377             }
378         }
379         else
380             SetLastError(ERROR_OUTOFMEMORY);
381     }
382     return signerInfo;
383 }
384
385 static BOOL WINTRUST_SaveSigner(CRYPT_PROVIDER_DATA *data, DWORD signerIdx)
386 {
387     BOOL ret;
388     CMSG_SIGNER_INFO *signerInfo = WINTRUST_GetSigner(data, signerIdx);
389
390     if (signerInfo)
391     {
392         CRYPT_PROVIDER_SGNR sgnr = { sizeof(sgnr), { 0 } };
393
394         sgnr.psSigner = signerInfo;
395         sgnr.sftVerifyAsOf = data->sftSystemTime;
396         ret = data->psPfns->pfnAddSgnr2Chain(data, FALSE, signerIdx, &sgnr);
397     }
398     else
399         ret = FALSE;
400     return ret;
401 }
402
403 static CERT_INFO *WINTRUST_GetSignerCertInfo(CRYPT_PROVIDER_DATA *data,
404  DWORD signerIdx)
405 {
406     BOOL ret;
407     CERT_INFO *certInfo = NULL;
408     DWORD size;
409
410     ret = CryptMsgGetParam(data->hMsg, CMSG_SIGNER_CERT_INFO_PARAM, signerIdx,
411      NULL, &size);
412     if (ret)
413     {
414         certInfo = data->psPfns->pfnAlloc(size);
415         if (certInfo)
416         {
417             ret = CryptMsgGetParam(data->hMsg, CMSG_SIGNER_CERT_INFO_PARAM,
418              signerIdx, certInfo, &size);
419             if (!ret)
420             {
421                 data->psPfns->pfnFree(certInfo);
422                 certInfo = NULL;
423             }
424         }
425         else
426             SetLastError(ERROR_OUTOFMEMORY);
427     }
428     return certInfo;
429 }
430
431 static BOOL WINTRUST_VerifySigner(CRYPT_PROVIDER_DATA *data, DWORD signerIdx)
432 {
433     BOOL ret;
434     CERT_INFO *certInfo = WINTRUST_GetSignerCertInfo(data, signerIdx);
435
436     if (certInfo)
437     {
438         PCCERT_CONTEXT subject = CertGetSubjectCertificateFromStore(
439          data->pahStores[0], data->dwEncoding, certInfo);
440
441         if (subject)
442         {
443             CMSG_CTRL_VERIFY_SIGNATURE_EX_PARA para = { sizeof(para), 0,
444              signerIdx, CMSG_VERIFY_SIGNER_CERT, (LPVOID)subject };
445
446             ret = CryptMsgControl(data->hMsg, 0, CMSG_CTRL_VERIFY_SIGNATURE_EX,
447              &para);
448             if (!ret)
449                 SetLastError(TRUST_E_CERT_SIGNATURE);
450             else
451                 data->psPfns->pfnAddCert2Chain(data, signerIdx, FALSE, 0,
452                  subject);
453             CertFreeCertificateContext(subject);
454         }
455         else
456         {
457             SetLastError(TRUST_E_NO_SIGNER_CERT);
458             ret = FALSE;
459         }
460         data->psPfns->pfnFree(certInfo);
461     }
462     else
463         ret = FALSE;
464     return ret;
465 }
466
467 HRESULT WINAPI SoftpubLoadSignature(CRYPT_PROVIDER_DATA *data)
468 {
469     BOOL ret;
470
471     TRACE("(%p)\n", data);
472
473     if (!data->padwTrustStepErrors)
474         return S_FALSE;
475
476     if (data->hMsg)
477     {
478         DWORD signerCount, size;
479
480         size = sizeof(signerCount);
481         ret = CryptMsgGetParam(data->hMsg, CMSG_SIGNER_COUNT_PARAM, 0,
482          &signerCount, &size);
483         if (ret)
484         {
485             DWORD i;
486
487             for (i = 0; ret && i < signerCount; i++)
488             {
489                 if ((ret = WINTRUST_SaveSigner(data, i)))
490                     ret = WINTRUST_VerifySigner(data, i);
491             }
492         }
493         else
494             SetLastError(TRUST_E_NOSIGNATURE);
495     }
496     else
497         ret = TRUE;
498     if (!ret)
499         data->padwTrustStepErrors[TRUSTERROR_STEP_FINAL_SIGPROV] =
500          GetLastError();
501     return ret ? S_OK : S_FALSE;
502 }
503
504 static DWORD WINTRUST_TrustStatusToConfidence(DWORD errorStatus)
505 {
506     DWORD confidence = 0;
507
508     confidence = 0;
509     if (!(errorStatus & CERT_TRUST_IS_NOT_SIGNATURE_VALID))
510         confidence |= CERT_CONFIDENCE_SIG;
511     if (!(errorStatus & CERT_TRUST_IS_NOT_TIME_VALID))
512         confidence |= CERT_CONFIDENCE_TIME;
513     if (!(errorStatus & CERT_TRUST_IS_NOT_TIME_NESTED))
514         confidence |= CERT_CONFIDENCE_TIMENEST;
515     return confidence;
516 }
517
518 BOOL WINAPI SoftpubCheckCert(CRYPT_PROVIDER_DATA *data, DWORD idxSigner,
519  BOOL fCounterSignerChain, DWORD idxCounterSigner)
520 {
521     BOOL ret;
522
523     TRACE("(%p, %d, %d, %d)\n", data, idxSigner, fCounterSignerChain,
524      idxCounterSigner);
525
526     if (fCounterSignerChain)
527     {
528         FIXME("unimplemented for counter signers\n");
529         ret = FALSE;
530     }
531     else
532     {
533         PCERT_SIMPLE_CHAIN simpleChain =
534          data->pasSigners[idxSigner].pChainContext->rgpChain[0];
535         DWORD i;
536
537         ret = TRUE;
538         for (i = 0; i < simpleChain->cElement; i++)
539         {
540             /* Set confidence */
541             data->pasSigners[idxSigner].pasCertChain[i].dwConfidence =
542              WINTRUST_TrustStatusToConfidence(
543              simpleChain->rgpElement[i]->TrustStatus.dwErrorStatus);
544             /* Set additional flags */
545             if (!(simpleChain->rgpElement[i]->TrustStatus.dwErrorStatus &
546              CERT_TRUST_IS_UNTRUSTED_ROOT))
547                 data->pasSigners[idxSigner].pasCertChain[i].fTrustedRoot = TRUE;
548             if (simpleChain->rgpElement[i]->TrustStatus.dwInfoStatus &
549              CERT_TRUST_IS_SELF_SIGNED)
550                 data->pasSigners[idxSigner].pasCertChain[i].fSelfSigned = TRUE;
551             if (simpleChain->rgpElement[i]->TrustStatus.dwErrorStatus &
552              CERT_TRUST_IS_CYCLIC)
553                 data->pasSigners[idxSigner].pasCertChain[i].fIsCyclic = TRUE;
554         }
555     }
556     return ret;
557 }
558
559 static DWORD WINTRUST_TrustStatusToError(DWORD errorStatus)
560 {
561     DWORD error;
562
563     if (errorStatus & CERT_TRUST_IS_NOT_SIGNATURE_VALID)
564         error = TRUST_E_CERT_SIGNATURE;
565     else if (errorStatus & CERT_TRUST_IS_UNTRUSTED_ROOT)
566         error = CERT_E_UNTRUSTEDROOT;
567     else if (errorStatus & CERT_TRUST_IS_NOT_TIME_VALID)
568         error = CERT_E_EXPIRED;
569     else if (errorStatus & CERT_TRUST_IS_NOT_TIME_NESTED)
570         error = CERT_E_VALIDITYPERIODNESTING;
571     else if (errorStatus & CERT_TRUST_IS_REVOKED)
572         error = CERT_E_REVOKED;
573     else if (errorStatus & CERT_TRUST_IS_OFFLINE_REVOCATION ||
574      errorStatus & CERT_TRUST_REVOCATION_STATUS_UNKNOWN)
575         error = CERT_E_REVOCATION_FAILURE;
576     else if (errorStatus & CERT_TRUST_IS_NOT_VALID_FOR_USAGE)
577         error = CERT_E_WRONG_USAGE;
578     else if (errorStatus & CERT_TRUST_IS_CYCLIC)
579         error = CERT_E_CHAINING;
580     else if (errorStatus & CERT_TRUST_INVALID_EXTENSION)
581         error = CERT_E_CRITICAL;
582     else if (errorStatus & CERT_TRUST_INVALID_POLICY_CONSTRAINTS)
583         error = CERT_E_INVALID_POLICY;
584     else if (errorStatus & CERT_TRUST_INVALID_BASIC_CONSTRAINTS)
585         error = TRUST_E_BASIC_CONSTRAINTS;
586     else if (errorStatus & CERT_TRUST_INVALID_NAME_CONSTRAINTS ||
587      errorStatus & CERT_TRUST_HAS_NOT_SUPPORTED_NAME_CONSTRAINT ||
588      errorStatus & CERT_TRUST_HAS_NOT_DEFINED_NAME_CONSTRAINT ||
589      errorStatus & CERT_TRUST_HAS_NOT_PERMITTED_NAME_CONSTRAINT ||
590      errorStatus & CERT_TRUST_HAS_EXCLUDED_NAME_CONSTRAINT)
591         error = CERT_E_INVALID_NAME;
592     else if (errorStatus & CERT_TRUST_NO_ISSUANCE_CHAIN_POLICY)
593         error = CERT_E_INVALID_POLICY;
594     else if (errorStatus)
595     {
596         FIXME("unknown error status %08x\n", errorStatus);
597         error = TRUST_E_SYSTEM_ERROR;
598     }
599     else
600         error = S_OK;
601     return error;
602 }
603
604 static BOOL WINTRUST_CopyChain(CRYPT_PROVIDER_DATA *data, DWORD signerIdx)
605 {
606     BOOL ret;
607     PCERT_SIMPLE_CHAIN simpleChain =
608      data->pasSigners[signerIdx].pChainContext->rgpChain[0];
609     DWORD i;
610
611     data->pasSigners[signerIdx].pasCertChain[0].dwConfidence =
612      WINTRUST_TrustStatusToConfidence(
613      simpleChain->rgpElement[0]->TrustStatus.dwErrorStatus);
614     data->pasSigners[signerIdx].pasCertChain[0].dwError =
615      WINTRUST_TrustStatusToError(
616      simpleChain->rgpElement[0]->TrustStatus.dwErrorStatus);
617     data->pasSigners[signerIdx].pasCertChain[0].pChainElement =
618      simpleChain->rgpElement[0];
619     ret = TRUE;
620     for (i = 1; ret && i < simpleChain->cElement; i++)
621     {
622         ret = data->psPfns->pfnAddCert2Chain(data, signerIdx, FALSE, 0,
623          simpleChain->rgpElement[i]->pCertContext);
624         if (ret)
625         {
626             data->pasSigners[signerIdx].pasCertChain[i].pChainElement =
627              simpleChain->rgpElement[i];
628             data->pasSigners[signerIdx].pasCertChain[i].dwConfidence =
629              WINTRUST_TrustStatusToConfidence(
630              simpleChain->rgpElement[i]->TrustStatus.dwErrorStatus);
631             data->pasSigners[signerIdx].pasCertChain[i].dwError =
632              WINTRUST_TrustStatusToError(
633              simpleChain->rgpElement[i]->TrustStatus.dwErrorStatus);
634         }
635     }
636     return ret;
637 }
638
639 static void WINTRUST_CreateChainPolicyCreateInfo(
640  const CRYPT_PROVIDER_DATA *data, PWTD_GENERIC_CHAIN_POLICY_CREATE_INFO info,
641  PCERT_CHAIN_PARA chainPara)
642 {
643     chainPara->cbSize = sizeof(CERT_CHAIN_PARA);
644     if (data->pRequestUsage)
645         chainPara->RequestedUsage = *data->pRequestUsage;
646     else
647     {
648         chainPara->RequestedUsage.dwType = 0;
649         chainPara->RequestedUsage.Usage.cUsageIdentifier = 0;
650     }
651     info->u.cbSize = sizeof(WTD_GENERIC_CHAIN_POLICY_CREATE_INFO);
652     info->hChainEngine = NULL;
653     info->pChainPara = chainPara;
654     if (data->dwProvFlags & CPD_REVOCATION_CHECK_END_CERT)
655         info->dwFlags = CERT_CHAIN_REVOCATION_CHECK_END_CERT;
656     else if (data->dwProvFlags & CPD_REVOCATION_CHECK_CHAIN)
657         info->dwFlags = CERT_CHAIN_REVOCATION_CHECK_CHAIN;
658     else if (data->dwProvFlags & CPD_REVOCATION_CHECK_CHAIN_EXCLUDE_ROOT)
659         info->dwFlags = CERT_CHAIN_REVOCATION_CHECK_CHAIN_EXCLUDE_ROOT;
660     else
661         info->dwFlags = 0;
662     info->pvReserved = NULL;
663 }
664
665 static BOOL WINTRUST_CreateChainForSigner(CRYPT_PROVIDER_DATA *data,
666  DWORD signer, PWTD_GENERIC_CHAIN_POLICY_CREATE_INFO createInfo,
667  PCERT_CHAIN_PARA chainPara)
668 {
669     BOOL ret = TRUE;
670     HCERTSTORE store = NULL;
671
672     if (data->chStores)
673     {
674         store = CertOpenStore(CERT_STORE_PROV_COLLECTION, 0, 0,
675          CERT_STORE_CREATE_NEW_FLAG, NULL);
676         if (store)
677         {
678             DWORD i;
679
680             for (i = 0; i < data->chStores; i++)
681                 CertAddStoreToCollection(store, data->pahStores[i], 0, 0);
682         }
683     }
684     /* Expect the end certificate for each signer to be the only cert in the
685      * chain:
686      */
687     if (data->pasSigners[signer].csCertChain)
688     {
689         /* Create a certificate chain for each signer */
690         ret = CertGetCertificateChain(createInfo->hChainEngine,
691          data->pasSigners[signer].pasCertChain[0].pCert,
692          &data->pasSigners[signer].sftVerifyAsOf, store,
693          chainPara, createInfo->dwFlags, createInfo->pvReserved,
694          &data->pasSigners[signer].pChainContext);
695         if (ret)
696         {
697             if (data->pasSigners[signer].pChainContext->cChain != 1)
698             {
699                 FIXME("unimplemented for more than 1 simple chain\n");
700                 ret = FALSE;
701             }
702             else
703             {
704                 if ((ret = WINTRUST_CopyChain(data, signer)))
705                 {
706                     if (data->psPfns->pfnCertCheckPolicy)
707                         ret = data->psPfns->pfnCertCheckPolicy(data, signer,
708                          FALSE, 0);
709                     else
710                         TRACE("no cert check policy, skipping policy check\n");
711                 }
712             }
713         }
714     }
715     CertCloseStore(store, 0);
716     return ret;
717 }
718
719 HRESULT WINAPI WintrustCertificateTrust(CRYPT_PROVIDER_DATA *data)
720 {
721     BOOL ret;
722
723     TRACE("(%p)\n", data);
724
725     if (!data->csSigners)
726     {
727         ret = FALSE;
728         SetLastError(TRUST_E_NOSIGNATURE);
729     }
730     else
731     {
732         DWORD i;
733         WTD_GENERIC_CHAIN_POLICY_CREATE_INFO createInfo;
734         CERT_CHAIN_PARA chainPara;
735
736         WINTRUST_CreateChainPolicyCreateInfo(data, &createInfo, &chainPara);
737         ret = TRUE;
738         for (i = 0; i < data->csSigners; i++)
739             ret = WINTRUST_CreateChainForSigner(data, i, &createInfo,
740              &chainPara);
741     }
742     if (!ret)
743         data->padwTrustStepErrors[TRUSTERROR_STEP_FINAL_CERTPROV] =
744          GetLastError();
745     TRACE("returning %d (%08x)\n", ret ? S_OK : S_FALSE,
746      data->padwTrustStepErrors[TRUSTERROR_STEP_FINAL_CERTPROV]);
747     return ret ? S_OK : S_FALSE;
748 }
749
750 HRESULT WINAPI GenericChainCertificateTrust(CRYPT_PROVIDER_DATA *data)
751 {
752     BOOL ret;
753     WTD_GENERIC_CHAIN_POLICY_DATA *policyData =
754      (WTD_GENERIC_CHAIN_POLICY_DATA *)data->pWintrustData->pPolicyCallbackData;
755
756     TRACE("(%p)\n", data);
757
758     if (policyData && policyData->u.cbSize !=
759      sizeof(WTD_GENERIC_CHAIN_POLICY_CREATE_INFO))
760     {
761         SetLastError(ERROR_INVALID_PARAMETER);
762         ret = FALSE;
763         goto end;
764     }
765     if (!data->csSigners)
766     {
767         ret = FALSE;
768         SetLastError(TRUST_E_NOSIGNATURE);
769     }
770     else
771     {
772         DWORD i;
773         WTD_GENERIC_CHAIN_POLICY_CREATE_INFO createInfo, *pCreateInfo;
774         CERT_CHAIN_PARA chainPara, *pChainPara;
775
776         if (policyData)
777         {
778             pCreateInfo = policyData->pSignerChainInfo;
779             pChainPara = pCreateInfo->pChainPara;
780         }
781         else
782         {
783             WINTRUST_CreateChainPolicyCreateInfo(data, &createInfo, &chainPara);
784             pChainPara = &chainPara;
785             pCreateInfo = &createInfo;
786         }
787         ret = TRUE;
788         for (i = 0; i < data->csSigners; i++)
789             ret = WINTRUST_CreateChainForSigner(data, i, pCreateInfo,
790              pChainPara);
791     }
792
793 end:
794     if (!ret)
795         data->padwTrustStepErrors[TRUSTERROR_STEP_FINAL_CERTPROV] =
796          GetLastError();
797     TRACE("returning %d (%08x)\n", ret ? S_OK : S_FALSE,
798      data->padwTrustStepErrors[TRUSTERROR_STEP_FINAL_CERTPROV]);
799     return ret ? S_OK : S_FALSE;
800 }
801
802 HRESULT WINAPI SoftpubAuthenticode(CRYPT_PROVIDER_DATA *data)
803 {
804     BOOL ret;
805     CERT_CHAIN_POLICY_STATUS policyStatus = { sizeof(policyStatus), 0 };
806
807     TRACE("(%p)\n", data);
808
809     if (data->pWintrustData->dwUIChoice != WTD_UI_NONE)
810         FIXME("unimplemented for UI choice %d\n",
811          data->pWintrustData->dwUIChoice);
812     if (!data->csSigners)
813     {
814         ret = FALSE;
815         policyStatus.dwError = TRUST_E_NOSIGNATURE;
816     }
817     else
818     {
819         DWORD i;
820
821         ret = TRUE;
822         for (i = 0; ret && i < data->csSigners; i++)
823         {
824             BYTE hash[20];
825             DWORD size = sizeof(hash);
826
827             /* First make sure cert isn't disallowed */
828             if ((ret = CertGetCertificateContextProperty(
829              data->pasSigners[i].pasCertChain[0].pCert,
830              CERT_SIGNATURE_HASH_PROP_ID, hash, &size)))
831             {
832                 static const WCHAR disallowedW[] =
833                  { 'D','i','s','a','l','l','o','w','e','d',0 };
834                 HCERTSTORE disallowed = CertOpenStore(CERT_STORE_PROV_SYSTEM_W,
835                  X509_ASN_ENCODING, 0, CERT_SYSTEM_STORE_CURRENT_USER,
836                  disallowedW);
837
838                 if (disallowed)
839                 {
840                     PCCERT_CONTEXT found = CertFindCertificateInStore(
841                      disallowed, X509_ASN_ENCODING, 0, CERT_FIND_SIGNATURE_HASH,
842                      hash, NULL);
843
844                     if (found)
845                     {
846                         /* Disallowed!  Can't verify it. */
847                         policyStatus.dwError = TRUST_E_SUBJECT_NOT_TRUSTED;
848                         ret = FALSE;
849                         CertFreeCertificateContext(found);
850                     }
851                     CertCloseStore(disallowed, 0);
852                 }
853             }
854             if (ret)
855             {
856                 CERT_CHAIN_POLICY_PARA policyPara = { sizeof(policyPara), 0 };
857
858                 if (data->dwRegPolicySettings & WTPF_TRUSTTEST)
859                     policyPara.dwFlags |= CERT_CHAIN_POLICY_TRUST_TESTROOT_FLAG;
860                 if (data->dwRegPolicySettings & WTPF_TESTCANBEVALID)
861                     policyPara.dwFlags |= CERT_CHAIN_POLICY_ALLOW_TESTROOT_FLAG;
862                 if (data->dwRegPolicySettings & WTPF_IGNOREEXPIRATION)
863                     policyPara.dwFlags |=
864                      CERT_CHAIN_POLICY_IGNORE_NOT_TIME_VALID_FLAG |
865                      CERT_CHAIN_POLICY_IGNORE_CTL_NOT_TIME_VALID_FLAG |
866                      CERT_CHAIN_POLICY_IGNORE_NOT_TIME_NESTED_FLAG;
867                 if (data->dwRegPolicySettings & WTPF_IGNOREREVOKATION)
868                     policyPara.dwFlags |=
869                      CERT_CHAIN_POLICY_IGNORE_END_REV_UNKNOWN_FLAG |
870                      CERT_CHAIN_POLICY_IGNORE_CTL_SIGNER_REV_UNKNOWN_FLAG |
871                      CERT_CHAIN_POLICY_IGNORE_CA_REV_UNKNOWN_FLAG |
872                      CERT_CHAIN_POLICY_IGNORE_ROOT_REV_UNKNOWN_FLAG;
873                 CertVerifyCertificateChainPolicy(CERT_CHAIN_POLICY_AUTHENTICODE,
874                  data->pasSigners[i].pChainContext, &policyPara, &policyStatus);
875                 if (policyStatus.dwError != NO_ERROR)
876                     ret = FALSE;
877             }
878         }
879     }
880     if (!ret)
881         data->padwTrustStepErrors[TRUSTERROR_STEP_FINAL_POLICYPROV] =
882          policyStatus.dwError;
883     TRACE("returning %d (%08x)\n", ret ? S_OK : S_FALSE,
884      data->padwTrustStepErrors[TRUSTERROR_STEP_FINAL_POLICYPROV]);
885     return ret ? S_OK : S_FALSE;
886 }
887
888 static HRESULT WINAPI WINTRUST_DefaultPolicy(CRYPT_PROVIDER_DATA *pProvData,
889  DWORD dwStepError, DWORD dwRegPolicySettings, DWORD cSigner,
890  PWTD_GENERIC_CHAIN_POLICY_SIGNER_INFO rgpSigner, void *pvPolicyArg)
891 {
892     DWORD i;
893     CERT_CHAIN_POLICY_STATUS policyStatus = { sizeof(policyStatus), 0 };
894
895     for (i = 0; !policyStatus.dwError && i < cSigner; i++)
896     {
897         CERT_CHAIN_POLICY_PARA policyPara = { sizeof(policyPara), 0 };
898
899         if (dwRegPolicySettings & WTPF_IGNOREEXPIRATION)
900             policyPara.dwFlags |=
901              CERT_CHAIN_POLICY_IGNORE_NOT_TIME_VALID_FLAG |
902              CERT_CHAIN_POLICY_IGNORE_CTL_NOT_TIME_VALID_FLAG |
903              CERT_CHAIN_POLICY_IGNORE_NOT_TIME_NESTED_FLAG;
904         if (dwRegPolicySettings & WTPF_IGNOREREVOKATION)
905             policyPara.dwFlags |=
906              CERT_CHAIN_POLICY_IGNORE_END_REV_UNKNOWN_FLAG |
907              CERT_CHAIN_POLICY_IGNORE_CTL_SIGNER_REV_UNKNOWN_FLAG |
908              CERT_CHAIN_POLICY_IGNORE_CA_REV_UNKNOWN_FLAG |
909              CERT_CHAIN_POLICY_IGNORE_ROOT_REV_UNKNOWN_FLAG;
910         CertVerifyCertificateChainPolicy(CERT_CHAIN_POLICY_BASE,
911          rgpSigner[i].pChainContext, &policyPara, &policyStatus);
912     }
913     return policyStatus.dwError;
914 }
915
916 HRESULT WINAPI GenericChainFinalProv(CRYPT_PROVIDER_DATA *data)
917 {
918     HRESULT err = NO_ERROR; /* not a typo, MS confused the types */
919     WTD_GENERIC_CHAIN_POLICY_DATA *policyData =
920      (WTD_GENERIC_CHAIN_POLICY_DATA *)data->pWintrustData->pPolicyCallbackData;
921
922     TRACE("(%p)\n", data);
923
924     if (data->pWintrustData->dwUIChoice != WTD_UI_NONE)
925         FIXME("unimplemented for UI choice %d\n",
926          data->pWintrustData->dwUIChoice);
927     if (!data->csSigners)
928         err = TRUST_E_NOSIGNATURE;
929     else
930     {
931         PFN_WTD_GENERIC_CHAIN_POLICY_CALLBACK policyCallback;
932         void *policyArg;
933         WTD_GENERIC_CHAIN_POLICY_SIGNER_INFO *signers = NULL;
934
935         if (policyData)
936         {
937             policyCallback = policyData->pfnPolicyCallback;
938             policyArg = policyData->pvPolicyArg;
939         }
940         else
941         {
942             policyCallback = WINTRUST_DefaultPolicy;
943             policyArg = NULL;
944         }
945         if (data->csSigners)
946         {
947             DWORD i;
948
949             signers = data->psPfns->pfnAlloc(
950              data->csSigners * sizeof(WTD_GENERIC_CHAIN_POLICY_SIGNER_INFO));
951             if (signers)
952             {
953                 for (i = 0; i < data->csSigners; i++)
954                 {
955                     signers[i].u.cbSize =
956                      sizeof(WTD_GENERIC_CHAIN_POLICY_SIGNER_INFO);
957                     signers[i].pChainContext =
958                      data->pasSigners[i].pChainContext;
959                     signers[i].dwSignerType = data->pasSigners[i].dwSignerType;
960                     signers[i].pMsgSignerInfo = data->pasSigners[i].psSigner;
961                     signers[i].dwError = data->pasSigners[i].dwError;
962                     if (data->pasSigners[i].csCounterSigners)
963                         FIXME("unimplemented for counter signers\n");
964                     signers[i].cCounterSigner = 0;
965                     signers[i].rgpCounterSigner = NULL;
966                 }
967             }
968             else
969                 err = ERROR_OUTOFMEMORY;
970         }
971         if (err == NO_ERROR)
972             err = policyCallback(data, TRUSTERROR_STEP_FINAL_POLICYPROV,
973              data->dwRegPolicySettings, data->csSigners, signers, policyArg);
974         data->psPfns->pfnFree(signers);
975     }
976     if (err != NO_ERROR)
977         data->padwTrustStepErrors[TRUSTERROR_STEP_FINAL_POLICYPROV] = err;
978     TRACE("returning %d (%08x)\n", err == NO_ERROR ? S_OK : S_FALSE,
979      data->padwTrustStepErrors[TRUSTERROR_STEP_FINAL_POLICYPROV]);
980     return err == NO_ERROR ? S_OK : S_FALSE;
981 }
982
983 HRESULT WINAPI SoftpubCleanup(CRYPT_PROVIDER_DATA *data)
984 {
985     DWORD i, j;
986
987     for (i = 0; i < data->csSigners; i++)
988     {
989         for (j = 0; j < data->pasSigners[i].csCertChain; j++)
990             CertFreeCertificateContext(data->pasSigners[i].pasCertChain[j].pCert);
991         data->psPfns->pfnFree(data->pasSigners[i].pasCertChain);
992         data->psPfns->pfnFree(data->pasSigners[i].psSigner);
993         CertFreeCertificateChain(data->pasSigners[i].pChainContext);
994     }
995     data->psPfns->pfnFree(data->pasSigners);
996
997     for (i = 0; i < data->chStores; i++)
998         CertCloseStore(data->pahStores[i], 0);
999     data->psPfns->pfnFree(data->pahStores);
1000
1001     if (data->u.pPDSip)
1002     {
1003         data->psPfns->pfnFree(data->u.pPDSip->pSip);
1004         data->psPfns->pfnFree(data->u.pPDSip->pCATSip);
1005         data->psPfns->pfnFree(data->u.pPDSip->psSipSubjectInfo);
1006         data->psPfns->pfnFree(data->u.pPDSip->psSipCATSubjectInfo);
1007         data->psPfns->pfnFree(data->u.pPDSip->psIndirectData);
1008     }
1009
1010     CryptMsgClose(data->hMsg);
1011
1012     if (data->fOpenedFile)
1013         CloseHandle(data->pWintrustData->u.pFile->hFile);
1014
1015     return S_OK;
1016 }