src/TortoiseShell/PreserveChdir.cpp

   1 // TortoiseGit - a Windows shell extension for easy version control
   2
   3 // Copyright (C) 2003-2007 - 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 "PreserveChdir.h"
  21
  22 PreserveChdir::PreserveChdir()
  23 {
  24         DWORD len = GetCurrentDirectory(0, NULL);
  25         if (len)
  26         {
  27                 originalCurrentDirectory = new TCHAR[len];
  28                 if (GetCurrentDirectory(len, originalCurrentDirectory)==0)
  29                 {
  30                         delete [] originalCurrentDirectory;
  31                         originalCurrentDirectory = NULL;
  32                 }
  33         }
  34         else
  35                 originalCurrentDirectory = NULL;
  36 }
  37
  38 PreserveChdir::~PreserveChdir()
  39 {
  40         if (originalCurrentDirectory)
  41         {
  42                 DWORD len = GetCurrentDirectory(0, NULL);
  43                 TCHAR * currentDirectory = new TCHAR[len];
  44
  45                 // _tchdir is an expensive function - don't call it unless we really have to
  46                 GetCurrentDirectory(len, currentDirectory);
  47                 if(_tcscmp(currentDirectory, originalCurrentDirectory) != 0)
  48                 {
  49                         SetCurrentDirectory(originalCurrentDirectory);
  50                 }
  51                 delete [] currentDirectory;
  52                 delete [] originalCurrentDirectory;
  53                 originalCurrentDirectory = NULL;
  54         }
  55 }