src/TortoiseProc/BstrSafeVector.h

   1 // TortoiseGit - a Windows shell extension for easy version control
   2
   3 // Copyright (C) 2010 - 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 #pragma once
  20
  21 class CBstrSafeVector {
  22 public:
  23     CBstrSafeVector() : controlled( 0 ) {}
  24     CBstrSafeVector( ULONG count );
  25     ~CBstrSafeVector() { release(); }
  26
  27     const SAFEARRAY* operator ->() { return controlled; }
  28     operator SAFEARRAY*() { return controlled; }
  29     operator const SAFEARRAY*() const { return controlled; }
  30
  31     SAFEARRAY** operator&();
  32
  33     HRESULT PutElement( LONG index, const CString& value );
  34
  35 private:
  36     SAFEARRAY* controlled;
  37
  38     void release();
  39 };
  40
  41 inline CBstrSafeVector::CBstrSafeVector( ULONG count )
  42 {
  43     controlled = SafeArrayCreateVector(VT_BSTR, 0, count);
  44 }
  45
  46 inline SAFEARRAY** CBstrSafeVector::operator&()
  47 {
  48     release();
  49     return &controlled;
  50 }
  51
  52 inline HRESULT CBstrSafeVector::PutElement( LONG index, const CString& value )
  53 {
  54     ATL::CComBSTR valueBstr;
  55     valueBstr.Attach( value.AllocSysString() );
  56     return SafeArrayPutElement(controlled, &index, valueBstr);
  57 }
  58
  59 inline void CBstrSafeVector::release()
  60 {
  61     if(controlled == 0)
  62         return;
  63     SafeArrayDestroy(controlled);
  64     controlled = 0;
  65 }