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>
+
+#include "content/public/browser/web_contents.h"
+
 namespace content {
 namespace devtools {
 namespace security {
 
 typedef DevToolsProtocolClient::Response Response;
 
-SecurityHandler::SecurityHandler() {
+SecurityHandler::SecurityHandler() : enabled_(false), host_(nullptr) {

[pfeldman, 2015/06/11 06:45:50]

http://google-styleguide.googlecode.com/svn/trunk/cppguide.html#Constructor_I...

[lgarron, 2015/06/11 22:00:53]

This matches the "all on one line" case from the style guide, and `git cl
format` is happy with it.
Other handlers are inconsistent about when to break lines, and where to place
the brace.

I'm assuming you are asking me to put this on multiple lines anyhow; I've gone
ahead and done so.

 }
 
 SecurityHandler::~SecurityHandler() {
 }
 
-void SecurityHandler::SetClient(scoped_ptr<DevToolsProtocolClient> client) {
+void SecurityHandler::SetClient(scoped_ptr<Client> client) {
+  client_.swap(client);
+}
+
+void SecurityHandler::SetRenderFrameHost(RenderFrameHost* host) {
+  host_ = host;
+  if (enabled_ && host_)
+    WebContentsObserver::Observe(WebContents::FromRenderFrameHost(host_));
+}
+
+void SecurityHandler::SecurityStyleChanged(SecurityStyle security_style) {
+  if (!enabled_)

[pfeldman, 2015/06/11 06:45:50]

DCHECK(enabled_);

[lgarron, 2015/06/11 22:00:53]

Done.

+    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(

[pfeldman, 2015/06/11 06:45:50]

Utility function could be defined in a local static method above or in an
anonymous namespace above.

[lgarron, 2015/06/11 22:00:53]

Done.

+    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();
+      return kSecurityStateUnknown;
+  }
+}
+
 }  // namespace security
 }  // namespace devtools
 }  // namespace content