Add a flag to filter out high-cost networks.

webrtc/p2p/client/basicportallocator_unittest.cc

 /*
  *  Copyright 2009 The WebRTC Project Authors. All rights reserved.
  *
  *  Use of this source code is governed by a BSD-style license
  *  that can be found in the LICENSE file in the root of the source
  *  tree. An additional intellectual property rights grant can be found
  *  in the file PATENTS.  All contributing project authors may
  *  be found in the AUTHORS file in the root of the source tree.
  */
 
@@ skipping 501 matching lines @@
                                    rtc::ADAPTER_TYPE_WIFI);
   EXPECT_TRUE(CreateSession(ICE_CANDIDATE_COMPONENT_RTP));
   session_->set_flags(PORTALLOCATOR_DISABLE_STUN | PORTALLOCATOR_DISABLE_RELAY |
                       PORTALLOCATOR_DISABLE_TCP);
   session_->StartGettingPorts();
   EXPECT_TRUE_WAIT(candidate_allocation_done_, kDefaultAllocationTimeout);
   EXPECT_EQ(1U, candidates_.size());
   EXPECT_EQ(0x12345602U, candidates_[0].address().ip());
 }
 
+// Test that high cost networks are filtered if the flag
+// PORTALLOCATOR_DISABLE_COSTLY_NETWORKS is set.
+TEST_F(BasicPortAllocatorTest, TestGatherLowCostNetworkOnly) {
+  SocketAddress addr_wifi(IPAddress(0x12345600U), 0);
+  SocketAddress addr_cellular(IPAddress(0x12345601U), 0);
+  SocketAddress addr_unknown1(IPAddress(0x12345602U), 0);
+  SocketAddress addr_unknown2(IPAddress(0x12345603U), 0);
+  // If both Wi-Fi and cellular interfaces are present, only gather on the Wi-Fi
+  // interface.
+  AddInterface(addr_wifi, "test_wlan0", rtc::ADAPTER_TYPE_WIFI);
+  AddInterface(addr_cellular, "test_cell0", rtc::ADAPTER_TYPE_CELLULAR);
+  allocator().set_flags(cricket::PORTALLOCATOR_DISABLE_STUN |
+                        cricket::PORTALLOCATOR_DISABLE_RELAY |
+                        cricket::PORTALLOCATOR_DISABLE_TCP |
+                        cricket::PORTALLOCATOR_DISABLE_COSTLY_NETWORKS);
+  EXPECT_TRUE(CreateSession(cricket::ICE_CANDIDATE_COMPONENT_RTP));
+  session_->StartGettingPorts();
+  EXPECT_TRUE_WAIT(candidate_allocation_done_, kDefaultAllocationTimeout);
+  EXPECT_EQ(1U, candidates_.size());
+  EXPECT_TRUE(addr_wifi.EqualIPs(candidates_[0].address()));
+
+  // If both cellular and unknown interfaces are present, only gather on the
+  // unknown interfaces.
+  candidates_.clear();
+  candidate_allocation_done_ = false;
+  RemoveInterface(addr_wifi);
+  AddInterface(addr_unknown1, "test_unknown0", rtc::ADAPTER_TYPE_UNKNOWN);
+  AddInterface(addr_unknown2, "test_unknown1", rtc::ADAPTER_TYPE_UNKNOWN);
+  session_->StartGettingPorts();
+  EXPECT_TRUE_WAIT(candidate_allocation_done_, kDefaultAllocationTimeout);
+  EXPECT_EQ(2U, candidates_.size());
+  EXPECT_TRUE((addr_unknown1.EqualIPs(candidates_[0].address()) &&
+               addr_unknown2.EqualIPs(candidates_[1].address())) ||
+              (addr_unknown1.EqualIPs(candidates_[1].address()) &&
+               addr_unknown2.EqualIPs(candidates_[0].address())));
+
+  // If Wi-Fi, cellular, unknown interfaces are all present, only gather on the
+  // Wi-Fi interface.
+  candidates_.clear();
+  candidate_allocation_done_ = false;
+  AddInterface(addr_wifi, "test_wlan0", rtc::ADAPTER_TYPE_WIFI);
+  session_->StartGettingPorts();
+  EXPECT_TRUE_WAIT(candidate_allocation_done_, kDefaultAllocationTimeout);
+  EXPECT_EQ(1U, candidates_.size());
+  EXPECT_TRUE(addr_wifi.EqualIPs(candidates_[0].address()));
+}
+
 // Tests that we allocator session not trying to allocate ports for every 250ms.
 TEST_F(BasicPortAllocatorTest, TestNoNetworkInterface) {
   EXPECT_TRUE(CreateSession(ICE_CANDIDATE_COMPONENT_RTP));
   session_->StartGettingPorts();
   // Waiting for one second to make sure BasicPortAllocatorSession has not
   // called OnAllocate multiple times. In old behavior it's called every 250ms.
   // When there are no network interfaces, each execution of OnAllocate will
   // result in SignalCandidatesAllocationDone signal.
   rtc::Thread::Current()->ProcessMessages(1000);
   EXPECT_TRUE(candidate_allocation_done_);
@@ skipping 951 matching lines @@
   for (const Candidate& candidate : candidates) {
     // Expect only relay candidates now that the filter is applied.
     EXPECT_EQ(std::string(RELAY_PORT_TYPE), candidate.type());
     // Expect that the raddr is emptied due to the CF_RELAY filter.
     EXPECT_EQ(candidate.related_address(),
               rtc::EmptySocketAddressWithFamily(candidate.address().family()));
   }
 }
 
 }  // namespace cricket