Category

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

Windows Kernel

Development Platform:

Visual C++

gfldset.cpp：Code Content
  1. // gfldset.cpp
  2. //
  3. //  A class that manages global folder settings.
  5. #include "priv.h"
  6. #include "sccls.h"
  8. class CGlobalFolderSettings : public IGlobalFolderSettings
  9. {
  10.     public:
  11.     // *** IUnknown methods ***
  12.     STDMETHOD(QueryInterface)(THIS_ REFIID riid, LPVOID * ppvObj);
  13.     STDMETHOD_(ULONG,AddRef)(THIS);
  14.     STDMETHOD_(ULONG,Release)(THIS);
  16.     // *** IGlobalFolderSettings meTHODs ***
  17.     STDMETHOD(Get)(THIS_ DEFFOLDERSETTINGS *pdfs, int cbDfs);
  18.     STDMETHOD(Set)(THIS_ const DEFFOLDERSETTINGS *pdfs, int cbDfs, UINT flags);
  20.     protected:
  21.         friend HRESULT CGlobalFolderSettings_CreateInstance(IUnknown* pUnkOuter, IUnknown** ppunk, LPCOBJECTINFO poi);
  23.         CGlobalFolderSettings();
  24.         ~CGlobalFolderSettings();
  26.         LONG            m_cRef;
  27. };
  29. STDAPI CGlobalFolderSettings_CreateInstance(IUnknown* pUnkOuter, IUnknown** ppunk, LPCOBJECTINFO poi)
  30. {
  31.     // class factory will take care of denying aggregation and giving a null
  32.     // back on failure
  33.     ASSERT(pUnkOuter == NULL);
  35.     CGlobalFolderSettings* pid = new CGlobalFolderSettings();
  37.     if (pid)
  38.     {
  39.         *ppunk = SAFECAST(pid, IGlobalFolderSettings*);
  40.         return S_OK;
  41.     }
  42.     else
  43.     {
  44.         return E_OUTOFMEMORY;
  45.     }
  46. }
  48. /////////////////////////////////////////////////////////////////////////////////////////////////////
  49. CGlobalFolderSettings::CGlobalFolderSettings() : m_cRef(1)
  50. {
  51. }
  53. /////////////////////////////////////////////////////////////////////////////////////////////////////
  54. CGlobalFolderSettings::~CGlobalFolderSettings()
  55. {
  56. }
  58. /////////////////////////////////////////////////////////////////////////////////////////////////////
  59. STDMETHODIMP CGlobalFolderSettings::QueryInterface ( REFIID riid, LPVOID * ppvObj )
  60. {
  61.     if ( riid == IID_IUnknown || riid == IID_IGlobalFolderSettings)
  62.     {
  63.         *ppvObj = SAFECAST( this, IGlobalFolderSettings *);
  64.         AddRef();
  65.     }
  66.     else
  67.     {
  68.         return E_NOINTERFACE;
  69.     }
  71.     return NOERROR;
  72. }
  74. /////////////////////////////////////////////////////////////////////////////////////////////////////
  75. STDMETHODIMP_( ULONG ) CGlobalFolderSettings:: AddRef ()
  76. {
  77.     InterlockedIncrement( &m_cRef );
  78.     return m_cRef;
  79. }
  81. /////////////////////////////////////////////////////////////////////////////////////////////////////
  82. STDMETHODIMP_( ULONG ) CGlobalFolderSettings:: Release ()
  83. {
  84.     if ( InterlockedDecrement( &m_cRef ) == 0 )
  85.     {
  86.         delete this;
  87.         return 0;
  88.     }
  90.     return m_cRef;
  91. }
  93. /////////////////////////////////////////////////////////////////////////////////////////////////////
  94. STDMETHODIMP CGlobalFolderSettings::Get(DEFFOLDERSETTINGS *pdfs, int cbDfs)
  95. {
  96.     if (cbDfs < sizeof(DEFFOLDERSETTINGS))
  97.     {
  98.         return E_INVALIDARG;
  99.     }
  100.     else if (cbDfs > sizeof(DEFFOLDERSETTINGS))
  101.     {
  102.         ZeroMemory(pdfs, cbDfs);
  103.     }
  105.     *pdfs = g_dfs;
  107.     return S_OK;
  108. }
  110. /////////////////////////////////////////////////////////////////////////////////////////////////////
  111. STDMETHODIMP CGlobalFolderSettings::Set(const DEFFOLDERSETTINGS *pdfs, int cbDfs, UINT flags)
  112. {
  113.     if (flags & ~GFSS_VALID) {
  114.         ASSERT(!"Invalid flags passed to CGlobalFolderSettings::Set");
  115.         return E_INVALIDARG;
  116.     }
  118.     //
  119.     //  Special hack:  If you pass (NULL, 0) then it means "reset to
  120.     //  default".
  121.     //
  122.     if (pdfs == NULL && cbDfs == 0)
  123.     {
  124.         static DEFFOLDERSETTINGS dfs = INIT_DEFFOLDERSETTINGS;
  125.         dfs.vid = g_bRunOnNT5 ? VID_LargeIcons : DFS_VID_Default;
  126.         pdfs = &dfs;
  127.     }
  128.     else if (cbDfs < sizeof(DEFFOLDERSETTINGS))
  129.     {
  130.         ASSERT(!"Invalid cbDfs passed to CGlobalFolderSettings::Set");
  131.         return E_INVALIDARG;
  132.     }
  134.     // Preserve the dwDefRevCount, otherwise we'll never be able
  135.     // to tell if the structure has been revised!
  136.     DWORD dwDefRevCount = g_dfs.dwDefRevCount;
  137.     g_dfs = *pdfs;
  138.     g_dfs.dwDefRevCount = dwDefRevCount;
  139.     SaveDefaultFolderSettings(flags);
  140.     return S_OK;
  141. }