CAssocClientElement Class Reference

Inheritance diagram for CAssocClientElement:

Collaboration diagram for CAssocClientElement:

Public Member Functions
STDMETHODIMP	QueryString (ASSOCQUERY query, PCWSTR key, PWSTR *ppszValue) override
STDMETHODIMP	GetClassID (CLSID *pClassID) override
HRESULT	_InitSource () override
HRESULT	_FixNetscapeRegistration ()
BOOL	_CreateRepairedNetscapeRegistration (HKEY hKey)
HRESULT	_InitSourceFromKey (HKEY hKey, PCWSTR lpSubKey, DWORD dwFlags)
Public Member Functions inherited from CAssocShellElement
	~CAssocShellElement ()
STDMETHODIMP	QueryInterface (REFIID riid, PVOID *ppv) override
	STDMETHODIMP_ (ULONG) AddRef() override
	STDMETHODIMP_ (ULONG) Release() override
STDMETHODIMP	QueryString (ASSOCQUERY query, PCWSTR key, PWSTR *ppszValue) override
STDMETHODIMP	QueryDword (ASSOCQUERY query, PCWSTR key, DWORD *pdwValue) override
STDMETHODIMP	QueryExists (ASSOCQUERY query, PCWSTR key) override
STDMETHODIMP	QueryDirect (ASSOCQUERY query, PCWSTR key, FLAGGED_BYTE_BLOB **ppBlob) override
STDMETHODIMP	QueryObject (ASSOCQUERY query, PCWSTR key, REFIID riid, PVOID *ppvObj) override
STDMETHODIMP	GetClassID (CLSID *pClassID) override
STDMETHODIMP	SetString (PCWSTR psz) override
STDMETHODIMP	GetString (PWSTR *ppsz) override
virtual BOOL	_UseEnumForDefaultVerb ()
virtual HRESULT	_InitSource ()
virtual BOOL	_IsAppSource ()
virtual HRESULT	_GetVerbDelegate (PCWSTR pszSrc, IAssociationElementOld **ppElement)
Public Member Functions inherited from CAssocElement
virtual	~CAssocElement ()
STDMETHODIMP	QueryInterface (REFIID riid, PVOID *ppv) override
	STDMETHODIMP_ (ULONG) AddRef() override
	STDMETHODIMP_ (ULONG) Release() override
STDMETHODIMP	SetSource (IQuerySourceOld *pSource) override
STDMETHODIMP	GetSource (REFIID riid, PVOID *ppSource) override
STDMETHODIMP	QueryString (ASSOCQUERY query, PCWSTR key, PWSTR *ppszValue) override
STDMETHODIMP	QueryDword (ASSOCQUERY query, PCWSTR key, DWORD *pdwValue) override
STDMETHODIMP	QueryExists (ASSOCQUERY query, PCWSTR key) override
STDMETHODIMP	QueryDirect (ASSOCQUERY query, PCWSTR key, FLAGGED_BYTE_BLOB **ppBlob) override
STDMETHODIMP	QueryObject (ASSOCQUERY query, PCWSTR key, REFIID riid, PVOID *ppvObj) override

Additional Inherited Members
Protected Member Functions inherited from CAssocShellElement
UINT	_GetQueryKeyVal (const QUERYKEYVAL **ppItems) override
HRESULT	_DefaultVerbSource (IQuerySourceOld **ppSource)
Protected Member Functions inherited from CAssocElement
HRESULT	_QueryKeyValAny (QUERY_CALLBACK callback, const QUERYKEYVAL *pItems, UINT cItems, IQuerySourceOld *pQS, ASSOCQUERY query, PCWSTR valueName, PVOID pValue)
HRESULT	_QuerySourceAny (QUERY_CALLBACK callback, IQuerySourceOld *pSource, DWORD dwFlags, ASSOCQUERY query, PCWSTR valueName, PVOID pValue)
virtual UINT	_GetQueryKeyVal (const QUERYKEYVAL **ppItems)
Protected Attributes inherited from CAssocShellElement
PWSTR	m_pszName = NULL
WCHAR	m_szBuff [64] = L""
Protected Attributes inherited from CAssocElement
LONG	m_cRefs = 1
IQuerySourceOld *	m_pSource = NULL

