|
|
#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; } } }}
|
