Implement SecurityHandler to send the lock icon status to DevTools.

content/browser/devtools/protocol/security_handler.cc

Index: content/browser/devtools/protocol/security_handler.cc
diff --git a/content/browser/devtools/protocol/security_handler.cc b/content/browser/devtools/protocol/security_handler.cc
index d3b52e4a8c8fb879a8848f2e36cd6b4801c178ea..d33257c8d862dcec10d5dd380e67b1c429f3f9b8 100644
--- a/content/browser/devtools/protocol/security_handler.cc
+++ b/content/browser/devtools/protocol/security_handler.cc
@@ -4,29 +4,74 @@
 
 #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