blob: e9b5946a31e29d637b49305cd4e5817f3de42f6e (
plain)
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
|
#include <stdio.h>
#include <windows.h>
#include "testwinlib.h"
int exceptionCount;
int continueCount;
int unhandledCount;
LONG WINAPI customExceptionHandlder(struct _EXCEPTION_POINTERS *ExceptionInfo)
{
if (ExceptionInfo->ExceptionRecord->ExceptionCode == EXCEPTION_BREAKPOINT)
{
exceptionCount++;
// prepare context to resume execution
CONTEXT *c = ExceptionInfo->ContextRecord;
#ifdef _AMD64_
c->Rip = *(DWORD64 *)c->Rsp;
c->Rsp += 8;
#elif defined(_X86_)
c->Eip = *(DWORD *)c->Esp;
c->Esp += 4;
#else
c->Pc = c->Lr;
#endif
#ifdef _ARM64_
// TODO: remove when windows/arm64 supports SEH stack unwinding.
return EXCEPTION_CONTINUE_EXECUTION;
#endif
}
return EXCEPTION_CONTINUE_SEARCH;
}
LONG WINAPI customContinueHandlder(struct _EXCEPTION_POINTERS *ExceptionInfo)
{
if (ExceptionInfo->ExceptionRecord->ExceptionCode == EXCEPTION_BREAKPOINT)
{
continueCount++;
}
return EXCEPTION_CONTINUE_SEARCH;
}
LONG WINAPI unhandledExceptionHandler(struct _EXCEPTION_POINTERS *ExceptionInfo) {
if (ExceptionInfo->ExceptionRecord->ExceptionCode == EXCEPTION_BREAKPOINT)
{
unhandledCount++;
return EXCEPTION_CONTINUE_EXECUTION;
}
return EXCEPTION_CONTINUE_SEARCH;
}
void throwFromC()
{
DebugBreak();
}
int main()
{
// simulate a "lazily" attached debugger, by calling some go code before attaching the exception/continue handler
Dummy();
exceptionCount = 0;
continueCount = 0;
void *exceptionHandlerHandle = AddVectoredExceptionHandler(0, customExceptionHandlder);
if (NULL == exceptionHandlerHandle)
{
printf("cannot add vectored exception handler\n");
fflush(stdout);
return 2;
}
void *continueHandlerHandle = AddVectoredContinueHandler(0, customContinueHandlder);
if (NULL == continueHandlerHandle)
{
printf("cannot add vectored continue handler\n");
fflush(stdout);
return 2;
}
void *prevUnhandledHandler = SetUnhandledExceptionFilter(unhandledExceptionHandler);
CallMeBack(throwFromC);
RemoveVectoredContinueHandler(continueHandlerHandle);
RemoveVectoredExceptionHandler(exceptionHandlerHandle);
if (prevUnhandledHandler != NULL)
{
SetUnhandledExceptionFilter(prevUnhandledHandler);
}
printf("exceptionCount: %d\ncontinueCount: %d\nunhandledCount: %d\n", exceptionCount, continueCount, unhandledCount);
fflush(stdout);
return 0;
}
|
