reactos/dll/win32/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 }