urlmon: Implemented IUri_GetAbsoluteUri.
[wine] / dlls / shlwapi / assoc.c
1 /*
2  * IQueryAssociations helper functions
3  *
4  * Copyright 2002 Jon Griffiths
5  *
6  * This library is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * This library is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with this library; if not, write to the Free Software
18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
19  */
20 #include <stdarg.h>
21 #include <assert.h>
22
23 #include "windef.h"
24 #include "winbase.h"
25 #include "winnls.h"
26 #include "winreg.h"
27 #include "objbase.h"
28 #include "shlguid.h"
29 #include "shlobj.h"
30 #include "shlwapi.h"
31 #include "wine/unicode.h"
32 #include "wine/debug.h"
33
34 WINE_DEFAULT_DEBUG_CHANNEL(shell);
35
36 /* Default IQueryAssociations::Init() flags */
37 #define SHLWAPI_DEF_ASSOCF (ASSOCF_INIT_BYEXENAME|ASSOCF_INIT_DEFAULTTOSTAR| \
38                             ASSOCF_INIT_DEFAULTTOFOLDER)
39
40 /*************************************************************************
41  * SHLWAPI_ParamAToW
42  *
43  * Internal helper function: Convert ASCII parameter to Unicode.
44  */
45 static BOOL SHLWAPI_ParamAToW(LPCSTR lpszParam, LPWSTR lpszBuff, DWORD dwLen,
46                               LPWSTR* lpszOut)
47 {
48   if (lpszParam)
49   {
50     DWORD dwStrLen = MultiByteToWideChar(CP_ACP, 0, lpszParam, -1, NULL, 0);
51
52     if (dwStrLen < dwLen)
53     {
54       *lpszOut = lpszBuff; /* Use Buffer, it is big enough */
55     }
56     else
57     {
58       /* Create a new buffer big enough for the string */
59       *lpszOut = HeapAlloc(GetProcessHeap(), 0,
60                                    dwStrLen * sizeof(WCHAR));
61       if (!*lpszOut)
62         return FALSE;
63     }
64     MultiByteToWideChar(CP_ACP, 0, lpszParam, -1, *lpszOut, dwStrLen);
65   }
66   else
67     *lpszOut = NULL;
68   return TRUE;
69 }
70
71 /*************************************************************************
72  * AssocCreate  [SHLWAPI.@]
73  *
74  * Create a new IQueryAssociations object.
75  *
76  * PARAMS
77  *  clsid       [I] CLSID of object
78  *  refiid      [I] REFIID of interface
79  *  lpInterface [O] Destination for the created IQueryAssociations object
80  *
81  * RETURNS
82  *  Success: S_OK. lpInterface contains the new object.
83  *  Failure: An HRESULT error code indicating the error.
84  *
85  * NOTES
86  *  clsid  must be equal to CLSID_QueryAssociations and
87  *  refiid must be equal to IID_IQueryAssociations, IID_IUnknown or this function will fail
88  */
89 HRESULT WINAPI AssocCreate(CLSID clsid, REFIID refiid, void **lpInterface)
90 {
91   TRACE("(%s,%s,%p)\n", debugstr_guid(&clsid), debugstr_guid(refiid),
92         lpInterface);
93
94   if (!lpInterface)
95     return E_INVALIDARG;
96
97   *(DWORD*)lpInterface = 0;
98
99   if (!IsEqualGUID(&clsid,  &CLSID_QueryAssociations))
100     return CLASS_E_CLASSNOTAVAILABLE;
101
102   return SHCoCreateInstance( NULL, &clsid, NULL, refiid, lpInterface );
103 }
104
105 /*************************************************************************
106  * AssocQueryKeyW  [SHLWAPI.@]
107  *
108  * See AssocQueryKeyA.
109  */
110 HRESULT WINAPI AssocQueryKeyW(ASSOCF cfFlags, ASSOCKEY assockey, LPCWSTR pszAssoc,
111                               LPCWSTR pszExtra, HKEY *phkeyOut)
112 {
113   HRESULT hRet;
114   IQueryAssociations* lpAssoc;
115
116   TRACE("(0x%8x,0x%8x,%s,%s,%p)\n", cfFlags, assockey, debugstr_w(pszAssoc),
117         debugstr_w(pszExtra), phkeyOut);
118
119   hRet = AssocCreate( CLSID_QueryAssociations, &IID_IQueryAssociations, (void **)&lpAssoc );
120   if (FAILED(hRet)) return hRet;
121
122   cfFlags &= SHLWAPI_DEF_ASSOCF;
123   hRet = IQueryAssociations_Init(lpAssoc, cfFlags, pszAssoc, NULL, NULL);
124
125   if (SUCCEEDED(hRet))
126     hRet = IQueryAssociations_GetKey(lpAssoc, cfFlags, assockey, pszExtra, phkeyOut);
127
128   IQueryAssociations_Release(lpAssoc);
129   return hRet;
130 }
131
132 /*************************************************************************
133  * AssocQueryKeyA  [SHLWAPI.@]
134  *
135  * Get a file association key from the registry.
136  *
137  * PARAMS
138  *  cfFlags  [I] ASSOCF_ flags from "shlwapi.h"
139  *  assockey [I] Type of key to get
140  *  pszAssoc [I] Key name to search below
141  *  pszExtra [I] Extra information about the key location
142  *  phkeyOut [O] Destination for the association key
143  *
144  * RETURNS
145  *  Success: S_OK. phkeyOut contains the key.
146  *  Failure: An HRESULT error code indicating the error.
147  */
148 HRESULT WINAPI AssocQueryKeyA(ASSOCF cfFlags, ASSOCKEY assockey, LPCSTR pszAssoc,
149                               LPCSTR pszExtra, HKEY *phkeyOut)
150 {
151   WCHAR szAssocW[MAX_PATH], *lpszAssocW = NULL;
152   WCHAR szExtraW[MAX_PATH], *lpszExtraW = NULL;
153   HRESULT hRet = E_OUTOFMEMORY;
154
155   TRACE("(0x%8x,0x%8x,%s,%s,%p)\n", cfFlags, assockey, debugstr_a(pszAssoc),
156         debugstr_a(pszExtra), phkeyOut);
157
158   if (SHLWAPI_ParamAToW(pszAssoc, szAssocW, MAX_PATH, &lpszAssocW) &&
159       SHLWAPI_ParamAToW(pszExtra, szExtraW, MAX_PATH, &lpszExtraW))
160   {
161     hRet = AssocQueryKeyW(cfFlags, assockey, lpszAssocW, lpszExtraW, phkeyOut);
162   }
163
164   if (lpszAssocW != szAssocW)
165     HeapFree(GetProcessHeap(), 0, lpszAssocW);
166
167   if (lpszExtraW != szExtraW)
168     HeapFree(GetProcessHeap(), 0, lpszExtraW);
169
170   return hRet;
171 }
172
173 /*************************************************************************
174  * AssocQueryStringW  [SHLWAPI.@]
175  *
176  * See AssocQueryStringA.
177  */
178 HRESULT WINAPI AssocQueryStringW(ASSOCF cfFlags, ASSOCSTR str, LPCWSTR pszAssoc,
179                                  LPCWSTR pszExtra, LPWSTR pszOut, DWORD *pcchOut)
180 {
181   HRESULT hRet;
182   IQueryAssociations* lpAssoc;
183
184   TRACE("(0x%8x,0x%8x,%s,%s,%p,%p)\n", cfFlags, str, debugstr_w(pszAssoc),
185         debugstr_w(pszExtra), pszOut, pcchOut);
186
187   if (!pcchOut)
188     return E_UNEXPECTED;
189
190   hRet = AssocCreate( CLSID_QueryAssociations, &IID_IQueryAssociations, (void **)&lpAssoc );
191   if (FAILED(hRet)) return hRet;
192
193   hRet = IQueryAssociations_Init(lpAssoc, cfFlags & SHLWAPI_DEF_ASSOCF,
194                                  pszAssoc, NULL, NULL);
195
196   if (SUCCEEDED(hRet))
197     hRet = IQueryAssociations_GetString(lpAssoc, cfFlags, str, pszExtra,
198                                         pszOut, pcchOut);
199
200   IQueryAssociations_Release(lpAssoc);
201   return hRet;
202 }
203
204 /*************************************************************************
205  * AssocQueryStringA  [SHLWAPI.@]
206  *
207  * Get a file association string from the registry.
208  *
209  * PARAMS
210  *  cfFlags  [I] ASSOCF_ flags from "shlwapi.h"
211  *  str      [I] Type of string to get (ASSOCSTR enum from "shlwapi.h")
212  *  pszAssoc [I] Key name to search below
213  *  pszExtra [I] Extra information about the string location
214  *  pszOut   [O] Destination for the association string
215  *  pcchOut  [O] Length of pszOut
216  *
217  * RETURNS
218  *  Success: S_OK. pszOut contains the string, pcchOut contains its length.
219  *  Failure: An HRESULT error code indicating the error.
220  */
221 HRESULT WINAPI AssocQueryStringA(ASSOCF cfFlags, ASSOCSTR str, LPCSTR pszAssoc,
222                                  LPCSTR pszExtra, LPSTR pszOut, DWORD *pcchOut)
223 {
224   WCHAR szAssocW[MAX_PATH], *lpszAssocW = NULL;
225   WCHAR szExtraW[MAX_PATH], *lpszExtraW = NULL;
226   HRESULT hRet = E_OUTOFMEMORY;
227
228   TRACE("(0x%8x,0x%8x,%s,%s,%p,%p)\n", cfFlags, str, debugstr_a(pszAssoc),
229         debugstr_a(pszExtra), pszOut, pcchOut);
230
231   if (!pcchOut)
232     hRet = E_UNEXPECTED;
233   else if (SHLWAPI_ParamAToW(pszAssoc, szAssocW, MAX_PATH, &lpszAssocW) &&
234            SHLWAPI_ParamAToW(pszExtra, szExtraW, MAX_PATH, &lpszExtraW))
235   {
236     WCHAR szReturnW[MAX_PATH], *lpszReturnW = szReturnW;
237     DWORD dwLenOut = *pcchOut;
238
239     if (dwLenOut >= MAX_PATH)
240       lpszReturnW = HeapAlloc(GetProcessHeap(), 0,
241                                       (dwLenOut + 1) * sizeof(WCHAR));
242     else
243       dwLenOut = sizeof(szReturnW) / sizeof(szReturnW[0]);
244
245     if (!lpszReturnW)
246       hRet = E_OUTOFMEMORY;
247     else
248     {
249       hRet = AssocQueryStringW(cfFlags, str, lpszAssocW, lpszExtraW,
250                                lpszReturnW, &dwLenOut);
251
252       if (SUCCEEDED(hRet))
253         dwLenOut = WideCharToMultiByte(CP_ACP, 0, lpszReturnW, -1,
254                                        pszOut, *pcchOut, NULL, NULL);
255
256       *pcchOut = dwLenOut;
257       if (lpszReturnW != szReturnW)
258         HeapFree(GetProcessHeap(), 0, lpszReturnW);
259     }
260   }
261
262   if (lpszAssocW != szAssocW)
263     HeapFree(GetProcessHeap(), 0, lpszAssocW);
264   if (lpszExtraW != szExtraW)
265     HeapFree(GetProcessHeap(), 0, lpszExtraW);
266   return hRet;
267 }
268
269 /*************************************************************************
270  * AssocQueryStringByKeyW  [SHLWAPI.@]
271  *
272  * See AssocQueryStringByKeyA.
273  */
274 HRESULT WINAPI AssocQueryStringByKeyW(ASSOCF cfFlags, ASSOCSTR str, HKEY hkAssoc,
275                                       LPCWSTR pszExtra, LPWSTR pszOut,
276                                       DWORD *pcchOut)
277 {
278   HRESULT hRet;
279   IQueryAssociations* lpAssoc;
280
281   TRACE("(0x%8x,0x%8x,%p,%s,%p,%p)\n", cfFlags, str, hkAssoc,
282         debugstr_w(pszExtra), pszOut, pcchOut);
283
284   hRet = AssocCreate( CLSID_QueryAssociations, &IID_IQueryAssociations, (void **)&lpAssoc );
285   if (FAILED(hRet)) return hRet;
286
287   cfFlags &= SHLWAPI_DEF_ASSOCF;
288   hRet = IQueryAssociations_Init(lpAssoc, cfFlags, 0, hkAssoc, NULL);
289
290   if (SUCCEEDED(hRet))
291     hRet = IQueryAssociations_GetString(lpAssoc, cfFlags, str, pszExtra,
292                                         pszOut, pcchOut);
293
294   IQueryAssociations_Release(lpAssoc);
295   return hRet;
296 }
297
298 /*************************************************************************
299  * AssocQueryStringByKeyA  [SHLWAPI.@]
300  *
301  * Get a file association string from the registry, given a starting key.
302  *
303  * PARAMS
304  *  cfFlags  [I] ASSOCF_ flags from "shlwapi.h"
305  *  str      [I] Type of string to get
306  *  hkAssoc  [I] Key to search below
307  *  pszExtra [I] Extra information about the string location
308  *  pszOut   [O] Destination for the association string
309  *  pcchOut  [O] Length of pszOut
310  *
311  * RETURNS
312  *  Success: S_OK. pszOut contains the string, pcchOut contains its length.
313  *  Failure: An HRESULT error code indicating the error.
314  */
315 HRESULT WINAPI AssocQueryStringByKeyA(ASSOCF cfFlags, ASSOCSTR str, HKEY hkAssoc,
316                                       LPCSTR pszExtra, LPSTR pszOut,
317                                       DWORD *pcchOut)
318 {
319   WCHAR szExtraW[MAX_PATH], *lpszExtraW = szExtraW;
320   WCHAR szReturnW[MAX_PATH], *lpszReturnW = szReturnW;
321   HRESULT hRet = E_OUTOFMEMORY;
322
323   TRACE("(0x%8x,0x%8x,%p,%s,%p,%p)\n", cfFlags, str, hkAssoc,
324         debugstr_a(pszExtra), pszOut, pcchOut);
325
326   if (!pcchOut)
327     hRet = E_INVALIDARG;
328   else if (SHLWAPI_ParamAToW(pszExtra, szExtraW, MAX_PATH, &lpszExtraW))
329   {
330     DWORD dwLenOut = *pcchOut;
331     if (dwLenOut >= MAX_PATH)
332       lpszReturnW = HeapAlloc(GetProcessHeap(), 0,
333                                       (dwLenOut + 1) * sizeof(WCHAR));
334
335     if (lpszReturnW)
336     {
337       hRet = AssocQueryStringByKeyW(cfFlags, str, hkAssoc, lpszExtraW,
338                                     lpszReturnW, &dwLenOut);
339
340       if (SUCCEEDED(hRet))
341         WideCharToMultiByte(CP_ACP,0,szReturnW,-1,pszOut,dwLenOut,0,0);
342       *pcchOut = dwLenOut;
343
344       if (lpszReturnW != szReturnW)
345         HeapFree(GetProcessHeap(), 0, lpszReturnW);
346     }
347   }
348
349   if (lpszExtraW != szExtraW)
350     HeapFree(GetProcessHeap(), 0, lpszExtraW);
351   return hRet;
352 }
353
354
355 /**************************************************************************
356  *  AssocIsDangerous  (SHLWAPI.@)
357  *  
358  * Determine if a file association is dangerous (potentially malware).
359  *
360  * PARAMS
361  *  lpszAssoc [I] Name of file or file extension to check.
362  *
363  * RETURNS
364  *  TRUE, if lpszAssoc may potentially be malware (executable),
365  *  FALSE, Otherwise.
366  */
367 BOOL WINAPI AssocIsDangerous(LPCWSTR lpszAssoc)
368 {
369     FIXME("%s\n", debugstr_w(lpszAssoc));
370     return FALSE;
371 }