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