Add MinidumpExceptionWriter and its test

minidump/minidump_exception_writer_test.cc

Index: minidump/minidump_exception_writer_test.cc
diff --git a/minidump/minidump_exception_writer_test.cc b/minidump/minidump_exception_writer_test.cc
new file mode 100644
index 0000000000000000000000000000000000000000..6c289650f658b07c96577f8cd0e1c162b47fd446
--- /dev/null
+++ b/minidump/minidump_exception_writer_test.cc
@@ -0,0 +1,220 @@
+// 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_exception_writer.h"
+
+#include <dbghelp.h>
+#include <stdint.h>
+
+#include <string>
+#include <vector>
+
+#include "gtest/gtest.h"
+#include "minidump/minidump_context.h"
+#include "minidump/minidump_context_test_util.h"
+#include "minidump/minidump_context_writer.h"
+#include "minidump/minidump_extensions.h"
+#include "minidump/minidump_file_writer.h"
+#include "minidump/minidump_test_util.h"
+#include "util/file/string_file_writer.h"
+
+namespace crashpad {
+namespace test {
+namespace {
+
+// This returns the MINIDUMP_EXCEPTION_STREAM stream in |exception_stream|.
+void GetExceptionStream(const std::string& file_contents,
+                        const MINIDUMP_EXCEPTION_STREAM** exception_stream) {
+  const size_t kDirectoryOffset = sizeof(MINIDUMP_HEADER);
+  const size_t kExceptionStreamOffset =
+      kDirectoryOffset + sizeof(MINIDUMP_DIRECTORY);
+  const size_t kContextOffset =
+      kExceptionStreamOffset + sizeof(MINIDUMP_EXCEPTION_STREAM);
+  const size_t kFileSize = kContextOffset + sizeof(MinidumpContextX86);
+  ASSERT_EQ(file_contents.size(), kFileSize);
+
+  const MINIDUMP_HEADER* header =
+      reinterpret_cast<const MINIDUMP_HEADER*>(&file_contents[0]);
+
+  ASSERT_NO_FATAL_FAILURE(VerifyMinidumpHeader(header, 1, 0));
+
+  const MINIDUMP_DIRECTORY* directory =
+      reinterpret_cast<const MINIDUMP_DIRECTORY*>(
+          &file_contents[kDirectoryOffset]);
+
+  ASSERT_EQ(kMinidumpStreamTypeException, directory[0].StreamType);
+  ASSERT_GE(directory[0].Location.DataSize, sizeof(MINIDUMP_EXCEPTION_STREAM));
+  ASSERT_EQ(kExceptionStreamOffset, directory[0].Location.Rva);
+
+  *exception_stream = reinterpret_cast<const MINIDUMP_EXCEPTION_STREAM*>(
+      &file_contents[kExceptionStreamOffset]);
+}
+
+// The MINIDUMP_EXCEPTION_STREAMs |expected| and |observed| are compared against
+// each other using gtest assertions. The context will be recovered from
+// |file_contents| and stored in |context|.
+void ExpectExceptionStream(const MINIDUMP_EXCEPTION_STREAM* expected,
+                           const MINIDUMP_EXCEPTION_STREAM* observed,
+                           const std::string& file_contents,
+                           const MinidumpContextX86** context) {
+  EXPECT_EQ(expected->ThreadId, observed->ThreadId);
+  EXPECT_EQ(0u, observed->__alignment);
+  EXPECT_EQ(expected->ExceptionRecord.ExceptionCode,
+            observed->ExceptionRecord.ExceptionCode);
+  EXPECT_EQ(expected->ExceptionRecord.ExceptionFlags,
+            observed->ExceptionRecord.ExceptionFlags);
+  EXPECT_EQ(expected->ExceptionRecord.ExceptionRecord,
+            observed->ExceptionRecord.ExceptionRecord);
+  EXPECT_EQ(expected->ExceptionRecord.ExceptionAddress,
+            observed->ExceptionRecord.ExceptionAddress);
+  EXPECT_EQ(expected->ExceptionRecord.NumberParameters,
+            observed->ExceptionRecord.NumberParameters);
+  EXPECT_EQ(0u, observed->ExceptionRecord.__unusedAlignment);
+  for (size_t index = 0;
+       index < arraysize(observed->ExceptionRecord.ExceptionInformation);
+       ++index) {
+    EXPECT_EQ(expected->ExceptionRecord.ExceptionInformation[index],
+              observed->ExceptionRecord.ExceptionInformation[index]);
+  }
+  EXPECT_EQ(expected->ThreadContext.DataSize, observed->ThreadContext.DataSize);
+  ASSERT_NE(0u, observed->ThreadContext.DataSize);
+  ASSERT_NE(0u, observed->ThreadContext.Rva);
+  ASSERT_GE(file_contents.size(),
+            observed->ThreadContext.Rva + observed->ThreadContext.DataSize);
+  *context = reinterpret_cast<const MinidumpContextX86*>(
+      &file_contents[observed->ThreadContext.Rva]);
+}
+
+TEST(MinidumpExceptionWriter, Minimal) {
+  MinidumpFileWriter minidump_file_writer;
+  MinidumpExceptionWriter exception_writer;
+
+  const uint32_t kSeed = 100;
+
+  MinidumpContextX86Writer context_x86_writer;
+  InitializeMinidumpContextX86(context_x86_writer.context(), kSeed);
+  exception_writer.SetContext(&context_x86_writer);
+
+  minidump_file_writer.AddStream(&exception_writer);
+
+  StringFileWriter file_writer;
+  ASSERT_TRUE(minidump_file_writer.WriteEverything(&file_writer));
+
+  const MINIDUMP_EXCEPTION_STREAM* observed_exception_stream;
+  ASSERT_NO_FATAL_FAILURE(
+      GetExceptionStream(file_writer.string(), &observed_exception_stream));
+
+  MINIDUMP_EXCEPTION_STREAM expected_exception_stream = {};
+  expected_exception_stream.ThreadContext.DataSize = sizeof(MinidumpContextX86);
+
+  const MinidumpContextX86* observed_context;
+  ASSERT_NO_FATAL_FAILURE(ExpectExceptionStream(&expected_exception_stream,
+                                                observed_exception_stream,
+                                                file_writer.string(),
+                                                &observed_context));
+
+  ASSERT_NO_FATAL_FAILURE(ExpectMinidumpContextX86(kSeed, observed_context));
+}
+
+TEST(MinidumpExceptionWriter, Standard) {
+  MinidumpFileWriter minidump_file_writer;
+  MinidumpExceptionWriter exception_writer;
+
+  const uint32_t kSeed = 200;
+  const uint32_t kThreadID = 1;
+  const uint32_t kExceptionCode = 2;
+  const uint32_t kExceptionFlags = 3;
+  const uint32_t kExceptionRecord = 4;
+  const uint32_t kExceptionAddress = 5;
+  const uint64_t kExceptionInformation0 = 6;
+  const uint64_t kExceptionInformation1 = 7;
+  const uint64_t kExceptionInformation2 = 7;
+
+  MinidumpContextX86Writer context_x86_writer;
+  InitializeMinidumpContextX86(context_x86_writer.context(), kSeed);
+  exception_writer.SetContext(&context_x86_writer);
+
+  exception_writer.SetThreadID(kThreadID);
+  exception_writer.SetExceptionCode(kExceptionCode);
+  exception_writer.SetExceptionFlags(kExceptionFlags);
+  exception_writer.SetExceptionRecord(kExceptionRecord);
+  exception_writer.SetExceptionAddress(kExceptionAddress);
+
+  // Set a lot of exception information at first, and then replace it with less.
+  // This tests that the exception that is written does not contain the
+  // “garbage” from the initial SetExceptionInformation() call.
+  std::vector<uint64_t> exception_information(EXCEPTION_MAXIMUM_PARAMETERS,
+                                              0x5a5a5a5a5a5a5a5a);
+  exception_writer.SetExceptionInformation(exception_information);
+
+  exception_information.clear();
+  exception_information.push_back(kExceptionInformation0);
+  exception_information.push_back(kExceptionInformation1);
+  exception_information.push_back(kExceptionInformation2);
+  exception_writer.SetExceptionInformation(exception_information);
+
+  minidump_file_writer.AddStream(&exception_writer);
+
+  StringFileWriter file_writer;
+  ASSERT_TRUE(minidump_file_writer.WriteEverything(&file_writer));
+
+  const MINIDUMP_EXCEPTION_STREAM* observed_exception_stream;
+  ASSERT_NO_FATAL_FAILURE(
+      GetExceptionStream(file_writer.string(), &observed_exception_stream));
+
+  MINIDUMP_EXCEPTION_STREAM expected_exception_stream = {};
+  expected_exception_stream.ThreadId = kThreadID;
+  expected_exception_stream.ExceptionRecord.ExceptionCode = kExceptionCode;
+  expected_exception_stream.ExceptionRecord.ExceptionFlags = kExceptionFlags;
+  expected_exception_stream.ExceptionRecord.ExceptionRecord = kExceptionRecord;
+  expected_exception_stream.ExceptionRecord.ExceptionAddress =
+      kExceptionAddress;
+  expected_exception_stream.ExceptionRecord.NumberParameters =
+      exception_information.size();
+  for (size_t index = 0; index < exception_information.size(); ++index) {
+    expected_exception_stream.ExceptionRecord.ExceptionInformation[index] =
+        exception_information[index];
+  }
+  expected_exception_stream.ThreadContext.DataSize = sizeof(MinidumpContextX86);
+
+  const MinidumpContextX86* observed_context;
+  ASSERT_NO_FATAL_FAILURE(ExpectExceptionStream(&expected_exception_stream,
+                                                observed_exception_stream,
+                                                file_writer.string(),
+                                                &observed_context));
+
+  ASSERT_NO_FATAL_FAILURE(ExpectMinidumpContextX86(kSeed, observed_context));
+}
+
+TEST(MinidumpExceptionWriterDeathTest, NoContext) {
+  MinidumpFileWriter minidump_file_writer;
+  MinidumpExceptionWriter exception_writer;
+
+  minidump_file_writer.AddStream(&exception_writer);
+
+  StringFileWriter file_writer;
+  ASSERT_DEATH(minidump_file_writer.WriteEverything(&file_writer), "context_");
+}
+
+TEST(MinidumpExceptionWriterDeathTest, TooMuchInformation) {
+  MinidumpExceptionWriter exception_writer;
+  std::vector<uint64_t> exception_information(EXCEPTION_MAXIMUM_PARAMETERS + 1,
+                                              0x5a5a5a5a5a5a5a5a);
+  ASSERT_DEATH(exception_writer.SetExceptionInformation(exception_information),
+               "kMaxParameters");
+}
+
+}  // namespace
+}  // namespace test
+}  // namespace crashpad