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
|
// Copyright 2018 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef SERVICES_NETWORK_HOST_RESOLVER_H_
#define SERVICES_NETWORK_HOST_RESOLVER_H_
#include <memory>
#include <set>
#include <string>
#include "base/component_export.h"
#include "base/containers/unique_ptr_adapters.h"
#include "base/functional/callback.h"
#include "base/memory/raw_ptr.h"
#include "base/memory/weak_ptr.h"
#include "mojo/public/cpp/bindings/pending_receiver.h"
#include "mojo/public/cpp/bindings/pending_remote.h"
#include "mojo/public/cpp/bindings/receiver.h"
#include "net/dns/public/dns_query_type.h"
#include "services/network/public/mojom/host_resolver.mojom.h"
namespace net {
class HostResolver;
class HostPortPair;
class NetLog;
class NetworkAnonymizationKey;
} // namespace net
namespace network {
class HostResolverMdnsListener;
class ResolveHostRequest;
class COMPONENT_EXPORT(NETWORK_SERVICE) HostResolver
: public mojom::HostResolver {
public:
using ConnectionShutdownCallback = base::OnceCallback<void(HostResolver*)>;
// Constructs and binds to the given mojom::HostResolver pipe. On pipe close,
// cancels all outstanding receivers (whether made through the pipe or by
// directly calling ResolveHost()) with ERR_FAILED. Also on pipe close, calls
// |connection_shutdown_callback| and passes |this| to notify that the
// resolver has cancelled all receivers and may be cleaned up.
//
// `owned_internal_resolver` if set should be equal to `internal_resolver` and
// denotes that `this` takes ownership.
HostResolver(mojo::PendingReceiver<mojom::HostResolver> resolver_receiver,
ConnectionShutdownCallback connection_shutdown_callback,
net::HostResolver* internal_resolver,
std::unique_ptr<net::HostResolver> owned_internal_resolver,
net::NetLog* net_log);
// Constructor for when the resolver will not be bound to a
// mojom::HostResolver pipe, eg because it is handling ResolveHost requests
// made directly on NetworkContext.
HostResolver(net::HostResolver* internal_resolver, net::NetLog* net_log);
HostResolver(const HostResolver&) = delete;
HostResolver& operator=(const HostResolver&) = delete;
~HostResolver() override;
void ResolveHost(
mojom::HostResolverHostPtr host,
const net::NetworkAnonymizationKey& network_anonymization_key,
mojom::ResolveHostParametersPtr optional_parameters,
mojo::PendingRemote<mojom::ResolveHostClient> response_client) override;
void MdnsListen(const net::HostPortPair& host,
net::DnsQueryType query_type,
mojo::PendingRemote<mojom::MdnsListenClient> response_client,
MdnsListenCallback callback) override;
size_t GetNumOutstandingRequestsForTesting() const;
// Sets a global callback when a ResolveHost call arrives.
using ResolveHostCallback =
base::RepeatingCallback<void(const std::string& host)>;
static void SetResolveHostCallbackForTesting(ResolveHostCallback callback);
private:
void AsyncSetUp();
void OnResolveHostComplete(ResolveHostRequest* request, int error);
void OnMdnsListenerCancelled(HostResolverMdnsListener* listener);
void OnConnectionError();
mojo::Receiver<mojom::HostResolver> receiver_;
mojo::PendingReceiver<mojom::HostResolver> pending_receiver_;
ConnectionShutdownCallback connection_shutdown_callback_;
std::set<std::unique_ptr<ResolveHostRequest>, base::UniquePtrComparator>
requests_;
std::set<std::unique_ptr<HostResolverMdnsListener>, base::UniquePtrComparator>
listeners_;
const std::unique_ptr<net::HostResolver> owned_internal_resolver_;
const raw_ptr<net::HostResolver> internal_resolver_;
const raw_ptr<net::NetLog> net_log_;
base::WeakPtrFactory<HostResolver> weak_factory_{this};
};
} // namespace network
#endif // SERVICES_NETWORK_HOST_RESOLVER_H_
|
