yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 1 | // Copyright 2016 The Chromium Authors. All rights reserved. |
| 2 | // Use of this source code is governed by a BSD-style license that can be |
| 3 | // found in the LICENSE file. |
| 4 | |
| 5 | #include "content/browser/loader/mojo_async_resource_handler.h" |
| 6 | |
tzik | bedcbce | 2017-01-24 15:08:05 | [diff] [blame] | 7 | #include <algorithm> |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 8 | #include <utility> |
yhirano | 2a525f9 | 2017-01-18 08:16:46 | [diff] [blame] | 9 | #include <vector> |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 10 | |
tzik | ad092abe | 2017-01-24 02:34:17 | [diff] [blame] | 11 | #include "base/bind.h" |
tzik | bedcbce | 2017-01-24 15:08:05 | [diff] [blame] | 12 | #include "base/location.h" |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 13 | #include "base/logging.h" |
| 14 | #include "base/macros.h" |
tzik | bedcbce | 2017-01-24 15:08:05 | [diff] [blame] | 15 | #include "base/memory/ptr_util.h" |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 16 | #include "base/time/time.h" |
tzik | 47a9f9b | 2016-12-01 04:16:24 | [diff] [blame] | 17 | #include "content/browser/loader/downloaded_temp_file_impl.h" |
arthursonzogni | 84dda21 | 2018-03-01 02:06:36 | [diff] [blame] | 18 | #include "content/browser/loader/navigation_metrics.h" |
tyoshino | 3b3ee8e9 | 2016-12-07 16:04:07 | [diff] [blame] | 19 | #include "content/browser/loader/resource_controller.h" |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 20 | #include "content/browser/loader/resource_dispatcher_host_impl.h" |
| 21 | #include "content/browser/loader/resource_request_info_impl.h" |
yhirano | 5939370 | 2016-11-16 06:10:24 | [diff] [blame] | 22 | #include "content/public/browser/global_request_id.h" |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 23 | #include "mojo/public/c/system/data_pipe.h" |
yhirano | 7153dc47 | 2016-11-16 09:42:46 | [diff] [blame] | 24 | #include "mojo/public/cpp/bindings/message.h" |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 25 | #include "net/base/mime_sniffer.h" |
mmenke | 6add172 | 2017-03-21 21:31:17 | [diff] [blame] | 26 | #include "net/base/net_errors.h" |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 27 | #include "net/url_request/redirect_info.h" |
John Abd-El-Malek | 4624803 | 2018-01-17 19:11:23 | [diff] [blame] | 28 | #include "services/network/public/cpp/resource_response.h" |
Takashi Toyoshima | aa27866 | 2017-11-20 11:11:26 | [diff] [blame] | 29 | #include "services/network/public/cpp/url_loader_completion_status.h" |
Yutaka Hirano | 4cd4e30 | 2018-02-01 17:38:00 | [diff] [blame] | 30 | #include "services/network/resource_scheduler.h" |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 31 | |
| 32 | namespace content { |
| 33 | namespace { |
| 34 | |
| 35 | int g_allocation_size = MojoAsyncResourceHandler::kDefaultAllocationSize; |
| 36 | |
| 37 | // MimeTypeResourceHandler *implicitly* requires that the buffer size |
| 38 | // returned from OnWillRead should be larger than certain size. |
| 39 | // TODO(yhirano): Fix MimeTypeResourceHandler. |
| 40 | constexpr size_t kMinAllocationSize = 2 * net::kMaxBytesToSniff; |
| 41 | |
| 42 | constexpr size_t kMaxChunkSize = 32 * 1024; |
| 43 | |
John Abd-El-Malek | b165dc5 | 2018-01-18 17:12:18 | [diff] [blame] | 44 | void NotReached(network::mojom::URLLoaderRequest mojo_request, |
| 45 | network::mojom::URLLoaderClientPtr url_loader_client) { |
tzik | ad092abe | 2017-01-24 02:34:17 | [diff] [blame] | 46 | NOTREACHED(); |
| 47 | } |
| 48 | |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 49 | } // namespace |
| 50 | |
| 51 | // This class is for sharing the ownership of a ScopedDataPipeProducerHandle |
| 52 | // between WriterIOBuffer and MojoAsyncResourceHandler. |
| 53 | class MojoAsyncResourceHandler::SharedWriter final |
| 54 | : public base::RefCountedThreadSafe<SharedWriter> { |
| 55 | public: |
| 56 | explicit SharedWriter(mojo::ScopedDataPipeProducerHandle writer) |
| 57 | : writer_(std::move(writer)) {} |
| 58 | mojo::DataPipeProducerHandle writer() { return writer_.get(); } |
| 59 | |
| 60 | private: |
| 61 | friend class base::RefCountedThreadSafe<SharedWriter>; |
| 62 | ~SharedWriter() {} |
| 63 | |
| 64 | const mojo::ScopedDataPipeProducerHandle writer_; |
| 65 | |
| 66 | DISALLOW_COPY_AND_ASSIGN(SharedWriter); |
| 67 | }; |
| 68 | |
| 69 | // This class is a IOBuffer subclass for data gotten from a |
| 70 | // ScopedDataPipeProducerHandle. |
| 71 | class MojoAsyncResourceHandler::WriterIOBuffer final |
| 72 | : public net::IOBufferWithSize { |
| 73 | public: |
Reilly Grant | 789c63e | 2017-08-04 15:30:16 | [diff] [blame] | 74 | // |data| and |size| should be gotten from |writer| via BeginWriteData. |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 75 | // They will be accesible via IOBuffer methods. As |writer| is stored in this |
| 76 | // instance, |data| will be kept valid as long as the following conditions |
| 77 | // hold: |
| 78 | // 1. |data| is not invalidated via EndWriteDataRaw. |
| 79 | // 2. |this| instance is alive. |
| 80 | WriterIOBuffer(scoped_refptr<SharedWriter> writer, void* data, size_t size) |
| 81 | : net::IOBufferWithSize(static_cast<char*>(data), size), |
| 82 | writer_(std::move(writer)) {} |
| 83 | |
| 84 | private: |
| 85 | ~WriterIOBuffer() override { |
| 86 | // Avoid deleting |data_| in the IOBuffer destructor. |
| 87 | data_ = nullptr; |
| 88 | } |
| 89 | |
| 90 | // This member is for keeping the writer alive. |
| 91 | scoped_refptr<SharedWriter> writer_; |
| 92 | |
| 93 | DISALLOW_COPY_AND_ASSIGN(WriterIOBuffer); |
| 94 | }; |
| 95 | |
| 96 | MojoAsyncResourceHandler::MojoAsyncResourceHandler( |
| 97 | net::URLRequest* request, |
| 98 | ResourceDispatcherHostImpl* rdh, |
John Abd-El-Malek | b165dc5 | 2018-01-18 17:12:18 | [diff] [blame] | 99 | network::mojom::URLLoaderRequest mojo_request, |
| 100 | network::mojom::URLLoaderClientPtr url_loader_client, |
arthursonzogni | 2e1524a7 | 2017-12-18 16:53:26 | [diff] [blame] | 101 | ResourceType resource_type, |
arthursonzogni | b521c6a | 2018-01-08 12:23:40 | [diff] [blame] | 102 | uint32_t url_loader_options) |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 103 | : ResourceHandler(request), |
| 104 | rdh_(rdh), |
| 105 | binding_(this, std::move(mojo_request)), |
arthursonzogni | b521c6a | 2018-01-08 12:23:40 | [diff] [blame] | 106 | url_loader_options_(url_loader_options), |
Hajime Hoshi | 1c97b84 | 2018-03-16 07:10:10 | [diff] [blame] | 107 | handle_watcher_(FROM_HERE, |
| 108 | mojo::SimpleWatcher::ArmingPolicy::MANUAL, |
| 109 | base::SequencedTaskRunnerHandle::Get()), |
yhirano | bbc904d0 | 2016-11-26 05:59:26 | [diff] [blame] | 110 | url_loader_client_(std::move(url_loader_client)), |
| 111 | weak_factory_(this) { |
arthursonzogni | b521c6a | 2018-01-08 12:23:40 | [diff] [blame] | 112 | DCHECK(IsResourceTypeFrame(resource_type) || |
John Abd-El-Malek | b165dc5 | 2018-01-18 17:12:18 | [diff] [blame] | 113 | !(url_loader_options_ & |
| 114 | network::mojom::kURLLoadOptionSendSSLInfoWithResponse)); |
arthursonzogni | b521c6a | 2018-01-08 12:23:40 | [diff] [blame] | 115 | DCHECK(resource_type == RESOURCE_TYPE_MAIN_FRAME || |
| 116 | !(url_loader_options_ & |
John Abd-El-Malek | b165dc5 | 2018-01-18 17:12:18 | [diff] [blame] | 117 | network::mojom::kURLLoadOptionSendSSLInfoForCertificateError)); |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 118 | DCHECK(url_loader_client_); |
| 119 | InitializeResourceBufferConstants(); |
yhirano | 5939370 | 2016-11-16 06:10:24 | [diff] [blame] | 120 | // This unretained pointer is safe, because |binding_| is owned by |this| and |
| 121 | // the callback will never be called after |this| is destroyed. |
Yuzhu Shen | 07e9631 | 2018-02-05 23:20:29 | [diff] [blame] | 122 | binding_.set_connection_error_with_reason_handler(base::BindOnce( |
tzik | 0f14f19 | 2017-08-15 02:43:33 | [diff] [blame] | 123 | &MojoAsyncResourceHandler::Cancel, base::Unretained(this))); |
yhirano | bbc904d0 | 2016-11-26 05:59:26 | [diff] [blame] | 124 | |
tzik | ad092abe | 2017-01-24 02:34:17 | [diff] [blame] | 125 | if (IsResourceTypeFrame(resource_type)) { |
| 126 | GetRequestInfo()->set_on_transfer(base::Bind( |
| 127 | &MojoAsyncResourceHandler::OnTransfer, weak_factory_.GetWeakPtr())); |
| 128 | } else { |
| 129 | GetRequestInfo()->set_on_transfer(base::Bind(&NotReached)); |
| 130 | } |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 131 | } |
| 132 | |
| 133 | MojoAsyncResourceHandler::~MojoAsyncResourceHandler() { |
| 134 | if (has_checked_for_sufficient_resources_) |
| 135 | rdh_->FinishedWithResourcesForRequest(request()); |
| 136 | } |
| 137 | |
Daniel Bratell | f77777a | 2017-10-19 09:12:06 | [diff] [blame] | 138 | void MojoAsyncResourceHandler::InitializeResourceBufferConstants() { |
| 139 | static bool did_init = false; |
| 140 | if (did_init) |
| 141 | return; |
| 142 | did_init = true; |
| 143 | |
| 144 | GetNumericArg("resource-buffer-size", &g_allocation_size); |
| 145 | } |
| 146 | |
mmenke | 87f5c77a | 2017-01-31 16:11:26 | [diff] [blame] | 147 | void MojoAsyncResourceHandler::OnRequestRedirected( |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 148 | const net::RedirectInfo& redirect_info, |
John Abd-El-Malek | 4624803 | 2018-01-17 19:11:23 | [diff] [blame] | 149 | network::ResourceResponse* response, |
mmenke | 87f5c77a | 2017-01-31 16:11:26 | [diff] [blame] | 150 | std::unique_ptr<ResourceController> controller) { |
yhirano | 7153dc47 | 2016-11-16 09:42:46 | [diff] [blame] | 151 | // Unlike OnResponseStarted, OnRequestRedirected will NOT be preceded by |
| 152 | // OnWillRead. |
mmenke | 87f5c77a | 2017-01-31 16:11:26 | [diff] [blame] | 153 | DCHECK(!has_controller()); |
yhirano | 7153dc47 | 2016-11-16 09:42:46 | [diff] [blame] | 154 | DCHECK(!shared_writer_); |
| 155 | |
yhirano | 7153dc47 | 2016-11-16 09:42:46 | [diff] [blame] | 156 | request()->LogBlockedBy("MojoAsyncResourceHandler"); |
mmenke | 87f5c77a | 2017-01-31 16:11:26 | [diff] [blame] | 157 | HoldController(std::move(controller)); |
yhirano | 7153dc47 | 2016-11-16 09:42:46 | [diff] [blame] | 158 | did_defer_on_redirect_ = true; |
| 159 | |
yhirano | 7153dc47 | 2016-11-16 09:42:46 | [diff] [blame] | 160 | response->head.encoded_data_length = request()->GetTotalReceivedBytes(); |
| 161 | response->head.request_start = request()->creation_time(); |
| 162 | response->head.response_start = base::TimeTicks::Now(); |
| 163 | // TODO(davidben): Is it necessary to pass the new first party URL for |
| 164 | // cookies? The only case where it can change is top-level navigation requests |
| 165 | // and hopefully those will eventually all be owned by the browser. It's |
| 166 | // possible this is still needed while renderer-owned ones exist. |
| 167 | url_loader_client_->OnReceiveRedirect(redirect_info, response->head); |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 168 | } |
| 169 | |
mmenke | 87f5c77a | 2017-01-31 16:11:26 | [diff] [blame] | 170 | void MojoAsyncResourceHandler::OnResponseStarted( |
John Abd-El-Malek | 4624803 | 2018-01-17 19:11:23 | [diff] [blame] | 171 | network::ResourceResponse* response, |
mmenke | 87f5c77a | 2017-01-31 16:11:26 | [diff] [blame] | 172 | std::unique_ptr<ResourceController> controller) { |
| 173 | DCHECK(!has_controller()); |
arthursonzogni | 84dda21 | 2018-03-01 02:06:36 | [diff] [blame] | 174 | time_response_started_ = base::TimeTicks::Now(); |
mmenke | 87f5c77a | 2017-01-31 16:11:26 | [diff] [blame] | 175 | |
tzik | bedcbce | 2017-01-24 15:08:05 | [diff] [blame] | 176 | if (upload_progress_tracker_) { |
| 177 | upload_progress_tracker_->OnUploadCompleted(); |
| 178 | upload_progress_tracker_ = nullptr; |
| 179 | } |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 180 | |
tzik | bedcbce | 2017-01-24 15:08:05 | [diff] [blame] | 181 | const ResourceRequestInfoImpl* info = GetRequestInfo(); |
yhirano | a275b3c | 2016-12-20 03:58:40 | [diff] [blame] | 182 | response->head.encoded_data_length = request()->raw_header_size(); |
| 183 | reported_total_received_bytes_ = response->head.encoded_data_length; |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 184 | |
| 185 | response->head.request_start = request()->creation_time(); |
arthursonzogni | 84dda21 | 2018-03-01 02:06:36 | [diff] [blame] | 186 | response->head.response_start = time_response_started_; |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 187 | sent_received_response_message_ = true; |
tzik | 47a9f9b | 2016-12-01 04:16:24 | [diff] [blame] | 188 | |
John Abd-El-Malek | b165dc5 | 2018-01-18 17:12:18 | [diff] [blame] | 189 | network::mojom::DownloadedTempFilePtr downloaded_file_ptr; |
tzik | 47a9f9b | 2016-12-01 04:16:24 | [diff] [blame] | 190 | if (!response->head.download_file_path.empty()) { |
yzshen | 20d07510 | 2017-02-16 21:36:44 | [diff] [blame] | 191 | downloaded_file_ptr = DownloadedTempFileImpl::Create(info->GetChildID(), |
| 192 | info->GetRequestID()); |
tzik | 47a9f9b | 2016-12-01 04:16:24 | [diff] [blame] | 193 | rdh_->RegisterDownloadedTempFile(info->GetChildID(), info->GetRequestID(), |
| 194 | response->head.download_file_path); |
| 195 | } |
| 196 | |
arthursonzogni | b521c6a | 2018-01-08 12:23:40 | [diff] [blame] | 197 | base::Optional<net::SSLInfo> ssl_info; |
John Abd-El-Malek | b165dc5 | 2018-01-18 17:12:18 | [diff] [blame] | 198 | if (url_loader_options_ & |
| 199 | network::mojom::kURLLoadOptionSendSSLInfoWithResponse) |
arthursonzogni | b521c6a | 2018-01-08 12:23:40 | [diff] [blame] | 200 | ssl_info = request()->ssl_info(); |
| 201 | |
| 202 | url_loader_client_->OnReceiveResponse(response->head, std::move(ssl_info), |
tzik | 47a9f9b | 2016-12-01 04:16:24 | [diff] [blame] | 203 | std::move(downloaded_file_ptr)); |
yhirano | 2a525f9 | 2017-01-18 08:16:46 | [diff] [blame] | 204 | |
| 205 | net::IOBufferWithSize* metadata = GetResponseMetadata(request()); |
| 206 | if (metadata) { |
| 207 | const uint8_t* data = reinterpret_cast<const uint8_t*>(metadata->data()); |
| 208 | |
| 209 | url_loader_client_->OnReceiveCachedMetadata( |
| 210 | std::vector<uint8_t>(data, data + metadata->size())); |
| 211 | } |
mmenke | 87f5c77a | 2017-01-31 16:11:26 | [diff] [blame] | 212 | |
John Abd-El-Malek | b165dc5 | 2018-01-18 17:12:18 | [diff] [blame] | 213 | if (url_loader_options_ & |
| 214 | network::mojom::kURLLoadOptionPauseOnResponseStarted) { |
arthursonzogni | 2e1524a7 | 2017-12-18 16:53:26 | [diff] [blame] | 215 | did_defer_on_response_started_ = true; |
| 216 | DCHECK(!has_controller()); |
| 217 | request()->LogBlockedBy("MojoAsyncResourceHandler"); |
| 218 | HoldController(std::move(controller)); |
| 219 | return; |
| 220 | } |
| 221 | |
mmenke | 87f5c77a | 2017-01-31 16:11:26 | [diff] [blame] | 222 | controller->Resume(); |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 223 | } |
| 224 | |
mmenke | 87f5c77a | 2017-01-31 16:11:26 | [diff] [blame] | 225 | void MojoAsyncResourceHandler::OnWillStart( |
| 226 | const GURL& url, |
| 227 | std::unique_ptr<ResourceController> controller) { |
tzik | bedcbce | 2017-01-24 15:08:05 | [diff] [blame] | 228 | if (GetRequestInfo()->is_upload_progress_enabled() && |
| 229 | request()->has_upload()) { |
| 230 | upload_progress_tracker_ = CreateUploadProgressTracker( |
| 231 | FROM_HERE, |
| 232 | base::BindRepeating(&MojoAsyncResourceHandler::SendUploadProgress, |
| 233 | base::Unretained(this))); |
| 234 | } |
| 235 | |
mmenke | 87f5c77a | 2017-01-31 16:11:26 | [diff] [blame] | 236 | controller->Resume(); |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 237 | } |
| 238 | |
mmenke | 3c1d10c | 2017-03-09 16:25:45 | [diff] [blame] | 239 | void MojoAsyncResourceHandler::OnWillRead( |
| 240 | scoped_refptr<net::IOBuffer>* buf, |
| 241 | int* buf_size, |
| 242 | std::unique_ptr<ResourceController> controller) { |
mmenke | 6add172 | 2017-03-21 21:31:17 | [diff] [blame] | 243 | // |buffer_| is set to nullptr on successful read completion (Except for the |
| 244 | // final 0-byte read, so this DCHECK will also catch OnWillRead being called |
| 245 | // after OnReadCompelted(0)). |
| 246 | DCHECK(!buffer_); |
| 247 | DCHECK_EQ(0u, buffer_offset_); |
| 248 | |
mmenke | 3c1d10c | 2017-03-09 16:25:45 | [diff] [blame] | 249 | if (!CheckForSufficientResource()) { |
| 250 | controller->CancelWithError(net::ERR_INSUFFICIENT_RESOURCES); |
| 251 | return; |
| 252 | } |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 253 | |
mmenke | 6add172 | 2017-03-21 21:31:17 | [diff] [blame] | 254 | bool first_call = false; |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 255 | if (!shared_writer_) { |
mmenke | 6add172 | 2017-03-21 21:31:17 | [diff] [blame] | 256 | first_call = true; |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 257 | MojoCreateDataPipeOptions options; |
| 258 | options.struct_size = sizeof(MojoCreateDataPipeOptions); |
| 259 | options.flags = MOJO_CREATE_DATA_PIPE_OPTIONS_FLAG_NONE; |
| 260 | options.element_num_bytes = 1; |
| 261 | options.capacity_num_bytes = g_allocation_size; |
Ken Rockot | d61caeda | 2017-05-01 22:52:22 | [diff] [blame] | 262 | mojo::ScopedDataPipeProducerHandle producer; |
| 263 | mojo::ScopedDataPipeConsumerHandle consumer; |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 264 | |
Ken Rockot | d61caeda | 2017-05-01 22:52:22 | [diff] [blame] | 265 | MojoResult result = mojo::CreateDataPipe(&options, &producer, &consumer); |
| 266 | if (result != MOJO_RESULT_OK) { |
| 267 | controller->CancelWithError(net::ERR_INSUFFICIENT_RESOURCES); |
| 268 | return; |
| 269 | } |
| 270 | DCHECK(producer.is_valid()); |
| 271 | DCHECK(consumer.is_valid()); |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 272 | |
Ken Rockot | d61caeda | 2017-05-01 22:52:22 | [diff] [blame] | 273 | response_body_consumer_handle_ = std::move(consumer); |
| 274 | shared_writer_ = new SharedWriter(std::move(producer)); |
rockot | 9eadaba | 2017-03-15 23:57:47 | [diff] [blame] | 275 | handle_watcher_.Watch(shared_writer_->writer(), MOJO_HANDLE_SIGNAL_WRITABLE, |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 276 | base::Bind(&MojoAsyncResourceHandler::OnWritable, |
| 277 | base::Unretained(this))); |
rockot | 9eadaba | 2017-03-15 23:57:47 | [diff] [blame] | 278 | handle_watcher_.ArmOrNotify(); |
mmenke | 6add172 | 2017-03-21 21:31:17 | [diff] [blame] | 279 | } |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 280 | |
mmenke | 6add172 | 2017-03-21 21:31:17 | [diff] [blame] | 281 | bool defer = false; |
| 282 | if (!AllocateWriterIOBuffer(&buffer_, &defer)) { |
| 283 | controller->CancelWithError(net::ERR_INSUFFICIENT_RESOURCES); |
| 284 | return; |
| 285 | } |
| 286 | |
| 287 | if (defer) { |
| 288 | DCHECK(!buffer_); |
| 289 | parent_buffer_ = buf; |
| 290 | parent_buffer_size_ = buf_size; |
| 291 | HoldController(std::move(controller)); |
| 292 | request()->LogBlockedBy("MojoAsyncResourceHandler"); |
| 293 | did_defer_on_will_read_ = true; |
| 294 | return; |
| 295 | } |
| 296 | |
| 297 | // The first call to OnWillRead must return a buffer of at least |
| 298 | // kMinAllocationSize. If the Mojo buffer is too small, need to allocate an |
| 299 | // intermediary buffer. |
| 300 | if (first_call && static_cast<size_t>(buffer_->size()) < kMinAllocationSize) { |
| 301 | // The allocated buffer is too small, so need to create an intermediary one. |
| 302 | if (EndWrite(0) != MOJO_RESULT_OK) { |
mmenke | 3c1d10c | 2017-03-09 16:25:45 | [diff] [blame] | 303 | controller->CancelWithError(net::ERR_INSUFFICIENT_RESOURCES); |
| 304 | return; |
| 305 | } |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 306 | DCHECK(!is_using_io_buffer_not_from_writer_); |
| 307 | is_using_io_buffer_not_from_writer_ = true; |
| 308 | buffer_ = new net::IOBufferWithSize(kMinAllocationSize); |
| 309 | } |
| 310 | |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 311 | *buf = buffer_; |
| 312 | *buf_size = buffer_->size(); |
mmenke | 3c1d10c | 2017-03-09 16:25:45 | [diff] [blame] | 313 | controller->Resume(); |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 314 | } |
| 315 | |
mmenke | 87f5c77a | 2017-01-31 16:11:26 | [diff] [blame] | 316 | void MojoAsyncResourceHandler::OnReadCompleted( |
| 317 | int bytes_read, |
| 318 | std::unique_ptr<ResourceController> controller) { |
| 319 | DCHECK(!has_controller()); |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 320 | DCHECK_GE(bytes_read, 0); |
| 321 | DCHECK(buffer_); |
| 322 | |
mmenke | 6add172 | 2017-03-21 21:31:17 | [diff] [blame] | 323 | if (bytes_read == 0) { |
| 324 | // Note that |buffer_| is not cleared here, which will cause a DCHECK on |
| 325 | // subsequent OnWillRead calls. |
mmenke | 87f5c77a | 2017-01-31 16:11:26 | [diff] [blame] | 326 | controller->Resume(); |
| 327 | return; |
| 328 | } |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 329 | |
yhirano | a275b3c | 2016-12-20 03:58:40 | [diff] [blame] | 330 | const ResourceRequestInfoImpl* info = GetRequestInfo(); |
| 331 | if (info->ShouldReportRawHeaders()) { |
| 332 | auto transfer_size_diff = CalculateRecentlyReceivedBytes(); |
| 333 | if (transfer_size_diff > 0) |
| 334 | url_loader_client_->OnTransferSizeUpdated(transfer_size_diff); |
| 335 | } |
| 336 | |
yhirano | 246883ff6 | 2017-01-19 04:19:08 | [diff] [blame] | 337 | if (response_body_consumer_handle_.is_valid()) { |
arthursonzogni | 84dda21 | 2018-03-01 02:06:36 | [diff] [blame] | 338 | if (url_loader_options_ & |
| 339 | network::mojom::kURLLoadOptionPauseOnResponseStarted) { |
| 340 | base::TimeTicks time_first_read_completed = base::TimeTicks::Now(); |
| 341 | RecordNavigationResourceHandlerMetrics(time_response_started_, |
| 342 | time_proceed_with_response_, |
| 343 | time_first_read_completed); |
| 344 | } |
yhirano | 246883ff6 | 2017-01-19 04:19:08 | [diff] [blame] | 345 | // Send the data pipe on the first OnReadCompleted call. |
| 346 | url_loader_client_->OnStartLoadingResponseBody( |
| 347 | std::move(response_body_consumer_handle_)); |
| 348 | response_body_consumer_handle_.reset(); |
| 349 | } |
| 350 | |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 351 | if (is_using_io_buffer_not_from_writer_) { |
mmenke | 6add172 | 2017-03-21 21:31:17 | [diff] [blame] | 352 | // Couldn't allocate a large enough buffer on the data pipe in OnWillRead. |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 353 | DCHECK_EQ(0u, buffer_bytes_read_); |
| 354 | buffer_bytes_read_ = bytes_read; |
mmenke | 87f5c77a | 2017-01-31 16:11:26 | [diff] [blame] | 355 | bool defer = false; |
| 356 | if (!CopyReadDataToDataPipe(&defer)) { |
mmenke | 6add172 | 2017-03-21 21:31:17 | [diff] [blame] | 357 | controller->CancelWithError(net::ERR_INSUFFICIENT_RESOURCES); |
mmenke | 87f5c77a | 2017-01-31 16:11:26 | [diff] [blame] | 358 | return; |
| 359 | } |
| 360 | if (defer) { |
yhirano | 7153dc47 | 2016-11-16 09:42:46 | [diff] [blame] | 361 | request()->LogBlockedBy("MojoAsyncResourceHandler"); |
| 362 | did_defer_on_writing_ = true; |
mmenke | 87f5c77a | 2017-01-31 16:11:26 | [diff] [blame] | 363 | HoldController(std::move(controller)); |
| 364 | return; |
yhirano | 7153dc47 | 2016-11-16 09:42:46 | [diff] [blame] | 365 | } |
mmenke | 87f5c77a | 2017-01-31 16:11:26 | [diff] [blame] | 366 | controller->Resume(); |
| 367 | return; |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 368 | } |
| 369 | |
mmenke | 87f5c77a | 2017-01-31 16:11:26 | [diff] [blame] | 370 | if (EndWrite(bytes_read) != MOJO_RESULT_OK) { |
| 371 | controller->Cancel(); |
| 372 | return; |
| 373 | } |
mmenke | 87f5c77a | 2017-01-31 16:11:26 | [diff] [blame] | 374 | |
mmenke | 6add172 | 2017-03-21 21:31:17 | [diff] [blame] | 375 | buffer_ = nullptr; |
mmenke | 87f5c77a | 2017-01-31 16:11:26 | [diff] [blame] | 376 | controller->Resume(); |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 377 | } |
| 378 | |
| 379 | void MojoAsyncResourceHandler::OnDataDownloaded(int bytes_downloaded) { |
yhirano | a275b3c | 2016-12-20 03:58:40 | [diff] [blame] | 380 | url_loader_client_->OnDataDownloaded(bytes_downloaded, |
| 381 | CalculateRecentlyReceivedBytes()); |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 382 | } |
| 383 | |
| 384 | void MojoAsyncResourceHandler::FollowRedirect() { |
yhirano | 7153dc47 | 2016-11-16 09:42:46 | [diff] [blame] | 385 | if (!request()->status().is_success()) { |
| 386 | DVLOG(1) << "FollowRedirect for invalid request"; |
| 387 | return; |
| 388 | } |
| 389 | if (!did_defer_on_redirect_) { |
| 390 | DVLOG(1) << "Malformed FollowRedirect request"; |
| 391 | ReportBadMessage("Malformed FollowRedirect request"); |
| 392 | return; |
| 393 | } |
| 394 | |
mmenke | 6add172 | 2017-03-21 21:31:17 | [diff] [blame] | 395 | DCHECK(!did_defer_on_will_read_); |
yhirano | 7153dc47 | 2016-11-16 09:42:46 | [diff] [blame] | 396 | DCHECK(!did_defer_on_writing_); |
| 397 | did_defer_on_redirect_ = false; |
| 398 | request()->LogUnblocked(); |
mmenke | 87f5c77a | 2017-01-31 16:11:26 | [diff] [blame] | 399 | Resume(); |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 400 | } |
| 401 | |
arthursonzogni | 2e1524a7 | 2017-12-18 16:53:26 | [diff] [blame] | 402 | void MojoAsyncResourceHandler::ProceedWithResponse() { |
| 403 | DCHECK(did_defer_on_response_started_); |
arthursonzogni | 84dda21 | 2018-03-01 02:06:36 | [diff] [blame] | 404 | |
| 405 | time_proceed_with_response_ = base::TimeTicks::Now(); |
| 406 | |
arthursonzogni | 2e1524a7 | 2017-12-18 16:53:26 | [diff] [blame] | 407 | request()->LogUnblocked(); |
| 408 | Resume(); |
| 409 | } |
| 410 | |
yhirano | 933ee58 | 2017-01-27 13:28:40 | [diff] [blame] | 411 | void MojoAsyncResourceHandler::SetPriority(net::RequestPriority priority, |
| 412 | int32_t intra_priority_value) { |
John Abd-El-Malek | eb1a538 | 2018-01-05 16:58:00 | [diff] [blame] | 413 | auto* scheduler = ResourceDispatcherHostImpl::Get()->scheduler(); |
| 414 | if (intra_priority_value == -1) { |
| 415 | scheduler->ReprioritizeRequest(request(), priority); |
| 416 | } else { |
| 417 | scheduler->ReprioritizeRequest(request(), priority, intra_priority_value); |
| 418 | } |
yhirano | 933ee58 | 2017-01-27 13:28:40 | [diff] [blame] | 419 | } |
| 420 | |
Yuzhu Shen | 99681b1 | 2017-09-28 06:11:10 | [diff] [blame] | 421 | void MojoAsyncResourceHandler::PauseReadingBodyFromNet() { |
Yuzhu Shen | 19cafda0 | 2018-02-01 19:45:20 | [diff] [blame] | 422 | ResourceHandler::PauseReadingBodyFromNet(); |
Yuzhu Shen | 0bec954 | 2017-09-16 01:31:59 | [diff] [blame] | 423 | } |
| 424 | |
Yuzhu Shen | 99681b1 | 2017-09-28 06:11:10 | [diff] [blame] | 425 | void MojoAsyncResourceHandler::ResumeReadingBodyFromNet() { |
Yuzhu Shen | 19cafda0 | 2018-02-01 19:45:20 | [diff] [blame] | 426 | ResourceHandler::ResumeReadingBodyFromNet(); |
Yuzhu Shen | 0bec954 | 2017-09-16 01:31:59 | [diff] [blame] | 427 | } |
| 428 | |
yhirano | 7153dc47 | 2016-11-16 09:42:46 | [diff] [blame] | 429 | void MojoAsyncResourceHandler::OnWritableForTesting() { |
| 430 | OnWritable(MOJO_RESULT_OK); |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 431 | } |
| 432 | |
| 433 | void MojoAsyncResourceHandler::SetAllocationSizeForTesting(size_t size) { |
| 434 | g_allocation_size = size; |
| 435 | } |
| 436 | |
| 437 | MojoResult MojoAsyncResourceHandler::BeginWrite(void** data, |
| 438 | uint32_t* available) { |
Reilly Grant | 789c63e | 2017-08-04 15:30:16 | [diff] [blame] | 439 | MojoResult result = shared_writer_->writer().BeginWriteData( |
| 440 | data, available, MOJO_WRITE_DATA_FLAG_NONE); |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 441 | if (result == MOJO_RESULT_OK) |
| 442 | *available = std::min(*available, static_cast<uint32_t>(kMaxChunkSize)); |
rockot | 9eadaba | 2017-03-15 23:57:47 | [diff] [blame] | 443 | else if (result == MOJO_RESULT_SHOULD_WAIT) |
| 444 | handle_watcher_.ArmOrNotify(); |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 445 | return result; |
| 446 | } |
| 447 | |
| 448 | MojoResult MojoAsyncResourceHandler::EndWrite(uint32_t written) { |
Reilly Grant | 789c63e | 2017-08-04 15:30:16 | [diff] [blame] | 449 | MojoResult result = shared_writer_->writer().EndWriteData(written); |
horo | e8442e6 | 2017-04-27 19:10:48 | [diff] [blame] | 450 | if (result == MOJO_RESULT_OK) { |
| 451 | total_written_bytes_ += written; |
rockot | 9eadaba | 2017-03-15 23:57:47 | [diff] [blame] | 452 | handle_watcher_.ArmOrNotify(); |
horo | e8442e6 | 2017-04-27 19:10:48 | [diff] [blame] | 453 | } |
rockot | 9eadaba | 2017-03-15 23:57:47 | [diff] [blame] | 454 | return result; |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 455 | } |
| 456 | |
yhirano | 2a525f9 | 2017-01-18 08:16:46 | [diff] [blame] | 457 | net::IOBufferWithSize* MojoAsyncResourceHandler::GetResponseMetadata( |
| 458 | net::URLRequest* request) { |
| 459 | return request->response_info().metadata.get(); |
| 460 | } |
| 461 | |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 462 | void MojoAsyncResourceHandler::OnResponseCompleted( |
Takashi Toyoshima | 8f98853 | 2017-11-13 07:32:37 | [diff] [blame] | 463 | const net::URLRequestStatus& request_status, |
mmenke | 87f5c77a | 2017-01-31 16:11:26 | [diff] [blame] | 464 | std::unique_ptr<ResourceController> controller) { |
tzik | bedcbce | 2017-01-24 15:08:05 | [diff] [blame] | 465 | // Ensure sending the final upload progress message here, since |
| 466 | // OnResponseCompleted can be called without OnResponseStarted on cancellation |
| 467 | // or error cases. |
| 468 | if (upload_progress_tracker_) { |
| 469 | upload_progress_tracker_->OnUploadCompleted(); |
| 470 | upload_progress_tracker_ = nullptr; |
| 471 | } |
| 472 | |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 473 | shared_writer_ = nullptr; |
| 474 | buffer_ = nullptr; |
| 475 | handle_watcher_.Cancel(); |
| 476 | |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 477 | // TODO(gavinp): Remove this CHECK when we figure out the cause of |
| 478 | // http://crbug.com/124680 . This check mirrors closely check in |
| 479 | // WebURLLoaderImpl::OnCompletedRequest that routes this message to a WebCore |
| 480 | // ResourceHandleInternal which asserts on its state and crashes. By crashing |
| 481 | // when the message is sent, we should get better crash reports. |
Takashi Toyoshima | 8f98853 | 2017-11-13 07:32:37 | [diff] [blame] | 482 | CHECK(request_status.status() != net::URLRequestStatus::SUCCESS || |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 483 | sent_received_response_message_); |
| 484 | |
Takashi Toyoshima | 8f98853 | 2017-11-13 07:32:37 | [diff] [blame] | 485 | int error_code = request_status.error(); |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 486 | |
Takashi Toyoshima | 8f98853 | 2017-11-13 07:32:37 | [diff] [blame] | 487 | DCHECK_NE(request_status.status(), net::URLRequestStatus::IO_PENDING); |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 488 | |
Takashi Toyoshima | aa27866 | 2017-11-20 11:11:26 | [diff] [blame] | 489 | network::URLLoaderCompletionStatus loader_status; |
Takashi Toyoshima | 8f98853 | 2017-11-13 07:32:37 | [diff] [blame] | 490 | loader_status.error_code = error_code; |
Brad Lassey | 16c13f7 | 2018-03-19 19:37:50 | [diff] [blame^] | 491 | if (error_code == net::ERR_QUIC_PROTOCOL_ERROR) { |
| 492 | net::NetErrorDetails details; |
| 493 | request()->PopulateNetErrorDetails(&details); |
| 494 | loader_status.extended_error_code = details.quic_connection_error; |
| 495 | } |
Arthur Sonzogni | 0b2b1a92 | 2018-03-15 15:24:15 | [diff] [blame] | 496 | loader_status.exists_in_cache = request()->response_info().was_cached; |
Takashi Toyoshima | 8f98853 | 2017-11-13 07:32:37 | [diff] [blame] | 497 | loader_status.completion_time = base::TimeTicks::Now(); |
| 498 | loader_status.encoded_data_length = request()->GetTotalReceivedBytes(); |
| 499 | loader_status.encoded_body_length = request()->GetRawBodyBytes(); |
| 500 | loader_status.decoded_body_length = total_written_bytes_; |
Pavel Feldman | 8f36166 | 2018-01-03 01:55:07 | [diff] [blame] | 501 | loader_status.blocked_cross_site_document = |
| 502 | GetRequestInfo()->blocked_cross_site_document(); |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 503 | |
arthursonzogni | b521c6a | 2018-01-08 12:23:40 | [diff] [blame] | 504 | if ((url_loader_options_ & |
John Abd-El-Malek | b165dc5 | 2018-01-18 17:12:18 | [diff] [blame] | 505 | network::mojom::kURLLoadOptionSendSSLInfoForCertificateError) && |
arthursonzogni | b521c6a | 2018-01-08 12:23:40 | [diff] [blame] | 506 | net::IsCertStatusError(request()->ssl_info().cert_status) && |
| 507 | !net::IsCertStatusMinorError(request()->ssl_info().cert_status)) { |
| 508 | loader_status.ssl_info = request()->ssl_info(); |
| 509 | } |
| 510 | |
Takashi Toyoshima | 8f98853 | 2017-11-13 07:32:37 | [diff] [blame] | 511 | url_loader_client_->OnComplete(loader_status); |
mmenke | 87f5c77a | 2017-01-31 16:11:26 | [diff] [blame] | 512 | controller->Resume(); |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 513 | } |
| 514 | |
| 515 | bool MojoAsyncResourceHandler::CopyReadDataToDataPipe(bool* defer) { |
mmenke | 6add172 | 2017-03-21 21:31:17 | [diff] [blame] | 516 | while (buffer_bytes_read_ > 0) { |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 517 | scoped_refptr<net::IOBufferWithSize> dest; |
| 518 | if (!AllocateWriterIOBuffer(&dest, defer)) |
| 519 | return false; |
| 520 | if (*defer) |
| 521 | return true; |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 522 | |
| 523 | size_t copied_size = |
| 524 | std::min(buffer_bytes_read_, static_cast<size_t>(dest->size())); |
| 525 | memcpy(dest->data(), buffer_->data() + buffer_offset_, copied_size); |
| 526 | buffer_offset_ += copied_size; |
| 527 | buffer_bytes_read_ -= copied_size; |
| 528 | if (EndWrite(copied_size) != MOJO_RESULT_OK) |
| 529 | return false; |
yhirano | 4fede0f | 2017-03-21 03:33:15 | [diff] [blame] | 530 | } |
mmenke | 6add172 | 2017-03-21 21:31:17 | [diff] [blame] | 531 | |
| 532 | // All bytes are copied. |
| 533 | buffer_ = nullptr; |
| 534 | buffer_offset_ = 0; |
| 535 | is_using_io_buffer_not_from_writer_ = false; |
| 536 | return true; |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 537 | } |
| 538 | |
| 539 | bool MojoAsyncResourceHandler::AllocateWriterIOBuffer( |
| 540 | scoped_refptr<net::IOBufferWithSize>* buf, |
| 541 | bool* defer) { |
| 542 | void* data = nullptr; |
| 543 | uint32_t available = 0; |
| 544 | MojoResult result = BeginWrite(&data, &available); |
| 545 | if (result == MOJO_RESULT_SHOULD_WAIT) { |
| 546 | *defer = true; |
| 547 | return true; |
| 548 | } |
| 549 | if (result != MOJO_RESULT_OK) |
| 550 | return false; |
mmenke | 6add172 | 2017-03-21 21:31:17 | [diff] [blame] | 551 | DCHECK_GT(available, 0u); |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 552 | *buf = new WriterIOBuffer(shared_writer_, data, available); |
| 553 | return true; |
| 554 | } |
| 555 | |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 556 | bool MojoAsyncResourceHandler::CheckForSufficientResource() { |
| 557 | if (has_checked_for_sufficient_resources_) |
| 558 | return true; |
| 559 | has_checked_for_sufficient_resources_ = true; |
| 560 | |
| 561 | if (rdh_->HasSufficientResourcesForRequest(request())) |
| 562 | return true; |
| 563 | |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 564 | return false; |
| 565 | } |
| 566 | |
yhirano | 7153dc47 | 2016-11-16 09:42:46 | [diff] [blame] | 567 | void MojoAsyncResourceHandler::OnWritable(MojoResult result) { |
mmenke | 6add172 | 2017-03-21 21:31:17 | [diff] [blame] | 568 | if (did_defer_on_will_read_) { |
| 569 | DCHECK(has_controller()); |
| 570 | DCHECK(!did_defer_on_writing_); |
| 571 | DCHECK(!did_defer_on_redirect_); |
| 572 | |
| 573 | did_defer_on_will_read_ = false; |
| 574 | |
| 575 | scoped_refptr<net::IOBuffer>* parent_buffer = parent_buffer_; |
| 576 | parent_buffer_ = nullptr; |
| 577 | int* parent_buffer_size = parent_buffer_size_; |
| 578 | parent_buffer_size_ = nullptr; |
| 579 | |
| 580 | request()->LogUnblocked(); |
| 581 | OnWillRead(parent_buffer, parent_buffer_size, ReleaseController()); |
| 582 | return; |
| 583 | } |
| 584 | |
yhirano | 7153dc47 | 2016-11-16 09:42:46 | [diff] [blame] | 585 | if (!did_defer_on_writing_) |
| 586 | return; |
mmenke | 87f5c77a | 2017-01-31 16:11:26 | [diff] [blame] | 587 | DCHECK(has_controller()); |
yhirano | 7153dc47 | 2016-11-16 09:42:46 | [diff] [blame] | 588 | DCHECK(!did_defer_on_redirect_); |
| 589 | did_defer_on_writing_ = false; |
| 590 | |
mmenke | 6add172 | 2017-03-21 21:31:17 | [diff] [blame] | 591 | DCHECK(is_using_io_buffer_not_from_writer_); |
| 592 | // |buffer_| is set to a net::IOBufferWithSize. Write the buffer contents |
| 593 | // to the data pipe. |
| 594 | DCHECK_GT(buffer_bytes_read_, 0u); |
| 595 | if (!CopyReadDataToDataPipe(&did_defer_on_writing_)) { |
| 596 | CancelWithError(net::ERR_INSUFFICIENT_RESOURCES); |
| 597 | return; |
yhirano | 7153dc47 | 2016-11-16 09:42:46 | [diff] [blame] | 598 | } |
| 599 | |
| 600 | if (did_defer_on_writing_) { |
| 601 | // Continue waiting. |
| 602 | return; |
| 603 | } |
| 604 | request()->LogUnblocked(); |
mmenke | 87f5c77a | 2017-01-31 16:11:26 | [diff] [blame] | 605 | Resume(); |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 606 | } |
| 607 | |
Yuzhu Shen | 07e9631 | 2018-02-05 23:20:29 | [diff] [blame] | 608 | void MojoAsyncResourceHandler::Cancel(uint32_t custom_reason, |
| 609 | const std::string& description) { |
| 610 | ResourceRequestInfoImpl* info = GetRequestInfo(); |
| 611 | |
| 612 | if (custom_reason == network::mojom::URLLoader::kClientDisconnectReason) |
| 613 | info->set_custom_cancel_reason(description); |
| 614 | |
yhirano | 5939370 | 2016-11-16 06:10:24 | [diff] [blame] | 615 | ResourceDispatcherHostImpl::Get()->CancelRequestFromRenderer( |
| 616 | GlobalRequestID(info->GetChildID(), info->GetRequestID())); |
| 617 | } |
| 618 | |
yhirano | a275b3c | 2016-12-20 03:58:40 | [diff] [blame] | 619 | int64_t MojoAsyncResourceHandler::CalculateRecentlyReceivedBytes() { |
| 620 | int64_t total_received_bytes = request()->GetTotalReceivedBytes(); |
| 621 | int64_t bytes_to_report = |
| 622 | total_received_bytes - reported_total_received_bytes_; |
| 623 | reported_total_received_bytes_ = total_received_bytes; |
| 624 | DCHECK_LE(0, bytes_to_report); |
| 625 | return bytes_to_report; |
| 626 | } |
| 627 | |
yhirano | 7153dc47 | 2016-11-16 09:42:46 | [diff] [blame] | 628 | void MojoAsyncResourceHandler::ReportBadMessage(const std::string& error) { |
| 629 | mojo::ReportBadMessage(error); |
| 630 | } |
| 631 | |
John Abd-El-Malek | 3bbbdf9 | 2018-01-30 03:27:35 | [diff] [blame] | 632 | std::unique_ptr<network::UploadProgressTracker> |
tzik | bedcbce | 2017-01-24 15:08:05 | [diff] [blame] | 633 | MojoAsyncResourceHandler::CreateUploadProgressTracker( |
Brett Wilson | 1c99002 | 2017-09-12 20:11:15 | [diff] [blame] | 634 | const base::Location& from_here, |
John Abd-El-Malek | 3bbbdf9 | 2018-01-30 03:27:35 | [diff] [blame] | 635 | network::UploadProgressTracker::UploadProgressReportCallback callback) { |
| 636 | return std::make_unique<network::UploadProgressTracker>( |
| 637 | from_here, std::move(callback), request()); |
tzik | bedcbce | 2017-01-24 15:08:05 | [diff] [blame] | 638 | } |
| 639 | |
yhirano | bbc904d0 | 2016-11-26 05:59:26 | [diff] [blame] | 640 | void MojoAsyncResourceHandler::OnTransfer( |
John Abd-El-Malek | b165dc5 | 2018-01-18 17:12:18 | [diff] [blame] | 641 | network::mojom::URLLoaderRequest mojo_request, |
| 642 | network::mojom::URLLoaderClientPtr url_loader_client) { |
yhirano | bbc904d0 | 2016-11-26 05:59:26 | [diff] [blame] | 643 | binding_.Unbind(); |
| 644 | binding_.Bind(std::move(mojo_request)); |
Yuzhu Shen | 07e9631 | 2018-02-05 23:20:29 | [diff] [blame] | 645 | binding_.set_connection_error_with_reason_handler(base::BindOnce( |
tzik | 0f14f19 | 2017-08-15 02:43:33 | [diff] [blame] | 646 | &MojoAsyncResourceHandler::Cancel, base::Unretained(this))); |
yhirano | bbc904d0 | 2016-11-26 05:59:26 | [diff] [blame] | 647 | url_loader_client_ = std::move(url_loader_client); |
| 648 | } |
| 649 | |
tzik | bedcbce | 2017-01-24 15:08:05 | [diff] [blame] | 650 | void MojoAsyncResourceHandler::SendUploadProgress( |
| 651 | const net::UploadProgress& progress) { |
| 652 | url_loader_client_->OnUploadProgress( |
| 653 | progress.position(), progress.size(), |
tzik | 0f14f19 | 2017-08-15 02:43:33 | [diff] [blame] | 654 | base::BindOnce(&MojoAsyncResourceHandler::OnUploadProgressACK, |
| 655 | weak_factory_.GetWeakPtr())); |
tzik | bedcbce | 2017-01-24 15:08:05 | [diff] [blame] | 656 | } |
| 657 | |
| 658 | void MojoAsyncResourceHandler::OnUploadProgressACK() { |
| 659 | if (upload_progress_tracker_) |
| 660 | upload_progress_tracker_->OnAckReceived(); |
| 661 | } |
| 662 | |
yhirano | 72f6227 | 2016-08-13 12:50:06 | [diff] [blame] | 663 | } // namespace content |