Allocate a Dart_CMessage structure when decoding a message into C structures

runtime/vm/snapshot.cc

 // Copyright (c) 2012, the Dart project authors.  Please see the AUTHORS file
 // for details. All rights reserved. Use of this source code is governed by a
 // BSD-style license that can be found in the LICENSE file.
 
 #include "vm/snapshot.h"
 
 #include "platform/assert.h"
 #include "vm/bootstrap.h"
 #include "vm/exceptions.h"
 #include "vm/heap.h"
@@ skipping 419 matching lines @@
 #undef SNAPSHOT_READ
     default: UNREACHABLE(); break;
   }
   if (kind_ == Snapshot::kFull) {
     obj_.SetCreatedFromSnapshot();
   }
   return obj_.raw();
 }
 
 
+static uint8_t* allocator(uint8_t* ptr, intptr_t old_size, intptr_t new_size) {
+  return reinterpret_cast<uint8_t*>(realloc(ptr, new_size));
+}
+
+
 CMessageReader::CMessageReader(const uint8_t* buffer,
                                intptr_t length,
                                ReAlloc alloc)
     : BaseReader(buffer, length),
       alloc_(alloc),
-      backward_references_(kNumInitialReferences) {
+      backward_references_(kNumInitialReferences),
+      allocated_(kNumInitialReferences * 2),
+      original_message_(buffer),
+      original_message_length_(length) {
+  // Use default allocator if none specified.
+  if (alloc_ == NULL) alloc_ = allocator;
   // Initialize marker objects used to handle Lists.
   // TODO(sjesse): Remove this when message serialization format is
   // updated.
   memset(&type_arguments_marker, 0, sizeof(type_arguments_marker));
   memset(&dynamic_type_marker, 0, sizeof(dynamic_type_marker));
   type_arguments_marker.type =
       static_cast<Dart_CObject::Type>(Dart_CObject_Internal::kTypeArguments);
   dynamic_type_marker.type =
       static_cast<Dart_CObject::Type>(Dart_CObject_Internal::kDynamicType);
 }
 
 
+Dart_CMessage* CMessageReader::ReadMessage() {
+  // Read the object out of the message.
+  Dart_CObject* object = ReadObject();
+
+  intptr_t size =
+      sizeof(Dart_CMessage) + allocated_.length() * sizeof(Dart_CObject*);  // NOLINT
+  Dart_CMessage* message =
+      reinterpret_cast<Dart_CMessage*>(alloc_(NULL, 0, size));
+  message->allocated = reinterpret_cast<Dart_CObject**>(message + 1);
+  if (message == NULL) return NULL;

[siva, 2012/02/01 01:48:36]

The null check for message should happen before message is accessed in the line
above?

[Søren Gjesse, 2012/02/01 12:12:23]

Done.

+  message->object = object;
+  message->allocated_length = allocated_.length();
+  for (int i = 0; i < allocated_.length(); i++) {
+    message->allocated[i] = allocated_[i];
+  }

[siva, 2012/02/01 01:48:36]

Not sure if we need to do this copying over here. Please see comment in
dart_api.h

[Søren Gjesse, 2012/02/01 12:12:23]

Only needed if pointing back into the message and embedder is responsible for
freeing. Removed.

+  message->original_message = const_cast<uint8_t*>(original_message_);
+  message->original_message_length = original_message_length_;
+  return message;
+}
+
 intptr_t CMessageReader::LookupInternalClass(intptr_t class_header) {
   SerializedHeaderType header_type = SerializedHeaderTag::decode(class_header);
   ASSERT(header_type == kObjectId);
   intptr_t header_value = SerializedHeaderData::decode(class_header);
   return header_value;
 }
 
 
 Dart_CObject* CMessageReader::AllocateDartCObject(Dart_CObject::Type type) {
   Dart_CObject* value =
       reinterpret_cast<Dart_CObject*>(
           alloc_(NULL, 0, sizeof(Dart_CObject)));
+  allocated_.Add(value);
   value->type = type;
   return value;
 }
 
 
 Dart_CObject* CMessageReader::AllocateDartCObjectNull() {
   return AllocateDartCObject(Dart_CObject::kNull);
 }
 
 
@@ skipping 18 matching lines @@
 }
 
 
 Dart_CObject* CMessageReader::AllocateDartCObjectString(intptr_t length) {
   // Allocate a Dart_CObject structure followed by an array of chars
   // for the string content. The pointer to the string content is set
   // up to this area.
   Dart_CObject* value =
       reinterpret_cast<Dart_CObject*>(
           alloc_(NULL, 0, sizeof(Dart_CObject) + length + 1));
+  allocated_.Add(value);
   value->value.as_string = reinterpret_cast<char*>(value) + sizeof(*value);
   value->type = Dart_CObject::kString;
   return value;
 }
 
 
 Dart_CObject* CMessageReader::AllocateDartCObjectArray(intptr_t length) {
   // Allocate a Dart_CObject structure followed by an array of
   // pointers to Dart_CObject structures. The pointer to the array
   // content is set up to this area.
   Dart_CObject* value =
       reinterpret_cast<Dart_CObject*>(
           alloc_(NULL, 0, sizeof(Dart_CObject) + length * sizeof(value)));
+  allocated_.Add(value);
   value->type = Dart_CObject::kArray;
   value->value.as_array.length = length;
   if (length > 0) {
     value->value.as_array.values = reinterpret_cast<Dart_CObject**>(value + 1);
   } else {
     value->value.as_array.values = NULL;
   }
   return value;
 }
 
@@ skipping 364 matching lines @@
 
 
 void SnapshotWriterVisitor::VisitPointers(RawObject** first, RawObject** last) {
   for (RawObject** current = first; current <= last; current++) {
     RawObject* raw_obj = *current;
     writer_->WriteObject(raw_obj);
   }
 }
 
 }  // namespace dart