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)
71320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci#include <stdint.h>
81320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci
95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/bind.h"
105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/logging.h"
115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/metrics/histogram.h"
12f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles)#include "base/sequenced_task_runner.h"
132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/single_thread_task_runner.h"
145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/stl_util.h"
155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/thread_task_runner_handle.h"
165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/tracked_objects.h"
175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/base/io_buffer.h"
185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/base/load_flags.h"
195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/base/net_errors.h"
200f1bc08d4cfcc34181b0b5cbf065c40f687bf740Torne (Richard Coles)#include "net/base/request_priority.h"
212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "net/base/upload_bytes_element_reader.h"
222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "net/base/upload_data_stream.h"
232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "net/base/upload_file_element_reader.h"
245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/http/http_response_headers.h"
256e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles)#include "net/url_request/redirect_info.h"
265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/url_request/url_fetcher_delegate.h"
272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "net/url_request/url_fetcher_response_writer.h"
285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/url_request/url_request_context.h"
295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/url_request/url_request_context_getter.h"
305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/url_request/url_request_throttler_manager.h"
315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace {
335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const int kBufferSize = 4096;
355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const int kUploadProgressTimerInterval = 100;
365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool g_interception_enabled = false;
372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)bool g_ignore_certificate_requests = false;
385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
391e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles)void EmptyCompletionCallback(int result) {}
401e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles)
415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}  // namespace
425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace net {
445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// URLFetcherCore::Registry ---------------------------------------------------
465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)URLFetcherCore::Registry::Registry() {}
485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)URLFetcherCore::Registry::~Registry() {}
495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::Registry::AddURLFetcherCore(URLFetcherCore* core) {
515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(!ContainsKey(fetchers_, core));
525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  fetchers_.insert(core);
535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::Registry::RemoveURLFetcherCore(URLFetcherCore* core) {
565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(ContainsKey(fetchers_, core));
575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  fetchers_.erase(core);
585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::Registry::CancelAll() {
615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  while (!fetchers_.empty())
624e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)    (*fetchers_.begin())->CancelURLRequest(ERR_ABORTED);
635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// URLFetcherCore -------------------------------------------------------------
665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// static
685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)base::LazyInstance<URLFetcherCore::Registry>
695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    URLFetcherCore::g_registry = LAZY_INSTANCE_INITIALIZER;
705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)URLFetcherCore::URLFetcherCore(URLFetcher* fetcher,
725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                               const GURL& original_url,
735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                               URLFetcher::RequestType request_type,
745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                               URLFetcherDelegate* d)
755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    : fetcher_(fetcher),
765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      original_url_(original_url),
775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      request_type_(request_type),
785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      delegate_(d),
797d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)      delegate_task_runner_(base::ThreadTaskRunnerHandle::Get()),
805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      load_flags_(LOAD_NORMAL),
815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      response_code_(URLFetcher::RESPONSE_CODE_INVALID),
825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      buffer_(new IOBuffer(kBufferSize)),
835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      url_request_data_key_(NULL),
845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      was_fetched_via_proxy_(false),
852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      upload_content_set_(false),
86c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)      upload_range_offset_(0),
87c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)      upload_range_length_(0),
885d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)      referrer_policy_(
895d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)          URLRequest::CLEAR_REFERRER_ON_TRANSITION_FROM_SECURE_TO_INSECURE),
905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      is_chunked_upload_(false),
915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      was_cancelled_(false),
925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      stop_on_redirect_(false),
935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      stopped_on_redirect_(false),
945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      automatically_retry_on_5xx_(true),
952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      num_retries_on_5xx_(0),
962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      max_retries_on_5xx_(0),
972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      num_retries_on_network_changes_(0),
982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      max_retries_on_network_changes_(0),
995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      current_upload_bytes_(-1),
1005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      current_response_bytes_(0),
1015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      total_response_bytes_(-1) {
1025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  CHECK(original_url_.is_valid());
1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
1045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::Start() {
106868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)  DCHECK(delegate_task_runner_.get());
107868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)  DCHECK(request_context_getter_.get()) << "We need an URLRequestContext!";
108868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)  if (network_task_runner_.get()) {
1095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    DCHECK_EQ(network_task_runner_,
1105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)              request_context_getter_->GetNetworkTaskRunner());
1115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  } else {
1125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    network_task_runner_ = request_context_getter_->GetNetworkTaskRunner();
1135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
1145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(network_task_runner_.get()) << "We need an IO task runner";
1155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  network_task_runner_->PostTask(
1175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      FROM_HERE, base::Bind(&URLFetcherCore::StartOnIOThread, this));
1185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
1195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::Stop() {
121868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)  if (delegate_task_runner_.get())  // May be NULL in tests.
1225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    DCHECK(delegate_task_runner_->BelongsToCurrentThread());
1235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  delegate_ = NULL;
1255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  fetcher_ = NULL;
1265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (!network_task_runner_.get())
1275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return;
1285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (network_task_runner_->RunsTasksOnCurrentThread()) {
1294e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)    CancelURLRequest(ERR_ABORTED);
1305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  } else {
1315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    network_task_runner_->PostTask(
1324e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)        FROM_HERE,
1334e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)        base::Bind(&URLFetcherCore::CancelURLRequest, this, ERR_ABORTED));
1345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
1355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
1365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::SetUploadData(const std::string& upload_content_type,
1385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                                   const std::string& upload_content) {
1395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(!is_chunked_upload_);
1402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  DCHECK(!upload_content_set_);
1412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  DCHECK(upload_content_.empty());
1422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  DCHECK(upload_file_path_.empty());
1432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  DCHECK(upload_content_type_.empty());
1442a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
1452a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  // Empty |upload_content_type| is allowed iff the |upload_content| is empty.
1462a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  DCHECK(upload_content.empty() || !upload_content_type.empty());
1472a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
1485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  upload_content_type_ = upload_content_type;
1495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  upload_content_ = upload_content;
1502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  upload_content_set_ = true;
1512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)}
1522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
1532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void URLFetcherCore::SetUploadFilePath(
1542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    const std::string& upload_content_type,
1552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    const base::FilePath& file_path,
156c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)    uint64 range_offset,
157c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)    uint64 range_length,
1582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    scoped_refptr<base::TaskRunner> file_task_runner) {
1592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  DCHECK(!is_chunked_upload_);
1602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  DCHECK(!upload_content_set_);
1612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  DCHECK(upload_content_.empty());
1622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  DCHECK(upload_file_path_.empty());
163c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)  DCHECK_EQ(upload_range_offset_, 0ULL);
164c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)  DCHECK_EQ(upload_range_length_, 0ULL);
1652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  DCHECK(upload_content_type_.empty());
1662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  DCHECK(!upload_content_type.empty());
1672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
1682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  upload_content_type_ = upload_content_type;
1692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  upload_file_path_ = file_path;
170c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)  upload_range_offset_ = range_offset;
171c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)  upload_range_length_ = range_length;
1722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  upload_file_task_runner_ = file_task_runner;
1732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  upload_content_set_ = true;
1745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
1755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::SetChunkedUpload(const std::string& content_type) {
1775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(is_chunked_upload_ ||
1785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)         (upload_content_type_.empty() &&
1795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          upload_content_.empty()));
1802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
1812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  // Empty |content_type| is not allowed here, because it is impossible
1822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  // to ensure non-empty upload content as it is not yet supplied.
1832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  DCHECK(!content_type.empty());
1842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
1855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  upload_content_type_ = content_type;
1865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  upload_content_.clear();
1875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  is_chunked_upload_ = true;
1885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
1895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::AppendChunkToUpload(const std::string& content,
1915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                                         bool is_last_chunk) {
192868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)  DCHECK(delegate_task_runner_.get());
1935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(network_task_runner_.get());
1945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  network_task_runner_->PostTask(
1955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      FROM_HERE,
1965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      base::Bind(&URLFetcherCore::CompleteAddingUploadDataChunk, this, content,
1975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                 is_last_chunk));
1985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
1995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::SetLoadFlags(int load_flags) {
2015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  load_flags_ = load_flags;
2025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)int URLFetcherCore::GetLoadFlags() const {
2055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return load_flags_;
2065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::SetReferrer(const std::string& referrer) {
2095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  referrer_ = referrer;
2105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2125d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)void URLFetcherCore::SetReferrerPolicy(
2135d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)    URLRequest::ReferrerPolicy referrer_policy) {
2145d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)  referrer_policy_ = referrer_policy;
2155d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)}
2165d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)
2175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::SetExtraRequestHeaders(
2185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    const std::string& extra_request_headers) {
2195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  extra_request_headers_.Clear();
2205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  extra_request_headers_.AddHeadersFromString(extra_request_headers);
2215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::AddExtraRequestHeader(const std::string& header_line) {
2245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  extra_request_headers_.AddHeaderFromString(header_line);
2255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::SetRequestContext(
2285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    URLRequestContextGetter* request_context_getter) {
229868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)  DCHECK(!request_context_getter_.get());
2305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(request_context_getter);
2315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  request_context_getter_ = request_context_getter;
2325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::SetFirstPartyForCookies(
2355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    const GURL& first_party_for_cookies) {
2365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(first_party_for_cookies_.is_empty());
2375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  first_party_for_cookies_ = first_party_for_cookies;
2385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::SetURLRequestUserData(
2415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    const void* key,
2425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    const URLFetcher::CreateDataCallback& create_data_callback) {
2435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(key);
2445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(!create_data_callback.is_null());
2455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  url_request_data_key_ = key;
2465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  url_request_create_data_callback_ = create_data_callback;
2475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::SetStopOnRedirect(bool stop_on_redirect) {
2505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  stop_on_redirect_ = stop_on_redirect;
2515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::SetAutomaticallyRetryOn5xx(bool retry) {
2545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  automatically_retry_on_5xx_ = retry;
2555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void URLFetcherCore::SetMaxRetriesOn5xx(int max_retries) {
2582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  max_retries_on_5xx_ = max_retries;
2595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)int URLFetcherCore::GetMaxRetriesOn5xx() const {
2622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  return max_retries_on_5xx_;
2635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)base::TimeDelta URLFetcherCore::GetBackoffDelay() const {
2665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return backoff_delay_;
2675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void URLFetcherCore::SetAutomaticallyRetryOnNetworkChanges(int max_retries) {
2702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  max_retries_on_network_changes_ = max_retries;
2712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)}
2722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
2735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::SaveResponseToFileAtPath(
2742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    const base::FilePath& file_path,
275f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles)    scoped_refptr<base::SequencedTaskRunner> file_task_runner) {
2765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(delegate_task_runner_->BelongsToCurrentThread());
2774e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)  SaveResponseWithWriter(scoped_ptr<URLFetcherResponseWriter>(
2784e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)      new URLFetcherFileWriter(file_task_runner, file_path)));
2795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::SaveResponseToTemporaryFile(
282f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles)    scoped_refptr<base::SequencedTaskRunner> file_task_runner) {
2835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(delegate_task_runner_->BelongsToCurrentThread());
2844e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)  SaveResponseWithWriter(scoped_ptr<URLFetcherResponseWriter>(
2854e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)      new URLFetcherFileWriter(file_task_runner, base::FilePath())));
2864e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)}
2874e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)
2884e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)void URLFetcherCore::SaveResponseWithWriter(
2894e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)    scoped_ptr<URLFetcherResponseWriter> response_writer) {
2904e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)  DCHECK(delegate_task_runner_->BelongsToCurrentThread());
2914e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)  response_writer_ = response_writer.Pass();
2925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)HttpResponseHeaders* URLFetcherCore::GetResponseHeaders() const {
295868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)  return response_headers_.get();
2965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// TODO(panayiotis): socket_address_ is written in the IO thread,
2995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// if this is accessed in the UI thread, this could result in a race.
3005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Same for response_headers_ above and was_fetched_via_proxy_ below.
3015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)HostPortPair URLFetcherCore::GetSocketAddress() const {
3025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return socket_address_;
3035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
3045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool URLFetcherCore::WasFetchedViaProxy() const {
3065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return was_fetched_via_proxy_;
3075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
3085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const GURL& URLFetcherCore::GetOriginalURL() const {
3105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return original_url_;
3115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
3125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const GURL& URLFetcherCore::GetURL() const {
3145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return url_;
3155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
3165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const URLRequestStatus& URLFetcherCore::GetStatus() const {
3185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return status_;
3195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
3205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)int URLFetcherCore::GetResponseCode() const {
3225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return response_code_;
3235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
3245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const ResponseCookies& URLFetcherCore::GetCookies() const {
3265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return cookies_;
3275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
3285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::ReceivedContentWasMalformed() {
3305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(delegate_task_runner_->BelongsToCurrentThread());
3315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (network_task_runner_.get()) {
3325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    network_task_runner_->PostTask(
3335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        FROM_HERE, base::Bind(&URLFetcherCore::NotifyMalformedContent, this));
3345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
3355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
3365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool URLFetcherCore::GetResponseAsString(
3385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    std::string* out_response_string) const {
3394e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)  URLFetcherStringWriter* string_writer =
3404e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)      response_writer_ ? response_writer_->AsStringWriter() : NULL;
3414e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)  if (!string_writer)
3425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return false;
3435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3444e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)  *out_response_string = string_writer->data();
3455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  UMA_HISTOGRAM_MEMORY_KB("UrlFetcher.StringResponseSize",
3464e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)                          (string_writer->data().length() / 1024));
3475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return true;
3485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
3495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool URLFetcherCore::GetResponseAsFilePath(bool take_ownership,
3512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)                                           base::FilePath* out_response_path) {
3525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(delegate_task_runner_->BelongsToCurrentThread());
3534e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)
3544e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)  URLFetcherFileWriter* file_writer =
3554e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)      response_writer_ ? response_writer_->AsFileWriter() : NULL;
3564e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)  if (!file_writer)
3575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return false;
3585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3594e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)  *out_response_path = file_writer->file_path();
3605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (take_ownership) {
3627dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch    // Intentionally calling a file_writer_ method directly without posting
3637dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch    // the task to network_task_runner_.
364eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch    //
3657dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch    // This is for correctly handling the case when file_writer_->DisownFile()
3667dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch    // is soon followed by URLFetcherCore::Stop(). We have to make sure that
3677dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch    // DisownFile takes effect before Stop deletes file_writer_.
3687dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch    //
3697dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch    // This direct call should be thread-safe, since DisownFile itself does no
3707dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch    // file operation. It just flips the state to be referred in destruction.
3714e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)    file_writer->DisownFile();
3725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
3735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return true;
3745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
3755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::OnReceivedRedirect(URLRequest* request,
3776e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles)                                        const RedirectInfo& redirect_info,
3785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                                        bool* defer_redirect) {
3795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK_EQ(request, request_.get());
3805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(network_task_runner_->BelongsToCurrentThread());
3815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (stop_on_redirect_) {
3825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    stopped_on_redirect_ = true;
3836e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles)    url_ = redirect_info.new_url;
3845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    response_code_ = request_->GetResponseCode();
3855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    was_fetched_via_proxy_ = request_->was_fetched_via_proxy();
3865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    request->Cancel();
3875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    OnReadCompleted(request, 0);
3885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
3895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
3905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::OnResponseStarted(URLRequest* request) {
3925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK_EQ(request, request_.get());
3935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(network_task_runner_->BelongsToCurrentThread());
3945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (request_->status().is_success()) {
3955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    response_code_ = request_->GetResponseCode();
3965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    response_headers_ = request_->response_headers();
3975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    socket_address_ = request_->GetSocketAddress();
3985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    was_fetched_via_proxy_ = request_->was_fetched_via_proxy();
3995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    total_response_bytes_ = request_->GetExpectedContentSize();
4005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
4015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  ReadResponse();
4035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
4045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4052a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void URLFetcherCore::OnCertificateRequested(
4062a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    URLRequest* request,
4072a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    SSLCertRequestInfo* cert_request_info) {
4082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  DCHECK_EQ(request, request_.get());
4092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  DCHECK(network_task_runner_->BelongsToCurrentThread());
4102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
4112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  if (g_ignore_certificate_requests) {
4122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    request->ContinueWithCertificate(NULL);
4132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  } else {
4142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    request->Cancel();
4152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  }
4162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)}
4172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
4185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::OnReadCompleted(URLRequest* request,
4195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                                     int bytes_read) {
4205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(request == request_);
4215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(network_task_runner_->BelongsToCurrentThread());
4225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (!stopped_on_redirect_)
4245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    url_ = request->url();
4255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  URLRequestThrottlerManager* throttler_manager =
4265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      request->context()->throttler_manager();
4275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (throttler_manager) {
4285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    url_throttler_entry_ = throttler_manager->RegisterRequestUrl(url_);
4295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
4305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  do {
4325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (!request_->status().is_success() || bytes_read <= 0)
4335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      break;
4345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    current_response_bytes_ += bytes_read;
4365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    InformDelegateDownloadProgress();
4375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
438868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)    const int result =
439868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)        WriteBuffer(new DrainableIOBuffer(buffer_.get(), bytes_read));
4402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    if (result < 0) {
4412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      // Write failed or waiting for write completion.
4424e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)      return;
4435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
444868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)  } while (request_->Read(buffer_.get(), kBufferSize, &bytes_read));
4455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  const URLRequestStatus status = request_->status();
4475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (status.is_success())
4495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    request_->GetResponseCookies(&cookies_);
4505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // See comments re: HEAD requests in ReadResponse().
4524e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)  if (!status.is_io_pending() || request_type_ == URLFetcher::HEAD) {
4535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    status_ = status;
4545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    ReleaseRequest();
4555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    // No more data to write.
4572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    const int result = response_writer_->Finish(
4582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        base::Bind(&URLFetcherCore::DidFinishWriting, this));
4592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    if (result != ERR_IO_PENDING)
4602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      DidFinishWriting(result);
4615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
4625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
4635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::CancelAll() {
4655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  g_registry.Get().CancelAll();
4665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
4675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)int URLFetcherCore::GetNumFetcherCores() {
4695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return g_registry.Get().size();
4705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
4715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::SetEnableInterceptionForTests(bool enabled) {
4735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  g_interception_enabled = enabled;
4745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
4755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4762a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void URLFetcherCore::SetIgnoreCertificateRequests(bool ignored) {
4772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  g_ignore_certificate_requests = ignored;
4782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)}
4792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
4805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)URLFetcherCore::~URLFetcherCore() {
4811320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci  // |request_| should be NULL. If not, it's unsafe to delete it here since we
4825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // may not be on the IO thread.
4835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(!request_.get());
4845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
4855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::StartOnIOThread() {
4875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(network_task_runner_->BelongsToCurrentThread());
4885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4894e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)  if (!response_writer_)
4904e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)    response_writer_.reset(new URLFetcherStringWriter);
4915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  const int result = response_writer_->Initialize(
4932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      base::Bind(&URLFetcherCore::DidInitializeWriter, this));
4942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  if (result != ERR_IO_PENDING)
4952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    DidInitializeWriter(result);
4965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
4975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::StartURLRequest() {
4995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(network_task_runner_->BelongsToCurrentThread());
5005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (was_cancelled_) {
5025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // Since StartURLRequest() is posted as a *delayed* task, it may
5035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // run after the URLFetcher was already stopped.
5045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return;
5055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
5065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
507868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)  DCHECK(request_context_getter_.get());
5085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(!request_.get());
5095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  g_registry.Get().AddURLFetcherCore(this);
5115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  current_response_bytes_ = 0;
5120f1bc08d4cfcc34181b0b5cbf065c40f687bf740Torne (Richard Coles)  request_ = request_context_getter_->GetURLRequestContext()->CreateRequest(
513a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)      original_url_, DEFAULT_PRIORITY, this, NULL);
5145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  request_->set_stack_trace(stack_trace_);
5155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int flags = request_->load_flags() | load_flags_;
5165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (!g_interception_enabled)
5175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    flags = flags | LOAD_DISABLE_INTERCEPT;
5185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (is_chunked_upload_)
5205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    request_->EnableChunkedUpload();
521f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles)  request_->SetLoadFlags(flags);
522c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)  request_->SetReferrer(referrer_);
5235d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)  request_->set_referrer_policy(referrer_policy_);
5245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  request_->set_first_party_for_cookies(first_party_for_cookies_.is_empty() ?
5255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      original_url_ : first_party_for_cookies_);
5265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (url_request_data_key_ && !url_request_create_data_callback_.is_null()) {
5275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    request_->SetUserData(url_request_data_key_,
5285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                          url_request_create_data_callback_.Run());
5295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
5305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  switch (request_type_) {
5325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    case URLFetcher::GET:
5335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      break;
5345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    case URLFetcher::POST:
5365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    case URLFetcher::PUT:
5372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    case URLFetcher::PATCH:
5382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      // Upload content must be set.
5392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      DCHECK(is_chunked_upload_ || upload_content_set_);
5405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      request_->set_method(
5422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)          request_type_ == URLFetcher::POST ? "POST" :
5432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)          request_type_ == URLFetcher::PUT ? "PUT" : "PATCH");
5442a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      if (!upload_content_type_.empty()) {
5452a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        extra_request_headers_.SetHeader(HttpRequestHeaders::kContentType,
5462a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)                                         upload_content_type_);
5472a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      }
5485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      if (!upload_content_.empty()) {
5492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        scoped_ptr<UploadElementReader> reader(new UploadBytesElementReader(
5502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)            upload_content_.data(), upload_content_.size()));
5512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        request_->set_upload(make_scoped_ptr(
5522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)            UploadDataStream::CreateWithReader(reader.Pass(), 0)));
5532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      } else if (!upload_file_path_.empty()) {
554868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)        scoped_ptr<UploadElementReader> reader(
555868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)            new UploadFileElementReader(upload_file_task_runner_.get(),
556868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)                                        upload_file_path_,
557868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)                                        upload_range_offset_,
558868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)                                        upload_range_length_,
559868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)                                        base::Time()));
5602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        request_->set_upload(make_scoped_ptr(
5612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)            UploadDataStream::CreateWithReader(reader.Pass(), 0)));
5625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      }
5635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      current_upload_bytes_ = -1;
5655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      // TODO(kinaba): http://crbug.com/118103. Implement upload callback in the
5665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      //  layer and avoid using timer here.
5675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      upload_progress_checker_timer_.reset(
5685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          new base::RepeatingTimer<URLFetcherCore>());
5695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      upload_progress_checker_timer_->Start(
5705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          FROM_HERE,
5715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          base::TimeDelta::FromMilliseconds(kUploadProgressTimerInterval),
5725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          this,
5735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          &URLFetcherCore::InformDelegateUploadProgress);
5745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      break;
5755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    case URLFetcher::HEAD:
5775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      request_->set_method("HEAD");
5785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      break;
5795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    case URLFetcher::DELETE_REQUEST:
5815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      request_->set_method("DELETE");
5825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      break;
5835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    default:
5855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      NOTREACHED();
5865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
5875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (!extra_request_headers_.IsEmpty())
5895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    request_->SetExtraRequestHeaders(extra_request_headers_);
5905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  request_->Start();
5925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
5935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void URLFetcherCore::DidInitializeWriter(int result) {
5952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  if (result != OK) {
5964e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)    CancelURLRequest(result);
5972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    delegate_task_runner_->PostTask(
5982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        FROM_HERE,
5992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        base::Bind(&URLFetcherCore::InformDelegateFetchIsComplete, this));
6002a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    return;
6012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  }
6022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  StartURLRequestWhenAppropriate();
6032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)}
6042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
6055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::StartURLRequestWhenAppropriate() {
6065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(network_task_runner_->BelongsToCurrentThread());
6075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (was_cancelled_)
6095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return;
6105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
611868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)  DCHECK(request_context_getter_.get());
6125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6131320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci  int64 delay = INT64_C(0);
6141320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci  if (!original_url_throttler_entry_.get()) {
6155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    URLRequestThrottlerManager* manager =
6165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        request_context_getter_->GetURLRequestContext()->throttler_manager();
6175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (manager) {
6185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      original_url_throttler_entry_ =
6195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          manager->RegisterRequestUrl(original_url_);
6205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
6215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
6221320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci  if (original_url_throttler_entry_.get()) {
6235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    delay = original_url_throttler_entry_->ReserveSendingTimeForNextRequest(
6245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        GetBackoffReleaseTime());
6255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
6265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6271320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci  if (delay == INT64_C(0)) {
6285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    StartURLRequest();
6295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  } else {
6305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    base::ThreadTaskRunnerHandle::Get()->PostDelayedTask(
6315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        FROM_HERE, base::Bind(&URLFetcherCore::StartURLRequest, this),
6325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        base::TimeDelta::FromMilliseconds(delay));
6335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
6345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
6355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6364e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)void URLFetcherCore::CancelURLRequest(int error) {
6375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(network_task_runner_->BelongsToCurrentThread());
6385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (request_.get()) {
6404e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)    request_->CancelWithError(error);
6415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    ReleaseRequest();
6425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
6434e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)
6444e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)  // Set the error manually.
6454e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)  // Normally, calling URLRequest::CancelWithError() results in calling
6464e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)  // OnReadCompleted() with bytes_read = -1 via an asynchronous task posted by
6474e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)  // URLRequestJob::NotifyDone(). But, because the request was released
6484e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)  // immediately after being canceled, the request could not call
6494e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)  // OnReadCompleted() which overwrites |status_| with the error status.
6504e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)  status_.set_status(URLRequestStatus::CANCELED);
6514e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)  status_.set_error(error);
6524e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)
6535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Release the reference to the request context. There could be multiple
6545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // references to URLFetcher::Core at this point so it may take a while to
6555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // delete the object, but we cannot delay the destruction of the request
6565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // context.
6575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  request_context_getter_ = NULL;
6585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  first_party_for_cookies_ = GURL();
6595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  url_request_data_key_ = NULL;
6605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  url_request_create_data_callback_.Reset();
6615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  was_cancelled_ = true;
6625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
6635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::OnCompletedURLRequest(
6655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    base::TimeDelta backoff_delay) {
6665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(delegate_task_runner_->BelongsToCurrentThread());
6675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Save the status and backoff_delay so that delegates can read it.
6695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (delegate_) {
6705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    backoff_delay_ = backoff_delay;
6715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    InformDelegateFetchIsComplete();
6725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
6735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
6745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::InformDelegateFetchIsComplete() {
6765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(delegate_task_runner_->BelongsToCurrentThread());
6775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (delegate_)
6785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    delegate_->OnURLFetchComplete(fetcher_);
6795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
6805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::NotifyMalformedContent() {
6825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(network_task_runner_->BelongsToCurrentThread());
6831320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci  if (url_throttler_entry_.get()) {
6845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    int status_code = response_code_;
6855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (status_code == URLFetcher::RESPONSE_CODE_INVALID) {
6865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      // The status code will generally be known by the time clients
6875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      // call the |ReceivedContentWasMalformed()| function (which ends up
6885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      // calling the current function) but if it's not, we need to assume
6895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      // the response was successful so that the total failure count
6905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      // used to calculate exponential back-off goes up.
6915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      status_code = 200;
6925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
6935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    url_throttler_entry_->ReceivedContentWasMalformed(status_code);
6945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
6955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
6965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void URLFetcherCore::DidFinishWriting(int result) {
6982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  if (result != OK) {
6994e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)    CancelURLRequest(result);
7002a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    delegate_task_runner_->PostTask(
7012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        FROM_HERE,
7022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        base::Bind(&URLFetcherCore::InformDelegateFetchIsComplete, this));
7032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    return;
7042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  }
7052a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  // If the file was successfully closed, then the URL request is complete.
7062a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  RetryOrCompleteUrlFetch();
7072a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)}
7082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
7095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::RetryOrCompleteUrlFetch() {
7105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(network_task_runner_->BelongsToCurrentThread());
7115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  base::TimeDelta backoff_delay;
7125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
7135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Checks the response from server.
7145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (response_code_ >= 500 ||
7155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      status_.error() == ERR_TEMPORARILY_THROTTLED) {
7165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // When encountering a server error, we will send the request again
7175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // after backoff time.
7182a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    ++num_retries_on_5xx_;
7195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
7205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // Note that backoff_delay may be 0 because (a) the
7215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // URLRequestThrottlerManager and related code does not
7225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // necessarily back off on the first error, (b) it only backs off
7235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // on some of the 5xx status codes, (c) not all URLRequestContexts
7245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // have a throttler manager.
7255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    base::TimeTicks backoff_release_time = GetBackoffReleaseTime();
7265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    backoff_delay = backoff_release_time - base::TimeTicks::Now();
7275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (backoff_delay < base::TimeDelta())
7285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      backoff_delay = base::TimeDelta();
7295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
7302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    if (automatically_retry_on_5xx_ &&
7312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        num_retries_on_5xx_ <= max_retries_on_5xx_) {
7325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      StartOnIOThread();
7335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      return;
7345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
7355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  } else {
7365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    backoff_delay = base::TimeDelta();
7375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
7382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
7392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  // Retry if the request failed due to network changes.
7402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  if (status_.error() == ERR_NETWORK_CHANGED &&
7412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      num_retries_on_network_changes_ < max_retries_on_network_changes_) {
7422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    ++num_retries_on_network_changes_;
7432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
7442a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    // Retry soon, after flushing all the current tasks which may include
7452a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    // further network change observers.
7462a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    network_task_runner_->PostTask(
7472a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        FROM_HERE, base::Bind(&URLFetcherCore::StartOnIOThread, this));
7482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    return;
7492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  }
7502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
7515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  request_context_getter_ = NULL;
7525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  first_party_for_cookies_ = GURL();
7535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  url_request_data_key_ = NULL;
7545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  url_request_create_data_callback_.Reset();
7555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  bool posted = delegate_task_runner_->PostTask(
7565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      FROM_HERE,
7575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      base::Bind(&URLFetcherCore::OnCompletedURLRequest, this, backoff_delay));
7585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
7595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // If the delegate message loop does not exist any more, then the delegate
7605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // should be gone too.
7615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(posted || !delegate_);
7625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
7635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
7645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::ReleaseRequest() {
7655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  upload_progress_checker_timer_.reset();
7665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  request_.reset();
7675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  g_registry.Get().RemoveURLFetcherCore(this);
7685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
7695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
7705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)base::TimeTicks URLFetcherCore::GetBackoffReleaseTime() {
7715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(network_task_runner_->BelongsToCurrentThread());
7725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
7731320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci  if (!original_url_throttler_entry_.get())
7745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return base::TimeTicks();
7751320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci
7761320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci  base::TimeTicks original_url_backoff =
7771320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci      original_url_throttler_entry_->GetExponentialBackoffReleaseTime();
7781320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci  base::TimeTicks destination_url_backoff;
7791320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci  if (url_throttler_entry_.get() &&
7801320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci      original_url_throttler_entry_.get() != url_throttler_entry_.get()) {
7811320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci    destination_url_backoff =
7821320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci        url_throttler_entry_->GetExponentialBackoffReleaseTime();
7835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
7841320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci
7851320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci  return original_url_backoff > destination_url_backoff ?
7861320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci      original_url_backoff : destination_url_backoff;
7875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
7885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
7895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::CompleteAddingUploadDataChunk(
7905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    const std::string& content, bool is_last_chunk) {
7915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (was_cancelled_) {
7925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // Since CompleteAddingUploadDataChunk() is posted as a *delayed* task, it
7935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // may run after the URLFetcher was already stopped.
7945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return;
7955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
7965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(is_chunked_upload_);
7975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(request_.get());
7985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(!content.empty());
7995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  request_->AppendChunkToUpload(content.data(),
8005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                                static_cast<int>(content.length()),
8015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                                is_last_chunk);
8025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
8035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
80490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)int URLFetcherCore::WriteBuffer(scoped_refptr<DrainableIOBuffer> data) {
80590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)  while (data->BytesRemaining() > 0) {
80690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)    const int result = response_writer_->Write(
807868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)        data.get(),
808868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)        data->BytesRemaining(),
80990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)        base::Bind(&URLFetcherCore::DidWriteBuffer, this, data));
8104e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)    if (result < 0) {
8111e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles)      if (result != ERR_IO_PENDING)
8121e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles)        DidWriteBuffer(data, result);
81390dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)      return result;
8144e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)    }
81590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)    data->DidConsume(result);
81690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)  }
81790dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)  return OK;
81890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)}
81990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)
82090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)void URLFetcherCore::DidWriteBuffer(scoped_refptr<DrainableIOBuffer> data,
82190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)                                    int result) {
82290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)  if (result < 0) {  // Handle errors.
8234e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)    CancelURLRequest(result);
8241e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles)    response_writer_->Finish(base::Bind(&EmptyCompletionCallback));
8252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    delegate_task_runner_->PostTask(
8262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        FROM_HERE,
8272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        base::Bind(&URLFetcherCore::InformDelegateFetchIsComplete, this));
8282a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    return;
8295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
8304e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)
8314e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)  // Continue writing.
8324e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)  data->DidConsume(result);
8334e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)  if (WriteBuffer(data) < 0)
8344e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)    return;
8354e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)
836f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles)  // Finished writing buffer_. Read some more, unless the request has been
837f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles)  // cancelled and deleted.
83890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)  DCHECK_EQ(0, data->BytesRemaining());
839f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles)  if (request_.get())
840f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles)    ReadResponse();
8415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
8425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
8435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::ReadResponse() {
8441320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci  // Some servers may treat HEAD requests as GET requests. To free up the
8455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // network connection as soon as possible, signal that the request has
8465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // completed immediately, without trying to read any data back (all we care
8475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // about is the response code and headers, which we already have).
8485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int bytes_read = 0;
8495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (request_->status().is_success() &&
8501320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci      (request_type_ != URLFetcher::HEAD)) {
8511320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci    if (!request_->Read(buffer_.get(), kBufferSize, &bytes_read))
8521320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci      bytes_read = -1;  // Match OnReadCompleted() interface contract.
8531320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci  }
8545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  OnReadCompleted(request_.get(), bytes_read);
8555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
8565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
8575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::InformDelegateUploadProgress() {
8585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(network_task_runner_->BelongsToCurrentThread());
8595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (request_.get()) {
8605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    int64 current = request_->GetUploadProgress().position();
8615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (current_upload_bytes_ != current) {
8625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      current_upload_bytes_ = current;
8635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      int64 total = -1;
8642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      if (!is_chunked_upload_) {
8652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        total = static_cast<int64>(request_->GetUploadProgress().size());
8662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        // Total may be zero if the UploadDataStream::Init has not been called
8671320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci        // yet. Don't send the upload progress until the size is initialized.
8682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        if (!total)
8692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)          return;
8702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      }
8715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      delegate_task_runner_->PostTask(
8725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          FROM_HERE,
8735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          base::Bind(
8745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)              &URLFetcherCore::InformDelegateUploadProgressInDelegateThread,
8755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)              this, current, total));
8765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
8775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
8785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
8795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
8805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::InformDelegateUploadProgressInDelegateThread(
8815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    int64 current, int64 total) {
8825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(delegate_task_runner_->BelongsToCurrentThread());
8835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (delegate_)
8845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    delegate_->OnURLFetchUploadProgress(fetcher_, current, total);
8855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
8865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
8875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::InformDelegateDownloadProgress() {
8885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(network_task_runner_->BelongsToCurrentThread());
8895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  delegate_task_runner_->PostTask(
8905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      FROM_HERE,
8915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      base::Bind(
8925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          &URLFetcherCore::InformDelegateDownloadProgressInDelegateThread,
8935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          this, current_response_bytes_, total_response_bytes_));
8945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
8955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
8965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLFetcherCore::InformDelegateDownloadProgressInDelegateThread(
8975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    int64 current, int64 total) {
8985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(delegate_task_runner_->BelongsToCurrentThread());
8995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (delegate_)
9005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    delegate_->OnURLFetchDownloadProgress(fetcher_, current, total);
9015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
9025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
9035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}  // namespace net
904