Remember user decisions on invalid certificates behind a flag

content/browser/ssl/ssl_host_state.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 "content/browser/ssl/ssl_host_state.h"
 
 #include "base/logging.h"
 #include "base/lazy_instance.h"
+#include "base/pickle.h"
 #include "content/public/browser/browser_context.h"
+#include "content/public/browser/ssl_host_state_decisions.h"
+#include "net/http/http_transaction_factory.h"
+#include "net/url_request/url_request_context.h"
+#include "net/url_request/url_request_context_getter.h"
+#include "url/gurl.h"
 
 const char kKeyName[] = "content_ssl_host_state";
 
 namespace content {
 
 SSLHostState* SSLHostState::GetFor(BrowserContext* context) {
   SSLHostState* rv = static_cast<SSLHostState*>(context->GetUserData(kKeyName));
   if (!rv) {
     rv = new SSLHostState();
-    context->SetUserData(kKeyName, rv);
+    rv->decisions_ = context->GetSSLHostStateDecisions();
+    // |context| may be NULL, implementing the default storage strategy.
+    if (context)
+      context->SetUserData(kKeyName, rv);
   }
   return rv;
 }
 
 SSLHostState::SSLHostState() {
 }
 
 SSLHostState::~SSLHostState() {
 }
 
 void SSLHostState::HostRanInsecureContent(const std::string& host, int pid) {
   DCHECK(CalledOnValidThread());
   ran_insecure_content_hosts_.insert(BrokenHostEntry(host, pid));
 }
 
 bool SSLHostState::DidHostRunInsecureContent(const std::string& host,
                                              int pid) const {
   DCHECK(CalledOnValidThread());
   return !!ran_insecure_content_hosts_.count(BrokenHostEntry(host, pid));
 }
 
 void SSLHostState::DenyCertForHost(net::X509Certificate* cert,
-                                   const std::string& host,
+                                   const GURL& url,
                                    net::CertStatus error) {
   DCHECK(CalledOnValidThread());
 
-  cert_policy_for_host_[host].Deny(cert, error);
+  if (!decisions_)
+    return;
+
+  decisions_->DenyCert(url, cert, error);
 }
 
 void SSLHostState::AllowCertForHost(net::X509Certificate* cert,
-                                    const std::string& host,
+                                    const GURL& url,
                                     net::CertStatus error) {
   DCHECK(CalledOnValidThread());
 
-  cert_policy_for_host_[host].Allow(cert, error);
+  if (!decisions_)
+    return;
+
+  decisions_->AllowCert(url, cert, error);
+}
+
+void SSLHostState::RevokeAllowAndDenyPreferences(const GURL& url) {
+  DCHECK(CalledOnValidThread());
+
+  if (!decisions_)
+    return;
+
+  // TODO(jww): This will revoke all of the decisions in the browser context.
+  // However, the networking stack actually keeps track of its own list of
+  // exceptions per-HttpNetworkTransaction in the SSLConfig structure (see the
+  // allowed_bad_certs Vector in net/ssl/ssl_config.h). This dual-tracking of
+  // exceptions introduces a problem where the browser context can revoke a
+  // certificate, but if a transaction reuses a cached version of the SSLConfig
+  // (probably from a pooled socket), it may bypass the intestitial layer.
+  //
+  // Over time, the cached versions should expire and it should converge on
+  // showing the interstitial. We probably need to
+  // introduce into the networking stack a way revoke SSLConfig's
+  // allowed_bad_certs lists per socket.
+  decisions_->RevokeAllowAndDenyPreferences(url);
+}
+
+bool SSLHostState::HasAllowedOrDeniedCert(const GURL& url) {
+  DCHECK(CalledOnValidThread());
+
+  if (!decisions_)
+    return false;
+
+  return decisions_->HasAllowedOrDeniedCert(url);
 }
 
 void SSLHostState::Clear() {
-  DCHECK(CalledOnValidThread());
+  if (!decisions_)
+    return;
 
-  cert_policy_for_host_.clear();
+  decisions_->Clear();
 }
 
 net::CertPolicy::Judgment SSLHostState::QueryPolicy(net::X509Certificate* cert,
-                                                    const std::string& host,
+                                                    const GURL& url,
                                                     net::CertStatus error) {
   DCHECK(CalledOnValidThread());
 
-  return cert_policy_for_host_[host].Check(cert, error);
+  if (!decisions_)
+    return net::CertPolicy::Judgment::UNKNOWN;
+
+  return decisions_->QueryPolicy(url, cert, error);
 }
 
 }  // namespace content