IPC: Add MojoHandleAttachment

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/ipc_message_attachment.h"
@@ skipping 27 matching lines @@
                << consumed_descriptor_highwater_ << "/" << size();
 }
 
 unsigned MessageAttachmentSet::num_descriptors() const {
   return std::count_if(attachments_.begin(), attachments_.end(),
                        [](scoped_refptr<MessageAttachment> i) {
     return i->GetType() == MessageAttachment::TYPE_PLATFORM_FILE;
   });
 }
 
+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::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 ||
+  if (attachment->GetType() == MessageAttachment::TYPE_PLATFORM_FILE &&
       num_descriptors() == kMaxDescriptorsPerMessage) {
     DLOG(WARNING) << "Cannot add file descriptor. MessageAttachmentSet full.";
     return false;
   }
 #endif
 
   attachments_.push_back(attachment);
   return true;
 }
 
@@ skipping 29 matching lines @@
     consumed_descriptor_highwater_ = 0;
 
   if (index != consumed_descriptor_highwater_)
     return scoped_refptr<MessageAttachment>();
 
   consumed_descriptor_highwater_ = index + 1;
 
   return attachments_[index];
 }
 
+void MessageAttachmentSet::CommitAll() {
+  attachments_.clear();
+  consumed_descriptor_highwater_ = 0;
+}
+
 #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;
 
   for (auto i = attachments_.begin(); i != attachments_.end(); ++i) {
     if (fstat(internal::GetPlatformFile(*i), &st) == 0 && S_ISDIR(st.st_mode))
       return true;
   }
 
   return false;
 }
 
-void MessageAttachmentSet::CommitAll() {
-  attachments_.clear();
-  consumed_descriptor_highwater_ = 0;
-}
-
 void MessageAttachmentSet::ReleaseFDsToClose(
     std::vector<base::PlatformFile>* fds) {
   for (size_t i = 0; i < attachments_.size(); ++i) {
     internal::PlatformFileAttachment* file =
         static_cast<internal::PlatformFileAttachment*>(attachments_[i].get());
     if (file->Owns())
       fds->push_back(file->TakePlatformFile());
   }
 
   CommitAll();
 }
 
 void MessageAttachmentSet::AddDescriptorsToOwn(const base::PlatformFile* buffer,
                                                unsigned count) {
   DCHECK(count <= kMaxDescriptorsPerMessage);
   DCHECK_EQ(num_descriptors(), 0u);
   DCHECK_EQ(consumed_descriptor_highwater_, 0u);
 
   attachments_.reserve(count);
   for (unsigned i = 0; i < count; ++i)
     AddAttachment(
         new internal::PlatformFileAttachment(base::ScopedFD(buffer[i])));
 }
 
 #endif  // OS_POSIX
 
 }  // namespace IPC