[mojo-edk] Add explicit message object APIs

mojo/edk/system/core_test_base.cc

 // Copyright 2013 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/core_test_base.h"
 
 #include <stddef.h>
 #include <stdint.h>
 
 #include <vector>
 
 #include "base/logging.h"
 #include "base/macros.h"
 #include "base/memory/ref_counted.h"
 #include "mojo/edk/embedder/embedder_internal.h"
 #include "mojo/edk/system/configuration.h"
 #include "mojo/edk/system/core.h"
 #include "mojo/edk/system/dispatcher.h"
+#include "mojo/edk/system/message_for_transit.h"
 
 namespace mojo {
 namespace edk {
 namespace test {
 
 namespace {
 
 // MockDispatcher --------------------------------------------------------------
 
 class MockDispatcher : public Dispatcher {
  public:
   static scoped_refptr<MockDispatcher> Create(
       CoreTestBase::MockHandleInfo* info) {
     return make_scoped_refptr(new MockDispatcher(info));
   }
 
   // Dispatcher:
   Type GetType() const override { return Type::UNKNOWN; }
 
   MojoResult Close() override {
     info_->IncrementCloseCallCount();
     return MOJO_RESULT_OK;
   }
 
   MojoResult WriteMessage(
-      const void* bytes,
-      uint32_t num_bytes,
-      const DispatcherInTransit* dispatchers,
-      uint32_t num_dispatchers,
+      std::unique_ptr<MessageForTransit> message,
       MojoWriteMessageFlags /*flags*/) override {
     info_->IncrementWriteMessageCallCount();
 
-    if (num_bytes > GetConfiguration().max_message_num_bytes)
+    if (message->num_bytes() > GetConfiguration().max_message_num_bytes)
       return MOJO_RESULT_RESOURCE_EXHAUSTED;
 
-    if (dispatchers)
+    if (message->num_handles())
       return MOJO_RESULT_UNIMPLEMENTED;
 
     return MOJO_RESULT_OK;
   }
 
-  MojoResult ReadMessage(void* bytes,
+  MojoResult ReadMessage(std::unique_ptr<MessageForTransit>* message,
                          uint32_t* num_bytes,
                          MojoHandle* handle,
                          uint32_t* num_handles,
-                         MojoReadMessageFlags /*flags*/) override {
+                         MojoReadMessageFlags /*flags*/,
+                         bool ignore_num_bytes) override {
     info_->IncrementReadMessageCallCount();
 
     if (num_handles)
       *num_handles = 1;
 
     return MOJO_RESULT_OK;
   }
 
   MojoResult WriteData(const void* elements,
                        uint32_t* num_bytes,
@@ skipping 260 matching lines @@
 }
 
 void CoreTestBase_MockHandleInfo::AwakableWasAdded(Awakable* awakable) {
   base::AutoLock locker(lock_);
   added_awakables_.push_back(awakable);
 }
 
 }  // namespace test
 }  // namespace edk
 }  // namespace mojo