CRegistrySource Class Reference

Inheritance diagram for CRegistrySource:

Collaboration diagram for CRegistrySource:

Public Member Functions
virtual	~CRegistrySource ()
HRESULT	Init (HKEY hKey, PCWSTR pszSubKey, BOOL bCreate)
STDMETHODIMP	QueryInterface (REFIID riid, void **ppvObject) override
	STDMETHODIMP_ (ULONG) AddRef() override
	STDMETHODIMP_ (ULONG) Release() override
STDMETHODIMP	EnumValues (IEnumString **ppEnum) override
STDMETHODIMP	EnumSources (IEnumString **ppEnum) override
STDMETHODIMP	QueryValueString (PCWSTR keyName, PCWSTR valueName, PWSTR *ppszValue) override
STDMETHODIMP	QueryValueDword (PCWSTR keyName, PCWSTR valueName, DWORD *pdwValue) override
STDMETHODIMP	QueryValueExists (PCWSTR keyName, PCWSTR valueName) override
STDMETHODIMP	QueryValueDirect (PCWSTR keyName, PCWSTR valueName, FLAGGED_BYTE_BLOB **ppBlob) override
STDMETHODIMP	OpenSource (PCWSTR keyName, BOOL bCreate, IQuerySourceOld **ppSource) override
STDMETHODIMP	SetValueDirect (PCWSTR keyName, PCWSTR valueName, DWORD dwType, DWORD cbData, LPCVOID pvData) override
STDMETHODIMP	SetKey (HKEY hKey) override
STDMETHODIMP	GetKey (HKEY *phKey) override

Private Attributes
LONG	m_cRefs = 1
HKEY	m_hKey = NULL

Detailed Description

Definition at line 91 of file querysrc.cpp.

Constructor & Destructor Documentation

~CRegistrySource()

CRegistrySource::~CRegistrySource ( )

virtual

Definition at line 254 of file querysrc.cpp.

{
    if (m_hKey)
        RegCloseKey(m_hKey);
}

Member Function Documentation

EnumSources()

STDMETHODIMP CRegistrySource::EnumSources ( IEnumString **  ppEnum )

override

Definition at line 322 of file querysrc.cpp.

{
    CRegistryEnumKeys* pEnum = new(std::nothrow) CRegistryEnumKeys();
    if (!pEnum)
        return E_OUTOFMEMORY;
    HRESULT hr = pEnum->Init(m_hKey, this);
    if (FAILED(hr))
    {
        pEnum->Release();
        pEnum = NULL;
    }
    *ppEnum = pEnum;
    return pEnum ? S_OK : E_OUTOFMEMORY;
}

EnumValues()

STDMETHODIMP CRegistrySource::EnumValues ( IEnumString **  ppEnum )

override

Definition at line 307 of file querysrc.cpp.

{
    CRegistryEnumValues* pEnum = new(std::nothrow) CRegistryEnumValues();
    if (!pEnum)
        return E_OUTOFMEMORY;
    HRESULT hr = pEnum->Init(m_hKey, this);
    if (FAILED(hr))
    {
        pEnum->Release();
        pEnum = NULL;
    }
    *ppEnum = pEnum;
    return pEnum ? S_OK : E_OUTOFMEMORY;
}

GetKey()

STDMETHODIMP CRegistrySource::GetKey ( HKEY *  phKey )

override

Definition at line 480 of file querysrc.cpp.

{
    if (!m_hKey)
        return E_UNEXPECTED;
 
    *phKey = SHRegDuplicateHKey(m_hKey);
    if (!*phKey)
        return E_UNEXPECTED;
 
    return S_OK;
}

Init()

HRESULT CRegistrySource::Init	(	HKEY	hKey,
		PCWSTR	pszSubKey,
		BOOL	bCreate
	)

Definition at line 260 of file querysrc.cpp.

{
    LSTATUS error;
    if (bCreate)
        error = RegCreateKeyExW(hKey, pszSubKey, 0, NULL, 0, MAXIMUM_ALLOWED, NULL, &m_hKey, NULL);
    else
        error = RegOpenKeyExW(hKey, pszSubKey, 0, MAXIMUM_ALLOWED, &m_hKey);
 
    return error ? HRESULT_FROM_WIN32(error) : S_OK;
}

Referenced by QuerySourceCreateFromKey().

OpenSource()

STDMETHODIMP CRegistrySource::OpenSource ( PCWSTR  keyName, BOOL  bCreate, IQuerySourceOld **  ppSource  )

override

Definition at line 453 of file querysrc.cpp.

{
    return QuerySourceCreateFromKey(m_hKey, keyName, bCreate, IID_IQuerySourceOld, (PVOID*)ppSource);
}

QueryInterface()

STDMETHODIMP CRegistrySource::QueryInterface ( REFIID  riid, void **  ppvObject  )

override

Definition at line 271 of file querysrc.cpp.

{
    if (!ppvObject)
        return E_POINTER;
 
    if (riid == IID_IQuerySourceOld)
    {
        *ppvObject = static_cast<IQuerySourceOld*>(this);
        AddRef();
        return S_OK;
    }
 
    if (riid == IID_IObjectWithRegistryKeyOld)
    {
        *ppvObject = static_cast<IObjectWithRegistryKeyOld*>(this);
        AddRef();
        return S_OK;
    }
 
    *ppvObject = NULL;
    return E_NOINTERFACE;
}

Referenced by QuerySourceCreateFromKey().

QueryValueDirect()

STDMETHODIMP CRegistrySource::QueryValueDirect ( PCWSTR  keyName, PCWSTR  valueName, FLAGGED_BYTE_BLOB **  ppBlob  )

override

Definition at line 401 of file querysrc.cpp.

