Stub in more IPC attachment brokering functionality.

ipc/ipc_message_attachment_set.cc

 // Copyright (c) 2011 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 "ipc/ipc_message_attachment_set.h"
 
 #include <algorithm>
 #include "base/logging.h"
 #include "base/posix/eintr_wrapper.h"
+#include "ipc/brokerable_attachment.h"
 #include "ipc/ipc_message_attachment.h"
 
 #if defined(OS_POSIX)
 #include <sys/types.h>
 #include <sys/stat.h>
 #include <unistd.h>
 #include "ipc/ipc_platform_file_attachment_posix.h"
 #endif // OS_POSIX
 
 namespace IPC {
@@ skipping 25 matching lines @@
   });
 }
 
 unsigned MessageAttachmentSet::num_mojo_handles() const {
   return std::count_if(attachments_.begin(), attachments_.end(),
                        [](scoped_refptr<MessageAttachment> i) {
     return i->GetType() == MessageAttachment::TYPE_MOJO_HANDLE;
   });
 }
 
+unsigned MessageAttachmentSet::num_brokerable_attachments() const {
+  return std::count_if(attachments_.begin(), attachments_.end(),

[Tom Sepez, 2015/06/19 18:04:11]

C++11 std library not yet allowed AFAIK (as much as I like these constructs). I
also worry about incrementing/decrementing the refcount
on each call.

[erikchen, 2015/06/23 22:37:00]

The latter problem could be fixed, but the former cannot. I was copying the code
from the two previous methods. I've created a new function in this translation
unit that does the counting, and I've changed all three methods to use this new
function.

+                       [](scoped_refptr<MessageAttachment> i) {
+                         return i->GetType() ==
+                                MessageAttachment::TYPE_BROKERABLE_ATTACHMENT;
+                       });
+}
+
 unsigned MessageAttachmentSet::size() const {
   return static_cast<unsigned>(attachments_.size());
 }
 
 bool MessageAttachmentSet::AddAttachment(
     scoped_refptr<MessageAttachment> attachment) {
 #if defined(OS_POSIX)
   if (attachment->GetType() == MessageAttachment::TYPE_PLATFORM_FILE &&
       num_descriptors() == kMaxDescriptorsPerMessage) {
     DLOG(WARNING) << "Cannot add file descriptor. MessageAttachmentSet full.";
@@ skipping 42 matching lines @@
   consumed_descriptor_highwater_ = index + 1;
 
   return attachments_[index];
 }
 
 void MessageAttachmentSet::CommitAll() {
   attachments_.clear();
   consumed_descriptor_highwater_ = 0;
 }
 
+std::vector<scoped_refptr<BrokerableAttachment>>
+MessageAttachmentSet::PeekBrokerableAttachments() const {
+  std::vector<scoped_refptr<BrokerableAttachment>> output;
+  for (const scoped_refptr<MessageAttachment>& attachment : attachments_) {
+    if (attachment->GetType() ==
+        MessageAttachment::TYPE_BROKERABLE_ATTACHMENT) {
+      scoped_refptr<BrokerableAttachment> brokerable_attachment(
+          static_cast<BrokerableAttachment*>(attachment.get()));
+      output.push_back(brokerable_attachment);
+    }
+  }
+  return output;
+}
+
 #if defined(OS_POSIX)
 
 void MessageAttachmentSet::PeekDescriptors(base::PlatformFile* buffer) const {
   for (size_t i = 0; i != attachments_.size(); ++i)
     buffer[i] = internal::GetPlatformFile(attachments_[i]);
 }
 
 bool MessageAttachmentSet::ContainsDirectoryDescriptor() const {
   struct stat st;
 
@@ skipping 27 matching lines @@
   for (unsigned i = 0; i < count; ++i)
     AddAttachment(
         new internal::PlatformFileAttachment(base::ScopedFD(buffer[i])));
 }
 
 #endif  // OS_POSIX
 
 }  // namespace IPC