2002-09-05 01:32:19 +02:00
|
|
|
/*
|
2003-07-19 05:07:41 +02:00
|
|
|
* IQueryAssociations object and helper functions
|
2002-09-05 01:32:19 +02:00
|
|
|
*
|
|
|
|
* 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
|
|
|
|
*/
|
2003-09-06 01:08:26 +02:00
|
|
|
#include <stdarg.h>
|
|
|
|
|
2002-09-05 01:32:19 +02:00
|
|
|
#include "windef.h"
|
2003-08-20 20:22:31 +02:00
|
|
|
#include "winbase.h"
|
2002-09-05 01:32:19 +02:00
|
|
|
#include "winnls.h"
|
|
|
|
#include "winreg.h"
|
2003-09-06 01:08:26 +02:00
|
|
|
#include "objbase.h"
|
2002-09-05 01:32:19 +02:00
|
|
|
#include "shlguid.h"
|
|
|
|
#include "shlwapi.h"
|
|
|
|
#include "wine/debug.h"
|
|
|
|
|
|
|
|
WINE_DEFAULT_DEBUG_CHANNEL(shell);
|
|
|
|
|
2003-07-19 05:07:41 +02:00
|
|
|
/**************************************************************************
|
|
|
|
* 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 tranparently 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
|
|
|
|
*/
|
|
|
|
|
2002-09-05 01:32:19 +02:00
|
|
|
/* Default IQueryAssociations::Init() flags */
|
|
|
|
#define SHLWAPI_DEF_ASSOCF (ASSOCF_INIT_BYEXENAME|ASSOCF_INIT_DEFAULTTOSTAR| \
|
|
|
|
ASSOCF_INIT_DEFAULTTOFOLDER)
|
|
|
|
|
2003-07-19 05:07:41 +02:00
|
|
|
typedef struct
|
|
|
|
{
|
2004-08-12 21:52:49 +02:00
|
|
|
IQueryAssociationsVtbl *lpVtbl;
|
2003-07-19 05:07:41 +02:00
|
|
|
LONG ref;
|
2003-09-11 04:56:15 +02:00
|
|
|
HKEY hkeySource;
|
|
|
|
HKEY hkeyProgID;
|
2003-07-19 05:07:41 +02:00
|
|
|
} IQueryAssociationsImpl;
|
|
|
|
|
2004-08-13 01:00:51 +02:00
|
|
|
static struct IQueryAssociationsVtbl IQueryAssociations_vtbl;
|
2003-07-19 05:07:41 +02:00
|
|
|
|
|
|
|
/**************************************************************************
|
|
|
|
* IQueryAssociations_Constructor [internal]
|
|
|
|
*
|
|
|
|
* Construct a new IQueryAssociations object.
|
2002-09-05 01:32:19 +02:00
|
|
|
*/
|
2003-07-19 05:07:41 +02:00
|
|
|
static IQueryAssociations* IQueryAssociations_Constructor(void)
|
2002-09-05 01:32:19 +02:00
|
|
|
{
|
2003-07-19 05:07:41 +02:00
|
|
|
IQueryAssociationsImpl* iface;
|
|
|
|
|
2004-07-04 02:06:29 +02:00
|
|
|
iface = HeapAlloc(GetProcessHeap(),0,sizeof(IQueryAssociationsImpl));
|
2003-07-19 05:07:41 +02:00
|
|
|
iface->lpVtbl = &IQueryAssociations_vtbl;
|
|
|
|
iface->ref = 1;
|
2003-09-11 04:56:15 +02:00
|
|
|
iface->hkeySource = NULL;
|
|
|
|
iface->hkeyProgID = NULL;
|
2003-07-19 05:07:41 +02:00
|
|
|
|
|
|
|
TRACE("Returning IQueryAssociations* %p\n", iface);
|
|
|
|
return (IQueryAssociations*)iface;
|
2002-09-05 01:32:19 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
/*************************************************************************
|
|
|
|
* SHLWAPI_ParamAToW
|
|
|
|
*
|
|
|
|
* Internal helper function: Convert ASCII parameter to Unicode.
|
|
|
|
*/
|
|
|
|
static BOOL SHLWAPI_ParamAToW(LPCSTR lpszParam, LPWSTR lpszBuff, DWORD dwLen,
|
|
|
|
LPWSTR* lpszOut)
|
|
|
|
{
|
|
|
|
if (lpszParam)
|
|
|
|
{
|
2004-07-04 02:06:29 +02:00
|
|
|
DWORD dwStrLen = MultiByteToWideChar(CP_ACP, 0, lpszParam, -1, NULL, 0);
|
2002-09-05 01:32:19 +02:00
|
|
|
|
|
|
|
if (dwStrLen < dwLen)
|
|
|
|
{
|
|
|
|
*lpszOut = lpszBuff; /* Use Buffer, it is big enough */
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
/* Create a new buffer big enough for the string */
|
2004-07-04 02:06:29 +02:00
|
|
|
*lpszOut = HeapAlloc(GetProcessHeap(), 0,
|
|
|
|
dwStrLen * sizeof(WCHAR));
|
2002-09-05 01:32:19 +02:00
|
|
|
if (!*lpszOut)
|
|
|
|
return FALSE;
|
|
|
|
}
|
2004-07-04 02:06:29 +02:00
|
|
|
MultiByteToWideChar(CP_ACP, 0, lpszParam, -1, *lpszOut, dwStrLen);
|
2002-09-05 01:32:19 +02:00
|
|
|
}
|
|
|
|
else
|
|
|
|
*lpszOut = NULL;
|
|
|
|
return TRUE;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*************************************************************************
|
2003-09-11 04:56:15 +02:00
|
|
|
* AssocCreate [SHLWAPI.@]
|
2002-09-05 01:32:19 +02:00
|
|
|
*
|
|
|
|
* 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
|
2003-07-19 05:07:41 +02:00
|
|
|
* 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.
|
2002-09-05 01:32:19 +02:00
|
|
|
*/
|
|
|
|
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;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*************************************************************************
|
2003-09-11 04:56:15 +02:00
|
|
|
* AssocQueryKeyW [SHLWAPI.@]
|
2002-09-05 01:32:19 +02:00
|
|
|
*
|
|
|
|
* See AssocQueryKeyA.
|
|
|
|
*/
|
2003-07-19 05:07:41 +02:00
|
|
|
HRESULT WINAPI AssocQueryKeyW(ASSOCF cfFlags, ASSOCKEY assockey, LPCWSTR pszAssoc,
|
2002-09-05 01:32:19 +02:00
|
|
|
LPCWSTR pszExtra, HKEY *phkeyOut)
|
|
|
|
{
|
|
|
|
HRESULT hRet;
|
|
|
|
IQueryAssociations* lpAssoc;
|
|
|
|
|
2003-07-19 05:07:41 +02:00
|
|
|
TRACE("(0x%8lx,0x%8x,%s,%s,%p)\n", cfFlags, assockey, debugstr_w(pszAssoc),
|
|
|
|
debugstr_w(pszExtra), phkeyOut);
|
|
|
|
|
2002-09-05 01:32:19 +02:00
|
|
|
lpAssoc = IQueryAssociations_Constructor();
|
|
|
|
|
|
|
|
if (!lpAssoc)
|
|
|
|
return E_OUTOFMEMORY;
|
|
|
|
|
2003-07-19 05:07:41 +02:00
|
|
|
cfFlags &= SHLWAPI_DEF_ASSOCF;
|
|
|
|
hRet = IQueryAssociations_Init(lpAssoc, cfFlags, pszAssoc, NULL, NULL);
|
2002-09-05 01:32:19 +02:00
|
|
|
|
|
|
|
if (SUCCEEDED(hRet))
|
2003-07-19 05:07:41 +02:00
|
|
|
hRet = IQueryAssociations_GetKey(lpAssoc, cfFlags, assockey, pszExtra, phkeyOut);
|
2002-09-05 01:32:19 +02:00
|
|
|
|
|
|
|
IQueryAssociations_Release(lpAssoc);
|
|
|
|
return hRet;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*************************************************************************
|
2003-09-11 04:56:15 +02:00
|
|
|
* AssocQueryKeyA [SHLWAPI.@]
|
2002-09-05 01:32:19 +02:00
|
|
|
*
|
|
|
|
* Get a file association key from the registry.
|
|
|
|
*
|
|
|
|
* PARAMS
|
2003-07-19 05:07:41 +02:00
|
|
|
* cfFlags [I] ASSOCF_ flags from "shlwapi.h"
|
|
|
|
* assockey [I] Type of key to get
|
2002-09-05 01:32:19 +02:00
|
|
|
* pszAssoc [I] Key name to search below
|
|
|
|
* pszExtra [I] Extra information about the key location
|
|
|
|
* phkeyOut [O] Destination for the association key
|
|
|
|
*
|
|
|
|
* RETURNS
|
2003-07-19 05:07:41 +02:00
|
|
|
* Success: S_OK. phkeyOut contains the key.
|
|
|
|
* Failure: An HRESULT error code indicating the error.
|
2002-09-05 01:32:19 +02:00
|
|
|
*/
|
2003-07-19 05:07:41 +02:00
|
|
|
HRESULT WINAPI AssocQueryKeyA(ASSOCF cfFlags, ASSOCKEY assockey, LPCSTR pszAssoc,
|
2002-09-05 01:32:19 +02:00
|
|
|
LPCSTR pszExtra, HKEY *phkeyOut)
|
|
|
|
{
|
|
|
|
WCHAR szAssocW[MAX_PATH], *lpszAssocW = NULL;
|
|
|
|
WCHAR szExtraW[MAX_PATH], *lpszExtraW = NULL;
|
|
|
|
HRESULT hRet = E_OUTOFMEMORY;
|
|
|
|
|
2003-07-19 05:07:41 +02:00
|
|
|
TRACE("(0x%8lx,0x%8x,%s,%s,%p)\n", cfFlags, assockey, debugstr_a(pszAssoc),
|
|
|
|
debugstr_a(pszExtra), phkeyOut);
|
|
|
|
|
2002-09-05 01:32:19 +02:00
|
|
|
if (SHLWAPI_ParamAToW(pszAssoc, szAssocW, MAX_PATH, &lpszAssocW) &&
|
|
|
|
SHLWAPI_ParamAToW(pszExtra, szExtraW, MAX_PATH, &lpszExtraW))
|
|
|
|
{
|
2003-07-19 05:07:41 +02:00
|
|
|
hRet = AssocQueryKeyW(cfFlags, assockey, lpszAssocW, lpszExtraW, phkeyOut);
|
2002-09-05 01:32:19 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
if (lpszAssocW && lpszAssocW != szAssocW)
|
|
|
|
HeapFree(GetProcessHeap(), 0, lpszAssocW);
|
|
|
|
|
|
|
|
if (lpszExtraW && lpszExtraW != szExtraW)
|
|
|
|
HeapFree(GetProcessHeap(), 0, lpszExtraW);
|
|
|
|
|
|
|
|
return hRet;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*************************************************************************
|
2003-09-11 04:56:15 +02:00
|
|
|
* AssocQueryStringW [SHLWAPI.@]
|
2002-09-05 01:32:19 +02:00
|
|
|
*
|
|
|
|
* See AssocQueryStringA.
|
|
|
|
*/
|
2003-07-19 05:07:41 +02:00
|
|
|
HRESULT WINAPI AssocQueryStringW(ASSOCF cfFlags, ASSOCSTR str, LPCWSTR pszAssoc,
|
2002-09-05 01:32:19 +02:00
|
|
|
LPCWSTR pszExtra, LPWSTR pszOut, DWORD *pcchOut)
|
|
|
|
{
|
|
|
|
HRESULT hRet;
|
|
|
|
IQueryAssociations* lpAssoc;
|
|
|
|
|
2003-07-19 05:07:41 +02:00
|
|
|
TRACE("(0x%8lx,0x%8x,%s,%s,%p,%p)\n", cfFlags, str, debugstr_w(pszAssoc),
|
|
|
|
debugstr_w(pszExtra), pszOut, pcchOut);
|
|
|
|
|
2002-09-05 01:32:19 +02:00
|
|
|
if (!pcchOut)
|
|
|
|
return E_INVALIDARG;
|
|
|
|
|
|
|
|
lpAssoc = IQueryAssociations_Constructor();
|
|
|
|
|
|
|
|
if (!lpAssoc)
|
|
|
|
return E_OUTOFMEMORY;
|
|
|
|
|
2003-07-19 05:07:41 +02:00
|
|
|
hRet = IQueryAssociations_Init(lpAssoc, cfFlags & SHLWAPI_DEF_ASSOCF,
|
2002-12-02 19:10:57 +01:00
|
|
|
pszAssoc, NULL, NULL);
|
2002-09-05 01:32:19 +02:00
|
|
|
|
|
|
|
if (SUCCEEDED(hRet))
|
2003-07-19 05:07:41 +02:00
|
|
|
hRet = IQueryAssociations_GetString(lpAssoc, cfFlags, str, pszExtra,
|
2002-09-05 01:32:19 +02:00
|
|
|
pszOut, pcchOut);
|
|
|
|
|
|
|
|
IQueryAssociations_Release(lpAssoc);
|
|
|
|
return hRet;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*************************************************************************
|
2003-09-11 04:56:15 +02:00
|
|
|
* AssocQueryStringA [SHLWAPI.@]
|
2002-09-05 01:32:19 +02:00
|
|
|
*
|
|
|
|
* Get a file association string from the registry.
|
|
|
|
*
|
|
|
|
* PARAMS
|
2003-07-19 05:07:41 +02:00
|
|
|
* cfFlags [I] ASSOCF_ flags from "shlwapi.h"
|
|
|
|
* str [I] Type of string to get (ASSOCSTR enum from "shlwapi.h")
|
2002-09-05 01:32:19 +02:00
|
|
|
* 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
|
2003-07-19 05:07:41 +02:00
|
|
|
* Success: S_OK. pszOut contains the string, pcchOut contains its length.
|
|
|
|
* Failure: An HRESULT error code indicating the error.
|
2002-09-05 01:32:19 +02:00
|
|
|
*/
|
2003-07-19 05:07:41 +02:00
|
|
|
HRESULT WINAPI AssocQueryStringA(ASSOCF cfFlags, ASSOCSTR str, LPCSTR pszAssoc,
|
2002-09-05 01:32:19 +02:00
|
|
|
LPCSTR pszExtra, LPSTR pszOut, DWORD *pcchOut)
|
|
|
|
{
|
|
|
|
WCHAR szAssocW[MAX_PATH], *lpszAssocW = NULL;
|
|
|
|
WCHAR szExtraW[MAX_PATH], *lpszExtraW = NULL;
|
|
|
|
HRESULT hRet = E_OUTOFMEMORY;
|
|
|
|
|
2003-07-19 05:07:41 +02:00
|
|
|
TRACE("(0x%8lx,0x%8x,%s,%s,%p,%p)\n", cfFlags, str, debugstr_a(pszAssoc),
|
|
|
|
debugstr_a(pszExtra), pszOut, pcchOut);
|
|
|
|
|
2002-09-05 01:32:19 +02:00
|
|
|
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)
|
2004-07-04 02:06:29 +02:00
|
|
|
lpszReturnW = HeapAlloc(GetProcessHeap(), 0,
|
2002-09-05 01:32:19 +02:00
|
|
|
(dwLenOut + 1) * sizeof(WCHAR));
|
|
|
|
|
|
|
|
if (!lpszReturnW)
|
|
|
|
hRet = E_OUTOFMEMORY;
|
|
|
|
else
|
|
|
|
{
|
2003-07-19 05:07:41 +02:00
|
|
|
hRet = AssocQueryStringW(cfFlags, str, lpszAssocW, lpszExtraW,
|
2002-09-05 01:32:19 +02:00
|
|
|
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;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*************************************************************************
|
2003-09-11 04:56:15 +02:00
|
|
|
* AssocQueryStringByKeyW [SHLWAPI.@]
|
2002-09-05 01:32:19 +02:00
|
|
|
*
|
|
|
|
* See AssocQueryStringByKeyA.
|
|
|
|
*/
|
2003-07-19 05:07:41 +02:00
|
|
|
HRESULT WINAPI AssocQueryStringByKeyW(ASSOCF cfFlags, ASSOCSTR str, HKEY hkAssoc,
|
2002-09-05 01:32:19 +02:00
|
|
|
LPCWSTR pszExtra, LPWSTR pszOut,
|
|
|
|
DWORD *pcchOut)
|
|
|
|
{
|
|
|
|
HRESULT hRet;
|
|
|
|
IQueryAssociations* lpAssoc;
|
|
|
|
|
2003-07-19 05:07:41 +02:00
|
|
|
TRACE("(0x%8lx,0x%8x,%p,%s,%p,%p)\n", cfFlags, str, hkAssoc,
|
|
|
|
debugstr_w(pszExtra), pszOut, pcchOut);
|
|
|
|
|
2002-09-05 01:32:19 +02:00
|
|
|
lpAssoc = IQueryAssociations_Constructor();
|
|
|
|
|
|
|
|
if (!lpAssoc)
|
|
|
|
return E_OUTOFMEMORY;
|
|
|
|
|
2003-07-19 05:07:41 +02:00
|
|
|
cfFlags &= SHLWAPI_DEF_ASSOCF;
|
|
|
|
hRet = IQueryAssociations_Init(lpAssoc, cfFlags, 0, hkAssoc, NULL);
|
2002-09-05 01:32:19 +02:00
|
|
|
|
|
|
|
if (SUCCEEDED(hRet))
|
2003-07-19 05:07:41 +02:00
|
|
|
hRet = IQueryAssociations_GetString(lpAssoc, cfFlags, str, pszExtra,
|
2002-09-05 01:32:19 +02:00
|
|
|
pszOut, pcchOut);
|
|
|
|
|
|
|
|
IQueryAssociations_Release(lpAssoc);
|
|
|
|
return hRet;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*************************************************************************
|
2003-09-11 04:56:15 +02:00
|
|
|
* AssocQueryStringByKeyA [SHLWAPI.@]
|
2002-09-05 01:32:19 +02:00
|
|
|
*
|
|
|
|
* Get a file association string from the registry, given a starting key.
|
|
|
|
*
|
|
|
|
* PARAMS
|
2003-07-19 05:07:41 +02:00
|
|
|
* cfFlags [I] ASSOCF_ flags from "shlwapi.h"
|
2002-09-05 01:32:19 +02:00
|
|
|
* 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
|
2003-07-19 05:07:41 +02:00
|
|
|
* Success: S_OK. pszOut contains the string, pcchOut contains its length.
|
|
|
|
* Failure: An HRESULT error code indicating the error.
|
2002-09-05 01:32:19 +02:00
|
|
|
*/
|
2003-07-19 05:07:41 +02:00
|
|
|
HRESULT WINAPI AssocQueryStringByKeyA(ASSOCF cfFlags, ASSOCSTR str, HKEY hkAssoc,
|
2002-09-05 01:32:19 +02:00
|
|
|
LPCSTR pszExtra, LPSTR pszOut,
|
|
|
|
DWORD *pcchOut)
|
|
|
|
{
|
|
|
|
WCHAR szExtraW[MAX_PATH], *lpszExtraW;
|
|
|
|
WCHAR szReturnW[MAX_PATH], *lpszReturnW = szReturnW;
|
|
|
|
HRESULT hRet = E_OUTOFMEMORY;
|
|
|
|
|
2003-07-19 05:07:41 +02:00
|
|
|
TRACE("(0x%8lx,0x%8x,%p,%s,%p,%p)\n", cfFlags, str, hkAssoc,
|
|
|
|
debugstr_a(pszExtra), pszOut, pcchOut);
|
|
|
|
|
2002-09-05 01:32:19 +02:00
|
|
|
if (!pcchOut)
|
|
|
|
hRet = E_INVALIDARG;
|
|
|
|
else if (SHLWAPI_ParamAToW(pszExtra, szExtraW, MAX_PATH, &lpszExtraW))
|
|
|
|
{
|
|
|
|
DWORD dwLenOut = *pcchOut;
|
|
|
|
if (dwLenOut >= MAX_PATH)
|
2004-07-04 02:06:29 +02:00
|
|
|
lpszReturnW = HeapAlloc(GetProcessHeap(), 0,
|
2002-09-05 01:32:19 +02:00
|
|
|
(dwLenOut + 1) * sizeof(WCHAR));
|
|
|
|
|
|
|
|
if (lpszReturnW)
|
|
|
|
{
|
2003-07-19 05:07:41 +02:00
|
|
|
hRet = AssocQueryStringByKeyW(cfFlags, str, hkAssoc, lpszExtraW,
|
2002-09-05 01:32:19 +02:00
|
|
|
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;
|
|
|
|
}
|
2003-07-19 05:07:41 +02:00
|
|
|
|
2003-07-22 01:42:24 +02:00
|
|
|
|
|
|
|
/**************************************************************************
|
|
|
|
* AssocIsDangerous (SHLWAPI.@)
|
2004-02-10 03:22:17 +01:00
|
|
|
*
|
|
|
|
* Determine if a file association is dangerous (potentially malware).
|
|
|
|
*
|
|
|
|
* PARAMS
|
2004-03-02 21:55:57 +01:00
|
|
|
* lpszAssoc [I] Name of file or file extension to check.
|
2004-02-10 03:22:17 +01:00
|
|
|
*
|
|
|
|
* RETURNS
|
|
|
|
* TRUE, if lpszAssoc may potentially be malware (executable),
|
|
|
|
* FALSE, Otherwise.
|
2003-07-22 01:42:24 +02:00
|
|
|
*/
|
2004-02-10 03:22:17 +01:00
|
|
|
BOOL WINAPI AssocIsDangerous(LPCWSTR lpszAssoc)
|
2003-07-22 01:42:24 +02:00
|
|
|
{
|
2004-02-10 03:22:17 +01:00
|
|
|
FIXME("%s\n", debugstr_w(lpszAssoc));
|
|
|
|
return FALSE;
|
2003-07-22 01:42:24 +02:00
|
|
|
}
|
|
|
|
|
2003-07-19 05:07:41 +02:00
|
|
|
/**************************************************************************
|
|
|
|
* IQueryAssociations_QueryInterface {SHLWAPI}
|
|
|
|
*
|
|
|
|
* See IUnknown_QueryInterface.
|
|
|
|
*/
|
|
|
|
static HRESULT WINAPI IQueryAssociations_fnQueryInterface(
|
|
|
|
IQueryAssociations* iface,
|
|
|
|
REFIID riid,
|
|
|
|
LPVOID *ppvObj)
|
|
|
|
{
|
|
|
|
ICOM_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)
|
|
|
|
{
|
|
|
|
ICOM_THIS(IQueryAssociationsImpl,iface);
|
|
|
|
|
|
|
|
TRACE("(%p)->(ref before=%lu)\n",This, This->ref);
|
|
|
|
|
|
|
|
return InterlockedIncrement(&This->ref);
|
|
|
|
}
|
|
|
|
|
|
|
|
/**************************************************************************
|
|
|
|
* IQueryAssociations_Release {SHLWAPI}
|
|
|
|
*
|
|
|
|
* See IUnknown_Release.
|
|
|
|
*/
|
|
|
|
static ULONG WINAPI IQueryAssociations_fnRelease(IQueryAssociations *iface)
|
|
|
|
{
|
|
|
|
ICOM_THIS(IQueryAssociationsImpl,iface);
|
|
|
|
ULONG ulRet;
|
|
|
|
|
|
|
|
TRACE("(%p)->(ref before=%lu)\n",This, This->ref);
|
|
|
|
|
|
|
|
if (!(ulRet = InterlockedDecrement(&This->ref)))
|
|
|
|
{
|
|
|
|
TRACE("Destroying IQueryAssociations (%p)\n", This);
|
|
|
|
HeapFree(GetProcessHeap(), 0, This);
|
|
|
|
}
|
|
|
|
return ulRet;
|
|
|
|
}
|
|
|
|
|
|
|
|
/**************************************************************************
|
|
|
|
* IQueryAssociations_Init {SHLWAPI}
|
|
|
|
*
|
|
|
|
* Initialise an IQueryAssociations object.
|
|
|
|
*
|
|
|
|
* PARAMS
|
2003-09-11 04:56:15 +02:00
|
|
|
* 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.
|
2003-07-19 05:07:41 +02:00
|
|
|
*
|
|
|
|
* 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,
|
2003-09-11 04:56:15 +02:00
|
|
|
HKEY hkeyProgid,
|
2003-07-19 05:07:41 +02:00
|
|
|
HWND hWnd)
|
|
|
|
{
|
2003-09-11 04:56:15 +02:00
|
|
|
static const WCHAR szProgID[] = {'P','r','o','g','I','D',0};
|
|
|
|
ICOM_THIS(IQueryAssociationsImpl,iface);
|
|
|
|
HRESULT hr;
|
|
|
|
|
|
|
|
TRACE("(%p)->(%ld,%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: %lx\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;
|
2003-07-19 05:07:41 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
/**************************************************************************
|
|
|
|
* 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)
|
|
|
|
{
|
|
|
|
ICOM_THIS(IQueryAssociationsImpl, iface);
|
|
|
|
|
|
|
|
FIXME("(%p,0x%8lx,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)
|
|
|
|
{
|
|
|
|
ICOM_THIS(IQueryAssociationsImpl, iface);
|
|
|
|
|
|
|
|
FIXME("(%p,0x%8lx,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)
|
|
|
|
{
|
|
|
|
ICOM_THIS(IQueryAssociationsImpl, iface);
|
|
|
|
|
|
|
|
FIXME("(%p,0x%8lx,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)
|
|
|
|
{
|
|
|
|
ICOM_THIS(IQueryAssociationsImpl, iface);
|
|
|
|
|
|
|
|
FIXME("(%p,0x%8lx,0x%8x,%s,%s,%p)-stub!\n", This, cfFlags, assocenum,
|
|
|
|
debugstr_w(pszExtra), debugstr_guid(riid), ppvOut);
|
|
|
|
return E_NOTIMPL;
|
|
|
|
}
|
|
|
|
|
2004-08-13 01:00:51 +02:00
|
|
|
static struct IQueryAssociationsVtbl IQueryAssociations_vtbl =
|
2003-07-19 05:07:41 +02:00
|
|
|
{
|
|
|
|
IQueryAssociations_fnQueryInterface,
|
|
|
|
IQueryAssociations_fnAddRef,
|
|
|
|
IQueryAssociations_fnRelease,
|
|
|
|
IQueryAssociations_fnInit,
|
|
|
|
IQueryAssociations_fnGetString,
|
|
|
|
IQueryAssociations_fnGetKey,
|
|
|
|
IQueryAssociations_fnGetData,
|
|
|
|
IQueryAssociations_fnGetEnum
|
|
|
|
};
|