Category

Development Platform
Home > SourceCode/Document > Windows Develop > Windows Kernel > View Code
isf.cpp
Package: shell.rar　[view]
Upload User: xhy777
Upload Date: 2007-02-14
Package Size: 24088k
Code Size: 15k
Category:

Windows Kernel

Development Platform:

Visual C++

isf.cpp：Code Content
  1. /*****************************************************************************
  2.     FILE:   isf.cpp
  4.     DESCRIPTION:
  5.         This is a base class that implements the default behavior of IShellFolder.
  6. *****************************************************************************/
  8. #include "priv.h"
  9. #include "isf.h"
  10. #include <shlobj.h>
  14. /*****************************************************************************
  15.     FUNCTION: IShellFolder::ParseDisplayName
  17.     DESCRIPTION:
  18. *****************************************************************************/
  19. HRESULT CBaseFolder::ParseDisplayName(HWND hwnd, LPBC pbcReserved, LPOLESTR pwszDisplayName,
  20.                         ULONG * pchEaten, LPITEMIDLIST * ppidl, ULONG *pdwAttributes)
  21. {
  22.     if (pdwAttributes)
  23.         *pdwAttributes = 0;
  25.     if (ppidl)
  26.         *ppidl = NULL;
  28.     return E_NOTIMPL;
  29. }
  31. /*****************************************************************************
  32.     FUNCTION: IShellFolder::EnumObjects
  34.     DESCRIPTION:
  35. *****************************************************************************/
  36. HRESULT CBaseFolder::EnumObjects(HWND hwndOwner, DWORD grfFlags, IEnumIDList ** ppenumIDList)
  37. {
  38.     if (ppenumIDList)
  39.         *ppenumIDList = NULL;
  41.     return E_NOTIMPL;
  42. }
  45. /*****************************************************************************
  46.     FUNCTION: IShellFolder::BindToObject
  48.     DESCRIPTION:
  49. *****************************************************************************/
  50. HRESULT CBaseFolder::BindToObject(LPCITEMIDLIST pidl, LPBC pbc, REFIID riid, LPVOID * ppvObj)
  51. {
  52.     if (ppvObj)
  53.         *ppvObj = NULL;
  55.     return E_NOTIMPL;
  56. }
  59. /*****************************************************************************
  60.      FUNCTION: IShellFolder::BindToStorage
  61.  
  62.     DESCRIPTION:
  63.          This should be implemented so people can use the File.Open and File.SaveAs
  64.     dialogs with this ShellFolder.
  65. *****************************************************************************/
  66. HRESULT CBaseFolder::BindToStorage(LPCITEMIDLIST pidl, LPBC pbc, REFIID riid, LPVOID * ppvObj)
  67. {
  68.     if (ppvObj)
  69.         *ppvObj = NULL;
  71.     return E_NOTIMPL;
  72. }
  75. /*****************************************************************************
  76.      FUNCTION: IShellFolder::CompareIDs
  77.  
  78.     DESCRIPTION:
  79.          This should be implemented so people can use the File.Open and File.SaveAs
  80.     dialogs with this ShellFolder.
  81. *****************************************************************************/
  82. HRESULT CBaseFolder::CompareIDs(LPARAM ici, LPCITEMIDLIST pidl1, LPCITEMIDLIST pidl2)
  83. {
  84.     return E_NOTIMPL;
  85. }
  88. /*****************************************************************************
  89.      FUNCTION: IShellFolder::CreateViewObject
  90.  
  91.     DESCRIPTION:
  92.          This should be implemented so people can use the File.Open and File.SaveAs
  93.     dialogs with this ShellFolder.
  94. *****************************************************************************/
  95. HRESULT CBaseFolder::CreateViewObject(HWND hwndOwner, REFIID riid, LPVOID * ppvObj)
  96. {
  97.     HRESULT hr = E_NOINTERFACE;
  99.     *ppvObj = NULL;
  100.     if (IsEqualIID(riid, IID_IShellView))
  101.         hr = _CreateShellView(hwndOwner, ppvObj);
  102.     else if (IsEqualIID(riid, IID_IContextMenu))
  103.         hr = _GetUIObjectOf(hwndOwner, 0, NULL, riid, 0, ppvObj, TRUE);
  104.     else
  105.         hr = E_NOINTERFACE;
  107.     return hr;
  108. }
  111. BOOL IsShellIntegration(void)
  112. {
  113.     BOOL fResult = FALSE;
  114.     HINSTANCE hInst = LoadLibrary(TEXT("shell32.dll"));
  116.     if (EVAL(hInst))
  117.     {
  118.         LPVOID pv = GetProcAddress(hInst, "DllGetVersion");
  119.         if (pv)
  120.             fResult = TRUE;
  121.         FreeLibrary(hInst);
  122.     }
  123.     
  124.     return fResult;
  125. }
  127. HRESULT CBaseFolder::_CreateShellView(HWND hwndOwner, void ** ppvObj, LONG lEvents, FOLDERVIEWMODE fvm, 
  128.                                        IShellFolderViewCB * psfvCallBack, LPCITEMIDLIST pidl, LPFNVIEWCALLBACK pfnCallback)
  129. {
  130.     HRESULT hr;
  131.     IShellFolder * psf;
  133.     hr = this->QueryInterface(IID_IShellFolder, (LPVOID *) &psf);
  134.     if (EVAL(SUCCEEDED(hr)))
  135.     {
  136.         SFV_CREATE sfvCreate =      // SHCreateShellFolderView struct
  137.         {
  138.                 sizeof(SFV_CREATE),
  139.                 psf,            // psf
  140.                 NULL,           // psvOuter
  141.                 psfvCallBack    // psfvcb - (IShellFolderViewCB *)
  142.         };
  144.         // SHCreateShellFolderView isn't in the original shell.  We can't rely on the 
  145.         // the Delayload code because it's exported by ordinal and the original
  146.         // shell had a different exports by the same number.
  147.         if (IsShellIntegration())
  148.             hr = _SHCreateShellFolderView(&sfvCreate, (LPSHELLVIEW FAR*)ppvObj);
  149.         else
  150.             hr = E_FAIL;  // Force us to go into the next try.
  152.         // If we aren't running on a machine with Shell Integration, SHCreateShellFolderView will fail.
  153.         if (FAILED(hr))
  154.         {
  155.             CSFV csfv;
  157.             csfv.cbSize = sizeof(csfv);
  158.             csfv.pshf = psf;
  159.             csfv.psvOuter = (IShellView *) psfvCallBack;      // Hack but it works...
  160.             csfv.pidl = pidl;           // This is feed to SFVM_GETNOTIFY so it needs to be a pidlTarget.
  161.             csfv.lEvents = lEvents;
  162.             csfv.pfnCallback = pfnCallback;
  163.             csfv.fvm = fvm;         // vs. FVM_ICON, ...
  165.             hr = SHCreateShellFolderViewEx(&csfv, (LPSHELLVIEW FAR*)ppvObj);
  166.             if (SUCCEEDED(hr))
  167.                 psfvCallBack->AddRef();     // We gave them a ref.
  168.         }
  170.         psf->Release();
  171.     }
  173.     return hr;
  174. }
  177. /*****************************************************************************
  178.      FUNCTION: IShellFolder::GetAttributesOf
  179.  
  180.     DESCRIPTION:
  181. *****************************************************************************/
  182. HRESULT CBaseFolder::GetAttributesOf(UINT cpidl, LPCITEMIDLIST *apidl, ULONG *rgfInOut)
  183. {
  184.     return E_NOTIMPL;
  185. }
  188. /*****************************************************************************
  189.      FUNCTION: IShellFolder::GetUIObjectOf
  190.  
  191.     DESCRIPTION:
  192. *****************************************************************************/
  193. HRESULT CBaseFolder::GetUIObjectOf(HWND hwndOwner, UINT cidl, LPCITEMIDLIST rgpidl[],
  194.                                 REFIID riid, UINT * prgfInOut, LPVOID * ppvObj)
  195. {
  196.     return E_NOTIMPL;
  197. }
  200. /*****************************************************************************
  201.     DESCRIPTION:
  202. *****************************************************************************/
  203. HRESULT CBaseFolder::_GetUIObjectOf(HWND hwndOwner, UINT cidl, LPCITEMIDLIST rgpidl[],
  204.                                 REFIID riid, UINT * prgfInOut, LPVOID * ppvOut, BOOL fFromCreateViewObject)
  205. {
  206.     return GetUIObjectOf(hwndOwner, cidl, rgpidl, riid, prgfInOut, ppvOut);
  207. }
  210. /*****************************************************************************
  211.      FUNCTION: IShellFolder::GetDisplayNameOf
  212.  
  213.     DESCRIPTION:
  214. *****************************************************************************/
  215. HRESULT CBaseFolder::GetDisplayNameOf(LPCITEMIDLIST pidl, DWORD shgno, LPSTRRET pStrRet)
  216. {
  217.     return E_NOTIMPL;
  218. }
  221. /*****************************************************************************
  222.      FUNCTION: IShellFolder::SetNameOf
  223.  
  224.     DESCRIPTION:
  225. *****************************************************************************/
  226. HRESULT CBaseFolder::SetNameOf(HWND hwndOwner, LPCITEMIDLIST pidl, LPCOLESTR pwszName,
  227.                                 DWORD dwReserved, LPITEMIDLIST *ppidlOut)
  228. {
  229.     return E_NOTIMPL;
  230. }
  233. //===========================
  234. // *** IShellFolder2 Interface ***
  235. //===========================
  237. //===========================
  238. // *** IPersist Interface ***
  239. //===========================
  241. /*****************************************************************************
  242.      FUNCTION: IPersist::GetClassID
  243.  
  244.     DESCRIPTION:
  245. *****************************************************************************/
  246. HRESULT CBaseFolder::GetClassID(LPCLSID pClassID)
  247. {
  248.     HRESULT hr = E_INVALIDARG;
  250.     if (EVAL(pClassID))
  251.     {
  252.         if (EVAL(m_pClassID))
  253.         {
  254.             *pClassID = *m_pClassID;
  255.             hr = S_OK;
  256.         }
  257.         else
  258.             hr = E_FAIL;
  259.     }
  261.     return hr;
  262. }
  264. //===========================
  265. // *** IPersistFolder Interface ***
  266. //===========================
  268. /*****************************************************************************
  269.     DESCRIPTION:
  270. *****************************************************************************/
  271. HRESULT CBaseFolder::Initialize(LPCITEMIDLIST pidl)
  272. {
  273.     ASSERT(!m_pidl);   // Don't reroot us.
  274.     return _Initialize(pidl, NULL, ILGetSize(pidl) - sizeof(pidl->mkid.cb));
  275. }
  277. //===========================
  278. // *** IPersistFolder2 Interface ***
  279. //===========================
  281. /***************************************************************************** 
  282.     DESCRIPTION:
  283. *****************************************************************************/
  284. HRESULT CBaseFolder::GetCurFolder(LPITEMIDLIST *ppidl)
  285. {
  286.     HRESULT hr = E_INVALIDARG;
  288.     if (EVAL(ppidl))
  289.     {
  290.         hr = E_FAIL;
  292.         if (m_pidlRoot)
  293.         {
  294.             *ppidl = ILClone(m_pidlRoot);
  295.         }
  296.         else if (EVAL(m_pidl))
  297.         {
  298.             *ppidl = GetPublicTargetPidlClone();
  299.         }
  301.         if (*ppidl)
  302.             hr = S_OK;
  303.     }
  305.     return hr;
  306. }
  308. //===========================
  309. // *** IPersistFolder3 Interface ***
  310. //===========================
  311. HRESULT GetPidlFromPersistFolderTargetInfo(const PERSIST_FOLDER_TARGET_INFO *ppfti, LPITEMIDLIST * ppidl, BOOL fFree)
  312. {
  313.     HRESULT hr = E_INVALIDARG;
  315.     if (ppidl)
  316.     {
  317.         *ppidl = NULL;
  318.         if (ppfti->pidlTargetFolder)
  319.         {
  320.             *ppidl = (fFree ? ppfti->pidlTargetFolder : ILClone(ppfti->pidlTargetFolder));
  321.             if (*ppidl)
  322.                 hr = S_OK;
  323.             else
  324.                 hr = E_OUTOFMEMORY;
  325.         }
  326.         else
  327.         {
  328.             if (ppfti->szTargetParsingName[0])
  329.             {
  330.                 hr = IEParseDisplayNameWithBCW(CP_ACP, ppfti->szTargetParsingName, NULL, ppidl);
  331.             }
  333.             if (!*ppidl && (-1 != ppfti->csidl))
  334.             {
  335.                 hr = SHGetSpecialFolderLocation(NULL, ppfti->csidl, ppidl);
  336.             }
  337.         }
  338.     }
  339.     
  340.     return hr;
  341. }
  344. /*****************************************************************************
  345.     DESCRIPTION:
  346. *****************************************************************************/
  347. HRESULT CBaseFolder::InitializeEx(IBindCtx *pbc, LPCITEMIDLIST pidlRoot, const PERSIST_FOLDER_TARGET_INFO *ppfti)
  348. {
  349.     HRESULT hr = E_INVALIDARG;
  351.     if (EVAL(pidlRoot))
  352.     {
  353.         if (ppfti)
  354.         {
  355.             // We are a Folder Shortcut.
  356.             LPITEMIDLIST pidlTarget;
  358.             hr = GetPidlFromPersistFolderTargetInfo(ppfti, &pidlTarget, FALSE);  // Get the real root.           
  359.             TraceMsg(TF_FOLDER_SHRTCUTS, "CBaseFolder::InitializeEx() this=%#08lx, pidlTarget=%#08lx, pidlRoot=%#08lx", this, pidlTarget, pidlRoot);
  361.             AssertMsg((NULL != pidlTarget), TEXT("CBaseFolder::InitializeEx() We are useless without a pidlTarget so watch me go limp."));
  362.             if (pidlTarget)
  363.             {
  364.                 hr = _Initialize(pidlTarget, pidlRoot, m_nIDOffsetToPrivate);
  365.                 ILFree(pidlTarget);
  366.             }
  367.         }
  368.         else
  369.         {
  370.             // We aren't a folder shortcut.
  371.             hr = Initialize(pidlRoot);
  372.         }
  373.     }
  375.     return hr;
  376. }
  379. HRESULT CBaseFolder::GetFolderTargetInfo(PERSIST_FOLDER_TARGET_INFO *ppfti)
  380. {
  381.     HRESULT hr = E_INVALIDARG;
  383.     AssertMsg((NULL != ppfti), TEXT("CBaseFolder::GetFolderTargetInfo() Caller passed an invalid param."));
  384.     if (ppfti)
  385.     {
  386.         ZeroMemory(ppfti, sizeof(*ppfti)); 
  388.         ppfti->pidlTargetFolder = ILClone(m_pidlRoot);
  389.         ppfti->dwAttributes = -1;
  390.         ppfti->csidl = -1;
  391.         hr = S_OK;
  392.     }
  394.     return hr;
  395. }
  398. LPCITEMIDLIST CBaseFolder::GetPrivatePidlReference(void)
  399. {
  400.     return _ILSkip(m_pidl, m_nIDOffsetToPrivate);
  401. }
  403. // This function always needs the InternetExplorer pidl.
  404. LPITEMIDLIST CBaseFolder::GetPublicPidlRootIDClone(void)
  405. {
  406.     LPITEMIDLIST pidlFull = ILClone(m_pidl);
  407.     LPITEMIDLIST pidlPrivStart = _ILSkip(pidlFull, m_nIDOffsetToPrivate);
  409.     // Strip all Private ItemIDs
  410.     while (!ILIsEmpty(pidlPrivStart))
  411.         ILRemoveLastID(pidlPrivStart);
  413.     return pidlFull;
  414. }
  416. LPITEMIDLIST CBaseFolder::CreateFullPrivatePidl(LPCITEMIDLIST pidlPrivateSubPidl)
  417. {
  418.     return ILCombine(GetPrivatePidlReference(), pidlPrivateSubPidl);
  419. }
  421. LPITEMIDLIST CBaseFolder::CreateFullPublicPidlFromRelative(LPCITEMIDLIST pidlPrivateSubPidl)
  422. {
  423.     return ILCombine(GetPublicRootPidlReference(), pidlPrivateSubPidl);
  424. }
  426. LPITEMIDLIST CBaseFolder::CreateFullPublicPidl(LPCITEMIDLIST pidlPrivatePidl)
  427. {
  428.     LPITEMIDLIST pidlRoot = GetPublicPidlRootIDClone();
  429.     LPITEMIDLIST pidlResult = NULL;
  430.     
  431.     if (pidlRoot)
  432.     {
  433.         pidlResult = ILCombine(pidlRoot, pidlPrivatePidl);
  434.         ILFree(pidlRoot);
  435.     }
  437.     return pidlResult;
  438. }
  440. HRESULT CBaseFolder::_Initialize(LPCITEMIDLIST pidlTarget, LPCITEMIDLIST pidlRoot, int nBytesToPrivate)
  441. {
  442.     HRESULT hr = E_INVALIDARG;
  444.     if (pidlTarget)
  445.     {
  446.         ILFree(m_pidl);
  447.         ILFree(m_pidlRoot);    
  448.         m_pidl = ILClone(pidlTarget);
  449.         m_pidlRoot = ILClone(pidlRoot); // This is the Folder Shortcut pidl.  We don't use it outselves.
  451.         if (m_pidl)
  452.         {
  453.             m_nIDOffsetToPrivate = nBytesToPrivate;
  454.             hr = S_OK;
  455.         }
  456.         else
  457.             hr = E_OUTOFMEMORY;
  458.     }
  460.     return hr;
  461. }
  464. /****************************************************
  465.     Constructor
  466. ****************************************************/
  467. CBaseFolder::CBaseFolder(LPCLSID pClassID) : m_cRef(1)
  468. {
  469.     DllAddRef();
  471.     // This needs to be allocated in Zero Inited Memory.
  472.     // Assert that all Member Variables are inited to Zero.
  473.     ASSERT(!m_pidl);
  474.     ASSERT(!m_nIDOffsetToPrivate);
  475.     ASSERT(!m_pClassID);
  477.     m_pClassID = pClassID;
  478.     ASSERT(pClassID);
  479. }
  482. /****************************************************
  483.     Destructor
  484. ****************************************************/
  485. CBaseFolder::~CBaseFolder()
  486. {
  487.     Pidl_Set(&m_pidlRoot, NULL);    // Folder Shortcut pidl
  488.     Pidl_Set(&m_pidl, NULL);
  489.     DllRelease();
  490. }
  493. //===========================
  494. // *** IUnknown Interface ***
  495. //===========================
  497. ULONG CBaseFolder::AddRef()
  498. {
  499.     m_cRef++;
  500.     return m_cRef;
  501. }
  503. ULONG CBaseFolder::Release()
  504. {
  505.     ASSERT(m_cRef > 0);
  506.     m_cRef--;
  508.     if (m_cRef > 0)
  509.         return m_cRef;
  511.     delete this;
  512.     return 0;
  513. }
  515. HRESULT CBaseFolder::QueryInterface(REFIID riid, void **ppvObj)
  516. {
  517.     static const QITAB qit[] = {
  518.         QITABENTMULTI(CBaseFolder, IShellFolder, IShellFolder2),
  519.         QITABENTMULTI(CBaseFolder, IPersist, IPersistFolder),
  520.         QITABENTMULTI(CBaseFolder, IPersist, IPersistFolder3),
  521.         QITABENTMULTI(CBaseFolder, IPersistFolder, IPersistFolder3),
  522.         QITABENTMULTI(CBaseFolder, IPersistFolder2, IPersistFolder3),
  523.         QITABENT(CBaseFolder, IShellFolder2),
  524.         QITABENT(CBaseFolder, IPersistFolder3),
  525.         QITABENT(CBaseFolder, IObjectWithSite),
  526.         { 0 },
  527.     };
  529.     return QISearch(this, qit, riid, ppvObj);
  530. }