{
    HRESULT hr = E_FAIL;
    HKEY hKey = m_hKey;
    DWORD dwType = REG_NONE;
    DWORD cbData = 256;
    BYTE abData[256];
 
    *ppBlob = NULL;
 
    if (keyName && *keyName &&
        RegOpenKeyExW(m_hKey, keyName, 0, KEY_READ, &hKey) != ERROR_SUCCESS)
    {
        return E_FAIL;
    }
 
    LSTATUS error = RegQueryValueExW(hKey, valueName, NULL, &dwType, abData, &cbData);
    if (error == ERROR_SUCCESS)
    {
        hr = SHAllocBlob(cbData, abData, ppBlob);
    }
    else if (error == ERROR_MORE_DATA)
    {
        hr = SHAllocBlob(cbData, NULL, ppBlob);
        if (SUCCEEDED(hr))
        {
            error = RegQueryValueExW(hKey, valueName, NULL, &dwType, (*ppBlob)->abData, &cbData);
            if (error != ERROR_SUCCESS)
            {
                CoTaskMemFree(*ppBlob);
                *ppBlob = NULL;
                hr = error ? HRESULT_FROM_WIN32(error) : S_OK;
            }
        }
    }
    else
    {
        hr = HRESULT_FROM_WIN32(error);
    }
 
    if (hKey != m_hKey)
        RegCloseKey(hKey);
 
    if (SUCCEEDED(hr))
        (*ppBlob)->fFlags = dwType;
 
    return hr;
}

QueryValueDword()

STDMETHODIMP CRegistrySource::QueryValueDword ( PCWSTR  keyName, PCWSTR  valueName, DWORD *  pdwValue  )

override

Definition at line 382 of file querysrc.cpp.

{
    DWORD dwType, cbValue = sizeof(*pdwValue);
    LSTATUS error = SHGetValueW(m_hKey, keyName, valueName, &dwType, pdwValue, &cbValue);
    if (error)
        return HRESULT_FROM_WIN32(error);
    if (dwType != REG_DWORD)
        return E_DATATYPE_MISMATCH;
    return S_OK;
}

QueryValueExists()

STDMETHODIMP CRegistrySource::QueryValueExists ( PCWSTR  keyName, PCWSTR  valueName  )

override

Definition at line 393 of file querysrc.cpp.

{
    LSTATUS error = SHGetValueW(m_hKey, keyName, valueName, NULL, NULL, NULL);
    if (error)
        return HRESULT_FROM_WIN32(error);
    return S_OK;
}

QueryValueString()

STDMETHODIMP CRegistrySource::QueryValueString ( PCWSTR  keyName, PCWSTR  valueName, PWSTR *  ppszValue  )

override

Definition at line 337 of file querysrc.cpp.

{
    *ppszValue = NULL;
 
    WCHAR szData[128];
    DWORD dwType = REG_NONE, cbData = sizeof(szData);
    LSTATUS error = SHGetValueW(m_hKey, keyName, valueName, &dwType, szData, &cbData);
    if (error == ERROR_SUCCESS)
    {
        if (dwType != REG_SZ)
            return E_DATATYPE_MISMATCH;
        if (!valueName && !szData[0])
            return HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND);
        return SHStrDupW(szData, ppszValue);
    }
 
    if (error != ERROR_MORE_DATA)
        return error ? HRESULT_FROM_WIN32(error) : S_OK;
 
    *ppszValue = (PWSTR)CoTaskMemAlloc(cbData);
    if (!*ppszValue)
        return E_OUTOFMEMORY;
 
    HRESULT hr = S_OK;
    error = SHGetValueW(m_hKey, keyName, valueName, &dwType, *ppszValue, &cbData);
    if (error)
    {
        CoTaskMemFree(*ppszValue);
        *ppszValue = NULL;
        hr = HRESULT_FROM_WIN32(error);
    }
 
    if (SUCCEEDED(hr) && dwType != REG_SZ)
    {
        CoTaskMemFree(*ppszValue);
        *ppszValue = NULL;
        hr = E_DATATYPE_MISMATCH;
    }
 
    return hr;
}

SetKey()

STDMETHODIMP CRegistrySource::SetKey ( HKEY  hKey )

override

Definition at line 471 of file querysrc.cpp.

{
    if (m_hKey)
        return E_UNEXPECTED;
 
    m_hKey = SHRegDuplicateHKey(hKey);
    return m_hKey ? S_OK : E_UNEXPECTED;
}

SetValueDirect()

STDMETHODIMP CRegistrySource::SetValueDirect ( PCWSTR  keyName, PCWSTR  valueName, DWORD  dwType, DWORD  cbData, LPCVOID  pvData  )

override

Definition at line 458 of file querysrc.cpp.

{
    LSTATUS error = SHSetValueW(m_hKey, keyName, valueName, dwType, pvData, cbData);
    if (error)
        return HRESULT_FROM_WIN32(error);
    return S_OK;
}

STDMETHODIMP_() [1/2]

CRegistrySource::STDMETHODIMP_ ( ULONG  )

override

STDMETHODIMP_() [2/2]

CRegistrySource::STDMETHODIMP_ ( ULONG  )

override

Member Data Documentation

m_cRefs

LONG CRegistrySource::m_cRefs = 1

private

Definition at line 95 of file querysrc.cpp.

m_hKey

HKEY CRegistrySource::m_hKey = NULL

private

Definition at line 96 of file querysrc.cpp.

Referenced by EnumSources(), EnumValues(), GetKey(), Init(), OpenSource(), QueryValueDirect(), QueryValueDword(), QueryValueExists(), QueryValueString(), SetKey(), SetValueDirect(), and ~CRegistrySource().

---

The documentation for this class was generated from the following file:

• dll/win32/shlwapi/querysrc.cpp