| Index: components/browser_watcher/postmortem_minidump_writer_win_unittest.cc
|
| diff --git a/components/browser_watcher/postmortem_minidump_writer_win_unittest.cc b/components/browser_watcher/postmortem_minidump_writer_win_unittest.cc
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..c93e5cadabe3913255ba637286db2edbc8bb7850
|
| --- /dev/null
|
| +++ b/components/browser_watcher/postmortem_minidump_writer_win_unittest.cc
|
| @@ -0,0 +1,71 @@
|
| +// Copyright 2016 The Chromium Authors. All rights reserved.
|
| +// Use of this source code is governed by a BSD-style license that can be
|
| +// found in the LICENSE file.
|
| +
|
| +#include "components/browser_watcher/postmortem_minidump_writer.h"
|
| +
|
| +#include <windows.h> // NOLINT
|
| +#include <dbghelp.h>
|
| +
|
| +#include "base/files/file_path.h"
|
| +#include "base/files/file_util.h"
|
| +#include "base/files/scoped_temp_dir.h"
|
| +#include "components/browser_watcher/stability_report.pb.h"
|
| +#include "testing/gtest/include/gtest/gtest.h"
|
| +
|
| +namespace browser_watcher {
|
| +
|
| +TEST(PostmortemMinidumpWriterWinTest, WriteDumpBasicTest) {
|
| + // Create a stability report.
|
| + // TODO(manzagop): flesh out the report once proto is more detailed.
|
| + StabilityReport expected_report;
|
| + ProcessState* process_state = expected_report.add_process_states();
|
| + CodeModule* module = process_state->add_modules();
|
| + module->set_base_address(1024);
|
| + module->set_code_file("some_code_file.dll");
|
| +
|
| + // Write to minidump.
|
| + base::ScopedTempDir temp_dir;
|
| + ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
|
| + base::FilePath minidump_path = temp_dir.path().AppendASCII("minidump.dmp");
|
| + {
|
| + // Note: the inner scope to ensure the writer goes out of scope, which will
|
| + // close the minidump file.
|
| + PostmortemMinidumpWriter writer;
|
| + ASSERT_TRUE(writer.WriteDump(minidump_path, expected_report));
|
| + }
|
| +
|
| + // Read back the minidump.
|
| + base::ScopedFILE minidump_file;
|
| + minidump_file.reset(base::OpenFile(minidump_path, "rb"));
|
| + ASSERT_TRUE(minidump_file.get());
|
| +
|
| + MINIDUMP_HEADER header = {};
|
| + ASSERT_EQ(1, fread(&header, sizeof(header), 1, minidump_file.get()));
|
| + ASSERT_EQ(MINIDUMP_SIGNATURE, header.Signature);
|
| + ASSERT_EQ(1, header.NumberOfStreams);
|
| + RVA directory_rva = header.StreamDirectoryRva;
|
| +
|
| + MINIDUMP_DIRECTORY directory = {};
|
| + ASSERT_EQ(0, fseek(minidump_file.get(), directory_rva, SEEK_SET));
|
| + ASSERT_EQ(1, fread(&directory, sizeof(directory), 1, minidump_file.get()));
|
| + ASSERT_EQ(0x4B6B0002, directory.StreamType);
|
| + RVA report_rva = directory.Location.Rva;
|
| + ULONG32 report_size_bytes = directory.Location.DataSize;
|
| +
|
| + std::string recovered_serialized_report;
|
| + recovered_serialized_report.resize(report_size_bytes);
|
| + ASSERT_EQ(0, fseek(minidump_file.get(), report_rva, SEEK_SET));
|
| + ASSERT_EQ(report_size_bytes, fread(&recovered_serialized_report.at(0), 1,
|
| + report_size_bytes, minidump_file.get()));
|
| +
|
| + // Validate the recovered report.
|
| + std::string expected_serialized_report;
|
| + expected_report.SerializeToString(&expected_serialized_report);
|
| + ASSERT_EQ(expected_serialized_report, recovered_serialized_report);
|
| +
|
| + StabilityReport recovered_report;
|
| + ASSERT_TRUE(recovered_report.ParseFromString(recovered_serialized_report));
|
| +}
|
| +
|
| +} // namespace browser_watcher
|
|
|