mirror of https://github.com/tongzx/nt5src
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
817 lines
20 KiB
817 lines
20 KiB
#include "stdafx.h"
|
|
// defined with iisrtl2.lib
|
|
//DEBUG_PRINTS * g_pDebug;
|
|
|
|
// open an existing key
|
|
// if failure, return NULL pointer and put error into SetLastError();
|
|
CRegKey :: CRegKey (HKEY hKeyBase, LPCTSTR pchSubKey,REGSAM regSam ) : m_hKey( NULL ), m_dwDisposition( 0 )
|
|
{
|
|
LONG err = ERROR_SUCCESS ;
|
|
SetLastError(ERROR_SUCCESS);
|
|
m_iDisplayWarnings = TRUE;
|
|
|
|
if ( pchSubKey )
|
|
{err = ::RegOpenKeyEx( hKeyBase, pchSubKey, 0, regSam, & m_hKey ) ;}
|
|
else
|
|
{m_hKey = hKeyBase ;}
|
|
|
|
if ( err != ERROR_SUCCESS )
|
|
{
|
|
if (m_hKey) {::RegCloseKey(m_hKey);}
|
|
m_hKey = NULL ;
|
|
// Check if the error is because it simply doesn't exist.
|
|
// if this is the case then don't say failed. Say WARNING.
|
|
if ( err == ERROR_FILE_NOT_FOUND )
|
|
{
|
|
if (m_iDisplayWarnings)
|
|
{iisDebugOut((LOG_TYPE_TRACE_WIN32_API, _T("CRegKey:CRegKey() %s Key doesn't exist. WARNING. Code=0x%x\n"), pchSubKey, err));}
|
|
}
|
|
else
|
|
{
|
|
iisDebugOut((LOG_TYPE_ERROR, _T("CRegKey:CRegKey() %s FAILED. Err=0x%x\n"), pchSubKey, err));
|
|
}
|
|
SetLastError(err);
|
|
}
|
|
}
|
|
|
|
// Constructor creating a new key/opening a key if already exist, and set value if specified
|
|
CRegKey :: CRegKey (LPCTSTR lpSubKey,HKEY hKeyBase,LPCTSTR lpValueName,DWORD dwType,LPBYTE lpValueData,DWORD cbValueData): m_hKey( NULL ),m_dwDisposition( 0 )
|
|
{
|
|
LONG err = ERROR_SUCCESS;
|
|
SetLastError(ERROR_SUCCESS);
|
|
m_iDisplayWarnings = TRUE;
|
|
|
|
err = ::RegCreateKeyEx( hKeyBase, lpSubKey, 0, _T(""), REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS, NULL, & m_hKey, & m_dwDisposition ) ;
|
|
if ( err != ERROR_SUCCESS)
|
|
{
|
|
if ( m_hKey ) {::RegCloseKey( m_hKey ) ;}
|
|
m_hKey = NULL ;
|
|
// Check if the error is because it simply doesn't exist.
|
|
// if this is the case then don't say failed. Say WARNING.
|
|
if ( err == ERROR_FILE_NOT_FOUND )
|
|
{
|
|
if (m_iDisplayWarnings)
|
|
{iisDebugOut((LOG_TYPE_TRACE_WIN32_API, _T("CRegKey:CRegKey() %s Key doesn't exist. WARNING. Code=0x%x\n"), lpSubKey, err));}
|
|
}
|
|
else
|
|
{
|
|
iisDebugOut((LOG_TYPE_ERROR, _T("CRegKey::CRegKey() Open %s FAILED. err=0x%x\n"), lpSubKey, err));
|
|
}
|
|
SetLastError(err);
|
|
}
|
|
else
|
|
{
|
|
if (lpValueName)
|
|
{
|
|
::RegSetValueEx(m_hKey, lpValueName, 0, dwType, (const LPBYTE)lpValueData, cbValueData);
|
|
if ( err != ERROR_SUCCESS)
|
|
{
|
|
iisDebugOut((LOG_TYPE_ERROR, _T("CRegKey::CRegKey() RegSetValueEx %s, %s FAILED. err=0x%x\n"), lpSubKey,lpValueName, err));
|
|
SetLastError(err);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
CRegKey :: ~ CRegKey ()
|
|
{
|
|
if (m_hKey) {::RegCloseKey( m_hKey );}
|
|
}
|
|
|
|
|
|
// Prepare to read a value by finding the value's size.
|
|
LONG CRegKey :: PrepareValue (LPCTSTR pchValueName, DWORD * pdwType,DWORD * pcbSize,BYTE ** ppbData )
|
|
{
|
|
LONG err = 0 ;
|
|
|
|
BYTE chDummy[2] ;
|
|
DWORD cbData = 0 ;
|
|
|
|
do
|
|
{
|
|
// Set the resulting buffer size to 0.
|
|
*pcbSize = 0 ;
|
|
*ppbData = NULL ;
|
|
|
|
err = ::RegQueryValueEx( *this, (TCHAR *) pchValueName, 0, pdwType, chDummy, & cbData ) ;
|
|
|
|
// The only error we should get here is ERROR_MORE_DATA, but
|
|
// we may get no error if the value has no data.
|
|
if ( err == 0 )
|
|
{
|
|
cbData = sizeof (LONG) ; // Just a fudgy number
|
|
}
|
|
else
|
|
if ( err != ERROR_MORE_DATA )
|
|
break ;
|
|
|
|
// Allocate a buffer large enough for the data.
|
|
|
|
*ppbData = new BYTE [ (*pcbSize = cbData) + sizeof (LONG) ] ;
|
|
|
|
if ( *ppbData == NULL )
|
|
{
|
|
err = ERROR_NOT_ENOUGH_MEMORY ;
|
|
break ;
|
|
}
|
|
|
|
// Now that have a buffer, re-fetch the value.
|
|
|
|
err = ::RegQueryValueEx( *this, (TCHAR *) pchValueName, 0, pdwType, *ppbData, pcbSize ) ;
|
|
|
|
} while ( FALSE ) ;
|
|
|
|
if ( err ) {delete [] *ppbData ;}
|
|
|
|
return err ;
|
|
}
|
|
|
|
// Overloaded value query members; each returns ERROR_INVALID_PARAMETER
|
|
// if data exists but not in correct form to deliver into result object.
|
|
LONG CRegKey :: QueryValue ( LPCTSTR pchValueName, CString & strResult )
|
|
{
|
|
LONG err = 0 ;
|
|
|
|
DWORD dwType ;
|
|
DWORD cbData ;
|
|
BYTE * pabData = NULL ;
|
|
|
|
do
|
|
{
|
|
if ( err = PrepareValue( pchValueName, & dwType, & cbData, & pabData ) )
|
|
{break ;}
|
|
|
|
if (( dwType != REG_SZ ) && ( dwType != REG_EXPAND_SZ ))
|
|
{
|
|
err = ERROR_INVALID_PARAMETER ;
|
|
break ;
|
|
}
|
|
|
|
// Guarantee that the data looks like a string
|
|
pabData[cbData] = 0 ;
|
|
|
|
// Catch exceptions trying to assign to the caller's string
|
|
TRY
|
|
{
|
|
strResult = (TCHAR *) pabData ;
|
|
}
|
|
CATCH_ALL(e)
|
|
{
|
|
err = ERROR_NOT_ENOUGH_MEMORY ;
|
|
}
|
|
END_CATCH_ALL
|
|
}
|
|
while ( FALSE ) ;
|
|
|
|
// Memory leak....
|
|
//if ( err )
|
|
//{
|
|
delete [] pabData ;
|
|
//}
|
|
|
|
if (err)
|
|
{
|
|
if ( err == ERROR_FILE_NOT_FOUND )
|
|
{
|
|
if (m_iDisplayWarnings)
|
|
{iisDebugOut((LOG_TYPE_WARN, _T("CRegKey::QueryValue(): %s Not found. WARNING. code=0x%x\n"), pchValueName, err));}
|
|
}
|
|
else
|
|
{
|
|
if (err != ERROR_INVALID_PARAMETER)
|
|
{
|
|
iisDebugOut((LOG_TYPE_ERROR, _T("CRegKey::QueryValue(): %s FAILED. err=0x%x\n"), pchValueName, err));
|
|
}
|
|
}
|
|
}
|
|
return err ;
|
|
}
|
|
|
|
LONG CRegKey :: QueryValue ( LPCTSTR pchValueName, CStringList & strList )
|
|
{
|
|
LONG err = 0 ;
|
|
|
|
DWORD dwType ;
|
|
DWORD cbData ;
|
|
BYTE * pabData = NULL ;
|
|
TCHAR * pbTemp, * pbTempLimit ;
|
|
|
|
do
|
|
{
|
|
if ( err = PrepareValue( pchValueName, & dwType, & cbData, & pabData ) )
|
|
{break ;}
|
|
|
|
if ( dwType != REG_MULTI_SZ )
|
|
{
|
|
err = ERROR_INVALID_PARAMETER ;
|
|
break ;
|
|
}
|
|
|
|
// Guarantee that the trailing data looks like a string
|
|
pabData[cbData] = 0 ;
|
|
pbTemp = (TCHAR *) pabData ;
|
|
pbTempLimit = (TCHAR *) (& pabData[cbData]) ;
|
|
|
|
// Catch exceptions trying to build the list
|
|
TRY
|
|
{
|
|
for ( ; pbTemp < pbTempLimit ; )
|
|
{
|
|
strList.AddTail( pbTemp ) ;
|
|
pbTemp += ::_tcslen( pbTemp ) + 1 ;
|
|
}
|
|
}
|
|
CATCH_ALL(e)
|
|
{
|
|
err = ERROR_NOT_ENOUGH_MEMORY ;
|
|
}
|
|
END_CATCH_ALL
|
|
}
|
|
while ( FALSE ) ;
|
|
|
|
delete [] pabData ;
|
|
|
|
if (err)
|
|
{
|
|
if ( err == ERROR_FILE_NOT_FOUND )
|
|
{
|
|
if (m_iDisplayWarnings)
|
|
{iisDebugOut((LOG_TYPE_WARN, _T("CRegKey::QueryValue(): %s Not found. WARNING. code=0x%x\n"), pchValueName, err));}
|
|
}
|
|
else
|
|
{
|
|
if (err != ERROR_INVALID_PARAMETER)
|
|
{
|
|
iisDebugOut((LOG_TYPE_ERROR, _T("CRegKey::QueryValue(): %s FAILED. err=0x%x.\n"), pchValueName, err));
|
|
}
|
|
}
|
|
}
|
|
return err ;
|
|
}
|
|
|
|
LONG CRegKey :: QueryValue ( LPCTSTR pchValueName, DWORD & dwResult )
|
|
{
|
|
LONG err = 0 ;
|
|
|
|
DWORD dwType ;
|
|
DWORD cbData ;
|
|
BYTE * pabData = NULL ;
|
|
|
|
do
|
|
{
|
|
if ( err = PrepareValue( pchValueName, & dwType, & cbData, & pabData ) )
|
|
break ;
|
|
|
|
if ( dwType != REG_DWORD || cbData != sizeof dwResult )
|
|
{
|
|
err = ERROR_INVALID_PARAMETER ;
|
|
break ;
|
|
}
|
|
|
|
dwResult = *((DWORD *) pabData) ;
|
|
}
|
|
while ( FALSE ) ;
|
|
|
|
// Memory leak...
|
|
//if ( err )
|
|
//{
|
|
delete [] pabData ;
|
|
//}
|
|
|
|
if (err)
|
|
{
|
|
if ( err == ERROR_FILE_NOT_FOUND )
|
|
{
|
|
if (m_iDisplayWarnings)
|
|
{iisDebugOut((LOG_TYPE_WARN, _T("CRegKey::QueryValue(): %s Not found. WARNING. code=0x%x\n"), pchValueName, err));}
|
|
}
|
|
else
|
|
{
|
|
if (err != ERROR_INVALID_PARAMETER)
|
|
{
|
|
iisDebugOut((LOG_TYPE_ERROR, _T("CRegKey::QueryValue(): %s FAILED. err=0x%x.\n"), pchValueName, err));
|
|
}
|
|
}
|
|
}
|
|
return err ;
|
|
}
|
|
|
|
LONG CRegKey :: QueryValue ( LPCTSTR pchValueName, CByteArray & abResult )
|
|
{
|
|
LONG err = 0 ;
|
|
|
|
DWORD dwType ;
|
|
DWORD cbData ;
|
|
BYTE * pabData = NULL ;
|
|
|
|
do
|
|
{
|
|
if ( err = PrepareValue( pchValueName, & dwType, & cbData, & pabData ) )
|
|
break ;
|
|
|
|
if ( dwType != REG_BINARY )
|
|
{
|
|
err = ERROR_INVALID_PARAMETER ;
|
|
break ;
|
|
}
|
|
|
|
// Catch exceptions trying to grow the result array
|
|
TRY
|
|
{
|
|
abResult.SetSize( cbData ) ;
|
|
}
|
|
CATCH_ALL(e)
|
|
{
|
|
err = ERROR_NOT_ENOUGH_MEMORY ;
|
|
}
|
|
END_CATCH_ALL
|
|
|
|
if ( err )
|
|
break ;
|
|
|
|
// Move the data to the result array.
|
|
for ( DWORD i = 0 ; i < cbData ; i++ )
|
|
{
|
|
abResult[i] = pabData[i] ;
|
|
}
|
|
}
|
|
while ( FALSE ) ;
|
|
|
|
// Memory leak....
|
|
//if ( err )
|
|
//{
|
|
delete [] pabData ;
|
|
//}
|
|
|
|
if (err)
|
|
{
|
|
if ( err == ERROR_FILE_NOT_FOUND )
|
|
{
|
|
if (m_iDisplayWarnings)
|
|
{iisDebugOut((LOG_TYPE_WARN, _T("CRegKey::QueryValue(): %s Not found. WARNING. code=0x%x\n"), pchValueName, err));}
|
|
}
|
|
else
|
|
{
|
|
if (err != ERROR_INVALID_PARAMETER)
|
|
{
|
|
iisDebugOut((LOG_TYPE_ERROR, _T("CRegKey::QueryValue(): %s FAILED. err=0x%x.\n"), pchValueName, err));
|
|
}
|
|
}
|
|
}
|
|
return err ;
|
|
}
|
|
|
|
LONG CRegKey :: QueryValue ( LPCTSTR pchValueName, void * pvResult, DWORD cbSize )
|
|
{
|
|
LONG err = 0 ;
|
|
|
|
DWORD dwType ;
|
|
DWORD cbData ;
|
|
BYTE * pabData = NULL ;
|
|
|
|
do
|
|
{
|
|
if ( err = PrepareValue( pchValueName, & dwType, & cbData, & pabData ) )
|
|
break ;
|
|
|
|
if ( dwType != REG_BINARY )
|
|
{
|
|
err = ERROR_INVALID_PARAMETER ;
|
|
break ;
|
|
}
|
|
|
|
if ( cbSize < cbData )
|
|
{
|
|
err = ERROR_MORE_DATA;
|
|
break;
|
|
}
|
|
|
|
::memcpy(pvResult, pabData, cbData);
|
|
}
|
|
while ( FALSE ) ;
|
|
|
|
// Memory leak....
|
|
//if ( err )
|
|
//{
|
|
delete [] pabData ;
|
|
//}
|
|
|
|
if (err)
|
|
{
|
|
if ( err == ERROR_FILE_NOT_FOUND )
|
|
{
|
|
if (m_iDisplayWarnings)
|
|
{iisDebugOut((LOG_TYPE_WARN, _T("CRegKey::QueryValue(): %s Not found. WARNING. code=0x%x\n"), pchValueName, err));}
|
|
}
|
|
else
|
|
{
|
|
if (err != ERROR_INVALID_PARAMETER)
|
|
{
|
|
iisDebugOut((LOG_TYPE_ERROR, _T("CRegKey::QueryValue(): %s FAILED. err=0x%x.\n"), pchValueName, err));
|
|
}
|
|
}
|
|
}
|
|
return err ;
|
|
}
|
|
|
|
|
|
// Overloaded value setting members.
|
|
LONG CRegKey :: SetValue ( LPCTSTR pchValueName, LPCTSTR szResult, BOOL fExpand )
|
|
{
|
|
LONG err = 0;
|
|
err = ::RegSetValueEx( *this, pchValueName,0,fExpand ? REG_EXPAND_SZ : REG_SZ,(const BYTE *) szResult,(_tcsclen(szResult) + 1) * sizeof(_TCHAR) ) ;
|
|
if (err != ERROR_SUCCESS) {iisDebugOut((LOG_TYPE_ERROR, _T("CRegKey::SetValue(): %s FAILED, err=0x%x\n"), pchValueName,err));}
|
|
return err ;
|
|
}
|
|
|
|
LONG CRegKey :: SetValue ( LPCTSTR pchValueName, CStringList & strList )
|
|
{
|
|
LONG err = 0;
|
|
|
|
DWORD cbSize ;
|
|
BYTE * pbData = NULL ;
|
|
|
|
err = FlattenValue( strList, & cbSize, & pbData ) ;
|
|
if ( err == 0 )
|
|
{
|
|
err = ::RegSetValueEx( *this, pchValueName,0,REG_MULTI_SZ,pbData, cbSize ) ;
|
|
if (err != ERROR_SUCCESS) {iisDebugOut((LOG_TYPE_ERROR, _T("CRegKey::SetValue(): %s FAILED, err=0x%x\n"), pchValueName,err));}
|
|
}
|
|
|
|
delete pbData ;
|
|
|
|
return err ;
|
|
}
|
|
|
|
LONG CRegKey :: SetValue ( LPCTSTR pchValueName, DWORD dwResult )
|
|
{
|
|
LONG err = 0;
|
|
err = ::RegSetValueEx( *this, pchValueName,0,REG_DWORD,(const BYTE *) & dwResult,sizeof dwResult ) ;
|
|
if (err != ERROR_SUCCESS) {iisDebugOut((LOG_TYPE_ERROR, _T("CRegKey::SetValue(): %s FAILED, err=0x%x\n"), pchValueName,err));}
|
|
return err ;
|
|
}
|
|
|
|
LONG CRegKey :: SetValue ( LPCTSTR pchValueName, CByteArray & abResult )
|
|
{
|
|
LONG err = 0;
|
|
|
|
DWORD cbSize ;
|
|
BYTE * pbData = NULL ;
|
|
|
|
err = FlattenValue( abResult, & cbSize, & pbData ) ;
|
|
|
|
if ( err == 0 )
|
|
{
|
|
err = ::RegSetValueEx( *this, pchValueName,0,REG_BINARY,pbData, cbSize ) ;
|
|
if (err != ERROR_SUCCESS) {iisDebugOut((LOG_TYPE_ERROR, _T("CRegKey::SetValue(): %s FAILED, err=0x%x\n"), pchValueName,err));}
|
|
}
|
|
|
|
delete pbData ;
|
|
|
|
return err ;
|
|
}
|
|
|
|
LONG CRegKey :: SetValue ( LPCTSTR pchValueName, void * pvResult, DWORD cbSize )
|
|
{
|
|
LONG err = 0;
|
|
err = ::RegSetValueEx( *this, pchValueName,0,REG_BINARY,(const BYTE *)pvResult, cbSize ) ;
|
|
if (err != ERROR_SUCCESS) {iisDebugOut((LOG_TYPE_ERROR, _T("CRegKey::SetValue(): %s FAILED, err=0x%x\n"), pchValueName,err));}
|
|
return err ;
|
|
}
|
|
|
|
LONG CRegKey::DeleteValue( LPCTSTR pchKeyName )
|
|
{
|
|
LONG err = 0;
|
|
err = ::RegDeleteValue( *this, pchKeyName );
|
|
if (err != ERROR_SUCCESS)
|
|
{
|
|
if ( err != ERROR_FILE_NOT_FOUND )
|
|
{
|
|
if (m_iDisplayWarnings)
|
|
iisDebugOut((LOG_TYPE_WARN, _T("CRegKey::DeleteValue(): %s FAILED, err=0x%x\n"), pchKeyName,err));
|
|
}
|
|
}
|
|
return(err);
|
|
}
|
|
|
|
LONG CRegKey::DeleteTree( LPCTSTR pchKeyName )
|
|
{
|
|
LONG err = 0;
|
|
CRegKey regSubKey( *this, pchKeyName );
|
|
|
|
if ( NULL != (HKEY) regSubKey )
|
|
{
|
|
CString strName;
|
|
CTime cTime;
|
|
|
|
while (TRUE)
|
|
{
|
|
CRegKeyIter regEnum( regSubKey );
|
|
|
|
if ( regEnum.Next( &strName, &cTime ) != ERROR_SUCCESS )
|
|
{
|
|
break;
|
|
}
|
|
|
|
regSubKey.DeleteTree( strName );
|
|
}
|
|
// delete myself
|
|
err = ::RegDeleteKey( *this, pchKeyName );
|
|
}
|
|
return(err);
|
|
}
|
|
|
|
LONG CRegKey :: FlattenValue ( CStringList & strList, DWORD * pcbSize, BYTE ** ppbData )
|
|
{
|
|
LONG err = 0 ;
|
|
|
|
POSITION pos ;
|
|
CString * pstr ;
|
|
int cbTotal = 0 ;
|
|
|
|
// Walk the list accumulating sizes
|
|
for ( pos = strList.GetHeadPosition() ; pos != NULL && (pstr = & strList.GetNext( pos )) ; )
|
|
{
|
|
cbTotal += ((pstr->GetLength() + 1)*sizeof(TCHAR));
|
|
}
|
|
|
|
// Allocate and fill a temporary buffer
|
|
if (*pcbSize = cbTotal)
|
|
{
|
|
TRY
|
|
{
|
|
*ppbData = new BYTE[ *pcbSize ] ;
|
|
|
|
if (*ppbData)
|
|
{
|
|
BYTE * pbData = *ppbData ;
|
|
|
|
// Populate the buffer with the strings.
|
|
for ( pos = strList.GetHeadPosition() ; pos != NULL && (pstr = & strList.GetNext( pos )) ; )
|
|
{
|
|
int cb = (pstr->GetLength() + 1)*sizeof(TCHAR) ;
|
|
::memcpy( pbData, (LPCTSTR) *pstr, cb ) ;
|
|
pbData += cb ;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
err = ERROR_NOT_ENOUGH_MEMORY ;
|
|
}
|
|
}
|
|
CATCH_ALL(e)
|
|
{
|
|
err = ERROR_NOT_ENOUGH_MEMORY ;
|
|
}
|
|
END_CATCH_ALL
|
|
}
|
|
else
|
|
{
|
|
*ppbData = NULL;
|
|
}
|
|
|
|
if (err)
|
|
{
|
|
iisDebugOut((LOG_TYPE_WARN, _T("CRegKey::FlattenValue(): %s FAILED. err=0x%x.\n"), strList, err));
|
|
}
|
|
return err ;
|
|
}
|
|
|
|
LONG CRegKey :: FlattenValue ( CByteArray & abData,DWORD * pcbSize,BYTE ** ppbData )
|
|
{
|
|
LONG err = 0 ;
|
|
|
|
DWORD i ;
|
|
|
|
// Allocate and fill a temporary buffer
|
|
if (*pcbSize = (DWORD)abData.GetSize())
|
|
{
|
|
TRY
|
|
{
|
|
*ppbData = new BYTE[*pcbSize] ;
|
|
if (*ppbData)
|
|
{
|
|
for ( i = 0 ; i < *pcbSize ; i++ )
|
|
{
|
|
(*ppbData)[i] = abData[i] ;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
err = ERROR_NOT_ENOUGH_MEMORY ;
|
|
}
|
|
|
|
}
|
|
CATCH_ALL(e)
|
|
{
|
|
err = ERROR_NOT_ENOUGH_MEMORY ;
|
|
}
|
|
END_CATCH_ALL
|
|
}
|
|
else
|
|
{
|
|
*ppbData = NULL;
|
|
}
|
|
|
|
if (err)
|
|
{
|
|
iisDebugOut((LOG_TYPE_WARN, _T("CRegKey::FlattenValue(): FAILED. err=0x%x.\n"), err));
|
|
}
|
|
return err ;
|
|
}
|
|
|
|
|
|
LONG CRegKey :: QueryKeyInfo ( CREGKEY_KEY_INFO * pRegKeyInfo )
|
|
{
|
|
LONG err = 0 ;
|
|
pRegKeyInfo->dwClassNameSize = sizeof pRegKeyInfo->chBuff - 1 ;
|
|
err = ::RegQueryInfoKey( *this,pRegKeyInfo->chBuff,& pRegKeyInfo->dwClassNameSize,NULL,& pRegKeyInfo->dwNumSubKeys,& pRegKeyInfo->dwMaxSubKey,& pRegKeyInfo->dwMaxClass,& pRegKeyInfo->dwMaxValues,& pRegKeyInfo->dwMaxValueName,& pRegKeyInfo->dwMaxValueData,& pRegKeyInfo->dwSecDesc,& pRegKeyInfo->ftKey ) ;
|
|
if (err != ERROR_SUCCESS) {iisDebugOut((LOG_TYPE_ERROR, _T("CRegKey::QueryKeyInfo(): FAILED. err=0x%x.\n"), err));}
|
|
return err ;
|
|
}
|
|
|
|
CRegKeyIter :: CRegKeyIter ( CRegKey & regKey ) : m_rk_iter( regKey ),m_p_buffer( NULL ),m_cb_buffer( 0 )
|
|
{
|
|
LONG err = 0 ;
|
|
|
|
CRegKey::CREGKEY_KEY_INFO regKeyInfo ;
|
|
|
|
Reset() ;
|
|
|
|
err = regKey.QueryKeyInfo( & regKeyInfo ) ;
|
|
if ( err == 0 )
|
|
{
|
|
TRY
|
|
{
|
|
m_cb_buffer = regKeyInfo.dwMaxSubKey + sizeof (DWORD) ;
|
|
m_p_buffer = new TCHAR [ m_cb_buffer ] ;
|
|
}
|
|
CATCH_ALL(e)
|
|
{
|
|
err = ERROR_NOT_ENOUGH_MEMORY ;
|
|
}
|
|
END_CATCH_ALL
|
|
}
|
|
|
|
if (err)
|
|
{
|
|
if (ERROR_NO_MORE_ITEMS != err)
|
|
{
|
|
iisDebugOut((LOG_TYPE_ERROR, _T("CRegKeyIter::CRegKeyIter(): FAILED. err=0x%x.\n"), err));
|
|
}
|
|
}
|
|
}
|
|
|
|
CRegKeyIter :: ~ CRegKeyIter ()
|
|
{
|
|
delete [] m_p_buffer ;
|
|
}
|
|
|
|
|
|
// Get the name (and optional last write time) of the next key.
|
|
LONG CRegKeyIter :: Next ( CString * pstrName, CTime * pTime )
|
|
{
|
|
LONG err = 0;
|
|
|
|
FILETIME ftDummy ;
|
|
DWORD dwNameSize = m_cb_buffer ;
|
|
|
|
err = ::RegEnumKeyEx( m_rk_iter, m_dw_index, m_p_buffer,& dwNameSize, NULL,NULL,NULL,& ftDummy ) ;
|
|
if ( err == 0 )
|
|
{
|
|
m_dw_index++ ;
|
|
|
|
if ( pTime )
|
|
{
|
|
*pTime = ftDummy ;
|
|
}
|
|
|
|
TRY
|
|
{
|
|
*pstrName = m_p_buffer ;
|
|
}
|
|
CATCH_ALL(e)
|
|
{
|
|
err = ERROR_NOT_ENOUGH_MEMORY ;
|
|
}
|
|
END_CATCH_ALL
|
|
}
|
|
|
|
if (err)
|
|
{
|
|
if (ERROR_NO_MORE_ITEMS != err)
|
|
{
|
|
iisDebugOut((LOG_TYPE_ERROR, _T("CRegKeyIter::Next(): FAILED. err=0x%x.\n"), err));
|
|
}
|
|
}
|
|
return err ;
|
|
}
|
|
|
|
|
|
CRegValueIter :: CRegValueIter ( CRegKey & regKey )
|
|
: m_rk_iter( regKey ),
|
|
m_p_buffer( NULL ),
|
|
m_cb_buffer( 0 )
|
|
{
|
|
LONG err = 0 ;
|
|
|
|
CRegKey::CREGKEY_KEY_INFO regKeyInfo ;
|
|
|
|
Reset() ;
|
|
|
|
err = regKey.QueryKeyInfo( & regKeyInfo ) ;
|
|
|
|
if ( err == 0 )
|
|
{
|
|
TRY
|
|
{
|
|
m_cb_buffer = regKeyInfo.dwMaxValueName + sizeof (DWORD) ;
|
|
m_p_buffer = new TCHAR [ m_cb_buffer ] ;
|
|
}
|
|
CATCH_ALL(e)
|
|
{
|
|
err = ERROR_NOT_ENOUGH_MEMORY ;
|
|
}
|
|
END_CATCH_ALL
|
|
}
|
|
|
|
if (err)
|
|
{
|
|
if (ERROR_NO_MORE_ITEMS != err)
|
|
{
|
|
iisDebugOut((LOG_TYPE_ERROR, _T("CRegValueIter::CRegValueIter(): FAILED. err=0x%x.\n"), err));
|
|
}
|
|
}
|
|
}
|
|
|
|
CRegValueIter :: ~ CRegValueIter ()
|
|
{
|
|
delete [] m_p_buffer ;
|
|
}
|
|
|
|
LONG CRegValueIter :: Next ( CString * pstrName, DWORD * pdwType )
|
|
{
|
|
LONG err = 0 ;
|
|
|
|
DWORD dwNameLength = m_cb_buffer ;
|
|
err = ::RegEnumValue( m_rk_iter,m_dw_index,m_p_buffer,& dwNameLength,NULL,pdwType,NULL,NULL ) ;
|
|
if ( err == 0 )
|
|
{
|
|
m_dw_index++ ;
|
|
|
|
TRY
|
|
{
|
|
*pstrName = m_p_buffer ;
|
|
}
|
|
CATCH_ALL(e)
|
|
{
|
|
err = ERROR_NOT_ENOUGH_MEMORY ;
|
|
}
|
|
END_CATCH_ALL
|
|
}
|
|
|
|
if (err)
|
|
{
|
|
if (ERROR_NO_MORE_ITEMS != err)
|
|
{
|
|
iisDebugOut((LOG_TYPE_ERROR, _T("CRegValueIter::Next(): FAILED. err=0x%x.\n"), err));
|
|
}
|
|
}
|
|
return err ;
|
|
}
|
|
|
|
LONG CRegValueIter :: Next ( CString * pstrName, CString * pstrValue )
|
|
{
|
|
LONG err = 0 ;
|
|
|
|
DWORD dwNameLength = m_cb_buffer ;
|
|
|
|
TCHAR szValue[_MAX_PATH];
|
|
DWORD dwValue = _MAX_PATH * sizeof(TCHAR);
|
|
|
|
err = ::RegEnumValue( m_rk_iter,m_dw_index,m_p_buffer,& dwNameLength,NULL,NULL,(LPBYTE)szValue,&dwValue ) ;
|
|
if ( err == 0 )
|
|
{
|
|
m_dw_index++ ;
|
|
|
|
TRY
|
|
{
|
|
*pstrName = m_p_buffer ;
|
|
*pstrValue = szValue;
|
|
}
|
|
CATCH_ALL(e)
|
|
{
|
|
err = ERROR_NOT_ENOUGH_MEMORY ;
|
|
}
|
|
END_CATCH_ALL
|
|
}
|
|
|
|
if (err)
|
|
{
|
|
if (ERROR_NO_MORE_ITEMS != err)
|
|
{
|
|
iisDebugOut((LOG_TYPE_ERROR, _T("CRegValueIter::Next(): FAILED. err=0x%x.\n"), err));
|
|
}
|
|
}
|
|
return err ;
|
|
}
|