Muldersoft/MUtilities
Muldersoft/MUtilities
1
0
Fork 0
Code Releases Activity

Registry functions: Optionally allow caller to force 32-bit registry view or 64-bit registry view. Default behavior should be unchanged.

Browse Source
This commit is contained in:
LoRd_MuldeR 2016-03-11 19:11:18 +01:00
parent 52c0c777f5
commit 819a3106b3
2 changed files with 51 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
include/MUtils/Registry.h
View File

@ -49,6 +49,15 @@ namespace MUtils
} }
reg_access_t; reg_access_t;
//Regsitry scope
typedef enum
{
scope_default = 0,
scope_wow_x32 = 1,
scope_wow_x64 = 2
}
reg_scope_t;
//Forward declaration //Forward declaration
namespace Internal namespace Internal
{ {
@ -59,7 +68,7 @@ namespace MUtils
class MUTILS_API RegistryKey class MUTILS_API RegistryKey
{ {
public: public:
RegistryKey(const reg_root_t &rootKey, const QString &keyName, const reg_access_t &access); RegistryKey(const reg_root_t &rootKey, const QString &keyName, const reg_access_t &access, const reg_scope_t &scope = scope_default);
~RegistryKey(void); ~RegistryKey(void);
inline bool isOpen(void); inline bool isOpen(void);
@ -78,14 +87,14 @@ namespace MUtils
}; };
//Regsitry functions //Regsitry functions
MUTILS_API bool reg_value_write (const reg_root_t &rootKey, const QString &keyName, const QString &valueName, const quint32 &value); MUTILS_API bool reg_value_write (const reg_root_t &rootKey, const QString &keyName, const QString &valueName, const quint32 &value, const reg_scope_t &scope = scope_default);
MUTILS_API bool reg_value_write (const reg_root_t &rootKey, const QString &keyName, const QString &valueName, const QString &value); MUTILS_API bool reg_value_write (const reg_root_t &rootKey, const QString &keyName, const QString &valueName, const QString &value, const reg_scope_t &scope = scope_default);
MUTILS_API bool reg_value_read (const reg_root_t &rootKey, const QString &keyName, const QString &valueName, quint32 &value); MUTILS_API bool reg_value_read (const reg_root_t &rootKey, const QString &keyName, const QString &valueName, quint32 &value, const reg_scope_t &scope = scope_default);
MUTILS_API bool reg_value_read (const reg_root_t &rootKey, const QString &keyName, const QString &valueName, QString &value); MUTILS_API bool reg_value_read (const reg_root_t &rootKey, const QString &keyName, const QString &valueName, QString &value, const reg_scope_t &scope = scope_default);
MUTILS_API bool reg_key_exists (const reg_root_t &rootKey, const QString &keyName); MUTILS_API bool reg_key_exists (const reg_root_t &rootKey, const QString &keyName, const reg_scope_t &scope = scope_default);
MUTILS_API bool reg_key_delete (const reg_root_t &rootKey, const QString &keyName, const bool &recusrive = true, const bool &ascend = false); MUTILS_API bool reg_key_delete (const reg_root_t &rootKey, const QString &keyName, const bool &recusrive = true, const bool &ascend = false, const reg_scope_t &scope = scope_default);
MUTILS_API bool reg_enum_values (const reg_root_t &rootKey, const QString &keyName, QStringList &list); MUTILS_API bool reg_enum_values (const reg_root_t &rootKey, const QString &keyName, QStringList &list, const reg_scope_t &scope = scope_default);
MUTILS_API bool reg_enum_subkeys(const reg_root_t &rootKey, const QString &keyName, QStringList &list); MUTILS_API bool reg_enum_subkeys(const reg_root_t &rootKey, const QString &keyName, QStringList &list, const reg_scope_t &scope = scope_default);
} }
} }

50
src/Registry_Win32.cpp
View File

