137 lines
3.8 KiB
C++
Raw Normal View History

/*
** 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);
}