Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(1081)

Unified Diff: mojo/public/cpp/bindings/lib/message.cc

Issue 1524613002: [mojo] Use base::Pickle for Message storage. (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: some cleanup Created 5 years ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
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)

Powered by Google App Engine
This is Rietveld 408576698