Detailed Description

Definition at line 604 of file elements.cpp.

Member Function Documentation

_CreateRepairedNetscapeRegistration()

BOOL CAssocClientElement::_CreateRepairedNetscapeRegistration

(

HKEY

hKey

)

Definition at line 1533 of file elements.cpp.

{
    HKEY hkResult;
    LSTATUS error = RegOpenKeyExW(HKEY_LOCAL_MACHINE,
                                  L"Software\\Clients\\Mail\\Netscape Messenger",
                                  0, KEY_READ, &hkResult);
    if (error)
        return FALSE;
 
    WCHAR szIconFile[MAX_PATH];
    error = _RegQueryString(hkResult, L"Protocols\\mailto\\DefaultIcon",
                            szIconFile, _countof(szIconFile));
    if (error == ERROR_SUCCESS)
    {
        PathParseIconLocationW(szIconFile);
        StringCchCatW(szIconFile, _countof(szIconFile), L",-1349");
        _RegSetVolatileString(hKey, L"DefaultIcon", szIconFile);
    }
 
    BOOL ret = FALSE;
    PWSTR lpString2 = NULL;
    if (_RegQueryString(hkResult, 0, szIconFile, _countof(szIconFile)) == ERROR_SUCCESS &&
        _RegSetVolatileString(hKey, 0, szIconFile) == ERROR_SUCCESS &&
        _RegQueryString(hkResult, L"Protocols\\mailto\\shell\\open\\command",
                        szIconFile, _countof(szIconFile)) == ERROR_SUCCESS &&
        SUCCEEDED(_ExeFromCmd(szIconFile, &lpString2)))
    {
        StringCchCopyW(szIconFile, _countof(szIconFile), lpString2);
        CoTaskMemFree(lpString2);
        PathQuoteSpacesW(szIconFile);
        StringCchCatW(szIconFile, _countof(szIconFile), L" -mail");
        if (_RegSetVolatileString(hKey, L"shell\\open\\command", szIconFile) == ERROR_SUCCESS)
            ret = TRUE;
    }
 
    RegCloseKey(hkResult);
    return ret;
}

Referenced by _FixNetscapeRegistration().

_FixNetscapeRegistration()

HRESULT CAssocClientElement::_FixNetscapeRegistration

(

)

Definition at line 1492 of file elements.cpp.

{
    HKEY hKeyParent;
    LSTATUS error = RegCreateKeyExW(HKEY_CURRENT_USER, L"Software\\Clients\\Mail", 0, NULL, 0,
                                    KEY_ALL_ACCESS, NULL, &hKeyParent, NULL);
    if (error)
        return E_FAIL;
 
    DWORD dwDisposition;
    HKEY hKey;
    error = RegCreateKeyExW(hKeyParent, L"Netscape Messenger", 0, NULL, REG_OPTION_VOLATILE,
                            KEY_ALL_ACCESS, NULL, &hKey, &dwDisposition);
    if (error)
    {
        SHDeleteKeyW(hKeyParent, L"Netscape Messenger");
        RegCloseKey(hKeyParent);
        return E_FAIL;
    }
 
    HRESULT hr = E_FAIL;
    if (dwDisposition == REG_OPENED_EXISTING_KEY || _CreateRepairedNetscapeRegistration(hKey))
    {
        IQuerySourceOld* pNewSource = NULL;
        hr = _QuerySourceCreateFromKey(hKey, NULL, NULL, &pNewSource);
        if (SUCCEEDED(hr))
        {
            if (m_pSource)
                m_pSource->Release();
            m_pSource = pNewSource;
        }
    }
 
    RegCloseKey(hKey);
 
    if (FAILED_UNEXPECTEDLY(hr))
        SHDeleteKeyW(hKeyParent, L"Netscape Messenger");
 
    RegCloseKey(hKeyParent);
    return hr;
}

