CQueryAssociations.cpp File Reference

#include "precomp.h"
#include <shlwapi_undoc.h>
#include "evalcmd/elements.h"
#include <evalcmd.h>

Include dependency graph for CQueryAssociations.cpp:

Go to the source code of this file.

Classes
struct	ClassesRegKey

Functions
	WINE_DEFAULT_DEBUG_CHANNEL (shell)
static HRESULT	SHRegDuplicateHKey (HKEY hKey, HKEY *phKey)
static HRESULT	SH32_AssocCreateElement (_In_ REFCLSID rclsid, _In_ REFIID riid, _Outptr_ PVOID *ppv)
static HRESULT	AssocCreateElementInternal (REFCLSID rclsid, IQuerySource *pQS, REFIID riid, PVOID *ppv)
static HRESULT	AssocCreateElementOnKeyInternal (_In_ REFCLSID rclsid, _In_ HKEY hKey, _In_opt_ PCWSTR pszSubKey, _In_ REFIID riid, _Outptr_ PVOID *ppv, _In_ BYTE Major)
static HRESULT	SH32_AssocCreateElementOnKey (_In_ REFCLSID rclsid, _In_ HKEY hKey, _In_opt_ PCWSTR pszSubKey, _In_ REFIID riid, _Outptr_ PVOID *ppv)
static HRESULT	SH32_QueryAssocElementString (_In_ REFCLSID rclsid, _In_ UINT Query, _In_ HKEY hKey, _In_opt_ PCWSTR pszSubKey, _In_ PCWSTR pszExtra, _Out_ PWSTR *ppszValue)
template<class T >
static HRESULT	GetKeyFromObjectWithQuerySource (T &Obj, HKEY *phKey)
EXTERN_C HRESULT	SHELL32_AssocGetFSDirectoryDescription (PWSTR Buf, UINT cchBuf)
static HRESULT	GetExtensionDefaultDescription (PCWSTR DotExt, PWSTR Buf, UINT cchBuf)
static HRESULT	SHELL32_AssocGetExtensionDescription (PCWSTR DotExt, PWSTR Buf, UINT cchBuf)
EXTERN_C HRESULT	SHELL32_AssocGetFileDescription (PCWSTR Name, PWSTR Buf, UINT cchBuf)
static HRESULT	OpenShellVerbKey (HKEY hClass, LPCWSTR pszVerb, HKEY *phKey)
static HRESULT	OpenClassWithShellExecKey (ASSOCF fA, HKEY hClass, LPCWSTR pszVerb, HKEY *phKey)
static HRESULT	GetClassKeyName (ASSOCF fA, PCWSTR pszSource, HKEY hSource, LPWSTR pszClass, UINT cchMax)
static HRESULT	OpenClassKey (ASSOCF fA, PCWSTR pszSource, HKEY hCR, HKEY hSource, HKEY *phKey)
static HRESULT	OpenClassKey (ASSOCF fA, const WCHAR *const pszSource, HKEY *phKey)

Function Documentation

AssocCreateElementInternal()

static HRESULT AssocCreateElementInternal ( REFCLSID  rclsid, IQuerySource *  pQS, REFIID  riid, PVOID *  ppv  )

static

Definition at line 34 of file CQueryAssociations.cpp.

{
    HRESULT hr;
    C_ASSERT(&IID_IObjectWithQuerySource == &IID_IObjectWithQuerySourceOld);
    CComPtr<IObjectWithQuerySource> pOWQS;
    if (FAILED(hr = SH32_AssocCreateElement(rclsid, IID_PPV_ARG(IObjectWithQuerySource, &pOWQS))))
        return hr;
    return (SUCCEEDED(hr = pOWQS->SetSource(pQS))) ? pOWQS->QueryInterface(riid, ppv) : hr;
}

Referenced by AssocCreateElementOnKeyInternal().

AssocCreateElementOnKeyInternal()

static HRESULT AssocCreateElementOnKeyInternal ( _In_ REFCLSID  rclsid, _In_ HKEY  hKey, _In_opt_ PCWSTR  pszSubKey, _In_ REFIID  riid, _Outptr_ PVOID *  ppv, _In_ BYTE  Major  )

static

Definition at line 44 of file CQueryAssociations.cpp.

