Chromium Code Reviews| Index: mojo/public/cpp/bindings/lib/message.cc |
| diff --git a/mojo/public/cpp/bindings/lib/message.cc b/mojo/public/cpp/bindings/lib/message.cc |
| index 6b563e779aca7d196284726c0a899fe03ce2cdde..dd80f9787b606bc161275af3ecb407281a2c16b4 100644 |
| --- a/mojo/public/cpp/bindings/lib/message.cc |
| +++ b/mojo/public/cpp/bindings/lib/message.cc |
| @@ -17,50 +17,39 @@ Message::Message() { |
| } |
| Message::~Message() { |
| - FreeDataAndCloseHandles(); |
| + CloseHandles(); |
| } |
| void Message::Reset() { |
| - FreeDataAndCloseHandles(); |
| + CloseHandles(); |
| handles_.clear(); |
| Initialize(); |
| } |
| -void Message::AllocData(uint32_t num_bytes) { |
| - MOJO_DCHECK(!data_); |
| - data_num_bytes_ = num_bytes; |
| - data_ = static_cast<internal::MessageData*>(calloc(num_bytes, 1)); |
| +void Message::ReserveCapacity(uint32_t num_bytes) { |
|
yzshen1
2015/12/15 22:27:08
According to our discussion, let's make it Initial
Ken Rockot(use gerrit already)
2015/12/16 00:09:33
Done
|
| + buffer_->ReserveCapacity(num_bytes); |
| } |
| -void Message::AllocUninitializedData(uint32_t num_bytes) { |
| - MOJO_DCHECK(!data_); |
| - data_num_bytes_ = num_bytes; |
| - data_ = static_cast<internal::MessageData*>(malloc(num_bytes)); |
| +void Message::ReserveUninitializedCapacity(uint32_t num_bytes) { |
| + buffer_->ReserveUninitializedCapacity(num_bytes); |
| } |
| void Message::MoveTo(Message* destination) { |
| MOJO_DCHECK(this != destination); |
| - destination->FreeDataAndCloseHandles(); |
| - |
| // No copy needed. |
| - destination->data_num_bytes_ = data_num_bytes_; |
| - destination->data_ = data_; |
| + std::swap(destination->buffer_, buffer_); |
| std::swap(destination->handles_, handles_); |
| - handles_.clear(); |
| - Initialize(); |
| + Reset(); |
| } |
| void Message::Initialize() { |
| - data_num_bytes_ = 0; |
| - data_ = nullptr; |
| + buffer_.reset(new internal::PickleBuffer); |
| } |
| -void Message::FreeDataAndCloseHandles() { |
| - free(data_); |
| - |
| +void Message::CloseHandles() { |
| for (std::vector<Handle>::iterator it = handles_.begin(); |
| it != handles_.end(); ++it) { |
| if (it->is_valid()) |
| @@ -84,16 +73,17 @@ MojoResult ReadAndDispatchMessage(MessagePipeHandle handle, |
| return rv; |
| Message message; |
| - message.AllocUninitializedData(num_bytes); |
| + message.ReserveUninitializedCapacity(num_bytes); |
| + void* mutable_data = message.buffer()->Allocate(num_bytes); |
| message.mutable_handles()->resize(num_handles); |
| rv = ReadMessageRaw( |
| handle, |
| - message.mutable_data(), |
| + mutable_data, |
| &num_bytes, |
| message.mutable_handles()->empty() |
| ? nullptr |
| - : reinterpret_cast<MojoHandle*>(&message.mutable_handles()->front()), |
| + : reinterpret_cast<MojoHandle*>(message.mutable_handles()->data()), |
| &num_handles, |
| MOJO_READ_MESSAGE_FLAG_NONE); |
| if (receiver && rv == MOJO_RESULT_OK) |