minidump: Add InitializeFromSnapshot() for MinidumpModuleWriter and MinidumpModuleListWriter

minidump/minidump_module_writer.h

 // 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,
@@ skipping 14 matching lines @@
 #include "base/basictypes.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/strings/string16.h"
 #include "minidump/minidump_extensions.h"
 #include "minidump/minidump_stream_writer.h"
 #include "minidump/minidump_writable.h"
 #include "util/stdlib/pointer_container.h"
 
 namespace crashpad {
 
+class ModuleSnapshot;
+
 namespace internal {
 class MinidumpUTF16StringWriter;
 }  // namespace internal
 
 //! \brief The base class for writers of CodeView records referenced by
 //!     MINIDUMP_MODULE::CvRecord in minidump files.
 class MinidumpModuleCodeViewRecordWriter : public internal::MinidumpWritable {
  public:
   ~MinidumpModuleCodeViewRecordWriter() override;
 
@@ skipping 65 matching lines @@
 class MinidumpModuleCodeViewRecordPDB70Writer final
     : public internal::MinidumpModuleCodeViewRecordPDBLinkWriter<
           MinidumpModuleCodeViewRecordPDB70> {
  public:
   MinidumpModuleCodeViewRecordPDB70Writer()
       : internal::MinidumpModuleCodeViewRecordPDBLinkWriter<
             MinidumpModuleCodeViewRecordPDB70>() {}
 
   ~MinidumpModuleCodeViewRecordPDB70Writer() override;
 
+  //! \brief Initializes the MinidumpModuleCodeViewRecordPDB70 based on \a
+  //!     module_snapshot.
+  //!
+  //! \param[in] module_snapshot The module snapshot to use as source data.
+  //!
+  //! \note Valid in #kStateMutable. No mutator methods may be called before
+  //!     this method, and it is not normally necessary to call any mutator
+  //!     methods after this method.
+  void InitializeFromSnapshot(const ModuleSnapshot* module_snapshot);
+
   //! \brief Sets MinidumpModuleCodeViewRecordPDB70::uuid and
   //!     MinidumpModuleCodeViewRecordPDB70::age.
   void SetUUIDAndAge(const UUID& uuid, uint32_t age) {
     codeview_record()->uuid = uuid;
     codeview_record()->age = age;
   }
 
  private:
   DISALLOW_COPY_AND_ASSIGN(MinidumpModuleCodeViewRecordPDB70Writer);
 };
@@ skipping 39 matching lines @@
 //!
 //! Because MINIDUMP_MODULE objects only appear as elements of
 //! MINIDUMP_MODULE_LIST objects, this class does not write any data on its own.
 //! It makes its MINIDUMP_MODULE data available to its MinidumpModuleListWriter
 //! parent, which writes it as part of a MINIDUMP_MODULE_LIST.
 class MinidumpModuleWriter final : public internal::MinidumpWritable {
  public:
   MinidumpModuleWriter();
   ~MinidumpModuleWriter() override;
 
+  //! \brief Initializes the MINIDUMP_MODULE based on \a module_snapshot.
+  //!
+  //! \param[in] module_snapshot The module snapshot to use as source data.
+  //!
+  //! \note Valid in #kStateMutable. No mutator methods may be called before
+  //!     this method, and it is not normally necessary to call any mutator
+  //!     methods after this method.
+  void InitializeFromSnapshot(const ModuleSnapshot* module_snapshot);
+
   //! \brief Returns a MINIDUMP_MODULE referencing this object’s data.
   //!
   //! This method is expected to be called by a MinidumpModuleListWriter in
   //! order to obtain a MINIDUMP_MODULE to include in its list.
   //!
   //! \note Valid in #kStateWritable.
   const MINIDUMP_MODULE* MinidumpModule() const;
 
   //! \brief Arranges for MINIDUMP_MODULE::ModuleNameRva to point to a
   //!     MINIDUMP_STRING containing \a name.
@@ skipping 98 matching lines @@
   DISALLOW_COPY_AND_ASSIGN(MinidumpModuleWriter);
 };
 
 //! \brief The writer for a MINIDUMP_MODULE_LIST stream in a minidump file,
 //!     containing a list of MINIDUMP_MODULE objects.
 class MinidumpModuleListWriter final : public internal::MinidumpStreamWriter {
  public:
   MinidumpModuleListWriter();
   ~MinidumpModuleListWriter() override;
 
+  //! \brief Adds an initialized MINIDUMP_MODULE for each module in \a
+  //!     module_snapshots to the MINIDUMP_MODULE_LIST.
+  //!
+  //! \param[in] module_snapshots The module snapshots to use as source data.
+  //!
+  //! \note Valid in #kStateMutable. No mutator methods may be called before
+  //!     this method, and it is not normally necessary to call any mutator
+  //!     methods after this method.
+  void InitializeFromSnapshot(
+      const std::vector<const ModuleSnapshot*>& module_snapshots);
+
   //! \brief Adds a MinidumpModuleWriter to the MINIDUMP_MODULE_LIST.
   //!
   //! This object takes ownership of \a module and becomes its parent in the
   //! overall tree of internal::MinidumpWritable objects.
   //!
   //! \note Valid in #kStateMutable.
   void AddModule(scoped_ptr<MinidumpModuleWriter> module);
 
  protected:
   // MinidumpWritable:
   bool Freeze() override;
   size_t SizeOfObject() override;
   std::vector<MinidumpWritable*> Children() override;
   bool WriteObject(FileWriterInterface* file_writer) override;
 
   // MinidumpStreamWriter:
   MinidumpStreamType StreamType() const override;
 
  private:
   MINIDUMP_MODULE_LIST module_list_base_;
   PointerVector<MinidumpModuleWriter> modules_;
 
   DISALLOW_COPY_AND_ASSIGN(MinidumpModuleListWriter);
 };
 
 }  // namespace crashpad
 
 #endif  // CRASHPAD_MINIDUMP_MINIDUMP_MODULE_WRITER_H_