shell32/tests: Fix shlexec on Vista and higher.
[wine] / dlls / crypt32 / chain.c
1 /*
2  * Copyright 2006 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 <stdarg.h>
20 #define NONAMELESSUNION
21 #include "windef.h"
22 #include "winbase.h"
23 #define CERT_CHAIN_PARA_HAS_EXTRA_FIELDS
24 #define CERT_REVOCATION_PARA_HAS_EXTRA_FIELDS
25 #include "wincrypt.h"
26 #include "wine/debug.h"
27 #include "wine/unicode.h"
28 #include "crypt32_private.h"
29
30 WINE_DEFAULT_DEBUG_CHANNEL(crypt);
31 WINE_DECLARE_DEBUG_CHANNEL(chain);
32
33 #define DEFAULT_CYCLE_MODULUS 7
34
35 static HCERTCHAINENGINE CRYPT_defaultChainEngine;
36
37 /* This represents a subset of a certificate chain engine:  it doesn't include
38  * the "hOther" store described by MSDN, because I'm not sure how that's used.
39  * It also doesn't include the "hTrust" store, because I don't yet implement
40  * CTLs or complex certificate chains.
41  */
42 typedef struct _CertificateChainEngine
43 {
44     LONG       ref;
45     HCERTSTORE hRoot;
46     HCERTSTORE hWorld;
47     DWORD      dwFlags;
48     DWORD      dwUrlRetrievalTimeout;
49     DWORD      MaximumCachedCertificates;
50     DWORD      CycleDetectionModulus;
51 } CertificateChainEngine, *PCertificateChainEngine;
52
53 static inline void CRYPT_AddStoresToCollection(HCERTSTORE collection,
54  DWORD cStores, HCERTSTORE *stores)
55 {
56     DWORD i;
57
58     for (i = 0; i < cStores; i++)
59         CertAddStoreToCollection(collection, stores[i], 0, 0);
60 }
61
62 static inline void CRYPT_CloseStores(DWORD cStores, HCERTSTORE *stores)
63 {
64     DWORD i;
65
66     for (i = 0; i < cStores; i++)
67         CertCloseStore(stores[i], 0);
68 }
69
70 static const WCHAR rootW[] = { 'R','o','o','t',0 };
71
72 static BOOL CRYPT_CheckRestrictedRoot(HCERTSTORE store)
73 {
74     BOOL ret = TRUE;
75
76     if (store)
77     {
78         HCERTSTORE rootStore = CertOpenSystemStoreW(0, rootW);
79         PCCERT_CONTEXT cert = NULL, check;
80         BYTE hash[20];
81         DWORD size;
82
83         do {
84             cert = CertEnumCertificatesInStore(store, cert);
85             if (cert)
86             {
87                 size = sizeof(hash);
88
89                 ret = CertGetCertificateContextProperty(cert, CERT_HASH_PROP_ID,
90                  hash, &size);
91                 if (ret)
92                 {
93                     CRYPT_HASH_BLOB blob = { sizeof(hash), hash };
94
95                     check = CertFindCertificateInStore(rootStore,
96                      cert->dwCertEncodingType, 0, CERT_FIND_SHA1_HASH, &blob,
97                      NULL);
98                     if (!check)
99                         ret = FALSE;
100                     else
101                         CertFreeCertificateContext(check);
102                 }
103             }
104         } while (ret && cert);
105         if (cert)
106             CertFreeCertificateContext(cert);
107         CertCloseStore(rootStore, 0);
108     }
109     return ret;
110 }
111
112 HCERTCHAINENGINE CRYPT_CreateChainEngine(HCERTSTORE root,
113  PCERT_CHAIN_ENGINE_CONFIG pConfig)
114 {
115     static const WCHAR caW[] = { 'C','A',0 };
116     static const WCHAR myW[] = { 'M','y',0 };
117     static const WCHAR trustW[] = { 'T','r','u','s','t',0 };
118     PCertificateChainEngine engine =
119      CryptMemAlloc(sizeof(CertificateChainEngine));
120
121     if (engine)
122     {
123         HCERTSTORE worldStores[4];
124
125         engine->ref = 1;
126         engine->hRoot = root;
127         engine->hWorld = CertOpenStore(CERT_STORE_PROV_COLLECTION, 0, 0,
128          CERT_STORE_CREATE_NEW_FLAG, NULL);
129         worldStores[0] = CertDuplicateStore(engine->hRoot);
130         worldStores[1] = CertOpenSystemStoreW(0, caW);
131         worldStores[2] = CertOpenSystemStoreW(0, myW);
132         worldStores[3] = CertOpenSystemStoreW(0, trustW);
133         CRYPT_AddStoresToCollection(engine->hWorld,
134          sizeof(worldStores) / sizeof(worldStores[0]), worldStores);
135         CRYPT_AddStoresToCollection(engine->hWorld,
136          pConfig->cAdditionalStore, pConfig->rghAdditionalStore);
137         CRYPT_CloseStores(sizeof(worldStores) / sizeof(worldStores[0]),
138          worldStores);
139         engine->dwFlags = pConfig->dwFlags;
140         engine->dwUrlRetrievalTimeout = pConfig->dwUrlRetrievalTimeout;
141         engine->MaximumCachedCertificates =
142          pConfig->MaximumCachedCertificates;
143         if (pConfig->CycleDetectionModulus)
144             engine->CycleDetectionModulus = pConfig->CycleDetectionModulus;
145         else
146             engine->CycleDetectionModulus = DEFAULT_CYCLE_MODULUS;
147     }
148     return engine;
149 }
150
151 BOOL WINAPI CertCreateCertificateChainEngine(PCERT_CHAIN_ENGINE_CONFIG pConfig,
152  HCERTCHAINENGINE *phChainEngine)
153 {
154     BOOL ret;
155
156     TRACE("(%p, %p)\n", pConfig, phChainEngine);
157
158     if (pConfig->cbSize != sizeof(*pConfig))
159     {
160         SetLastError(E_INVALIDARG);
161         return FALSE;
162     }
163     *phChainEngine = NULL;
164     ret = CRYPT_CheckRestrictedRoot(pConfig->hRestrictedRoot);
165     if (ret)
166     {
167         HCERTSTORE root;
168         HCERTCHAINENGINE engine;
169
170         if (pConfig->hRestrictedRoot)
171             root = CertDuplicateStore(pConfig->hRestrictedRoot);
172         else
173             root = CertOpenSystemStoreW(0, rootW);
174         engine = CRYPT_CreateChainEngine(root, pConfig);
175         if (engine)
176         {
177             *phChainEngine = engine;
178             ret = TRUE;
179         }
180         else
181             ret = FALSE;
182     }
183     return ret;
184 }
185
186 VOID WINAPI CertFreeCertificateChainEngine(HCERTCHAINENGINE hChainEngine)
187 {
188     PCertificateChainEngine engine = (PCertificateChainEngine)hChainEngine;
189
190     TRACE("(%p)\n", hChainEngine);
191
192     if (engine && InterlockedDecrement(&engine->ref) == 0)
193     {
194         CertCloseStore(engine->hWorld, 0);
195         CertCloseStore(engine->hRoot, 0);
196         CryptMemFree(engine);
197     }
198 }
199
200 static HCERTCHAINENGINE CRYPT_GetDefaultChainEngine(void)
201 {
202     if (!CRYPT_defaultChainEngine)
203     {
204         CERT_CHAIN_ENGINE_CONFIG config = { 0 };
205         HCERTCHAINENGINE engine;
206
207         config.cbSize = sizeof(config);
208         CertCreateCertificateChainEngine(&config, &engine);
209         InterlockedCompareExchangePointer(&CRYPT_defaultChainEngine, engine,
210          NULL);
211         if (CRYPT_defaultChainEngine != engine)
212             CertFreeCertificateChainEngine(engine);
213     }
214     return CRYPT_defaultChainEngine;
215 }
216
217 void default_chain_engine_free(void)
218 {
219     CertFreeCertificateChainEngine(CRYPT_defaultChainEngine);
220 }
221
222 typedef struct _CertificateChain
223 {
224     CERT_CHAIN_CONTEXT context;
225     HCERTSTORE world;
226     LONG ref;
227 } CertificateChain, *PCertificateChain;
228
229 static inline BOOL CRYPT_IsCertificateSelfSigned(PCCERT_CONTEXT cert)
230 {
231     return CertCompareCertificateName(cert->dwCertEncodingType,
232      &cert->pCertInfo->Subject, &cert->pCertInfo->Issuer);
233 }
234
235 static void CRYPT_FreeChainElement(PCERT_CHAIN_ELEMENT element)
236 {
237     CertFreeCertificateContext(element->pCertContext);
238     CryptMemFree(element);
239 }
240
241 static void CRYPT_CheckSimpleChainForCycles(PCERT_SIMPLE_CHAIN chain)
242 {
243     DWORD i, j, cyclicCertIndex = 0;
244
245     /* O(n^2) - I don't think there's a faster way */
246     for (i = 0; !cyclicCertIndex && i < chain->cElement; i++)
247         for (j = i + 1; !cyclicCertIndex && j < chain->cElement; j++)
248             if (CertCompareCertificate(X509_ASN_ENCODING,
249              chain->rgpElement[i]->pCertContext->pCertInfo,
250              chain->rgpElement[j]->pCertContext->pCertInfo))
251                 cyclicCertIndex = j;
252     if (cyclicCertIndex)
253     {
254         chain->rgpElement[cyclicCertIndex]->TrustStatus.dwErrorStatus
255          |= CERT_TRUST_IS_CYCLIC | CERT_TRUST_INVALID_BASIC_CONSTRAINTS;
256         /* Release remaining certs */
257         for (i = cyclicCertIndex + 1; i < chain->cElement; i++)
258             CRYPT_FreeChainElement(chain->rgpElement[i]);
259         /* Truncate chain */
260         chain->cElement = cyclicCertIndex + 1;
261     }
262 }
263
264 /* Checks whether the chain is cyclic by examining the last element's status */
265 static inline BOOL CRYPT_IsSimpleChainCyclic(const CERT_SIMPLE_CHAIN *chain)
266 {
267     if (chain->cElement)
268         return chain->rgpElement[chain->cElement - 1]->TrustStatus.dwErrorStatus
269          & CERT_TRUST_IS_CYCLIC;
270     else
271         return FALSE;
272 }
273
274 static inline void CRYPT_CombineTrustStatus(CERT_TRUST_STATUS *chainStatus,
275  const CERT_TRUST_STATUS *elementStatus)
276 {
277     /* Any error that applies to an element also applies to a chain.. */
278     chainStatus->dwErrorStatus |= elementStatus->dwErrorStatus;
279     /* but the bottom nibble of an element's info status doesn't apply to the
280      * chain.
281      */
282     chainStatus->dwInfoStatus |= (elementStatus->dwInfoStatus & 0xfffffff0);
283 }
284
285 static BOOL CRYPT_AddCertToSimpleChain(const CertificateChainEngine *engine,
286  PCERT_SIMPLE_CHAIN chain, PCCERT_CONTEXT cert, DWORD subjectInfoStatus)
287 {
288     BOOL ret = FALSE;
289     PCERT_CHAIN_ELEMENT element = CryptMemAlloc(sizeof(CERT_CHAIN_ELEMENT));
290
291     if (element)
292     {
293         if (!chain->cElement)
294             chain->rgpElement = CryptMemAlloc(sizeof(PCERT_CHAIN_ELEMENT));
295         else
296             chain->rgpElement = CryptMemRealloc(chain->rgpElement,
297              (chain->cElement + 1) * sizeof(PCERT_CHAIN_ELEMENT));
298         if (chain->rgpElement)
299         {
300             chain->rgpElement[chain->cElement++] = element;
301             memset(element, 0, sizeof(CERT_CHAIN_ELEMENT));
302             element->cbSize = sizeof(CERT_CHAIN_ELEMENT);
303             element->pCertContext = CertDuplicateCertificateContext(cert);
304             if (chain->cElement > 1)
305                 chain->rgpElement[chain->cElement - 2]->TrustStatus.dwInfoStatus
306                  = subjectInfoStatus;
307             /* FIXME: initialize the rest of element */
308             if (!(chain->cElement % engine->CycleDetectionModulus))
309             {
310                 CRYPT_CheckSimpleChainForCycles(chain);
311                 /* Reinitialize the element pointer in case the chain is
312                  * cyclic, in which case the chain is truncated.
313                  */
314                 element = chain->rgpElement[chain->cElement - 1];
315             }
316             CRYPT_CombineTrustStatus(&chain->TrustStatus,
317              &element->TrustStatus);
318             ret = TRUE;
319         }
320         else
321             CryptMemFree(element);
322     }
323     return ret;
324 }
325
326 static void CRYPT_FreeSimpleChain(PCERT_SIMPLE_CHAIN chain)
327 {
328     DWORD i;
329
330     for (i = 0; i < chain->cElement; i++)
331         CRYPT_FreeChainElement(chain->rgpElement[i]);
332     CryptMemFree(chain->rgpElement);
333     CryptMemFree(chain);
334 }
335
336 static void CRYPT_CheckTrustedStatus(HCERTSTORE hRoot,
337  PCERT_CHAIN_ELEMENT rootElement)
338 {
339     BYTE hash[20];
340     DWORD size = sizeof(hash);
341     CRYPT_HASH_BLOB blob = { sizeof(hash), hash };
342     PCCERT_CONTEXT trustedRoot;
343
344     CertGetCertificateContextProperty(rootElement->pCertContext,
345      CERT_HASH_PROP_ID, hash, &size);
346     trustedRoot = CertFindCertificateInStore(hRoot,
347      rootElement->pCertContext->dwCertEncodingType, 0, CERT_FIND_SHA1_HASH,
348      &blob, NULL);
349     if (!trustedRoot)
350         rootElement->TrustStatus.dwErrorStatus |=
351          CERT_TRUST_IS_UNTRUSTED_ROOT;
352     else
353         CertFreeCertificateContext(trustedRoot);
354 }
355
356 static void CRYPT_CheckRootCert(HCERTCHAINENGINE hRoot,
357  PCERT_CHAIN_ELEMENT rootElement)
358 {
359     PCCERT_CONTEXT root = rootElement->pCertContext;
360
361     if (!CryptVerifyCertificateSignatureEx(0, root->dwCertEncodingType,
362      CRYPT_VERIFY_CERT_SIGN_SUBJECT_CERT, (void *)root,
363      CRYPT_VERIFY_CERT_SIGN_ISSUER_CERT, (void *)root, 0, NULL))
364     {
365         TRACE_(chain)("Last certificate's signature is invalid\n");
366         rootElement->TrustStatus.dwErrorStatus |=
367          CERT_TRUST_IS_NOT_SIGNATURE_VALID;
368     }
369     CRYPT_CheckTrustedStatus(hRoot, rootElement);
370 }
371
372 /* Decodes a cert's basic constraints extension (either szOID_BASIC_CONSTRAINTS
373  * or szOID_BASIC_CONSTRAINTS2, whichever is present) into a
374  * CERT_BASIC_CONSTRAINTS2_INFO.  If it neither extension is present, sets
375  * constraints->fCA to defaultIfNotSpecified.
376  * Returns FALSE if the extension is present but couldn't be decoded.
377  */
378 static BOOL CRYPT_DecodeBasicConstraints(PCCERT_CONTEXT cert,
379  CERT_BASIC_CONSTRAINTS2_INFO *constraints, BOOL defaultIfNotSpecified)
380 {
381     BOOL ret = TRUE;
382     PCERT_EXTENSION ext = CertFindExtension(szOID_BASIC_CONSTRAINTS,
383      cert->pCertInfo->cExtension, cert->pCertInfo->rgExtension);
384
385     constraints->fPathLenConstraint = FALSE;
386     if (ext)
387     {
388         CERT_BASIC_CONSTRAINTS_INFO *info;
389         DWORD size = 0;
390
391         ret = CryptDecodeObjectEx(X509_ASN_ENCODING, szOID_BASIC_CONSTRAINTS,
392          ext->Value.pbData, ext->Value.cbData, CRYPT_DECODE_ALLOC_FLAG,
393          NULL, &info, &size);
394         if (ret)
395         {
396             if (info->SubjectType.cbData == 1)
397                 constraints->fCA =
398                  info->SubjectType.pbData[0] & CERT_CA_SUBJECT_FLAG;
399             LocalFree(info);
400         }
401     }
402     else
403     {
404         ext = CertFindExtension(szOID_BASIC_CONSTRAINTS2,
405          cert->pCertInfo->cExtension, cert->pCertInfo->rgExtension);
406         if (ext)
407         {
408             DWORD size = sizeof(CERT_BASIC_CONSTRAINTS2_INFO);
409
410             ret = CryptDecodeObjectEx(X509_ASN_ENCODING,
411              szOID_BASIC_CONSTRAINTS2, ext->Value.pbData, ext->Value.cbData,
412              0, NULL, constraints, &size);
413         }
414         else
415             constraints->fCA = defaultIfNotSpecified;
416     }
417     return ret;
418 }
419
420 /* Checks element's basic constraints to see if it can act as a CA, with
421  * remainingCAs CAs left in this chain.  A root certificate is assumed to be
422  * allowed to be a CA whether or not the basic constraints extension is present,
423  * whereas an intermediate CA cert is not.  This matches the expected usage in
424  * RFC 3280:  a conforming intermediate CA MUST contain the basic constraints
425  * extension.  It also appears to match Microsoft's implementation.
426  * Updates chainConstraints with the element's constraints, if:
427  * 1. chainConstraints doesn't have a path length constraint, or
428  * 2. element's path length constraint is smaller than chainConstraints's
429  * Sets *pathLengthConstraintViolated to TRUE if a path length violation
430  * occurs.
431  * Returns TRUE if the element can be a CA, and the length of the remaining
432  * chain is valid.
433  */
434 static BOOL CRYPT_CheckBasicConstraintsForCA(PCCERT_CONTEXT cert,
435  CERT_BASIC_CONSTRAINTS2_INFO *chainConstraints, DWORD remainingCAs,
436  BOOL isRoot, BOOL *pathLengthConstraintViolated)
437 {
438     BOOL validBasicConstraints;
439     CERT_BASIC_CONSTRAINTS2_INFO constraints;
440
441     if ((validBasicConstraints = CRYPT_DecodeBasicConstraints(cert,
442      &constraints, isRoot)))
443     {
444         if (!constraints.fCA)
445         {
446             TRACE_(chain)("chain element %d can't be a CA\n", remainingCAs + 1);
447             validBasicConstraints = FALSE;
448         }
449         else if (constraints.fPathLenConstraint)
450         {
451             /* If the element has path length constraints, they apply to the
452              * entire remaining chain.
453              */
454             if (!chainConstraints->fPathLenConstraint ||
455              constraints.dwPathLenConstraint <
456              chainConstraints->dwPathLenConstraint)
457             {
458                 TRACE_(chain)("setting path length constraint to %d\n",
459                  chainConstraints->dwPathLenConstraint);
460                 chainConstraints->fPathLenConstraint = TRUE;
461                 chainConstraints->dwPathLenConstraint =
462                  constraints.dwPathLenConstraint;
463             }
464         }
465     }
466     if (chainConstraints->fPathLenConstraint &&
467      remainingCAs > chainConstraints->dwPathLenConstraint)
468     {
469         TRACE_(chain)("remaining CAs %d exceed max path length %d\n",
470          remainingCAs, chainConstraints->dwPathLenConstraint);
471         validBasicConstraints = FALSE;
472         *pathLengthConstraintViolated = TRUE;
473     }
474     return validBasicConstraints;
475 }
476
477 static BOOL url_matches(LPCWSTR constraint, LPCWSTR name,
478  DWORD *trustErrorStatus)
479 {
480     BOOL match = FALSE;
481
482     TRACE("%s, %s\n", debugstr_w(constraint), debugstr_w(name));
483
484     if (!constraint)
485         *trustErrorStatus |= CERT_TRUST_INVALID_NAME_CONSTRAINTS;
486     else if (!name)
487         ; /* no match */
488     else if (constraint[0] == '.')
489     {
490         if (lstrlenW(name) > lstrlenW(constraint))
491             match = !lstrcmpiW(name + lstrlenW(name) - lstrlenW(constraint),
492              constraint);
493     }
494     else
495         match = !lstrcmpiW(constraint, name);
496     return match;
497 }
498
499 static BOOL rfc822_name_matches(LPCWSTR constraint, LPCWSTR name,
500  DWORD *trustErrorStatus)
501 {
502     BOOL match = FALSE;
503     LPCWSTR at;
504
505     TRACE("%s, %s\n", debugstr_w(constraint), debugstr_w(name));
506
507     if (!constraint)
508         *trustErrorStatus |= CERT_TRUST_INVALID_NAME_CONSTRAINTS;
509     else if (!name)
510         ; /* no match */
511     else if ((at = strchrW(constraint, '@')))
512         match = !lstrcmpiW(constraint, name);
513     else
514     {
515         if ((at = strchrW(name, '@')))
516             match = url_matches(constraint, at + 1, trustErrorStatus);
517         else
518             match = !lstrcmpiW(constraint, name);
519     }
520     return match;
521 }
522
523 static BOOL dns_name_matches(LPCWSTR constraint, LPCWSTR name,
524  DWORD *trustErrorStatus)
525 {
526     BOOL match = FALSE;
527
528     TRACE("%s, %s\n", debugstr_w(constraint), debugstr_w(name));
529
530     if (!constraint)
531         *trustErrorStatus |= CERT_TRUST_INVALID_NAME_CONSTRAINTS;
532     else if (!name)
533         ; /* no match */
534     else if (lstrlenW(name) >= lstrlenW(constraint))
535         match = !lstrcmpiW(name + lstrlenW(name) - lstrlenW(constraint),
536          constraint);
537     /* else:  name is too short, no match */
538
539     return match;
540 }
541
542 static BOOL ip_address_matches(const CRYPT_DATA_BLOB *constraint,
543  const CRYPT_DATA_BLOB *name, DWORD *trustErrorStatus)
544 {
545     BOOL match = FALSE;
546
547     TRACE("(%d, %p), (%d, %p)\n", constraint->cbData, constraint->pbData,
548      name->cbData, name->pbData);
549
550     if (constraint->cbData != sizeof(DWORD) * 2)
551         *trustErrorStatus |= CERT_TRUST_INVALID_NAME_CONSTRAINTS;
552     else if (name->cbData == sizeof(DWORD))
553     {
554         DWORD subnet, mask, addr;
555
556         memcpy(&subnet, constraint->pbData, sizeof(subnet));
557         memcpy(&mask, constraint->pbData + sizeof(subnet), sizeof(mask));
558         memcpy(&addr, name->pbData, sizeof(addr));
559         /* These are really in big-endian order, but for equality matching we
560          * don't need to swap to host order
561          */
562         match = (subnet & mask) == (addr & mask);
563     }
564     /* else: name is wrong size, no match */
565
566     return match;
567 }
568
569 static void CRYPT_FindMatchingNameEntry(const CERT_ALT_NAME_ENTRY *constraint,
570  const CERT_ALT_NAME_INFO *subjectName, DWORD *trustErrorStatus,
571  DWORD errorIfFound, DWORD errorIfNotFound)
572 {
573     DWORD i;
574     BOOL match = FALSE;
575
576     for (i = 0; i < subjectName->cAltEntry; i++)
577     {
578         if (subjectName->rgAltEntry[i].dwAltNameChoice ==
579          constraint->dwAltNameChoice)
580         {
581             switch (constraint->dwAltNameChoice)
582             {
583             case CERT_ALT_NAME_RFC822_NAME:
584                 match = rfc822_name_matches(constraint->u.pwszURL,
585                  subjectName->rgAltEntry[i].u.pwszURL, trustErrorStatus);
586                 break;
587             case CERT_ALT_NAME_DNS_NAME:
588                 match = dns_name_matches(constraint->u.pwszURL,
589                  subjectName->rgAltEntry[i].u.pwszURL, trustErrorStatus);
590                 break;
591             case CERT_ALT_NAME_URL:
592                 match = url_matches(constraint->u.pwszURL,
593                  subjectName->rgAltEntry[i].u.pwszURL, trustErrorStatus);
594                 break;
595             case CERT_ALT_NAME_IP_ADDRESS:
596                 match = ip_address_matches(&constraint->u.IPAddress,
597                  &subjectName->rgAltEntry[i].u.IPAddress, trustErrorStatus);
598                 break;
599             case CERT_ALT_NAME_DIRECTORY_NAME:
600             default:
601                 ERR("name choice %d unsupported in this context\n",
602                  constraint->dwAltNameChoice);
603                 *trustErrorStatus |=
604                  CERT_TRUST_HAS_NOT_SUPPORTED_NAME_CONSTRAINT;
605             }
606         }
607     }
608     *trustErrorStatus |= match ? errorIfFound : errorIfNotFound;
609 }
610
611 static void CRYPT_CheckNameConstraints(
612  const CERT_NAME_CONSTRAINTS_INFO *nameConstraints, const CERT_INFO *cert,
613  DWORD *trustErrorStatus)
614 {
615     /* If there aren't any existing constraints, don't bother checking */
616     if (nameConstraints->cPermittedSubtree || nameConstraints->cExcludedSubtree)
617     {
618         CERT_EXTENSION *ext;
619
620         if ((ext = CertFindExtension(szOID_SUBJECT_ALT_NAME, cert->cExtension,
621          cert->rgExtension)))
622         {
623             CERT_ALT_NAME_INFO *subjectName;
624             DWORD size;
625
626             if (CryptDecodeObjectEx(X509_ASN_ENCODING, X509_ALTERNATE_NAME,
627              ext->Value.pbData, ext->Value.cbData,
628              CRYPT_DECODE_ALLOC_FLAG | CRYPT_DECODE_NOCOPY_FLAG, NULL,
629              &subjectName, &size))
630             {
631                 DWORD i;
632
633                 for (i = 0; i < nameConstraints->cExcludedSubtree; i++)
634                     CRYPT_FindMatchingNameEntry(
635                      &nameConstraints->rgExcludedSubtree[i].Base, subjectName,
636                      trustErrorStatus,
637                      CERT_TRUST_HAS_EXCLUDED_NAME_CONSTRAINT, 0);
638                 for (i = 0; i < nameConstraints->cPermittedSubtree; i++)
639                     CRYPT_FindMatchingNameEntry(
640                      &nameConstraints->rgPermittedSubtree[i].Base, subjectName,
641                      trustErrorStatus,
642                      0, CERT_TRUST_HAS_NOT_PERMITTED_NAME_CONSTRAINT);
643                 LocalFree(subjectName);
644             }
645         }
646         else
647         {
648             if (nameConstraints->cPermittedSubtree)
649                 *trustErrorStatus |=
650                  CERT_TRUST_HAS_NOT_PERMITTED_NAME_CONSTRAINT;
651             if (nameConstraints->cExcludedSubtree)
652                 *trustErrorStatus |=
653                  CERT_TRUST_HAS_EXCLUDED_NAME_CONSTRAINT;
654         }
655     }
656 }
657
658 /* Gets cert's name constraints, if any.  Free with LocalFree. */
659 static CERT_NAME_CONSTRAINTS_INFO *CRYPT_GetNameConstraints(CERT_INFO *cert)
660 {
661     CERT_NAME_CONSTRAINTS_INFO *info = NULL;
662
663     CERT_EXTENSION *ext;
664
665     if ((ext = CertFindExtension(szOID_NAME_CONSTRAINTS, cert->cExtension,
666      cert->rgExtension)))
667     {
668         DWORD size;
669
670         CryptDecodeObjectEx(X509_ASN_ENCODING, X509_NAME_CONSTRAINTS,
671          ext->Value.pbData, ext->Value.cbData,
672          CRYPT_DECODE_ALLOC_FLAG | CRYPT_DECODE_NOCOPY_FLAG, NULL, &info,
673          &size);
674     }
675     return info;
676 }
677
678 static void CRYPT_CheckChainNameConstraints(PCERT_SIMPLE_CHAIN chain)
679 {
680     int i, j;
681
682     /* Microsoft's implementation appears to violate RFC 3280:  according to
683      * MSDN, the various CERT_TRUST_*_NAME_CONSTRAINT errors are set if a CA's
684      * name constraint is violated in the end cert.  According to RFC 3280,
685      * the constraints should be checked against every subsequent certificate
686      * in the chain, not just the end cert.
687      * Microsoft's implementation also sets the name constraint errors on the
688      * certs whose constraints were violated, not on the certs that violated
689      * them.
690      * In order to be error-compatible with Microsoft's implementation, while
691      * still adhering to RFC 3280, I use a O(n ^ 2) algorithm to check name
692      * constraints.
693      */
694     for (i = chain->cElement - 1; i > 0; i--)
695     {
696         CERT_NAME_CONSTRAINTS_INFO *nameConstraints;
697
698         if ((nameConstraints = CRYPT_GetNameConstraints(
699          chain->rgpElement[i]->pCertContext->pCertInfo)))
700         {
701             for (j = i - 1; j >= 0; j--)
702             {
703                 DWORD errorStatus = 0;
704
705                 /* According to RFC 3280, self-signed certs don't have name
706                  * constraints checked unless they're the end cert.
707                  */
708                 if (j == 0 || !CRYPT_IsCertificateSelfSigned(
709                  chain->rgpElement[j]->pCertContext))
710                 {
711                     CRYPT_CheckNameConstraints(nameConstraints,
712                      chain->rgpElement[i]->pCertContext->pCertInfo,
713                      &errorStatus);
714                     chain->rgpElement[i]->TrustStatus.dwErrorStatus |=
715                      errorStatus;
716                 }
717             }
718             LocalFree(nameConstraints);
719         }
720     }
721 }
722
723 static void dump_basic_constraints(const CERT_EXTENSION *ext)
724 {
725     CERT_BASIC_CONSTRAINTS_INFO *info;
726     DWORD size = 0;
727
728     if (CryptDecodeObjectEx(X509_ASN_ENCODING, szOID_BASIC_CONSTRAINTS,
729      ext->Value.pbData, ext->Value.cbData, CRYPT_DECODE_ALLOC_FLAG,
730      NULL, &info, &size))
731     {
732         TRACE_(chain)("SubjectType: %02x\n", info->SubjectType.pbData[0]);
733         TRACE_(chain)("%s path length constraint\n",
734          info->fPathLenConstraint ? "has" : "doesn't have");
735         TRACE_(chain)("path length=%d\n", info->dwPathLenConstraint);
736         LocalFree(info);
737     }
738 }
739
740 static void dump_basic_constraints2(const CERT_EXTENSION *ext)
741 {
742     CERT_BASIC_CONSTRAINTS2_INFO constraints;
743     DWORD size = sizeof(CERT_BASIC_CONSTRAINTS2_INFO);
744
745     if (CryptDecodeObjectEx(X509_ASN_ENCODING,
746      szOID_BASIC_CONSTRAINTS2, ext->Value.pbData, ext->Value.cbData,
747      0, NULL, &constraints, &size))
748     {
749         TRACE_(chain)("basic constraints:\n");
750         TRACE_(chain)("can%s be a CA\n", constraints.fCA ? "" : "not");
751         TRACE_(chain)("%s path length constraint\n",
752          constraints.fPathLenConstraint ? "has" : "doesn't have");
753         TRACE_(chain)("path length=%d\n", constraints.dwPathLenConstraint);
754     }
755 }
756
757 static void dump_extension(const CERT_EXTENSION *ext)
758 {
759     TRACE_(chain)("%s (%scritical)\n", debugstr_a(ext->pszObjId),
760      ext->fCritical ? "" : "not ");
761     if (!strcmp(ext->pszObjId, szOID_BASIC_CONSTRAINTS))
762         dump_basic_constraints(ext);
763     else if (!strcmp(ext->pszObjId, szOID_BASIC_CONSTRAINTS2))
764         dump_basic_constraints2(ext);
765 }
766
767 static LPCWSTR filetime_to_str(const FILETIME *time)
768 {
769     static WCHAR date[80];
770     WCHAR dateFmt[80]; /* sufficient for all versions of LOCALE_SSHORTDATE */
771     SYSTEMTIME sysTime;
772
773     if (!time) return NULL;
774
775     GetLocaleInfoW(LOCALE_SYSTEM_DEFAULT, LOCALE_SSHORTDATE, dateFmt,
776      sizeof(dateFmt) / sizeof(dateFmt[0]));
777     FileTimeToSystemTime(time, &sysTime);
778     GetDateFormatW(LOCALE_SYSTEM_DEFAULT, 0, &sysTime, dateFmt, date,
779      sizeof(date) / sizeof(date[0]));
780     return date;
781 }
782
783 static void dump_element(PCCERT_CONTEXT cert)
784 {
785     LPWSTR name = NULL;
786     DWORD len, i;
787
788     TRACE_(chain)("%p\n", cert);
789     len = CertGetNameStringW(cert, CERT_NAME_SIMPLE_DISPLAY_TYPE,
790      CERT_NAME_ISSUER_FLAG, NULL, NULL, 0);
791     name = CryptMemAlloc(len * sizeof(WCHAR));
792     if (name)
793     {
794         CertGetNameStringW(cert, CERT_NAME_SIMPLE_DISPLAY_TYPE,
795          CERT_NAME_ISSUER_FLAG, NULL, name, len);
796         TRACE_(chain)("issued by %s\n", debugstr_w(name));
797         CryptMemFree(name);
798     }
799     len = CertGetNameStringW(cert, CERT_NAME_SIMPLE_DISPLAY_TYPE, 0, NULL,
800      NULL, 0);
801     name = CryptMemAlloc(len * sizeof(WCHAR));
802     if (name)
803     {
804         CertGetNameStringW(cert, CERT_NAME_SIMPLE_DISPLAY_TYPE, 0, NULL,
805          name, len);
806         TRACE_(chain)("issued to %s\n", debugstr_w(name));
807         CryptMemFree(name);
808     }
809     TRACE_(chain)("valid from %s to %s\n",
810      debugstr_w(filetime_to_str(&cert->pCertInfo->NotBefore)),
811      debugstr_w(filetime_to_str(&cert->pCertInfo->NotAfter)));
812     TRACE_(chain)("%d extensions\n", cert->pCertInfo->cExtension);
813     for (i = 0; i < cert->pCertInfo->cExtension; i++)
814         dump_extension(&cert->pCertInfo->rgExtension[i]);
815 }
816
817 static void CRYPT_CheckSimpleChain(PCertificateChainEngine engine,
818  PCERT_SIMPLE_CHAIN chain, LPFILETIME time)
819 {
820     PCERT_CHAIN_ELEMENT rootElement = chain->rgpElement[chain->cElement - 1];
821     int i;
822     BOOL pathLengthConstraintViolated = FALSE;
823     CERT_BASIC_CONSTRAINTS2_INFO constraints = { TRUE, FALSE, 0 };
824
825     TRACE_(chain)("checking chain with %d elements for time %s\n",
826      chain->cElement, debugstr_w(filetime_to_str(time)));
827     for (i = chain->cElement - 1; i >= 0; i--)
828     {
829         if (TRACE_ON(chain))
830             dump_element(chain->rgpElement[i]->pCertContext);
831         if (CertVerifyTimeValidity(time,
832          chain->rgpElement[i]->pCertContext->pCertInfo) != 0)
833             chain->rgpElement[i]->TrustStatus.dwErrorStatus |=
834              CERT_TRUST_IS_NOT_TIME_VALID;
835         if (i != 0)
836         {
837             BOOL isRoot;
838
839             if (i == chain->cElement - 1)
840                 isRoot = CRYPT_IsCertificateSelfSigned(
841                  chain->rgpElement[i]->pCertContext);
842             else
843                 isRoot = FALSE;
844             /* Check the signature of the cert this issued */
845             if (!CryptVerifyCertificateSignatureEx(0, X509_ASN_ENCODING,
846              CRYPT_VERIFY_CERT_SIGN_SUBJECT_CERT,
847              (void *)chain->rgpElement[i - 1]->pCertContext,
848              CRYPT_VERIFY_CERT_SIGN_ISSUER_CERT,
849              (void *)chain->rgpElement[i]->pCertContext, 0, NULL))
850                 chain->rgpElement[i - 1]->TrustStatus.dwErrorStatus |=
851                  CERT_TRUST_IS_NOT_SIGNATURE_VALID;
852             /* Once a path length constraint has been violated, every remaining
853              * CA cert's basic constraints is considered invalid.
854              */
855             if (pathLengthConstraintViolated)
856                 chain->rgpElement[i]->TrustStatus.dwErrorStatus |=
857                  CERT_TRUST_INVALID_BASIC_CONSTRAINTS;
858             else if (!CRYPT_CheckBasicConstraintsForCA(
859              chain->rgpElement[i]->pCertContext, &constraints, i - 1,
860              isRoot, &pathLengthConstraintViolated))
861                 chain->rgpElement[i]->TrustStatus.dwErrorStatus |=
862                  CERT_TRUST_INVALID_BASIC_CONSTRAINTS;
863             else if (constraints.fPathLenConstraint &&
864              constraints.dwPathLenConstraint)
865             {
866                 /* This one's valid - decrement max length */
867                 constraints.dwPathLenConstraint--;
868             }
869         }
870         if (CRYPT_IsSimpleChainCyclic(chain))
871         {
872             /* If the chain is cyclic, then the path length constraints
873              * are violated, because the chain is infinitely long.
874              */
875             pathLengthConstraintViolated = TRUE;
876             chain->TrustStatus.dwErrorStatus |=
877              CERT_TRUST_IS_PARTIAL_CHAIN |
878              CERT_TRUST_INVALID_BASIC_CONSTRAINTS;
879         }
880         /* FIXME: check valid usages */
881         CRYPT_CombineTrustStatus(&chain->TrustStatus,
882          &chain->rgpElement[i]->TrustStatus);
883     }
884     CRYPT_CheckChainNameConstraints(chain);
885     if (CRYPT_IsCertificateSelfSigned(rootElement->pCertContext))
886     {
887         rootElement->TrustStatus.dwInfoStatus |=
888          CERT_TRUST_IS_SELF_SIGNED | CERT_TRUST_HAS_NAME_MATCH_ISSUER;
889         CRYPT_CheckRootCert(engine->hRoot, rootElement);
890     }
891     CRYPT_CombineTrustStatus(&chain->TrustStatus, &rootElement->TrustStatus);
892 }
893
894 static PCCERT_CONTEXT CRYPT_GetIssuer(HCERTSTORE store, PCCERT_CONTEXT subject,
895  PCCERT_CONTEXT prevIssuer, DWORD *infoStatus)
896 {
897     PCCERT_CONTEXT issuer = NULL;
898     PCERT_EXTENSION ext;
899     DWORD size;
900
901     *infoStatus = 0;
902     if ((ext = CertFindExtension(szOID_AUTHORITY_KEY_IDENTIFIER,
903      subject->pCertInfo->cExtension, subject->pCertInfo->rgExtension)))
904     {
905         CERT_AUTHORITY_KEY_ID_INFO *info;
906         BOOL ret;
907
908         ret = CryptDecodeObjectEx(subject->dwCertEncodingType,
909          X509_AUTHORITY_KEY_ID, ext->Value.pbData, ext->Value.cbData,
910          CRYPT_DECODE_ALLOC_FLAG | CRYPT_DECODE_NOCOPY_FLAG, NULL,
911          &info, &size);
912         if (ret)
913         {
914             CERT_ID id;
915
916             if (info->CertIssuer.cbData && info->CertSerialNumber.cbData)
917             {
918                 id.dwIdChoice = CERT_ID_ISSUER_SERIAL_NUMBER;
919                 memcpy(&id.u.IssuerSerialNumber.Issuer, &info->CertIssuer,
920                  sizeof(CERT_NAME_BLOB));
921                 memcpy(&id.u.IssuerSerialNumber.SerialNumber,
922                  &info->CertSerialNumber, sizeof(CRYPT_INTEGER_BLOB));
923                 issuer = CertFindCertificateInStore(store,
924                  subject->dwCertEncodingType, 0, CERT_FIND_CERT_ID, &id,
925                  prevIssuer);
926                 if (issuer)
927                     *infoStatus = CERT_TRUST_HAS_EXACT_MATCH_ISSUER;
928             }
929             else if (info->KeyId.cbData)
930             {
931                 id.dwIdChoice = CERT_ID_KEY_IDENTIFIER;
932                 memcpy(&id.u.KeyId, &info->KeyId, sizeof(CRYPT_HASH_BLOB));
933                 issuer = CertFindCertificateInStore(store,
934                  subject->dwCertEncodingType, 0, CERT_FIND_CERT_ID, &id,
935                  prevIssuer);
936                 if (issuer)
937                     *infoStatus = CERT_TRUST_HAS_KEY_MATCH_ISSUER;
938             }
939             LocalFree(info);
940         }
941     }
942     else if ((ext = CertFindExtension(szOID_AUTHORITY_KEY_IDENTIFIER2,
943      subject->pCertInfo->cExtension, subject->pCertInfo->rgExtension)))
944     {
945         CERT_AUTHORITY_KEY_ID2_INFO *info;
946         BOOL ret;
947
948         ret = CryptDecodeObjectEx(subject->dwCertEncodingType,
949          X509_AUTHORITY_KEY_ID2, ext->Value.pbData, ext->Value.cbData,
950          CRYPT_DECODE_ALLOC_FLAG | CRYPT_DECODE_NOCOPY_FLAG, NULL,
951          &info, &size);
952         if (ret)
953         {
954             CERT_ID id;
955
956             if (info->AuthorityCertIssuer.cAltEntry &&
957              info->AuthorityCertSerialNumber.cbData)
958             {
959                 PCERT_ALT_NAME_ENTRY directoryName = NULL;
960                 DWORD i;
961
962                 for (i = 0; !directoryName &&
963                  i < info->AuthorityCertIssuer.cAltEntry; i++)
964                     if (info->AuthorityCertIssuer.rgAltEntry[i].dwAltNameChoice
965                      == CERT_ALT_NAME_DIRECTORY_NAME)
966                         directoryName =
967                          &info->AuthorityCertIssuer.rgAltEntry[i];
968                 if (directoryName)
969                 {
970                     id.dwIdChoice = CERT_ID_ISSUER_SERIAL_NUMBER;
971                     memcpy(&id.u.IssuerSerialNumber.Issuer,
972                      &directoryName->u.DirectoryName, sizeof(CERT_NAME_BLOB));
973                     memcpy(&id.u.IssuerSerialNumber.SerialNumber,
974                      &info->AuthorityCertSerialNumber,
975                      sizeof(CRYPT_INTEGER_BLOB));
976                     issuer = CertFindCertificateInStore(store,
977                      subject->dwCertEncodingType, 0, CERT_FIND_CERT_ID, &id,
978                      prevIssuer);
979                     if (issuer)
980                         *infoStatus = CERT_TRUST_HAS_EXACT_MATCH_ISSUER;
981                 }
982                 else
983                     FIXME("no supported name type in authority key id2\n");
984             }
985             else if (info->KeyId.cbData)
986             {
987                 id.dwIdChoice = CERT_ID_KEY_IDENTIFIER;
988                 memcpy(&id.u.KeyId, &info->KeyId, sizeof(CRYPT_HASH_BLOB));
989                 issuer = CertFindCertificateInStore(store,
990                  subject->dwCertEncodingType, 0, CERT_FIND_CERT_ID, &id,
991                  prevIssuer);
992                 if (issuer)
993                     *infoStatus = CERT_TRUST_HAS_KEY_MATCH_ISSUER;
994             }
995             LocalFree(info);
996         }
997     }
998     else
999     {
1000         issuer = CertFindCertificateInStore(store,
1001          subject->dwCertEncodingType, 0, CERT_FIND_SUBJECT_NAME,
1002          &subject->pCertInfo->Issuer, prevIssuer);
1003         *infoStatus = CERT_TRUST_HAS_NAME_MATCH_ISSUER;
1004     }
1005     return issuer;
1006 }
1007
1008 /* Builds a simple chain by finding an issuer for the last cert in the chain,
1009  * until reaching a self-signed cert, or until no issuer can be found.
1010  */
1011 static BOOL CRYPT_BuildSimpleChain(const CertificateChainEngine *engine,
1012  HCERTSTORE world, PCERT_SIMPLE_CHAIN chain)
1013 {
1014     BOOL ret = TRUE;
1015     PCCERT_CONTEXT cert = chain->rgpElement[chain->cElement - 1]->pCertContext;
1016
1017     while (ret && !CRYPT_IsSimpleChainCyclic(chain) &&
1018      !CRYPT_IsCertificateSelfSigned(cert))
1019     {
1020         PCCERT_CONTEXT issuer = CRYPT_GetIssuer(world, cert, NULL,
1021          &chain->rgpElement[chain->cElement - 1]->TrustStatus.dwInfoStatus);
1022
1023         if (issuer)
1024         {
1025             ret = CRYPT_AddCertToSimpleChain(engine, chain, issuer,
1026              chain->rgpElement[chain->cElement - 1]->TrustStatus.dwInfoStatus);
1027             /* CRYPT_AddCertToSimpleChain add-ref's the issuer, so free it to
1028              * close the enumeration that found it
1029              */
1030             CertFreeCertificateContext(issuer);
1031             cert = issuer;
1032         }
1033         else
1034         {
1035             TRACE_(chain)("Couldn't find issuer, halting chain creation\n");
1036             chain->TrustStatus.dwErrorStatus |= CERT_TRUST_IS_PARTIAL_CHAIN;
1037             break;
1038         }
1039     }
1040     return ret;
1041 }
1042
1043 static BOOL CRYPT_GetSimpleChainForCert(PCertificateChainEngine engine,
1044  HCERTSTORE world, PCCERT_CONTEXT cert, LPFILETIME pTime,
1045  PCERT_SIMPLE_CHAIN *ppChain)
1046 {
1047     BOOL ret = FALSE;
1048     PCERT_SIMPLE_CHAIN chain;
1049
1050     TRACE("(%p, %p, %p, %p)\n", engine, world, cert, pTime);
1051
1052     chain = CryptMemAlloc(sizeof(CERT_SIMPLE_CHAIN));
1053     if (chain)
1054     {
1055         memset(chain, 0, sizeof(CERT_SIMPLE_CHAIN));
1056         chain->cbSize = sizeof(CERT_SIMPLE_CHAIN);
1057         ret = CRYPT_AddCertToSimpleChain(engine, chain, cert, 0);
1058         if (ret)
1059         {
1060             ret = CRYPT_BuildSimpleChain(engine, world, chain);
1061             if (ret)
1062                 CRYPT_CheckSimpleChain(engine, chain, pTime);
1063         }
1064         if (!ret)
1065         {
1066             CRYPT_FreeSimpleChain(chain);
1067             chain = NULL;
1068         }
1069         *ppChain = chain;
1070     }
1071     return ret;
1072 }
1073
1074 static BOOL CRYPT_BuildCandidateChainFromCert(HCERTCHAINENGINE hChainEngine,
1075  PCCERT_CONTEXT cert, LPFILETIME pTime, HCERTSTORE hAdditionalStore,
1076  PCertificateChain *ppChain)
1077 {
1078     PCertificateChainEngine engine = (PCertificateChainEngine)hChainEngine;
1079     PCERT_SIMPLE_CHAIN simpleChain = NULL;
1080     HCERTSTORE world;
1081     BOOL ret;
1082
1083     world = CertOpenStore(CERT_STORE_PROV_COLLECTION, 0, 0,
1084      CERT_STORE_CREATE_NEW_FLAG, NULL);
1085     CertAddStoreToCollection(world, engine->hWorld, 0, 0);
1086     if (hAdditionalStore)
1087         CertAddStoreToCollection(world, hAdditionalStore, 0, 0);
1088     /* FIXME: only simple chains are supported for now, as CTLs aren't
1089      * supported yet.
1090      */
1091     if ((ret = CRYPT_GetSimpleChainForCert(engine, world, cert, pTime,
1092      &simpleChain)))
1093     {
1094         PCertificateChain chain = CryptMemAlloc(sizeof(CertificateChain));
1095
1096         if (chain)
1097         {
1098             chain->ref = 1;
1099             chain->world = world;
1100             chain->context.cbSize = sizeof(CERT_CHAIN_CONTEXT);
1101             chain->context.TrustStatus = simpleChain->TrustStatus;
1102             chain->context.cChain = 1;
1103             chain->context.rgpChain = CryptMemAlloc(sizeof(PCERT_SIMPLE_CHAIN));
1104             chain->context.rgpChain[0] = simpleChain;
1105             chain->context.cLowerQualityChainContext = 0;
1106             chain->context.rgpLowerQualityChainContext = NULL;
1107             chain->context.fHasRevocationFreshnessTime = FALSE;
1108             chain->context.dwRevocationFreshnessTime = 0;
1109         }
1110         else
1111             ret = FALSE;
1112         *ppChain = chain;
1113     }
1114     return ret;
1115 }
1116
1117 /* Makes and returns a copy of chain, up to and including element iElement. */
1118 static PCERT_SIMPLE_CHAIN CRYPT_CopySimpleChainToElement(
1119  const CERT_SIMPLE_CHAIN *chain, DWORD iElement)
1120 {
1121     PCERT_SIMPLE_CHAIN copy = CryptMemAlloc(sizeof(CERT_SIMPLE_CHAIN));
1122
1123     if (copy)
1124     {
1125         memset(copy, 0, sizeof(CERT_SIMPLE_CHAIN));
1126         copy->cbSize = sizeof(CERT_SIMPLE_CHAIN);
1127         copy->rgpElement =
1128          CryptMemAlloc((iElement + 1) * sizeof(PCERT_CHAIN_ELEMENT));
1129         if (copy->rgpElement)
1130         {
1131             DWORD i;
1132             BOOL ret = TRUE;
1133
1134             memset(copy->rgpElement, 0,
1135              (iElement + 1) * sizeof(PCERT_CHAIN_ELEMENT));
1136             for (i = 0; ret && i <= iElement; i++)
1137             {
1138                 PCERT_CHAIN_ELEMENT element =
1139                  CryptMemAlloc(sizeof(CERT_CHAIN_ELEMENT));
1140
1141                 if (element)
1142                 {
1143                     *element = *chain->rgpElement[i];
1144                     element->pCertContext = CertDuplicateCertificateContext(
1145                      chain->rgpElement[i]->pCertContext);
1146                     /* Reset the trust status of the copied element, it'll get
1147                      * rechecked after the new chain is done.
1148                      */
1149                     memset(&element->TrustStatus, 0, sizeof(CERT_TRUST_STATUS));
1150                     copy->rgpElement[copy->cElement++] = element;
1151                 }
1152                 else
1153                     ret = FALSE;
1154             }
1155             if (!ret)
1156             {
1157                 for (i = 0; i <= iElement; i++)
1158                     CryptMemFree(copy->rgpElement[i]);
1159                 CryptMemFree(copy->rgpElement);
1160                 CryptMemFree(copy);
1161                 copy = NULL;
1162             }
1163         }
1164         else
1165         {
1166             CryptMemFree(copy);
1167             copy = NULL;
1168         }
1169     }
1170     return copy;
1171 }
1172
1173 static void CRYPT_FreeLowerQualityChains(PCertificateChain chain)
1174 {
1175     DWORD i;
1176
1177     for (i = 0; i < chain->context.cLowerQualityChainContext; i++)
1178         CertFreeCertificateChain(chain->context.rgpLowerQualityChainContext[i]);
1179     CryptMemFree(chain->context.rgpLowerQualityChainContext);
1180     chain->context.cLowerQualityChainContext = 0;
1181     chain->context.rgpLowerQualityChainContext = NULL;
1182 }
1183
1184 static void CRYPT_FreeChainContext(PCertificateChain chain)
1185 {
1186     DWORD i;
1187
1188     CRYPT_FreeLowerQualityChains(chain);
1189     for (i = 0; i < chain->context.cChain; i++)
1190         CRYPT_FreeSimpleChain(chain->context.rgpChain[i]);
1191     CryptMemFree(chain->context.rgpChain);
1192     CertCloseStore(chain->world, 0);
1193     CryptMemFree(chain);
1194 }
1195
1196 /* Makes and returns a copy of chain, up to and including element iElement of
1197  * simple chain iChain.
1198  */
1199 static PCertificateChain CRYPT_CopyChainToElement(PCertificateChain chain,
1200  DWORD iChain, DWORD iElement)
1201 {
1202     PCertificateChain copy = CryptMemAlloc(sizeof(CertificateChain));
1203
1204     if (copy)
1205     {
1206         copy->ref = 1;
1207         copy->world = CertDuplicateStore(chain->world);
1208         copy->context.cbSize = sizeof(CERT_CHAIN_CONTEXT);
1209         /* Leave the trust status of the copied chain unset, it'll get
1210          * rechecked after the new chain is done.
1211          */
1212         memset(&copy->context.TrustStatus, 0, sizeof(CERT_TRUST_STATUS));
1213         copy->context.cLowerQualityChainContext = 0;
1214         copy->context.rgpLowerQualityChainContext = NULL;
1215         copy->context.fHasRevocationFreshnessTime = FALSE;
1216         copy->context.dwRevocationFreshnessTime = 0;
1217         copy->context.rgpChain = CryptMemAlloc(
1218          (iChain + 1) * sizeof(PCERT_SIMPLE_CHAIN));
1219         if (copy->context.rgpChain)
1220         {
1221             BOOL ret = TRUE;
1222             DWORD i;
1223
1224             memset(copy->context.rgpChain, 0,
1225              (iChain + 1) * sizeof(PCERT_SIMPLE_CHAIN));
1226             if (iChain)
1227             {
1228                 for (i = 0; ret && iChain && i < iChain - 1; i++)
1229                 {
1230                     copy->context.rgpChain[i] =
1231                      CRYPT_CopySimpleChainToElement(chain->context.rgpChain[i],
1232                      chain->context.rgpChain[i]->cElement - 1);
1233                     if (!copy->context.rgpChain[i])
1234                         ret = FALSE;
1235                 }
1236             }
1237             else
1238                 i = 0;
1239             if (ret)
1240             {
1241                 copy->context.rgpChain[i] =
1242                  CRYPT_CopySimpleChainToElement(chain->context.rgpChain[i],
1243                  iElement);
1244                 if (!copy->context.rgpChain[i])
1245                     ret = FALSE;
1246             }
1247             if (!ret)
1248             {
1249                 CRYPT_FreeChainContext(copy);
1250                 copy = NULL;
1251             }
1252             else
1253                 copy->context.cChain = iChain + 1;
1254         }
1255         else
1256         {
1257             CryptMemFree(copy);
1258             copy = NULL;
1259         }
1260     }
1261     return copy;
1262 }
1263
1264 static PCertificateChain CRYPT_BuildAlternateContextFromChain(
1265  HCERTCHAINENGINE hChainEngine, LPFILETIME pTime, HCERTSTORE hAdditionalStore,
1266  PCertificateChain chain)
1267 {
1268     PCertificateChainEngine engine = (PCertificateChainEngine)hChainEngine;
1269     PCertificateChain alternate;
1270
1271     TRACE("(%p, %p, %p, %p)\n", hChainEngine, pTime, hAdditionalStore, chain);
1272
1273     /* Always start with the last "lower quality" chain to ensure a consistent
1274      * order of alternate creation:
1275      */
1276     if (chain->context.cLowerQualityChainContext)
1277         chain = (PCertificateChain)chain->context.rgpLowerQualityChainContext[
1278          chain->context.cLowerQualityChainContext - 1];
1279     /* A chain with only one element can't have any alternates */
1280     if (chain->context.cChain <= 1 && chain->context.rgpChain[0]->cElement <= 1)
1281         alternate = NULL;
1282     else
1283     {
1284         DWORD i, j, infoStatus;
1285         PCCERT_CONTEXT alternateIssuer = NULL;
1286
1287         alternate = NULL;
1288         for (i = 0; !alternateIssuer && i < chain->context.cChain; i++)
1289             for (j = 0; !alternateIssuer &&
1290              j < chain->context.rgpChain[i]->cElement - 1; j++)
1291             {
1292                 PCCERT_CONTEXT subject =
1293                  chain->context.rgpChain[i]->rgpElement[j]->pCertContext;
1294                 PCCERT_CONTEXT prevIssuer = CertDuplicateCertificateContext(
1295                  chain->context.rgpChain[i]->rgpElement[j + 1]->pCertContext);
1296
1297                 alternateIssuer = CRYPT_GetIssuer(prevIssuer->hCertStore,
1298                  subject, prevIssuer, &infoStatus);
1299             }
1300         if (alternateIssuer)
1301         {
1302             i--;
1303             j--;
1304             alternate = CRYPT_CopyChainToElement(chain, i, j);
1305             if (alternate)
1306             {
1307                 BOOL ret = CRYPT_AddCertToSimpleChain(engine,
1308                  alternate->context.rgpChain[i], alternateIssuer, infoStatus);
1309
1310                 /* CRYPT_AddCertToSimpleChain add-ref's the issuer, so free it
1311                  * to close the enumeration that found it
1312                  */
1313                 CertFreeCertificateContext(alternateIssuer);
1314                 if (ret)
1315                 {
1316                     ret = CRYPT_BuildSimpleChain(engine, alternate->world,
1317                      alternate->context.rgpChain[i]);
1318                     if (ret)
1319                         CRYPT_CheckSimpleChain(engine,
1320                          alternate->context.rgpChain[i], pTime);
1321                     CRYPT_CombineTrustStatus(&alternate->context.TrustStatus,
1322                      &alternate->context.rgpChain[i]->TrustStatus);
1323                 }
1324                 if (!ret)
1325                 {
1326                     CRYPT_FreeChainContext(alternate);
1327                     alternate = NULL;
1328                 }
1329             }
1330         }
1331     }
1332     TRACE("%p\n", alternate);
1333     return alternate;
1334 }
1335
1336 #define CHAIN_QUALITY_SIGNATURE_VALID 8
1337 #define CHAIN_QUALITY_TIME_VALID      4
1338 #define CHAIN_QUALITY_COMPLETE_CHAIN  2
1339 #define CHAIN_QUALITY_TRUSTED_ROOT    1
1340
1341 #define CHAIN_QUALITY_HIGHEST \
1342  CHAIN_QUALITY_SIGNATURE_VALID | CHAIN_QUALITY_TIME_VALID | \
1343  CHAIN_QUALITY_COMPLETE_CHAIN | CHAIN_QUALITY_TRUSTED_ROOT
1344
1345 #define IS_TRUST_ERROR_SET(TrustStatus, bits) \
1346  (TrustStatus)->dwErrorStatus & (bits)
1347
1348 static DWORD CRYPT_ChainQuality(const CertificateChain *chain)
1349 {
1350     DWORD quality = CHAIN_QUALITY_HIGHEST;
1351
1352     if (IS_TRUST_ERROR_SET(&chain->context.TrustStatus,
1353      CERT_TRUST_IS_UNTRUSTED_ROOT))
1354         quality &= ~CHAIN_QUALITY_TRUSTED_ROOT;
1355     if (IS_TRUST_ERROR_SET(&chain->context.TrustStatus,
1356      CERT_TRUST_IS_PARTIAL_CHAIN))
1357     if (chain->context.TrustStatus.dwErrorStatus & CERT_TRUST_IS_PARTIAL_CHAIN)
1358         quality &= ~CHAIN_QUALITY_COMPLETE_CHAIN;
1359     if (IS_TRUST_ERROR_SET(&chain->context.TrustStatus,
1360      CERT_TRUST_IS_NOT_TIME_VALID | CERT_TRUST_IS_NOT_TIME_NESTED))
1361         quality &= ~CHAIN_QUALITY_TIME_VALID;
1362     if (IS_TRUST_ERROR_SET(&chain->context.TrustStatus,
1363      CERT_TRUST_IS_NOT_SIGNATURE_VALID))
1364         quality &= ~CHAIN_QUALITY_SIGNATURE_VALID;
1365     return quality;
1366 }
1367
1368 /* Chooses the highest quality chain among chain and its "lower quality"
1369  * alternate chains.  Returns the highest quality chain, with all other
1370  * chains as lower quality chains of it.
1371  */
1372 static PCertificateChain CRYPT_ChooseHighestQualityChain(
1373  PCertificateChain chain)
1374 {
1375     DWORD i;
1376
1377     /* There are always only two chains being considered:  chain, and an
1378      * alternate at chain->rgpLowerQualityChainContext[i].  If the alternate
1379      * has a higher quality than chain, the alternate gets assigned the lower
1380      * quality contexts, with chain taking the alternate's place among the
1381      * lower quality contexts.
1382      */
1383     for (i = 0; i < chain->context.cLowerQualityChainContext; i++)
1384     {
1385         PCertificateChain alternate =
1386          (PCertificateChain)chain->context.rgpLowerQualityChainContext[i];
1387
1388         if (CRYPT_ChainQuality(alternate) > CRYPT_ChainQuality(chain))
1389         {
1390             alternate->context.cLowerQualityChainContext =
1391              chain->context.cLowerQualityChainContext;
1392             alternate->context.rgpLowerQualityChainContext =
1393              chain->context.rgpLowerQualityChainContext;
1394             alternate->context.rgpLowerQualityChainContext[i] =
1395              (PCCERT_CHAIN_CONTEXT)chain;
1396             chain->context.cLowerQualityChainContext = 0;
1397             chain->context.rgpLowerQualityChainContext = NULL;
1398             chain = alternate;
1399         }
1400     }
1401     return chain;
1402 }
1403
1404 static BOOL CRYPT_AddAlternateChainToChain(PCertificateChain chain,
1405  const CertificateChain *alternate)
1406 {
1407     BOOL ret;
1408
1409     if (chain->context.cLowerQualityChainContext)
1410         chain->context.rgpLowerQualityChainContext =
1411          CryptMemRealloc(chain->context.rgpLowerQualityChainContext,
1412          (chain->context.cLowerQualityChainContext + 1) *
1413          sizeof(PCCERT_CHAIN_CONTEXT));
1414     else
1415         chain->context.rgpLowerQualityChainContext =
1416          CryptMemAlloc(sizeof(PCCERT_CHAIN_CONTEXT));
1417     if (chain->context.rgpLowerQualityChainContext)
1418     {
1419         chain->context.rgpLowerQualityChainContext[
1420          chain->context.cLowerQualityChainContext++] =
1421          (PCCERT_CHAIN_CONTEXT)alternate;
1422         ret = TRUE;
1423     }
1424     else
1425         ret = FALSE;
1426     return ret;
1427 }
1428
1429 static PCERT_CHAIN_ELEMENT CRYPT_FindIthElementInChain(
1430  const CERT_CHAIN_CONTEXT *chain, DWORD i)
1431 {
1432     DWORD j, iElement;
1433     PCERT_CHAIN_ELEMENT element = NULL;
1434
1435     for (j = 0, iElement = 0; !element && j < chain->cChain; j++)
1436     {
1437         if (iElement + chain->rgpChain[j]->cElement < i)
1438             iElement += chain->rgpChain[j]->cElement;
1439         else
1440             element = chain->rgpChain[j]->rgpElement[i - iElement];
1441     }
1442     return element;
1443 }
1444
1445 typedef struct _CERT_CHAIN_PARA_NO_EXTRA_FIELDS {
1446     DWORD            cbSize;
1447     CERT_USAGE_MATCH RequestedUsage;
1448 } CERT_CHAIN_PARA_NO_EXTRA_FIELDS, *PCERT_CHAIN_PARA_NO_EXTRA_FIELDS;
1449
1450 static void CRYPT_VerifyChainRevocation(PCERT_CHAIN_CONTEXT chain,
1451  LPFILETIME pTime, const CERT_CHAIN_PARA *pChainPara, DWORD chainFlags)
1452 {
1453     DWORD cContext;
1454
1455     if (chainFlags & CERT_CHAIN_REVOCATION_CHECK_END_CERT)
1456         cContext = 1;
1457     else if ((chainFlags & CERT_CHAIN_REVOCATION_CHECK_CHAIN) ||
1458      (chainFlags & CERT_CHAIN_REVOCATION_CHECK_CHAIN_EXCLUDE_ROOT))
1459     {
1460         DWORD i;
1461
1462         for (i = 0, cContext = 0; i < chain->cChain; i++)
1463         {
1464             if (i < chain->cChain - 1 ||
1465              chainFlags & CERT_CHAIN_REVOCATION_CHECK_CHAIN)
1466                 cContext += chain->rgpChain[i]->cElement;
1467             else
1468                 cContext += chain->rgpChain[i]->cElement - 1;
1469         }
1470     }
1471     else
1472         cContext = 0;
1473     if (cContext)
1474     {
1475         PCCERT_CONTEXT *contexts =
1476          CryptMemAlloc(cContext * sizeof(PCCERT_CONTEXT *));
1477
1478         if (contexts)
1479         {
1480             DWORD i, j, iContext, revocationFlags;
1481             CERT_REVOCATION_PARA revocationPara = { sizeof(revocationPara), 0 };
1482             CERT_REVOCATION_STATUS revocationStatus =
1483              { sizeof(revocationStatus), 0 };
1484             BOOL ret;
1485
1486             for (i = 0, iContext = 0; iContext < cContext && i < chain->cChain;
1487              i++)
1488             {
1489                 for (j = 0; iContext < cContext &&
1490                  j < chain->rgpChain[i]->cElement; j++)
1491                     contexts[iContext++] =
1492                      chain->rgpChain[i]->rgpElement[j]->pCertContext;
1493             }
1494             revocationFlags = CERT_VERIFY_REV_CHAIN_FLAG;
1495             if (chainFlags & CERT_CHAIN_REVOCATION_CHECK_CACHE_ONLY)
1496                 revocationFlags |= CERT_VERIFY_CACHE_ONLY_BASED_REVOCATION;
1497             if (chainFlags & CERT_CHAIN_REVOCATION_ACCUMULATIVE_TIMEOUT)
1498                 revocationFlags |= CERT_VERIFY_REV_ACCUMULATIVE_TIMEOUT_FLAG;
1499             revocationPara.pftTimeToUse = pTime;
1500             if (pChainPara->cbSize == sizeof(CERT_CHAIN_PARA))
1501             {
1502                 revocationPara.dwUrlRetrievalTimeout =
1503                  pChainPara->dwUrlRetrievalTimeout;
1504                 revocationPara.fCheckFreshnessTime =
1505                  pChainPara->fCheckRevocationFreshnessTime;
1506                 revocationPara.dwFreshnessTime =
1507                  pChainPara->dwRevocationFreshnessTime;
1508             }
1509             ret = CertVerifyRevocation(X509_ASN_ENCODING,
1510              CERT_CONTEXT_REVOCATION_TYPE, cContext, (void **)contexts,
1511              revocationFlags, &revocationPara, &revocationStatus);
1512             if (!ret)
1513             {
1514                 PCERT_CHAIN_ELEMENT element =
1515                  CRYPT_FindIthElementInChain(chain, revocationStatus.dwIndex);
1516                 DWORD error;
1517
1518                 switch (revocationStatus.dwError)
1519                 {
1520                 case CRYPT_E_NO_REVOCATION_CHECK:
1521                 case CRYPT_E_NO_REVOCATION_DLL:
1522                 case CRYPT_E_NOT_IN_REVOCATION_DATABASE:
1523                     error = CERT_TRUST_REVOCATION_STATUS_UNKNOWN;
1524                     break;
1525                 case CRYPT_E_REVOCATION_OFFLINE:
1526                     error = CERT_TRUST_IS_OFFLINE_REVOCATION;
1527                     break;
1528                 case CRYPT_E_REVOKED:
1529                     error = CERT_TRUST_IS_REVOKED;
1530                     break;
1531                 default:
1532                     WARN("unmapped error %08x\n", revocationStatus.dwError);
1533                     error = 0;
1534                 }
1535                 if (element)
1536                 {
1537                     /* FIXME: set element's pRevocationInfo member */
1538                     element->TrustStatus.dwErrorStatus |= error;
1539                 }
1540                 chain->TrustStatus.dwErrorStatus |= error;
1541             }
1542             CryptMemFree(contexts);
1543         }
1544     }
1545 }
1546
1547 BOOL WINAPI CertGetCertificateChain(HCERTCHAINENGINE hChainEngine,
1548  PCCERT_CONTEXT pCertContext, LPFILETIME pTime, HCERTSTORE hAdditionalStore,
1549  PCERT_CHAIN_PARA pChainPara, DWORD dwFlags, LPVOID pvReserved,
1550  PCCERT_CHAIN_CONTEXT* ppChainContext)
1551 {
1552     BOOL ret;
1553     PCertificateChain chain = NULL;
1554
1555     TRACE("(%p, %p, %p, %p, %p, %08x, %p, %p)\n", hChainEngine, pCertContext,
1556      pTime, hAdditionalStore, pChainPara, dwFlags, pvReserved, ppChainContext);
1557
1558     if (ppChainContext)
1559         *ppChainContext = NULL;
1560     if (!pChainPara)
1561     {
1562         SetLastError(E_INVALIDARG);
1563         return FALSE;
1564     }
1565     if (!pCertContext->pCertInfo->SignatureAlgorithm.pszObjId)
1566     {
1567         SetLastError(ERROR_INVALID_DATA);
1568         return FALSE;
1569     }
1570
1571     if (!hChainEngine)
1572         hChainEngine = CRYPT_GetDefaultChainEngine();
1573     /* FIXME: what about HCCE_LOCAL_MACHINE? */
1574     ret = CRYPT_BuildCandidateChainFromCert(hChainEngine, pCertContext, pTime,
1575      hAdditionalStore, &chain);
1576     if (ret)
1577     {
1578         PCertificateChain alternate = NULL;
1579         PCERT_CHAIN_CONTEXT pChain;
1580
1581         do {
1582             alternate = CRYPT_BuildAlternateContextFromChain(hChainEngine,
1583              pTime, hAdditionalStore, chain);
1584
1585             /* Alternate contexts are added as "lower quality" contexts of
1586              * chain, to avoid loops in alternate chain creation.
1587              * The highest-quality chain is chosen at the end.
1588              */
1589             if (alternate)
1590                 ret = CRYPT_AddAlternateChainToChain(chain, alternate);
1591         } while (ret && alternate);
1592         chain = CRYPT_ChooseHighestQualityChain(chain);
1593         if (!(dwFlags & CERT_CHAIN_RETURN_LOWER_QUALITY_CONTEXTS))
1594             CRYPT_FreeLowerQualityChains(chain);
1595         pChain = (PCERT_CHAIN_CONTEXT)chain;
1596         CRYPT_VerifyChainRevocation(pChain, pTime, pChainPara, dwFlags);
1597         if (ppChainContext)
1598             *ppChainContext = pChain;
1599         else
1600             CertFreeCertificateChain(pChain);
1601     }
1602     TRACE("returning %d\n", ret);
1603     return ret;
1604 }
1605
1606 PCCERT_CHAIN_CONTEXT WINAPI CertDuplicateCertificateChain(
1607  PCCERT_CHAIN_CONTEXT pChainContext)
1608 {
1609     PCertificateChain chain = (PCertificateChain)pChainContext;
1610
1611     TRACE("(%p)\n", pChainContext);
1612
1613     if (chain)
1614         InterlockedIncrement(&chain->ref);
1615     return pChainContext;
1616 }
1617
1618 VOID WINAPI CertFreeCertificateChain(PCCERT_CHAIN_CONTEXT pChainContext)
1619 {
1620     PCertificateChain chain = (PCertificateChain)pChainContext;
1621
1622     TRACE("(%p)\n", pChainContext);
1623
1624     if (chain)
1625     {
1626         if (InterlockedDecrement(&chain->ref) == 0)
1627             CRYPT_FreeChainContext(chain);
1628     }
1629 }
1630
1631 static void find_element_with_error(PCCERT_CHAIN_CONTEXT chain, DWORD error,
1632  LONG *iChain, LONG *iElement)
1633 {
1634     DWORD i, j;
1635
1636     for (i = 0; i < chain->cChain; i++)
1637         for (j = 0; j < chain->rgpChain[i]->cElement; j++)
1638             if (chain->rgpChain[i]->rgpElement[j]->TrustStatus.dwErrorStatus &
1639              error)
1640             {
1641                 *iChain = i;
1642                 *iElement = j;
1643                 return;
1644             }
1645 }
1646
1647 static BOOL WINAPI verify_base_policy(LPCSTR szPolicyOID,
1648  PCCERT_CHAIN_CONTEXT pChainContext, PCERT_CHAIN_POLICY_PARA pPolicyPara,
1649  PCERT_CHAIN_POLICY_STATUS pPolicyStatus)
1650 {
1651     pPolicyStatus->lChainIndex = pPolicyStatus->lElementIndex = -1;
1652     if (pChainContext->TrustStatus.dwErrorStatus &
1653      CERT_TRUST_IS_NOT_SIGNATURE_VALID)
1654     {
1655         pPolicyStatus->dwError = TRUST_E_CERT_SIGNATURE;
1656         find_element_with_error(pChainContext,
1657          CERT_TRUST_IS_NOT_SIGNATURE_VALID, &pPolicyStatus->lChainIndex,
1658          &pPolicyStatus->lElementIndex);
1659     }
1660     else if (pChainContext->TrustStatus.dwErrorStatus &
1661      CERT_TRUST_IS_UNTRUSTED_ROOT)
1662     {
1663         pPolicyStatus->dwError = CERT_E_UNTRUSTEDROOT;
1664         find_element_with_error(pChainContext,
1665          CERT_TRUST_IS_UNTRUSTED_ROOT, &pPolicyStatus->lChainIndex,
1666          &pPolicyStatus->lElementIndex);
1667     }
1668     else if (pChainContext->TrustStatus.dwErrorStatus & CERT_TRUST_IS_CYCLIC)
1669     {
1670         pPolicyStatus->dwError = CERT_E_CHAINING;
1671         find_element_with_error(pChainContext, CERT_TRUST_IS_CYCLIC,
1672          &pPolicyStatus->lChainIndex, &pPolicyStatus->lElementIndex);
1673         /* For a cyclic chain, which element is a cycle isn't meaningful */
1674         pPolicyStatus->lElementIndex = -1;
1675     }
1676     else
1677         pPolicyStatus->dwError = NO_ERROR;
1678     return TRUE;
1679 }
1680
1681 static BYTE msTestPubKey1[] = {
1682 0x30,0x47,0x02,0x40,0x81,0x55,0x22,0xb9,0x8a,0xa4,0x6f,0xed,0xd6,0xe7,0xd9,
1683 0x66,0x0f,0x55,0xbc,0xd7,0xcd,0xd5,0xbc,0x4e,0x40,0x02,0x21,0xa2,0xb1,0xf7,
1684 0x87,0x30,0x85,0x5e,0xd2,0xf2,0x44,0xb9,0xdc,0x9b,0x75,0xb6,0xfb,0x46,0x5f,
1685 0x42,0xb6,0x9d,0x23,0x36,0x0b,0xde,0x54,0x0f,0xcd,0xbd,0x1f,0x99,0x2a,0x10,
1686 0x58,0x11,0xcb,0x40,0xcb,0xb5,0xa7,0x41,0x02,0x03,0x01,0x00,0x01 };
1687 static BYTE msTestPubKey2[] = {
1688 0x30,0x47,0x02,0x40,0x9c,0x50,0x05,0x1d,0xe2,0x0e,0x4c,0x53,0xd8,0xd9,0xb5,
1689 0xe5,0xfd,0xe9,0xe3,0xad,0x83,0x4b,0x80,0x08,0xd9,0xdc,0xe8,0xe8,0x35,0xf8,
1690 0x11,0xf1,0xe9,0x9b,0x03,0x7a,0x65,0x64,0x76,0x35,0xce,0x38,0x2c,0xf2,0xb6,
1691 0x71,0x9e,0x06,0xd9,0xbf,0xbb,0x31,0x69,0xa3,0xf6,0x30,0xa0,0x78,0x7b,0x18,
1692 0xdd,0x50,0x4d,0x79,0x1e,0xeb,0x61,0xc1,0x02,0x03,0x01,0x00,0x01 };
1693
1694 static BOOL WINAPI verify_authenticode_policy(LPCSTR szPolicyOID,
1695  PCCERT_CHAIN_CONTEXT pChainContext, PCERT_CHAIN_POLICY_PARA pPolicyPara,
1696  PCERT_CHAIN_POLICY_STATUS pPolicyStatus)
1697 {
1698     BOOL ret = verify_base_policy(szPolicyOID, pChainContext, pPolicyPara,
1699      pPolicyStatus);
1700
1701     if (ret && pPolicyStatus->dwError == CERT_E_UNTRUSTEDROOT)
1702     {
1703         CERT_PUBLIC_KEY_INFO msPubKey = { { 0 } };
1704         BOOL isMSTestRoot = FALSE;
1705         PCCERT_CONTEXT failingCert =
1706          pChainContext->rgpChain[pPolicyStatus->lChainIndex]->
1707          rgpElement[pPolicyStatus->lElementIndex]->pCertContext;
1708         DWORD i;
1709         CRYPT_DATA_BLOB keyBlobs[] = {
1710          { sizeof(msTestPubKey1), msTestPubKey1 },
1711          { sizeof(msTestPubKey2), msTestPubKey2 },
1712         };
1713
1714         /* Check whether the root is an MS test root */
1715         for (i = 0; !isMSTestRoot && i < sizeof(keyBlobs) / sizeof(keyBlobs[0]);
1716          i++)
1717         {
1718             msPubKey.PublicKey.cbData = keyBlobs[i].cbData;
1719             msPubKey.PublicKey.pbData = keyBlobs[i].pbData;
1720             if (CertComparePublicKeyInfo(
1721              X509_ASN_ENCODING | PKCS_7_ASN_ENCODING,
1722              &failingCert->pCertInfo->SubjectPublicKeyInfo, &msPubKey))
1723                 isMSTestRoot = TRUE;
1724         }
1725         if (isMSTestRoot)
1726             pPolicyStatus->dwError = CERT_E_UNTRUSTEDTESTROOT;
1727     }
1728     return ret;
1729 }
1730
1731 static BOOL WINAPI verify_basic_constraints_policy(LPCSTR szPolicyOID,
1732  PCCERT_CHAIN_CONTEXT pChainContext, PCERT_CHAIN_POLICY_PARA pPolicyPara,
1733  PCERT_CHAIN_POLICY_STATUS pPolicyStatus)
1734 {
1735     pPolicyStatus->lChainIndex = pPolicyStatus->lElementIndex = -1;
1736     if (pChainContext->TrustStatus.dwErrorStatus &
1737      CERT_TRUST_INVALID_BASIC_CONSTRAINTS)
1738     {
1739         pPolicyStatus->dwError = TRUST_E_BASIC_CONSTRAINTS;
1740         find_element_with_error(pChainContext,
1741          CERT_TRUST_INVALID_BASIC_CONSTRAINTS, &pPolicyStatus->lChainIndex,
1742          &pPolicyStatus->lElementIndex);
1743     }
1744     else
1745         pPolicyStatus->dwError = NO_ERROR;
1746     return TRUE;
1747 }
1748
1749 static BYTE msPubKey1[] = {
1750 0x30,0x82,0x01,0x0a,0x02,0x82,0x01,0x01,0x00,0xdf,0x08,0xba,0xe3,0x3f,0x6e,
1751 0x64,0x9b,0xf5,0x89,0xaf,0x28,0x96,0x4a,0x07,0x8f,0x1b,0x2e,0x8b,0x3e,0x1d,
1752 0xfc,0xb8,0x80,0x69,0xa3,0xa1,0xce,0xdb,0xdf,0xb0,0x8e,0x6c,0x89,0x76,0x29,
1753 0x4f,0xca,0x60,0x35,0x39,0xad,0x72,0x32,0xe0,0x0b,0xae,0x29,0x3d,0x4c,0x16,
1754 0xd9,0x4b,0x3c,0x9d,0xda,0xc5,0xd3,0xd1,0x09,0xc9,0x2c,0x6f,0xa6,0xc2,0x60,
1755 0x53,0x45,0xdd,0x4b,0xd1,0x55,0xcd,0x03,0x1c,0xd2,0x59,0x56,0x24,0xf3,0xe5,
1756 0x78,0xd8,0x07,0xcc,0xd8,0xb3,0x1f,0x90,0x3f,0xc0,0x1a,0x71,0x50,0x1d,0x2d,
1757 0xa7,0x12,0x08,0x6d,0x7c,0xb0,0x86,0x6c,0xc7,0xba,0x85,0x32,0x07,0xe1,0x61,
1758 0x6f,0xaf,0x03,0xc5,0x6d,0xe5,0xd6,0xa1,0x8f,0x36,0xf6,0xc1,0x0b,0xd1,0x3e,
1759 0x69,0x97,0x48,0x72,0xc9,0x7f,0xa4,0xc8,0xc2,0x4a,0x4c,0x7e,0xa1,0xd1,0x94,
1760 0xa6,0xd7,0xdc,0xeb,0x05,0x46,0x2e,0xb8,0x18,0xb4,0x57,0x1d,0x86,0x49,0xdb,
1761 0x69,0x4a,0x2c,0x21,0xf5,0x5e,0x0f,0x54,0x2d,0x5a,0x43,0xa9,0x7a,0x7e,0x6a,
1762 0x8e,0x50,0x4d,0x25,0x57,0xa1,0xbf,0x1b,0x15,0x05,0x43,0x7b,0x2c,0x05,0x8d,
1763 0xbd,0x3d,0x03,0x8c,0x93,0x22,0x7d,0x63,0xea,0x0a,0x57,0x05,0x06,0x0a,0xdb,
1764 0x61,0x98,0x65,0x2d,0x47,0x49,0xa8,0xe7,0xe6,0x56,0x75,0x5c,0xb8,0x64,0x08,
1765 0x63,0xa9,0x30,0x40,0x66,0xb2,0xf9,0xb6,0xe3,0x34,0xe8,0x67,0x30,0xe1,0x43,
1766 0x0b,0x87,0xff,0xc9,0xbe,0x72,0x10,0x5e,0x23,0xf0,0x9b,0xa7,0x48,0x65,0xbf,
1767 0x09,0x88,0x7b,0xcd,0x72,0xbc,0x2e,0x79,0x9b,0x7b,0x02,0x03,0x01,0x00,0x01 };
1768 static BYTE msPubKey2[] = {
1769 0x30,0x82,0x01,0x0a,0x02,0x82,0x01,0x01,0x00,0xa9,0x02,0xbd,0xc1,0x70,0xe6,
1770 0x3b,0xf2,0x4e,0x1b,0x28,0x9f,0x97,0x78,0x5e,0x30,0xea,0xa2,0xa9,0x8d,0x25,
1771 0x5f,0xf8,0xfe,0x95,0x4c,0xa3,0xb7,0xfe,0x9d,0xa2,0x20,0x3e,0x7c,0x51,0xa2,
1772 0x9b,0xa2,0x8f,0x60,0x32,0x6b,0xd1,0x42,0x64,0x79,0xee,0xac,0x76,0xc9,0x54,
1773 0xda,0xf2,0xeb,0x9c,0x86,0x1c,0x8f,0x9f,0x84,0x66,0xb3,0xc5,0x6b,0x7a,0x62,
1774 0x23,0xd6,0x1d,0x3c,0xde,0x0f,0x01,0x92,0xe8,0x96,0xc4,0xbf,0x2d,0x66,0x9a,
1775 0x9a,0x68,0x26,0x99,0xd0,0x3a,0x2c,0xbf,0x0c,0xb5,0x58,0x26,0xc1,0x46,0xe7,
1776 0x0a,0x3e,0x38,0x96,0x2c,0xa9,0x28,0x39,0xa8,0xec,0x49,0x83,0x42,0xe3,0x84,
1777 0x0f,0xbb,0x9a,0x6c,0x55,0x61,0xac,0x82,0x7c,0xa1,0x60,0x2d,0x77,0x4c,0xe9,
1778 0x99,0xb4,0x64,0x3b,0x9a,0x50,0x1c,0x31,0x08,0x24,0x14,0x9f,0xa9,0xe7,0x91,
1779 0x2b,0x18,0xe6,0x3d,0x98,0x63,0x14,0x60,0x58,0x05,0x65,0x9f,0x1d,0x37,0x52,
1780 0x87,0xf7,0xa7,0xef,0x94,0x02,0xc6,0x1b,0xd3,0xbf,0x55,0x45,0xb3,0x89,0x80,
1781 0xbf,0x3a,0xec,0x54,0x94,0x4e,0xae,0xfd,0xa7,0x7a,0x6d,0x74,0x4e,0xaf,0x18,
1782 0xcc,0x96,0x09,0x28,0x21,0x00,0x57,0x90,0x60,0x69,0x37,0xbb,0x4b,0x12,0x07,
1783 0x3c,0x56,0xff,0x5b,0xfb,0xa4,0x66,0x0a,0x08,0xa6,0xd2,0x81,0x56,0x57,0xef,
1784 0xb6,0x3b,0x5e,0x16,0x81,0x77,0x04,0xda,0xf6,0xbe,0xae,0x80,0x95,0xfe,0xb0,
1785 0xcd,0x7f,0xd6,0xa7,0x1a,0x72,0x5c,0x3c,0xca,0xbc,0xf0,0x08,0xa3,0x22,0x30,
1786 0xb3,0x06,0x85,0xc9,0xb3,0x20,0x77,0x13,0x85,0xdf,0x02,0x03,0x01,0x00,0x01 };
1787 static BYTE msPubKey3[] = {
1788 0x30,0x82,0x02,0x0a,0x02,0x82,0x02,0x01,0x00,0xf3,0x5d,0xfa,0x80,0x67,0xd4,
1789 0x5a,0xa7,0xa9,0x0c,0x2c,0x90,0x20,0xd0,0x35,0x08,0x3c,0x75,0x84,0xcd,0xb7,
1790 0x07,0x89,0x9c,0x89,0xda,0xde,0xce,0xc3,0x60,0xfa,0x91,0x68,0x5a,0x9e,0x94,
1791 0x71,0x29,0x18,0x76,0x7c,0xc2,0xe0,0xc8,0x25,0x76,0x94,0x0e,0x58,0xfa,0x04,
1792 0x34,0x36,0xe6,0xdf,0xaf,0xf7,0x80,0xba,0xe9,0x58,0x0b,0x2b,0x93,0xe5,0x9d,
1793 0x05,0xe3,0x77,0x22,0x91,0xf7,0x34,0x64,0x3c,0x22,0x91,0x1d,0x5e,0xe1,0x09,
1794 0x90,0xbc,0x14,0xfe,0xfc,0x75,0x58,0x19,0xe1,0x79,0xb7,0x07,0x92,0xa3,0xae,
1795 0x88,0x59,0x08,0xd8,0x9f,0x07,0xca,0x03,0x58,0xfc,0x68,0x29,0x6d,0x32,0xd7,
1796 0xd2,0xa8,0xcb,0x4b,0xfc,0xe1,0x0b,0x48,0x32,0x4f,0xe6,0xeb,0xb8,0xad,0x4f,
1797 0xe4,0x5c,0x6f,0x13,0x94,0x99,0xdb,0x95,0xd5,0x75,0xdb,0xa8,0x1a,0xb7,0x94,
1798 0x91,0xb4,0x77,0x5b,0xf5,0x48,0x0c,0x8f,0x6a,0x79,0x7d,0x14,0x70,0x04,0x7d,
1799 0x6d,0xaf,0x90,0xf5,0xda,0x70,0xd8,0x47,0xb7,0xbf,0x9b,0x2f,0x6c,0xe7,0x05,
1800 0xb7,0xe1,0x11,0x60,0xac,0x79,0x91,0x14,0x7c,0xc5,0xd6,0xa6,0xe4,0xe1,0x7e,
1801 0xd5,0xc3,0x7e,0xe5,0x92,0xd2,0x3c,0x00,0xb5,0x36,0x82,0xde,0x79,0xe1,0x6d,
1802 0xf3,0xb5,0x6e,0xf8,0x9f,0x33,0xc9,0xcb,0x52,0x7d,0x73,0x98,0x36,0xdb,0x8b,
1803 0xa1,0x6b,0xa2,0x95,0x97,0x9b,0xa3,0xde,0xc2,0x4d,0x26,0xff,0x06,0x96,0x67,
1804 0x25,0x06,0xc8,0xe7,0xac,0xe4,0xee,0x12,0x33,0x95,0x31,0x99,0xc8,0x35,0x08,
1805 0x4e,0x34,0xca,0x79,0x53,0xd5,0xb5,0xbe,0x63,0x32,0x59,0x40,0x36,0xc0,0xa5,
1806 0x4e,0x04,0x4d,0x3d,0xdb,0x5b,0x07,0x33,0xe4,0x58,0xbf,0xef,0x3f,0x53,0x64,
1807 0xd8,0x42,0x59,0x35,0x57,0xfd,0x0f,0x45,0x7c,0x24,0x04,0x4d,0x9e,0xd6,0x38,
1808 0x74,0x11,0x97,0x22,0x90,0xce,0x68,0x44,0x74,0x92,0x6f,0xd5,0x4b,0x6f,0xb0,
1809 0x86,0xe3,0xc7,0x36,0x42,0xa0,0xd0,0xfc,0xc1,0xc0,0x5a,0xf9,0xa3,0x61,0xb9,
1810 0x30,0x47,0x71,0x96,0x0a,0x16,0xb0,0x91,0xc0,0x42,0x95,0xef,0x10,0x7f,0x28,
1811 0x6a,0xe3,0x2a,0x1f,0xb1,0xe4,0xcd,0x03,0x3f,0x77,0x71,0x04,0xc7,0x20,0xfc,
1812 0x49,0x0f,0x1d,0x45,0x88,0xa4,0xd7,0xcb,0x7e,0x88,0xad,0x8e,0x2d,0xec,0x45,
1813 0xdb,0xc4,0x51,0x04,0xc9,0x2a,0xfc,0xec,0x86,0x9e,0x9a,0x11,0x97,0x5b,0xde,
1814 0xce,0x53,0x88,0xe6,0xe2,0xb7,0xfd,0xac,0x95,0xc2,0x28,0x40,0xdb,0xef,0x04,
1815 0x90,0xdf,0x81,0x33,0x39,0xd9,0xb2,0x45,0xa5,0x23,0x87,0x06,0xa5,0x55,0x89,
1816 0x31,0xbb,0x06,0x2d,0x60,0x0e,0x41,0x18,0x7d,0x1f,0x2e,0xb5,0x97,0xcb,0x11,
1817 0xeb,0x15,0xd5,0x24,0xa5,0x94,0xef,0x15,0x14,0x89,0xfd,0x4b,0x73,0xfa,0x32,
1818 0x5b,0xfc,0xd1,0x33,0x00,0xf9,0x59,0x62,0x70,0x07,0x32,0xea,0x2e,0xab,0x40,
1819 0x2d,0x7b,0xca,0xdd,0x21,0x67,0x1b,0x30,0x99,0x8f,0x16,0xaa,0x23,0xa8,0x41,
1820 0xd1,0xb0,0x6e,0x11,0x9b,0x36,0xc4,0xde,0x40,0x74,0x9c,0xe1,0x58,0x65,0xc1,
1821 0x60,0x1e,0x7a,0x5b,0x38,0xc8,0x8f,0xbb,0x04,0x26,0x7c,0xd4,0x16,0x40,0xe5,
1822 0xb6,0x6b,0x6c,0xaa,0x86,0xfd,0x00,0xbf,0xce,0xc1,0x35,0x02,0x03,0x01,0x00,
1823 0x01 };
1824
1825 static BOOL WINAPI verify_ms_root_policy(LPCSTR szPolicyOID,
1826  PCCERT_CHAIN_CONTEXT pChainContext, PCERT_CHAIN_POLICY_PARA pPolicyPara,
1827  PCERT_CHAIN_POLICY_STATUS pPolicyStatus)
1828 {
1829     BOOL ret = verify_base_policy(szPolicyOID, pChainContext, pPolicyPara,
1830      pPolicyStatus);
1831
1832     if (ret && !pPolicyStatus->dwError)
1833     {
1834         CERT_PUBLIC_KEY_INFO msPubKey = { { 0 } };
1835         BOOL isMSRoot = FALSE;
1836         DWORD i;
1837         CRYPT_DATA_BLOB keyBlobs[] = {
1838          { sizeof(msPubKey1), msPubKey1 },
1839          { sizeof(msPubKey2), msPubKey2 },
1840          { sizeof(msPubKey3), msPubKey3 },
1841         };
1842         PCERT_SIMPLE_CHAIN rootChain =
1843          pChainContext->rgpChain[pChainContext->cChain -1 ];
1844         PCCERT_CONTEXT root =
1845          rootChain->rgpElement[rootChain->cElement - 1]->pCertContext;
1846
1847         for (i = 0; !isMSRoot && i < sizeof(keyBlobs) / sizeof(keyBlobs[0]);
1848          i++)
1849         {
1850             msPubKey.PublicKey.cbData = keyBlobs[i].cbData;
1851             msPubKey.PublicKey.pbData = keyBlobs[i].pbData;
1852             if (CertComparePublicKeyInfo(
1853              X509_ASN_ENCODING | PKCS_7_ASN_ENCODING,
1854              &root->pCertInfo->SubjectPublicKeyInfo, &msPubKey))
1855                 isMSRoot = TRUE;
1856         }
1857         if (isMSRoot)
1858             pPolicyStatus->lChainIndex = pPolicyStatus->lElementIndex = 0;
1859     }
1860     return ret;
1861 }
1862
1863 typedef BOOL (WINAPI *CertVerifyCertificateChainPolicyFunc)(LPCSTR szPolicyOID,
1864  PCCERT_CHAIN_CONTEXT pChainContext, PCERT_CHAIN_POLICY_PARA pPolicyPara,
1865  PCERT_CHAIN_POLICY_STATUS pPolicyStatus);
1866
1867 BOOL WINAPI CertVerifyCertificateChainPolicy(LPCSTR szPolicyOID,
1868  PCCERT_CHAIN_CONTEXT pChainContext, PCERT_CHAIN_POLICY_PARA pPolicyPara,
1869  PCERT_CHAIN_POLICY_STATUS pPolicyStatus)
1870 {
1871     static HCRYPTOIDFUNCSET set = NULL;
1872     BOOL ret = FALSE;
1873     CertVerifyCertificateChainPolicyFunc verifyPolicy = NULL;
1874     HCRYPTOIDFUNCADDR hFunc = NULL;
1875
1876     TRACE("(%s, %p, %p, %p)\n", debugstr_a(szPolicyOID), pChainContext,
1877      pPolicyPara, pPolicyStatus);
1878
1879     if (!HIWORD(szPolicyOID))
1880     {
1881         switch (LOWORD(szPolicyOID))
1882         {
1883         case LOWORD(CERT_CHAIN_POLICY_BASE):
1884             verifyPolicy = verify_base_policy;
1885             break;
1886         case LOWORD(CERT_CHAIN_POLICY_AUTHENTICODE):
1887             verifyPolicy = verify_authenticode_policy;
1888             break;
1889         case LOWORD(CERT_CHAIN_POLICY_BASIC_CONSTRAINTS):
1890             verifyPolicy = verify_basic_constraints_policy;
1891             break;
1892         case LOWORD(CERT_CHAIN_POLICY_MICROSOFT_ROOT):
1893             verifyPolicy = verify_ms_root_policy;
1894             break;
1895         default:
1896             FIXME("unimplemented for %d\n", LOWORD(szPolicyOID));
1897         }
1898     }
1899     if (!verifyPolicy)
1900     {
1901         if (!set)
1902             set = CryptInitOIDFunctionSet(
1903              CRYPT_OID_VERIFY_CERTIFICATE_CHAIN_POLICY_FUNC, 0);
1904         CryptGetOIDFunctionAddress(set, X509_ASN_ENCODING, szPolicyOID, 0,
1905          (void **)&verifyPolicy, &hFunc);
1906     }
1907     if (verifyPolicy)
1908         ret = verifyPolicy(szPolicyOID, pChainContext, pPolicyPara,
1909          pPolicyStatus);
1910     if (hFunc)
1911         CryptFreeOIDFunctionAddress(hFunc, 0);
1912     return ret;
1913 }