Add button to page info to revoke user certificate decisions.

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 "content/public/browser/browser_context.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"
 
 const char kKeyName[] = "content_ssl_host_state";
 
+namespace {
+
+void CloseIdleConnections(
+    const std::string& host,
+    scoped_refptr<net::URLRequestContextGetter> url_request_context_getter) {
+  url_request_context_getter->GetURLRequestContext()
+      ->http_transaction_factory()
+      ->GetSession()
+      ->CloseIdleConnections();
+}
+
+}  // namespace
+
 namespace content {
 
 SSLHostState* SSLHostState::GetFor(BrowserContext* context) {
   SSLHostState* rv = static_cast<SSLHostState*>(context->GetUserData(kKeyName));
   if (!rv) {
     rv = new SSLHostState();
+    rv->browser_context_ = context;
     context->SetUserData(kKeyName, rv);
   }
   return rv;
 }
 
 SSLHostState::SSLHostState() {
 }
 
 SSLHostState::~SSLHostState() {
 }
@@ skipping 24 matching lines @@
 
   cert_policy_for_host_[host].Allow(cert, error);
 }
 
 void SSLHostState::Clear() {
   DCHECK(CalledOnValidThread());
 
   cert_policy_for_host_.clear();
 }
 
+void SSLHostState::RevokeAllowAndDenyPreferences(const std::string& host) {
+  DCHECK(CalledOnValidThread());
+
+  cert_policy_for_host_.erase(cert_policy_for_host_.find(host));
+  scoped_refptr<net::URLRequestContextGetter> getter(
+      browser_context_->GetRequestContext());
+  browser_context_->GetRequestContext()->GetNetworkTaskRunner()->PostTask(
+      FROM_HERE, base::Bind(&CloseIdleConnections, host, getter));
+}
+
+bool SSLHostState::HasAllowedOrDeniedCert(const std::string& host) {
+  DCHECK(CalledOnValidThread());
+
+  return cert_policy_for_host_[host].HasAllowedCert() ||
+         cert_policy_for_host_[host].HasDeniedCert();
+}
+
 net::CertPolicy::Judgment SSLHostState::QueryPolicy(net::X509Certificate* cert,
                                                     const std::string& host,
                                                     net::CertStatus error) {
   DCHECK(CalledOnValidThread());
 
   return cert_policy_for_host_[host].Check(cert, error);
 }
 
 }  // namespace content