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