| 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 | if (auto EncodePointer_Real = GET_PROC_ADDRESS(KERNEL32, EncodePointer))
|
|---|
| 33 | {
|
|---|
| 34 | return EncodePointer_Real(p);
|
|---|
| 35 | }
|
|---|
| 36 | else
|
|---|
| 37 | {
|
|---|
| 38 | return reinterpret_cast<void*>(reinterpret_cast<DWORD>(p) ^ GetCurrentProcessId());
|
|---|
| 39 | }
|
|---|
| 40 | }
|
|---|
| 41 |
|
|---|
| 42 | void* WINAPI DecodePointer_Helper(void* p)
|
|---|
| 43 | {
|
|---|
| 44 | if (auto DecodePointer_Real = GET_PROC_ADDRESS(KERNEL32, DecodePointer))
|
|---|
| 45 | {
|
|---|
| 46 | return DecodePointer_Real(p);
|
|---|
| 47 | }
|
|---|
| 48 | else
|
|---|
| 49 | {
|
|---|
| 50 | return reinterpret_cast<void*>(reinterpret_cast<DWORD>(p) ^ GetCurrentProcessId());
|
|---|
| 51 | }
|
|---|
| 52 | }
|
|---|
| 53 |
|
|---|
| 54 | BOOL WINAPI IsProcessorFeaturePresent_Helper(DWORD feture)
|
|---|
| 55 | {
|
|---|
| 56 | if (auto IsProcessorFeaturePresent_Real = GET_PROC_ADDRESS(KERNEL32, IsProcessorFeaturePresent))
|
|---|
| 57 | {
|
|---|
| 58 | return IsProcessorFeaturePresent_Real(feture);
|
|---|
| 59 | }
|
|---|
| 60 | else
|
|---|
| 61 | {
|
|---|
| 62 | assert(feture == PF_NX_ENABLED); // ATLから呼ばれる
|
|---|
| 63 | return FALSE;
|
|---|
| 64 | }
|
|---|
| 65 | }
|
|---|
| 66 |
|
|---|
| 67 | BOOL WINAPI HeapSetInformation_Helper(HANDLE hHeap, HEAP_INFORMATION_CLASS hic, void* information, SSIZE_T informationLength)
|
|---|
| 68 | {
|
|---|
| 69 | if (auto HeapSetInformation_Real = GET_PROC_ADDRESS(KERNEL32, HeapSetInformation))
|
|---|
| 70 | {
|
|---|
| 71 | return HeapSetInformation_Real(hHeap, hic, information, informationLength);
|
|---|
| 72 | }
|
|---|
| 73 | else
|
|---|
| 74 | {
|
|---|
| 75 | return TRUE;
|
|---|
| 76 | }
|
|---|
| 77 | }
|
|---|
| 78 |
|
|---|
| 79 | BOOL WINAPI InitializeCriticalSectionAndSpinCount_Helper(LPCRITICAL_SECTION lpcs, DWORD spinCount)
|
|---|
| 80 | {
|
|---|
| 81 | __try
|
|---|
| 82 | {
|
|---|
| 83 | if (auto InitializeCriticalSectionAndSpinCount_Real = GET_PROC_ADDRESS(KERNEL32, InitializeCriticalSectionAndSpinCount))
|
|---|
| 84 | {
|
|---|
| 85 | OSVERSIONINFO vi = {sizeof vi};
|
|---|
| 86 | ::GetVersionEx(&vi);
|
|---|
| 87 | BOOL ret = InitializeCriticalSectionAndSpinCount_Real(lpcs, spinCount);
|
|---|
| 88 | return vi.dwPlatformId == VER_PLATFORM_WIN32_WINDOWS
|
|---|
| 89 | ? TRUE
|
|---|
| 90 | : ret;
|
|---|
| 91 | }
|
|---|
| 92 | else
|
|---|
| 93 | {
|
|---|
| 94 | ::InitializeCriticalSection(lpcs);
|
|---|
| 95 | return TRUE;
|
|---|
| 96 | }
|
|---|
| 97 | }
|
|---|
| 98 | __except(EXCEPTION_EXECUTE_HANDLER)
|
|---|
| 99 | {
|
|---|
| 100 | }
|
|---|
| 101 | return FALSE;
|
|---|
| 102 | }
|
|---|
| 103 |
|
|---|
| 104 | // Interlocked(Push|Pop)EntrySListは、NXが有効な場合のみ呼ばれる。
|
|---|
| 105 | // Windows XP以前はNX対応していないためInterlocked(Push|Pop)EntrySListが呼ばれることはない。
|
|---|
| 106 | // そのため、GET_PROC_ADDRESSがnullptrを返すことはないと仮定している。
|
|---|
| 107 | // なお、Interlocked(Push|Pop)EntrySListもWindows XPから搭載されている。
|
|---|
| 108 |
|
|---|
| 109 | PSLIST_ENTRY WINAPI InterlockedPushEntrySList_Helper(PSLIST_HEADER head, PSLIST_ENTRY entry)
|
|---|
| 110 | {
|
|---|
| 111 | auto InterlockedPushEntrySList_Real = GET_PROC_ADDRESS(KERNEL32, InterlockedPushEntrySList);
|
|---|
| 112 | assert(InterlockedPushEntrySList_Real != nullptr);
|
|---|
| 113 | return InterlockedPushEntrySList_Real(head, entry);
|
|---|
| 114 | }
|
|---|
| 115 |
|
|---|
| 116 | PSLIST_ENTRY WINAPI InterlockedPopEntrySList_Helper(PSLIST_HEADER head)
|
|---|
| 117 | {
|
|---|
| 118 | auto InterlockedPopEntrySList_Real = GET_PROC_ADDRESS(KERNEL32, InterlockedPopEntrySList);
|
|---|
| 119 | assert(InterlockedPopEntrySList_Real != nullptr);
|
|---|
| 120 | return InterlockedPopEntrySList_Real(head);
|
|---|
| 121 | }
|
|---|
| 122 | }
|
|---|