Leaked source code of windows server 2003
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

3903 lines
131 KiB

#include "stdafx.h"
#pragma hdrstop
#include "bitbuck.h"
#include <iethread.h>
#include "apithk.h"
#include "mtpt.h"
#include "..\util.h"
#include "fassoc.h"
#include "..\filetbl.h"
#define DM_FOCUS 0 // focus
#define DM_SHUTDOWN DM_TRACE // shutdown
#define TF_SHDAUTO 0
#define DM_MISC DM_TRACE // misc/tmp
#define IDT_STARTBACKGROUNDSHELLTASKS 7
#define IDT_TASKBARWAKEUP 8
#define ENABLE_CHANNELS
#define DESKTOPCLASS
STDAPI_(void) CheckWinIniForAssocs();
BOOL GetOldWorkAreas(LPRECT lprc, DWORD* pdwNoOfOldWA);
void SaveOldWorkAreas(LPCRECT lprc, DWORD nOldWA);
BOOL UpdateAllDesktopSubscriptions(IADesktopP2 *);
//This is in deskreg.cpp
BOOL AdjustDesktopComponents(LPCRECT prcNewWorkAreas, int nNewWorkAreas,
LPCRECT prcOldMonitors, LPCRECT prcOldWorkAreas, int nOldWorkAreas);
// in defview.cpp
BOOL IsFolderWindow(HWND hwnd);
// copied from tray.c if changing here, change there also
#define GHID_FIRST 500
#define g_xVirtualScreen GetSystemMetrics(SM_XVIRTUALSCREEN)
#define g_yVirtualScreen GetSystemMetrics(SM_YVIRTUALSCREEN)
#define g_cxVirtualScreen GetSystemMetrics(SM_CXVIRTUALSCREEN)
#define g_cyVirtualScreen GetSystemMetrics(SM_CYVIRTUALSCREEN)
#define g_cxEdge GetSystemMetrics(SM_CXEDGE)
#define g_cyEdge GetSystemMetrics(SM_CYEDGE)
// TOID_Desktop 6aec6a60-b7a4-11d1-be89-0000f805ca57 is the id for ShellTasks added by the desktop
const GUID TOID_Desktop = { 0x6aec6a60, 0xb7a4, 0x11d1, {0xbe, 0x89, 0x00, 0x00, 0xf8, 0x05, 0xca, 0x57} };
// these are the CLSIDs that are supported for creating LocalServer thread
// objects. the shell supports these in the RunDll32 and then the
// invocation of a thread on the desktop object.
CLSID const *c_localServers[] =
{
&CLSID_ShellDesktop,
&CLSID_NetCrawler,
&CLSID_HWShellExecute,
&CLSID_ShellAutoplay,
};
typedef struct
{
INT iLocalServer; // index into the local server table
DWORD *pdwThreadID; // where to stash the thread id
} LOCALSERVERDATA;
// Private interface to talk to explorer.exe
IDeskTray* g_pdtray = NULL;
void FireEventSz(LPCTSTR szEvent)
{
HANDLE hEvent = OpenEvent(EVENT_MODIFY_STATE, FALSE, szEvent);
if (hEvent)
{
SetEvent(hEvent);
CloseHandle(hEvent);
}
}
#define PERF_ENABLESETMARK
#ifdef PERF_ENABLESETMARK
void DoSetMark(LPCSTR pszMark, ULONG cbSz);
#define PERFSETMARK(text) DoSetMark(text, sizeof(text))
#else
#define PERFSETMARK(text)
#endif // PERF_ENABLESETMARK
#ifdef PERF_ENABLESETMARK
#include <wmistr.h>
#include <ntwmi.h> // PWMI_SET_MARK_INFORMATION is defined in ntwmi.h
#include <wmiumkm.h>
#define NTPERF
#include <ntperf.h>
void DoSetMark(LPCSTR pszMark, ULONG cbSz)
{
PWMI_SET_MARK_INFORMATION MarkInfo;
HANDLE hTemp;
ULONG cbBufferSize;
ULONG cbReturnSize;
cbBufferSize = FIELD_OFFSET(WMI_SET_MARK_INFORMATION, Mark) + cbSz;
MarkInfo = (PWMI_SET_MARK_INFORMATION) LocalAlloc(LPTR, cbBufferSize);
// Failed to init, no big deal
if (MarkInfo == NULL)
return;
BYTE *pMarkBuffer = (BYTE *) (&MarkInfo->Mark[0]);
memcpy(pMarkBuffer, pszMark, cbSz);
// WMI_SET_MARK_WITH_FLUSH will flush the working set when setting the mark
MarkInfo->Flag = PerformanceMmInfoMark;
hTemp = CreateFile(WMIDataDeviceName,
GENERIC_READ | GENERIC_WRITE,
0,
NULL,
OPEN_EXISTING,
FILE_ATTRIBUTE_NORMAL |
FILE_FLAG_OVERLAPPED,
NULL);
if (hTemp != INVALID_HANDLE_VALUE)
{
// here's the piece that actually puts the mark in the buffer
BOOL fIoctlSuccess = DeviceIoControl(hTemp,
IOCTL_WMI_SET_MARK,
MarkInfo,
cbBufferSize,
NULL,
0,
&cbReturnSize,
NULL);
CloseHandle(hTemp);
}
LocalFree(MarkInfo);
}
#endif // PERF_ENABLESETMARK
// MISC stuff duplicated in browseui {
HRESULT _ConvertPathToPidlW(IBrowserService2 *pbs, HWND hwnd, LPCWSTR pszPath, LPITEMIDLIST * ppidl)
{
WCHAR wszCmdLine[MAX_URL_STRING]; // must be with pszPath
TCHAR szFixedUrl[MAX_URL_STRING];
TCHAR szParsedUrl[MAX_URL_STRING] = {'\0'};
DWORD dwUrlLen = ARRAYSIZE(szParsedUrl);
// Copy the command line into a temporary buffer
// so we can remove the surrounding quotes (if
// they exist)
SHUnicodeToTChar(pszPath, szFixedUrl, ARRAYSIZE(szFixedUrl));
PathUnquoteSpaces(szFixedUrl);
HRESULT hr = S_OK;
if (ParseURLFromOutsideSource(szFixedUrl, szParsedUrl, &dwUrlLen, NULL))
SHTCharToUnicode(szParsedUrl, wszCmdLine, ARRAYSIZE(wszCmdLine));
else
hr = StringCchCopy(wszCmdLine, ARRAYSIZE(wszCmdLine), pszPath);
if (SUCCEEDED(hr))
{
hr = pbs->IEParseDisplayName(CP_ACP, wszCmdLine, ppidl);
}
pbs->DisplayParseError(hr, wszCmdLine);
return hr;
}
// END of MISC stuff duplicated in browseui }
// Several places rely on the fact that IShellBrowser is the first interface
// we inherit (and therefore is what we use as our canonical IUnknown).
// Grep for IUnknownIdentity to find them.
class CDesktopBrowser :
public IShellBrowser
,public IServiceProvider
,public IOleCommandTarget
,public IDockingWindowSite
,public IInputObjectSite
,public IDropTarget
,public IDockingWindowFrame
,public IMultiMonitorDockingSite
,public IBrowserService2
,public IShellBrowserService
{
public:
// IUnknown
STDMETHOD(QueryInterface)(REFIID riid, void * *ppvObj);
virtual STDMETHODIMP_(ULONG) AddRef(void);
virtual STDMETHODIMP_(ULONG) Release(void);
// IShellBrowser (same as IOleInPlaceFrame)
virtual STDMETHODIMP GetWindow(HWND * lphwnd);
virtual STDMETHODIMP ContextSensitiveHelp(BOOL fEnterMode);
virtual STDMETHODIMP InsertMenusSB(HMENU hmenuShared, LPOLEMENUGROUPWIDTHS lpMenuWidths);
virtual STDMETHODIMP SetMenuSB(HMENU hmenuShared, HOLEMENU holemenu, HWND hwnd);
virtual STDMETHODIMP RemoveMenusSB(HMENU hmenuShared);
virtual STDMETHODIMP SetStatusTextSB(LPCOLESTR lpszStatusText);
virtual STDMETHODIMP EnableModelessSB(BOOL fEnable);
virtual STDMETHODIMP TranslateAcceleratorSB(LPMSG lpmsg, WORD wID);
virtual STDMETHODIMP BrowseObject(LPCITEMIDLIST pidl, UINT wFlags);
virtual STDMETHODIMP GetViewStateStream(DWORD grfMode, LPSTREAM *ppStrm);
virtual STDMETHODIMP GetControlWindow(UINT id, HWND * lphwnd);
virtual STDMETHODIMP SendControlMsg(UINT id, UINT uMsg, WPARAM wParam, LPARAM lParam, LRESULT *pret);
virtual STDMETHODIMP QueryActiveShellView(struct IShellView ** ppshv);
virtual STDMETHODIMP OnViewWindowActive(struct IShellView * ppshv);
virtual STDMETHODIMP SetToolbarItems(LPTBBUTTON lpButtons, UINT nButtons, UINT uFlags);
// IServiceProvider
virtual STDMETHODIMP QueryService(REFGUID guidService, REFIID riid, void **ppvObj);
// IOleCommandTarget
virtual STDMETHODIMP QueryStatus(const GUID *pguidCmdGroup, ULONG cCmds, OLECMD rgCmds[], OLECMDTEXT *pcmdtext);
virtual STDMETHODIMP Exec(const GUID *pguidCmdGroup, DWORD nCmdID, DWORD nCmdexecopt, VARIANTARG *pvarargIn, VARIANTARG *pvarargOut);
// IDockingWindowSite (also IOleWindow)
virtual STDMETHODIMP GetBorderDW(IUnknown* punkSrc, LPRECT lprectBorder);
virtual STDMETHODIMP RequestBorderSpaceDW(IUnknown* punkSrc, LPCBORDERWIDTHS pborderwidths);
virtual STDMETHODIMP SetBorderSpaceDW(IUnknown* punkSrc, LPCBORDERWIDTHS pborderwidths);
// IInputObjectSite
virtual STDMETHODIMP OnFocusChangeIS(IUnknown* punkSrc, BOOL fSetFocus);
// IDropTarget
virtual STDMETHODIMP DragEnter(IDataObject *pdtobj, DWORD grfKeyState, POINTL pt, DWORD *pdwEffect);
virtual STDMETHODIMP DragOver(DWORD grfKeyState, POINTL pt, DWORD *pdwEffect);
virtual STDMETHODIMP DragLeave(void);
virtual STDMETHODIMP Drop(IDataObject *pdtobj, DWORD grfKeyState, POINTL pt, DWORD *pdwEffect);
// IDockingWindowFrame (also IOleWindow)
virtual STDMETHODIMP AddToolbar(IUnknown* punkSrc, LPCWSTR pwszItem, DWORD dwReserved);
virtual STDMETHODIMP RemoveToolbar(IUnknown* punkSrc, DWORD dwFlags);
virtual STDMETHODIMP FindToolbar(LPCWSTR pwszItem, REFIID riid, void **ppvObj);
// IMultiMonitorDockingSite
virtual STDMETHODIMP GetMonitor(IUnknown* punkSrc, HMONITOR * phMon);
virtual STDMETHODIMP RequestMonitor(IUnknown* punkSrc, HMONITOR * phMon);
virtual STDMETHODIMP SetMonitor(IUnknown* punkSrc, HMONITOR hMon, HMONITOR * phMonOld);
static LRESULT CALLBACK DesktopWndProc(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam);
static LRESULT CALLBACK RaisedWndProc(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam);
void _MessageLoop();
HWND GetTrayWindow(void) { return _hwndTray; }
HWND GetDesktopWindow(void) { return _pbbd->_hwnd; }
// IBrowserService
// *** IBrowserService specific methods ***
virtual STDMETHODIMP GetParentSite(IOleInPlaceSite** ppipsite);
virtual STDMETHODIMP SetTitle(IShellView* psv, LPCWSTR pszName);
virtual STDMETHODIMP GetTitle(IShellView* psv, LPWSTR pszName, DWORD cchName);
virtual STDMETHODIMP GetOleObject( IOleObject** ppobjv);
virtual STDMETHODIMP GetTravelLog(ITravelLog** pptl);
virtual STDMETHODIMP ShowControlWindow(UINT id, BOOL fShow);
virtual STDMETHODIMP IsControlWindowShown(UINT id, BOOL *pfShown);
virtual STDMETHODIMP IEGetDisplayName(LPCITEMIDLIST pidl, LPWSTR pwszName, UINT uFlags);
virtual STDMETHODIMP IEParseDisplayName(UINT uiCP, LPCWSTR pwszPath, LPITEMIDLIST * ppidlOut);
virtual STDMETHODIMP DisplayParseError(HRESULT hres, LPCWSTR pwszPath);
virtual STDMETHODIMP NavigateToPidl(LPCITEMIDLIST pidl, DWORD grfHLNF);
virtual STDMETHODIMP SetNavigateState(BNSTATE bnstate);
virtual STDMETHODIMP GetNavigateState (BNSTATE *pbnstate);
virtual STDMETHODIMP NotifyRedirect ( IShellView* psv, LPCITEMIDLIST pidl, BOOL *pfDidBrowse);
virtual STDMETHODIMP UpdateWindowList ();
virtual STDMETHODIMP UpdateBackForwardState ();
virtual STDMETHODIMP SetFlags(DWORD dwFlags, DWORD dwFlagMask);
virtual STDMETHODIMP GetFlags(DWORD *pdwFlags);
virtual STDMETHODIMP CanNavigateNow ();
virtual STDMETHODIMP GetPidl(LPITEMIDLIST *ppidl);
virtual STDMETHODIMP SetReferrer (LPITEMIDLIST pidl);
virtual STDMETHODIMP_(DWORD) GetBrowserIndex();
virtual STDMETHODIMP GetBrowserByIndex(DWORD dwID, IUnknown **ppunk);
virtual STDMETHODIMP GetHistoryObject(IOleObject **ppole, IStream **pstm, IBindCtx **ppbc);
virtual STDMETHODIMP SetHistoryObject(IOleObject *pole, BOOL fIsLocalAnchor);
virtual STDMETHODIMP CacheOLEServer(IOleObject *pole);
virtual STDMETHODIMP GetSetCodePage(VARIANT* pvarIn, VARIANT* pvarOut);
virtual STDMETHODIMP OnHttpEquiv(IShellView* psv, BOOL fDone, VARIANT* pvarargIn, VARIANT* pvarargOut);
virtual STDMETHODIMP GetPalette( HPALETTE * hpal);
virtual STDMETHODIMP RegisterWindow(BOOL fUnregister, int swc);
virtual STDMETHODIMP_(LRESULT) WndProcBS(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam);
virtual STDMETHODIMP OnSize(WPARAM wParam);
virtual STDMETHODIMP OnCreate(LPCREATESTRUCT pcs);
virtual STDMETHODIMP_(LRESULT) OnCommand(WPARAM wParam, LPARAM lParam);
virtual STDMETHODIMP OnDestroy();
virtual STDMETHODIMP_(LRESULT) OnNotify(NMHDR * pnm);
virtual STDMETHODIMP OnSetFocus();
virtual STDMETHODIMP OnFrameWindowActivateBS(BOOL fActive);
virtual STDMETHODIMP ReleaseShellView();
virtual STDMETHODIMP ActivatePendingView();
virtual STDMETHODIMP CreateViewWindow(IShellView* psvNew, IShellView* psvOld, LPRECT prcView, HWND* phwnd);
virtual STDMETHODIMP GetBaseBrowserData(LPCBASEBROWSERDATA* ppbd);
virtual STDMETHODIMP_(LPBASEBROWSERDATA) PutBaseBrowserData();
virtual STDMETHODIMP SetAsDefFolderSettings() { ASSERT(FALSE); return E_NOTIMPL;}
virtual STDMETHODIMP SetTopBrowser();
virtual STDMETHODIMP UpdateSecureLockIcon(int eSecureLock);
virtual STDMETHODIMP Offline(int iCmd);
virtual STDMETHODIMP InitializeDownloadManager();
virtual STDMETHODIMP InitializeTransitionSite();
virtual STDMETHODIMP GetFolderSetData(struct tagFolderSetData* pfsd) { *pfsd = _fsd; return S_OK; };
virtual STDMETHODIMP _OnFocusChange(UINT itb);
virtual STDMETHODIMP v_ShowHideChildWindows(BOOL fChildOnly);
virtual STDMETHODIMP CreateBrowserPropSheetExt(THIS_ REFIID riid, void **ppvObj);
virtual STDMETHODIMP SetActivateState(UINT uActivate);
virtual STDMETHODIMP AllowViewResize(BOOL f);
virtual STDMETHODIMP _Initialize(HWND hwnd, IUnknown *pauto);
virtual STDMETHODIMP_(UINT) _get_itbLastFocus();
virtual STDMETHODIMP _put_itbLastFocus(UINT itbLastFocus);
virtual STDMETHODIMP _UIActivateView(UINT uState);
virtual STDMETHODIMP _CancelPendingNavigationAsync();
virtual STDMETHODIMP _MaySaveChanges();
virtual STDMETHODIMP _PauseOrResumeView(BOOL fPaused);
virtual STDMETHODIMP _DisableModeless();
virtual STDMETHODIMP _NavigateToPidl(LPCITEMIDLIST pidl, DWORD grfHLNF, DWORD dwFlags);
virtual STDMETHODIMP _TryShell2Rename(IShellView* psv, LPCITEMIDLIST pidlNew);
virtual STDMETHODIMP _SwitchActivationNow();
virtual STDMETHODIMP _CancelPendingView();
virtual STDMETHODIMP _ExecChildren(IUnknown *punkBar, BOOL fBroadcast,
const GUID *pguidCmdGroup, DWORD nCmdID, DWORD nCmdexecopt,
VARIANTARG *pvarargIn, VARIANTARG *pvarargOut);
virtual STDMETHODIMP _SendChildren(HWND hwndBar, BOOL fBroadcast,
UINT uMsg, WPARAM wParam, LPARAM lParam);
virtual STDMETHODIMP v_MayGetNextToolbarFocus(LPMSG lpMsg, UINT itbNext, int citb, LPTOOLBARITEM * pptbi, HWND * phwnd);
virtual STDMETHODIMP _SetFocus(LPTOOLBARITEM ptbi, HWND hwnd, LPMSG lpMsg) { ASSERT(FALSE); return E_NOTIMPL; }
virtual STDMETHODIMP _GetViewBorderRect(RECT* prc);
virtual STDMETHODIMP _UpdateViewRectSize();
virtual STDMETHODIMP _ResizeNextBorder(UINT itb);
virtual STDMETHODIMP _ResizeView();
virtual STDMETHODIMP _GetEffectiveClientArea(LPRECT lprectBorder, HMONITOR hmon);
virtual STDMETHODIMP GetCurrentFolderSettings(DEFFOLDERSETTINGS *pdfs, int cbDfs) { ASSERT(FALSE); return E_NOTIMPL; }
virtual STDMETHODIMP GetViewRect(RECT* prc);
virtual STDMETHODIMP GetViewWindow(HWND * phwndView);
virtual STDMETHODIMP InitializeTravelLog(ITravelLog* ptl, DWORD dw);
//IShellBrowserService
virtual STDMETHODIMP GetPropertyBag(DWORD dwFlags, REFIID riid, void** ppv);
// Desktop needs to override these:
virtual STDMETHODIMP_(IStream*) v_GetViewStream(LPCITEMIDLIST pidl, DWORD grfMode, LPCWSTR pwszName);
// Desktop needs access to these:
virtual STDMETHODIMP_(LRESULT) ForwardViewMsg(UINT uMsg, WPARAM wParam, LPARAM lParam) { ASSERT(FALSE); return 0; };
virtual STDMETHODIMP SetAcceleratorMenu(HACCEL hacc) { ASSERT(FALSE); return E_FAIL; }
virtual STDMETHODIMP_(int) _GetToolbarCount(THIS) { ASSERT(FALSE); return 0; }
virtual STDMETHODIMP_(LPTOOLBARITEM) _GetToolbarItem(THIS_ int itb) { ASSERT(FALSE); return NULL; }
virtual STDMETHODIMP _SaveToolbars(IStream* pstm) { ASSERT(FALSE); return E_NOTIMPL; }
virtual STDMETHODIMP _LoadToolbars(IStream* pstm) { ASSERT(FALSE); return E_NOTIMPL; }
virtual STDMETHODIMP _CloseAndReleaseToolbars(BOOL fClose) { ASSERT(FALSE); return E_NOTIMPL; }
virtual STDMETHODIMP_(UINT) _FindTBar(IUnknown* punkSrc) { ASSERT(FALSE); return (UINT)-1; };
virtual STDMETHODIMP v_MayTranslateAccelerator(MSG* pmsg) { ASSERT(FALSE); return E_NOTIMPL; }
virtual STDMETHODIMP _GetBorderDWHelper(IUnknown* punkSrc, LPRECT lprectBorder, BOOL bUseHmonitor) { ASSERT(FALSE); return E_NOTIMPL; }
// Shell browser overrides this.
virtual STDMETHODIMP v_CheckZoneCrossing(LPCITEMIDLIST pidl) {return S_OK;};
// Desktop and basesb need access to these:
virtual STDMETHODIMP _ResizeNextBorderHelper(UINT itb, BOOL bUseHmonitor);
// it just for us of course!
void StartBackgroundShellTasks(void);
void TaskbarWakeup(void);
protected:
CDesktopBrowser();
~CDesktopBrowser();
friend HRESULT CDesktopBrowser_CreateInstance(HWND hwnd, void **ppsb);
HRESULT SetInner(IUnknown* punk);
long _cRef;
// cached pointers on inner object
IUnknown* _punkInner;
IBrowserService2* _pbsInner;
IShellBrowser* _psbInner;
IServiceProvider* _pspInner;
IOleCommandTarget* _pctInner;
IDockingWindowSite* _pdwsInner;
IDockingWindowFrame* _pdwfInner;
IInputObjectSite* _piosInner;
IDropTarget* _pdtInner;
LPCBASEBROWSERDATA _pbbd;
LRESULT _RaisedWndProc(UINT msg, WPARAM wParam, LPARAM lParam);
void _SetViewArea();
void _GetViewBorderRects(int nRects, LPRECT prc); // does not have the tool bars
void _SetWorkAreas(int nWorkAreas, RECT * prcWork);
void _SubtractBottommostTray(LPRECT prc);
void _SaveState();
void _InitDeskbars();
void _SaveDesktopToolbars();
void _OnRaise(WPARAM wParam, LPARAM lParam);
void _SetupAppRan(WPARAM wParam, LPARAM lParam);
BOOL _QueryHKCRChanged(HWND hwnd, LPDWORD pdwCookie);
void _Lower();
void _Raise();
void _SwapParents(HWND hwndOldParent, HWND hwndNewParent);
BOOL _OnCopyData(PCOPYDATASTRUCT pcds);
void _OnAddToRecent(HANDLE hMem, DWORD dwProcId);
BOOL _InitScheduler(void);
HRESULT _AddDesktopTask(IRunnableTask *ptask, DWORD dwPriority);
BOOL _PtOnDesktopEdge(POINTL* ppt, LPUINT puEdge);
#ifdef DEBUG
void _CreateDeskbars();
#endif
HRESULT _CreateDeskBarForBand(UINT uEdge, IUnknown *punk, POINTL *pptl, IBandSite **pbs);
virtual void v_PropagateMessage(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL fSend);
HRESULT _OnFocusMsg(UINT uMsg, WPARAM wParam, LPARAM lParam);
HRESULT _CreateDesktopView();
HRESULT _GetPropertyBag(LPCITEMIDLIST pidl, DWORD dwFlags, REFIID riid, void** ppv);
HWND _GetDesktopListview();
UINT _PeekForAMessage();
void _InitMonitors();
#ifdef ENABLE_CHANNELS
void _MaybeLaunchChannelBand(void);
#endif
virtual void _ViewChange(DWORD dwAspect, LONG lindex);
HWND _hwndTray;
int _iWaitCount;
ULONG _uNotifyID;
DWORD _dwThreadIdTray; // Used to wakeup tray thread when the machine is really stressed
int _iTrayPriority;
DWORD _grfKeyState;
DWORD _dwEffectOnEdge; // what's the drop effect that desktop should return on dragging over the edge
BOOL _fRaised;
HWND _hwndRaised; // this is the parent of all of desktop's children when raised
struct tagFolderSetData _fsd;
int _nMonitors; // number of monitors on this desktop
HMONITOR _hMonitors[LV_MAX_WORKAREAS]; // the order of these hmonitors need to be preserved
RECT _rcWorkArea; // cached work-area
RECT _rcOldWorkAreas[LV_MAX_WORKAREAS]; // Old work areas before settings change
DWORD _nOldWork;
RECT _rcOldMonitors[LV_MAX_WORKAREAS]; // Old monitor sizes before settings change
// for _OnAddToRecent()
IShellTaskScheduler *_psched;
DWORD _idLocalServerThreads[ARRAYSIZE(c_localServers)];
DWORD _cChangeEvents;
HANDLE _rghChangeEvents[2]; // we watch HKCR and HKCR\CLSID
DWORD _dwChangeCookie;
DWORD _rgdwQHKCRCookies[QHKCRID_MAX - QHKCRID_MIN];
HKEY _hkClsid;
WCHAR _wzDesktopTitle[64]; // Localized Title
// IUnknownIdentity - for uniformity w.r.t. aggregation
// We are not aggregatable, so we are our own Outer.
IUnknown *_GetOuter() { return SAFECAST(this, IShellBrowser*); }
};
HRESULT CDesktopBrowser_CreateInstance(HWND hwnd, void **ppsb)
{
HRESULT hr = E_OUTOFMEMORY;
CDesktopBrowser *pdb = new CDesktopBrowser();
if (pdb)
{
hr = pdb->_Initialize(hwnd, NULL); // aggregation, etc.
if (FAILED(hr))
ATOMICRELEASE(pdb);
}
*ppsb = pdb;
return hr;
}
CDesktopBrowser::CDesktopBrowser() : _cRef(1)
{
TraceMsg(TF_LIFE, "ctor CDesktopBrowser %x", this);
for (INT i = 0; i < ARRAYSIZE(_idLocalServerThreads); i++)
_idLocalServerThreads[i] = -1;
}
CDesktopBrowser::~CDesktopBrowser()
{
SaveOldWorkAreas(_rcOldWorkAreas, _nOldWork);
// cleanup for QueryHKCRChanged()
for (int i = 0; i < ARRAYSIZE(_rghChangeEvents); i++)
{
if (_rghChangeEvents[i])
CloseHandle(_rghChangeEvents[i]);
}
if (_hkClsid)
RegCloseKey(_hkClsid);
// close down the local server threads that may be running
for (i = 0; i < ARRAYSIZE(_idLocalServerThreads); i++)
{
if (_idLocalServerThreads[i] != -1)
PostThreadMessage(_idLocalServerThreads[i], WM_QUIT, 0, 0);
}
TraceMsg(TF_LIFE, "dtor CDesktopBrowser %x", this);
}
HRESULT CDesktopBrowser::_Initialize(HWND hwnd, IUnknown* pauto)
{
IUnknown* punk;
HRESULT hres = CoCreateInstance(CLSID_CCommonBrowser, _GetOuter(), CLSCTX_INPROC_SERVER, IID_PPV_ARG(IUnknown, &punk));
if (SUCCEEDED(hres))
{
hres = SetInner(punk); // paired w/ Release in outer (TBS::Release)
if (SUCCEEDED(hres))
{
// we must initialize the inner guy BEFORE we call through any of these pointers.
hres = _pbsInner->_Initialize(hwnd, pauto);
if (SUCCEEDED(hres))
{
_pbsInner->GetBaseBrowserData(&_pbbd);
ASSERT(_pbbd);
// Restore the old settings from the registry that we persist.
if (!GetOldWorkAreas(_rcOldWorkAreas, &_nOldWork) || _nOldWork == 0)
{
// We didn't find it in the registry
_nOldWork = 0; // Since this is 0, we don't have to set _rcOldWorkAreas.
//We will recover from this in _SetWorkAreas()
}
SetTopBrowser();
_put_itbLastFocus(ITB_VIEW); // focus on desktop (w95 compat)
HACCEL hacc = LoadAccelerators(HINST_THISDLL, MAKEINTRESOURCE(ACCEL_DESKTOP));
ASSERT(hacc);
_pbsInner->SetAcceleratorMenu(hacc);
// Perf: never fire events from the desktop.
ASSERT(_pbbd->_pautoEDS);
ATOMICRELEASE(const_cast<IExpDispSupport *>(_pbbd->_pautoEDS));
_InitMonitors();
// Initialise _rcOldMonitors
for (int i = 0; i < _nMonitors; i++)
{
GetMonitorRect(_hMonitors[i], &_rcOldMonitors[i]);
}
// NOTE: if we have any more keys to watch, then
// we should create a static struct to walk
// so that it is easier to add more event/key pairs
_rghChangeEvents[0] = CreateEvent(NULL, TRUE, FALSE, NULL);
_rghChangeEvents[1] = CreateEvent(NULL, TRUE, FALSE, NULL);
if (_rghChangeEvents[0] && _rghChangeEvents[1])
{
if (ERROR_SUCCESS == RegNotifyChangeKeyValue(HKEY_CLASSES_ROOT, TRUE,
REG_NOTIFY_CHANGE_LAST_SET |REG_NOTIFY_CHANGE_NAME, _rghChangeEvents[0], TRUE))
{
_cChangeEvents = 1;
if (ERROR_SUCCESS == RegOpenKeyEx(HKEY_CLASSES_ROOT, TEXT("CLSID"), 0, MAXIMUM_ALLOWED, &_hkClsid)
&& ERROR_SUCCESS == RegNotifyChangeKeyValue(_hkClsid, TRUE, REG_NOTIFY_CHANGE_LAST_SET |REG_NOTIFY_CHANGE_NAME, _rghChangeEvents[1], TRUE))
{
// we need to leave the key open,
// or the event is signaled right away
_cChangeEvents++;
}
}
}
}
}
}
return hres;
}
//
// The refcount in the punk is transferred to us. We do not need to
// and indeed should not AddRef it.
//
// If any of these steps fails, we will clean up in our destructor.
//
HRESULT CDesktopBrowser::SetInner(IUnknown* punk)
{
HRESULT hres;
ASSERT(_punkInner == NULL);
_punkInner = punk;
#define INNERCACHE(iid, p) do { \
hres = SHQueryInnerInterface(_GetOuter(), punk, iid, (void **)&p); \
if (!EVAL(SUCCEEDED(hres))) return E_FAIL; \
} while (0)
INNERCACHE(IID_IBrowserService2, _pbsInner);
INNERCACHE(IID_IShellBrowser, _psbInner);
INNERCACHE(IID_IServiceProvider, _pspInner);
INNERCACHE(IID_IOleCommandTarget, _pctInner);
INNERCACHE(IID_IDockingWindowSite, _pdwsInner);
INNERCACHE(IID_IDockingWindowFrame, _pdwfInner);
INNERCACHE(IID_IInputObjectSite, _piosInner);
INNERCACHE(IID_IDropTarget, _pdtInner);
#undef INNERCACHE
return S_OK;
}
ULONG CDesktopBrowser::AddRef()
{
return InterlockedIncrement(&_cRef);
}
ULONG CDesktopBrowser::Release()
{
ASSERT( 0 != _cRef );
ULONG cRef = InterlockedDecrement(&_cRef);
if ( 0 == cRef )
{
_cRef = 1000; // guard against recursion
RELEASEINNERINTERFACE(_GetOuter(), _pbsInner);
RELEASEINNERINTERFACE(_GetOuter(), _psbInner);
RELEASEINNERINTERFACE(_GetOuter(), _pspInner);
RELEASEINNERINTERFACE(_GetOuter(), _pctInner);
RELEASEINNERINTERFACE(_GetOuter(), _pdwsInner);
RELEASEINNERINTERFACE(_GetOuter(), _pdwfInner);
RELEASEINNERINTERFACE(_GetOuter(), _piosInner);
RELEASEINNERINTERFACE(_GetOuter(), _pdtInner);
// this must come last
ATOMICRELEASE(_punkInner); // paired w/ CCI aggregation
ASSERT(_cRef == 1000);
delete this;
}
return cRef;
}
HRESULT CDesktopBrowser::QueryInterface(REFIID riid, void **ppvObj)
{
// IUnknownIdentity - The interface we use for IUnknown must come first.
static const QITAB qit[] = {
QITABENT(CDesktopBrowser, IShellBrowser),
QITABENT(CDesktopBrowser, IBrowserService2),
QITABENTMULTI(CDesktopBrowser, IBrowserService, IBrowserService2),
QITABENTMULTI(CDesktopBrowser, IOleWindow, IShellBrowser),
QITABENTMULTI2(CDesktopBrowser, SID_SShellDesktop, IShellBrowser), // effectively an IUnknown supported only by this class
QITABENT(CDesktopBrowser, IServiceProvider),
QITABENT(CDesktopBrowser, IShellBrowserService),
QITABENT(CDesktopBrowser, IOleCommandTarget),
QITABENT(CDesktopBrowser, IDockingWindowSite),
QITABENT(CDesktopBrowser, IInputObjectSite),
QITABENT(CDesktopBrowser, IMultiMonitorDockingSite),
QITABENT(CDesktopBrowser, IDropTarget),
{ 0 },
};
HRESULT hres = QISearch(this, qit, riid, ppvObj);
if (FAILED(hres))
{
if (_punkInner)
{
// don't let these get through to our base class...
// IID_IOleCommandTarget, IID_IOleInPlaceUIWindow
// 970414 adp: spoke to SatoNa, these *can* go thru
// i'll remove this week
// It's been working like this for a while now.
if (IsEqualIID(riid, IID_IOleInPlaceUIWindow))
{
*ppvObj = NULL;
hres = E_NOINTERFACE;
}
else
{
hres = _punkInner->QueryInterface(riid, ppvObj);
}
}
}
return hres;
}
void _InitDesktopMetrics(WPARAM wParam, LPCTSTR pszSection)
{
BOOL fForce = (!pszSection || !*pszSection);
if (fForce || (wParam == SPI_SETNONCLIENTMETRICS) || !lstrcmpi(pszSection, TEXT("WindowMetrics")))
{
FileIconInit(TRUE); // Tell the shell we want to play with a full deck
}
}
typedef struct
{
int iMonitors;
HMONITOR * phMonitors;
} EnumMonitorsData;
BOOL CALLBACK MultiMonEnumCallBack(HMONITOR hMonitor, HDC hdc, LPRECT lprc, LPARAM lData)
{
EnumMonitorsData * pEmd = (EnumMonitorsData *)lData;
if (pEmd->iMonitors > LV_MAX_WORKAREAS - 1)
//ignore the other monitors because we can only handle up to LV_MAX_WORKAREAS
//REARCHITECT: should we dynamically allocated this?
return FALSE;
pEmd->phMonitors[pEmd->iMonitors++] = hMonitor;
return TRUE;
}
// Initialize the number of monitors and the hmonitors array
void CDesktopBrowser::_InitMonitors()
{
HMONITOR hMonPrimary = GetPrimaryMonitor();
EnumMonitorsData emd;
emd.iMonitors = 0;
emd.phMonitors = _hMonitors;
EnumDisplayMonitors(NULL, NULL, MultiMonEnumCallBack, (LPARAM)&emd);
_nMonitors = GetNumberOfMonitors();
// Always move the primary monitor to the first location.
if (_hMonitors[0] != hMonPrimary)
{
for (int iMon = 1; iMon < _nMonitors; iMon++)
{
if (_hMonitors[iMon] == hMonPrimary)
{
_hMonitors[iMon] = _hMonitors[0];
_hMonitors[0] = hMonPrimary;
break;
}
}
}
}
// Gets the persisted old work areas, from the registry
BOOL GetOldWorkAreas(LPRECT lprc, DWORD* pdwNoOfOldWA)
{
BOOL fRet = FALSE;
*pdwNoOfOldWA = 0;
HKEY hkey;
if (RegOpenKeyEx(HKEY_CURRENT_USER, REG_DESKCOMP_OLDWORKAREAS, 0, KEY_READ, &hkey) == ERROR_SUCCESS)
{
DWORD dwType, cbSize = sizeof(*pdwNoOfOldWA);
// Read in the no of old work areas
if (SHQueryValueEx(hkey, REG_VAL_OLDWORKAREAS_COUNT, NULL, &dwType, (LPBYTE)pdwNoOfOldWA, &cbSize) == ERROR_SUCCESS)
{
// Read in the old work area rects
cbSize = sizeof(*lprc) * (*pdwNoOfOldWA);
if (SHQueryValueEx(hkey, REG_VAL_OLDWORKAREAS_RECTS, NULL, &dwType, (LPBYTE)lprc, &cbSize) == ERROR_SUCCESS)
{
fRet = TRUE;
}
}
RegCloseKey(hkey);
}
return fRet;
}
// Saves the old work areas into the registry
void SaveOldWorkAreas(LPCRECT lprc, DWORD nOldWA)
{
// Recreate the registry key.
HKEY hkey;
if (RegCreateKey(HKEY_CURRENT_USER, REG_DESKCOMP_OLDWORKAREAS, &hkey) == ERROR_SUCCESS)
{
// Write out the no. of old work areas
RegSetValueEx(hkey, REG_VAL_OLDWORKAREAS_COUNT, 0, REG_DWORD, (LPBYTE)&nOldWA, sizeof(nOldWA));
// Write out the no work area rectangles
RegSetValueEx(hkey, REG_VAL_OLDWORKAREAS_RECTS, 0, REG_BINARY, (LPBYTE)lprc, sizeof(*lprc) * nOldWA);
// Close out the reg key
RegCloseKey(hkey);
}
}
//*** CDesktopBrowser::IOleCommandTarget::* {
STDMETHODIMP CDesktopBrowser::QueryStatus(const GUID *pguidCmdGroup,
ULONG cCmds, OLECMD rgCmds[], OLECMDTEXT *pcmdtext)
{
return E_NOTIMPL;
}
STDMETHODIMP CDesktopBrowser::Exec(const GUID *pguidCmdGroup, DWORD nCmdID,
DWORD nCmdexecopt, VARIANTARG *pvarargIn, VARIANTARG *pvarargOut)
{
if (pguidCmdGroup == NULL)
{
/*NOTHING*/
}
else if (IsEqualGUID(CGID_ShellDocView, *pguidCmdGroup))
{
switch (nCmdID)
{
case SHDVID_RAISE:
// n.b.: DTRF_RAISE/DTRF_LOWER go down; DTRF_QUERY goes up
ASSERT(pvarargIn != NULL && pvarargIn->vt == VT_I4);
if (pvarargIn->vt == VT_I4 && pvarargIn->lVal == DTRF_QUERY)
{
ASSERT(pvarargOut != NULL);
pvarargOut->vt = VT_I4;
pvarargOut->lVal = _fRaised ? DTRF_RAISE : DTRF_LOWER;
return S_OK;
}
// o.w. let parent handle it
break;
case SHDVID_UPDATEOFFLINEDESKTOP:
UpdateAllDesktopSubscriptions(NULL);
return S_OK;
}
}
else if (IsEqualGUID(CGID_Explorer, *pguidCmdGroup))
{
switch (nCmdID)
{
case SBCMDID_OPTIONS:
case SBCMDID_ADDTOFAVORITES:
return _pctInner->Exec(pguidCmdGroup, nCmdID, nCmdexecopt, pvarargIn, pvarargOut);
}
}
// do *not* forward up to SUPERCLASS::Exec (see QI's cryptic commment
// about "don't let these get thru to our base class")
return OLECMDERR_E_NOTSUPPORTED;
}
// }
STDMETHODIMP CDesktopBrowser::BrowseObject(LPCITEMIDLIST pidl, UINT wFlags)
{
// Force SBSP_NEWBROWSER, SBSP_ABSOLUTE, and SBSP_NOTRANSFERHIST
wFlags &= ~(SBSP_DEFBROWSER | SBSP_SAMEBROWSER | SBSP_RELATIVE | SBSP_PARENT);
wFlags |= (SBSP_NEWBROWSER | SBSP_ABSOLUTE | SBSP_NOTRANSFERHIST);
return _psbInner->BrowseObject(pidl, wFlags);
}
IStream *GetDesktopViewStream(DWORD grfMode, LPCTSTR pszName)
{
HKEY hkStreams;
if (RegCreateKey(HKEY_CURRENT_USER, REGSTR_PATH_EXPLORER TEXT("\\Streams"), &hkStreams) == ERROR_SUCCESS)
{
IStream *pstm = OpenRegStream(hkStreams, TEXT("Desktop"), pszName, grfMode);
RegCloseKey(hkStreams);
return pstm;
}
return NULL;
}
void DeleteDesktopViewStream(LPCTSTR pszName)
{
SHDeleteValue(HKEY_CURRENT_USER, REGSTR_PATH_EXPLORER TEXT("\\Streams\\Desktop"), pszName);
}
IStream *CDesktopBrowser::v_GetViewStream(LPCITEMIDLIST pidl, DWORD grfMode, LPCWSTR pszName)
{
return GetDesktopViewStream(grfMode, pszName);
}
HRESULT CDesktopBrowser::_GetPropertyBag(LPCITEMIDLIST pidl, DWORD dwFlags, REFIID riid, void** ppv)
{
return SHGetViewStatePropertyBag(pidl, VS_BAGSTR_DESKTOP, dwFlags | SHGVSPB_ROAM, riid, ppv);
}
HRESULT CDesktopBrowser::GetPropertyBag(DWORD dwFlags, REFIID riid, void** ppv)
{
HRESULT hr;
LPITEMIDLIST pidl;
hr = SHGetSpecialFolderLocation(NULL, CSIDL_DESKTOP, &pidl);
if (SUCCEEDED(hr))
{
hr = _GetPropertyBag(pidl, dwFlags, riid, ppv);
ILFree(pidl);
}
return hr;
}
LRESULT CDesktopBrowser::OnCommand(WPARAM wParam, LPARAM lParam)
{
switch (GET_WM_COMMAND_ID(wParam, lParam))
{
case FCIDM_FINDFILES:
SHFindFiles(_pbbd->_pidlCur, NULL);
break;
case FCIDM_REFRESH:
{
VARIANT v = {0};
v.vt = VT_I4;
v.lVal = OLECMDIDF_REFRESH_NO_CACHE|OLECMDIDF_REFRESH_PROMPTIFOFFLINE;
// Our Exec is neutered (on purpose), so call our parent
_pctInner->Exec(NULL, OLECMDID_REFRESH, OLECMDEXECOPT_DONTPROMPTUSER, &v, NULL);
break;
}
case IDC_KBSTART:
case FCIDM_NEXTCTL:
if (_hwndTray)
{
// n.b. VK_TAB handled this way (among other things)
SendMessage(_hwndTray, WM_COMMAND, wParam, lParam);
}
break;
case IDM_CLOSE:
PostMessage(_hwndTray, TM_DOEXITWINDOWS, 0, 0);
break;
default:
_pbsInner->OnCommand(wParam, lParam);
break;
}
return S_OK;
}
// Create desktop IShellView instance
HRESULT CDesktopBrowser::_CreateDesktopView()
{
LPCITEMIDLIST pidl = SHCloneSpecialIDList(NULL, CSIDL_DESKTOP, TRUE);
if (pidl)
{
IPropertyBag* ppb;
if (0 == GetSystemMetrics(SM_CLEANBOOT) &&
SUCCEEDED(_GetPropertyBag(pidl, SHGVSPB_PERUSER | SHGVSPB_PERFOLDER, IID_PPV_ARG(IPropertyBag, &ppb))))
{
SHPropertyBag_ReadDWORDDef(ppb, VS_PROPSTR_FFLAGS, reinterpret_cast<DWORD*>(&_fsd._fs.fFlags), FWF_DESKTOP | FWF_NOCLIENTEDGE | FWF_SNAPTOGRID);
ppb->Release();
}
else
{
_fsd._fs.fFlags = FWF_DESKTOP | FWF_NOCLIENTEDGE | FWF_SNAPTOGRID; // default
}
_fsd._fs.ViewMode = FVM_ICON; // can't change this, sorry
SHELLSTATE ss = {0};
SHGetSetSettings(&ss, SSF_HIDEICONS, FALSE);
if (ss.fHideIcons)
_fsd._fs.fFlags |= FWF_NOICONS;
else
_fsd._fs.fFlags &= ~FWF_NOICONS;
// We keep the active desktop in offline mode!
ASSERT(_pbbd->_pautoWB2);
_pbbd->_pautoWB2->put_Offline(TRUE);
return _psbInner->BrowseObject(pidl, SBSP_SAMEBROWSER);
}
else
{
TCHAR szYouLoose[256];
LoadString(HINST_THISDLL, IDS_YOULOSE, szYouLoose, ARRAYSIZE(szYouLoose));
MessageBox(NULL, szYouLoose, NULL, MB_ICONSTOP);
return E_FAIL;
}
}
HRESULT CDesktopBrowser::ActivatePendingView(void)
{
HRESULT hres = _pbsInner->ActivatePendingView();
if (SUCCEEDED(hres))
{
// calling SetShellWindow will cause the desktop
// to initially paint white, then the background window.
// This causes an ugly white trail when you move windows
// around until the desktop finally paints.
//
// Calling SetShellWindowEx resolves this problem.
//
SHSetShellWindowEx(_pbbd->_hwnd, _GetDesktopListview());
}
return hres;
}
#ifdef DEBUG
void CDesktopBrowser::_CreateDeskbars()
{
HRESULT hres;
BOOL fCreate = FALSE;
HKEY hkey;
if (ERROR_SUCCESS == RegOpenKey(HKEY_CURRENT_USER, REGSTR_PATH_EXPLORER TEXT("\\DeskBar\\Bands"), &hkey))
{
fCreate = TRUE;
RegCloseKey(hkey);
}
if (fCreate)
{
IPersistStreamInit *ppstm;
hres = CoCreateInstance(CLSID_DeskBarApp, NULL, CLSCTX_INPROC_SERVER, IID_PPV_ARG(IPersistStreamInit, &ppstm));
if (SUCCEEDED(hres)) {
hres = ppstm->InitNew();
AddToolbar(ppstm, L"test", NULL); // "Microsoft.DeskBarApp"
ppstm->Release();
}
}
}
#endif
void CDesktopBrowser::_InitDeskbars()
{
//
// Load toolbars
//
// 1st, try persisted state
IStream* pstm = GetDesktopViewStream(STGM_READ, TEXT("Toolbars"));
HRESULT hres = E_FAIL;
if (pstm)
{
hres = _pbsInner->_LoadToolbars(pstm);
pstm->Release();
}
// 2nd, if there is none (or if version mismatch or other failure),
// try settings from setup
// NOTE: this works fine for ie4 where we have no old toolbars,
// but for future releases we'll need some kind of merging scheme,
// so we probably want to change this after ie4-beta-1.
if (FAILED(hres))
{
// n.b. HKLM not HKCU
// like GetDesktopViewStream but for HKLM
HKEY hk = SHGetShellKey(SHELLKEY_HKLM_EXPLORER, TEXT("Streams\\Desktop"), TRUE);
if (hk)
{
pstm = OpenRegStream(hk, NULL, TEXT("Default Toolbars"), STGM_READ);
if (pstm)
{
hres = _pbsInner->_LoadToolbars(pstm);
pstm->Release();
}
RegCloseKey(hk);
}
}
// o.w., throw up our hands
if (FAILED(hres))
{
ASSERT(0);
#ifdef DEBUG
// but for debug, need a way to bootstrap the entire process
_CreateDeskbars();
#endif
}
}
// Handle creation of a new Desktop folder window. Creates everything except
// the viewer part.
// Returns -1 if something goes wrong.
HWND g_hwndTray = NULL;
HRESULT CDesktopBrowser::OnCreate(CREATESTRUCT *pcs)
{
LRESULT lr;
g_pdtray->GetTrayWindow(&_hwndTray);
g_hwndTray = _hwndTray;
g_pdtray->SetDesktopWindow(_pbbd->_hwnd);
SetTimer(_pbbd->_hwnd, IDT_ENUMHKCR, 5 * 60 * 1000, NULL);
//
// Notify IEDDE that the automation services are now available.
//
IEOnFirstBrowserCreation(NULL);
ASSERT(_hwndTray);
// REARCHITECT: we need to split out "ie registry settings" into a
// browser component and a shell component.
//
//EnsureWebViewRegSettings();
if (SUCCEEDED(_CreateDesktopView()))
{
lr = _pbsInner->OnCreate(pcs); // success
PostMessage(_pbbd->_hwnd, DTM_CREATESAVEDWINDOWS, 0, 0);
return (HRESULT) lr;
}
return (LRESULT)-1; // failure
}
UINT GetDDEExecMsg()
{
static UINT uDDEExec = 0;
if (!uDDEExec)
uDDEExec = RegisterWindowMessage(TEXT("DDEEXECUTESHORTCIRCUIT"));
return uDDEExec;
}
LRESULT CDesktopBrowser::OnNotify(NMHDR * pnm)
{
switch (pnm->code)
{
case SEN_DDEEXECUTE:
if (pnm->idFrom == 0)
{
// short cut notifier around the dde conv.
LPNMVIEWFOLDER pnmPost = DDECreatePostNotify((LPNMVIEWFOLDER)pnm);
if (pnmPost)
{
PostMessage(_pbbd->_hwnd, GetDDEExecMsg(), 0, (LPARAM)pnmPost);
return TRUE;
}
}
break;
case NM_STARTWAIT:
case NM_ENDWAIT:
_iWaitCount += (pnm->code == NM_STARTWAIT ? 1 :-1);
ASSERT(_iWaitCount >= 0);
// Don't let it go negative or we'll never get rid of it.
if (_iWaitCount < 0)
_iWaitCount = 0;
// what we really want is for user to simulate a mouse move/setcursor
SetCursor(LoadCursor(NULL, _iWaitCount ? IDC_APPSTARTING : IDC_ARROW));
break;
default:
return _pbsInner->OnNotify(pnm);
}
return 0;
}
// HACKHACK: this hard codes in that we know a listview is the child
// of the view.
HWND CDesktopBrowser::_GetDesktopListview()
{
HWND hwndView = _pbbd->_hwndView ? _pbbd->_hwndView : _pbbd->_hwndViewPending;
if (!hwndView)
return NULL;
return FindWindowEx(hwndView, NULL, WC_LISTVIEW, NULL);
}
#ifndef ENUM_REGISTRY_SETTINGS
#define ENUM_REGISTRY_SETTINGS ((DWORD)-2)
#endif
STDAPI_(BOOL) SHIsTempDisplayMode()
{
BOOL fTempMode = FALSE;
DEVMODE dm = {0};
dm.dmSize = sizeof(dm);
if (EnumDisplaySettings(NULL, ENUM_REGISTRY_SETTINGS, &dm) &&
dm.dmPelsWidth > 0 && dm.dmPelsHeight > 0)
{
HDC hdc = GetDC(NULL);
int xres = GetDeviceCaps(hdc, HORZRES);
int yres = GetDeviceCaps(hdc, VERTRES);
ReleaseDC(NULL, hdc);
if (xres != (int)dm.dmPelsWidth || yres != (int)dm.dmPelsHeight)
fTempMode = TRUE;
}
return fTempMode;
}
// NOTE: this is the hack andyp put in
// (dli) Currently, bottommost Tray is really wierd, it is not treated as toolbars.
// In a sense, it has higher priority than those toolbars. So they should be taken
// off the EffectiveClientArea
void CDesktopBrowser::_SubtractBottommostTray(LPRECT prc)
{
LRESULT lTmp;
APPBARDATA abd;
abd.cbSize = sizeof(APPBARDATA);
abd.hWnd = _hwndTray;
// lTmp = SHAppBarMessage(ABM_GETSTATE, &abd);
lTmp = g_pdtray->AppBarGetState();
if ((lTmp & (ABS_ALWAYSONTOP|ABS_AUTOHIDE)) == 0) {
// tray is on bottom and takes 'real' space
RECT rcTray = {0};
GetWindowRect(_hwndTray, &rcTray);
IntersectRect(&rcTray, prc, &rcTray);
SubtractRect(prc, prc, &rcTray);
}
}
HRESULT CDesktopBrowser::_GetEffectiveClientArea(LPRECT lprectBorder, HMONITOR hmon)
{
//
// Cache the work area if
// (1) this is the very first call
// (2) cached value is blew off by WM_SIZE (in _OnSize)
//
if (hmon) {
GetMonitorWorkArea(hmon, lprectBorder);
}
else {
if (::IsRectEmpty(&_rcWorkArea)) {
::SystemParametersInfo(SPI_GETWORKAREA, 0, &_rcWorkArea, 0);
}
*lprectBorder = _rcWorkArea;
}
_SubtractBottommostTray(lprectBorder);
MapWindowPoints(NULL, _pbbd->_hwnd, (LPPOINT)lprectBorder, 2);
return S_OK;
}
BOOL EqualRects(LPRECT prcNew, LPRECT prcOld, int nRects)
{
int i;
for (i = 0; i < nRects; i++)
if (!EqualRect(&prcNew[i], &prcOld[i]))
return FALSE;
return TRUE;
}
//
// When Snap-To-Grid is on, we want to reduce the size of gutter space around the primary monitor.
// We achieve that by adding a few pixels to the grid size sothat the gutter size left out is
// as small as possible.
// Note: This is currently done only for Desktop listview.
//
// fMinimizeCutterSpace == FALSE => Nothing to do. Just return.
// fMinimizeGutterSpace == TRUE => We calculate and set the icon spacing so as to minimize gutter.
//
void UpdateGridSizes(BOOL fDesktop, HWND hwndListview, int nWorkAreas, LPRECT prcWork, BOOL fMinimizeGutterSpace)
{
if(!fDesktop) //If this is not desktop, we do not change any of this.
return;
// Trying to reset the icon spacing to system icon spacing results in ReComputing everything.
// So, just return without doing anything!
if(!fMinimizeGutterSpace)
return; //If we don't have to minimize the gutter space, nothing to do!
int cxSysIconSpacing = GetSystemMetrics(SM_CXICONSPACING);
if (cxSysIconSpacing <= 0) cxSysIconSpacing = 1; // avoid div0
int cySysIconSpacing = GetSystemMetrics(SM_CYICONSPACING);
if (cySysIconSpacing <= 0) cySysIconSpacing = 1; // avoid div0
int cxNewIconSpacing = 0, cyNewIconSpacing = 0;
RECT rcWorkAreas[LV_MAX_WORKAREAS];
//If the work areas are not given, we need to get them.
if(prcWork == NULL)
{
prcWork = &rcWorkAreas[0];
ListView_GetNumberOfWorkAreas(hwndListview, &nWorkAreas);
if(nWorkAreas > 0)
ListView_GetWorkAreas(hwndListview, nWorkAreas, prcWork);
else
ListView_GetViewRect(hwndListview, prcWork);
}
//Get the primary work area.
for(int iPrimary = 0; iPrimary < nWorkAreas; iPrimary++)
{
//LATER: Is this check enough! What about when tray is at the top or left?
if((prcWork[iPrimary].left == 0) && (prcWork[iPrimary].top == 0))
break;
}
if(iPrimary == nWorkAreas)
iPrimary = 0; //Assume that the first work area is primary work area.
//Find the number of columns based on current system horizontal icon spacing.
int nCols = (prcWork[iPrimary].right - prcWork[iPrimary].left)/cxSysIconSpacing;
if (nCols <= 0) nCols = 1; // avoid div0
//Divide the remaining pixels and add them to each column to minimize the reminder area.
cxNewIconSpacing = cxSysIconSpacing + ((prcWork[iPrimary].right - prcWork[iPrimary].left)%cxSysIconSpacing)/nCols;
if (cxNewIconSpacing <= 0) cxNewIconSpacing = 1; // avoid div0
//Find the number of Rows based on current system vertical icon spacing.
int nRows = (prcWork[iPrimary].bottom - prcWork[iPrimary].top)/cySysIconSpacing;
if (nRows <= 0) nRows = 1; // avoid div0
//Divide the remaining pixles and add them to each row to minimize the reminder area.
cyNewIconSpacing = cySysIconSpacing + ((prcWork[iPrimary].bottom - prcWork[iPrimary].top)%cySysIconSpacing)/nRows;
if (cyNewIconSpacing <= 0) cyNewIconSpacing = 1; // avoid div0
//Set the new icon spacing to the desktop's listview.
ListView_SetIconSpacing(hwndListview, cxNewIconSpacing, cyNewIconSpacing);
}
void CDesktopBrowser::_SetWorkAreas(int nWorkAreas, LPRECT prcWork)
{
RECT rcListViewWork[LV_MAX_WORKAREAS];
RECT rcNewWork[LV_MAX_WORKAREAS];
int nListViewWork = 0;
HWND hwndList;
int i;
ASSERT(prcWork);
ASSERT(nWorkAreas <= LV_MAX_WORKAREAS);
ASSERT(nWorkAreas > 0);
if (nWorkAreas <= 0)
return;
if (SHIsTempDisplayMode())
return;
hwndList = _GetDesktopListview();
ASSERT(IsWindow(hwndList));
ListView_GetNumberOfWorkAreas(hwndList, &nListViewWork);
BOOL fUpgradeGridSize = (BOOL)((ListView_GetExtendedListViewStyle(hwndList)) & LVS_EX_SNAPTOGRID);
BOOL fRedraw = FALSE;
if (nListViewWork > 0)
{
ListView_GetWorkAreas(hwndList, nListViewWork, rcListViewWork);
// Map these rects back to DESKTOP coordinate
// We need to convert the following only if WorkAreas > 1
if (nListViewWork > 1)
{
// [msadek]; MapWindowPoints() is mirroring-aware only if you pass two points
for(i = 0; i < nListViewWork; i++)
{
MapWindowPoints(hwndList, HWND_DESKTOP, (LPPOINT)(&rcListViewWork[i]), 2);
}
}
if(nListViewWork == nWorkAreas && EqualRects(prcWork, rcListViewWork, nWorkAreas))
return;
}
else if (_nOldWork > 1)
// In single monitor case, listview workares always starts from (0,0)
// It will be wrong to set the persisted workarea.
{
for (nListViewWork = 0; nListViewWork < (int)_nOldWork && nListViewWork < LV_MAX_WORKAREAS; nListViewWork++)
CopyRect(&rcListViewWork[nListViewWork], &_rcOldWorkAreas[nListViewWork]);
// This may not be needed, because at this point, ListView is in Desktop coordinate
// [msadek]; MapWindowPoints() is mirroring-aware only if you pass two points
for(i = 0; i < nListViewWork; i++)
{
MapWindowPoints(HWND_DESKTOP, hwndList, (LPPOINT)(&rcListViewWork[i]), 2);
}
//Before setting the WorkAreas, change the grid-size if needed.
if(fUpgradeGridSize)
{
// We set the grid size based on the new work area. This way Recycle-Bin gets snapped
// to the correct location just once and we don't need to change it later.
SendMessage(hwndList, WM_SETREDRAW, FALSE, 0);
fRedraw = TRUE;
UpdateGridSizes(TRUE, hwndList, nWorkAreas, prcWork, TRUE);
fUpgradeGridSize = FALSE; //Don't need to do it again!
}
// This call to SetWorkAreas sets the old work areas in the listview, which is not persisted there.
ListView_SetWorkAreas(hwndList, nListViewWork, rcListViewWork);
}
//Make a copy of the new work area array because it gets modified by
// the MapWindowPoints below.
for(i = 0; i < nWorkAreas; i++)
rcNewWork[i] = *(prcWork + i);
// It's already in ListView coordinate if we just have one monitor
if (nWorkAreas > 1)
{
for(i = 0; i < nWorkAreas; i++)
{
MapWindowPoints(HWND_DESKTOP, hwndList, (LPPOINT)(&prcWork[i]), 2);
}
}
//Because the work areas change, update the grid size to reduce the gutter area.
if(fUpgradeGridSize) //If we haven't done already!
{
SendMessage(hwndList, WM_SETREDRAW, FALSE, 0);
fRedraw = TRUE;
UpdateGridSizes(TRUE, hwndList, nWorkAreas, prcWork, TRUE);
}
// We need to set the new work area before we call AdjustDesktopComponents below
// because that function results in a refresh and that ends up setting
// the work areas again to the same values.
ListView_SetWorkAreas(hwndList, nWorkAreas, prcWork);
if (fRedraw)
SendMessage(hwndList, WM_SETREDRAW, TRUE, 0);
if (nWorkAreas == 1)
MapWindowPoints(hwndList, HWND_DESKTOP, (LPPOINT)rcNewWork, 2 * nWorkAreas);
//Move and size desktop components based on the new work areas.
AdjustDesktopComponents((LPCRECT)rcNewWork, nWorkAreas, (LPCRECT)_rcOldMonitors, (LPCRECT)_rcOldWorkAreas, _nOldWork);
// Backup the new Monitor rect's in _rcOldMonitors
for (i = 0; i < _nMonitors; i++)
{
GetMonitorRect(_hMonitors[i], &_rcOldMonitors[i]);
}
// Backup the new workareas in _rcOldWorkAreas
for (i = 0; i < nWorkAreas; i++)
{
_rcOldWorkAreas[i] = rcNewWork[i];
}
_nOldWork = nWorkAreas;
static const LPTSTR lpszSubkey = TEXT("Software\\Microsoft\\Windows\\CurrentVersion\\Explorer\\ScreenResFixer");
static const LPTSTR lpszValue = TEXT("AdjustRecycleBinPosition");
//Check if we need to change the recycle bin position because of this work-area change
// 0 => Recycle-bin hasn't been positioned because of resolution fixer.
// 1 => Recycle-bin needs to be repositioned. It has't happened yet!
// 2 => Recycle-bin has already been re-positioned. Nothing needs to be done here!
DWORD dwAdjustPos = 2; //Assume that Recycle-bin has been already positioned.
DWORD dwSize = sizeof(dwAdjustPos);
SHRegGetUSValue(lpszSubkey, lpszValue, NULL, &dwAdjustPos, &dwSize, FALSE, &dwAdjustPos, dwSize);
//If the AdjustRecycleBinPosition value has 1, that means we need to reposition recyclebin.
if(dwAdjustPos == 1)
{
//Move the recycle-bin to default position
SendMessage(_pbbd->_hwndView, WM_DSV_ADJUSTRECYCLEBINPOSITION, 0, 0);
dwAdjustPos = 2; //We have moved this just now! No need to move it anymore!
SHRegSetUSValue(lpszSubkey, lpszValue, REG_DWORD, &dwAdjustPos, sizeof(dwAdjustPos), SHREGSET_HKCU | SHREGSET_FORCE_HKCU);
}
}
// Get all the view border rectangles (not including toolbars) for the monitors
// this is used for multi-monitor case only.
void CDesktopBrowser::_GetViewBorderRects(int nRects, LPRECT prcBorders)
{
int iMon;
HMONITOR hmonTray;
Tray_GetHMonitor(_hwndTray, &hmonTray);
for (iMon = 0; iMon < min(_nMonitors, nRects); iMon++)
{
GetMonitorWorkArea(_hMonitors[iMon], &prcBorders[iMon]);
if (hmonTray == _hMonitors[iMon])
{
_SubtractBottommostTray(&prcBorders[iMon]);
}
// Extract the border taken by all "frame" toolbars
for (int itb=0; itb < _pbsInner->_GetToolbarCount(); itb++)
{
LPTOOLBARITEM ptbi = _pbsInner->_GetToolbarItem(itb);
if (ptbi && ptbi->hMon == _hMonitors[iMon])
{
prcBorders[iMon].left += ptbi->rcBorderTool.left;
prcBorders[iMon].top += ptbi->rcBorderTool.top;
prcBorders[iMon].right -= ptbi->rcBorderTool.right;
prcBorders[iMon].bottom -= ptbi->rcBorderTool.bottom;
}
}
}
}
HRESULT CDesktopBrowser::_UpdateViewRectSize()
{
HWND hwndView = _pbbd->_hwndView;
if (!hwndView && ((hwndView = _pbbd->_hwndViewPending) == NULL))
return S_FALSE;
_pbsInner->_UpdateViewRectSize();
if (_nMonitors <= 1)
{
RECT rcView, rcWork;
GetViewRect(&rcView);
rcWork.top = rcWork.left = 0;
rcWork.right = RECTWIDTH(rcView);
rcWork.bottom = RECTHEIGHT(rcView);
_SetWorkAreas(1, &rcWork);
}
else
{
RECT rcWorks[LV_MAX_WORKAREAS];
_GetViewBorderRects(_nMonitors, rcWorks);
_SetWorkAreas(_nMonitors, rcWorks);
}
return S_OK;
}
void CDesktopBrowser::_SetViewArea()
{
//
// Invalidate the cached work area size
//
::SetRectEmpty(&_rcWorkArea);
v_ShowHideChildWindows(FALSE);
}
// we get called here when new drives come and go;
// things like net connections, hot insertions, etc.
void _OnDeviceBroadcast(HWND hwnd, ULONG_PTR code, DEV_BROADCAST_HDR *pbh)
{
// do a bunch of this stuff here in desktop so it only happens
// once...
switch (code)
{
case DBT_DEVICEREMOVECOMPLETE: // drive or media went away
case DBT_DEVICEARRIVAL: // new drive or media (or UNC) has arrived
case DBT_DEVICEQUERYREMOVE: // drive or media about to go away
case DBT_DEVICEQUERYREMOVEFAILED: // drive or media didn't go away
{
BOOL fFilteredOut = FALSE;
// Don't process if we are being shutdown...
if (!IsWindowVisible(hwnd))
break;
// Filter this one out
if (DBT_DEVICEARRIVAL == code)
{
if (DBT_DEVTYP_NET == pbh->dbch_devicetype)
{
LPITEMIDLIST pidl;
// Tell the hood to update as things have probably changed!
// PERF: this can be slow in the case of the nethood FS folder
// no longer exists. this hangs the desktop/tray. this is a robustness bug
if (SUCCEEDED(SHGetSpecialFolderLocation(NULL, CSIDL_NETHOOD, &pidl)))
{
SHChangeNotify(SHCNE_UPDATEDIR, SHCNF_IDLIST, pidl, NULL);
SHFree(pidl);
}
// use UNCToNetID(loop if (pbn->dbcn_resource)
fFilteredOut = TRUE;
}
}
if (!fFilteredOut)
{
CMountPoint::HandleWMDeviceChange(code, pbh);
}
break;
}
}
}
// Note: this overrides the one in CBaseBrowser
HRESULT CDesktopBrowser::GetViewRect(RECT* prc)
{
//
// Check if we are on a multiple-monitor system. In multiple monitors the
// view needs to cover all displays (ie the size of _pbbd->_hwnd) while on
// single-monitor systems the view just needs to cover the work area (like
// in Win95).
//
if (_nMonitors <= 1)
_pbsInner->GetViewRect(prc);
else
GetClientRect(_pbbd->_hwnd, prc);
return S_OK;
}
HRESULT CDesktopBrowser::ReleaseShellView()
{
_SaveState();
return _pbsInner->ReleaseShellView();
}
void CDesktopBrowser::_ViewChange(DWORD dwAspect, LONG lindex)
{
// do nothing here...
}
void CDesktopBrowser::_SaveDesktopToolbars()
{
IStream * pstm = GetDesktopViewStream(STGM_WRITE, TEXT("Toolbars"));
if (pstm)
{
_pbsInner->_SaveToolbars(pstm);
pstm->Release();
}
}
void CDesktopBrowser::_SaveState()
{
// save off the Recycle Bin information to the registry
SaveRecycleBinInfo();
if (!SHRestricted(REST_NOSAVESET) && _pbbd->_psv)
{
if (0 == GetSystemMetrics(SM_CLEANBOOT))
{
FOLDERSETTINGS fs;
_pbbd->_psv->GetCurrentInfo(&fs);
IPropertyBag* ppb;
if (SUCCEEDED(GetPropertyBag(SHGVSPB_PERUSER | SHGVSPB_PERFOLDER, IID_PPV_ARG(IPropertyBag, &ppb))))
{
SHPropertyBag_WriteInt(ppb, VS_PROPSTR_FFLAGS, fs.fFlags);
ppb->Release();
}
}
_pbbd->_psv->SaveViewState();
_SaveDesktopToolbars();
}
}
HRESULT CDesktopBrowser::OnSize(WPARAM wParam)
{
if (wParam == SIZE_MINIMIZED)
{
TraceMsg(DM_TRACE, "c.dwp: Desktop minimized by somebody!");
// Put it back.
ShowWindow(_pbbd->_hwnd, SW_RESTORE);
}
_SetViewArea();
return S_OK;
}
HRESULT CDesktopBrowser::OnDestroy()
{
TraceMsg(DM_SHUTDOWN, "cdtb._od (WM_DESTROY)");
if (_uNotifyID)
{
SHChangeNotifyDeregister(_uNotifyID);
_uNotifyID = 0;
}
if (_hwndRaised)
DestroyWindow(_hwndRaised);
// get rid of the scheduler and the desktop tasks
if (_psched)
{
_psched->RemoveTasks(TOID_Desktop, 0, TRUE);
ATOMICRELEASE(_psched);
}
_pbsInner->OnDestroy();
_pbsInner->_CloseAndReleaseToolbars(TRUE);
return S_OK;
}
#define DM_SWAP DM_TRACE
void CDesktopBrowser::_SwapParents(HWND hwndOldParent, HWND hwndNewParent)
{
HWND hwnd = ::GetWindow(hwndOldParent, GW_CHILD);
while (hwnd)
{
//
// Note that we must get the next sibling BEFORE we set the new
// parent.
//
HWND hwndNext = ::GetWindow(hwnd, GW_HWNDNEXT);
::SetParent(hwnd, hwndNewParent);
hwnd = hwndNext;
}
}
LRESULT CDesktopBrowser::RaisedWndProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
{
CDesktopBrowser *psb = (CDesktopBrowser*)GetWindowLongPtr(hwnd, 0);
return psb->_RaisedWndProc(uMsg, wParam, lParam);
}
LRESULT CDesktopBrowser::_RaisedWndProc(UINT uMsg, WPARAM wParam, LPARAM lParam)
{
switch (uMsg)
{
case WM_ACTIVATE:
break;
case WM_SIZE:
if (wParam == SIZE_MINIMIZED)
ShowWindow(_hwndRaised, SW_RESTORE);
break;
case WM_NOTIFY:
case WM_ERASEBKGND:
goto SendToDesktop;
default:
if (uMsg >= WM_USER)
{
SendToDesktop:
return SendMessage(_pbbd->_hwnd, uMsg, wParam, lParam);
}
else
{
return ::SHDefWindowProc(_hwndRaised, uMsg, wParam, lParam);
}
}
return 0;
}
void CDesktopBrowser::_Raise()
{
RECT rc;
HWND hwndDesktop = GetDesktopWindow();
BOOL fLocked;
HWND hwndLastActive = GetLastActivePopup(_pbbd->_hwnd);
if (SHIsRestricted(NULL, REST_NODESKTOP))
return;
if (hwndLastActive != _pbbd->_hwnd)
{
SetForegroundWindow(hwndLastActive);
return;
}
if (!_hwndRaised)
_hwndRaised = SHCreateWorkerWindow(RaisedWndProc, NULL, WS_EX_TOOLWINDOW, WS_POPUP | WS_CLIPCHILDREN, NULL, this);
//SystemParametersInfo(SPI_GETWORKAREA, 0, &rc, 0);
//rc.left = 0; // need to always start from 0, 0 to get the wallpaper centered right
//rc.top = 0;
fLocked = LockWindowUpdate(hwndDesktop);
_SwapParents(_pbbd->_hwnd, _hwndRaised);
// set the view window to the bottom of the z order
SetWindowPos(_pbbd->_hwndView, HWND_BOTTOM, 0, 0, 0, 0, SWP_NOMOVE | SWP_NOSIZE | SWP_NOACTIVATE);
GetWindowRect(_pbbd->_hwnd, &rc);
SetWindowPos(_hwndRaised, HWND_TOP, rc.left, rc.top, RECTWIDTH(rc), RECTHEIGHT(rc), SWP_SHOWWINDOW);
SetForegroundWindow(_hwndRaised);
if (fLocked)
LockWindowUpdate(NULL);
THR(RegisterDragDrop(_hwndRaised, (IDropTarget *)this));
SetFocus(_pbbd->_hwndView);
//SetWindowPos(_hwndTray, HWND_TOPMOST, 0, 0, 0, 0, SWP_NOACTIVATE | SWP_NOSIZE | SWP_NOMOVE);
_fRaised = TRUE;
}
void CDesktopBrowser::_Lower()
{
BOOL fLocked;
fLocked = LockWindowUpdate(_hwndRaised);
_SwapParents(_hwndRaised, _pbbd->_hwnd);
// set the view window to the bottom of the z order
SetWindowPos(_pbbd->_hwndView, HWND_BOTTOM, 0, 0, 0, 0, SWP_NOMOVE | SWP_NOSIZE | SWP_NOACTIVATE);
ShowWindow(_hwndRaised, SW_HIDE);
if (fLocked)
LockWindowUpdate(NULL);
RevokeDragDrop(_hwndRaised);
_fRaised = FALSE;
}
void CDesktopBrowser::_OnRaise(WPARAM wParam, LPARAM lParam)
{
VARIANTARG vaIn;
VariantInit(&vaIn);
vaIn.vt = VT_I4;
vaIn.lVal = (DWORD) lParam;
switch (lParam) {
case DTRF_RAISE:
_Raise();
_ExecChildren(NULL, TRUE, &CGID_ShellDocView, SHDVID_RAISE, 0, &vaIn, NULL);
break;
case DTRF_LOWER:
_ExecChildren(NULL, TRUE, &CGID_ShellDocView, SHDVID_RAISE, 0, &vaIn, NULL);
_Lower();
break;
}
VariantClear(&vaIn);
if (!wParam) {
wParam = (WPARAM)_hwndTray;
}
PostMessage((HWND)wParam, TM_DESKTOPSTATE, 0, _fRaised ? DTRF_RAISE : DTRF_LOWER);
}
BOOL CDesktopBrowser::_QueryHKCRChanged(HWND hwnd, DWORD *pdwCookie)
{
// we assume that the key has changed if
// we were unable to init correctly
BOOL fRet = TRUE;
ASSERT(pdwCookie);
if (_cChangeEvents)
{
DWORD dw = WaitForMultipleObjects(_cChangeEvents, _rghChangeEvents, FALSE, 0);
dw -= WAIT_OBJECT_0;
// Note: Since "dw" is a DWORD, an underflow in the line above
// will result in dw becoming a huge value, so the test below will
// fail.
if (dw < _cChangeEvents)
{
// this means the key changed...
ResetEvent(_rghChangeEvents[dw]);
_dwChangeCookie = GetTickCount();
PostMessage(hwnd, DTM_SETUPAPPRAN, 0, NULL);
}
//
// if nothing has changed yet, or if nothing has changed
// since the client last checked,
// than this client doesnt need to update its cache
//
if (!_dwChangeCookie || (*pdwCookie && *pdwCookie == _dwChangeCookie))
fRet = FALSE;
// update the cookie
*pdwCookie = _dwChangeCookie;
}
return fRet;
}
// This msg gets posted to us after a setup application runs so that we can
// fix things up.
void CDesktopBrowser::_SetupAppRan(WPARAM wParam, LPARAM lParam)
{
// Lotus 123R5 sometimes gets confused when installing over IE4 and
// they leave their country setting blank. Detect this case and put
// in USA so that they at least boot.
{
TCHAR szPath[MAX_PATH];
GetWindowsDirectory(szPath, ARRAYSIZE(szPath));
if (PathAppend(szPath, TEXT("123r5.ini")) && PathFileExistsAndAttributes(szPath, NULL))
{
TCHAR szData[100];
GetPrivateProfileString(TEXT("Country"), TEXT("driver"), TEXT(""), szData, ARRAYSIZE(szData), TEXT("123r5.ini"));
if (!szData[0])
{
WritePrivateProfileString(TEXT("Country"), TEXT("driver"), TEXT("L1WUSF"), TEXT("123r5.ini"));
}
}
}
// this location in the registry is a good place to cache info
// that needs to be invalided once
SHDeleteKey(HKEY_CURRENT_USER, STRREG_DISCARDABLE STRREG_POSTSETUP);
HKEY hk = SHGetShellKey(SHELLKEY_HKCULM_MUICACHE, NULL, FALSE);
if (hk)
{
SHDeleteKeyA(hk, NULL);
RegCloseKey(hk);
}
// Take this opportunity to freshen our component categories cache:
IShellTaskScheduler* pScheduler ;
if(SUCCEEDED(CoCreateInstance(CLSID_SharedTaskScheduler, NULL, CLSCTX_INPROC_SERVER,
IID_PPV_ARG(IShellTaskScheduler, &pScheduler))))
{
#ifdef _WIN64
// Before creating 64-bit component cache, we should delete the 32-bit cache,
// because we used to erroneously cache 64-bit components in the 32-bit cache,
// and it won't get rebuilt with correc 32-bit components unless we delete it now.
SHDeleteKey(HKEY_CURRENT_USER, STRREG_DISCARDABLE STRREG_POSTSETUP TEXT("\\Component Categories"));
#endif
IRunnableTask* pTask ;
if(SUCCEEDED(CoCreateInstance(CLSID_ComCatCacheTask, NULL, CLSCTX_INPROC_SERVER,
IID_PPV_ARG(IRunnableTask, &pTask))))
{
pScheduler->AddTask(pTask, CLSID_ComCatCacheTask, 0L, ITSAT_DEFAULT_PRIORITY);
pTask->Release(); // Scheduler has AddRef'd him
}
if (SUCCEEDED(CTaskEnumHKCR_Create(&pTask)))
{
pScheduler->AddTask(pTask, CLSID_QueryAssociations, 0L, ITSAT_DEFAULT_PRIORITY);
pTask->Release(); // Scheduler has AddRef'd him
}
pScheduler->Release(); // OK to release global task scheduler.
}
// We get this notification from the OS that a setup app ran.
// Legacy app support for freshly written entries under [extensions] section;
// Scoop these up and shove into registry. (re: bug 140986)
CheckWinIniForAssocs();
}
//-----------------------------------------------------------------------------
struct propagatemsg
{
UINT uMsg;
WPARAM wParam;
LPARAM lParam;
BOOL fSend;
};
BOOL PropagateCallback(HWND hwndChild, LPARAM lParam)
{
struct propagatemsg *pmsg = (struct propagatemsg*)lParam;
if (pmsg->fSend)
SendMessage(hwndChild, pmsg->uMsg, pmsg->wParam, pmsg->lParam);
else
PostMessage(hwndChild, pmsg->uMsg, pmsg->wParam, pmsg->lParam);
return TRUE;
}
void PropagateMessage(HWND hwndParent, UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL fSend)
{
if (!hwndParent)
return;
struct propagatemsg msg = { uMsg, wParam, lParam, fSend };
EnumChildWindows(hwndParent, PropagateCallback, (LPARAM)&msg);
}
void CDesktopBrowser::v_PropagateMessage(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL fSend)
{
// WARNING: We can't propagate the message to a NULL hwnd because it will
// turn into a broadcast. This will come back to us and we will re-send it
// causing an infinite loop. BryanSt.
if (_fRaised && _hwndRaised)
PropagateMessage(_hwndRaised, uMsg, wParam, lParam, fSend);
else if (_pbbd->_hwnd)
PropagateMessage(_pbbd->_hwnd, uMsg, wParam, lParam, fSend);
}
//***
// NOTES
// in the tray case, we actually do SetFocus etc.
HRESULT CDesktopBrowser::v_MayGetNextToolbarFocus(LPMSG lpMsg,
UINT itbCur, int citb,
LPTOOLBARITEM * pptbi, HWND * phwnd)
{
HRESULT hr;
// _fTrayHack?
if (itbCur == ITB_VIEW) {
if (citb == -1) {
TraceMsg(DM_FOCUS, "cdtb.v_mgntf: ITB_VIEW,-1 => tray");
goto Ltray;
}
}
hr = _pbsInner->v_MayGetNextToolbarFocus(lpMsg, itbCur, citb, pptbi, phwnd);
TraceMsg(DM_FOCUS, "cdtb.v_mgntf: SUPER hr=%x", hr);
if (SUCCEEDED(hr)) {
// S_OK: we got and handled a candidate
// S_FALSE: we got a candidate and our parent will finish up
ASSERT(hr != S_OK); // currently never happens (but should work)
return hr;
}
// E_xxx: no candidate
ASSERT(citb == 1 || citb == -1);
*pptbi = NULL;
if (citb == 1) {
Ltray:
*phwnd = _hwndTray;
// AndyP REVIEW: why do we do this here instead of overriding
// _SetFocus and letting commonsb call that function? Sure, this
// is one less override, but why have different code paths?
SendMessage(_hwndTray, TM_UIACTIVATEIO, TRUE, citb);
return S_OK;
}
else
{
//Lview:
*phwnd = _pbbd->_hwndView;
return S_FALSE;
}
/*NOTREACHED*/
ASSERT(0);
}
//
// NOTE: Please think before calling this function, in a multi-monitor system this function
// returns TRUE if you are within a certain edge for any monitor, so puEdge means puEdge
// of a certain monitor instead of the whole desktop. -- dli
//
BOOL CDesktopBrowser::_PtOnDesktopEdge(POINTL* ppt, LPUINT puEdge)
{
RECT rcMonitor;
POINT pt = {ppt->x, ppt->y};
HMONITOR hMon = MonitorFromPoint(pt, MONITOR_DEFAULTTONULL);
// We got this point from drop, so it definitely should belong to a valid monitor -- dli
ASSERT(hMon);
GetMonitorRect(hMon, &rcMonitor);
// if it's near/on the edge on this monitor
if (ppt->x < rcMonitor.left + g_cxEdge) {
*puEdge = ABE_LEFT;
} else if (ppt->x > rcMonitor.right - g_cxEdge) {
*puEdge = ABE_RIGHT;
} else if (ppt->y < rcMonitor.top + g_cyEdge) {
*puEdge = ABE_TOP;
} else if (ppt->y > rcMonitor.bottom - g_cyEdge) {
*puEdge = ABE_BOTTOM;
} else {
*puEdge = (UINT)-1;
return FALSE;
}
return TRUE;
}
UINT g_cfDeskBand = 0;
HRESULT CDesktopBrowser::DragEnter(IDataObject *pdtobj, DWORD grfKeyState, POINTL ptl, DWORD *pdwEffect)
{
ASSERT(pdtobj);
if (!g_cfDeskBand)
g_cfDeskBand = RegisterClipboardFormat(TEXT("DeskBand"));
FORMATETC fmte = { (CLIPFORMAT) g_cfDeskBand, NULL, 0, -1, TYMED_ISTREAM};
if (pdtobj->QueryGetData(&fmte) == S_OK) {
_dwEffectOnEdge = DROPEFFECT_COPY | DROPEFFECT_MOVE;
} else {
_dwEffectOnEdge = DROPEFFECT_NONE;
}
_grfKeyState = grfKeyState;
HRESULT hr;
if (_pdtInner)
hr = _pdtInner->DragEnter(pdtobj, grfKeyState, ptl, pdwEffect);
else
{
DAD_DragEnterEx3(_pbbd->_hwndView, ptl, pdtobj);
hr = S_OK;
}
return hr;
}
HRESULT CDesktopBrowser::DragOver(DWORD grfKeyState, POINTL ptl, DWORD *pdwEffect)
{
HRESULT hr = S_OK;
DAD_DragMoveEx(_pbbd->_hwndView, ptl);
_grfKeyState = grfKeyState;
if (_dwEffectOnEdge != DROPEFFECT_NONE) {
*pdwEffect &= _dwEffectOnEdge;
return S_OK;
}
if (_pdtInner)
hr = _pdtInner->DragOver(grfKeyState, ptl, pdwEffect);
return hr;
}
HRESULT DeskBarApp_Create(IUnknown** ppunkBar, IUnknown** ppunkBandSite)
{
IDeskBar* pdb;
HRESULT hres = CoCreateInstance(CLSID_DeskBarApp, NULL, CLSCTX_INPROC_SERVER, IID_PPV_ARG(IDeskBar, &pdb));
*ppunkBar = pdb;
if (SUCCEEDED(hres))
{
pdb->GetClient(ppunkBandSite);
}
return hres;
}
//***
// ENTRY/EXIT
// hres AddBand result on success; o.w. failure code
// NOTES
// n.b. on success we *must* return AddBand's hres (which is a dwBandID)
HRESULT CDesktopBrowser::_CreateDeskBarForBand(UINT uEdge, IUnknown *punk, POINTL *pptl, IBandSite **ppbsOut)
{
IBandSite *pbs;
IUnknown *punkBar;
IUnknown *punkBandSite;
HRESULT hres;
#ifdef DEBUG
HRESULT hresRet = -1;
#endif
if (ppbsOut)
*ppbsOut = NULL;
hres = DeskBarApp_Create(&punkBar, &punkBandSite);
if (SUCCEEDED(hres))
{
IDockingBarPropertyBagInit* ppbi;
if (SUCCEEDED(CoCreateInstance(CLSID_CDockingBarPropertyBag, NULL, CLSCTX_INPROC_SERVER,
IID_PPV_ARG(IDockingBarPropertyBagInit, &ppbi))))
{
if ((UINT)uEdge != -1)
{
ppbi->SetDataDWORD(PROPDATA_MODE, WBM_BOTTOMMOST);
ppbi->SetDataDWORD(PROPDATA_SIDE, uEdge);
}
else
{
ppbi->SetDataDWORD(PROPDATA_MODE, WBM_FLOATING);
}
ppbi->SetDataDWORD(PROPDATA_X, pptl->x);
ppbi->SetDataDWORD(PROPDATA_Y, pptl->y);
IPropertyBag * ppb;
if (SUCCEEDED(ppbi->QueryInterface(IID_PPV_ARG(IPropertyBag, &ppb))))
{
SHLoadFromPropertyBag(punkBar, ppb);
punkBandSite->QueryInterface(IID_PPV_ARG(IBandSite, &pbs));
if (pbs)
{
hres = pbs->AddBand(punk);
#ifdef DEBUG
hresRet = hres;
#endif
AddToolbar(punkBar, L"", NULL);
if (ppbsOut)
{
// IUnknown_Set (sort of...)
*ppbsOut = pbs;
(*ppbsOut)->AddRef();
}
pbs->Release();
if (_fRaised)
{
VARIANTARG vaIn = { 0 };
vaIn.vt = VT_I4;
vaIn.lVal = DTRF_RAISE;
ASSERT(punkBar != NULL); // o.w. we'd do a broadcast
_ExecChildren(punkBar, FALSE, &CGID_ShellDocView, SHDVID_RAISE, 0, &vaIn, NULL);
}
}
ppb->Release();
}
ppbi->Release();
}
punkBandSite->Release();
punkBar->Release();
}
ASSERT(hres == hresRet || FAILED(hres));
return hres;
}
HRESULT CDesktopBrowser::Drop(IDataObject *pdtobj, DWORD grfKeyState, POINTL pt, DWORD *pdwEffect)
{
UINT uEdge;
HRESULT hres = E_FAIL;
if (((_PtOnDesktopEdge(&pt, &uEdge) && (_grfKeyState & MK_LBUTTON)) ||
(_dwEffectOnEdge != DROPEFFECT_NONE)) && !SHRestricted(REST_NOCLOSE_DRAGDROPBAND))
{
// if the point is on the edge of the desktop and the item dropped was
// a single url object, then create a webbar
// TODO: (reuse) w/ a little restructuring we might share this code
// w/ CBandSite::Drop etc.
FORMATETC fmte = {(CLIPFORMAT)g_cfDeskBand, NULL, 0, -1, TYMED_ISTREAM};
STGMEDIUM stg;
LPITEMIDLIST pidl;
IUnknown* punk = NULL;
// we can move a band from bar to bar, but we can only copy or link a folder
// because the creation of a band relies on the source still abeing there
if ((*pdwEffect & (DROPEFFECT_COPY | DROPEFFECT_MOVE)) &&
SUCCEEDED(pdtobj->GetData(&fmte, &stg)))
{
// this is a drag of a band from another bar, create it!
hres = OleLoadFromStream(stg.pstm, IID_PPV_ARG(IUnknown, &punk));
if (SUCCEEDED(hres))
{
if (*pdwEffect & DROPEFFECT_COPY)
*pdwEffect = DROPEFFECT_COPY;
else
*pdwEffect = DROPEFFECT_MOVE;
}
ReleaseStgMedium(&stg);
}
else if ((*pdwEffect & (DROPEFFECT_COPY | DROPEFFECT_LINK)) &&
SUCCEEDED(SHPidlFromDataObject(pdtobj, &pidl, NULL, 0)))
{
hres = SHCreateBandForPidl(pidl, &punk, (grfKeyState & (MK_CONTROL | MK_SHIFT)) == (MK_CONTROL | MK_SHIFT));
ILFree(pidl);
if (SUCCEEDED(hres))
{
if (*pdwEffect & DROPEFFECT_LINK)
*pdwEffect = DROPEFFECT_LINK;
else
*pdwEffect = DROPEFFECT_COPY;
}
}
if (SUCCEEDED(hres))
{
if (punk)
{
BOOL fCreateRebarWindow = TRUE;
IDeskBandEx* pdbex;
if (SUCCEEDED(punk->QueryInterface(IID_PPV_ARG(IDeskBandEx, &pdbex))))
{
fCreateRebarWindow = (pdbex->MoveBand() == S_OK) ? TRUE : FALSE;
pdbex->Release();
}
if (fCreateRebarWindow)
{
IBandSite *pbs;
hres = _CreateDeskBarForBand(uEdge, punk, &pt, &pbs);
if (SUCCEEDED(hres))
{
DWORD dwState = IDataObject_GetDeskBandState(pdtobj);
pbs->SetBandState(ShortFromResult(hres), BSSF_NOTITLE, dwState & BSSF_NOTITLE);
pbs->Release();
}
punk->Release();
}
}
IDropTarget *pdtView;
HRESULT hr = E_FAIL;
//Get the view's drop target
if (_pbbd->_psv)
{
hr = _pbbd->_psv->QueryInterface(IID_PPV_ARG(IDropTarget, &pdtView));
if (SUCCEEDED(hr))
{
pdtView->DragLeave();
pdtView->Release();
}
}
return hres;
}
// if we failed, pass this on to our child.
// this allows things like d/d of wallpaper to the edge to do
// right thing
}
if (_pdtInner)
hres = _pdtInner->Drop(pdtobj, grfKeyState, pt, pdwEffect);
return hres;
}
BOOL CDesktopBrowser::_OnCopyData(PCOPYDATASTRUCT pcds)
{
IETHREADPARAM *piei = SHCreateIETHREADPARAM(NULL, (ULONG)pcds->dwData, NULL, NULL);
if (piei)
{
LPCWSTR pwszSrc = (LPCWSTR)pcds->lpData;
LPCWSTR pwszDdeRegEvent = NULL;
LPCWSTR pwszCloseEvent = NULL;
DWORD cchSrc = pcds->cbData / sizeof(WCHAR);
piei->uFlags = COF_NORMAL | COF_WAITFORPENDING | COF_IEXPLORE;
//
// Remember where the command line parameters are.
//
LPCWSTR pszCmd = pwszSrc;
int cch = lstrlenW(pwszSrc) + 1;
pwszSrc += cch;
cchSrc -= cch;
TraceMsg(TF_SHDAUTO, "CDB::_OnCopyData got %hs", pszCmd);
//
// Get the dde reg event name into piei->szDdeRegEvent.
//
// NOTE: this is now conditional because we now launch the channel band from the desktop
// NOTE: as a fake WM_COPYDATA command
if (cchSrc)
{
ASSERT(cchSrc);
pwszDdeRegEvent = pwszSrc;
StrCpyNW(piei->szDdeRegEvent, pwszSrc, ARRAYSIZE(piei->szDdeRegEvent));
cch = lstrlenW(pwszSrc) + 1;
pwszSrc += cch;
cchSrc -= cch;
piei->uFlags |= COF_FIREEVENTONDDEREG;
//
// Get the name of the event to fire on close, if any.
//
if (cchSrc)
{
pwszCloseEvent = pwszSrc;
StrCpyNW(piei->szCloseEvent, pwszSrc, ARRAYSIZE(piei->szCloseEvent));
cch = lstrlenW(pwszSrc) + 1;
pwszSrc += cch;
cchSrc -= cch;
piei->uFlags |= COF_FIREEVENTONCLOSE;
}
}
ASSERT(cchSrc == 0);
if (pszCmd && pszCmd[0])
{
// for compatibility with apps that spawn the browser with a command line
// tell wininet to refresh its proxy settings. (this is particularly needed
// for TravelSoft WebEx)
MyInternetSetOption(NULL, INTERNET_OPTION_SETTINGS_CHANGED, NULL, 0);
if (SHParseIECommandLine(&pszCmd, piei))
piei->uFlags |= COF_NOFINDWINDOW;
if (pszCmd[0] && FAILED(_ConvertPathToPidlW(_pbsInner, _pbbd->_hwnd, pszCmd, &piei->pidl)))
piei->pidl = NULL;
}
else
{
piei->fCheckFirstOpen = TRUE;
}
// SHOpenFolderWindow takes ownership of piei
BOOL fRes = SHOpenFolderWindow(piei);
if (!fRes)
{
//
// Something went wrong creating the browser,
// let's fire all the events ourselves.
//
if (pwszDdeRegEvent)
FireEventSz(pwszDdeRegEvent);
if (pwszCloseEvent)
FireEventSz(pwszCloseEvent);
}
}
else
{
TraceMsg(TF_WARNING, "OnCopyData unable to create IETHREADPARAM");
}
return TRUE;
}
BOOL CDesktopBrowser::_InitScheduler(void)
{
if (!_psched)
{
// get the system background scheduler thread
CoCreateInstance(CLSID_SharedTaskScheduler, NULL, CLSCTX_INPROC,
IID_PPV_ARG(IShellTaskScheduler, &_psched));
}
return (_psched != NULL);
}
HRESULT CDesktopBrowser::_AddDesktopTask(IRunnableTask *ptask, DWORD dwPriority)
{
if (_InitScheduler())
return _psched->AddTask(ptask, TOID_Desktop, 0, dwPriority);
return E_FAIL;
}
void CDesktopBrowser::_OnAddToRecent(HANDLE hMem, DWORD dwProcId)
{
IRunnableTask *ptask;
if (SUCCEEDED(CTaskAddDoc_Create(hMem, dwProcId, &ptask)))
{
_AddDesktopTask(ptask, ITSAT_DEFAULT_PRIORITY);
ptask->Release();
}
}
//
// local server handling
//
DWORD WINAPI _LocalServerThread(void *pv)
{
LOCALSERVERDATA *ptd = (LOCALSERVERDATA *)pv;
CLSID clsid = *c_localServers[ptd->iLocalServer];
IUnknown *punk;
if (SUCCEEDED(DllGetClassObject(clsid, IID_PPV_ARG(IUnknown, &punk))))
{
DWORD dwReg;
if (SUCCEEDED(CoRegisterClassObject(clsid, punk, CLSCTX_LOCAL_SERVER, REGCLS_MULTIPLEUSE, &dwReg)))
{
__try
{
MSG msg;
while(GetMessage(&msg, NULL, 0, 0))
{
TranslateMessage(&msg);
DispatchMessage(&msg);
}
}
__except (EXCEPTION_EXECUTE_HANDLER)
{
// we only send the message on an exception,
// because otherwise we got a WM_QUIT which
// means the desktop is being destroyed.
PostMessage(GetShellWindow(), CWM_CREATELOCALSERVER, FALSE, ptd->iLocalServer);
}
CoRevokeClassObject(dwReg);
}
}
LocalFree(ptd);
return 0;
}
DWORD WINAPI _SetThreadID(void *pv)
{
LOCALSERVERDATA *ptd = (LOCALSERVERDATA *)pv;
*ptd->pdwThreadID = GetCurrentThreadId();
return 0;
}
STDAPI_(HWND) SCNGetWindow(BOOL fUseDesktop);
LRESULT CDesktopBrowser::WndProcBS(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
{
INSTRUMENT_WNDPROC(SHCNFI_DESKTOP_WNDPROC, _pbbd->_hwnd, uMsg, wParam, lParam);
ASSERT(IsWindowTchar(hwnd));
switch (uMsg)
{
#ifdef DEBUG
case WM_QUERYENDSESSION:
TraceMsg(DM_SHUTDOWN, "cdtb.wp: WM_QUERYENDSESSION");
goto DoDefault;
#endif
case WM_ENDSESSION:
TraceMsg(DM_SHUTDOWN, "cdtb.wp: WM_ENDSESSION wP=%d lP=%d", wParam, lParam);
if (wParam)
{
#ifdef DEBUG
// call out inner an give him a chance to set _fMightBeShuttingDown so we
// don't assert later on in the case where the system is shutting down
_pbsInner->WndProcBS(hwnd, uMsg, wParam, lParam);
#endif
SHELLSTATE ss = {0};
// When we shut down, if the desktop is in WebView, we leave some temp
// files undeleted because we never get the WM_DESTROY message below.
// So, I just destroy the shellview which in turn destroys the temp
// file here. Note: This is done only if we are in web view.
SHGetSetSettings(&ss, SSF_DESKTOPHTML, FALSE); //Get the desktop_html flag
if (ss.fDesktopHTML)
{
ReleaseShellView();
}
g_pdtray->SetVar(SVTRAY_EXITEXPLORER, FALSE); // don't exit process
// flush log before we exit
if (StopWatchMode())
{
StopWatchFlush();
}
// Kill this window so that we free active desktop threads properly
DestroyWindow(hwnd);
}
TraceMsg(DM_SHUTDOWN, "cdtb.wp: WM_ENDSESSION return 0");
break;
case WM_ERASEBKGND:
PaintDesktop((HDC)wParam);
return 1;
case WM_TIMER:
switch (wParam)
{
case IDT_DDETIMEOUT:
DDEHandleTimeout(_pbbd->_hwnd);
break;
case IDT_ENUMHKCR:
KillTimer(_pbbd->_hwnd, IDT_ENUMHKCR);
{
IShellTaskScheduler* pScheduler ;
if(SUCCEEDED(CoCreateInstance(CLSID_SharedTaskScheduler, NULL, CLSCTX_INPROC_SERVER,
IID_PPV_ARG(IShellTaskScheduler, &pScheduler))))
{
IRunnableTask* pTask ;
if (SUCCEEDED(CTaskEnumHKCR_Create(&pTask)))
{
pScheduler->AddTask(pTask, CLSID_QueryAssociations, 0L, ITSAT_DEFAULT_PRIORITY);
pTask->Release(); // Scheduler has AddRef'd him
}
pScheduler->Release(); // OK to release global task scheduler.
}
}
break;
default:
goto DoDefault;
break;
}
break;
case WM_SHELLNOTIFY:
switch(wParam)
{
case SHELLNOTIFY_WALLPAPERCHANGED:
// this is done only to the shell window when someone sets
// the wall paper but doesn't specify to broadcast
_pbsInner->ForwardViewMsg(uMsg, wParam, lParam);
break;
}
break;
case WM_PALETTECHANGED:
case WM_QUERYNEWPALETTE:
//
// in Win95 the desktop wndproc will invalidate the shell window when
// a palette change occurs so we didn't have to do anything here before
//
// in Nashville the desktop can be HTML and needs the palette messages
//
// so now we fall through and propagate...
//
case WM_ACTIVATEAPP:
if (!_pbbd->_hwndView)
goto DoDefault;
return _pbsInner->ForwardViewMsg(uMsg, wParam, lParam);
case WM_DEVICECHANGE:
_pbsInner->ForwardViewMsg(uMsg, wParam, lParam);
_OnDeviceBroadcast(_pbbd->_hwnd, wParam, (DEV_BROADCAST_HDR *)lParam);
goto DoDefault;
case WM_WINDOWPOSCHANGING:
#define ppos ((LPWINDOWPOS)lParam)
ppos->x = g_xVirtualScreen;
ppos->y = g_yVirtualScreen;
ppos->cx = g_cxVirtualScreen;
ppos->cy = g_cyVirtualScreen;
break;
case WM_HOTKEY:
// NOTE: forward hotkeys to the tray. This fixes the logitech mouseman who sends
// NOTE: hotkeys directly to the desktop.
// SPECIAL NOTE: the offset for GHID_FIRST is added because hotkeys that are sent to the
// SPECIAL NOTE: desktop are not proper hotkeys generated from the keyboard, they are
// SPECIAL NOTE: sent by an app, and the IDs have changed since win95....
ASSERT(g_hwndTray);
ASSERT(wParam < GHID_FIRST);
PostMessage(g_hwndTray, uMsg, wParam + GHID_FIRST, lParam);
return 0;
case WM_SYSCOMMAND:
switch (wParam & 0xFFF0) {
// NB Dashboard 1.0 sends a WM_SYSCOMMAND SC_CLOSE to the desktop when it starts up.
// What it was trying to do was to close down any non-shell versions of Progman. The
// proper shell version would just ignore the close. Under Chicago, they think that
// the desktop is Progman and send it the close, so we put up the exit windows dialog!
// Dashboard 2.0 has been fixed to avoid this bogisity.
case SC_CLOSE:
break;
// America alive tries to minimise Progman after installing - they end up minimising
// the desktop on Chicago!
case SC_MINIMIZE:
break;
default:
goto DoDefault;
}
break;
case WM_SETCURSOR:
// REVIEW: is this really needed?
if (_iWaitCount)
{
SetCursor(LoadCursor(NULL, IDC_APPSTARTING));
return TRUE;
}
else
goto DoDefault;
case WM_CLOSE:
SendMessage(_hwndTray, TM_DOEXITWINDOWS, 0, 0);
return 0;
// REVIEW: do we need this, can all of these cases be the same?
case WM_DRAWITEM:
case WM_MEASUREITEM:
if (!_pbsInner->ForwardViewMsg(uMsg, wParam, lParam))
goto DoDefault;
break;
case WM_INITMENUPOPUP:
case WM_ENTERMENULOOP:
case WM_EXITMENULOOP:
// let the fsview deal with any popup menus it created
_pbsInner->ForwardViewMsg(uMsg, wParam, lParam);
break;
// Looking at messages to try to capture when the workarea may
// have changed...
case WM_DISPLAYCHANGE:
lParam = 0;
if (GetNumberOfMonitors() != _nMonitors)
_InitMonitors();
// fall through
case WM_WININICHANGE:
_InitDesktopMetrics(wParam, (LPCTSTR)lParam);
if (wParam == SPI_SETNONCLIENTMETRICS)
{
VARIANTARG varIn;
VARIANTARG varOut = {0};
varIn.vt = VT_BOOL;
varIn.boolVal = VARIANT_TRUE;
_pctInner->Exec(&CGID_Explorer, SBCMDID_CACHEINETZONEICON, OLECMDEXECOPT_DODEFAULT , &varIn, &varOut);
}
if (lParam)
{
if (lstrcmpi((LPCTSTR)lParam, TEXT("Extensions")) == 0)
{
// Post a message to our selves so we can do more stuff
// slightly delayed.
PostMessage(hwnd, DTM_SETUPAPPRAN, 0, 0);
}
else if (lstrcmpi((LPCTSTR)lParam, TEXT("ShellState")) == 0)
{
// this should cover external apps changing
// our settings.
SHRefreshSettings();
}
else
{
// SPI_GETICONTITLELONGFONT is broadcast from IE when the home page is changed. We look for that so
// we can be sure to update the MyCurrentHomePage component.
if((wParam == SPI_SETDESKWALLPAPER) || (wParam == SPI_SETDESKPATTERN) || (wParam == SPI_GETICONTITLELOGFONT))
{
// Some desktop attribute has changed. So, regenerate desktop.
if(lstrcmpi((LPCTSTR)lParam, TEXT("ToggleDesktop")) &&
lstrcmpi((LPCTSTR)lParam, TEXT("RefreshDesktop")) &&
lstrcmpi((LPCTSTR)lParam, TEXT("BufferedRefresh")))
{
DWORD dwFlags = AD_APPLY_HTMLGEN | AD_APPLY_REFRESH;
switch (wParam)
{
case SPI_SETDESKPATTERN:
dwFlags |= (AD_APPLY_FORCE | AD_APPLY_DYNAMICREFRESH);
break;
case SPI_SETDESKWALLPAPER:
dwFlags |= AD_APPLY_SAVE;
break;
case SPI_GETICONTITLELOGFONT:
dwFlags |= AD_APPLY_FORCE;
break;
}
PokeWebViewDesktop(dwFlags);
// If we are not currently in ActiveDesktop Mode, then we need to set the dirty bit
// sothat a new HTML file will be generated showing the new wallpaper,
// the next time the active desktop is turned ON!
if (wParam == SPI_SETDESKWALLPAPER)
{
SHELLSTATE ss = {0};
SHGetSetSettings(&ss, SSF_DESKTOPHTML, FALSE); //Get the desktop_html flag
if (!ss.fDesktopHTML)
SetDesktopFlags(COMPONENTS_DIRTY, COMPONENTS_DIRTY);
}
}
}
}
}
// control panel applet that allows users to change the
// environment with-which to spawn new applications. On NT, we need
// to pick up that environment change so that anything we spawn in
// the future will pick up those updated environment values.
//
if (lParam && (lstrcmpi((LPTSTR)lParam, TEXT("Environment")) == 0))
{
void *pv;
RegenerateUserEnvironment(&pv, TRUE);
}
v_PropagateMessage(uMsg, wParam, lParam, TRUE);
SetWindowPos(_pbbd->_hwnd, NULL, 0, 0, 0, 0, SWP_NOACTIVATE | SWP_NOZORDER);
if ((uMsg == WM_DISPLAYCHANGE) || (wParam == SPI_SETWORKAREA))
_SetViewArea();
SHSettingsChanged(wParam, lParam);
break;
case WM_SYSCOLORCHANGE:
//
// In memphis, when apps go into low-res mode, we get this message. We should not re-generate
// desktop.htt in this scenario, or else the centered wallpaper gets messed up because we do not
// get the message when the app exists and the resoultion goes up. So, we make the following check.
//
if(!SHIsTempDisplayMode())
OnDesktopSysColorChange();
//This is done sothat the defview can set the listview in proper
//colors.
_pbsInner->ForwardViewMsg(uMsg, wParam, lParam);
break;
// Don't go to default wnd proc for this one...
case WM_INPUTLANGCHANGEREQUEST:
if (wParam)
goto DoDefault;
else
return 0;
case WM_COPYDATA:
return _OnCopyData((PCOPYDATASTRUCT)lParam);
case CWM_COMMANDLINE:
SHOnCWMCommandLine(lParam);
break;
case CWM_GETSCNWINDOW:
return (LRESULT) SCNGetWindow(FALSE);
break;
case CWM_ADDTORECENT:
_OnAddToRecent((HANDLE)wParam, (DWORD) lParam);
return 0;
case CWM_WAITOP:
SHWaitOp_Operate((HANDLE)wParam, (DWORD)lParam);
return 0;
case CWM_CREATELOCALSERVER:
{
// wParam = TRUE to create, FALSE to mark as terminated
// lParam = (INT) index into LocalServer table
INT i = (INT)lParam;
if ((i >= 0) && (i < ARRAYSIZE(_idLocalServerThreads)))
{
if (wParam)
{
if (_idLocalServerThreads[i] == -1)
{
LOCALSERVERDATA *ptd = (LOCALSERVERDATA *)LocalAlloc(LPTR, sizeof(*ptd));
if (ptd)
{
ptd->iLocalServer = i;
ptd->pdwThreadID = &_idLocalServerThreads[i];
if (!SHCreateThread(_LocalServerThread, ptd, CTF_COINIT, _SetThreadID))
{
LocalFree(ptd);
}
}
}
}
else
{
_idLocalServerThreads[i] = -1;
}
}
return 0;
}
case CWM_SHOWFOLDEROPT:
switch (wParam)
{
case CWMW_FOLDEROPTIONS: // appwiz.cpl used to send this message to us
// now it's done by FolderOptionsRunDll
DoGlobalFolderOptions();
break;
case CWMW_TASKBAROPTIONS: // TaskbarOptionsRunDll sends this message
PostMessage(_hwndTray, TM_DOTRAYPROPERTIES, 0, 0);
break;
default:
break;
}
return 0;
case CWM_TASKBARWAKEUP:
_dwThreadIdTray = (DWORD)wParam;
if (_dwThreadIdTray)
{
_iTrayPriority = HIWORD(lParam);
SetTimer(hwnd, IDT_TASKBARWAKEUP, LOWORD(lParam), NULL);
}
break;
case DTM_CREATESAVEDWINDOWS:
_InitDeskbars();
SHCreateSavedWindows();
#ifdef ENABLE_CHANNELS
_MaybeLaunchChannelBand();
#endif
// we need to update the recycle bin icon because the recycle bin
// is per user on NTFS, and thus the state could change w/ each new user
// who logs in.
SHUpdateRecycleBinIcon();
break;
case DTM_SAVESTATE:
TraceMsg(DM_SHUTDOWN, "cdtb.wp: DTM_SAVESTATE");
_SaveState();
break;
case DTM_RAISE:
_OnRaise(wParam, lParam);
break;
#ifdef DEBUG
case DTM_NEXTCTL:
#endif
case DTM_UIACTIVATEIO:
case DTM_ONFOCUSCHANGEIS:
_OnFocusMsg(uMsg, wParam, lParam);
break;
case DTM_SETUPAPPRAN:
_SetupAppRan(wParam, lParam);
break;
case DTM_QUERYHKCRCHANGED:
// some clients are out of process, so we
// can cache their cookies for them.
if (!lParam && wParam > QHKCRID_NONE && wParam < QHKCRID_MAX)
lParam = (LPARAM)&_rgdwQHKCRCookies[wParam - QHKCRID_MIN];
return _QueryHKCRChanged(hwnd, (LPDWORD)lParam);
case DTM_UPDATENOW:
UpdateWindow(hwnd);
break;
case DTM_GETVIEWAREAS:
{
// wParam is an in/out param. in - the max. # of areas, out - the actual # of areas.
// if "in" value < "out" value, lParam is not set.
// The ViewAreas are already stored in the desktop Listview.
int* pnViewAreas = (int*) wParam;
LPRECT lprcViewAreas = (LPRECT) lParam;
if (pnViewAreas)
{
int nMaxAreas = *pnViewAreas;
HWND hwndList = _GetDesktopListview();
ASSERT(IsWindow(hwndList));
ListView_GetNumberOfWorkAreas(hwndList, pnViewAreas);
if (*pnViewAreas >= 0 && *pnViewAreas <= nMaxAreas && lprcViewAreas)
{
ListView_GetWorkAreas(hwndList, *pnViewAreas, lprcViewAreas);
// These are in Listview co-ordinates. We have to map them to screen co-ordinates.
// [msadek]; MapWindowPoints() is mirroring-aware only if you pass two points
for (int i = 0; i < *pnViewAreas; i++)
{
MapWindowPoints(hwndList, HWND_DESKTOP, (LPPOINT)(&lprcViewAreas[i]), 2);
}
}
}
}
break;
case DTM_MAKEHTMLCHANGES:
//Make changes to desktop's HTML using Dynamic HTML
SendMessage(_pbbd->_hwndView, WM_DSV_DESKHTML_CHANGES, wParam, lParam);
break;
case DTM_STARTPAGEONOFF:
SendMessage(_pbbd->_hwndView, WM_DSV_STARTPAGE_TURNONOFF, wParam, lParam);
break;
case DTM_REFRESHACTIVEDESKTOP:
// This message is posted to refresh active desktop in a delayed fashion.
REFRESHACTIVEDESKTOP();
break;
case DTM_SETAPPSTARTCUR:
{
INotifyAppStart * pnasTop;
HRESULT hr = QueryService(SID_STopLevelBrowser, IID_PPV_ARG(INotifyAppStart, &pnasTop));
if (SUCCEEDED(hr))
{
pnasTop->AppStarting();
pnasTop->Release();
}
}
break;
// Handle DDE messages for badly written apps (that assume the shell's
// window is of class Progman and called Program Manager.
case WM_DDE_INITIATE:
case WM_DDE_TERMINATE:
case WM_DDE_ADVISE:
case WM_DDE_UNADVISE:
case WM_DDE_ACK:
case WM_DDE_DATA:
case WM_DDE_REQUEST:
case WM_DDE_POKE:
case WM_DDE_EXECUTE:
return DDEHandleMsgs(_pbbd->_hwnd, uMsg, wParam, lParam);
break;
default:
if (uMsg == GetDDEExecMsg())
{
ASSERT(lParam && 0 == ((LPNMHDR)lParam)->idFrom);
DDEHandleViewFolderNotify(NULL, _pbbd->_hwnd, (LPNMVIEWFOLDER)lParam);
LocalFree((LPNMVIEWFOLDER)lParam);
return TRUE;
}
DoDefault:
return _pbsInner->WndProcBS(hwnd, uMsg, wParam, lParam);
}
return 0;
}
#ifdef ENABLE_CHANNELS
// launch the channelbar, this is called when the desktop has finished starting up.
const WCHAR c_szwChannelBand[] = L"-channelband";
void CDesktopBrowser::_MaybeLaunchChannelBand()
{
DWORD dwType = REG_SZ;
TCHAR szYesOrNo[20];
DWORD cbSize = sizeof(szYesOrNo);
BOOL bLaunchChannelBar = FALSE;
if(SHRegGetUSValue(TEXT("Software\\Microsoft\\Internet Explorer\\Main"), TEXT("Show_ChannelBand"),
&dwType, (void *)szYesOrNo, &cbSize, FALSE, NULL, 0) == ERROR_SUCCESS)
{
bLaunchChannelBar = !lstrcmpi(szYesOrNo, TEXT("yes"));
}
// Don't launch by default post IE4
//else if (IsOS(OS_WINDOWS))
//{
// bLaunchChannelBar = TRUE; // launch channel bar by default on Memphis and win95
//}
if (bLaunchChannelBar)
{
// fake up a WM_COPYDATA struct
COPYDATASTRUCT cds;
cds.dwData = SW_NORMAL;
cds.cbData = sizeof(c_szwChannelBand);
cds.lpData = (void *) c_szwChannelBand;
// fake it as if we had launched iexplore.exe, it saves us a whole process doing it this way....
_OnCopyData(&cds);
}
}
#endif // ENABLE_CHANNELS
//***
// NOTES
// REARCHITECT should this be CBaseBrowser::IInputObject::UIActIO etc.?
HRESULT CDesktopBrowser::_OnFocusMsg(UINT uMsg, WPARAM wParam, LPARAM lParam)
{
BOOL fActivate = (BOOL) wParam;
switch (uMsg) {
case DTM_UIACTIVATEIO:
fActivate = (BOOL) wParam;
TraceMsg(DM_FOCUS, "cdtb.oxiois: DTM_UIActIO fAct=%d dtb=%d", fActivate, (int) lParam);
if (fActivate)
{
MSG msg = {_pbbd->_hwnd, WM_KEYDOWN, VK_TAB, 0xf0001};
BOOL bShift = (GetAsyncKeyState(VK_SHIFT) < 0);
if (bShift)
{
int cToolbars = _pbsInner->_GetToolbarCount();
while (--cToolbars >= 0)
{
// activate last toolbar in tab order
LPTOOLBARITEM ptbi = _pbsInner->_GetToolbarItem(cToolbars);
if (ptbi && ptbi->ptbar)
{
IInputObject* pio;
if (SUCCEEDED(ptbi->ptbar->QueryInterface(IID_PPV_ARG(IInputObject, &pio))))
{
pio->UIActivateIO(TRUE, &msg);
pio->Release();
return S_OK;
}
}
}
}
#ifdef KEYBOARDCUES
// Since we are Tab or Shift Tab we should turn the focus rect on.
SendMessage(_pbbd->_hwnd, WM_UPDATEUISTATE, MAKEWPARAM(UIS_CLEAR,
UISF_HIDEFOCUS), 0);
#endif
// activate view
if (bShift && _pbbd->_psv)
{
_pbbd->_psv->TranslateAccelerator(&msg);
}
else
{
_pbsInner->_SetFocus(NULL, _pbbd->_hwndView, NULL);
}
}
else {
Ldeact:
// if we don't have focus, we're fine;
// if we do have focus, there's nothing we can do about it...
/*NOTHING*/
;
#ifdef DEBUG
TraceMsg(DM_FOCUS, "cdtb.oxiois: GetFocus()=%x _pbbd->_hwnd=%x _pbbd->_hwndView=%x", GetFocus(), _pbbd->_hwnd, _pbbd->_hwndView);
#endif
}
break;
case DTM_ONFOCUSCHANGEIS:
TraceMsg(DM_FOCUS, "cdtb.oxiois: DTM_OnFocChgIS hwnd=%x fAct=%d", (HWND) lParam, fActivate);
if (fActivate) {
// someone else is activating, so we need to deactivate
goto Ldeact;
}
break;
default:
ASSERT(0);
break;
}
return S_OK;
}
LRESULT CALLBACK CDesktopBrowser::DesktopWndProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
{
CDesktopBrowser *psb = (CDesktopBrowser*)GetWindowLongPtr(hwnd, 0);
switch(uMsg)
{
case WM_CREATE:
#ifdef KEYBOARDCUES
// Initialize our keyboard cues bits
SendMessage(hwnd, WM_CHANGEUISTATE, MAKEWPARAM(UIS_INITIALIZE, 0), 0);
#endif
// Set the localized name of the Desktop so it can be used in Error messages
// that have the desktop window as the title.
if (EVAL(LoadStringW(HINST_THISDLL, IDS_DESKTOP, psb->_wzDesktopTitle, ARRAYSIZE(psb->_wzDesktopTitle))))
{
EVAL(SetProp(hwnd, TEXT("pszDesktopTitleW"), (HANDLE)psb->_wzDesktopTitle));
}
if (psb)
return psb->WndProcBS(hwnd, uMsg, wParam, lParam);
else
return DefWindowProc(hwnd, uMsg, wParam, lParam); // known charset
#ifdef KEYBOARDCUES
case WM_ACTIVATE:
if (WA_INACTIVE == LOWORD(wParam))
{
SendMessage(hwnd, WM_CHANGEUISTATE,
MAKEWPARAM(UIS_SET, UISF_HIDEFOCUS | UISF_HIDEACCEL), 0);
}
goto DoDefault;
break;
#endif
case WM_NCCREATE:
ASSERT(psb == NULL);
CDesktopBrowser_CreateInstance(hwnd, (void **)&psb);
if (psb)
{
SetWindowLongPtr(hwnd, 0, (LONG_PTR)psb);
goto DoDefault;
}
else
{
return FALSE;
}
break;
case WM_TIMER:
if (psb)
{
switch(wParam)
{
case IDT_STARTBACKGROUNDSHELLTASKS:
KillTimer(hwnd, IDT_STARTBACKGROUNDSHELLTASKS);
psb->StartBackgroundShellTasks();
break;
case IDT_TASKBARWAKEUP:
KillTimer(hwnd, IDT_TASKBARWAKEUP);
psb->TaskbarWakeup();
break;
default:
return psb->WndProcBS(hwnd, uMsg, wParam, lParam);
break;
}
}
break;
case WM_NCDESTROY:
if (psb)
{
RemoveProp(hwnd, TEXT("pszDesktopTitleW"));
// In case someone does a get shell window and post a WM_QUIT, we need to
// make sure that we also close down our other thread.
TraceMsg(DM_SHUTDOWN, "cdtb.wp(WM_NCDESTROY): ?post WM_QUIT hwndTray=%x(IsWnd=%d)", psb->_hwndTray, IsWindow(psb->_hwndTray));
if (psb->_hwndTray && IsWindow(psb->_hwndTray))
PostMessage(psb->_hwndTray, WM_QUIT, 0, 0);
psb->ReleaseShellView();
psb->Release();
}
PostQuitMessage(0); // exit out message loop
break;
default:
if (psb)
return psb->WndProcBS(hwnd, uMsg, wParam, lParam);
else {
DoDefault:
return DefWindowProc(hwnd, uMsg, wParam, lParam); // known charset
}
}
return 0;
}
void RegisterDesktopClass()
{
WNDCLASS wc = {0};
wc.style = CS_DBLCLKS;
wc.lpfnWndProc = CDesktopBrowser::DesktopWndProc;
//wc.cbClsExtra = 0;
wc.cbWndExtra = sizeof(void *);
wc.hInstance = HINST_THISDLL;
//wc.hIcon = NULL;
wc.hCursor = GetClassCursor(GetDesktopWindow());
wc.hbrBackground = (HBRUSH)(COLOR_DESKTOP + 1);
//wc.lpszMenuName = NULL;
wc.lpszClassName = TEXT(STR_DESKTOPCLASS);
RegisterClass(&wc);
}
#define PEEK_NORMAL 0
#define PEEK_QUIT 1
#define PEEK_CONTINUE 2
#define PEEK_CLOSE 3
// RETURNS BOOL whehter to continue the search or not.
// so FALSE means we've found one.
// TRUE means we haven't.
BOOL CALLBACK FindBrowserWindow_Callback(HWND hwnd, LPARAM lParam)
{
if (IsExplorerWindow(hwnd) || IsFolderWindow(hwnd) || IsTrayWindow(hwnd))
{
DWORD dwProcID;
GetWindowThreadProcessId(hwnd, &dwProcID);
if (dwProcID == GetCurrentProcessId())
{
if (lParam)
*((BOOL*)lParam) = TRUE; // found one!
return FALSE; // stop search
}
}
return TRUE; // continue search
}
#define IsBrowserWindow(hwnd) !FindBrowserWindow_Callback(hwnd, NULL)
BOOL CALLBACK CloseWindow_Callback(HWND hwnd, LPARAM lParam)
{
if (IsBrowserWindow(hwnd)) {
TraceMsg(DM_SHUTDOWN, "s.cw_cb: post WM_CLOSE hwnd=%x", hwnd);
PostMessage(hwnd, WM_CLOSE, 0, 0);
}
return TRUE;
}
UINT CDesktopBrowser::_PeekForAMessage()
{
MSG msg;
BOOL fPeek;
fPeek = PeekMessage(&msg, NULL, 0, 0, PM_REMOVE);
if (fPeek)
{
if (msg.message == WM_QUIT)
{
TraceMsg(DM_SHUTDOWN, "cdtb.pfam: WM_QUIT wP=%d [lP=%d]", msg.wParam, msg.lParam);
if (msg.lParam == 1)
{
return PEEK_CLOSE;
}
TraceMsg(DM_TRACE, "c.ml: Got quit message for %#08x", GetCurrentThreadId());
return PEEK_QUIT; // break all the way out of the main loop
}
if (_pbbd->_hwnd)
{
if (S_OK == _pbsInner->v_MayTranslateAccelerator(&msg))
return PEEK_CONTINUE;
}
TranslateMessage(&msg);
DispatchMessage(&msg);
return PEEK_CONTINUE; // Go back and get the next message
}
return PEEK_NORMAL;
}
void CDesktopBrowser::_MessageLoop()
{
for (; ;)
{
switch (_PeekForAMessage())
{
case PEEK_QUIT:
return;
case PEEK_NORMAL:
{
static BOOL s_fSignaled = FALSE;
if (!s_fSignaled)
{
FireEventSz(TEXT("msgina: ShellReadyEvent"));
s_fSignaled = TRUE;
PERFSETMARK("ExplorerDesktopReady");
}
WaitMessage();
break;
}
case PEEK_CONTINUE:
break;
case PEEK_CLOSE:
// we need to close all the shell windows too
TraceMsg(DM_SHUTDOWN, "cdtb._ml: PEEK_CLOSE, close/wait all");
EnumWindows(CloseWindow_Callback, 0);
{
#define MAXIMUM_DESKTOP_WAIT 15000
DWORD iStartTime = GetTickCount();
// block until all other browser windows are closed
for (;;)
{
BOOL f = FALSE;
EnumWindows(FindBrowserWindow_Callback, (LPARAM)&f);
if (!f || (GetTickCount() - iStartTime > MAXIMUM_DESKTOP_WAIT))
return;
switch (_PeekForAMessage())
{
case PEEK_NORMAL:
// don't do a waitmessage because we want to exit when thelast other window is gone
// and we don't get a message to signal that
Sleep(100);
break;
}
}
}
return;
}
}
}
HRESULT CDesktopBrowser::QueryService(REFGUID guidService, REFIID riid, void **ppvObj)
{
if (IsEqualGUID(guidService, SID_SShellDesktop))
return QueryInterface(riid, ppvObj);
return _pspInner->QueryService(guidService, riid, ppvObj);
}
void CDesktopBrowser::StartBackgroundShellTasks(void)
{
HKEY hkey = SHGetShellKey(SHELLKEY_HKLM_EXPLORER, TEXT("SharedTaskScheduler"), FALSE);
if (hkey)
{
TCHAR szClass[GUIDSTR_MAX];
DWORD cchClass, dwType;
int i = 0;
while (cchClass = ARRAYSIZE(szClass),
ERROR_SUCCESS == RegEnumValue(hkey, i++, szClass, &cchClass, NULL, &dwType, NULL, NULL))
{
CLSID clsid;
if (SUCCEEDED(SHCLSIDFromString(szClass, &clsid)))
{
IRunnableTask* ptask;
if (SUCCEEDED(CoCreateInstance(clsid, NULL, CLSCTX_INPROC, IID_PPV_ARG(IRunnableTask, &ptask))))
{
// Personally I think the start menu should have priority
// over itbar icon extraction, so set this priority list
// a tad lower than the default priority (which start menu is at)
_AddDesktopTask(ptask, ITSAT_DEFAULT_PRIORITY-1);
ptask->Release();
}
}
}
RegCloseKey(hkey);
}
}
void CDesktopBrowser::TaskbarWakeup(void)
{
if (_dwThreadIdTray)
{
HANDLE hThread = OpenThread(THREAD_SET_INFORMATION, FALSE, _dwThreadIdTray);
SetThreadPriority(hThread, _iTrayPriority);
CloseHandle(hThread);
_dwThreadIdTray = 0;
}
}
// create the desktop window and its shell view
DWORD_PTR DesktopWindowCreate(CDesktopBrowser **ppBrowser)
{
*ppBrowser = NULL;
DWORD dwExStyle = WS_EX_TOOLWINDOW;
OleInitialize(NULL);
RegisterDesktopClass();
_InitDesktopMetrics(0, NULL);
dwExStyle |= IS_BIDI_LOCALIZED_SYSTEM() ? dwExStyleRTLMirrorWnd : 0L;
//
// NB This windows class is Progman and it's title is Program Manager. This makes
// sure apps (like ATM) think that program is running and don't fail their install.
//
HWND hwnd = CreateWindowEx(dwExStyle, TEXT(STR_DESKTOPCLASS), TEXT("Program Manager"),
WS_POPUP | WS_CLIPCHILDREN,
g_xVirtualScreen, g_yVirtualScreen,
g_cxVirtualScreen, g_cyVirtualScreen,
NULL, NULL, HINST_THISDLL, NULL);
// The returned hwnd can already be bogus if the system is shutting
// down and user has somehow already called xxxDestroyWindow on us
// even though we never even received a WM_NCCREATE!!
// CreateWindowEx ends up returning a handle to a window that
// has already been destroyed. So in that case, act as if
// CreateWindowEx failed (because it did!)
if (!IsWindow(hwnd)) hwnd = NULL;
if (hwnd)
{
CDesktopBrowser *psb = (CDesktopBrowser*)GetWindowLongPtr(hwnd, 0);
ASSERT(psb);
if (!SHRestricted(REST_NODESKTOP))
{
// do this here to avoid painting the desktop, then repainting
// when the tray appears and causes everything to move
ShowWindow(hwnd, SW_SHOW);
UpdateWindow(hwnd);
}
SetTimer(hwnd, IDT_STARTBACKGROUNDSHELLTASKS, 5 * 1000, NULL);
*ppBrowser = (CDesktopBrowser*)GetWindowLongPtr(hwnd, 0);
}
return (DWORD_PTR)hwnd;
}
STDAPI_(HANDLE) SHCreateDesktop(IDeskTray* pdtray)
{
if (g_dwProfileCAP & 0x00000010)
StartCAP();
ASSERT(pdtray);
ASSERT(g_pdtray==NULL);
g_pdtray = pdtray;
pdtray->AddRef(); // this is no-op, but we want to follow the COM rule.
// put the desktop on the main thread (this is needed for win95 so that
// the DDeExecuteHack() in user.exe works, it needs the desktop and the
// DDE window on the mainthread
CDesktopBrowser *pBrowser;
if (DesktopWindowCreate(&pBrowser))
{
if (g_dwProfileCAP & 0x00040000)
StopCAP();
// hack, cast the object to a handle (otherwise we have to export the class
// declaration so that explorer.exe can use it)
return (HANDLE) pBrowser;
}
return NULL;
}
// nash:49485 (IME focus) and nash:nobug (win95 compat)
// make sure keyboard input goes to the desktop. this is
// a) win95 compat: where focus was on win95 and
// b) nash:49485: focus was in the empty taskband on login so
// the keys went into the bitbucket
//
// If some other process has stolen the foreground window,
// don't be rude and grab it back.
void FriendlySetForegroundWindow(HWND hwnd)
{
HWND hwndOld = GetForegroundWindow();
if (hwndOld)
{
DWORD dwProcessId;
GetWindowThreadProcessId(hwndOld, &dwProcessId);
if (dwProcessId == GetCurrentProcessId())
hwndOld = NULL;
}
if (!hwndOld)
SetForegroundWindow(hwnd);
}
STDAPI_(BOOL) SHDesktopMessageLoop(HANDLE hDesktop)
{
CDesktopBrowser *pBrowser = (CDesktopBrowser *) hDesktop;
if (pBrowser)
{
// We must AddRef the pBrowser because _MessageLoop() will
// Release() it if another app initiated a system shutdown.
// We will do our own Release() when we don't need the pointer
// any more.
pBrowser->AddRef();
FriendlySetForegroundWindow(pBrowser->GetDesktopWindow());
pBrowser->_MessageLoop();
IconCacheSave();
// In case someone posted us a WM_QUIT message, before terminating
// the thread, make sure it is properly destroyed so that trident etc
// gets properly freed up.
HWND hwnd = pBrowser->GetDesktopWindow();
if (hwnd)
{
DestroyWindow(hwnd);
}
pBrowser->Release();
OleUninitialize();
}
return BOOLFROMPTR(pBrowser);
}
//
// Whenever we remove the toolbar from the desktop, we persist it.
//
HRESULT CDesktopBrowser::RemoveToolbar(IUnknown* punkSrc, DWORD dwRemoveFlags)
{
HRESULT hres = E_FAIL;
UINT itb = _pbsInner->_FindTBar(punkSrc);
if (itb==(UINT)-1)
{
return E_INVALIDARG;
}
LPTOOLBARITEM ptbi = _pbsInner->_GetToolbarItem(itb);
if (ptbi && ptbi->pwszItem)
{
LPCWSTR szItem = ptbi->pwszItem;
if (dwRemoveFlags & STFRF_DELETECONFIGDATA)
{
DeleteDesktopViewStream(szItem);
}
else
{
IStream* pstm = GetDesktopViewStream(STGM_WRITE, szItem);
if (pstm)
{
IPersistStreamInit* ppstm;
HRESULT hresT = punkSrc->QueryInterface(IID_PPV_ARG(IPersistStreamInit, &ppstm));
if (SUCCEEDED(hresT)) {
ppstm->Save(pstm, TRUE);
ppstm->Release();
}
pstm->Release();
}
}
}
hres = _pdwfInner->RemoveToolbar(punkSrc, dwRemoveFlags);
_UpdateViewRectSize();
return hres;
}
HRESULT CDesktopBrowser::GetBorderDW(IUnknown* punkSrc, LPRECT lprectBorder)
{
BOOL bUseHmonitor = (GetNumberOfMonitors() > 1);
return _pbsInner->_GetBorderDWHelper(punkSrc, lprectBorder, bUseHmonitor);
}
HRESULT CDesktopBrowser::_ResizeNextBorder(UINT itb)
{
_ResizeNextBorderHelper(itb, TRUE);
return S_OK;
}
HRESULT CDesktopBrowser::GetMonitor(IUnknown* punkSrc, HMONITOR * phMon)
{
ASSERT(phMon);
*phMon = NULL; // just in case
UINT itb = _pbsInner->_FindTBar(punkSrc);
if (itb==(UINT)-1) {
return E_INVALIDARG;
}
LPTOOLBARITEM ptbi = _pbsInner->_GetToolbarItem(itb);
if (ptbi) {
*phMon = ptbi->hMon;
return S_OK;
} else {
return E_FAIL;
}
}
HRESULT CDesktopBrowser::RequestMonitor(IUnknown* punkSrc, HMONITOR * phMonitor)
{
UINT itb = _pbsInner->_FindTBar(punkSrc);
if (itb==(UINT)-1) {
return E_INVALIDARG;
}
ASSERT(phMonitor);
if (IsMonitorValid(*phMonitor))
return S_OK;
else
{
*phMonitor = GetPrimaryMonitor();
return S_FALSE;
}
}
HRESULT CDesktopBrowser::SetMonitor(IUnknown* punkSrc, HMONITOR hMonNew, HMONITOR * phMonOld)
{
ASSERT(phMonOld);
*phMonOld = NULL; // just in case
UINT itb = _pbsInner->_FindTBar(punkSrc);
if (itb==(UINT)-1) {
return E_INVALIDARG;
}
LPTOOLBARITEM ptbThis = _pbsInner->_GetToolbarItem(itb);
if (ptbThis) {
*phMonOld = ptbThis->hMon;
ptbThis->hMon = hMonNew;
return S_OK;
} else {
return E_FAIL;
}
}
///////////////////////////////////////////////////////////////////////
//
// CDesktopBrowser FORWARDERS to commonsb
//
// {
#define CALL_INNER(_result, _function, _arglist, _args) \
_result CDesktopBrowser:: _function _arglist { return _psbInner-> _function _args ; }
#define CALL_INNER_HRESULT(_function, _arglist, _args) CALL_INNER(HRESULT, _function, _arglist, _args)
// IShellBrowser (same as IOleInPlaceFrame)
CALL_INNER_HRESULT(GetWindow, (HWND * lphwnd), (lphwnd));
CALL_INNER_HRESULT(ContextSensitiveHelp, (BOOL fEnterMode), (fEnterMode));
CALL_INNER_HRESULT(InsertMenusSB, (HMENU hmenuShared, LPOLEMENUGROUPWIDTHS lpMenuWidths), (hmenuShared, lpMenuWidths));
CALL_INNER_HRESULT(SetMenuSB, (HMENU hmenuShared, HOLEMENU holemenu, HWND hwnd), (hmenuShared, holemenu, hwnd));
CALL_INNER_HRESULT(RemoveMenusSB, (HMENU hmenuShared), (hmenuShared));
CALL_INNER_HRESULT(SetStatusTextSB, (LPCOLESTR lpszStatusText), (lpszStatusText));
CALL_INNER_HRESULT(EnableModelessSB, (BOOL fEnable), (fEnable));
CALL_INNER_HRESULT(TranslateAcceleratorSB, (LPMSG lpmsg, WORD wID), (lpmsg, wID));
CALL_INNER_HRESULT(GetViewStateStream, (DWORD grfMode, LPSTREAM *ppStrm), (grfMode, ppStrm));
CALL_INNER_HRESULT(GetControlWindow, (UINT id, HWND * lphwnd), (id, lphwnd));
CALL_INNER_HRESULT(SendControlMsg, (UINT id, UINT uMsg, WPARAM wParam, LPARAM lParam, LRESULT *pret), (id, uMsg, wParam, lParam, pret));
CALL_INNER_HRESULT(QueryActiveShellView, (struct IShellView ** ppshv), (ppshv));
CALL_INNER_HRESULT(OnViewWindowActive, (struct IShellView * ppshv), (ppshv));
CALL_INNER_HRESULT(SetToolbarItems, (LPTBBUTTON lpButtons, UINT nButtons, UINT uFlags), (lpButtons, nButtons, uFlags));
#undef CALL_INNER
#undef CALL_INNER_HRESULT
// }
// {
#define CALL_INNER(_result, _function, _arglist, _args) \
_result CDesktopBrowser:: _function _arglist { return _pdwsInner-> _function _args ; }
#define CALL_INNER_HRESULT(_function, _arglist, _args) CALL_INNER(HRESULT, _function, _arglist, _args)
// IDockingWindowSite
// TODO: move these up from basesb to commonsb - requires toolbars
CALL_INNER_HRESULT(RequestBorderSpaceDW, (IUnknown* punkSrc, LPCBORDERWIDTHS pborderwidths), (punkSrc, pborderwidths));
CALL_INNER_HRESULT(SetBorderSpaceDW, (IUnknown* punkSrc, LPCBORDERWIDTHS pborderwidths), (punkSrc, pborderwidths));
#undef CALL_INNER
#undef CALL_INNER_HRESULT
// }
// {
#define CALL_INNER(_result, _function, _arglist, _args) \
_result CDesktopBrowser:: _function _arglist { return _pdwfInner-> _function _args ; }
#define CALL_INNER_HRESULT(_function, _arglist, _args) CALL_INNER(HRESULT, _function, _arglist, _args)
// IDockingWindowFrame
CALL_INNER_HRESULT(AddToolbar, (IUnknown* punkSrc, LPCWSTR pwszItem, DWORD dwReserved), (punkSrc, pwszItem, dwReserved));
CALL_INNER_HRESULT(FindToolbar, (LPCWSTR pwszItem, REFIID riid, void **ppvObj), (pwszItem, riid, ppvObj));
#undef CALL_INNER
#undef CALL_INNER_HRESULT
// }
// {
#define CALL_INNER(_result, _function, _arglist, _args) \
_result CDesktopBrowser:: _function _arglist { return _piosInner-> _function _args ; }
#define CALL_INNER_HRESULT(_function, _arglist, _args) CALL_INNER(HRESULT, _function, _arglist, _args)
// IInputObjectSite
CALL_INNER_HRESULT(OnFocusChangeIS, (IUnknown* punkSrc, BOOL fSetFocus), (punkSrc, fSetFocus));
#undef CALL_INNER
#undef CALL_INNER_HRESULT
// }
// {
#define CALL_INNER(_result, _function, _arglist, _args) \
_result CDesktopBrowser:: _function _arglist { return _pdtInner-> _function _args ; }
#define CALL_INNER_HRESULT(_function, _arglist, _args) CALL_INNER(HRESULT, _function, _arglist, _args)
// *** IDropTarget ***
CALL_INNER_HRESULT(DragLeave, (void), ());
#undef CALL_INNER
#undef CALL_INNER_HRESULT
// }
// {
#define CALL_INNER(_result, _function, _arglist, _args) \
_result CDesktopBrowser:: _function _arglist { return _pbsInner-> _function _args ; }
#define CALL_INNER_HRESULT(_function, _arglist, _args) CALL_INNER(HRESULT, _function, _arglist, _args)
// *** IBrowserService2 specific methods ***
CALL_INNER_HRESULT(GetParentSite, ( IOleInPlaceSite** ppipsite), ( ppipsite));
CALL_INNER_HRESULT(SetTitle, (IShellView* psv, LPCWSTR pszName), (psv, pszName));
CALL_INNER_HRESULT(GetTitle, (IShellView* psv, LPWSTR pszName, DWORD cchName), (psv, pszName, cchName));
CALL_INNER_HRESULT(GetOleObject, ( IOleObject** ppobjv), ( ppobjv));
// think about this one.. I'm not sure we want to expose this -- Chee
// My impression is that we won't document this whole interface???
CALL_INNER_HRESULT(GetTravelLog, (ITravelLog** pptl), (pptl));
CALL_INNER_HRESULT(ShowControlWindow, (UINT id, BOOL fShow), (id, fShow));
CALL_INNER_HRESULT(IsControlWindowShown, (UINT id, BOOL *pfShown), (id, pfShown));
CALL_INNER_HRESULT(IEGetDisplayName, (LPCITEMIDLIST pidl, LPWSTR pwszName, UINT uFlags), (pidl, pwszName, uFlags));
CALL_INNER_HRESULT(IEParseDisplayName, (UINT uiCP, LPCWSTR pwszPath, LPITEMIDLIST * ppidlOut), (uiCP, pwszPath, ppidlOut));
CALL_INNER_HRESULT(DisplayParseError, (HRESULT hres, LPCWSTR pwszPath), (hres, pwszPath));
CALL_INNER_HRESULT(NavigateToPidl, (LPCITEMIDLIST pidl, DWORD grfHLNF), (pidl, grfHLNF));
CALL_INNER_HRESULT(SetNavigateState, (BNSTATE bnstate), (bnstate));
CALL_INNER_HRESULT(GetNavigateState, (BNSTATE *pbnstate), (pbnstate));
CALL_INNER_HRESULT(NotifyRedirect, ( IShellView* psv, LPCITEMIDLIST pidl, BOOL *pfDidBrowse), ( psv, pidl, pfDidBrowse));
CALL_INNER_HRESULT(UpdateWindowList, (), ());
CALL_INNER_HRESULT(UpdateBackForwardState, (), ());
CALL_INNER_HRESULT(SetFlags, (DWORD dwFlags, DWORD dwFlagMask), (dwFlags, dwFlagMask));
CALL_INNER_HRESULT(GetFlags, (DWORD *pdwFlags), (pdwFlags));
// Tells if it can navigate now or not.
CALL_INNER_HRESULT(CanNavigateNow, (), ());
CALL_INNER_HRESULT(GetPidl, (LPITEMIDLIST *ppidl), (ppidl));
CALL_INNER_HRESULT(SetReferrer, (LPITEMIDLIST pidl), (pidl));
CALL_INNER(DWORD, GetBrowserIndex ,(), ());
CALL_INNER_HRESULT(GetBrowserByIndex, (DWORD dwID, IUnknown **ppunk), (dwID, ppunk));
CALL_INNER_HRESULT(GetHistoryObject, (IOleObject **ppole, IStream **pstm, IBindCtx **ppbc), (ppole, pstm, ppbc));
CALL_INNER_HRESULT(SetHistoryObject, (IOleObject *pole, BOOL fIsLocalAnchor), (pole, fIsLocalAnchor));
CALL_INNER_HRESULT(CacheOLEServer, (IOleObject *pole), (pole));
CALL_INNER_HRESULT(GetSetCodePage, (VARIANT* pvarIn, VARIANT* pvarOut), (pvarIn, pvarOut));
CALL_INNER_HRESULT(OnHttpEquiv, (IShellView* psv, BOOL fDone, VARIANT* pvarargIn, VARIANT* pvarargOut), (psv, fDone, pvarargIn, pvarargOut));
CALL_INNER_HRESULT(GetPalette, ( HPALETTE * hpal), ( hpal));
CALL_INNER_HRESULT(OnSetFocus, (), ());
CALL_INNER_HRESULT(OnFrameWindowActivateBS, (BOOL fActive), (fActive));
CALL_INNER_HRESULT(RegisterWindow, (BOOL fUnregister, int swc), (fUnregister, swc));
CALL_INNER_HRESULT(GetBaseBrowserData,(LPCBASEBROWSERDATA* ppbd), (ppbd));
CALL_INNER(LPBASEBROWSERDATA, PutBaseBrowserData,(), ());
CALL_INNER_HRESULT(CreateViewWindow, (IShellView* psvNew, IShellView* psvOld, LPRECT prcView, HWND* phwnd), (psvNew, psvOld, prcView, phwnd));;
CALL_INNER_HRESULT(SetTopBrowser, (), ());
CALL_INNER_HRESULT(InitializeDownloadManager, (), ());
CALL_INNER_HRESULT(InitializeTransitionSite, (), ());
CALL_INNER_HRESULT(Offline, (int iCmd), (iCmd));
CALL_INNER_HRESULT(AllowViewResize, (BOOL f), (f));
CALL_INNER_HRESULT(SetActivateState, (UINT u), (u));
CALL_INNER_HRESULT(UpdateSecureLockIcon, (int eSecureLock), (eSecureLock));
CALL_INNER_HRESULT(CreateBrowserPropSheetExt, (REFIID riid, void **ppvObj), (riid, ppvObj));
CALL_INNER_HRESULT(GetViewWindow,(HWND * phwnd), (phwnd));
CALL_INNER_HRESULT(InitializeTravelLog,(ITravelLog* ptl, DWORD dw), (ptl, dw));
CALL_INNER_HRESULT(_UIActivateView, (UINT uState), (uState));
CALL_INNER_HRESULT(_ResizeView,(), ());
CALL_INNER_HRESULT(_ExecChildren, (IUnknown *punkBar, BOOL fBroadcast, const GUID *pguidCmdGroup,
DWORD nCmdID, DWORD nCmdexecopt, VARIANTARG *pvarargIn, VARIANTARG *pvarargOut),
(punkBar, fBroadcast, pguidCmdGroup, nCmdID, nCmdexecopt, pvarargIn, pvarargOut));
CALL_INNER_HRESULT(_SendChildren,
(HWND hwndBar, BOOL fBroadcast, UINT uMsg, WPARAM wParam, LPARAM lParam),
(hwndBar, fBroadcast, uMsg, wParam, lParam));
CALL_INNER_HRESULT(_OnFocusChange, (UINT itb), (itb));
CALL_INNER_HRESULT(v_ShowHideChildWindows, (BOOL fChildOnly), (fChildOnly));
CALL_INNER_HRESULT(_GetViewBorderRect, (RECT* prc), (prc));
// BEGIN REVIEW: review names and need of each.
//
// this first set could be basebrowser only members. no one overrides
CALL_INNER_HRESULT(_CancelPendingNavigationAsync, (), ());
CALL_INNER_HRESULT(_MaySaveChanges, (), ());
CALL_INNER_HRESULT(_PauseOrResumeView, (BOOL fPaused), (fPaused));
CALL_INNER_HRESULT(_DisableModeless, (), ());
// rethink these... are all of these necessary?
CALL_INNER_HRESULT(_NavigateToPidl, (LPCITEMIDLIST pidl, DWORD grfHLNF, DWORD dwFlags), (pidl, grfHLNF, dwFlags));
CALL_INNER_HRESULT(_TryShell2Rename, (IShellView* psv, LPCITEMIDLIST pidlNew), (psv, pidlNew));
CALL_INNER_HRESULT(_SwitchActivationNow, () , ());
CALL_INNER_HRESULT(_CancelPendingView, (), ());
//END REVIEW:
CALL_INNER(UINT, _get_itbLastFocus, (), ());
CALL_INNER_HRESULT(_put_itbLastFocus, (UINT itbLastFocus), (itbLastFocus));
CALL_INNER_HRESULT(_ResizeNextBorderHelper, (UINT itb, BOOL bUseHmonitor), (itb, bUseHmonitor));
#undef CALL_INNER
#undef CALL_INNER_HRESULT
// }
// RunDll32 entry point to create a new local server on the desktop
// thread. We convert the CLSID to the index into the LocalServer list
// and then send it to the desktop, which inturn spins the thread off.
STDAPI_(void) SHCreateLocalServerRunDll(HWND hwndStub, HINSTANCE hAppInstance, LPSTR pszCmdLine, int nCmdShow)
{
CLSID clsidLocalServer;
if (GUIDFromStringA(pszCmdLine, &clsidLocalServer))
{
for (int i = 0; i < ARRAYSIZE(c_localServers); i++)
{
if (IsEqualCLSID(clsidLocalServer, *c_localServers[i]))
{
PostMessage(GetShellWindow(), CWM_CREATELOCALSERVER, (WPARAM)TRUE, (LPARAM)i);
break;
}
}
}
}