1 // Copyright 2017 The Chromium Authors
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "net/nqe/network_quality_estimator_util.h"
6
7 #include <memory>
8
9 #include "base/check_op.h"
10 #include "base/functional/bind.h"
11 #include "base/notreached.h"
12 #include "net/base/address_list.h"
13 #include "net/base/host_port_pair.h"
14 #include "net/base/ip_address.h"
15 #include "net/base/ip_endpoint.h"
16 #include "net/base/net_errors.h"
17 #include "net/dns/host_resolver.h"
18 #include "net/dns/public/host_resolver_source.h"
19 #include "net/url_request/url_request.h"
20 #include "net/url_request/url_request_context.h"
21
22 namespace net::nqe {
23
24 namespace {
25
IsPrivateHost(HostResolver * host_resolver,const HostPortPair & host_port_pair,const NetworkAnonymizationKey & network_anonymization_key,NetLogWithSource net_log)26 bool IsPrivateHost(HostResolver* host_resolver,
27 const HostPortPair& host_port_pair,
28 const NetworkAnonymizationKey& network_anonymization_key,
29 NetLogWithSource net_log) {
30 // Try resolving |host_port_pair.host()| synchronously.
31 HostResolver::ResolveHostParameters parameters;
32 parameters.source = HostResolverSource::LOCAL_ONLY;
33 std::unique_ptr<HostResolver::ResolveHostRequest> request =
34 host_resolver->CreateRequest(host_port_pair, network_anonymization_key,
35 net_log, parameters);
36
37 int rv = request->Start(base::BindOnce([](int error) { NOTREACHED(); }));
38 DCHECK_NE(rv, ERR_IO_PENDING);
39
40 if (rv == OK && request->GetAddressResults() &&
41 !request->GetAddressResults()->empty()) {
42 // Checking only the first address should be sufficient.
43 IPEndPoint ip_endpoint = request->GetAddressResults()->front();
44 IPAddress ip_address = ip_endpoint.address();
45 if (!ip_address.IsPubliclyRoutable())
46 return true;
47 }
48
49 return false;
50 }
51
52 } // namespace
53
54 namespace internal {
55
IsRequestForPrivateHost(const URLRequest & request,NetLogWithSource net_log)56 bool IsRequestForPrivateHost(const URLRequest& request,
57 NetLogWithSource net_log) {
58 // Using the request's NetworkAnonymizationKey isn't necessary for privacy
59 // reasons, but is needed to maximize the chances of a cache hit.
60 return IsPrivateHost(
61 request.context()->host_resolver(), HostPortPair::FromURL(request.url()),
62 request.isolation_info().network_anonymization_key(), net_log);
63 }
64
IsPrivateHostForTesting(HostResolver * host_resolver,const HostPortPair & host_port_pair,const NetworkAnonymizationKey & network_anonymization_key)65 bool IsPrivateHostForTesting(
66 HostResolver* host_resolver,
67 const HostPortPair& host_port_pair,
68 const NetworkAnonymizationKey& network_anonymization_key) {
69 return IsPrivateHost(host_resolver, host_port_pair, network_anonymization_key,
70 NetLogWithSource());
71 }
72
73 } // namespace internal
74
75 } // namespace net::nqe
76