12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049 |
- // Copyright (c) 2012 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.
- // This file contains some tests for TCPClientSocket.
- // transport_client_socket_unittest.cc contans some other tests that
- // are common for TCP and other types of sockets.
- #include "net/socket/tcp_client_socket.h"
- #include <stddef.h>
- #include <set>
- #include <string>
- #include <vector>
- #include "base/strings/string_number_conversions.h"
- #include "base/test/bind.h"
- #include "base/test/power_monitor_test.h"
- #include "base/test/scoped_feature_list.h"
- #include "base/test/task_environment.h"
- #include "build/build_config.h"
- #include "net/base/features.h"
- #include "net/base/ip_address.h"
- #include "net/base/ip_endpoint.h"
- #include "net/base/net_errors.h"
- #include "net/base/test_completion_callback.h"
- #include "net/log/net_log_source.h"
- #include "net/nqe/network_quality_estimator_test_util.h"
- #include "net/socket/socket_performance_watcher.h"
- #include "net/socket/socket_test_util.h"
- #include "net/socket/tcp_server_socket.h"
- #include "net/test/embedded_test_server/embedded_test_server.h"
- #include "net/test/gtest_util.h"
- #include "net/traffic_annotation/network_traffic_annotation_test_helper.h"
- #include "testing/gmock/include/gmock/gmock.h"
- #include "testing/gtest/include/gtest/gtest.h"
- // This matches logic in tcp_client_socket.cc. Only used once, but defining it
- // in this file instead of just inlining the OS checks where its used makes it
- // more grep-able.
- #if !BUILDFLAG(IS_ANDROID)
- #define TCP_CLIENT_SOCKET_OBSERVES_SUSPEND
- #endif
- using net::test::IsError;
- using net::test::IsOk;
- using testing::Not;
- namespace base {
- class TimeDelta;
- }
- namespace net {
- namespace {
- class TCPClientSocketTest : public testing::Test {
- public:
- TCPClientSocketTest()
- : task_environment_(base::test::TaskEnvironment::MainThreadType::IO) {}
- ~TCPClientSocketTest() override { base::PowerMonitor::ShutdownForTesting(); }
- void Suspend() { power_monitor_source_.Suspend(); }
- void Resume() { power_monitor_source_.Resume(); }
- void CreateConnectedSockets(
- std::unique_ptr<StreamSocket>* accepted_socket,
- std::unique_ptr<TCPClientSocket>* client_socket,
- std::unique_ptr<ServerSocket>* server_socket_opt = nullptr) {
- IPAddress local_address = IPAddress::IPv4Localhost();
- std::unique_ptr<TCPServerSocket> server_socket =
- std::make_unique<TCPServerSocket>(nullptr, NetLogSource());
- ASSERT_THAT(server_socket->Listen(IPEndPoint(local_address, 0), 1), IsOk());
- IPEndPoint server_address;
- ASSERT_THAT(server_socket->GetLocalAddress(&server_address), IsOk());
- *client_socket = std::make_unique<TCPClientSocket>(
- AddressList(server_address), nullptr, nullptr, nullptr, NetLogSource());
- EXPECT_THAT((*client_socket)->Bind(IPEndPoint(local_address, 0)), IsOk());
- IPEndPoint local_address_result;
- EXPECT_THAT((*client_socket)->GetLocalAddress(&local_address_result),
- IsOk());
- EXPECT_EQ(local_address, local_address_result.address());
- TestCompletionCallback connect_callback;
- int connect_result = (*client_socket)->Connect(connect_callback.callback());
- TestCompletionCallback accept_callback;
- int result =
- server_socket->Accept(accepted_socket, accept_callback.callback());
- result = accept_callback.GetResult(result);
- ASSERT_THAT(result, IsOk());
- ASSERT_THAT(connect_callback.GetResult(connect_result), IsOk());
- EXPECT_TRUE((*client_socket)->IsConnected());
- EXPECT_TRUE((*accepted_socket)->IsConnected());
- if (server_socket_opt)
- *server_socket_opt = std::move(server_socket);
- }
- private:
- base::test::TaskEnvironment task_environment_;
- base::test::ScopedPowerMonitorTestSource power_monitor_source_;
- };
- // Try binding a socket to loopback interface and verify that we can
- // still connect to a server on the same interface.
- TEST_F(TCPClientSocketTest, BindLoopbackToLoopback) {
- IPAddress lo_address = IPAddress::IPv4Localhost();
- TCPServerSocket server(nullptr, NetLogSource());
- ASSERT_THAT(server.Listen(IPEndPoint(lo_address, 0), 1), IsOk());
- IPEndPoint server_address;
- ASSERT_THAT(server.GetLocalAddress(&server_address), IsOk());
- TCPClientSocket socket(AddressList(server_address), nullptr, nullptr, nullptr,
- NetLogSource());
- EXPECT_THAT(socket.Bind(IPEndPoint(lo_address, 0)), IsOk());
- IPEndPoint local_address_result;
- EXPECT_THAT(socket.GetLocalAddress(&local_address_result), IsOk());
- EXPECT_EQ(lo_address, local_address_result.address());
- TestCompletionCallback connect_callback;
- int connect_result = socket.Connect(connect_callback.callback());
- TestCompletionCallback accept_callback;
- std::unique_ptr<StreamSocket> accepted_socket;
- int result = server.Accept(&accepted_socket, accept_callback.callback());
- result = accept_callback.GetResult(result);
- ASSERT_THAT(result, IsOk());
- EXPECT_THAT(connect_callback.GetResult(connect_result), IsOk());
- EXPECT_TRUE(socket.IsConnected());
- socket.Disconnect();
- EXPECT_FALSE(socket.IsConnected());
- EXPECT_EQ(ERR_SOCKET_NOT_CONNECTED,
- socket.GetLocalAddress(&local_address_result));
- }
- // Try to bind socket to the loopback interface and connect to an
- // external address, verify that connection fails.
- TEST_F(TCPClientSocketTest, BindLoopbackToExternal) {
- IPAddress external_ip(72, 14, 213, 105);
- TCPClientSocket socket(AddressList::CreateFromIPAddress(external_ip, 80),
- nullptr, nullptr, nullptr, NetLogSource());
- EXPECT_THAT(socket.Bind(IPEndPoint(IPAddress::IPv4Localhost(), 0)), IsOk());
- TestCompletionCallback connect_callback;
- int result = socket.Connect(connect_callback.callback());
- // We may get different errors here on different system, but
- // connect() is not expected to succeed.
- EXPECT_THAT(connect_callback.GetResult(result), Not(IsOk()));
- }
- // Bind a socket to the IPv4 loopback interface and try to connect to
- // the IPv6 loopback interface, verify that connection fails.
- TEST_F(TCPClientSocketTest, BindLoopbackToIPv6) {
- TCPServerSocket server(nullptr, NetLogSource());
- int listen_result =
- server.Listen(IPEndPoint(IPAddress::IPv6Localhost(), 0), 1);
- if (listen_result != OK) {
- LOG(ERROR) << "Failed to listen on ::1 - probably because IPv6 is disabled."
- " Skipping the test";
- return;
- }
- IPEndPoint server_address;
- ASSERT_THAT(server.GetLocalAddress(&server_address), IsOk());
- TCPClientSocket socket(AddressList(server_address), nullptr, nullptr, nullptr,
- NetLogSource());
- EXPECT_THAT(socket.Bind(IPEndPoint(IPAddress::IPv4Localhost(), 0)), IsOk());
- TestCompletionCallback connect_callback;
- int result = socket.Connect(connect_callback.callback());
- EXPECT_THAT(connect_callback.GetResult(result), Not(IsOk()));
- }
- TEST_F(TCPClientSocketTest, WasEverUsed) {
- IPAddress lo_address = IPAddress::IPv4Localhost();
- TCPServerSocket server(nullptr, NetLogSource());
- ASSERT_THAT(server.Listen(IPEndPoint(lo_address, 0), 1), IsOk());
- IPEndPoint server_address;
- ASSERT_THAT(server.GetLocalAddress(&server_address), IsOk());
- TCPClientSocket socket(AddressList(server_address), nullptr, nullptr, nullptr,
- NetLogSource());
- EXPECT_FALSE(socket.WasEverUsed());
- EXPECT_THAT(socket.Bind(IPEndPoint(lo_address, 0)), IsOk());
- // Just connecting the socket should not set WasEverUsed.
- TestCompletionCallback connect_callback;
- int connect_result = socket.Connect(connect_callback.callback());
- EXPECT_FALSE(socket.WasEverUsed());
- TestCompletionCallback accept_callback;
- std::unique_ptr<StreamSocket> accepted_socket;
- int result = server.Accept(&accepted_socket, accept_callback.callback());
- ASSERT_THAT(accept_callback.GetResult(result), IsOk());
- EXPECT_THAT(connect_callback.GetResult(connect_result), IsOk());
- EXPECT_FALSE(socket.WasEverUsed());
- EXPECT_TRUE(socket.IsConnected());
- // Writing some data to the socket _should_ set WasEverUsed.
- const char kRequest[] = "GET / HTTP/1.0";
- auto write_buffer = base::MakeRefCounted<StringIOBuffer>(kRequest);
- TestCompletionCallback write_callback;
- socket.Write(write_buffer.get(), write_buffer->size(),
- write_callback.callback(), TRAFFIC_ANNOTATION_FOR_TESTS);
- EXPECT_TRUE(socket.WasEverUsed());
- socket.Disconnect();
- EXPECT_FALSE(socket.IsConnected());
- EXPECT_TRUE(socket.WasEverUsed());
- // Re-use the socket, which should set WasEverUsed to false.
- EXPECT_THAT(socket.Bind(IPEndPoint(lo_address, 0)), IsOk());
- TestCompletionCallback connect_callback2;
- connect_result = socket.Connect(connect_callback2.callback());
- EXPECT_FALSE(socket.WasEverUsed());
- }
- // Tests that DNS aliases can be stored in a socket for reuse.
- TEST_F(TCPClientSocketTest, DnsAliasesPersistForReuse) {
- IPAddress lo_address = IPAddress::IPv4Localhost();
- TCPServerSocket server(nullptr, NetLogSource());
- ASSERT_THAT(server.Listen(IPEndPoint(lo_address, 0), 1), IsOk());
- IPEndPoint server_address;
- ASSERT_THAT(server.GetLocalAddress(&server_address), IsOk());
- // Create a socket.
- TCPClientSocket socket(AddressList(server_address), nullptr, nullptr, nullptr,
- NetLogSource());
- EXPECT_FALSE(socket.WasEverUsed());
- EXPECT_THAT(socket.Bind(IPEndPoint(lo_address, 0)), IsOk());
- // The socket's DNS aliases are unset.
- EXPECT_TRUE(socket.GetDnsAliases().empty());
- // Set the aliases.
- std::set<std::string> dns_aliases({"alias1", "alias2", "host"});
- socket.SetDnsAliases(dns_aliases);
- // Verify that the aliases are set.
- EXPECT_THAT(socket.GetDnsAliases(),
- testing::UnorderedElementsAre("alias1", "alias2", "host"));
- // Connect the socket.
- TestCompletionCallback connect_callback;
- int connect_result = socket.Connect(connect_callback.callback());
- EXPECT_FALSE(socket.WasEverUsed());
- TestCompletionCallback accept_callback;
- std::unique_ptr<StreamSocket> accepted_socket;
- int result = server.Accept(&accepted_socket, accept_callback.callback());
- ASSERT_THAT(accept_callback.GetResult(result), IsOk());
- EXPECT_THAT(connect_callback.GetResult(connect_result), IsOk());
- EXPECT_FALSE(socket.WasEverUsed());
- EXPECT_TRUE(socket.IsConnected());
- // Write some data to the socket to set WasEverUsed, so that the
- // socket can be re-used.
- const char kRequest[] = "GET / HTTP/1.0";
- auto write_buffer = base::MakeRefCounted<StringIOBuffer>(kRequest);
- TestCompletionCallback write_callback;
- socket.Write(write_buffer.get(), write_buffer->size(),
- write_callback.callback(), TRAFFIC_ANNOTATION_FOR_TESTS);
- EXPECT_TRUE(socket.WasEverUsed());
- socket.Disconnect();
- EXPECT_FALSE(socket.IsConnected());
- EXPECT_TRUE(socket.WasEverUsed());
- // Re-use the socket, and verify that the aliases are still set.
- EXPECT_THAT(socket.Bind(IPEndPoint(lo_address, 0)), IsOk());
- TestCompletionCallback connect_callback2;
- connect_result = socket.Connect(connect_callback2.callback());
- EXPECT_FALSE(socket.WasEverUsed());
- EXPECT_THAT(socket.GetDnsAliases(),
- testing::ElementsAre("alias1", "alias2", "host"));
- }
- class TestSocketPerformanceWatcher : public SocketPerformanceWatcher {
- public:
- TestSocketPerformanceWatcher() = default;
- TestSocketPerformanceWatcher(const TestSocketPerformanceWatcher&) = delete;
- TestSocketPerformanceWatcher& operator=(const TestSocketPerformanceWatcher&) =
- delete;
- ~TestSocketPerformanceWatcher() override = default;
- bool ShouldNotifyUpdatedRTT() const override { return true; }
- void OnUpdatedRTTAvailable(const base::TimeDelta& rtt) override {}
- void OnConnectionChanged() override { connection_changed_count_++; }
- size_t connection_changed_count() const { return connection_changed_count_; }
- private:
- size_t connection_changed_count_ = 0u;
- };
- // TestSocketPerformanceWatcher requires kernel support for tcp_info struct, and
- // so it is enabled only on certain platforms.
- #if defined(TCP_INFO) || BUILDFLAG(IS_LINUX) || BUILDFLAG(IS_CHROMEOS)
- #define MAYBE_TestSocketPerformanceWatcher TestSocketPerformanceWatcher
- #else
- #define MAYBE_TestSocketPerformanceWatcher TestSocketPerformanceWatcher
- #endif
- // Tests if the socket performance watcher is notified if the same socket is
- // used for a different connection.
- TEST_F(TCPClientSocketTest, MAYBE_TestSocketPerformanceWatcher) {
- const size_t kNumIPs = 2;
- IPAddressList ip_list;
- for (size_t i = 0; i < kNumIPs; ++i)
- ip_list.push_back(IPAddress(72, 14, 213, i));
- auto watcher = std::make_unique<TestSocketPerformanceWatcher>();
- TestSocketPerformanceWatcher* watcher_ptr = watcher.get();
- std::vector<std::string> aliases({"example.com"});
- TCPClientSocket socket(
- AddressList::CreateFromIPAddressList(ip_list, std::move(aliases)),
- std::move(watcher), nullptr, nullptr, NetLogSource());
- EXPECT_THAT(socket.Bind(IPEndPoint(IPAddress::IPv4Localhost(), 0)), IsOk());
- TestCompletionCallback connect_callback;
- ASSERT_NE(OK, connect_callback.GetResult(
- socket.Connect(connect_callback.callback())));
- EXPECT_EQ(kNumIPs - 1, watcher_ptr->connection_changed_count());
- }
- // On Android, where socket tagging is supported, verify that
- // TCPClientSocket::Tag works as expected.
- #if BUILDFLAG(IS_ANDROID)
- TEST_F(TCPClientSocketTest, Tag) {
- if (!CanGetTaggedBytes()) {
- DVLOG(0) << "Skipping test - GetTaggedBytes unsupported.";
- return;
- }
- // Start test server.
- EmbeddedTestServer test_server;
- test_server.AddDefaultHandlers(base::FilePath());
- ASSERT_TRUE(test_server.Start());
- AddressList addr_list;
- ASSERT_TRUE(test_server.GetAddressList(&addr_list));
- TCPClientSocket s(addr_list, nullptr, nullptr, nullptr, NetLogSource());
- // Verify TCP connect packets are tagged and counted properly.
- int32_t tag_val1 = 0x12345678;
- uint64_t old_traffic = GetTaggedBytes(tag_val1);
- SocketTag tag1(SocketTag::UNSET_UID, tag_val1);
- s.ApplySocketTag(tag1);
- TestCompletionCallback connect_callback;
- int connect_result = s.Connect(connect_callback.callback());
- EXPECT_THAT(connect_callback.GetResult(connect_result), IsOk());
- EXPECT_GT(GetTaggedBytes(tag_val1), old_traffic);
- // Verify socket can be retagged with a new value and the current process's
- // UID.
- int32_t tag_val2 = 0x87654321;
- old_traffic = GetTaggedBytes(tag_val2);
- SocketTag tag2(getuid(), tag_val2);
- s.ApplySocketTag(tag2);
- const char kRequest1[] = "GET / HTTP/1.0";
- scoped_refptr<IOBuffer> write_buffer1 =
- base::MakeRefCounted<StringIOBuffer>(kRequest1);
- TestCompletionCallback write_callback1;
- EXPECT_EQ(s.Write(write_buffer1.get(), strlen(kRequest1),
- write_callback1.callback(), TRAFFIC_ANNOTATION_FOR_TESTS),
- static_cast<int>(strlen(kRequest1)));
- EXPECT_GT(GetTaggedBytes(tag_val2), old_traffic);
- // Verify socket can be retagged with a new value and the current process's
- // UID.
- old_traffic = GetTaggedBytes(tag_val1);
- s.ApplySocketTag(tag1);
- const char kRequest2[] = "\n\n";
- scoped_refptr<IOBufferWithSize> write_buffer2 =
- base::MakeRefCounted<IOBufferWithSize>(strlen(kRequest2));
- memmove(write_buffer2->data(), kRequest2, strlen(kRequest2));
- TestCompletionCallback write_callback2;
- EXPECT_EQ(s.Write(write_buffer2.get(), strlen(kRequest2),
- write_callback2.callback(), TRAFFIC_ANNOTATION_FOR_TESTS),
- static_cast<int>(strlen(kRequest2)));
- EXPECT_GT(GetTaggedBytes(tag_val1), old_traffic);
- s.Disconnect();
- }
- TEST_F(TCPClientSocketTest, TagAfterConnect) {
- if (!CanGetTaggedBytes()) {
- DVLOG(0) << "Skipping test - GetTaggedBytes unsupported.";
- return;
- }
- // Start test server.
- EmbeddedTestServer test_server;
- test_server.AddDefaultHandlers(base::FilePath());
- ASSERT_TRUE(test_server.Start());
- AddressList addr_list;
- ASSERT_TRUE(test_server.GetAddressList(&addr_list));
- TCPClientSocket s(addr_list, nullptr, nullptr, nullptr, NetLogSource());
- // Connect socket.
- TestCompletionCallback connect_callback;
- int connect_result = s.Connect(connect_callback.callback());
- EXPECT_THAT(connect_callback.GetResult(connect_result), IsOk());
- // Verify socket can be tagged with a new value and the current process's
- // UID.
- int32_t tag_val2 = 0x87654321;
- uint64_t old_traffic = GetTaggedBytes(tag_val2);
- SocketTag tag2(getuid(), tag_val2);
- s.ApplySocketTag(tag2);
- const char kRequest1[] = "GET / HTTP/1.0";
- scoped_refptr<IOBuffer> write_buffer1 =
- base::MakeRefCounted<StringIOBuffer>(kRequest1);
- TestCompletionCallback write_callback1;
- EXPECT_EQ(s.Write(write_buffer1.get(), strlen(kRequest1),
- write_callback1.callback(), TRAFFIC_ANNOTATION_FOR_TESTS),
- static_cast<int>(strlen(kRequest1)));
- EXPECT_GT(GetTaggedBytes(tag_val2), old_traffic);
- // Verify socket can be retagged with a new value and the current process's
- // UID.
- int32_t tag_val1 = 0x12345678;
- old_traffic = GetTaggedBytes(tag_val1);
- SocketTag tag1(SocketTag::UNSET_UID, tag_val1);
- s.ApplySocketTag(tag1);
- const char kRequest2[] = "\n\n";
- scoped_refptr<IOBuffer> write_buffer2 =
- base::MakeRefCounted<StringIOBuffer>(kRequest2);
- TestCompletionCallback write_callback2;
- EXPECT_EQ(s.Write(write_buffer2.get(), strlen(kRequest2),
- write_callback2.callback(), TRAFFIC_ANNOTATION_FOR_TESTS),
- static_cast<int>(strlen(kRequest2)));
- EXPECT_GT(GetTaggedBytes(tag_val1), old_traffic);
- s.Disconnect();
- }
- #endif // BUILDFLAG(IS_ANDROID)
- // TCP socket that hangs indefinitely when establishing a connection.
- class NeverConnectingTCPClientSocket : public TCPClientSocket {
- public:
- NeverConnectingTCPClientSocket(
- const AddressList& addresses,
- std::unique_ptr<SocketPerformanceWatcher> socket_performance_watcher,
- NetworkQualityEstimator* network_quality_estimator,
- net::NetLog* net_log,
- const net::NetLogSource& source)
- : TCPClientSocket(addresses,
- std::move(socket_performance_watcher),
- network_quality_estimator,
- net_log,
- source) {}
- // Returns the number of times that ConnectInternal() was called.
- int connect_internal_counter() const { return connect_internal_counter_; }
- private:
- int ConnectInternal(const IPEndPoint& endpoint) override {
- connect_internal_counter_++;
- return ERR_IO_PENDING;
- }
- int connect_internal_counter_ = 0;
- };
- // Tests for closing sockets on suspend mode.
- #if defined(TCP_CLIENT_SOCKET_OBSERVES_SUSPEND)
- // Entering suspend mode shouldn't affect sockets that haven't connected yet, or
- // listening server sockets.
- TEST_F(TCPClientSocketTest, SuspendBeforeConnect) {
- IPAddress lo_address = IPAddress::IPv4Localhost();
- TCPServerSocket server(nullptr, NetLogSource());
- ASSERT_THAT(server.Listen(IPEndPoint(lo_address, 0), 1), IsOk());
- IPEndPoint server_address;
- ASSERT_THAT(server.GetLocalAddress(&server_address), IsOk());
- TCPClientSocket socket(AddressList(server_address), nullptr, nullptr, nullptr,
- NetLogSource());
- EXPECT_THAT(socket.Bind(IPEndPoint(lo_address, 0)), IsOk());
- IPEndPoint local_address_result;
- EXPECT_THAT(socket.GetLocalAddress(&local_address_result), IsOk());
- EXPECT_EQ(lo_address, local_address_result.address());
- TestCompletionCallback accept_callback;
- std::unique_ptr<StreamSocket> accepted_socket;
- ASSERT_THAT(server.Accept(&accepted_socket, accept_callback.callback()),
- IsError(ERR_IO_PENDING));
- Suspend();
- // Power notifications happen asynchronously, so have to wait for the socket
- // to be notified of the suspend event.
- base::RunLoop().RunUntilIdle();
- TestCompletionCallback connect_callback;
- int connect_result = socket.Connect(connect_callback.callback());
- ASSERT_THAT(accept_callback.WaitForResult(), IsOk());
- ASSERT_THAT(connect_callback.GetResult(connect_result), IsOk());
- EXPECT_TRUE(socket.IsConnected());
- EXPECT_TRUE(accepted_socket->IsConnected());
- }
- TEST_F(TCPClientSocketTest, SuspendDuringConnect) {
- IPAddress lo_address = IPAddress::IPv4Localhost();
- TCPServerSocket server(nullptr, NetLogSource());
- ASSERT_THAT(server.Listen(IPEndPoint(lo_address, 0), 1), IsOk());
- IPEndPoint server_address;
- ASSERT_THAT(server.GetLocalAddress(&server_address), IsOk());
- NeverConnectingTCPClientSocket socket(AddressList(server_address), nullptr,
- nullptr, nullptr, NetLogSource());
- EXPECT_THAT(socket.Bind(IPEndPoint(lo_address, 0)), IsOk());
- IPEndPoint local_address_result;
- EXPECT_THAT(socket.GetLocalAddress(&local_address_result), IsOk());
- EXPECT_EQ(lo_address, local_address_result.address());
- TestCompletionCallback connect_callback;
- int rv = socket.Connect(connect_callback.callback());
- ASSERT_THAT(rv, IsError(ERR_IO_PENDING));
- Suspend();
- EXPECT_THAT(connect_callback.WaitForResult(),
- IsError(ERR_NETWORK_IO_SUSPENDED));
- }
- TEST_F(TCPClientSocketTest, SuspendDuringConnectMultipleAddresses) {
- IPAddress lo_address = IPAddress::IPv4Localhost();
- TCPServerSocket server(nullptr, NetLogSource());
- ASSERT_THAT(server.Listen(IPEndPoint(IPAddress(0, 0, 0, 0), 0), 1), IsOk());
- IPEndPoint server_address;
- ASSERT_THAT(server.GetLocalAddress(&server_address), IsOk());
- AddressList address_list;
- address_list.push_back(
- IPEndPoint(IPAddress(127, 0, 0, 1), server_address.port()));
- address_list.push_back(
- IPEndPoint(IPAddress(127, 0, 0, 2), server_address.port()));
- NeverConnectingTCPClientSocket socket(address_list, nullptr, nullptr, nullptr,
- NetLogSource());
- EXPECT_THAT(socket.Bind(IPEndPoint(lo_address, 0)), IsOk());
- IPEndPoint local_address_result;
- EXPECT_THAT(socket.GetLocalAddress(&local_address_result), IsOk());
- EXPECT_EQ(lo_address, local_address_result.address());
- TestCompletionCallback connect_callback;
- int rv = socket.Connect(connect_callback.callback());
- ASSERT_THAT(rv, IsError(ERR_IO_PENDING));
- Suspend();
- EXPECT_THAT(connect_callback.WaitForResult(),
- IsError(ERR_NETWORK_IO_SUSPENDED));
- }
- TEST_F(TCPClientSocketTest, SuspendWhileIdle) {
- std::unique_ptr<StreamSocket> accepted_socket;
- std::unique_ptr<TCPClientSocket> client_socket;
- std::unique_ptr<ServerSocket> server_socket;
- CreateConnectedSockets(&accepted_socket, &client_socket, &server_socket);
- Suspend();
- // Power notifications happen asynchronously.
- base::RunLoop().RunUntilIdle();
- scoped_refptr<IOBuffer> buffer = base::MakeRefCounted<IOBuffer>(1);
- buffer->data()[0] = '1';
- TestCompletionCallback callback;
- // Check that the client socket is disconnected, and actions fail with
- // ERR_NETWORK_IO_SUSPENDED.
- EXPECT_FALSE(client_socket->IsConnected());
- EXPECT_THAT(client_socket->Read(buffer.get(), 1, callback.callback()),
- IsError(ERR_NETWORK_IO_SUSPENDED));
- EXPECT_THAT(client_socket->Write(buffer.get(), 1, callback.callback(),
- TRAFFIC_ANNOTATION_FOR_TESTS),
- IsError(ERR_NETWORK_IO_SUSPENDED));
- // Check that the accepted socket is disconnected, and actions fail with
- // ERR_NETWORK_IO_SUSPENDED.
- EXPECT_FALSE(accepted_socket->IsConnected());
- EXPECT_THAT(accepted_socket->Read(buffer.get(), 1, callback.callback()),
- IsError(ERR_NETWORK_IO_SUSPENDED));
- EXPECT_THAT(accepted_socket->Write(buffer.get(), 1, callback.callback(),
- TRAFFIC_ANNOTATION_FOR_TESTS),
- IsError(ERR_NETWORK_IO_SUSPENDED));
- // Reconnecting the socket should work.
- TestCompletionCallback connect_callback;
- int connect_result = client_socket->Connect(connect_callback.callback());
- accepted_socket.reset();
- TestCompletionCallback accept_callback;
- int accept_result =
- server_socket->Accept(&accepted_socket, accept_callback.callback());
- ASSERT_THAT(accept_callback.GetResult(accept_result), IsOk());
- EXPECT_THAT(connect_callback.GetResult(connect_result), IsOk());
- }
- TEST_F(TCPClientSocketTest, SuspendDuringRead) {
- std::unique_ptr<StreamSocket> accepted_socket;
- std::unique_ptr<TCPClientSocket> client_socket;
- CreateConnectedSockets(&accepted_socket, &client_socket);
- // Start a read. This shouldn't complete, since the other end of the pipe
- // writes no data.
- scoped_refptr<IOBuffer> read_buffer = base::MakeRefCounted<IOBuffer>(1);
- read_buffer->data()[0] = '1';
- TestCompletionCallback callback;
- ASSERT_THAT(client_socket->Read(read_buffer.get(), 1, callback.callback()),
- IsError(ERR_IO_PENDING));
- // Simulate a suspend event. Can't use a real power event, as it would affect
- // |accepted_socket| as well.
- client_socket->OnSuspend();
- EXPECT_THAT(callback.WaitForResult(), IsError(ERR_NETWORK_IO_SUSPENDED));
- // Check that the client socket really is disconnected.
- EXPECT_FALSE(client_socket->IsConnected());
- EXPECT_THAT(client_socket->Read(read_buffer.get(), 1, callback.callback()),
- IsError(ERR_NETWORK_IO_SUSPENDED));
- EXPECT_THAT(client_socket->Write(read_buffer.get(), 1, callback.callback(),
- TRAFFIC_ANNOTATION_FOR_TESTS),
- IsError(ERR_NETWORK_IO_SUSPENDED));
- }
- TEST_F(TCPClientSocketTest, SuspendDuringWrite) {
- std::unique_ptr<StreamSocket> accepted_socket;
- std::unique_ptr<TCPClientSocket> client_socket;
- CreateConnectedSockets(&accepted_socket, &client_socket);
- // Write to the socket until a write doesn't complete synchronously.
- const int kBufferSize = 4096;
- scoped_refptr<IOBuffer> write_buffer =
- base::MakeRefCounted<IOBuffer>(kBufferSize);
- memset(write_buffer->data(), '1', kBufferSize);
- TestCompletionCallback callback;
- while (true) {
- int rv =
- client_socket->Write(write_buffer.get(), kBufferSize,
- callback.callback(), TRAFFIC_ANNOTATION_FOR_TESTS);
- if (rv == ERR_IO_PENDING)
- break;
- ASSERT_GT(rv, 0);
- }
- // Simulate a suspend event. Can't use a real power event, as it would affect
- // |accepted_socket| as well.
- client_socket->OnSuspend();
- EXPECT_THAT(callback.WaitForResult(), IsError(ERR_NETWORK_IO_SUSPENDED));
- // Check that the client socket really is disconnected.
- EXPECT_FALSE(client_socket->IsConnected());
- EXPECT_THAT(client_socket->Read(write_buffer.get(), 1, callback.callback()),
- IsError(ERR_NETWORK_IO_SUSPENDED));
- EXPECT_THAT(client_socket->Write(write_buffer.get(), 1, callback.callback(),
- TRAFFIC_ANNOTATION_FOR_TESTS),
- IsError(ERR_NETWORK_IO_SUSPENDED));
- }
- TEST_F(TCPClientSocketTest, SuspendDuringReadAndWrite) {
- enum class ReadCallbackAction {
- kNone,
- kDestroySocket,
- kDisconnectSocket,
- kReconnectSocket,
- };
- for (ReadCallbackAction read_callback_action : {
- ReadCallbackAction::kNone,
- ReadCallbackAction::kDestroySocket,
- ReadCallbackAction::kDisconnectSocket,
- ReadCallbackAction::kReconnectSocket,
- }) {
- std::unique_ptr<StreamSocket> accepted_socket;
- std::unique_ptr<TCPClientSocket> client_socket;
- std::unique_ptr<ServerSocket> server_socket;
- CreateConnectedSockets(&accepted_socket, &client_socket, &server_socket);
- // Start a read. This shouldn't complete, since the other end of the pipe
- // writes no data.
- scoped_refptr<IOBuffer> read_buffer = base::MakeRefCounted<IOBuffer>(1);
- read_buffer->data()[0] = '1';
- TestCompletionCallback read_callback;
- // Used int the ReadCallbackAction::kReconnectSocket case, since can't run a
- // nested message loop in the read callback.
- TestCompletionCallback nested_connect_callback;
- int nested_connect_result;
- CompletionOnceCallback read_completion_once_callback =
- base::BindLambdaForTesting([&](int result) {
- EXPECT_FALSE(client_socket->IsConnected());
- switch (read_callback_action) {
- case ReadCallbackAction::kNone:
- break;
- case ReadCallbackAction::kDestroySocket:
- client_socket.reset();
- break;
- case ReadCallbackAction::kDisconnectSocket:
- client_socket->Disconnect();
- break;
- case ReadCallbackAction::kReconnectSocket: {
- TestCompletionCallback connect_callback;
- nested_connect_result =
- client_socket->Connect(nested_connect_callback.callback());
- break;
- }
- }
- read_callback.callback().Run(result);
- });
- ASSERT_THAT(client_socket->Read(read_buffer.get(), 1,
- std::move(read_completion_once_callback)),
- IsError(ERR_IO_PENDING));
- // Write to the socket until a write doesn't complete synchronously.
- const int kBufferSize = 4096;
- scoped_refptr<IOBuffer> write_buffer =
- base::MakeRefCounted<IOBuffer>(kBufferSize);
- memset(write_buffer->data(), '1', kBufferSize);
- TestCompletionCallback write_callback;
- while (true) {
- int rv = client_socket->Write(write_buffer.get(), kBufferSize,
- write_callback.callback(),
- TRAFFIC_ANNOTATION_FOR_TESTS);
- if (rv == ERR_IO_PENDING)
- break;
- ASSERT_GT(rv, 0);
- }
- // Simulate a suspend event. Can't use a real power event, as it would
- // affect |accepted_socket| as well.
- client_socket->OnSuspend();
- EXPECT_THAT(read_callback.WaitForResult(),
- IsError(ERR_NETWORK_IO_SUSPENDED));
- if (read_callback_action == ReadCallbackAction::kNone) {
- EXPECT_THAT(write_callback.WaitForResult(),
- IsError(ERR_NETWORK_IO_SUSPENDED));
- // Check that the client socket really is disconnected.
- EXPECT_FALSE(client_socket->IsConnected());
- EXPECT_THAT(
- client_socket->Read(read_buffer.get(), 1, read_callback.callback()),
- IsError(ERR_NETWORK_IO_SUSPENDED));
- EXPECT_THAT(
- client_socket->Write(write_buffer.get(), 1, write_callback.callback(),
- TRAFFIC_ANNOTATION_FOR_TESTS),
- IsError(ERR_NETWORK_IO_SUSPENDED));
- } else {
- // Each of the actions taken in the read callback will cancel the pending
- // write callback.
- EXPECT_FALSE(write_callback.have_result());
- }
- if (read_callback_action == ReadCallbackAction::kReconnectSocket) {
- // Finish establishing a connection, just to make sure the reconnect case
- // completely works.
- accepted_socket.reset();
- TestCompletionCallback accept_callback;
- int accept_result =
- server_socket->Accept(&accepted_socket, accept_callback.callback());
- ASSERT_THAT(accept_callback.GetResult(accept_result), IsOk());
- EXPECT_THAT(nested_connect_callback.GetResult(nested_connect_result),
- IsOk());
- }
- }
- }
- #endif // defined(TCP_CLIENT_SOCKET_OBSERVES_SUSPEND)
- // Scoped helper to override the TCP connect attempt policy.
- class OverrideTcpConnectAttemptTimeout {
- public:
- OverrideTcpConnectAttemptTimeout(double rtt_multipilier,
- base::TimeDelta min_timeout,
- base::TimeDelta max_timeout) {
- base::FieldTrialParams params;
- params[features::kTimeoutTcpConnectAttemptRTTMultiplier.name] =
- base::NumberToString(rtt_multipilier);
- params[features::kTimeoutTcpConnectAttemptMin.name] =
- base::NumberToString(min_timeout.InMilliseconds()) + "ms";
- params[features::kTimeoutTcpConnectAttemptMax.name] =
- base::NumberToString(max_timeout.InMilliseconds()) + "ms";
- scoped_feature_list_.InitAndEnableFeatureWithParameters(
- features::kTimeoutTcpConnectAttempt, params);
- }
- private:
- base::test::ScopedFeatureList scoped_feature_list_;
- };
- // Test fixture that uses a MOCK_TIME test environment, so time can
- // be advanced programmatically.
- class TCPClientSocketMockTimeTest : public testing::Test {
- public:
- TCPClientSocketMockTimeTest()
- : task_environment_(base::test::TaskEnvironment::MainThreadType::IO,
- base::test::TaskEnvironment::TimeSource::MOCK_TIME) {}
- protected:
- base::test::TaskEnvironment task_environment_;
- };
- // Tests that no TCP connect timeout is enforced by default (i.e.
- // when the feature is disabled).
- TEST_F(TCPClientSocketMockTimeTest, NoConnectAttemptTimeoutByDefault) {
- IPEndPoint server_address(IPAddress::IPv4Localhost(), 80);
- NeverConnectingTCPClientSocket socket(AddressList(server_address), nullptr,
- nullptr, nullptr, NetLogSource());
- TestCompletionCallback connect_callback;
- int rv = socket.Connect(connect_callback.callback());
- ASSERT_THAT(rv, IsError(ERR_IO_PENDING));
- // After 4 minutes, the socket should still be connecting.
- task_environment_.FastForwardBy(base::Minutes(4));
- EXPECT_FALSE(connect_callback.have_result());
- EXPECT_FALSE(socket.IsConnected());
- // 1 attempt was made.
- EXPECT_EQ(1, socket.connect_internal_counter());
- }
- // Tests that the maximum timeout is used when there is no estimated
- // RTT.
- TEST_F(TCPClientSocketMockTimeTest, ConnectAttemptTimeoutUsesMaxWhenNoRTT) {
- OverrideTcpConnectAttemptTimeout override_timeout(1, base::Seconds(4),
- base::Seconds(10));
- IPEndPoint server_address(IPAddress::IPv4Localhost(), 80);
- // Pass a null NetworkQualityEstimator, so the TCPClientSocket is unable to
- // estimate the RTT.
- NeverConnectingTCPClientSocket socket(AddressList(server_address), nullptr,
- nullptr, nullptr, NetLogSource());
- // Start connecting.
- TestCompletionCallback connect_callback;
- int rv = socket.Connect(connect_callback.callback());
- ASSERT_THAT(rv, IsError(ERR_IO_PENDING));
- // Advance to t=3.1s
- // Should still be pending, as this is before the minimum timeout.
- task_environment_.FastForwardBy(base::Milliseconds(3100));
- EXPECT_FALSE(connect_callback.have_result());
- EXPECT_FALSE(socket.IsConnected());
- // Advance to t=4.1s
- // Should still be pending. This is after the minimum timeout, but before the
- // maximum.
- task_environment_.FastForwardBy(base::Seconds(1));
- EXPECT_FALSE(connect_callback.have_result());
- EXPECT_FALSE(socket.IsConnected());
- // Advance to t=10.1s
- // Should now be timed out, as this is after the maximum timeout.
- task_environment_.FastForwardBy(base::Seconds(6));
- rv = connect_callback.GetResult(rv);
- ASSERT_THAT(rv, IsError(ERR_TIMED_OUT));
- // 1 attempt was made.
- EXPECT_EQ(1, socket.connect_internal_counter());
- }
- // Tests that the minimum timeout is used when the adaptive timeout using RTT
- // ends up being too low.
- TEST_F(TCPClientSocketMockTimeTest, ConnectAttemptTimeoutUsesMinWhenRTTLow) {
- OverrideTcpConnectAttemptTimeout override_timeout(5, base::Seconds(4),
- base::Seconds(10));
- // Set the estimated RTT to 1 millisecond.
- TestNetworkQualityEstimator network_quality_estimator;
- network_quality_estimator.SetStartTimeNullTransportRtt(base::Milliseconds(1));
- IPEndPoint server_address(IPAddress::IPv4Localhost(), 80);
- NeverConnectingTCPClientSocket socket(AddressList(server_address), nullptr,
- &network_quality_estimator, nullptr,
- NetLogSource());
- // Start connecting.
- TestCompletionCallback connect_callback;
- int rv = socket.Connect(connect_callback.callback());
- ASSERT_THAT(rv, IsError(ERR_IO_PENDING));
- // Advance to t=1.1s
- // Should be pending, since although the adaptive timeout has been reached, it
- // is lower than the minimum timeout.
- task_environment_.FastForwardBy(base::Milliseconds(1100));
- EXPECT_FALSE(connect_callback.have_result());
- EXPECT_FALSE(socket.IsConnected());
- // Advance to t=4.1s
- // Should have timed out due to hitting the minimum timeout.
- task_environment_.FastForwardBy(base::Seconds(3));
- rv = connect_callback.GetResult(rv);
- ASSERT_THAT(rv, IsError(ERR_TIMED_OUT));
- // 1 attempt was made.
- EXPECT_EQ(1, socket.connect_internal_counter());
- }
- // Tests that the maximum timeout is used when the adaptive timeout from RTT is
- // too high.
- TEST_F(TCPClientSocketMockTimeTest, ConnectAttemptTimeoutUsesMinWhenRTTHigh) {
- OverrideTcpConnectAttemptTimeout override_timeout(5, base::Seconds(4),
- base::Seconds(10));
- // Set the estimated RTT to 5 seconds.
- TestNetworkQualityEstimator network_quality_estimator;
- network_quality_estimator.SetStartTimeNullTransportRtt(base::Seconds(5));
- IPEndPoint server_address(IPAddress::IPv4Localhost(), 80);
- NeverConnectingTCPClientSocket socket(AddressList(server_address), nullptr,
- &network_quality_estimator, nullptr,
- NetLogSource());
- // Start connecting.
- TestCompletionCallback connect_callback;
- int rv = socket.Connect(connect_callback.callback());
- ASSERT_THAT(rv, IsError(ERR_IO_PENDING));
- // Advance to t=10.1s
- // The socket should have timed out due to hitting the maximum timeout. Had
- // the adaptive timeout been used, the socket would instead be timing out at
- // t=25s.
- task_environment_.FastForwardBy(base::Milliseconds(10100));
- rv = connect_callback.GetResult(rv);
- ASSERT_THAT(rv, IsError(ERR_TIMED_OUT));
- // 1 attempt was made.
- EXPECT_EQ(1, socket.connect_internal_counter());
- }
- // Tests that an adaptive timeout is used for TCP connection attempts based on
- // the estimated RTT.
- TEST_F(TCPClientSocketMockTimeTest, ConnectAttemptTimeoutUsesRTT) {
- OverrideTcpConnectAttemptTimeout override_timeout(5, base::Seconds(4),
- base::Seconds(10));
- // Set the estimated RTT to 1 second. Since the multiplier is set to 5, the
- // total adaptive timeout will be 5 seconds.
- TestNetworkQualityEstimator network_quality_estimator;
- network_quality_estimator.SetStartTimeNullTransportRtt(base::Seconds(1));
- IPEndPoint server_address(IPAddress::IPv4Localhost(), 80);
- NeverConnectingTCPClientSocket socket(AddressList(server_address), nullptr,
- &network_quality_estimator, nullptr,
- NetLogSource());
- // Start connecting.
- TestCompletionCallback connect_callback;
- int rv = socket.Connect(connect_callback.callback());
- ASSERT_THAT(rv, IsError(ERR_IO_PENDING));
- // Advance to t=4.1s
- // The socket should still be pending. Had the minimum timeout been enforced,
- // it would instead have timed out now.
- task_environment_.FastForwardBy(base::Milliseconds(4100));
- EXPECT_FALSE(connect_callback.have_result());
- EXPECT_FALSE(socket.IsConnected());
- // Advance to t=5.1s
- // The adaptive timeout was at t=5s, so it should now be timed out.
- task_environment_.FastForwardBy(base::Seconds(1));
- rv = connect_callback.GetResult(rv);
- ASSERT_THAT(rv, IsError(ERR_TIMED_OUT));
- // 1 attempt was made.
- EXPECT_EQ(1, socket.connect_internal_counter());
- }
- // Tests that when multiple TCP connect attempts are made, the timeout for each
- // one is applied independently.
- TEST_F(TCPClientSocketMockTimeTest, ConnectAttemptTimeoutIndependent) {
- OverrideTcpConnectAttemptTimeout override_timeout(5, base::Seconds(4),
- base::Seconds(10));
- // This test will attempt connecting to 5 endpoints.
- const size_t kNumIps = 5;
- AddressList addresses;
- for (size_t i = 0; i < kNumIps; ++i)
- addresses.push_back(IPEndPoint(IPAddress::IPv4Localhost(), 80 + i));
- NeverConnectingTCPClientSocket socket(addresses, nullptr, nullptr, nullptr,
- NetLogSource());
- // Start connecting.
- TestCompletionCallback connect_callback;
- int rv = socket.Connect(connect_callback.callback());
- ASSERT_THAT(rv, IsError(ERR_IO_PENDING));
- // Advance to t=49s
- // Should still be pending.
- task_environment_.FastForwardBy(base::Seconds(49));
- EXPECT_FALSE(connect_callback.have_result());
- EXPECT_FALSE(socket.IsConnected());
- // Advance to t=50.1s
- // All attempts should take 50 seconds to complete (5 attempts, 10 seconds
- // each). So by this point the overall connect attempt will have timed out.
- task_environment_.FastForwardBy(base::Milliseconds(1100));
- rv = connect_callback.GetResult(rv);
- ASSERT_THAT(rv, IsError(ERR_TIMED_OUT));
- // 5 attempts were made.
- EXPECT_EQ(5, socket.connect_internal_counter());
- }
- } // namespace
- } // namespace net
|