Serious-Engine/Sources/Engine/Base/Registry.cpp

130 lines
3.8 KiB
C++
Raw Normal View History

2016-03-12 01:20:51 +01:00
/* Copyright (c) 2002-2012 Croteam Ltd.
This program is free software; you can redistribute it and/or modify
it under the terms of version 2 of the GNU General Public License as published by
the Free Software Foundation
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License along
with this program; if not, write to the Free Software Foundation, Inc.,
51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. */
2016-03-11 14:57:17 +01:00
#include "Engine/StdH.h"
2016-03-11 14:57:17 +01:00
#include <Engine/Base/CTString.h>
#include <Engine/Base/FileName.h>
#include <Engine/Base/Registry.h>
static void ParseKeyName(const CTString &strKey, HKEY &hKeyRoot, CTString &strKeyPath, CTString &strKeyName)
{
CTString strRemain = strKey;
// separate key into the value part and path
strKeyName = CTFileName(strRemain).FileName();
strRemain = CTFileName(strRemain).FileDir();
strRemain.TrimRight(strlen(strRemain)-1); // removes trailing backslash
// try to find root key value
if (strRemain.RemovePrefix("HKEY_CLASSES_ROOT\\")) {
hKeyRoot = HKEY_CLASSES_ROOT;
} else if (strRemain.RemovePrefix("HKEY_CURRENT_CONFIG\\")) {
hKeyRoot = HKEY_CURRENT_CONFIG;
} else if (strRemain.RemovePrefix("HKEY_CURRENT_USER\\")) {
hKeyRoot = HKEY_CURRENT_USER;
} else if (strRemain.RemovePrefix("HKEY_LOCAL_MACHINE\\")) {
hKeyRoot = HKEY_LOCAL_MACHINE;
} else if (strRemain.RemovePrefix("HKEY_USERS\\")) {
hKeyRoot = HKEY_USERS;
} else {
ASSERT(FALSE);
hKeyRoot = HKEY_CURRENT_USER;
}
strKeyPath = strRemain;
}
ENGINE_API BOOL REG_GetString(const CTString &strKey, CTString &strString)
{
// parse the key name
HKEY hKeyRoot; CTString strKeyPath; CTString strKeyName;
ParseKeyName(strKey, hKeyRoot, strKeyPath, strKeyName);
// open the key
HKEY hkey;
LONG lRes = RegOpenKeyExA(hKeyRoot, strKeyPath, 0, KEY_READ, &hkey);
if (lRes!=ERROR_SUCCESS) {
return FALSE;
}
// get the value
char achrKeyValue[1024];
ULONG ctType;
ULONG ctLength = sizeof(achrKeyValue);
LONG lResult = RegQueryValueExA(hkey, strKeyName, 0, &ctType, (unsigned char *)achrKeyValue, &ctLength);
if(lResult != ERROR_SUCCESS) {
return FALSE;
}
// close the key
RegCloseKey(hkey);
strString = CTString( achrKeyValue);
return TRUE;
}
ENGINE_API BOOL REG_GetLong(const CTString &strKey, ULONG &ulLong)
{
// parse the key name
HKEY hKeyRoot; CTString strKeyPath; CTString strKeyName;
ParseKeyName(strKey, hKeyRoot, strKeyPath, strKeyName);
// open the key
HKEY hkey;
LONG lRes = RegOpenKeyExA(hKeyRoot, strKeyPath, 0, KEY_READ, &hkey);
if (lRes!=ERROR_SUCCESS) {
return FALSE;
}
// get the value
ULONG ulKeyValue;
ULONG ctType;
ULONG ctLength = sizeof(ulKeyValue);
LONG lResult = RegQueryValueExA(hkey, strKeyName, 0, &ctType, (unsigned char *)&ulKeyValue, &ctLength);
if(lResult != ERROR_SUCCESS) {
return FALSE;
}
// close the key
RegCloseKey(hkey);
ulLong = ulKeyValue;
return TRUE;
}
ENGINE_API BOOL REG_SetString(const CTString &strKey, const CTString &strString)
{
// parse the key name
HKEY hKeyRoot; CTString strKeyPath; CTString strKeyName;
ParseKeyName(strKey, hKeyRoot, strKeyPath, strKeyName);
// create the registry key
HKEY hkey;
DWORD dwDisposition;
LONG lRes = RegCreateKeyExA(hKeyRoot, (const char*)strKeyPath, 0,
"", REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS, NULL, &hkey, &dwDisposition);
if (lRes!=ERROR_SUCCESS) {
return FALSE;
}
// set the value
lRes = RegSetValueExA(hkey, strKeyName, 0, REG_SZ, (const UBYTE*)(const char*)strString, strlen(strString));
// close the key
RegCloseKey(hkey);
return lRes==ERROR_SUCCESS;
}