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) |