@ -71,6 +71,17 @@ namespace MUtils
} }
} }
static DWORD registry_scope(const reg_scope_t &scope)
{
switch (scope)
{
case scope_default: return 0; break;
case scope_wow_x32: return KEY_WOW64_32KEY; break;
case scope_wow_x64: return KEY_WOW64_64KEY; break;
default: MUTILS_THROW("Unknown scope value was specified!");
}
}
static const char* reg_root2str(const reg_root_t &rootKey) static const char* reg_root2str(const reg_root_t &rootKey)
{ {
ENUM2STR(rootKey, root_classes); ENUM2STR(rootKey, root_classes);
@ -134,12 +145,12 @@ while(0)
// Registry Key Class // Registry Key Class
/////////////////////////////////////////////////////////////////////////////// ///////////////////////////////////////////////////////////////////////////////
MUtils::Registry::RegistryKey::RegistryKey(const reg_root_t &rootKey, const QString &keyName, const reg_access_t &access) MUtils::Registry::RegistryKey::RegistryKey(const reg_root_t &rootKey, const QString &keyName, const reg_access_t &access, const reg_scope_t &scope)
: :
p(new Internal::RegistryKeyPrivate()) p(new Internal::RegistryKeyPrivate())
{ {
p->m_hKey = NULL; p->m_hKey = NULL;
p->m_access = registry_access(access); p->m_access = registry_access(access) | registry_scope(scope);
p->m_isOpen = false; p->m_isOpen = false;
p->m_isOpen = (RegCreateKeyEx(registry_root(rootKey), MUTILS_WCHR(keyName), 0, NULL, 0, p->m_access, NULL, &p->m_hKey, NULL) == ERROR_SUCCESS); p->m_isOpen = (RegCreateKeyEx(registry_root(rootKey), MUTILS_WCHR(keyName), 0, NULL, 0, p->m_access, NULL, &p->m_hKey, NULL) == ERROR_SUCCESS);
@ -243,10 +254,10 @@ bool MUtils::Registry::RegistryKey::enum_subkeys(QStringList &list) const
/* /*
* Write registry value * Write registry value
*/ */
bool MUtils::Registry::reg_value_write(const reg_root_t &rootKey, const QString &keyName, const QString &valueName, const quint32 &value) bool MUtils::Registry::reg_value_write(const reg_root_t &rootKey, const QString &keyName, const QString &valueName, const quint32 &value, const reg_scope_t &scope)
{ {
bool success = false; bool success = false;
RegistryKey regKey(rootKey, keyName, access_readwrite); RegistryKey regKey(rootKey, keyName, access_readwrite, scope);
if(regKey.isOpen()) if(regKey.isOpen())
{ {
success = regKey.value_write(valueName, value); success = regKey.value_write(valueName, value);
@ -257,10 +268,10 @@ bool MUtils::Registry::reg_value_write(const reg_root_t &rootKey, const QString
/* /*
* Write registry value * Write registry value
*/ */
bool MUtils::Registry::reg_value_write(const reg_root_t &rootKey, const QString &keyName, const QString &valueName, const QString &value) bool MUtils::Registry::reg_value_write(const reg_root_t &rootKey, const QString &keyName, const QString &valueName, const QString &value, const reg_scope_t &scope)
{ {
bool success = false; bool success = false;
RegistryKey regKey(rootKey, keyName, access_readwrite); RegistryKey regKey(rootKey, keyName, access_readwrite, scope);
if(regKey.isOpen()) if(regKey.isOpen())
{ {
success = regKey.value_write(valueName, value); success = regKey.value_write(valueName, value);
@ -271,10 +282,10 @@ bool MUtils::Registry::reg_value_write(const reg_root_t &rootKey, const QString
/* /*
* Read registry value * Read registry value
*/ */
bool MUtils::Registry::reg_value_read(const reg_root_t &rootKey, const QString &keyName, const QString &valueName, quint32 &value) bool MUtils::Registry::reg_value_read(const reg_root_t &rootKey, const QString &keyName, const QString &valueName, quint32 &value, const reg_scope_t &scope)
{ {
bool success = false; bool success = false;
RegistryKey regKey(rootKey, keyName, access_readonly); RegistryKey regKey(rootKey, keyName, access_readonly, scope);
if(regKey.isOpen()) if(regKey.isOpen())
{ {
success = regKey.value_read(valueName, value); success = regKey.value_read(valueName, value);
@ -289,10 +300,10 @@ bool MUtils::Registry::reg_value_read(const reg_root_t &rootKey, const QString &
/* /*
* Read registry value * Read registry value
*/ */
bool MUtils::Registry::reg_value_read(const reg_root_t &rootKey, const QString &keyName, const QString &valueName, QString &value) bool MUtils::Registry::reg_value_read(const reg_root_t &rootKey, const QString &keyName, const QString &valueName, QString &value, const reg_scope_t &scope)
{ {
bool success = false; bool success = false;
RegistryKey regKey(rootKey, keyName, access_readonly); RegistryKey regKey(rootKey, keyName, access_readonly, scope);
if(regKey.isOpen()) if(regKey.isOpen())
{ {
success = regKey.value_read(valueName, value); success = regKey.value_read(valueName, value);
@ -307,10 +318,10 @@ bool MUtils::Registry::reg_value_read(const reg_root_t &rootKey, const QString &
/* /*
* Enumerate value names * Enumerate value names
*/ */
bool MUtils::Registry::reg_enum_values(const reg_root_t &rootKey, const QString &keyName, QStringList &values) bool MUtils::Registry::reg_enum_values(const reg_root_t &rootKey, const QString &keyName, QStringList &values, const reg_scope_t &scope)
{ {
bool success = false; bool success = false;
RegistryKey regKey(rootKey, keyName, access_readonly); RegistryKey regKey(rootKey, keyName, access_readonly, scope);
if(regKey.isOpen()) if(regKey.isOpen())
{ {
success = regKey.enum_values(values); success = regKey.enum_values(values);
@ -325,10 +336,10 @@ bool MUtils::Registry::reg_enum_values(const reg_root_t &rootKey, const QString
/* /*
* Enumerate subkey names * Enumerate subkey names
*/ */
bool MUtils::Registry::reg_enum_subkeys(const reg_root_t &rootKey, const QString &keyName, QStringList &subkeys) bool MUtils::Registry::reg_enum_subkeys(const reg_root_t &rootKey, const QString &keyName, QStringList &subkeys, const reg_scope_t &scope)
{ {
bool success = false; bool success = false;
RegistryKey regKey(rootKey, keyName, access_enumerate); RegistryKey regKey(rootKey, keyName, access_enumerate, scope);
if(regKey.isOpen()) if(regKey.isOpen())
{ {
success = regKey.enum_subkeys(subkeys); success = regKey.enum_subkeys(subkeys);
@ -343,10 +354,10 @@ bool MUtils::Registry::reg_enum_subkeys(const reg_root_t &rootKey, const QString
/* /*
* Check registry key existence * Check registry key existence
*/ */
bool MUtils::Registry::reg_key_exists(const reg_root_t &rootKey, const QString &keyName) bool MUtils::Registry::reg_key_exists(const reg_root_t &rootKey, const QString &keyName, const reg_scope_t &scope)
{ {
HKEY hKey = NULL; HKEY hKey = NULL;
if(RegOpenKeyEx(registry_root(rootKey), MUTILS_WCHR(keyName), 0, STANDARD_RIGHTS_READ, &hKey) == ERROR_SUCCESS) if(RegOpenKeyEx(registry_root(rootKey), MUTILS_WCHR(keyName), 0, STANDARD_RIGHTS_READ | registry_scope(scope), &hKey) == ERROR_SUCCESS)
{ {
RegCloseKey(hKey); RegCloseKey(hKey);
return true; return true;
@ -357,10 +368,15 @@ bool MUtils::Registry::reg_key_exists(const reg_root_t &rootKey, const QString &
/* /*
* Delete registry key * Delete registry key
*/ */
bool MUtils::Registry::reg_key_delete(const reg_root_t &rootKey, const QString &keyName, const bool &recusrive, const bool &ascend) bool MUtils::Registry::reg_key_delete(const reg_root_t &rootKey, const QString &keyName, const bool &recusrive, const bool &ascend, const reg_scope_t &scope)
{ {
bool okay = false; bool okay = false;
if (scope != scope_default)
{
MUTILS_THROW("Scope option not currently supported by reg_key_delete() function!");
}
if(recusrive) if(recusrive)
{ {
okay = (SHDeleteKey(registry_root(rootKey), MUTILS_WCHR(keyName)) == ERROR_SUCCESS); okay = (SHDeleteKey(registry_root(rootKey), MUTILS_WCHR(keyName)) == ERROR_SUCCESS);