{
    CComPtr<IUnknown> pQSO;
    REFIID riidQS = Major > 5 ? IID_IQuerySource : IID_IQuerySourceOld;
    HRESULT hr = QuerySourceCreateFromKey(hKey, pszSubKey, false, riidQS, (void**)&pQSO);
    if (FAILED(hr))
        return hr;
    return AssocCreateElementInternal(rclsid, (IQuerySource*)static_cast<IUnknown*>(pQSO), riid, ppv);
}

Referenced by SH32_AssocCreateElementOnKey().

GetClassKeyName()

static HRESULT GetClassKeyName ( ASSOCF  fA, PCWSTR  pszSource, HKEY  hSource, LPWSTR  pszClass, UINT  cchMax  )

inlinestatic

Definition at line 198 of file CQueryAssociations.cpp.

{
    PCWSTR pszSubKey = *pszSource == L'{' ? L"ProgID" : NULL;
    return SHELL_RegGetString(hSource, pszSubKey, NULL, pszClass, cchMax);
}

Referenced by OpenClassKey().

GetExtensionDefaultDescription()

static HRESULT GetExtensionDefaultDescription ( PCWSTR  DotExt, PWSTR  Buf, UINT  cchBuf  )

static

Definition at line 108 of file CQueryAssociations.cpp.

{
    static WCHAR fmt[33] = {};
    if (!*fmt)
        LoadStringW(shell32_hInstance, IDS_ANY_FILE, fmt, _countof(fmt));
    return StringCchPrintfW(Buf, cchBuf, fmt, DotExt);
}

Referenced by SHELL32_AssocGetExtensionDescription().

GetKeyFromObjectWithQuerySource()

template<class T >

static HRESULT GetKeyFromObjectWithQuerySource ( T &  Obj, HKEY *  phKey  )

static

Definition at line 82 of file CQueryAssociations.cpp.

{
    HRESULT hr;
    C_ASSERT(&IID_IObjectWithQuerySource == &IID_IObjectWithQuerySourceOld);
    CComPtr<IObjectWithQuerySource> pOWQS;
    if (FAILED(hr = Obj.QueryInterface(IID_PPV_ARG(IObjectWithQuerySource, &pOWQS))))
        return hr;
    CComPtr<IObjectWithRegistryKeyOld> pOWRKO;
    CComPtr<IObjectWithRegistryKey> pOWRK;
    if (SUCCEEDED(hr = pOWQS->GetSource(IID_PPV_ARG(IObjectWithRegistryKeyOld, &pOWRKO))))
        hr = pOWRKO->GetKey(phKey);
#if DLL_EXPORT_VERSION >= _WIN32_WINNT_VISTA
    else if (SUCCEEDED(hr = pOWQS->GetSource(IID_PPV_ARG(IObjectWithRegistryKey, &pOWRK))))
        hr = pOWRK->GetKey(MAXIMUM_ALLOWED, phKey);
#endif
    return hr;
}

Referenced by CQueryAssociations::GetKey().

OpenClassKey() [1/2]

static HRESULT OpenClassKey ( ASSOCF  fA, const WCHAR *const  pszSource, HKEY *  phKey  )

static

Definition at line 241 of file CQueryAssociations.cpp.

{
    // TODO: Handle the FileExts key and ASSOCF_NOUSERSETTINGS?
    PCWSTR pszSourcePath = pszSource;
    WCHAR buf[100];
    if (*pszSource == L'{')
    {
        HRESULT hr = StringCchPrintfW(buf, _countof(buf), L"CLSID\\%s", pszSource);
        if (FAILED(hr))
            return hr;
        pszSourcePath = buf;
    }
    ClassesRegKey cr;
    if (HRESULT hr = cr.Initialize(fA, pszSource))
        return hr;
    HKEY hSourceKey;
    LONG err = RegOpenKeyExW(cr.hKey, pszSourcePath, 0, KEY_READ, &hSourceKey);
    if (err)
        return HResultFromWin32(err);
    HRESULT hr = OpenClassKey(fA, pszSource, cr.hKey, hSourceKey, phKey);
    RegCloseKey(hSourceKey);
    return hr;
}

OpenClassKey() [2/2]

static HRESULT OpenClassKey ( ASSOCF  fA, PCWSTR  pszSource, HKEY  hCR, HKEY  hSource, HKEY *  phKey  )

static

Definition at line 204 of file CQueryAssociations.cpp.