Referenced by _InitSourceFromKey().

_InitSource()

HRESULT CAssocClientElement::_InitSource ( )

overridevirtual

Reimplemented from CAssocShellElement.

Definition at line 1481 of file elements.cpp.

{
    WCHAR szBuff[MAX_PATH];
    StringCchPrintfW(szBuff, _countof(szBuff), L"SOFTWARE\\Clients\\%s", m_pszName);
 
    HRESULT hr = _InitSourceFromKey(HKEY_CURRENT_USER, szBuff, 0);
    if (SUCCEEDED(hr))
        return hr;
    return _InitSourceFromKey(HKEY_LOCAL_MACHINE, szBuff, 0);
}

_InitSourceFromKey()

HRESULT CAssocClientElement::_InitSourceFromKey	(	HKEY	hKey,
		PCWSTR	lpSubKey,
		DWORD	dwFlags
	)

Definition at line 1572 of file elements.cpp.

{
    HKEY hKey2;
    LSTATUS error = RegOpenKeyExW(hKey, lpSubKey, 0, KEY_READ | dwFlags, &hKey2);
    if (error)
        return E_INVALIDARG;
 
    HRESULT hr = E_INVALIDARG;
    HKEY hkeySrc = NULL;
    WCHAR szData[80];
    DWORD dwType, cbData = sizeof(szData);
    error = RegQueryValueExW(hKey2, NULL, NULL, &dwType, (PBYTE)szData, &cbData);
    if (error == ERROR_SUCCESS && dwType == REG_SZ && szData[0] &&
        RegOpenKeyExW(HKEY_LOCAL_MACHINE, lpSubKey, 0, KEY_READ, &hkeySrc) == ERROR_SUCCESS)
    {
        hr = _QuerySourceCreateFromKey2(hkeySrc, NULL, szData, &m_pSource);
        RegCloseKey(hkeySrc);
    }
 
    RegCloseKey(hKey2);
 
    if (FAILED_UNEXPECTEDLY(hr) ||
        lstrcmpiW(m_pszName, L"mail") != 0 ||
        lstrcmpiW(szData, L"Netscape Messenger") != 0 ||
        SUCCEEDED(QueryExists(ASSOCQUERY_COMMAND, L"open")))
    {
        return hr;
    }
 
    return _FixNetscapeRegistration();
}

Referenced by _InitSource().

GetClassID()

STDMETHODIMP CAssocClientElement::GetClassID ( CLSID *  pClassID )

override

Definition at line 1475 of file elements.cpp.

{
    *pClassID = CLSID_AssocClientElement;
    return S_OK;
}

QueryString()

STDMETHODIMP CAssocClientElement::QueryString ( ASSOCQUERY  query, PCWSTR  key, PWSTR *  ppszValue  )

override

Definition at line 1444 of file elements.cpp.

{
    HRESULT hr;
 
    if (query == ASSOCQUERY_DEFAULTICON)
    {
        hr = CAssocElement::QueryString(ASSOCQUERY_DEFAULTICON, key, ppszValue);
        if (SUCCEEDED(hr))
            return hr;
 
        IAssociationElementOld* pDelegate;
        HRESULT hr = _GetVerbDelegate(key, &pDelegate);
        if (FAILED_UNEXPECTEDLY(hr))
            return hr;
 
        hr = pDelegate->QueryString(ASSOCQUERY_DELEGATE, L"open", ppszValue);
        pDelegate->Release();
        return hr;
    }
 
    if (query == ASSOCQUERY_FRIENDLYAPPNAME)
    {
        hr = CAssocElement::QueryString(ASSOCQUERY_LOCALIZEDSTRING, L"LocalizedString", ppszValue);
        if (SUCCEEDED(hr))
            return hr;
        return CAssocElement::QueryString((ASSOCQUERY_SDED | ASSOCQUERY_EXTRA_NON_VERB), NULL, ppszValue);
    }
 
    return CAssocShellElement::QueryString(query, key, ppszValue);
}

---

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

• dll/win32/shell32/evalcmd/elements.cpp