[Devtools] Allow User-Agent header override for Websockets

content/browser/devtools/protocol/network_handler.cc

 // Copyright 2014 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/network_handler.h"
 
 #include <stddef.h>
 
 #include "base/containers/hash_tables.h"
 #include "base/strings/stringprintf.h"
@@ skipping 187 matching lines @@
 }
 
 void NetworkHandler::SetRenderFrameHost(RenderFrameHostImpl* host) {
   host_ = host;
 }
 
 void NetworkHandler::SetClient(std::unique_ptr<Client> client) {
   client_.swap(client);
 }
 
+void NetworkHandler::Detached() {
+  user_agent_override_ = "";
+}
+
+std::string NetworkHandler::GetUserAgentOverride() {
+  return user_agent_override_;
+}
+
 Response NetworkHandler::ClearBrowserCache() {
   if (host_)
     GetContentClient()->browser()->ClearCache(host_);
   return Response::OK();
 }
 
 Response NetworkHandler::ClearBrowserCookies() {
   if (host_)
     GetContentClient()->browser()->ClearCookies(host_);
   return Response::OK();
@@ skipping 109 matching lines @@
 
 Response NetworkHandler::ShowCertificateViewer(int certificate_id) {
   if (!host_)
     return Response::InternalError("Could not connect to view");
   WebContents* web_contents = WebContents::FromRenderFrameHost(host_);
   web_contents->GetDelegate()->ShowCertificateViewerInDevTools(
       web_contents, certificate_id);
   return Response::OK();
 }
 
+Response NetworkHandler::SetUserAgentOverride(const std::string& user_agent) {
+  if (!host_)
+    return Response::InternalError("Could not connect to view");
+  user_agent_override_ = user_agent;
+  return Response::OK();
+}
+
 }  // namespace network
 }  // namespace devtools
 }  // namespace content