Better determination of Symmetric NAT

webrtc/p2p/stunprober/stunprober.cc

 /*
  *  Copyright 2015 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.
  */
 
 #include <iostream>
 #include <map>
 #include <set>
 #include <string>
 
 #include "webrtc/base/bind.h"
 #include "webrtc/base/checks.h"
 #include "webrtc/base/helpers.h"
 #include "webrtc/base/timeutils.h"
 #include "webrtc/p2p/base/stun.h"
 #include "webrtc/p2p/stunprober/stunprober.h"
 
 namespace stunprober {
 
 namespace {
 
-void IncrementCounterByAddress(std::map<rtc::IPAddress, int>* counter_per_ip,
-                               const rtc::IPAddress& ip) {
+template <typename T>
+void IncrementCounterByAddress(std::map<T, int>* counter_per_ip, const T& ip) {
   counter_per_ip->insert(std::make_pair(ip, 0)).first->second++;
 }
 
+bool behind_nat(NatType nat_type) {
+  return nat_type > stunprober::NATTYPE_NONE;
+}
+
 }  // namespace
 
 // A requester tracks the requests and responses from a single socket to many
 // STUN servers
 class StunProber::Requester {
  public:
   // Each Request maps to a request and response.
   struct Request {
     // Actual time the STUN bind request was sent.
     int64 sent_time_ms = 0;
@@ skipping 369 matching lines @@
     if (!SendNextRequest()) {
       End(GENERIC_FAILURE, 0);
       return;
     }
     next_request_time_ms_ = now + interval_ms_;
   }
   task_runner_->PostTask(
       rtc::Bind(&StunProber::MaybeScheduleStunRequests, this), 1 /* ms */);
 }
 
-bool StunProber::GetStats(StunProber::Stats* prob_stats) {
+bool StunProber::GetStats(StunProber::Stats* prob_stats) const {
   // No need to be on the same thread.
   if (!prob_stats) {
     return false;
   }
 
   StunProber::Stats stats;
 
   int rtt_sum = 0;
-  bool behind_nat_set = false;
   int64 first_sent_time = 0;
   int64 last_sent_time = 0;
+  NatType nat_type = NATTYPE_INVALID;
 
   // Track of how many srflx IP that we have seen.
   std::set<rtc::IPAddress> srflx_ips;
 
   // If we're not receiving any response on a given IP, all requests sent to
   // that IP should be ignored as this could just be an DNS error.
-  std::map<rtc::IPAddress, int> num_response_per_ip;
-  std::map<rtc::IPAddress, int> num_request_per_ip;
+  std::map<rtc::IPAddress, int> num_response_per_server;
+  std::map<rtc::IPAddress, int> num_request_per_server;
 
   for (auto* requester : requesters_) {
+    std::map<rtc::SocketAddress, int> num_response_per_srflx_addr;
     for (auto request : requester->requests()) {
       if (request->sent_time_ms <= 0) {
         continue;
       }
 
-      IncrementCounterByAddress(&num_request_per_ip, request->server_addr);
+      IncrementCounterByAddress(&num_request_per_server, request->server_addr);
 
       if (!first_sent_time) {
         first_sent_time = request->sent_time_ms;
       }
       last_sent_time = request->sent_time_ms;
 
       if (request->received_time_ms < request->sent_time_ms) {
         continue;
       }
 
-      IncrementCounterByAddress(&num_response_per_ip, request->server_addr);
+      IncrementCounterByAddress(&num_response_per_server, request->server_addr);
+      IncrementCounterByAddress(&num_response_per_srflx_addr,
+                                request->srflx_addr);
 
       rtt_sum += request->rtt();
-      if (!behind_nat_set) {
-        stats.behind_nat = request->behind_nat;
-        behind_nat_set = true;
-      } else if (stats.behind_nat != request->behind_nat) {
+      if (nat_type == NATTYPE_INVALID) {
+        nat_type = request->behind_nat ? NATTYPE_UNKNOWN : NATTYPE_NONE;
+      } else if (behind_nat(nat_type) != request->behind_nat) {
         // Detect the inconsistency in NAT presence.
         return false;
       }
       stats.srflx_addrs.insert(request->srflx_addr.ToString());
       srflx_ips.insert(request->srflx_addr.ipaddr());
     }
+
+    // If we're using shared mode and seeing >1 srflx addresses for a single
+    // requester, it's symmetric NAT.
+    if (shared_socket_mode_ && num_response_per_srflx_addr.size() > 1) {
+      nat_type = NATTYPE_SYMMETRIC;
+    }
   }
 
   // We're probably not behind a regular NAT. We have more than 1 distinct
   // server reflexive IPs.
   if (srflx_ips.size() > 1) {
     return false;
   }
 
   int num_sent = 0;
   int num_received = 0;
   int num_server_ip_with_response = 0;
 
-  for (const auto& kv : num_response_per_ip) {
+  for (const auto& kv : num_response_per_server) {
     DCHECK_GT(kv.second, 0);
     num_server_ip_with_response++;
     num_received += kv.second;
-    num_sent += num_request_per_ip[kv.first];
+    num_sent += num_request_per_server[kv.first];
   }
 
   // Not receiving any response, the trial is inconclusive.
   if (!num_received) {
     return false;
   }
 
+  stats.nat_type = nat_type;
+
   // Shared mode is only true if we use the shared socket and there are more
   // than 1 responding servers.
   stats.shared_socket_mode =
       shared_socket_mode_ && (num_server_ip_with_response > 1);
 
+  if (stats.shared_socket_mode && nat_type == NATTYPE_UNKNOWN) {
+    stats.nat_type = NATTYPE_NON_SYMMETRIC;
+  }
+
   stats.host_ip = local_addr_.ToString();
   stats.num_request_sent = num_sent;
   stats.num_response_received = num_received;
   stats.target_request_interval_ns = interval_ms_ * 1000;
-  stats.symmetric_nat =
-      stats.srflx_addrs.size() > static_cast<size_t>(GetTotalServerSockets());
 
   if (num_sent) {
     stats.success_percent = static_cast<int>(100 * num_received / num_sent);
   }
 
   if (num_sent > 1) {
     stats.actual_request_interval_ns =
         (1000 * (last_sent_time - first_sent_time)) / (num_sent - 1);
   }
 
@@ skipping 10 matching lines @@
   if (!finished_callback_.empty()) {
     AsyncCallback callback = finished_callback_;
     finished_callback_ = AsyncCallback();
 
     // Callback at the last since the prober might be deleted in the callback.
     callback(status);
   }
 }
 
 }  // namespace stunprober