[mojo-edk] Broadcast surprise port disruptions

mojo/edk/system/channel.cc

 // Copyright 2016 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/edk/system/channel.h"
 
 #include <string.h>
 
 #include <algorithm>
 #include <limits>
@@ skipping 103 matching lines @@
   }
 }
 
 Channel::Message::~Message() {
   base::AlignedFree(data_);
 }
 
 // static
 Channel::MessagePtr Channel::Message::Deserialize(const void* data,
                                                   size_t data_num_bytes) {
-#if !defined(OS_WIN) && !(defined(OS_MACOSX) && !defined(OS_IOS))
-  // We only serialize messages into other messages when performing message
-  // relay on Windows and OSX.
-  NOTREACHED();
-  return nullptr;
-#else
   if (data_num_bytes < sizeof(Header))
     return nullptr;
 
   const Header* header = reinterpret_cast<const Header*>(data);
   if (header->num_bytes != data_num_bytes) {
     DLOG(ERROR) << "Decoding invalid message: " << header->num_bytes
                 << " != " << data_num_bytes;
     return nullptr;
   }
 
+#if defined(MOJO_EDK_LEGACY_PROTOCOL)
+  size_t payload_size = data_num_bytes - sizeof(Header);
+  const char* payload = static_cast<const char*>(data) + sizeof(Header);
+#else
   if (header->num_bytes < header->num_header_bytes ||
       header->num_header_bytes < sizeof(Header)) {
     DLOG(ERROR) << "Decoding invalid message: " << header->num_bytes << " < "
                 << header->num_header_bytes;
     return nullptr;
   }
 
   uint32_t extra_header_size = header->num_header_bytes - sizeof(Header);
+  size_t payload_size = data_num_bytes - header->num_header_bytes;
+  const char* payload =
+      static_cast<const char*>(data) + header->num_header_bytes;
+#endif  // defined(MOJO_EDK_LEGACY_PROTOCOL)
+
 #if defined(OS_WIN)
   uint32_t max_handles = extra_header_size / sizeof(HandleEntry);
 #elif defined(OS_MACOSX) && !defined(OS_IOS)
   if (extra_header_size < sizeof(MachPortsExtraHeader)) {
     DLOG(ERROR) << "Decoding invalid message: " << extra_header_size << " < "
                 << sizeof(MachPortsExtraHeader);
     return nullptr;
   }
   uint32_t max_handles = (extra_header_size - sizeof(MachPortsExtraHeader)) /
       sizeof(MachPortsEntry);
-#endif
+#else
+  const uint32_t max_handles = 0;
+#endif  // defined(OS_WIN)
+
   if (header->num_handles > max_handles || max_handles > kMaxAttachedHandles) {
     DLOG(ERROR) << "Decoding invalid message:" << header->num_handles
                 << " > " << max_handles;
     return nullptr;
   }
 
-  MessagePtr message(new Message(data_num_bytes - header->num_header_bytes,
-                                 max_handles));
+  MessagePtr message(new Message(payload_size, max_handles));
   DCHECK_EQ(message->data_num_bytes(), data_num_bytes);
+
+  // Copy all payload bytes.
+  if (payload_size)
+    memcpy(message->mutable_payload(), payload, payload_size);
+
+#if !defined(MOJO_EDK_LEGACY_PROTOCOL)
   DCHECK_EQ(message->extra_header_size(), extra_header_size);
   DCHECK_EQ(message->header_->num_header_bytes, header->num_header_bytes);
 
-  if (data_num_bytes > header->num_header_bytes) {
-    // Copy all payload bytes.
-    memcpy(message->mutable_payload(),
-           static_cast<const char*>(data) + header->num_header_bytes,
-           data_num_bytes - header->num_header_bytes);
-  }
-
   if (message->extra_header_size()) {
     // Copy extra header bytes.
     memcpy(message->mutable_extra_header(),
            static_cast<const char*>(data) + sizeof(Header),
            message->extra_header_size());
   }
+#endif
 
   message->header_->num_handles = header->num_handles;
 #if defined(OS_WIN)
   ScopedPlatformHandleVectorPtr handles(
       new PlatformHandleVector(header->num_handles));
   for (size_t i = 0; i < header->num_handles; i++) {
     (*handles)[i].handle = reinterpret_cast<HANDLE>(
         static_cast<uintptr_t>(message->handles_[i].handle));
   }
   message->SetHandles(std::move(handles));
 #endif
 
   return message;
-#endif
 }
 
 size_t Channel::Message::payload_size() const {
 #if defined(MOJO_EDK_LEGACY_PROTOCOL)
   return header_->num_bytes - sizeof(Header);
 #else
   return size_ - header_->num_header_bytes;
 #endif
 }
 
@@ skipping 366 matching lines @@
 
 bool Channel::OnControlMessage(Message::Header::MessageType message_type,
                                const void* payload,
                                size_t payload_size,
                                ScopedPlatformHandleVectorPtr handles) {
   return false;
 }
 
 }  // namespace edk
 }  // namespace mojo