From 244f5fac39cd6d70430c89b177ae0893f1cccb46 Mon Sep 17 00:00:00 2001
From: klzgrad <kizdiv@gmail.com>
Date: Sat, 8 Dec 2018 00:52:08 -0500
Subject: [PATCH] socket: Add RawConnect method

---
 src/net/socket/client_socket_pool_manager.cc | 22 ++++++++++++++++++++
 src/net/socket/client_socket_pool_manager.h  | 14 +++++++++++++
 2 files changed, 36 insertions(+)

diff --git a/src/net/socket/client_socket_pool_manager.cc b/src/net/socket/client_socket_pool_manager.cc
index 3d896b83a5..dae45948b5 100644
--- a/src/net/socket/client_socket_pool_manager.cc
+++ b/src/net/socket/client_socket_pool_manager.cc
@@ -278,6 +278,28 @@ int InitSocketHandleForWebSocketRequest(
       std::move(callback), proxy_auth_callback);
 }
 
+int InitSocketHandleForRawConnect2(url::SchemeHostPort endpoint,
+                                   HttpNetworkSession* session,
+                                   int request_load_flags,
+                                   RequestPriority request_priority,
+                                   const ProxyInfo& proxy_info,
+                                   const SSLConfig& ssl_config_for_origin,
+                                   const SSLConfig& ssl_config_for_proxy,
+                                   PrivacyMode privacy_mode,
+                                   NetworkIsolationKey network_isolation_key,
+                                   const NetLogWithSource& net_log,
+                                   ClientSocketHandle* socket_handle,
+                                   CompletionOnceCallback callback) {
+  DCHECK(socket_handle);
+  return InitSocketPoolHelper(
+      std::move(endpoint), request_load_flags, request_priority, session,
+      proxy_info, ssl_config_for_origin, ssl_config_for_proxy,
+      /*is_for_websockets=*/true, privacy_mode,
+      std::move(network_isolation_key), SecureDnsPolicy::kDisable, SocketTag(),
+      net_log, 0, socket_handle, HttpNetworkSession::NORMAL_SOCKET_POOL,
+      std::move(callback), ClientSocketPool::ProxyAuthCallback());
+}
+
 int PreconnectSocketsForHttpRequest(url::SchemeHostPort endpoint,
                                     int request_load_flags,
                                     RequestPriority request_priority,
diff --git a/src/net/socket/client_socket_pool_manager.h b/src/net/socket/client_socket_pool_manager.h
index 8ce5d56646..5fc737bc0e 100644
--- a/src/net/socket/client_socket_pool_manager.h
+++ b/src/net/socket/client_socket_pool_manager.h
@@ -120,6 +120,20 @@ int InitSocketHandleForWebSocketRequest(
     CompletionOnceCallback callback,
     const ClientSocketPool::ProxyAuthCallback& proxy_auth_callback);
 
+NET_EXPORT int InitSocketHandleForRawConnect2(
+    url::SchemeHostPort endpoint,
+    HttpNetworkSession* session,
+    int request_load_flags,
+    RequestPriority request_priority,
+    const ProxyInfo& proxy_info,
+    const SSLConfig& ssl_config_for_origin,
+    const SSLConfig& ssl_config_for_proxy,
+    PrivacyMode privacy_mode,
+    NetworkIsolationKey network_isolation_key,
+    const NetLogWithSource& net_log,
+    ClientSocketHandle* socket_handle,
+    CompletionOnceCallback callback);
+
 // Similar to InitSocketHandleForHttpRequest except that it initiates the
 // desired number of preconnect streams from the relevant socket pool.
 int PreconnectSocketsForHttpRequest(url::SchemeHostPort endpoint,