OLD | NEW |
| (Empty) |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 #include "mojo/public/cpp/bindings/message.h" | |
6 | |
7 #include <stdlib.h> | |
8 | |
9 #include <algorithm> | |
10 | |
11 #include "mojo/public/cpp/environment/logging.h" | |
12 | |
13 namespace mojo { | |
14 | |
15 Message::Message() { | |
16 Initialize(); | |
17 } | |
18 | |
19 Message::~Message() { | |
20 FreeDataAndCloseHandles(); | |
21 } | |
22 | |
23 void Message::Reset() { | |
24 FreeDataAndCloseHandles(); | |
25 | |
26 handles_.clear(); | |
27 Initialize(); | |
28 } | |
29 | |
30 void Message::AllocData(uint32_t num_bytes) { | |
31 MOJO_DCHECK(!data_); | |
32 data_num_bytes_ = num_bytes; | |
33 data_ = static_cast<internal::MessageData*>(calloc(num_bytes, 1)); | |
34 } | |
35 | |
36 void Message::AllocUninitializedData(uint32_t num_bytes) { | |
37 MOJO_DCHECK(!data_); | |
38 data_num_bytes_ = num_bytes; | |
39 data_ = static_cast<internal::MessageData*>(malloc(num_bytes)); | |
40 } | |
41 | |
42 void Message::MoveTo(Message* destination) { | |
43 MOJO_DCHECK(this != destination); | |
44 | |
45 destination->FreeDataAndCloseHandles(); | |
46 | |
47 // No copy needed. | |
48 destination->data_num_bytes_ = data_num_bytes_; | |
49 destination->data_ = data_; | |
50 std::swap(destination->handles_, handles_); | |
51 | |
52 handles_.clear(); | |
53 Initialize(); | |
54 } | |
55 | |
56 void Message::Initialize() { | |
57 data_num_bytes_ = 0; | |
58 data_ = nullptr; | |
59 } | |
60 | |
61 void Message::FreeDataAndCloseHandles() { | |
62 free(data_); | |
63 | |
64 for (std::vector<Handle>::iterator it = handles_.begin(); | |
65 it != handles_.end(); ++it) { | |
66 if (it->is_valid()) | |
67 CloseRaw(*it); | |
68 } | |
69 } | |
70 | |
71 MojoResult ReadMessage(MessagePipeHandle handle, Message* message) { | |
72 MOJO_DCHECK(handle.is_valid()); | |
73 MOJO_DCHECK(message); | |
74 MOJO_DCHECK(message->handles()->empty()); | |
75 MOJO_DCHECK(message->data_num_bytes() == 0); | |
76 | |
77 uint32_t num_bytes = 0; | |
78 uint32_t num_handles = 0; | |
79 MojoResult rv = ReadMessageRaw(handle, nullptr, &num_bytes, nullptr, | |
80 &num_handles, MOJO_READ_MESSAGE_FLAG_NONE); | |
81 if (rv != MOJO_RESULT_RESOURCE_EXHAUSTED) | |
82 return rv; | |
83 | |
84 message->AllocUninitializedData(num_bytes); | |
85 message->mutable_handles()->resize(num_handles); | |
86 | |
87 uint32_t num_bytes_actual = num_bytes; | |
88 uint32_t num_handles_actual = num_handles; | |
89 rv = ReadMessageRaw( | |
90 handle, message->mutable_data(), &num_bytes_actual, | |
91 message->mutable_handles()->empty() | |
92 ? nullptr | |
93 : reinterpret_cast<MojoHandle*>(&message->mutable_handles()->front()), | |
94 &num_handles_actual, MOJO_READ_MESSAGE_FLAG_NONE); | |
95 | |
96 MOJO_DCHECK(num_bytes == num_bytes_actual); | |
97 MOJO_DCHECK(num_handles == num_handles_actual); | |
98 | |
99 return rv; | |
100 } | |
101 | |
102 MojoResult ReadAndDispatchMessage(MessagePipeHandle handle, | |
103 MessageReceiver* receiver, | |
104 bool* receiver_result) { | |
105 Message message; | |
106 MojoResult rv = ReadMessage(handle, &message); | |
107 if (receiver && rv == MOJO_RESULT_OK) | |
108 *receiver_result = receiver->Accept(&message); | |
109 | |
110 return rv; | |
111 } | |
112 | |
113 } // namespace mojo | |
OLD | NEW |