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