url_fetcher_core.cc revision 2a99a7e74a7f215066514fe81d2bfa6639d9eddd
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"
102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/single_thread_task_runner.h"
115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/stl_util.h"
125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/thread_task_runner_handle.h"
135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/tracked_objects.h"
145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/base/io_buffer.h"
155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/base/load_flags.h"
165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/base/net_errors.h"
172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "net/base/upload_bytes_element_reader.h"
182a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "net/base/upload_data_stream.h"
192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "net/base/upload_file_element_reader.h"
205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/http/http_response_headers.h"
215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/url_request/url_fetcher_delegate.h"
222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "net/url_request/url_fetcher_response_writer.h"
235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/url_request/url_request_context.h"
245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/url_request/url_request_context_getter.h"
255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/url_request/url_request_throttler_manager.h"
265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace {
285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const int kBufferSize = 4096;
305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const int kUploadProgressTimerInterval = 100;
315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool g_interception_enabled = false;
322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)bool g_ignore_certificate_requests = false;
335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}  // namespace
355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace net {
375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// URLFetcherCore::Registry ---------------------------------------------------
395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)URLFetcherCore::Registry::Registry() {}
415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)URLFetcherCore::Registry::~Registry() {}
425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::Registry::AddURLFetcherCore(URLFetcherCore* core) {
445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(!ContainsKey(fetchers_, core));
455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  fetchers_.insert(core);
465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::Registry::RemoveURLFetcherCore(URLFetcherCore* core) {
495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(ContainsKey(fetchers_, core));
505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  fetchers_.erase(core);
515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::Registry::CancelAll() {
545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  while (!fetchers_.empty())
555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    (*fetchers_.begin())->CancelURLRequest();
565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// URLFetcherCore -------------------------------------------------------------
595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// static
615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)base::LazyInstance<URLFetcherCore::Registry>
625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    URLFetcherCore::g_registry = LAZY_INSTANCE_INITIALIZER;
635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)URLFetcherCore::URLFetcherCore(URLFetcher* fetcher,
655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                               const GURL& original_url,
665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                               URLFetcher::RequestType request_type,
675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                               URLFetcherDelegate* d)
685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    : fetcher_(fetcher),
695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      original_url_(original_url),
705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      request_type_(request_type),
715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      delegate_(d),
725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      delegate_task_runner_(
735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          base::ThreadTaskRunnerHandle::Get()),
745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      request_(NULL),
755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      load_flags_(LOAD_NORMAL),
765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      response_code_(URLFetcher::RESPONSE_CODE_INVALID),
775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      buffer_(new IOBuffer(kBufferSize)),
785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      url_request_data_key_(NULL),
795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      was_fetched_via_proxy_(false),
802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      upload_content_set_(false),
815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      is_chunked_upload_(false),
825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      was_cancelled_(false),
832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      file_writer_(NULL),
845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      response_destination_(STRING),
855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      stop_on_redirect_(false),
865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      stopped_on_redirect_(false),
875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      automatically_retry_on_5xx_(true),
882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      num_retries_on_5xx_(0),
892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      max_retries_on_5xx_(0),
902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      num_retries_on_network_changes_(0),
912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      max_retries_on_network_changes_(0),
925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      current_upload_bytes_(-1),
935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      current_response_bytes_(0),
945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      total_response_bytes_(-1) {
955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  CHECK(original_url_.is_valid());
965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::Start() {
995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(delegate_task_runner_);
1005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(request_context_getter_) << "We need an URLRequestContext!";
1015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (network_task_runner_) {
1025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    DCHECK_EQ(network_task_runner_,
1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)              request_context_getter_->GetNetworkTaskRunner());
1045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  } else {
1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    network_task_runner_ = request_context_getter_->GetNetworkTaskRunner();
1065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
1075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(network_task_runner_.get()) << "We need an IO task runner";
1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  network_task_runner_->PostTask(
1105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      FROM_HERE, base::Bind(&URLFetcherCore::StartOnIOThread, this));
1115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
1125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::Stop() {
1145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (delegate_task_runner_)  // May be NULL in tests.
1155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    DCHECK(delegate_task_runner_->BelongsToCurrentThread());
1165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  delegate_ = NULL;
1185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  fetcher_ = NULL;
1195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (!network_task_runner_.get())
1205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return;
1215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (network_task_runner_->RunsTasksOnCurrentThread()) {
1225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    CancelURLRequest();
1235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  } else {
1245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    network_task_runner_->PostTask(
1255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        FROM_HERE, base::Bind(&URLFetcherCore::CancelURLRequest, this));
1265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
1275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
1285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::SetUploadData(const std::string& upload_content_type,
1305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                                   const std::string& upload_content) {
1315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(!is_chunked_upload_);
1322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  DCHECK(!upload_content_set_);
1332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  DCHECK(upload_content_.empty());
1342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  DCHECK(upload_file_path_.empty());
1352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  DCHECK(upload_content_type_.empty());
1362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
1372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  // Empty |upload_content_type| is allowed iff the |upload_content| is empty.
1382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  DCHECK(upload_content.empty() || !upload_content_type.empty());
1392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
1405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  upload_content_type_ = upload_content_type;
1415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  upload_content_ = upload_content;
1422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  upload_content_set_ = true;
1432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)}
1442a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
1452a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void URLFetcherCore::SetUploadFilePath(
1462a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    const std::string& upload_content_type,
1472a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    const base::FilePath& file_path,
1482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    scoped_refptr<base::TaskRunner> file_task_runner) {
1492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  DCHECK(!is_chunked_upload_);
1502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  DCHECK(!upload_content_set_);
1512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  DCHECK(upload_content_.empty());
1522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  DCHECK(upload_file_path_.empty());
1532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  DCHECK(upload_content_type_.empty());
1542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  DCHECK(!upload_content_type.empty());
1552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
1562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  upload_content_type_ = upload_content_type;
1572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  upload_file_path_ = file_path;
1582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  upload_file_task_runner_ = file_task_runner;
1592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  upload_content_set_ = true;
1605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
1615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::SetChunkedUpload(const std::string& content_type) {
1635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(is_chunked_upload_ ||
1645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)         (upload_content_type_.empty() &&
1655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          upload_content_.empty()));
1662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
1672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  // Empty |content_type| is not allowed here, because it is impossible
1682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  // to ensure non-empty upload content as it is not yet supplied.
1692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  DCHECK(!content_type.empty());
1702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
1715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  upload_content_type_ = content_type;
1725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  upload_content_.clear();
1735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  is_chunked_upload_ = true;
1745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
1755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::AppendChunkToUpload(const std::string& content,
1775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                                         bool is_last_chunk) {
1785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(delegate_task_runner_);
1795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(network_task_runner_.get());
1805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  network_task_runner_->PostTask(
1815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      FROM_HERE,
1825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      base::Bind(&URLFetcherCore::CompleteAddingUploadDataChunk, this, content,
1835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                 is_last_chunk));
1845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
1855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::SetLoadFlags(int load_flags) {
1875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  load_flags_ = load_flags;
1885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
1895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)int URLFetcherCore::GetLoadFlags() const {
1915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return load_flags_;
1925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
1935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::SetReferrer(const std::string& referrer) {
1955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  referrer_ = referrer;
1965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
1975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::SetExtraRequestHeaders(
1995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    const std::string& extra_request_headers) {
2005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  extra_request_headers_.Clear();
2015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  extra_request_headers_.AddHeadersFromString(extra_request_headers);
2025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::AddExtraRequestHeader(const std::string& header_line) {
2055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  extra_request_headers_.AddHeaderFromString(header_line);
2065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::GetExtraRequestHeaders(
2095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    HttpRequestHeaders* headers) const {
2105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  headers->CopyFrom(extra_request_headers_);
2115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::SetRequestContext(
2145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    URLRequestContextGetter* request_context_getter) {
2155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(!request_context_getter_);
2165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(request_context_getter);
2175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  request_context_getter_ = request_context_getter;
2185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::SetFirstPartyForCookies(
2215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    const GURL& first_party_for_cookies) {
2225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(first_party_for_cookies_.is_empty());
2235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  first_party_for_cookies_ = first_party_for_cookies;
2245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::SetURLRequestUserData(
2275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    const void* key,
2285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    const URLFetcher::CreateDataCallback& create_data_callback) {
2295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(key);
2305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(!create_data_callback.is_null());
2315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  url_request_data_key_ = key;
2325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  url_request_create_data_callback_ = create_data_callback;
2335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::SetStopOnRedirect(bool stop_on_redirect) {
2365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  stop_on_redirect_ = stop_on_redirect;
2375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::SetAutomaticallyRetryOn5xx(bool retry) {
2405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  automatically_retry_on_5xx_ = retry;
2415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void URLFetcherCore::SetMaxRetriesOn5xx(int max_retries) {
2442a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  max_retries_on_5xx_ = max_retries;
2455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2472a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)int URLFetcherCore::GetMaxRetriesOn5xx() const {
2482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  return max_retries_on_5xx_;
2495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)base::TimeDelta URLFetcherCore::GetBackoffDelay() const {
2525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return backoff_delay_;
2535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void URLFetcherCore::SetAutomaticallyRetryOnNetworkChanges(int max_retries) {
2562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  max_retries_on_network_changes_ = max_retries;
2572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)}
2582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
2595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::SaveResponseToFileAtPath(
2602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    const base::FilePath& file_path,
2615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    scoped_refptr<base::TaskRunner> file_task_runner) {
2625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(delegate_task_runner_->BelongsToCurrentThread());
2635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  file_task_runner_ = file_task_runner;
2645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  response_destination_ = URLFetcherCore::PERMANENT_FILE;
2655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  response_destination_file_path_ = file_path;
2665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::SaveResponseToTemporaryFile(
2695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    scoped_refptr<base::TaskRunner> file_task_runner) {
2705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(delegate_task_runner_->BelongsToCurrentThread());
2715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  file_task_runner_ = file_task_runner;
2725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  response_destination_ = URLFetcherCore::TEMP_FILE;
2735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)HttpResponseHeaders* URLFetcherCore::GetResponseHeaders() const {
2765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return response_headers_;
2775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// TODO(panayiotis): socket_address_ is written in the IO thread,
2805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// if this is accessed in the UI thread, this could result in a race.
2815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Same for response_headers_ above and was_fetched_via_proxy_ below.
2825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)HostPortPair URLFetcherCore::GetSocketAddress() const {
2835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return socket_address_;
2845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool URLFetcherCore::WasFetchedViaProxy() const {
2875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return was_fetched_via_proxy_;
2885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const GURL& URLFetcherCore::GetOriginalURL() const {
2915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return original_url_;
2925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const GURL& URLFetcherCore::GetURL() const {
2955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return url_;
2965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const URLRequestStatus& URLFetcherCore::GetStatus() const {
2995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return status_;
3005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
3015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)int URLFetcherCore::GetResponseCode() const {
3035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return response_code_;
3045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
3055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const ResponseCookies& URLFetcherCore::GetCookies() const {
3075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return cookies_;
3085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
3095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)bool URLFetcherCore::FileErrorOccurred(int* out_error_code) const {
3115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Can't have a file error if no file is being created or written to.
3122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  if (!file_writer_)
3135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return false;
3145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  int error_code = file_writer_->error_code();
3162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  if (error_code == OK)
3175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return false;
3185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  *out_error_code = error_code;
3205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return true;
3215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
3225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::ReceivedContentWasMalformed() {
3245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(delegate_task_runner_->BelongsToCurrentThread());
3255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (network_task_runner_.get()) {
3265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    network_task_runner_->PostTask(
3275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        FROM_HERE, base::Bind(&URLFetcherCore::NotifyMalformedContent, this));
3285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
3295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
3305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool URLFetcherCore::GetResponseAsString(
3325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    std::string* out_response_string) const {
3335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (response_destination_ != URLFetcherCore::STRING)
3345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return false;
3355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  *out_response_string = data_;
3375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  UMA_HISTOGRAM_MEMORY_KB("UrlFetcher.StringResponseSize",
3385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                          (data_.length() / 1024));
3395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return true;
3415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
3425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool URLFetcherCore::GetResponseAsFilePath(bool take_ownership,
3442a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)                                           base::FilePath* out_response_path) {
3455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(delegate_task_runner_->BelongsToCurrentThread());
3465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  const bool destination_is_file =
3475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      response_destination_ == URLFetcherCore::TEMP_FILE ||
3485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      response_destination_ == URLFetcherCore::PERMANENT_FILE;
3492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  if (!destination_is_file || !file_writer_)
3505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return false;
3515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  *out_response_path = file_writer_->file_path();
3535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (take_ownership) {
3555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    network_task_runner_->PostTask(
3565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        FROM_HERE,
3575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        base::Bind(&URLFetcherCore::DisownFile, this));
3585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
3595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return true;
3605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
3615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::OnReceivedRedirect(URLRequest* request,
3635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                                        const GURL& new_url,
3645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                                        bool* defer_redirect) {
3655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK_EQ(request, request_.get());
3665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(network_task_runner_->BelongsToCurrentThread());
3675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (stop_on_redirect_) {
3685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    stopped_on_redirect_ = true;
3695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    url_ = new_url;
3705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    response_code_ = request_->GetResponseCode();
3715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    was_fetched_via_proxy_ = request_->was_fetched_via_proxy();
3725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    request->Cancel();
3735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    OnReadCompleted(request, 0);
3745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
3755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
3765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::OnResponseStarted(URLRequest* request) {
3785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK_EQ(request, request_.get());
3795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(network_task_runner_->BelongsToCurrentThread());
3805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (request_->status().is_success()) {
3815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    response_code_ = request_->GetResponseCode();
3825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    response_headers_ = request_->response_headers();
3835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    socket_address_ = request_->GetSocketAddress();
3845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    was_fetched_via_proxy_ = request_->was_fetched_via_proxy();
3855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    total_response_bytes_ = request_->GetExpectedContentSize();
3865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
3875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  ReadResponse();
3895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
3905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void URLFetcherCore::OnCertificateRequested(
3922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    URLRequest* request,
3932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    SSLCertRequestInfo* cert_request_info) {
3942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  DCHECK_EQ(request, request_.get());
3952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  DCHECK(network_task_runner_->BelongsToCurrentThread());
3962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
3972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  if (g_ignore_certificate_requests) {
3982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    request->ContinueWithCertificate(NULL);
3992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  } else {
4002a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    request->Cancel();
4012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  }
4022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)}
4032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
4045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::OnReadCompleted(URLRequest* request,
4055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                                     int bytes_read) {
4065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(request == request_);
4075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(network_task_runner_->BelongsToCurrentThread());
4085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (!stopped_on_redirect_)
4105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    url_ = request->url();
4115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  URLRequestThrottlerManager* throttler_manager =
4125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      request->context()->throttler_manager();
4135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (throttler_manager) {
4145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    url_throttler_entry_ = throttler_manager->RegisterRequestUrl(url_);
4155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
4165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  bool waiting_on_write = false;
4185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  do {
4195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (!request_->status().is_success() || bytes_read <= 0)
4205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      break;
4215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    current_response_bytes_ += bytes_read;
4235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    InformDelegateDownloadProgress();
4245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    InformDelegateDownloadDataIfNecessary(bytes_read);
4255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    const int result = response_writer_->Write(
4272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        buffer_, bytes_read,
4282a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        base::Bind(&URLFetcherCore::DidWriteBuffer, this));
4292a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    if (result < 0) {
4302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      // Write failed or waiting for write completion.
4312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      if (result == ERR_IO_PENDING)
4322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        waiting_on_write = true;
4335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      break;
4345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
4355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  } while (request_->Read(buffer_, kBufferSize, &bytes_read));
4365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  const URLRequestStatus status = request_->status();
4385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (status.is_success())
4405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    request_->GetResponseCookies(&cookies_);
4415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // See comments re: HEAD requests in ReadResponse().
4435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if ((!status.is_io_pending() && !waiting_on_write) ||
4445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      (request_type_ == URLFetcher::HEAD)) {
4455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    status_ = status;
4465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    ReleaseRequest();
4475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    // No more data to write.
4492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    const int result = response_writer_->Finish(
4502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        base::Bind(&URLFetcherCore::DidFinishWriting, this));
4512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    if (result != ERR_IO_PENDING)
4522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      DidFinishWriting(result);
4535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
4545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
4555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::CancelAll() {
4575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  g_registry.Get().CancelAll();
4585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
4595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)int URLFetcherCore::GetNumFetcherCores() {
4615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return g_registry.Get().size();
4625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
4635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::SetEnableInterceptionForTests(bool enabled) {
4655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  g_interception_enabled = enabled;
4665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
4675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void URLFetcherCore::SetIgnoreCertificateRequests(bool ignored) {
4692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  g_ignore_certificate_requests = ignored;
4702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)}
4712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
4725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)URLFetcherCore::~URLFetcherCore() {
4735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // |request_| should be NULL.  If not, it's unsafe to delete it here since we
4745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // may not be on the IO thread.
4755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(!request_.get());
4765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
4775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::StartOnIOThread() {
4795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(network_task_runner_->BelongsToCurrentThread());
4805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  switch (response_destination_) {
4825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    case STRING:
4832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      response_writer_.reset(new URLFetcherStringWriter(&data_));
4845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      break;
4855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    case PERMANENT_FILE:
4875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    case TEMP_FILE:
4885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      DCHECK(file_task_runner_.get())
4895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          << "Need to set the file task runner.";
4905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      file_writer_ = new URLFetcherFileWriter(file_task_runner_);
4925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      // If the file is successfully created,
4945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      // URLFetcherCore::StartURLRequestWhenAppropriate() will be called.
4952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      if (response_destination_ == PERMANENT_FILE) {
4962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        file_writer_->set_destination_file_path(
4972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)            response_destination_file_path_);
4985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      }
4992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      response_writer_.reset(file_writer_);
5005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      break;
5015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    default:
5035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      NOTREACHED();
5045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
5052a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  DCHECK(response_writer_);
5062a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  const int result = response_writer_->Initialize(
5072a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      base::Bind(&URLFetcherCore::DidInitializeWriter, this));
5082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  if (result != ERR_IO_PENDING)
5092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    DidInitializeWriter(result);
5105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
5115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::StartURLRequest() {
5135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(network_task_runner_->BelongsToCurrentThread());
5145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (was_cancelled_) {
5165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // Since StartURLRequest() is posted as a *delayed* task, it may
5175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // run after the URLFetcher was already stopped.
5185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return;
5195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
5205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(request_context_getter_);
5225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(!request_.get());
5235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  g_registry.Get().AddURLFetcherCore(this);
5255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  current_response_bytes_ = 0;
5265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  request_.reset(request_context_getter_->GetURLRequestContext()->CreateRequest(
5275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      original_url_, this));
5285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  request_->set_stack_trace(stack_trace_);
5295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int flags = request_->load_flags() | load_flags_;
5305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (!g_interception_enabled)
5315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    flags = flags | LOAD_DISABLE_INTERCEPT;
5325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (is_chunked_upload_)
5345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    request_->EnableChunkedUpload();
5355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  request_->set_load_flags(flags);
5365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  request_->set_referrer(referrer_);
5375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  request_->set_first_party_for_cookies(first_party_for_cookies_.is_empty() ?
5385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      original_url_ : first_party_for_cookies_);
5395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (url_request_data_key_ && !url_request_create_data_callback_.is_null()) {
5405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    request_->SetUserData(url_request_data_key_,
5415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                          url_request_create_data_callback_.Run());
5425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
5435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  switch (request_type_) {
5455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    case URLFetcher::GET:
5465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      break;
5475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    case URLFetcher::POST:
5495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    case URLFetcher::PUT:
5502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    case URLFetcher::PATCH:
5512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      // Upload content must be set.
5522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      DCHECK(is_chunked_upload_ || upload_content_set_);
5535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      request_->set_method(
5552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)          request_type_ == URLFetcher::POST ? "POST" :
5562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)          request_type_ == URLFetcher::PUT ? "PUT" : "PATCH");
5575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      extra_request_headers_.SetHeader(HttpRequestHeaders::kContentType,
5585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                                       upload_content_type_);
5592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      if (!upload_content_type_.empty()) {
5602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        extra_request_headers_.SetHeader(HttpRequestHeaders::kContentType,
5612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)                                         upload_content_type_);
5622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      }
5635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      if (!upload_content_.empty()) {
5642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        scoped_ptr<UploadElementReader> reader(new UploadBytesElementReader(
5652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)            upload_content_.data(), upload_content_.size()));
5662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        request_->set_upload(make_scoped_ptr(
5672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)            UploadDataStream::CreateWithReader(reader.Pass(), 0)));
5682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      } else if (!upload_file_path_.empty()) {
5692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        scoped_ptr<UploadElementReader> reader(new UploadFileElementReader(
5702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)            upload_file_task_runner_,
5712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)            upload_file_path_,
5722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)            0, kuint64max, base::Time()));
5732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        request_->set_upload(make_scoped_ptr(
5742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)            UploadDataStream::CreateWithReader(reader.Pass(), 0)));
5755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      }
5765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      current_upload_bytes_ = -1;
5785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      // TODO(kinaba): http://crbug.com/118103. Implement upload callback in the
5795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      //  layer and avoid using timer here.
5805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      upload_progress_checker_timer_.reset(
5815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          new base::RepeatingTimer<URLFetcherCore>());
5825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      upload_progress_checker_timer_->Start(
5835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          FROM_HERE,
5845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          base::TimeDelta::FromMilliseconds(kUploadProgressTimerInterval),
5855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          this,
5865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          &URLFetcherCore::InformDelegateUploadProgress);
5875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      break;
5885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    case URLFetcher::HEAD:
5905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      request_->set_method("HEAD");
5915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      break;
5925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    case URLFetcher::DELETE_REQUEST:
5945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      request_->set_method("DELETE");
5955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      break;
5965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    default:
5985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      NOTREACHED();
5995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
6005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (!extra_request_headers_.IsEmpty())
6025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    request_->SetExtraRequestHeaders(extra_request_headers_);
6035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // There might be data left over from a previous request attempt.
6055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  data_.clear();
6065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // If we are writing the response to a file, the only caller
6085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // of this function should have created it and not written yet.
6092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  DCHECK(!file_writer_ || file_writer_->total_bytes_written() == 0);
6105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  request_->Start();
6125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
6135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void URLFetcherCore::DidInitializeWriter(int result) {
6152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  if (result != OK) {
6162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    delegate_task_runner_->PostTask(
6172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        FROM_HERE,
6182a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        base::Bind(&URLFetcherCore::InformDelegateFetchIsComplete, this));
6192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    return;
6202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  }
6212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  StartURLRequestWhenAppropriate();
6222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)}
6232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
6245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::StartURLRequestWhenAppropriate() {
6255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(network_task_runner_->BelongsToCurrentThread());
6265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (was_cancelled_)
6285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return;
6295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(request_context_getter_);
6315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int64 delay = 0LL;
6335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (original_url_throttler_entry_ == NULL) {
6345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    URLRequestThrottlerManager* manager =
6355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        request_context_getter_->GetURLRequestContext()->throttler_manager();
6365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (manager) {
6375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      original_url_throttler_entry_ =
6385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          manager->RegisterRequestUrl(original_url_);
6395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
6405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
6415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (original_url_throttler_entry_ != NULL) {
6425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    delay = original_url_throttler_entry_->ReserveSendingTimeForNextRequest(
6435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        GetBackoffReleaseTime());
6445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
6455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (delay == 0) {
6475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    StartURLRequest();
6485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  } else {
6495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    base::ThreadTaskRunnerHandle::Get()->PostDelayedTask(
6505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        FROM_HERE, base::Bind(&URLFetcherCore::StartURLRequest, this),
6515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        base::TimeDelta::FromMilliseconds(delay));
6525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
6535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
6545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::CancelURLRequest() {
6565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(network_task_runner_->BelongsToCurrentThread());
6575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (request_.get()) {
6595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    request_->Cancel();
6605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    ReleaseRequest();
6615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
6625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Release the reference to the request context. There could be multiple
6635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // references to URLFetcher::Core at this point so it may take a while to
6645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // delete the object, but we cannot delay the destruction of the request
6655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // context.
6665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  request_context_getter_ = NULL;
6675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  first_party_for_cookies_ = GURL();
6685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  url_request_data_key_ = NULL;
6695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  url_request_create_data_callback_.Reset();
6705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  was_cancelled_ = true;
6712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  response_writer_.reset();
6722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  file_writer_ = NULL;
6735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
6745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::OnCompletedURLRequest(
6765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    base::TimeDelta backoff_delay) {
6775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(delegate_task_runner_->BelongsToCurrentThread());
6785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Save the status and backoff_delay so that delegates can read it.
6805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (delegate_) {
6815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    backoff_delay_ = backoff_delay;
6825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    InformDelegateFetchIsComplete();
6835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
6845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
6855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::InformDelegateFetchIsComplete() {
6875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(delegate_task_runner_->BelongsToCurrentThread());
6885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (delegate_)
6895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    delegate_->OnURLFetchComplete(fetcher_);
6905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
6915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::NotifyMalformedContent() {
6935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(network_task_runner_->BelongsToCurrentThread());
6945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (url_throttler_entry_ != NULL) {
6955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    int status_code = response_code_;
6965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (status_code == URLFetcher::RESPONSE_CODE_INVALID) {
6975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      // The status code will generally be known by the time clients
6985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      // call the |ReceivedContentWasMalformed()| function (which ends up
6995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      // calling the current function) but if it's not, we need to assume
7005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      // the response was successful so that the total failure count
7015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      // used to calculate exponential back-off goes up.
7025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      status_code = 200;
7035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
7045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    url_throttler_entry_->ReceivedContentWasMalformed(status_code);
7055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
7065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
7075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
7082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void URLFetcherCore::DidFinishWriting(int result) {
7092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  if (result != OK) {
7102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    delegate_task_runner_->PostTask(
7112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        FROM_HERE,
7122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        base::Bind(&URLFetcherCore::InformDelegateFetchIsComplete, this));
7132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    return;
7142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  }
7152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  // If the file was successfully closed, then the URL request is complete.
7162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  RetryOrCompleteUrlFetch();
7172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)}
7182a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
7195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::RetryOrCompleteUrlFetch() {
7205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(network_task_runner_->BelongsToCurrentThread());
7215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  base::TimeDelta backoff_delay;
7225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
7235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Checks the response from server.
7245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (response_code_ >= 500 ||
7255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      status_.error() == ERR_TEMPORARILY_THROTTLED) {
7265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // When encountering a server error, we will send the request again
7275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // after backoff time.
7282a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    ++num_retries_on_5xx_;
7295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
7305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // Note that backoff_delay may be 0 because (a) the
7315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // URLRequestThrottlerManager and related code does not
7325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // necessarily back off on the first error, (b) it only backs off
7335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // on some of the 5xx status codes, (c) not all URLRequestContexts
7345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // have a throttler manager.
7355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    base::TimeTicks backoff_release_time = GetBackoffReleaseTime();
7365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    backoff_delay = backoff_release_time - base::TimeTicks::Now();
7375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (backoff_delay < base::TimeDelta())
7385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      backoff_delay = base::TimeDelta();
7395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
7402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    if (automatically_retry_on_5xx_ &&
7412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        num_retries_on_5xx_ <= max_retries_on_5xx_) {
7425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      StartOnIOThread();
7435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      return;
7445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
7455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  } else {
7465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    backoff_delay = base::TimeDelta();
7475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
7482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
7492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  // Retry if the request failed due to network changes.
7502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  if (status_.error() == ERR_NETWORK_CHANGED &&
7512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      num_retries_on_network_changes_ < max_retries_on_network_changes_) {
7522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    ++num_retries_on_network_changes_;
7532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
7542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    // Retry soon, after flushing all the current tasks which may include
7552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    // further network change observers.
7562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    network_task_runner_->PostTask(
7572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        FROM_HERE, base::Bind(&URLFetcherCore::StartOnIOThread, this));
7582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    return;
7592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  }
7602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
7615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  request_context_getter_ = NULL;
7625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  first_party_for_cookies_ = GURL();
7635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  url_request_data_key_ = NULL;
7645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  url_request_create_data_callback_.Reset();
7655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  bool posted = delegate_task_runner_->PostTask(
7665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      FROM_HERE,
7675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      base::Bind(&URLFetcherCore::OnCompletedURLRequest, this, backoff_delay));
7685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
7695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // If the delegate message loop does not exist any more, then the delegate
7705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // should be gone too.
7715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(posted || !delegate_);
7725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
7735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
7745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::ReleaseRequest() {
7755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  upload_progress_checker_timer_.reset();
7765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  request_.reset();
7775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  g_registry.Get().RemoveURLFetcherCore(this);
7785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
7795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
7805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)base::TimeTicks URLFetcherCore::GetBackoffReleaseTime() {
7815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(network_task_runner_->BelongsToCurrentThread());
7825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
7835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (original_url_throttler_entry_) {
7845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    base::TimeTicks original_url_backoff =
7855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        original_url_throttler_entry_->GetExponentialBackoffReleaseTime();
7865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    base::TimeTicks destination_url_backoff;
7875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (url_throttler_entry_ != NULL &&
7885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        original_url_throttler_entry_ != url_throttler_entry_) {
7895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      destination_url_backoff =
7905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          url_throttler_entry_->GetExponentialBackoffReleaseTime();
7915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
7925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
7935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return original_url_backoff > destination_url_backoff ?
7945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        original_url_backoff : destination_url_backoff;
7955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  } else {
7965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return base::TimeTicks();
7975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
7985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
7995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
8005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::CompleteAddingUploadDataChunk(
8015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    const std::string& content, bool is_last_chunk) {
8025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (was_cancelled_) {
8035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // Since CompleteAddingUploadDataChunk() is posted as a *delayed* task, it
8045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // may run after the URLFetcher was already stopped.
8055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return;
8065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
8075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(is_chunked_upload_);
8085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(request_.get());
8095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(!content.empty());
8105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  request_->AppendChunkToUpload(content.data(),
8115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                                static_cast<int>(content.length()),
8125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                                is_last_chunk);
8135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
8145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
8152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void URLFetcherCore::DidWriteBuffer(int result) {
8162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  if (result < 0) {
8172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    delegate_task_runner_->PostTask(
8182a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        FROM_HERE,
8192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        base::Bind(&URLFetcherCore::InformDelegateFetchIsComplete, this));
8202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    return;
8215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
8222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  // Finished writing buffer_. Read some more.
8232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  ReadResponse();
8245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
8255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
8265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::ReadResponse() {
8275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Some servers may treat HEAD requests as GET requests.  To free up the
8285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // network connection as soon as possible, signal that the request has
8295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // completed immediately, without trying to read any data back (all we care
8305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // about is the response code and headers, which we already have).
8315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int bytes_read = 0;
8325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (request_->status().is_success() &&
8335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      (request_type_ != URLFetcher::HEAD))
8345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    request_->Read(buffer_, kBufferSize, &bytes_read);
8355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  OnReadCompleted(request_.get(), bytes_read);
8365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
8375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
8385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::DisownFile() {
8392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  DCHECK(file_writer_);
8405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  file_writer_->DisownFile();
8415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
8425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
8435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::InformDelegateUploadProgress() {
8445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(network_task_runner_->BelongsToCurrentThread());
8455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (request_.get()) {
8465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    int64 current = request_->GetUploadProgress().position();
8475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (current_upload_bytes_ != current) {
8485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      current_upload_bytes_ = current;
8495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      int64 total = -1;
8502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      if (!is_chunked_upload_) {
8512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        total = static_cast<int64>(request_->GetUploadProgress().size());
8522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        // Total may be zero if the UploadDataStream::Init has not been called
8532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        // yet.  Don't send the upload progress until the size is initialized.
8542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        if (!total)
8552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)          return;
8562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      }
8575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      delegate_task_runner_->PostTask(
8585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          FROM_HERE,
8595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          base::Bind(
8605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)              &URLFetcherCore::InformDelegateUploadProgressInDelegateThread,
8615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)              this, current, total));
8625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
8635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
8645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
8655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
8665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::InformDelegateUploadProgressInDelegateThread(
8675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    int64 current, int64 total) {
8685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(delegate_task_runner_->BelongsToCurrentThread());
8695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (delegate_)
8705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    delegate_->OnURLFetchUploadProgress(fetcher_, current, total);
8715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
8725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
8735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::InformDelegateDownloadProgress() {
8745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(network_task_runner_->BelongsToCurrentThread());
8755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  delegate_task_runner_->PostTask(
8765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      FROM_HERE,
8775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      base::Bind(
8785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          &URLFetcherCore::InformDelegateDownloadProgressInDelegateThread,
8795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          this, current_response_bytes_, total_response_bytes_));
8805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
8815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
8825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::InformDelegateDownloadProgressInDelegateThread(
8835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    int64 current, int64 total) {
8845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(delegate_task_runner_->BelongsToCurrentThread());
8855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (delegate_)
8865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    delegate_->OnURLFetchDownloadProgress(fetcher_, current, total);
8875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
8885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
8895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::InformDelegateDownloadDataIfNecessary(int bytes_read) {
8905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(network_task_runner_->BelongsToCurrentThread());
8915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (delegate_ && delegate_->ShouldSendDownloadData()) {
8925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    scoped_ptr<std::string> download_data(
8935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        new std::string(buffer_->data(), bytes_read));
8945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    delegate_task_runner_->PostTask(
8955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        FROM_HERE,
8965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        base::Bind(
8975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)            &URLFetcherCore::InformDelegateDownloadDataInDelegateThread,
8985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)            this, base::Passed(&download_data)));
8995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
9005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
9015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
9025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::InformDelegateDownloadDataInDelegateThread(
9035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    scoped_ptr<std::string> download_data) {
9045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(delegate_task_runner_->BelongsToCurrentThread());
9055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (delegate_)
9065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    delegate_->OnURLFetchDownloadData(fetcher_, download_data.Pass());
9075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
9085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
9095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}  // namespace net
910