url_fetcher_core.cc revision f2477e01787aa58f445919b809d89e252beef54f
15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved.
25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be
35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file.
45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/url_request/url_fetcher_core.h"
65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/bind.h"
85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/logging.h"
95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/metrics/histogram.h"
10f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles)#include "base/sequenced_task_runner.h"
112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/single_thread_task_runner.h"
125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/stl_util.h"
135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/thread_task_runner_handle.h"
145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/tracked_objects.h"
155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/base/io_buffer.h"
165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/base/load_flags.h"
175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/base/net_errors.h"
180f1bc08d4cfcc34181b0b5cbf065c40f687bf740Torne (Richard Coles)#include "net/base/request_priority.h"
192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "net/base/upload_bytes_element_reader.h"
202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "net/base/upload_data_stream.h"
212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "net/base/upload_file_element_reader.h"
225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/http/http_response_headers.h"
235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/url_request/url_fetcher_delegate.h"
242a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "net/url_request/url_fetcher_response_writer.h"
255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/url_request/url_request_context.h"
265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/url_request/url_request_context_getter.h"
275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/url_request/url_request_throttler_manager.h"
285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace {
305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const int kBufferSize = 4096;
325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const int kUploadProgressTimerInterval = 100;
335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool g_interception_enabled = false;
342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)bool g_ignore_certificate_requests = false;
355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
361e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles)void EmptyCompletionCallback(int result) {}
371e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles)
385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}  // namespace
395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace net {
415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// URLFetcherCore::Registry ---------------------------------------------------
435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)URLFetcherCore::Registry::Registry() {}
455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)URLFetcherCore::Registry::~Registry() {}
465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::Registry::AddURLFetcherCore(URLFetcherCore* core) {
485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(!ContainsKey(fetchers_, core));
495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  fetchers_.insert(core);
505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::Registry::RemoveURLFetcherCore(URLFetcherCore* core) {
535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(ContainsKey(fetchers_, core));
545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  fetchers_.erase(core);
555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::Registry::CancelAll() {
585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  while (!fetchers_.empty())
594e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)    (*fetchers_.begin())->CancelURLRequest(ERR_ABORTED);
605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// URLFetcherCore -------------------------------------------------------------
635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// static
655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)base::LazyInstance<URLFetcherCore::Registry>
665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    URLFetcherCore::g_registry = LAZY_INSTANCE_INITIALIZER;
675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)URLFetcherCore::URLFetcherCore(URLFetcher* fetcher,
695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                               const GURL& original_url,
705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                               URLFetcher::RequestType request_type,
715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                               URLFetcherDelegate* d)
725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    : fetcher_(fetcher),
735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      original_url_(original_url),
745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      request_type_(request_type),
755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      delegate_(d),
767d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)      delegate_task_runner_(base::ThreadTaskRunnerHandle::Get()),
775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      load_flags_(LOAD_NORMAL),
785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      response_code_(URLFetcher::RESPONSE_CODE_INVALID),
795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      buffer_(new IOBuffer(kBufferSize)),
805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      url_request_data_key_(NULL),
815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      was_fetched_via_proxy_(false),
822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      upload_content_set_(false),
83c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)      upload_range_offset_(0),
84c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)      upload_range_length_(0),
855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      is_chunked_upload_(false),
865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      was_cancelled_(false),
875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      stop_on_redirect_(false),
885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      stopped_on_redirect_(false),
895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      automatically_retry_on_5xx_(true),
902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      num_retries_on_5xx_(0),
912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      max_retries_on_5xx_(0),
922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      num_retries_on_network_changes_(0),
932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      max_retries_on_network_changes_(0),
945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      current_upload_bytes_(-1),
955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      current_response_bytes_(0),
965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      total_response_bytes_(-1) {
975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  CHECK(original_url_.is_valid());
985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::Start() {
101868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)  DCHECK(delegate_task_runner_.get());
102868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)  DCHECK(request_context_getter_.get()) << "We need an URLRequestContext!";
103868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)  if (network_task_runner_.get()) {
1045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    DCHECK_EQ(network_task_runner_,
1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)              request_context_getter_->GetNetworkTaskRunner());
1065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  } else {
1075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    network_task_runner_ = request_context_getter_->GetNetworkTaskRunner();
1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
1095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(network_task_runner_.get()) << "We need an IO task runner";
1105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  network_task_runner_->PostTask(
1125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      FROM_HERE, base::Bind(&URLFetcherCore::StartOnIOThread, this));
1135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
1145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::Stop() {
116868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)  if (delegate_task_runner_.get())  // May be NULL in tests.
1175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    DCHECK(delegate_task_runner_->BelongsToCurrentThread());
1185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  delegate_ = NULL;
1205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  fetcher_ = NULL;
1215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (!network_task_runner_.get())
1225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return;
1235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (network_task_runner_->RunsTasksOnCurrentThread()) {
1244e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)    CancelURLRequest(ERR_ABORTED);
1255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  } else {
1265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    network_task_runner_->PostTask(
1274e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)        FROM_HERE,
1284e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)        base::Bind(&URLFetcherCore::CancelURLRequest, this, ERR_ABORTED));
1295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
1305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
1315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::SetUploadData(const std::string& upload_content_type,
1335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                                   const std::string& upload_content) {
1345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(!is_chunked_upload_);
1352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  DCHECK(!upload_content_set_);
1362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  DCHECK(upload_content_.empty());
1372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  DCHECK(upload_file_path_.empty());
1382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  DCHECK(upload_content_type_.empty());
1392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
1402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  // Empty |upload_content_type| is allowed iff the |upload_content| is empty.
1412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  DCHECK(upload_content.empty() || !upload_content_type.empty());
1422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
1435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  upload_content_type_ = upload_content_type;
1445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  upload_content_ = upload_content;
1452a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  upload_content_set_ = true;
1462a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)}
1472a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
1482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void URLFetcherCore::SetUploadFilePath(
1492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    const std::string& upload_content_type,
1502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    const base::FilePath& file_path,
151c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)    uint64 range_offset,
152c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)    uint64 range_length,
1532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    scoped_refptr<base::TaskRunner> file_task_runner) {
1542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  DCHECK(!is_chunked_upload_);
1552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  DCHECK(!upload_content_set_);
1562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  DCHECK(upload_content_.empty());
1572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  DCHECK(upload_file_path_.empty());
158c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)  DCHECK_EQ(upload_range_offset_, 0ULL);
159c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)  DCHECK_EQ(upload_range_length_, 0ULL);
1602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  DCHECK(upload_content_type_.empty());
1612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  DCHECK(!upload_content_type.empty());
1622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
1632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  upload_content_type_ = upload_content_type;
1642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  upload_file_path_ = file_path;
165c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)  upload_range_offset_ = range_offset;
166c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)  upload_range_length_ = range_length;
1672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  upload_file_task_runner_ = file_task_runner;
1682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  upload_content_set_ = true;
1695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
1705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::SetChunkedUpload(const std::string& content_type) {
1725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(is_chunked_upload_ ||
1735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)         (upload_content_type_.empty() &&
1745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          upload_content_.empty()));
1752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
1762a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  // Empty |content_type| is not allowed here, because it is impossible
1772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  // to ensure non-empty upload content as it is not yet supplied.
1782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  DCHECK(!content_type.empty());
1792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
1805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  upload_content_type_ = content_type;
1815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  upload_content_.clear();
1825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  is_chunked_upload_ = true;
1835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
1845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::AppendChunkToUpload(const std::string& content,
1865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                                         bool is_last_chunk) {
187868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)  DCHECK(delegate_task_runner_.get());
1885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(network_task_runner_.get());
1895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  network_task_runner_->PostTask(
1905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      FROM_HERE,
1915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      base::Bind(&URLFetcherCore::CompleteAddingUploadDataChunk, this, content,
1925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                 is_last_chunk));
1935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
1945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::SetLoadFlags(int load_flags) {
1965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  load_flags_ = load_flags;
1975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
1985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)int URLFetcherCore::GetLoadFlags() const {
2005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return load_flags_;
2015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::SetReferrer(const std::string& referrer) {
2045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  referrer_ = referrer;
2055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::SetExtraRequestHeaders(
2085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    const std::string& extra_request_headers) {
2095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  extra_request_headers_.Clear();
2105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  extra_request_headers_.AddHeadersFromString(extra_request_headers);
2115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::AddExtraRequestHeader(const std::string& header_line) {
2145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  extra_request_headers_.AddHeaderFromString(header_line);
2155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::GetExtraRequestHeaders(
2185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    HttpRequestHeaders* headers) const {
2195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  headers->CopyFrom(extra_request_headers_);
2205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::SetRequestContext(
2235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    URLRequestContextGetter* request_context_getter) {
224868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)  DCHECK(!request_context_getter_.get());
2255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(request_context_getter);
2265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  request_context_getter_ = request_context_getter;
2275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::SetFirstPartyForCookies(
2305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    const GURL& first_party_for_cookies) {
2315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(first_party_for_cookies_.is_empty());
2325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  first_party_for_cookies_ = first_party_for_cookies;
2335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::SetURLRequestUserData(
2365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    const void* key,
2375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    const URLFetcher::CreateDataCallback& create_data_callback) {
2385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(key);
2395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(!create_data_callback.is_null());
2405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  url_request_data_key_ = key;
2415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  url_request_create_data_callback_ = create_data_callback;
2425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::SetStopOnRedirect(bool stop_on_redirect) {
2455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  stop_on_redirect_ = stop_on_redirect;
2465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::SetAutomaticallyRetryOn5xx(bool retry) {
2495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  automatically_retry_on_5xx_ = retry;
2505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void URLFetcherCore::SetMaxRetriesOn5xx(int max_retries) {
2532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  max_retries_on_5xx_ = max_retries;
2545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)int URLFetcherCore::GetMaxRetriesOn5xx() const {
2572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  return max_retries_on_5xx_;
2585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)base::TimeDelta URLFetcherCore::GetBackoffDelay() const {
2615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return backoff_delay_;
2625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void URLFetcherCore::SetAutomaticallyRetryOnNetworkChanges(int max_retries) {
2652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  max_retries_on_network_changes_ = max_retries;
2662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)}
2672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
2685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::SaveResponseToFileAtPath(
2692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    const base::FilePath& file_path,
270f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles)    scoped_refptr<base::SequencedTaskRunner> file_task_runner) {
2715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(delegate_task_runner_->BelongsToCurrentThread());
2724e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)  SaveResponseWithWriter(scoped_ptr<URLFetcherResponseWriter>(
2734e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)      new URLFetcherFileWriter(file_task_runner, file_path)));
2745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::SaveResponseToTemporaryFile(
277f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles)    scoped_refptr<base::SequencedTaskRunner> file_task_runner) {
2785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(delegate_task_runner_->BelongsToCurrentThread());
2794e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)  SaveResponseWithWriter(scoped_ptr<URLFetcherResponseWriter>(
2804e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)      new URLFetcherFileWriter(file_task_runner, base::FilePath())));
2814e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)}
2824e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)
2834e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)void URLFetcherCore::SaveResponseWithWriter(
2844e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)    scoped_ptr<URLFetcherResponseWriter> response_writer) {
2854e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)  DCHECK(delegate_task_runner_->BelongsToCurrentThread());
2864e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)  response_writer_ = response_writer.Pass();
2875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)HttpResponseHeaders* URLFetcherCore::GetResponseHeaders() const {
290868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)  return response_headers_.get();
2915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// TODO(panayiotis): socket_address_ is written in the IO thread,
2945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// if this is accessed in the UI thread, this could result in a race.
2955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Same for response_headers_ above and was_fetched_via_proxy_ below.
2965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)HostPortPair URLFetcherCore::GetSocketAddress() const {
2975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return socket_address_;
2985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool URLFetcherCore::WasFetchedViaProxy() const {
3015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return was_fetched_via_proxy_;
3025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
3035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const GURL& URLFetcherCore::GetOriginalURL() const {
3055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return original_url_;
3065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
3075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const GURL& URLFetcherCore::GetURL() const {
3095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return url_;
3105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
3115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const URLRequestStatus& URLFetcherCore::GetStatus() const {
3135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return status_;
3145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
3155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)int URLFetcherCore::GetResponseCode() const {
3175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return response_code_;
3185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
3195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const ResponseCookies& URLFetcherCore::GetCookies() const {
3215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return cookies_;
3225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
3235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::ReceivedContentWasMalformed() {
3255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(delegate_task_runner_->BelongsToCurrentThread());
3265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (network_task_runner_.get()) {
3275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    network_task_runner_->PostTask(
3285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        FROM_HERE, base::Bind(&URLFetcherCore::NotifyMalformedContent, this));
3295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
3305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
3315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool URLFetcherCore::GetResponseAsString(
3335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    std::string* out_response_string) const {
3344e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)  URLFetcherStringWriter* string_writer =
3354e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)      response_writer_ ? response_writer_->AsStringWriter() : NULL;
3364e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)  if (!string_writer)
3375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return false;
3385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3394e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)  *out_response_string = string_writer->data();
3405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  UMA_HISTOGRAM_MEMORY_KB("UrlFetcher.StringResponseSize",
3414e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)                          (string_writer->data().length() / 1024));
3425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return true;
3435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
3445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool URLFetcherCore::GetResponseAsFilePath(bool take_ownership,
3462a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)                                           base::FilePath* out_response_path) {
3475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(delegate_task_runner_->BelongsToCurrentThread());
3484e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)
3494e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)  URLFetcherFileWriter* file_writer =
3504e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)      response_writer_ ? response_writer_->AsFileWriter() : NULL;
3514e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)  if (!file_writer)
3525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return false;
3535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3544e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)  *out_response_path = file_writer->file_path();
3555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (take_ownership) {
3577dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch    // Intentionally calling a file_writer_ method directly without posting
3587dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch    // the task to network_task_runner_.
359eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch    //
3607dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch    // This is for correctly handling the case when file_writer_->DisownFile()
3617dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch    // is soon followed by URLFetcherCore::Stop(). We have to make sure that
3627dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch    // DisownFile takes effect before Stop deletes file_writer_.
3637dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch    //
3647dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch    // This direct call should be thread-safe, since DisownFile itself does no
3657dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch    // file operation. It just flips the state to be referred in destruction.
3664e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)    file_writer->DisownFile();
3675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
3685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return true;
3695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
3705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::OnReceivedRedirect(URLRequest* request,
3725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                                        const GURL& new_url,
3735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                                        bool* defer_redirect) {
3745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK_EQ(request, request_.get());
3755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(network_task_runner_->BelongsToCurrentThread());
3765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (stop_on_redirect_) {
3775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    stopped_on_redirect_ = true;
3785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    url_ = new_url;
3795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    response_code_ = request_->GetResponseCode();
3805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    was_fetched_via_proxy_ = request_->was_fetched_via_proxy();
3815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    request->Cancel();
3825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    OnReadCompleted(request, 0);
3835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
3845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
3855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::OnResponseStarted(URLRequest* request) {
3875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK_EQ(request, request_.get());
3885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(network_task_runner_->BelongsToCurrentThread());
3895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (request_->status().is_success()) {
3905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    response_code_ = request_->GetResponseCode();
3915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    response_headers_ = request_->response_headers();
3925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    socket_address_ = request_->GetSocketAddress();
3935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    was_fetched_via_proxy_ = request_->was_fetched_via_proxy();
3945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    total_response_bytes_ = request_->GetExpectedContentSize();
3955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
3965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  ReadResponse();
3985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
3995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4002a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void URLFetcherCore::OnCertificateRequested(
4012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    URLRequest* request,
4022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    SSLCertRequestInfo* cert_request_info) {
4032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  DCHECK_EQ(request, request_.get());
4042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  DCHECK(network_task_runner_->BelongsToCurrentThread());
4052a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
4062a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  if (g_ignore_certificate_requests) {
4072a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    request->ContinueWithCertificate(NULL);
4082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  } else {
4092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    request->Cancel();
4102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  }
4112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)}
4122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
4135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::OnReadCompleted(URLRequest* request,
4145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                                     int bytes_read) {
4155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(request == request_);
4165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(network_task_runner_->BelongsToCurrentThread());
4175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (!stopped_on_redirect_)
4195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    url_ = request->url();
4205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  URLRequestThrottlerManager* throttler_manager =
4215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      request->context()->throttler_manager();
4225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (throttler_manager) {
4235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    url_throttler_entry_ = throttler_manager->RegisterRequestUrl(url_);
4245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
4255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  do {
4275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (!request_->status().is_success() || bytes_read <= 0)
4285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      break;
4295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    current_response_bytes_ += bytes_read;
4315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    InformDelegateDownloadProgress();
4325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
433868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)    const int result =
434868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)        WriteBuffer(new DrainableIOBuffer(buffer_.get(), bytes_read));
4352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    if (result < 0) {
4362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      // Write failed or waiting for write completion.
4374e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)      return;
4385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
439868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)  } while (request_->Read(buffer_.get(), kBufferSize, &bytes_read));
4405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  const URLRequestStatus status = request_->status();
4425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (status.is_success())
4445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    request_->GetResponseCookies(&cookies_);
4455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // See comments re: HEAD requests in ReadResponse().
4474e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)  if (!status.is_io_pending() || request_type_ == URLFetcher::HEAD) {
4485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    status_ = status;
4495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    ReleaseRequest();
4505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    // No more data to write.
4522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    const int result = response_writer_->Finish(
4532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        base::Bind(&URLFetcherCore::DidFinishWriting, this));
4542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    if (result != ERR_IO_PENDING)
4552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      DidFinishWriting(result);
4565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
4575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
4585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::CancelAll() {
4605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  g_registry.Get().CancelAll();
4615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
4625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)int URLFetcherCore::GetNumFetcherCores() {
4645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return g_registry.Get().size();
4655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
4665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::SetEnableInterceptionForTests(bool enabled) {
4685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  g_interception_enabled = enabled;
4695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
4705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void URLFetcherCore::SetIgnoreCertificateRequests(bool ignored) {
4722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  g_ignore_certificate_requests = ignored;
4732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)}
4742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
4755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)URLFetcherCore::~URLFetcherCore() {
4765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // |request_| should be NULL.  If not, it's unsafe to delete it here since we
4775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // may not be on the IO thread.
4785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(!request_.get());
4795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
4805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::StartOnIOThread() {
4825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(network_task_runner_->BelongsToCurrentThread());
4835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4844e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)  if (!response_writer_)
4854e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)    response_writer_.reset(new URLFetcherStringWriter);
4865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  const int result = response_writer_->Initialize(
4882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      base::Bind(&URLFetcherCore::DidInitializeWriter, this));
4892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  if (result != ERR_IO_PENDING)
4902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    DidInitializeWriter(result);
4915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
4925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::StartURLRequest() {
4945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(network_task_runner_->BelongsToCurrentThread());
4955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (was_cancelled_) {
4975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // Since StartURLRequest() is posted as a *delayed* task, it may
4985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // run after the URLFetcher was already stopped.
4995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return;
5005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
5015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
502868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)  DCHECK(request_context_getter_.get());
5035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(!request_.get());
5045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  g_registry.Get().AddURLFetcherCore(this);
5065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  current_response_bytes_ = 0;
5070f1bc08d4cfcc34181b0b5cbf065c40f687bf740Torne (Richard Coles)  request_ = request_context_getter_->GetURLRequestContext()->CreateRequest(
5080f1bc08d4cfcc34181b0b5cbf065c40f687bf740Torne (Richard Coles)      original_url_, DEFAULT_PRIORITY, this);
5095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  request_->set_stack_trace(stack_trace_);
5105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int flags = request_->load_flags() | load_flags_;
5115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (!g_interception_enabled)
5125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    flags = flags | LOAD_DISABLE_INTERCEPT;
5135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (is_chunked_upload_)
5155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    request_->EnableChunkedUpload();
516f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles)  request_->SetLoadFlags(flags);
517c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)  request_->SetReferrer(referrer_);
5185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  request_->set_first_party_for_cookies(first_party_for_cookies_.is_empty() ?
5195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      original_url_ : first_party_for_cookies_);
5205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (url_request_data_key_ && !url_request_create_data_callback_.is_null()) {
5215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    request_->SetUserData(url_request_data_key_,
5225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                          url_request_create_data_callback_.Run());
5235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
5245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  switch (request_type_) {
5265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    case URLFetcher::GET:
5275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      break;
5285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    case URLFetcher::POST:
5305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    case URLFetcher::PUT:
5312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    case URLFetcher::PATCH:
5322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      // Upload content must be set.
5332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      DCHECK(is_chunked_upload_ || upload_content_set_);
5345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      request_->set_method(
5362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)          request_type_ == URLFetcher::POST ? "POST" :
5372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)          request_type_ == URLFetcher::PUT ? "PUT" : "PATCH");
5382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      if (!upload_content_type_.empty()) {
5392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        extra_request_headers_.SetHeader(HttpRequestHeaders::kContentType,
5402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)                                         upload_content_type_);
5412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      }
5425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      if (!upload_content_.empty()) {
5432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        scoped_ptr<UploadElementReader> reader(new UploadBytesElementReader(
5442a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)            upload_content_.data(), upload_content_.size()));
5452a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        request_->set_upload(make_scoped_ptr(
5462a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)            UploadDataStream::CreateWithReader(reader.Pass(), 0)));
5472a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      } else if (!upload_file_path_.empty()) {
548868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)        scoped_ptr<UploadElementReader> reader(
549868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)            new UploadFileElementReader(upload_file_task_runner_.get(),
550868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)                                        upload_file_path_,
551868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)                                        upload_range_offset_,
552868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)                                        upload_range_length_,
553868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)                                        base::Time()));
5542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        request_->set_upload(make_scoped_ptr(
5552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)            UploadDataStream::CreateWithReader(reader.Pass(), 0)));
5565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      }
5575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      current_upload_bytes_ = -1;
5595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      // TODO(kinaba): http://crbug.com/118103. Implement upload callback in the
5605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      //  layer and avoid using timer here.
5615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      upload_progress_checker_timer_.reset(
5625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          new base::RepeatingTimer<URLFetcherCore>());
5635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      upload_progress_checker_timer_->Start(
5645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          FROM_HERE,
5655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          base::TimeDelta::FromMilliseconds(kUploadProgressTimerInterval),
5665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          this,
5675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          &URLFetcherCore::InformDelegateUploadProgress);
5685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      break;
5695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    case URLFetcher::HEAD:
5715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      request_->set_method("HEAD");
5725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      break;
5735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    case URLFetcher::DELETE_REQUEST:
5755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      request_->set_method("DELETE");
5765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      break;
5775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    default:
5795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      NOTREACHED();
5805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
5815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (!extra_request_headers_.IsEmpty())
5835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    request_->SetExtraRequestHeaders(extra_request_headers_);
5845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  request_->Start();
5865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
5875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void URLFetcherCore::DidInitializeWriter(int result) {
5892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  if (result != OK) {
5904e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)    CancelURLRequest(result);
5912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    delegate_task_runner_->PostTask(
5922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        FROM_HERE,
5932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        base::Bind(&URLFetcherCore::InformDelegateFetchIsComplete, this));
5942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    return;
5952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  }
5962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  StartURLRequestWhenAppropriate();
5972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)}
5982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
5995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::StartURLRequestWhenAppropriate() {
6005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(network_task_runner_->BelongsToCurrentThread());
6015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (was_cancelled_)
6035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return;
6045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
605868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)  DCHECK(request_context_getter_.get());
6065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int64 delay = 0LL;
608868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)  if (original_url_throttler_entry_.get() == NULL) {
6095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    URLRequestThrottlerManager* manager =
6105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        request_context_getter_->GetURLRequestContext()->throttler_manager();
6115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (manager) {
6125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      original_url_throttler_entry_ =
6135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          manager->RegisterRequestUrl(original_url_);
6145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
6155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
616868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)  if (original_url_throttler_entry_.get() != NULL) {
6175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    delay = original_url_throttler_entry_->ReserveSendingTimeForNextRequest(
6185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        GetBackoffReleaseTime());
6195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
6205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (delay == 0) {
6225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    StartURLRequest();
6235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  } else {
6245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    base::ThreadTaskRunnerHandle::Get()->PostDelayedTask(
6255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        FROM_HERE, base::Bind(&URLFetcherCore::StartURLRequest, this),
6265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        base::TimeDelta::FromMilliseconds(delay));
6275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
6285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
6295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6304e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)void URLFetcherCore::CancelURLRequest(int error) {
6315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(network_task_runner_->BelongsToCurrentThread());
6325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (request_.get()) {
6344e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)    request_->CancelWithError(error);
6355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    ReleaseRequest();
6365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
6374e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)
6384e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)  // Set the error manually.
6394e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)  // Normally, calling URLRequest::CancelWithError() results in calling
6404e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)  // OnReadCompleted() with bytes_read = -1 via an asynchronous task posted by
6414e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)  // URLRequestJob::NotifyDone(). But, because the request was released
6424e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)  // immediately after being canceled, the request could not call
6434e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)  // OnReadCompleted() which overwrites |status_| with the error status.
6444e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)  status_.set_status(URLRequestStatus::CANCELED);
6454e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)  status_.set_error(error);
6464e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)
6475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Release the reference to the request context. There could be multiple
6485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // references to URLFetcher::Core at this point so it may take a while to
6495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // delete the object, but we cannot delay the destruction of the request
6505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // context.
6515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  request_context_getter_ = NULL;
6525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  first_party_for_cookies_ = GURL();
6535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  url_request_data_key_ = NULL;
6545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  url_request_create_data_callback_.Reset();
6555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  was_cancelled_ = true;
6565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
6575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::OnCompletedURLRequest(
6595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    base::TimeDelta backoff_delay) {
6605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(delegate_task_runner_->BelongsToCurrentThread());
6615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Save the status and backoff_delay so that delegates can read it.
6635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (delegate_) {
6645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    backoff_delay_ = backoff_delay;
6655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    InformDelegateFetchIsComplete();
6665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
6675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
6685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::InformDelegateFetchIsComplete() {
6705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(delegate_task_runner_->BelongsToCurrentThread());
6715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (delegate_)
6725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    delegate_->OnURLFetchComplete(fetcher_);
6735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
6745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::NotifyMalformedContent() {
6765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(network_task_runner_->BelongsToCurrentThread());
677868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)  if (url_throttler_entry_.get() != NULL) {
6785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    int status_code = response_code_;
6795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (status_code == URLFetcher::RESPONSE_CODE_INVALID) {
6805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      // The status code will generally be known by the time clients
6815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      // call the |ReceivedContentWasMalformed()| function (which ends up
6825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      // calling the current function) but if it's not, we need to assume
6835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      // the response was successful so that the total failure count
6845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      // used to calculate exponential back-off goes up.
6855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      status_code = 200;
6865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
6875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    url_throttler_entry_->ReceivedContentWasMalformed(status_code);
6885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
6895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
6905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void URLFetcherCore::DidFinishWriting(int result) {
6922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  if (result != OK) {
6934e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)    CancelURLRequest(result);
6942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    delegate_task_runner_->PostTask(
6952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        FROM_HERE,
6962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        base::Bind(&URLFetcherCore::InformDelegateFetchIsComplete, this));
6972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    return;
6982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  }
6992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  // If the file was successfully closed, then the URL request is complete.
7002a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  RetryOrCompleteUrlFetch();
7012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)}
7022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
7035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::RetryOrCompleteUrlFetch() {
7045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(network_task_runner_->BelongsToCurrentThread());
7055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  base::TimeDelta backoff_delay;
7065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
7075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Checks the response from server.
7085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (response_code_ >= 500 ||
7095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      status_.error() == ERR_TEMPORARILY_THROTTLED) {
7105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // When encountering a server error, we will send the request again
7115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // after backoff time.
7122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    ++num_retries_on_5xx_;
7135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
7145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // Note that backoff_delay may be 0 because (a) the
7155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // URLRequestThrottlerManager and related code does not
7165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // necessarily back off on the first error, (b) it only backs off
7175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // on some of the 5xx status codes, (c) not all URLRequestContexts
7185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // have a throttler manager.
7195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    base::TimeTicks backoff_release_time = GetBackoffReleaseTime();
7205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    backoff_delay = backoff_release_time - base::TimeTicks::Now();
7215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (backoff_delay < base::TimeDelta())
7225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      backoff_delay = base::TimeDelta();
7235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
7242a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    if (automatically_retry_on_5xx_ &&
7252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        num_retries_on_5xx_ <= max_retries_on_5xx_) {
7265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      StartOnIOThread();
7275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      return;
7285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
7295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  } else {
7305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    backoff_delay = base::TimeDelta();
7315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
7322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
7332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  // Retry if the request failed due to network changes.
7342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  if (status_.error() == ERR_NETWORK_CHANGED &&
7352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      num_retries_on_network_changes_ < max_retries_on_network_changes_) {
7362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    ++num_retries_on_network_changes_;
7372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
7382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    // Retry soon, after flushing all the current tasks which may include
7392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    // further network change observers.
7402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    network_task_runner_->PostTask(
7412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        FROM_HERE, base::Bind(&URLFetcherCore::StartOnIOThread, this));
7422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    return;
7432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  }
7442a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
7455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  request_context_getter_ = NULL;
7465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  first_party_for_cookies_ = GURL();
7475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  url_request_data_key_ = NULL;
7485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  url_request_create_data_callback_.Reset();
7495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  bool posted = delegate_task_runner_->PostTask(
7505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      FROM_HERE,
7515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      base::Bind(&URLFetcherCore::OnCompletedURLRequest, this, backoff_delay));
7525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
7535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // If the delegate message loop does not exist any more, then the delegate
7545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // should be gone too.
7555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(posted || !delegate_);
7565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
7575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
7585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::ReleaseRequest() {
7595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  upload_progress_checker_timer_.reset();
7605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  request_.reset();
7615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  g_registry.Get().RemoveURLFetcherCore(this);
7625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
7635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
7645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)base::TimeTicks URLFetcherCore::GetBackoffReleaseTime() {
7655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(network_task_runner_->BelongsToCurrentThread());
7665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
767868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)  if (original_url_throttler_entry_.get()) {
7685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    base::TimeTicks original_url_backoff =
7695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        original_url_throttler_entry_->GetExponentialBackoffReleaseTime();
7705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    base::TimeTicks destination_url_backoff;
771868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)    if (url_throttler_entry_.get() != NULL &&
772868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)        original_url_throttler_entry_.get() != url_throttler_entry_.get()) {
7735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      destination_url_backoff =
7745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          url_throttler_entry_->GetExponentialBackoffReleaseTime();
7755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
7765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
7775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return original_url_backoff > destination_url_backoff ?
7785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        original_url_backoff : destination_url_backoff;
7795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  } else {
7805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return base::TimeTicks();
7815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
7825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
7835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
7845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::CompleteAddingUploadDataChunk(
7855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    const std::string& content, bool is_last_chunk) {
7865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (was_cancelled_) {
7875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // Since CompleteAddingUploadDataChunk() is posted as a *delayed* task, it
7885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // may run after the URLFetcher was already stopped.
7895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return;
7905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
7915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(is_chunked_upload_);
7925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(request_.get());
7935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(!content.empty());
7945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  request_->AppendChunkToUpload(content.data(),
7955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                                static_cast<int>(content.length()),
7965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                                is_last_chunk);
7975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
7985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
79990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)int URLFetcherCore::WriteBuffer(scoped_refptr<DrainableIOBuffer> data) {
80090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)  while (data->BytesRemaining() > 0) {
80190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)    const int result = response_writer_->Write(
802868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)        data.get(),
803868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)        data->BytesRemaining(),
80490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)        base::Bind(&URLFetcherCore::DidWriteBuffer, this, data));
8054e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)    if (result < 0) {
8061e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles)      if (result != ERR_IO_PENDING)
8071e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles)        DidWriteBuffer(data, result);
80890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)      return result;
8094e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)    }
81090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)    data->DidConsume(result);
81190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)  }
81290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)  return OK;
81390dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)}
81490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)
81590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)void URLFetcherCore::DidWriteBuffer(scoped_refptr<DrainableIOBuffer> data,
81690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)                                    int result) {
81790dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)  if (result < 0) {  // Handle errors.
8184e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)    CancelURLRequest(result);
8191e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles)    response_writer_->Finish(base::Bind(&EmptyCompletionCallback));
8202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    delegate_task_runner_->PostTask(
8212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        FROM_HERE,
8222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        base::Bind(&URLFetcherCore::InformDelegateFetchIsComplete, this));
8232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    return;
8245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
8254e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)
8264e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)  // Continue writing.
8274e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)  data->DidConsume(result);
8284e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)  if (WriteBuffer(data) < 0)
8294e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)    return;
8304e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)
831f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles)  // Finished writing buffer_. Read some more, unless the request has been
832f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles)  // cancelled and deleted.
83390dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)  DCHECK_EQ(0, data->BytesRemaining());
834f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles)  if (request_.get())
835f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles)    ReadResponse();
8365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
8375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
8385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::ReadResponse() {
8395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Some servers may treat HEAD requests as GET requests.  To free up the
8405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // network connection as soon as possible, signal that the request has
8415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // completed immediately, without trying to read any data back (all we care
8425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // about is the response code and headers, which we already have).
8435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int bytes_read = 0;
8445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (request_->status().is_success() &&
8455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      (request_type_ != URLFetcher::HEAD))
846868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)    request_->Read(buffer_.get(), kBufferSize, &bytes_read);
8475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  OnReadCompleted(request_.get(), bytes_read);
8485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
8495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
8505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::InformDelegateUploadProgress() {
8515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(network_task_runner_->BelongsToCurrentThread());
8525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (request_.get()) {
8535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    int64 current = request_->GetUploadProgress().position();
8545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (current_upload_bytes_ != current) {
8555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      current_upload_bytes_ = current;
8565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      int64 total = -1;
8572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      if (!is_chunked_upload_) {
8582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        total = static_cast<int64>(request_->GetUploadProgress().size());
8592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        // Total may be zero if the UploadDataStream::Init has not been called
8602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        // yet.  Don't send the upload progress until the size is initialized.
8612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        if (!total)
8622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)          return;
8632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      }
8645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      delegate_task_runner_->PostTask(
8655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          FROM_HERE,
8665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          base::Bind(
8675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)              &URLFetcherCore::InformDelegateUploadProgressInDelegateThread,
8685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)              this, current, total));
8695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
8705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
8715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
8725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
8735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::InformDelegateUploadProgressInDelegateThread(
8745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    int64 current, int64 total) {
8755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(delegate_task_runner_->BelongsToCurrentThread());
8765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (delegate_)
8775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    delegate_->OnURLFetchUploadProgress(fetcher_, current, total);
8785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
8795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
8805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::InformDelegateDownloadProgress() {
8815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(network_task_runner_->BelongsToCurrentThread());
8825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  delegate_task_runner_->PostTask(
8835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      FROM_HERE,
8845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      base::Bind(
8855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          &URLFetcherCore::InformDelegateDownloadProgressInDelegateThread,
8865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          this, current_response_bytes_, total_response_bytes_));
8875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
8885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
8895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::InformDelegateDownloadProgressInDelegateThread(
8905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    int64 current, int64 total) {
8915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(delegate_task_runner_->BelongsToCurrentThread());
8925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (delegate_)
8935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    delegate_->OnURLFetchDownloadProgress(fetcher_, current, total);
8945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
8955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
8965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}  // namespace net
897