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/debug.h"
  32
  33 WINE_DEFAULT_DEBUG_CHANNEL(shell);
  34
  35 /* Default IQueryAssociations::Init() flags */
  36 #define SHLWAPI_DEF_ASSOCF (ASSOCF_INIT_BYEXENAME|ASSOCF_INIT_DEFAULTTOSTAR| \
  37                             ASSOCF_INIT_DEFAULTTOFOLDER)
  38
  39 /*************************************************************************
  40  * SHLWAPI_ParamAToW
  41  *
  42  * Internal helper function: Convert ANSI parameter to Unicode.
  43  */
  44 static BOOL SHLWAPI_ParamAToW(LPCSTR lpszParam, LPWSTR lpszBuff, DWORD dwLen,
  45                               LPWSTR* lpszOut)
  46 {
  47   if (lpszParam)
  48   {
  49     DWORD dwStrLen = MultiByteToWideChar(CP_ACP, 0, lpszParam, -1, NULL, 0);
  50
  51     if (dwStrLen < dwLen)
  52     {
  53       *lpszOut = lpszBuff; /* Use Buffer, it is big enough */
  54     }
  55     else
  56     {
  57       /* Create a new buffer big enough for the string */
  58       *lpszOut = malloc(dwStrLen * sizeof(WCHAR));
  59       if (!*lpszOut)
  60         return FALSE;
  61     }
  62     MultiByteToWideChar(CP_ACP, 0, lpszParam, -1, *lpszOut, dwStrLen);
  63   }
  64   else
  65     *lpszOut = NULL;
  66   return TRUE;
  67 }
  68
  69 /*************************************************************************
  70  * AssocCreate  [SHLWAPI.@]
  71  *
  72  * Create a new IQueryAssociations object.
  73  *
  74  * PARAMS
  75  *  clsid       [I] CLSID of object
  76  *  refiid      [I] REFIID of interface
  77  *  lpInterface [O] Destination for the created IQueryAssociations object
  78  *
  79  * RETURNS
  80  *  Success: S_OK. lpInterface contains the new object.
  81  *  Failure: An HRESULT error code indicating the error.
  82  *
  83  * NOTES
  84  *  clsid  must be equal to CLSID_QueryAssociations and
  85  *  refiid must be equal to IID_IQueryAssociations, IID_IUnknown or this function will fail
  86  */
  87 HRESULT WINAPI AssocCreate(CLSID clsid, REFIID refiid, void **lpInterface)
  88 {
  89   TRACE("(%s,%s,%p)\n", debugstr_guid(&clsid), debugstr_guid(refiid),
  90         lpInterface);
  91
  92   if (!lpInterface)
  93     return E_INVALIDARG;
  94
  95   *(DWORD*)lpInterface = 0;
  96
  97   if (!IsEqualGUID(&clsid,  &CLSID_QueryAssociations))
  98     return CLASS_E_CLASSNOTAVAILABLE;
  99
 100   return SHCoCreateInstance( NULL, &clsid, NULL, refiid, lpInterface );
 101 }
 102
 103 /*************************************************************************
 104  * AssocGetPerceivedType  [SHLWAPI.@]
 105  *
 106  * Detect the type of a file by inspecting its extension
 107  *
 108  * PARAMS
 109  *  lpszExt     [I] File extension to evaluate.
 110  *  lpType      [O] Pointer to perceived type
 111  *  lpFlag      [O] Pointer to perceived type flag
 112  *  lppszType   [O] Address to pointer for perceived type text
 113  *
 114  * RETURNS
 115  *  Success: S_OK. lpType and lpFlag contain the perceived type and
 116  *           its information. If lppszType is not NULL, it will point
 117  *           to a string with perceived type text.
 118  *  Failure: An HRESULT error code indicating the error.
 119  *
 120  * NOTES
 121  *  lppszType is optional and it can be NULL.
 122  *  if lpType or lpFlag are NULL, the function will crash.
 123  *  if lpszExt is NULL, an error is returned.
 124  *
 125  * BUGS
 126  *   Unimplemented.
 127  */
 128 HRESULT WINAPI AssocGetPerceivedType(LPCWSTR lpszExt, PERCEIVED *lpType,
 129                                      INT *lpFlag, LPWSTR *lppszType)
 130 {
 131   FIXME("(%s, %p, %p, %p) not supported\n", debugstr_w(lpszExt), lpType, lpFlag, lppszType);
 132
 133   if (lpszExt == NULL)
 134     return HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND);
 135
 136   return E_NOTIMPL;
 137 }
 138
 139 /*************************************************************************
 140  * AssocQueryKeyW  [SHLWAPI.@]
 141  *
 142  * See AssocQueryKeyA.
 143  */
 144 HRESULT WINAPI AssocQueryKeyW(ASSOCF cfFlags, ASSOCKEY assockey, LPCWSTR pszAssoc,
 145                               LPCWSTR pszExtra, HKEY *phkeyOut)
 146 {
 147   HRESULT hRet;
 148   IQueryAssociations* lpAssoc;
 149
 150   TRACE("(0x%lx,%d,%s,%s,%p)\n", cfFlags, assockey, debugstr_w(pszAssoc),
 151         debugstr_w(pszExtra), phkeyOut);
 152
 153   hRet = AssocCreate( CLSID_QueryAssociations, &IID_IQueryAssociations, (void **)&lpAssoc );
 154   if (FAILED(hRet)) return hRet;
 155
 156   cfFlags &= SHLWAPI_DEF_ASSOCF;
 157   hRet = IQueryAssociations_Init(lpAssoc, cfFlags, pszAssoc, NULL, NULL);
 158
 159   if (SUCCEEDED(hRet))
 160     hRet = IQueryAssociations_GetKey(lpAssoc, cfFlags, assockey, pszExtra, phkeyOut);
 161
 162   IQueryAssociations_Release(lpAssoc);
 163   return hRet;
 164 }
 165
 166 /*************************************************************************
 167  * AssocQueryKeyA  [SHLWAPI.@]
 168  *
 169  * Get a file association key from the registry.
 170  *
 171  * PARAMS
 172  *  cfFlags  [I] ASSOCF_ flags from "shlwapi.h"
 173  *  assockey [I] Type of key to get
 174  *  pszAssoc [I] Key name to search below
 175  *  pszExtra [I] Extra information about the key location
 176  *  phkeyOut [O] Destination for the association key
 177  *
 178  * RETURNS
 179  *  Success: S_OK. phkeyOut contains the key.
 180  *  Failure: An HRESULT error code indicating the error.
 181  */
 182 HRESULT WINAPI AssocQueryKeyA(ASSOCF cfFlags, ASSOCKEY assockey, LPCSTR pszAssoc,
 183                               LPCSTR pszExtra, HKEY *phkeyOut)
 184 {
 185   WCHAR szAssocW[MAX_PATH], *lpszAssocW = NULL;
 186   WCHAR szExtraW[MAX_PATH], *lpszExtraW = NULL;
 187   HRESULT hRet = E_OUTOFMEMORY;
 188
 189   TRACE("(0x%lx,%d,%s,%s,%p)\n", cfFlags, assockey, debugstr_a(pszAssoc),
 190         debugstr_a(pszExtra), phkeyOut);
 191
 192   if (SHLWAPI_ParamAToW(pszAssoc, szAssocW, MAX_PATH, &lpszAssocW) &&
 193       SHLWAPI_ParamAToW(pszExtra, szExtraW, MAX_PATH, &lpszExtraW))
 194   {
 195     hRet = AssocQueryKeyW(cfFlags, assockey, lpszAssocW, lpszExtraW, phkeyOut);
 196   }
 197
 198   if (lpszAssocW != szAssocW)
 199     free(lpszAssocW);
 200
 201   if (lpszExtraW != szExtraW)
 202     free(lpszExtraW);
 203
 204   return hRet;
 205 }
 206
 207 /*************************************************************************
 208  * AssocQueryStringW  [SHLWAPI.@]
 209  *
 210  * See AssocQueryStringA.
 211  */
 212 HRESULT WINAPI AssocQueryStringW(ASSOCF cfFlags, ASSOCSTR str, LPCWSTR pszAssoc,
 213                                  LPCWSTR pszExtra, LPWSTR pszOut, DWORD *pcchOut)
 214 {
 215   HRESULT hRet;
 216   IQueryAssociations* lpAssoc;
 217
 218   TRACE("(0x%lx,%d,%s,%s,%p,%p)\n", cfFlags, str, debugstr_w(pszAssoc),
 219         debugstr_w(pszExtra), pszOut, pcchOut);
 220
 221   if (!pcchOut)
 222     return E_UNEXPECTED;
 223
 224   hRet = AssocCreate( CLSID_QueryAssociations, &IID_IQueryAssociations, (void **)&lpAssoc );
 225   if (FAILED(hRet)) return hRet;
 226
 227   hRet = IQueryAssociations_Init(lpAssoc, cfFlags & SHLWAPI_DEF_ASSOCF,
 228                                  pszAssoc, NULL, NULL);
 229
 230   if (SUCCEEDED(hRet))
 231     hRet = IQueryAssociations_GetString(lpAssoc, cfFlags, str, pszExtra,
 232                                         pszOut, pcchOut);
 233
 234   IQueryAssociations_Release(lpAssoc);
 235   return hRet;
 236 }
 237
 238 /*************************************************************************
 239  * AssocQueryStringA  [SHLWAPI.@]
 240  *
 241  * Get a file association string from the registry.
 242  *
 243  * PARAMS
 244  *  cfFlags  [I] ASSOCF_ flags from "shlwapi.h"
 245  *  str      [I] Type of string to get (ASSOCSTR enum from "shlwapi.h")
 246  *  pszAssoc [I] Key name to search below
 247  *  pszExtra [I] Extra information about the string location
 248  *  pszOut   [O] Destination for the association string
 249  *  pcchOut  [O] Length of pszOut
 250  *
 251  * RETURNS
 252  *  Success: S_OK. pszOut contains the string, pcchOut contains its length.
 253  *  Failure: An HRESULT error code indicating the error.
 254  */
 255 HRESULT WINAPI AssocQueryStringA(ASSOCF cfFlags, ASSOCSTR str, LPCSTR pszAssoc,
 256                                  LPCSTR pszExtra, LPSTR pszOut, DWORD *pcchOut)
 257 {
 258   WCHAR szAssocW[MAX_PATH], *lpszAssocW = NULL;
 259   WCHAR szExtraW[MAX_PATH], *lpszExtraW = NULL;
 260   HRESULT hRet = E_OUTOFMEMORY;
 261
 262   TRACE("(0x%lx,0x%d,%s,%s,%p,%p)\n", cfFlags, str, debugstr_a(pszAssoc),
 263         debugstr_a(pszExtra), pszOut, pcchOut);
 264
 265   if (!pcchOut)
 266     hRet = E_UNEXPECTED;
 267   else if (SHLWAPI_ParamAToW(pszAssoc, szAssocW, MAX_PATH, &lpszAssocW) &&
 268            SHLWAPI_ParamAToW(pszExtra, szExtraW, MAX_PATH, &lpszExtraW))
 269   {
 270     WCHAR szReturnW[MAX_PATH], *lpszReturnW = szReturnW;
 271     DWORD dwLenOut = *pcchOut;
 272
 273     if (dwLenOut >= MAX_PATH)
 274       lpszReturnW = malloc((dwLenOut + 1) * sizeof(WCHAR));
 275     else
 276       dwLenOut = ARRAY_SIZE(szReturnW);
 277
 278     if (!lpszReturnW)
 279       hRet = E_OUTOFMEMORY;
 280     else
 281     {
 282       hRet = AssocQueryStringW(cfFlags, str, lpszAssocW, lpszExtraW,
 283                                lpszReturnW, &dwLenOut);
 284
 285       if (SUCCEEDED(hRet))
 286         dwLenOut = WideCharToMultiByte(CP_ACP, 0, lpszReturnW, -1,
 287                                        pszOut, *pcchOut, NULL, NULL);
 288
 289       *pcchOut = dwLenOut;
 290       if (lpszReturnW != szReturnW)
 291         free(lpszReturnW);
 292     }
 293   }
 294
 295   if (lpszAssocW != szAssocW)
 296     free(lpszAssocW);
 297   if (lpszExtraW != szExtraW)
 298     free(lpszExtraW);
 299   return hRet;
 300 }
 301
 302 /*************************************************************************
 303  * AssocQueryStringByKeyW  [SHLWAPI.@]
 304  *
 305  * See AssocQueryStringByKeyA.
 306  */
 307 HRESULT WINAPI AssocQueryStringByKeyW(ASSOCF cfFlags, ASSOCSTR str, HKEY hkAssoc,
 308                                       LPCWSTR pszExtra, LPWSTR pszOut,
 309                                       DWORD *pcchOut)
 310 {
 311   HRESULT hRet;
 312   IQueryAssociations* lpAssoc;
 313
 314   TRACE("(0x%lx,0x%d,%p,%s,%p,%p)\n", cfFlags, str, hkAssoc,
 315         debugstr_w(pszExtra), pszOut, pcchOut);
 316
 317   hRet = AssocCreate( CLSID_QueryAssociations, &IID_IQueryAssociations, (void **)&lpAssoc );
 318   if (FAILED(hRet)) return hRet;
 319
 320   cfFlags &= SHLWAPI_DEF_ASSOCF;
 321   hRet = IQueryAssociations_Init(lpAssoc, cfFlags, 0, hkAssoc, NULL);
 322
 323   if (SUCCEEDED(hRet))
 324     hRet = IQueryAssociations_GetString(lpAssoc, cfFlags, str, pszExtra,
 325                                         pszOut, pcchOut);
 326
 327   IQueryAssociations_Release(lpAssoc);
 328   return hRet;
 329 }
 330
 331 /*************************************************************************
 332  * AssocQueryStringByKeyA  [SHLWAPI.@]
 333  *
 334  * Get a file association string from the registry, given a starting key.
 335  *
 336  * PARAMS
 337  *  cfFlags  [I] ASSOCF_ flags from "shlwapi.h"
 338  *  str      [I] Type of string to get
 339  *  hkAssoc  [I] Key to search below
 340  *  pszExtra [I] Extra information about the string location
 341  *  pszOut   [O] Destination for the association string
 342  *  pcchOut  [O] Length of pszOut
 343  *
 344  * RETURNS
 345  *  Success: S_OK. pszOut contains the string, pcchOut contains its length.
 346  *  Failure: An HRESULT error code indicating the error.
 347  */
 348 HRESULT WINAPI AssocQueryStringByKeyA(ASSOCF cfFlags, ASSOCSTR str, HKEY hkAssoc,
 349                                       LPCSTR pszExtra, LPSTR pszOut,
 350                                       DWORD *pcchOut)
 351 {
 352   WCHAR szExtraW[MAX_PATH], *lpszExtraW = szExtraW;
 353   WCHAR szReturnW[MAX_PATH], *lpszReturnW = szReturnW;
 354   HRESULT hRet = E_OUTOFMEMORY;
 355
 356   TRACE("(0x%lx,0x%d,%p,%s,%p,%p)\n", cfFlags, str, hkAssoc,
 357         debugstr_a(pszExtra), pszOut, pcchOut);
 358
 359   if (!pcchOut)
 360     hRet = E_INVALIDARG;
 361   else if (SHLWAPI_ParamAToW(pszExtra, szExtraW, MAX_PATH, &lpszExtraW))
 362   {
 363     DWORD dwLenOut = *pcchOut;
 364     if (dwLenOut >= MAX_PATH)
 365       lpszReturnW = malloc((dwLenOut + 1) * sizeof(WCHAR));
 366
 367     if (lpszReturnW)
 368     {
 369       hRet = AssocQueryStringByKeyW(cfFlags, str, hkAssoc, lpszExtraW,
 370                                     lpszReturnW, &dwLenOut);
 371
 372       if (SUCCEEDED(hRet))
 373         WideCharToMultiByte(CP_ACP,0,szReturnW,-1,pszOut,dwLenOut,0,0);
 374       *pcchOut = dwLenOut;
 375
 376       if (lpszReturnW != szReturnW)
 377         free(lpszReturnW);
 378     }
 379   }
 380
 381   if (lpszExtraW != szExtraW)
 382     free(lpszExtraW);
 383   return hRet;
 384 }
 385
 386
 387 /**************************************************************************
 388  *  AssocIsDangerous  (SHLWAPI.@)
 389  *
 390  * Determine if a file association is dangerous (potentially malware).
 391  *
 392  * PARAMS
 393  *  lpszAssoc [I] Name of file or file extension to check.
 394  *
 395  * RETURNS
 396  *  TRUE, if lpszAssoc may potentially be malware (executable),
 397  *  FALSE, Otherwise.
 398  */
 399 BOOL WINAPI AssocIsDangerous(LPCWSTR lpszAssoc)
 400 {
 401     FIXME("%s\n", debugstr_w(lpszAssoc));
 402     return FALSE;
 403 }