src: use std::vector<size_t> instead of IndexArray

joyeecheung · Apr 19, 2019 · Apr 18, 2019 · Apr 18, 2019 · Apr 18, 2019 · Apr 20, 2019
commit 2ac1d81cc208e7baee6e38ddc1f82a43109d43be
diff --git a/src/env.cc b/src/env.cc
@@ -75,8 +75,7 @@ std::vector<size_t> IsolateData::Serialize(SnapshotCreator* creator) {
   return indexes;
 }
 
-void IsolateData::DeserializeProperties(
-    const NodeMainInstance::IndexArray* indexes) {
+void IsolateData::DeserializeProperties(const std::vector<size_t>* indexes) {
   size_t i = 0;
   HandleScope handle_scope(isolate_);
 
@@ -86,7 +85,7 @@ void IsolateData::DeserializeProperties(
 #define V(TypeName, PropertyName)                                              \
   do {                                                                         \
     MaybeLocal<TypeName> field =                                               \
-        isolate_->GetDataFromSnapshotOnce<TypeName>(indexes->Get(i++));        \
+        isolate_->GetDataFromSnapshotOnce<TypeName>((*indexes)[i++]);          \
     if (field.IsEmpty()) {                                                     \
       fprintf(stderr, "Failed to deserialize " #PropertyName "\n");            \
     }                                                                          \
@@ -155,7 +154,7 @@ IsolateData::IsolateData(Isolate* isolate,
                          uv_loop_t* event_loop,
                          MultiIsolatePlatform* platform,
                          ArrayBufferAllocator* node_allocator,
-                         const NodeMainInstance::IndexArray* indexes)
+                         const std::vector<size_t>* indexes)
     : isolate_(isolate),
       event_loop_(event_loop),
       allocator_(isolate->GetArrayBufferAllocator()),

diff --git a/src/env.h b/src/env.h
@@ -420,7 +420,7 @@ class IsolateData : public MemoryRetainer {
               uv_loop_t* event_loop,
               MultiIsolatePlatform* platform = nullptr,
               ArrayBufferAllocator* node_allocator = nullptr,
-              const NodeMainInstance::IndexArray* indexes = nullptr);
+              const std::vector<size_t>* indexes = nullptr);
   SET_MEMORY_INFO_NAME(IsolateData);
   SET_SELF_SIZE(IsolateData);
   void MemoryInfo(MemoryTracker* tracker) const override;
@@ -454,7 +454,7 @@ class IsolateData : public MemoryRetainer {
   IsolateData& operator=(const IsolateData&) = delete;
 
  private:
-  void DeserializeProperties(const NodeMainInstance::IndexArray* indexes);
+  void DeserializeProperties(const std::vector<size_t>* indexes);
   void CreateProperties();
 
 #define VP(PropertyName, StringValue) V(v8::Private, PropertyName)

diff --git a/src/node.cc b/src/node.cc
@@ -891,7 +891,7 @@ int Start(int argc, char** argv) {
     std::vector<intptr_t> external_references = {
         reinterpret_cast<intptr_t>(nullptr)};
     v8::StartupData* blob = NodeMainInstance::GetEmbeddedSnapshotBlob();
-    const NodeMainInstance::IndexArray* indexes =
+    const std::vector<size_t>* indexes =
         NodeMainInstance::GetIsolateDataIndexes();
     if (blob != nullptr) {
       params.external_references = external_references.data();

diff --git a/src/node_main_instance.cc b/src/node_main_instance.cc
@@ -40,12 +40,13 @@ NodeMainInstance* NodeMainInstance::Create(
   return new NodeMainInstance(isolate, event_loop, platform, args, exec_args);
 }
 
-NodeMainInstance::NodeMainInstance(Isolate::CreateParams* params,
-                                   uv_loop_t* event_loop,
-                                   MultiIsolatePlatform* platform,
-                                   const std::vector<std::string>& args,
-                                   const std::vector<std::string>& exec_args,
-                                   const IndexArray* per_isolate_data_indexes)
+NodeMainInstance::NodeMainInstance(
+    Isolate::CreateParams* params,
+    uv_loop_t* event_loop,
+    MultiIsolatePlatform* platform,
+    const std::vector<std::string>& args,
+    const std::vector<std::string>& exec_args,
+    const std::vector<size_t>* per_isolate_data_indexes)
     : args_(args),
       exec_args_(exec_args),
       array_buffer_allocator_(ArrayBufferAllocator::Create()),

diff --git a/src/node_main_instance.h b/src/node_main_instance.h
@@ -15,18 +15,6 @@ namespace node {
 // We may be able to create an abstract class to reuse some of the routines.
 class NodeMainInstance {
  public:
-  // An array of indexes that can be used to deserialize data from a V8
-  // snapshot.
-  struct IndexArray {
-    const size_t* data;
-    size_t length;
-
-    size_t Get(size_t index) const {
-      DCHECK_LT(index, length);
-      return data[index];
-    }
-  };
-
   // To create a main instance that does not own the isoalte,
   // The caller needs to do:
   //
@@ -53,12 +41,13 @@ class NodeMainInstance {
   void Dispose();
 
   // Create a main instance that owns the isolate
-  NodeMainInstance(v8::Isolate::CreateParams* params,
-                   uv_loop_t* event_loop,
-                   MultiIsolatePlatform* platform,
-                   const std::vector<std::string>& args,
-                   const std::vector<std::string>& exec_args,
-                   const IndexArray* per_isolate_data_indexes = nullptr);
+  NodeMainInstance(
+      v8::Isolate::CreateParams* params,
+      uv_loop_t* event_loop,
+      MultiIsolatePlatform* platform,
+      const std::vector<std::string>& args,
+      const std::vector<std::string>& exec_args,
+      const std::vector<size_t>* per_isolate_data_indexes = nullptr);
   ~NodeMainInstance();
 
   // Start running the Node.js instances, return the exit code when finished.
@@ -72,7 +61,7 @@ class NodeMainInstance {
 
   // If nullptr is returned, the binary is not built with embedded
   // snapshot.
-  static const IndexArray* GetIsolateDataIndexes();
+  static const std::vector<size_t>* GetIsolateDataIndexes();
   static v8::StartupData* GetEmbeddedSnapshotBlob();
 
   static const size_t kNodeContextIndex;

diff --git a/src/node_snapshot_stub.cc b/src/node_snapshot_stub.cc
@@ -6,7 +6,7 @@ v8::StartupData* NodeMainInstance::GetEmbeddedSnapshotBlob() {
   return nullptr;
 }
 
-const NodeMainInstance::IndexArray* NodeMainInstance::GetIsolateDataIndexes() {
+const std::vector<size_t>* NodeMainInstance::GetIsolateDataIndexes() {
   return nullptr;
 }
 

diff --git a/tools/snapshot/snapshot_builder.cc b/tools/snapshot/snapshot_builder.cc
@@ -16,10 +16,16 @@ using v8::Locker;
 using v8::SnapshotCreator;
 using v8::StartupData;
 
+template <typename T>
+void WriteVector(std::stringstream* ss, const T* vec, size_t size) {
+  for (size_t i = 0; i < size; i++) {
+    *ss << std::to_string(vec[i]) << (i == size - 1 ? '\n' : ',');
+  }
+}
+
 std::string FormatBlob(v8::StartupData* blob,
                        const std::vector<size_t>& isolate_data_indexes) {
   std::stringstream ss;
-  size_t isolate_data_indexes_size = isolate_data_indexes.size();
 
   ss << R"(#include <cinttypes>
 #include <array>
@@ -30,47 +36,26 @@ std::string FormatBlob(v8::StartupData* blob,
 
 namespace node {
 
-static const uint8_t blob_data[] = {
+static const char blob_data[] = {
 )";
-
-  for (int i = 0; i < blob->raw_size; i++) {
-    uint8_t ch = blob->data[i];
-    ss << std::to_string(ch) << ((i == blob->raw_size - 1) ? '\n' : ',');
-  }
-
+  WriteVector(&ss, blob->data, blob->raw_size);
   ss << R"(};
 
 static const int blob_size = )"
      << blob->raw_size << R"(;
-static v8::StartupData blob = {
-  reinterpret_cast<const char*>(blob_data),
-  blob_size
-};
+static v8::StartupData blob = { blob_data, blob_size };
 )";
 
-  ss << R"(v8::StartupData*
-NodeMainInstance::GetEmbeddedSnapshotBlob() {
+  ss << R"(v8::StartupData* NodeMainInstance::GetEmbeddedSnapshotBlob() {
   return &blob;
 }
 
-static const size_t isolate_data_indexes_raw[] = {
+static const std::vector<size_t> isolate_data_indexes {
 )";
-  for (size_t i = 0; i < isolate_data_indexes_size; i++) {
-    ss << std::to_string(isolate_data_indexes[i])
-       << ((i == isolate_data_indexes_size - 1) ? '\n' : ',');
-  }
-  ss << "};\n\n";
-
-  ss << "static const size_t isolate_data_indexes_size = "
-     << isolate_data_indexes_size << R"(;
-
-NodeMainInstance::IndexArray isolate_data_indexes {
-    isolate_data_indexes_raw,
-    isolate_data_indexes_size
-};
+  WriteVector(&ss, isolate_data_indexes.data(), isolate_data_indexes.size());
+  ss << R"(};
 
-const NodeMainInstance::IndexArray*
-NodeMainInstance::GetIsolateDataIndexes() {
+const std::vector<size_t>* NodeMainInstance::GetIsolateDataIndexes() {
   return &isolate_data_indexes;
 }
 }  // namespace node