Make Cast certificate verification enforce constraints specified in the trusted root certificate.

net/cert/internal/verify_certificate_chain.h

Index: net/cert/internal/verify_certificate_chain.h
diff --git a/net/cert/internal/verify_certificate_chain.h b/net/cert/internal/verify_certificate_chain.h
index 0ef8cba49f821024260735e478e74ffd23e07e3f..2b4a7abe0b7a10aebb944e0718e38e0b1fb8c876 100644
--- a/net/cert/internal/verify_certificate_chain.h
+++ b/net/cert/internal/verify_certificate_chain.h
@@ -13,34 +13,107 @@
 
 #include "base/compiler_specific.h"
 #include "net/base/net_export.h"
+#include "net/cert/internal/parse_certificate.h"
+#include "net/der/input.h"
 
 namespace net {
 
 namespace der {
-class Input;
 struct GeneralizedTime;
 }
 
 class SignaturePolicy;
 
-struct NET_EXPORT TrustAnchor {
+// Represents a trust anchor (i.e. a trusted root certificate).
+class NET_EXPORT TrustAnchor {
+ public:
+  // The certificate data for this trust anchor may either be owned internally
+  // (INTERNAL_COPY) or owned externally (EXTERNAL_REFERENCE). When it is
+  // owned internally the data is held by |cert_data_|
+  enum class DataSource {
+    INTERNAL_COPY,
+    EXTERNAL_REFERENCE,
+  };
+
+  TrustAnchor();
   ~TrustAnchor();
 
-  // DER-encoded SubjectPublicKeyInfo for the trusted key.
-  std::string spki;
+  // Creates a TrustAnchor given a DER-encoded certificate. Returns nullptr on
+  // failure. Failure will occur if the certificate data cannot be parsed to
+  // find a subject.
+  //
+  // The provided certificate data is either copied, or aliased, depending on
+  // the value of |source|. See the comments for DataSource for details.
+  static std::unique_ptr<TrustAnchor> CreateFromCertificateData(
+      const uint8_t* data,
+      size_t length,
+      DataSource source);
+
+  // Returns true if the trust anchor matches |name|. In other words, returns
+  // true if the certificate's subject matches |name|.
+  bool MatchesName(const der::Input& name) const;
+
+  // Returns the DER-encoded certificate data for this trust anchor.
+  const der::Input& cert() const { return cert_; }
+
+ private:
+  // The backing store for the certificate data. This is only applicable when
+  // the trust anchor was initialized using DataSource::INTERNAL_COPY.
+  std::vector<uint8_t> cert_data_;
+
+  // Note that the backing data for |cert_| and |name_| may come either form
+  // |cert_data_| or some external buffer (depending on how the anchor was
+  // created).
+
+  // Points to the raw certificate DER.
+  der::Input cert_;
+
+  // Points to the subject TLV for the certificate.
+  der::Input name_;
 
-  // DER-encoded "Name" corresponding to the key.
-  std::string name;
+  DISALLOW_COPY_AND_ASSIGN(TrustAnchor);
 };
 
-// A very simple implementation of a TrustStore, which contains mappings from
-// names to trusted public keys.
-struct NET_EXPORT TrustStore {
+// A very simple implementation of a TrustStore, which contains a set of
+// trusted certificates.
+class NET_EXPORT TrustStore {
+ public:
   TrustStore();
-  TrustStore(const TrustStore& other);
   ~TrustStore();
 
-  std::vector<TrustAnchor> anchors;
+  // Empties the trust store, resetting it to original state.
+  void Clear();
+
+  // Adds a trusted certificate to the store. The trust store makes a copy of
+  // the provided certificate data.
+  bool AddTrustedCertificate(const uint8_t* data,
+                             size_t length) WARN_UNUSED_RESULT;
+  bool AddTrustedCertificate(const base::StringPiece& data) WARN_UNUSED_RESULT;
+
+  // This function is the same as AddTrustedCertificate() except the underlying
+  // data is not copied. The caller is responsible for ensuring that the data
+  // pointer remains alive and is not mutated for the lifetime of the
+  // TrustStore.
+  bool AddTrustedCertificateWithoutCopying(const uint8_t* data,
+                                           size_t length) WARN_UNUSED_RESULT;
+
+  // Returns the trust anchor that matches |name|, or nullptr if there is none.
+  // TODO(eroman): There may be multiple matches.
+  const TrustAnchor* FindTrustAnchorByName(const der::Input& name) const
+      WARN_UNUSED_RESULT;
+
+  // Returns true if |cert_der| matches a certificate in the TrustStore.
+  bool IsTrustedCertificate(const der::Input& cert_der) const
+      WARN_UNUSED_RESULT;
+
+ private:
+  bool AddTrustedCertificate(const uint8_t* data,
+                             size_t length,
+                             TrustAnchor::DataSource source) WARN_UNUSED_RESULT;
+
+  std::vector<std::unique_ptr<TrustAnchor>> anchors_;
+
+  DISALLOW_COPY_AND_ASSIGN(TrustStore);
 };
 
 // VerifyCertificateChain() verifies a certificate path (chain) based on the