minidump: Add InitializeFromSnapshot() for MinidumpModuleCrashpadInfoListWriter and downstream

minidump/minidump_module_crashpad_info_writer.cc

 // Copyright 2014 The Crashpad Authors. All rights reserved.
 //
 // Licensed under the Apache License, Version 2.0 (the "License");
 // you may not use this file except in compliance with the License.
 // You may obtain a copy of the License at
 //
 //     http://www.apache.org/licenses/LICENSE-2.0
 //
 // Unless required by applicable law or agreed to in writing, software
 // distributed under the License is distributed on an "AS IS" BASIS,
 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
 #include "minidump/minidump_module_crashpad_info_writer.h"
 
 #include <sys/types.h>
 
 #include "base/logging.h"
 #include "minidump/minidump_simple_string_dictionary_writer.h"
+#include "snapshot/module_snapshot.h"
 #include "util/file/file_writer.h"
 #include "util/numeric/safe_assignment.h"
 
 namespace crashpad {
 
 MinidumpModuleCrashpadInfoWriter::MinidumpModuleCrashpadInfoWriter()
     : MinidumpWritable(), module_(), simple_annotations_() {
   module_.version = MinidumpModuleCrashpadInfo::kVersion;
 }
 
 MinidumpModuleCrashpadInfoWriter::~MinidumpModuleCrashpadInfoWriter() {
 }
 
+void MinidumpModuleCrashpadInfoWriter::InitializeFromSnapshot(
+    const ModuleSnapshot* module_snapshot, size_t module_list_index) {
+  DCHECK_EQ(state(), kStateMutable);
+  DCHECK(!simple_annotations_);
+
+  uint32_t module_list_index_u32;
+  if (!AssignIfInRange(&module_list_index_u32, module_list_index)) {
+    LOG(ERROR) << "module_list_index " << module_list_index << " out of range";
+    return;
+  }
+  SetMinidumpModuleListIndex(module_list_index_u32);
+
+  auto simple_annotations =
+      make_scoped_ptr(new MinidumpSimpleStringDictionaryWriter());
+  simple_annotations->InitializeFromMap(
+      module_snapshot->AnnotationsSimpleMap());
+  if (simple_annotations->IsUseful()) {
+    SetSimpleAnnotations(simple_annotations.Pass());
+  }
+}
+
 void MinidumpModuleCrashpadInfoWriter::SetSimpleAnnotations(
     scoped_ptr<MinidumpSimpleStringDictionaryWriter> simple_annotations) {
   DCHECK_EQ(state(), kStateMutable);
 
   simple_annotations_ = simple_annotations.Pass();
 }
 
+bool MinidumpModuleCrashpadInfoWriter::IsUseful() const {
+  return simple_annotations_;
+}
+
 bool MinidumpModuleCrashpadInfoWriter::Freeze() {
   DCHECK_EQ(state(), kStateMutable);
 
   if (!MinidumpWritable::Freeze()) {
     return false;
   }
 
   if (simple_annotations_) {
     simple_annotations_->RegisterLocationDescriptor(
         &module_.simple_annotations);
@@ skipping 30 matching lines @@
 MinidumpModuleCrashpadInfoListWriter::MinidumpModuleCrashpadInfoListWriter()
     : MinidumpWritable(),
       module_list_base_(),
       modules_(),
       module_location_descriptors_() {
 }
 
 MinidumpModuleCrashpadInfoListWriter::~MinidumpModuleCrashpadInfoListWriter() {
 }
 
+void MinidumpModuleCrashpadInfoListWriter::InitializeFromSnapshot(
+    const std::vector<const ModuleSnapshot*>& module_snapshots) {
+  DCHECK_EQ(state(), kStateMutable);
+  DCHECK(modules_.empty());
+  DCHECK(module_location_descriptors_.empty());
+
+  size_t count = module_snapshots.size();
+  for (size_t index = 0; index < count; ++index) {
+    const ModuleSnapshot* module_snapshot = module_snapshots[index];
+
+    auto module = make_scoped_ptr(new MinidumpModuleCrashpadInfoWriter());
+    module->InitializeFromSnapshot(module_snapshot, index);
+    if (module->IsUseful()) {
+      AddModule(module.Pass());
+    }
+  }
+}
+
 void MinidumpModuleCrashpadInfoListWriter::AddModule(
     scoped_ptr<MinidumpModuleCrashpadInfoWriter> module) {
   DCHECK_EQ(state(), kStateMutable);
 
   modules_.push_back(module.release());
 }
 
 bool MinidumpModuleCrashpadInfoListWriter::Freeze() {
   DCHECK_EQ(state(), kStateMutable);
   DCHECK(module_location_descriptors_.empty());
@@ skipping 50 matching lines @@
     iov.iov_base = &module_location_descriptors_[0];
     iov.iov_len = module_location_descriptors_.size() *
                   sizeof(MINIDUMP_LOCATION_DESCRIPTOR);
     iovecs.push_back(iov);
   }
 
   return file_writer->WriteIoVec(&iovecs);
 }
 
 }  // namespace crashpad