Index: components/cronet/android/cronet_url_request_adapter.cc |
diff --git a/components/cronet/android/cronet_url_request_adapter.cc b/components/cronet/android/cronet_url_request_adapter.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..f4edd9079745fba2132e830c0d4ffbec8d28de80 |
--- /dev/null |
+++ b/components/cronet/android/cronet_url_request_adapter.cc |
@@ -0,0 +1,160 @@ |
+// Copyright 2014 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 "cronet_url_request_adapter.h" |
+ |
+#include "base/bind.h" |
+#include "base/location.h" |
+#include "base/logging.h" |
+#include "components/cronet/android/cronet_url_request_context_adapter.h" |
+#include "components/cronet/android/wrapped_channel_upload_element_reader.h" |
+#include "net/base/io_buffer.h" |
+#include "net/base/load_flags.h" |
+#include "net/http/http_status_code.h" |
+#include "net/url_request/redirect_info.h" |
+#include "net/url_request/url_request_context.h" |
+ |
+namespace cronet { |
+ |
+static const int kReadBufferSize = 32768; |
+ |
+CronetURLRequestAdapter::CronetURLRequestAdapter( |
+ CronetURLRequestContextAdapter* context, |
+ scoped_ptr<CronetURLRequestAdapterDelegate> delegate, |
+ const GURL& url, |
+ net::RequestPriority priority) |
+ : context_(context), |
+ delegate_(delegate.Pass()), |
+ initial_url_(url), |
+ initial_priority_(priority), |
+ initial_method_("GET") { |
+} |
+ |
+CronetURLRequestAdapter::~CronetURLRequestAdapter() { |
+ DCHECK(IsOnNetworkThread()); |
+} |
+ |
+void CronetURLRequestAdapter::AddRequestHeader(const std::string& name, |
+ const std::string& value) { |
+ DCHECK(!IsOnNetworkThread()); |
+ initial_request_headers_.SetHeader(name, value); |
+} |
+ |
+bool CronetURLRequestAdapter::PostTaskToNetworkThread( |
+ const tracked_objects::Location& from_here, |
+ const base::Closure& task) { |
+ DCHECK(!IsOnNetworkThread()); |
+ return context_->GetNetworkTaskRunner()->PostTask(from_here, task); |
+} |
+ |
+bool CronetURLRequestAdapter::IsOnNetworkThread() const { |
+ return context_->GetNetworkTaskRunner()->BelongsToCurrentThread(); |
+} |
+ |
+void CronetURLRequestAdapter::Start() { |
+ DCHECK(IsOnNetworkThread()); |
+ VLOG(1) << "Starting chromium request: " |
+ << initial_url_.possibly_invalid_spec().c_str() |
+ << " priority: " << RequestPriorityToString(initial_priority_); |
+ url_request_ = context_->GetURLRequestContext()->CreateRequest( |
+ initial_url_, net::DEFAULT_PRIORITY, this, NULL); |
+ url_request_->SetLoadFlags(net::LOAD_DISABLE_CACHE | |
+ net::LOAD_DO_NOT_SAVE_COOKIES | |
+ net::LOAD_DO_NOT_SEND_COOKIES); |
+ url_request_->set_method(initial_method_); |
+ url_request_->SetExtraRequestHeaders(initial_request_headers_); |
+ url_request_->SetPriority(initial_priority_); |
+ url_request_->Start(); |
+} |
+ |
+void CronetURLRequestAdapter::FollowDeferredRedirect() { |
+ DCHECK(IsOnNetworkThread()); |
+ |
+ url_request_->FollowDeferredRedirect(); |
+} |
+ |
+void CronetURLRequestAdapter::ReadData() { |
+ DCHECK(IsOnNetworkThread()); |
+ if (!read_buffer_.get()) |
+ read_buffer_ = new net::IOBufferWithSize(kReadBufferSize); |
+ |
+ int bytes_read = 0; |
+ url_request_->Read(read_buffer_.get(), read_buffer_->size(), &bytes_read); |
+ // If IO is pending, wait for the URLRequest to call OnReadCompleted. |
+ if (url_request_->status().is_io_pending()) |
+ return; |
+ |
+ OnReadCompleted(url_request_.get(), bytes_read); |
+} |
+ |
+void CronetURLRequestAdapter::Destroy() { |
+ DCHECK(IsOnNetworkThread()); |
+ delete this; |
+} |
+ |
+const net::HttpResponseHeaders* |
+CronetURLRequestAdapter::GetResponseHeaders() const { |
+ DCHECK(IsOnNetworkThread()); |
+ return url_request_->response_headers(); |
+} |
+ |
+const std::string& CronetURLRequestAdapter::GetNegotiatedProtocol() const { |
+ DCHECK(IsOnNetworkThread()); |
+ return url_request_->response_info().npn_negotiated_protocol; |
+} |
+ |
+bool CronetURLRequestAdapter::GetWasCached() const { |
+ DCHECK(IsOnNetworkThread()); |
+ return url_request_->response_info().was_cached; |
+} |
+ |
+int64 CronetURLRequestAdapter::GetTotalReceivedBytes() const { |
+ DCHECK(IsOnNetworkThread()); |
+ return url_request_->GetTotalReceivedBytes(); |
+} |
+ |
+// net::URLRequest::Delegate overrides (called on network thread). |
+ |
+void CronetURLRequestAdapter::OnReceivedRedirect( |
+ net::URLRequest* request, |
+ const net::RedirectInfo& redirect_info, |
+ bool* defer_redirect) { |
+ DCHECK(IsOnNetworkThread()); |
+ DCHECK(request->status().is_success()); |
+ delegate_->OnRedirect(redirect_info.new_url, redirect_info.status_code); |
+ *defer_redirect = true; |
+} |
+ |
+void CronetURLRequestAdapter::OnResponseStarted(net::URLRequest* request) { |
+ DCHECK(IsOnNetworkThread()); |
+ if (MaybeReportError(request)) |
+ return; |
+ delegate_->OnResponseStarted(request->GetResponseCode()); |
+} |
+ |
+void CronetURLRequestAdapter::OnReadCompleted(net::URLRequest* request, |
+ int bytes_read) { |
+ DCHECK(IsOnNetworkThread()); |
+ if (MaybeReportError(request)) |
+ return; |
+ if (bytes_read != 0) { |
+ delegate_->OnBytesRead( |
+ reinterpret_cast<unsigned char*>(read_buffer_->data()), bytes_read); |
+ } else { |
+ delegate_->OnRequestFinished(); |
+ } |
+} |
+ |
+bool CronetURLRequestAdapter::MaybeReportError(net::URLRequest* request) const { |
+ DCHECK_NE(net::URLRequestStatus::IO_PENDING, url_request_->status().status()); |
+ DCHECK_EQ(request, url_request_); |
+ if (url_request_->status().is_success()) |
+ return false; |
+ VLOG(1) << "Error " << url_request_->status().error() |
+ << " on chromium request: " << initial_url_.possibly_invalid_spec(); |
+ delegate_->OnError(url_request_->status().error()); |
+ return true; |
+} |
+ |
+} // namespace cronet |