blob: 3bb996932f9ac65cc57c7ccd8929af3baa90924a [file] [log] [blame]
// Copyright 2016 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include <memory>
#include "base/callback.h"
#include "base/memory/ref_counted.h"
#include "base/memory/weak_ptr.h"
#include "base/optional.h"
#include "base/single_thread_task_runner.h"
#include "base/threading/thread_checker.h"
#include "base/time/time.h"
#include "net/nqe/network_quality_estimator_util.h"
#include "net/socket/socket_performance_watcher.h"
#include "net/socket/socket_performance_watcher_factory.h"
namespace base {
class TickClock;
class TimeDelta;
} // namespace base
namespace net {
namespace {
typedef base::Callback<void(SocketPerformanceWatcherFactory::Protocol protocol,
const base::TimeDelta& rtt,
const base::Optional<nqe::internal::IPHash>& host)>
namespace nqe {
namespace internal {
// SocketWatcherFactory implements SocketPerformanceWatcherFactory.
// SocketWatcherFactory is thread safe.
class SocketWatcherFactory : public SocketPerformanceWatcherFactory {
// Creates a SocketWatcherFactory. All socket watchers created by
// SocketWatcherFactory call |updated_rtt_observation_callback| on
// |task_runner| every time a new RTT observation is available.
// |min_notification_interval| is the minimum interval betweeen consecutive
// notifications to the socket watchers created by this factory. |tick_clock|
// is guaranteed to be non-null.
scoped_refptr<base::SingleThreadTaskRunner> task_runner,
base::TimeDelta min_notification_interval,
OnUpdatedRTTAvailableCallback updated_rtt_observation_callback,
base::TickClock* tick_clock);
~SocketWatcherFactory() override;
// SocketPerformanceWatcherFactory implementation:
std::unique_ptr<SocketPerformanceWatcher> CreateSocketPerformanceWatcher(
const Protocol protocol,
const AddressList& address_list) override;
void SetUseLocalHostRequestsForTesting(bool use_localhost_requests) {
allow_rtt_private_address_ = use_localhost_requests;
scoped_refptr<base::SingleThreadTaskRunner> task_runner_;
// Minimum interval betweeen consecutive notifications to the socket watchers
// created by this factory.
const base::TimeDelta min_notification_interval_;
// True if socket watchers constructed by this factory can use the RTT from
// the sockets that are connected to the private addresses.
bool allow_rtt_private_address_;
// Called every time a new RTT observation is available.
OnUpdatedRTTAvailableCallback updated_rtt_observation_callback_;
base::TickClock* tick_clock_;
} // namespace internal
} // namespace nqe
} // namespace net