test/UnitTests/AutoTempDir.cpp

   1 // TortoiseGit - a Windows shell extension for easy version control
   2
   3 // Copyright (C) 2015-2016 - TortoiseGit
   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
  20 #include "stdafx.h"
  21 #include "AutoTempDir.h"
  22 #include "TGitPath.h"
  23
  24 CAutoTempDir::CAutoTempDir()
  25 {
  26         CString temppath;
  27         GetTempPath(MAX_PATH, temppath.GetBufferSetLength(MAX_PATH));
  28         TCHAR szTempName[MAX_PATH] = { 0 };
  29         temppath.ReleaseBuffer();
  30         GetTempFileName(temppath, L"tgit-tests", 0, szTempName);
  31         DeleteFile(szTempName);
  32         CreateDirectory(szTempName, nullptr);
  33         tempdir = szTempName;
  34 }
  35
  36 void CAutoTempDir::DeleteDirectoryRecursive(const CString& dir)
  37 {
  38         WIN32_FIND_DATA ffd;
  39         HANDLE hp = FindFirstFile(dir + L"\\*", &ffd);
  40         do
  41         {
  42                 if (!wcscmp(ffd.cFileName, L".") || !wcscmp(ffd.cFileName, L".."))
  43                         continue;
  44                 if ((ffd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) == FILE_ATTRIBUTE_DIRECTORY)
  45                 {
  46                         CString subdir = dir + L'\\' + ffd.cFileName;
  47                         DeleteDirectoryRecursive(subdir);
  48                 }
  49                 else
  50                 {
  51                         CString file = dir + L'\\' + ffd.cFileName;
  52                         bool failed = !DeleteFile(file);
  53                         if (failed && GetLastError() == ERROR_ACCESS_DENIED)
  54                         {
  55                                 SetFileAttributes(file, GetFileAttributes(file) & ~FILE_ATTRIBUTE_READONLY);
  56                                 failed = !DeleteFile(file);
  57                         }
  58                 }
  59         } while(FindNextFile(hp, &ffd));
  60         FindClose(hp);
  61
  62         RemoveDirectory(dir);
  63 }
  64
  65 CAutoTempDir::~CAutoTempDir()
  66 {
  67         if (!tempdir.IsEmpty())
  68                 DeleteDirectoryRecursive(tempdir);
  69 }
  70
  71 CString CAutoTempDir::GetTempDir() const
  72 {
  73         return tempdir;
  74 }