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
104
|
#include <ntddk.h>
#include <DriverThread.hpp>
class TestSmth
{
public:
TestSmth()
{
DbgPrint("%s\n", "ctor");
}
~TestSmth()
{
DbgPrint("%s\n", "dtor");
}
void doSmth(void)
{
DbgPrint("%s\n", "Hello Class!");
}
};
static TestSmth * cdtor_test;
class Derived : public TestSmth
{
public:
Derived()
{
}
~Derived()
{
}
void doSmth(void)
{
DbgPrint("%s\n", "Hello Derived!");
}
};
struct threadContext
{
DriverThread::Semaphore sem;
DriverThread::Thread dth;
};
static NTSTATUS threadRoutine(PVOID threadContext)
{
DbgPrint("ThreadRoutine %p, ThreadContext: %p\n", threadRoutine, threadContext);
for (size_t i = 3; i > 0; --i)
{
DbgPrint("ThreadLoop: %zu\n", i);
}
struct threadContext * const ctx = (struct threadContext *)threadContext;
DbgPrint("Fin. ThreadId: %p\n", ctx->dth.GetThreadId());
ctx->sem.Release();
DbgPrint("Thread WaitForTermination: 0x%X\n", ctx->dth.WaitForTermination()); // must return STATUS_UNSUCCESSFUL;
return STATUS_SUCCESS;
}
static void test_cplusplus(void)
{
TestSmth t;
t.doSmth();
Derived d;
d.doSmth();
struct threadContext ctx;
ctx.dth.Start(threadRoutine, (PVOID)&ctx);
ctx.sem.Wait();
DbgPrint("MainThread semaphore signaled.\n");
ctx.dth.WaitForTermination();
ctx.dth.WaitForTermination();
DbgPrint("MainThread EOF\n");
}
extern "C"
{
DRIVER_INITIALIZE DriverEntry;
DRIVER_UNLOAD DriverUnload;
NTSTATUS DriverEntry(_In_ struct _DRIVER_OBJECT * DriverObject, _In_ PUNICODE_STRING RegistryPath)
{
(void)DriverObject;
(void)RegistryPath;
DbgPrint("%s\n", "Hello ring0!");
cdtor_test = new TestSmth();
/* support for service stopping */
DriverObject->DriverUnload = DriverUnload;
test_cplusplus();
return STATUS_SUCCESS;
}
VOID DriverUnload(_In_ struct _DRIVER_OBJECT * DriverObject)
{
(void)DriverObject;
delete cdtor_test;
DbgPrint("%s\n", "Bye ring0!");
}
}
|