Category

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

Windows Kernel

Development Platform:

Visual C++

objwsite.cpp：Code Content
  1. #include "priv.h"
  3. #include "objwsite.h"
  5. //
  6. // Default IObjectWithSite
  7. //
  10. CObjWithSite::CObjWithSite()
  11. {
  12.     _punkSite = NULL;
  13. }
  15. CObjWithSite::~CObjWithSite()
  16. {
  17.     if (_punkSite) 
  18.         _punkSite->Release();
  19. }
  21. HRESULT CObjWithSite::SetSite(IUnknown *punkSite)
  22. {
  23.     ATOMICRELEASE(_punkSite);
  25.     ASSERT(_punkSite == NULL);  // don't lose a reference to this
  27.     _punkSite = punkSite;
  29.     if (_punkSite)
  30.         _punkSite->AddRef();
  32.     return S_OK;
  33. }
  35. HRESULT CObjWithSite::GetSite(REFIID riid, void **ppvSite)
  36. {
  37.     if (_punkSite) 
  38.         return _punkSite->QueryInterface(riid, ppvSite);
  40.     *ppvSite = NULL;
  41.     return E_FAIL;
  42. }