Changes to HttpNetworkTransaction for WebSocket Handshake

net/http/http_network_transaction.h

Index: net/http/http_network_transaction.h
diff --git a/net/http/http_network_transaction.h b/net/http/http_network_transaction.h
index b0950a2ecec652617613267991108fed431cca0b..73e87bb366e2d1ffa4b46bef61cf0195b56d1cf0 100644
--- a/net/http/http_network_transaction.h
+++ b/net/http/http_network_transaction.h
@@ -21,6 +21,7 @@
 #include "net/http/http_transaction.h"
 #include "net/proxy/proxy_service.h"
 #include "net/ssl/ssl_config_service.h"
+#include "net/websockets/websocket_handshake_stream_base.h"
 
 namespace net {
 
@@ -68,6 +69,8 @@ class NET_EXPORT_PRIVATE HttpNetworkTransaction
   virtual bool GetLoadTimingInfo(
       LoadTimingInfo* load_timing_info) const OVERRIDE;
   virtual void SetPriority(RequestPriority priority) OVERRIDE;
+  virtual void SetWebSocketHandshakeStreamFactory(
+      WebSocketHandshakeStreamBase::Factory* factory) OVERRIDE;
 
   // HttpStreamRequest::Delegate methods:
   virtual void OnStreamReady(const SSLConfig& used_ssl_config,
@@ -306,6 +309,11 @@ class NET_EXPORT_PRIVATE HttpNetworkTransaction
   // read from the socket until the tunnel is done.
   bool establishing_tunnel_;
 
+  // The factory to use to create WebSocketHandshakeStreamBase objects. Only
+  // relevant when establishing a WebSocket connection.
+  WebSocketHandshakeStreamBase::Factory*
+      websocket_handshake_stream_base_factory_;
+
   DISALLOW_COPY_AND_ASSIGN(HttpNetworkTransaction);
 };