Enable security state computation for HeadlessWebContents.

headless/lib/browser/headless_web_contents_impl.cc

 // Copyright 2015 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 "headless/lib/browser/headless_web_contents_impl.h"
 
 #include <string>
 #include <utility>
 
 #include "base/bind.h"
 #include "base/json/json_writer.h"
 #include "base/memory/ptr_util.h"
 #include "base/memory/weak_ptr.h"
 #include "base/strings/utf_string_conversions.h"
 #include "base/trace_event/trace_event.h"
+#include "components/security_state/content/content_utils.h"
+#include "components/security_state/core/security_state.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/devtools_agent_host.h"
 #include "content/public/browser/navigation_handle.h"
 #include "content/public/browser/render_frame_host.h"
 #include "content/public/browser/render_process_host.h"
 #include "content/public/browser/render_view_host.h"
 #include "content/public/browser/render_widget_host_view.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/browser/web_contents_delegate.h"
 #include "content/public/common/bindings_policy.h"
+#include "content/public/common/origin_util.h"
 #include "content/public/renderer/render_frame.h"
 #include "headless/lib/browser/headless_browser_context_impl.h"
 #include "headless/lib/browser/headless_browser_impl.h"
 #include "headless/lib/browser/headless_browser_main_parts.h"
 #include "headless/lib/browser/headless_devtools_client_impl.h"
 #include "services/service_manager/public/cpp/interface_registry.h"
 #include "ui/aura/window.h"
 
 namespace headless {
 
@@ skipping 37 matching lines @@
                           content::WebContents* new_contents) override {
     std::unique_ptr<HeadlessWebContentsImpl> web_contents =
         HeadlessWebContentsImpl::CreateFromWebContents(new_contents,
                                                        browser_context_);
 
     DCHECK(new_contents->GetBrowserContext() == browser_context_);
 
     browser_context_->RegisterWebContents(std::move(web_contents));
   }
 
+  // Return the security style of the given |web_contents|, populating
+  // |security_style_explanations| to explain why the SecurityStyle was chosen.
+  blink::WebSecurityStyle GetSecurityStyle(
+      content::WebContents* web_contents,
+      content::SecurityStyleExplanations* security_style_explanations)
+      override {
+    security_state::SecurityInfo security_info;
+    security_state::GetSecurityInfo(
+        security_state::GetVisibleSecurityState(web_contents),
+        false /* used_policy_installed_certificate */,
+        base::Bind(&content::IsOriginSecure), &security_info);
+    return security_state::GetSecurityStyle(security_info,
+                                            security_style_explanations);
+  }
+
  private:
   HeadlessBrowserContextImpl* browser_context_;  // Not owned.
   DISALLOW_COPY_AND_ASSIGN(Delegate);
 };
 
 // static
 std::unique_ptr<HeadlessWebContentsImpl> HeadlessWebContentsImpl::Create(
     HeadlessWebContents::Builder* builder,
     aura::Window* parent_window) {
   content::WebContents::CreateParams create_params(builder->browser_context_,
@@ skipping 164 matching lines @@
 HeadlessWebContents::Builder::MojoService::MojoService() {}
 
 HeadlessWebContents::Builder::MojoService::MojoService(
     const std::string& service_name,
     const base::Callback<void(mojo::ScopedMessagePipeHandle)>& service_factory)
     : service_name(service_name), service_factory(service_factory) {}
 
 HeadlessWebContents::Builder::MojoService::~MojoService() {}
 
 }  // namespace headless