git.oblomov.eu Git - wine/blob - dlls/shell32/classes.c

   1 /*
   2  *      file type mapping
   3  *      (HKEY_CLASSES_ROOT - Stuff)
   4  *
   5  * Copyright 1998, 1999, 2000 Juergen Schmied
   6  *
   7  * This library is free software; you can redistribute it and/or
   8  * modify it under the terms of the GNU Lesser General Public
   9  * License as published by the Free Software Foundation; either
  10  * version 2.1 of the License, or (at your option) any later version.
  11  *
  12  * This library is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15  * Lesser General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU Lesser General Public
  18  * License along with this library; if not, write to the Free Software
  19  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  20  */
  21
  22 #include "config.h"
  23 #include "wine/port.h"
  24
  25 #include <stdlib.h>
  26 #include <string.h>
  27 #include <stdarg.h>
  28 #include <stdio.h>
  29
  30 #define COBJMACROS
  31
  32 #include "wine/debug.h"
  33 #include "winerror.h"
  34 #include "windef.h"
  35 #include "winbase.h"
  36 #include "winreg.h"
  37 #include "wingdi.h"
  38 #include "winuser.h"
  39
  40 #include "shlobj.h"
  41 #include "shell32_main.h"
  42 #include "shlguid.h"
  43 #include "shresdef.h"
  44 #include "shlwapi.h"
  45 #include "pidl.h"
  46 #include "wine/unicode.h"
  47
  48 WINE_DEFAULT_DEBUG_CHANNEL(shell);
  49
  50 #define MAX_EXTENSION_LENGTH 20
  51
  52 BOOL HCR_MapTypeToValueW(LPCWSTR szExtension, LPWSTR szFileType, LONG len, BOOL bPrependDot)
  53 {
  54         HKEY    hkey;
  55         WCHAR   szTemp[MAX_EXTENSION_LENGTH + 2];
  56
  57         TRACE("%s %p\n", debugstr_w(szExtension), debugstr_w(szFileType));
  58
  59     /* added because we do not want to have double dots */
  60     if (szExtension[0] == '.')
  61         bPrependDot = 0;
  62
  63         if (bPrependDot)
  64           szTemp[0] = '.';
  65
  66         lstrcpynW(szTemp + (bPrependDot?1:0), szExtension, MAX_EXTENSION_LENGTH);
  67
  68         if (RegOpenKeyExW(HKEY_CLASSES_ROOT, szTemp, 0, 0x02000000, &hkey))
  69         {
  70           return FALSE;
  71         }
  72
  73         if (RegQueryValueW(hkey, NULL, szFileType, &len))
  74         {
  75           RegCloseKey(hkey);
  76           return FALSE;
  77         }
  78
  79         RegCloseKey(hkey);
  80
  81         TRACE("--UE;\n} %s\n", debugstr_w(szFileType));
  82
  83         return TRUE;
  84 }
  85
  86 BOOL HCR_MapTypeToValueA(LPCSTR szExtension, LPSTR szFileType, LONG len, BOOL bPrependDot)
  87 {
  88         HKEY    hkey;
  89         char    szTemp[MAX_EXTENSION_LENGTH + 2];
  90
  91         TRACE("%s %p\n", szExtension, szFileType);
  92
  93     /* added because we do not want to have double dots */
  94     if (szExtension[0] == '.')
  95         bPrependDot = 0;
  96
  97         if (bPrependDot)
  98           szTemp[0] = '.';
  99
 100         lstrcpynA(szTemp + (bPrependDot?1:0), szExtension, MAX_EXTENSION_LENGTH);
 101
 102         if (RegOpenKeyExA(HKEY_CLASSES_ROOT, szTemp, 0, 0x02000000, &hkey))
 103         {
 104           return FALSE;
 105         }
 106
 107         if (RegQueryValueA(hkey, NULL, szFileType, &len))
 108         {
 109           RegCloseKey(hkey);
 110           return FALSE;
 111         }
 112
 113         RegCloseKey(hkey);
 114
 115         TRACE("--UE;\n} %s\n", szFileType);
 116
 117         return TRUE;
 118 }
 119
 120
 121 BOOL HCR_GetExecuteCommandW( HKEY hkeyClass, LPCWSTR szClass, LPCWSTR szVerb, LPWSTR szDest, DWORD len )
 122 {
 123         static const WCHAR swShell[] = {'s','h','e','l','l','\\',0};
 124         static const WCHAR swCommand[] = {'\\','c','o','m','m','a','n','d',0};
 125         BOOL    ret = FALSE;
 126
 127         TRACE("%p %s %s %p\n", hkeyClass, debugstr_w(szClass), debugstr_w(szVerb), szDest);
 128
 129         if (szClass)
 130             RegOpenKeyExW(HKEY_CLASSES_ROOT, szClass, 0, 0x02000000, &hkeyClass);
 131
 132         if (hkeyClass)
 133         {
 134             WCHAR sTemp[MAX_PATH];
 135             lstrcpyW(sTemp, swShell);
 136             lstrcatW(sTemp, szVerb);
 137             lstrcatW(sTemp, swCommand);
 138
 139             ret = (ERROR_SUCCESS == SHGetValueW(hkeyClass, sTemp, NULL, NULL, szDest, &len));
 140
 141             if (szClass)
 142                RegCloseKey(hkeyClass);
 143         }
 144
 145         TRACE("-- %s\n", debugstr_w(szDest) );
 146         return ret;
 147 }
 148
 149 /***************************************************************************************
 150 *       HCR_GetDefaultIcon      [internal]
 151 *
 152 * Gets the icon for a filetype
 153 */
 154 static BOOL HCR_RegOpenClassIDKey(REFIID riid, HKEY *hkey)
 155 {
 156         char    xriid[50];
 157     sprintf( xriid, "CLSID\\{%08lx-%04x-%04x-%02x%02x-%02x%02x%02x%02x%02x%02x}",
 158                  riid->Data1, riid->Data2, riid->Data3,
 159                  riid->Data4[0], riid->Data4[1], riid->Data4[2], riid->Data4[3],
 160                  riid->Data4[4], riid->Data4[5], riid->Data4[6], riid->Data4[7] );
 161
 162         TRACE("%s\n",xriid );
 163
 164         return !RegOpenKeyExA(HKEY_CLASSES_ROOT, xriid, 0, KEY_READ, hkey);
 165 }
 166
 167 static BOOL HCR_RegGetDefaultIconW(HKEY hkey, LPWSTR szDest, DWORD len, LPDWORD dwNr)
 168 {
 169         DWORD dwType;
 170         WCHAR sTemp[MAX_PATH];
 171         WCHAR sNum[5];
 172
 173         if (!RegQueryValueExW(hkey, NULL, 0, &dwType, (LPBYTE)szDest, &len))
 174         {
 175       if (dwType == REG_EXPAND_SZ)
 176           {
 177             ExpandEnvironmentStringsW(szDest, sTemp, MAX_PATH);
 178             lstrcpynW(szDest, sTemp, len);
 179           }
 180           if (ParseFieldW (szDest, 2, sNum, 5))
 181              *dwNr = atoiW(sNum);
 182           else
 183              *dwNr=0; /* sometimes the icon number is missing */
 184           ParseFieldW (szDest, 1, szDest, len);
 185           PathUnquoteSpacesW(szDest);
 186           return TRUE;
 187         }
 188         return FALSE;
 189 }
 190
 191 static BOOL HCR_RegGetDefaultIconA(HKEY hkey, LPSTR szDest, DWORD len, LPDWORD dwNr)
 192 {
 193         DWORD dwType;
 194         char sTemp[MAX_PATH];
 195         char  sNum[5];
 196
 197         if (!RegQueryValueExA(hkey, NULL, 0, &dwType, szDest, &len))
 198         {
 199       if (dwType == REG_EXPAND_SZ)
 200           {
 201             ExpandEnvironmentStringsA(szDest, sTemp, MAX_PATH);
 202             lstrcpynA(szDest, sTemp, len);
 203           }
 204           if (ParseFieldA (szDest, 2, sNum, 5))
 205              *dwNr=atoi(sNum);
 206           else
 207              *dwNr=0; /* sometimes the icon number is missing */
 208           ParseFieldA (szDest, 1, szDest, len);
 209           PathUnquoteSpacesA(szDest);
 210           return TRUE;
 211         }
 212         return FALSE;
 213 }
 214
 215 BOOL HCR_GetDefaultIconW(LPCWSTR szClass, LPWSTR szDest, DWORD len, LPDWORD dwNr)
 216 {
 217         static const WCHAR swDefaultIcon[] = {'\\','D','e','f','a','u','l','t','I','c','o','n',0};
 218         HKEY    hkey;
 219         WCHAR   sTemp[MAX_PATH];
 220         BOOL    ret = FALSE;
 221
 222         TRACE("%s\n",debugstr_w(szClass) );
 223
 224         lstrcpynW(sTemp, szClass, MAX_PATH);
 225         lstrcatW(sTemp, swDefaultIcon);
 226
 227         if (!RegOpenKeyExW(HKEY_CLASSES_ROOT, sTemp, 0, 0x02000000, &hkey))
 228         {
 229           ret = HCR_RegGetDefaultIconW(hkey, szDest, len, dwNr);
 230           RegCloseKey(hkey);
 231         }
 232
 233         if(ret)
 234             TRACE("-- %s %li\n", debugstr_w(szDest), *dwNr );
 235         else
 236             TRACE("-- not found\n");
 237
 238         return ret;
 239 }
 240
 241 BOOL HCR_GetDefaultIconA(LPCSTR szClass, LPSTR szDest, DWORD len, LPDWORD dwNr)
 242 {
 243         HKEY    hkey;
 244         char    sTemp[MAX_PATH];
 245         BOOL    ret = FALSE;
 246
 247         TRACE("%s\n",szClass );
 248
 249         sprintf(sTemp, "%s\\DefaultIcon",szClass);
 250
 251         if (!RegOpenKeyExA(HKEY_CLASSES_ROOT, sTemp, 0, 0x02000000, &hkey))
 252         {
 253           ret = HCR_RegGetDefaultIconA(hkey, szDest, len, dwNr);
 254           RegCloseKey(hkey);
 255         }
 256         TRACE("-- %s %li\n", szDest, *dwNr );
 257         return ret;
 258 }
 259
 260 BOOL HCR_GetDefaultIconFromGUIDW(REFIID riid, LPWSTR szDest, DWORD len, LPDWORD dwNr)
 261 {
 262         HKEY    hkey;
 263         BOOL    ret = FALSE;
 264
 265         if (HCR_RegOpenClassIDKey(riid, &hkey))
 266         {
 267           ret = HCR_RegGetDefaultIconW(hkey, szDest, len, dwNr);
 268           RegCloseKey(hkey);
 269         }
 270         TRACE("-- %s %li\n", debugstr_w(szDest), *dwNr );
 271         return ret;
 272 }
 273
 274 /***************************************************************************************
 275 *       HCR_GetClassName        [internal]
 276 *
 277 * Gets the name of a registered class
 278 */
 279 static const WCHAR swEmpty[] = {0};
 280
 281 BOOL HCR_GetClassNameW(REFIID riid, LPWSTR szDest, DWORD len)
 282 {
 283         HKEY    hkey;
 284         BOOL ret = FALSE;
 285         DWORD buflen = len;
 286
 287         szDest[0] = 0;
 288         if (HCR_RegOpenClassIDKey(riid, &hkey))
 289         {
 290           if (!RegQueryValueExW(hkey, swEmpty, 0, NULL, (LPBYTE)szDest, &len))
 291           {
 292             ret = TRUE;
 293           }
 294           RegCloseKey(hkey);
 295         }
 296
 297         if (!ret || !szDest[0])
 298         {
 299           if(IsEqualIID(riid, &CLSID_ShellDesktop))
 300           {
 301             if (LoadStringW(shell32_hInstance, IDS_DESKTOP, szDest, buflen))
 302               ret = TRUE;
 303           }
 304           else if (IsEqualIID(riid, &CLSID_MyComputer))
 305           {
 306             if(LoadStringW(shell32_hInstance, IDS_MYCOMPUTER, szDest, buflen))
 307               ret = TRUE;
 308           }
 309         }
 310         TRACE("-- %s\n", debugstr_w(szDest));
 311         return ret;
 312 }
 313
 314 BOOL HCR_GetClassNameA(REFIID riid, LPSTR szDest, DWORD len)
 315 {       HKEY    hkey;
 316         BOOL ret = FALSE;
 317         DWORD buflen = len;
 318
 319         szDest[0] = 0;
 320         if (HCR_RegOpenClassIDKey(riid, &hkey))
 321         {
 322           if (!RegQueryValueExA(hkey,"",0,NULL,szDest,&len))
 323           {
 324             ret = TRUE;
 325           }
 326           RegCloseKey(hkey);
 327         }
 328
 329         if (!ret || !szDest[0])
 330         {
 331           if(IsEqualIID(riid, &CLSID_ShellDesktop))
 332           {
 333             if (LoadStringA(shell32_hInstance, IDS_DESKTOP, szDest, buflen))
 334               ret = TRUE;
 335           }
 336           else if (IsEqualIID(riid, &CLSID_MyComputer))
 337           {
 338             if(LoadStringA(shell32_hInstance, IDS_MYCOMPUTER, szDest, buflen))
 339               ret = TRUE;
 340           }
 341         }
 342
 343         TRACE("-- %s\n", szDest);
 344
 345         return ret;
 346 }
 347
 348 /******************************************************************************
 349  * HCR_GetFolderAttributes [Internal]
 350  *
 351  * Query the registry for a shell folders' attributes
 352  *
 353  * PARAMS
 354  *  pidlFolder    [I]  A simple pidl of type PT_GUID.
 355  *  pdwAttributes [IO] In: Attributes to be queried, OUT: Resulting attributes.
 356  *
 357  * RETURNS
 358  *  TRUE:  Found information for the attributes in the registry
 359  *  FALSE: No attribute information found
 360  *
 361  * NOTES
 362  *  If queried for an attribute, which is set in the CallForAttributes registry
 363  *  value, the function binds to the shellfolder objects and queries it.
 364  */
 365 BOOL HCR_GetFolderAttributes(LPCITEMIDLIST pidlFolder, LPDWORD pdwAttributes)
 366 {
 367     HKEY hSFKey;
 368     LPOLESTR pwszCLSID;
 369     LONG lResult;
 370     DWORD dwTemp, dwLen;
 371     static const WCHAR wszAttributes[] = { 'A','t','t','r','i','b','u','t','e','s',0 };
 372     static const WCHAR wszCallForAttributes[] = {
 373         'C','a','l','l','F','o','r','A','t','t','r','i','b','u','t','e','s',0 };
 374     WCHAR wszShellFolderKey[] = { 'C','L','S','I','D','\\','{','0','0','0','2','1','4','0','0','-',
 375         '0','0','0','0','-','0','0','0','0','-','C','0','0','0','-','0','0','0','0','0','0','0',
 376         '0','0','0','4','6','}','\\','S','h','e','l','l','F','o','l','d','e','r',0 };
 377
 378     TRACE("(pidlFolder=%p, pdwAttributes=%p)\n", pidlFolder, pdwAttributes);
 379
 380     if (!_ILIsPidlSimple(pidlFolder)) {
 381         ERR("HCR_GetFolderAttributes should be called for simple PIDL's only!\n");
 382         return FALSE;
 383     }
 384
 385     if (!_ILIsDesktop(pidlFolder)) {
 386         if (FAILED(StringFromCLSID(_ILGetGUIDPointer(pidlFolder), &pwszCLSID))) return FALSE;
 387         memcpy(&wszShellFolderKey[6], pwszCLSID, 38 * sizeof(WCHAR));
 388         CoTaskMemFree(pwszCLSID);
 389     }
 390
 391     lResult = RegOpenKeyExW(HKEY_CLASSES_ROOT, wszShellFolderKey, 0, KEY_READ, &hSFKey);
 392     if (lResult != ERROR_SUCCESS) return FALSE;
 393
 394     dwLen = sizeof(DWORD);
 395     lResult = RegQueryValueExW(hSFKey, wszCallForAttributes, 0, NULL, (LPBYTE)&dwTemp, &dwLen);
 396     if ((lResult == ERROR_SUCCESS) && (dwTemp & *pdwAttributes)) {
 397         LPSHELLFOLDER psfDesktop, psfFolder;
 398         HRESULT hr;
 399
 400         RegCloseKey(hSFKey);
 401         hr = SHGetDesktopFolder(&psfDesktop);
 402         if (SUCCEEDED(hr)) {
 403             hr = IShellFolder_BindToObject(psfDesktop, pidlFolder, NULL, &IID_IShellFolder,
 404                                            (LPVOID*)&psfFolder);
 405             if (SUCCEEDED(hr)) {
 406                 hr = IShellFolder_GetAttributesOf(psfFolder, 0, NULL, pdwAttributes);
 407             }
 408         }
 409         IShellFolder_Release(psfFolder);
 410         IShellFolder_Release(psfDesktop);
 411         if (FAILED(hr)) return FALSE;
 412     } else {
 413         lResult = RegQueryValueExW(hSFKey, wszAttributes, 0, NULL, (LPBYTE)&dwTemp, &dwLen);
 414         RegCloseKey(hSFKey);
 415         if (lResult == ERROR_SUCCESS) {
 416             *pdwAttributes &= dwTemp;
 417         } else {
 418             return FALSE;
 419         }
 420     }
 421
 422     TRACE("-- *pdwAttributes == 0x%08lx\n", *pdwAttributes);
 423
 424     return TRUE;
 425 }