Add support for chunked encoding in ChromeFrame for POST requests. This fixes...

chrome_frame/urlmon_upload_data_stream.cc

 // Copyright (c) 2009 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 "chrome_frame/urlmon_upload_data_stream.h"
 
 #include "net/base/io_buffer.h"
 #include "net/base/net_errors.h"
 
 void UrlmonUploadDataStream::Initialize(net::UploadData* upload_data) {
   upload_data_ = upload_data;
   request_body_stream_.reset(net::UploadDataStream::Create(upload_data, NULL));
   DCHECK(request_body_stream_.get());
 }
 
 STDMETHODIMP UrlmonUploadDataStream::Read(void* pv, ULONG cb, ULONG* read) {
   if (pv == NULL) {
     NOTREACHED();
     return E_POINTER;
   }
 
   // Have we already read past the end of the stream?
-  if (request_body_stream_->position() >= request_body_stream_->size()) {
+  if (request_body_stream_->eof()) {
     if (read) {
       *read = 0;
     }
     return S_FALSE;
   }
 
   uint64 total_bytes_to_copy = std::min(static_cast<uint64>(cb),
-      request_body_stream_->size() - request_body_stream_->position());
-  uint64 initial_position = request_body_stream_->position();
+      static_cast<uint64>(request_body_stream_->buf_len()));

[amit, 2011/01/26 21:42:24]

if bug_len is ULONG you can remove both 'static_cast's

[ananta, 2011/01/26 22:19:58]

Leaving this as is as per our discussion.

 
   uint64 bytes_copied = 0;
 
   char* write_pointer = reinterpret_cast<char*>(pv);
   while (bytes_copied < total_bytes_to_copy) {
     net::IOBuffer* buf = request_body_stream_->buf();
 
     // Make sure our length doesn't run past the end of the available data.
     size_t bytes_to_copy_now = static_cast<size_t>(
         std::min(static_cast<uint64>(request_body_stream_->buf_len()),
                  total_bytes_to_copy - bytes_copied));
 
     memcpy(write_pointer, buf->data(), bytes_to_copy_now);
 
     // Advance our copy tally
     bytes_copied += bytes_to_copy_now;
 
     // Advance our write pointer
     write_pointer += bytes_to_copy_now;
 
     // Advance the UploadDataStream read pointer:
     request_body_stream_->MarkConsumedAndFillBuffer(bytes_to_copy_now);
   }
 
   DCHECK(bytes_copied == total_bytes_to_copy);
-  DCHECK(request_body_stream_->position() ==
-         initial_position + total_bytes_to_copy);
 
   if (read) {
     *read = static_cast<ULONG>(total_bytes_to_copy);
   }
 
   return S_OK;
 }
 
 STDMETHODIMP UrlmonUploadDataStream::Seek(LARGE_INTEGER move, DWORD origin,
                                           ULARGE_INTEGER* new_pos) {
@@ skipping 24 matching lines @@
   if (0 == (grf_stat_flag & STATFLAG_NONAME)) {
     const wchar_t kStreamBuffer[] = L"PostStream";
     stat_stg->pwcsName =
         static_cast<wchar_t*>(::CoTaskMemAlloc(sizeof(kStreamBuffer)));
     lstrcpy(stat_stg->pwcsName, kStreamBuffer);
   }
   stat_stg->type = STGTY_STREAM;
   stat_stg->cbSize.QuadPart = upload_data_->GetContentLength();
   return S_OK;
 }