Implement SecurityHandler to send the lock icon status to DevTools.

content/browser/devtools/protocol/security_handler.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 "content/browser/devtools/protocol/security_handler.h"
 
+#include <string>
+
 namespace content {
 namespace devtools {
 namespace security {
 
 typedef DevToolsProtocolClient::Response Response;
 
-SecurityHandler::SecurityHandler() {
+SecurityHandler::SecurityHandler() : enabled_(false), host_(nullptr) {
 }
 
 SecurityHandler::~SecurityHandler() {
 }
 
-void SecurityHandler::SetClient(scoped_ptr<DevToolsProtocolClient> client) {
+void SecurityHandler::SetClient(scoped_ptr<Client> client) {
+  client_.swap(client);
+}
+
+void SecurityHandler::SetRenderFrameHost(RenderFrameHost* host) {

[estark, 2015/06/09 21:51:45]

swap these two definitions (SetRFH and SecurityStyleChanged) to match the order
in the .h

[lgarron, 2015/06/09 22:12:09]

I've swapped the declarations in the .h (which keeps the two Set* methods
together).

+  host_ = host;
+  if (enabled_ && host_)
+    WebContentsObserver::Observe(WebContents::FromRenderFrameHost(host_));
+}
+
+void SecurityHandler::SecurityStyleChanged(SecurityStyle security_style) {
+  if (!enabled_)
+    return;
+
+  const std::string security_state =
+      SecurityStyleToProtocolSecurityState(security_style);
+  client_->SecurityStateChanged(
+      SecurityStateChangedParams::Create()->set_security_state(security_state));
 }
 
 Response SecurityHandler::Enable() {
+  enabled_ = true;
+  if (host_)
+    WebContentsObserver::Observe(WebContents::FromRenderFrameHost(host_));
   return Response::OK();
 }
 
 Response SecurityHandler::Disable() {
+  enabled_ = false;
+  WebContentsObserver::Observe(nullptr);
   return Response::OK();
 }
 
+std::string SecurityHandler::SecurityStyleToProtocolSecurityState(
+    SecurityStyle security_style) {
+  switch (security_style) {
+    case SECURITY_STYLE_UNKNOWN:
+      return kSecurityStateUnknown;
+    case SECURITY_STYLE_UNAUTHENTICATED:
+      return kSecurityStateHttp;
+    case SECURITY_STYLE_AUTHENTICATION_BROKEN:
+      return kSecurityStateInsecure;
+    case SECURITY_STYLE_WARNING:
+      return kSecurityStateWarning;
+    case SECURITY_STYLE_AUTHENTICATED:
+      return kSecurityStateSecure;
+    default:
+      NOTREACHED();

[estark, 2015/06/09 21:51:45]

I think builds on some platforms will fail without a return value here.

[lgarron, 2015/06/09 22:12:09]

Done (although I admit, I kind of want to see some buildbots fail on this).

+  }
+}
+
 }  // namespace security
 }  // namespace devtools
 }  // namespace content