[Devtools] Allow User-Agent header override for Websockets

content/browser/renderer_host/websocket_host.cc

 // Copyright 2013 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/renderer_host/websocket_host.h"
 
 #include <inttypes.h>
 
 #include <utility>
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
 #include "base/location.h"
 #include "base/logging.h"
 #include "base/macros.h"
 #include "base/memory/ptr_util.h"
 #include "base/single_thread_task_runner.h"
 #include "base/strings/string_util.h"
 #include "base/strings/stringprintf.h"
 #include "base/threading/thread_task_runner_handle.h"
 #include "content/browser/bad_message.h"
 #include "content/browser/renderer_host/websocket_blob_sender.h"
 #include "content/browser/renderer_host/websocket_dispatcher_host.h"
 #include "content/browser/ssl/ssl_error_handler.h"
 #include "content/browser/ssl/ssl_manager.h"
 #include "content/common/websocket_messages.h"
 #include "content/public/browser/browser_thread.h"
+#include "content/public/browser/devtools_agent_host.h"
 #include "content/public/browser/render_frame_host.h"
 #include "content/public/browser/storage_partition.h"
+#include "content/public/browser/web_contents.h"
 #include "ipc/ipc_message_macros.h"
 #include "net/base/net_errors.h"
 #include "net/http/http_request_headers.h"
 #include "net/http/http_response_headers.h"
 #include "net/http/http_util.h"
 #include "net/ssl/ssl_info.h"
 #include "net/websockets/websocket_channel.h"
 #include "net/websockets/websocket_errors.h"
 #include "net/websockets/websocket_event_interface.h"
 #include "net/websockets/websocket_frame.h"  // for WebSocketFrameHeader::OpCode
@@ skipping 377 matching lines @@
     AddChannel(socket_url, requested_protocols, origin, render_frame_id);
   }
   // |this| may have been deleted here.
 }
 
 void WebSocketHost::AddChannel(
     const GURL& socket_url,
     const std::vector<std::string>& requested_protocols,
     const url::Origin& origin,
     int render_frame_id) {
+
+  DCHECK(!channel_);
+
+  net::HttpRequestHeaders additional_headers;
+  RenderFrameHost* frame_host = RenderFrameHost::FromID(
+            dispatcher_->render_process_id(), render_frame_id);
+  if (frame_host) {
+    WebContents* tab = WebContents::FromRenderFrameHost(frame_host);
+    DCHECK(tab);
+
+    if (DevToolsAgentHost::HasFor(tab)) {
+      scoped_refptr<DevToolsAgentHost> agent =
+                DevToolsAgentHost::GetOrCreateFor(tab);
+      DCHECK(agent);
+
+      std::string user_agent = agent->GetUserAgentOverride();

[dgozman, 2016/05/24 19:02:30]

I'd reverse this dependency, reaching from NetworkHandler::SetUserAgentOverride
into WebSocketHost (or WebSocketDispatcherHost).

+      if (user_agent.length())
+        additional_headers.SetHeader(
+                net::HttpRequestHeaders::kUserAgent, user_agent.c_str());
+    }
+  }
+
   DVLOG(3) << "WebSocketHost::AddChannel"
            << " routing_id=" << routing_id_ << " socket_url=\"" << socket_url
            << "\" requested_protocols=\""
            << base::JoinString(requested_protocols, ", ") << "\" origin=\""
-           << origin << "\"";
-
-  DCHECK(!channel_);
+           << origin << "\" additional_headers=\""
+           << additional_headers.ToString() << "\"";
 
   std::unique_ptr<net::WebSocketEventInterface> event_interface(
       new WebSocketEventHandler(dispatcher_, this, routing_id_,
                                 render_frame_id));
   channel_.reset(new net::WebSocketChannel(std::move(event_interface),
                                            url_request_context_));
 
   if (pending_flow_control_quota_ > 0) {
     // channel_->SendFlowControl(pending_flow_control_quota_) must be called
     // after channel_->SendAddChannelRequest() below.
     // We post OnFlowControl() here using |weak_ptr_factory_| instead of
     // calling SendFlowControl directly, because |this| may have been deleted
     // after channel_->SendAddChannelRequest().
     base::ThreadTaskRunnerHandle::Get()->PostTask(
         FROM_HERE, base::Bind(&WebSocketHost::OnFlowControl,
                               weak_ptr_factory_.GetWeakPtr(),
                               pending_flow_control_quota_));
     pending_flow_control_quota_ = 0;
   }
-
-  channel_->SendAddChannelRequest(socket_url, requested_protocols, origin);
+  channel_->SendAddChannelRequest(
+          socket_url, requested_protocols, origin,
+          additional_headers.ToString());
   // |this| may have been deleted here.
 }
 
 void WebSocketHost::OnSendBlob(const std::string& uuid,
                                uint64_t expected_size) {
   DVLOG(3) << "WebSocketHost::OnSendBlob"
            << " routing_id=" << routing_id_ << " uuid=" << uuid
            << " expected_size=" << expected_size;
 
   DCHECK(channel_);
@@ skipping 109 matching lines @@
     default:
       ignore_result(dispatcher_->NotifyFailure(
           routing_id_,
           "Failed to load Blob: error code = " + net::ErrorToString(result)));
       // |this| is destroyed here.
       return;
   }
 }
 
 }  // namespace content