Initial implementation of Cronet Async API.

components/cronet/android/cronet_url_request_adapter.cc

+// 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 size_t kBufferSizeIncrement = 32768;
+
+CronetURLRequestAdapter::CronetURLRequestAdapter(
+    CronetURLRequestContextAdapter* context,
+    CronetURLRequestAdapterDelegate* delegate,
+    const GURL& url,
+    net::RequestPriority priority)
+    : initial_method_("GET"),
+      read_buffer_(new net::IOBufferWithSize(kBufferSizeIncrement)),
+      http_status_code_(0),
+      called_to_delegate_(false),
+      cancel_soon_(false) {
+  context_ = context;
+  delegate_ = delegate;
+  initial_url_ = url;
+  initial_priority_ = priority;
+}
+
+CronetURLRequestAdapter::~CronetURLRequestAdapter() {
+  url_request_.reset();
+}
+
+void CronetURLRequestAdapter::SetMethod(const std::string& method) {
+  initial_method_ = method;
+}
+
+void CronetURLRequestAdapter::AddRequestHeader(const std::string& name,
+                                               const std::string& value) {
+  initial_request_headers_.SetHeader(name, value);
+}
+
+net::HttpResponseHeaders* CronetURLRequestAdapter::GetResponseHeaders() const {
+  DCHECK(IsOnNetworkThread());
+  if (url_request_ == NULL)
+    return NULL;
+  return url_request_->response_headers();
+}
+
+std::string CronetURLRequestAdapter::GetNegotiatedProtocol() const {
+  DCHECK(IsOnNetworkThread());
+  if (url_request_ == NULL)
+    return std::string();
+  return url_request_->response_info().npn_negotiated_protocol;
+}
+
+bool CronetURLRequestAdapter::GetWasCached() const {
+  DCHECK(IsOnNetworkThread());
+  if (url_request_ == NULL)
+    return false;
+  return url_request_->response_info().was_cached;
+}
+
+int64 CronetURLRequestAdapter::GetTotalReceivedBytes() const {
+  DCHECK(IsOnNetworkThread());
+  if (url_request_ == NULL)
+    return 0;
+  return url_request_->GetTotalReceivedBytes();
+}
+
+void CronetURLRequestAdapter::Start() {
+  DCHECK(!IsOnNetworkThread());
+  context_->GetNetworkTaskRunner()->PostTask(
+      FROM_HERE,
+      base::Bind(&CronetURLRequestAdapter::StartOnNetworkThread,
+                 base::Unretained(this)));
+}
+
+void CronetURLRequestAdapter::FollowDeferredRedirect() {
+  DCHECK(!IsOnNetworkThread());
+  context_->GetNetworkTaskRunner()->PostTask(
+      FROM_HERE,
+      base::Bind(
+          &CronetURLRequestAdapter::FollowDeferredRedirectOnNetworkThread,
+          base::Unretained(this)));
+}
+
+// Receive more data.
+void CronetURLRequestAdapter::ReadData() {
+  DCHECK(!IsOnNetworkThread());
+  context_->GetNetworkTaskRunner()->PostTask(
+      FROM_HERE,
+      base::Bind(&CronetURLRequestAdapter::ReadDataOnNetworkThread,
+                 base::Unretained(this)));
+}
+
+void CronetURLRequestAdapter::Cancel() {
+  DCHECK(!IsOnNetworkThread());
+  context_->GetNetworkTaskRunner()->PostTask(
+      FROM_HERE,
+      base::Bind(&CronetURLRequestAdapter::CancelOnNetworkThread,
+                 base::Unretained(this)));
+}
+
+void CronetURLRequestAdapter::Destroy() {
+  DCHECK(!IsOnNetworkThread());
+  context_->GetNetworkTaskRunner()->DeleteSoon(FROM_HERE, this);
+}
+
+// net::URLRequest::Delegate overrides (called on network thread).
+void CronetURLRequestAdapter::OnReceivedRedirect(
+    net::URLRequest* request,
+    const net::RedirectInfo& redirect_info,
+    bool* defer_redirect) {
+  DCHECK(request->status().is_success());
+  called_to_delegate_ = true;
+  delegate_->OnRedirect(redirect_info.new_url);
+  *defer_redirect = true;
+}
+
+void CronetURLRequestAdapter::OnResponseStarted(net::URLRequest* request) {
+  if (!CheckStatus(request))
+    return;
+  http_status_code_ = request->GetResponseCode();
+  called_to_delegate_ = true;
+  delegate_->OnResponseStarted();
+}
+
+void CronetURLRequestAdapter::OnReadCompleted(net::URLRequest* request,
+                                              int bytes_read) {
+  if (!CheckStatus(request))
+    return;
+  called_to_delegate_ = true;
+  if (bytes_read != 0) {
+    delegate_->OnBytesRead(Data(), bytes_read);
+  } else {
+    delegate_->OnRequestFinished();
+  }
+}
+bool CronetURLRequestAdapter::IsOnNetworkThread() const {
+  return context_->GetNetworkTaskRunner()->BelongsToCurrentThread();
+}
+
+void CronetURLRequestAdapter::StartOnNetworkThread() {
+  DCHECK(IsOnNetworkThread());
+  if (cancel_soon_)
+    return;
+
+  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::FollowDeferredRedirectOnNetworkThread() {
+  DCHECK(IsOnNetworkThread());
+  if (WasCanceledWhileCalledToDelegate())
+    return;
+
+  url_request_->FollowDeferredRedirect();
+}
+
+// Reads all available data or starts an asynchronous read.
+void CronetURLRequestAdapter::ReadDataOnNetworkThread() {
+  DCHECK(IsOnNetworkThread());
+  if (WasCanceledWhileCalledToDelegate())
+    return;
+  int bytes_read = 0;
+  // If read completes synchronously, pass data to delegate.
+  if (url_request_->Read(
+          read_buffer_.get(), read_buffer_->size(), &bytes_read)) {
+    OnReadCompleted(url_request_.get(), bytes_read);
+  } else if (url_request_->status().status() !=
+             net::URLRequestStatus::IO_PENDING) {
+    OnReadCompleted(url_request_.get(), -1);
+  }
+}
+
+void CronetURLRequestAdapter::CancelOnNetworkThread() {
+  DCHECK(IsOnNetworkThread());
+  VLOG(1) << "Cancelling chromium request: " <<
+      initial_url_.possibly_invalid_spec();
+
+  if (called_to_delegate_) {
+    cancel_soon_ = true;
+    return;
+  }
+
+  if (url_request_ != NULL)
+    url_request_->Cancel();
+
+  delegate_->OnRequestCanceled();
+}
+
+bool CronetURLRequestAdapter::WasCanceledWhileCalledToDelegate() {
+  DCHECK(IsOnNetworkThread());
+  DCHECK(called_to_delegate_);
+  called_to_delegate_ = false;
+  if (cancel_soon_) {
+    CancelOnNetworkThread();
+    return true;
+  }
+  return false;
+}
+
+bool CronetURLRequestAdapter::CheckStatus(net::URLRequest* request) {
+  DCHECK_NE(net::URLRequestStatus::IO_PENDING, url_request_->status().status());
+  DCHECK_EQ(request, url_request_);
+  if (url_request_->status().is_success())
+    return true;
+  VLOG(1) << "Error " << url_request_->status().error()
+          << " on chromium request: " << initial_url_.possibly_invalid_spec();
+  called_to_delegate_ = true;
+  delegate_->OnError(url_request_->status().error());
+  return false;
+}
+
+unsigned char* CronetURLRequestAdapter::Data() const {
+  return reinterpret_cast<unsigned char*>(read_buffer_->data());
+}
+
+}  // namespace cronet