minidump: Add InitializeFromSnapshot() for MinidumpModuleWriter and MinidumpModuleListWriter

minidump/minidump_module_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_writer.h"
 
 #include <sys/types.h>
 
+#include <limits>
+
 #include "base/logging.h"
 #include "minidump/minidump_string_writer.h"
 #include "minidump/minidump_writer_util.h"
+#include "snapshot/module_snapshot.h"
 #include "util/file/file_writer.h"
+#include "util/numeric/in_range_cast.h"
 #include "util/numeric/safe_assignment.h"
 
 namespace crashpad {
 
 MinidumpModuleCodeViewRecordWriter::~MinidumpModuleCodeViewRecordWriter() {
 }
 
 namespace internal {
 
 template <typename CodeViewRecordType>
@@ skipping 56 matching lines @@
   codeview_record()->age = age;
 }
 
 template class internal::MinidumpModuleCodeViewRecordPDBLinkWriter<
     MinidumpModuleCodeViewRecordPDB70>;
 
 MinidumpModuleCodeViewRecordPDB70Writer::
     ~MinidumpModuleCodeViewRecordPDB70Writer() {
 }
 
+void MinidumpModuleCodeViewRecordPDB70Writer::InitializeFromSnapshot(
+    const ModuleSnapshot* module_snapshot) {
+  DCHECK_EQ(state(), kStateMutable);
+
+  std::string name = module_snapshot->Name();
+  std::string leaf_name;
+  size_t last_slash = name.find_last_of('/');
+  if (last_slash != std::string::npos) {
+    leaf_name = name.substr(last_slash + 1);
+  } else {
+    leaf_name = name;
+  }
+  SetPDBName(leaf_name);
+
+  UUID uuid;
+  module_snapshot->UUID(&uuid);
+  SetUUIDAndAge(uuid, 0);
+}
+
 MinidumpModuleMiscDebugRecordWriter::MinidumpModuleMiscDebugRecordWriter()
     : internal::MinidumpWritable(),
       image_debug_misc_(),
       data_(),
       data_utf16_() {
 }
 
 MinidumpModuleMiscDebugRecordWriter::~MinidumpModuleMiscDebugRecordWriter() {
 }
 
@@ skipping 70 matching lines @@
       name_(),
       codeview_record_(nullptr),
       misc_debug_record_(nullptr) {
   module_.VersionInfo.dwSignature = VS_FFI_SIGNATURE;
   module_.VersionInfo.dwStrucVersion = VS_FFI_STRUCVERSION;
 }
 
 MinidumpModuleWriter::~MinidumpModuleWriter() {
 }
 
+void MinidumpModuleWriter::InitializeFromSnapshot(
+    const ModuleSnapshot* module_snapshot) {
+  DCHECK_EQ(state(), kStateMutable);
+  DCHECK(!name_);
+  DCHECK(!codeview_record_);
+  DCHECK(!misc_debug_record_);
+
+  SetName(module_snapshot->Name());
+
+  SetImageBaseAddress(module_snapshot->Address());
+  SetImageSize(InRangeCast<uint32_t>(module_snapshot->Size(),
+                                     std::numeric_limits<uint32_t>::max()));
+  SetTimestamp(module_snapshot->Timestamp());
+
+  uint16_t v[4];
+  module_snapshot->FileVersion(&v[0], &v[1], &v[2], &v[3]);
+  SetFileVersion(v[0], v[1], v[2], v[3]);
+
+  module_snapshot->SourceVersion(&v[0], &v[1], &v[2], &v[3]);
+  SetProductVersion(v[0], v[1], v[2], v[3]);
+
+  uint32_t file_type;
+  switch (module_snapshot->GetModuleType()) {
+    case ModuleSnapshot::kModuleTypeExecutable:
+      file_type = VFT_APP;
+      break;
+    case ModuleSnapshot::kModuleTypeSharedLibrary:
+    case ModuleSnapshot::kModuleTypeLoadableModule:
+      file_type = VFT_DLL;
+      break;
+    default:
+      file_type = VFT_UNKNOWN;
+      break;
+  }
+  SetFileTypeAndSubtype(file_type, VFT2_UNKNOWN);
+
+  auto codeview_record =
+      make_scoped_ptr(new MinidumpModuleCodeViewRecordPDB70Writer());
+  codeview_record->InitializeFromSnapshot(module_snapshot);
+  SetCodeViewRecord(codeview_record.Pass());
+}
+
 const MINIDUMP_MODULE* MinidumpModuleWriter::MinidumpModule() const {
   DCHECK_EQ(state(), kStateWritable);
 
   return &module_;
 }
 
 void MinidumpModuleWriter::SetName(const std::string& name) {
   DCHECK_EQ(state(), kStateMutable);
 
   if (!name_) {
@@ skipping 110 matching lines @@
   return true;
 }
 
 MinidumpModuleListWriter::MinidumpModuleListWriter()
     : MinidumpStreamWriter(), module_list_base_(), modules_() {
 }
 
 MinidumpModuleListWriter::~MinidumpModuleListWriter() {
 }
 
+void MinidumpModuleListWriter::InitializeFromSnapshot(
+    const std::vector<const ModuleSnapshot*>& module_snapshots) {
+  DCHECK_EQ(state(), kStateMutable);
+  DCHECK(modules_.empty());
+
+  for (const ModuleSnapshot* module_snapshot : module_snapshots) {
+    auto module = make_scoped_ptr(new MinidumpModuleWriter());
+    module->InitializeFromSnapshot(module_snapshot);
+    AddModule(module.Pass());
+  }
+}
+
 void MinidumpModuleListWriter::AddModule(
     scoped_ptr<MinidumpModuleWriter> module) {
   DCHECK_EQ(state(), kStateMutable);
 
   modules_.push_back(module.release());
 }
 
 bool MinidumpModuleListWriter::Freeze() {
   DCHECK_EQ(state(), kStateMutable);
 
@@ skipping 42 matching lines @@
   }
 
   return file_writer->WriteIoVec(&iovecs);
 }
 
 MinidumpStreamType MinidumpModuleListWriter::StreamType() const {
   return kMinidumpStreamTypeModuleList;
 }
 
 }  // namespace crashpad