Index: third_party/crashpad/crashpad/handler/crashpad_handler_test.cc |
diff --git a/third_party/crashpad/crashpad/handler/crashpad_handler_test.cc b/third_party/crashpad/crashpad/handler/crashpad_handler_test.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..7237e8c803f2de7e620c2f1c180930e652ea61c1 |
--- /dev/null |
+++ b/third_party/crashpad/crashpad/handler/crashpad_handler_test.cc |
@@ -0,0 +1,132 @@ |
+// Copyright 2017 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 <windows.h> |
+#include <dbghelp.h> |
+#include <string.h> |
+ |
+#include <map> |
+#include <string> |
+#include <vector> |
+ |
+#include "base/files/file_path.h" |
+#include "client/crash_report_database.h" |
+#include "client/crashpad_client.h" |
+#include "gtest/gtest.h" |
+#include "minidump/test/minidump_file_writer_test_util.h" |
+#include "test/test_paths.h" |
+#include "test/win/win_multiprocess_with_temp_dir.h" |
+#include "util/file/file_reader.h" |
+ |
+namespace crashpad { |
+namespace test { |
+namespace { |
+ |
+void StartAndCrashWithExtendedHandler(const base::FilePath& temp_dir) { |
+ base::FilePath handler_path = TestPaths::Executable().DirName().Append( |
+ FILE_PATH_LITERAL("crashpad_handler_test_extended_handler.exe")); |
+ |
+ CrashpadClient client; |
+ ASSERT_TRUE(client.StartHandler(handler_path, |
+ temp_dir, |
+ base::FilePath(), |
+ "", |
+ std::map<std::string, std::string>(), |
+ std::vector<std::string>(), |
+ false, |
+ false)); |
+ |
+ __debugbreak(); |
+} |
+ |
+class CrashWithExtendedHandler final : public WinMultiprocessWithTempDir { |
+ public: |
+ CrashWithExtendedHandler() : WinMultiprocessWithTempDir() {} |
+ ~CrashWithExtendedHandler() {} |
+ |
+ private: |
+ void ValidateGeneratedDump(); |
+ |
+ void WinMultiprocessParent() override { |
+ SetExpectedChildExitCode(EXCEPTION_BREAKPOINT); |
+ } |
+ |
+ void WinMultiprocessChild() override { |
+ StartAndCrashWithExtendedHandler(GetTempDirPath()); |
+ } |
+ |
+ void WinMultiprocessParentAfterChild(HANDLE child) override { |
+ // At this point the child has exited, which means the crash report should |
+ // have been written. |
+ ValidateGeneratedDump(); |
+ |
+ // Delegate the cleanup to the superclass. |
+ WinMultiprocessWithTempDir::WinMultiprocessParentAfterChild(child); |
+ } |
+}; |
+ |
+void CrashWithExtendedHandler::ValidateGeneratedDump() { |
+ // Open the database and find the sole dump that should have been created. |
+ std::unique_ptr<CrashReportDatabase> database( |
+ CrashReportDatabase::Initialize(GetTempDirPath())); |
+ ASSERT_TRUE(database); |
+ |
+ std::vector<CrashReportDatabase::Report> reports; |
+ ASSERT_EQ(database->GetCompletedReports(&reports), |
+ CrashReportDatabase::kNoError); |
+ ASSERT_EQ(reports.size(), 1u); |
+ |
+ // Open the dump and validate that it has the extension stream with the |
+ // expected contents. |
+ FileReader reader; |
+ ASSERT_TRUE(reader.Open(reports[0].file_path)); |
+ |
+ // Read the header. |
+ MINIDUMP_HEADER header = {}; |
+ ASSERT_TRUE(reader.ReadExactly(&header, sizeof(header))); |
+ |
+ // Read the directory. |
+ std::vector<MINIDUMP_DIRECTORY> directory(header.NumberOfStreams); |
+ ASSERT_TRUE(reader.SeekSet(header.StreamDirectoryRva)); |
+ ASSERT_TRUE(reader.ReadExactly(directory.data(), |
+ directory.size() * sizeof(directory[0]))); |
+ |
+ // Search for the extension stream. |
+ size_t found_extension_streams = 0; |
+ for (const auto& entry : directory) { |
+ if (entry.StreamType == 0xCAFEBABE) { |
+ ++found_extension_streams; |
+ |
+ ASSERT_TRUE(reader.SeekSet(entry.Location.Rva)); |
+ |
+ std::vector<char> data; |
+ data.resize(entry.Location.DataSize); |
+ |
+ ASSERT_TRUE(reader.ReadExactly(data.data(), data.size())); |
+ |
+ static const char kExpectedData[] = "Injected extension stream!"; |
+ EXPECT_EQ(memcmp(kExpectedData, data.data(), sizeof(kExpectedData)), 0); |
+ } |
+ } |
+ |
+ EXPECT_EQ(found_extension_streams, 1u); |
+} |
+ |
+TEST(CrashpadHandler, ExtensibilityCalloutsWork) { |
+ WinMultiprocessWithTempDir::Run<CrashWithExtendedHandler>(); |
+} |
+ |
+} // namespace |
+} // namespace test |
+} // namespace crashpad |