{
    WCHAR szBuf[MAX_PATH];
    HRESULT hr = E_FAIL;
    if (*pszSource == L'.' || (*pszSource == L'{' && !(fA & ASSOCF_INIT_NOREMAPCLSID)))
    {
        hr = GetClassKeyName(fA, pszSource, hSource, szBuf, _countof(szBuf));
        if (FAILED(hr) && *pszSource == L'.')
            return hr; // .ext can only have its ProgId in another location
    }
 
    HKEY hClassKey;
    if (SUCCEEDED(hr))
        hr = HResultFromWin32(RegOpenKeyExW(hCR, szBuf, 0, KEY_READ, &hClassKey));
    else
        hr = SHRegDuplicateHKey(hSource, &hClassKey);
 
    if (FAILED(hr))
        return hr;
 
    // Check for CurVer redirection 
    if (SUCCEEDED(SHELL_RegGetString(hClassKey, L"CurVer", NULL, szBuf, _countof(szBuf))))
    {
        if (RegOpenKeyExW(hCR, szBuf, 0, KEY_READ, phKey) == ERROR_SUCCESS)
        {
            if (RegKeyExists(*phKey, L"shell")) // Does it look like a valid ProgId?
            {
                RegCloseKey(hClassKey);
                return S_OK;
            }
            RegCloseKey(*phKey);
        }
    }
    *phKey = hClassKey;
    return hr;
}

Referenced by CQueryAssociations::Init(), and OpenClassKey().

OpenClassWithShellExecKey()

static HRESULT OpenClassWithShellExecKey ( ASSOCF  fA, HKEY  hClass, LPCWSTR  pszVerb, HKEY *  phKey  )

static

Definition at line 186 of file CQueryAssociations.cpp.

{
    if (fA & ASSOCF_VERIFY)
    {
        HRESULT hr = OpenShellVerbKey(hClass, pszVerb, phKey);
        if (FAILED(hr))
            return HRESULT_FROM_WIN32(ERROR_NO_ASSOCIATION);
        RegCloseKey(*phKey);
    }
    return SHRegDuplicateHKey(hClass, phKey);
}

Referenced by CQueryAssociations::GetKey().

OpenShellVerbKey()

static HRESULT OpenShellVerbKey ( HKEY  hClass, LPCWSTR  pszVerb, HKEY *  phKey  )

static

Definition at line 173 of file CQueryAssociations.cpp.

{
    const SIZE_T cchPrefix = sizeof("shell\\") - 1;
    WCHAR szPath[cchPrefix + MAX_PATH];
    if (FAILED(StringCchPrintfW(szPath, _countof(szPath), L"shell\\%s", pszVerb ? pszVerb : L"")))
        return E_FAIL;
 
    if (!pszVerb && !HCR_GetDefaultVerbW(hClass, NULL, szPath + cchPrefix, _countof(szPath) - cchPrefix))
        return HRESULT_FROM_WIN32(ERROR_NO_ASSOCIATION);
 
    return HResultFromWin32(RegOpenKeyExW(hClass, szPath, 0, KEY_READ, phKey));
}

Referenced by OpenClassWithShellExecKey().

SH32_AssocCreateElement()

static HRESULT SH32_AssocCreateElement ( _In_ REFCLSID  rclsid, _In_ REFIID  riid, _Outptr_ PVOID *  ppv  )

inlinestatic

Definition at line 25 of file CQueryAssociations.cpp.

{
#if DLL_EXPORT_VERSION >= _WIN32_WINNT_VISTA
    return AssocCreateElement(rclsid, riid, ppv);
#else
    return AssocCreate(rclsid, riid, ppv);
#endif
}

Referenced by AssocCreateElementInternal(), and CQueryAssociations::GetKey().

SH32_AssocCreateElementOnKey()

static HRESULT SH32_AssocCreateElementOnKey ( _In_ REFCLSID  rclsid, _In_ HKEY  hKey, _In_opt_ PCWSTR  pszSubKey, _In_ REFIID  riid, _Outptr_ PVOID *  ppv  )

static

Definition at line 55 of file CQueryAssociations.cpp.

{
#if DLL_EXPORT_VERSION >= _WIN32_WINNT_VISTA
    HRESULT hr = AssocCreateElementOnKeyInternal(rclsid, hKey, pszSubKey, riid, ppv, 6);
    if (SUCCEEDED(hr))
        return hr;
#endif
    return AssocCreateElementOnKeyInternal(rclsid, hKey, pszSubKey, riid, ppv, 5);
}

Referenced by SH32_QueryAssocElementString().

SH32_QueryAssocElementString()

