1 | #include "stdafx.h"
|
---|
2 |
|
---|
3 | static OSVERSIONINFO GetVersionEx2()
|
---|
4 | {
|
---|
5 | OSVERSIONINFO vi = {sizeof vi};
|
---|
6 | GetVersionEx(&vi);
|
---|
7 | return vi;
|
---|
8 | }
|
---|
9 |
|
---|
10 | static OSVERSIONINFO const vi = GetVersionEx2();
|
---|
11 |
|
---|
12 | #define KERNEL32 (GetModuleHandle(TEXT("kernel32")))
|
---|
13 |
|
---|
14 | #define GET_PROC_ADDRESS(hmod, functionName) reinterpret_cast<decltype(functionName)*>(GetProcAddress((hmod), #functionName))
|
---|
15 |
|
---|
16 | extern "C"
|
---|
17 | {
|
---|
18 | BOOL WINAPI IsDebuggerPresent_Helper()
|
---|
19 | {
|
---|
20 | if (auto IsDebuggerPresent_Real = GET_PROC_ADDRESS(KERNEL32, IsDebuggerPresent))
|
---|
21 | {
|
---|
22 | return IsDebuggerPresent_Real();
|
---|
23 | }
|
---|
24 | else
|
---|
25 | {
|
---|
26 | return FALSE;
|
---|
27 | }
|
---|
28 | }
|
---|
29 |
|
---|
30 | void* WINAPI EncodePointer_Helper(void* p)
|
---|
31 | {
|
---|
32 | static bool first = false;
|
---|
33 | if (!first)
|
---|
34 | {
|
---|
35 | ::MessageBox(nullptr, "test", "", 0);
|
---|
36 | first = true;
|
---|
37 | }
|
---|
38 | if (auto EncodePointer_Real = GET_PROC_ADDRESS(KERNEL32, EncodePointer))
|
---|
39 | {
|
---|
40 | return EncodePointer_Real(p);
|
---|
41 | }
|
---|
42 | else
|
---|
43 | {
|
---|
44 | return reinterpret_cast<void*>(reinterpret_cast<DWORD>(p) ^ GetCurrentProcessId());
|
---|
45 | }
|
---|
46 | }
|
---|
47 |
|
---|
48 | void* WINAPI DecodePointer_Helper(void* p)
|
---|
49 | {
|
---|
50 | if (auto DecodePointer_Real = GET_PROC_ADDRESS(KERNEL32, DecodePointer))
|
---|
51 | {
|
---|
52 | return DecodePointer_Real(p);
|
---|
53 | }
|
---|
54 | else
|
---|
55 | {
|
---|
56 | return reinterpret_cast<void*>(reinterpret_cast<DWORD>(p) ^ GetCurrentProcessId());
|
---|
57 | }
|
---|
58 | }
|
---|
59 |
|
---|
60 | BOOL WINAPI IsProcessorFeaturePresent_Helper(DWORD feture)
|
---|
61 | {
|
---|
62 | if (auto IsProcessorFeaturePresent_Real = GET_PROC_ADDRESS(KERNEL32, IsProcessorFeaturePresent))
|
---|
63 | {
|
---|
64 | return IsProcessorFeaturePresent_Real(feture);
|
---|
65 | }
|
---|
66 | else
|
---|
67 | {
|
---|
68 | assert(feture == PF_NX_ENABLED); // ATLから呼ばれる
|
---|
69 | return FALSE;
|
---|
70 | }
|
---|
71 | }
|
---|
72 |
|
---|
73 | BOOL WINAPI HeapSetInformation_Helper(HANDLE hHeap, HEAP_INFORMATION_CLASS hic, void* information, SSIZE_T informationLength)
|
---|
74 | {
|
---|
75 | if (auto HeapSetInformation_Real = GET_PROC_ADDRESS(KERNEL32, HeapSetInformation))
|
---|
76 | {
|
---|
77 | return HeapSetInformation_Real(hHeap, hic, information, informationLength);
|
---|
78 | }
|
---|
79 | else
|
---|
80 | {
|
---|
81 | return TRUE;
|
---|
82 | }
|
---|
83 | }
|
---|
84 |
|
---|
85 | BOOL WINAPI InitializeCriticalSectionAndSpinCount_Helper(LPCRITICAL_SECTION lpcs, DWORD spinCount)
|
---|
86 | {
|
---|
87 | __try
|
---|
88 | {
|
---|
89 | if (auto InitializeCriticalSectionAndSpinCount_Real = GET_PROC_ADDRESS(KERNEL32, InitializeCriticalSectionAndSpinCount))
|
---|
90 | {
|
---|
91 | OSVERSIONINFO vi = {sizeof vi};
|
---|
92 | ::GetVersionEx(&vi);
|
---|
93 | BOOL ret = InitializeCriticalSectionAndSpinCount_Real(lpcs, spinCount);
|
---|
94 | return vi.dwPlatformId == VER_PLATFORM_WIN32_WINDOWS
|
---|
95 | ? TRUE
|
---|
96 | : ret;
|
---|
97 | }
|
---|
98 | else
|
---|
99 | {
|
---|
100 | ::InitializeCriticalSection(lpcs);
|
---|
101 | return TRUE;
|
---|
102 | }
|
---|
103 | }
|
---|
104 | __except(EXCEPTION_EXECUTE_HANDLER)
|
---|
105 | {
|
---|
106 | }
|
---|
107 | return FALSE;
|
---|
108 | }
|
---|
109 |
|
---|
110 | // Interlocked(Push|Pop)EntrySListは、NXが有効な場合のみ呼ばれる。
|
---|
111 | // Windows XP以前はNX対応していないためInterlocked(Push|Pop)EntrySListが呼ばれることはない。
|
---|
112 | // そのため、GET_PROC_ADDRESSがnullptrを返すことはないと仮定している。
|
---|
113 | // なお、Interlocked(Push|Pop)EntrySListもWindows XPから搭載されている。
|
---|
114 |
|
---|
115 | PSLIST_ENTRY WINAPI InterlockedPushEntrySList_Helper(PSLIST_HEADER head, PSLIST_ENTRY entry)
|
---|
116 | {
|
---|
117 | auto InterlockedPushEntrySList_Real = GET_PROC_ADDRESS(KERNEL32, InterlockedPushEntrySList);
|
---|
118 | assert(InterlockedPushEntrySList_Real != nullptr);
|
---|
119 | return InterlockedPushEntrySList_Real(head, entry);
|
---|
120 | }
|
---|
121 |
|
---|
122 | PSLIST_ENTRY WINAPI InterlockedPopEntrySList_Helper(PSLIST_HEADER head)
|
---|
123 | {
|
---|
124 | auto InterlockedPopEntrySList_Real = GET_PROC_ADDRESS(KERNEL32, InterlockedPopEntrySList);
|
---|
125 | assert(InterlockedPopEntrySList_Real != nullptr);
|
---|
126 | return InterlockedPopEntrySList_Real(head);
|
---|
127 | }
|
---|
128 | }
|
---|