src/TortoiseOverlays/ShellExt.cpp

   1 // TortoiseOverlays - an overlay handler for Tortoise clients
   2 // Copyright (C) 2007 - TortoiseSVN
   3 #include "stdafx.h"
   4
   5 #pragma warning (disable : 4786)
   6
   7 // Initialize GUIDs (should be done only and at-least once per DLL/EXE)
   8 #include <initguid.h>
   9 #include "Guids.h"
  10
  11 #include "ShellExt.h"
  12
  13
  14 // *********************** CShellExt *************************
  15 CShellExt::CShellExt(FileState state)
  16 {
  17     m_State = state;
  18
  19     m_cRef = 0L;
  20     g_cRefThisDll++;
  21
  22 }
  23
  24 CShellExt::~CShellExt()
  25 {
  26         g_cRefThisDll--;
  27 }
  28
  29
  30 STDMETHODIMP CShellExt::QueryInterface(REFIID riid, LPVOID FAR *ppv)
  31 {
  32     *ppv = NULL;
  33
  34     if (IsEqualIID(riid, IID_IShellExtInit) || IsEqualIID(riid, IID_IUnknown))
  35     {
  36         *ppv = (LPSHELLEXTINIT)this;
  37     }
  38     else if (IsEqualIID(riid, IID_IContextMenu))
  39     {
  40         *ppv = (LPCONTEXTMENU)this;
  41     }
  42     else if (IsEqualIID(riid, IID_IContextMenu2))
  43     {
  44         *ppv = (LPCONTEXTMENU2)this;
  45     }
  46     else if (IsEqualIID(riid, IID_IContextMenu3))
  47     {
  48         *ppv = (LPCONTEXTMENU3)this;
  49     }
  50     else if (IsEqualIID(riid, IID_IShellIconOverlayIdentifier))
  51     {
  52         *ppv = (IShellIconOverlayIdentifier*)this;
  53     }
  54     else if (IsEqualIID(riid, IID_IShellPropSheetExt))
  55     {
  56         *ppv = (LPSHELLPROPSHEETEXT)this;
  57     }
  58     if (*ppv)
  59     {
  60         AddRef();
  61
  62         return NOERROR;
  63     }
  64
  65     return E_NOINTERFACE;
  66 }
  67
  68 STDMETHODIMP_(ULONG) CShellExt::AddRef()
  69 {
  70     return ++m_cRef;
  71 }
  72
  73 STDMETHODIMP_(ULONG) CShellExt::Release()
  74 {
  75     if (--m_cRef)
  76         return m_cRef;
  77
  78         for (vector<DLLPointers>::iterator it = m_dllpointers.begin(); it != m_dllpointers.end(); ++it)
  79         {
  80                 if (it->pShellIconOverlayIdentifier != NULL)
  81                         it->pShellIconOverlayIdentifier->Release();
  82                 if (it->hDll != NULL)
  83                         FreeLibrary(it->hDll);
  84
  85                 it->hDll = NULL;
  86                 it->pDllGetClassObject = NULL;
  87                 it->pDllCanUnloadNow = NULL;
  88                 it->pShellIconOverlayIdentifier = NULL;
  89         }
  90
  91         m_dllpointers.clear();
  92
  93     delete this;
  94
  95     return 0L;
  96 }
  97
  98