chrome_frame/crash_reporting/veh_test.cc

   1 // Copyright (c) 2010 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include <atlbase.h>
   6 #include "chrome_frame/crash_reporting/veh_test.h"
   7 #include "testing/gmock/include/gmock/gmock.h"
   8 #include "testing/gtest/include/gtest/gtest.h"
   9 #include "chrome_frame/crash_reporting/vectored_handler-impl.h"
  10
  11 #pragma code_seg(push, ".m$_0")
  12 static void ModuleStart() {}
  13 #pragma code_seg(pop)
  14
  15 #pragma code_seg(push, ".m$_2")
  16 DECLSPEC_NOINLINE static void Undetectable(DWORD code) {
  17   __try {
  18     ::RaiseException(code, 0, 0, NULL);
  19   } __except(EXCEPTION_EXECUTE_HANDLER) {
  20
  21   }
  22 };
  23 #pragma code_seg(pop)
  24
  25 #pragma code_seg(push, ".m$_3")
  26 static void UndetectableEnd() {}
  27 #pragma code_seg(pop)
  28
  29 #pragma code_seg(push, ".m$_4")
  30 DECLSPEC_NOINLINE static void CatchThis() {
  31   __try {
  32     ::RaiseException(STATUS_ACCESS_VIOLATION, 0, 0, NULL);
  33   } __except(EXCEPTION_EXECUTE_HANDLER) {
  34
  35   }
  36
  37   // this will be detected since we are on the stack!
  38   Undetectable(STATUS_ILLEGAL_INSTRUCTION);
  39 }
  40 #pragma code_seg(pop)
  41
  42 #pragma code_seg(push, ".m$_9")
  43 static void ModuleEnd() {}
  44 #pragma code_seg(pop)
  45
  46 using testing::_;
  47 namespace {
  48 MATCHER_P(ExceptionCodeIs, code, "") {
  49   return (arg->ExceptionRecord->ExceptionCode == code);
  50 }
  51
  52 class MockApi : public Win32VEHTraits,
  53                 public ModuleOfInterestWithExcludedRegion {
  54  public:
  55   MockApi() {
  56     ModuleOfInterestWithExcludedRegion::SetModule(&ModuleStart, &ModuleEnd);
  57     ModuleOfInterestWithExcludedRegion::SetExcludedRegion(&Undetectable,
  58         &UndetectableEnd);
  59   }
  60
  61   MOCK_METHOD1(WriteDump, void(const EXCEPTION_POINTERS*));
  62   MOCK_METHOD0(RtlpGetExceptionList, const EXCEPTION_REGISTRATION_RECORD*());
  63 };
  64 };  // namespace
  65
  66 typedef VectoredHandlerT<MockApi> VectoredHandlerMock;
  67
  68 #pragma optimize("y", off)
  69 static VectoredHandlerMock* g_mock_veh = NULL;
  70 static LONG WINAPI VEH(EXCEPTION_POINTERS* exptrs) {
  71   return g_mock_veh->Handler(exptrs);
  72 }
  73 #pragma optimize("y", on)
  74
  75 TEST(ChromeFrame, ExceptionExcludedCode) {
  76   MockApi api;
  77   VectoredHandlerMock veh(&api);
  78
  79   g_mock_veh = &veh;
  80   void* id = ::AddVectoredExceptionHandler(FALSE, VEH);
  81
  82   EXPECT_CALL(api, RtlpGetExceptionList())
  83       .WillRepeatedly(testing::Return(EXCEPTION_CHAIN_END));
  84
  85   testing::Sequence s;
  86
  87   EXPECT_CALL(api, WriteDump(ExceptionCodeIs(STATUS_ACCESS_VIOLATION)))
  88     .Times(1);
  89
  90   EXPECT_CALL(api, WriteDump(ExceptionCodeIs(STATUS_ILLEGAL_INSTRUCTION)))
  91       .Times(1);
  92
  93   CatchThis();
  94   EXPECT_EQ(2, veh.get_exceptions_seen());
  95
  96   // Not detected since we are not on the stack.
  97   Undetectable(STATUS_INTEGER_DIVIDE_BY_ZERO);
  98   EXPECT_EQ(3, veh.get_exceptions_seen());
  99
 100   ::RemoveVectoredExceptionHandler(id);
 101   g_mock_veh = NULL;
 102 }