mirror of
https://github.com/klzgrad/naiveproxy.git
synced 2024-12-01 01:36:09 +03:00
96 lines
2.9 KiB
C++
96 lines
2.9 KiB
C++
// Copyright (c) 2012 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 "net/dns/mapped_host_resolver.h"
|
|
|
|
#include <utility>
|
|
|
|
#include "base/strings/string_util.h"
|
|
#include "base/values.h"
|
|
#include "net/base/host_port_pair.h"
|
|
#include "net/base/net_errors.h"
|
|
|
|
namespace net {
|
|
|
|
MappedHostResolver::MappedHostResolver(std::unique_ptr<HostResolver> impl)
|
|
: impl_(std::move(impl)) {}
|
|
|
|
MappedHostResolver::~MappedHostResolver() = default;
|
|
|
|
int MappedHostResolver::Resolve(const RequestInfo& original_info,
|
|
RequestPriority priority,
|
|
AddressList* addresses,
|
|
const CompletionCallback& callback,
|
|
std::unique_ptr<Request>* request,
|
|
const NetLogWithSource& net_log) {
|
|
RequestInfo info = original_info;
|
|
int rv = ApplyRules(&info);
|
|
if (rv != OK)
|
|
return rv;
|
|
|
|
return impl_->Resolve(info, priority, addresses, callback, request, net_log);
|
|
}
|
|
|
|
int MappedHostResolver::ResolveFromCache(const RequestInfo& original_info,
|
|
AddressList* addresses,
|
|
const NetLogWithSource& net_log) {
|
|
RequestInfo info = original_info;
|
|
int rv = ApplyRules(&info);
|
|
if (rv != OK)
|
|
return rv;
|
|
|
|
return impl_->ResolveFromCache(info, addresses, net_log);
|
|
}
|
|
|
|
int MappedHostResolver::ResolveStaleFromCache(
|
|
const RequestInfo& original_info,
|
|
AddressList* addresses,
|
|
HostCache::EntryStaleness* stale_info,
|
|
const NetLogWithSource& net_log) {
|
|
RequestInfo info = original_info;
|
|
int rv = ApplyRules(&info);
|
|
if (rv != OK)
|
|
return rv;
|
|
|
|
return impl_->ResolveStaleFromCache(info, addresses, stale_info, net_log);
|
|
}
|
|
|
|
void MappedHostResolver::SetDnsClientEnabled(bool enabled) {
|
|
impl_->SetDnsClientEnabled(enabled);
|
|
}
|
|
|
|
HostCache* MappedHostResolver::GetHostCache() {
|
|
return impl_->GetHostCache();
|
|
}
|
|
|
|
bool MappedHostResolver::HasCached(base::StringPiece hostname,
|
|
HostCache::Entry::Source* source_out,
|
|
HostCache::EntryStaleness* stale_out) const {
|
|
return impl_->HasCached(hostname, source_out, stale_out);
|
|
}
|
|
|
|
std::unique_ptr<base::Value> MappedHostResolver::GetDnsConfigAsValue() const {
|
|
return impl_->GetDnsConfigAsValue();
|
|
}
|
|
|
|
void MappedHostResolver::SetNoIPv6OnWifi(bool no_ipv6_on_wifi) {
|
|
impl_->SetNoIPv6OnWifi(no_ipv6_on_wifi);
|
|
}
|
|
|
|
bool MappedHostResolver::GetNoIPv6OnWifi() {
|
|
return impl_->GetNoIPv6OnWifi();
|
|
}
|
|
|
|
int MappedHostResolver::ApplyRules(RequestInfo* info) const {
|
|
HostPortPair host_port(info->host_port_pair());
|
|
if (rules_.RewriteHost(&host_port)) {
|
|
if (host_port.host() == "~NOTFOUND")
|
|
return ERR_NAME_NOT_RESOLVED;
|
|
info->set_host_port_pair(host_port);
|
|
}
|
|
return OK;
|
|
}
|
|
|
|
} // namespace net
|