dll/win32/shell32/shell32.cpp

   1 /*
   2  *                 Shell basics
   3  *
   4  * Copyright 1998 Marcus Meissner
   5  * Copyright 1998 Juergen Schmied (jsch)  *  <juergen.schmied@metronet.de>
   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., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  20  */
  21
  22 #include "precomp.h"
  23
  24 #include "shell32_version.h"
  25
  26 WINE_DEFAULT_DEBUG_CHANNEL(shell);
  27
  28 /*
  29  * Implemented
  30  */
  31 EXTERN_C LPWSTR
  32 WINAPI
  33 AddCommasW(DWORD lValue, LPWSTR lpNumber)
  34 {
  35     WCHAR szValue[MAX_PATH], szSeparator[8 + 1];
  36     NUMBERFMTW numFormat;
  37
  38     GetLocaleInfoW(LOCALE_USER_DEFAULT,
  39                    LOCALE_STHOUSAND,
  40                    szSeparator,
  41                    _countof(szSeparator));
  42
  43     numFormat.NumDigits     = 0;
  44     numFormat.LeadingZero   = 0;
  45     numFormat.Grouping      = 3; // FIXME! Use GetLocaleInfoW with LOCALE_SGROUPING and interpret the result.
  46     numFormat.lpDecimalSep  = szSeparator;
  47     numFormat.lpThousandSep = szSeparator;
  48     numFormat.NegativeOrder = 0;
  49
  50     swprintf(szValue, L"%lu", lValue);
  51
  52     if (GetNumberFormatW(LOCALE_USER_DEFAULT,
  53                          0,
  54                          szValue,
  55                          &numFormat,
  56                          lpNumber,
  57                          MAX_PATH) != 0)
  58     {
  59         return lpNumber;
  60     }
  61
  62     wcscpy(lpNumber, szValue);
  63     return lpNumber;
  64 }
  65
  66 /*
  67  * Implemented
  68  */
  69 EXTERN_C BOOL
  70 WINAPI
  71 RegenerateUserEnvironment(LPVOID *lpEnvironment, BOOL bUpdateSelf)
  72 {
  73     HANDLE hUserToken;
  74     if (!OpenProcessToken(GetCurrentProcess(), TOKEN_READ | TOKEN_WRITE, &hUserToken))
  75         return FALSE;
  76
  77     BOOL bResult = CreateEnvironmentBlock(lpEnvironment, hUserToken, TRUE);
  78     if (!bResult || !lpEnvironment)
  79     {
  80         CloseHandle(hUserToken);
  81         return FALSE;
  82     }
  83
  84     if (bUpdateSelf)
  85     {
  86         LPWSTR pszz = (LPWSTR)*lpEnvironment;
  87         if (!pszz)
  88             return FALSE;
  89
  90         while (*pszz)
  91         {
  92             size_t cch = wcslen(pszz);
  93             LPWSTR pchEqual = wcschr(pszz, L'=');
  94             if (pchEqual)
  95             {
  96                 CStringW strName(pszz, pchEqual - pszz);
  97                 SetEnvironmentVariableW(strName, pchEqual + 1);
  98             }
  99             pszz += cch + 1;
 100         }
 101     }
 102
 103     CloseHandle(hUserToken);
 104
 105     return bResult;
 106 }
 107
 108 /**************************************************************************
 109  * Default ClassFactory types
 110  */
 111 typedef HRESULT (CALLBACK *LPFNCREATEINSTANCE)(IUnknown* pUnkOuter, REFIID riid, LPVOID* ppvObject);
 112 HRESULT IDefClF_fnConstructor(LPFNCREATEINSTANCE lpfnCI, PLONG pcRefDll, const IID *riidInst, IClassFactory **theFactory);
 113
 114
 115 /**************************************************************************
 116  * Default ClassFactory Implementation
 117  *
 118  * SHCreateDefClassObject
 119  *
 120  * NOTES
 121  *  Helper function for dlls without their own classfactory.
 122  *  A generic classfactory is returned.
 123  *  When the CreateInstance of the cf is called the callback is executed.
 124  */
 125
 126 class IDefClFImpl :
 127     public CComObjectRootEx<CComMultiThreadModelNoCS>,
 128     public IClassFactory
 129 {
 130 private:
 131     CLSID                    *rclsid;
 132     LPFNCREATEINSTANCE        lpfnCI;
 133     const IID                *riidInst;
 134     LONG                    *pcRefDll;        /* pointer to refcounter in external dll (ugrrr...) */
 135 public:
 136     IDefClFImpl();
 137     HRESULT Initialize(LPFNCREATEINSTANCE lpfnCI, PLONG pcRefDll, const IID *riidInstx);
 138
 139     // IClassFactory
 140     virtual HRESULT WINAPI CreateInstance(IUnknown * pUnkOuter, REFIID riid, LPVOID *ppvObject);
 141     virtual HRESULT WINAPI LockServer(BOOL fLock);
 142
 143 BEGIN_COM_MAP(IDefClFImpl)
 144     COM_INTERFACE_ENTRY_IID(IID_IClassFactory, IClassFactory)
 145 END_COM_MAP()
 146 };
 147
 148 IDefClFImpl::IDefClFImpl()
 149 {
 150     lpfnCI = NULL;
 151     riidInst = NULL;
 152     pcRefDll = NULL;
 153     rclsid = NULL;
 154 }
 155
 156 HRESULT IDefClFImpl::Initialize(LPFNCREATEINSTANCE lpfnCIx, PLONG pcRefDllx, const IID *riidInstx)
 157 {
 158     lpfnCI = lpfnCIx;
 159     riidInst = riidInstx;
 160     pcRefDll = pcRefDllx;
 161
 162     if (pcRefDll)
 163         InterlockedIncrement(pcRefDll);
 164
 165     TRACE("(%p)%s\n", this, shdebugstr_guid(riidInst));
 166     return S_OK;
 167 }
 168
 169 /******************************************************************************
 170  * IDefClF_fnCreateInstance
 171  */
 172 HRESULT WINAPI IDefClFImpl::CreateInstance(IUnknown * pUnkOuter, REFIID riid, LPVOID *ppvObject)
 173 {
 174     TRACE("%p->(%p,%s,%p)\n", this, pUnkOuter, shdebugstr_guid(&riid), ppvObject);
 175
 176     *ppvObject = NULL;
 177
 178     if (riidInst == NULL || IsEqualCLSID(riid, *riidInst) || IsEqualCLSID(riid, IID_IUnknown))
 179     {
 180         return lpfnCI(pUnkOuter, riid, ppvObject);
 181     }
 182
 183     ERR("unknown IID requested %s\n", shdebugstr_guid(&riid));
 184     return E_NOINTERFACE;
 185 }
 186
 187 /******************************************************************************
 188  * IDefClF_fnLockServer
 189  */
 190 HRESULT WINAPI IDefClFImpl::LockServer(BOOL fLock)
 191 {
 192     TRACE("%p->(0x%x), not implemented\n", this, fLock);
 193     return E_NOTIMPL;
 194 }
 195
 196 /**************************************************************************
 197  *  IDefClF_fnConstructor
 198  */
 199
 200 HRESULT IDefClF_fnConstructor(LPFNCREATEINSTANCE lpfnCI, PLONG pcRefDll, const IID *riidInst, IClassFactory **theFactory)
 201 {
 202     return ShellObjectCreatorInit<IDefClFImpl>(lpfnCI, pcRefDll, riidInst, IID_PPV_ARG(IClassFactory, theFactory));
 203 }
 204
 205 /******************************************************************************
 206  * SHCreateDefClassObject            [SHELL32.70]
 207  */
 208 HRESULT WINAPI SHCreateDefClassObject(
 209     REFIID    riid,
 210     LPVOID*    ppv,
 211     LPFNCREATEINSTANCE lpfnCI,    /* [in] create instance callback entry */
 212     LPDWORD    pcRefDll,        /* [in/out] ref count of the dll */
 213     REFIID    riidInst)        /* [in] optional interface to the instance */
 214 {
 215     IClassFactory                *pcf;
 216     HRESULT                        hResult;
 217
 218     TRACE("%s %p %p %p %s\n", shdebugstr_guid(&riid), ppv, lpfnCI, pcRefDll, shdebugstr_guid(&riidInst));
 219
 220     if (!IsEqualCLSID(riid, IID_IClassFactory))
 221         return E_NOINTERFACE;
 222     hResult = IDefClF_fnConstructor(lpfnCI, (PLONG)pcRefDll, &riidInst, &pcf);
 223     if (FAILED(hResult))
 224         return hResult;
 225     *ppv = pcf;
 226     return S_OK;
 227 }
 228
 229 /**************************************************************************
 230  *  CStartMenuDummy
 231  */
 232 class CStartMenuDummy :
 233     public CComCoClass<CStartMenuDummy, &CLSID_StartMenu>,
 234     public CComObjectRootEx<CComMultiThreadModelNoCS>
 235 {
 236 private:
 237     CStartMenuDummy();
 238     virtual ~CStartMenuDummy();
 239
 240 public:
 241     DECLARE_REGISTRY_RESOURCEID(IDR_STARTMENU)
 242
 243     class _CreatorClass
 244     {
 245     public:
 246         static STDMETHODIMP CreateInstance(void *pv, REFIID riid, LPVOID *ppv)
 247         {
 248             if (ppv == NULL)
 249                 return E_POINTER;
 250             *ppv = NULL;
 251             if (pv != NULL)
 252                 return CLASS_E_NOAGGREGATION;
 253             return RSHELL_CStartMenu_CreateInstance(riid, ppv);
 254         }
 255     };
 256 };
 257
 258 /**************************************************************************
 259  *  CShell32Module
 260  */
 261 class CShell32Module : public CComModule
 262 {
 263 public:
 264     void Term()
 265     {
 266         CComCreatorCentralInstance< ATL::CComObject< CDrivesFolder > >::Term();
 267         CComCreatorCentralInstance< ATL::CComObject< CDesktopFolder > >::Term();
 268         CComModule::Term();
 269     }
 270 };
 271
 272
 273 BEGIN_OBJECT_MAP(ObjectMap)
 274     OBJECT_ENTRY(CLSID_ShellFSFolder, CFSFolder)
 275     OBJECT_ENTRY(CLSID_MyComputer, CDrivesFolder)
 276     OBJECT_ENTRY(CLSID_ShellDesktop, CDesktopFolder)
 277     OBJECT_ENTRY(CLSID_ShellItem, CShellItem)
 278     OBJECT_ENTRY(CLSID_ShellLink, CShellLink)
 279     OBJECT_ENTRY(CLSID_Shell, CShellDispatch)
 280     OBJECT_ENTRY(CLSID_DragDropHelper, CDropTargetHelper)
 281     OBJECT_ENTRY(CLSID_ControlPanel, CControlPanelFolder)
 282     OBJECT_ENTRY(CLSID_MyDocuments, CMyDocsFolder)
 283     OBJECT_ENTRY(CLSID_NetworkPlaces, CNetFolder)
 284     OBJECT_ENTRY(CLSID_FontsFolderShortcut, CFontsFolder)
 285     OBJECT_ENTRY(CLSID_Printers, CPrinterFolder)
 286     OBJECT_ENTRY(CLSID_AdminFolderShortcut, CAdminToolsFolder)
 287     OBJECT_ENTRY(CLSID_ShellFldSetExt, CFolderOptions)
 288     OBJECT_ENTRY(CLSID_RecycleBin, CRecycleBin)
 289     OBJECT_ENTRY(CLSID_OpenWithMenu, COpenWithMenu)
 290     OBJECT_ENTRY(CLSID_NewMenu, CNewMenu)
 291     OBJECT_ENTRY(CLSID_StartMenu, CStartMenuDummy)
 292     OBJECT_ENTRY(CLSID_MenuBandSite, CMenuSite)
 293     OBJECT_ENTRY(CLSID_MenuBand, CMenuBand)
 294     OBJECT_ENTRY(CLSID_MenuDeskBar, CMenuDeskBar)
 295     OBJECT_ENTRY(CLSID_MergedFolder, CMergedFolder)
 296     OBJECT_ENTRY(CLSID_ExeDropHandler, CExeDropHandler)
 297     OBJECT_ENTRY(CLSID_QueryAssociations, CQueryAssociations)
 298     OBJECT_ENTRY(CLSID_UserNotification, CUserNotification)
 299 END_OBJECT_MAP()
 300
 301 CShell32Module  gModule;
 302
 303
 304 /***********************************************************************
 305  * DllGetVersion [SHELL32.@]
 306  *
 307  * Retrieves version information of the 'SHELL32.DLL'
 308  *
 309  * PARAMS
 310  *     pdvi [O] pointer to version information structure.
 311  *
 312  * RETURNS
 313  *     Success: S_OK
 314  *     Failure: E_INVALIDARG
 315  *
 316  * NOTES
 317  *     Returns version of a shell32.dll from IE4.01 SP1.
 318  */
 319
 320 STDAPI DllGetVersion(DLLVERSIONINFO *pdvi)
 321 {
 322     /* FIXME: shouldn't these values come from the version resource? */
 323     if (pdvi->cbSize == sizeof(DLLVERSIONINFO) ||
 324         pdvi->cbSize == sizeof(DLLVERSIONINFO2))
 325     {
 326         pdvi->dwMajorVersion = WINE_FILEVERSION_MAJOR;
 327         pdvi->dwMinorVersion = WINE_FILEVERSION_MINOR;
 328         pdvi->dwBuildNumber = WINE_FILEVERSION_BUILD;
 329         pdvi->dwPlatformID = WINE_FILEVERSION_PLATFORMID;
 330         if (pdvi->cbSize == sizeof(DLLVERSIONINFO2))
 331         {
 332             DLLVERSIONINFO2 *pdvi2 = (DLLVERSIONINFO2 *)pdvi;
 333
 334             pdvi2->dwFlags = 0;
 335             pdvi2->ullVersion = MAKEDLLVERULL(WINE_FILEVERSION_MAJOR,
 336                                               WINE_FILEVERSION_MINOR,
 337                                               WINE_FILEVERSION_BUILD,
 338                                               WINE_FILEVERSION_PLATFORMID);
 339         }
 340         TRACE("%u.%u.%u.%u\n",
 341               pdvi->dwMajorVersion, pdvi->dwMinorVersion,
 342               pdvi->dwBuildNumber, pdvi->dwPlatformID);
 343         return S_OK;
 344     }
 345     else
 346     {
 347         WARN("wrong DLLVERSIONINFO size from app\n");
 348         return E_INVALIDARG;
 349     }
 350 }
 351
 352 /*************************************************************************
 353  * global variables of the shell32.dll
 354  * all are once per process
 355  *
 356  */
 357 HINSTANCE   shell32_hInstance;
 358
 359 /*************************************************************************
 360  * SHELL32 DllMain
 361  *
 362  * NOTES
 363  *  calling oleinitialize here breaks sone apps.
 364  */
 365 STDAPI_(BOOL) DllMain(HINSTANCE hInstance, DWORD dwReason, LPVOID fImpLoad)
 366 {
 367     TRACE("%p 0x%x %p\n", hInstance, dwReason, fImpLoad);
 368     if (dwReason == DLL_PROCESS_ATTACH)
 369     {
 370         shell32_hInstance = hInstance;
 371         gModule.Init(ObjectMap, hInstance, &LIBID_Shell32);
 372
 373         DisableThreadLibraryCalls (hInstance);
 374
 375         /* get full path to this DLL for IExtractIconW_fnGetIconLocation() */
 376         GetModuleFileNameW(hInstance, swShell32Name, MAX_PATH);
 377         swShell32Name[MAX_PATH - 1] = '\0';
 378
 379         /* Initialize comctl32 */
 380         INITCOMMONCONTROLSEX InitCtrls;
 381         InitCtrls.dwSize = sizeof(INITCOMMONCONTROLSEX);
 382         InitCtrls.dwICC = ICC_WIN95_CLASSES | ICC_DATE_CLASSES | ICC_USEREX_CLASSES;
 383         InitCommonControlsEx(&InitCtrls);
 384
 385         /* Bad idea, initialization in DllMain! */
 386         InitChangeNotifications();
 387     }
 388     else if (dwReason == DLL_PROCESS_DETACH)
 389     {
 390         shell32_hInstance = NULL;
 391         SIC_Destroy();
 392         FreeChangeNotifications();
 393         gModule.Term();
 394     }
 395     return TRUE;
 396 }
 397
 398 /***********************************************************************
 399  *              DllCanUnloadNow (SHELL32.@)
 400  */
 401 STDAPI DllCanUnloadNow()
 402 {
 403     return gModule.DllCanUnloadNow();
 404 }
 405
 406 /*************************************************************************
 407  *              DllGetClassObject     [SHELL32.@]
 408  *              SHDllGetClassObject   [SHELL32.128]
 409  */
 410 STDAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID *ppv)
 411 {
 412     HRESULT                                hResult;
 413
 414     TRACE("CLSID:%s,IID:%s\n", shdebugstr_guid(&rclsid), shdebugstr_guid(&riid));
 415
 416     hResult = gModule.DllGetClassObject(rclsid, riid, ppv);
 417     TRACE("-- pointer to class factory: %p\n", *ppv);
 418     return hResult;
 419 }
 420
 421 /***********************************************************************
 422  *              DllRegisterServer (SHELL32.@)
 423  */
 424 STDAPI DllRegisterServer()
 425 {
 426     HRESULT hr;
 427
 428     hr = gModule.DllRegisterServer(TRUE);
 429     if (FAILED(hr))
 430         return hr;
 431
 432     hr = gModule.UpdateRegistryFromResource(IDR_FOLDEROPTIONS, TRUE, NULL);
 433     if (FAILED(hr))
 434         return hr;
 435
 436     hr = SHELL_RegisterShellFolders();
 437     if (FAILED(hr))
 438         return hr;
 439
 440     return S_OK;
 441 }
 442
 443 /***********************************************************************
 444  *              DllUnregisterServer (SHELL32.@)
 445  */
 446 STDAPI DllUnregisterServer()
 447 {
 448     HRESULT hr;
 449
 450     hr = gModule.DllUnregisterServer(TRUE);
 451     if (FAILED(hr))
 452         return hr;
 453
 454     hr = gModule.UpdateRegistryFromResource(IDR_FOLDEROPTIONS, FALSE, NULL);
 455     if (FAILED(hr))
 456         return hr;
 457
 458     return S_OK;
 459 }
 460
 461 /*************************************************************************
 462  * DllInstall         [SHELL32.@]
 463  *
 464  * PARAMETERS
 465  *
 466  *    BOOL bInstall - TRUE for install, FALSE for uninstall
 467  *    LPCWSTR pszCmdLine - command line (unused by shell32?)
 468  */
 469
 470 HRESULT WINAPI DllInstall(BOOL bInstall, LPCWSTR cmdline)
 471 {
 472     FIXME("%s %s: stub\n", bInstall ? "TRUE":"FALSE", debugstr_w(cmdline));
 473     return S_OK;        /* indicate success */
 474 }