Cronet modifications to support AGSA.

components/cronet/android/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 "url_request_adapter.h"
 
+#include <string.h>
+
+#include "base/bind.h"
+#include "base/location.h"
+#include "base/logging.h"
 #include "base/strings/string_number_conversions.h"
 #include "components/cronet/android/url_request_context_adapter.h"
 #include "components/cronet/android/wrapped_channel_upload_element_reader.h"
 #include "net/base/load_flags.h"
 #include "net/base/upload_bytes_element_reader.h"
 #include "net/http/http_status_code.h"
 
 namespace cronet {
 
 static const size_t kBufferSizeIncrement = 8192;
 
 URLRequestAdapter::URLRequestAdapter(URLRequestContextAdapter* context,
                                      URLRequestAdapterDelegate* 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) {
+      expected_size_(0),
+      chunked_upload_(false) {
   context_ = context;
   delegate_ = delegate;
   url_ = url;
   priority_ = priority;
 }
 
 URLRequestAdapter::~URLRequestAdapter() {
   CHECK(url_request_ == NULL);
 }
 
@@ skipping 14 matching lines @@
       net::UploadDataStream::CreateWithReader(reader.Pass(), 0));
 }
 
 void URLRequestAdapter::SetUploadChannel(JNIEnv* env, int64 content_length) {
   scoped_ptr<net::UploadElementReader> reader(
       new WrappedChannelElementReader(delegate_, content_length));
   upload_data_stream_.reset(
       net::UploadDataStream::CreateWithReader(reader.Pass(), 0));
 }
 
+void URLRequestAdapter::EnableChunkedUpload() {
+  chunked_upload_ = true;
+}
+
+void URLRequestAdapter::AppendChunk(const char* bytes, int bytes_len,
+                                    bool is_last_chunk) {
+  VLOG(1) << "AppendChunk, len: " << bytes_len << ", last: " << is_last_chunk;
+  scoped_ptr<char[]> buf(new char[bytes_len]);
+  memcpy(buf.get(), bytes, bytes_len);
+  context_->GetNetworkTaskRunner()->PostTask(
+      FROM_HERE,
+      base::Bind(&URLRequestAdapter::OnAppendChunk,
+                 base::Unretained(this),
+                 Passed(buf.Pass()),
+                 bytes_len,
+                 is_last_chunk));
+}
+
 std::string URLRequestAdapter::GetHeader(const std::string& name) const {
   std::string value;
   if (url_request_ != NULL) {
     url_request_->GetResponseHeaderByName(name, &value);
   }
   return value;
 }
 
 net::HttpResponseHeaders* URLRequestAdapter::GetResponseHeaders() const {
   if (url_request_ == NULL) {
     return NULL;
   }
   return url_request_->response_headers();
 }
 
 void URLRequestAdapter::Start() {
   context_->GetNetworkTaskRunner()->PostTask(
       FROM_HERE,
       base::Bind(&URLRequestAdapter::OnInitiateConnection,
                  base::Unretained(this)));
 }
 
+void URLRequestAdapter::OnAppendChunk(const scoped_ptr<char[]> bytes,
+                                      int bytes_len, bool is_last_chunk) {
+  url_request_->AppendChunkToUpload(bytes.get(), bytes_len, is_last_chunk);
+}
+
 void URLRequestAdapter::OnInitiateConnection() {
   if (canceled_) {
     return;
   }
 
   VLOG(1) << "Starting chromium request: "
           << url_.possibly_invalid_spec().c_str()
           << " priority: " << 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_);
   if (!headers_.HasHeader(net::HttpRequestHeaders::kUserAgent)) {
     std::string user_agent;
     user_agent = context_->GetUserAgent(url_);
     url_request_->SetExtraRequestHeaderByName(
         net::HttpRequestHeaders::kUserAgent, user_agent, true /* override */);
   }
 
-  if (upload_data_stream_)
+  if (upload_data_stream_) {
     url_request_->set_upload(upload_data_stream_.Pass());
+  } else if (chunked_upload_) {
+    url_request_->EnableChunkedUpload();
+  }
 
   url_request_->SetPriority(priority_);
 
   url_request_->Start();
 }
 
 void URLRequestAdapter::Cancel() {
   if (canceled_) {
     return;
   }
@@ skipping 135 matching lines @@
 
   delegate_->OnBytesRead(this);
   delegate_->OnRequestFinished(this);
 }
 
 unsigned char* URLRequestAdapter::Data() const {
   return reinterpret_cast<unsigned char*>(read_buffer_->StartOfBuffer());
 }
 
 }  // namespace cronet