1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
|
// Copyright 2016 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef NET_NQE_SOCKET_WATCHER_FACTORY_H_
#define NET_NQE_SOCKET_WATCHER_FACTORY_H_
#include <memory>
#include <optional>
#include "base/functional/callback.h"
#include "base/memory/raw_ptr.h"
#include "base/memory/scoped_refptr.h"
#include "base/memory/weak_ptr.h"
#include "base/sequence_checker.h"
#include "base/task/single_thread_task_runner.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::RepeatingCallback<void(
SocketPerformanceWatcherFactory::Protocol protocol,
const base::TimeDelta& rtt,
const std::optional<nqe::internal::IPHash>& host)>
OnUpdatedRTTAvailableCallback;
typedef base::RepeatingCallback<bool(base::TimeTicks)> ShouldNotifyRTTCallback;
} // namespace
namespace nqe::internal {
// SocketWatcherFactory implements SocketPerformanceWatcherFactory.
// SocketWatcherFactory is thread safe.
class SocketWatcherFactory : public SocketPerformanceWatcherFactory {
public:
// 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. |should_notify_rtt_callback| is the callback
// that should be called back on |task_runner| to check if RTT observation
// should be taken and notified.
SocketWatcherFactory(
scoped_refptr<base::SingleThreadTaskRunner> task_runner,
base::TimeDelta min_notification_interval,
OnUpdatedRTTAvailableCallback updated_rtt_observation_callback,
ShouldNotifyRTTCallback should_notify_rtt_callback,
const base::TickClock* tick_clock);
SocketWatcherFactory(const SocketWatcherFactory&) = delete;
SocketWatcherFactory& operator=(const SocketWatcherFactory&) = delete;
~SocketWatcherFactory() override;
// SocketPerformanceWatcherFactory implementation:
std::unique_ptr<SocketPerformanceWatcher> CreateSocketPerformanceWatcher(
const Protocol protocol,
const IPAddress& address) override;
void SetUseLocalHostRequestsForTesting(bool use_localhost_requests) {
allow_rtt_private_address_ = use_localhost_requests;
}
// Overrides the tick clock used by |this| for testing.
void SetTickClockForTesting(const base::TickClock* tick_clock);
private:
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_ = false;
// Called every time a new RTT observation is available.
OnUpdatedRTTAvailableCallback updated_rtt_observation_callback_;
// Callback that should be called by socket watchers to determine if the RTT
// notification should be notified using |updated_rtt_observation_callback_|.
ShouldNotifyRTTCallback should_notify_rtt_callback_;
raw_ptr<const base::TickClock> tick_clock_;
};
} // namespace nqe::internal
} // namespace net
#endif // NET_NQE_SOCKET_WATCHER_FACTORY_H_
|
