ExcCrashRecoverOriginalException()

util/mach/exc_server_variants_test.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 "util/mach/exc_server_variants.h"
 
 #include <mach/mach.h>
+#include <signal.h>
 #include <string.h>
 
 #include "base/basictypes.h"
 #include "base/strings/stringprintf.h"
 #include "gmock/gmock.h"
 #include "gtest/gtest.h"
 #include "util/mach/exception_behaviors.h"
 #include "util/mach/mach_extensions.h"
 #include "util/test/mac/mach_errors.h"
 #include "util/test/mac/mach_multiprocess.h"
@@ skipping 869 matching lines @@
       EXPECT_NE(kMachPortNull, thread);
       EXPECT_EQ(ChildTask(), task);
     } else {
       EXPECT_EQ(kMachPortNull, thread);
       EXPECT_EQ(kMachPortNull, task);
     }
 
     EXPECT_EQ(EXC_CRASH, exception);
     EXPECT_EQ(2u, code_count);
 
-    // The code_count check above would ideally use ASSERT_EQ so that the next
-    // conditional would not be necessary, but ASSERT_* requires a function
-    // returning type void, and the interface dictates otherwise here.
-    if (code_count >= 1) {
-      // The signal that terminated the process is stored in code[0] along with
-      // some other data. See 10.9.4 xnu-2422.110.17/bsd/kern/kern_exit.c
-      // proc_prepareexit().
-      int sig = (code[0] >> 24) & 0xff;
-      SetExpectedChildTermination(kTerminationSignal, sig);
+    // The exception and code_count checks above would ideally use ASSERT_EQ so
+    // that the next conditional would not be necessary, but ASSERT_* requires a
+    // function returning type void, and the interface dictates otherwise here.
+    if (exception == EXC_CRASH && code_count >= 1) {
+      int signal;
+      ExcCrashRecoverOriginalException(code[0], NULL, &signal);
+      SetExpectedChildTermination(kTerminationSignal, signal);
     }
 
     const bool has_state = ExceptionBehaviorHasState(behavior);
     if (has_state) {
       EXPECT_EQ(flavor_, *flavor);
       EXPECT_EQ(state_count_, old_state_count);
       EXPECT_NE(static_cast<const natural_t*>(NULL), old_state);
       EXPECT_EQ(static_cast<mach_msg_type_number_t>(THREAD_STATE_MAX),
                 *new_state_count);
       EXPECT_NE(static_cast<natural_t*>(NULL), new_state);
@@ skipping 142 matching lines @@
         base::StringPrintf("index %zu, flavor %d", index, test.flavor));
 
     TestExcServerVariants test_exc_server_variants(
         MACH_EXCEPTION_CODES | EXCEPTION_STATE_IDENTITY,
         test.flavor,
         test.count);
     test_exc_server_variants.Run();
   }
 }
 
+TEST(ExcServerVariants, ExcCrashRecoverOriginalException) {
+  struct TestData {
+    mach_exception_data_type_t code_0;
+    exception_type_t exception;
+    mach_exception_data_type_t original_code_0;
+    int signal;
+  };
+  const TestData kTestData[] = {
+    {0xb100001, EXC_BAD_ACCESS, KERN_INVALID_ADDRESS, SIGSEGV},
+    {0xb100002, EXC_BAD_ACCESS, KERN_PROTECTION_FAILURE, SIGSEGV},
+    {0xa100002, EXC_BAD_ACCESS, KERN_PROTECTION_FAILURE, SIGBUS},
+    {0x4200001, EXC_BAD_INSTRUCTION, 1, SIGILL},
+    {0x8300001, EXC_ARITHMETIC, 1, SIGFPE},
+    {0x5600002, EXC_BREAKPOINT, 2, SIGTRAP},
+    {0x6000000, 0, 0, SIGABRT},
+    {0, 0, 0, 0},
+  };
+
+  for (size_t index = 0; index < arraysize(kTestData); ++index) {
+    const TestData& test_data = kTestData[index];
+    SCOPED_TRACE(base::StringPrintf("index %zu, code_0 0x%llx",
+                                    index,
+                                    test_data.code_0));
+
+    mach_exception_data_type_t original_code_0;
+    int signal;
+    exception_type_t exception =
+        ExcCrashRecoverOriginalException(test_data.code_0,
+                                         &original_code_0,
+                                         &signal);
+
+    EXPECT_EQ(test_data.exception, exception);
+    EXPECT_EQ(test_data.original_code_0, original_code_0);
+    EXPECT_EQ(test_data.signal, signal);
+  }
+
+  // Now make sure that ExcCrashRecoverOriginalException() properly ignores
+  // optional arguments.
+  COMPILE_ASSERT(arraysize(kTestData) >= 1, must_have_something_to_test);
+  const TestData& test_data = kTestData[0];
+  EXPECT_EQ(test_data.exception,
+            ExcCrashRecoverOriginalException(test_data.code_0, NULL, NULL));
+
+  mach_exception_data_type_t original_code_0;
+  EXPECT_EQ(test_data.exception,
+            ExcCrashRecoverOriginalException(
+                test_data.code_0, &original_code_0, NULL));
+  EXPECT_EQ(test_data.original_code_0, original_code_0);
+
+  int signal;
+  EXPECT_EQ(test_data.exception,
+            ExcCrashRecoverOriginalException(test_data.code_0, NULL, &signal));
+  EXPECT_EQ(test_data.signal, signal);
+}
+
 TEST(ExcServerVariants, ExcServerSuccessfulReturnValue) {
   struct TestData {
     exception_behavior_t behavior;
     bool set_thread_state;
     kern_return_t kr;
   };
   const TestData kTestData[] = {
       {EXCEPTION_DEFAULT, false, KERN_SUCCESS},
       {EXCEPTION_STATE, false, MACH_RCV_PORT_DIED},
       {EXCEPTION_STATE_IDENTITY, false, MACH_RCV_PORT_DIED},
@@ skipping 18 matching lines @@
                            test_data.behavior,
                            test_data.set_thread_state ? "true" : "false"));
 
     EXPECT_EQ(test_data.kr,
               ExcServerSuccessfulReturnValue(test_data.behavior,
                                              test_data.set_thread_state));
   }
 }
 
 }  // namespace