137 lines
3.8 KiB
C++
137 lines
3.8 KiB
C++
![]() |
/*
|
||
|
** Command & Conquer Generals Zero Hour(tm)
|
||
|
** Copyright 2025 Electronic Arts Inc.
|
||
|
**
|
||
|
** This program is free software: you can redistribute it and/or modify
|
||
|
** it under the terms of the GNU General Public License as published by
|
||
|
** the Free Software Foundation, either version 3 of the License, or
|
||
|
** (at your option) any later version.
|
||
|
**
|
||
|
** 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, see <http://www.gnu.org/licenses/>.
|
||
|
*/
|
||
|
|
||
|
// Registry.cpp
|
||
|
// Simple interface for storing/retreiving registry values
|
||
|
// Author: Matthew D. Campbell, December 2001
|
||
|
|
||
|
#include <string>
|
||
|
|
||
|
#define WIN32_LEAN_AND_MEAN
|
||
|
#include <windows.h>
|
||
|
|
||
|
#include "Registry.h"
|
||
|
|
||
|
bool getStringFromRegistry(HKEY root, std::string path, std::string key, std::string& val)
|
||
|
{
|
||
|
HKEY handle;
|
||
|
unsigned char buffer[256];
|
||
|
unsigned long size = 256;
|
||
|
unsigned long type;
|
||
|
int returnValue;
|
||
|
|
||
|
if ((returnValue = RegOpenKeyEx( root, path.c_str(), 0, KEY_ALL_ACCESS, &handle )) == ERROR_SUCCESS)
|
||
|
{
|
||
|
returnValue = RegQueryValueEx(handle, key.c_str(), NULL, &type, (unsigned char *) &buffer, &size);
|
||
|
RegCloseKey( handle );
|
||
|
}
|
||
|
|
||
|
if (returnValue == ERROR_SUCCESS)
|
||
|
{
|
||
|
val = (char *)buffer;
|
||
|
return true;
|
||
|
}
|
||
|
|
||
|
return false;
|
||
|
}
|
||
|
|
||
|
bool getUnsignedIntFromRegistry(HKEY root, std::string path, std::string key, unsigned int& val)
|
||
|
{
|
||
|
HKEY handle;
|
||
|
unsigned long buffer;
|
||
|
unsigned long size = sizeof(buffer);
|
||
|
unsigned long type;
|
||
|
int returnValue;
|
||
|
|
||
|
if ((returnValue = RegOpenKeyEx( root, path.c_str(), 0, KEY_ALL_ACCESS, &handle )) == ERROR_SUCCESS)
|
||
|
{
|
||
|
returnValue = RegQueryValueEx(handle, key.c_str(), NULL, &type, (unsigned char *) &buffer, &size);
|
||
|
RegCloseKey( handle );
|
||
|
}
|
||
|
|
||
|
if (returnValue == ERROR_SUCCESS)
|
||
|
{
|
||
|
val = buffer;
|
||
|
return true;
|
||
|
}
|
||
|
|
||
|
return false;
|
||
|
}
|
||
|
|
||
|
bool setStringInRegistry( HKEY root, std::string path, std::string key, std::string val)
|
||
|
{
|
||
|
HKEY handle;
|
||
|
unsigned long type;
|
||
|
unsigned long returnValue;
|
||
|
int size;
|
||
|
|
||
|
if ((returnValue = RegCreateKeyEx( root, path.c_str(), 0, "REG_NONE", REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS, NULL, &handle, NULL )) == ERROR_SUCCESS)
|
||
|
{
|
||
|
type = REG_SZ;
|
||
|
size = val.length()+1;
|
||
|
returnValue = RegSetValueEx(handle, key.c_str(), 0, type, (unsigned char *)val.c_str(), size);
|
||
|
RegCloseKey( handle );
|
||
|
}
|
||
|
|
||
|
return (returnValue == ERROR_SUCCESS);
|
||
|
}
|
||
|
|
||
|
bool setUnsignedIntInRegistry( HKEY root, std::string path, std::string key, unsigned int val)
|
||
|
{
|
||
|
HKEY handle;
|
||
|
unsigned long type;
|
||
|
unsigned long returnValue;
|
||
|
int size;
|
||
|
|
||
|
if ((returnValue = RegCreateKeyEx( root, path.c_str(), 0, "REG_NONE", REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS, NULL, &handle, NULL )) == ERROR_SUCCESS)
|
||
|
{
|
||
|
type = REG_DWORD;
|
||
|
size = 4;
|
||
|
returnValue = RegSetValueEx(handle, key.c_str(), 0, type, (unsigned char *)&val, size);
|
||
|
RegCloseKey( handle );
|
||
|
}
|
||
|
|
||
|
return (returnValue == ERROR_SUCCESS);
|
||
|
}
|
||
|
|
||
|
bool GetStringFromRegistry(std::string path, std::string key, std::string& val)
|
||
|
{
|
||
|
std::string fullPath = "SOFTWARE\\Electronic Arts\\EA Games\\Generals";
|
||
|
|
||
|
fullPath.append(path);
|
||
|
if (getStringFromRegistry(HKEY_LOCAL_MACHINE, fullPath.c_str(), key.c_str(), val))
|
||
|
{
|
||
|
return true;
|
||
|
}
|
||
|
|
||
|
return getStringFromRegistry(HKEY_CURRENT_USER, fullPath.c_str(), key.c_str(), val);
|
||
|
}
|
||
|
|
||
|
bool GetUnsignedIntFromRegistry(std::string path, std::string key, unsigned int& val)
|
||
|
{
|
||
|
std::string fullPath = "SOFTWARE\\Electronic Arts\\EA Games\\Generals";
|
||
|
|
||
|
fullPath.append(path);
|
||
|
if (getUnsignedIntFromRegistry(HKEY_LOCAL_MACHINE, fullPath.c_str(), key.c_str(), val))
|
||
|
{
|
||
|
return true;
|
||
|
}
|
||
|
|
||
|
return getUnsignedIntFromRegistry(HKEY_CURRENT_USER, fullPath.c_str(), key.c_str(), val);
|
||
|
}
|