uid issue
This commit is contained in:
KittenPopo
193
common/environment_utils.cpp
Normal file
193
common/environment_utils.cpp
Normal file
@ -0,0 +1,193 @@
|
||||
//========= (C) Copyright 2009-2016 Valve, L.L.C. All rights reserved. ========
|
||||
//
|
||||
// The copyright to the contents herein is the property of Valve, L.L.C.
|
||||
// The contents may be used and/or copied only with the written permission of
|
||||
// Valve, L.L.C., or in accordance with the terms and conditions stipulated in
|
||||
// the agreement/contract under which the contents have been supplied.
|
||||
//
|
||||
// $Header: $
|
||||
// $NoKeywords: $
|
||||
//
|
||||
//=============================================================================
|
||||
|
||||
#include "tier0/dbg.h"
|
||||
#include "environment_utils.h"
|
||||
|
||||
#ifdef PLATFORM_WINDOWS_PC
|
||||
#include <windows.h> // for RegGetValue
|
||||
#endif
|
||||
#undef GetCurrentDirectory
|
||||
#undef OUT
|
||||
#undef ERR
|
||||
|
||||
#if defined( PLATFORM_WINDOWS_PC )
|
||||
bool VGetRegistryKeyValue( HKEY baseKey, const char *pSubKey, const char *pValue, char *pOutBuf, int nMaxBuf )
|
||||
{
|
||||
DWORD nBufSize = nMaxBuf;
|
||||
HKEY hKey = NULL;
|
||||
LONG nResult;
|
||||
|
||||
nResult = RegOpenKeyEx( baseKey, pSubKey, NULL, KEY_READ, &hKey );
|
||||
if ( nResult == ERROR_SUCCESS )
|
||||
{
|
||||
nResult = RegQueryValueExA( hKey, pValue, NULL, NULL, (LPBYTE)pOutBuf, &nBufSize);
|
||||
RegCloseKey( hKey );
|
||||
return nResult == ERROR_SUCCESS;
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
#endif
|
||||
|
||||
//--------------------------------------------------------------------------------------------------
|
||||
//--------------------------------------------------------------------------------------------------
|
||||
bool GetWindowsSDKDir( char *pOutBuf, int nMaxBuf )
|
||||
{
|
||||
#ifdef PLATFORM_WINDOWS_PC
|
||||
const char *pWindowsSDKDirRegKey = "SOFTWARE\\Microsoft\\Microsoft SDKs\\Windows";
|
||||
|
||||
if( !VGetRegistryKeyValue( HKEY_LOCAL_MACHINE, pWindowsSDKDirRegKey, "CurrentInstallFolder", pOutBuf, nMaxBuf ) &&
|
||||
!VGetRegistryKeyValue( HKEY_CURRENT_USER, pWindowsSDKDirRegKey, "CurrentInstallFolder", pOutBuf, nMaxBuf ) )
|
||||
{
|
||||
Warning( "ERROR: Failed to read VS Windows SDK from registry key '%s'\n", pWindowsSDKDirRegKey );
|
||||
return false;
|
||||
}
|
||||
|
||||
return true;
|
||||
#else
|
||||
return false;
|
||||
#endif
|
||||
}
|
||||
|
||||
//--------------------------------------------------------------------------------------------------
|
||||
//--------------------------------------------------------------------------------------------------
|
||||
bool GetMSVCIncludePaths( CUtlVector<CUtlString> &paths, const char *pVSToolsEnv )
|
||||
{
|
||||
const char *pVSToolsPath = Plat_GetEnv( pVSToolsEnv );
|
||||
if ( !pVSToolsPath )
|
||||
{
|
||||
Warning( "ERROR: GetMSVCIncludePaths failed to read VS location from environment variable '%s'\n", pVSToolsEnv );
|
||||
return false;
|
||||
}
|
||||
|
||||
char pWindowsSDKDir[MAX_PATH];
|
||||
if ( !GetWindowsSDKDir( pWindowsSDKDir, sizeof(pWindowsSDKDir) ) )
|
||||
{
|
||||
return false;
|
||||
}
|
||||
|
||||
char pVSRootPath[MAX_PATH];
|
||||
V_MakeAbsolutePath( pVSRootPath, sizeof(pVSRootPath), "../../", pVSToolsPath );
|
||||
|
||||
char pIncludePath[MAX_PATH];
|
||||
V_MakeAbsolutePath( pIncludePath, sizeof(pIncludePath), "VC/INCLUDE", pVSRootPath );
|
||||
paths.AddToTail( pIncludePath );
|
||||
V_MakeAbsolutePath( pIncludePath, sizeof(pIncludePath), "VC/ATLMFC/INCLUDE", pVSRootPath );
|
||||
paths.AddToTail( pIncludePath );
|
||||
V_MakeAbsolutePath( pIncludePath, sizeof(pIncludePath), "INCLUDE", pWindowsSDKDir );
|
||||
paths.AddToTail( pIncludePath );
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
//--------------------------------------------------------------------------------------------------
|
||||
//--------------------------------------------------------------------------------------------------
|
||||
bool GetX360IncludePaths( CUtlVector<CUtlString> &paths )
|
||||
{
|
||||
const char *pXdkPathEnv = "XEDK";
|
||||
const char *pXdkPath = Plat_GetEnv( pXdkPathEnv );
|
||||
if ( !pXdkPath )
|
||||
{
|
||||
Warning( "ERROR: GetX360IncludePaths failed to read XDK location from environment variable '%s'\n", pXdkPath );
|
||||
return false;
|
||||
}
|
||||
|
||||
char pIncludePath[MAX_PATH];
|
||||
V_MakeAbsolutePath( pIncludePath, sizeof(pIncludePath), "include/xbox", pXdkPath );
|
||||
paths.AddToTail( pIncludePath );
|
||||
V_MakeAbsolutePath( pIncludePath, sizeof(pIncludePath), "include/win32", pXdkPath );
|
||||
paths.AddToTail( pIncludePath );
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
//--------------------------------------------------------------------------------------------------
|
||||
//--------------------------------------------------------------------------------------------------
|
||||
bool GetOSXIncludePaths( CUtlVector<CUtlString> &paths )
|
||||
{
|
||||
const char *pPath = Plat_GetEnv( "SDKROOT" );
|
||||
if ( !pPath )
|
||||
{
|
||||
pPath = Plat_GetEnv( "OSX_SDK_PATH" );
|
||||
if ( !pPath )
|
||||
{
|
||||
Warning( "ERROR: %s failed to read SDK location from environment variable 'SDKROOT' or 'OSX_SDK_PATH'\n",
|
||||
__FUNCTION__ );
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
char pIncludePath[MAX_PATH];
|
||||
V_MakeAbsolutePath( pIncludePath, sizeof(pIncludePath), "usr/include", pPath, false );
|
||||
paths.AddToTail( pIncludePath );
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
//--------------------------------------------------------------------------------------------------
|
||||
//--------------------------------------------------------------------------------------------------
|
||||
bool GetSystemIncludePaths( CUtlVector<CUtlString> &paths, const char *pPlatform, const char *pCompiler )
|
||||
{
|
||||
#if !defined( PLATFORM_WINDOWS ) && !defined( PLATFORM_POSIX ) && !defined( PLATFORM_OSX )
|
||||
Warning( "ERROR: GetSystemIncludePaths not implemented for this platform!\n" );
|
||||
return false;
|
||||
#endif
|
||||
|
||||
if ( !V_stricmp_fast( pPlatform, "WIN32" ) || !V_stricmp_fast( pPlatform, "WIN64" ) )
|
||||
{
|
||||
if ( !V_stricmp_fast( pCompiler, "VS2005" ) )
|
||||
{
|
||||
return GetMSVCIncludePaths( paths, "VS80COMNTOOLS" );
|
||||
}
|
||||
else if ( !V_stricmp_fast( pCompiler, "VS2010" ) )
|
||||
{
|
||||
return GetMSVCIncludePaths( paths, "VS100COMNTOOLS" );
|
||||
}
|
||||
else if ( !V_stricmp_fast( pCompiler, "VS2012" ) )
|
||||
{
|
||||
return GetMSVCIncludePaths( paths, "VS110COMNTOOLS" );
|
||||
}
|
||||
else if ( !V_stricmp_fast( pCompiler, "VS2013" ) )
|
||||
{
|
||||
return GetMSVCIncludePaths( paths, "VS120COMNTOOLS" );
|
||||
}
|
||||
else if ( !V_stricmp_fast( pCompiler, "VS2015" ) )
|
||||
{
|
||||
return GetMSVCIncludePaths( paths, "VS140COMNTOOLS" );
|
||||
}
|
||||
AssertMsg1( false, "ERROR: GetSystemIncludePaths not implemented for this compiler yet! (%s)\n", pCompiler );
|
||||
return false;
|
||||
}
|
||||
else if ( !V_stricmp_fast( pPlatform, "X360" ) )
|
||||
{
|
||||
return GetX360IncludePaths( paths );
|
||||
}
|
||||
else if ( !V_stricmp_fast( pPlatform, "LINUXSTEAMRT64" ) ||
|
||||
!V_stricmp_fast( pPlatform, "LINUXSERVER64" ) )
|
||||
{
|
||||
// The Steam runtime tool.sh script will rewrite /usr/include
|
||||
// to whatever is appropriate for runtime compiles, so
|
||||
// we can just blindly use it here and it will work both
|
||||
// for runtime and non-runtime compiles.
|
||||
paths.AddToTail( "/usr/include" );
|
||||
return true;
|
||||
}
|
||||
else if ( !V_stricmp_fast( pPlatform, "OSX32" ) ||
|
||||
!V_stricmp_fast( pPlatform, "OSX64" ) )
|
||||
{
|
||||
return GetOSXIncludePaths( paths );
|
||||
}
|
||||
|
||||
AssertMsg1( false, "ERROR: GetSystemIncludePaths not implemented for this platform yet! (%s)\n", pPlatform );
|
||||
return false;
|
||||
}
|
||||
Reference in New Issue
Block a user