blob: a9e88a067d047b8a5bd2f237dfadc79f3cff3f07 [file] [log] [blame]
// Copyright 2017 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 "content/common/throttling_url_loader.h"
#include "base/single_thread_task_runner.h"
namespace content {
ThrottlingURLLoader::StartInfo::StartInfo(
mojom::URLLoaderFactory* in_url_loader_factory,
int32_t in_routing_id,
int32_t in_request_id,
uint32_t in_options,
StartLoaderCallback in_start_loader_callback,
const ResourceRequest& in_url_request,
scoped_refptr<base::SingleThreadTaskRunner> in_task_runner)
: url_loader_factory(in_url_loader_factory),
routing_id(in_routing_id),
request_id(in_request_id),
options(in_options),
start_loader_callback(std::move(in_start_loader_callback)),
url_request(in_url_request),
task_runner(std::move(in_task_runner)) {}
ThrottlingURLLoader::StartInfo::~StartInfo() = default;
ThrottlingURLLoader::ResponseInfo::ResponseInfo(
const ResourceResponseHead& in_response_head,
const base::Optional<net::SSLInfo>& in_ssl_info,
mojom::DownloadedTempFilePtr in_downloaded_file)
: response_head(in_response_head),
ssl_info(in_ssl_info),
downloaded_file(std::move(in_downloaded_file)) {}
ThrottlingURLLoader::ResponseInfo::~ResponseInfo() = default;
ThrottlingURLLoader::RedirectInfo::RedirectInfo(
const net::RedirectInfo& in_redirect_info,
const ResourceResponseHead& in_response_head)
: redirect_info(in_redirect_info), response_head(in_response_head) {}
ThrottlingURLLoader::RedirectInfo::~RedirectInfo() = default;
ThrottlingURLLoader::PriorityInfo::PriorityInfo(
net::RequestPriority in_priority,
int32_t in_intra_priority_value)
: priority(in_priority), intra_priority_value(in_intra_priority_value) {}
ThrottlingURLLoader::PriorityInfo::~PriorityInfo() = default;
// static
std::unique_ptr<ThrottlingURLLoader> ThrottlingURLLoader::CreateLoaderAndStart(
mojom::URLLoaderFactory* factory,
std::vector<std::unique_ptr<URLLoaderThrottle>> throttles,
int32_t routing_id,
int32_t request_id,
uint32_t options,
const ResourceRequest& url_request,
mojom::URLLoaderClient* client,
const net::NetworkTrafficAnnotationTag& traffic_annotation,
scoped_refptr<base::SingleThreadTaskRunner> task_runner) {
std::unique_ptr<ThrottlingURLLoader> loader(new ThrottlingURLLoader(
std::move(throttles), client, traffic_annotation));
loader->Start(factory, routing_id, request_id, options, StartLoaderCallback(),
url_request, std::move(task_runner));
return loader;
}
// static
std::unique_ptr<ThrottlingURLLoader> ThrottlingURLLoader::CreateLoaderAndStart(
StartLoaderCallback start_loader_callback,
std::vector<std::unique_ptr<URLLoaderThrottle>> throttles,
const ResourceRequest& url_request,
mojom::URLLoaderClient* client,
const net::NetworkTrafficAnnotationTag& traffic_annotation,
scoped_refptr<base::SingleThreadTaskRunner> task_runner) {
std::unique_ptr<ThrottlingURLLoader> loader(new ThrottlingURLLoader(
std::move(throttles), client, traffic_annotation));
loader->Start(nullptr, 0, 0, mojom::kURLLoadOptionNone,
std::move(start_loader_callback), url_request,
std::move(task_runner));
return loader;
}
ThrottlingURLLoader::~ThrottlingURLLoader() {}
void ThrottlingURLLoader::FollowRedirect() {
if (url_loader_)
url_loader_->FollowRedirect();
}
void ThrottlingURLLoader::SetPriority(net::RequestPriority priority,
int32_t intra_priority_value) {
if (!url_loader_ && !loader_cancelled_) {
DCHECK_EQ(DEFERRED_START, deferred_stage_);
priority_info_ =
base::MakeUnique<PriorityInfo>(priority, intra_priority_value);
return;
}
url_loader_->SetPriority(priority, intra_priority_value);
}
void ThrottlingURLLoader::DisconnectClient() {
client_binding_.Close();
url_loader_ = nullptr;
loader_cancelled_ = true;
}
ThrottlingURLLoader::ThrottlingURLLoader(
std::vector<std::unique_ptr<URLLoaderThrottle>> throttles,
mojom::URLLoaderClient* client,
const net::NetworkTrafficAnnotationTag& traffic_annotation)
: forwarding_client_(client),
client_binding_(this),
traffic_annotation_(traffic_annotation) {
if (throttles.size() > 0) {
// TODO(yzshen): Implement a URLLoaderThrottle subclass which handles a list
// of URLLoaderThrottles.
CHECK_EQ(1u, throttles.size());
throttle_ = std::move(throttles[0]);
throttle_->set_delegate(this);
}
}
void ThrottlingURLLoader::Start(
mojom::URLLoaderFactory* factory,
int32_t routing_id,
int32_t request_id,
uint32_t options,
StartLoaderCallback start_loader_callback,
const ResourceRequest& url_request,
scoped_refptr<base::SingleThreadTaskRunner> task_runner) {
DCHECK_EQ(DEFERRED_NONE, deferred_stage_);
DCHECK(!loader_cancelled_);
if (throttle_) {
bool deferred = false;
throttle_->WillStartRequest(url_request, &deferred);
if (loader_cancelled_)
return;
if (deferred) {
deferred_stage_ = DEFERRED_START;
start_info_ =
base::MakeUnique<StartInfo>(factory, routing_id, request_id, options,
std::move(start_loader_callback),
url_request, std::move(task_runner));
return;
}
}
StartNow(factory, routing_id, request_id, options,
std::move(start_loader_callback), url_request,
std::move(task_runner));
}
void ThrottlingURLLoader::StartNow(
mojom::URLLoaderFactory* factory,
int32_t routing_id,
int32_t request_id,
uint32_t options,
StartLoaderCallback start_loader_callback,
const ResourceRequest& url_request,
scoped_refptr<base::SingleThreadTaskRunner> task_runner) {
mojom::URLLoaderClientPtr client;
client_binding_.Bind(mojo::MakeRequest(&client), std::move(task_runner));
if (factory) {
DCHECK(!start_loader_callback);
mojom::URLLoaderPtr url_loader;
auto url_loader_request = mojo::MakeRequest(&url_loader);
url_loader_ = std::move(url_loader);
factory->CreateLoaderAndStart(
std::move(url_loader_request), routing_id, request_id, options,
url_request, std::move(client),
net::MutableNetworkTrafficAnnotationTag(traffic_annotation_));
} else {
mojom::URLLoaderPtr url_loader;
auto url_loader_request = mojo::MakeRequest(&url_loader);
url_loader_ = std::move(url_loader);
std::move(start_loader_callback)
.Run(std::move(url_loader_request), std::move(client));
}
}
void ThrottlingURLLoader::OnReceiveResponse(
const ResourceResponseHead& response_head,
const base::Optional<net::SSLInfo>& ssl_info,
mojom::DownloadedTempFilePtr downloaded_file) {
DCHECK_EQ(DEFERRED_NONE, deferred_stage_);
DCHECK(!loader_cancelled_);
if (throttle_) {
bool deferred = false;
throttle_->WillProcessResponse(&deferred);
if (loader_cancelled_)
return;
if (deferred) {
deferred_stage_ = DEFERRED_RESPONSE;
response_info_ = base::MakeUnique<ResponseInfo>(
response_head, ssl_info, std::move(downloaded_file));
client_binding_.PauseIncomingMethodCallProcessing();
return;
}
}
forwarding_client_->OnReceiveResponse(response_head, ssl_info,
std::move(downloaded_file));
}
void ThrottlingURLLoader::OnReceiveRedirect(
const net::RedirectInfo& redirect_info,
const ResourceResponseHead& response_head) {
DCHECK_EQ(DEFERRED_NONE, deferred_stage_);
DCHECK(!loader_cancelled_);
if (throttle_) {
bool deferred = false;
throttle_->WillRedirectRequest(redirect_info, &deferred);
if (loader_cancelled_)
return;
if (deferred) {
deferred_stage_ = DEFERRED_REDIRECT;
redirect_info_ =
base::MakeUnique<RedirectInfo>(redirect_info, response_head);
client_binding_.PauseIncomingMethodCallProcessing();
return;
}
}
forwarding_client_->OnReceiveRedirect(redirect_info, response_head);
}
void ThrottlingURLLoader::OnDataDownloaded(int64_t data_len,
int64_t encoded_data_len) {
DCHECK_EQ(DEFERRED_NONE, deferred_stage_);
DCHECK(!loader_cancelled_);
forwarding_client_->OnDataDownloaded(data_len, encoded_data_len);
}
void ThrottlingURLLoader::OnUploadProgress(
int64_t current_position,
int64_t total_size,
OnUploadProgressCallback ack_callback) {
DCHECK_EQ(DEFERRED_NONE, deferred_stage_);
DCHECK(!loader_cancelled_);
forwarding_client_->OnUploadProgress(current_position, total_size,
std::move(ack_callback));
}
void ThrottlingURLLoader::OnReceiveCachedMetadata(
const std::vector<uint8_t>& data) {
DCHECK_EQ(DEFERRED_NONE, deferred_stage_);
DCHECK(!loader_cancelled_);
forwarding_client_->OnReceiveCachedMetadata(data);
}
void ThrottlingURLLoader::OnTransferSizeUpdated(int32_t transfer_size_diff) {
DCHECK_EQ(DEFERRED_NONE, deferred_stage_);
DCHECK(!loader_cancelled_);
forwarding_client_->OnTransferSizeUpdated(transfer_size_diff);
}
void ThrottlingURLLoader::OnStartLoadingResponseBody(
mojo::ScopedDataPipeConsumerHandle body) {
DCHECK_EQ(DEFERRED_NONE, deferred_stage_);
DCHECK(!loader_cancelled_);
forwarding_client_->OnStartLoadingResponseBody(std::move(body));
}
void ThrottlingURLLoader::OnComplete(
const ResourceRequestCompletionStatus& status) {
DCHECK_EQ(DEFERRED_NONE, deferred_stage_);
DCHECK(!loader_cancelled_);
forwarding_client_->OnComplete(status);
}
void ThrottlingURLLoader::CancelWithError(int error_code) {
// TODO(yzshen): Support a mode that cancellation also deletes the disk cache
// entry.
if (loader_cancelled_)
return;
ResourceRequestCompletionStatus request_complete_data;
request_complete_data.error_code = error_code;
request_complete_data.completion_time = base::TimeTicks::Now();
deferred_stage_ = DEFERRED_NONE;
DisconnectClient();
forwarding_client_->OnComplete(request_complete_data);
}
void ThrottlingURLLoader::Resume() {
if (loader_cancelled_ || deferred_stage_ == DEFERRED_NONE)
return;
switch (deferred_stage_) {
case DEFERRED_START: {
StartNow(start_info_->url_loader_factory, start_info_->routing_id,
start_info_->request_id, start_info_->options,
std::move(start_info_->start_loader_callback),
start_info_->url_request, std::move(start_info_->task_runner));
if (priority_info_) {
auto priority_info = std::move(priority_info_);
url_loader_->SetPriority(priority_info->priority,
priority_info->intra_priority_value);
}
break;
}
case DEFERRED_REDIRECT: {
client_binding_.ResumeIncomingMethodCallProcessing();
forwarding_client_->OnReceiveRedirect(redirect_info_->redirect_info,
redirect_info_->response_head);
break;
}
case DEFERRED_RESPONSE: {
client_binding_.ResumeIncomingMethodCallProcessing();
forwarding_client_->OnReceiveResponse(
response_info_->response_head, response_info_->ssl_info,
std::move(response_info_->downloaded_file));
break;
}
default:
NOTREACHED();
break;
}
deferred_stage_ = DEFERRED_NONE;
}
} // namespace content