tcp_client_socket_brokered.cc 7.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262
  1. // Copyright 2022 The Chromium Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style license that can be
  3. // found in the LICENSE file.
  4. #include "services/network/tcp_client_socket_brokered.h"
  5. #include "base/bind.h"
  6. #include "base/memory/weak_ptr.h"
  7. #include "build/build_config.h"
  8. #include "net/base/address_list.h"
  9. #include "net/base/completion_once_callback.h"
  10. #include "net/socket/tcp_client_socket.h"
  11. #include "net/traffic_annotation/network_traffic_annotation.h"
  12. #include "services/network/brokered_client_socket_factory.h"
  13. namespace network {
  14. TCPClientSocketBrokered::TCPClientSocketBrokered(
  15. const net::AddressList& addresses,
  16. std::unique_ptr<net::SocketPerformanceWatcher> socket_performance_watcher,
  17. net::NetworkQualityEstimator* network_quality_estimator,
  18. net::NetLog* net_log,
  19. const net::NetLogSource& source,
  20. BrokeredClientSocketFactory* client_socket_factory)
  21. : addresses_(addresses),
  22. socket_performance_watcher_(std::move(socket_performance_watcher)),
  23. network_quality_estimator_(network_quality_estimator),
  24. net_log_(net_log),
  25. source_(source),
  26. client_socket_factory_(client_socket_factory) {}
  27. TCPClientSocketBrokered::~TCPClientSocketBrokered() {
  28. Disconnect();
  29. }
  30. int TCPClientSocketBrokered::Bind(const net::IPEndPoint& address) {
  31. // TODO(liza): Implement this.
  32. NOTREACHED();
  33. return net::OK;
  34. }
  35. bool TCPClientSocketBrokered::SetKeepAlive(bool enable, int delay) {
  36. if (!brokered_socket_) {
  37. return false;
  38. }
  39. return brokered_socket_->SetKeepAlive(enable, delay);
  40. }
  41. bool TCPClientSocketBrokered::SetNoDelay(bool no_delay) {
  42. if (!brokered_socket_) {
  43. return false;
  44. }
  45. return brokered_socket_->SetNoDelay(no_delay);
  46. }
  47. void TCPClientSocketBrokered::SetBeforeConnectCallback(
  48. const BeforeConnectCallback& before_connect_callback) {
  49. // TODO(liza): Implement this.
  50. NOTREACHED();
  51. }
  52. int TCPClientSocketBrokered::Connect(net::CompletionOnceCallback callback) {
  53. // TODO(liza): add support for reconnecting disconnected socket, or look into
  54. // removing support for reconnection from TCPClientSocket if it's not needed.
  55. DCHECK(!callback.is_null());
  56. // If connecting or already connected, then just return OK.
  57. if (IsConnected() || is_connect_in_progress_)
  58. return net::OK;
  59. is_connect_in_progress_ = true;
  60. // TODO(https://crbug.com/1321274): Pass in AddressFamily of single IPEndPoint
  61. client_socket_factory_->BrokerCreateTcpSocket(
  62. addresses_.begin()->GetFamily(),
  63. base::BindOnce(&TCPClientSocketBrokered::DidCompleteCreate,
  64. brokered_weak_ptr_factory_.GetWeakPtr(),
  65. std::move(callback)));
  66. return net::ERR_IO_PENDING;
  67. }
  68. int TCPClientSocketBrokered::OpenSocketForBind(const net::IPEndPoint& address) {
  69. // TODO(liza): Implement this.
  70. NOTREACHED();
  71. return net::OK;
  72. }
  73. void TCPClientSocketBrokered::DidCompleteOpenForBind(
  74. const net::IPEndPoint& address,
  75. std::unique_ptr<net::TCPSocket> new_socket,
  76. net::Error result) {
  77. // TODO(liza): Implement this.
  78. NOTREACHED();
  79. }
  80. void TCPClientSocketBrokered::DidCompleteConnect(
  81. net::CompletionOnceCallback callback,
  82. int result) {
  83. DCHECK_NE(result, net::ERR_IO_PENDING);
  84. std::move(callback).Run(result);
  85. is_connect_in_progress_ = false;
  86. }
  87. void TCPClientSocketBrokered ::DidCompleteCreate(
  88. net::CompletionOnceCallback callback,
  89. mojo::PlatformHandle fd,
  90. int result) {
  91. if (result != net::OK) {
  92. std::move(callback).Run(result);
  93. return;
  94. }
  95. // Create an unconnected TCPSocket with the socket fd that was opened in the
  96. // browser process.
  97. std::unique_ptr<net::TCPSocket> tcp_socket = std::make_unique<net::TCPSocket>(
  98. std::move(socket_performance_watcher_), net_log_, source_);
  99. // TODO(https://crbug.com/1311014): call TCPSocketWin::AdoptUnconnectedSocket
  100. #if BUILDFLAG(IS_WIN)
  101. tcp_socket->Open(addresses_.begin()->GetFamily());
  102. #else
  103. tcp_socket->AdoptUnconnectedSocket(fd.ReleaseFD());
  104. #endif
  105. // TODO(liza): Pass through the NetworkHandle.
  106. brokered_socket_ = std::make_unique<net::TCPClientSocket>(
  107. std::move(tcp_socket), addresses_, network_quality_estimator_);
  108. brokered_socket_->ApplySocketTag(tag_);
  109. brokered_socket_->Connect(base::BindOnce(
  110. &TCPClientSocketBrokered::DidCompleteConnect,
  111. brokered_weak_ptr_factory_.GetWeakPtr(), std::move(callback)));
  112. }
  113. void TCPClientSocketBrokered::Disconnect() {
  114. if (brokered_socket_) {
  115. brokered_socket_->Disconnect();
  116. }
  117. is_connect_in_progress_ = false;
  118. }
  119. bool TCPClientSocketBrokered::IsConnected() const {
  120. if (!brokered_socket_) {
  121. return false;
  122. }
  123. return brokered_socket_->IsConnected();
  124. }
  125. bool TCPClientSocketBrokered::IsConnectedAndIdle() const {
  126. if (!brokered_socket_) {
  127. return false;
  128. }
  129. return brokered_socket_->IsConnectedAndIdle();
  130. }
  131. int TCPClientSocketBrokered::GetPeerAddress(net::IPEndPoint* address) const {
  132. if (!brokered_socket_) {
  133. return net::ERR_SOCKET_NOT_CONNECTED;
  134. }
  135. return brokered_socket_->GetPeerAddress(std::move(address));
  136. }
  137. int TCPClientSocketBrokered::GetLocalAddress(net::IPEndPoint* address) const {
  138. if (!brokered_socket_) {
  139. return net::ERR_SOCKET_NOT_CONNECTED;
  140. }
  141. return brokered_socket_->GetLocalAddress(std::move(address));
  142. }
  143. const net::NetLogWithSource& TCPClientSocketBrokered::NetLog() const {
  144. if (!brokered_socket_) {
  145. return NetLog();
  146. }
  147. return brokered_socket_->NetLog();
  148. }
  149. bool TCPClientSocketBrokered::WasEverUsed() const {
  150. if (!brokered_socket_) {
  151. return false;
  152. }
  153. return brokered_socket_->WasEverUsed();
  154. }
  155. bool TCPClientSocketBrokered::WasAlpnNegotiated() const {
  156. return false;
  157. }
  158. net::NextProto TCPClientSocketBrokered::GetNegotiatedProtocol() const {
  159. return net::kProtoUnknown;
  160. }
  161. bool TCPClientSocketBrokered::GetSSLInfo(net::SSLInfo* ssl_info) {
  162. return false;
  163. }
  164. int64_t TCPClientSocketBrokered::GetTotalReceivedBytes() const {
  165. if (!brokered_socket_) {
  166. return 0;
  167. }
  168. return brokered_socket_->GetTotalReceivedBytes();
  169. }
  170. void TCPClientSocketBrokered::ApplySocketTag(const net::SocketTag& tag) {
  171. if (!brokered_socket_) {
  172. tag_ = tag;
  173. } else {
  174. brokered_socket_->ApplySocketTag(tag);
  175. }
  176. }
  177. int TCPClientSocketBrokered::Read(net::IOBuffer* buf,
  178. int buf_len,
  179. net::CompletionOnceCallback callback) {
  180. if (!brokered_socket_) {
  181. return net::ERR_SOCKET_NOT_CONNECTED;
  182. }
  183. return brokered_socket_->Read(buf, buf_len, std::move(callback));
  184. }
  185. int TCPClientSocketBrokered::ReadIfReady(net::IOBuffer* buf,
  186. int buf_len,
  187. net::CompletionOnceCallback callback) {
  188. if (!brokered_socket_) {
  189. return net::ERR_SOCKET_NOT_CONNECTED;
  190. }
  191. return brokered_socket_->ReadIfReady(buf, buf_len, std::move(callback));
  192. }
  193. int TCPClientSocketBrokered::CancelReadIfReady() {
  194. if (!brokered_socket_) {
  195. return net::ERR_SOCKET_NOT_CONNECTED;
  196. }
  197. return brokered_socket_->CancelReadIfReady();
  198. }
  199. int TCPClientSocketBrokered::Write(
  200. net::IOBuffer* buf,
  201. int buf_len,
  202. net::CompletionOnceCallback callback,
  203. const net::NetworkTrafficAnnotationTag& traffic_annotation) {
  204. if (!brokered_socket_) {
  205. return net::ERR_SOCKET_NOT_CONNECTED;
  206. }
  207. return brokered_socket_->Write(std::move(buf), buf_len, std::move(callback),
  208. traffic_annotation);
  209. }
  210. int TCPClientSocketBrokered::SetReceiveBufferSize(int32_t size) {
  211. if (!brokered_socket_) {
  212. return net::ERR_SOCKET_NOT_CONNECTED;
  213. }
  214. return brokered_socket_->SetReceiveBufferSize(size);
  215. }
  216. int TCPClientSocketBrokered::SetSendBufferSize(int32_t size) {
  217. if (!brokered_socket_) {
  218. return net::ERR_SOCKET_NOT_CONNECTED;
  219. }
  220. return brokered_socket_->SetSendBufferSize(size);
  221. }
  222. } // namespace network