static HRESULT SH32_QueryAssocElementString ( _In_ REFCLSID  rclsid, _In_ UINT  Query, _In_ HKEY  hKey, _In_opt_ PCWSTR  pszSubKey, _In_ PCWSTR  pszExtra, _Out_ PWSTR *  ppszValue  )

static

Definition at line 66 of file CQueryAssociations.cpp.

{
#if SHELL32_CAssocElement_Version >= _WIN32_WINNT_VISTA
    CComPtr<IAssociationElement> pAE;
    REFIID riid = IID_IAssociationElement;
#else
    CComPtr<IAssociationElementOld> pAE;
    REFIID riid = IID_IAssociationElementOld;
#endif
    HRESULT hr = SH32_AssocCreateElementOnKey(rclsid, hKey, pszSubKey, riid, (void**)&pAE);
    if (FAILED(hr))
        return hr;
    return pAE->QueryString(Query, pszExtra, ppszValue);
}

Referenced by CQueryAssociations::GetCommand().

SHELL32_AssocGetExtensionDescription()

static HRESULT SHELL32_AssocGetExtensionDescription ( PCWSTR  DotExt, PWSTR  Buf, UINT  cchBuf  )

static

Definition at line 116 of file CQueryAssociations.cpp.

{
    HRESULT hr;
    if (!DotExt[0] || (!DotExt[1] && DotExt[0] == '.'))
    {
        if (SUCCEEDED(hr = GetExtensionDefaultDescription(L"", Buf, cchBuf)))
            StrTrimW(Buf, L" -"); // Remove the empty %s so we are left with "File"
        return hr;
    }
    HKEY hKey;
    if (SUCCEEDED(hr = HCR_GetProgIdKeyOfExtension(DotExt, &hKey, TRUE)))
    {
        DWORD err = RegLoadMUIStringW(hKey, L"FriendlyTypeName", Buf, cchBuf, NULL, 0, NULL);
        if (err && hr == S_OK) // ProgId default value fallback (but not if we only have a .ext key)
        {
            DWORD cb = cchBuf * sizeof(*Buf);
            err = RegGetValueW(hKey, NULL, NULL, RRF_RT_REG_SZ, NULL, Buf, &cb);
        }
        RegCloseKey(hKey);
        if (!err)
            return err;
    }
    // No information in the registry, default to "UPPERCASEEXT File"
    WCHAR ext[MAX_PATH + 33];
    if (LCMapStringW(LOCALE_USER_DEFAULT, LCMAP_UPPERCASE, ++DotExt, -1, ext, _countof(ext)))
        DotExt = ext;
    return GetExtensionDefaultDescription(DotExt, Buf, cchBuf);
}

Referenced by SHELL32_AssocGetFileDescription().

SHELL32_AssocGetFileDescription()

EXTERN_C HRESULT SHELL32_AssocGetFileDescription	(	PCWSTR	Name,
		PWSTR	Buf,
		UINT	cchBuf
	)

Definition at line 145 of file CQueryAssociations.cpp.

{
    return SHELL32_AssocGetExtensionDescription(PathFindExtensionW(Name), Buf, cchBuf);
}

Referenced by GetItemDescription(), and SHGetFileInfoW().

SHELL32_AssocGetFSDirectoryDescription()

EXTERN_C HRESULT SHELL32_AssocGetFSDirectoryDescription	(	PWSTR	Buf,
		UINT	cchBuf
	)

Definition at line 100 of file CQueryAssociations.cpp.

{
    static WCHAR cache[33] = {};
    if (!*cache)
        LoadStringW(shell32_hInstance, IDS_DIRECTORY, cache, _countof(cache));
    return StringCchCopyW(Buf, cchBuf, cache);
}

Referenced by GetItemDescription(), and SHGetFileInfoW().

SHRegDuplicateHKey()

static HRESULT SHRegDuplicateHKey ( HKEY  hKey, HKEY *  phKey  )

inlinestatic

Definition at line 19 of file CQueryAssociations.cpp.

{
    hKey = *phKey = ::SHRegDuplicateHKey(hKey);
    return hKey ? S_OK : HResultFromWin32(GetLastError());
}

Referenced by OpenClassKey(), OpenClassWithShellExecKey(), and SHRegDuplicateHKey().

WINE_DEFAULT_DEBUG_CHANNEL()

WINE_DEFAULT_DEBUG_CHANNEL

(

shell

)