Set SSL info when an HTTP auth dialog is triggered by direct navigation.

content/browser/ssl/ssl_manager.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_manager.h"
 
 #include <set>
 
 #include "base/bind.h"
 #include "base/strings/utf_string_conversions.h"
 #include "base/supports_user_data.h"
 #include "content/browser/frame_host/navigation_entry_impl.h"
 #include "content/browser/loader/resource_dispatcher_host_impl.h"
 #include "content/browser/loader/resource_request_info_impl.h"
 #include "content/browser/ssl/ssl_cert_error_handler.h"
 #include "content/browser/ssl/ssl_policy.h"
 #include "content/browser/ssl/ssl_request_info.h"
 #include "content/browser/web_contents/web_contents_impl.h"
 #include "content/common/ssl_status_serialization.h"
 #include "content/public/browser/browser_context.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/load_from_memory_cache_details.h"
 #include "content/public/browser/navigation_details.h"
+#include "content/public/browser/navigation_entry.h"
 #include "content/public/browser/resource_request_details.h"
 #include "content/public/common/ssl_status.h"
 #include "net/url_request/url_request.h"
 
 namespace content {
 
 namespace {
 
 const char kSSLManagerKeyName[] = "content_ssl_manager";
 
@@ skipping 38 matching lines @@
                  new SSLCertErrorHandler(delegate,
                                          resource_type,
                                          url,
                                          render_process_id,
                                          render_frame_id,
                                          ssl_info,
                                          fatal)));
 }
 
 // static
+void SSLManager::OnAuthDialog(int render_process_id,
+                              int render_frame_id,
+                              const SSLStatus& ssl_status,
+                              bool is_main_frame) {
+  DCHECK_CURRENTLY_ON(BrowserThread::IO);
+  BrowserThread::PostTask(
+      BrowserThread::UI, FROM_HERE,
+      base::Bind(SSLManager::OnAuthDialogOnUI, render_process_id,
+                 render_frame_id, ssl_status, is_main_frame));
+}
+
+// static
+void SSLManager::OnAuthDialogOnUI(int render_process_id,
+                                  int render_frame_id,
+                                  const SSLStatus& ssl_status,
+                                  bool is_main_frame) {
+  DCHECK_CURRENTLY_ON(BrowserThread::UI);
+  RenderFrameHost* render_frame_host =
+      RenderFrameHostImpl::FromID(render_process_id, render_frame_id);
+  WebContents* web_contents =
+      WebContents::FromRenderFrameHost(render_frame_host);
+  if (!web_contents)
+    return;
+  NavigationControllerImpl* controller =
+      static_cast<NavigationControllerImpl*>(&web_contents->GetController());
+  // TODO(palmer, creis, meacer): If you've just done a PostTask to get here,
+  // then there's no guarantee that the pending entry still exists or is for the
+  // same navigation. (It may have been discarded or replaced with a different
+  // one in the meantime.) Ensure this is safe.
+  NavigationEntryImpl* entry = controller->GetPendingEntry();
+  controller->ssl_manager()->UpdateEntry(ssl_status, is_main_frame, entry);
+}
+
+// static
 void SSLManager::NotifySSLInternalStateChanged(BrowserContext* context) {
   SSLManagerSet* managers = static_cast<SSLManagerSet*>(
       context->GetUserData(kSSLManagerKeyName));
 
   for (std::set<SSLManager*>::iterator i = managers->get().begin();
        i != managers->get().end(); ++i) {
     (*i)->UpdateEntry((*i)->controller()->GetLastCommittedEntry());
   }
 }
 
@@ skipping 13 matching lines @@
 }
 
 SSLManager::~SSLManager() {
   SSLManagerSet* managers = static_cast<SSLManagerSet*>(
       controller_->GetBrowserContext()->GetUserData(kSSLManagerKeyName));
   managers->get().erase(this);
 }
 
 void SSLManager::DidCommitProvisionalLoad(const LoadCommittedDetails& details) {
   NavigationEntryImpl* entry = controller_->GetLastCommittedEntry();
+  UpdateEntry(details.ssl_status, details.is_main_frame, entry);
+}
 
-  if (details.is_main_frame) {
-    if (entry) {
-      // We may not have an entry if this is a navigation to an initial blank
-      // page. Add the new data we have.
-      entry->GetSSL() = details.ssl_status;
-    }
+void SSLManager::UpdateEntry(const SSLStatus& ssl_status,
+                             bool is_main_frame,
+                             NavigationEntryImpl* entry) {
+  if (is_main_frame && entry) {
+    // We may not have an entry if this is a navigation to an initial blank
+    // page. Reset the SSL information and add the new data we have.
+    entry->GetSSL() = SSLStatus();
+    entry->GetSSL().cert_id = ssl_status.cert_id;
+    entry->GetSSL().cert_status = ssl_status.cert_status;
+    entry->GetSSL().security_bits = ssl_status.security_bits;
+    entry->GetSSL().connection_status = ssl_status.connection_status;
+    entry->GetSSL().signed_certificate_timestamp_ids =
+        ssl_status.signed_certificate_timestamp_ids;

[meacer, 2015/09/29 00:17:02]

estark added a constructor for SSLStatus, this can now simply call it: 

entry->GetSSL() = SSLStatus(...);

https://code.google.com/p/chromium/codesearch#chromium/src/content/public/com...

[palmer, 2015/09/29 00:46:36]

How shall we get the SecurityStyle argument?

[meacer, 2015/09/29 00:48:51]

You can copy directly from ssl_status.security_style (which this code would miss
otherwise, hence the constructor :)

[palmer, 2015/09/29 00:56:46]

Ah, right. But, now we are missing the SSLInfo... And the callers for this
function don't seem to have one handy.

I'm adding in SecurityStyle and the couple other missing fields in |entry|'s
SSLStatus, though.

   }
 
   policy()->UpdateEntry(entry, controller_->delegate()->GetWebContents());
   // Always notify the WebContents that the SSL state changed when a
   // load is committed, in case the active navigation entry has changed.
   NotifyDidChangeVisibleSSLState();
 }
 
 void SSLManager::DidDisplayInsecureContent() {
   UpdateEntry(controller_->GetLastCommittedEntry());
@@ skipping 61 matching lines @@
     NotifyDidChangeVisibleSSLState();
 }
 
 void SSLManager::NotifyDidChangeVisibleSSLState() {
   WebContentsImpl* contents =
       static_cast<WebContentsImpl*>(controller_->delegate()->GetWebContents());
   contents->DidChangeVisibleSSLState();
 }
 
 }  // namespace content