upload_data_stream.cc revision 3345a6884c488ff3a535c2c9acdd33d74b37e311
1c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch// Copyright (c) 2010 The Chromium Authors. All rights reserved. 2c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott// Use of this source code is governed by a BSD-style license that can be 3c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott// found in the LICENSE file. 4c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 500d26a728db2814620f390b418a7d6325ce5aca6Ben Murdoch 600d26a728db2814620f390b418a7d6325ce5aca6Ben Murdoch 7c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#include "net/base/upload_data_stream.h" 8c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 9c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#include "base/file_util.h" 10c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#include "base/logging.h" 113345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick#include "net/base/file_stream.h" 12c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#include "net/base/io_buffer.h" 13c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#include "net/base/net_errors.h" 14c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 15c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottnamespace net { 16c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 17c407dc5cd9bdc5668497f21b26b09d988ab439deBen MurdochUploadDataStream* UploadDataStream::Create(UploadData* data, int* error_code) { 18c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch scoped_ptr<UploadDataStream> stream(new UploadDataStream(data)); 19c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch int rv = stream->FillBuf(); 20c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch if (error_code) 21c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch *error_code = rv; 22c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch if (rv != OK) 23c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch return NULL; 24c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 25c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch return stream.release(); 26c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch} 27c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 28c407dc5cd9bdc5668497f21b26b09d988ab439deBen MurdochUploadDataStream::UploadDataStream(UploadData* data) 29c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott : data_(data), 30c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott buf_(new IOBuffer(kBufSize)), 31c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott buf_len_(0), 32c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch next_element_(data->elements()->begin()), 33c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott next_element_offset_(0), 34c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott next_element_remaining_(0), 35c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott total_size_(data->GetContentLength()), 36c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch current_position_(0), 37c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch eof_(false) { 38c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 39c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 40c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick ScottUploadDataStream::~UploadDataStream() { 41c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 42c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 43c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottvoid UploadDataStream::DidConsume(size_t num_bytes) { 44c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch DCHECK_LE(num_bytes, buf_len_); 45c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch DCHECK(!eof_); 46c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 47c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott buf_len_ -= num_bytes; 48c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (buf_len_) 49c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott memmove(buf_->data(), buf_->data() + num_bytes, buf_len_); 50c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 51c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott FillBuf(); 52c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 53c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott current_position_ += num_bytes; 54c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 55c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 56c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochint UploadDataStream::FillBuf() { 57c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch std::vector<UploadData::Element>::iterator end = 58c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch data_->elements()->end(); 59c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 60c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott while (buf_len_ < kBufSize && next_element_ != end) { 61c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott bool advance_to_next_element = false; 62c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 63c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch UploadData::Element& element = *next_element_; 64c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 65c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott size_t size_remaining = kBufSize - buf_len_; 66c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (element.type() == UploadData::TYPE_BYTES) { 67c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott const std::vector<char>& d = element.bytes(); 68c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott size_t count = d.size() - next_element_offset_; 69c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 70c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott size_t bytes_copied = std::min(count, size_remaining); 71c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 72c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott memcpy(buf_->data() + buf_len_, &d[next_element_offset_], bytes_copied); 73c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott buf_len_ += bytes_copied; 74c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 75c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (bytes_copied == count) { 76c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott advance_to_next_element = true; 77c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } else { 78c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott next_element_offset_ += bytes_copied; 79c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 80c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } else { 81c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott DCHECK(element.type() == UploadData::TYPE_FILE); 82c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 83c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch if (!next_element_remaining_) { 84c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch // If the underlying file has been changed, treat it as error. 85c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch // Note that the expected modification time from WebKit is based on 86c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch // time_t precision. So we have to convert both to time_t to compare. 87c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch if (!element.expected_file_modification_time().is_null()) { 883345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick base::PlatformFileInfo info; 89c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch if (file_util::GetFileInfo(element.file_path(), &info) && 90c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch element.expected_file_modification_time().ToTimeT() != 91c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch info.last_modified.ToTimeT()) { 92c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch return ERR_UPLOAD_FILE_CHANGED; 93c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 94c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 95c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch next_element_remaining_ = element.GetContentLength(); 96c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch next_element_stream_.reset(element.NewFileStreamForReading()); 97c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 98c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 99c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int rv = 0; 100c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch int count = 101c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch static_cast<int>(std::min(next_element_remaining_, 102c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch static_cast<uint64>(size_remaining))); 103c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch if (count > 0) { 104c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch if (next_element_stream_.get()) 105c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch rv = next_element_stream_->Read(buf_->data() + buf_len_, count, NULL); 106c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch if (rv <= 0) { 107c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch // If there's less data to read than we initially observed, then 108c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch // pad with zero. Otherwise the server will hang waiting for the 109c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch // rest of the data. 110c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch memset(buf_->data() + buf_len_, 0, count); 111c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch rv = count; 112c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch } 113c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott buf_len_ += rv; 114c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch } 115c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 116c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch if (static_cast<int>(next_element_remaining_) == rv) { 117c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott advance_to_next_element = true; 118c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch } else { 119c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch next_element_remaining_ -= rv; 120c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 121c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 122c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 123c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (advance_to_next_element) { 124c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott ++next_element_; 125c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott next_element_offset_ = 0; 126c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch next_element_remaining_ = 0; 127c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch next_element_stream_.reset(); 128c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 129c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 130c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 131c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch if (next_element_ == end && !buf_len_) 132c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch eof_ = true; 133c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 134c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch return OK; 135c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 136c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 137c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} // namespace net 138