#include "priv.h" #include #include "sccls.h" #include "iface.h" #include "itbar.h" #include "itbdrop.h" #include "bands.h" #include "isfband.h" #include "menubar.h" #include "resource.h" #include "menuisf.h" #include "dpastuff.h" #include "shlwapi.h" #include "cobjsafe.h" #include #include "uemapp.h" #include "mnfolder.h" #include "channel.h" #include "browmenu.h" #define DM_VERBOSE 0 // misc verbose traces #define DM_PERSIST 0 #define TF_BANDDD TF_BAND #define DM_RENAME 0 #define DM_MISC 0 // miscellany #define SZ_PROPERTIESA "properties" #define SZ_PROPERTIES TEXT(SZ_PROPERTIESA) #define SZ_REGKEY_ADVFOLDER TEXT("Software\\Microsoft\\Windows\\CurrentVersion\\Explorer\\Advanced") // {F47162A0-C18F-11d0-A3A5-00C04FD706EC} static const GUID TOID_ExtractImage = { 0xf47162a0, 0xc18f, 0x11d0, { 0xa3, 0xa5, 0x0, 0xc0, 0x4f, 0xd7, 0x6, 0xec } }; #define SUPERCLASS CToolBand HRESULT FakeGetUIObjectOf( IShellFolder *psf, LPCITEMIDLIST pidl, UINT * prgfFlags, REFIID riid, void **ppvObj ); extern UINT g_idFSNotify; HRESULT CExtractImageTask_Create( CLogoBase *plb, LPEXTRACTIMAGE pExtract, LPCWSTR pszCache, DWORD dwItem, int iIcon, DWORD dwFlags, LPRUNNABLETASK * ppTask ); class CExtractImageTask : public IRunnableTask { public: STDMETHOD ( QueryInterface ) ( REFIID riid, void **ppvObj ); STDMETHOD_( ULONG, AddRef ) (); STDMETHOD_( ULONG, Release ) (); STDMETHOD (Run)( void ); STDMETHOD (Kill)( BOOL fWait ); STDMETHOD (Suspend)( ); STDMETHOD (Resume)( ); STDMETHOD_( ULONG, IsRunning )( void ); protected: CExtractImageTask( HRESULT * pHr, CLogoBase *plb, IExtractImage * pImage, LPCWSTR pszCache, DWORD dwItem, int iIcon, DWORD dwFlags ); ~CExtractImageTask(); HRESULT InternalResume(); friend HRESULT CExtractImageTask_Create( CLogoBase* plb, LPEXTRACTIMAGE pExtract, LPCWSTR pszCache, DWORD dwItem, int iIcon, DWORD dwFlags, LPRUNNABLETASK * ppTask ); LONG m_cRef; LONG m_lState; LPEXTRACTIMAGE m_pExtract; LPRUNNABLETASK m_pTask; WCHAR m_szPath[MAX_PATH]; DWORD m_dwFlags; DWORD m_dwItem; CLogoBase* m_plb; HBITMAP m_hBmp; int m_iIcon; }; //================================================================= // Implementation of CISFBand //================================================================= CISFBand::CISFBand() : CToolbarBand() { _fCanFocus = TRUE; _eUemLog = UEMIND_NIL; _dwPriv = -1; _fHasOrder = TRUE; // ISFBand always has an order... _fAllowDropdown = BOOLIFY(SHRegGetBoolUSValue(SZ_REGKEY_ADVFOLDER, TEXT("CascadeFolderBands"), FALSE, FALSE)); // Should we enable logging of arbirary events? // _pguidUEMGroup = &UEMIID_SHELL; ASSERT(_pguidUEMGroup == NULL); // Assert that this class is ZERO INITed. ASSERT(!_pbp); ASSERT(FALSE == _fCreatedBandProxy); } CISFBand::~CISFBand() { if(_pbp && _fCreatedBandProxy) _pbp->SetSite(NULL); ATOMICRELEASE(_pbp); } HRESULT CISFBand_CreateInstance(IUnknown* pUnkOuter, IUnknown** ppunk, LPCOBJECTINFO poi) { // aggregation checking is handled in class factory HRESULT hres; CISFBand* pObj; hres = E_OUTOFMEMORY; pObj = new CISFBand(); if (pObj) { *ppunk = SAFECAST(pObj, IShellFolderBand*); hres = S_OK; } return hres; } /*---------------------------------------------------------- Purpose: See CISFBand::Init for an explanation on the parameters. */ CISFBand* CISFBand_CreateEx(IShellFolder* psf, LPCITEMIDLIST pidl) { CISFBand * p = NULL; if (psf || pidl) { p = new CISFBand(); if (p) { IShellFolderBand * psfband = SAFECAST(p, IShellFolderBand *); if (psfband && FAILED(psfband->InitializeSFB(psf, pidl))) { delete p; p = NULL; } } } return p; } #ifdef DEBUG #define _AddRef(psz) { ++_cRef; TraceMsg(TF_SHDREF, "CDocObjectView(%x)::QI(%s) is AddRefing _cRef=%d", this, psz, _cRef); } #else #define _AddRef(psz) ++_cRef #endif HRESULT CISFBand::QueryInterface(REFIID riid, void **ppvObj) { static const QITAB qit[] = { QITABENT(CISFBand, IShellFolderBand), QITABENT(CISFBand, IFolderBandPriv), { 0 }, }; HRESULT hres = QISearch(this, qit, riid, ppvObj); if (FAILED(hres)) hres = CToolBand::QueryInterface(riid, ppvObj); if (FAILED(hres)) hres = CSFToolbar::QueryInterface(riid, ppvObj); if (S_OK != hres) { // HACKHACK: this is yucko! if (IsEqualIID(riid, CLSID_ISFBand)) { *ppvObj = (void*)this; _AddRef(TEXT("CLSID_ISFBand")); return S_OK; } } return hres; } #if 0 LPITEMIDLIST PidlFromFolderAndSubPath(int iFolder, TCHAR *pszSubPath) { LPITEMIDLIST pidl = NULL; if (SUCCEEDED(SHGetSpecialFolderLocation(NULL, iFolder, &pidl))) { if (pszSubPath) { TCHAR szPath[MAX_PATH]; SHGetPathFromIDList(pidl, szPath); PathCombine(szPath, szPath, pszSubPath); ILFree(pidl); pidl = ILCreateFromPath(szPath); } } return pidl; } #endif //*** ILIsParentCSIDL -- like ILIsParent, but accepts a CSIDL_* for pidl1 // NOTES // TODO move to shlwapi (if/when idlist.c moves there)? // STDAPI_(BOOL) ILIsParentCSIDL(int csidl1, LPCITEMIDLIST pidl2, BOOL fImmediate) { LPITEMIDLIST pidlSpec; BOOL fRet = FALSE; if (SUCCEEDED(SHGetSpecialFolderLocation(NULL, csidl1, &pidlSpec))) { fRet = ILIsParent(pidlSpec, pidl2, fImmediate); ILFree(pidlSpec); } return fRet; } /*---------------------------------------------------------- Purpose: IShellFolderBand::InitializeSFB - supply IShellFolder with no PIDL if you want to view some ISF (either already instantiated from the filesystem or some non-filesystem ISF) that you do NOT want to receive notifies from (either from SHChangeNotify nor from IShellChangeNotify) - supply a PIDL with no IShellFolder for a full-blown band looking at a shell namespace (rooted on desktop) item. */ HRESULT CISFBand::InitializeSFB(IShellFolder *psf, LPCITEMIDLIST pidl) { HRESULT hres = S_OK; // Did they try to add the Recycle Bin? If so we need to reject it // for consistance reasons. We also reject the Temp. Internet Files // for security reasons. if (pidl && (ILIsParentCSIDL(CSIDL_BITBUCKET, pidl, FALSE) || ILIsParentCSIDL(CSIDL_INTERNET_CACHE, pidl, FALSE))) { // this will eventually show up as IDS_CANTISFBAND TraceMsg(DM_TRACE, "cib.isfb: recycle => E_INVALIDARG"); hres = E_INVALIDARG; } if (SUCCEEDED(hres)) hres = CSFToolbar::SetShellFolder(psf, pidl); if (SUCCEEDED(hres)) _AfterLoad(); return hres; } /*---------------------------------------------------------- Purpose: IShellFolderBand::SetBandInfoSFB */ HRESULT CISFBand::SetBandInfoSFB(BANDINFOSFB * pbi) { ASSERT(pbi); if (!pbi) return E_POINTER; if ((pbi->dwMask & ISFB_MASK_INVALID) || (pbi->dwMask & ISFB_MASK_VIEWMODE) && (pbi->wViewMode & ~3)) return E_INVALIDARG; // We don't handle ISFB_MASK_SHELLFOLDER and ISFB_MASK_IDLIST // in Set because there's a lot of work to resync pidl, psf, and // notifcations in the toolbar. If somebody wants to do it, // more power to ya. :) if (pbi->dwMask & (ISFB_MASK_SHELLFOLDER | ISFB_MASK_IDLIST)) return E_INVALIDARG; if (pbi->dwMask & ISFB_MASK_STATE) { if (pbi->dwStateMask & ISFB_STATE_DEBOSSED) _fDebossed = BOOLIFY(pbi->dwState & ISFB_STATE_DEBOSSED); if (pbi->dwStateMask & ISFB_STATE_ALLOWRENAME) _fAllowRename = BOOLIFY(pbi->dwState & ISFB_STATE_ALLOWRENAME); if (pbi->dwStateMask & ISFB_STATE_NOSHOWTEXT) _fNoShowText = BOOLIFY(pbi->dwState & ISFB_STATE_NOSHOWTEXT); if (pbi->dwStateMask & ISFB_STATE_CHANNELBAR) _fChannels = BOOLIFY(pbi->dwState & ISFB_STATE_CHANNELBAR); /* ISFB_STATE_NOTITLE: removed 970619, use cbs::SetBandState */ if (pbi->dwStateMask & ISFB_STATE_QLINKSMODE) _fLinksMode = BOOLIFY(pbi->dwState & ISFB_STATE_QLINKSMODE); if (pbi->dwStateMask & ISFB_STATE_FULLOPEN) _fFullOpen = BOOLIFY(pbi->dwState & ISFB_STATE_FULLOPEN); if (pbi->dwStateMask & ISFB_STATE_NONAMESORT) _fNoNameSort = BOOLIFY(pbi->dwState & ISFB_STATE_NONAMESORT); if (pbi->dwStateMask & ISFB_STATE_BTNMINSIZE) _fBtnMinSize = BOOLIFY(pbi->dwState & ISFB_STATE_BTNMINSIZE); } if (pbi->dwMask & ISFB_MASK_BKCOLOR) { _crBkgnd = pbi->crBkgnd; _fHaveBkColor = TRUE; if (EVAL(_hwndTB)) SHSetWindowBits(_hwndTB, GWL_STYLE, TBSTYLE_CUSTOMERASE, TBSTYLE_CUSTOMERASE); ASSERT(_hwnd); if (_hwndPager) { TraceMsg(TF_BAND, "cib.sbisfb: Pager_SetBkColor(_hwnd=%x crBkgnd=%x)", _hwnd, _crBkgnd); Pager_SetBkColor(_hwnd, _crBkgnd); } } // BUGBUG (kkahl): We don't support changing these once TB is created if (pbi->dwMask & ISFB_MASK_COLORS) { _crBtnLt = pbi->crBtnLt; _crBtnDk = pbi->crBtnDk; _fHaveColors = TRUE; } if (pbi->dwMask & ISFB_MASK_VIEWMODE) { _uIconSize = (pbi->wViewMode & 3); // stored in a 2-bit field currently... // only force no recalc if one of the recalcable fields was set _fNoRecalcDefaults = TRUE; } // If the bandsite queried us before, let it know the info may have changed if (_fInitialized) _BandInfoChanged(); return S_OK; } /*---------------------------------------------------------- Purpose: IShellFolderBand::GetBandInfoSFB */ HRESULT CISFBand::GetBandInfoSFB(BANDINFOSFB * pbi) { ASSERT(pbi); if (!pbi) return E_POINTER; if (pbi->dwMask & ISFB_MASK_STATE) { pbi->dwState = 0; pbi->dwStateMask = ISFB_STATE_ALL; if (_fDebossed) pbi->dwState |= ISFB_STATE_DEBOSSED; if (_fAllowRename) pbi->dwState |= ISFB_STATE_ALLOWRENAME; if (_fNoShowText) pbi->dwState |= ISFB_STATE_NOSHOWTEXT; if (_fLinksMode) pbi->dwState |= ISFB_STATE_QLINKSMODE; if (_fFullOpen) pbi->dwState |= ISFB_STATE_FULLOPEN; if (_fNoNameSort) pbi->dwState |= ISFB_STATE_NONAMESORT; if (_fBtnMinSize) pbi->dwState |= ISFB_STATE_BTNMINSIZE; } if (pbi->dwMask & ISFB_MASK_BKCOLOR) { pbi->crBkgnd = (_fHaveBkColor) ? _crBkgnd : CLR_DEFAULT; } if (pbi->dwMask & ISFB_MASK_COLORS) { if (_fHaveColors) { pbi->crBtnLt = _crBtnLt; pbi->crBtnDk = _crBtnDk; } else { pbi->crBtnLt = CLR_DEFAULT; pbi->crBtnDk = CLR_DEFAULT; } } if (pbi->dwMask & ISFB_MASK_VIEWMODE) { pbi->wViewMode = _uIconSize; } if (pbi->dwMask & ISFB_MASK_SHELLFOLDER) { pbi->psf = _psf; if (pbi->psf) pbi->psf->AddRef(); } if (pbi->dwMask & ISFB_MASK_IDLIST) { if (_pidl) pbi->pidl = ILClone(_pidl); else pbi->pidl = NULL; } return S_OK; } // *** IInputObject methods *** HRESULT CISFBand::TranslateAcceleratorIO(LPMSG lpMsg) { if (SendMessage(_hwnd, TB_TRANSLATEACCELERATOR, 0, (LPARAM)lpMsg)) return S_OK; return SUPERCLASS::TranslateAcceleratorIO(lpMsg); } void CISFBand::_SetCacheMenuPopup(IMenuPopup* pmp) { if (!SHIsSameObject(pmp, _pmpCache)) { _ReleaseMenuPopup(&_pmpCache); _pmpCache = pmp; if (_pmpCache) _pmpCache->AddRef(); } } void CISFBand::_ReleaseMenuPopup(IMenuPopup** ppmp) { IUnknown_SetSite(*ppmp, NULL); ATOMICRELEASE(*ppmp); } /*---------------------------------------------------------- Purpose: Releases the held menu popup. */ void CISFBand::_ReleaseMenu() { if (!SHIsSameObject(_pmp, _pmpCache)) { TraceMsg(TF_MENUBAND, "Releasing pmp %#lx", _pmp); _ReleaseMenuPopup(&_pmp); } else ATOMICRELEASE(_pmp); } //*** // ENTRY/EXIT // S_OK desktop browser // S_FALSE other browser (explorer, OC, etc.) // E_xxx not a browser at all (e.g. band asking tray) HRESULT IsDesktopBrowser(IUnknown *punkSite) { HRESULT hr; IServiceProvider *psp; IUnknown *punk; hr = E_FAIL; if (SUCCEEDED(IUnknown_QueryService(punkSite, SID_STopLevelBrowser, IID_IServiceProvider, (void**)&psp))) { hr = S_FALSE; if (SUCCEEDED(psp->QueryInterface(SID_SShellDesktop, (void**)&punk))) { hr = S_OK; punk->Release(); } psp->Release(); } TraceMsg(DM_VERBOSE, "idb: ret hrDesk=%x (0=dt 1=sh e=!brow)", hr); return hr; } /*---------------------------------------------------------- Purpose: IDockingWindow::SetSite method. */ HRESULT CISFBand::SetSite(IUnknown* punkSite) { _ReleaseMenu(); SUPERCLASS::SetSite(punkSite); if (_punkSite) { if (!_hwndTB) _CreateToolbar(_hwndParent); IUnknown_SetOwner(_psf, SAFECAST(this, IDeskBand*)); _Initialize(); // BUGBUG always or just on 1st SetSite? } else IUnknown_SetOwner(_psf, NULL); // BUGBUG: the below is bogus - no need to throw away and recreate. // First destroy the band proxy // Call SetSite(NULL) only if you own // if not, it's the parent from whom you got it via QS who will call SetSite(NULL) if(_pbp && _fCreatedBandProxy) _pbp->SetSite(NULL); ATOMICRELEASE(_pbp); _fCreatedBandProxy = FALSE; // Need a bandproxy QueryService_SID_IBandProxy(punkSite, IID_IBandProxy, &_pbp, NULL); if(!_pbp) { // We need to create it ourselves since our parent couldn't help ASSERT(FALSE == _fCreatedBandProxy); HRESULT hres; hres = CreateIBandProxyAndSetSite(punkSite, IID_IBandProxy, &_pbp, NULL); if(_pbp) { ASSERT(S_OK == hres); _fCreatedBandProxy = TRUE; } } ASSERT(_pbp); return S_OK; } void CISFBand::_Initialize() { _fDesktop = (IsDesktopBrowser(_punkSite) == S_OK); return; } /*---------------------------------------------------------- Purpose: IDockingWindow::CloseDW method. */ HRESULT CISFBand::CloseDW(DWORD dw) { _fClosing = TRUE; // close down the task scheduler ... if ( _pTaskScheduler ) ATOMICRELEASE( _pTaskScheduler ); _UnregisterToolbar(); EmptyToolbar(); IUnknown_SetOwner(_psf, NULL); _SetCacheMenuPopup(NULL); // should get freed in EmptyToolbar(); ASSERT(!_hdpa); return SUPERCLASS::CloseDW(dw); } /*---------------------------------------------------------- Purpose: IDockingWindow::ShowDW method */ HRESULT CISFBand::ShowDW(BOOL fShow) { HRESULT hres = S_OK; SUPERCLASS::ShowDW(fShow); if (fShow) { _fShow = TRUE; if (_fDirty) { _FillToolbar(); } if (!_fDelayInit) { _RegisterToolbar(); } } else { _fShow = FALSE; } return hres; } void CISFBand::_StopDelayPainting() { if (_fDelayPainting) { _fDelayPainting = FALSE; // May be called by background thread // Use PostMessage instead of SendMessage to avoid deadlock PostMessage(_hwndTB, WM_SETREDRAW, TRUE, 0); if (_hwndPager) PostMessage(_hwnd, PGM_RECALCSIZE, 0L, 0L); } } HWND CISFBand::_CreatePager(HWND hwndParent) { // don't create a pager for isfbands return hwndParent; } void CISFBand::_CreateToolbar(HWND hwndParent) { if (_fHaveBkColor) _dwStyle |= TBSTYLE_CUSTOMERASE; CSFToolbar::_CreateToolbar(hwndParent); if ( _fHaveBkColor ) ToolBar_SetInsertMarkColor(_hwndTB, GetSysColor( COLOR_BTNFACE )); ASSERT(_hwndTB); SendMessage(_hwndTB, TB_SETEXTENDEDSTYLE, TBSTYLE_EX_DRAWDDARROWS, TBSTYLE_EX_DRAWDDARROWS); if(_fChannels) { SHSetWindowBits(_hwndTB, GWL_EXSTYLE, dwExStyleRTLMirrorWnd, 0); } _hwnd = _hwndPager ? _hwndPager : _hwndTB; if (_fHaveColors) { COLORSCHEME cs; cs.dwSize = SIZEOF(cs); cs.clrBtnHighlight = _crBtnLt; cs.clrBtnShadow = _crBtnDk; SendMessage(_hwndTB, TB_SETCOLORSCHEME, 0, (LPARAM) &cs); } } int CISFBand::_GetBitmap(int iCommandID, PIBDATA pibdata, BOOL fUseCache) { int iBitmap; if ( _uIconSize == ISFBVIEWMODE_LOGOS ) { LPRUNNABLETASK pTask = NULL; DWORD dwPriority = 0; // fetch the logo instead... ASSERT(!_fDelayPainting); // Warning - cannot hold ptask in a member variable - it will be a circular reference iBitmap = GetLogoIndex( iCommandID, pibdata->GetPidl(), &pTask, &dwPriority, NULL ); if (pTask) { AddTaskToQueue(pTask, dwPriority, (DWORD)iCommandID); ATOMICRELEASE(pTask); } } else iBitmap = CSFToolbar::_GetBitmap(iCommandID, pibdata, fUseCache); return iBitmap; } void CISFBand::_SetDirty(BOOL fDirty) { CSFToolbar::_SetDirty(fDirty); if (fDirty) IUnknown_Exec(_punkSite, &CGID_PrivCITCommands, CITIDM_SET_DIRTYBIT, TRUE, NULL, NULL); } BOOL CISFBand::_UpdateIconSize(UINT uIconSize, BOOL fUpdateButtons) { BOOL fChanged = (_uIconSize != uIconSize); _uIconSize = uIconSize; HIMAGELIST himl = NULL; if ( uIconSize == ISFBVIEWMODE_LOGOS ) { if ( SUCCEEDED( InitLogoView())) { himl = GetLogoHIML(); } if ( himl ) { SendMessage(_hwndTB, TB_SETIMAGELIST, 0, (LPARAM)himl); _UpdateButtons(); } } if ( !himl ) fChanged |= CSFToolbar::_UpdateIconSize(uIconSize,fUpdateButtons); return fChanged; } void CISFBand::_UpdateVerticalMode(BOOL fVertical) { _fVertical = (fVertical != 0); TraceMsg(TF_BAND, "ISFBand::_UpdateVerticalMode going %hs", _fVertical ? "VERTICAL" : "HORIZONTAL"); ASSERT(_hwnd); if (_hwndPager) { SHSetWindowBits(_hwnd, GWL_STYLE, PGS_HORZ|PGS_VERT, _fVertical ? PGS_VERT : PGS_HORZ); } if (_hwndTB) { SHSetWindowBits(_hwndTB, GWL_STYLE, TBSTYLE_WRAPABLE | CCS_VERT, TBSTYLE_WRAPABLE | (_fVertical ? CCS_VERT : 0)); } } HRESULT IUnknown_QueryBand(IUnknown *punk, DWORD dwBandID, IDeskBand** ppstb, DWORD* pdwState, LPWSTR pszName, int cchName) { HRESULT hr; IBandSite *pbs; hr = punk->QueryInterface(IID_IBandSite, (void**)&pbs); if (SUCCEEDED(hr)) { hr = pbs->QueryBand(dwBandID, ppstb, pdwState, pszName, cchName); pbs->Release(); } return hr; } #define CISFBAND_GETBUTTONSIZE() (_hwndTB ? (LONG)SendMessage(_hwndTB, TB_GETBUTTONSIZE, 0, 0L) : MAKELONG(16, 16)) // // _GetIdealSize // // calculates ideal height and width for band and passes back in // psize, if psize isn't NULL; return value is band's 'ideal length' // (ideal height if vertical, else ideal width) // int CISFBand::_GetIdealSize(PSIZE psize) { SIZE size; LONG lButtonSize = CISFBAND_GETBUTTONSIZE(); RECT rc = {0}; if (_hwndTB) GetClientRect(_hwndTB, &rc); if (_fVertical) { // set width to be max of toolbar width and toolbar button width size.cx = max(RECTWIDTH(rc), LOWORD(lButtonSize)); // have toolbar calculate height given that width SendMessage(_hwndTB, TB_GETIDEALSIZE, TRUE, (LPARAM)&size); } else { // set height to be max of toolbar width and toolbar button width size.cy = max(RECTHEIGHT(rc), HIWORD(lButtonSize)); // have toolbar calculate width given that height SendMessage(_hwndTB, TB_GETIDEALSIZE, FALSE, (LPARAM)&size); } // BUGBUG: I'm ripping out this check as it causes nt5 bug #225449 (disappearing chevron). // _fDirty == TRUE doesn't mean "we're still waiting to call _FillToolbar", it just means // "we need to persist out this order stream". The bit gets set after a drag-and-drop // reordering, but we don't call a matching _FillToolbar in that case. #if 0 if (_fDirty) { // until the TB is populated, we get back bogus data from the // above. so use -1 until we actually have a correct answer. size.cx = size.cy = -1; } #endif if (psize) *psize = size; return _fVertical ? size.cy : size.cx; } /*---------------------------------------------------------- Purpose: IDeskBand::GetBandInfo method */ HRESULT CISFBand::GetBandInfo(DWORD dwBandID, DWORD fViewMode, DESKBANDINFO* pdbi) { HRESULT hr = S_OK; _dwBandID = dwBandID; // We don't know the default icon size until GetBandInfo is called. // After we set the default, we pay attention to the context menu. // if (!_fNoRecalcDefaults) { _uIconSize = (fViewMode & (DBIF_VIEWMODE_FLOATING |DBIF_VIEWMODE_VERTICAL)) ? ISFBVIEWMODE_LARGEICONS : ISFBVIEWMODE_SMALLICONS; _fNoRecalcDefaults = TRUE; } if (!_fInitialized) { _fInitialized = TRUE; _UpdateIconSize(_uIconSize, FALSE); _UpdateShowText(_fNoShowText); } // we treat floating the same as vertical _UpdateVerticalMode(fViewMode & (DBIF_VIEWMODE_FLOATING |DBIF_VIEWMODE_VERTICAL)); LONG lButtonSize = CISFBAND_GETBUTTONSIZE(); pdbi->dwModeFlags = DBIMF_VARIABLEHEIGHT | DBIMF_USECHEVRON; if (_fDebossed) pdbi->dwModeFlags |= DBIMF_DEBOSSED; pdbi->ptMinSize.x = 0; pdbi->ptMaxSize.y = 32000; // random pdbi->ptIntegral.y = 1; pdbi->ptIntegral.x = 1; if (!_fFullOpen) _iIdealLength = _GetIdealSize((PSIZE)&pdbi->ptActual); // CalcMinWidthHeight { // BUGBUG need pager msg for cx/cy scroll #define g_cxScrollbar (GetSystemMetrics(SM_CXVSCROLL) * 3 / 4) #define g_cyScrollbar (GetSystemMetrics(SM_CYVSCROLL) * 3 / 4) #define CX_TBBUTTON_MAX (16 + CX_FILENAME_AVG) // button + name #define CY_TBBUTTON_MAX (16) // button int csBut, csButMin, clBut, clButMin, clScroll; // set up short/long aliases if (_fVertical) { csBut = LOWORD(lButtonSize); if (_fBtnMinSize) csButMin = min(csBut, CX_TBBUTTON_MAX); else csButMin = 0; // people like to shrink things way down, so let 'em clBut = HIWORD(lButtonSize); clButMin = clBut; //ASSERT(min(clBut, CY_TBBUTTON_MAX) == clButMin); // fails! clScroll = g_cyScrollbar; } else { csBut = HIWORD(lButtonSize); csButMin = csBut; //ASSERT(min(csBut, CY_TBBUTTON_MAX) == csButMin); // fails! clBut = LOWORD(lButtonSize); clButMin = min(clBut, CX_TBBUTTON_MAX); clScroll = g_cxScrollbar; // nt5:176448: integral for horz //pdbi->ptIntegral.y = csBut; this is the cause for 287082 and 341592 } // n.b. virt pdbi->pt.x,y is really phys y,x (i.e. phys long,short) pdbi->ptMinSize.x = 0; pdbi->ptMinSize.y = csButMin; DWORD dwState = BSSF_NOTITLE; IUnknown_QueryBand(_punkSite, dwBandID, NULL, &dwState, NULL, 0); if (dwState & BSSF_NOTITLE) { // _fNoTitle int i, cBut, clTmp; // cbut= text notext // horz 1 4 // vert 1 1 cBut = 1; if (!_fVertical && _fNoShowText) { // special-case for QLaunch so see several buttons cBut = 4; // for both QLaunch and arbitrary ISF band } pdbi->ptMinSize.x = cBut * clButMin; if (_hwndPager) { // tack on extra space for pager arrows pdbi->ptMinSize.x += 2 * clScroll; } i = (int)SendMessage(_hwndTB, TB_BUTTONCOUNT, 0, 0); if (i <= cBut) { clTmp = i * clBut; if (clTmp < pdbi->ptMinSize.x) { // scrollbars take as much space as button would // so just do the button pdbi->ptMinSize.x = clTmp; } } } // } #if 0 // BUGBUG don't we need this? if (_fHaveBkColor) { pdbi->crBkgnd = _crBkgnd; pdbi->dwModeFlags |= DBIMF_BKCOLOR; } #endif hr = _GetTitleW(pdbi->wszTitle, ARRAYSIZE(pdbi->wszTitle)); if (FAILED(hr)) { // we don't support title #ifdef DEBUG if (pdbi->dwMask & DBIM_TITLE) TraceMsg(DM_VERBOSE, "cisfb.gbi: patch ~DBIM_TITLE"); #endif pdbi->dwMask &= ~DBIM_TITLE; } return hr; } LRESULT CISFBand::_OnCustomDraw(NMCUSTOMDRAW* pnmcd) { NMTBCUSTOMDRAW * ptbcd = (NMTBCUSTOMDRAW *)pnmcd; LRESULT lres = CDRF_DODEFAULT; switch (pnmcd->dwDrawStage) { case CDDS_PREPAINT: // if there is a palette, then quietly select it into the DC ... if ( _hpalHalftone && _uIconSize == ISFBVIEWMODE_LOGOS ) { ASSERT( pnmcd->hdc ); _hpalOld = SelectPalette( pnmcd->hdc, _hpalHalftone, TRUE ); // LINTASSERT(_hpalOld || !_hpalOld); // 0 semi-ok for SelectPalette RealizePalette( pnmcd->hdc ); } // make sure we get the postpaint as well so we can de-select the palette... lres = CDRF_NOTIFYPOSTPAINT; break; case CDDS_POSTPAINT: // if there is a palette, then quietly select it into the DC ... if ( _hpalHalftone && _uIconSize == ISFBVIEWMODE_LOGOS ) { ASSERT( pnmcd->hdc ); (void) SelectPalette( pnmcd->hdc, _hpalOld, TRUE ); // we don't need a realize here, we can keep the other palette realzied, we // re select the old palette above, otherwise we bleed the resource.... // RealizePalette( pnmcd->hdc ); } break; case CDDS_PREERASE: if (_fHaveBkColor) { RECT rcClient; GetClientRect(_hwndTB, &rcClient); SHFillRectClr(pnmcd->hdc, &rcClient, _crBkgnd); lres = CDRF_SKIPDEFAULT; } break; } return lres; } void CISFBand::_OnDragBegin(int iItem, DWORD dwPreferedEffect) { LPCITEMIDLIST pidl = _IDToPidl(iItem, &_iDragSource); ToolBar_MarkButton(_hwndTB, iItem, TRUE); DragDrop(_hwnd, _psf, pidl, dwPreferedEffect, NULL); ToolBar_MarkButton(_hwndTB, iItem, FALSE); _iDragSource = -1; } LRESULT CISFBand::_OnHotItemChange(NMTBHOTITEM * pnm) { LPNMTBHOTITEM lpnmhi = (LPNMTBHOTITEM)pnm; LRESULT lres = 0; if (_hwndPager && (lpnmhi->dwFlags & HICF_ARROWKEYS)) { int iOldPos, iNewPos; RECT rc, rcPager; int heightPager; int iSelected = lpnmhi->idNew; iOldPos = (int)SendMessage(_hwnd, PGM_GETPOS, (WPARAM)0, (LPARAM)0); iNewPos = iOldPos; SendMessage(_hwndTB, TB_GETITEMRECT, (WPARAM)iSelected, (LPARAM)&rc); if (rc.top < iOldPos) { iNewPos =rc.top; } GetClientRect(_hwnd, &rcPager); heightPager = RECTHEIGHT(rcPager); if (rc.top >= iOldPos + heightPager) { iNewPos += (rc.bottom - (iOldPos + heightPager)) ; } if (iNewPos != iOldPos) SendMessage(_hwnd, PGM_SETPOS, (WPARAM)0, (LPARAM)iNewPos); } else { lres = CToolbarBand::_OnHotItemChange(pnm); } return lres; } LRESULT CISFBand::_OnNotify(LPNMHDR pnm) { LRESULT lres = 0; switch (pnm->code) { case TBN_DROPDOWN: { LPNMTOOLBAR pnmtb = (LPNMTOOLBAR)pnm; lres = TBDDRET_DEFAULT; _DropdownItem(_IDToPidl(pnmtb->iItem), pnmtb->iItem); } break; default: lres = CSFToolbar::_OnNotify(pnm); } return lres; } HRESULT CISFBand::_TBStyleForPidl(LPCITEMIDLIST pidl, DWORD * pdwTBStyle, DWORD* pdwTBState, DWORD * pdwMIFFlags, int* piIcon) { HRESULT hres = CSFToolbar::_TBStyleForPidl(pidl, pdwTBStyle, pdwTBState, pdwMIFFlags, piIcon); if (_fAllowDropdown && !_fCascadeFolder && ((_GetAttributesOfPidl(pidl, SFGAO_FOLDER) & SFGAO_FOLDER) || IsBrowsableShellExt(pidl))) { *pdwTBStyle &= ~BTNS_BUTTON; *pdwTBStyle |= BTNS_DROPDOWN; } return hres; } LRESULT CISFBand::_OnContextMenu(WPARAM wParam, LPARAM lParam) { LRESULT lres; lres = CSFToolbar::_OnContextMenu(wParam, lParam); // todo: csidl? TraceMsg(DM_MISC, "cib._ocm: _dwPriv=%d", _dwPriv); UEMFireEvent(&UEMIID_SHELL, UEME_INSTRBROWSER, UEMF_INSTRUMENT, UIBW_UICONTEXT, (_dwPriv == CSIDL_APPDATA || _dwPriv == CSIDL_FAVORITES) ? UIBL_CTXTQCUTITEM : UIBL_CTXTISFITEM); return lres; } LRESULT CISFBand::_DefWindowProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam) { switch (uMsg) { case WM_SIZE: // forward to toolbar SendMessage(_hwndTB, TB_AUTOSIZE, wParam, lParam); if (_GetIdealSize(NULL) != _iIdealLength) { // our ideal size has changed since the last time bandsite // asked; so tell bandsite ask us for our bandinfo again _BandInfoChanged(); } return 0; } return CSFToolbar::_DefWindowProc(hwnd, uMsg, wParam, lParam); } /*---------------------------------------------------------- Purpose: Set the given IMenuPopup as the submenu to expand. Returns S_FALSE if the menu was modal, S_OK if it was modeless, or failure. */ HRESULT CISFBand::_SetSubMenuPopup(IMenuPopup* pmp, UINT uiCmd, LPCITEMIDLIST pidl, DWORD dwFlagsMPPF) { HRESULT hres = E_FAIL; _ReleaseMenu(); _pmp = pmp; if (pmp) { pmp->AddRef(); RECT rc; POINT pt; SendMessage(_hwndTB, TB_GETRECT, uiCmd, (LPARAM)&rc); MapWindowPoints(_hwndTB, HWND_DESKTOP, (POINT*)&rc, 2); // Align the sub menu appropriately if (_fVertical) { pt.x = rc.right; pt.y = rc.top; } else { pt.x = rc.left; pt.y = rc.bottom; } // // Use a reflect point for the sub-menu to start // if the window is RTL mirrored. [samera] // if (IS_WINDOW_RTL_MIRRORED(_hwndTB)) { pt.x = (_fVertical) ? rc.left : rc.right; } // Tell the sub menu deskbar who we are, so it can // inform us later when the user navigates out of // its scope. IUnknown_SetSite(_pmp, SAFECAST(this, IDeskBand*)); // This must be called after SetSite is done above _SendInitMenuPopup(pmp, pidl); // Show the menubar hres = _pmp->Popup((POINTL*)&pt, (RECTL*)&rc, dwFlagsMPPF); } return hres; } void CISFBand::_SendInitMenuPopup(IMenuPopup * pmp, LPCITEMIDLIST pidl) { } IMenuPopup* ISFBandCreateMenuPopup(IUnknown *punk, IShellFolder* psf, LPCITEMIDLIST pidl, BANDINFOSFB * pbi, BOOL bMenuBand) { return ISFBandCreateMenuPopup2(punk, NULL, psf, pidl, pbi, bMenuBand); } IMenuPopup* ISFBandCreateMenuPopup2(IUnknown *punk, IMenuBand* pmb, IShellFolder* psf, LPCITEMIDLIST pidl, BANDINFOSFB * pbi, BOOL bMenuBand) { IMenuPopup* pmpParent = NULL; VARIANTARG v = {0}; BOOL fUseCache = FALSE; if (punk && pidl) { fUseCache = TRUE; IUnknown_Exec(punk, &CGID_ISFBand, ISFBID_CACHEPOPUP, 0, NULL, &v); if (v.vt == VT_UNKNOWN && v.punkVal) v.punkVal->QueryInterface(IID_IMenuPopup, (void **)&pmpParent); } IMenuPopup * pmp = CreateMenuPopup2(pmpParent, pmb, psf, pidl, pbi, bMenuBand); if (fUseCache) { // cache it now // clear from the variant above to release v.punkVal of pmpParent VariantClear(&v); if (pmp) { VariantInit(&v); v.vt = VT_UNKNOWN; v.punkVal = pmp; pmp->AddRef(); IUnknown_Exec(punk, &CGID_ISFBand, ISFBID_CACHEPOPUP, 0, &v, NULL); VariantClear(&v); } } ATOMICRELEASE(pmpParent); return pmp; } IMenuPopup * CISFBand::_CreateMenuPopup( IShellFolder * psfChild, LPCITEMIDLIST pidlFull, BANDINFOSFB * pbi) { return ISFBandCreateMenuPopup(SAFECAST(this, IOleCommandTarget*), psfChild, pidlFull, pbi, FALSE); } HRESULT CISFBand::_DropdownItem(LPCITEMIDLIST pidl, UINT idCmd) { HRESULT hres = E_FAIL; if (_pidl && _psf) { LPITEMIDLIST pidlFull = ILCombine(_pidl, pidl); if (pidlFull) { IShellFolder* psf; if (SUCCEEDED(_psf->BindToObject(pidl, NULL, IID_IShellFolder, (void **)&psf))) { RECT rc; SendMessage(_hwndTB, TB_GETRECT, idCmd, (LPARAM)&rc); MapWindowPoints(_hwndTB, HWND_DESKTOP, (POINT*)&rc, 2); ITrackShellMenu* ptsm; if (SUCCEEDED(CoCreateInstance(CLSID_TrackShellMenu, NULL, CLSCTX_INPROC_SERVER, IID_ITrackShellMenu, (void**)&ptsm))) { CFavoritesCallback *pfcb = new CFavoritesCallback(); if(pfcb) { ptsm->Initialize(pfcb, 0, 0, SMINIT_TOPLEVEL | SMINIT_VERTICAL|SMINIT_NOSETSITE); pfcb->SetSite(_punkSite); } else ptsm->Initialize(NULL, 0, 0, SMINIT_TOPLEVEL | SMINIT_VERTICAL); if (SUCCEEDED(ptsm->SetShellFolder(psf, pidlFull, NULL, SMSET_TOP | SMSET_USEBKICONEXTRACTION))) { POINTL pt = {rc.left, rc.right}; hres = ptsm->Popup(_hwndTB, &pt, (RECTL*)&rc, MPPF_BOTTOM); } if(pfcb) pfcb->Release(); ptsm->Release(); } psf->Release(); } ILFree(pidlFull); } } return hres; } /*---------------------------------------------------------- Purpose: Try treating the pidl as a cascading menu item. Returns: non-zero if succeeded */ LRESULT CISFBand::_TryCascadingItem(LPCITEMIDLIST pidl, UINT uiCmd) { LRESULT lRet = 0; // Do we cascade to another submenu? if ((GetKeyState(VK_CONTROL) < 0) || _fCascadeFolder) { // Is the item a browsable folder? if ((_GetAttributesOfPidl(pidl, SFGAO_FOLDER) & SFGAO_FOLDER) || IsBrowsableShellExt(pidl)) { // Yes; cascade the browsable folder as a submenu lRet = (S_OK == _DropdownItem(pidl, uiCmd)); } } return lRet; } /*---------------------------------------------------------- Purpose: Try just invoking the pidl Returns: non-zero if succeeded */ LRESULT CISFBand::_TrySimpleInvoke(LPCITEMIDLIST pidl) { LRESULT lRet = 0; if (S_OK == _pbp->IsConnected()) // Force IE { LPITEMIDLIST pidlDest; if (SUCCEEDED(SHGetNavigateTarget(_psf, pidl, &pidlDest, NULL)) && pidlDest && ILIsWeb(pidlDest)) { TCHAR szPath[MAX_PATH]; // We want to ensure that we first give NavFrameWithFile a chance // since this will do the right thing if the PIDL points to a // shortcut. // If the PIDL is a shortcut, NavFrameWithFile will restore any // persistence information stored in the shortcut // if that fails - we take the default code path that simply // uses the PIDL lRet = SUCCEEDED(GetPathForItem(_psf, pidl, szPath, NULL)) && SUCCEEDED(NavFrameWithFile(szPath, (IServiceProvider *)this)); if (!lRet) { if (EVAL(_pbp) && (SUCCEEDED(_pbp->NavigateToPIDL(pidlDest)))) lRet = 1; } ILFree(pidlDest); } } if (!lRet) { IContextMenu *pcm = (LPCONTEXTMENU)_GetUIObjectOfPidl(pidl, IID_IContextMenu); if (pcm) { LPCSTR pVerb = NULL; UINT fFlags = 0; // If ALT double click, accelerator for "Properties..." if (GetKeyState(VK_MENU) < 0) { pVerb = SZ_PROPERTIESA; } // // SHIFT+dblclick does a Explore by default // if (GetKeyState(VK_SHIFT) < 0) { fFlags |= CMF_EXPLORE; } IContextMenu_Invoke(pcm, _hwndTB, pVerb, fFlags); pcm->Release(); } } return lRet; } /*---------------------------------------------------------- Purpose: Helper function to call the menubar site's IMenuPopup::OnSelect method. */ HRESULT CISFBand::_SiteOnSelect(DWORD dwType) { IMenuPopup * pmp; HRESULT hres = IUnknown_QueryService(_punkSite, SID_SMenuPopup, IID_IMenuPopup, (void **)&pmp); if (SUCCEEDED(hres)) { pmp->OnSelect(dwType); pmp->Release(); } return hres; } LRESULT CISFBand::_OnCommand(WPARAM wParam, LPARAM lParam) { UINT uiCmd = GET_WM_COMMAND_ID(wParam, lParam); LRESULT lres = 0; TraceMsg(TF_BAND, "_OnCommand 0x%x", uiCmd); LPCITEMIDLIST pidl = _IDToPidl(uiCmd); if (pidl) { if (_eUemLog != UEMIND_NIL) { // FEATURE_UASSIST should be grp,uiCmd UEMFireEvent(&UEMIID_SHELL, UEME_UIQCUT, UEMF_XEVENT, -1, (LPARAM)-1); } // Only do this if we are the quick links in the browser. The derived class will set this if (_pguidUEMGroup) { LPITEMIDLIST pidlFull = ILCombine(_pidl, pidl); if (pidlFull) { UEMFireEvent(_pguidUEMGroup, UEME_RUNPIDL, UEMF_XEVENT, (WPARAM)_psf, (LPARAM)pidl); SHSendChangeMenuNotify(NULL, SHCNEE_PROMOTEDITEM, 0, pidlFull); ILFree(pidlFull); } } lres = _TryCascadingItem(pidl, uiCmd); if (!lres && _fChannels) lres = _TryChannelSurfing(pidl); if (!lres) lres = _TrySimpleInvoke(pidl); } else { MessageBeep(MB_OK); } return(lres); } // *** IPersistStream // HRESULT CISFBand::GetClassID(CLSID *pClassID) { *pClassID = CLSID_ISFBand; return S_OK; } // // This might be a directory inside CSIDL_APPDATA that was created on // a Win9x machine. Win9x doesn't do the special folder signature info, // so when it shows up on NT, it's just a boring directory that now points // to the wrong place. // // So if we get a bad directory, see if it's one of these corrupted // Win9x pidls and if so, try to reconstitute the original CSIDL_APPDATA // by searching for "Application Data". // void CISFBand::_FixupAppDataDirectory() { TCHAR szDirPath[MAX_PATH]; // We use PathFileExists to check for existence because it turns off // hard error boxes if the target is not available (e.g., floppy not // in drive) if (SHGetPathFromIDList(_pidl, szDirPath) && !PathFileExists(szDirPath)) { static TCHAR szBSAppData[] = TEXT("\\Application Data"); LPTSTR pszAppData; // For every instance of "Application Data", try to graft it // into the real CSIDL_APPDATA. If it works, run with it. for (pszAppData = szDirPath; pszAppData = StrStrI(pszAppData, szBSAppData); pszAppData++) { // Found a candidate. The thing after "\\Application Data" // had better be another backslash (in which case we step // over it) or the end of the string (in which case we don't). TCHAR szPathBuffer[MAX_PATH]; LPTSTR pszTail = pszAppData + ARRAYSIZE(szBSAppData) - 1; // If we did our math right, we should be right after the // "a" at the end of "Application Data". ASSERT(pszTail[-1] == TEXT('a')); if (pszTail[0] == TEXT('\\')) pszTail++; // Step over separator else if (pszTail[0] == TEXT('\0')) { } // at end of string; stay there else continue; // we were faked out; keep looking if (SHGetSpecialFolderPath(NULL, szPathBuffer, CSIDL_APPDATA, FALSE)) { PathCombine(szPathBuffer, szPathBuffer, pszTail); if (PathFileExists(szPathBuffer)) { LPITEMIDLIST pidlReal; pidlReal = ILCreateFromPath(szPathBuffer); if (pidlReal) { ILFree(_pidl); _pidl = pidlReal; } ASSERT(_pidl); break; // found it; stop looking } } } } } typedef struct tagBANDISFSTREAM { WORD wVersion; // version of this structure WORD cbSize; // size of this structure DWORD dwFlags; // BANDISF_ flags DWORD dwPriv; // special folder identifier WORD wViewMode; // small/large/logo WORD wUnused; // For DWORD alignment COLORREF crBkgnd; // band background color COLORREF crBtnLt; // band button hilite color COLORREF crBtnDk; // band button lolite color } BANDISFSTREAM, * PBANDISFSTREAM; #define BANDISF_VERSION 0x22 #define BANDISF_MASK_PSF 0x00000001 // TRUE if _psf is saved #define BANDISF_BOOL_NOSHOWTEXT 0x00000002 // TRUE if _fNoShowText #define BANDISF_BOOL_LARGEICON 0x00000004 // last used in version 0x20 #define BANDISF_MASK_PIDLASLINK 0x00000008 // TRUE if _pidl is saved as a link #define BANDISF_UNUSED10 0x00000010 // (obsolete) was BOOL_NOTITLE #define BANDISF_BOOL_CHANNELS 0x00000020 // TRUE if in channel mode #define BANDISF_BOOL_ALLOWRENAME 0x00000040 // TRUE if _psf context menu should be enabled #define BANDISF_BOOL_DEBOSSED 0x00000080 // TRUE if band should have embossed background #define BANDISF_MASK_ORDERLIST 0x00000100 // TRUE if an order list is saved #define BANDISF_BOOL_BKCOLOR 0x00000200 // TRUE if bk color is persisted #define BANDISF_BOOL_FULLOPEN 0x00000400 // TRUE if band should maximize when opened #define BANDISF_BOOL_NONAMESORT 0x00000800 // TRUE if band should _not_ sort icons by name #define BANDISF_BOOL_BTNMINSIZE 0x00001000 // TRUE if band should report min thickness of button #define BANDISF_BOOL_COLORS 0x00002000 // TRUE if colors are persisted #define BANDISF_VALIDBITS 0x00003FFF HRESULT CISFBand::Load(IStream *pstm) { HRESULT hres; DWORD cbRead; BANDISFSTREAM bisfs = {0}; // figure out what we need to load // // read first DWORD only (old stream format started with ONE dword) hres = pstm->Read(&bisfs, SIZEOF(DWORD), &cbRead); if (SUCCEEDED(hres)) { if (bisfs.cbSize == 0) { // upgrade case, IE4 beta1 shipped this way // bisfs.dwFlags = *((LPDWORD)&bisfs); bisfs.cbSize = SIZEOF(bisfs); bisfs.wVersion = BANDISF_VERSION; bisfs.dwPriv = -1; bisfs.wViewMode = (bisfs.dwFlags & BANDISF_BOOL_LARGEICON) ? ISFBVIEWMODE_LARGEICONS : ISFBVIEWMODE_SMALLICONS; } else { // read rest of stream // DWORD dw = (DWORD)bisfs.cbSize; if (dw > SIZEOF(bisfs)) dw = SIZEOF(bisfs); dw -= SIZEOF(DWORD); hres = pstm->Read(&(bisfs.dwFlags), dw, &cbRead); if (FAILED(hres)) return(hres); } // HEY, DON'T BE LAME ANY MORE. When you next touch this code, // I suggest you figure out what sizes of this structure have // been actually shipped and only upgrade those. Also use // the offsetof macro so you don't have to keep calculating these // things... // old upgrade, I don't know what state is persisted at setup time! // if (bisfs.cbSize == SIZEOF(bisfs) - 3*SIZEOF(COLORREF) - SIZEOF(DWORD) - SIZEOF(DWORD)) { bisfs.dwPriv = -1; bisfs.cbSize += SIZEOF(DWORD); } // most recent upgrade, this is NOT persisted in registry at setup time!!! // if (bisfs.cbSize == SIZEOF(bisfs) - 3*SIZEOF(COLORREF) - SIZEOF(DWORD)) { bisfs.wViewMode = (bisfs.dwFlags & BANDISF_BOOL_LARGEICON) ? ISFBVIEWMODE_LARGEICONS : ISFBVIEWMODE_SMALLICONS; bisfs.cbSize = SIZEOF(bisfs); } // upgrade from version 0x21 + crBkgnd only to 0x22 // if (bisfs.cbSize == SIZEOF(bisfs) - 2*SIZEOF(COLORREF)) { bisfs.cbSize = SIZEOF(bisfs); } // upgrade from version 0x21 to 0x22 // if (bisfs.cbSize == SIZEOF(bisfs) - 3*SIZEOF(COLORREF)) { bisfs.cbSize = SIZEOF(bisfs); } if (!EVAL(bisfs.cbSize >= SIZEOF(bisfs))) { return(E_FAIL); } ASSERT(!(bisfs.dwFlags & ~BANDISF_VALIDBITS)); if (bisfs.dwFlags & BANDISF_BOOL_NOSHOWTEXT) _fNoShowText = TRUE; if (bisfs.dwFlags & BANDISF_BOOL_ALLOWRENAME) _fAllowRename = TRUE; if (bisfs.dwFlags & BANDISF_BOOL_DEBOSSED) _fDebossed = TRUE; if (bisfs.dwFlags & BANDISF_BOOL_FULLOPEN) _fFullOpen = TRUE; if (bisfs.dwFlags & BANDISF_BOOL_NONAMESORT) _fNoNameSort = TRUE; if (bisfs.dwFlags & BANDISF_BOOL_BTNMINSIZE) _fBtnMinSize = TRUE; if (bisfs.dwFlags & BANDISF_BOOL_BKCOLOR) { _crBkgnd = bisfs.crBkgnd; _fHaveBkColor = TRUE; } if (bisfs.dwFlags & BANDISF_BOOL_COLORS) { _crBtnLt = bisfs.crBtnLt; _crBtnDk = bisfs.crBtnDk; _fHaveColors = TRUE; } _dwPriv = bisfs.dwPriv; #if 1 // BUGBUG FEATURE_UASSIST hack this should be persisted not recalc'ed #define UEMIsLogCsidl(dwPrivID) ((dwPrivID) == CSIDL_APPDATA) if (UEMIsLogCsidl(_dwPriv)) { _eUemLog = UEMIND_SHELL; } #endif _uIconSize = bisfs.wViewMode; _fNoRecalcDefaults = TRUE; if (bisfs.dwFlags & BANDISF_MASK_PIDLASLINK) { ASSERT(NULL==_pidl); hres = LoadPidlAsLink(_punkSite, pstm, &_pidl); // If we hit hits, LoadPidlAsLink() read a chuck of our data. - BryanSt ASSERT(SUCCEEDED(hres)); // DEBUG_CODE(TCHAR szDbgBuffer[MAX_PATH];) // TraceMsg(TF_BAND|TF_GENERAL, "CISFBand::Load() _pidl=>%s<", Dbg_PidlStr(_pidl, szDbgBuffer, SIZECHARS(szDbgBuffer))); _FixupAppDataDirectory(); } if (SUCCEEDED(hres) && (bisfs.dwFlags & BANDISF_MASK_PSF)) { ASSERT(NULL == _psf); hres = OleLoadFromStream(pstm, IID_IShellFolder, (void **)&_psf); } // map this to working info // if (SUCCEEDED(hres)) _AfterLoad(); // we need _psf before we can read the order list. if (SUCCEEDED(hres) && (bisfs.dwFlags & BANDISF_MASK_ORDERLIST)) { hres = OrderList_LoadFromStream(pstm, &_hdpaOrder, _psf); if (SUCCEEDED(hres)) { // _fDropped "persists" along with the orderlist - if this flag // is set, we assume we have a non-default ordering _fDropped = TRUE; } } } return hres; } HRESULT SaveIsfToStream(IShellFolder *psf, IStream *pstm) { IPersistStream* pps; HRESULT hres = psf->QueryInterface(IID_IPersistStream, (void **)&pps); if (SUCCEEDED(hres)) { hres = OleSaveToStream(pps, pstm); pps->Release(); } return hres; } HRESULT CISFBand::Save(IStream *pstm, BOOL fClearDirty) { IPersistStream* pps = NULL; HRESULT hres; BANDISFSTREAM bisfs = {0}; // figure out what we will save // if (_pidl) bisfs.dwFlags |= BANDISF_MASK_PIDLASLINK; // BUGBUG(lamadio): This case is busted. None of the IShellFolders implement IPersistStream (at least as far as // TJ and I can see). Qhen quick links initializes, it will set the pidlQuickLinks as the _pidl. So, in the // After load, _fPSFBandDesktop gets set to TRUE. Why? I don't know. Well, then we never attempt to persist the // IShellFolder and we will never fail the save. We should remove this case so we don't run into this again. if (_psf && !_fPSFBandDesktop) bisfs.dwFlags |= BANDISF_MASK_PSF; if (_fDropped && (_hdpa || _hdpaOrder)) // only if a drop occurred do we have non-default ordering bisfs.dwFlags |= BANDISF_MASK_ORDERLIST; if (_fNoShowText) bisfs.dwFlags |= BANDISF_BOOL_NOSHOWTEXT; if (_fAllowRename) bisfs.dwFlags |= BANDISF_BOOL_ALLOWRENAME; if (_fDebossed) bisfs.dwFlags |= BANDISF_BOOL_DEBOSSED; if (_fFullOpen) bisfs.dwFlags |= BANDISF_BOOL_FULLOPEN; if (_fNoNameSort) bisfs.dwFlags |= BANDISF_BOOL_NONAMESORT; if (_fBtnMinSize) bisfs.dwFlags |= BANDISF_BOOL_BTNMINSIZE; if (_fHaveBkColor) { bisfs.dwFlags |= BANDISF_BOOL_BKCOLOR; bisfs.crBkgnd = _crBkgnd; } if (_fHaveColors) { bisfs.dwFlags |= BANDISF_BOOL_COLORS; bisfs.crBtnLt = _crBtnLt; bisfs.crBtnDk = _crBtnDk; } bisfs.cbSize = SIZEOF(bisfs); bisfs.wVersion = BANDISF_VERSION; bisfs.dwPriv = _dwPriv; bisfs.wViewMode = _uIconSize; // now save it // hres = pstm->Write(&bisfs, SIZEOF(bisfs), NULL); if (SUCCEEDED(hres) && bisfs.dwFlags & BANDISF_MASK_PIDLASLINK) { hres = SavePidlAsLink(_punkSite, pstm, _pidl); // BUGBUG: We need to save a terminator. } if (SUCCEEDED(hres) && bisfs.dwFlags & BANDISF_MASK_PSF) { hres = SaveIsfToStream(_psf, pstm); } if (SUCCEEDED(hres) && (bisfs.dwFlags & BANDISF_MASK_ORDERLIST)) { hres = OrderList_SaveToStream(pstm, (_hdpa ? _hdpa : _hdpaOrder), _psf); } return(hres); } #if 0 // IPersistPropertyBag implementation // HRESULT CISFBand::Load(IPropertyBag *pPropBag, IErrorLog *pErrorLog) { ASSERT(0); // obsolete! _fCascadeFolder = PropBag_ReadInt4(pPropBag, L"Cascade", FALSE); // n.b. old "Title" property nuked _uIconSize = (PropBag_ReadInt4(pPropBag, L"Large", TRUE) ? ISFBVIEWMODE_LARGEICONS : ISFBVIEWMODE_SMALLICONS); _fNoShowText = PropBag_ReadInt4(pPropBag, L"Text", TRUE); return(S_OK); } HRESULT CISFBand::Save(IPropertyBag *pPropBag, BOOL fClearDirty, BOOL fSaveAllProperties) { return(E_NOTIMPL); } HRESULT CISFBand::InitNew() { ASSERT(0); // obsolete! return(E_NOTIMPL); } #endif // IContextMenu implementation // HRESULT CISFBand::InvokeCommand(LPCMINVOKECOMMANDINFO lpici) { BOOL fChanged = FALSE; int idCmd = -1; UINT uNewMode = 0; if (!HIWORD(lpici->lpVerb)) idCmd = LOWORD(lpici->lpVerb); switch (idCmd) { case ISFBIDM_LARGE: uNewMode = ISFBVIEWMODE_LARGEICONS; goto newViewMode; case ISFBIDM_SMALL: uNewMode = ISFBVIEWMODE_SMALLICONS; newViewMode: if (uNewMode != _uIconSize) { BOOL fRefresh = FALSE; if (uNewMode == ISFBVIEWMODE_LOGOS || _uIconSize == ISFBVIEWMODE_LOGOS) { // invalidate all before switching the imagelist... _RememberOrder(); EmptyToolbar(); fRefresh = TRUE; } // we Logo view has now left the building... if ( uNewMode != ISFBVIEWMODE_LOGOS && _uIconSize == ISFBVIEWMODE_LOGOS ) { ExitLogoView(); } fChanged = _UpdateIconSize(uNewMode, TRUE); if ( fRefresh ) { _FillToolbar(); } if (fChanged) _BandInfoChanged(); } // fall thru default: return CSFToolbar::InvokeCommand(lpici); } return(S_OK); } // *** IOleCommandTarget methods *** STDMETHODIMP CISFBand::QueryStatus(const GUID *pguidCmdGroup, ULONG cCmds, OLECMD rgCmds[], OLECMDTEXT *pcmdtext) { HRESULT hr = OLECMDERR_E_UNKNOWNGROUP; if (pguidCmdGroup == NULL) { // nothing } else if (IsEqualGUID(CGID_ISFBand, *pguidCmdGroup)) { for (UINT i = 0; i < cCmds; i++) { switch (rgCmds[i].cmdID) { case ISFBID_CACHEPOPUP: case ISFBID_ISITEMVISIBLE: case ISFBID_PRIVATEID: rgCmds[i].cmdf |= OLECMDF_SUPPORTED; break; } } hr = S_OK; } else if (IsEqualGUID(CGID_ShellDocView, *pguidCmdGroup)) { for (UINT i = 0; i < cCmds; i++) { switch (rgCmds[i].cmdID) { case SHDVID_UEMLOG: rgCmds[i].cmdf |= OLECMDF_SUPPORTED; break; } } hr = S_OK; } return hr; } HRESULT CISFBand::_IsPidlVisible(LPITEMIDLIST pidl) { int i; if (_GetButtonFromPidl(pidl, NULL, &i)) { RECT rc; GetClientRect(_hwndTB, &rc); if (SHIsButtonObscured(_hwndTB, &rc, i)) return S_FALSE; else return S_OK; } return E_FAIL; } HRESULT CISFBand::_OrderListFromIStream(VARIANT* pvarargIn) { HRESULT hres = E_FAIL; if (pvarargIn->vt == VT_UNKNOWN) { IStream* pstm; if (SUCCEEDED(pvarargIn->punkVal->QueryInterface(IID_IStream, (void**)&pstm))) { OrderList_Destroy(&_hdpaOrder); hres = OrderList_LoadFromStream(pstm, &_hdpaOrder, _psf); if (SUCCEEDED(hres)) { _SetDirty(TRUE); if (_fShow) { _FillToolbar(); } } pstm->Release(); } } return hres; } HRESULT CISFBand::_IStreamFromOrderList(VARIANT* pvarargOut) { HRESULT hres = E_OUTOFMEMORY; ASSERT(pvarargOut != NULL); IStream* pstm = SHCreateMemStream(NULL, 0); if (pstm) { hres = OrderList_SaveToStream(pstm, _hdpa, _psf); if (SUCCEEDED(hres)) { pvarargOut->vt = VT_UNKNOWN; pvarargOut->punkVal = pstm; pvarargOut->punkVal->AddRef(); } pstm->Release(); } return hres; } STDMETHODIMP CISFBand::Exec(const GUID *pguidCmdGroup, DWORD nCmdID, DWORD nCmdexecopt, VARIANTARG *pvarargIn, VARIANTARG *pvarargOut) { if (pguidCmdGroup == NULL) { // nothing } else if (IsEqualGUID(CGID_ISFBand, *pguidCmdGroup)) { switch (nCmdID) { case ISFBID_CACHEPOPUP: if (pvarargIn && pvarargIn->vt == VT_UNKNOWN) { IMenuPopup* pmp = NULL; if (pvarargIn->punkVal) pvarargIn->punkVal->QueryInterface(IID_IMenuPopup, (void **)&pmp); _SetCacheMenuPopup(pmp); ATOMICRELEASE(pmp); } if (pvarargOut) { pvarargOut->vt = VT_UNKNOWN; pvarargOut->punkVal = _pmpCache; if (_pmpCache) _pmpCache->AddRef(); } return S_OK; case ISFBID_ISITEMVISIBLE: { HRESULT hr = E_INVALIDARG; if (pvarargIn && pvarargIn->vt == VT_INT_PTR) hr = _IsPidlVisible((LPITEMIDLIST)pvarargIn->byref); return hr; } case ISFBID_PRIVATEID: // hack hack for BSMenu to differentiate between specially created // isfbands. see bsmenu's _FindBand // if pvarargOut is set, we give back the id we have stored. if (pvarargOut) { pvarargOut->vt = VT_I4; pvarargOut->lVal = _dwPriv; } // if pvarargIn is set, then we take and keep this id. if (pvarargIn && pvarargIn->vt == VT_I4) _dwPriv = pvarargIn->lVal; return S_OK; case ISFBID_GETORDERSTREAM: return _IStreamFromOrderList(pvarargOut); case ISFBID_SETORDERSTREAM: return _OrderListFromIStream(pvarargIn); } } else if (IsEqualGUID(CGID_ShellDocView, *pguidCmdGroup)) { switch (nCmdID) { case SHDVID_UEMLOG: ASSERT(pvarargOut == NULL); // if pvarargIn is set, then we take and keep this id. if (pvarargIn && pvarargIn->vt == VT_I4) { _eUemLog = pvarargIn->lVal; ASSERT(_eUemLog == UEMIND_SHELL || _eUemLog == UEMIND_BROWSER); } return S_OK; } } else if (IsEqualGUID(CGID_DeskBand, *pguidCmdGroup)) { switch (nCmdID) { case DBID_DELAYINIT: _fDelayInit = TRUE; break; case DBID_FINISHINIT: _fDelayInit = FALSE; _RegisterToolbar(); break; } return S_OK; } return OLECMDERR_E_NOTSUPPORTED; } IShellFolder * CISFBand::GetSF() { ASSERT( _psf ); return _psf; } HWND CISFBand::GetHWND() { return _hwndTB; } REFTASKOWNERID CISFBand::GetTOID() { return TOID_ExtractImage; } HRESULT CISFBand::OnTranslatedChange(LONG lEvent, LPCITEMIDLIST pidl1, LPCITEMIDLIST pidl2) { if (lEvent == SHCNE_RMDIR && _IsEqualID(pidl1)) { HRESULT hres = E_FAIL; IBandSite *pbandSite; if (_punkSite) { hres = _punkSite->QueryInterface(IID_IBandSite, (void **)&pbandSite); if (EVAL(SUCCEEDED(hres))) { pbandSite->RemoveBand(_dwBandID); pbandSite->Release(); } } return hres; } else { return CSFToolbar::OnTranslatedChange(lEvent, pidl1, pidl2); } } HRESULT CISFBand::UpdateLogoCallback( DWORD dwItem, int iIcon, HBITMAP hImage, LPCWSTR pszCache, BOOL fCache ) { int iItem = (int)dwItem; HRESULT hr; UINT uImage; // catch if we are closing... if ( _fClosing ) return NOERROR; IMAGECACHEINFO rgInfo; rgInfo.dwMask = ICIFLAG_NAME | ICIFLAG_LARGE | ICIFLAG_BITMAP | ICIFLAG_NOUSAGE; rgInfo.cbSize = sizeof( rgInfo ); rgInfo.pszName = pszCache; rgInfo.hBitmapLarge = hImage; ASSERT(_pLogoCache); if (_pLogoCache) hr = _pLogoCache->AddImage( &rgInfo, &uImage ); else hr = E_FAIL; // catch if we are closing... if ( _fClosing ) return NOERROR; if ( SUCCEEDED( hr )) { // remember the icon to logo mapping.... AddIndicesToLogoList( iIcon, uImage ); // catch we are closing before we try and doa bloc PostMessage( _hwndTB, TB_CHANGEBITMAP, iItem, uImage ); } // stop delay painting when the last extract image task calls back if (_fDelayPainting) { if (_pTaskScheduler && _pTaskScheduler->CountTasks(TOID_NULL) == 1) { _StopDelayPainting(); } } return hr; } // } HRESULT CISFBand::_GetTitleW(LPWSTR pwszTitle, DWORD cchSize) { HRESULT hr = E_FAIL; TraceMsg(TF_BAND, "Calling baseclass CISFBand::_GetTitleW"); if (!EVAL(pwszTitle)) return E_INVALIDARG; *pwszTitle = 0; if (_pidl) { hr = SHGetNameAndFlagsW(_pidl, SHGDN_NORMAL, pwszTitle, cchSize, NULL); } else if (_psf && !_fPSFBandDesktop) { #ifdef BUSTED // BUGBUG (scotth): We cannot call GetDisplayNameOf with NULL pidl. // We must change this code so _pidl is always // valid, and key off a flag to determine whether // to receive notifies. Remove this code once // that is done. STRRET strret; if (SUCCEEDED(_psf->GetDisplayNameOf(NULL, SHGDN_NORMAL, &strret))) StrRetToBufW(&strret, NULL, pwszTitle, cchSize); #endif } return hr; } STDAPI NavigateToPIDL(IWebBrowser2* pwb, LPCITEMIDLIST pidl); HRESULT FakeGetNavigateTarget(IShellFolder *psf, LPCITEMIDLIST pidl, LPITEMIDLIST *ppidl); LRESULT CISFBand::_TryChannelSurfing(LPCITEMIDLIST pidl) { LRESULT lRet = 0; ASSERT(_fChannels); LPITEMIDLIST pidlTarget; HRESULT hr = SHGetNavigateTarget(_psf, pidl, &pidlTarget, NULL); // channel category folders hack. if (FAILED(hr)) hr = FakeGetNavigateTarget(_psf, pidl, &pidlTarget); if (SUCCEEDED(hr)) { IWebBrowser2* pwb; // n.b. careful! only one of GCB and C_OB up the refcnt _GetChannelBrowser(&pwb); if (SUCCEEDED(Channels_OpenBrowser(&pwb, pwb != NULL))) { lRet = 1; // success at this point if (SUCCEEDED(NavigateToPIDL(pwb, pidlTarget))) { LPITEMIDLIST pidlFull = ILCombine(_pidl, pidl); if (pidlFull) { VARIANT varURLpidl, flags; flags.vt = VT_I4; flags.lVal = navBrowserBar; if (SUCCEEDED(InitVariantFromIDList(&varURLpidl, pidlFull))) { pwb->Navigate2(&varURLpidl, &flags, PVAREMPTY, PVAREMPTY, PVAREMPTY); VariantClear(&varURLpidl); } ILFree(pidlFull); } } } if (pwb) pwb->Release(); ILFree(pidlTarget); } return lRet; } //*** _GetChannelBrowser -- find appropriate browser for surfing // DESCRIPTION // for the DTBrowser case, we fail (pwb=NULL, hr=S_FALSE) so that our // caller will create a new SHBrowser (which can be put into theater mode). // for the SHBrowser case, we find the top-level browser (so we'll navigate // in-place). HRESULT CISFBand::_GetChannelBrowser(IWebBrowser2 **ppwb) { HRESULT hr; IServiceProvider *psp; *ppwb = NULL; // assume failure if (_fDesktop) { ASSERT(*ppwb == NULL); hr = S_FALSE; } else { hr = IUnknown_QueryService(_punkSite, SID_STopLevelBrowser, IID_IServiceProvider, (void**)&psp); ASSERT(SUCCEEDED(hr)); if (SUCCEEDED(hr)) { hr = psp->QueryService(SID_SWebBrowserApp, IID_IWebBrowser2, (void **)ppwb); ASSERT(SUCCEEDED(hr)); psp->Release(); } } return hr; } HRESULT IUnknown_SetBandInfoSFB(IUnknown *punkBand, BANDINFOSFB *pbi) { HRESULT hr = E_FAIL; IShellFolderBand *pisfBand; if (punkBand) { hr = punkBand->QueryInterface(IID_IShellFolderBand, (void **)&pisfBand); if (EVAL(SUCCEEDED(hr))) { hr = pisfBand->SetBandInfoSFB(pbi); pisfBand->Release(); } } return hr; } /////////////////////////////////////////////////////////////////////////////////////////// ///////////////////////////CExtractImageTask/////////////////////////////////////////////// /////////////////////////////////////////////////////////////////////////////////////////// // Warning // // The CLogoBase class cannot have a ref on the returned task // since that would be a circular reference // // Warning HRESULT CExtractImageTask_Create( CLogoBase *plb, LPEXTRACTIMAGE pExtract, LPCWSTR pszCache, DWORD dwItem, int iIcon, DWORD dwFlags, LPRUNNABLETASK * ppTask ) { if ( !ppTask || !plb || !pExtract ) { return E_INVALIDARG; } HRESULT hr = NOERROR; CExtractImageTask * pNewTask = new CExtractImageTask( &hr, plb, pExtract, pszCache, dwItem, iIcon, dwFlags ); if ( !pNewTask ) { return E_OUTOFMEMORY; } if ( FAILED( hr )) { pNewTask->Release(); return hr; } *ppTask = SAFECAST( pNewTask, IRunnableTask *); return NOERROR; } ///////////////////////////////////////////////////////////////////////////////////////////////////// CExtractImageTask::CExtractImageTask( HRESULT * pHr, CLogoBase *plb, IExtractImage * pImage, LPCWSTR pszCache, DWORD dwItem, int iIcon, DWORD dwFlags ) { m_lState = IRTIR_TASK_NOT_RUNNING; m_plb = plb; m_plb->AddRef(); // cannot assume the band will kill us before it dies.... // hence we hold a reference StrCpyW( m_szPath, pszCache ); m_pExtract = pImage; pImage->AddRef(); m_cRef = 1; // use the upper bit of the flags to determine if we should always call.... m_dwFlags = dwFlags; m_dwItem = dwItem; m_iIcon = iIcon; // Since the task moves from thread to thread, // don't charge this thread for the objects we're using } ///////////////////////////////////////////////////////////////////////////////////////////////////// CExtractImageTask::~CExtractImageTask() { ATOMICRELEASE( m_pExtract ); ATOMICRELEASE( m_pTask ); if ( m_hBmp && !( m_dwFlags & EITF_SAVEBITMAP )) { DeleteObject( m_hBmp ); } if(m_plb) m_plb->Release(); } ////////////////////////////////////////////////////////////////////////////////////////// STDMETHODIMP CExtractImageTask::QueryInterface( REFIID riid, void **ppvObj ) { if ( !ppvObj ) { return E_INVALIDARG; } if ( IsEqualIID( riid, IID_IUnknown )) { *ppvObj = SAFECAST( this, IUnknown *); } else if ( IsEqualIID( riid, IID_IRunnableTask )) { *ppvObj = SAFECAST( this, IRunnableTask *); } else { return E_NOINTERFACE; } AddRef(); return NOERROR; } ////////////////////////////////////////////////////////////////////////////////////////// STDMETHODIMP_ (ULONG) CExtractImageTask::AddRef() { InterlockedIncrement( &m_cRef ); return m_cRef; } ////////////////////////////////////////////////////////////////////////////////////////// STDMETHODIMP_ (ULONG) CExtractImageTask::Release() { if (InterlockedDecrement( &m_cRef ) == 0 ) { delete this; return 0; } return m_cRef; } ////////////////////////////////////////////////////////////////////////////////////////// STDMETHODIMP CExtractImageTask::Run ( void ) { HRESULT hr = E_FAIL; if ( m_lState == IRTIR_TASK_RUNNING ) { hr = S_FALSE; } else if ( m_lState == IRTIR_TASK_PENDING ) { hr = E_FAIL; } else if ( m_lState == IRTIR_TASK_NOT_RUNNING ) { LONG lRes = InterlockedExchange( & m_lState, IRTIR_TASK_RUNNING); if ( lRes == IRTIR_TASK_PENDING ) { m_lState = IRTIR_TASK_FINISHED; return NOERROR; } // see if it supports IRunnableTask m_pExtract->QueryInterface( IID_IRunnableTask, (void **) & m_pTask ); #ifdef UNIX //Hey Guys : IE4.01 has an error - it returns the wrong VTABLE //when this QI is done. We know how our VTABLEs are laid out #else // IE4.01 has an error - it returns the wrong VTABLE // when this QI is done. if((LPVOID)m_pTask == (LPVOID)m_pExtract) { m_pTask = m_pTask + 2; // This vtable is two ptrs away and is in fstree.cpp in shell32 in IE4.01 } #endif if ( m_lState == IRTIR_TASK_RUNNING ) { // start the extractor.... hr = m_pExtract->Extract( &m_hBmp ); } if (( SUCCEEDED( hr ) || ( hr != E_PENDING && (m_dwFlags & EITF_ALWAYSCALL))) && m_lState == IRTIR_TASK_RUNNING ) { hr = InternalResume(); } if ( m_lState != IRTIR_TASK_SUSPENDED || hr != E_PENDING ) { m_lState = IRTIR_TASK_FINISHED; } } return hr; } ///////////////////////////////////////////////////////////////////////////////////////////////////// STDMETHODIMP CExtractImageTask::Kill ( BOOL fWait ) { if ( m_lState != IRTIR_TASK_RUNNING ) { return S_FALSE; } LONG lRes = InterlockedExchange( &m_lState, IRTIR_TASK_PENDING ); if ( lRes == IRTIR_TASK_FINISHED ) { m_lState = lRes; return NOERROR; } // does it support IRunnableTask ? Can we kill it ? HRESULT hr = E_NOTIMPL; if ( m_pTask != NULL ) { hr = m_pTask->Kill( FALSE ); } return hr; } /////////////////////////////////////////////////////////////////////////////////////// STDMETHODIMP CExtractImageTask::Suspend( void ) { if ( !m_pTask ) { return E_NOTIMPL; } if ( m_lState != IRTIR_TASK_RUNNING ) { return E_FAIL; } LONG lRes = InterlockedExchange( &m_lState, IRTIR_TASK_SUSPENDED ); HRESULT hr = m_pTask->Suspend(); if ( SUCCEEDED( hr )) { lRes = (LONG) m_pTask->IsRunning(); if ( lRes == IRTIR_TASK_SUSPENDED ) { m_lState = lRes; } } else { m_lState = lRes; } return hr; } //////////////////////////////////////////////////////////////////////////////////////// STDMETHODIMP CExtractImageTask::Resume( void ) { if ( !m_pTask ) { return E_NOTIMPL; } if ( m_lState != IRTIR_TASK_SUSPENDED ) { return E_FAIL; } m_lState = IRTIR_TASK_RUNNING; HRESULT hr = m_pTask->Resume(); if ( SUCCEEDED( hr ) || ( hr != E_PENDING && ( m_dwFlags & EITF_ALWAYSCALL ))) { hr = InternalResume(); } return hr; } ///////////////////////////////////////////////////////////////////////////////////////////////////// HRESULT CExtractImageTask::InternalResume() { HRESULT hr = NOERROR; if ( m_dwFlags & EITF_ALWAYSCALL || m_hBmp ) { // call the update function hr = m_plb->UpdateLogoCallback( m_dwItem, m_iIcon, m_hBmp, m_szPath, TRUE ); } m_lState = IRTIR_TASK_FINISHED; return hr; } ///////////////////////////////////////////////////////////////////////////////////////////////////// STDMETHODIMP_( ULONG ) CExtractImageTask:: IsRunning ( void ) { return m_lState; } //////////////////////////////////////////////////////////////////////////////////////////////////// //////////////////////////////CLogoBase///////////////////////////////////////////////////////////// //////////////////////////////////////////////////////////////////////////////////////////////////// // static data... IImageCache * CLogoBase::s_pSharedWideLogoCache = NULL; long CLogoBase::s_lSharedWideLogosRef = 0; HDSA CLogoBase::s_hdsaWideLogoIndices = NULL; CRITICAL_SECTION CLogoBase::s_csSharedLogos = {0}; extern "C" void CLogoBase_Initialize( void ) { CLogoBase::_Initialize(); } extern "C" void CLogoBase_Cleanup( void ) { CLogoBase::_Cleanup( ); } void CLogoBase::_Initialize( void ) { InitializeCriticalSection( &s_csSharedLogos ); } void CLogoBase::_Cleanup( void ) { DeleteCriticalSection( & s_csSharedLogos ); } CLogoBase::CLogoBase( BOOL fWide ) { // are we paletized, then use the global halftone palette .... HDC hdcTmp = GetDC( NULL ); if (hdcTmp) { if (GetDeviceCaps( hdcTmp, RASTERCAPS) & RC_PALETTE) { ASSERT( g_hpalHalftone ); _hpalHalftone = g_hpalHalftone; } ReleaseDC( NULL, hdcTmp ); } _fWide = fWide; } CLogoBase::~CLogoBase() { if (_pLogoCache || _pTaskScheduler) { ExitLogoView(); } // NOTE: no palette release because we are using the global Halftone palette...... } HRESULT CLogoBase::AddRefLogoCache( void ) { if ( _fWide ) { EnterCriticalSection( &s_csSharedLogos ); if ( !s_lSharedWideLogosRef ) { if ( !s_hdsaWideLogoIndices ) { s_hdsaWideLogoIndices = DSA_Create( sizeof( LogoIndex ), 5 ); if ( !s_hdsaWideLogoIndices ) { LeaveCriticalSection( &s_csSharedLogos ); return E_OUTOFMEMORY; } } ASSERT( s_hdsaWideLogoIndices ); ASSERT( !s_pSharedWideLogoCache ); // BUGBUG for now CoCreate one per view HRESULT hr = CoCreateInstance( CLSID_ImageListCache, NULL, CLSCTX_INPROC, IID_IImageCache, (void **) & s_pSharedWideLogoCache ); if ( FAILED( hr )) { LeaveCriticalSection( &s_csSharedLogos ); return hr; } } ASSERT( s_pSharedWideLogoCache ); // bump up the ref and get a pointer to it... s_lSharedWideLogosRef ++; _pLogoCache = s_pSharedWideLogoCache; _pLogoCache->AddRef(); _hdsaLogoIndices = s_hdsaWideLogoIndices; LeaveCriticalSection( &s_csSharedLogos ); return NOERROR; } else { // non wide logo version we don't share because w eonly expect there ever to be one... _hdsaLogoIndices = DSA_Create( sizeof( LogoIndex ), 5 ); if ( !_hdsaLogoIndices ) { return E_OUTOFMEMORY; } // BUGBUG for now CoCreate one per view return CoCreateInstance( CLSID_ImageListCache, NULL, CLSCTX_INPROC, IID_IImageCache, (void **) & _pLogoCache ); } } HRESULT CLogoBase::ReleaseLogoCache( void ) { if ( !_pLogoCache ) { return S_FALSE; } ATOMICRELEASE(_pLogoCache); if ( _fWide ) { EnterCriticalSection( &s_csSharedLogos ); ASSERT( s_lSharedWideLogosRef > 0 ); s_lSharedWideLogosRef --; if ( ! s_lSharedWideLogosRef ) { // let go of the final ref..... ATOMICRELEASE(s_pSharedWideLogoCache); ASSERT( s_hdsaWideLogoIndices ); DSA_Destroy( s_hdsaWideLogoIndices ); s_hdsaWideLogoIndices = NULL; } LeaveCriticalSection( &s_csSharedLogos ); } else { // free the HDSA DSA_Destroy( _hdsaLogoIndices ); _hdsaLogoIndices = NULL; } return NOERROR; } HRESULT CLogoBase::InitLogoView( void ) { HRESULT hr = AddRefLogoCache(); if (SUCCEEDED(hr)) { hr = CoCreateInstance(CLSID_ShellTaskScheduler, NULL, CLSCTX_INPROC, IID_IShellTaskScheduler, (void **) &_pTaskScheduler); if (FAILED(hr)) { ATOMICRELEASE(_pLogoCache); } else { _rgLogoSize.cx = ( _fWide ) ? LOGO_WIDE_WIDTH : LOGO_WIDTH ; _rgLogoSize.cy = LOGO_HEIGHT; IMAGECACHEINITINFO rgInfo; rgInfo.cbSize = sizeof( rgInfo ); rgInfo.dwMask = ICIIFLAG_LARGE; rgInfo.iStart = 0; rgInfo.iGrow = 5; // the color depth is currently the screen resolution... int iColorRes = SHGetCurColorRes(); _dwClrDepth = (DWORD) iColorRes; switch (iColorRes) { case 16 : rgInfo.dwFlags = ILC_COLOR16; break; case 24 : case 32 : rgInfo.dwFlags = ILC_COLOR24; break; default : rgInfo.dwFlags = ILC_COLOR8; } rgInfo.rgSizeLarge = _rgLogoSize; if (_pLogoCache) hr = _pLogoCache->GetImageList(&rgInfo); else hr = E_UNEXPECTED; if (FAILED(hr)) { ATOMICRELEASE(_pLogoCache); ATOMICRELEASE(_pTaskScheduler); } else { _himlLogos = rgInfo.himlLarge; // GetImageList() will return S_FALSE if it was already created... if ((hr == S_OK) && (iColorRes <= 8)) { // init the color table so that it matches The "special halftone palette" HPALETTE hpal = SHCreateShellPalette(NULL); PALETTEENTRY rgColours[256]; RGBQUAD rgDIBColours[256]; ASSERT( hpal ); int nColours = GetPaletteEntries(hpal, 0, ARRAYSIZE(rgColours), rgColours); // SHGetShellPalette should always return a 256 colour palette ASSERT(nColours == ARRAYSIZE(rgColours)); // translate from the LOGPALETTE structure to the RGBQUAD structure ... for (int iColour = 0; iColour < nColours; iColour ++) { rgDIBColours[iColour].rgbRed = rgColours[iColour].peRed; rgDIBColours[iColour].rgbBlue = rgColours[iColour].peBlue; rgDIBColours[iColour].rgbGreen = rgColours[iColour].peGreen; rgDIBColours[iColour].rgbReserved = 0; } DeletePalette(hpal); ImageList_SetColorTable(_himlLogos, 0, 256, rgDIBColours); } } } } return hr; } HRESULT CLogoBase::ExitLogoView( void ) { ATOMICRELEASE( _pTaskScheduler ); // the task scheduler callbacks can reference // the logocache, so make sure you free the // logo cache AFTER the task scheduler! ReleaseLogoCache(); return NOERROR; } int CLogoBase::GetCachedLogoIndex( DWORD dwItem, LPCITEMIDLIST pidl, LPRUNNABLETASK *ppTask, DWORD * pdwPriority, DWORD *pdwFlags ) { DWORD dwPassedFlags = 0; if ( pdwFlags ) { dwPassedFlags = *pdwFlags; *pdwFlags = 0; } // No logo cache? if (!_pLogoCache) return 0; ASSERT( pidl ); // HACK: this is used on browser only mode to tell what sort of logos we need... UINT rgfFlags = _fWide; LPEXTRACTIMAGE pImage = NULL; int iImage = -1; HRESULT hr = E_FAIL; // IID_IEXtractLogo and IID_IExtractImage are the same interface, by using a new guid // it means we can selectively decided what can logo in logo view... hr = FakeGetUIObjectOf( GetSF(), pidl, &rgfFlags, IID_IExtractLogo, (void **) &pImage ); if ( SUCCEEDED( hr )) { // extract .... HBITMAP hImage; WCHAR szPath[MAX_PATH]; DWORD dwFlags = IEIFLAG_ASYNC | IEIFLAG_ASPECT | dwPassedFlags; IMAGECACHEINFO rgInfo; UINT uIndex; BOOL fAsync; DWORD dwPriority; rgInfo.cbSize = sizeof( rgInfo ); hr = pImage->GetLocation( szPath, MAX_PATH, &dwPriority, &_rgLogoSize, _dwClrDepth, &dwFlags ); fAsync = ( hr == E_PENDING ); if ( SUCCEEDED( hr ) || fAsync ) { // mask off the flags passed to use by the flags returned from the extractor... if ( pdwFlags ) *pdwFlags = dwPassedFlags & dwFlags; rgInfo.dwMask = ICIFLAG_NAME; rgInfo.pszName = szPath; hr = _pLogoCache->FindImage( &rgInfo, &uIndex ); if ( hr == S_OK ) { ATOMICRELEASE( pImage ); return (int) uIndex; } if ( fAsync ) { LPRUNNABLETASK pTaskTmp = NULL; ASSERT( _pTaskScheduler ); // pass the icon index so we can find the right logo later... int iIcon = SHMapPIDLToSystemImageListIndex(GetSF(), pidl, NULL); hr = CExtractImageTask_Create( this, pImage, szPath, dwItem, iIcon, 0, &pTaskTmp ); if ( SUCCEEDED( hr )) { if ( !ppTask ) { hr = AddTaskToQueue( pTaskTmp, dwPriority, dwItem ); pTaskTmp->Release(); } else { * ppTask = pTaskTmp; ASSERT( pdwPriority ); *pdwPriority = dwPriority; } } else if ( ppTask ) { *ppTask = NULL; } // if all this failed, then we will just end up with a default // logo. This is only likely to fail in low memory conditions, // so that will be fine. // if this SUCCEEDED we will drop through to pick up a defualt piccy for now. } else { // otherwise extract synchronously....... hr = pImage->Extract( &hImage ); if ( SUCCEEDED( hr )) { rgInfo.dwMask = ICIFLAG_NAME | ICIFLAG_LARGE | ICIFLAG_BITMAP | ICIFLAG_NOUSAGE; rgInfo.hBitmapLarge = hImage; hr = _pLogoCache->AddImage( &rgInfo, &uIndex ); DeleteObject( hImage ); } if ( SUCCEEDED( hr )) { iImage = (int ) uIndex; } } } } ATOMICRELEASE( pImage ); return iImage; } int CLogoBase::GetLogoIndex( DWORD dwItem, LPCITEMIDLIST pidl, LPRUNNABLETASK *ppTask, DWORD * pdwPriority, DWORD *pdwFlags ) { int iImage = GetCachedLogoIndex(dwItem, pidl, ppTask, pdwPriority, pdwFlags ); if ( iImage == -1 ) { // always pass FALSE, we want the proper ICON, cdfview no longer hits the // wire for the icon so we can safely ask for the correct icon. iImage = GetDefaultLogo( pidl, FALSE); } return iImage; } HRESULT CLogoBase::AddTaskToQueue( LPRUNNABLETASK pTask, DWORD dwPriority, DWORD dwItem ) { ASSERT( _pTaskScheduler ); return _pTaskScheduler->AddTask( pTask, GetTOID(), dwItem, dwPriority ); } int CLogoBase::GetDefaultLogo( LPCITEMIDLIST pidl, BOOL fQuick ) { USES_CONVERSION; // Get icon to draw from int iIndex = -1; if ( !fQuick ) { iIndex = SHMapPIDLToSystemImageListIndex(GetSF(), pidl, NULL); } if (iIndex < 0) { iIndex = II_DOCNOASSOC; } WCHAR wszText[MAX_PATH]; wszText[0] = 0; STRRET strret; HRESULT hr = GetSF()->GetDisplayNameOf( pidl, SHGDN_NORMAL, &strret ); if ( SUCCEEDED( hr )) { StrRetToBufW(&strret, pidl, wszText, ARRAYSIZE(wszText)); } UINT uCacheIndex = (UINT) -1; if (_pLogoCache) // We didn't have one in stress. { IMAGECACHEINFO rgInfo; rgInfo.cbSize = sizeof( rgInfo ); rgInfo.dwMask = ICIFLAG_NAME | ICIFLAG_INDEX; rgInfo.pszName = wszText; rgInfo.iIndex = iIndex; hr = _pLogoCache->FindImage( &rgInfo, &uCacheIndex ); if ( hr == S_OK ) { return uCacheIndex; } HBITMAP hDef; hr = CreateDefaultLogo( iIndex, _rgLogoSize.cx, _rgLogoSize.cy, W2T(wszText), &hDef ); if ( SUCCEEDED( hr )) { rgInfo.hBitmapLarge = hDef; rgInfo.hMaskLarge = NULL; rgInfo.dwMask = ICIFLAG_NAME | ICIFLAG_INDEX | ICIFLAG_BITMAP | ICIFLAG_LARGE; hr = _pLogoCache->AddImage( &rgInfo, &uCacheIndex ); if ( FAILED(hr )) { uCacheIndex = (UINT) -1; } else { // remember the index of the logo AddIndicesToLogoList( iIndex, uCacheIndex ); } DeleteObject( hDef ); } } return (int) uCacheIndex; } #define DXFUDGE 4 #define COLORTEXT RGB(255,255,255) #define COLORBK RGB(0,0,0) HRESULT CLogoBase::CreateDefaultLogo(int iIcon, int cxLogo, int cyLogo, LPCTSTR pszText, HBITMAP * phBmpLogo) { HRESULT hr = E_OUTOFMEMORY; HBITMAP hbmp = NULL; HIMAGELIST himl; int cxIcon, cyIcon; int x, y, dx, dy; // get the small icons.... Shell_GetImageLists(NULL, &himl); ImageList_GetIconSize(himl, &cxIcon, &cyIcon); // Calculate position info. We assume logos are wider than they are tall. // ASSERT(cxLogo >= cyLogo); // Put the icon on the left x = 2; // Center the icon vertically if (cyIcon <= cyLogo) { y = (cyLogo - cyIcon) / 2; dy = cyIcon; dx = cxIcon; } else { y = 0; dy = cyLogo; // keep shrinkage proportional dx = MulDiv(cxIcon, cyIcon, cyLogo); } // get ready to draw HDC hTBDC = GetDC( GetHWND()); if ( !hTBDC ) { return E_FAIL; } HDC hdc = CreateCompatibleDC( hTBDC ); if (hdc) { RECT rc; int dx, dy, x, y; SIZE size; hbmp = CreateCompatibleBitmap(hTBDC, cxLogo, cyLogo); if (hbmp) { HGDIOBJ hTmp = SelectObject(hdc, hbmp); HPALETTE hpalOld; HFONT hfont, hfontOld; if ( _hpalHalftone ) { hpalOld = SelectPalette( hdc, _hpalHalftone, TRUE ); // LINTASSERT(hpalOld || !hpalOld); // 0 semi-ok for SelectPalette RealizePalette( hdc ); } SetMapMode( hdc, MM_TEXT ); rc.left = rc.top = 0; rc.bottom = cyLogo; rc.right = cxLogo; SHFillRectClr(hdc, &rc, COLORBK); // draw the icon into the memory DC. ImageList_GetIconSize(himl, &dx, &dy); x = DXFUDGE; y = ((cyLogo- dy) >> 1); ImageList_Draw( himl, iIcon, hdc, x, y, ILD_TRANSPARENT ); hfont = (HFONT)GetStockObject(DEFAULT_GUI_FONT); hfontOld = hfont ? (HFONT)SelectObject(hdc, hfont) : NULL; GetTextExtentPoint32(hdc, pszText, lstrlen(pszText), &size); x += (dx + DXFUDGE); y = ((cyLogo- size.cy) >> 1); rc.left = x; UINT eto = ETO_CLIPPED; SetTextColor(hdc, COLORTEXT); SetBkMode(hdc, TRANSPARENT); ExtTextOut(hdc, x, y, eto, &rc , pszText, lstrlen(pszText), NULL); if (hfontOld) SelectObject(hdc, hfontOld); if (hfont) DeleteObject(hfont); if ( _hpalHalftone ) { (void) SelectPalette( hdc, hpalOld, TRUE ); RealizePalette( hdc ); } // remove the final bitmap SelectObject( hdc, hTmp ); hr = S_OK; if (FAILED(hr)) { DeleteObject(hbmp); hbmp = NULL; } } DeleteDC(hdc); } ReleaseDC( GetHWND(), hTBDC ); *phBmpLogo = hbmp; return hr; } HRESULT CLogoBase::FlushLogoCache( ) { HRESULT hr = E_UNEXPECTED; if (_pLogoCache) { // forcibly clear out the logo cache so the items get refetched ... _pLogoCache->Flush(TRUE); hr = S_OK; } return hr; } HRESULT CLogoBase::DitherBitmap( HBITMAP hBmp, HBITMAP * phBmpNew ) { // if ( !phBmpNew ) // { // return E_INVALIDARG; // } // // if ( _dwClrDepth > 8) // { // *phBmpNew = hBmp; // return S_FALSE; // } // // IIntDitherer * pDither; // HRESULT hr = CoCreateInstance( CLSID_IntDitherer, // NULL, // CLSCTX_INPROC_SERVER, // IID_IIntDitherer, // (void **) & pDither ); // if ( FAILED( hr )) // { // return hr; // } // // static BYTE rgb[32768]; // static BOOL fInit = FALSE; // // if ( !fInit ) // { // // init the inverse color map table // SHGetInverseCMAP( rgb, sizeof( rgb )); // fInit = TRUE; // } // // HDC hMemDc = CreateCompatibleDC( NULL ); // if ( !hMemDc ) // { // pDither->Release(); // return E_FAIL; // } // // HBITMAP hOld = SelectObject( hdc, hBmp ); // // BITMAPINFO bi; // // ZeroMemory( &bi, sizeof( bi )); // bi.bmiHeader.biSize = sizeof( BITMAPINFOHEADER ); // bi.bmiHeader.biBitCount = 0; // bi.bmiHeader.biCompression = 0; // // // get the header information.... // iRet = GetDIBits( hMemDc, hBmp, 0, 0, NULL, &bi, DIB_RGB_COLORS ); // if ( iRet != 0 ) // { // LPVOID pBuffer, pBits; // int iOffset = 0; // // if ( bi.bmiHeader.biCompression == BI_BITFIELDS ) // { // iOffset = sizeof( DWORD ) * 3; // } // else if ( bi.bmiHeader.biBitCount <= 8 ) // { // if ( bi.bmiHeader.biClrUsed ) // { // iOffset = sizeof( RGBQUAD ) * bi.bmiHeader.biClrUsed; // } // else // { // iOffset = (1 << bi.bmiHeader.biBitCount) * sizeof( RGBQUAD ); // } // } // // bi.bmiHeader.biHeight = iHeight; // // // calc // pBuffer = LocalAlloc( LPTR, sizeof( BITMAPINFOHEADER ) + // bi.bmiHeader.biSizeImage + // iOffset ); // // // calc the size of the colour table so we put the data afterwards... // pBits = (( LPBYTE )pBuffer ) + sizeof( BITMAPINFOHEADER ) + iOffset; // // CopyMemory( pBuffer, &bi, sizeof( BITMAPINFOHEADER ) ); // iRet = GetDIBits( hMemDc, hBmp, 0, iHeight, pBits, // ( LPBITMAPINFO )pBuffer, DIB_RGB_COLORS ); // // // // we know we are going to 256 colour bitmap, so create a DIBSECTION as the destination ... // pDither->DitherTo8bpp( BYTE * pDestBits, LONG nDestPitch, // BYTE * pSrcBits, LONG nSrcPitch, REFGUID bfidSrc, // RGBQUAD * prgbDestColors, RGBQUAD * prgbSrcColors, // rgb, // LONG x, LONG y, LONG cx, LONG cy, // -1, -1); // } // pDither->Release(); ASSERT( FALSE ); return E_NOTIMPL; } int CLogoBase::AddIndicesToLogoList( int iIcon, UINT uIndex ) { int iRet = -1; LogoIndex * pIndex; LogoIndex rgNew; rgNew.iIcon = iIcon; rgNew.iLogo = (int) uIndex; if ( _fWide ) { EnterCriticalSection( &s_csSharedLogos ); } // scan to see if we have an extact match already in there... for ( int n = 0; n < DSA_GetItemCount( _hdsaLogoIndices ); n ++ ) { pIndex = (LogoIndex *) DSA_GetItemPtr( _hdsaLogoIndices, n ); ASSERT( pIndex ); if ( pIndex->iLogo == (int) uIndex ) { // set the icon just incase it changed... pIndex->iIcon = iIcon; iRet = n; break; } } if ( iRet == -1 ) { iRet = DSA_AppendItem( _hdsaLogoIndices, &rgNew ); } if ( _fWide ) { LeaveCriticalSection( &s_csSharedLogos ); } return iRet; } int CLogoBase::FindLogoFromIcon( int iIcon, int * piLastLogo ) { int iRet = -1; if ( !piLastLogo ) { return -1; } LogoIndex * pIndex; if ( _fWide ) { EnterCriticalSection( &s_csSharedLogos ); } for ( int n = *piLastLogo + 1; n < DSA_GetItemCount( _hdsaLogoIndices ); n ++ ) { pIndex = (LogoIndex *) DSA_GetItemPtr( _hdsaLogoIndices, n ); ASSERT( pIndex ); if ( pIndex->iIcon == iIcon ) { *piLastLogo = n; iRet = pIndex->iLogo; break; } } if ( _fWide ) { LeaveCriticalSection( &s_csSharedLogos ); } return iRet; } HRESULT CISFBand_CreateEx(IShellFolder* psf, LPCITEMIDLIST pidl, REFIID riid, void **ppv) { *ppv = NULL; HRESULT hr = E_FAIL; if (psf || pidl) { IShellFolderBand *psfb; hr = CoCreateInstance(CLSID_ISFBand, NULL, CLSCTX_INPROC_SERVER, IID_PPV_ARG(IShellFolderBand, &psfb)); if (SUCCEEDED(hr)) { hr = psfb->InitializeSFB(psf, pidl); if (SUCCEEDED(hr)) { hr = psfb->QueryInterface(riid, ppv); } psfb->Release(); } } return hr; }