Add MinidumpCrashpadInfoWriter::InitializeFromSnapshot() and its test

minidump/minidump_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_crashpad_info_writer.h"
 
 #include "base/logging.h"
 #include "minidump/minidump_module_crashpad_info_writer.h"
+#include "snapshot/process_snapshot.h"
 #include "util/file/file_writer.h"
 
 namespace crashpad {
 
 MinidumpCrashpadInfoWriter::MinidumpCrashpadInfoWriter()
     : MinidumpStreamWriter(), crashpad_info_(), module_list_(nullptr) {
   crashpad_info_.version = MinidumpCrashpadInfo::kVersion;
 }
 
 MinidumpCrashpadInfoWriter::~MinidumpCrashpadInfoWriter() {
 }
 
+void MinidumpCrashpadInfoWriter::InitializeFromSnapshot(
+    const ProcessSnapshot* process_snapshot) {
+  DCHECK_EQ(state(), kStateMutable);
+  DCHECK(!module_list_);
+
+  auto modules = make_scoped_ptr(new MinidumpModuleCrashpadInfoListWriter());
+  modules->InitializeFromSnapshot(process_snapshot->Modules());
+
+  if (modules->IsUseful()) {
+    SetModuleList(modules.Pass());
+  }
+}
+
 void MinidumpCrashpadInfoWriter::SetModuleList(
     scoped_ptr<MinidumpModuleCrashpadInfoListWriter> module_list) {
   DCHECK_EQ(state(), kStateMutable);
 
   module_list_ = module_list.Pass();
 }
 
 bool MinidumpCrashpadInfoWriter::Freeze() {
   DCHECK_EQ(state(), kStateMutable);
 
@@ skipping 29 matching lines @@
 bool MinidumpCrashpadInfoWriter::WriteObject(FileWriterInterface* file_writer) {
   DCHECK_EQ(state(), kStateWritable);
 
   return file_writer->Write(&crashpad_info_, sizeof(crashpad_info_));
 }
 
 MinidumpStreamType MinidumpCrashpadInfoWriter::StreamType() const {
   return kMinidumpStreamTypeCrashpadInfo;
 }
 
+bool MinidumpCrashpadInfoWriter::IsUseful() const {
+  return module_list_;
+}
+
 }  // namespace crashpad