Changes to HttpNetworkTransaction for WebSocket Handshake

net/http/http_transaction.h

Index: net/http/http_transaction.h
diff --git a/net/http/http_transaction.h b/net/http/http_transaction.h
index d44050080ba4d1cfa9a5c18573211f3adb11c398..ae66655053bea0f7c34aceefb80556e4e70b27bd 100644
--- a/net/http/http_transaction.h
+++ b/net/http/http_transaction.h
@@ -10,6 +10,7 @@
 #include "net/base/net_export.h"
 #include "net/base/request_priority.h"
 #include "net/base/upload_progress.h"
+#include "net/websockets/websocket_handshake_stream_base.h"
 
 namespace net {
 
@@ -134,6 +135,12 @@ class NET_EXPORT_PRIVATE HttpTransaction {
 
   // Called when the priority of the parent job changes.
   virtual void SetPriority(RequestPriority priority) = 0;
+
+  // Set the WebSocketHandshakeStreamBase::Factory to be used for the request.
+  // Only relevant to WebSocket transactions. Must be called before
+  // Start(). Ownership of |factory| remains with the caller.
+  virtual void SetWebSocketHandshakeStreamFactory(
+      WebSocketHandshakeStreamBase::Factory* factory) = 0;
 };
 
 }  // namespace net