| Index: net/test/url_request/url_request_slow_download_job.cc
|
| diff --git a/net/test/url_request/url_request_slow_download_job.cc b/net/test/url_request/url_request_slow_download_job.cc
|
| deleted file mode 100644
|
| index 718a9f05502cef03e616160b891d75e8da05ae22..0000000000000000000000000000000000000000
|
| --- a/net/test/url_request/url_request_slow_download_job.cc
|
| +++ /dev/null
|
| @@ -1,265 +0,0 @@
|
| -// 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/test/url_request/url_request_slow_download_job.h"
|
| -
|
| -#include "base/bind.h"
|
| -#include "base/compiler_specific.h"
|
| -#include "base/logging.h"
|
| -#include "base/message_loop/message_loop.h"
|
| -#include "base/strings/string_util.h"
|
| -#include "base/strings/stringprintf.h"
|
| -#include "net/base/io_buffer.h"
|
| -#include "net/base/net_errors.h"
|
| -#include "net/http/http_response_headers.h"
|
| -#include "net/url_request/url_request.h"
|
| -#include "net/url_request/url_request_filter.h"
|
| -#include "url/gurl.h"
|
| -
|
| -namespace net {
|
| -
|
| -const char URLRequestSlowDownloadJob::kUnknownSizeUrl[] =
|
| - "http://url.handled.by.slow.download/download-unknown-size";
|
| -const char URLRequestSlowDownloadJob::kKnownSizeUrl[] =
|
| - "http://url.handled.by.slow.download/download-known-size";
|
| -const char URLRequestSlowDownloadJob::kFinishDownloadUrl[] =
|
| - "http://url.handled.by.slow.download/download-finish";
|
| -const char URLRequestSlowDownloadJob::kErrorDownloadUrl[] =
|
| - "http://url.handled.by.slow.download/download-error";
|
| -
|
| -const int URLRequestSlowDownloadJob::kFirstDownloadSize = 1024 * 35;
|
| -const int URLRequestSlowDownloadJob::kSecondDownloadSize = 1024 * 10;
|
| -
|
| -// static
|
| -base::LazyInstance<URLRequestSlowDownloadJob::SlowJobsSet>::Leaky
|
| - URLRequestSlowDownloadJob::pending_requests_ = LAZY_INSTANCE_INITIALIZER;
|
| -
|
| -void URLRequestSlowDownloadJob::Start() {
|
| - base::MessageLoop::current()->PostTask(
|
| - FROM_HERE, base::Bind(&URLRequestSlowDownloadJob::StartAsync,
|
| - weak_factory_.GetWeakPtr()));
|
| -}
|
| -
|
| -// static
|
| -void URLRequestSlowDownloadJob::AddUrlHandler() {
|
| - URLRequestFilter* filter = URLRequestFilter::GetInstance();
|
| - filter->AddUrlHandler(GURL(kUnknownSizeUrl),
|
| - &URLRequestSlowDownloadJob::Factory);
|
| - filter->AddUrlHandler(GURL(kKnownSizeUrl),
|
| - &URLRequestSlowDownloadJob::Factory);
|
| - filter->AddUrlHandler(GURL(kFinishDownloadUrl),
|
| - &URLRequestSlowDownloadJob::Factory);
|
| - filter->AddUrlHandler(GURL(kErrorDownloadUrl),
|
| - &URLRequestSlowDownloadJob::Factory);
|
| -}
|
| -
|
| -// static
|
| -URLRequestJob* URLRequestSlowDownloadJob::Factory(
|
| - URLRequest* request,
|
| - NetworkDelegate* network_delegate,
|
| - const std::string& scheme) {
|
| - URLRequestSlowDownloadJob* job =
|
| - new URLRequestSlowDownloadJob(request, network_delegate);
|
| - if (request->url().spec() != kFinishDownloadUrl &&
|
| - request->url().spec() != kErrorDownloadUrl)
|
| - pending_requests_.Get().insert(job);
|
| - return job;
|
| -}
|
| -
|
| -// static
|
| -size_t URLRequestSlowDownloadJob::NumberOutstandingRequests() {
|
| - return pending_requests_.Get().size();
|
| -}
|
| -
|
| -// static
|
| -void URLRequestSlowDownloadJob::FinishPendingRequests() {
|
| - typedef std::set<URLRequestSlowDownloadJob*> JobList;
|
| - for (JobList::iterator it = pending_requests_.Get().begin();
|
| - it != pending_requests_.Get().end(); ++it) {
|
| - (*it)->set_should_finish_download();
|
| - }
|
| -}
|
| -
|
| -void URLRequestSlowDownloadJob::ErrorPendingRequests() {
|
| - typedef std::set<URLRequestSlowDownloadJob*> JobList;
|
| - for (JobList::iterator it = pending_requests_.Get().begin();
|
| - it != pending_requests_.Get().end(); ++it) {
|
| - (*it)->set_should_error_download();
|
| - }
|
| -}
|
| -
|
| -URLRequestSlowDownloadJob::URLRequestSlowDownloadJob(
|
| - URLRequest* request,
|
| - NetworkDelegate* network_delegate)
|
| - : URLRequestJob(request, network_delegate),
|
| - bytes_already_sent_(0),
|
| - should_error_download_(false),
|
| - should_finish_download_(false),
|
| - buffer_size_(0),
|
| - weak_factory_(this) {
|
| -}
|
| -
|
| -void URLRequestSlowDownloadJob::StartAsync() {
|
| - if (LowerCaseEqualsASCII(kFinishDownloadUrl, request_->url().spec().c_str()))
|
| - URLRequestSlowDownloadJob::FinishPendingRequests();
|
| - if (LowerCaseEqualsASCII(kErrorDownloadUrl, request_->url().spec().c_str()))
|
| - URLRequestSlowDownloadJob::ErrorPendingRequests();
|
| -
|
| - NotifyHeadersComplete();
|
| -}
|
| -
|
| -// ReadRawData and CheckDoneStatus together implement a state
|
| -// machine. ReadRawData may be called arbitrarily by the network stack.
|
| -// It responds by:
|
| -// * If there are bytes remaining in the first chunk, they are
|
| -// returned.
|
| -// [No bytes remaining in first chunk. ]
|
| -// * If should_finish_download_ is not set, it returns IO_PENDING,
|
| -// and starts calling CheckDoneStatus on a regular timer.
|
| -// [should_finish_download_ set.]
|
| -// * If there are bytes remaining in the second chunk, they are filled.
|
| -// * Otherwise, return *bytes_read = 0 to indicate end of request.
|
| -// CheckDoneStatus is called on a regular basis, in the specific
|
| -// case where we have transmitted all of the first chunk and none of the
|
| -// second. If should_finish_download_ becomes set, it will "complete"
|
| -// the ReadRawData call that spawned off the CheckDoneStatus() repeated call.
|
| -//
|
| -// FillBufferHelper is a helper function that does the actual work of figuring
|
| -// out where in the state machine we are and how we should fill the buffer.
|
| -// It returns an enum indicating the state of the read.
|
| -URLRequestSlowDownloadJob::ReadStatus
|
| -URLRequestSlowDownloadJob::FillBufferHelper(IOBuffer* buf,
|
| - int buf_size,
|
| - int* bytes_written) {
|
| - if (bytes_already_sent_ < kFirstDownloadSize) {
|
| - int bytes_to_write =
|
| - std::min(kFirstDownloadSize - bytes_already_sent_, buf_size);
|
| - for (int i = 0; i < bytes_to_write; ++i) {
|
| - buf->data()[i] = '*';
|
| - }
|
| - *bytes_written = bytes_to_write;
|
| - bytes_already_sent_ += bytes_to_write;
|
| - return BUFFER_FILLED;
|
| - }
|
| -
|
| - if (!should_finish_download_)
|
| - return REQUEST_BLOCKED;
|
| -
|
| - if (bytes_already_sent_ < kFirstDownloadSize + kSecondDownloadSize) {
|
| - int bytes_to_write =
|
| - std::min(kFirstDownloadSize + kSecondDownloadSize - bytes_already_sent_,
|
| - buf_size);
|
| - for (int i = 0; i < bytes_to_write; ++i) {
|
| - buf->data()[i] = '*';
|
| - }
|
| - *bytes_written = bytes_to_write;
|
| - bytes_already_sent_ += bytes_to_write;
|
| - return BUFFER_FILLED;
|
| - }
|
| -
|
| - return REQUEST_COMPLETE;
|
| -}
|
| -
|
| -bool URLRequestSlowDownloadJob::ReadRawData(IOBuffer* buf,
|
| - int buf_size,
|
| - int* bytes_read) {
|
| - if (LowerCaseEqualsASCII(kFinishDownloadUrl,
|
| - request_->url().spec().c_str()) ||
|
| - LowerCaseEqualsASCII(kErrorDownloadUrl, request_->url().spec().c_str())) {
|
| - VLOG(10) << __FUNCTION__ << " called w/ kFinish/ErrorDownloadUrl.";
|
| - *bytes_read = 0;
|
| - return true;
|
| - }
|
| -
|
| - VLOG(10) << __FUNCTION__ << " called at position " << bytes_already_sent_
|
| - << " in the stream.";
|
| - ReadStatus status = FillBufferHelper(buf, buf_size, bytes_read);
|
| - switch (status) {
|
| - case BUFFER_FILLED:
|
| - return true;
|
| - case REQUEST_BLOCKED:
|
| - buffer_ = buf;
|
| - buffer_size_ = buf_size;
|
| - SetStatus(URLRequestStatus(URLRequestStatus::IO_PENDING, 0));
|
| - base::MessageLoop::current()->PostDelayedTask(
|
| - FROM_HERE, base::Bind(&URLRequestSlowDownloadJob::CheckDoneStatus,
|
| - weak_factory_.GetWeakPtr()),
|
| - base::TimeDelta::FromMilliseconds(100));
|
| - return false;
|
| - case REQUEST_COMPLETE:
|
| - *bytes_read = 0;
|
| - return true;
|
| - }
|
| - NOTREACHED();
|
| - return true;
|
| -}
|
| -
|
| -void URLRequestSlowDownloadJob::CheckDoneStatus() {
|
| - if (should_finish_download_) {
|
| - VLOG(10) << __FUNCTION__ << " called w/ should_finish_download_ set.";
|
| - DCHECK(NULL != buffer_.get());
|
| - int bytes_written = 0;
|
| - ReadStatus status =
|
| - FillBufferHelper(buffer_.get(), buffer_size_, &bytes_written);
|
| - DCHECK_EQ(BUFFER_FILLED, status);
|
| - buffer_ = NULL; // Release the reference.
|
| - SetStatus(URLRequestStatus());
|
| - NotifyReadComplete(bytes_written);
|
| - } else if (should_error_download_) {
|
| - VLOG(10) << __FUNCTION__ << " called w/ should_finish_ownload_ set.";
|
| - NotifyDone(
|
| - URLRequestStatus(URLRequestStatus::FAILED, ERR_CONNECTION_RESET));
|
| - } else {
|
| - base::MessageLoop::current()->PostDelayedTask(
|
| - FROM_HERE, base::Bind(&URLRequestSlowDownloadJob::CheckDoneStatus,
|
| - weak_factory_.GetWeakPtr()),
|
| - base::TimeDelta::FromMilliseconds(100));
|
| - }
|
| -}
|
| -
|
| -// Public virtual version.
|
| -void URLRequestSlowDownloadJob::GetResponseInfo(HttpResponseInfo* info) {
|
| - // Forward to private const version.
|
| - GetResponseInfoConst(info);
|
| -}
|
| -
|
| -URLRequestSlowDownloadJob::~URLRequestSlowDownloadJob() {
|
| - pending_requests_.Get().erase(this);
|
| -}
|
| -
|
| -// Private const version.
|
| -void URLRequestSlowDownloadJob::GetResponseInfoConst(
|
| - HttpResponseInfo* info) const {
|
| - // Send back mock headers.
|
| - std::string raw_headers;
|
| - if (LowerCaseEqualsASCII(kFinishDownloadUrl,
|
| - request_->url().spec().c_str()) ||
|
| - LowerCaseEqualsASCII(kErrorDownloadUrl, request_->url().spec().c_str())) {
|
| - raw_headers.append(
|
| - "HTTP/1.1 200 OK\n"
|
| - "Content-type: text/plain\n");
|
| - } else {
|
| - raw_headers.append(
|
| - "HTTP/1.1 200 OK\n"
|
| - "Content-type: application/octet-stream\n"
|
| - "Cache-Control: max-age=0\n");
|
| -
|
| - if (LowerCaseEqualsASCII(kKnownSizeUrl, request_->url().spec().c_str())) {
|
| - raw_headers.append(base::StringPrintf(
|
| - "Content-Length: %d\n", kFirstDownloadSize + kSecondDownloadSize));
|
| - }
|
| - }
|
| -
|
| - // ParseRawHeaders expects \0 to end each header line.
|
| - ReplaceSubstringsAfterOffset(&raw_headers, 0, "\n", std::string("\0", 1));
|
| - info->headers = new HttpResponseHeaders(raw_headers);
|
| -}
|
| -
|
| -bool URLRequestSlowDownloadJob::GetMimeType(std::string* mime_type) const {
|
| - HttpResponseInfo info;
|
| - GetResponseInfoConst(&info);
|
| - return info.headers.get() && info.headers->GetMimeType(mime_type);
|
| -}
|
| -
|
| -} // namespace net
|
|
|