148 lines
4.0 KiB
C++
148 lines
4.0 KiB
C++
/*--------------------------------------------------------------------------*
|
|
*
|
|
* Microsoft Windows
|
|
* Copyright (C) Microsoft Corporation, 1992 - 00
|
|
*
|
|
* File: regkeyex.inl
|
|
*
|
|
* Contents: Inline files for CRegKeyEx class
|
|
*
|
|
* History: 7-Apr-2000 jeffro Created
|
|
*
|
|
*--------------------------------------------------------------------------*/
|
|
|
|
|
|
/*+-------------------------------------------------------------------------*
|
|
* CRegKeyEx::ScQueryString
|
|
*
|
|
* Loads a string from the given value name.
|
|
*
|
|
* The template type StringType can be any string class that supports
|
|
* MFC's CString interface (i.e. MFC's CString, WTL::CString, or MMC's
|
|
* CStr).
|
|
*--------------------------------------------------------------------------*/
|
|
|
|
template<class StringType>
|
|
SC ScQueryString (
|
|
LPCTSTR pszValueName, /* I:name of value to read */
|
|
StringType& strData, /* O:contents of the value */
|
|
DWORD* pdwType /*=NULL*/) /* O:REG_SZ, REG_EXPAND_SZ, REG_MULTI_SZ*/
|
|
{
|
|
DECLARE_SC (sc, _T("CRegKeyEx::ScQueryString"));
|
|
|
|
/*
|
|
* clear out existing contents
|
|
*/
|
|
strData.Empty ();
|
|
|
|
/*
|
|
* find out how much space we need to load the string
|
|
*/
|
|
DWORD dwType = REG_SZ;
|
|
DWORD cbBuffer = 0;
|
|
|
|
sc = ScQueryValue (pszValueName, &dwType, NULL, &cbBuffer);
|
|
if (sc)
|
|
return (sc);
|
|
|
|
/*
|
|
* if we're not loading a string type, return an error
|
|
*/
|
|
if (pdwType != NULL)
|
|
*pdwType = dwType;
|
|
|
|
if ((dwType != REG_SZ) && (dwType != REG_EXPAND_SZ) && (dwType != REG_MULTI_SZ))
|
|
return (sc = ScFromWin32 (ERROR_INVALID_DATATYPE));
|
|
|
|
/*
|
|
* allocate a buffer for the string
|
|
*/
|
|
DWORD cchBuffer = cbBuffer / sizeof (TCHAR);
|
|
LPTSTR pBuffer = strData.GetBuffer (cchBuffer);
|
|
if (pBuffer == NULL)
|
|
return (sc = E_OUTOFMEMORY);
|
|
|
|
/*
|
|
* load the string from the registry
|
|
*/
|
|
sc = ScQueryValue (pszValueName, &dwType, pBuffer, &cbBuffer);
|
|
strData.ReleaseBuffer (cchBuffer);
|
|
if (sc)
|
|
return (sc);
|
|
|
|
return (sc);
|
|
}
|
|
|
|
|
|
/*+-------------------------------------------------------------------------*
|
|
* CRegKeyEx::ScLoadRegUIString
|
|
*
|
|
* Wrapper around SHLoadRegUIString, which is used to support MUI.
|
|
*
|
|
* SHLoadRegUIString will read a string of the form
|
|
*
|
|
* @[path\]<dllname>,-<strId>
|
|
*
|
|
* The string with id <strId> is loaded from <dllname>. If no explicit
|
|
* path is provided then the DLL will be chosen according to pluggable UI
|
|
* specifications, if possible.
|
|
*
|
|
* If the registry string is not of the special form described here,
|
|
* SHLoadRegUIString will return the string intact.
|
|
*
|
|
* The template type StringType can be any string class that supports
|
|
* MFC's CString interface (i.e. MFC's CString, WTL::CString, or MMC's
|
|
* CStr).
|
|
*--------------------------------------------------------------------------*/
|
|
|
|
template<class StringType>
|
|
SC ScLoadRegUIString (
|
|
LPCTSTR pszValueName, /* I:name of value to read */
|
|
StringType& strData) /* O:logical contents of the value */
|
|
{
|
|
DECLARE_SC (sc, _T("CRegKeyEx::ScLoadRegUIString"));
|
|
|
|
/*
|
|
* clear out existing contents
|
|
*/
|
|
strData.Empty ();
|
|
|
|
const int cbGrow = 256;
|
|
int cbBuffer = 0;
|
|
|
|
do
|
|
{
|
|
/*
|
|
* allocate a larger buffer for the string
|
|
*/
|
|
cbBuffer += cbGrow;
|
|
LPTSTR pBuffer = strData.GetBuffer (cbBuffer);
|
|
if (pBuffer == NULL)
|
|
return (sc = E_OUTOFMEMORY);
|
|
|
|
/*
|
|
* load the string from the registry
|
|
* Most of the snapins do not have MUI string so we do not want to trace
|
|
* this error as the caller takes care of error condition and reads registry
|
|
* directly.
|
|
*/
|
|
SC scNoTrace = SHLoadRegUIString (m_hKey, pszValueName, pBuffer, cbBuffer);
|
|
strData.ReleaseBuffer();
|
|
if (scNoTrace)
|
|
return scNoTrace;
|
|
|
|
/*
|
|
* If we filled up the buffer, we'll pessimistically assume that
|
|
* there's more data available. We'll loop around, grow the buffer,
|
|
* and try again.
|
|
*/
|
|
} while (strData.GetLength() == cbBuffer-1);
|
|
|
|
/*
|
|
* free up extra space
|
|
*/
|
|
strData.FreeExtra();
|
|
|
|
return (sc);
|
|
}
|