1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
|
// Copyright 2013 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "mojo/public/cpp/bindings/message.h"
#include <stdlib.h>
#include <algorithm>
#include "mojo/public/cpp/environment/logging.h"
namespace mojo {
Message::Message() : data_num_bytes_(0), data_(nullptr) {
}
Message::~Message() {
free(data_);
for (std::vector<Handle>::iterator it = handles_.begin();
it != handles_.end();
++it) {
if (it->is_valid())
CloseRaw(*it);
}
}
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::AdoptData(uint32_t num_bytes, internal::MessageData* data) {
MOJO_DCHECK(!data_);
data_num_bytes_ = num_bytes;
data_ = data;
}
void Message::Swap(Message* other) {
std::swap(data_num_bytes_, other->data_num_bytes_);
std::swap(data_, other->data_);
std::swap(handles_, other->handles_);
}
MojoResult ReadAndDispatchMessage(MessagePipeHandle handle,
MessageReceiver* receiver,
bool* receiver_result) {
MojoResult rv;
uint32_t num_bytes = 0, num_handles = 0;
rv = ReadMessageRaw(handle,
nullptr,
&num_bytes,
nullptr,
&num_handles,
MOJO_READ_MESSAGE_FLAG_NONE);
if (rv != MOJO_RESULT_RESOURCE_EXHAUSTED)
return rv;
Message message;
message.AllocUninitializedData(num_bytes);
message.mutable_handles()->resize(num_handles);
rv = ReadMessageRaw(
handle,
message.mutable_data(),
&num_bytes,
message.mutable_handles()->empty()
? nullptr
: reinterpret_cast<MojoHandle*>(&message.mutable_handles()->front()),
&num_handles,
MOJO_READ_MESSAGE_FLAG_NONE);
if (receiver && rv == MOJO_RESULT_OK)
*receiver_result = receiver->Accept(&message);
return rv;
}
} // namespace mojo
|
