| Index: webrtc/p2p/client/basicportallocator_unittest.cc
|
| diff --git a/webrtc/p2p/client/basicportallocator_unittest.cc b/webrtc/p2p/client/basicportallocator_unittest.cc
|
| index 28a522fcd25426c6aae9f54d863f5ffea54ebefa..6d5d1086a9322e26ff9b1cf6d134f4862ce05507 100644
|
| --- a/webrtc/p2p/client/basicportallocator_unittest.cc
|
| +++ b/webrtc/p2p/client/basicportallocator_unittest.cc
|
| @@ -39,15 +39,12 @@
|
| using rtc::Thread;
|
|
|
| static const SocketAddress kClientAddr("11.11.11.11", 0);
|
| -static const SocketAddress kClientAddr2("22.22.22.22", 0);
|
| static const SocketAddress kLoopbackAddr("127.0.0.1", 0);
|
| static const SocketAddress kPrivateAddr("192.168.1.11", 0);
|
| static const SocketAddress kPrivateAddr2("192.168.1.12", 0);
|
| static const SocketAddress kClientIPv6Addr("2401:fa00:4:1000:be30:5bff:fee5:c3",
|
| 0);
|
| -static const SocketAddress kClientIPv6Addr2(
|
| - "2401:fa00:4:2000:be30:5bff:fee5:c3",
|
| - 0);
|
| +static const SocketAddress kClientAddr2("22.22.22.22", 0);
|
| static const SocketAddress kNatUdpAddr("77.77.77.77", rtc::NAT_SERVER_UDP_PORT);
|
| static const SocketAddress kNatTcpAddr("77.77.77.77", rtc::NAT_SERVER_TCP_PORT);
|
| static const SocketAddress kRemoteClientAddr("22.22.22.22", 0);
|
| @@ -59,13 +56,7 @@
|
| static const SocketAddress kRelaySslTcpIntAddr("99.99.99.2", 5004);
|
| static const SocketAddress kRelaySslTcpExtAddr("99.99.99.3", 5005);
|
| static const SocketAddress kTurnUdpIntAddr("99.99.99.4", 3478);
|
| -static const SocketAddress kTurnUdpIntIPv6Addr(
|
| - "2402:fb00:4:1000:be30:5bff:fee5:c3",
|
| - 3479);
|
| static const SocketAddress kTurnTcpIntAddr("99.99.99.5", 3478);
|
| -static const SocketAddress kTurnTcpIntIPv6Addr(
|
| - "2402:fb00:4:2000:be30:5bff:fee5:c3",
|
| - 3479);
|
| static const SocketAddress kTurnUdpExtAddr("99.99.99.6", 0);
|
|
|
| // Minimum and maximum port for port range tests.
|
| @@ -193,10 +184,12 @@
|
| turn_server.credentials = credentials;
|
|
|
| if (!udp_turn.IsNil()) {
|
| - turn_server.ports.push_back(ProtocolAddress(udp_turn, PROTO_UDP, false));
|
| + turn_server.ports.push_back(
|
| + ProtocolAddress(kTurnUdpIntAddr, PROTO_UDP, false));
|
| }
|
| if (!tcp_turn.IsNil()) {
|
| - turn_server.ports.push_back(ProtocolAddress(tcp_turn, PROTO_TCP, false));
|
| + turn_server.ports.push_back(
|
| + ProtocolAddress(kTurnTcpIntAddr, PROTO_TCP, false));
|
| }
|
| allocator_->AddTurnServer(turn_server);
|
| }
|
| @@ -239,8 +232,6 @@
|
| sid, content_name, component, ice_ufrag, ice_pwd);
|
| session->SignalPortReady.connect(this,
|
| &BasicPortAllocatorTest::OnPortReady);
|
| - session->SignalPortPruned.connect(this,
|
| - &BasicPortAllocatorTest::OnPortPruned);
|
| session->SignalCandidatesReady.connect(
|
| this, &BasicPortAllocatorTest::OnCandidatesReady);
|
| session->SignalCandidatesAllocationDone.connect(
|
| @@ -258,20 +249,6 @@
|
| ((pattern.port() == 0 &&
|
| (address.port() != 0 || IPIsAny(address.ipaddr()))) ||
|
| (pattern.port() != 0 && address.port() == pattern.port()));
|
| - }
|
| -
|
| - // Returns the number of ports that have matching type, protocol and
|
| - // address.
|
| - static int CountPorts(const std::vector<PortInterface*>& ports,
|
| - const std::string& type,
|
| - ProtocolType protocol,
|
| - const SocketAddress& client_addr) {
|
| - return std::count_if(
|
| - ports.begin(), ports.end(),
|
| - [type, protocol, client_addr](PortInterface* port) {
|
| - return port->Type() == type && port->GetProtocol() == protocol &&
|
| - port->Network()->GetBestIP() == client_addr.ipaddr();
|
| - });
|
| }
|
|
|
| // Find a candidate and return it.
|
| @@ -415,15 +392,6 @@
|
| EXPECT_NE(ready_ports.end(),
|
| std::find(ready_ports.begin(), ready_ports.end(), port));
|
| }
|
| - void OnPortPruned(PortAllocatorSession* ses, PortInterface* port) {
|
| - LOG(LS_INFO) << "OnPortPruned: " << port->ToString();
|
| - ports_.erase(std::remove(ports_.begin(), ports_.end(), port), ports_.end());
|
| - // Make sure the pruned port is not in ReadyPorts.
|
| - auto ready_ports = ses->ReadyPorts();
|
| - EXPECT_EQ(ready_ports.end(),
|
| - std::find(ready_ports.begin(), ready_ports.end(), port));
|
| - }
|
| -
|
| void OnCandidatesReady(PortAllocatorSession* ses,
|
| const std::vector<Candidate>& candidates) {
|
| for (const Candidate& candidate : candidates) {
|
| @@ -1206,145 +1174,6 @@
|
| EXPECT_EQ(3U, candidates_.size());
|
| }
|
|
|
| -// Test that if prune_turn_ports is set, TCP TurnPort will not
|
| -// be used if UDP TurnPort is used.
|
| -TEST_F(BasicPortAllocatorTest, TestUdpTurnPortDisablesTcpTurnPorts) {
|
| - turn_server_.AddInternalSocket(kTurnTcpIntAddr, PROTO_TCP);
|
| - AddInterface(kClientAddr);
|
| - allocator_.reset(new BasicPortAllocator(&network_manager_));
|
| - allocator_->SetConfiguration(allocator_->stun_servers(),
|
| - allocator_->turn_servers(), 0, true);
|
| - AddTurnServers(kTurnUdpIntAddr, kTurnTcpIntAddr);
|
| - allocator_->set_step_delay(kMinimumStepDelay);
|
| - allocator_->set_flags(allocator().flags() |
|
| - PORTALLOCATOR_ENABLE_SHARED_SOCKET |
|
| - PORTALLOCATOR_DISABLE_TCP);
|
| -
|
| - EXPECT_TRUE(CreateSession(ICE_CANDIDATE_COMPONENT_RTP));
|
| - session_->StartGettingPorts();
|
| - EXPECT_TRUE_WAIT(candidate_allocation_done_, kDefaultAllocationTimeout);
|
| - // Only 2 ports (one STUN and one TURN) are actually being used.
|
| - EXPECT_EQ(2U, session_->ReadyPorts().size());
|
| - // We have verified that each port, when it is added to |ports_|, it is found
|
| - // in |ready_ports|, and when it is pruned, it is not found in |ready_ports|,
|
| - // so we only need to verify the content in one of them.
|
| - EXPECT_EQ(2U, ports_.size());
|
| - EXPECT_EQ(1, CountPorts(ports_, "local", PROTO_UDP, kClientAddr));
|
| - EXPECT_EQ(1, CountPorts(ports_, "relay", PROTO_UDP, kClientAddr));
|
| - EXPECT_EQ(0, CountPorts(ports_, "relay", PROTO_TCP, kClientAddr));
|
| -
|
| - // We don't remove candidates, so the size of |candidates_| will depend on
|
| - // when the TCP TURN port becomes ready. If it is ready after the UDP TURN
|
| - // port becomes ready, its candidates will be used there will be 3 candidates.
|
| - // Otherwise there will be only 2 candidates.
|
| - EXPECT_LE(2U, candidates_.size());
|
| - // There will only be 2 candidates in |ready_candidates| because it only
|
| - // includes the candidates in the ready ports.
|
| - const std::vector<Candidate>& ready_candidates = session_->ReadyCandidates();
|
| - EXPECT_EQ(2U, ready_candidates.size());
|
| - EXPECT_PRED4(HasCandidate, ready_candidates, "local", "udp", kClientAddr);
|
| - EXPECT_PRED4(HasCandidate, ready_candidates, "relay", "udp",
|
| - rtc::SocketAddress(kTurnUdpExtAddr.ipaddr(), 0));
|
| -}
|
| -
|
| -// Tests that if prune_turn_ports is set, IPv4 TurnPort will not
|
| -// be used if IPv6 TurnPort is used.
|
| -TEST_F(BasicPortAllocatorTest, TestIPv6TurnPortDisablesIPv4TurnPorts) {
|
| - turn_server_.AddInternalSocket(kTurnUdpIntIPv6Addr, PROTO_UDP);
|
| - // Add two IP addresses on the same interface.
|
| - AddInterface(kClientAddr, "net1");
|
| - AddInterface(kClientIPv6Addr, "net1");
|
| - allocator_.reset(new BasicPortAllocator(&network_manager_));
|
| - allocator_->SetConfiguration(allocator_->stun_servers(),
|
| - allocator_->turn_servers(), 0, true);
|
| - AddTurnServers(kTurnUdpIntIPv6Addr, rtc::SocketAddress());
|
| -
|
| - allocator_->set_step_delay(kMinimumStepDelay);
|
| - allocator_->set_flags(allocator().flags() |
|
| - PORTALLOCATOR_ENABLE_SHARED_SOCKET |
|
| - PORTALLOCATOR_ENABLE_IPV6 | PORTALLOCATOR_DISABLE_TCP);
|
| -
|
| - EXPECT_TRUE(CreateSession(ICE_CANDIDATE_COMPONENT_RTP));
|
| - session_->StartGettingPorts();
|
| - EXPECT_TRUE_WAIT(candidate_allocation_done_, kDefaultAllocationTimeout);
|
| - rtc::Thread::Current()->ProcessMessages(1000);
|
| - // Three ports (one IPv4 STUN, one IPv6 STUN and one TURN) will be ready.
|
| - EXPECT_EQ(3U, session_->ReadyPorts().size());
|
| - EXPECT_EQ(3U, ports_.size());
|
| - EXPECT_EQ(1, CountPorts(ports_, "local", PROTO_UDP, kClientAddr));
|
| - EXPECT_EQ(1, CountPorts(ports_, "local", PROTO_UDP, kClientIPv6Addr));
|
| - EXPECT_EQ(1, CountPorts(ports_, "relay", PROTO_UDP, kClientIPv6Addr));
|
| - EXPECT_EQ(0, CountPorts(ports_, "relay", PROTO_UDP, kClientAddr));
|
| -
|
| - // We don't remove candidates, so there may be more than 3 elemenets in
|
| - // |candidates_|, although |ready_candidates| only includes the candidates
|
| - // in |ready_ports|.
|
| - EXPECT_LE(3U, candidates_.size());
|
| - const std::vector<Candidate>& ready_candidates = session_->ReadyCandidates();
|
| - EXPECT_EQ(3U, ready_candidates.size());
|
| - EXPECT_PRED4(HasCandidate, ready_candidates, "local", "udp", kClientAddr);
|
| - EXPECT_PRED4(HasCandidate, ready_candidates, "relay", "udp",
|
| - rtc::SocketAddress(kTurnUdpExtAddr.ipaddr(), 0));
|
| -}
|
| -
|
| -// Tests that if prune_turn_ports is set, each network interface
|
| -// will has its own set of TurnPorts based on their priorities.
|
| -TEST_F(BasicPortAllocatorTest, TestEachInterfaceHasItsOwnTurnPorts) {
|
| - turn_server_.AddInternalSocket(kTurnTcpIntAddr, PROTO_TCP);
|
| - turn_server_.AddInternalSocket(kTurnUdpIntIPv6Addr, PROTO_UDP);
|
| - turn_server_.AddInternalSocket(kTurnTcpIntIPv6Addr, PROTO_TCP);
|
| - // Add two interfaces both having IPv4 and IPv6 addresses.
|
| - AddInterface(kClientAddr, "net1");
|
| - AddInterface(kClientIPv6Addr, "net1");
|
| - AddInterface(kClientAddr2, "net2");
|
| - AddInterface(kClientIPv6Addr2, "net2");
|
| - allocator_.reset(new BasicPortAllocator(&network_manager_));
|
| - allocator_->SetConfiguration(allocator_->stun_servers(),
|
| - allocator_->turn_servers(), 0, true);
|
| - // Have both UDP/TCP and IPv4/IPv6 TURN ports.
|
| - AddTurnServers(kTurnUdpIntAddr, kTurnTcpIntAddr);
|
| - AddTurnServers(kTurnUdpIntIPv6Addr, kTurnTcpIntIPv6Addr);
|
| -
|
| - allocator_->set_step_delay(kMinimumStepDelay);
|
| - allocator_->set_flags(allocator().flags() |
|
| - PORTALLOCATOR_ENABLE_SHARED_SOCKET |
|
| - PORTALLOCATOR_ENABLE_IPV6);
|
| - EXPECT_TRUE(CreateSession(ICE_CANDIDATE_COMPONENT_RTP));
|
| - session_->StartGettingPorts();
|
| - EXPECT_TRUE_WAIT(candidate_allocation_done_, kDefaultAllocationTimeout);
|
| - // 10 ports (4 STUN and 1 TURN ports on each interface) will be ready to use.
|
| - EXPECT_EQ(10U, session_->ReadyPorts().size());
|
| - EXPECT_EQ(10U, ports_.size());
|
| - EXPECT_EQ(1, CountPorts(ports_, "local", PROTO_UDP, kClientAddr));
|
| - EXPECT_EQ(1, CountPorts(ports_, "local", PROTO_UDP, kClientAddr2));
|
| - EXPECT_EQ(1, CountPorts(ports_, "local", PROTO_UDP, kClientIPv6Addr));
|
| - EXPECT_EQ(1, CountPorts(ports_, "local", PROTO_UDP, kClientIPv6Addr2));
|
| - EXPECT_EQ(1, CountPorts(ports_, "local", PROTO_TCP, kClientAddr));
|
| - EXPECT_EQ(1, CountPorts(ports_, "local", PROTO_TCP, kClientAddr2));
|
| - EXPECT_EQ(1, CountPorts(ports_, "local", PROTO_TCP, kClientIPv6Addr));
|
| - EXPECT_EQ(1, CountPorts(ports_, "local", PROTO_TCP, kClientIPv6Addr2));
|
| - EXPECT_EQ(1, CountPorts(ports_, "relay", PROTO_UDP, kClientIPv6Addr));
|
| - EXPECT_EQ(1, CountPorts(ports_, "relay", PROTO_UDP, kClientIPv6Addr2));
|
| -
|
| - // We don't remove candidates, so there may be more than 10 candidates
|
| - // in |candidates_|.
|
| - EXPECT_LE(10U, candidates_.size());
|
| - const std::vector<Candidate>& ready_candidates = session_->ReadyCandidates();
|
| - EXPECT_EQ(10U, ready_candidates.size());
|
| - EXPECT_PRED4(HasCandidate, ready_candidates, "local", "udp", kClientAddr);
|
| - EXPECT_PRED4(HasCandidate, ready_candidates, "local", "udp", kClientAddr2);
|
| - EXPECT_PRED4(HasCandidate, ready_candidates, "local", "udp", kClientIPv6Addr);
|
| - EXPECT_PRED4(HasCandidate, ready_candidates, "local", "udp",
|
| - kClientIPv6Addr2);
|
| - EXPECT_PRED4(HasCandidate, ready_candidates, "local", "tcp", kClientAddr);
|
| - EXPECT_PRED4(HasCandidate, ready_candidates, "local", "tcp", kClientAddr2);
|
| - EXPECT_PRED4(HasCandidate, ready_candidates, "local", "tcp", kClientIPv6Addr);
|
| - EXPECT_PRED4(HasCandidate, ready_candidates, "local", "tcp",
|
| - kClientIPv6Addr2);
|
| - EXPECT_PRED4(HasCandidate, ready_candidates, "relay", "udp",
|
| - rtc::SocketAddress(kTurnUdpExtAddr.ipaddr(), 0));
|
| -}
|
| -
|
| // Testing DNS resolve for the TURN server, this will test AllocationSequence
|
| // handling the unresolved address signal from TurnPort.
|
| TEST_F(BasicPortAllocatorTest, TestSharedSocketWithServerAddressResolve) {
|
| @@ -1640,7 +1469,7 @@
|
| AddInterface(kClientAddr);
|
| int pool_size = 1;
|
| allocator_->SetConfiguration(allocator_->stun_servers(),
|
| - allocator_->turn_servers(), pool_size, false);
|
| + allocator_->turn_servers(), pool_size);
|
| const PortAllocatorSession* peeked_session = allocator_->GetPooledSession();
|
| ASSERT_NE(nullptr, peeked_session);
|
| EXPECT_EQ_WAIT(true, peeked_session->CandidatesAllocationDone(),
|
| @@ -1676,7 +1505,7 @@
|
| AddInterface(kClientAddr);
|
| int pool_size = 1;
|
| allocator_->SetConfiguration(allocator_->stun_servers(),
|
| - allocator_->turn_servers(), pool_size, false);
|
| + allocator_->turn_servers(), pool_size);
|
| const PortAllocatorSession* peeked_session = allocator_->GetPooledSession();
|
| ASSERT_NE(nullptr, peeked_session);
|
| EXPECT_EQ_WAIT(true, peeked_session->CandidatesAllocationDone(),
|
|
|