src/Utils/DropFiles.cpp

   1 // TortoiseGit - a Windows shell extension for easy version control
   2
   3 // Copyright (C) 2006, 2008 - 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 "DropFiles.h"
  21
  22 CDropFiles::CDropFiles()
  23 {
  24         m_pBuffer = NULL;
  25         m_nBufferSize = 0;
  26 }
  27
  28 CDropFiles::~CDropFiles()
  29 {
  30         delete [] m_pBuffer;
  31 }
  32
  33 void CDropFiles::AddFile(const CString &sFile)
  34 {
  35         m_arFiles.Add(sFile);
  36 }
  37
  38 INT_PTR CDropFiles::GetCount()
  39 {
  40         return m_arFiles.GetCount();
  41 }
  42
  43 void CDropFiles::CreateBuffer()
  44 {
  45         ASSERT(m_pBuffer == NULL);
  46         ASSERT(m_nBufferSize == 0);
  47         ASSERT(m_arFiles.GetCount()>0);
  48
  49         int nLength = 0;
  50
  51         for(int i=0;i<m_arFiles.GetSize();i++)
  52         {
  53                 nLength += m_arFiles[i].GetLength();
  54                 nLength += 1; // '\0' separator
  55         }
  56
  57         m_nBufferSize = sizeof(DROPFILES) + (nLength+1)*sizeof(TCHAR);
  58         m_pBuffer = new char[m_nBufferSize];
  59
  60         SecureZeroMemory(m_pBuffer, m_nBufferSize);
  61
  62         DROPFILES* df = (DROPFILES*)m_pBuffer;
  63         df->pFiles = sizeof(DROPFILES);
  64         df->fWide = 1;
  65
  66         TCHAR* pFilenames = (TCHAR*)(m_pBuffer + sizeof(DROPFILES));
  67         TCHAR* pCurrentFilename = pFilenames;
  68
  69         for(int i=0;i<m_arFiles.GetSize();i++)
  70         {
  71                 CString str = m_arFiles[i];
  72                 wcscpy_s(pCurrentFilename,str.GetLength()+1,str.GetBuffer());
  73                 pCurrentFilename += str.GetLength();
  74                 *pCurrentFilename = '\0'; // separator between file names
  75                 pCurrentFilename++;
  76         }
  77         *pCurrentFilename = '\0'; // terminate array
  78 }
  79
  80 void* CDropFiles::GetBuffer() const
  81 {
  82         return (void*)m_pBuffer;
  83 }
  84
  85 int     CDropFiles::GetBufferSize() const
  86 {
  87         return m_nBufferSize;
  88 }
  89
  90 void CDropFiles::CreateStructure()
  91 {
  92         CreateBuffer();
  93
  94         COleDataSource dropData;
  95         HGLOBAL hMem = ::GlobalAlloc(GMEM_ZEROINIT|GMEM_MOVEABLE|GMEM_DDESHARE, GetBufferSize());
  96         memcpy( ::GlobalLock(hMem), GetBuffer(), GetBufferSize() );
  97         ::GlobalUnlock(hMem);
  98         dropData.CacheGlobalData( CF_HDROP, hMem );
  99         dropData.DoDragDrop(DROPEFFECT_COPY|DROPEFFECT_MOVE|DROPEFFECT_LINK,NULL);
 100 }