blob: af05919c78cadb119d9fe10dab2f98fb55da1024 [file] [log] [blame]
// Copyright 2016 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/child/url_response_body_consumer.h"
#include "base/bind.h"
#include "base/callback_forward.h"
#include "base/macros.h"
#include "base/memory/ptr_util.h"
#include "base/run_loop.h"
#include "content/child/request_extra_data.h"
#include "content/child/resource_dispatcher.h"
#include "content/common/resource_messages.h"
#include "content/common/resource_request.h"
#include "content/common/resource_request_completion_status.h"
#include "content/common/service_worker/service_worker_types.h"
#include "content/public/child/request_peer.h"
#include "content/public/common/request_context_frame_type.h"
#include "net/base/request_priority.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "url/gurl.h"
#include "url/origin.h"
namespace content {
namespace {
class TestRequestPeer : public RequestPeer {
public:
struct Context;
TestRequestPeer(Context* context,
scoped_refptr<base::SingleThreadTaskRunner> task_runner)
: context_(context), task_runner_(std::move(task_runner)) {}
void OnUploadProgress(uint64_t position, uint64_t size) override {
ADD_FAILURE() << "OnUploadProgress should not be called.";
}
bool OnReceivedRedirect(const net::RedirectInfo& redirect_info,
const ResourceResponseInfo& info) override {
ADD_FAILURE() << "OnReceivedRedirect should not be called.";
return false;
}
void OnReceivedResponse(const ResourceResponseInfo& info) override {
ADD_FAILURE() << "OnReceivedResponse should not be called.";
}
void OnDownloadedData(int len, int encoded_data_length) override {
ADD_FAILURE() << "OnDownloadedData should not be called.";
}
void OnReceivedData(std::unique_ptr<ReceivedData> data) override {
EXPECT_FALSE(context_->complete);
context_->data.append(data->payload(), data->length());
if (context_->release_data_asynchronously)
task_runner_->DeleteSoon(FROM_HERE, data.release());
context_->run_loop_quit_closure.Run();
}
void OnTransferSizeUpdated(int transfer_size_diff) override {}
void OnCompletedRequest(int error_code,
bool was_ignored_by_handler,
bool stale_copy_in_cache,
const base::TimeTicks& completion_time,
int64_t total_transfer_size,
int64_t encoded_body_size) override {
EXPECT_FALSE(context_->complete);
context_->complete = true;
context_->error_code = error_code;
context_->run_loop_quit_closure.Run();
}
struct Context {
// Data received. If downloading to file, remains empty.
std::string data;
bool complete = false;
base::Closure run_loop_quit_closure;
int error_code = net::OK;
bool release_data_asynchronously = false;
};
private:
Context* context_;
scoped_refptr<base::SingleThreadTaskRunner> task_runner_;
DISALLOW_COPY_AND_ASSIGN(TestRequestPeer);
};
class URLResponseBodyConsumerTest : public ::testing::Test,
public ::IPC::Sender {
protected:
URLResponseBodyConsumerTest()
: dispatcher_(new ResourceDispatcher(this, message_loop_.task_runner())) {
}
~URLResponseBodyConsumerTest() override {
dispatcher_.reset();
base::RunLoop().RunUntilIdle();
}
bool Send(IPC::Message* message) override {
delete message;
return true;
}
std::unique_ptr<ResourceRequest> CreateResourceRequest() {
std::unique_ptr<ResourceRequest> request(new ResourceRequest);
request->method = "GET";
request->url = GURL("http://www.example.com/");
request->priority = net::LOW;
request->appcache_host_id = 0;
request->fetch_request_mode = FETCH_REQUEST_MODE_NO_CORS;
request->fetch_frame_type = REQUEST_CONTEXT_FRAME_TYPE_NONE;
const RequestExtraData extra_data;
extra_data.CopyToResourceRequest(request.get());
return request;
}
MojoCreateDataPipeOptions CreateDataPipeOptions() {
MojoCreateDataPipeOptions options;
options.struct_size = sizeof(MojoCreateDataPipeOptions);
options.flags = MOJO_CREATE_DATA_PIPE_OPTIONS_FLAG_NONE;
options.element_num_bytes = 1;
options.capacity_num_bytes = 1024;
return options;
}
// Returns the request id.
int SetUpRequestPeer(std::unique_ptr<ResourceRequest> request,
TestRequestPeer::Context* context) {
return dispatcher_->StartAsync(
std::move(request), 0, nullptr, url::Origin(),
base::MakeUnique<TestRequestPeer>(context, message_loop_.task_runner()),
blink::WebURLRequest::LoadingIPCType::kChromeIPC, nullptr,
mojo::ScopedDataPipeConsumerHandle());
}
void Run(TestRequestPeer::Context* context) {
base::RunLoop run_loop;
context->run_loop_quit_closure = run_loop.QuitClosure();
run_loop.Run();
}
base::MessageLoop message_loop_;
std::unique_ptr<ResourceDispatcher> dispatcher_;
static const MojoWriteDataFlags kNone = MOJO_WRITE_DATA_FLAG_NONE;
};
TEST_F(URLResponseBodyConsumerTest, ReceiveData) {
TestRequestPeer::Context context;
std::unique_ptr<ResourceRequest> request(CreateResourceRequest());
int request_id = SetUpRequestPeer(std::move(request), &context);
mojo::DataPipe data_pipe(CreateDataPipeOptions());
scoped_refptr<URLResponseBodyConsumer> consumer(new URLResponseBodyConsumer(
request_id, dispatcher_.get(), std::move(data_pipe.consumer_handle),
message_loop_.task_runner()));
mojo::ScopedDataPipeProducerHandle writer =
std::move(data_pipe.producer_handle);
std::string buffer = "hello";
uint32_t size = buffer.size();
MojoResult result =
mojo::WriteDataRaw(writer.get(), buffer.c_str(), &size, kNone);
ASSERT_EQ(MOJO_RESULT_OK, result);
ASSERT_EQ(buffer.size(), size);
Run(&context);
EXPECT_FALSE(context.complete);
EXPECT_EQ("hello", context.data);
}
TEST_F(URLResponseBodyConsumerTest, OnCompleteThenClose) {
TestRequestPeer::Context context;
std::unique_ptr<ResourceRequest> request(CreateResourceRequest());
int request_id = SetUpRequestPeer(std::move(request), &context);
mojo::DataPipe data_pipe(CreateDataPipeOptions());
scoped_refptr<URLResponseBodyConsumer> consumer(new URLResponseBodyConsumer(
request_id, dispatcher_.get(), std::move(data_pipe.consumer_handle),
message_loop_.task_runner()));
consumer->OnComplete(ResourceRequestCompletionStatus());
mojo::ScopedDataPipeProducerHandle writer =
std::move(data_pipe.producer_handle);
std::string buffer = "hello";
uint32_t size = buffer.size();
MojoResult result =
mojo::WriteDataRaw(writer.get(), buffer.c_str(), &size, kNone);
ASSERT_EQ(MOJO_RESULT_OK, result);
ASSERT_EQ(buffer.size(), size);
Run(&context);
writer.reset();
EXPECT_FALSE(context.complete);
EXPECT_EQ("hello", context.data);
Run(&context);
EXPECT_TRUE(context.complete);
EXPECT_EQ("hello", context.data);
}
// Release the received data asynchronously. This leads to MOJO_RESULT_BUSY
// from the BeginReadDataRaw call in OnReadable.
TEST_F(URLResponseBodyConsumerTest, OnCompleteThenCloseWithAsyncRelease) {
TestRequestPeer::Context context;
context.release_data_asynchronously = true;
std::unique_ptr<ResourceRequest> request(CreateResourceRequest());
int request_id = SetUpRequestPeer(std::move(request), &context);
mojo::DataPipe data_pipe(CreateDataPipeOptions());
scoped_refptr<URLResponseBodyConsumer> consumer(new URLResponseBodyConsumer(
request_id, dispatcher_.get(), std::move(data_pipe.consumer_handle),
message_loop_.task_runner()));
consumer->OnComplete(ResourceRequestCompletionStatus());
mojo::ScopedDataPipeProducerHandle writer =
std::move(data_pipe.producer_handle);
std::string buffer = "hello";
uint32_t size = buffer.size();
MojoResult result =
mojo::WriteDataRaw(writer.get(), buffer.c_str(), &size, kNone);
ASSERT_EQ(MOJO_RESULT_OK, result);
ASSERT_EQ(buffer.size(), size);
Run(&context);
writer.reset();
EXPECT_FALSE(context.complete);
EXPECT_EQ("hello", context.data);
Run(&context);
EXPECT_TRUE(context.complete);
EXPECT_EQ("hello", context.data);
}
TEST_F(URLResponseBodyConsumerTest, CloseThenOnComplete) {
TestRequestPeer::Context context;
std::unique_ptr<ResourceRequest> request(CreateResourceRequest());
int request_id = SetUpRequestPeer(std::move(request), &context);
mojo::DataPipe data_pipe(CreateDataPipeOptions());
scoped_refptr<URLResponseBodyConsumer> consumer(new URLResponseBodyConsumer(
request_id, dispatcher_.get(), std::move(data_pipe.consumer_handle),
message_loop_.task_runner()));
ResourceRequestCompletionStatus status;
status.error_code = net::ERR_FAILED;
data_pipe.producer_handle.reset();
consumer->OnComplete(status);
Run(&context);
EXPECT_TRUE(context.complete);
EXPECT_EQ(net::ERR_FAILED, context.error_code);
EXPECT_EQ("", context.data);
}
TEST_F(URLResponseBodyConsumerTest, TooBigChunkShouldBeSplit) {
constexpr auto kMaxNumConsumedBytesInTask =
URLResponseBodyConsumer::kMaxNumConsumedBytesInTask;
TestRequestPeer::Context context;
std::unique_ptr<ResourceRequest> request(CreateResourceRequest());
int request_id = SetUpRequestPeer(std::move(request), &context);
auto options = CreateDataPipeOptions();
options.capacity_num_bytes = 2 * kMaxNumConsumedBytesInTask;
mojo::DataPipe data_pipe(options);
mojo::ScopedDataPipeProducerHandle writer =
std::move(data_pipe.producer_handle);
void* buffer = nullptr;
uint32_t size = 0;
MojoResult result =
mojo::BeginWriteDataRaw(writer.get(), &buffer, &size, kNone);
ASSERT_EQ(MOJO_RESULT_OK, result);
ASSERT_EQ(options.capacity_num_bytes, size);
memset(buffer, 'a', kMaxNumConsumedBytesInTask);
memset(static_cast<char*>(buffer) + kMaxNumConsumedBytesInTask, 'b',
kMaxNumConsumedBytesInTask);
result = mojo::EndWriteDataRaw(writer.get(), size);
ASSERT_EQ(MOJO_RESULT_OK, result);
scoped_refptr<URLResponseBodyConsumer> consumer(new URLResponseBodyConsumer(
request_id, dispatcher_.get(), std::move(data_pipe.consumer_handle),
message_loop_.task_runner()));
Run(&context);
EXPECT_EQ(std::string(kMaxNumConsumedBytesInTask, 'a'), context.data);
context.data.clear();
Run(&context);
EXPECT_EQ(std::string(kMaxNumConsumedBytesInTask, 'b'), context.data);
}
} // namespace
} // namespace content