blob: 2252e617aa40797aae81998a8845d89b47958b78 [file] [log] [blame]
Guo-wei Shiehf0bd1342015-05-15 17:26:521/*
2 * Copyright 2015 The WebRTC Project Authors. All rights reserved.
3 *
4 * Use of this source code is governed by a BSD-style license
5 * that can be found in the LICENSE file in the root of the source
6 * tree. An additional intellectual property rights grant can be found
7 * in the file PATENTS. All contributing project authors may
8 * be found in the AUTHORS file in the root of the source tree.
9 */
10
pbos67d12ed2017-01-02 16:42:3211#include <stdint.h>
12
kwiberg1ab92652016-04-27 14:22:5313#include <memory>
14
Guo-wei Shieh5aca27b2015-06-18 21:44:4115#include "webrtc/p2p/base/basicpacketsocketfactory.h"
Guo-wei Shiehf0bd1342015-05-15 17:26:5216#include "webrtc/p2p/base/teststunserver.h"
17#include "webrtc/p2p/stunprober/stunprober.h"
Edward Lemur76de83e2017-07-06 17:44:3418#include "webrtc/rtc_base/asyncresolverinterface.h"
19#include "webrtc/rtc_base/bind.h"
20#include "webrtc/rtc_base/checks.h"
21#include "webrtc/rtc_base/gunit.h"
22#include "webrtc/rtc_base/ssladapter.h"
23#include "webrtc/rtc_base/virtualsocketserver.h"
Guo-wei Shiehf0bd1342015-05-15 17:26:5224
Guo-wei Shiehf0bd1342015-05-15 17:26:5225using stunprober::StunProber;
26using stunprober::AsyncCallback;
Guo-wei Shiehf0bd1342015-05-15 17:26:5227
28namespace stunprober {
29
30namespace {
31
32const rtc::SocketAddress kLocalAddr("192.168.0.1", 0);
33const rtc::SocketAddress kStunAddr1("1.1.1.1", 3478);
34const rtc::SocketAddress kStunAddr2("1.1.1.2", 3478);
Guo-wei Shiehf42236b2015-05-19 04:36:2035const rtc::SocketAddress kFailedStunAddr("1.1.1.3", 3478);
36const rtc::SocketAddress kStunMappedAddr("77.77.77.77", 0);
Guo-wei Shiehf0bd1342015-05-15 17:26:5237
Guo-wei Shiehf0bd1342015-05-15 17:26:5238} // namespace
39
40class StunProberTest : public testing::Test {
41 public:
42 StunProberTest()
deadbeef92b90e42017-05-17 01:00:0643 : ss_(new rtc::VirtualSocketServer()),
nisse59a36ed2017-05-08 12:25:4144 main_(ss_.get()),
Guo-wei Shiehf0bd1342015-05-15 17:26:5245 result_(StunProber::SUCCESS),
46 stun_server_1_(cricket::TestStunServer::Create(rtc::Thread::Current(),
47 kStunAddr1)),
48 stun_server_2_(cricket::TestStunServer::Create(rtc::Thread::Current(),
49 kStunAddr2)) {
Guo-wei Shiehf42236b2015-05-19 04:36:2050 stun_server_1_->set_fake_stun_addr(kStunMappedAddr);
51 stun_server_2_->set_fake_stun_addr(kStunMappedAddr);
Guo-wei Shiehf0bd1342015-05-15 17:26:5252 rtc::InitializeSSL();
53 }
54
55 void set_expected_result(int result) { result_ = result; }
56
Guo-wei Shieh5aca27b2015-06-18 21:44:4157 void StartProbing(rtc::PacketSocketFactory* socket_factory,
58 const std::vector<rtc::SocketAddress>& addrs,
59 const rtc::NetworkManager::NetworkList& networks,
Guo-wei Shiehf0bd1342015-05-15 17:26:5260 bool shared_socket,
Peter Boström07e22e62015-10-07 10:23:2161 uint16_t interval,
62 uint16_t pings_per_ip) {
Guo-wei Shieh5aca27b2015-06-18 21:44:4163 prober.reset(
64 new StunProber(socket_factory, rtc::Thread::Current(), networks));
Guo-wei Shiehf42236b2015-05-19 04:36:2065 prober->Start(addrs, shared_socket, interval, pings_per_ip,
Guo-wei Shieh5aca27b2015-06-18 21:44:4166 100 /* timeout_ms */, [this](StunProber* prober, int result) {
67 this->StopCallback(prober, result);
68 });
Guo-wei Shiehf0bd1342015-05-15 17:26:5269 }
70
71 void RunProber(bool shared_mode) {
72 const int pings_per_ip = 3;
Guo-wei Shiehf42236b2015-05-19 04:36:2073 std::vector<rtc::SocketAddress> addrs;
Guo-wei Shieh5aca27b2015-06-18 21:44:4174 addrs.push_back(kStunAddr1);
75 addrs.push_back(kStunAddr2);
Guo-wei Shiehf42236b2015-05-19 04:36:2076 // Add a non-existing server. This shouldn't pollute the result.
Guo-wei Shieh5aca27b2015-06-18 21:44:4177 addrs.push_back(kFailedStunAddr);
Guo-wei Shiehf0bd1342015-05-15 17:26:5278
Guo-wei Shieh5aca27b2015-06-18 21:44:4179 rtc::Network ipv4_network1("test_eth0", "Test Network Adapter 1",
80 rtc::IPAddress(0x12345600U), 24);
81 ipv4_network1.AddIP(rtc::IPAddress(0x12345678));
82 rtc::NetworkManager::NetworkList networks;
83 networks.push_back(&ipv4_network1);
Guo-wei Shiehf0bd1342015-05-15 17:26:5284
kwiberg1ab92652016-04-27 14:22:5385 std::unique_ptr<rtc::BasicPacketSocketFactory> socket_factory(
Guo-wei Shieh5aca27b2015-06-18 21:44:4186 new rtc::BasicPacketSocketFactory());
Guo-wei Shiehf0bd1342015-05-15 17:26:5287
88 // Set up the expected results for verification.
89 std::set<std::string> srflx_addresses;
Guo-wei Shiehf42236b2015-05-19 04:36:2090 srflx_addresses.insert(kStunMappedAddr.ToString());
Peter Boström07e22e62015-10-07 10:23:2191 const uint32_t total_pings_tried =
92 static_cast<uint32_t>(pings_per_ip * addrs.size());
Guo-wei Shiehf0bd1342015-05-15 17:26:5293
Guo-wei Shiehf42236b2015-05-19 04:36:2094 // The reported total_pings should not count for pings sent to the
95 // kFailedStunAddr.
Peter Boström07e22e62015-10-07 10:23:2196 const uint32_t total_pings_reported = total_pings_tried - pings_per_ip;
Guo-wei Shiehf42236b2015-05-19 04:36:2097
Guo-wei Shieh5aca27b2015-06-18 21:44:4198 StartProbing(socket_factory.get(), addrs, networks, shared_mode, 3,
99 pings_per_ip);
Guo-wei Shiehf0bd1342015-05-15 17:26:52100
101 WAIT(stopped_, 1000);
102
103 StunProber::Stats stats;
Guo-wei Shiehf0bd1342015-05-15 17:26:52104 EXPECT_TRUE(prober->GetStats(&stats));
105 EXPECT_EQ(stats.success_percent, 100);
Guo-wei Shiehc4d23052015-06-09 04:03:48106 EXPECT_TRUE(stats.nat_type > stunprober::NATTYPE_NONE);
Guo-wei Shiehf42236b2015-05-19 04:36:20107 EXPECT_EQ(stats.srflx_addrs, srflx_addresses);
Peter Boström07e22e62015-10-07 10:23:21108 EXPECT_EQ(static_cast<uint32_t>(stats.num_request_sent),
Guo-wei Shiehf42236b2015-05-19 04:36:20109 total_pings_reported);
Peter Boström07e22e62015-10-07 10:23:21110 EXPECT_EQ(static_cast<uint32_t>(stats.num_response_received),
Guo-wei Shiehf42236b2015-05-19 04:36:20111 total_pings_reported);
Guo-wei Shiehf0bd1342015-05-15 17:26:52112 }
113
114 private:
Guo-wei Shieh5aca27b2015-06-18 21:44:41115 void StopCallback(StunProber* prober, int result) {
Guo-wei Shiehf0bd1342015-05-15 17:26:52116 EXPECT_EQ(result, result_);
117 stopped_ = true;
118 }
119
kwiberg1ab92652016-04-27 14:22:53120 std::unique_ptr<rtc::VirtualSocketServer> ss_;
nisse59a36ed2017-05-08 12:25:41121 rtc::AutoSocketServerThread main_;
kwiberg1ab92652016-04-27 14:22:53122 std::unique_ptr<StunProber> prober;
Guo-wei Shiehf0bd1342015-05-15 17:26:52123 int result_ = 0;
124 bool stopped_ = false;
kwiberg1ab92652016-04-27 14:22:53125 std::unique_ptr<cricket::TestStunServer> stun_server_1_;
126 std::unique_ptr<cricket::TestStunServer> stun_server_2_;
Guo-wei Shiehf0bd1342015-05-15 17:26:52127};
128
Guo-wei Shiehf0bd1342015-05-15 17:26:52129TEST_F(StunProberTest, NonSharedMode) {
130 RunProber(false);
131}
132
133TEST_F(StunProberTest, SharedMode) {
134 RunProber(true);
135}
136
137} // namespace stunprober