Add control to use If-Range header for range request.

content/browser/download/download_request_core.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "content/browser/download/download_request_core.h"
 
 #include <string>
 
 #include "base/bind.h"
 #include "base/callback_helpers.h"
@@ skipping 93 matching lines @@
 const int DownloadRequestCore::kDownloadByteStreamSize = 100 * 1024;
 
 // static
 std::unique_ptr<net::URLRequest> DownloadRequestCore::CreateRequestOnIOThread(
     uint32_t download_id,
     DownloadUrlParameters* params) {
   DCHECK_CURRENTLY_ON(BrowserThread::IO);
   DCHECK(download_id == DownloadItem::kInvalidId ||
          !params->content_initiated())
       << "Content initiated downloads shouldn't specify a download ID";
+  DCHECK(params->offset() >= 0);
 
   // ResourceDispatcherHost{Base} is-not-a URLRequest::Delegate, and
   // DownloadUrlParameters can-not include resource_dispatcher_host_impl.h, so
   // we must down cast. RDHI is the only subclass of RDH as of 2012 May 4.
   std::unique_ptr<net::URLRequest> request(
       params->url_request_context_getter()
           ->GetURLRequestContext()
           ->CreateRequest(params->url(), net::DEFAULT_PRIORITY, nullptr));
   request->set_method(params->method());
 
@@ skipping 25 matching lines @@
     // validating the entry if present.
     if (request->get_upload())
       load_flags |= net::LOAD_ONLY_FROM_CACHE | net::LOAD_SKIP_CACHE_VALIDATION;
     else
       load_flags |= net::LOAD_SKIP_CACHE_VALIDATION;
   } else {
     load_flags |= net::LOAD_DISABLE_CACHE;
   }
   request->SetLoadFlags(load_flags);
 
-  bool has_last_modified = !params->last_modified().empty();
-  bool has_etag = !params->etag().empty();
-
-  // Strong validator(i.e. etag or last modified) is required in range requests
-  // for download resumption and parallel download.
-  DCHECK((params->offset() == 0 &&
-          params->length() == DownloadSaveInfo::kLengthFullContent) ||
-         has_etag || has_last_modified);
-
-  // Add "Range" and "If-Range" request header fields if the range request is to
-  // retrieve bytes from {offset} to the end of the file.
-  // E.g. "Range:bytes=50-".
-  if (params->offset() > 0 &&
-      params->length() == DownloadSaveInfo::kLengthFullContent &&
-      (has_etag || has_last_modified)) {
-    request->SetExtraRequestHeaderByName(
-        net::HttpRequestHeaders::kRange,
-        base::StringPrintf("bytes=%" PRId64 "-", params->offset()), true);
-
-    // In accordance with RFC 7233 Section 3.2, use If-Range to specify that
-    // the server return the entire entity if the validator doesn't match.
-    // Last-Modified can be used in the absence of ETag as a validator if the
-    // response headers satisfied the HttpUtil::HasStrongValidators() predicate.
-    //
-    // This function assumes that HasStrongValidators() was true and that the
-    // ETag and Last-Modified header values supplied are valid.
-    request->SetExtraRequestHeaderByName(
-        net::HttpRequestHeaders::kIfRange,
-        has_etag ? params->etag() : params->last_modified(), true);
-  }
-
-  // Add "Range", "If-Match", and "If-Unmodified-Since" for range requests with
-  // last byte position specified. e.g. "Range:bytes=50-59". If the file is
-  // updated on the server, we should get http 412 response code.
-  if (params->length() != DownloadSaveInfo::kLengthFullContent &&
-      (has_etag || has_last_modified)) {
-    request->SetExtraRequestHeaderByName(
-        net::HttpRequestHeaders::kRange,
-        base::StringPrintf("bytes=%" PRId64 "-%" PRId64, params->offset(),
-                           params->offset() + params->length() - 1),
-        true);
-    if (has_etag) {
-      request->SetExtraRequestHeaderByName(net::HttpRequestHeaders::kIfMatch,
-                                           params->etag(), true);
-    }
-    // According to RFC 7232 section 3.4, "If-Unmodified-Since" is mainly for
-    // old servers that didn't implement "If-Match" and must be ignored when
-    // "If-Match" presents.
-    if (has_last_modified) {
-      request->SetExtraRequestHeaderByName(
-          net::HttpRequestHeaders::kIfUnmodifiedSince, params->last_modified(),
-          true);
-    }
-  }
+  // Add partial requests headers.
+  AddPartialRequestHeaders(request.get(), params);
 
   // Downloads are treated as top level navigations. Hence the first-party
   // origin for cookies is always based on the target URL and is updated on
   // redirects.
   request->set_first_party_for_cookies(params->url());
   request->set_first_party_url_policy(
       net::URLRequest::UPDATE_FIRST_PARTY_URL_ON_REDIRECT);
   request->set_initiator(params->initiator());
 
   for (const auto& header : params->request_headers())
