1 /*
2  *  Copyright 2019 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 
11 #include "p2p/base/basic_ice_controller.h"
12 
13 namespace {
14 
15 // The minimum improvement in RTT that justifies a switch.
16 const int kMinImprovement = 10;
17 
IsRelayRelay(const cricket::Connection * conn)18 bool IsRelayRelay(const cricket::Connection* conn) {
19   return conn->local_candidate().type() == cricket::RELAY_PORT_TYPE &&
20          conn->remote_candidate().type() == cricket::RELAY_PORT_TYPE;
21 }
22 
IsUdp(const cricket::Connection * conn)23 bool IsUdp(const cricket::Connection* conn) {
24   return conn->local_candidate().relay_protocol() == cricket::UDP_PROTOCOL_NAME;
25 }
26 
27 // TODO(qingsi) Use an enum to replace the following constants for all
28 // comparision results.
29 static constexpr int a_is_better = 1;
30 static constexpr int b_is_better = -1;
31 static constexpr int a_and_b_equal = 0;
32 
LocalCandidateUsesPreferredNetwork(const cricket::Connection * conn,absl::optional<rtc::AdapterType> network_preference)33 bool LocalCandidateUsesPreferredNetwork(
34     const cricket::Connection* conn,
35     absl::optional<rtc::AdapterType> network_preference) {
36   rtc::AdapterType network_type = conn->network()->type();
37   return network_preference.has_value() && (network_type == network_preference);
38 }
39 
CompareCandidatePairsByNetworkPreference(const cricket::Connection * a,const cricket::Connection * b,absl::optional<rtc::AdapterType> network_preference)40 int CompareCandidatePairsByNetworkPreference(
41     const cricket::Connection* a,
42     const cricket::Connection* b,
43     absl::optional<rtc::AdapterType> network_preference) {
44   bool a_uses_preferred_network =
45       LocalCandidateUsesPreferredNetwork(a, network_preference);
46   bool b_uses_preferred_network =
47       LocalCandidateUsesPreferredNetwork(b, network_preference);
48   if (a_uses_preferred_network && !b_uses_preferred_network) {
49     return a_is_better;
50   } else if (!a_uses_preferred_network && b_uses_preferred_network) {
51     return b_is_better;
52   }
53   return a_and_b_equal;
54 }
55 
56 }  // namespace
57 
58 namespace cricket {
59 
BasicIceController(const IceControllerFactoryArgs & args)60 BasicIceController::BasicIceController(const IceControllerFactoryArgs& args)
61     : ice_transport_state_func_(args.ice_transport_state_func),
62       ice_role_func_(args.ice_role_func),
63       is_connection_pruned_func_(args.is_connection_pruned_func),
64       field_trials_(args.ice_field_trials) {}
65 
~BasicIceController()66 BasicIceController::~BasicIceController() {}
67 
SetIceConfig(const IceConfig & config)68 void BasicIceController::SetIceConfig(const IceConfig& config) {
69   config_ = config;
70 }
71 
SetSelectedConnection(const Connection * selected_connection)72 void BasicIceController::SetSelectedConnection(
73     const Connection* selected_connection) {
74   selected_connection_ = selected_connection;
75 }
76 
AddConnection(const Connection * connection)77 void BasicIceController::AddConnection(const Connection* connection) {
78   connections_.push_back(connection);
79   unpinged_connections_.insert(connection);
80 }
81 
OnConnectionDestroyed(const Connection * connection)82 void BasicIceController::OnConnectionDestroyed(const Connection* connection) {
83   pinged_connections_.erase(connection);
84   unpinged_connections_.erase(connection);
85   connections_.erase(absl::c_find(connections_, connection));
86 }
87 
HasPingableConnection() const88 bool BasicIceController::HasPingableConnection() const {
89   int64_t now = rtc::TimeMillis();
90   return absl::c_any_of(connections_, [this, now](const Connection* c) {
91     return IsPingable(c, now);
92   });
93 }
94 
SelectConnectionToPing(int64_t last_ping_sent_ms)95 IceControllerInterface::PingResult BasicIceController::SelectConnectionToPing(
96     int64_t last_ping_sent_ms) {
97   // When the selected connection is not receiving or not writable, or any
98   // active connection has not been pinged enough times, use the weak ping
99   // interval.
100   bool need_more_pings_at_weak_interval =
101       absl::c_any_of(connections_, [](const Connection* conn) {
102         return conn->active() &&
103                conn->num_pings_sent() < MIN_PINGS_AT_WEAK_PING_INTERVAL;
104       });
105   int ping_interval = (weak() || need_more_pings_at_weak_interval)
106                           ? weak_ping_interval()
107                           : strong_ping_interval();
108 
109   const Connection* conn = nullptr;
110   if (rtc::TimeMillis() >= last_ping_sent_ms + ping_interval) {
111     conn = FindNextPingableConnection();
112   }
113   PingResult res(conn, std::min(ping_interval, check_receiving_interval()));
114   return res;
115 }
116 
MarkConnectionPinged(const Connection * conn)117 void BasicIceController::MarkConnectionPinged(const Connection* conn) {
118   if (conn && pinged_connections_.insert(conn).second) {
119     unpinged_connections_.erase(conn);
120   }
121 }
122 
123 // Returns the next pingable connection to ping.
FindNextPingableConnection()124 const Connection* BasicIceController::FindNextPingableConnection() {
125   int64_t now = rtc::TimeMillis();
126 
127   // Rule 1: Selected connection takes priority over non-selected ones.
128   if (selected_connection_ && selected_connection_->connected() &&
129       selected_connection_->writable() &&
130       WritableConnectionPastPingInterval(selected_connection_, now)) {
131     return selected_connection_;
132   }
133 
134   // Rule 2: If the channel is weak, we need to find a new writable and
135   // receiving connection, probably on a different network. If there are lots of
136   // connections, it may take several seconds between two pings for every
137   // non-selected connection. This will cause the receiving state of those
138   // connections to be false, and thus they won't be selected. This is
139   // problematic for network fail-over. We want to make sure at least one
140   // connection per network is pinged frequently enough in order for it to be
141   // selectable. So we prioritize one connection per network.
142   // Rule 2.1: Among such connections, pick the one with the earliest
143   // last-ping-sent time.
144   if (weak()) {
145     std::vector<const Connection*> pingable_selectable_connections;
146     absl::c_copy_if(GetBestWritableConnectionPerNetwork(),
147                     std::back_inserter(pingable_selectable_connections),
148                     [this, now](const Connection* conn) {
149                       return WritableConnectionPastPingInterval(conn, now);
150                     });
151     auto iter = absl::c_min_element(
152         pingable_selectable_connections,
153         [](const Connection* conn1, const Connection* conn2) {
154           return conn1->last_ping_sent() < conn2->last_ping_sent();
155         });
156     if (iter != pingable_selectable_connections.end()) {
157       return *iter;
158     }
159   }
160 
161   // Rule 3: Triggered checks have priority over non-triggered connections.
162   // Rule 3.1: Among triggered checks, oldest takes precedence.
163   const Connection* oldest_triggered_check =
164       FindOldestConnectionNeedingTriggeredCheck(now);
165   if (oldest_triggered_check) {
166     return oldest_triggered_check;
167   }
168 
169   // Rule 4: Unpinged connections have priority over pinged ones.
170   RTC_CHECK(connections_.size() ==
171             pinged_connections_.size() + unpinged_connections_.size());
172   // If there are unpinged and pingable connections, only ping those.
173   // Otherwise, treat everything as unpinged.
174   // TODO(honghaiz): Instead of adding two separate vectors, we can add a state
175   // "pinged" to filter out unpinged connections.
176   if (absl::c_none_of(unpinged_connections_,
177                       [this, now](const Connection* conn) {
178                         return this->IsPingable(conn, now);
179                       })) {
180     unpinged_connections_.insert(pinged_connections_.begin(),
181                                  pinged_connections_.end());
182     pinged_connections_.clear();
183   }
184 
185   // Among un-pinged pingable connections, "more pingable" takes precedence.
186   std::vector<const Connection*> pingable_connections;
187   absl::c_copy_if(
188       unpinged_connections_, std::back_inserter(pingable_connections),
189       [this, now](const Connection* conn) { return IsPingable(conn, now); });
190   auto iter = absl::c_max_element(
191       pingable_connections,
192       [this](const Connection* conn1, const Connection* conn2) {
193         // Some implementations of max_element
194         // compare an element with itself.
195         if (conn1 == conn2) {
196           return false;
197         }
198         return MorePingable(conn1, conn2) == conn2;
199       });
200   if (iter != pingable_connections.end()) {
201     return *iter;
202   }
203   return nullptr;
204 }
205 
206 // Find "triggered checks".  We ping first those connections that have
207 // received a ping but have not sent a ping since receiving it
208 // (last_ping_received > last_ping_sent).  But we shouldn't do
209 // triggered checks if the connection is already writable.
FindOldestConnectionNeedingTriggeredCheck(int64_t now)210 const Connection* BasicIceController::FindOldestConnectionNeedingTriggeredCheck(
211     int64_t now) {
212   const Connection* oldest_needing_triggered_check = nullptr;
213   for (auto* conn : connections_) {
214     if (!IsPingable(conn, now)) {
215       continue;
216     }
217     bool needs_triggered_check =
218         (!conn->writable() &&
219          conn->last_ping_received() > conn->last_ping_sent());
220     if (needs_triggered_check &&
221         (!oldest_needing_triggered_check ||
222          (conn->last_ping_received() <
223           oldest_needing_triggered_check->last_ping_received()))) {
224       oldest_needing_triggered_check = conn;
225     }
226   }
227 
228   if (oldest_needing_triggered_check) {
229     RTC_LOG(LS_INFO) << "Selecting connection for triggered check: "
230                      << oldest_needing_triggered_check->ToString();
231   }
232   return oldest_needing_triggered_check;
233 }
234 
WritableConnectionPastPingInterval(const Connection * conn,int64_t now) const235 bool BasicIceController::WritableConnectionPastPingInterval(
236     const Connection* conn,
237     int64_t now) const {
238   int interval = CalculateActiveWritablePingInterval(conn, now);
239   return conn->last_ping_sent() + interval <= now;
240 }
241 
CalculateActiveWritablePingInterval(const Connection * conn,int64_t now) const242 int BasicIceController::CalculateActiveWritablePingInterval(
243     const Connection* conn,
244     int64_t now) const {
245   // Ping each connection at a higher rate at least
246   // MIN_PINGS_AT_WEAK_PING_INTERVAL times.
247   if (conn->num_pings_sent() < MIN_PINGS_AT_WEAK_PING_INTERVAL) {
248     return weak_ping_interval();
249   }
250 
251   int stable_interval =
252       config_.stable_writable_connection_ping_interval_or_default();
253   int weak_or_stablizing_interval = std::min(
254       stable_interval, WEAK_OR_STABILIZING_WRITABLE_CONNECTION_PING_INTERVAL);
255   // If the channel is weak or the connection is not stable yet, use the
256   // weak_or_stablizing_interval.
257   return (!weak() && conn->stable(now)) ? stable_interval
258                                         : weak_or_stablizing_interval;
259 }
260 
261 // Is the connection in a state for us to even consider pinging the other side?
262 // We consider a connection pingable even if it's not connected because that's
263 // how a TCP connection is kicked into reconnecting on the active side.
IsPingable(const Connection * conn,int64_t now) const264 bool BasicIceController::IsPingable(const Connection* conn, int64_t now) const {
265   const Candidate& remote = conn->remote_candidate();
266   // We should never get this far with an empty remote ufrag.
267   RTC_DCHECK(!remote.username().empty());
268   if (remote.username().empty() || remote.password().empty()) {
269     // If we don't have an ICE ufrag and pwd, there's no way we can ping.
270     return false;
271   }
272 
273   // A failed connection will not be pinged.
274   if (conn->state() == IceCandidatePairState::FAILED) {
275     return false;
276   }
277 
278   // An never connected connection cannot be written to at all, so pinging is
279   // out of the question. However, if it has become WRITABLE, it is in the
280   // reconnecting state so ping is needed.
281   if (!conn->connected() && !conn->writable()) {
282     return false;
283   }
284 
285   // If we sent a number of pings wo/ reply, skip sending more
286   // until we get one.
287   if (conn->TooManyOutstandingPings(field_trials_->max_outstanding_pings)) {
288     return false;
289   }
290 
291   // If the channel is weakly connected, ping all connections.
292   if (weak()) {
293     return true;
294   }
295 
296   // Always ping active connections regardless whether the channel is completed
297   // or not, but backup connections are pinged at a slower rate.
298   if (IsBackupConnection(conn)) {
299     return conn->rtt_samples() == 0 ||
300            (now >= conn->last_ping_response_received() +
301                        config_.backup_connection_ping_interval_or_default());
302   }
303   // Don't ping inactive non-backup connections.
304   if (!conn->active()) {
305     return false;
306   }
307 
308   // Do ping unwritable, active connections.
309   if (!conn->writable()) {
310     return true;
311   }
312 
313   // Ping writable, active connections if it's been long enough since the last
314   // ping.
315   return WritableConnectionPastPingInterval(conn, now);
316 }
317 
318 // A connection is considered a backup connection if the channel state
319 // is completed, the connection is not the selected connection and it is active.
IsBackupConnection(const Connection * conn) const320 bool BasicIceController::IsBackupConnection(const Connection* conn) const {
321   return ice_transport_state_func_() == IceTransportState::STATE_COMPLETED &&
322          conn != selected_connection_ && conn->active();
323 }
324 
MorePingable(const Connection * conn1,const Connection * conn2)325 const Connection* BasicIceController::MorePingable(const Connection* conn1,
326                                                    const Connection* conn2) {
327   RTC_DCHECK(conn1 != conn2);
328   if (config_.prioritize_most_likely_candidate_pairs) {
329     const Connection* most_likely_to_work_conn = MostLikelyToWork(conn1, conn2);
330     if (most_likely_to_work_conn) {
331       return most_likely_to_work_conn;
332     }
333   }
334 
335   const Connection* least_recently_pinged_conn =
336       LeastRecentlyPinged(conn1, conn2);
337   if (least_recently_pinged_conn) {
338     return least_recently_pinged_conn;
339   }
340 
341   // During the initial state when nothing has been pinged yet, return the first
342   // one in the ordered |connections_|.
343   auto connections = connections_;
344   return *(std::find_if(connections.begin(), connections.end(),
345                         [conn1, conn2](const Connection* conn) {
346                           return conn == conn1 || conn == conn2;
347                         }));
348 }
349 
MostLikelyToWork(const Connection * conn1,const Connection * conn2)350 const Connection* BasicIceController::MostLikelyToWork(
351     const Connection* conn1,
352     const Connection* conn2) {
353   bool rr1 = IsRelayRelay(conn1);
354   bool rr2 = IsRelayRelay(conn2);
355   if (rr1 && !rr2) {
356     return conn1;
357   } else if (rr2 && !rr1) {
358     return conn2;
359   } else if (rr1 && rr2) {
360     bool udp1 = IsUdp(conn1);
361     bool udp2 = IsUdp(conn2);
362     if (udp1 && !udp2) {
363       return conn1;
364     } else if (udp2 && udp1) {
365       return conn2;
366     }
367   }
368   return nullptr;
369 }
370 
LeastRecentlyPinged(const Connection * conn1,const Connection * conn2)371 const Connection* BasicIceController::LeastRecentlyPinged(
372     const Connection* conn1,
373     const Connection* conn2) {
374   if (conn1->last_ping_sent() < conn2->last_ping_sent()) {
375     return conn1;
376   }
377   if (conn1->last_ping_sent() > conn2->last_ping_sent()) {
378     return conn2;
379   }
380   return nullptr;
381 }
382 
383 std::map<const rtc::Network*, const Connection*>
GetBestConnectionByNetwork() const384 BasicIceController::GetBestConnectionByNetwork() const {
385   // |connections_| has been sorted, so the first one in the list on a given
386   // network is the best connection on the network, except that the selected
387   // connection is always the best connection on the network.
388   std::map<const rtc::Network*, const Connection*> best_connection_by_network;
389   if (selected_connection_) {
390     best_connection_by_network[selected_connection_->network()] =
391         selected_connection_;
392   }
393   // TODO(honghaiz): Need to update this if |connections_| are not sorted.
394   for (const Connection* conn : connections_) {
395     const rtc::Network* network = conn->network();
396     // This only inserts when the network does not exist in the map.
397     best_connection_by_network.insert(std::make_pair(network, conn));
398   }
399   return best_connection_by_network;
400 }
401 
402 std::vector<const Connection*>
GetBestWritableConnectionPerNetwork() const403 BasicIceController::GetBestWritableConnectionPerNetwork() const {
404   std::vector<const Connection*> connections;
405   for (auto kv : GetBestConnectionByNetwork()) {
406     const Connection* conn = kv.second;
407     if (conn->writable() && conn->connected()) {
408       connections.push_back(conn);
409     }
410   }
411   return connections;
412 }
413 
414 IceControllerInterface::SwitchResult
HandleInitialSelectDampening(IceControllerEvent reason,const Connection * new_connection)415 BasicIceController::HandleInitialSelectDampening(
416     IceControllerEvent reason,
417     const Connection* new_connection) {
418   if (!field_trials_->initial_select_dampening.has_value() &&
419       !field_trials_->initial_select_dampening_ping_received.has_value()) {
420     // experiment not enabled => select connection.
421     return {new_connection, absl::nullopt};
422   }
423 
424   int64_t now = rtc::TimeMillis();
425   int64_t max_delay = 0;
426   if (new_connection->last_ping_received() > 0 &&
427       field_trials_->initial_select_dampening_ping_received.has_value()) {
428     max_delay = *field_trials_->initial_select_dampening_ping_received;
429   } else if (field_trials_->initial_select_dampening.has_value()) {
430     max_delay = *field_trials_->initial_select_dampening;
431   }
432 
433   int64_t start_wait =
434       initial_select_timestamp_ms_ == 0 ? now : initial_select_timestamp_ms_;
435   int64_t max_wait_until = start_wait + max_delay;
436 
437   if (now >= max_wait_until) {
438     RTC_LOG(LS_INFO) << "reset initial_select_timestamp_ = "
439                      << initial_select_timestamp_ms_
440                      << " selection delayed by: " << (now - start_wait) << "ms";
441     initial_select_timestamp_ms_ = 0;
442     return {new_connection, absl::nullopt};
443   }
444 
445   // We are not yet ready to select first connection...
446   if (initial_select_timestamp_ms_ == 0) {
447     // Set timestamp on first time...
448     // but run the delayed invokation everytime to
449     // avoid possibility that we miss it.
450     initial_select_timestamp_ms_ = now;
451     RTC_LOG(LS_INFO) << "set initial_select_timestamp_ms_ = "
452                      << initial_select_timestamp_ms_;
453   }
454 
455   int min_delay = max_delay;
456   if (field_trials_->initial_select_dampening.has_value()) {
457     min_delay = std::min(min_delay, *field_trials_->initial_select_dampening);
458   }
459   if (field_trials_->initial_select_dampening_ping_received.has_value()) {
460     min_delay = std::min(
461         min_delay, *field_trials_->initial_select_dampening_ping_received);
462   }
463 
464   RTC_LOG(LS_INFO) << "delay initial selection up to " << min_delay << "ms";
465   reason.type = IceControllerEvent::ICE_CONTROLLER_RECHECK;
466   reason.recheck_delay_ms = min_delay;
467   return {absl::nullopt, reason};
468 }
469 
ShouldSwitchConnection(IceControllerEvent reason,const Connection * new_connection)470 IceControllerInterface::SwitchResult BasicIceController::ShouldSwitchConnection(
471     IceControllerEvent reason,
472     const Connection* new_connection) {
473   if (!ReadyToSend(new_connection) || selected_connection_ == new_connection) {
474     return {absl::nullopt, absl::nullopt};
475   }
476 
477   if (selected_connection_ == nullptr) {
478     return HandleInitialSelectDampening(reason, new_connection);
479   }
480 
481   // Do not switch to a connection that is not receiving if it is not on a
482   // preferred network or it has higher cost because it may be just spuriously
483   // better.
484   int compare_a_b_by_networks = CompareCandidatePairNetworks(
485       new_connection, selected_connection_, config_.network_preference);
486   if (compare_a_b_by_networks == b_is_better && !new_connection->receiving()) {
487     return {absl::nullopt, absl::nullopt};
488   }
489 
490   bool missed_receiving_unchanged_threshold = false;
491   absl::optional<int64_t> receiving_unchanged_threshold(
492       rtc::TimeMillis() - config_.receiving_switching_delay_or_default());
493   int cmp = CompareConnections(selected_connection_, new_connection,
494                                receiving_unchanged_threshold,
495                                &missed_receiving_unchanged_threshold);
496 
497   absl::optional<IceControllerEvent> recheck_event;
498   if (missed_receiving_unchanged_threshold &&
499       config_.receiving_switching_delay_or_default()) {
500     // If we do not switch to the connection because it missed the receiving
501     // threshold, the new connection is in a better receiving state than the
502     // currently selected connection. So we need to re-check whether it needs
503     // to be switched at a later time.
504     recheck_event = reason;
505     recheck_event->recheck_delay_ms =
506         config_.receiving_switching_delay_or_default();
507   }
508 
509   if (cmp < 0) {
510     return {new_connection, absl::nullopt};
511   } else if (cmp > 0) {
512     return {absl::nullopt, recheck_event};
513   }
514 
515   // If everything else is the same, switch only if rtt has improved by
516   // a margin.
517   if (new_connection->rtt() <= selected_connection_->rtt() - kMinImprovement) {
518     return {new_connection, absl::nullopt};
519   }
520 
521   return {absl::nullopt, recheck_event};
522 }
523 
524 IceControllerInterface::SwitchResult
SortAndSwitchConnection(IceControllerEvent reason)525 BasicIceController::SortAndSwitchConnection(IceControllerEvent reason) {
526   // Find the best alternative connection by sorting.  It is important to note
527   // that amongst equal preference, writable connections, this will choose the
528   // one whose estimated latency is lowest.  So it is the only one that we
529   // need to consider switching to.
530   // TODO(honghaiz): Don't sort;  Just use std::max_element in the right places.
531   absl::c_stable_sort(
532       connections_, [this](const Connection* a, const Connection* b) {
533         int cmp = CompareConnections(a, b, absl::nullopt, nullptr);
534         if (cmp != 0) {
535           return cmp > 0;
536         }
537         // Otherwise, sort based on latency estimate.
538         return a->rtt() < b->rtt();
539       });
540 
541   RTC_LOG(LS_VERBOSE) << "Sorting " << connections_.size()
542                       << " available connections";
543   for (size_t i = 0; i < connections_.size(); ++i) {
544     RTC_LOG(LS_VERBOSE) << connections_[i]->ToString();
545   }
546 
547   const Connection* top_connection =
548       (!connections_.empty()) ? connections_[0] : nullptr;
549 
550   return ShouldSwitchConnection(reason, top_connection);
551 }
552 
ReadyToSend(const Connection * connection) const553 bool BasicIceController::ReadyToSend(const Connection* connection) const {
554   // Note that we allow sending on an unreliable connection, because it's
555   // possible that it became unreliable simply due to bad chance.
556   // So this shouldn't prevent attempting to send media.
557   return connection != nullptr &&
558          (connection->writable() ||
559           connection->write_state() == Connection::STATE_WRITE_UNRELIABLE ||
560           PresumedWritable(connection));
561 }
562 
PresumedWritable(const Connection * conn) const563 bool BasicIceController::PresumedWritable(const Connection* conn) const {
564   return (conn->write_state() == Connection::STATE_WRITE_INIT &&
565           config_.presume_writable_when_fully_relayed &&
566           conn->local_candidate().type() == RELAY_PORT_TYPE &&
567           (conn->remote_candidate().type() == RELAY_PORT_TYPE ||
568            conn->remote_candidate().type() == PRFLX_PORT_TYPE));
569 }
570 
571 // Compare two connections based on their writing, receiving, and connected
572 // states.
CompareConnectionStates(const Connection * a,const Connection * b,absl::optional<int64_t> receiving_unchanged_threshold,bool * missed_receiving_unchanged_threshold) const573 int BasicIceController::CompareConnectionStates(
574     const Connection* a,
575     const Connection* b,
576     absl::optional<int64_t> receiving_unchanged_threshold,
577     bool* missed_receiving_unchanged_threshold) const {
578   // First, prefer a connection that's writable or presumed writable over
579   // one that's not writable.
580   bool a_writable = a->writable() || PresumedWritable(a);
581   bool b_writable = b->writable() || PresumedWritable(b);
582   if (a_writable && !b_writable) {
583     return a_is_better;
584   }
585   if (!a_writable && b_writable) {
586     return b_is_better;
587   }
588 
589   // Sort based on write-state. Better states have lower values.
590   if (a->write_state() < b->write_state()) {
591     return a_is_better;
592   }
593   if (b->write_state() < a->write_state()) {
594     return b_is_better;
595   }
596 
597   // We prefer a receiving connection to a non-receiving, higher-priority
598   // connection when sorting connections and choosing which connection to
599   // switch to.
600   if (a->receiving() && !b->receiving()) {
601     return a_is_better;
602   }
603   if (!a->receiving() && b->receiving()) {
604     if (!receiving_unchanged_threshold ||
605         (a->receiving_unchanged_since() <= *receiving_unchanged_threshold &&
606          b->receiving_unchanged_since() <= *receiving_unchanged_threshold)) {
607       return b_is_better;
608     }
609     *missed_receiving_unchanged_threshold = true;
610   }
611 
612   // WARNING: Some complexity here about TCP reconnecting.
613   // When a TCP connection fails because of a TCP socket disconnecting, the
614   // active side of the connection will attempt to reconnect for 5 seconds while
615   // pretending to be writable (the connection is not set to the unwritable
616   // state).  On the passive side, the connection also remains writable even
617   // though it is disconnected, and a new connection is created when the active
618   // side connects.  At that point, there are two TCP connections on the passive
619   // side: 1. the old, disconnected one that is pretending to be writable, and
620   // 2.  the new, connected one that is maybe not yet writable.  For purposes of
621   // pruning, pinging, and selecting the selected connection, we want to treat
622   // the new connection as "better" than the old one. We could add a method
623   // called something like Connection::ImReallyBadEvenThoughImWritable, but that
624   // is equivalent to the existing Connection::connected(), which we already
625   // have. So, in code throughout this file, we'll check whether the connection
626   // is connected() or not, and if it is not, treat it as "worse" than a
627   // connected one, even though it's writable.  In the code below, we're doing
628   // so to make sure we treat a new writable connection as better than an old
629   // disconnected connection.
630 
631   // In the case where we reconnect TCP connections, the original best
632   // connection is disconnected without changing to WRITE_TIMEOUT. In this case,
633   // the new connection, when it becomes writable, should have higher priority.
634   if (a->write_state() == Connection::STATE_WRITABLE &&
635       b->write_state() == Connection::STATE_WRITABLE) {
636     if (a->connected() && !b->connected()) {
637       return a_is_better;
638     }
639     if (!a->connected() && b->connected()) {
640       return b_is_better;
641     }
642   }
643 
644   return 0;
645 }
646 
647 // Compares two connections based only on the candidate and network information.
648 // Returns positive if |a| is better than |b|.
CompareConnectionCandidates(const Connection * a,const Connection * b) const649 int BasicIceController::CompareConnectionCandidates(const Connection* a,
650                                                     const Connection* b) const {
651   int compare_a_b_by_networks =
652       CompareCandidatePairNetworks(a, b, config_.network_preference);
653   if (compare_a_b_by_networks != a_and_b_equal) {
654     return compare_a_b_by_networks;
655   }
656 
657   // Compare connection priority. Lower values get sorted last.
658   if (a->priority() > b->priority()) {
659     return a_is_better;
660   }
661   if (a->priority() < b->priority()) {
662     return b_is_better;
663   }
664 
665   // If we're still tied at this point, prefer a younger generation.
666   // (Younger generation means a larger generation number).
667   int cmp = (a->remote_candidate().generation() + a->generation()) -
668             (b->remote_candidate().generation() + b->generation());
669   if (cmp != 0) {
670     return cmp;
671   }
672 
673   // A periodic regather (triggered by the regather_all_networks_interval_range)
674   // will produce candidates that appear the same but would use a new port. We
675   // want to use the new candidates and purge the old candidates as they come
676   // in, so use the fact that the old ports get pruned immediately to rank the
677   // candidates with an active port/remote candidate higher.
678   bool a_pruned = is_connection_pruned_func_(a);
679   bool b_pruned = is_connection_pruned_func_(b);
680   if (!a_pruned && b_pruned) {
681     return a_is_better;
682   }
683   if (a_pruned && !b_pruned) {
684     return b_is_better;
685   }
686 
687   // Otherwise, must be equal
688   return 0;
689 }
690 
CompareConnections(const Connection * a,const Connection * b,absl::optional<int64_t> receiving_unchanged_threshold,bool * missed_receiving_unchanged_threshold) const691 int BasicIceController::CompareConnections(
692     const Connection* a,
693     const Connection* b,
694     absl::optional<int64_t> receiving_unchanged_threshold,
695     bool* missed_receiving_unchanged_threshold) const {
696   RTC_CHECK(a != nullptr);
697   RTC_CHECK(b != nullptr);
698 
699   // We prefer to switch to a writable and receiving connection over a
700   // non-writable or non-receiving connection, even if the latter has
701   // been nominated by the controlling side.
702   int state_cmp = CompareConnectionStates(a, b, receiving_unchanged_threshold,
703                                           missed_receiving_unchanged_threshold);
704   if (state_cmp != 0) {
705     return state_cmp;
706   }
707 
708   if (ice_role_func_() == ICEROLE_CONTROLLED) {
709     // Compare the connections based on the nomination states and the last data
710     // received time if this is on the controlled side.
711     if (a->remote_nomination() > b->remote_nomination()) {
712       return a_is_better;
713     }
714     if (a->remote_nomination() < b->remote_nomination()) {
715       return b_is_better;
716     }
717 
718     if (a->last_data_received() > b->last_data_received()) {
719       return a_is_better;
720     }
721     if (a->last_data_received() < b->last_data_received()) {
722       return b_is_better;
723     }
724   }
725 
726   // Compare the network cost and priority.
727   return CompareConnectionCandidates(a, b);
728 }
729 
CompareCandidatePairNetworks(const Connection * a,const Connection * b,absl::optional<rtc::AdapterType> network_preference) const730 int BasicIceController::CompareCandidatePairNetworks(
731     const Connection* a,
732     const Connection* b,
733     absl::optional<rtc::AdapterType> network_preference) const {
734   int compare_a_b_by_network_preference =
735       CompareCandidatePairsByNetworkPreference(a, b,
736                                                config_.network_preference);
737   // The network preference has a higher precedence than the network cost.
738   if (compare_a_b_by_network_preference != a_and_b_equal) {
739     return compare_a_b_by_network_preference;
740   }
741 
742   uint32_t a_cost = a->ComputeNetworkCost();
743   uint32_t b_cost = b->ComputeNetworkCost();
744   // Prefer lower network cost.
745   if (a_cost < b_cost) {
746     return a_is_better;
747   }
748   if (a_cost > b_cost) {
749     return b_is_better;
750   }
751   return a_and_b_equal;
752 }
753 
PruneConnections()754 std::vector<const Connection*> BasicIceController::PruneConnections() {
755   // We can prune any connection for which there is a connected, writable
756   // connection on the same network with better or equal priority.  We leave
757   // those with better priority just in case they become writable later (at
758   // which point, we would prune out the current selected connection).  We leave
759   // connections on other networks because they may not be using the same
760   // resources and they may represent very distinct paths over which we can
761   // switch. If |best_conn_on_network| is not connected, we may be reconnecting
762   // a TCP connection and should not prune connections in this network.
763   // See the big comment in CompareConnectionStates.
764   //
765   // An exception is made for connections on an "any address" network, meaning
766   // not bound to any specific network interface. We don't want to keep one of
767   // these alive as a backup, since it could be using the same network
768   // interface as the higher-priority, selected candidate pair.
769   std::vector<const Connection*> connections_to_prune;
770   auto best_connection_by_network = GetBestConnectionByNetwork();
771   for (const Connection* conn : connections_) {
772     const Connection* best_conn = selected_connection_;
773     if (!rtc::IPIsAny(conn->network()->ip())) {
774       // If the connection is bound to a specific network interface (not an
775       // "any address" network), compare it against the best connection for
776       // that network interface rather than the best connection overall. This
777       // ensures that at least one connection per network will be left
778       // unpruned.
779       best_conn = best_connection_by_network[conn->network()];
780     }
781     // Do not prune connections if the connection being compared against is
782     // weak. Otherwise, it may delete connections prematurely.
783     if (best_conn && conn != best_conn && !best_conn->weak() &&
784         CompareConnectionCandidates(best_conn, conn) >= 0) {
785       connections_to_prune.push_back(conn);
786     }
787   }
788   return connections_to_prune;
789 }
790 
GetUseCandidateAttr(const Connection * conn,NominationMode mode,IceMode remote_ice_mode) const791 bool BasicIceController::GetUseCandidateAttr(const Connection* conn,
792                                              NominationMode mode,
793                                              IceMode remote_ice_mode) const {
794   switch (mode) {
795     case NominationMode::REGULAR:
796       // TODO(honghaiz): Implement regular nomination.
797       return false;
798     case NominationMode::AGGRESSIVE:
799       if (remote_ice_mode == ICEMODE_LITE) {
800         return GetUseCandidateAttr(conn, NominationMode::REGULAR,
801                                    remote_ice_mode);
802       }
803       return true;
804     case NominationMode::SEMI_AGGRESSIVE: {
805       // Nominate if
806       // a) Remote is in FULL ICE AND
807       //    a.1) |conn| is the selected connection OR
808       //    a.2) there is no selected connection OR
809       //    a.3) the selected connection is unwritable OR
810       //    a.4) |conn| has higher priority than selected_connection.
811       // b) Remote is in LITE ICE AND
812       //    b.1) |conn| is the selected_connection AND
813       //    b.2) |conn| is writable.
814       bool selected = conn == selected_connection_;
815       if (remote_ice_mode == ICEMODE_LITE) {
816         return selected && conn->writable();
817       }
818       bool better_than_selected =
819           !selected_connection_ || !selected_connection_->writable() ||
820           CompareConnectionCandidates(selected_connection_, conn) < 0;
821       return selected || better_than_selected;
822     }
823     default:
824       RTC_NOTREACHED();
825       return false;
826   }
827 }
828 
829 }  // namespace cricket
830