OLD | NEW |
| (Empty) |
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 } | |
OLD | NEW |