mirror of
https://github.com/klzgrad/naiveproxy.git
synced 2024-11-24 14:26:09 +03:00
77 lines
2.2 KiB
C++
77 lines
2.2 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);
|
||
|
|
}
|
||
|
|
|
||
|
|
void MappedHostResolver::SetDnsClientEnabled(bool enabled) {
|
||
|
|
impl_->SetDnsClientEnabled(enabled);
|
||
|
|
}
|
||
|
|
|
||
|
|
HostCache* MappedHostResolver::GetHostCache() {
|
||
|
|
return impl_->GetHostCache();
|
||
|
|
}
|
||
|
|
|
||
|
|
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
|