base/shell/rshell/CStartMenu.cpp

   1 /*
   2  * ReactOS Explorer
   3  *
   4  * Copyright 2014 Giannis Adamopoulos
   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 Street, Fifth Floor, Boston, MA  02110-1301  USA
  19  */
  20 #include "precomp.h"
  21
  22 #include "CMergedFolder.h"
  23
  24 // TODO: declare these GUIDs and interfaces in the right place (whatever that may be)
  25 IID IID_IAugmentedShellFolder = { 0x91EA3F8C, 0xC99B, 0x11D0, { 0x98, 0x15, 0x00, 0xC0, 0x4F, 0xD9, 0x19, 0x72 } };
  26 IID IID_IAugmentedShellFolder2 = { 0x8DB3B3F4, 0x6CFE, 0x11D1, { 0x8A, 0xE9, 0x00, 0xC0, 0x4F, 0xD9, 0x18, 0xD0 } };
  27 CLSID CLSID_MergedFolder = { 0x26FDC864, 0xBE88, 0x46E7, { 0x92, 0x35, 0x03, 0x2D, 0x8E, 0xA5, 0x16, 0x2E } };
  28
  29 //#define TEST_TRACKPOPUPMENU_SUBMENUS
  30
  31
  32 /* NOTE: The following constants *MUST NOT* be changed because
  33          they're hardcoded and need to be the exact values
  34          in order to get the start menu to work! */
  35 #define IDM_RUN                     401
  36 #define IDM_LOGOFF                  402
  37 #define IDM_UNDOCKCOMPUTER          410
  38 #define IDM_TASKBARANDSTARTMENU     413
  39 #define IDM_LASTSTARTMENU_SEPARATOR 450
  40 #define IDM_DOCUMENTS               501
  41 #define IDM_HELPANDSUPPORT          503
  42 #define IDM_PROGRAMS                504
  43 #define IDM_CONTROLPANEL            505
  44 #define IDM_SHUTDOWN                506
  45 #define IDM_FAVORITES               507
  46 #define IDM_SETTINGS                508
  47 #define IDM_PRINTERSANDFAXES        510
  48 #define IDM_SEARCH                  520
  49 #define IDM_SYNCHRONIZE             553
  50 #define IDM_NETWORKCONNECTIONS      557
  51 #define IDM_DISCONNECT              5000
  52 #define IDM_SECURITY                5001
  53
  54 /*
  55  * TODO:
  56  * 1. append the start menu contents from all users
  57  * 2. implement the context menu for start menu entries (programs, control panel, network connetions, printers)
  58  * 3. filter out programs folder from the shell folder part of the start menu
  59  * 4. showing the programs start menu is SLOW compared to windows. this needs some investigation
  60  */
  61
  62 class CShellMenuCallback :
  63     public CComObjectRootEx<CComMultiThreadModelNoCS>,
  64     public IShellMenuCallback
  65 {
  66 private:
  67
  68     HWND m_hwndTray;
  69     CComPtr<IShellMenu> m_pShellMenu;
  70     CComPtr<IBandSite> m_pBandSite;
  71     CComPtr<IDeskBar> m_pDeskBar;
  72     CComPtr<ITrayPriv> m_pTrayPriv;
  73     CComPtr<IShellFolder> m_psfPrograms;
  74
  75     LPITEMIDLIST m_pidlPrograms;
  76
  77     HRESULT OnInitMenu()
  78     {
  79         HMENU hmenu;
  80         HRESULT hr;
  81
  82         if (m_pTrayPriv.p)
  83             return S_OK;
  84
  85         hr = IUnknown_GetSite(m_pDeskBar, IID_PPV_ARG(ITrayPriv, &m_pTrayPriv));
  86         hr = IUnknown_GetWindow(m_pTrayPriv, &m_hwndTray);
  87         hr = m_pTrayPriv->AppendMenuW(&hmenu);
  88 #ifndef TEST_TRACKPOPUPMENU_SUBMENUS
  89         hr = m_pShellMenu->SetMenu(hmenu, NULL, SMSET_BOTTOM);
  90 #else
  91         hr = m_pShellMenu->SetMenu(hmenu, m_hwndTray, SMSET_BOTTOM);
  92 #endif
  93
  94         return hr;
  95     }
  96
  97     HRESULT OnGetInfo(LPSMDATA psmd, SMINFO *psminfo)
  98     {
  99         int iconIndex = 0;
 100
 101         switch (psmd->uId)
 102         {
 103             // Smaller "24x24" icons used for the start menu
 104             // The bitmaps are still 32x32, but the image is centered
 105         case IDM_FAVORITES: iconIndex = -322; break;
 106         case IDM_SEARCH: iconIndex = -323; break;
 107         case IDM_HELPANDSUPPORT: iconIndex = -324; break;
 108         case IDM_LOGOFF: iconIndex = -325; break;
 109         case IDM_PROGRAMS:  iconIndex = -326; break;
 110         case IDM_DOCUMENTS: iconIndex = -327; break;
 111         case IDM_RUN: iconIndex = -328; break;
 112         case IDM_SHUTDOWN: iconIndex = -329; break;
 113         case IDM_SETTINGS: iconIndex = -330; break;
 114
 115         case IDM_CONTROLPANEL: iconIndex = -22; break;
 116         case IDM_NETWORKCONNECTIONS: iconIndex = -257; break;
 117         case IDM_PRINTERSANDFAXES: iconIndex = -138; break;
 118         case IDM_TASKBARANDSTARTMENU: iconIndex = -40; break;
 119         //case IDM_SECURITY: iconIndex = -21; break;
 120         //case IDM_SYNCHRONIZE: iconIndex = -21; break;
 121         //case IDM_DISCONNECT: iconIndex = -21; break;
 122         //case IDM_UNDOCKCOMPUTER: iconIndex = -21; break;
 123         default:
 124             return S_FALSE;
 125         }
 126
 127         if (iconIndex)
 128         {
 129             if ((psminfo->dwMask & SMIM_TYPE) != 0)
 130                 psminfo->dwType = SMIT_STRING;
 131             if ((psminfo->dwMask & SMIM_ICON) != 0)
 132                 psminfo->iIcon = Shell_GetCachedImageIndex(L"shell32.dll", iconIndex, FALSE);
 133             if ((psminfo->dwMask & SMIM_FLAGS) != 0)
 134                 psminfo->dwFlags |= SMIF_ICON;
 135 #ifdef TEST_TRACKPOPUPMENU_SUBMENUS
 136             if ((psminfo->dwMask & SMIM_FLAGS) != 0)
 137                 psminfo->dwFlags |= SMIF_TRACKPOPUP;
 138 #endif
 139         }
 140         else
 141         {
 142             if ((psminfo->dwMask & SMIM_TYPE) != 0)
 143                 psminfo->dwType = SMIT_SEPARATOR;
 144         }
 145         return S_OK;
 146     }
 147
 148     HRESULT OnGetSubMenu(LPSMDATA psmd, REFIID iid, void ** pv)
 149     {
 150         HRESULT hr;
 151         int csidl = 0;
 152         IShellMenu *pShellMenu;
 153
 154 #if USE_SYSTEM_MENUBAND
 155         hr = CoCreateInstance(CLSID_MenuBand,
 156             NULL,
 157             CLSCTX_INPROC_SERVER,
 158             IID_PPV_ARG(IShellMenu, &pShellMenu));
 159 #else
 160         hr = CMenuBand_Constructor(IID_PPV_ARG(IShellMenu, &pShellMenu));
 161 #endif
 162         if (FAILED_UNEXPECTEDLY(hr))
 163             return hr;
 164
 165         hr = pShellMenu->Initialize(this, 0, ANCESTORDEFAULT, SMINIT_VERTICAL);
 166
 167         switch (psmd->uId)
 168         {
 169         case IDM_PROGRAMS:  csidl = CSIDL_PROGRAMS; break;
 170         case IDM_FAVORITES: csidl = CSIDL_FAVORITES; break;
 171         case IDM_DOCUMENTS: csidl = CSIDL_RECENT; break;
 172         }
 173
 174         if (csidl)
 175         {
 176             IShellFolder *psfStartMenu;
 177
 178             if (csidl == CSIDL_PROGRAMS && m_psfPrograms)
 179             {
 180                 psfStartMenu = m_psfPrograms;
 181             }
 182             else
 183             {
 184                 LPITEMIDLIST pidlStartMenu;
 185                 IShellFolder *psfDestop;
 186                 hr = SHGetFolderLocation(NULL, csidl, 0, 0, &pidlStartMenu);
 187                 hr = SHGetDesktopFolder(&psfDestop);
 188                 hr = psfDestop->BindToObject(pidlStartMenu, NULL, IID_PPV_ARG(IShellFolder, &psfStartMenu));
 189             }
 190
 191             hr = pShellMenu->SetShellFolder(psfStartMenu, NULL, NULL, 0);
 192         }
 193         else
 194         {
 195             MENUITEMINFO mii;
 196             mii.cbSize = sizeof(mii);
 197             mii.fMask = MIIM_SUBMENU;
 198             if (GetMenuItemInfoW(psmd->hmenu, psmd->uId, FALSE, &mii))
 199             {
 200                 hr = pShellMenu->SetMenu(mii.hSubMenu, NULL, SMSET_BOTTOM);
 201             }
 202         }
 203         return pShellMenu->QueryInterface(iid, pv);
 204     }
 205
 206     HRESULT OnGetContextMenu(LPSMDATA psmd, REFIID iid, void ** pv)
 207     {
 208         if (psmd->uId == IDM_PROGRAMS ||
 209             psmd->uId == IDM_CONTROLPANEL ||
 210             psmd->uId == IDM_NETWORKCONNECTIONS ||
 211             psmd->uId == IDM_PRINTERSANDFAXES)
 212         {
 213             //UNIMPLEMENTED
 214         }
 215
 216         return S_FALSE;
 217     }
 218
 219     HRESULT OnGetObject(LPSMDATA psmd, REFIID iid, void ** pv)
 220     {
 221         if (IsEqualIID(iid, IID_IShellMenu))
 222             return OnGetSubMenu(psmd, iid, pv);
 223         else if (IsEqualIID(iid, IID_IContextMenu))
 224             return OnGetContextMenu(psmd, iid, pv);
 225
 226         return S_FALSE;
 227     }
 228
 229     HRESULT OnExec(LPSMDATA psmd)
 230     {
 231         // H   H   A    CCC  K   K  ! ! !
 232         // H   H  A A  C   C K  K   ! ! !
 233         // HHHHH AAAAA C     KKK    ! ! !
 234         // H   H A   A C   C K  K   ! ! !
 235         // H   H A   A  CCC  K   K  . . .
 236         //
 237         // HACK: Instead of running explorer.exe with the path, we should be using ShellExecute to "open" the path directly!
 238         // Remove once ShellExecute can handle CLSID path components.
 239
 240         if (psmd->uId == IDM_CONTROLPANEL)
 241             ShellExecuteW(NULL, NULL, L"explorer.exe", L"::{20D04FE0-3AEA-1069-A2D8-08002B30309D}\\::{21EC2020-3AEA-1069-A2DD-08002B30309D}", NULL, SW_SHOWNORMAL);
 242         else if (psmd->uId == IDM_NETWORKCONNECTIONS)
 243             ShellExecuteW(NULL, NULL, L"explorer.exe", L"::{20D04FE0-3AEA-1069-A2D8-08002B30309D}\\::{21EC2020-3AEA-1069-A2DD-08002B30309D}\\::{7007ACC7-3202-11D1-AAD2-00805FC1270E}", NULL, SW_SHOWNORMAL);
 244         else if (psmd->uId == IDM_PRINTERSANDFAXES)
 245             {}/* FIXME: crashes: ShellExecuteW(NULL, NULL, L"explorer.exe", L"::{20D04FE0-3AEA-1069-A2D8-08002B30309D}\\::{21EC2020-3AEA-1069-A2DD-08002B30309D}\\::{2227A280-3AEA-1069-A2DE-08002B30309D}", NULL, SW_SHOWNORMAL);*/
 246         else
 247             PostMessageW(m_hwndTray, WM_COMMAND, psmd->uId, 0);
 248
 249         return S_OK;
 250     }
 251
 252 public:
 253
 254     DECLARE_NOT_AGGREGATABLE(CShellMenuCallback)
 255     DECLARE_PROTECT_FINAL_CONSTRUCT()
 256     BEGIN_COM_MAP(CShellMenuCallback)
 257         COM_INTERFACE_ENTRY_IID(IID_IShellMenuCallback, IShellMenuCallback)
 258     END_COM_MAP()
 259
 260     void Initialize(
 261         IShellMenu* pShellMenu,
 262         IBandSite* pBandSite,
 263         IDeskBar* pDeskBar)
 264     {
 265         m_pShellMenu = pShellMenu;
 266         m_pBandSite = pBandSite;
 267         m_pDeskBar = pDeskBar;
 268     }
 269
 270     ~CShellMenuCallback()
 271     {
 272     }
 273
 274     HRESULT _SetProgramsFolder(IShellFolder * psf, LPITEMIDLIST pidl)
 275     {
 276         m_psfPrograms = psf;
 277         m_pidlPrograms = pidl;
 278         return S_OK;
 279     }
 280
 281     HRESULT STDMETHODCALLTYPE CallbackSM(
 282         LPSMDATA psmd,
 283         UINT uMsg,
 284         WPARAM wParam,
 285         LPARAM lParam)
 286     {
 287         switch (uMsg)
 288         {
 289         case SMC_INITMENU:
 290             return OnInitMenu();
 291         case SMC_GETINFO:
 292             return OnGetInfo(psmd, reinterpret_cast<SMINFO*>(lParam));
 293         case SMC_GETOBJECT:
 294             return OnGetObject(psmd, *reinterpret_cast<IID *>(wParam), reinterpret_cast<void **>(lParam));
 295         case SMC_EXEC:
 296             return OnExec(psmd);
 297         case SMC_SFEXEC:
 298             m_pTrayPriv->Execute(psmd->psf, psmd->pidlItem);
 299             break;
 300         case 0x10000000: // _FilterPIDL from CMenuSFToolbar
 301             if (psmd->psf->CompareIDs(0, psmd->pidlItem, m_pidlPrograms) == 0)
 302                 return S_FALSE;
 303             return S_OK;
 304         }
 305
 306         return S_FALSE;
 307     }
 308 };
 309
 310 HRESULT BindToDesktop(LPCITEMIDLIST pidl, IShellFolder ** ppsfResult)
 311 {
 312     HRESULT hr;
 313     CComPtr<IShellFolder> psfDesktop;
 314
 315     *ppsfResult = NULL;
 316
 317     hr = SHGetDesktopFolder(&psfDesktop);
 318     if (FAILED(hr))
 319         return hr;
 320
 321     hr = psfDesktop->BindToObject(pidl, NULL, IID_PPV_ARG(IShellFolder, ppsfResult));
 322
 323     return hr;
 324 }
 325
 326 static HRESULT GetStartMenuFolder(IShellFolder ** ppsfStartMenu)
 327 {
 328     HRESULT hr;
 329     LPITEMIDLIST pidlUserStartMenu;
 330     LPITEMIDLIST pidlCommonStartMenu;
 331     CComPtr<IShellFolder> psfUserStartMenu;
 332     CComPtr<IShellFolder> psfCommonStartMenu;
 333     CComPtr<IAugmentedShellFolder> pasf;
 334
 335     *ppsfStartMenu = NULL;
 336
 337     hr = SHGetSpecialFolderLocation(NULL, CSIDL_STARTMENU, &pidlUserStartMenu);
 338     if (FAILED(hr))
 339         return hr;
 340
 341     if (FAILED(SHGetSpecialFolderLocation(NULL, CSIDL_COMMON_STARTMENU, &pidlCommonStartMenu)))
 342     {
 343         hr = BindToDesktop(pidlUserStartMenu, ppsfStartMenu);
 344         ILFree(pidlUserStartMenu);
 345         return hr;
 346     }
 347
 348     hr = BindToDesktop(pidlUserStartMenu, &psfUserStartMenu);
 349     if (FAILED_UNEXPECTEDLY(hr))
 350         return hr;
 351
 352     hr = BindToDesktop(pidlCommonStartMenu, &psfCommonStartMenu);
 353     if (FAILED_UNEXPECTEDLY(hr))
 354         return hr;
 355
 356 #if 1
 357     hr = CMergedFolder_Constructor(IID_PPV_ARG(IAugmentedShellFolder, &pasf));
 358 #else
 359     hr = CoCreateInstance(CLSID_MergedFolder, NULL, CLSCTX_INPROC_SERVER, IID_PPV_ARG(IAugmentedShellFolder, &pasf));
 360 #endif
 361     if (FAILED_UNEXPECTEDLY(hr))
 362     {
 363         *ppsfStartMenu = psfUserStartMenu.Detach();
 364         ILFree(pidlCommonStartMenu);
 365         ILFree(pidlUserStartMenu);
 366         return hr;
 367     }
 368
 369     hr = pasf->AddNameSpace(NULL, psfUserStartMenu, pidlUserStartMenu, 0xFF00);
 370     if (FAILED_UNEXPECTEDLY(hr))
 371         return hr;
 372
 373     hr = pasf->AddNameSpace(NULL, psfCommonStartMenu, pidlCommonStartMenu, 0);
 374     if (FAILED_UNEXPECTEDLY(hr))
 375         return hr;
 376
 377     hr = pasf->QueryInterface(IID_PPV_ARG(IShellFolder, ppsfStartMenu));
 378     pasf.Release();
 379
 380     ILFree(pidlCommonStartMenu);
 381     ILFree(pidlUserStartMenu);
 382
 383     return hr;
 384 }
 385
 386 extern "C"
 387 HRESULT WINAPI
 388 CStartMenu_Constructor(REFIID riid, void **ppv)
 389 {
 390     CComPtr<IShellMenu> pShellMenu;
 391     CComPtr<IBandSite> pBandSite;
 392     CComPtr<IDeskBar> pDeskBar;
 393
 394     HRESULT hr;
 395     IShellFolder * psf;
 396
 397     LPITEMIDLIST pidlProgramsAbsolute;
 398     LPITEMIDLIST pidlPrograms;
 399     CComPtr<IShellFolder> psfPrograms;
 400
 401 #if USE_SYSTEM_MENUBAND
 402     hr = CoCreateInstance(CLSID_MenuBand,
 403                           NULL,
 404                           CLSCTX_INPROC_SERVER,
 405                           IID_PPV_ARG(IShellMenu, &pShellMenu));
 406 #else
 407     hr = CMenuBand_Constructor(IID_PPV_ARG(IShellMenu, &pShellMenu));
 408 #endif
 409     if (FAILED_UNEXPECTEDLY(hr))
 410         return hr;
 411
 412 #if USE_SYSTEM_MENUSITE
 413     hr = CoCreateInstance(CLSID_MenuBandSite,
 414                           NULL,
 415                           CLSCTX_INPROC_SERVER,
 416                           IID_PPV_ARG(IBandSite, &pBandSite));
 417 #else
 418     hr = CMenuSite_Constructor(IID_PPV_ARG(IBandSite, &pBandSite));
 419 #endif
 420     if (FAILED_UNEXPECTEDLY(hr))
 421         return hr;
 422
 423 #if USE_SYSTEM_MENUDESKBAR
 424     hr = CoCreateInstance(CLSID_MenuDeskBar,
 425                           NULL,
 426                           CLSCTX_INPROC_SERVER,
 427                           IID_PPV_ARG(IDeskBar, &pDeskBar));
 428 #else
 429     hr = CMenuDeskBar_Constructor(IID_PPV_ARG(IDeskBar, &pDeskBar));
 430 #endif
 431     if (FAILED_UNEXPECTEDLY(hr))
 432         return hr;
 433
 434     CComObject<CShellMenuCallback> *pCallback;
 435     hr = CComObject<CShellMenuCallback>::CreateInstance(&pCallback);
 436     if (FAILED_UNEXPECTEDLY(hr))
 437         return hr;
 438     pCallback->AddRef(); // CreateInstance returns object with 0 ref count */
 439     pCallback->Initialize(pShellMenu, pBandSite, pDeskBar);
 440
 441     pShellMenu->Initialize(pCallback, (UINT) -1, 0, SMINIT_TOPLEVEL | SMINIT_VERTICAL);
 442     if (FAILED_UNEXPECTEDLY(hr))
 443         return hr;
 444
 445     hr = GetStartMenuFolder(&psf);
 446     if (FAILED_UNEXPECTEDLY(hr))
 447         return hr;
 448
 449     hr = SHGetSpecialFolderLocation(NULL, CSIDL_PROGRAMS, &pidlProgramsAbsolute);
 450     if (FAILED_UNEXPECTEDLY(hr))
 451         return hr;
 452
 453     pidlPrograms = ILClone(ILFindLastID(pidlProgramsAbsolute));
 454     ILFree(pidlProgramsAbsolute);
 455
 456     hr = psf->BindToObject(pidlPrograms, NULL, IID_PPV_ARG(IShellFolder, &psfPrograms));
 457     if (FAILED_UNEXPECTEDLY(hr))
 458         return hr;
 459
 460     hr = pCallback->_SetProgramsFolder(psfPrograms, pidlPrograms);
 461     if (FAILED_UNEXPECTEDLY(hr))
 462         return hr;
 463
 464     hr = pShellMenu->SetShellFolder(psf, NULL, NULL, 0);
 465     if (FAILED_UNEXPECTEDLY(hr))
 466         return hr;
 467
 468     hr = pDeskBar->SetClient(pBandSite);
 469     if (FAILED_UNEXPECTEDLY(hr))
 470         return hr;
 471
 472     hr = pBandSite->AddBand(pShellMenu);
 473     if (FAILED_UNEXPECTEDLY(hr))
 474         return hr;
 475
 476     return pDeskBar->QueryInterface(riid, ppv);
 477 }