Changeset 773 in dev for trunk/ab5.0/abdev/ab_common/src/Environment.cpp

Timestamp:

Jan 15, 2011, 6:04:58 PM (13 years ago)

Author:

イグトランス (egtra)

Message:

セキュリティ対策の機能を有効化

File:

• trunk/ab5.0/abdev/ab_common/src/Environment.cpp (modified) (2 diffs)

trunk/ab5.0/abdev/ab_common/src/Environment.cpp

@@ -35 +35 @@
     }
     throw;
-}
-
-BOOL EnableLFH(HANDLE hHeap)
-{
-    ULONG enableLFH = 2;
-    return HeapSetInformation(hHeap, HeapCompatibilityInformation, &enableLFH, sizeof enableLFH);
 }
 
@@ -84 +78 @@
     return E_NOTIMPL;
 }
+
+BOOL ActiveBasic::Common::EnableNX()
+{
+    typedef BOOL (WINAPI* PFNSETDEP)(DWORD);
+
+    HMODULE hmodKernel = GetModuleHandle(TEXT("KERNEL32.DLL"));
+    if (PFNSETDEP pfnSetDEP = reinterpret_cast<PFNSETDEP>(GetProcAddress(hmodKernel, "SetProcessDEPPolicy")))
+    {
+        return pfnSetDEP(PROCESS_DEP_ENABLE);
+    }
+    else
+    {
+        return FALSE;
+    }
+}
+
+void ActiveBasic::Common::SetHeapOptions()
+{
+//  SetDllDirectory(_T(""));
+    HMODULE hmodKernel = GetModuleHandle(TEXT("KERNEL32.DLL"));
+
+    typedef BOOL (WINAPI* HSI)(HANDLE, HEAP_INFORMATION_CLASS ,PVOID, SIZE_T);
+    HSI pHsi = reinterpret_cast<HSI>(GetProcAddress(hmodKernel, "HeapSetInformation"));
+    if (!pHsi)
+    {
+        return;
+    }
+
+    ULONG enableLFH = 2;
+    pHsi(GetProcessHeap(), HeapCompatibilityInformation, &enableLFH, sizeof enableLFH);
+
+#ifndef HeapEnableTerminationOnCorruption
+#   define HeapEnableTerminationOnCorruption (HEAP_INFORMATION_CLASS)1
+#endif
+
+    pHsi(NULL, HeapEnableTerminationOnCorruption, NULL, 0);
+}