| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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_bytes_element_reader.h" | 5 #include "net/base/upload_bytes_element_reader.h" |
| 6 | 6 |
| 7 #include "base/logging.h" | 7 #include "base/logging.h" |
| 8 #include "base/stl_util.h" | 8 #include "base/stl_util.h" |
| 9 #include "net/base/io_buffer.h" | 9 #include "net/base/io_buffer.h" |
| 10 #include "net/base/net_errors.h" | 10 #include "net/base/net_errors.h" |
| 11 | 11 |
| 12 namespace net { | 12 namespace net { |
| 13 | 13 |
| 14 UploadBytesElementReader::UploadBytesElementReader(const char* bytes, | 14 UploadBytesElementReader::UploadBytesElementReader(const char* bytes, |
| 15 uint64 length) | 15 uint64_t length) |
| 16 : bytes_(bytes), | 16 : bytes_(bytes), length_(length), offset_(0) { |
| 17 length_(length), | |
| 18 offset_(0) { | |
| 19 } | 17 } |
| 20 | 18 |
| 21 UploadBytesElementReader::~UploadBytesElementReader() { | 19 UploadBytesElementReader::~UploadBytesElementReader() { |
| 22 } | 20 } |
| 23 | 21 |
| 24 const UploadBytesElementReader* | 22 const UploadBytesElementReader* |
| 25 UploadBytesElementReader::AsBytesReader() const { | 23 UploadBytesElementReader::AsBytesReader() const { |
| 26 return this; | 24 return this; |
| 27 } | 25 } |
| 28 | 26 |
| 29 int UploadBytesElementReader::Init(const CompletionCallback& callback) { | 27 int UploadBytesElementReader::Init(const CompletionCallback& callback) { |
| 30 offset_ = 0; | 28 offset_ = 0; |
| 31 return OK; | 29 return OK; |
| 32 } | 30 } |
| 33 | 31 |
| 34 uint64 UploadBytesElementReader::GetContentLength() const { | 32 uint64_t UploadBytesElementReader::GetContentLength() const { |
| 35 return length_; | 33 return length_; |
| 36 } | 34 } |
| 37 | 35 |
| 38 uint64 UploadBytesElementReader::BytesRemaining() const { | 36 uint64_t UploadBytesElementReader::BytesRemaining() const { |
| 39 return length_ - offset_; | 37 return length_ - offset_; |
| 40 } | 38 } |
| 41 | 39 |
| 42 bool UploadBytesElementReader::IsInMemory() const { | 40 bool UploadBytesElementReader::IsInMemory() const { |
| 43 return true; | 41 return true; |
| 44 } | 42 } |
| 45 | 43 |
| 46 int UploadBytesElementReader::Read(IOBuffer* buf, | 44 int UploadBytesElementReader::Read(IOBuffer* buf, |
| 47 int buf_length, | 45 int buf_length, |
| 48 const CompletionCallback& callback) { | 46 const CompletionCallback& callback) { |
| 49 DCHECK_LT(0, buf_length); | 47 DCHECK_LT(0, buf_length); |
| 50 | 48 |
| 51 const int num_bytes_to_read = static_cast<int>( | 49 const int num_bytes_to_read = static_cast<int>( |
| 52 std::min(BytesRemaining(), static_cast<uint64>(buf_length))); | 50 std::min(BytesRemaining(), static_cast<uint64_t>(buf_length))); |
| 53 | 51 |
| 54 // Check if we have anything to copy first, because we are getting | 52 // Check if we have anything to copy first, because we are getting |
| 55 // the address of an element in |bytes_| and that will throw an | 53 // the address of an element in |bytes_| and that will throw an |
| 56 // exception if |bytes_| is an empty vector. | 54 // exception if |bytes_| is an empty vector. |
| 57 if (num_bytes_to_read > 0) | 55 if (num_bytes_to_read > 0) |
| 58 memcpy(buf->data(), bytes_ + offset_, num_bytes_to_read); | 56 memcpy(buf->data(), bytes_ + offset_, num_bytes_to_read); |
| 59 | 57 |
| 60 offset_ += num_bytes_to_read; | 58 offset_ += num_bytes_to_read; |
| 61 return num_bytes_to_read; | 59 return num_bytes_to_read; |
| 62 } | 60 } |
| 63 | 61 |
| 64 | 62 |
| 65 UploadOwnedBytesElementReader::UploadOwnedBytesElementReader( | 63 UploadOwnedBytesElementReader::UploadOwnedBytesElementReader( |
| 66 std::vector<char>* data) | 64 std::vector<char>* data) |
| 67 : UploadBytesElementReader(vector_as_array(data), data->size()) { | 65 : UploadBytesElementReader(vector_as_array(data), data->size()) { |
| 68 data_.swap(*data); | 66 data_.swap(*data); |
| 69 } | 67 } |
| 70 | 68 |
| 71 UploadOwnedBytesElementReader::~UploadOwnedBytesElementReader() {} | 69 UploadOwnedBytesElementReader::~UploadOwnedBytesElementReader() {} |
| 72 | 70 |
| 73 UploadOwnedBytesElementReader* | 71 UploadOwnedBytesElementReader* |
| 74 UploadOwnedBytesElementReader::CreateWithString(const std::string& string) { | 72 UploadOwnedBytesElementReader::CreateWithString(const std::string& string) { |
| 75 std::vector<char> data(string.begin(), string.end()); | 73 std::vector<char> data(string.begin(), string.end()); |
| 76 return new UploadOwnedBytesElementReader(&data); | 74 return new UploadOwnedBytesElementReader(&data); |
| 77 } | 75 } |
| 78 | 76 |
| 79 } // namespace net | 77 } // namespace net |
| OLD | NEW |