[Devtools] Allow User-Agent header override for Websockets

net/websockets/websocket_channel.h

 // 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.
 
 #ifndef NET_WEBSOCKETS_WEBSOCKET_CHANNEL_H_
 #define NET_WEBSOCKETS_WEBSOCKET_CHANNEL_H_
 
 #include <stdint.h>
 
 #include <memory>
@@ skipping 32 matching lines @@
 // definition; please see the RFC at http://tools.ietf.org/html/rfc6455 for
 // clarification.
 class NET_EXPORT WebSocketChannel {
  public:
   // The type of a WebSocketStream creator callback. Must match the signature of
   // WebSocketStream::CreateAndConnectStream().
   typedef base::Callback<std::unique_ptr<WebSocketStreamRequest>(
       const GURL&,
       const std::vector<std::string>&,
       const url::Origin&,
+      const std::string&,
       URLRequestContext*,
       const BoundNetLog&,
       std::unique_ptr<WebSocketStream::ConnectDelegate>)>
       WebSocketStreamCreator;
 
   // Methods which return a value of type ChannelState may delete |this|. If the
   // return value is CHANNEL_DELETED, then the caller must return without making
   // any further access to member variables or methods.
   using ChannelState = WebSocketEventInterface::ChannelState;
 
   // Creates a new WebSocketChannel in an idle state.
   // SendAddChannelRequest() must be called immediately afterwards to start the
   // connection process.
   WebSocketChannel(std::unique_ptr<WebSocketEventInterface> event_interface,
                    URLRequestContext* url_request_context);
   virtual ~WebSocketChannel();
 
   // Starts the connection process.
   void SendAddChannelRequest(
       const GURL& socket_url,
       const std::vector<std::string>& requested_protocols,
-      const url::Origin& origin);
+      const url::Origin& origin,
+      const std::string& additional_headers);
 
   // Sends a data frame to the remote side. It is the responsibility of the
   // caller to ensure that they have sufficient send quota to send this data,
   // otherwise the connection will be closed without sending. |fin| indicates
   // the last frame in a message, equivalent to "FIN" as specified in section
   // 5.2 of RFC6455. |data| is the "Payload Data". If |op_code| is kOpCodeText,
   // or it is kOpCodeContinuation and the type the message is Text, then |data|
   // must be a chunk of a valid UTF-8 message, however there is no requirement
   // for |data| to be split on character boundaries. Calling SendFrame may
   // result in synchronous calls to |event_interface_| which may result in this
@@ skipping 29 matching lines @@
   // valid for the execution of the current Task or until SendFrame() is called,
   // whichever happens sooner.
   int current_send_quota() const { return current_send_quota_; }
 
   // Starts the connection process, using a specified creator callback rather
   // than the default. This is exposed for testing.
   void SendAddChannelRequestForTesting(
       const GURL& socket_url,
       const std::vector<std::string>& requested_protocols,
       const url::Origin& origin,
+      const std::string& additional_headers,
       const WebSocketStreamCreator& creator);
 
   // The default timout for the closing handshake is a sensible value (see
   // kClosingHandshakeTimeoutSeconds in websocket_channel.cc). However, we can
   // set it to a very small value for testing purposes.
   void SetClosingHandshakeTimeoutForTesting(base::TimeDelta delay);
 
   // The default timout for the underlying connection close is a sensible value
   // (see kUnderlyingConnectionCloseTimeoutSeconds in websocket_channel.cc).
   // However, we can set it to a very small value for testing purposes.
@@ skipping 72 matching lines @@
   // interface with the implementation of those methods, and because the
   // lifetime of a WebSocketChannel is longer than the lifetime of the
   // connection process.
   class ConnectDelegate;
 
   // Starts the connection process, using the supplied creator callback.
   void SendAddChannelRequestWithSuppliedCreator(
       const GURL& socket_url,
       const std::vector<std::string>& requested_protocols,
       const url::Origin& origin,
+      const std::string& additional_headers,
       const WebSocketStreamCreator& creator);
 
   // Success callback from WebSocketStream::CreateAndConnectStream(). Reports
   // success to the event interface. May delete |this|.
   void OnConnectSuccess(std::unique_ptr<WebSocketStream> stream);
 
   // Failure callback from WebSocketStream::CreateAndConnectStream(). Reports
   // failure to the event interface. May delete |this|.
   void OnConnectFailure(const std::string& message);
 
@@ skipping 211 matching lines @@
   // For UMA. The time when OnConnectSuccess() method was called and |stream_|
   // was set.
   base::TimeTicks established_on_;
 
   DISALLOW_COPY_AND_ASSIGN(WebSocketChannel);
 };
 
 }  // namespace net
 
 #endif  // NET_WEBSOCKETS_WEBSOCKET_CHANNEL_H_