Convert WriteMessage...() to use the new user pointer handling (see r285350).

mojo/system/message_in_transit.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/system/message_in_transit.h"
 
 #include <string.h>
 
 #include "base/compiler_specific.h"
 #include "base/logging.h"
@@ skipping 74 matching lines @@
   return true;
 }
 
 MessageInTransit::MessageInTransit(Type type,
                                    Subtype subtype,
                                    uint32_t num_bytes,
                                    const void* bytes)
     : main_buffer_size_(RoundUpMessageAlignment(sizeof(Header) + num_bytes)),
       main_buffer_(static_cast<char*>(base::AlignedAlloc(main_buffer_size_,
                                                          kMessageAlignment))) {
-  DCHECK_LE(num_bytes, kMaxMessageNumBytes);
-
-  // |total_size| is updated below, from the other values.
-  header()->type = type;
-  header()->subtype = subtype;
-  header()->source_id = kInvalidEndpointId;
-  header()->destination_id = kInvalidEndpointId;
-  header()->num_bytes = num_bytes;
-  header()->unused = 0;
-  // Note: If dispatchers are subsequently attached, then |total_size| will have
-  // to be adjusted.
-  UpdateTotalSize();
-
+  ConstructorHelper(type, subtype, num_bytes);
   if (bytes) {
     memcpy(MessageInTransit::bytes(), bytes, num_bytes);
     memset(static_cast<char*>(MessageInTransit::bytes()) + num_bytes, 0,
            main_buffer_size_ - sizeof(Header) - num_bytes);
   } else {
     memset(MessageInTransit::bytes(), 0, main_buffer_size_ - sizeof(Header));
   }
 }
 
+MessageInTransit::MessageInTransit(Type type,
+                                   Subtype subtype,
+                                   uint32_t num_bytes,
+                                   UserPointer<const void> bytes)
+    : main_buffer_size_(RoundUpMessageAlignment(sizeof(Header) + num_bytes)),
+      main_buffer_(static_cast<char*>(base::AlignedAlloc(main_buffer_size_,
+                                                         kMessageAlignment))) {
+  ConstructorHelper(type, subtype, num_bytes);
+  bytes.GetArray(MessageInTransit::bytes(), num_bytes);
+}
+
 MessageInTransit::MessageInTransit(const View& message_view)
     : main_buffer_size_(message_view.main_buffer_size()),
       main_buffer_(static_cast<char*>(base::AlignedAlloc(main_buffer_size_,
                                                          kMessageAlignment))) {
   DCHECK_GE(main_buffer_size_, sizeof(Header));
   DCHECK_EQ(main_buffer_size_ % kMessageAlignment, 0u);
 
   memcpy(main_buffer_.get(), message_view.main_buffer(), main_buffer_size_);
   DCHECK_EQ(main_buffer_size_,
             RoundUpMessageAlignment(sizeof(Header) + num_bytes()));
@@ skipping 59 matching lines @@
 
   if (!dispatchers_ || !dispatchers_->size())
     return;
 
   transport_data_.reset(new TransportData(dispatchers_.Pass(), channel));
 
   // Update the sizes in the message header.
   UpdateTotalSize();
 }
 
+void MessageInTransit::ConstructorHelper(Type type,
+                                         Subtype subtype,
+                                         uint32_t num_bytes) {
+  DCHECK_LE(num_bytes, kMaxMessageNumBytes);
+
+  // |total_size| is updated below, from the other values.
+  header()->type = type;
+  header()->subtype = subtype;
+  header()->source_id = kInvalidEndpointId;
+  header()->destination_id = kInvalidEndpointId;
+  header()->num_bytes = num_bytes;
+  header()->unused = 0;
+  // Note: If dispatchers are subsequently attached, then |total_size| will have
+  // to be adjusted.
+  UpdateTotalSize();
+}
+
 void MessageInTransit::UpdateTotalSize() {
   DCHECK_EQ(main_buffer_size_ % kMessageAlignment, 0u);
   header()->total_size = static_cast<uint32_t>(main_buffer_size_);
   if (transport_data_) {
     header()->total_size +=
         static_cast<uint32_t>(transport_data_->buffer_size());
   }
 }
 
 }  // namespace system
 }  // namespace mojo