428 lines
10 KiB
C++
428 lines
10 KiB
C++
|
/*++
|
||
|
|
|
||
|
|
Copyright (c) 1998-1999 Microsoft Corporation
|
||
|
|
All rights reserved.
|
||
|
|
|
||
|
|
Module Name:
|
||
|
|
|
||
|
|
dbgreg.cxx
|
||
|
|
|
||
|
|
Abstract:
|
||
|
|
|
||
|
|
Debug Registry class
|
||
|
|
|
||
|
|
Author:
|
||
|
|
|
||
|
|
Steve Kiraly (SteveKi) 18-Jun-1998
|
||
|
|
|
||
|
|
Revision History:
|
||
|
|
|
||
|
|
--*/
|
||
|
|
#include "precomp.hxx"
|
||
|
|
#pragma hdrstop
|
||
|
|
|
||
|
|
#include "dbgloadl.hxx"
|
||
|
|
#include "dbgreg.hxx"
|
||
|
|
|
||
|
|
TDebugRegApis::
|
||
|
|
TDebugRegApis(
|
||
|
|
VOID
|
||
|
|
) : m_Lib( _T("advapi32.dll") ),
|
||
|
|
m_bValid( FALSE ),
|
||
|
|
m_CreateKeyEx (NULL ),
|
||
|
|
m_OpenKeyEx( NULL ),
|
||
|
|
m_CloseKey( NULL ),
|
||
|
|
m_QueryValueEx( NULL ),
|
||
|
|
m_SetValueEx( NULL ),
|
||
|
|
m_EnumKeyEx( NULL ),
|
||
|
|
m_DeleteKey( NULL )
|
||
|
|
{
|
||
|
|
if( m_Lib.bValid() )
|
||
|
|
{
|
||
|
|
m_CreateKeyEx = reinterpret_cast<pfRegCreateKeyEx> ( m_Lib.pfnGetProc( "RegCreateKeyEx" SUFFIX ) );
|
||
|
|
m_OpenKeyEx = reinterpret_cast<pfRegOpenKeyEx> ( m_Lib.pfnGetProc( "RegOpenKeyEx" SUFFIX ) );
|
||
|
|
m_CloseKey = reinterpret_cast<pfRegCloseKey> ( m_Lib.pfnGetProc( "RegCloseKey" ) );
|
||
|
|
m_QueryValueEx = reinterpret_cast<pfRegQueryValueEx> ( m_Lib.pfnGetProc( "RegQueryValueEx" SUFFIX ) );
|
||
|
|
m_SetValueEx = reinterpret_cast<pfRegSetValueEx> ( m_Lib.pfnGetProc( "RegSetValueEx" SUFFIX ) );
|
||
|
|
m_EnumKeyEx = reinterpret_cast<pfRegEnumKeyEx> ( m_Lib.pfnGetProc( "RegEnumKeyEx" SUFFIX ) );
|
||
|
|
m_DeleteKey = reinterpret_cast<pfRegDeleteKey> ( m_Lib.pfnGetProc( "RegDeleteKey" SUFFIX ) );
|
||
|
|
m_DeleteValue = reinterpret_cast<pfRegDeleteValue> ( m_Lib.pfnGetProc( "RegDeleteValue" SUFFIX ) );
|
||
|
|
|
||
|
|
if( m_CreateKeyEx &&
|
||
|
|
m_OpenKeyEx &&
|
||
|
|
m_CloseKey &&
|
||
|
|
m_QueryValueEx &&
|
||
|
|
m_SetValueEx &&
|
||
|
|
m_EnumKeyEx &&
|
||
|
|
m_DeleteKey &&
|
||
|
|
m_DeleteValue )
|
||
|
|
{
|
||
|
|
m_bValid = TRUE;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
BOOL
|
||
|
|
TDebugRegApis::
|
||
|
|
bValid(
|
||
|
|
VOID
|
||
|
|
) const
|
||
|
|
{
|
||
|
|
return m_bValid;
|
||
|
|
}
|
||
|
|
|
||
|
|
TDebugRegistry::
|
||
|
|
TDebugRegistry(
|
||
|
|
IN LPCTSTR pszSection,
|
||
|
|
IN UINT ioFlags,
|
||
|
|
IN HKEY hOpenedKey
|
||
|
|
) : m_strSection( pszSection ),
|
||
|
|
m_hKey( NULL ),
|
||
|
|
m_Status( ERROR_SUCCESS )
|
||
|
|
{
|
||
|
|
DBG_ASSERT( pszSection );
|
||
|
|
DBG_ASSERT( hOpenedKey );
|
||
|
|
|
||
|
|
if( m_Reg.bValid() )
|
||
|
|
{
|
||
|
|
DWORD dwDisposition = 0;
|
||
|
|
UINT uAccessIndex = 0;
|
||
|
|
|
||
|
|
static DWORD adwAccess [] = { 0, KEY_READ, KEY_WRITE, KEY_ALL_ACCESS };
|
||
|
|
|
||
|
|
uAccessIndex = ioFlags & ( kRead | kWrite );
|
||
|
|
|
||
|
|
if( uAccessIndex )
|
||
|
|
{
|
||
|
|
if( ioFlags & kCreate )
|
||
|
|
{
|
||
|
|
m_Status = m_Reg.m_CreateKeyEx( hOpenedKey,
|
||
|
|
m_strSection,
|
||
|
|
0,
|
||
|
|
NULL,
|
||
|
|
0,
|
||
|
|
adwAccess[ uAccessIndex ],
|
||
|
|
NULL,
|
||
|
|
&m_hKey,
|
||
|
|
&dwDisposition );
|
||
|
|
}
|
||
|
|
|
||
|
|
if( ioFlags & kOpen )
|
||
|
|
{
|
||
|
|
m_Status = m_Reg.m_OpenKeyEx( hOpenedKey,
|
||
|
|
m_strSection,
|
||
|
|
0,
|
||
|
|
adwAccess[ uAccessIndex ],
|
||
|
|
&m_hKey );
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
TDebugRegistry::
|
||
|
|
~TDebugRegistry(
|
||
|
|
VOID
|
||
|
|
)
|
||
|
|
{
|
||
|
|
if( m_hKey )
|
||
|
|
{
|
||
|
|
m_Reg.m_CloseKey( m_hKey );
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
BOOL
|
||
|
|
TDebugRegistry::
|
||
|
|
bValid(
|
||
|
|
VOID
|
||
|
|
) const
|
||
|
|
{
|
||
|
|
//
|
||
|
|
// A valid hKey and a valid section name is the class valid check.
|
||
|
|
//
|
||
|
|
return m_hKey != NULL && m_strSection.bValid() && m_Reg.bValid();
|
||
|
|
}
|
||
|
|
|
||
|
|
DWORD
|
||
|
|
TDebugRegistry::
|
||
|
|
LastError(
|
||
|
|
VOID
|
||
|
|
) const
|
||
|
|
{
|
||
|
|
return m_Status;
|
||
|
|
}
|
||
|
|
|
||
|
|
BOOL
|
||
|
|
TDebugRegistry::
|
||
|
|
bRead(
|
||
|
|
IN LPCTSTR pValueName,
|
||
|
|
IN OUT DWORD &dwValue
|
||
|
|
)
|
||
|
|
{
|
||
|
|
DBG_ASSERT( bValid() );
|
||
|
|
DBG_ASSERT( pValueName );
|
||
|
|
|
||
|
|
DWORD dwType = REG_DWORD;
|
||
|
|
DWORD dwSize = sizeof( dwValue );
|
||
|
|
|
||
|
|
m_Status = m_Reg.m_QueryValueEx( m_hKey,
|
||
|
|
pValueName,
|
||
|
|
NULL,
|
||
|
|
&dwType,
|
||
|
|
reinterpret_cast<LPBYTE>( &dwValue ),
|
||
|
|
&dwSize );
|
||
|
|
|
||
|
|
return m_Status == ERROR_SUCCESS && dwType == REG_DWORD;
|
||
|
|
}
|
||
|
|
|
||
|
|
BOOL
|
||
|
|
TDebugRegistry::
|
||
|
|
bRead(
|
||
|
|
IN LPCTSTR pValueName,
|
||
|
|
IN OUT BOOL &bValue
|
||
|
|
)
|
||
|
|
{
|
||
|
|
DBG_ASSERT( bValid() );
|
||
|
|
DBG_ASSERT( pValueName );
|
||
|
|
|
||
|
|
DWORD dwType = REG_DWORD;
|
||
|
|
DWORD dwSize = sizeof( bValue );
|
||
|
|
|
||
|
|
m_Status = m_Reg.m_QueryValueEx( m_hKey,
|
||
|
|
pValueName,
|
||
|
|
NULL,
|
||
|
|
&dwType,
|
||
|
|
reinterpret_cast<LPBYTE>( &bValue ),
|
||
|
|
&dwSize );
|
||
|
|
|
||
|
|
return m_Status == ERROR_SUCCESS && dwType == REG_DWORD;
|
||
|
|
}
|
||
|
|
|
||
|
|
BOOL
|
||
|
|
TDebugRegistry::
|
||
|
|
bRead(
|
||
|
|
IN LPCTSTR pValueName,
|
||
|
|
IN OUT TDebugString &strValue
|
||
|
|
)
|
||
|
|
{
|
||
|
|
DBG_ASSERT( bValid() );
|
||
|
|
DBG_ASSERT( pValueName );
|
||
|
|
|
||
|
|
TCHAR szBuffer[kHint];
|
||
|
|
DWORD dwType = REG_SZ;
|
||
|
|
BOOL bStatus = FALSE;
|
||
|
|
DWORD dwSize = sizeof( szBuffer );
|
||
|
|
|
||
|
|
m_Status = m_Reg.m_QueryValueEx( m_hKey,
|
||
|
|
pValueName,
|
||
|
|
NULL,
|
||
|
|
&dwType,
|
||
|
|
reinterpret_cast<LPBYTE>( szBuffer ),
|
||
|
|
&dwSize );
|
||
|
|
|
||
|
|
if( m_Status == ERROR_MORE_DATA && dwType == REG_SZ && dwSize )
|
||
|
|
{
|
||
|
|
LPTSTR pszBuff = INTERNAL_NEW TCHAR[ dwSize ];
|
||
|
|
|
||
|
|
if( pszBuff )
|
||
|
|
{
|
||
|
|
m_Status = m_Reg.m_QueryValueEx( m_hKey,
|
||
|
|
pValueName,
|
||
|
|
NULL,
|
||
|
|
&dwType,
|
||
|
|
reinterpret_cast<LPBYTE>( pszBuff ),
|
||
|
|
&dwSize );
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
m_Status = ERROR_NOT_ENOUGH_MEMORY;
|
||
|
|
}
|
||
|
|
|
||
|
|
if( m_Status == ERROR_SUCCESS )
|
||
|
|
{
|
||
|
|
bStatus = strValue.bUpdate( pszBuff );
|
||
|
|
}
|
||
|
|
|
||
|
|
INTERNAL_DELETE [] pszBuff;
|
||
|
|
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
if( m_Status == ERROR_SUCCESS && dwType == REG_SZ )
|
||
|
|
{
|
||
|
|
bStatus = strValue.bUpdate( szBuffer );
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return bStatus;
|
||
|
|
}
|
||
|
|
|
||
|
|
BOOL
|
||
|
|
TDebugRegistry::
|
||
|
|
bRead(
|
||
|
|
IN LPCTSTR pValueName,
|
||
|
|
IN OUT PVOID pValue,
|
||
|
|
IN DWORD cbSize,
|
||
|
|
IN LPDWORD pcbNeeded OPTIONAL
|
||
|
|
)
|
||
|
|
{
|
||
|
|
DBG_ASSERT( bValid() );
|
||
|
|
DBG_ASSERT( pValueName );
|
||
|
|
DBG_ASSERT( pValue );
|
||
|
|
|
||
|
|
DWORD dwType = REG_BINARY;
|
||
|
|
BOOL bStatus = FALSE;
|
||
|
|
DWORD dwSize = cbSize;
|
||
|
|
|
||
|
|
m_Status = m_Reg.m_QueryValueEx( m_hKey,
|
||
|
|
pValueName,
|
||
|
|
NULL,
|
||
|
|
&dwType,
|
||
|
|
reinterpret_cast<LPBYTE>( pValue ),
|
||
|
|
&dwSize );
|
||
|
|
|
||
|
|
if( m_Status == ERROR_MORE_DATA && pcbNeeded )
|
||
|
|
{
|
||
|
|
*pcbNeeded = dwSize;
|
||
|
|
}
|
||
|
|
|
||
|
|
return m_Status == ERROR_SUCCESS && dwType == REG_BINARY && dwSize == cbSize;
|
||
|
|
}
|
||
|
|
|
||
|
|
BOOL
|
||
|
|
TDebugRegistry::
|
||
|
|
bWrite(
|
||
|
|
IN LPCTSTR pValueName,
|
||
|
|
IN const DWORD dwValue
|
||
|
|
)
|
||
|
|
{
|
||
|
|
DBG_ASSERT( bValid() );
|
||
|
|
DBG_ASSERT( pValueName );
|
||
|
|
|
||
|
|
m_Status = m_Reg.m_SetValueEx( m_hKey,
|
||
|
|
pValueName,
|
||
|
|
0,
|
||
|
|
REG_DWORD,
|
||
|
|
reinterpret_cast<const BYTE *>( &dwValue ),
|
||
|
|
sizeof( dwValue ) );
|
||
|
|
|
||
|
|
return m_Status == ERROR_SUCCESS;
|
||
|
|
}
|
||
|
|
|
||
|
|
BOOL
|
||
|
|
TDebugRegistry::
|
||
|
|
bWrite(
|
||
|
|
IN LPCTSTR pValueName,
|
||
|
|
IN LPCTSTR pszValue
|
||
|
|
)
|
||
|
|
{
|
||
|
|
DBG_ASSERT( bValid() );
|
||
|
|
DBG_ASSERT( pValueName );
|
||
|
|
|
||
|
|
m_Status = m_Reg.m_SetValueEx( m_hKey,
|
||
|
|
pValueName,
|
||
|
|
0,
|
||
|
|
REG_SZ,
|
||
|
|
reinterpret_cast<const BYTE *>( pszValue ),
|
||
|
|
_tcslen( pszValue ) * sizeof( TCHAR ) + sizeof( TCHAR ) );
|
||
|
|
|
||
|
|
return m_Status == ERROR_SUCCESS;
|
||
|
|
}
|
||
|
|
|
||
|
|
BOOL
|
||
|
|
TDebugRegistry::
|
||
|
|
bWrite(
|
||
|
|
IN LPCTSTR pValueName,
|
||
|
|
IN const PVOID pValue,
|
||
|
|
IN DWORD cbSize
|
||
|
|
)
|
||
|
|
{
|
||
|
|
DBG_ASSERT( bValid() );
|
||
|
|
DBG_ASSERT( pValueName );
|
||
|
|
DBG_ASSERT( pValue );
|
||
|
|
|
||
|
|
m_Status = m_Reg.m_SetValueEx( m_hKey,
|
||
|
|
pValueName,
|
||
|
|
0,
|
||
|
|
REG_BINARY,
|
||
|
|
reinterpret_cast<const BYTE *>( pValue ),
|
||
|
|
cbSize );
|
||
|
|
|
||
|
|
return m_Status == ERROR_SUCCESS;
|
||
|
|
}
|
||
|
|
|
||
|
|
BOOL
|
||
|
|
TDebugRegistry::
|
||
|
|
bRemove(
|
||
|
|
IN LPCTSTR pValueName
|
||
|
|
)
|
||
|
|
{
|
||
|
|
DBG_ASSERT( bValid() );
|
||
|
|
DBG_ASSERT( pValueName );
|
||
|
|
|
||
|
|
m_Status = m_Reg.m_DeleteValue( m_hKey, pValueName );
|
||
|
|
|
||
|
|
return m_Status == ERROR_SUCCESS;
|
||
|
|
}
|
||
|
|
|
||
|
|
BOOL
|
||
|
|
TDebugRegistry::
|
||
|
|
bRemoveKey(
|
||
|
|
IN LPCTSTR pKeyName
|
||
|
|
)
|
||
|
|
{
|
||
|
|
DBG_ASSERT( bValid() );
|
||
|
|
DBG_ASSERT( pKeyName );
|
||
|
|
|
||
|
|
m_Status = dwRecursiveRegDeleteKey( m_hKey, pKeyName );
|
||
|
|
|
||
|
|
return m_Status == ERROR_SUCCESS;
|
||
|
|
}
|
||
|
|
|
||
|
|
DWORD
|
||
|
|
TDebugRegistry::
|
||
|
|
dwRecursiveRegDeleteKey(
|
||
|
|
IN HKEY hKey,
|
||
|
|
IN LPCTSTR pszSubkey
|
||
|
|
) const
|
||
|
|
{
|
||
|
|
HKEY hSubkey = NULL;
|
||
|
|
|
||
|
|
DWORD dwStatus = m_Reg.m_OpenKeyEx( hKey,
|
||
|
|
pszSubkey,
|
||
|
|
0,
|
||
|
|
KEY_ALL_ACCESS,
|
||
|
|
&hSubkey );
|
||
|
|
|
||
|
|
while ( dwStatus == ERROR_SUCCESS )
|
||
|
|
{
|
||
|
|
TCHAR szSubkey[ MAX_PATH ];
|
||
|
|
DWORD cbSubkeySize = COUNTOF( szSubkey );
|
||
|
|
|
||
|
|
dwStatus = m_Reg.m_EnumKeyEx( hSubkey,
|
||
|
|
0,
|
||
|
|
szSubkey,
|
||
|
|
&cbSubkeySize,
|
||
|
|
0,
|
||
|
|
0,
|
||
|
|
0,
|
||
|
|
0 );
|
||
|
|
|
||
|
|
if( dwStatus == ERROR_NO_MORE_ITEMS )
|
||
|
|
break;
|
||
|
|
|
||
|
|
if( dwStatus != ERROR_SUCCESS )
|
||
|
|
return dwStatus;
|
||
|
|
|
||
|
|
dwStatus = dwRecursiveRegDeleteKey( hSubkey, szSubkey );
|
||
|
|
}
|
||
|
|
|
||
|
|
if( hSubkey )
|
||
|
|
{
|
||
|
|
m_Reg.m_CloseKey( hSubkey );
|
||
|
|
}
|
||
|
|
|
||
|
|
dwStatus = m_Reg.m_DeleteKey( hKey, pszSubkey );
|
||
|
|
|
||
|
|
return dwStatus;
|
||
|
|
}
|