| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "components/cronet/android/url_request_adapter.h" | 5 #include "components/cronet/android/url_request_adapter.h" |
| 6 | 6 |
| 7 #include <stddef.h> | 7 #include <stddef.h> |
| 8 #include <string.h> | 8 #include <string.h> |
| 9 #include <utility> | 9 #include <utility> |
| 10 | 10 |
| (...skipping 111 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 122 void URLRequestAdapter::Start() { | 122 void URLRequestAdapter::Start() { |
| 123 context_->PostTaskToNetworkThread( | 123 context_->PostTaskToNetworkThread( |
| 124 FROM_HERE, | 124 FROM_HERE, |
| 125 base::Bind(&URLRequestAdapter::OnInitiateConnection, | 125 base::Bind(&URLRequestAdapter::OnInitiateConnection, |
| 126 base::Unretained(this))); | 126 base::Unretained(this))); |
| 127 } | 127 } |
| 128 | 128 |
| 129 void URLRequestAdapter::OnAppendChunk(const scoped_ptr<char[]> bytes, | 129 void URLRequestAdapter::OnAppendChunk(const scoped_ptr<char[]> bytes, |
| 130 int bytes_len, bool is_last_chunk) { | 130 int bytes_len, bool is_last_chunk) { |
| 131 DCHECK(OnNetworkThread()); | 131 DCHECK(OnNetworkThread()); |
| 132 // Request could have completed and been destroyed on the network thread | 132 // If AppendData returns false, the request has been cancelled or completed |
| 133 // while appendChunk was posting the task from an application thread. | 133 // without uploading the entire request body. Either way, that result will |
| 134 if (!url_request_) { | 134 // have been sent to the embedder, so there's nothing else to do here. |
| 135 VLOG(1) << "Cannot append chunk to destroyed request: " | 135 chunked_upload_writer_->AppendData(bytes.get(), bytes_len, is_last_chunk); |
| 136 << url_.possibly_invalid_spec().c_str(); | |
| 137 return; | |
| 138 } | |
| 139 url_request_->AppendChunkToUpload(bytes.get(), bytes_len, is_last_chunk); | |
| 140 } | 136 } |
| 141 | 137 |
| 142 void URLRequestAdapter::OnInitiateConnection() { | 138 void URLRequestAdapter::OnInitiateConnection() { |
| 143 DCHECK(OnNetworkThread()); | 139 DCHECK(OnNetworkThread()); |
| 144 if (canceled_) { | 140 if (canceled_) { |
| 145 return; | 141 return; |
| 146 } | 142 } |
| 147 | 143 |
| 148 VLOG(1) << "Starting chromium request: " | 144 VLOG(1) << "Starting chromium request: " |
| 149 << url_.possibly_invalid_spec().c_str() | 145 << url_.possibly_invalid_spec().c_str() |
| 150 << " priority: " << RequestPriorityToString(priority_); | 146 << " priority: " << RequestPriorityToString(priority_); |
| 151 url_request_ = context_->GetURLRequestContext()->CreateRequest( | 147 url_request_ = context_->GetURLRequestContext()->CreateRequest( |
| 152 url_, net::DEFAULT_PRIORITY, this); | 148 url_, net::DEFAULT_PRIORITY, this); |
| 153 int flags = net::LOAD_DO_NOT_SAVE_COOKIES | net::LOAD_DO_NOT_SEND_COOKIES; | 149 int flags = net::LOAD_DO_NOT_SAVE_COOKIES | net::LOAD_DO_NOT_SEND_COOKIES; |
| 154 if (context_->load_disable_cache()) | 150 if (context_->load_disable_cache()) |
| 155 flags |= net::LOAD_DISABLE_CACHE; | 151 flags |= net::LOAD_DISABLE_CACHE; |
| 156 url_request_->SetLoadFlags(flags); | 152 url_request_->SetLoadFlags(flags); |
| 157 url_request_->set_method(method_); | 153 url_request_->set_method(method_); |
| 158 url_request_->SetExtraRequestHeaders(headers_); | 154 url_request_->SetExtraRequestHeaders(headers_); |
| 159 if (!headers_.HasHeader(net::HttpRequestHeaders::kUserAgent)) { | 155 if (!headers_.HasHeader(net::HttpRequestHeaders::kUserAgent)) { |
| 160 std::string user_agent; | 156 std::string user_agent; |
| 161 user_agent = context_->GetUserAgent(url_); | 157 user_agent = context_->GetUserAgent(url_); |
| 162 url_request_->SetExtraRequestHeaderByName( | 158 url_request_->SetExtraRequestHeaderByName( |
| 163 net::HttpRequestHeaders::kUserAgent, user_agent, true /* override */); | 159 net::HttpRequestHeaders::kUserAgent, user_agent, true /* override */); |
| 164 } | 160 } |
| 165 | 161 |
| 166 if (upload_data_stream_) { | 162 if (upload_data_stream_) { |
| 167 url_request_->set_upload(std::move(upload_data_stream_)); | 163 url_request_->set_upload(std::move(upload_data_stream_)); |
| 168 } else if (chunked_upload_) { | 164 } else if (chunked_upload_) { |
| 169 url_request_->EnableChunkedUpload(); | 165 scoped_ptr<net::ChunkedUploadDataStream> chunked_upload_data_stream( |
| 166 new net::ChunkedUploadDataStream(0)); |
| 167 // Create a ChunkedUploadDataStream::Writer, which keeps a weak reference to |
| 168 // the UploadDataStream, before passing ownership of the stream to the |
| 169 // URLRequest. |
| 170 chunked_upload_writer_ = chunked_upload_data_stream->CreateWriter(); |
| 171 url_request_->set_upload(std::move(chunked_upload_data_stream)); |
| 170 } | 172 } |
| 171 | 173 |
| 172 url_request_->SetPriority(priority_); | 174 url_request_->SetPriority(priority_); |
| 173 | 175 |
| 174 url_request_->Start(); | 176 url_request_->Start(); |
| 175 } | 177 } |
| 176 | 178 |
| 177 void URLRequestAdapter::Cancel() { | 179 void URLRequestAdapter::Cancel() { |
| 178 context_->PostTaskToNetworkThread( | 180 context_->PostTaskToNetworkThread( |
| 179 FROM_HERE, | 181 FROM_HERE, |
| (...skipping 141 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 321 unsigned char* URLRequestAdapter::Data() const { | 323 unsigned char* URLRequestAdapter::Data() const { |
| 322 DCHECK(OnNetworkThread()); | 324 DCHECK(OnNetworkThread()); |
| 323 return reinterpret_cast<unsigned char*>(read_buffer_->data()); | 325 return reinterpret_cast<unsigned char*>(read_buffer_->data()); |
| 324 } | 326 } |
| 325 | 327 |
| 326 bool URLRequestAdapter::OnNetworkThread() const { | 328 bool URLRequestAdapter::OnNetworkThread() const { |
| 327 return context_->GetNetworkTaskRunner()->BelongsToCurrentThread(); | 329 return context_->GetNetworkTaskRunner()->BelongsToCurrentThread(); |
| 328 } | 330 } |
| 329 | 331 |
| 330 } // namespace cronet | 332 } // namespace cronet |
| OLD | NEW |