mirror of https://github.com/electron/electron
93 lines
3.2 KiB
C++
93 lines
3.2 KiB
C++
// Copyright (c) 2023 Signal Messenger, LLC
|
|
// Use of this source code is governed by the MIT license that can be
|
|
// found in the LICENSE file.
|
|
|
|
#include "shell/browser/net/resolve_host_function.h"
|
|
|
|
#include <utility>
|
|
|
|
#include "base/functional/bind.h"
|
|
#include "content/public/browser/browser_context.h"
|
|
#include "content/public/browser/browser_thread.h"
|
|
#include "content/public/browser/storage_partition.h"
|
|
#include "net/base/address_list.h"
|
|
#include "net/base/host_port_pair.h"
|
|
#include "net/base/net_errors.h"
|
|
#include "net/base/network_isolation_key.h"
|
|
#include "net/dns/public/host_resolver_results.h"
|
|
#include "net/dns/public/resolve_error_info.h"
|
|
#include "services/network/public/mojom/network_context.mojom.h"
|
|
#include "shell/browser/electron_browser_context.h"
|
|
#include "shell/common/process_util.h"
|
|
#include "shell/services/node/node_service.h"
|
|
#include "url/origin.h"
|
|
|
|
using content::BrowserThread;
|
|
|
|
namespace electron {
|
|
|
|
ResolveHostFunction::ResolveHostFunction(
|
|
ElectronBrowserContext* browser_context,
|
|
std::string host,
|
|
network::mojom::ResolveHostParametersPtr params,
|
|
ResolveHostCallback callback)
|
|
: browser_context_(browser_context),
|
|
host_(std::move(host)),
|
|
params_(std::move(params)),
|
|
callback_(std::move(callback)) {
|
|
DETACH_FROM_SEQUENCE(sequence_checker_);
|
|
}
|
|
|
|
ResolveHostFunction::~ResolveHostFunction() {
|
|
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
|
|
DCHECK(!receiver_.is_bound());
|
|
}
|
|
|
|
void ResolveHostFunction::Run() {
|
|
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
|
|
DCHECK(!receiver_.is_bound());
|
|
|
|
// Start the request.
|
|
net::HostPortPair host_port_pair(host_, 0);
|
|
mojo::PendingRemote<network::mojom::ResolveHostClient> resolve_host_client =
|
|
receiver_.BindNewPipeAndPassRemote();
|
|
receiver_.set_disconnect_handler(base::BindOnce(
|
|
&ResolveHostFunction::OnComplete, this, net::ERR_NAME_NOT_RESOLVED,
|
|
net::ResolveErrorInfo(net::ERR_FAILED),
|
|
/*resolved_addresses=*/std::nullopt,
|
|
/*endpoint_results_with_metadata=*/std::nullopt));
|
|
if (electron::IsUtilityProcess()) {
|
|
URLLoaderBundle::GetInstance()->GetHostResolver()->ResolveHost(
|
|
network::mojom::HostResolverHost::NewHostPortPair(
|
|
std::move(host_port_pair)),
|
|
net::NetworkAnonymizationKey(), std::move(params_),
|
|
std::move(resolve_host_client));
|
|
} else {
|
|
DCHECK_CURRENTLY_ON(BrowserThread::UI);
|
|
browser_context_->GetDefaultStoragePartition()
|
|
->GetNetworkContext()
|
|
->ResolveHost(network::mojom::HostResolverHost::NewHostPortPair(
|
|
std::move(host_port_pair)),
|
|
net::NetworkAnonymizationKey(), std::move(params_),
|
|
std::move(resolve_host_client));
|
|
}
|
|
}
|
|
|
|
void ResolveHostFunction::OnComplete(
|
|
int result,
|
|
const net::ResolveErrorInfo& resolve_error_info,
|
|
const std::optional<net::AddressList>& resolved_addresses,
|
|
const std::optional<net::HostResolverEndpointResults>&
|
|
endpoint_results_with_metadata) {
|
|
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
|
|
|
|
// Ensure that we outlive the `receiver_.reset()` call.
|
|
scoped_refptr<ResolveHostFunction> self(this);
|
|
|
|
receiver_.reset();
|
|
|
|
std::move(callback_).Run(resolve_error_info.error, resolved_addresses);
|
|
}
|
|
|
|
} // namespace electron
|