Make sure we do not forget to use ReleaseBuffer and make use of CStrBuf
[TortoiseGit.git] / src / Git / Git.cpp
blob04770407efacd0cd8c9cff4a6f2b981cc9216fca
1 // TortoiseGit - a Windows shell extension for easy version control
3 // Copyright (C) 2008-2015 - TortoiseGit
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.
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.
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.
20 #include "stdafx.h"
21 #include "Git.h"
22 #include "registry.h"
23 #include "GitForWindows.h"
24 #include "UnicodeUtils.h"
25 #include "gitdll.h"
26 #include <fstream>
27 #include <iterator>
28 #include "FormatMessageWrapper.h"
29 #include "SmartHandle.h"
30 #include "MassiveGitTaskBase.h"
31 #include "git2/sys/filter.h"
32 #include "git2/sys/transport.h"
33 #include "../libgit2/filter-filter.h"
34 #include "../libgit2/ssh-wintunnel.h"
36 bool CGit::ms_bCygwinGit = (CRegDWORD(_T("Software\\TortoiseGit\\CygwinHack"), FALSE) == TRUE);
37 int CGit::m_LogEncode=CP_UTF8;
38 typedef CComCritSecLock<CComCriticalSection> CAutoLocker;
40 static LPCTSTR nextpath(const wchar_t* path, wchar_t* buf, size_t buflen)
42 if (path == NULL || buf == NULL || buflen == 0)
43 return NULL;
45 const wchar_t* base = path;
46 wchar_t term = (*path == L'"') ? *path++ : L';';
48 for (buflen--; *path && *path != term && buflen; buflen--)
49 *buf++ = *path++;
51 *buf = L'\0'; /* reserved a byte via initial subtract */
53 while (*path == term || *path == L';')
54 ++path;
56 return (path != base) ? path : NULL;
59 static inline BOOL FileExists(LPCTSTR lpszFileName)
61 struct _stat st;
62 return _tstat(lpszFileName, &st) == 0;
65 static CString FindFileOnPath(const CString& filename, LPCTSTR env, bool wantDirectory = false)
67 TCHAR buf[MAX_PATH] = { 0 };
69 // search in all paths defined in PATH
70 while ((env = nextpath(env, buf, MAX_PATH - 1)) != NULL && *buf)
72 TCHAR *pfin = buf + _tcslen(buf) - 1;
74 // ensure trailing slash
75 if (*pfin != _T('/') && *pfin != _T('\\'))
76 _tcscpy_s(++pfin, 2, _T("\\")); // we have enough space left, MAX_PATH-1 is used in nextpath above
78 const size_t len = _tcslen(buf);
80 if ((len + filename.GetLength()) < MAX_PATH)
81 _tcscpy_s(pfin + 1, MAX_PATH - len, filename);
82 else
83 break;
85 if (FileExists(buf))
87 if (wantDirectory)
88 pfin[1] = 0;
89 return buf;
93 return _T("");
96 static BOOL FindGitPath()
98 size_t size;
99 _tgetenv_s(&size, NULL, 0, _T("PATH"));
100 if (!size)
101 return FALSE;
103 TCHAR* env = (TCHAR*)alloca(size * sizeof(TCHAR));
104 if (!env)
105 return FALSE;
106 _tgetenv_s(&size, env, size, _T("PATH"));
108 CString gitExeDirectory = FindFileOnPath(_T("git.exe"), env, true);
109 if (!gitExeDirectory.IsEmpty())
111 CGit::ms_LastMsysGitDir = gitExeDirectory;
112 CGit::ms_LastMsysGitDir.TrimRight(_T("\\"));
113 if (CGit::ms_LastMsysGitDir.GetLength() > 12 && (CGit::ms_LastMsysGitDir.Right(12) == _T("\\mingw32\\bin") || CGit::ms_LastMsysGitDir.Right(12) == _T("\\mingw64\\bin")))
115 // prefer cmd directory as early Git for Windows 2.x releases only had this
116 CString installRoot = CGit::ms_LastMsysGitDir.Mid(0, CGit::ms_LastMsysGitDir.GetLength() - 12) + _T("\\cmd\\git.exe");
117 if (FileExists(installRoot))
118 CGit::ms_LastMsysGitDir = CGit::ms_LastMsysGitDir.Mid(0, CGit::ms_LastMsysGitDir.GetLength() - 12) + _T("\\cmd");
120 if (CGit::ms_LastMsysGitDir.GetLength() > 4 && CGit::ms_LastMsysGitDir.Right(4) == _T("\\cmd"))
122 // often the msysgit\cmd folder is on the %PATH%, but
123 // that git.exe does not work, so try to guess the bin folder
124 CString binDir = CGit::ms_LastMsysGitDir.Mid(0, CGit::ms_LastMsysGitDir.GetLength() - 4) + _T("\\bin\\git.exe");
125 if (FileExists(binDir))
126 CGit::ms_LastMsysGitDir = CGit::ms_LastMsysGitDir.Mid(0, CGit::ms_LastMsysGitDir.GetLength() - 4) + _T("\\bin");
128 return TRUE;
131 return FALSE;
134 static CString FindExecutableOnPath(const CString& executable, LPCTSTR env)
136 CString filename = executable;
138 if (executable.GetLength() < 4 || executable.Find(_T(".exe"), executable.GetLength() - 4) != executable.GetLength() - 4)
139 filename += _T(".exe");
141 if (FileExists(filename))
142 return filename;
144 filename = FindFileOnPath(filename, env);
145 if (!filename.IsEmpty())
146 return filename;
148 return executable;
151 static bool g_bSortLogical;
152 static bool g_bSortLocalBranchesFirst;
153 static bool g_bSortTagsReversed;
154 static git_cred_acquire_cb g_Git2CredCallback;
155 static git_transport_certificate_check_cb g_Git2CheckCertificateCallback;
157 static void GetSortOptions()
159 g_bSortLogical = !CRegDWORD(L"SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Policies\\Explorer\\NoStrCmpLogical", 0, false, HKEY_CURRENT_USER);
160 if (g_bSortLogical)
161 g_bSortLogical = !CRegDWORD(L"SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Policies\\Explorer\\NoStrCmpLogical", 0, false, HKEY_LOCAL_MACHINE);
162 g_bSortLocalBranchesFirst = !CRegDWORD(L"Software\\TortoiseGit\\NoSortLocalBranchesFirst", 0, false, HKEY_CURRENT_USER);
163 if (g_bSortLocalBranchesFirst)
164 g_bSortLocalBranchesFirst = !CRegDWORD(L"Software\\TortoiseGit\\NoSortLocalBranchesFirst", 0, false, HKEY_LOCAL_MACHINE);
165 g_bSortTagsReversed = !!CRegDWORD(L"Software\\TortoiseGit\\SortTagsReversed", 0, false, HKEY_LOCAL_MACHINE);
166 if (!g_bSortTagsReversed)
167 g_bSortTagsReversed = !!CRegDWORD(L"Software\\TortoiseGit\\SortTagsReversed", 0, false, HKEY_CURRENT_USER);
170 static int LogicalComparePredicate(const CString &left, const CString &right)
172 if (g_bSortLogical)
173 return StrCmpLogicalW(left, right) < 0;
174 return StrCmpI(left, right) < 0;
177 static int LogicalCompareReversedPredicate(const CString &left, const CString &right)
179 if (g_bSortLogical)
180 return StrCmpLogicalW(left, right) > 0;
181 return StrCmpI(left, right) > 0;
184 static int LogicalCompareBranchesPredicate(const CString &left, const CString &right)
186 if (g_bSortLocalBranchesFirst)
188 int leftIsRemote = left.Find(_T("remotes/"));
189 int rightIsRemote = right.Find(_T("remotes/"));
191 if (leftIsRemote == 0 && rightIsRemote < 0)
192 return false;
193 else if (leftIsRemote < 0 && rightIsRemote == 0)
194 return true;
196 if (g_bSortLogical)
197 return StrCmpLogicalW(left, right) < 0;
198 return StrCmpI(left, right) < 0;
201 #define CALL_OUTPUT_READ_CHUNK_SIZE 1024
203 CString CGit::ms_LastMsysGitDir;
204 CString CGit::ms_MsysGitRootDir;
205 int CGit::ms_LastMsysGitVersion = 0;
206 CGit g_Git;
209 CGit::CGit(void)
211 GetCurrentDirectory(MAX_PATH, CStrBuf(m_CurrentDir, MAX_PATH));
212 m_IsGitDllInited = false;
213 m_GitDiff=0;
214 m_GitSimpleListDiff=0;
215 m_IsUseGitDLL = !!CRegDWORD(_T("Software\\TortoiseGit\\UsingGitDLL"),1);
216 m_IsUseLibGit2 = !!CRegDWORD(_T("Software\\TortoiseGit\\UseLibgit2"), TRUE);
217 m_IsUseLibGit2_mask = CRegDWORD(_T("Software\\TortoiseGit\\UseLibgit2_mask"), DEFAULT_USE_LIBGIT2_MASK);
219 SecureZeroMemory(&m_CurrentGitPi, sizeof(PROCESS_INFORMATION));
221 GetSortOptions();
222 this->m_bInitialized =false;
223 CheckMsysGitDir();
224 m_critGitDllSec.Init();
227 CGit::~CGit(void)
229 if(this->m_GitDiff)
231 git_close_diff(m_GitDiff);
232 m_GitDiff=0;
234 if(this->m_GitSimpleListDiff)
236 git_close_diff(m_GitSimpleListDiff);
237 m_GitSimpleListDiff=0;
241 bool CGit::IsBranchNameValid(const CString& branchname)
243 if (branchname.Left(1) == _T("-")) // branch names starting with a dash are discouraged when used with git.exe, see https://github.com/git/git/commit/6348624010888bd2353e5cebdc2b5329490b0f6d
244 return false;
245 if (branchname.FindOneOf(_T("\"|<>")) >= 0) // not valid on Windows
246 return false;
247 CStringA branchA = CUnicodeUtils::GetUTF8(_T("refs/heads/") + branchname);
248 return !!git_reference_is_valid_name(branchA);
251 static bool IsPowerShell(CString cmd)
253 cmd.MakeLower();
254 int powerShellPos = cmd.Find(_T("powershell"));
255 if (powerShellPos < 0)
256 return false;
258 // found the word powershell, check that it is the command and not any parameter
259 int end = cmd.GetLength();
260 if (cmd.Find(_T('"')) == 0)
262 int secondDoubleQuote = cmd.Find(_T('"'), 1);
263 if (secondDoubleQuote > 0)
264 end = secondDoubleQuote;
266 else
268 int firstSpace = cmd.Find(_T(' '));
269 if (firstSpace > 0)
270 end = firstSpace;
273 return (end - 4 - 10 == powerShellPos || end - 10 == powerShellPos); // len(".exe")==4, len("powershell")==10
276 int CGit::RunAsync(CString cmd, PROCESS_INFORMATION *piOut, HANDLE *hReadOut, HANDLE *hErrReadOut, CString *StdioFile)
278 SECURITY_ATTRIBUTES sa;
279 CAutoGeneralHandle hRead, hWrite, hReadErr, hWriteErr;
280 CAutoGeneralHandle hStdioFile;
282 sa.nLength = sizeof(SECURITY_ATTRIBUTES);
283 sa.lpSecurityDescriptor=NULL;
284 sa.bInheritHandle=TRUE;
285 if (!CreatePipe(hRead.GetPointer(), hWrite.GetPointer(), &sa, 0))
287 CString err = CFormatMessageWrapper();
288 CTraceToOutputDebugString::Instance()(_T(__FUNCTION__) _T(": could not open stdout pipe: %s\n"), err.Trim());
289 return TGIT_GIT_ERROR_OPEN_PIP;
291 if (hErrReadOut && !CreatePipe(hReadErr.GetPointer(), hWriteErr.GetPointer(), &sa, 0))
293 CString err = CFormatMessageWrapper();
294 CTraceToOutputDebugString::Instance()(_T(__FUNCTION__) _T(": could not open stderr pipe: %s\n"), err.Trim());
295 return TGIT_GIT_ERROR_OPEN_PIP;
298 if(StdioFile)
300 hStdioFile=CreateFile(*StdioFile,GENERIC_WRITE,FILE_SHARE_READ | FILE_SHARE_WRITE,
301 &sa,CREATE_ALWAYS,FILE_ATTRIBUTE_NORMAL,NULL);
304 STARTUPINFO si = { 0 };
305 PROCESS_INFORMATION pi;
306 si.cb=sizeof(STARTUPINFO);
308 if (hErrReadOut)
309 si.hStdError = hWriteErr;
310 else
311 si.hStdError = hWrite;
312 if(StdioFile)
313 si.hStdOutput=hStdioFile;
314 else
315 si.hStdOutput=hWrite;
317 si.wShowWindow=SW_HIDE;
318 si.dwFlags=STARTF_USESTDHANDLES|STARTF_USESHOWWINDOW;
320 LPTSTR pEnv = m_Environment;
321 DWORD dwFlags = pEnv ? CREATE_UNICODE_ENVIRONMENT : 0;
323 dwFlags |= CREATE_NEW_PROCESS_GROUP;
325 // CREATE_NEW_CONSOLE makes git (but not ssh.exe, see issue #2257) recognize that it has no console in order to launch askpass to ask for the password,
326 // DETACHED_PROCESS which was originally used here has the same effect (but works with git.exe AND ssh.exe), however, it prevents PowerShell from working (cf. issue #2143)
327 // => we keep using DETACHED_PROCESS as the default, but if cmd contains pwershell we use CREATE_NEW_CONSOLE
328 if (IsPowerShell(cmd))
329 dwFlags |= CREATE_NEW_CONSOLE;
330 else
331 dwFlags |= DETACHED_PROCESS;
333 memset(&this->m_CurrentGitPi,0,sizeof(PROCESS_INFORMATION));
334 memset(&pi, 0, sizeof(PROCESS_INFORMATION));
336 if (ms_bCygwinGit && cmd.Find(_T("git")) == 0 && cmd.Find(L"git.exe config ") == -1)
338 cmd.Replace(_T('\\'), _T('/'));
339 cmd.Replace(_T("\""), _T("\\\""));
340 cmd = _T('"') + CGit::ms_LastMsysGitDir + _T("\\bash.exe\" -c \"/bin/") + cmd + _T('"');
342 else if(cmd.Find(_T("git")) == 0)
344 int firstSpace = cmd.Find(_T(" "));
345 if (firstSpace > 0)
346 cmd = _T('"')+CGit::ms_LastMsysGitDir+_T("\\")+ cmd.Left(firstSpace) + _T('"')+ cmd.Mid(firstSpace);
347 else
348 cmd=_T('"')+CGit::ms_LastMsysGitDir+_T("\\")+cmd+_T('"');
351 CTraceToOutputDebugString::Instance()(_T(__FUNCTION__) _T(": executing %s\n"), cmd);
352 if(!CreateProcess(NULL,(LPWSTR)cmd.GetString(), NULL,NULL,TRUE,dwFlags,pEnv,(LPWSTR)m_CurrentDir.GetString(),&si,&pi))
354 CString err = CFormatMessageWrapper();
355 CTraceToOutputDebugString::Instance()(_T(__FUNCTION__) _T(": error while executing command: %s\n"), err.Trim());
356 return TGIT_GIT_ERROR_CREATE_PROCESS;
359 m_CurrentGitPi = pi;
361 if(piOut)
362 *piOut=pi;
363 if(hReadOut)
364 *hReadOut = hRead.Detach();
365 if(hErrReadOut)
366 *hErrReadOut = hReadErr.Detach();
367 return 0;
370 //Must use sperate function to convert ANSI str to union code string
371 //Becuase A2W use stack as internal convert buffer.
372 void CGit::StringAppend(CString *str, const BYTE *p, int code,int length)
374 if(str == NULL)
375 return ;
377 int len ;
378 if(length<0)
379 len = (int)strlen((const char*)p);
380 else
381 len=length;
382 if (len == 0)
383 return;
384 int currentContentLen = str->GetLength();
385 WCHAR * buf = str->GetBuffer(len * 4 + 1 + currentContentLen) + currentContentLen;
386 int appendedLen = MultiByteToWideChar(code, 0, (LPCSTR)p, len, buf, len * 4);
387 str->ReleaseBuffer(currentContentLen + appendedLen); // no - 1 because MultiByteToWideChar is called with a fixed length (thus no nul char included)
390 // This method was originally used to check for orphaned branches
391 BOOL CGit::CanParseRev(CString ref)
393 if (ref.IsEmpty())
394 ref = _T("HEAD");
396 CString cmdout;
397 if (Run(_T("git.exe rev-parse --revs-only ") + ref, &cmdout, CP_UTF8))
399 return FALSE;
401 if(cmdout.IsEmpty())
402 return FALSE;
404 return TRUE;
407 // Checks if we have an orphaned HEAD
408 BOOL CGit::IsInitRepos()
410 CGitHash hash;
411 if (GetHash(hash, _T("HEAD")) != 0)
412 return FALSE;
413 return hash.IsEmpty() ? TRUE : FALSE;
416 DWORD WINAPI CGit::AsyncReadStdErrThread(LPVOID lpParam)
418 PASYNCREADSTDERRTHREADARGS pDataArray;
419 pDataArray = (PASYNCREADSTDERRTHREADARGS)lpParam;
421 DWORD readnumber;
422 BYTE data[CALL_OUTPUT_READ_CHUNK_SIZE];
423 while (ReadFile(pDataArray->fileHandle, data, CALL_OUTPUT_READ_CHUNK_SIZE, &readnumber, NULL))
425 if (pDataArray->pcall->OnOutputErrData(data,readnumber))
426 break;
429 return 0;
432 int CGit::Run(CGitCall* pcall)
434 PROCESS_INFORMATION pi;
435 CAutoGeneralHandle hRead, hReadErr;
436 if (RunAsync(pcall->GetCmd(), &pi, hRead.GetPointer(), hReadErr.GetPointer()))
437 return TGIT_GIT_ERROR_CREATE_PROCESS;
439 CAutoGeneralHandle piThread(pi.hThread);
440 CAutoGeneralHandle piProcess(pi.hProcess);
442 ASYNCREADSTDERRTHREADARGS threadArguments;
443 threadArguments.fileHandle = hReadErr;
444 threadArguments.pcall = pcall;
445 CAutoGeneralHandle thread = CreateThread(NULL, 0, AsyncReadStdErrThread, &threadArguments, 0, NULL);
447 DWORD readnumber;
448 BYTE data[CALL_OUTPUT_READ_CHUNK_SIZE];
449 bool bAborted=false;
450 while(ReadFile(hRead,data,CALL_OUTPUT_READ_CHUNK_SIZE,&readnumber,NULL))
452 // TODO: when OnOutputData() returns 'true', abort git-command. Send CTRL-C signal?
453 if(!bAborted)//For now, flush output when command aborted.
454 if(pcall->OnOutputData(data,readnumber))
455 bAborted=true;
457 if(!bAborted)
458 pcall->OnEnd();
460 if (thread)
461 WaitForSingleObject(thread, INFINITE);
463 WaitForSingleObject(pi.hProcess, INFINITE);
464 DWORD exitcode =0;
466 if(!GetExitCodeProcess(pi.hProcess,&exitcode))
468 CString err = CFormatMessageWrapper();
469 CTraceToOutputDebugString::Instance()(_T(__FUNCTION__) _T(": could not get exit code: %s\n"), err.Trim());
470 return TGIT_GIT_ERROR_GET_EXIT_CODE;
472 else
473 CTraceToOutputDebugString::Instance()(_T(__FUNCTION__) _T(": process exited: %d\n"), exitcode);
475 return exitcode;
477 class CGitCall_ByteVector : public CGitCall
479 public:
480 CGitCall_ByteVector(CString cmd,BYTE_VECTOR* pvector, BYTE_VECTOR* pvectorErr = NULL):CGitCall(cmd),m_pvector(pvector),m_pvectorErr(pvectorErr){}
481 virtual bool OnOutputData(const BYTE* data, size_t size)
483 if (!m_pvector || size == 0)
484 return false;
485 size_t oldsize=m_pvector->size();
486 m_pvector->resize(m_pvector->size()+size);
487 memcpy(&*(m_pvector->begin()+oldsize),data,size);
488 return false;
490 virtual bool OnOutputErrData(const BYTE* data, size_t size)
492 if (!m_pvectorErr || size == 0)
493 return false;
494 size_t oldsize = m_pvectorErr->size();
495 m_pvectorErr->resize(m_pvectorErr->size() + size);
496 memcpy(&*(m_pvectorErr->begin() + oldsize), data, size);
497 return false;
499 BYTE_VECTOR* m_pvector;
500 BYTE_VECTOR* m_pvectorErr;
503 int CGit::Run(CString cmd,BYTE_VECTOR *vector, BYTE_VECTOR *vectorErr)
505 CGitCall_ByteVector call(cmd, vector, vectorErr);
506 return Run(&call);
508 int CGit::Run(CString cmd, CString* output, int code)
510 CString err;
511 int ret;
512 ret = Run(cmd, output, &err, code);
514 if (output && !err.IsEmpty())
516 if (!output->IsEmpty())
517 *output += _T("\n");
518 *output += err;
521 return ret;
523 int CGit::Run(CString cmd, CString* output, CString* outputErr, int code)
525 BYTE_VECTOR vector, vectorErr;
526 int ret;
527 if (outputErr)
528 ret = Run(cmd, &vector, &vectorErr);
529 else
530 ret = Run(cmd, &vector);
532 vector.push_back(0);
533 StringAppend(output, &(vector[0]), code);
535 if (outputErr)
537 vectorErr.push_back(0);
538 StringAppend(outputErr, &(vectorErr[0]), code);
541 return ret;
544 class CGitCallCb : public CGitCall
546 public:
547 CGitCallCb(CString cmd, const GitReceiverFunc& recv): CGitCall(cmd), m_recv(recv) {}
549 virtual bool OnOutputData(const BYTE* data, size_t size) override
551 // Add data
552 if (size == 0)
553 return false;
554 int oldEndPos = m_buffer.GetLength();
555 memcpy(m_buffer.GetBuffer(oldEndPos + (int)size) + oldEndPos, data, size);
556 m_buffer.ReleaseBuffer(oldEndPos + (int)size);
558 // Break into lines and feed to m_recv
559 int eolPos;
560 while ((eolPos = m_buffer.Find('\n')) >= 0)
562 m_recv(m_buffer.Left(eolPos));
563 m_buffer = m_buffer.Mid(eolPos + 1);
565 return false;
568 virtual bool OnOutputErrData(const BYTE*, size_t) override
570 return false; // Ignore error output for now
573 virtual void OnEnd() override
575 if (!m_buffer.IsEmpty())
576 m_recv(m_buffer);
577 m_buffer.Empty(); // Just for sure
580 private:
581 GitReceiverFunc m_recv;
582 CStringA m_buffer;
585 int CGit::Run(CString cmd, const GitReceiverFunc& recv)
587 CGitCallCb call(cmd, recv);
588 return Run(&call);
591 CString CGit::GetUserName(void)
593 CEnvironment env;
594 env.CopyProcessEnvironment();
595 CString envname = env.GetEnv(_T("GIT_AUTHOR_NAME"));
596 if (!envname.IsEmpty())
597 return envname;
598 return GetConfigValue(L"user.name");
600 CString CGit::GetUserEmail(void)
602 CEnvironment env;
603 env.CopyProcessEnvironment();
604 CString envmail = env.GetEnv(_T("GIT_AUTHOR_EMAIL"));
605 if (!envmail.IsEmpty())
606 return envmail;
608 return GetConfigValue(L"user.email");
611 CString CGit::GetConfigValue(const CString& name)
613 CString configValue;
614 if(this->m_IsUseGitDLL)
616 CAutoLocker lock(g_Git.m_critGitDllSec);
620 CheckAndInitDll();
621 }catch(...)
624 CStringA key, value;
625 key = CUnicodeUtils::GetUTF8(name);
629 if (git_get_config(key, CStrBufA(value, 4096), 4096))
630 return CString();
632 catch (const char *msg)
634 ::MessageBox(NULL, _T("Could not get config.\nlibgit reports:\n") + CString(msg), _T("TortoiseGit"), MB_OK | MB_ICONERROR);
635 return CString();
638 StringAppend(&configValue, (BYTE*)(LPCSTR)value);
639 return configValue;
641 else
643 CString cmd;
644 cmd.Format(L"git.exe config %s", name);
645 Run(cmd, &configValue, nullptr, CP_UTF8);
646 if (configValue.IsEmpty())
647 return configValue;
648 return configValue.Left(configValue.GetLength() - 1); // strip last newline character
652 bool CGit::GetConfigValueBool(const CString& name)
654 CString configValue = GetConfigValue(name);
655 configValue.MakeLower();
656 configValue.Trim();
657 if(configValue == _T("true") || configValue == _T("on") || configValue == _T("yes") || StrToInt(configValue) != 0)
658 return true;
659 else
660 return false;
663 int CGit::GetConfigValueInt32(const CString& name, int def)
665 CString configValue = GetConfigValue(name);
666 int value = def;
667 if (!git_config_parse_int32(&value, CUnicodeUtils::GetUTF8(configValue)))
668 return value;
669 return def;
672 int CGit::SetConfigValue(const CString& key, const CString& value, CONFIG_TYPE type)
674 if(this->m_IsUseGitDLL)
676 CAutoLocker lock(g_Git.m_critGitDllSec);
680 CheckAndInitDll();
682 }catch(...)
685 CStringA keya, valuea;
686 keya = CUnicodeUtils::GetMulti(key, CP_UTF8);
687 valuea = CUnicodeUtils::GetUTF8(value);
691 return [=]() { return get_set_config(keya, valuea, type); }();
693 catch (const char *msg)
695 ::MessageBox(NULL, _T("Could not set config.\nlibgit reports:\n") + CString(msg), _T("TortoiseGit"), MB_OK | MB_ICONERROR);
696 return -1;
699 else
701 CString cmd;
702 CString option;
703 switch(type)
705 case CONFIG_GLOBAL:
706 option = _T("--global");
707 break;
708 case CONFIG_SYSTEM:
709 option = _T("--system");
710 break;
711 default:
712 break;
714 CString mangledValue = value;
715 mangledValue.Replace(_T("\\\""), _T("\\\\\""));
716 mangledValue.Replace(_T("\""), _T("\\\""));
717 cmd.Format(_T("git.exe config %s %s \"%s\""), option, key, mangledValue);
718 CString out;
719 if (Run(cmd, &out, nullptr, CP_UTF8))
721 return -1;
724 return 0;
727 int CGit::UnsetConfigValue(const CString& key, CONFIG_TYPE type)
729 if(this->m_IsUseGitDLL)
731 CAutoLocker lock(g_Git.m_critGitDllSec);
735 CheckAndInitDll();
736 }catch(...)
739 CStringA keya;
740 keya = CUnicodeUtils::GetMulti(key, CP_UTF8);
744 return [=]() { return get_set_config(keya, nullptr, type); }();
746 catch (const char *msg)
748 ::MessageBox(NULL, _T("Could not unset config.\nlibgit reports:\n") + CString(msg), _T("TortoiseGit"), MB_OK | MB_ICONERROR);
749 return -1;
752 else
754 CString cmd;
755 CString option;
756 switch(type)
758 case CONFIG_GLOBAL:
759 option = _T("--global");
760 break;
761 case CONFIG_SYSTEM:
762 option = _T("--system");
763 break;
764 default:
765 break;
767 cmd.Format(_T("git.exe config %s --unset %s"), option, key);
768 CString out;
769 if (Run(cmd, &out, nullptr, CP_UTF8))
771 return -1;
774 return 0;
777 CString CGit::GetCurrentBranch(bool fallback)
779 CString output;
780 //Run(_T("git.exe branch"),&branch);
782 int result = GetCurrentBranchFromFile(m_CurrentDir, output, fallback);
783 if (result != 0 && ((result == 1 && !fallback) || result != 1))
785 return _T("(no branch)");
787 else
788 return output;
792 void CGit::GetRemoteTrackedBranch(const CString& localBranch, CString& pullRemote, CString& pullBranch)
794 if (localBranch.IsEmpty())
795 return;
797 CString configName;
798 configName.Format(L"branch.%s.remote", localBranch);
799 pullRemote = GetConfigValue(configName);
801 //Select pull-branch from current branch
802 configName.Format(L"branch.%s.merge", localBranch);
803 pullBranch = StripRefName(GetConfigValue(configName));
806 void CGit::GetRemoteTrackedBranchForHEAD(CString& remote, CString& branch)
808 CString refName;
809 if (GetCurrentBranchFromFile(m_CurrentDir, refName))
810 return;
811 GetRemoteTrackedBranch(StripRefName(refName), remote, branch);
814 CString CGit::GetFullRefName(const CString& shortRefName)
816 CString refName;
817 CString cmd;
818 cmd.Format(L"git.exe rev-parse --symbolic-full-name %s", shortRefName);
819 if (Run(cmd, &refName, NULL, CP_UTF8) != 0)
820 return CString();//Error
821 int iStart = 0;
822 return refName.Tokenize(L"\n", iStart);
825 CString CGit::StripRefName(CString refName)
827 if(wcsncmp(refName, L"refs/heads/", 11) == 0)
828 refName = refName.Mid(11);
829 else if(wcsncmp(refName, L"refs/", 5) == 0)
830 refName = refName.Mid(5);
831 int start =0;
832 return refName.Tokenize(_T("\n"),start);
835 int CGit::GetCurrentBranchFromFile(const CString &sProjectRoot, CString &sBranchOut, bool fallback)
837 // read current branch name like git-gui does, by parsing the .git/HEAD file directly
839 if ( sProjectRoot.IsEmpty() )
840 return -1;
842 CString sDotGitPath;
843 if (!GitAdminDir::GetAdminDirPath(sProjectRoot, sDotGitPath))
844 return -1;
846 CString sHeadFile = sDotGitPath + _T("HEAD");
848 FILE *pFile;
849 _tfopen_s(&pFile, sHeadFile.GetString(), _T("r"));
851 if (!pFile)
853 return -1;
856 char s[256] = {0};
857 fgets(s, sizeof(s), pFile);
859 fclose(pFile);
861 const char *pfx = "ref: refs/heads/";
862 const int len = 16;//strlen(pfx)
864 if ( !strncmp(s, pfx, len) )
866 //# We're on a branch. It might not exist. But
867 //# HEAD looks good enough to be a branch.
868 CStringA utf8Branch(s + len);
869 sBranchOut = CUnicodeUtils::GetUnicode(utf8Branch);
870 sBranchOut.TrimRight(_T(" \r\n\t"));
872 if ( sBranchOut.IsEmpty() )
873 return -1;
875 else if (fallback)
877 CStringA utf8Hash(s);
878 CString unicodeHash = CUnicodeUtils::GetUnicode(utf8Hash);
879 unicodeHash.TrimRight(_T(" \r\n\t"));
880 if (CGitHash::IsValidSHA1(unicodeHash))
881 sBranchOut = unicodeHash;
882 else
883 //# Assume this is a detached head.
884 sBranchOut = _T("HEAD");
885 return 1;
887 else
889 //# Assume this is a detached head.
890 sBranchOut = "HEAD";
892 return 1;
895 return 0;
898 int CGit::BuildOutputFormat(CString &format,bool IsFull)
900 CString log;
901 log.Format(_T("#<%c>%%x00"),LOG_REV_ITEM_BEGIN);
902 format += log;
903 if(IsFull)
905 log.Format(_T("#<%c>%%an%%x00"),LOG_REV_AUTHOR_NAME);
906 format += log;
907 log.Format(_T("#<%c>%%ae%%x00"),LOG_REV_AUTHOR_EMAIL);
908 format += log;
909 log.Format(_T("#<%c>%%ai%%x00"),LOG_REV_AUTHOR_DATE);
910 format += log;
911 log.Format(_T("#<%c>%%cn%%x00"),LOG_REV_COMMIT_NAME);
912 format += log;
913 log.Format(_T("#<%c>%%ce%%x00"),LOG_REV_COMMIT_EMAIL);
914 format += log;
915 log.Format(_T("#<%c>%%ci%%x00"),LOG_REV_COMMIT_DATE);
916 format += log;
917 log.Format(_T("#<%c>%%b%%x00"),LOG_REV_COMMIT_BODY);
918 format += log;
921 log.Format(_T("#<%c>%%m%%H%%x00"),LOG_REV_COMMIT_HASH);
922 format += log;
923 log.Format(_T("#<%c>%%P%%x00"),LOG_REV_COMMIT_PARENT);
924 format += log;
925 log.Format(_T("#<%c>%%s%%x00"),LOG_REV_COMMIT_SUBJECT);
926 format += log;
928 if(IsFull)
930 log.Format(_T("#<%c>%%x00"),LOG_REV_COMMIT_FILE);
931 format += log;
933 return 0;
936 CString CGit::GetLogCmd(const CString &range, const CTGitPath *path, int count, int mask, bool paramonly,
937 CFilterData *Filter)
939 CString cmd;
940 CString num;
941 CString since;
943 CString file = _T(" --");
945 if(path)
946 file.Format(_T(" -- \"%s\""),path->GetGitPathString());
948 if(count>0)
949 num.Format(_T("-n%d"),count);
951 CString param;
953 if(mask& LOG_INFO_STAT )
954 param += _T(" --numstat ");
955 if(mask& LOG_INFO_FILESTATE)
956 param += _T(" --raw ");
958 if(mask& LOG_INFO_FULLHISTORY)
959 param += _T(" --full-history ");
961 if(mask& LOG_INFO_BOUNDARY)
962 param += _T(" --left-right --boundary ");
964 if(mask& CGit::LOG_INFO_ALL_BRANCH)
965 param += _T(" --all ");
967 if(mask & CGit::LOG_INFO_LOCAL_BRANCHES)
968 param += _T(" --branches ");
970 if(mask& CGit::LOG_INFO_DETECT_COPYRENAME)
971 param += _T(" -C ");
973 if(mask& CGit::LOG_INFO_DETECT_RENAME )
974 param += _T(" -M ");
976 if(mask& CGit::LOG_INFO_FIRST_PARENT )
977 param += _T(" --first-parent ");
979 if(mask& CGit::LOG_INFO_NO_MERGE )
980 param += _T(" --no-merges ");
982 if(mask& CGit::LOG_INFO_FOLLOW)
983 param += _T(" --follow ");
985 if(mask& CGit::LOG_INFO_SHOW_MERGEDFILE)
986 param += _T(" -c ");
988 if(mask& CGit::LOG_INFO_FULL_DIFF)
989 param += _T(" --full-diff ");
991 if(mask& CGit::LOG_INFO_SIMPILFY_BY_DECORATION)
992 param += _T(" --simplify-by-decoration ");
994 param += range;
996 CString st1,st2;
998 if( Filter && (Filter->m_From != -1))
1000 st1.Format(_T(" --max-age=%I64u "), Filter->m_From);
1001 param += st1;
1004 if( Filter && (Filter->m_To != -1))
1006 st2.Format(_T(" --min-age=%I64u "), Filter->m_To);
1007 param += st2;
1010 bool isgrep = false;
1011 if( Filter && (!Filter->m_Author.IsEmpty()))
1013 st1.Format(_T(" --author=\"%s\"" ),Filter->m_Author);
1014 param += st1;
1015 isgrep = true;
1018 if( Filter && (!Filter->m_Committer.IsEmpty()))
1020 st1.Format(_T(" --committer=\"%s\"" ),Filter->m_Author);
1021 param += st1;
1022 isgrep = true;
1025 if( Filter && (!Filter->m_MessageFilter.IsEmpty()))
1027 st1.Format(_T(" --grep=\"%s\"" ),Filter->m_MessageFilter);
1028 param += st1;
1029 isgrep = true;
1032 if(Filter && isgrep)
1034 if(!Filter->m_IsRegex)
1035 param += _T(" --fixed-strings ");
1037 param += _T(" --regexp-ignore-case --extended-regexp ");
1040 DWORD logOrderBy = CRegDWORD(_T("Software\\TortoiseGit\\LogOrderBy"), LOG_ORDER_TOPOORDER);
1041 if (logOrderBy == LOG_ORDER_TOPOORDER)
1042 param += _T(" --topo-order");
1043 else if (logOrderBy == LOG_ORDER_DATEORDER)
1044 param += _T(" --date-order");
1046 if(paramonly) //tgit.dll.Git.cpp:setup_revisions() only looks at args[1] and greater. To account for this, pass a dummy parameter in the 0th place
1047 cmd.Format(_T("--ignore-this-parameter %s -z %s --parents "), num, param);
1048 else
1050 CString log;
1051 BuildOutputFormat(log,!(mask&CGit::LOG_INFO_ONLY_HASH));
1052 cmd.Format(_T("git.exe log %s -z %s --parents --pretty=format:\"%s\""),
1053 num,param,log);
1056 cmd += file;
1058 return cmd;
1060 #define BUFSIZE 512
1061 void GetTempPath(CString &path)
1063 TCHAR lpPathBuffer[BUFSIZE] = { 0 };
1064 DWORD dwRetVal;
1065 DWORD dwBufSize=BUFSIZE;
1066 dwRetVal = GetTortoiseGitTempPath(dwBufSize, // length of the buffer
1067 lpPathBuffer); // buffer for path
1068 if (dwRetVal > dwBufSize || (dwRetVal == 0))
1070 path=_T("");
1072 path.Format(_T("%s"),lpPathBuffer);
1074 CString GetTempFile()
1076 TCHAR lpPathBuffer[BUFSIZE] = { 0 };
1077 DWORD dwRetVal;
1078 DWORD dwBufSize=BUFSIZE;
1079 TCHAR szTempName[BUFSIZE] = { 0 };
1080 UINT uRetVal;
1082 dwRetVal = GetTortoiseGitTempPath(dwBufSize, // length of the buffer
1083 lpPathBuffer); // buffer for path
1084 if (dwRetVal > dwBufSize || (dwRetVal == 0))
1086 return _T("");
1088 // Create a temporary file.
1089 uRetVal = GetTempFileName(lpPathBuffer, // directory for tmp files
1090 TEXT("Patch"), // temp file name prefix
1091 0, // create unique name
1092 szTempName); // buffer for name
1095 if (uRetVal == 0)
1097 return _T("");
1100 return CString(szTempName);
1104 DWORD GetTortoiseGitTempPath(DWORD nBufferLength, LPTSTR lpBuffer)
1106 DWORD result = ::GetTempPath(nBufferLength, lpBuffer);
1107 if (result == 0) return 0;
1108 if (lpBuffer == NULL || (result + 13 > nBufferLength))
1110 if (lpBuffer)
1111 lpBuffer[0] = '\0';
1112 return result + 13;
1115 _tcscat_s(lpBuffer, nBufferLength, _T("TortoiseGit\\"));
1116 CreateDirectory(lpBuffer, NULL);
1118 return result + 13;
1121 int CGit::RunLogFile(CString cmd, const CString &filename, CString *stdErr)
1123 STARTUPINFO si;
1124 PROCESS_INFORMATION pi;
1125 si.cb=sizeof(STARTUPINFO);
1126 GetStartupInfo(&si);
1128 SECURITY_ATTRIBUTES psa={sizeof(psa),NULL,TRUE};;
1129 psa.bInheritHandle=TRUE;
1131 HANDLE hReadErr, hWriteErr;
1132 if (!CreatePipe(&hReadErr, &hWriteErr, &psa, 0))
1134 CString err = CFormatMessageWrapper();
1135 CTraceToOutputDebugString::Instance()(_T(__FUNCTION__) _T(": could not open stderr pipe: %s\n"), err.Trim());
1136 return TGIT_GIT_ERROR_OPEN_PIP;
1139 HANDLE houtfile=CreateFile(filename,GENERIC_WRITE,FILE_SHARE_READ | FILE_SHARE_WRITE,
1140 &psa,CREATE_ALWAYS,FILE_ATTRIBUTE_NORMAL,NULL);
1142 if (houtfile == INVALID_HANDLE_VALUE)
1144 CString err = CFormatMessageWrapper();
1145 CTraceToOutputDebugString::Instance()(_T(__FUNCTION__) _T(": could not open stdout pipe: %s\n"), err.Trim());
1146 CloseHandle(hReadErr);
1147 CloseHandle(hWriteErr);
1148 return TGIT_GIT_ERROR_OPEN_PIP;
1151 si.wShowWindow = SW_HIDE;
1152 si.dwFlags = STARTF_USESTDHANDLES|STARTF_USESHOWWINDOW;
1153 si.hStdOutput = houtfile;
1154 si.hStdError = hWriteErr;
1156 LPTSTR pEnv = m_Environment;
1157 DWORD dwFlags = pEnv ? CREATE_UNICODE_ENVIRONMENT : 0;
1159 if(cmd.Find(_T("git")) == 0)
1160 cmd=CGit::ms_LastMsysGitDir+_T("\\")+cmd;
1162 CTraceToOutputDebugString::Instance()(_T(__FUNCTION__) _T(": executing %s\n"), cmd);
1163 if(!CreateProcess(NULL,(LPWSTR)cmd.GetString(), NULL,NULL,TRUE,dwFlags,pEnv,(LPWSTR)m_CurrentDir.GetString(),&si,&pi))
1165 CString err = CFormatMessageWrapper();
1166 CTraceToOutputDebugString::Instance()(_T(__FUNCTION__) _T(": failed to create Process: %s\n"), err.Trim());
1167 stdErr = &err;
1168 CloseHandle(hReadErr);
1169 CloseHandle(hWriteErr);
1170 CloseHandle(houtfile);
1171 return TGIT_GIT_ERROR_CREATE_PROCESS;
1174 BYTE_VECTOR stderrVector;
1175 CGitCall_ByteVector pcall(L"", nullptr, &stderrVector);
1176 HANDLE thread;
1177 ASYNCREADSTDERRTHREADARGS threadArguments;
1178 threadArguments.fileHandle = hReadErr;
1179 threadArguments.pcall = &pcall;
1180 thread = CreateThread(nullptr, 0, AsyncReadStdErrThread, &threadArguments, 0, nullptr);
1182 WaitForSingleObject(pi.hProcess,INFINITE);
1184 CloseHandle(hWriteErr);
1185 CloseHandle(hReadErr);
1187 if (thread)
1188 WaitForSingleObject(thread, INFINITE);
1190 stderrVector.push_back(0);
1191 StringAppend(stdErr, &(stderrVector[0]), CP_UTF8);
1193 DWORD exitcode = 0;
1194 if (!GetExitCodeProcess(pi.hProcess, &exitcode))
1196 CString err = CFormatMessageWrapper();
1197 CTraceToOutputDebugString::Instance()(_T(__FUNCTION__) _T(": could not get exit code: %s\n"), err.Trim());
1198 return TGIT_GIT_ERROR_GET_EXIT_CODE;
1200 else
1201 CTraceToOutputDebugString::Instance()(_T(__FUNCTION__) _T(": process exited: %d\n"), exitcode);
1203 CloseHandle(pi.hThread);
1204 CloseHandle(pi.hProcess);
1205 CloseHandle(houtfile);
1206 return exitcode;
1209 git_repository * CGit::GetGitRepository() const
1211 git_repository * repo = nullptr;
1212 git_repository_open(&repo, GetGitPathStringA(m_CurrentDir));
1213 return repo;
1216 int CGit::GetHash(git_repository * repo, CGitHash &hash, const CString& friendname, bool skipFastCheck /* = false */)
1218 ATLASSERT(repo);
1220 // no need to parse a ref if it's already a 40-byte hash
1221 if (!skipFastCheck && CGitHash::IsValidSHA1(friendname))
1223 hash = CGitHash(friendname);
1224 return 0;
1227 int isHeadOrphan = git_repository_head_unborn(repo);
1228 if (isHeadOrphan != 0)
1230 hash.Empty();
1231 if (isHeadOrphan == 1)
1232 return 0;
1233 else
1234 return -1;
1237 CAutoObject gitObject;
1238 if (git_revparse_single(gitObject.GetPointer(), repo, CUnicodeUtils::GetUTF8(friendname)))
1239 return -1;
1241 const git_oid * oid = git_object_id(gitObject);
1242 if (!oid)
1243 return -1;
1245 hash = CGitHash((char *)oid->id);
1247 return 0;
1250 int CGit::GetHash(CGitHash &hash, const CString& friendname)
1252 // no need to parse a ref if it's already a 40-byte hash
1253 if (CGitHash::IsValidSHA1(friendname))
1255 hash = CGitHash(friendname);
1256 return 0;
1259 if (m_IsUseLibGit2)
1261 CAutoRepository repo(GetGitRepository());
1262 if (!repo)
1263 return -1;
1265 return GetHash(repo, hash, friendname, true);
1267 else
1269 CString branch = FixBranchName(friendname);
1270 if (friendname == _T("FETCH_HEAD") && branch.IsEmpty())
1271 branch = friendname;
1272 CString cmd;
1273 cmd.Format(_T("git.exe rev-parse %s"), branch);
1274 gitLastErr.Empty();
1275 int ret = Run(cmd, &gitLastErr, NULL, CP_UTF8);
1276 hash = CGitHash(gitLastErr.Trim());
1277 if (ret == 0)
1278 gitLastErr.Empty();
1279 else if (friendname == _T("HEAD")) // special check for unborn branch
1281 CString currentbranch;
1282 if (GetCurrentBranchFromFile(m_CurrentDir, currentbranch))
1283 return -1;
1284 gitLastErr.Empty();
1285 return 0;
1287 return ret;
1291 int CGit::GetInitAddList(CTGitPathList &outputlist)
1293 CString cmd;
1294 BYTE_VECTOR cmdout;
1296 cmd=_T("git.exe ls-files -s -t -z");
1297 outputlist.Clear();
1298 if (Run(cmd, &cmdout))
1299 return -1;
1301 if (outputlist.ParserFromLsFile(cmdout))
1302 return -1;
1303 for(int i = 0; i < outputlist.GetCount(); ++i)
1304 const_cast<CTGitPath&>(outputlist[i]).m_Action = CTGitPath::LOGACTIONS_ADDED;
1306 return 0;
1308 int CGit::GetCommitDiffList(const CString &rev1, const CString &rev2, CTGitPathList &outputlist, bool ignoreSpaceAtEol, bool ignoreSpaceChange, bool ignoreAllSpace , bool ignoreBlankLines)
1310 CString cmd;
1311 CString ignore;
1312 if (ignoreSpaceAtEol)
1313 ignore += _T(" --ignore-space-at-eol");
1314 if (ignoreSpaceChange)
1315 ignore += _T(" --ignore-space-change");
1316 if (ignoreAllSpace)
1317 ignore += _T(" --ignore-all-space");
1318 if (ignoreBlankLines)
1319 ignore += _T(" --ignore-blank-lines");
1321 if(rev1 == GIT_REV_ZERO || rev2 == GIT_REV_ZERO)
1323 //rev1=+_T("");
1324 if(rev1 == GIT_REV_ZERO)
1325 cmd.Format(_T("git.exe diff -r --raw -C -M --numstat -z %s %s --"), ignore, rev2);
1326 else
1327 cmd.Format(_T("git.exe diff -r -R --raw -C -M --numstat -z %s %s --"), ignore, rev1);
1329 else
1331 cmd.Format(_T("git.exe diff-tree -r --raw -C -M --numstat -z %s %s %s --"), ignore, rev2, rev1);
1334 BYTE_VECTOR out;
1335 if (Run(cmd, &out))
1336 return -1;
1338 return outputlist.ParserFromLog(out);
1341 int addto_list_each_ref_fn(const char *refname, const unsigned char * /*sha1*/, int /*flags*/, void *cb_data)
1343 STRING_VECTOR *list = (STRING_VECTOR*)cb_data;
1344 list->push_back(CUnicodeUtils::GetUnicode(refname));
1345 return 0;
1348 int CGit::GetTagList(STRING_VECTOR &list)
1350 if (this->m_IsUseLibGit2)
1352 CAutoRepository repo(GetGitRepository());
1353 if (!repo)
1354 return -1;
1356 CAutoStrArray tag_names;
1358 if (git_tag_list(tag_names, repo))
1359 return -1;
1361 for (size_t i = 0; i < tag_names->count; ++i)
1363 CStringA tagName(tag_names->strings[i]);
1364 list.push_back(CUnicodeUtils::GetUnicode(tagName));
1367 std::sort(list.begin(), list.end(), g_bSortTagsReversed ? LogicalCompareReversedPredicate : LogicalComparePredicate);
1369 return 0;
1371 else
1373 CString cmd, output;
1374 cmd=_T("git.exe tag -l");
1375 int ret = Run(cmd, &output, NULL, CP_UTF8);
1376 if(!ret)
1378 int pos=0;
1379 CString one;
1380 while( pos>=0 )
1382 one=output.Tokenize(_T("\n"),pos);
1383 if (!one.IsEmpty())
1384 list.push_back(one);
1386 std::sort(list.begin(), list.end(), g_bSortTagsReversed ? LogicalCompareReversedPredicate : LogicalComparePredicate);
1388 else if (ret == 1 && IsInitRepos())
1389 return 0;
1390 return ret;
1394 CString CGit::GetGitLastErr(const CString& msg)
1396 if (this->m_IsUseLibGit2)
1397 return GetLibGit2LastErr(msg);
1398 else if (gitLastErr.IsEmpty())
1399 return msg + _T("\nUnknown git.exe error.");
1400 else
1402 CString lastError = gitLastErr;
1403 gitLastErr.Empty();
1404 return msg + _T("\n") + lastError;
1408 CString CGit::GetGitLastErr(const CString& msg, LIBGIT2_CMD cmd)
1410 if (UsingLibGit2(cmd))
1411 return GetLibGit2LastErr(msg);
1412 else if (gitLastErr.IsEmpty())
1413 return msg + _T("\nUnknown git.exe error.");
1414 else
1416 CString lastError = gitLastErr;
1417 gitLastErr.Empty();
1418 return msg + _T("\n") + lastError;
1422 CString CGit::GetLibGit2LastErr()
1424 const git_error *libgit2err = giterr_last();
1425 if (libgit2err)
1427 CString lastError = CUnicodeUtils::GetUnicode(CStringA(libgit2err->message));
1428 giterr_clear();
1429 return _T("libgit2 returned: ") + lastError;
1431 else
1432 return _T("An error occoured in libgit2, but no message is available.");
1435 CString CGit::GetLibGit2LastErr(const CString& msg)
1437 if (!msg.IsEmpty())
1438 return msg + _T("\n") + GetLibGit2LastErr();
1439 return GetLibGit2LastErr();
1442 CString CGit::FixBranchName_Mod(CString& branchName)
1444 if(branchName == "FETCH_HEAD")
1445 branchName = DerefFetchHead();
1446 return branchName;
1449 CString CGit::FixBranchName(const CString& branchName)
1451 CString tempBranchName = branchName;
1452 FixBranchName_Mod(tempBranchName);
1453 return tempBranchName;
1456 bool CGit::IsBranchTagNameUnique(const CString& name)
1458 if (m_IsUseLibGit2)
1460 CAutoRepository repo(GetGitRepository());
1461 if (!repo)
1462 return true; // TODO: optimize error reporting
1464 CAutoReference tagRef;
1465 if (git_reference_lookup(tagRef.GetPointer(), repo, CUnicodeUtils::GetUTF8(L"refs/tags/" + name)))
1466 return true;
1468 CAutoReference branchRef;
1469 if (git_reference_lookup(branchRef.GetPointer(), repo, CUnicodeUtils::GetUTF8(L"refs/heads/" + name)))
1470 return true;
1472 return false;
1474 // else
1475 CString output;
1477 CString cmd;
1478 cmd.Format(_T("git.exe show-ref --tags --heads refs/heads/%s refs/tags/%s"), name, name);
1479 int ret = Run(cmd, &output, NULL, CP_UTF8);
1480 if (!ret)
1482 int i = 0, pos = 0;
1483 while (pos >= 0)
1485 if (!output.Tokenize(_T("\n"), pos).IsEmpty())
1486 ++i;
1488 if (i >= 2)
1489 return false;
1492 return true;
1495 bool CGit::BranchTagExists(const CString& name, bool isBranch /*= true*/)
1497 if (m_IsUseLibGit2)
1499 CAutoRepository repo(GetGitRepository());
1500 if (!repo)
1501 return false; // TODO: optimize error reporting
1503 CString prefix;
1504 if (isBranch)
1505 prefix = _T("refs/heads/");
1506 else
1507 prefix = _T("refs/tags/");
1509 CAutoReference ref;
1510 if (git_reference_lookup(ref.GetPointer(), repo, CUnicodeUtils::GetUTF8(prefix + name)))
1511 return false;
1513 return true;
1515 // else
1516 CString cmd, output;
1518 cmd = _T("git.exe show-ref ");
1519 if (isBranch)
1520 cmd += _T("--heads ");
1521 else
1522 cmd += _T("--tags ");
1524 cmd += _T("refs/heads/") + name;
1525 cmd += _T(" refs/tags/") + name;
1527 int ret = Run(cmd, &output, NULL, CP_UTF8);
1528 if (!ret)
1530 if (!output.IsEmpty())
1531 return true;
1534 return false;
1537 CString CGit::DerefFetchHead()
1539 CString dotGitPath;
1540 GitAdminDir::GetAdminDirPath(m_CurrentDir, dotGitPath);
1541 std::ifstream fetchHeadFile((dotGitPath + L"FETCH_HEAD").GetString(), std::ios::in | std::ios::binary);
1542 int forMergeLineCount = 0;
1543 std::string line;
1544 std::string hashToReturn;
1545 while(getline(fetchHeadFile, line))
1547 //Tokenize this line
1548 std::string::size_type prevPos = 0;
1549 std::string::size_type pos = line.find('\t');
1550 if(pos == std::string::npos) continue; //invalid line
1552 std::string hash = line.substr(0, pos);
1553 ++pos; prevPos = pos; pos = line.find('\t', pos); if(pos == std::string::npos) continue;
1555 bool forMerge = pos == prevPos;
1556 ++pos; prevPos = pos; pos = line.size(); if(pos == std::string::npos) continue;
1558 std::string remoteBranch = line.substr(prevPos, pos - prevPos);
1560 //Process this line
1561 if(forMerge)
1563 hashToReturn = hash;
1564 ++forMergeLineCount;
1565 if(forMergeLineCount > 1)
1566 return L""; //More then 1 branch to merge found. Octopus merge needed. Cannot pick single ref from FETCH_HEAD
1570 return CUnicodeUtils::GetUnicode(hashToReturn.c_str());
1573 int CGit::GetBranchList(STRING_VECTOR &list,int *current,BRANCH_TYPE type)
1575 int ret = 0;
1576 CString cur;
1577 if (m_IsUseLibGit2)
1579 CAutoRepository repo(GetGitRepository());
1580 if (!repo)
1581 return -1;
1583 if (git_repository_head_detached(repo) == 1)
1584 cur = _T("(detached from ");
1586 if ((type & (BRANCH_LOCAL | BRANCH_REMOTE)) != 0)
1588 git_branch_t flags = GIT_BRANCH_LOCAL;
1589 if ((type & BRANCH_LOCAL) && (type & BRANCH_REMOTE))
1590 flags = GIT_BRANCH_ALL;
1591 else if (type & BRANCH_REMOTE)
1592 flags = GIT_BRANCH_REMOTE;
1594 CAutoBranchIterator it;
1595 if (git_branch_iterator_new(it.GetPointer(), repo, flags))
1596 return -1;
1598 git_reference * ref = nullptr;
1599 git_branch_t branchType;
1600 while (git_branch_next(&ref, &branchType, it) == 0)
1602 CAutoReference autoRef(ref);
1603 const char * name = nullptr;
1604 if (git_branch_name(&name, ref))
1605 continue;
1607 CString branchname = CUnicodeUtils::GetUnicode(name);
1608 if (branchType & GIT_BRANCH_REMOTE)
1609 list.push_back(_T("remotes/") + branchname);
1610 else
1612 if (git_branch_is_head(ref))
1613 cur = branchname;
1614 list.push_back(branchname);
1619 else
1621 CString cmd, output;
1622 cmd = _T("git.exe branch --no-color");
1624 if ((type & BRANCH_ALL) == BRANCH_ALL)
1625 cmd += _T(" -a");
1626 else if (type & BRANCH_REMOTE)
1627 cmd += _T(" -r");
1629 ret = Run(cmd, &output, nullptr, CP_UTF8);
1630 if (!ret)
1632 int pos = 0;
1633 CString one;
1634 while (pos >= 0)
1636 one = output.Tokenize(_T("\n"), pos);
1637 one.Trim(L" \r\n\t");
1638 if (one.Find(L" -> ") >= 0 || one.IsEmpty())
1639 continue; // skip something like: refs/origin/HEAD -> refs/origin/master
1640 if (one[0] == _T('*'))
1642 one = one.Mid(2);
1643 cur = one;
1645 if (one.Left(10) != _T("(no branch") && one.Left(15) != _T("(detached from "))
1647 if ((type & BRANCH_REMOTE) != 0 && (type & BRANCH_LOCAL) == 0)
1648 one = _T("remotes/") + one;
1649 list.push_back(one);
1653 else if (ret == 1 && IsInitRepos())
1654 return 0;
1657 if(type & BRANCH_FETCH_HEAD && !DerefFetchHead().IsEmpty())
1658 list.push_back(L"FETCH_HEAD");
1660 std::sort(list.begin(), list.end(), LogicalCompareBranchesPredicate);
1662 if (current && cur.Left(10) != _T("(no branch") && cur.Left(15) != _T("(detached from "))
1664 for (unsigned int i = 0; i < list.size(); ++i)
1666 if (list[i] == cur)
1668 *current = i;
1669 break;
1674 return ret;
1677 int CGit::GetRemoteList(STRING_VECTOR &list)
1679 if (this->m_IsUseLibGit2)
1681 CAutoRepository repo(GetGitRepository());
1682 if (!repo)
1683 return -1;
1685 CAutoStrArray remotes;
1686 if (git_remote_list(remotes, repo))
1687 return -1;
1689 for (size_t i = 0; i < remotes->count; ++i)
1691 CStringA remote(remotes->strings[i]);
1692 list.push_back(CUnicodeUtils::GetUnicode(remote));
1695 std::sort(list.begin(), list.end(), LogicalComparePredicate);
1697 return 0;
1699 else
1701 int ret;
1702 CString cmd, output;
1703 cmd=_T("git.exe remote");
1704 ret = Run(cmd, &output, NULL, CP_UTF8);
1705 if(!ret)
1707 int pos=0;
1708 CString one;
1709 while( pos>=0 )
1711 one=output.Tokenize(_T("\n"),pos);
1712 if (!one.IsEmpty())
1713 list.push_back(one);
1716 return ret;
1720 int CGit::GetRemoteTags(const CString& remote, STRING_VECTOR& list)
1722 if (UsingLibGit2(GIT_CMD_FETCH))
1724 CAutoRepository repo(GetGitRepository());
1725 if (!repo)
1726 return -1;
1728 CStringA remoteA = CUnicodeUtils::GetUTF8(remote);
1729 CAutoRemote gitremote;
1730 if (git_remote_lookup(gitremote.GetPointer(), repo, remoteA) < 0)
1731 return -1;
1733 git_remote_callbacks callbacks = GIT_REMOTE_CALLBACKS_INIT;
1734 callbacks.credentials = g_Git2CredCallback;
1735 callbacks.certificate_check = g_Git2CheckCertificateCallback;
1736 git_remote_set_callbacks(gitremote, &callbacks);
1737 if (git_remote_connect(gitremote, GIT_DIRECTION_FETCH) < 0)
1738 return -1;
1740 const git_remote_head** heads = nullptr;
1741 size_t size = 0;
1742 if (git_remote_ls(&heads, &size, gitremote) < 0)
1743 return -1;
1745 for (size_t i = 0; i < size; i++)
1747 CString ref = CUnicodeUtils::GetUnicode(heads[i]->name);
1748 CString shortname;
1749 if (!GetShortName(ref, shortname, _T("refs/tags/")))
1750 continue;
1751 // dot not include annotated tags twice; this works, because an annotated tag appears twice (one normal tag and one with ^{} at the end)
1752 if (shortname.Find(_T("^{}")) >= 1)
1753 continue;
1754 list.push_back(shortname);
1756 std::sort(list.begin(), list.end(), g_bSortTagsReversed ? LogicalCompareReversedPredicate : LogicalComparePredicate);
1757 return 0;
1760 CString cmd, out, err;
1761 cmd.Format(_T("git.exe ls-remote -t \"%s\""), remote);
1762 if (Run(cmd, &out, &err, CP_UTF8))
1764 MessageBox(NULL, err, _T("TortoiseGit"), MB_ICONERROR);
1765 return -1;
1768 int pos = 0;
1769 while (pos >= 0)
1771 CString one = out.Tokenize(_T("\n"), pos).Mid(51).Trim(); // sha1, tab + refs/tags/
1772 // dot not include annotated tags twice; this works, because an annotated tag appears twice (one normal tag and one with ^{} at the end)
1773 if (one.Find(_T("^{}")) >= 1)
1774 continue;
1775 if (!one.IsEmpty())
1776 list.push_back(one);
1778 std::sort(list.begin(), list.end(), g_bSortTagsReversed ? LogicalCompareReversedPredicate : LogicalComparePredicate);
1779 return 0;
1782 int CGit::DeleteRemoteRefs(const CString& sRemote, const STRING_VECTOR& list)
1784 if (UsingLibGit2(GIT_CMD_PUSH))
1786 CAutoRepository repo(GetGitRepository());
1787 if (!repo)
1788 return -1;
1790 CStringA remoteA = CUnicodeUtils::GetUTF8(sRemote);
1791 CAutoRemote remote;
1792 if (git_remote_lookup(remote.GetPointer(), repo, remoteA) < 0)
1793 return -1;
1795 git_remote_callbacks callbacks = GIT_REMOTE_CALLBACKS_INIT;
1796 callbacks.credentials = g_Git2CredCallback;
1797 callbacks.certificate_check = g_Git2CheckCertificateCallback;
1798 git_remote_set_callbacks(remote, &callbacks);
1799 if (git_remote_connect(remote, GIT_DIRECTION_PUSH) < 0)
1800 return -1;
1802 std::vector<CStringA> refspecs;
1803 for (auto ref : list)
1804 refspecs.push_back(CUnicodeUtils::GetUTF8(_T(":") + ref));
1806 std::vector<char*> vc;
1807 vc.reserve(refspecs.size());
1808 std::transform(refspecs.begin(), refspecs.end(), std::back_inserter(vc), [](CStringA& s) -> char* { return s.GetBuffer(); });
1809 git_strarray specs = { &vc[0], vc.size() };
1811 if (git_remote_push(remote, &specs, nullptr) < 0)
1812 return -1;
1814 else
1816 CMassiveGitTaskBase mgtPush(_T("push ") + sRemote, FALSE);
1817 for (auto ref : list)
1819 CString refspec = _T(":") + ref;
1820 mgtPush.AddFile(refspec);
1823 BOOL cancel = FALSE;
1824 mgtPush.Execute(cancel);
1827 return 0;
1830 int libgit2_addto_list_each_ref_fn(git_reference *ref, void *payload)
1832 STRING_VECTOR *list = (STRING_VECTOR*)payload;
1833 list->push_back(CUnicodeUtils::GetUnicode(git_reference_name(ref)));
1834 return 0;
1837 int CGit::GetRefList(STRING_VECTOR &list)
1839 if (this->m_IsUseLibGit2)
1841 CAutoRepository repo(GetGitRepository());
1842 if (!repo)
1843 return -1;
1845 if (git_reference_foreach(repo, libgit2_addto_list_each_ref_fn, &list))
1846 return -1;
1848 std::sort(list.begin(), list.end(), LogicalComparePredicate);
1850 return 0;
1852 else
1854 CString cmd, output;
1855 cmd=_T("git.exe show-ref -d");
1856 int ret = Run(cmd, &output, NULL, CP_UTF8);
1857 if(!ret)
1859 int pos=0;
1860 CString one;
1861 while( pos>=0 )
1863 one=output.Tokenize(_T("\n"),pos);
1864 int start=one.Find(_T(" "),0);
1865 if(start>0)
1867 CString name;
1868 name=one.Right(one.GetLength()-start-1);
1869 if (list.empty() || name != *list.crbegin() + _T("^{}"))
1870 list.push_back(name);
1873 std::sort(list.begin(), list.end(), LogicalComparePredicate);
1875 else if (ret == 1 && IsInitRepos())
1876 return 0;
1877 return ret;
1881 typedef struct map_each_ref_payload {
1882 git_repository * repo;
1883 MAP_HASH_NAME * map;
1884 } map_each_ref_payload;
1886 int libgit2_addto_map_each_ref_fn(git_reference *ref, void *payload)
1888 map_each_ref_payload *payloadContent = (map_each_ref_payload*)payload;
1890 CString str = CUnicodeUtils::GetUnicode(git_reference_name(ref));
1892 CAutoObject gitObject;
1893 if (git_revparse_single(gitObject.GetPointer(), payloadContent->repo, git_reference_name(ref)))
1894 return 1;
1896 if (git_object_type(gitObject) == GIT_OBJ_TAG)
1898 str += _T("^{}"); // deref tag
1899 CAutoObject derefedTag;
1900 if (git_object_peel(derefedTag.GetPointer(), gitObject, GIT_OBJ_ANY))
1901 return 1;
1902 gitObject.Swap(derefedTag);
1905 const git_oid * oid = git_object_id(gitObject);
1906 if (oid == NULL)
1907 return 1;
1909 CGitHash hash((char *)oid->id);
1910 (*payloadContent->map)[hash].push_back(str);
1912 return 0;
1914 int CGit::GetMapHashToFriendName(git_repository* repo, MAP_HASH_NAME &map)
1916 ATLASSERT(repo);
1918 map_each_ref_payload payloadContent = { repo, &map };
1920 if (git_reference_foreach(repo, libgit2_addto_map_each_ref_fn, &payloadContent))
1921 return -1;
1923 for (auto it = map.begin(); it != map.end(); ++it)
1925 std::sort(it->second.begin(), it->second.end());
1928 return 0;
1931 int CGit::GetMapHashToFriendName(MAP_HASH_NAME &map)
1933 if (this->m_IsUseLibGit2)
1935 CAutoRepository repo(GetGitRepository());
1936 if (!repo)
1937 return -1;
1939 return GetMapHashToFriendName(repo, map);
1941 else
1943 CString cmd, output;
1944 cmd=_T("git.exe show-ref -d");
1945 int ret = Run(cmd, &output, NULL, CP_UTF8);
1946 if(!ret)
1948 int pos=0;
1949 CString one;
1950 while( pos>=0 )
1952 one=output.Tokenize(_T("\n"),pos);
1953 int start=one.Find(_T(" "),0);
1954 if(start>0)
1956 CString name;
1957 name=one.Right(one.GetLength()-start-1);
1959 CString hash;
1960 hash=one.Left(start);
1962 map[CGitHash(hash)].push_back(name);
1966 else if (ret == 1 && IsInitRepos())
1967 return 0;
1968 return ret;
1972 int CGit::GetBranchDescriptions(MAP_STRING_STRING& map)
1974 CAutoConfig config(true);
1975 if (git_config_add_file_ondisk(config, CGit::GetGitPathStringA(GetGitLocalConfig()), GIT_CONFIG_LEVEL_LOCAL, FALSE) < 0)
1976 return -1;
1977 return git_config_foreach_match(config, "^branch\\..*\\.description$", [](const git_config_entry* entry, void* data)
1979 MAP_STRING_STRING* descriptions = (MAP_STRING_STRING*)data;
1980 CString key = CUnicodeUtils::GetUnicode(entry->name);
1981 key = key.Mid(7, key.GetLength() - 7 - 12); // 7: branch., 12: .description
1982 descriptions->insert(std::make_pair(key, CUnicodeUtils::GetUnicode(entry->value)));
1983 return 0;
1984 }, &map);
1987 static void SetLibGit2SearchPath(int level, const CString &value)
1989 CStringA valueA = CUnicodeUtils::GetMulti(value, CP_UTF8);
1990 git_libgit2_opts(GIT_OPT_SET_SEARCH_PATH, level, valueA);
1993 static void SetLibGit2TemplatePath(const CString &value)
1995 CStringA valueA = CUnicodeUtils::GetMulti(value, CP_UTF8);
1996 git_libgit2_opts(GIT_OPT_SET_TEMPLATE_PATH, valueA);
1999 int CGit::FindAndSetGitExePath(BOOL bFallback)
2001 CRegString msysdir = CRegString(REG_MSYSGIT_PATH, _T(""), FALSE);
2002 CString str = msysdir;
2003 if (!str.IsEmpty() && FileExists(str + _T("\\git.exe")))
2005 CGit::ms_LastMsysGitDir = str;
2006 return TRUE;
2009 CTraceToOutputDebugString::Instance()(_T(__FUNCTION__) _T(": git.exe not exists: %s\n"), CGit::ms_LastMsysGitDir);
2010 if (!bFallback)
2011 return FALSE;
2013 // first, search PATH if git/bin directory is already present
2014 if (FindGitPath())
2016 CTraceToOutputDebugString::Instance()(_T(__FUNCTION__) _T(": FindGitPath() => %s\n"), CGit::ms_LastMsysGitDir);
2017 msysdir = CGit::ms_LastMsysGitDir;
2018 msysdir.write();
2019 return TRUE;
2022 CRegString msyslocalinstalldir = CRegString(REG_MSYSGIT_INSTALL_LOCAL, _T(""), FALSE, HKEY_CURRENT_USER);
2023 str = msyslocalinstalldir;
2024 str.TrimRight(_T("\\"));
2025 #ifdef _WIN64
2026 if (str.IsEmpty())
2028 CRegString msysinstalldir = CRegString(REG_MSYSGIT_INSTALL_LOCAL, _T(""), FALSE, HKEY_LOCAL_MACHINE);
2029 str = msysinstalldir;
2030 str.TrimRight(_T("\\"));
2032 #endif
2033 if (str.IsEmpty())
2035 CRegString msysinstalldir = CRegString(REG_MSYSGIT_INSTALL, _T(""), FALSE, HKEY_LOCAL_MACHINE);
2036 str = msysinstalldir;
2037 str.TrimRight(_T("\\"));
2039 if (!str.IsEmpty())
2041 if (FileExists(str + _T("\\bin\\git.exe")))
2042 str += _T("\\bin");
2043 else if (FileExists(str + _T("\\cmd\\git.exe"))) // only needed for older Git for Windows 2.x packages
2044 str += _T("\\cmd");
2045 else
2047 CTraceToOutputDebugString::Instance()(_T(__FUNCTION__) _T(": Git for Windows installation found, but git.exe not exists in %s\n"), str);
2048 return FALSE;
2050 msysdir = str;
2051 CGit::ms_LastMsysGitDir = str;
2052 msysdir.write();
2053 return TRUE;
2056 CTraceToOutputDebugString::Instance()(_T(__FUNCTION__) _T(": Found no git.exe\n"));
2057 return FALSE;
2060 BOOL CGit::CheckMsysGitDir(BOOL bFallback)
2062 if (m_bInitialized)
2064 return TRUE;
2067 CTraceToOutputDebugString::Instance()(_T(__FUNCTION__) _T(": CheckMsysGitDir(%d)\n"), bFallback);
2068 this->m_Environment.clear();
2069 m_Environment.CopyProcessEnvironment();
2070 m_Environment.SetEnv(_T("GIT_DIR"), nullptr); // Remove %GIT_DIR% before executing git.exe
2072 // set HOME if not set already
2073 size_t homesize;
2074 _tgetenv_s(&homesize, NULL, 0, _T("HOME"));
2075 if (!homesize)
2076 m_Environment.SetEnv(_T("HOME"), GetHomeDirectory());
2078 //setup ssh client
2079 CString sshclient=CRegString(_T("Software\\TortoiseGit\\SSH"));
2080 if (sshclient.IsEmpty())
2081 sshclient = CRegString(_T("Software\\TortoiseGit\\SSH"), _T(""), FALSE, HKEY_LOCAL_MACHINE);
2083 if(!sshclient.IsEmpty())
2085 m_Environment.SetEnv(_T("GIT_SSH"), sshclient);
2086 m_Environment.SetEnv(_T("SVN_SSH"), sshclient);
2088 else
2090 TCHAR sPlink[MAX_PATH] = {0};
2091 GetModuleFileName(NULL, sPlink, _countof(sPlink));
2092 LPTSTR ptr = _tcsrchr(sPlink, _T('\\'));
2093 if (ptr) {
2094 _tcscpy_s(ptr + 1, MAX_PATH - (ptr - sPlink + 1), _T("TortoiseGitPLink.exe"));
2095 m_Environment.SetEnv(_T("GIT_SSH"), sPlink);
2096 m_Environment.SetEnv(_T("SVN_SSH"), sPlink);
2101 TCHAR sAskPass[MAX_PATH] = {0};
2102 GetModuleFileName(NULL, sAskPass, _countof(sAskPass));
2103 LPTSTR ptr = _tcsrchr(sAskPass, _T('\\'));
2104 if (ptr)
2106 _tcscpy_s(ptr + 1, MAX_PATH - (ptr - sAskPass + 1), _T("SshAskPass.exe"));
2107 m_Environment.SetEnv(_T("DISPLAY"),_T(":9999"));
2108 m_Environment.SetEnv(_T("SSH_ASKPASS"),sAskPass);
2109 m_Environment.SetEnv(_T("GIT_ASKPASS"),sAskPass);
2113 if (!FindAndSetGitExePath(bFallback))
2114 return FALSE;
2116 CString msysGitDir;
2117 PathCanonicalize(CStrBuf(msysGitDir, MAX_PATH), CGit::ms_LastMsysGitDir + _T("\\..\\"));
2118 static CString prefixes[] = { L"mingw64\\etc", L"mingw32\\etc", L"etc" };
2119 static int prefixes_len[] = { 8, 8, 0 };
2120 for (int i = 0; i < _countof(prefixes); ++i)
2122 #ifndef _WIN64
2123 if (i == 0)
2124 continue;
2125 #endif
2126 if (PathIsDirectory(msysGitDir + prefixes[i])) {
2127 msysGitDir += prefixes[i].Left(prefixes_len[i]);
2128 break;
2131 CGit::ms_MsysGitRootDir = msysGitDir;
2133 if ((CString)CRegString(REG_SYSTEM_GITCONFIGPATH, _T(""), FALSE) != g_Git.GetGitSystemConfig())
2134 CRegString(REG_SYSTEM_GITCONFIGPATH, _T(""), FALSE) = g_Git.GetGitSystemConfig();
2136 CTraceToOutputDebugString::Instance()(_T(__FUNCTION__) _T(": ms_LastMsysGitDir = %s\n"), CGit::ms_LastMsysGitDir);
2137 CTraceToOutputDebugString::Instance()(_T(__FUNCTION__) _T(": ms_MsysGitRootDir = %s\n"), CGit::ms_MsysGitRootDir);
2138 CTraceToOutputDebugString::Instance()(_T(__FUNCTION__) _T(": System config = %s\n"), g_Git.GetGitSystemConfig());
2140 // Configure libgit2 search paths
2141 SetLibGit2SearchPath(GIT_CONFIG_LEVEL_SYSTEM, CTGitPath(g_Git.GetGitSystemConfig()).GetContainingDirectory().GetWinPathString());
2142 SetLibGit2SearchPath(GIT_CONFIG_LEVEL_GLOBAL, g_Git.GetHomeDirectory());
2143 SetLibGit2SearchPath(GIT_CONFIG_LEVEL_XDG, g_Git.GetGitGlobalXDGConfigPath());
2144 static git_smart_subtransport_definition ssh_wintunnel_subtransport_definition = { [](git_smart_subtransport **out, git_transport* owner) -> int { return git_smart_subtransport_ssh_wintunnel(out, owner, FindExecutableOnPath(g_Git.m_Environment.GetEnv(_T("GIT_SSH")), g_Git.m_Environment.GetEnv(_T("PATH"))), g_Git.m_Environment); }, 0 };
2145 git_transport_register("ssh", git_transport_smart, &ssh_wintunnel_subtransport_definition);
2146 if (!ms_bCygwinGit)
2147 SetLibGit2TemplatePath(CGit::ms_MsysGitRootDir + _T("share\\git-core\\templates"));
2148 else
2149 SetLibGit2TemplatePath(CGit::ms_MsysGitRootDir + _T("usr\\share\\git-core\\templates"));
2151 m_Environment.AddToPath(CGit::ms_LastMsysGitDir);
2152 m_Environment.AddToPath((CString)CRegString(REG_MSYSGIT_EXTRA_PATH, _T(""), FALSE));
2154 #if !defined(TGITCACHE) && !defined(TORTOISESHELL)
2155 // register filter only once
2156 if (!git_filter_lookup("filter"))
2158 static CString binDirPrefixes[] = { L"", L"\\..\\bin", L"\\..\\usr\\bin" };
2159 CString sh;
2160 for (const CString& binDirPrefix : binDirPrefixes)
2162 CString possibleShExe = CGit::ms_LastMsysGitDir + binDirPrefix + L"\\sh.exe";
2163 if (PathFileExists(possibleShExe))
2165 CString temp;
2166 PathCanonicalize(CStrBuf(temp, MAX_PATH), possibleShExe);
2167 sh.Format(L"\"%s\"", temp);
2168 // we need to put the usr\bin folder on the path for Git for Windows based on msys2
2169 m_Environment.AddToPath(temp.Left(temp.GetLength() - 7)); // 7 = len("\\sh.exe")
2170 break;
2173 if (git_filter_register("filter", git_filter_filter_new(sh, m_Environment), GIT_FILTER_DRIVER_PRIORITY))
2174 return FALSE;
2176 #endif
2178 m_bInitialized = TRUE;
2179 return true;
2182 CString CGit::GetHomeDirectory() const
2184 const wchar_t * homeDir = wget_windows_home_directory();
2185 return CString(homeDir, (int)wcslen(homeDir));
2188 CString CGit::GetGitLocalConfig() const
2190 CString path;
2191 GitAdminDir::GetAdminDirPath(m_CurrentDir, path);
2192 path += _T("config");
2193 return path;
2196 CStringA CGit::GetGitPathStringA(const CString &path)
2198 return CUnicodeUtils::GetUTF8(CTGitPath(path).GetGitPathString());
2201 CString CGit::GetGitGlobalConfig() const
2203 return g_Git.GetHomeDirectory() + _T("\\.gitconfig");
2206 CString CGit::GetGitGlobalXDGConfigPath() const
2208 return g_Git.GetHomeDirectory() + _T("\\.config\\git");
2211 CString CGit::GetGitGlobalXDGConfig() const
2213 return g_Git.GetGitGlobalXDGConfigPath() + _T("\\config");
2216 CString CGit::GetGitSystemConfig() const
2218 const wchar_t * systemConfig = wget_msysgit_etc();
2219 return CString(systemConfig, (int)wcslen(systemConfig));
2222 BOOL CGit::CheckCleanWorkTree(bool stagedOk /* false */)
2224 if (UsingLibGit2(GIT_CMD_CHECK_CLEAN_WT))
2226 CAutoRepository repo = GetGitRepository();
2227 if (!repo)
2228 return FALSE;
2230 if (git_repository_head_unborn(repo))
2231 return FALSE;
2233 git_status_options statusopt = GIT_STATUS_OPTIONS_INIT;
2234 statusopt.show = stagedOk ? GIT_STATUS_SHOW_WORKDIR_ONLY : GIT_STATUS_SHOW_INDEX_AND_WORKDIR;
2235 statusopt.flags = GIT_STATUS_OPT_UPDATE_INDEX | GIT_STATUS_OPT_EXCLUDE_SUBMODULES;
2237 CAutoStatusList status;
2238 if (git_status_list_new(status.GetPointer(), repo, &statusopt))
2239 return FALSE;
2241 return (0 == git_status_list_entrycount(status));
2244 CString out;
2245 CString cmd;
2246 cmd=_T("git.exe rev-parse --verify HEAD");
2248 if(Run(cmd,&out,CP_UTF8))
2249 return FALSE;
2251 cmd=_T("git.exe update-index --ignore-submodules --refresh");
2252 if(Run(cmd,&out,CP_UTF8))
2253 return FALSE;
2255 cmd=_T("git.exe diff-files --quiet --ignore-submodules");
2256 if(Run(cmd,&out,CP_UTF8))
2257 return FALSE;
2259 cmd = _T("git.exe diff-index --cached --quiet HEAD --ignore-submodules --");
2260 if (!stagedOk && Run(cmd, &out, CP_UTF8))
2261 return FALSE;
2263 return TRUE;
2265 int CGit::Revert(const CString& commit, const CTGitPathList &list, CString& err)
2267 int ret;
2268 for (int i = 0; i < list.GetCount(); ++i)
2270 ret = Revert(commit, (CTGitPath&)list[i], err);
2271 if(ret)
2272 return ret;
2274 return 0;
2276 int CGit::Revert(const CString& commit, const CTGitPath &path, CString& err)
2278 CString cmd;
2280 if(path.m_Action & CTGitPath::LOGACTIONS_REPLACED && !path.GetGitOldPathString().IsEmpty())
2282 if (CTGitPath(path.GetGitOldPathString()).IsDirectory())
2284 err.Format(_T("Cannot revert renaming of \"%s\". A directory with the old name \"%s\" exists."), path.GetGitPathString(), path.GetGitOldPathString());
2285 return -1;
2287 CString force;
2288 // if the filenames only differ in case, we have to pass "-f"
2289 if (path.GetGitPathString().CompareNoCase(path.GetGitOldPathString()) == 0)
2290 force = _T("-f ");
2291 cmd.Format(_T("git.exe mv %s-- \"%s\" \"%s\""), force, path.GetGitPathString(), path.GetGitOldPathString());
2292 if (Run(cmd, &err, CP_UTF8))
2293 return -1;
2295 cmd.Format(_T("git.exe checkout %s -f -- \"%s\""), commit, path.GetGitOldPathString());
2296 if (Run(cmd, &err, CP_UTF8))
2297 return -1;
2299 else if(path.m_Action & CTGitPath::LOGACTIONS_ADDED)
2300 { //To init git repository, there are not HEAD, so we can use git reset command
2301 cmd.Format(_T("git.exe rm -f --cached -- \"%s\""),path.GetGitPathString());
2303 if (Run(cmd, &err, CP_UTF8))
2304 return -1;
2306 else
2308 cmd.Format(_T("git.exe checkout %s -f -- \"%s\""), commit, path.GetGitPathString());
2309 if (Run(cmd, &err, CP_UTF8))
2310 return -1;
2313 if (path.m_Action & CTGitPath::LOGACTIONS_DELETED)
2315 cmd.Format(_T("git.exe add -f -- \"%s\""), path.GetGitPathString());
2316 if (Run(cmd, &err, CP_UTF8))
2317 return -1;
2320 return 0;
2323 int CGit::HasWorkingTreeConflicts(git_repository* repo)
2325 ATLASSERT(repo);
2327 CAutoIndex index;
2328 if (git_repository_index(index.GetPointer(), repo))
2329 return -1;
2331 return git_index_has_conflicts(index);
2334 int CGit::HasWorkingTreeConflicts()
2336 if (UsingLibGit2(GIT_CMD_CHECKCONFLICTS))
2338 CAutoRepository repo(GetGitRepository());
2339 if (!repo)
2340 return -1;
2342 return HasWorkingTreeConflicts(repo);
2345 CString cmd = _T("git.exe ls-files -u -t -z");
2347 CString output;
2348 if (Run(cmd, &output, &gitLastErr, CP_UTF8))
2349 return -1;
2351 return output.IsEmpty() ? 0 : 1;
2354 bool CGit::IsFastForward(const CString &from, const CString &to, CGitHash * commonAncestor)
2356 if (UsingLibGit2(GIT_CMD_MERGE_BASE))
2358 CAutoRepository repo(GetGitRepository());
2359 if (!repo)
2360 return false;
2362 CGitHash fromHash, toHash, baseHash;
2363 if (GetHash(repo, toHash, FixBranchName(to)))
2364 return false;
2366 if (GetHash(repo, fromHash, FixBranchName(from)))
2367 return false;
2369 git_oid baseOid;
2370 if (git_merge_base(&baseOid, repo, (const git_oid*)toHash.m_hash, (const git_oid*)fromHash.m_hash))
2371 return false;
2373 baseHash = baseOid.id;
2375 if (commonAncestor)
2376 *commonAncestor = baseHash;
2378 return fromHash == baseHash;
2380 // else
2381 CString base;
2382 CGitHash basehash,hash;
2383 CString cmd;
2384 cmd.Format(_T("git.exe merge-base %s %s"), FixBranchName(to), FixBranchName(from));
2386 if (Run(cmd, &base, &gitLastErr, CP_UTF8))
2388 return false;
2390 basehash = base.Trim();
2392 GetHash(hash, from);
2394 if (commonAncestor)
2395 *commonAncestor = basehash;
2397 return hash == basehash;
2400 unsigned int CGit::Hash2int(const CGitHash &hash)
2402 int ret=0;
2403 for (int i = 0; i < 4; ++i)
2405 ret = ret << 8;
2406 ret |= hash.m_hash[i];
2408 return ret;
2411 int CGit::RefreshGitIndex()
2413 if(g_Git.m_IsUseGitDLL)
2415 CAutoLocker lock(g_Git.m_critGitDllSec);
2418 return [] { return git_run_cmd("update-index","update-index -q --refresh"); }();
2420 }catch(...)
2422 return -1;
2426 else
2428 CString cmd,output;
2429 cmd=_T("git.exe update-index --refresh");
2430 return Run(cmd, &output, CP_UTF8);
2434 int CGit::GetOneFile(const CString &Refname, const CTGitPath &path, const CString &outputfile)
2436 if (UsingLibGit2(GIT_CMD_GETONEFILE))
2438 CAutoRepository repo(GetGitRepository());
2439 if (!repo)
2440 return -1;
2442 CGitHash hash;
2443 if (GetHash(repo, hash, Refname))
2444 return -1;
2446 CAutoCommit commit;
2447 if (git_commit_lookup(commit.GetPointer(), repo, (const git_oid *)hash.m_hash))
2448 return -1;
2450 CAutoTree tree;
2451 if (git_commit_tree(tree.GetPointer(), commit))
2452 return -1;
2454 CAutoTreeEntry entry;
2455 if (git_tree_entry_bypath(entry.GetPointer(), tree, CUnicodeUtils::GetUTF8(path.GetGitPathString())))
2456 return -1;
2458 CAutoBlob blob;
2459 if (git_tree_entry_to_object((git_object**)blob.GetPointer(), repo, entry))
2460 return -1;
2462 FILE *file = nullptr;
2463 _tfopen_s(&file, outputfile, _T("wb"));
2464 if (file == nullptr)
2466 giterr_set_str(GITERR_NONE, "Could not create file.");
2467 return -1;
2469 CAutoBuf buf;
2470 if (git_blob_filtered_content(buf, blob, CUnicodeUtils::GetUTF8(path.GetGitPathString()), 0))
2472 fclose(file);
2473 return -1;
2475 if (fwrite(buf->ptr, sizeof(char), buf->size, file) != buf->size)
2477 giterr_set_str(GITERR_OS, "Could not write to file.");
2478 fclose(file);
2480 return -1;
2482 fclose(file);
2484 return 0;
2486 else if (g_Git.m_IsUseGitDLL)
2488 CAutoLocker lock(g_Git.m_critGitDllSec);
2491 g_Git.CheckAndInitDll();
2492 CStringA ref, patha, outa;
2493 ref = CUnicodeUtils::GetMulti(Refname, CP_UTF8);
2494 patha = CUnicodeUtils::GetMulti(path.GetGitPathString(), CP_UTF8);
2495 outa = CUnicodeUtils::GetMulti(outputfile, CP_UTF8);
2496 ::DeleteFile(outputfile);
2497 int ret = git_checkout_file(ref, patha, outa.GetBuffer());
2498 outa.ReleaseBuffer();
2499 return ret;
2502 catch (const char * msg)
2504 gitLastErr = L"gitdll.dll reports: " + CString(msg);
2505 return -1;
2507 catch (...)
2509 gitLastErr = L"An unknown gitdll.dll error occurred.";
2510 return -1;
2513 else
2515 CString cmd;
2516 cmd.Format(_T("git.exe cat-file -p %s:\"%s\""), Refname, path.GetGitPathString());
2517 return RunLogFile(cmd, outputfile, &gitLastErr);
2521 void CEnvironment::clear()
2523 __super::clear();
2526 bool CEnvironment::empty()
2528 return size() < 3; // three is minimum for an empty environment with an empty key and empty value: "=\0\0"
2531 CEnvironment::operator LPTSTR()
2533 if (empty())
2534 return nullptr;
2535 return &__super::at(0);
2538 void CEnvironment::CopyProcessEnvironment()
2540 if (!empty())
2541 pop_back();
2542 TCHAR *porig = GetEnvironmentStrings();
2543 TCHAR *p = porig;
2544 while(*p !=0 || *(p+1) !=0)
2545 this->push_back(*p++);
2547 push_back(_T('\0'));
2548 push_back(_T('\0'));
2550 FreeEnvironmentStrings(porig);
2553 CString CEnvironment::GetEnv(const TCHAR *name)
2555 CString str;
2556 for (size_t i = 0; i < size(); ++i)
2558 str = &(*this)[i];
2559 int start =0;
2560 CString sname = str.Tokenize(_T("="),start);
2561 if(sname.CompareNoCase(name) == 0)
2563 return &(*this)[i+start];
2565 i+=str.GetLength();
2567 return _T("");
2570 void CEnvironment::SetEnv(const TCHAR *name, const TCHAR* value)
2572 unsigned int i;
2573 for (i = 0; i < size(); ++i)
2575 CString str = &(*this)[i];
2576 int start =0;
2577 CString sname = str.Tokenize(_T("="),start);
2578 if(sname.CompareNoCase(name) == 0)
2580 break;
2582 i+=str.GetLength();
2585 if(i == size())
2587 if (!value) // as we haven't found the variable we want to remove, just return
2588 return;
2589 if (i == 0) // make inserting into an empty environment work
2591 this->push_back(_T('\0'));
2592 ++i;
2594 i -= 1; // roll back terminate \0\0
2595 this->push_back(_T('\0'));
2598 CEnvironment::iterator it;
2599 it=this->begin();
2600 it += i;
2602 while(*it && i<size())
2604 this->erase(it);
2605 it=this->begin();
2606 it += i;
2609 if (value == nullptr) // remove the variable
2611 this->erase(it);
2612 return;
2615 while(*name)
2617 this->insert(it,*name++);
2618 ++i;
2619 it= begin()+i;
2622 this->insert(it, _T('='));
2623 ++i;
2624 it= begin()+i;
2626 while(*value)
2628 this->insert(it,*value++);
2629 ++i;
2630 it= begin()+i;
2635 void CEnvironment::AddToPath(CString value)
2637 value.TrimRight(L"\\");
2638 if (value.IsEmpty())
2639 return;
2641 CString path = GetEnv(L"PATH").TrimRight(L";") + L";";
2643 // do not double add paths to %PATH%
2644 if (path.Find(value + L";") >= 0 || path.Find(value + L"\\;") >= 0)
2645 return;
2647 path += value;
2649 SetEnv(L"PATH", path);
2652 int CGit::GetGitEncode(TCHAR* configkey)
2654 CString str=GetConfigValue(configkey);
2656 if(str.IsEmpty())
2657 return CP_UTF8;
2659 return CUnicodeUtils::GetCPCode(str);
2663 int CGit::GetDiffPath(CTGitPathList *PathList, CGitHash *hash1, CGitHash *hash2, char *arg)
2665 GIT_FILE file=0;
2666 int ret=0;
2667 GIT_DIFF diff=0;
2669 CAutoLocker lock(g_Git.m_critGitDllSec);
2673 if(!arg)
2674 diff = GetGitDiff();
2675 else
2676 git_open_diff(&diff, arg);
2678 catch (char* e)
2680 MessageBox(nullptr, _T("Could not get diff.\nlibgit reported:\n") + CString(e), _T("TortoiseGit"), MB_OK);
2681 return -1;
2684 if(diff ==NULL)
2685 return -1;
2687 bool isStat = 0;
2688 if(arg == NULL)
2689 isStat = true;
2690 else
2691 isStat = !!strstr(arg, "stat");
2693 int count=0;
2695 if(hash2 == NULL)
2696 ret = git_root_diff(diff, hash1->m_hash, &file, &count,isStat);
2697 else
2698 ret = git_do_diff(diff,hash2->m_hash,hash1->m_hash,&file,&count,isStat);
2700 if(ret)
2701 return -1;
2703 CTGitPath path;
2704 CString strnewname;
2705 CString stroldname;
2707 for (int j = 0; j < count; ++j)
2709 path.Reset();
2710 char *newname;
2711 char *oldname;
2713 strnewname.Empty();
2714 stroldname.Empty();
2716 int mode=0,IsBin=0,inc=0,dec=0;
2717 git_get_diff_file(diff,file,j,&newname,&oldname,
2718 &mode,&IsBin,&inc,&dec);
2720 StringAppend(&strnewname, (BYTE*)newname, CP_UTF8);
2721 StringAppend(&stroldname, (BYTE*)oldname, CP_UTF8);
2723 path.SetFromGit(strnewname,&stroldname);
2724 path.ParserAction((BYTE)mode);
2726 if(IsBin)
2728 path.m_StatAdd=_T("-");
2729 path.m_StatDel=_T("-");
2731 else
2733 path.m_StatAdd.Format(_T("%d"),inc);
2734 path.m_StatDel.Format(_T("%d"),dec);
2736 PathList->AddPath(path);
2738 git_diff_flush(diff);
2740 if(arg)
2741 git_close_diff(diff);
2743 return 0;
2746 int CGit::GetShortHASHLength() const
2748 return 7;
2751 CString CGit::GetShortName(const CString& ref, REF_TYPE *out_type)
2753 CString str=ref;
2754 CString shortname;
2755 REF_TYPE type = CGit::UNKNOWN;
2757 if (CGit::GetShortName(str, shortname, _T("refs/heads/")))
2759 type = CGit::LOCAL_BRANCH;
2762 else if (CGit::GetShortName(str, shortname, _T("refs/remotes/")))
2764 type = CGit::REMOTE_BRANCH;
2766 else if (CGit::GetShortName(str, shortname, _T("refs/tags/")))
2768 type = CGit::TAG;
2770 else if (CGit::GetShortName(str, shortname, _T("refs/stash")))
2772 type = CGit::STASH;
2773 shortname=_T("stash");
2775 else if (CGit::GetShortName(str, shortname, _T("refs/bisect/")))
2777 if (shortname.Find(_T("good")) == 0)
2779 type = CGit::BISECT_GOOD;
2780 shortname = _T("good");
2783 if (shortname.Find(_T("bad")) == 0)
2785 type = CGit::BISECT_BAD;
2786 shortname = _T("bad");
2789 else if (CGit::GetShortName(str, shortname, _T("refs/notes/")))
2791 type = CGit::NOTES;
2793 else if (CGit::GetShortName(str, shortname, _T("refs/")))
2795 type = CGit::UNKNOWN;
2797 else
2799 type = CGit::UNKNOWN;
2800 shortname = ref;
2803 if(out_type)
2804 *out_type = type;
2806 return shortname;
2809 bool CGit::UsingLibGit2(LIBGIT2_CMD cmd) const
2811 return m_IsUseLibGit2 && ((1 << cmd) & m_IsUseLibGit2_mask) ? true : false;
2814 void CGit::SetGit2CredentialCallback(void* callback)
2816 g_Git2CredCallback = (git_cred_acquire_cb)callback;
2819 void CGit::SetGit2CertificateCheckCertificate(void* callback)
2821 g_Git2CheckCertificateCallback = (git_transport_certificate_check_cb)callback;
2824 CString CGit::GetUnifiedDiffCmd(const CTGitPath& path, const git_revnum_t& rev1, const git_revnum_t& rev2, bool bMerge, bool bCombine, int diffContext)
2826 CString cmd;
2827 if (rev2 == GitRev::GetWorkingCopy())
2828 cmd.Format(_T("git.exe diff --stat -p %s --"), rev1);
2829 else if (rev1 == GitRev::GetWorkingCopy())
2830 cmd.Format(_T("git.exe diff -R --stat -p %s --"), rev2);
2831 else
2833 CString merge;
2834 if (bMerge)
2835 merge += _T(" -m");
2837 if (bCombine)
2838 merge += _T(" -c");
2840 CString unified;
2841 if (diffContext >= 0)
2842 unified.Format(_T(" --unified=%d"), diffContext);
2843 cmd.Format(_T("git.exe diff-tree -r -p %s %s --stat %s %s --"), merge, unified, rev1, rev2);
2846 if (!path.IsEmpty())
2848 cmd += _T(" \"");
2849 cmd += path.GetGitPathString();
2850 cmd += _T("\"");
2853 return cmd;
2856 static void UnifiedDiffStatToFile(const git_buf* text, void* payload)
2858 ATLASSERT(payload && text);
2859 fwrite(text->ptr, 1, text->size, (FILE *)payload);
2860 fwrite("\n", 1, 1, (FILE *)payload);
2863 static int UnifiedDiffToFile(const git_diff_delta * /* delta */, const git_diff_hunk * /* hunk */, const git_diff_line * line, void *payload)
2865 ATLASSERT(payload && line);
2866 if (line->origin == GIT_DIFF_LINE_CONTEXT || line->origin == GIT_DIFF_LINE_ADDITION || line->origin == GIT_DIFF_LINE_DELETION)
2867 fwrite(&line->origin, 1, 1, (FILE *)payload);
2868 fwrite(line->content, 1, line->content_len, (FILE *)payload);
2869 return 0;
2872 static int resolve_to_tree(git_repository *repo, const char *identifier, git_tree **tree)
2874 ATLASSERT(repo && identifier && tree);
2876 /* try to resolve identifier */
2877 CAutoObject obj;
2878 if (git_revparse_single(obj.GetPointer(), repo, identifier))
2879 return -1;
2881 if (obj == nullptr)
2882 return GIT_ENOTFOUND;
2884 int err = 0;
2885 switch (git_object_type(obj))
2887 case GIT_OBJ_TREE:
2888 *tree = (git_tree *)obj.Detach();
2889 break;
2890 case GIT_OBJ_COMMIT:
2891 err = git_commit_tree(tree, (git_commit *)(git_object*)obj);
2892 break;
2893 default:
2894 err = GIT_ENOTFOUND;
2897 return err;
2900 /* use libgit2 get unified diff */
2901 static int GetUnifiedDiffLibGit2(const CTGitPath& path, const git_revnum_t& revOld, const git_revnum_t& revNew, std::function<void(const git_buf*, void*)> statCallback, git_diff_line_cb callback, void *data, bool /* bMerge */)
2903 CStringA tree1 = CUnicodeUtils::GetMulti(revNew, CP_UTF8);
2904 CStringA tree2 = CUnicodeUtils::GetMulti(revOld, CP_UTF8);
2906 CAutoRepository repo(g_Git.GetGitRepository());
2907 if (!repo)
2908 return -1;
2910 int isHeadOrphan = git_repository_head_unborn(repo);
2911 if (isHeadOrphan == 1)
2912 return 0;
2913 else if (isHeadOrphan != 0)
2914 return -1;
2916 git_diff_options opts = GIT_DIFF_OPTIONS_INIT;
2917 CStringA pathA = CUnicodeUtils::GetMulti(path.GetGitPathString(), CP_UTF8);
2918 char *buf = pathA.GetBuffer();
2919 if (!pathA.IsEmpty())
2921 opts.pathspec.strings = &buf;
2922 opts.pathspec.count = 1;
2924 CAutoDiff diff;
2926 if (revNew == GitRev::GetWorkingCopy() || revOld == GitRev::GetWorkingCopy())
2928 CAutoTree t1;
2929 CAutoDiff diff2;
2931 if (revNew != GitRev::GetWorkingCopy() && resolve_to_tree(repo, tree1, t1.GetPointer()))
2932 return -1;
2934 if (revOld != GitRev::GetWorkingCopy() && resolve_to_tree(repo, tree2, t1.GetPointer()))
2935 return -1;
2937 if (git_diff_tree_to_index(diff.GetPointer(), repo, t1, nullptr, &opts))
2938 return -1;
2940 if (git_diff_index_to_workdir(diff2.GetPointer(), repo, nullptr, &opts))
2941 return -1;
2943 if (git_diff_merge(diff, diff2))
2944 return -1;
2946 else
2948 if (tree1.IsEmpty() && tree2.IsEmpty())
2949 return -1;
2951 if (tree1.IsEmpty())
2953 tree1 = tree2;
2954 tree2.Empty();
2957 CAutoTree t1;
2958 CAutoTree t2;
2959 if (!tree1.IsEmpty() && resolve_to_tree(repo, tree1, t1.GetPointer()))
2960 return -1;
2962 if (tree2.IsEmpty())
2964 /* don't check return value, there are not parent commit at first commit*/
2965 resolve_to_tree(repo, tree1 + "~1", t2.GetPointer());
2967 else if (resolve_to_tree(repo, tree2, t2.GetPointer()))
2968 return -1;
2969 if (git_diff_tree_to_tree(diff.GetPointer(), repo, t2, t1, &opts))
2970 return -1;
2973 CAutoDiffStats stats;
2974 if (git_diff_get_stats(stats.GetPointer(), diff))
2975 return -1;
2976 CAutoBuf statBuf;
2977 if (git_diff_stats_to_buf(statBuf, stats, GIT_DIFF_STATS_FULL, 0))
2978 return -1;
2979 statCallback(statBuf, data);
2981 for (size_t i = 0; i < git_diff_num_deltas(diff); ++i)
2983 CAutoPatch patch;
2984 if (git_patch_from_diff(patch.GetPointer(), diff, i))
2985 return -1;
2987 if (git_patch_print(patch, callback, data))
2988 return -1;
2991 pathA.ReleaseBuffer();
2993 return 0;
2996 int CGit::GetUnifiedDiff(const CTGitPath& path, const git_revnum_t& rev1, const git_revnum_t& rev2, CString patchfile, bool bMerge, bool bCombine, int diffContext)
2998 if (UsingLibGit2(GIT_CMD_DIFF))
3000 FILE *file = nullptr;
3001 _tfopen_s(&file, patchfile, _T("w"));
3002 if (file == nullptr)
3003 return -1;
3004 int ret = GetUnifiedDiffLibGit2(path, rev1, rev2, UnifiedDiffStatToFile, UnifiedDiffToFile, file, bMerge);
3005 fclose(file);
3006 return ret;
3008 else
3010 CString cmd;
3011 cmd = GetUnifiedDiffCmd(path, rev1, rev2, bMerge, bCombine, diffContext);
3012 return RunLogFile(cmd, patchfile, &gitLastErr);
3016 static void UnifiedDiffStatToStringA(const git_buf* text, void* payload)
3018 ATLASSERT(payload && text);
3019 CStringA *str = (CStringA*) payload;
3020 str->Append(text->ptr, (int)text->size);
3021 str->AppendChar('\n');
3024 static int UnifiedDiffToStringA(const git_diff_delta * /*delta*/, const git_diff_hunk * /*hunk*/, const git_diff_line *line, void *payload)
3026 ATLASSERT(payload && line);
3027 CStringA *str = (CStringA*) payload;
3028 if (line->origin == GIT_DIFF_LINE_CONTEXT || line->origin == GIT_DIFF_LINE_ADDITION || line->origin == GIT_DIFF_LINE_DELETION)
3029 str->Append(&line->origin, 1);
3030 str->Append(line->content, (int)line->content_len);
3031 return 0;
3034 int CGit::GetUnifiedDiff(const CTGitPath& path, const git_revnum_t& rev1, const git_revnum_t& rev2, CStringA * buffer, bool bMerge, bool bCombine, int diffContext)
3036 if (UsingLibGit2(GIT_CMD_DIFF))
3037 return GetUnifiedDiffLibGit2(path, rev1, rev2, UnifiedDiffStatToStringA, UnifiedDiffToStringA, buffer, bMerge);
3038 else
3040 CString cmd;
3041 cmd = GetUnifiedDiffCmd(path, rev1, rev2, bMerge, bCombine, diffContext);
3042 BYTE_VECTOR vector;
3043 int ret = Run(cmd, &vector);
3044 if (!vector.empty())
3046 vector.push_back(0); // vector is not NUL terminated
3047 buffer->Append((char *)&vector[0]);
3049 return ret;
3053 int CGit::GitRevert(int parent, const CGitHash &hash)
3055 if (UsingLibGit2(GIT_CMD_REVERT))
3057 CAutoRepository repo(GetGitRepository());
3058 if (!repo)
3059 return -1;
3061 CAutoCommit commit;
3062 if (git_commit_lookup(commit.GetPointer(), repo, (const git_oid *)hash.m_hash))
3063 return -1;
3065 git_revert_options revert_opts = GIT_REVERT_OPTIONS_INIT;
3066 revert_opts.mainline = parent;
3067 int result = git_revert(repo, commit, &revert_opts);
3069 return !result ? 0 : -1;
3071 else
3073 CString cmd, merge;
3074 if (parent)
3075 merge.Format(_T("-m %d "), parent);
3076 cmd.Format(_T("git.exe revert --no-edit --no-commit %s%s"), merge, hash.ToString());
3077 gitLastErr = cmd + _T("\n");
3078 if (Run(cmd, &gitLastErr, CP_UTF8))
3080 return -1;
3082 else
3084 gitLastErr.Empty();
3085 return 0;
3090 int CGit::DeleteRef(const CString& reference)
3092 if (UsingLibGit2(GIT_CMD_DELETETAGBRANCH))
3094 CAutoRepository repo(GetGitRepository());
3095 if (!repo)
3096 return -1;
3098 CStringA refA;
3099 if (reference.Right(3) == _T("^{}"))
3100 refA = CUnicodeUtils::GetUTF8(reference.Left(reference.GetLength() - 3));
3101 else
3102 refA = CUnicodeUtils::GetUTF8(reference);
3104 CAutoReference ref;
3105 if (git_reference_lookup(ref.GetPointer(), repo, refA))
3106 return -1;
3108 int result = -1;
3109 if (git_reference_is_tag(ref))
3110 result = git_tag_delete(repo, git_reference_shorthand(ref));
3111 else if (git_reference_is_branch(ref))
3112 result = git_branch_delete(ref);
3113 else if (git_reference_is_remote(ref))
3114 result = git_branch_delete(ref);
3115 else
3116 giterr_set_str(GITERR_REFERENCE, CUnicodeUtils::GetUTF8(L"unsupported reference type: " + reference));
3118 return result;
3120 else
3122 CString cmd, shortname;
3123 if (GetShortName(reference, shortname, _T("refs/heads/")))
3124 cmd.Format(_T("git.exe branch -D -- %s"), shortname);
3125 else if (GetShortName(reference, shortname, _T("refs/tags/")))
3126 cmd.Format(_T("git.exe tag -d -- %s"), shortname);
3127 else if (GetShortName(reference, shortname, _T("refs/remotes/")))
3128 cmd.Format(_T("git.exe branch -r -D -- %s"), shortname);
3129 else
3131 gitLastErr = L"unsupported reference type: " + reference;
3132 return -1;
3135 if (Run(cmd, &gitLastErr, CP_UTF8))
3136 return -1;
3138 gitLastErr.Empty();
3139 return 0;
3143 bool CGit::LoadTextFile(const CString &filename, CString &msg)
3145 if (PathFileExists(filename))
3147 FILE *pFile = nullptr;
3148 _tfopen_s(&pFile, filename, _T("r"));
3149 if (pFile)
3151 CStringA str;
3154 char s[8196] = { 0 };
3155 int read = (int)fread(s, sizeof(char), sizeof(s), pFile);
3156 if (read == 0)
3157 break;
3158 str += CStringA(s, read);
3159 } while (true);
3160 fclose(pFile);
3161 msg = CUnicodeUtils::GetUnicode(str);
3162 msg.Replace(_T("\r\n"), _T("\n"));
3163 msg.TrimRight(_T("\n"));
3164 msg += _T("\n");
3166 else
3167 ::MessageBox(nullptr, _T("Could not open ") + filename, _T("TortoiseGit"), MB_ICONERROR);
3168 return true; // load no further files
3170 return false;
3173 int CGit::GetWorkingTreeChanges(CTGitPathList& result, bool amend, CTGitPathList* filterlist)
3175 if (IsInitRepos())
3176 return GetInitAddList(result);
3178 BYTE_VECTOR out;
3180 int count = 1;
3181 if (filterlist)
3182 count = filterlist->GetCount();
3184 CString head = _T("HEAD");
3185 if (amend)
3186 head = _T("HEAD~1");
3188 for (int i = 0; i < count; ++i)
3190 BYTE_VECTOR cmdout;
3191 CString cmd;
3192 if (ms_bCygwinGit)
3194 // Prevent showing all files as modified when using cygwin's git
3195 if (!filterlist)
3196 cmd = _T("git.exe status --");
3197 else
3198 cmd.Format(_T("git.exe status -- \"%s\""), (*filterlist)[i].GetGitPathString());
3199 Run(cmd, &cmdout);
3200 cmdout.clear();
3203 // also list staged files which will be in the commit
3204 Run(_T("git.exe diff-index --cached --raw ") + head + _T(" --numstat -C -M -z --"), &cmdout);
3206 if (!filterlist)
3207 cmd = (_T("git.exe diff-index --raw ") + head + _T(" --numstat -C -M -z --"));
3208 else
3209 cmd.Format(_T("git.exe diff-index --raw ") + head + _T(" --numstat -C -M -z -- \"%s\""), (*filterlist)[i].GetGitPathString());
3211 BYTE_VECTOR cmdErr;
3212 if (Run(cmd, &cmdout, &cmdErr))
3214 int last = cmdErr.RevertFind(0, -1);
3215 CString str;
3216 CGit::StringAppend(&str, &cmdErr[last + 1], CP_UTF8, (int)cmdErr.size() - last - 1);
3217 MessageBox(nullptr, str, _T("TortoiseGit"), MB_OK | MB_ICONERROR);
3220 out.append(cmdout, 0);
3222 result.ParserFromLog(out);
3224 // handle delete conflict case, when remote : modified, local : deleted.
3225 for (int i = 0; i < count; ++i)
3227 BYTE_VECTOR cmdout;
3228 CString cmd;
3230 if (!filterlist)
3231 cmd = _T("git.exe ls-files -u -t -z");
3232 else
3233 cmd.Format(_T("git.exe ls-files -u -t -z -- \"%s\""), (*filterlist)[i].GetGitPathString());
3235 Run(cmd, &cmdout);
3237 CTGitPathList conflictlist;
3238 conflictlist.ParserFromLog(cmdout);
3239 for (int j = 0; j < conflictlist.GetCount(); ++j)
3241 CTGitPath* p = result.LookForGitPath(conflictlist[j].GetGitPathString());
3242 if (p)
3243 p->m_Action |= CTGitPath::LOGACTIONS_UNMERGED;
3244 else
3245 result.AddPath(conflictlist[j]);
3249 // handle source files of file renames/moves (issue #860)
3250 // if a file gets renamed and the new file "git add"ed, diff-index doesn't list the source file anymore
3251 for (int i = 0; i < count; ++i)
3253 BYTE_VECTOR cmdout;
3254 CString cmd;
3256 if (!filterlist)
3257 cmd = _T("git.exe ls-files -d -z");
3258 else
3259 cmd.Format(_T("git.exe ls-files -d -z -- \"%s\""), (*filterlist)[i].GetGitPathString());
3261 Run(cmd, &cmdout);
3263 CTGitPathList deletelist;
3264 deletelist.ParserFromLog(cmdout, true);
3265 for (int j = 0; j < deletelist.GetCount(); ++j)
3267 CTGitPath* p = result.LookForGitPath(deletelist[j].GetGitPathString());
3268 if (!p)
3269 result.AddPath(deletelist[j]);
3273 return 0;