/* * IQueryAssociations object and helper functions * * Copyright 2002 Jon Griffiths * * This library is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 of the License, or (at your option) any later version. * * This library is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with this library; if not, write to the Free Software * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA */ #include #include "windef.h" #include "winbase.h" #include "winnls.h" #include "winreg.h" #include "objbase.h" #include "shlguid.h" #include "shlwapi.h" #include "wine/debug.h" WINE_DEFAULT_DEBUG_CHANNEL(shell); /************************************************************************** * IQueryAssociations {SHLWAPI} * * DESCRIPTION * This object provides a layer of abstraction over the system registry in * order to simplify the process of parsing associations between files. * Associations in this context means the registry entries that link (for * example) the extension of a file with its description, list of * applications to open the file with, and actions that can be performed on it * (the shell displays such information in the context menu of explorer * when you right-click on a file). * * HELPERS * You can use this object transparently by calling the helper functions * AssocQueryKeyA(), AssocQueryStringA() and AssocQueryStringByKeyA(). These * create an IQueryAssociations object, perform the requested actions * and then dispose of the object. Alternatively, you can create an instance * of the object using AssocCreate() and call the following methods on it: * * METHODS */ /* Default IQueryAssociations::Init() flags */ #define SHLWAPI_DEF_ASSOCF (ASSOCF_INIT_BYEXENAME|ASSOCF_INIT_DEFAULTTOSTAR| \ ASSOCF_INIT_DEFAULTTOFOLDER) typedef struct { const IQueryAssociationsVtbl *lpVtbl; LONG ref; HKEY hkeySource; HKEY hkeyProgID; } IQueryAssociationsImpl; static const IQueryAssociationsVtbl IQueryAssociations_vtbl; /************************************************************************** * IQueryAssociations_Constructor [internal] * * Construct a new IQueryAssociations object. */ static IQueryAssociations* IQueryAssociations_Constructor(void) { IQueryAssociationsImpl* iface; iface = HeapAlloc(GetProcessHeap(),0,sizeof(IQueryAssociationsImpl)); iface->lpVtbl = &IQueryAssociations_vtbl; iface->ref = 1; iface->hkeySource = NULL; iface->hkeyProgID = NULL; TRACE("Returning IQueryAssociations* %p\n", iface); return (IQueryAssociations*)iface; } /************************************************************************* * SHLWAPI_ParamAToW * * Internal helper function: Convert ASCII parameter to Unicode. */ static BOOL SHLWAPI_ParamAToW(LPCSTR lpszParam, LPWSTR lpszBuff, DWORD dwLen, LPWSTR* lpszOut) { if (lpszParam) { DWORD dwStrLen = MultiByteToWideChar(CP_ACP, 0, lpszParam, -1, NULL, 0); if (dwStrLen < dwLen) { *lpszOut = lpszBuff; /* Use Buffer, it is big enough */ } else { /* Create a new buffer big enough for the string */ *lpszOut = HeapAlloc(GetProcessHeap(), 0, dwStrLen * sizeof(WCHAR)); if (!*lpszOut) return FALSE; } MultiByteToWideChar(CP_ACP, 0, lpszParam, -1, *lpszOut, dwStrLen); } else *lpszOut = NULL; return TRUE; } /************************************************************************* * AssocCreate [SHLWAPI.@] * * Create a new IQueryAssociations object. * * PARAMS * clsid [I] CLSID of object * refiid [I] REFIID of interface * lpInterface [O] Destination for the created IQueryAssociations object * * RETURNS * Success: S_OK. lpInterface contains the new object. * Failure: An HRESULT error code indicating the error. * * NOTES * refiid must be equal to IID_IQueryAssociations, or this function will fail. */ HRESULT WINAPI AssocCreate(CLSID clsid, REFIID refiid, void **lpInterface) { HRESULT hRet; IQueryAssociations* lpAssoc; TRACE("(%s,%s,%p)\n", debugstr_guid(&clsid), debugstr_guid(refiid), lpInterface); if (!lpInterface) return E_INVALIDARG; *(DWORD*)lpInterface = 0; if (!IsEqualGUID(&clsid, &IID_IQueryAssociations)) return E_NOTIMPL; lpAssoc = IQueryAssociations_Constructor(); if (!lpAssoc) return E_OUTOFMEMORY; hRet = IQueryAssociations_QueryInterface(lpAssoc, refiid, lpInterface); IQueryAssociations_Release(lpAssoc); return hRet; } /************************************************************************* * AssocQueryKeyW [SHLWAPI.@] * * See AssocQueryKeyA. */ HRESULT WINAPI AssocQueryKeyW(ASSOCF cfFlags, ASSOCKEY assockey, LPCWSTR pszAssoc, LPCWSTR pszExtra, HKEY *phkeyOut) { HRESULT hRet; IQueryAssociations* lpAssoc; TRACE("(0x%8x,0x%8x,%s,%s,%p)\n", cfFlags, assockey, debugstr_w(pszAssoc), debugstr_w(pszExtra), phkeyOut); lpAssoc = IQueryAssociations_Constructor(); if (!lpAssoc) return E_OUTOFMEMORY; cfFlags &= SHLWAPI_DEF_ASSOCF; hRet = IQueryAssociations_Init(lpAssoc, cfFlags, pszAssoc, NULL, NULL); if (SUCCEEDED(hRet)) hRet = IQueryAssociations_GetKey(lpAssoc, cfFlags, assockey, pszExtra, phkeyOut); IQueryAssociations_Release(lpAssoc); return hRet; } /************************************************************************* * AssocQueryKeyA [SHLWAPI.@] * * Get a file association key from the registry. * * PARAMS * cfFlags [I] ASSOCF_ flags from "shlwapi.h" * assockey [I] Type of key to get * pszAssoc [I] Key name to search below * pszExtra [I] Extra information about the key location * phkeyOut [O] Destination for the association key * * RETURNS * Success: S_OK. phkeyOut contains the key. * Failure: An HRESULT error code indicating the error. */ HRESULT WINAPI AssocQueryKeyA(ASSOCF cfFlags, ASSOCKEY assockey, LPCSTR pszAssoc, LPCSTR pszExtra, HKEY *phkeyOut) { WCHAR szAssocW[MAX_PATH], *lpszAssocW = NULL; WCHAR szExtraW[MAX_PATH], *lpszExtraW = NULL; HRESULT hRet = E_OUTOFMEMORY; TRACE("(0x%8x,0x%8x,%s,%s,%p)\n", cfFlags, assockey, debugstr_a(pszAssoc), debugstr_a(pszExtra), phkeyOut); if (SHLWAPI_ParamAToW(pszAssoc, szAssocW, MAX_PATH, &lpszAssocW) && SHLWAPI_ParamAToW(pszExtra, szExtraW, MAX_PATH, &lpszExtraW)) { hRet = AssocQueryKeyW(cfFlags, assockey, lpszAssocW, lpszExtraW, phkeyOut); } if (lpszAssocW != szAssocW) HeapFree(GetProcessHeap(), 0, lpszAssocW); if (lpszExtraW != szExtraW) HeapFree(GetProcessHeap(), 0, lpszExtraW); return hRet; } /************************************************************************* * AssocQueryStringW [SHLWAPI.@] * * See AssocQueryStringA. */ HRESULT WINAPI AssocQueryStringW(ASSOCF cfFlags, ASSOCSTR str, LPCWSTR pszAssoc, LPCWSTR pszExtra, LPWSTR pszOut, DWORD *pcchOut) { HRESULT hRet; IQueryAssociations* lpAssoc; TRACE("(0x%8x,0x%8x,%s,%s,%p,%p)\n", cfFlags, str, debugstr_w(pszAssoc), debugstr_w(pszExtra), pszOut, pcchOut); if (!pcchOut) return E_INVALIDARG; lpAssoc = IQueryAssociations_Constructor(); if (!lpAssoc) return E_OUTOFMEMORY; hRet = IQueryAssociations_Init(lpAssoc, cfFlags & SHLWAPI_DEF_ASSOCF, pszAssoc, NULL, NULL); if (SUCCEEDED(hRet)) hRet = IQueryAssociations_GetString(lpAssoc, cfFlags, str, pszExtra, pszOut, pcchOut); IQueryAssociations_Release(lpAssoc); return hRet; } /************************************************************************* * AssocQueryStringA [SHLWAPI.@] * * Get a file association string from the registry. * * PARAMS * cfFlags [I] ASSOCF_ flags from "shlwapi.h" * str [I] Type of string to get (ASSOCSTR enum from "shlwapi.h") * pszAssoc [I] Key name to search below * pszExtra [I] Extra information about the string location * pszOut [O] Destination for the association string * pcchOut [O] Length of pszOut * * RETURNS * Success: S_OK. pszOut contains the string, pcchOut contains its length. * Failure: An HRESULT error code indicating the error. */ HRESULT WINAPI AssocQueryStringA(ASSOCF cfFlags, ASSOCSTR str, LPCSTR pszAssoc, LPCSTR pszExtra, LPSTR pszOut, DWORD *pcchOut) { WCHAR szAssocW[MAX_PATH], *lpszAssocW = NULL; WCHAR szExtraW[MAX_PATH], *lpszExtraW = NULL; HRESULT hRet = E_OUTOFMEMORY; TRACE("(0x%8x,0x%8x,%s,%s,%p,%p)\n", cfFlags, str, debugstr_a(pszAssoc), debugstr_a(pszExtra), pszOut, pcchOut); if (!pcchOut) hRet = E_INVALIDARG; else if (SHLWAPI_ParamAToW(pszAssoc, szAssocW, MAX_PATH, &lpszAssocW) && SHLWAPI_ParamAToW(pszExtra, szExtraW, MAX_PATH, &lpszExtraW)) { WCHAR szReturnW[MAX_PATH], *lpszReturnW = szReturnW; DWORD dwLenOut = *pcchOut; if (dwLenOut >= MAX_PATH) lpszReturnW = HeapAlloc(GetProcessHeap(), 0, (dwLenOut + 1) * sizeof(WCHAR)); if (!lpszReturnW) hRet = E_OUTOFMEMORY; else { hRet = AssocQueryStringW(cfFlags, str, lpszAssocW, lpszExtraW, lpszReturnW, &dwLenOut); if (SUCCEEDED(hRet)) WideCharToMultiByte(CP_ACP,0,szReturnW,-1,pszOut,dwLenOut,0,0); *pcchOut = dwLenOut; if (lpszReturnW != szReturnW) HeapFree(GetProcessHeap(), 0, lpszReturnW); } } if (lpszAssocW != szAssocW) HeapFree(GetProcessHeap(), 0, lpszAssocW); if (lpszExtraW != szExtraW) HeapFree(GetProcessHeap(), 0, lpszExtraW); return hRet; } /************************************************************************* * AssocQueryStringByKeyW [SHLWAPI.@] * * See AssocQueryStringByKeyA. */ HRESULT WINAPI AssocQueryStringByKeyW(ASSOCF cfFlags, ASSOCSTR str, HKEY hkAssoc, LPCWSTR pszExtra, LPWSTR pszOut, DWORD *pcchOut) { HRESULT hRet; IQueryAssociations* lpAssoc; TRACE("(0x%8x,0x%8x,%p,%s,%p,%p)\n", cfFlags, str, hkAssoc, debugstr_w(pszExtra), pszOut, pcchOut); lpAssoc = IQueryAssociations_Constructor(); if (!lpAssoc) return E_OUTOFMEMORY; cfFlags &= SHLWAPI_DEF_ASSOCF; hRet = IQueryAssociations_Init(lpAssoc, cfFlags, 0, hkAssoc, NULL); if (SUCCEEDED(hRet)) hRet = IQueryAssociations_GetString(lpAssoc, cfFlags, str, pszExtra, pszOut, pcchOut); IQueryAssociations_Release(lpAssoc); return hRet; } /************************************************************************* * AssocQueryStringByKeyA [SHLWAPI.@] * * Get a file association string from the registry, given a starting key. * * PARAMS * cfFlags [I] ASSOCF_ flags from "shlwapi.h" * str [I] Type of string to get * hkAssoc [I] Key to search below * pszExtra [I] Extra information about the string location * pszOut [O] Destination for the association string * pcchOut [O] Length of pszOut * * RETURNS * Success: S_OK. pszOut contains the string, pcchOut contains its length. * Failure: An HRESULT error code indicating the error. */ HRESULT WINAPI AssocQueryStringByKeyA(ASSOCF cfFlags, ASSOCSTR str, HKEY hkAssoc, LPCSTR pszExtra, LPSTR pszOut, DWORD *pcchOut) { WCHAR szExtraW[MAX_PATH], *lpszExtraW; WCHAR szReturnW[MAX_PATH], *lpszReturnW = szReturnW; HRESULT hRet = E_OUTOFMEMORY; TRACE("(0x%8x,0x%8x,%p,%s,%p,%p)\n", cfFlags, str, hkAssoc, debugstr_a(pszExtra), pszOut, pcchOut); if (!pcchOut) hRet = E_INVALIDARG; else if (SHLWAPI_ParamAToW(pszExtra, szExtraW, MAX_PATH, &lpszExtraW)) { DWORD dwLenOut = *pcchOut; if (dwLenOut >= MAX_PATH) lpszReturnW = HeapAlloc(GetProcessHeap(), 0, (dwLenOut + 1) * sizeof(WCHAR)); if (lpszReturnW) { hRet = AssocQueryStringByKeyW(cfFlags, str, hkAssoc, lpszExtraW, lpszReturnW, &dwLenOut); if (SUCCEEDED(hRet)) WideCharToMultiByte(CP_ACP,0,szReturnW,-1,pszOut,dwLenOut,0,0); *pcchOut = dwLenOut; if (lpszReturnW != szReturnW) HeapFree(GetProcessHeap(), 0, lpszReturnW); } } if (lpszExtraW != szExtraW) HeapFree(GetProcessHeap(), 0, lpszExtraW); return hRet; } /************************************************************************** * AssocIsDangerous (SHLWAPI.@) * * Determine if a file association is dangerous (potentially malware). * * PARAMS * lpszAssoc [I] Name of file or file extension to check. * * RETURNS * TRUE, if lpszAssoc may potentially be malware (executable), * FALSE, Otherwise. */ BOOL WINAPI AssocIsDangerous(LPCWSTR lpszAssoc) { FIXME("%s\n", debugstr_w(lpszAssoc)); return FALSE; } /************************************************************************** * IQueryAssociations_QueryInterface {SHLWAPI} * * See IUnknown_QueryInterface. */ static HRESULT WINAPI IQueryAssociations_fnQueryInterface( IQueryAssociations* iface, REFIID riid, LPVOID *ppvObj) { IQueryAssociationsImpl *This = (IQueryAssociationsImpl *)iface; TRACE("(%p,%s,%p)\n",This, debugstr_guid(riid), ppvObj); *ppvObj = NULL; if (IsEqualIID(riid, &IID_IUnknown) || IsEqualIID(riid, &IID_IQueryAssociations)) { *ppvObj = (IQueryAssociations*)This; IQueryAssociations_AddRef((IQueryAssociations*)*ppvObj); TRACE("Returning IQueryAssociations (%p)\n", *ppvObj); return S_OK; } TRACE("Returning E_NOINTERFACE\n"); return E_NOINTERFACE; } /************************************************************************** * IQueryAssociations_AddRef {SHLWAPI} * * See IUnknown_AddRef. */ static ULONG WINAPI IQueryAssociations_fnAddRef(IQueryAssociations *iface) { IQueryAssociationsImpl *This = (IQueryAssociationsImpl *)iface; ULONG refCount = InterlockedIncrement(&This->ref); TRACE("(%p)->(ref before=%u)\n",This, refCount - 1); return refCount; } /************************************************************************** * IQueryAssociations_Release {SHLWAPI} * * See IUnknown_Release. */ static ULONG WINAPI IQueryAssociations_fnRelease(IQueryAssociations *iface) { IQueryAssociationsImpl *This = (IQueryAssociationsImpl *)iface; ULONG refCount = InterlockedDecrement(&This->ref); TRACE("(%p)->(ref before=%u)\n",This, refCount + 1); if (!refCount) { TRACE("Destroying IQueryAssociations (%p)\n", This); HeapFree(GetProcessHeap(), 0, This); } return refCount; } /************************************************************************** * IQueryAssociations_Init {SHLWAPI} * * Initialise an IQueryAssociations object. * * PARAMS * iface [I] IQueryAssociations interface to initialise * cfFlags [I] ASSOCF_ flags from "shlwapi.h" * pszAssoc [I] String for the root key name, or NULL if hkProgid is given * hkeyProgid [I] Handle for the root key, or NULL if pszAssoc is given * hWnd [I] Reserved, must be NULL. * * RETURNS * Success: S_OK. iface is initialised with the parameters given. * Failure: An HRESULT error code indicating the error. */ static HRESULT WINAPI IQueryAssociations_fnInit( IQueryAssociations *iface, ASSOCF cfFlags, LPCWSTR pszAssoc, HKEY hkeyProgid, HWND hWnd) { static const WCHAR szProgID[] = {'P','r','o','g','I','D',0}; IQueryAssociationsImpl *This = (IQueryAssociationsImpl *)iface; HRESULT hr; TRACE("(%p)->(%d,%s,%p,%p)\n", iface, cfFlags, debugstr_w(pszAssoc), hkeyProgid, hWnd); if (hWnd != NULL) FIXME("hwnd != NULL not supported\n"); if (cfFlags != 0) FIXME("unsupported flags: %x\n", cfFlags); if (pszAssoc != NULL) { hr = RegOpenKeyExW(HKEY_CLASSES_ROOT, pszAssoc, 0, KEY_READ, &This->hkeySource); if (FAILED(hr)) return HRESULT_FROM_WIN32(ERROR_NO_ASSOCIATION); /* if this is not a prog id */ if ((*pszAssoc == '.') || (*pszAssoc == '{')) { hr = RegOpenKeyExW(This->hkeySource, szProgID, 0, KEY_READ, &This->hkeyProgID); if (FAILED(hr)) FIXME("Don't know what to return\n"); } else This->hkeyProgID = This->hkeySource; return S_OK; } else if (hkeyProgid != NULL) { This->hkeyProgID = hkeyProgid; return S_OK; } else return E_FAIL; } /************************************************************************** * IQueryAssociations_GetString {SHLWAPI} * * Get a file association string from the registry. * * PARAMS * iface [I] IQueryAssociations interface to query * cfFlags [I] ASSOCF_ flags from "shlwapi.h" * str [I] Type of string to get (ASSOCSTR enum from "shlwapi.h") * pszExtra [I] Extra information about the string location * pszOut [O] Destination for the association string * pcchOut [I/O] Length of pszOut * * RETURNS * Success: S_OK. pszOut contains the string, pcchOut contains its length. * Failure: An HRESULT error code indicating the error. */ static HRESULT WINAPI IQueryAssociations_fnGetString( IQueryAssociations *iface, ASSOCF cfFlags, ASSOCSTR str, LPCWSTR pszExtra, LPWSTR pszOut, DWORD *pcchOut) { IQueryAssociationsImpl *This = (IQueryAssociationsImpl *)iface; FIXME("(%p,0x%8x,0x%8x,%s,%p,%p)-stub!\n", This, cfFlags, str, debugstr_w(pszExtra), pszOut, pcchOut); return E_NOTIMPL; } /************************************************************************** * IQueryAssociations_GetKey {SHLWAPI} * * Get a file association key from the registry. * * PARAMS * iface [I] IQueryAssociations interface to query * cfFlags [I] ASSOCF_ flags from "shlwapi.h" * assockey [I] Type of key to get (ASSOCKEY enum from "shlwapi.h") * pszExtra [I] Extra information about the key location * phkeyOut [O] Destination for the association key * * RETURNS * Success: S_OK. phkeyOut contains a handle to the key. * Failure: An HRESULT error code indicating the error. */ static HRESULT WINAPI IQueryAssociations_fnGetKey( IQueryAssociations *iface, ASSOCF cfFlags, ASSOCKEY assockey, LPCWSTR pszExtra, HKEY *phkeyOut) { IQueryAssociationsImpl *This = (IQueryAssociationsImpl *)iface; FIXME("(%p,0x%8x,0x%8x,%s,%p)-stub!\n", This, cfFlags, assockey, debugstr_w(pszExtra), phkeyOut); return E_NOTIMPL; } /************************************************************************** * IQueryAssociations_GetData {SHLWAPI} * * Get the data for a file association key from the registry. * * PARAMS * iface [I] IQueryAssociations interface to query * cfFlags [I] ASSOCF_ flags from "shlwapi.h" * assocdata [I] Type of data to get (ASSOCDATA enum from "shlwapi.h") * pszExtra [I] Extra information about the data location * pvOut [O] Destination for the association key * pcbOut [I/O] Size of pvOut * * RETURNS * Success: S_OK. pszOut contains the data, pcbOut contains its length. * Failure: An HRESULT error code indicating the error. */ static HRESULT WINAPI IQueryAssociations_fnGetData( IQueryAssociations *iface, ASSOCF cfFlags, ASSOCDATA assocdata, LPCWSTR pszExtra, LPVOID pvOut, DWORD *pcbOut) { IQueryAssociationsImpl *This = (IQueryAssociationsImpl *)iface; FIXME("(%p,0x%8x,0x%8x,%s,%p,%p)-stub!\n", This, cfFlags, assocdata, debugstr_w(pszExtra), pvOut, pcbOut); return E_NOTIMPL; } /************************************************************************** * IQueryAssociations_GetEnum {SHLWAPI} * * Not yet implemented in native Win32. * * PARAMS * iface [I] IQueryAssociations interface to query * cfFlags [I] ASSOCF_ flags from "shlwapi.h" * assocenum [I] Type of enum to get (ASSOCENUM enum from "shlwapi.h") * pszExtra [I] Extra information about the enum location * riid [I] REFIID to look for * ppvOut [O] Destination for the interface. * * RETURNS * Success: S_OK. * Failure: An HRESULT error code indicating the error. * * NOTES * Presumably this function returns an enumerator object. */ static HRESULT WINAPI IQueryAssociations_fnGetEnum( IQueryAssociations *iface, ASSOCF cfFlags, ASSOCENUM assocenum, LPCWSTR pszExtra, REFIID riid, LPVOID *ppvOut) { IQueryAssociationsImpl *This = (IQueryAssociationsImpl *)iface; FIXME("(%p,0x%8x,0x%8x,%s,%s,%p)-stub!\n", This, cfFlags, assocenum, debugstr_w(pszExtra), debugstr_guid(riid), ppvOut); return E_NOTIMPL; } static const IQueryAssociationsVtbl IQueryAssociations_vtbl = { IQueryAssociations_fnQueryInterface, IQueryAssociations_fnAddRef, IQueryAssociations_fnRelease, IQueryAssociations_fnInit, IQueryAssociations_fnGetString, IQueryAssociations_fnGetKey, IQueryAssociations_fnGetData, IQueryAssociations_fnGetEnum };