src/Utils/TempFile.cpp

   1 // TortoiseGit - a Windows shell extension for easy version control
   2
   3 // Copyright (C) 2009, 2011-2013, 2015-2016, 2018-2019 - TortoiseGit
   4 // Copyright (C) 2003-2008, 2020 - TortoiseSVN
   5
   6 // This program is free software; you can redistribute it and/or
   7 // modify it under the terms of the GNU General Public License
   8 // as published by the Free Software Foundation; either version 2
   9 // of the License, or (at your option) any later version.
  10
  11 // This program is distributed in the hope that it will be useful,
  12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 // GNU General Public License for more details.
  15
  16 // You should have received a copy of the GNU General Public License
  17 // along with this program; if not, write to the Free Software Foundation,
  18 // 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  19 //
  20 #include "stdafx.h"
  21 #include "registry.h"
  22 #include "TempFile.h"
  23 #include "TGitPath.h"
  24 #include "SmartHandle.h"
  25 #include "Git.h"
  26
  27 CTempFiles::CTempFiles(void)
  28 {
  29 }
  30
  31 CTempFiles::~CTempFiles(void)
  32 {
  33         m_TempFileList.DeleteAllFiles(false);
  34 }
  35
  36 CTempFiles& CTempFiles::Instance()
  37 {
  38         static CTempFiles instance;
  39         return instance;
  40 }
  41
  42 CTGitPath CTempFiles::GetTempFilePath(bool bRemoveAtEnd, const CTGitPath& path /* = CTGitPath() */, const CGitHash& hash /* = CGitHash() */)
  43 {
  44         DWORD len = GetTortoiseGitTempPath(0, nullptr);
  45
  46         auto temppath = std::make_unique<TCHAR[]>(len + 1);
  47         auto tempF = std::make_unique<TCHAR[]>(len + 50);
  48         GetTortoiseGitTempPath(len + 1, temppath.get());
  49         CTGitPath tempfile;
  50         CString possibletempfile;
  51         if (path.IsEmpty())
  52         {
  53                 ::GetTempFileName (temppath.get(), L"git", 0, tempF.get());
  54                 tempfile = CTGitPath(tempF.get());
  55         }
  56         else
  57         {
  58                 int i=0;
  59                 do
  60                 {
  61                         // use the UI path, which does unescaping for urls
  62                         CString filename = path.GetBaseFilename();
  63                         // remove illegal chars which could be present in urls
  64                         filename.Remove('?');
  65                         filename.Remove('*');
  66                         filename.Remove('<');
  67                         filename.Remove('>');
  68                         filename.Remove('|');
  69                         filename.Remove('"');
  70                         // the inner loop assures that the resulting path is < MAX_PATH
  71                         // if that's not possible without reducing the 'filename' to less than 5 chars, use a path
  72                         // that's longer than MAX_PATH (in that case, we can't really do much to avoid longer paths)
  73                         do
  74                         {
  75                                 if (!hash.IsEmpty())
  76                                         possibletempfile.Format(L"%s%s-%s.%3.3x%s", temppath.get(), static_cast<LPCTSTR>(filename), static_cast<LPCTSTR>(hash.ToString(g_Git.GetShortHASHLength())), i, static_cast<LPCTSTR>(path.GetFileExtension()));
  77                                 else
  78                                         possibletempfile.Format(L"%s%s.%3.3x%s", temppath.get(), static_cast<LPCTSTR>(filename), i, static_cast<LPCTSTR>(path.GetFileExtension()));
  79                                 tempfile.SetFromWin(possibletempfile);
  80                                 filename.Truncate(std::max(0, filename.GetLength() - 1));
  81                         } while (filename.GetLength() > 4 && tempfile.GetWinPathString().GetLength() >= MAX_PATH);
  82                         ++i;
  83                         // now create the temp file in a thread safe way, so that subsequent calls to GetTempFile() return different filenames.
  84                         CAutoFile hFile = CreateFile(tempfile.GetWinPath(), GENERIC_READ, FILE_SHARE_READ, nullptr, CREATE_NEW, FILE_ATTRIBUTE_TEMPORARY, nullptr);
  85                         auto lastErr = GetLastError();
  86                         if (hFile || ((lastErr != ERROR_FILE_EXISTS) && (lastErr != ERROR_ACCESS_DENIED)))
  87                                 break;
  88                 } while (true);
  89         }
  90         if (bRemoveAtEnd)
  91                 m_TempFileList.AddPath(tempfile);
  92         return tempfile;
  93 }