@@ skipping 438 matching lines @@
 
     return DOWNLOAD_INTERRUPT_REASON_NONE;
   }
 
   if (http_headers.response_code() == net::HTTP_PARTIAL_CONTENT)
     return DOWNLOAD_INTERRUPT_REASON_SERVER_BAD_CONTENT;
 
   return DOWNLOAD_INTERRUPT_REASON_NONE;
 }
 
+// static
+void DownloadRequestCore::AddPartialRequestHeaders(
+    net::URLRequest* request,
+    DownloadUrlParameters* params) {
+  if (params->offset() == 0 &&
+      params->length() == DownloadSaveInfo::kLengthFullContent)
+    return;
+
+  bool has_last_modified = !params->last_modified().empty();
+  bool has_etag = !params->etag().empty();
+
+  // Strong validator(i.e. etag or last modified) is required in range requests
+  // for download resumption and parallel download.
+  DCHECK(has_etag || has_last_modified);
+  if (!has_etag && !has_last_modified) {
+    DVLOG(1) << "Creating partial request without strong validators.";
+    return;
+  }
+
+  // Add "Range" header.
+  std::string range_header =
+      (params->length() == DownloadSaveInfo::kLengthFullContent)
+          ? base::StringPrintf("bytes=%" PRId64 "-", params->offset())
+          : base::StringPrintf("bytes=%" PRId64 "-%" PRId64, params->offset(),
+                               params->offset() + params->length() - 1);
+  request->SetExtraRequestHeaderByName(net::HttpRequestHeaders::kRange,
+                                       range_header, true);
+
+  // Add "If-Range" headers.
+  if (params->use_if_range()) {
+    // In accordance with RFC 7233 Section 3.2, use If-Range to specify that
+    // the server return the entire entity if the validator doesn't match.
+    // Last-Modified can be used in the absence of ETag as a validator if the
+    // response headers satisfied the HttpUtil::HasStrongValidators()
+    // predicate.
+    //
+    // This function assumes that HasStrongValidators() was true and that the
+    // ETag and Last-Modified header values supplied are valid.
+    request->SetExtraRequestHeaderByName(
+        net::HttpRequestHeaders::kIfRange,
+        has_etag ? params->etag() : params->last_modified(), true);
+    return;
+  }
+
+  // Add "If-Match"/"If-Unmodified-Since" headers.
+  if (has_etag) {
+    request->SetExtraRequestHeaderByName(net::HttpRequestHeaders::kIfMatch,
+                                         params->etag(), true);
+  }
+  // According to RFC 7232 section 3.4, "If-Unmodified-Since" is mainly for
+  // old servers that didn't implement "If-Match" and must be ignored when
+  // "If-Match" presents.
+  if (has_last_modified) {
+    request->SetExtraRequestHeaderByName(
+        net::HttpRequestHeaders::kIfUnmodifiedSince, params->last_modified(),
+        true);
+  }
+}
+
 }  // namespace content