Do not alter CGitHeadFileList after it was loaded and stored
[TortoiseGit.git] / src / Git / GitIndex.cpp
blob8c9648f1f9a93b6b0c4d7342be3d2f850f49cb1c
1 // TortoiseGit - a Windows shell extension for easy version control
3 // Copyright (C) 2008-2012 - 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 "atlconv.h"
23 #include "GitRev.h"
24 #include "registry.h"
25 #include "GitConfig.h"
26 #include <map>
27 #include "UnicodeUtils.h"
28 #include "TGitPath.h"
29 #include "gitindex.h"
30 #include <sys/types.h>
31 #include <sys/stat.h>
32 #include "SmartHandle.h"
34 class CAutoReadLock
36 SharedMutex *m_Lock;
37 public:
38 CAutoReadLock(SharedMutex * lock)
40 m_Lock = lock;
41 lock->AcquireShared();
43 ~CAutoReadLock()
45 m_Lock->ReleaseShared();
49 class CAutoWriteLock
51 SharedMutex *m_Lock;
52 public:
53 CAutoWriteLock(SharedMutex * lock)
55 m_Lock = lock;
56 lock->AcquireExclusive();
58 ~CAutoWriteLock()
60 m_Lock->ReleaseExclusive();
64 CGitAdminDirMap g_AdminDirMap;
66 int CGitIndex::Print()
68 _tprintf(_T("0x%08X 0x%08X %s %s\n"),
69 (int)this->m_ModifyTime,
70 this->m_Flags,
71 this->m_IndexHash.ToString(),
72 this->m_FileName);
74 return 0;
77 CGitIndexList::CGitIndexList()
79 this->m_LastModifyTime = 0;
80 m_critRepoSec.Init();
81 repository = NULL;
82 m_bCheckContent = !!(CRegDWORD(_T("Software\\TortoiseGit\\TGitCacheCheckContent"), TRUE) == TRUE);
85 CGitIndexList::~CGitIndexList()
87 if (repository != NULL)
89 git_repository_free(repository);
90 m_critRepoSec.Term();
94 static bool SortIndex(CGitIndex &Item1, CGitIndex &Item2)
96 return Item1.m_FileName.Compare(Item2.m_FileName) < 0;
99 static bool SortTree(CGitTreeItem &Item1, CGitTreeItem &Item2)
101 return Item1.m_FileName.Compare(Item2.m_FileName) < 0;
104 int CGitIndexList::ReadIndex(CString dgitdir)
106 this->clear();
108 CStringA gitdir = CUnicodeUtils::GetMulti(dgitdir, CP_UTF8);
110 m_critRepoSec.Lock();
111 if (repository != NULL)
113 git_repository_free(repository);
114 repository = NULL;
116 git_index *index = NULL;
118 int ret = git_repository_open(&repository, gitdir.GetBuffer());
119 gitdir.ReleaseBuffer();
120 if (ret)
121 return -1;
123 // add config files
124 git_config * config;
125 git_config_new(&config);
127 CString projectConfig = dgitdir + _T("config");
128 CString globalConfig = g_Git.GetGitGlobalConfig();
129 CString globalXDGConfig = g_Git.GetGitGlobalXDGConfig();
130 CString msysGitBinPath(CRegString(REG_MSYSGIT_PATH, _T(""), FALSE));
132 CStringA projectConfigA = CUnicodeUtils::GetMulti(projectConfig, CP_UTF8);
133 git_config_add_file_ondisk(config, projectConfigA.GetBuffer(), 4, FALSE);
134 projectConfigA.ReleaseBuffer();
135 CStringA globalConfigA = CUnicodeUtils::GetMulti(globalConfig, CP_UTF8);
136 git_config_add_file_ondisk(config, globalConfigA.GetBuffer(), 3, FALSE);
137 globalConfigA.ReleaseBuffer();
138 CStringA globalXDGConfigA = CUnicodeUtils::GetMulti(globalXDGConfig, CP_UTF8);
139 git_config_add_file_ondisk(config, globalXDGConfigA.GetBuffer(), 2, FALSE);
140 globalXDGConfigA.ReleaseBuffer();
141 if (!msysGitBinPath.IsEmpty())
143 CString systemConfig = msysGitBinPath + _T("\\..\\etc\\gitconfig");
144 CStringA systemConfigA = CUnicodeUtils::GetMulti(systemConfig, CP_UTF8);
145 git_config_add_file_ondisk(config, systemConfigA.GetBuffer(), 1, FALSE);
146 systemConfigA.ReleaseBuffer();
149 git_repository_set_config(repository, config);
151 // load index in order to enumerate files
152 if (git_repository_index(&index, repository))
154 repository = NULL;
155 m_critRepoSec.Unlock();
156 return -1;
159 size_t ecount = git_index_entrycount(index);
160 resize(ecount);
161 for (size_t i = 0; i < ecount; ++i)
163 const git_index_entry *e = git_index_get_byindex(index, i);
165 this->at(i).m_FileName.Empty();
166 g_Git.StringAppend(&this->at(i).m_FileName, (BYTE*)e->path, CP_UTF8);
167 this->at(i).m_FileName.MakeLower();
168 this->at(i).m_ModifyTime = e->mtime.seconds;
169 this->at(i).m_Flags = e->flags | e->flags_extended;
170 this->at(i).m_IndexHash = (char *) e->oid.id;
173 git_index_free(index);
175 g_Git.GetFileModifyTime(dgitdir + _T("index"), &this->m_LastModifyTime);
176 std::sort(this->begin(), this->end(), SortIndex);
178 m_critRepoSec.Unlock();
180 return 0;
183 int CGitIndexList::GetFileStatus(const CString &gitdir,const CString &pathorg,git_wc_status_kind *status,__int64 time,FIll_STATUS_CALLBACK callback,void *pData, CGitHash *pHash, bool * assumeValid, bool * skipWorktree)
185 if(status)
187 CString path = pathorg;
188 path.MakeLower();
190 int start = SearchInSortVector(*this, ((CString&)path).GetBuffer(), -1);
191 ((CString&)path).ReleaseBuffer();
193 if (start < 0)
195 *status = git_wc_status_unversioned;
196 if (pHash)
197 pHash->Empty();
200 else
202 int index = start;
203 if (index <0)
204 return -1;
205 if (index >= (int)size())
206 return -1;
208 // skip-worktree has higher priority than assume-valid
209 if (at(index).m_Flags & GIT_IDXENTRY_SKIP_WORKTREE)
211 *status = git_wc_status_normal;
212 if (skipWorktree)
213 *skipWorktree = true;
215 else if (at(index).m_Flags & GIT_IDXENTRY_VALID)
217 *status = git_wc_status_normal;
218 if (assumeValid)
219 *assumeValid = true;
221 else if (time == at(index).m_ModifyTime)
223 *status = git_wc_status_normal;
225 else if (m_bCheckContent && repository)
227 git_oid actual;
228 CStringA fileA = CUnicodeUtils::GetMulti(pathorg, CP_UTF8);
229 m_critRepoSec.Lock(); // prevent concurrent access to repository instance and especially filter-lists
230 if (!git_repository_hashfile(&actual, repository, fileA.GetBuffer(), GIT_OBJ_BLOB, NULL) && !git_oid_cmp(&actual, (const git_oid*)at(index).m_IndexHash.m_hash))
232 at(index).m_ModifyTime = time;
233 *status = git_wc_status_normal;
235 else
236 *status = git_wc_status_modified;
237 fileA.ReleaseBuffer();
238 m_critRepoSec.Unlock();
240 else
241 *status = git_wc_status_modified;
243 if (at(index).m_Flags & GIT_IDXENTRY_STAGEMASK)
244 *status = git_wc_status_conflicted;
245 else if (at(index).m_Flags & GIT_IDXENTRY_INTENT_TO_ADD)
246 *status = git_wc_status_added;
248 if(pHash)
249 *pHash = at(index).m_IndexHash;
254 if(callback && status)
255 callback(gitdir + _T("\\") + pathorg, *status, false, pData, *assumeValid, *skipWorktree);
256 return 0;
259 int CGitIndexList::GetStatus(const CString &gitdir,const CString &pathParam, git_wc_status_kind *status,
260 BOOL IsFull, BOOL /*IsRecursive*/,
261 FIll_STATUS_CALLBACK callback,void *pData,
262 CGitHash *pHash, bool * assumeValid, bool * skipWorktree)
264 int result;
265 git_wc_status_kind dirstatus = git_wc_status_none;
266 __int64 time;
267 bool isDir = false;
268 CString path = pathParam;
270 if (status)
272 if (path.IsEmpty())
273 result = g_Git.GetFileModifyTime(gitdir, &time, &isDir);
274 else
275 result = g_Git.GetFileModifyTime(gitdir + _T("\\") + path, &time, &isDir);
277 if (result)
279 *status = git_wc_status_deleted;
280 if (callback)
281 callback(gitdir + _T("\\") + path, git_wc_status_deleted, false, pData, *assumeValid, *skipWorktree);
283 return 0;
285 if (isDir)
287 if (!path.IsEmpty())
289 if (path.Right(1) != _T("\\"))
290 path += _T("\\");
292 int len = path.GetLength();
294 for (size_t i = 0; i < size(); i++)
296 if (at(i).m_FileName.GetLength() > len)
298 if (at(i).m_FileName.Left(len) == path)
300 if (!IsFull)
302 *status = git_wc_status_normal;
303 if (callback)
304 callback(gitdir + _T("\\") + path, *status, false, pData, (at(i).m_Flags & GIT_IDXENTRY_VALID) && !(at(i).m_Flags & GIT_IDXENTRY_SKIP_WORKTREE), (at(i).m_Flags & GIT_IDXENTRY_SKIP_WORKTREE) != 0);
305 return 0;
308 else
310 result = g_Git.GetFileModifyTime(gitdir+_T("\\") + at(i).m_FileName, &time);
311 if (result)
312 continue;
314 *status = git_wc_status_none;
315 if (assumeValid)
316 *assumeValid = false;
317 if (skipWorktree)
318 *skipWorktree = false;
319 GetFileStatus(gitdir, at(i).m_FileName, status, time, callback, pData, NULL, assumeValid, skipWorktree);
320 // if a file is assumed valid, we need to inform the caller, otherwise the assumevalid flag might not get to the explorer on first open of a repository
321 if (callback && (assumeValid || skipWorktree))
322 callback(gitdir + _T("\\") + path, *status, false, pData, *assumeValid, *skipWorktree);
323 if (*status != git_wc_status_none)
325 if (dirstatus == git_wc_status_none)
327 dirstatus = git_wc_status_normal;
329 if (*status != git_wc_status_normal)
331 dirstatus = git_wc_status_modified;
338 } /* End For */
340 if (dirstatus != git_wc_status_none)
342 *status = dirstatus;
344 else
346 *status = git_wc_status_unversioned;
348 if(callback)
349 callback(gitdir + _T("\\") + path, *status, false, pData, false, false);
351 return 0;
354 else
356 GetFileStatus(gitdir, path, status, time, callback, pData, pHash, assumeValid, skipWorktree);
359 return 0;
362 int CGitIndexFileMap::Check(const CString &gitdir, bool *isChanged)
364 __int64 time;
365 int result;
367 CString IndexFile = g_AdminDirMap.GetAdminDir(gitdir) + _T("index");
369 /* Get data associated with "crt_stat.c": */
370 result = g_Git.GetFileModifyTime(IndexFile, &time);
372 if (result)
373 return result;
375 SHARED_INDEX_PTR pIndex;
376 pIndex = this->SafeGet(gitdir);
378 if (pIndex.get() == NULL)
380 if(isChanged)
381 *isChanged = true;
382 return 0;
385 if (pIndex->m_LastModifyTime == time)
387 if (isChanged)
388 *isChanged = false;
390 else
392 if (isChanged)
393 *isChanged = true;
395 return 0;
398 int CGitIndexFileMap::LoadIndex(const CString &gitdir)
402 SHARED_INDEX_PTR pIndex(new CGitIndexList);
404 if(pIndex->ReadIndex(g_AdminDirMap.GetAdminDir(gitdir)))
405 return -1;
407 this->SafeSet(gitdir, pIndex);
409 }catch(...)
411 return -1;
413 return 0;
416 int CGitIndexFileMap::GetFileStatus(const CString &gitdir, const CString &path, git_wc_status_kind *status,BOOL IsFull, BOOL IsRecursive,
417 FIll_STATUS_CALLBACK callback,void *pData,
418 CGitHash *pHash,
419 bool isLoadUpdatedIndex, bool * assumeValid, bool * skipWorktree)
423 CheckAndUpdate(gitdir, isLoadUpdatedIndex);
425 SHARED_INDEX_PTR pIndex = this->SafeGet(gitdir);
426 if (pIndex.get() != NULL)
428 pIndex->GetStatus(gitdir, path, status, IsFull, IsRecursive, callback, pData, pHash, assumeValid, skipWorktree);
430 else
432 // git working tree has not index
433 *status = git_wc_status_unversioned;
436 catch(...)
438 return -1;
440 return 0;
443 int CGitIndexFileMap::IsUnderVersionControl(const CString &gitdir, const CString &path, bool isDir,bool *isVersion, bool isLoadUpdateIndex)
447 if (path.IsEmpty())
449 *isVersion = true;
450 return 0;
453 CString subpath = path;
454 subpath.Replace(_T('\\'), _T('/'));
455 if(isDir)
456 subpath += _T('/');
458 subpath.MakeLower();
460 CheckAndUpdate(gitdir, isLoadUpdateIndex);
462 SHARED_INDEX_PTR pIndex = this->SafeGet(gitdir);
464 if(pIndex.get())
466 if(isDir)
467 *isVersion = (SearchInSortVector(*pIndex, subpath.GetBuffer(), subpath.GetLength()) >= 0);
468 else
469 *isVersion = (SearchInSortVector(*pIndex, subpath.GetBuffer(), -1) >= 0);
470 subpath.ReleaseBuffer();
473 }catch(...)
475 return -1;
477 return 0;
480 int CGitHeadFileList::GetPackRef(const CString &gitdir)
482 CString PackRef = g_AdminDirMap.GetAdminDir(gitdir) + _T("packed-refs");
484 __int64 mtime;
485 if (g_Git.GetFileModifyTime(PackRef, &mtime))
487 CAutoWriteLock lock(&this->m_SharedMutex);
488 //packed refs is not existed
489 this->m_PackRefFile.Empty();
490 this->m_PackRefMap.clear();
491 return 0;
493 else if(mtime == m_LastModifyTimePackRef)
495 return 0;
497 else
499 CAutoWriteLock lock(&this->m_SharedMutex);
500 this->m_PackRefFile = PackRef;
501 this->m_LastModifyTimePackRef = mtime;
504 int ret = 0;
506 CAutoWriteLock lock(&this->m_SharedMutex);
507 this->m_PackRefMap.clear();
509 CAutoFile hfile = CreateFile(PackRef,
510 GENERIC_READ,
511 FILE_SHARE_READ|FILE_SHARE_DELETE|FILE_SHARE_WRITE,
512 NULL,
513 OPEN_EXISTING,
514 FILE_ATTRIBUTE_NORMAL,
515 NULL);
518 if (!hfile)
520 ret = -1;
521 break;
524 DWORD filesize = GetFileSize(hfile, NULL);
525 DWORD size =0;
526 char *buff;
527 buff = new char[filesize];
529 ReadFile(hfile, buff, filesize, &size, NULL);
531 if (size != filesize)
533 ret = -1;
534 break;
537 CString hash;
538 CString ref;
540 for(DWORD i=0;i<filesize;)
542 hash.Empty();
543 ref.Empty();
544 if (buff[i] == '#' || buff[i] == '^')
546 while (buff[i] != '\n')
548 i++;
549 if (i == filesize)
550 break;
552 i++;
555 if (i >= filesize)
556 break;
558 while (buff[i] != ' ')
560 hash.AppendChar(buff[i]);
561 i++;
562 if (i == filesize)
563 break;
566 i++;
567 if (i >= filesize)
568 break;
570 while (buff[i] != '\n')
572 ref.AppendChar(buff[i]);
573 i++;
574 if (i == filesize)
575 break;
578 if (!ref.IsEmpty() )
580 this->m_PackRefMap[ref] = hash;
583 while (buff[i] == '\n')
585 i++;
586 if (i == filesize)
587 break;
591 delete[] buff;
593 } while(0);
595 return ret;
598 int CGitHeadFileList::ReadHeadHash(CString gitdir)
600 int ret = 0;
601 CAutoWriteLock lock(&this->m_SharedMutex);
602 m_Gitdir = g_AdminDirMap.GetAdminDir(gitdir);
604 m_HeadFile = m_Gitdir + _T("HEAD");
606 if( g_Git.GetFileModifyTime(m_HeadFile, &m_LastModifyTimeHead))
607 return -1;
613 CAutoFile hfile = CreateFile(m_HeadFile,
614 GENERIC_READ,
615 FILE_SHARE_READ|FILE_SHARE_DELETE|FILE_SHARE_WRITE,
616 NULL,
617 OPEN_EXISTING,
618 FILE_ATTRIBUTE_NORMAL,
619 NULL);
621 if (!hfile)
623 ret = -1;
624 break;
627 DWORD size = 0,filesize = 0;
628 unsigned char buffer[40] ;
629 ReadFile(hfile, buffer, 4, &size, NULL);
630 if (size != 4)
632 ret = -1;
633 break;
635 buffer[4]=0;
636 if (strcmp((const char*)buffer,"ref:") == 0)
638 filesize = GetFileSize(hfile, NULL);
640 unsigned char *p = (unsigned char*)malloc(filesize -4);
642 ReadFile(hfile, p, filesize - 4, &size, NULL);
644 m_HeadRefFile.Empty();
645 g_Git.StringAppend(&this->m_HeadRefFile, p, CP_UTF8, filesize - 4);
646 CString ref = this->m_HeadRefFile;
647 ref = ref.Trim();
648 int start = 0;
649 ref = ref.Tokenize(_T("\n"), start);
650 free(p);
651 m_HeadRefFile = m_Gitdir + m_HeadRefFile.Trim();
652 m_HeadRefFile.Replace(_T('/'),_T('\\'));
654 __int64 time;
655 if (g_Git.GetFileModifyTime(m_HeadRefFile, &time, NULL))
657 m_HeadRefFile.Empty();
658 if (GetPackRef(gitdir))
660 ret = -1;
661 break;
663 if (this->m_PackRefMap.find(ref) == m_PackRefMap.end())
665 ret = -1;
666 break;
668 this ->m_Head = m_PackRefMap[ref];
669 ret = 0;
670 break;
673 CAutoFile href = CreateFile(m_HeadRefFile,
674 GENERIC_READ,
675 FILE_SHARE_READ|FILE_SHARE_DELETE|FILE_SHARE_WRITE,
676 NULL,
677 OPEN_EXISTING,
678 FILE_ATTRIBUTE_NORMAL,
679 NULL);
681 if (!href)
683 m_HeadRefFile.Empty();
685 if (GetPackRef(gitdir))
687 ret = -1;
688 break;
691 if (this->m_PackRefMap.find(ref) == m_PackRefMap.end())
693 ret = -1;
694 break;
696 this ->m_Head = m_PackRefMap[ref];
697 ret = 0;
698 break;
700 ReadFile(href, buffer, 40, &size, NULL);
701 if (size != 40)
703 ret = -1;
704 break;
706 this->m_Head.ConvertFromStrA((char*)buffer);
708 this->m_LastModifyTimeRef = time;
711 else
713 ReadFile(hfile, buffer + 4, 40 - 4, &size, NULL);
714 if(size != 36)
716 ret = -1;
717 break;
719 m_HeadRefFile.Empty();
721 this->m_Head.ConvertFromStrA((char*)buffer);
723 } while(0);
725 catch(...)
727 ret = -1;
730 return ret;
733 bool CGitHeadFileList::CheckHeadUpdate()
735 CAutoReadLock lock(&m_SharedMutex);
736 if (this->m_HeadFile.IsEmpty())
737 return true;
739 __int64 mtime=0;
741 if (g_Git.GetFileModifyTime(m_HeadFile, &mtime))
742 return true;
744 if (mtime != this->m_LastModifyTimeHead)
745 return true;
747 if (!this->m_HeadRefFile.IsEmpty())
749 if (g_Git.GetFileModifyTime(m_HeadRefFile, &mtime))
750 return true;
752 if (mtime != this->m_LastModifyTimeRef)
753 return true;
756 if(!this->m_PackRefFile.IsEmpty())
758 if (g_Git.GetFileModifyTime(m_PackRefFile, &mtime))
759 return true;
761 if (mtime != this->m_LastModifyTimePackRef)
762 return true;
765 // in an empty repo HEAD points to refs/heads/master, but this ref doesn't exist.
766 // So we need to retry again and again until the ref exists - otherwise we will never notice
767 if (this->m_Head.IsEmpty() && this->m_HeadRefFile.IsEmpty() && this->m_PackRefFile.IsEmpty())
768 return true;
770 return false;
773 bool CGitHeadFileList::HeadHashEqualsTreeHash()
775 CAutoReadLock lock(&this->m_SharedMutex);
776 return (m_Head == m_TreeHash);
779 bool CGitHeadFileList::HeadFileIsEmpty()
781 CAutoReadLock lock(&this->m_SharedMutex);
782 return m_HeadFile.IsEmpty();
785 bool CGitHeadFileList::HeadIsEmpty()
787 CAutoReadLock lock(&this->m_SharedMutex);
788 return m_Head.IsEmpty();
791 int CGitHeadFileList::CallBack(const unsigned char *sha1, const char *base, int baselen,
792 const char *pathname, unsigned mode, int /*stage*/, void *context)
794 #define S_IFGITLINK 0160000
796 CGitHeadFileList *p = (CGitHeadFileList*)context;
797 if( mode&S_IFDIR )
799 if( (mode&S_IFMT) != S_IFGITLINK)
800 return READ_TREE_RECURSIVE;
803 size_t cur = p->size();
804 p->resize(p->size() + 1);
805 p->at(cur).m_Hash = (char*)sha1;
806 p->at(cur).m_FileName.Empty();
808 if(base)
809 g_Git.StringAppend(&p->at(cur).m_FileName, (BYTE*)base, CP_UTF8, baselen);
811 g_Git.StringAppend(&p->at(cur).m_FileName,(BYTE*)pathname, CP_UTF8);
813 p->at(cur).m_FileName.MakeLower();
815 //p->at(cur).m_FileName.Replace(_T('/'), _T('\\'));
817 //p->m_Map[p->at(cur).m_FileName] = cur;
819 if( (mode&S_IFMT) == S_IFGITLINK)
820 return 0;
822 return READ_TREE_RECURSIVE;
825 int ReadTreeRecursive(git_repository &repo, git_tree * tree, CStringA base, int (*CallBack) (const unsigned char *, const char *, int, const char *, unsigned int, int, void *),void *data)
827 size_t count = git_tree_entrycount(tree);
828 for (size_t i = 0; i < count; i++)
830 const git_tree_entry *entry = git_tree_entry_byindex(tree, i);
831 if (entry == NULL)
832 continue;
833 int mode = git_tree_entry_filemode(entry);
834 if( CallBack(git_tree_entry_id(entry)->id,
835 base,
836 base.GetLength(),
837 git_tree_entry_name(entry),
838 mode,
840 data) == READ_TREE_RECURSIVE
843 if(mode&S_IFDIR)
845 git_object *object = NULL;
846 git_tree_entry_to_object(&object, &repo, entry);
847 if (object == NULL)
848 continue;
849 CStringA parent = base;
850 parent += git_tree_entry_name(entry);
851 parent += "/";
852 ReadTreeRecursive(repo, (git_tree*)object, parent, CallBack, data);
853 git_object_free(object);
859 return 0;
862 // ReadTree is/must only be executed on an empty list
863 int CGitHeadFileList::ReadTree()
865 CAutoWriteLock lock(&m_SharedMutex);
866 CStringA gitdir = CUnicodeUtils::GetMulti(m_Gitdir, CP_UTF8);
867 git_repository *repository = NULL;
868 git_commit *commit = NULL;
869 git_tree * tree = NULL;
870 int ret = 0;
871 ATLASSERT(this->empty());
874 ret = git_repository_open(&repository, gitdir.GetBuffer());
875 gitdir.ReleaseBuffer();
876 if(ret)
877 break;
878 ret = git_commit_lookup(&commit, repository, (const git_oid*)m_Head.m_hash);
879 if(ret)
880 break;
882 ret = git_commit_tree(&tree, commit);
883 if(ret)
884 break;
886 ret = ReadTreeRecursive(*repository, tree,"", CGitHeadFileList::CallBack,this);
887 if(ret)
888 break;
890 std::sort(this->begin(), this->end(), SortTree);
891 this->m_TreeHash = (char*)(git_commit_id(commit)->id);
893 } while(0);
895 if (tree)
896 git_tree_free(tree);
898 if (commit)
899 git_commit_free(commit);
901 if (repository)
902 git_repository_free(repository);
904 if (ret)
905 m_LastModifyTimeHead = 0;
907 return ret;
910 int CGitIgnoreItem::FetchIgnoreList(const CString &projectroot, const CString &file, bool isGlobal)
912 CAutoWriteLock lock(&this->m_SharedMutex);
914 if (this->m_pExcludeList)
916 free(m_pExcludeList);
917 m_pExcludeList=NULL;
920 this->m_BaseDir.Empty();
921 if (!isGlobal)
923 CString base = file.Mid(projectroot.GetLength() + 1);
924 base.Replace(_T('\\'), _T('/'));
926 int start = base.ReverseFind(_T('/'));
927 if(start > 0)
929 base = base.Left(start);
930 this->m_BaseDir = CUnicodeUtils::GetMulti(base, CP_UTF8) + "/";
935 if(g_Git.GetFileModifyTime(file, &m_LastModifyTime))
936 return -1;
938 if(git_create_exclude_list(&this->m_pExcludeList))
939 return -1;
942 CAutoFile hfile = CreateFile(file,
943 GENERIC_READ,
944 FILE_SHARE_READ|FILE_SHARE_DELETE|FILE_SHARE_WRITE,
945 NULL,
946 OPEN_EXISTING,
947 FILE_ATTRIBUTE_NORMAL,
948 NULL);
951 if (!hfile)
952 return -1 ;
954 DWORD size=0,filesize=0;
956 filesize=GetFileSize(hfile, NULL);
958 if(filesize == INVALID_FILE_SIZE)
959 return -1;
961 BYTE *buffer = new BYTE[filesize + 1];
963 if(buffer == NULL)
964 return -1;
966 if(! ReadFile(hfile, buffer,filesize,&size,NULL))
967 return GetLastError();
969 BYTE *p = buffer;
970 for (DWORD i = 0; i < size; i++)
972 if (buffer[i] == '\n' || buffer[i] == '\r' || i == (size - 1))
974 if (buffer[i] == '\n' || buffer[i] == '\r')
975 buffer[i] = 0;
976 if (i == size - 1)
977 buffer[size] = 0;
979 if(p[0] != '#' && p[0] != 0)
980 git_add_exclude((const char*)p,
981 this->m_BaseDir.GetBuffer(),
982 m_BaseDir.GetLength(),
983 this->m_pExcludeList);
985 p=buffer+i+1;
988 /* Can't free buffer, exluced list will use this buffer*/
989 //delete buffer;
990 //buffer = NULL;
992 return 0;
995 bool CGitIgnoreList::CheckFileChanged(const CString &path)
997 __int64 time = 0;
999 int ret = g_Git.GetFileModifyTime(path, &time);
1001 this->m_SharedMutex.AcquireShared();
1002 bool cacheExist = (m_Map.find(path) != m_Map.end());
1003 this->m_SharedMutex.ReleaseShared();
1005 if (!cacheExist && ret == 0)
1007 CAutoWriteLock lock(&this->m_SharedMutex);
1008 m_Map[path].m_LastModifyTime = 0;
1009 m_Map[path].m_SharedMutex.Init();
1011 // both cache and file is not exist
1012 if ((ret != 0) && (!cacheExist))
1013 return false;
1015 // file exist but cache miss
1016 if ((ret == 0) && (!cacheExist))
1017 return true;
1019 // file not exist but cache exist
1020 if ((ret != 0) && (cacheExist))
1022 return true;
1024 // file exist and cache exist
1027 CAutoReadLock lock(&this->m_SharedMutex);
1028 if (m_Map[path].m_LastModifyTime == time)
1029 return false;
1031 return true;
1034 bool CGitIgnoreList::CheckIgnoreChanged(const CString &gitdir,const CString &path)
1036 CString temp;
1037 temp = gitdir;
1038 temp += _T("\\");
1039 temp += path;
1041 temp.Replace(_T('/'), _T('\\'));
1043 while(!temp.IsEmpty())
1045 CString tempOrig = temp;
1046 temp += _T("\\.git");
1048 if (CGit::GitPathFileExists(temp))
1050 CString gitignore=temp;
1051 gitignore += _T("ignore");
1052 if (CheckFileChanged(gitignore))
1053 return true;
1055 CString adminDir = g_AdminDirMap.GetAdminDir(tempOrig);
1056 CString wcglobalgitignore = adminDir + _T("info\\exclude");
1057 if (CheckFileChanged(wcglobalgitignore))
1058 return true;
1060 if (CheckAndUpdateCoreExcludefile(adminDir))
1061 return true;
1063 return false;
1065 else
1067 temp += _T("ignore");
1068 if (CheckFileChanged(temp))
1069 return true;
1072 int found=0;
1073 int i;
1074 for (i = temp.GetLength() - 1; i >= 0; i--)
1076 if(temp[i] == _T('\\'))
1077 found ++;
1079 if(found == 2)
1080 break;
1083 temp = temp.Left(i);
1085 return true;
1088 int CGitIgnoreList::FetchIgnoreFile(const CString &gitdir, const CString &gitignore, bool isGlobal)
1090 if (CGit::GitPathFileExists(gitignore)) //if .gitignore remove, we need remote cache
1092 CAutoWriteLock lock(&this->m_SharedMutex);
1093 if (m_Map.find(gitignore) == m_Map.end())
1094 m_Map[gitignore].m_SharedMutex.Init();
1096 m_Map[gitignore].FetchIgnoreList(gitdir, gitignore, isGlobal);
1098 else
1100 CAutoWriteLock lock(&this->m_SharedMutex);
1101 if (m_Map.find(gitignore) != m_Map.end())
1102 m_Map[gitignore].m_SharedMutex.Release();
1104 m_Map.erase(gitignore);
1106 return 0;
1109 int CGitIgnoreList::LoadAllIgnoreFile(const CString &gitdir,const CString &path)
1111 CString temp;
1113 temp = gitdir;
1114 temp += _T("\\");
1115 temp += path;
1117 temp.Replace(_T('/'), _T('\\'));
1119 while (!temp.IsEmpty())
1121 CString tempOrig = temp;
1122 temp += _T("\\.git");
1124 if (CGit::GitPathFileExists(temp))
1126 CString gitignore = temp;
1127 gitignore += _T("ignore");
1128 if (CheckFileChanged(gitignore))
1130 FetchIgnoreFile(gitdir, gitignore, false);
1133 CString adminDir = g_AdminDirMap.GetAdminDir(tempOrig);
1134 CString wcglobalgitignore = adminDir + _T("info\\exclude");
1135 if (CheckFileChanged(wcglobalgitignore))
1137 FetchIgnoreFile(gitdir, wcglobalgitignore, true);
1140 if (CheckAndUpdateCoreExcludefile(adminDir))
1142 m_SharedMutex.AcquireShared();
1143 CString excludesFile = m_CoreExcludesfiles[adminDir];
1144 m_SharedMutex.ReleaseShared();
1145 if (!excludesFile.IsEmpty())
1146 FetchIgnoreFile(gitdir, excludesFile, true);
1149 return 0;
1151 else
1153 temp += _T("ignore");
1154 if (CheckFileChanged(temp))
1156 FetchIgnoreFile(gitdir, temp, false);
1160 int found = 0;
1161 int i;
1162 for (i = temp.GetLength() - 1; i >= 0; i--)
1164 if(temp[i] == _T('\\'))
1165 found ++;
1167 if(found == 2)
1168 break;
1171 temp = temp.Left(i);
1173 return 0;
1175 bool CGitIgnoreList::CheckAndUpdateMsysGitBinpath(bool force)
1177 // recheck every 30 seconds
1178 if (GetTickCount() - m_dMsysGitBinPathLastChecked > 30000 || force)
1180 m_dMsysGitBinPathLastChecked = GetTickCount();
1181 CString msysGitBinPath(CRegString(REG_MSYSGIT_PATH, _T(""), FALSE));
1182 if (msysGitBinPath != m_sMsysGitBinPath)
1184 m_sMsysGitBinPath = msysGitBinPath;
1185 return true;
1188 return false;
1190 bool CGitIgnoreList::CheckAndUpdateCoreExcludefile(const CString &adminDir)
1192 bool hasChanged = false;
1194 CString projectConfig = adminDir + _T("config");
1195 CString globalConfig = g_Git.GetGitGlobalConfig();
1196 CString globalXDGConfig = g_Git.GetGitGlobalXDGConfig();
1198 CAutoWriteLock lock(&m_coreExcludefilesSharedMutex);
1199 hasChanged = CheckAndUpdateMsysGitBinpath();
1200 CString systemConfig = m_sMsysGitBinPath + _T("\\..\\etc\\gitconfig");
1202 hasChanged = hasChanged || CheckFileChanged(projectConfig);
1203 hasChanged = hasChanged || CheckFileChanged(globalConfig);
1204 hasChanged = hasChanged || CheckFileChanged(globalXDGConfig);
1205 if (!m_sMsysGitBinPath.IsEmpty())
1206 hasChanged = hasChanged || CheckFileChanged(systemConfig);
1208 m_SharedMutex.AcquireShared();
1209 CString excludesFile = m_CoreExcludesfiles[adminDir];
1210 m_SharedMutex.ReleaseShared();
1211 if (!excludesFile.IsEmpty())
1212 hasChanged = hasChanged || CheckFileChanged(excludesFile);
1214 if (!hasChanged)
1215 return false;
1217 git_config * config;
1218 git_config_new(&config);
1219 CStringA projectConfigA = CUnicodeUtils::GetMulti(projectConfig, CP_UTF8);
1220 git_config_add_file_ondisk(config, projectConfigA.GetBuffer(), 4, FALSE);
1221 projectConfigA.ReleaseBuffer();
1222 CStringA globalConfigA = CUnicodeUtils::GetMulti(globalConfig, CP_UTF8);
1223 git_config_add_file_ondisk(config, globalConfigA.GetBuffer(), 3, FALSE);
1224 globalConfigA.ReleaseBuffer();
1225 CStringA globalXDGConfigA = CUnicodeUtils::GetMulti(globalXDGConfig, CP_UTF8);
1226 git_config_add_file_ondisk(config, globalXDGConfigA.GetBuffer(), 2, FALSE);
1227 globalXDGConfigA.ReleaseBuffer();
1228 if (!m_sMsysGitBinPath.IsEmpty())
1230 CStringA systemConfigA = CUnicodeUtils::GetMulti(systemConfig, CP_UTF8);
1231 git_config_add_file_ondisk(config, systemConfigA.GetBuffer(), 1, FALSE);
1232 systemConfigA.ReleaseBuffer();
1234 const char * out = NULL;
1235 CStringA name(_T("core.excludesfile"));
1236 git_config_get_string(&out, config, name.GetBuffer());
1237 name.ReleaseBuffer();
1238 CStringA excludesFileA(out);
1239 excludesFile = CUnicodeUtils::GetUnicode(excludesFileA);
1240 if (excludesFile.IsEmpty())
1241 excludesFile = GetWindowsHome() + _T("\\.config\\git\\ignore");
1242 else if (excludesFile.Find(_T("~/")) == 0)
1243 excludesFile = GetWindowsHome() + excludesFile.Mid(1);
1244 git_config_free(config);
1246 CAutoWriteLock lockMap(&m_SharedMutex);
1247 g_Git.GetFileModifyTime(projectConfig, &m_Map[projectConfig].m_LastModifyTime);
1248 g_Git.GetFileModifyTime(globalXDGConfig, &m_Map[globalXDGConfig].m_LastModifyTime);
1249 if (m_Map[globalXDGConfig].m_LastModifyTime == 0)
1251 m_Map[globalXDGConfig].m_SharedMutex.Release();
1252 m_Map.erase(globalXDGConfig);
1254 g_Git.GetFileModifyTime(globalConfig, &m_Map[globalConfig].m_LastModifyTime);
1255 if (m_Map[globalConfig].m_LastModifyTime == 0)
1257 m_Map[globalConfig].m_SharedMutex.Release();
1258 m_Map.erase(globalConfig);
1260 if (!m_sMsysGitBinPath.IsEmpty())
1261 g_Git.GetFileModifyTime(systemConfig, &m_Map[systemConfig].m_LastModifyTime);
1262 if (m_Map[systemConfig].m_LastModifyTime == 0 || m_sMsysGitBinPath.IsEmpty())
1264 m_Map[systemConfig].m_SharedMutex.Release();
1265 m_Map.erase(systemConfig);
1267 m_CoreExcludesfiles[adminDir] = excludesFile;
1269 return true;
1271 const CString CGitIgnoreList::GetWindowsHome()
1273 static CString sWindowsHome(g_Git.GetHomeDirectory());
1274 return sWindowsHome;
1276 bool CGitIgnoreList::IsIgnore(const CString &path,const CString &projectroot)
1278 CString str=path;
1280 str.Replace(_T('\\'),_T('/'));
1282 if (str.GetLength()>0)
1283 if (str[str.GetLength()-1] == _T('/'))
1284 str = str.Left(str.GetLength() - 1);
1286 int ret;
1287 ret = CheckIgnore(str, projectroot);
1288 while (ret < 0)
1290 int start = str.ReverseFind(_T('/'));
1291 if(start < 0)
1292 return (ret == 1);
1294 str = str.Left(start);
1295 ret = CheckIgnore(str, projectroot);
1298 return (ret == 1);
1300 int CGitIgnoreList::CheckFileAgainstIgnoreList(const CString &ignorefile, const CStringA &patha, const char * base, int &type)
1302 if (m_Map.find(ignorefile) != m_Map.end())
1304 int ret = -1;
1305 if(m_Map[ignorefile].m_pExcludeList)
1306 ret = git_check_excluded_1(patha, patha.GetLength(), base, &type, m_Map[ignorefile].m_pExcludeList);
1307 if (ret == 0 || ret == 1)
1308 return ret;
1310 return -1;
1312 int CGitIgnoreList::CheckIgnore(const CString &path,const CString &projectroot)
1314 __int64 time = 0;
1315 bool dir = 0;
1316 CString temp = projectroot + _T("\\") + path;
1317 temp.Replace(_T('/'), _T('\\'));
1319 CStringA patha = CUnicodeUtils::GetMulti(path, CP_UTF8);
1320 patha.Replace('\\', '/');
1322 if(g_Git.GetFileModifyTime(temp, &time, &dir))
1323 return -1;
1325 int type = 0;
1326 if (dir)
1328 type = DT_DIR;
1330 // strip directory name
1331 // we do not need to check for a .ignore file inside a directory we might ignore
1332 int i = temp.ReverseFind(_T('\\'));
1333 if (i >= 0)
1334 temp = temp.Left(i);
1336 else
1337 type = DT_REG;
1339 char * base = NULL;
1340 int pos = patha.ReverseFind('/');
1341 base = pos >= 0 ? patha.GetBuffer() + pos + 1 : patha.GetBuffer();
1343 int ret = -1;
1345 CAutoReadLock lock(&this->m_SharedMutex);
1346 while (!temp.IsEmpty())
1348 CString tempOrig = temp;
1349 temp += _T("\\.git");
1351 if (CGit::GitPathFileExists(temp))
1353 CString gitignore = temp;
1354 gitignore += _T("ignore");
1355 if ((ret = CheckFileAgainstIgnoreList(gitignore, patha, base, type)) != -1)
1356 break;
1358 CString adminDir = g_AdminDirMap.GetAdminDir(tempOrig);
1359 CString wcglobalgitignore = adminDir + _T("info\\exclude");
1360 if ((ret = CheckFileAgainstIgnoreList(wcglobalgitignore, patha, base, type)) != -1)
1361 break;
1363 m_SharedMutex.AcquireShared();
1364 CString excludesFile = m_CoreExcludesfiles[adminDir];
1365 m_SharedMutex.ReleaseShared();
1366 if (!excludesFile.IsEmpty())
1367 ret = CheckFileAgainstIgnoreList(excludesFile, patha, base, type);
1369 break;
1371 else
1373 temp += _T("ignore");
1374 if ((ret = CheckFileAgainstIgnoreList(temp, patha, base, type)) != -1)
1375 break;
1378 int found = 0;
1379 int i;
1380 for (i = temp.GetLength() - 1; i >= 0; i--)
1382 if (temp[i] == _T('\\'))
1383 found++;
1385 if (found == 2)
1386 break;
1389 temp = temp.Left(i);
1392 patha.ReleaseBuffer();
1394 return ret;
1397 bool CGitHeadFileMap::CheckHeadUpdate(const CString &gitdir)
1399 SHARED_TREE_PTR ptr;
1400 ptr = this->SafeGet(gitdir);
1402 if( ptr.get())
1404 return ptr->CheckHeadUpdate();
1406 else
1408 SHARED_TREE_PTR ptr1(new CGitHeadFileList);
1409 ptr1->ReadHeadHash(gitdir);
1411 this->SafeSet(gitdir, ptr1);
1412 return true;
1416 int CGitHeadFileMap::IsUnderVersionControl(const CString &gitdir, const CString &path, bool isDir, bool *isVersion)
1420 if (path.IsEmpty())
1422 *isVersion = true;
1423 return 0;
1426 CString subpath = path;
1427 subpath.Replace(_T('\\'), _T('/'));
1428 if(isDir)
1429 subpath += _T('/');
1431 subpath.MakeLower();
1433 CheckHeadUpdate(gitdir);
1435 SHARED_TREE_PTR treeptr;
1436 treeptr = SafeGet(gitdir);
1438 if (!treeptr->HeadHashEqualsTreeHash())
1440 SHARED_TREE_PTR treeptr(new CGitHeadFileList());
1441 treeptr->ReadHeadHash(gitdir);
1443 // Init Repository
1444 if (treeptr->HeadFileIsEmpty())
1446 *isVersion = false;
1447 return 0;
1449 else if (treeptr->ReadTree())
1451 *isVersion = false;
1452 return 1;
1454 SafeSet(gitdir, treeptr);
1457 if(isDir)
1458 *isVersion = (SearchInSortVector(*treeptr, subpath.GetBuffer(), subpath.GetLength()) >= 0);
1459 else
1460 *isVersion = (SearchInSortVector(*treeptr, subpath.GetBuffer(), -1) >= 0);
1461 subpath.ReleaseBuffer();
1463 catch(...)
1465 return -1;
1468 return 0;