356 lines
11 KiB
C
356 lines
11 KiB
C
|
/*
|
||
|
|
* SHLWAPI IQueryAssociations 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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
||
|
|
*
|
||
|
|
* NOTES
|
||
|
|
* These function simplify the process of using the IQueryAssociations
|
||
|
|
* interface, primarily by providing the means to get an interface pointer.
|
||
|
|
* For those not interested in the object, the AssocQuery* functions hide
|
||
|
|
* the object completely and simply provide the functionality.
|
||
|
|
*
|
||
|
|
* We require the implementation from shell32 because there is no interface
|
||
|
|
* for exporting an IQueryAssociations object available from that dll -
|
||
|
|
* as it is, we need the constructor.
|
||
|
|
*/
|
||
|
|
#include "windef.h"
|
||
|
|
#include "winnls.h"
|
||
|
|
#include "winreg.h"
|
||
|
|
#include "shlguid.h"
|
||
|
|
#include "shlwapi.h"
|
||
|
|
#include "wine/debug.h"
|
||
|
|
|
||
|
|
WINE_DEFAULT_DEBUG_CHANNEL(shell);
|
||
|
|
|
||
|
|
/* Default IQueryAssociations::Init() flags */
|
||
|
|
#define SHLWAPI_DEF_ASSOCF (ASSOCF_INIT_BYEXENAME|ASSOCF_INIT_DEFAULTTOSTAR| \
|
||
|
|
ASSOCF_INIT_DEFAULTTOFOLDER)
|
||
|
|
|
||
|
|
/* FIXME:
|
||
|
|
* This is a temporary placeholder. We need the whole IQueryAssociations object.
|
||
|
|
*/
|
||
|
|
static IQueryAssociations* IQueryAssociations_Constructor()
|
||
|
|
{
|
||
|
|
FIXME("() stub!\n");
|
||
|
|
return NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
/*************************************************************************
|
||
|
|
* 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 = lstrlenA(lpszParam);
|
||
|
|
|
||
|
|
if (dwStrLen < dwLen)
|
||
|
|
{
|
||
|
|
*lpszOut = lpszBuff; /* Use Buffer, it is big enough */
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
/* Create a new buffer big enough for the string */
|
||
|
|
*lpszOut = (LPWSTR)HeapAlloc(GetProcessHeap(), 0,
|
||
|
|
(dwStrLen + 1) * sizeof(WCHAR));
|
||
|
|
if (!*lpszOut)
|
||
|
|
return FALSE;
|
||
|
|
}
|
||
|
|
MultiByteToWideChar(0, 0, lpszParam, -1, *lpszOut, -1);
|
||
|
|
}
|
||
|
|
else
|
||
|
|
*lpszOut = NULL;
|
||
|
|
return TRUE;
|
||
|
|
}
|
||
|
|
|
||
|
|
/*************************************************************************
|
||
|
|
* AssocCreate [SHLWAPI.253]
|
||
|
|
*
|
||
|
|
* 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 value
|
||
|
|
*/
|
||
|
|
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.255]
|
||
|
|
*
|
||
|
|
* See AssocQueryKeyA.
|
||
|
|
*/
|
||
|
|
HRESULT WINAPI AssocQueryKeyW(ASSOCF flags, ASSOCKEY key, LPCWSTR pszAssoc,
|
||
|
|
LPCWSTR pszExtra, HKEY *phkeyOut)
|
||
|
|
{
|
||
|
|
HRESULT hRet;
|
||
|
|
IQueryAssociations* lpAssoc;
|
||
|
|
|
||
|
|
lpAssoc = IQueryAssociations_Constructor();
|
||
|
|
|
||
|
|
if (!lpAssoc)
|
||
|
|
return E_OUTOFMEMORY;
|
||
|
|
|
||
|
|
flags &= SHLWAPI_DEF_ASSOCF;
|
||
|
|
hRet = IQueryAssociations_Init(lpAssoc, flags, pszAssoc, (HKEY)0, (HWND)0);
|
||
|
|
|
||
|
|
if (SUCCEEDED(hRet))
|
||
|
|
hRet = IQueryAssociations_GetKey(lpAssoc, flags, key, pszExtra, phkeyOut);
|
||
|
|
|
||
|
|
IQueryAssociations_Release(lpAssoc);
|
||
|
|
return hRet;
|
||
|
|
}
|
||
|
|
|
||
|
|
/*************************************************************************
|
||
|
|
* AssocQueryKeyA [SHLWAPI.254]
|
||
|
|
*
|
||
|
|
* Get a file association key from the registry.
|
||
|
|
*
|
||
|
|
* PARAMS
|
||
|
|
* flags [I] Flags for the search
|
||
|
|
* key [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
|
||
|
|
*/
|
||
|
|
HRESULT WINAPI AssocQueryKeyA(ASSOCF flags, ASSOCKEY key, LPCSTR pszAssoc,
|
||
|
|
LPCSTR pszExtra, HKEY *phkeyOut)
|
||
|
|
{
|
||
|
|
WCHAR szAssocW[MAX_PATH], *lpszAssocW = NULL;
|
||
|
|
WCHAR szExtraW[MAX_PATH], *lpszExtraW = NULL;
|
||
|
|
HRESULT hRet = E_OUTOFMEMORY;
|
||
|
|
|
||
|
|
if (SHLWAPI_ParamAToW(pszAssoc, szAssocW, MAX_PATH, &lpszAssocW) &&
|
||
|
|
SHLWAPI_ParamAToW(pszExtra, szExtraW, MAX_PATH, &lpszExtraW))
|
||
|
|
{
|
||
|
|
hRet = AssocQueryKeyW(flags, key, lpszAssocW, lpszExtraW, phkeyOut);
|
||
|
|
}
|
||
|
|
|
||
|
|
if (lpszAssocW && lpszAssocW != szAssocW)
|
||
|
|
HeapFree(GetProcessHeap(), 0, lpszAssocW);
|
||
|
|
|
||
|
|
if (lpszExtraW && lpszExtraW != szExtraW)
|
||
|
|
HeapFree(GetProcessHeap(), 0, lpszExtraW);
|
||
|
|
|
||
|
|
return hRet;
|
||
|
|
}
|
||
|
|
|
||
|
|
/*************************************************************************
|
||
|
|
* AssocQueryStringW [SHLWAPI.384]
|
||
|
|
*
|
||
|
|
* See AssocQueryStringA.
|
||
|
|
*/
|
||
|
|
HRESULT WINAPI AssocQueryStringW(ASSOCF flags, ASSOCSTR str, LPCWSTR pszAssoc,
|
||
|
|
LPCWSTR pszExtra, LPWSTR pszOut, DWORD *pcchOut)
|
||
|
|
{
|
||
|
|
HRESULT hRet;
|
||
|
|
IQueryAssociations* lpAssoc;
|
||
|
|
|
||
|
|
if (!pcchOut)
|
||
|
|
return E_INVALIDARG;
|
||
|
|
|
||
|
|
lpAssoc = IQueryAssociations_Constructor();
|
||
|
|
|
||
|
|
if (!lpAssoc)
|
||
|
|
return E_OUTOFMEMORY;
|
||
|
|
|
||
|
|
hRet = IQueryAssociations_Init(lpAssoc, flags & SHLWAPI_DEF_ASSOCF,
|
||
|
|
pszAssoc, (HKEY)0, (HWND)0);
|
||
|
|
|
||
|
|
if (SUCCEEDED(hRet))
|
||
|
|
hRet = IQueryAssociations_GetString(lpAssoc, flags, str, pszExtra,
|
||
|
|
pszOut, pcchOut);
|
||
|
|
|
||
|
|
IQueryAssociations_Release(lpAssoc);
|
||
|
|
return hRet;
|
||
|
|
}
|
||
|
|
|
||
|
|
/*************************************************************************
|
||
|
|
* AssocQueryStringA [SHLWAPI.381]
|
||
|
|
*
|
||
|
|
* Get a file association string from the registry.
|
||
|
|
*
|
||
|
|
* PARAMS
|
||
|
|
* flags [I] Flags for the search
|
||
|
|
* str [I] Type of string to get
|
||
|
|
* 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.
|
||
|
|
* Failure: An HRESULT error code.
|
||
|
|
*/
|
||
|
|
HRESULT WINAPI AssocQueryStringA(ASSOCF flags, 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;
|
||
|
|
|
||
|
|
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 = (LPWSTR)HeapAlloc(GetProcessHeap(), 0,
|
||
|
|
(dwLenOut + 1) * sizeof(WCHAR));
|
||
|
|
|
||
|
|
if (!lpszReturnW)
|
||
|
|
hRet = E_OUTOFMEMORY;
|
||
|
|
else
|
||
|
|
{
|
||
|
|
hRet = AssocQueryStringW(flags, str, lpszAssocW, lpszExtraW,
|
||
|
|
lpszReturnW, &dwLenOut);
|
||
|
|
|
||
|
|
if (SUCCEEDED(hRet))
|
||
|
|
WideCharToMultiByte(CP_ACP,0,szReturnW,-1,pszOut,dwLenOut,0,0);
|
||
|
|
*pcchOut = dwLenOut;
|
||
|
|
|
||
|
|
if (lpszReturnW && lpszReturnW != szReturnW)
|
||
|
|
HeapFree(GetProcessHeap(), 0, lpszReturnW);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if (lpszAssocW && lpszAssocW != szAssocW)
|
||
|
|
HeapFree(GetProcessHeap(), 0, lpszAssocW);
|
||
|
|
if (lpszExtraW && lpszExtraW != szExtraW)
|
||
|
|
HeapFree(GetProcessHeap(), 0, lpszExtraW);
|
||
|
|
return hRet;
|
||
|
|
}
|
||
|
|
|
||
|
|
/*************************************************************************
|
||
|
|
* AssocQueryStringByKeyW [SHLWAPI.383]
|
||
|
|
*
|
||
|
|
* See AssocQueryStringByKeyA.
|
||
|
|
*/
|
||
|
|
HRESULT WINAPI AssocQueryStringByKeyW(ASSOCF flags, ASSOCSTR str, HKEY hkAssoc,
|
||
|
|
LPCWSTR pszExtra, LPWSTR pszOut,
|
||
|
|
DWORD *pcchOut)
|
||
|
|
{
|
||
|
|
HRESULT hRet;
|
||
|
|
IQueryAssociations* lpAssoc;
|
||
|
|
|
||
|
|
lpAssoc = IQueryAssociations_Constructor();
|
||
|
|
|
||
|
|
if (!lpAssoc)
|
||
|
|
return E_OUTOFMEMORY;
|
||
|
|
|
||
|
|
flags &= SHLWAPI_DEF_ASSOCF;
|
||
|
|
hRet = IQueryAssociations_Init(lpAssoc, flags, 0, hkAssoc, (HWND)0);
|
||
|
|
|
||
|
|
if (SUCCEEDED(hRet))
|
||
|
|
hRet = IQueryAssociations_GetString(lpAssoc, flags, str, pszExtra,
|
||
|
|
pszOut, pcchOut);
|
||
|
|
|
||
|
|
IQueryAssociations_Release(lpAssoc);
|
||
|
|
return hRet;
|
||
|
|
}
|
||
|
|
|
||
|
|
/*************************************************************************
|
||
|
|
* AssocQueryStringByKeyA [SHLWAPI.382]
|
||
|
|
*
|
||
|
|
* Get a file association string from the registry, given a starting key.
|
||
|
|
*
|
||
|
|
* PARAMS
|
||
|
|
* flags [I] Flags for the search
|
||
|
|
* 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.
|
||
|
|
* Failure: An HRESULT error code.
|
||
|
|
*/
|
||
|
|
HRESULT WINAPI AssocQueryStringByKeyA(ASSOCF flags, 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;
|
||
|
|
|
||
|
|
if (!pcchOut)
|
||
|
|
hRet = E_INVALIDARG;
|
||
|
|
else if (SHLWAPI_ParamAToW(pszExtra, szExtraW, MAX_PATH, &lpszExtraW))
|
||
|
|
{
|
||
|
|
DWORD dwLenOut = *pcchOut;
|
||
|
|
if (dwLenOut >= MAX_PATH)
|
||
|
|
lpszReturnW = (LPWSTR)HeapAlloc(GetProcessHeap(), 0,
|
||
|
|
(dwLenOut + 1) * sizeof(WCHAR));
|
||
|
|
|
||
|
|
if (lpszReturnW)
|
||
|
|
{
|
||
|
|
hRet = AssocQueryStringByKeyW(flags, 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 && lpszExtraW != szExtraW)
|
||
|
|
HeapFree(GetProcessHeap(), 0, lpszExtraW);
|
||
|
|
return hRet;
|
||
|
|
}
|