Initial upload of cronet for Android.

net/cronet/android/url_request_peer.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 "url_request_peer.h"
+
+#include "base/strings/string_number_conversions.h"
+#include "net/base/load_flags.h"
+#include "net/cronet/android/android_log.h"
+#include "net/http/http_status_code.h"
+
+#define LOG_TAG "ChromiumNetwork"
+
+static const size_t kBufferSizeIncrement = 8192;
+
+// Fragment automatically inserted in the User-Agent header to indicate
+// that the request is coming from this network stack.
+static const char kUserAgentFragment[] = "; ChromiumJNI/";
+
+URLRequestPeer::URLRequestPeer(URLRequestContextPeer *context,
+                               URLRequestPeerDelegate *delegate, GURL url,
+                               net::RequestPriority priority)
+    : method_("GET"),
+      url_request_(NULL),
+      read_buffer_(new net::GrowableIOBuffer()),
+      bytes_read_(0),
+      total_bytes_read_(0),
+      error_code_(0),
+      http_status_code_(0),
+      canceled_(false),
+      expected_size_(0) {
+  context_ = context;
+  delegate_ = delegate;
+  url_ = url;
+  priority_ = priority;
+}
+
+URLRequestPeer::~URLRequestPeer() {
+  CHECK(url_request_ == NULL);
+}
+
+void URLRequestPeer::SetMethod(const std::string &method) {
+  method_ = method;
+}
+
+void URLRequestPeer::AddHeader(const std::string &name,
+                               const std::string &value) {
+  headers_.SetHeader(name, value);
+}
+
+void URLRequestPeer::SetPostContent(const char *bytes, int bytes_len) {
+  if (!upload_data_stream_) {
+    upload_data_stream_.reset(
+        new net::UploadDataStream(net::UploadDataStream::CHUNKED, 0));
+  }
+  upload_data_stream_->AppendChunk(bytes, bytes_len, true /* is_last_chunk */);
+}
+
+GURL URLRequestPeer::url() const {
+  return url_;
+}
+
+int URLRequestPeer::error_code() const {
+  return error_code_;
+}
+
+int URLRequestPeer::http_status_code() const {
+  return http_status_code_;
+}
+
+int64 URLRequestPeer::content_length() const {
+  return expected_size_;
+}
+
+std::string URLRequestPeer::content_type() const {
+  return content_type_;
+}
+
+void URLRequestPeer::Start() {
+  context_->GetNetworkTaskRunner()->PostTask(
+      FROM_HERE,
+      base::Bind(&URLRequestPeer::OnInitiateConnectionWrapper, this));
+}
+
+/* static */
+void URLRequestPeer::OnInitiateConnectionWrapper(URLRequestPeer *self) {
+  self->OnInitiateConnection();
+}
+
+void URLRequestPeer::OnInitiateConnection() {
+  if (canceled_) {
+    return;
+  }
+
+  LOGV(LOG_TAG, context_->logging_level(),
+       "Starting chromium request: %s, priority: %s", url_.spec().c_str(),
+       RequestPriorityToString(priority_));
+
+  url_request_ =
+      new net::URLRequest(url_,
+                          net::DEFAULT_PRIORITY,
+                          this,
+                          context_->GetURLRequestContext());
+  url_request_->SetLoadFlags(net::LOAD_DISABLE_CACHE |
+                             net::LOAD_DO_NOT_SAVE_COOKIES |
+                             net::LOAD_DO_NOT_SEND_COOKIES);
+  url_request_->set_method(method_);
+  url_request_->SetExtraRequestHeaders(headers_);
+  std::string user_agent;
+  if (headers_.HasHeader(net::HttpRequestHeaders::kUserAgent)) {
+    headers_.GetHeader(net::HttpRequestHeaders::kUserAgent, &user_agent);
+  } else {
+    user_agent = context_->GetUserAgent(url_);
+  }
+  size_t pos = user_agent.find(')');
+  if (pos != std::string::npos) {
+    user_agent.insert(pos, context_->version());
+    user_agent.insert(pos, kUserAgentFragment);
+  }
+  url_request_->SetExtraRequestHeaderByName(net::HttpRequestHeaders::kUserAgent,
+                                            user_agent, true /* override */);
+  LOGV(LOG_TAG, context_->logging_level(), "User agent: %s",
+       user_agent.c_str());
+
+  if (upload_data_stream_) {
+    url_request_->set_upload(make_scoped_ptr(upload_data_stream_.release()));
+  }
+  url_request_->SetPriority(priority_);
+
+  url_request_->Start();
+}
+
+void URLRequestPeer::Cancel() {
+  if (canceled_) {
+    return;
+  }
+
+  canceled_ = true;
+
+  context_->GetNetworkTaskRunner()->PostTask(
+      FROM_HERE, base::Bind(&URLRequestPeer::OnCancelRequestWrapper, this));
+}
+
+/* static */
+void URLRequestPeer::OnCancelRequestWrapper(URLRequestPeer *self) {
+  self->OnCancelRequest();
+}
+
+void URLRequestPeer::OnCancelRequest() {
+  LOGV(LOG_TAG, context_->logging_level(), "Canceling chromium request: %s",
+       url_.spec().c_str());
+
+  if (url_request_ != NULL) {
+    url_request_->Cancel();
+  }
+
+  OnRequestCanceled();
+}
+
+void URLRequestPeer::Destroy() {
+  context_->GetNetworkTaskRunner()->PostTask(
+      FROM_HERE, base::Bind(&URLRequestPeer::OnDestroyRequest, this));
+}
+
+/* static */
+void URLRequestPeer::OnDestroyRequest(URLRequestPeer *self) {
+  LOGV(LOG_TAG, self->context_->logging_level(),
+       "Destroying chromium request: %s", self->url_.spec().c_str());
+  delete self;
+}
+
+void URLRequestPeer::OnResponseStarted(net::URLRequest *request) {
+  if (request->status().status() != net::URLRequestStatus::SUCCESS) {
+    OnRequestFailed();
+    return;
+  }
+
+  http_status_code_ = request->GetResponseCode();
+  LOGV(LOG_TAG, context_->logging_level(), "Response started with status %d",
+       http_status_code_);
+
+  request->GetResponseHeaderByName("Content-Type", &content_type_);
+  expected_size_ = request->GetExpectedContentSize();
+  delegate_->OnResponseStarted(this);
+
+  Read();
+}
+
+/*
+ * Reads all available data or starts an asynchronous read.
+ */
+void URLRequestPeer::Read() {
+  while (1) {
+    if (read_buffer_->RemainingCapacity() == 0) {
+      int new_capacity = read_buffer_->capacity() + kBufferSizeIncrement;
+      read_buffer_->SetCapacity(new_capacity);
+    }
+
+    int bytes_read;
+    if (url_request_->Read(read_buffer_, read_buffer_->RemainingCapacity(),
+                           &bytes_read)) {
+      if (bytes_read == 0) {
+        OnRequestSucceeded();
+        break;
+      }
+
+      LOGV(LOG_TAG, context_->logging_level(), "Synchronously read %d bytes",
+           bytes_read);
+      OnBytesRead(bytes_read);
+    } else if (url_request_->status().status() ==
+               net::URLRequestStatus::IO_PENDING) {
+      if (bytes_read_ != 0) {
+        LOGV(LOG_TAG, context_->logging_level(), "Flushing buffer: %d bytes",
+             bytes_read_);
+        delegate_->OnBytesRead(this);
+        read_buffer_->set_offset(0);
+        bytes_read_ = 0;
+      }
+      LOGV(LOG_TAG, context_->logging_level(), "Started async read");
+      break;
+    } else {
+      OnRequestFailed();
+      break;
+    }
+  }
+}
+
+void URLRequestPeer::OnReadCompleted(net::URLRequest *request, int bytes_read) {
+  LOGV(LOG_TAG, context_->logging_level(), "Asynchronously read %d bytes",
+       bytes_read);
+  if (bytes_read < 0) {
+    OnRequestFailed();
+    return;
+  } else if (bytes_read == 0) {
+    OnRequestSucceeded();
+    return;
+  }
+
+  OnBytesRead(bytes_read);
+  Read();
+}
+
+void URLRequestPeer::OnBytesRead(int bytes_read) {
+  read_buffer_->set_offset(read_buffer_->offset() + bytes_read);
+  bytes_read_ += bytes_read;
+  total_bytes_read_ += bytes_read;
+}
+
+void URLRequestPeer::OnRequestSucceeded() {
+  if (canceled_) {
+    return;
+  }
+
+  LOGV(LOG_TAG, context_->logging_level(),
+       "Request completed with HTTP status %d. Total bytes read: %d",
+       http_status_code_, total_bytes_read_);
+  OnRequestCompleted();
+}
+
+void URLRequestPeer::OnRequestFailed() {
+  if (canceled_) {
+    return;
+  }
+
+  error_code_ = url_request_->status().error();
+  LOGV(LOG_TAG, context_->logging_level(),
+       "Request failed with status %d and error %s",
+       url_request_->status().status(), net::ErrorToString(error_code_));
+  OnRequestCompleted();
+}
+
+void URLRequestPeer::OnRequestCanceled() { OnRequestCompleted(); }
+
+void URLRequestPeer::OnRequestCompleted() {
+  LOGV(LOG_TAG, context_->logging_level(), "Completed: %s",
+       url_.spec().c_str());
+  if (url_request_ != NULL) {
+    delete url_request_;
+    url_request_ = NULL;
+  }
+
+  delegate_->OnBytesRead(this);
+  delegate_->OnRequestFinished(this);
+}
+
+size_t URLRequestPeer::BytesRead() const { return bytes_read_; }
+
+unsigned char *URLRequestPeer::Data() const {
+  return reinterpret_cast<unsigned char *>(read_buffer_->StartOfBuffer());
+}