|
|
#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 { err = PrepareValue( pchValueName, & dwType, & cbData, & pabData );
if ( err ) { 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 { err = PrepareValue( pchValueName, & dwType, & cbData, & pabData );
if ( err != 0 ) { 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 { err = PrepareValue( pchValueName, & dwType, & cbData, & pabData );
if ( err != 0 ) 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 { err = PrepareValue( pchValueName, & dwType, & cbData, & pabData );
if ( err != 0 ) { 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 { err = PrepareValue( pchValueName, & dwType, & cbData, & pabData );
if ( err != 0 ) { 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 ) ) != NULL ) ; ) { cbTotal += ((pstr->GetLength() + 1)*sizeof(TCHAR)); }
// Allocate and fill a temporary buffer
*pcbSize = cbTotal;
if ( *pcbSize ) { 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 ) ) != NULL ); ) { 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
*pcbSize = (DWORD)abData.GetSize();
if ( *pcbSize ) { 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 ; }
|