Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "net/base/upload_data_stream.h" | 5 #include "net/base/upload_data_stream.h" |
| 6 | 6 |
| 7 #include "base/file_util.h" | 7 #include "base/file_util.h" |
| 8 #include "base/logging.h" | 8 #include "base/logging.h" |
| 9 #include "net/base/file_stream.h" | 9 #include "net/base/file_stream.h" |
| 10 #include "net/base/io_buffer.h" | 10 #include "net/base/io_buffer.h" |
| (...skipping 51 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 62 UploadData::Element& element = elements[next_element_]; | 62 UploadData::Element& element = elements[next_element_]; |
| 63 | 63 |
| 64 size_t size_remaining = kBufSize - buf_len_; | 64 size_t size_remaining = kBufSize - buf_len_; |
| 65 if (element.type() == UploadData::TYPE_BYTES || | 65 if (element.type() == UploadData::TYPE_BYTES || |
| 66 element.type() == UploadData::TYPE_CHUNK) { | 66 element.type() == UploadData::TYPE_CHUNK) { |
| 67 const std::vector<char>& d = element.bytes(); | 67 const std::vector<char>& d = element.bytes(); |
| 68 size_t count = d.size() - next_element_offset_; | 68 size_t count = d.size() - next_element_offset_; |
| 69 | 69 |
| 70 size_t bytes_copied = std::min(count, size_remaining); | 70 size_t bytes_copied = std::min(count, size_remaining); |
| 71 | 71 |
| 72 memcpy(buf_->data() + buf_len_, &d[next_element_offset_], bytes_copied); | 72 // Check if we have anything to copy first, because we are getting the |
| 73 buf_len_ += bytes_copied; | 73 // address of an element in |d| and that will throw an exception if |d| |
| 74 // is an empty vector. | |
| 75 if (bytes_copied) { | |
| 76 memcpy(buf_->data() + buf_len_, &d[next_element_offset_], bytes_copied); | |
| 77 buf_len_ += bytes_copied; | |
| 78 } | |
| 74 | 79 |
| 75 if (bytes_copied == count) { | 80 if (bytes_copied == count) { |
| 76 advance_to_next_element = true; | 81 advance_to_next_element = true; |
| 77 } else { | 82 } else { |
| 78 next_element_offset_ += bytes_copied; | 83 next_element_offset_ += bytes_copied; |
| 79 } | 84 } |
| 80 } else { | 85 } else { |
| 81 DCHECK(element.type() == UploadData::TYPE_FILE); | 86 DCHECK(element.type() == UploadData::TYPE_FILE); |
| 82 | 87 |
| 83 if (!next_element_remaining_) { | 88 if (!next_element_remaining_) { |
| (...skipping 47 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 131 if (next_element_ == elements.size() && !buf_len_) { | 136 if (next_element_ == elements.size() && !buf_len_) { |
| 132 if (!data_->is_chunked() || | 137 if (!data_->is_chunked() || |
| 133 (!elements.empty() && elements.back().is_last_chunk())) { | 138 (!elements.empty() && elements.back().is_last_chunk())) { |
| 134 eof_ = true; | 139 eof_ = true; |
| 135 } | 140 } |
| 136 } | 141 } |
| 137 | 142 |
| 138 return OK; | 143 return OK; |
| 139 } | 144 } |
| 140 | 145 |
| 146 bool UploadDataStream::peek_end_of_chunks() const { | |
|
willchan no longer on Chromium
2011/03/03 19:47:48
This is not a trivial accessor. Also, this may be
| |
| 147 std::vector<UploadData::Element>& elements = *data_->elements(); | |
|
willchan no longer on Chromium
2011/03/03 19:47:48
Can you get away with a const reference?
| |
| 148 if (!data_->is_chunked()) | |
|
willchan no longer on Chromium
2011/03/03 19:47:48
If this doesn't depend on |elements|, then how abo
| |
| 149 return false; | |
| 150 return (eof_ || | |
| 151 (!elements.empty() && | |
| 152 next_element_ >= elements.size() - 1 && | |
|
willchan no longer on Chromium
2011/03/03 19:47:48
I find |next_element_ > elements.size()| more read
Satish
2011/03/03 22:21:44
The condition could only be changed as "next_eleme
willchan no longer on Chromium
2011/03/03 22:26:54
Oops, missed that! Bleh, off by one =/
| |
| 153 elements.back().is_last_chunk())); | |
| 154 } | |
| 155 | |
| 141 } // namespace net | 156 } // namespace net |
| OLD | NEW |