source: dev/branches/egtra/ab5.0/abdev/ab_common/src/Environment.cpp@ 813

#include "stdafx.h"
#include <uxtheme.h>

DWORD const PROCESS_DEP_ENABLE = 0x00000001;

std::string ActiveBasic::Common::Environment::rootPath;
bool ActiveBasic::Common::Environment::isRemoveExternal = false;

using namespace ActiveBasic::Common;

void Environment::SetAbdevRootPath( const std::string &rootPath )
{
    Environment::rootPath = Jenga::Common::Path::MakeFullPath( rootPath, Jenga::Common::Environment::GetAppDir() );
}
const std::string Environment::GetAbdevRootPath()
{
    if( rootPath.empty() )
    {
        SetAbdevRootPath( "" );
    }
    return rootPath;
}

const std::string Environment::GetUserAppDir()
{
    return Jenga::Common::Environment::GetUserAppDir() + "\\ActiveBasic";
}

const std::string Environment::GetCompilerExePath( Platform::EnumType platform )
{
    switch( platform )
    {
    case Platform::X86:
        return rootPath + "\\bin\\x86\\abc.exe";
    case Platform::X64:
        return rootPath + "\\bin\\x64\\abc.exe";
    }
    throw;
}

void* operator new( std::size_t n )
{
    if ( void* p = dlmalloc(n) ){
        return p;
    }
    else{
        throw std::bad_alloc();
    }
}

void* operator new[]( std::size_t n )
{
    return ::operator new( n );
}

void operator delete( void* p )
{
    dlfree( p );
}

void operator delete[]( void* p )
{
    ::operator delete( p );
}

typedef HRESULT (WINAPI* PFN_EnableThemeDialogTexture)(HWND, DWORD);

HMODULE hmodUxTheme = LoadLibrary("uxtheme");

HRESULT ApplyDialogTexture( HWND hwnd )
{
    if( hmodUxTheme )
    {
        if( PFN_EnableThemeDialogTexture pfn = reinterpret_cast<PFN_EnableThemeDialogTexture>(
            GetProcAddress(hmodUxTheme, "EnableThemeDialogTexture")) )
        {
            return pfn(hwnd, ETDT_ENABLETAB);
        }
    }
    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);
}