src/TortoiseShell/ShellExtClassFactory.cpp

   1 // TortoiseGit - a Windows shell extension for easy version control
   2
   3 // Copyright (C) 2003-2006, 2009-2011 - TortoiseSVN
   4
   5 // This program is free software; you can redistribute it and/or
   6 // modify it under the terms of the GNU General Public License
   7 // as published by the Free Software Foundation; either version 2
   8 // of the License, or (at your option) any later version.
   9
  10 // This program is distributed in the hope that it will be useful,
  11 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 // GNU General Public License for more details.
  14
  15 // You should have received a copy of the GNU General Public License
  16 // along with this program; if not, write to the Free Software Foundation,
  17 // 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  18 //
  19 #include "stdafx.h"
  20 #include "ShellExt.h"
  21 #include "ShellExtClassFactory.h"
  22
  23
  24 CShellExtClassFactory::CShellExtClassFactory(FileState state)
  25         : m_StateToMake(state)
  26         , m_cRef(0L)
  27 {
  28         InterlockedIncrement(&g_cRefThisDll);
  29 }
  30
  31 CShellExtClassFactory::~CShellExtClassFactory()
  32 {
  33         InterlockedDecrement(&g_cRefThisDll);
  34 }
  35
  36 STDMETHODIMP CShellExtClassFactory::QueryInterface(REFIID riid,
  37                                                                                                    LPVOID FAR *ppv)
  38 {
  39         if (!ppv)
  40                 return E_POINTER;
  41
  42         *ppv = nullptr;
  43
  44         // Any interface on this object is the object pointer
  45
  46         if (IsEqualIID(riid, IID_IUnknown) || IsEqualIID(riid, IID_IClassFactory))
  47         {
  48                 *ppv = static_cast<LPCLASSFACTORY>(this);
  49                 AddRef();
  50                 return S_OK;
  51         }
  52
  53         return E_NOINTERFACE;
  54 }
  55
  56 STDMETHODIMP_(ULONG) CShellExtClassFactory::AddRef()
  57 {
  58         return InterlockedIncrement(&m_cRef);
  59 }
  60
  61 STDMETHODIMP_(ULONG) CShellExtClassFactory::Release()
  62 {
  63         if (InterlockedDecrement(&m_cRef))
  64                 return m_cRef;
  65
  66         delete this;
  67
  68         return 0L;
  69 }
  70
  71 STDMETHODIMP CShellExtClassFactory::CreateInstance(LPUNKNOWN pUnkOuter,
  72                                                                                                    REFIID riid,
  73                                                                                                    LPVOID *ppvObj)
  74 {
  75         if (!ppvObj)
  76                 return E_POINTER;
  77
  78         *ppvObj = nullptr;
  79
  80         // Shell extensions typically don't support aggregation (inheritance)
  81
  82         if (pUnkOuter)
  83                 return CLASS_E_NOAGGREGATION;
  84
  85         // Create the main shell extension object.  The shell will then call
  86         // QueryInterface with IID_IShellExtInit--this is how shell extensions are
  87         // initialized.
  88
  89         CShellExt* pShellExt = new (std::nothrow) CShellExt(m_StateToMake);  //Create the CShellExt object
  90
  91         if (!pShellExt)
  92                 return E_OUTOFMEMORY;
  93
  94         const HRESULT hr = pShellExt->QueryInterface(riid, ppvObj);
  95         if(FAILED(hr))
  96                 delete pShellExt;
  97         return hr;
  98 }
  99
 100 STDMETHODIMP CShellExtClassFactory::LockServer(BOOL /*fLock*/)
 101 {
 102         return E_NOTIMPL;
 103 }