OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "ipc/ipc_message.h" | 5 #include "ipc/ipc_message.h" |
6 | 6 |
7 #include <limits.h> | 7 #include <limits.h> |
8 #include <stddef.h> | 8 #include <stddef.h> |
9 #include <stdint.h> | 9 #include <stdint.h> |
10 | 10 |
(...skipping 148 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
159 const char* pickle_end = range_start + pickle_size; | 159 const char* pickle_end = range_start + pickle_size; |
160 | 160 |
161 info->message_end = pickle_end; | 161 info->message_end = pickle_end; |
162 | 162 |
163 info->pickle_end = pickle_end; | 163 info->pickle_end = pickle_end; |
164 info->message_found = true; | 164 info->message_found = true; |
165 } | 165 } |
166 | 166 |
167 bool Message::WriteAttachment( | 167 bool Message::WriteAttachment( |
168 scoped_refptr<base::Pickle::Attachment> attachment) { | 168 scoped_refptr<base::Pickle::Attachment> attachment) { |
| 169 bool brokerable; |
169 size_t index; | 170 size_t index; |
170 bool success = attachment_set()->AddAttachment( | 171 bool success = attachment_set()->AddAttachment( |
171 make_scoped_refptr(static_cast<MessageAttachment*>(attachment.get())), | 172 make_scoped_refptr(static_cast<MessageAttachment*>(attachment.get())), |
172 &index); | 173 &index, &brokerable); |
173 DCHECK(success); | 174 DCHECK(success); |
174 | 175 |
175 // NOTE: If you add more data to the pickle, make sure to update | 176 // NOTE: If you add more data to the pickle, make sure to update |
176 // PickleSizer::AddAttachment. | 177 // PickleSizer::AddAttachment. |
177 | 178 |
| 179 // Write the type of descriptor. |
| 180 WriteBool(brokerable); |
| 181 |
178 // Write the index of the descriptor so that we don't have to | 182 // Write the index of the descriptor so that we don't have to |
179 // keep the current descriptor as extra decoding state when deserialising. | 183 // keep the current descriptor as extra decoding state when deserialising. |
180 WriteInt(static_cast<int>(index)); | 184 WriteInt(static_cast<int>(index)); |
181 | 185 |
182 return success; | 186 return success; |
183 } | 187 } |
184 | 188 |
185 bool Message::ReadAttachment( | 189 bool Message::ReadAttachment( |
186 base::PickleIterator* iter, | 190 base::PickleIterator* iter, |
187 scoped_refptr<base::Pickle::Attachment>* attachment) const { | 191 scoped_refptr<base::Pickle::Attachment>* attachment) const { |
| 192 bool brokerable; |
| 193 if (!iter->ReadBool(&brokerable)) |
| 194 return false; |
| 195 |
188 int index; | 196 int index; |
189 if (!iter->ReadInt(&index)) | 197 if (!iter->ReadInt(&index)) |
190 return false; | 198 return false; |
191 | 199 |
192 MessageAttachmentSet* attachment_set = attachment_set_.get(); | 200 MessageAttachmentSet* attachment_set = attachment_set_.get(); |
193 if (!attachment_set) | 201 if (!attachment_set) |
194 return false; | 202 return false; |
195 | 203 |
196 *attachment = attachment_set->GetAttachmentAt(index); | 204 *attachment = brokerable |
| 205 ? attachment_set->GetBrokerableAttachmentAt(index) |
| 206 : attachment_set->GetNonBrokerableAttachmentAt(index); |
197 | 207 |
198 return nullptr != attachment->get(); | 208 return nullptr != attachment->get(); |
199 } | 209 } |
200 | 210 |
201 bool Message::HasAttachments() const { | 211 bool Message::HasAttachments() const { |
202 return attachment_set_.get() && !attachment_set_->empty(); | 212 return attachment_set_.get() && !attachment_set_->empty(); |
203 } | 213 } |
204 | 214 |
| 215 bool Message::HasMojoHandles() const { |
| 216 return attachment_set_.get() && attachment_set_->num_mojo_handles() > 0; |
| 217 } |
| 218 |
| 219 bool Message::HasBrokerableAttachments() const { |
| 220 return attachment_set_.get() && |
| 221 attachment_set_->num_brokerable_attachments() > 0; |
| 222 } |
| 223 |
205 } // namespace IPC | 224 } // namespace IPC |
OLD | NEW |