Make HSTS headers not clobber preloaded pins.

net/http/http_security_headers_unittest.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include <algorithm>
 
 #include "base/base64.h"
 #include "base/sha1.h"
 #include "base/strings/string_piece.h"
 #include "crypto/sha2.h"
@@ skipping 498 matching lines @@
                       HashValuesEqual(good_hash));
   EXPECT_NE(domain_state.dynamic_spki_hashes.end(), hash);
 
   hash = std::find_if(
       domain_state.dynamic_spki_hashes.begin(),
       domain_state.dynamic_spki_hashes.end(),
       HashValuesEqual(backup_hash));
   EXPECT_NE(domain_state.dynamic_spki_hashes.end(), hash);
 }
 
+TEST_F(HttpSecurityHeadersTest, NoClobberPins) {
+  TransportSecurityState state;
+  TransportSecurityState::DomainState domain_state;
+
+  std::string domain("accounts.google.com");
+
+  // Retrieve the DomainState as it is by default, including its known good
+  // pins. Assert sanity.
+  EXPECT_TRUE(state.GetDomainState(domain, true, &domain_state));
+  HashValueVector saved_hashes = domain_state.static_spki_hashes;
+  EXPECT_TRUE(domain_state.ShouldUpgradeToSSL());
+  EXPECT_TRUE(domain_state.HasPublicKeyPins());
+
+  // Add a dynamic header. Due to bug crbug.com/29386, this will mask the
+  // static pins. However, we temporarily work around that in
+  // CheckPublicKeyPins (invoked below). CheckPublicKeyPins should still
+  // pass when given the original |saved_hashes|.
+  EXPECT_TRUE(state.AddHSTSHeader(domain, "includesubdomains; max-age=10000"));
+  EXPECT_TRUE(domain_state.ShouldUpgradeToSSL());
+  EXPECT_TRUE(state.GetDomainState(domain, true, &domain_state));
+  EXPECT_TRUE(domain_state.CheckPublicKeyPins(saved_hashes));
+
+  // Add a header, which should only update the dynamic state.
+  HashValue good_hash = GetTestHashValue(1, HASH_VALUE_SHA1);
+  std::string good_pin = GetTestPin(1, HASH_VALUE_SHA1);
+  std::string backup_pin = GetTestPin(2, HASH_VALUE_SHA1);
+  std::string header = "max-age = 10000; " + good_pin + "; " + backup_pin;
+
+  // Construct a fake SSLInfo that will pass AddHPKPHeader's checks.
+  SSLInfo ssl_info;
+  ssl_info.public_key_hashes.push_back(good_hash);
+  ssl_info.public_key_hashes.push_back(saved_hashes[0]);
+  EXPECT_TRUE(state.AddHPKPHeader(domain, header, ssl_info));
+
+  EXPECT_TRUE(state.AddHPKPHeader(domain, header, ssl_info));
+  EXPECT_TRUE(domain_state.ShouldUpgradeToSSL());
+  EXPECT_TRUE(state.GetDomainState(domain, true, &domain_state));
+  EXPECT_TRUE(domain_state.CheckPublicKeyPins(saved_hashes));
+}
+
 };    // namespace net