src/TortoiseProc/gitlogcache.h

   1 // TortoiseGit - a Windows shell extension for easy version control
   2
   3 // Copyright (C) 2008-2013 - TortoiseGit
   4
   5 // This program is free software; you can redistribute it and/or
   6 // modify it under the terms of the GNU General Public License
   7 // as published by the Free Software Foundation; either version 2
   8 // of the License, or (at your option) any later version.
   9
  10 // This program is distributed in the hope that it will be useful,
  11 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 // GNU General Public License for more details.
  14
  15 // You should have received a copy of the GNU General Public License
  16 // along with this program; if not, write to the Free Software Foundation,
  17 // 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  18 //
  19 #pragma once
  20
  21 #include "Git.h"
  22 #include "TGitPath.h"
  23 #include "GitRev.h"
  24 #include "GitHash.h"
  25
  26 #define LOG_INDEX_MAGIC         0x88AA5566
  27 #define LOG_DATA_MAGIC          0x99BB0FFF
  28 #define LOG_DATA_ITEM_MAGIC 0x0FCC9ACC
  29 #define LOG_DATA_FILE_MAGIC 0x19EE9DFF
  30 #define LOG_INDEX_VERSION   0xF
  31
  32 #pragma pack (1)
  33 struct SLogCacheIndexHeader
  34 {
  35         DWORD m_Magic;
  36         DWORD m_Version;
  37         DWORD m_ItemCount;
  38 };
  39
  40 struct SLogCacheIndexItem
  41 {
  42         CGitHash  m_Hash;
  43         ULONGLONG m_Offset;
  44 };
  45
  46 struct SLogCacheIndexFile
  47 {
  48         struct SLogCacheIndexHeader m_Header;
  49         struct SLogCacheIndexItem       m_Item[1]; //dynmatic size
  50 };
  51
  52 struct SLogCacheRevFileHeader
  53 {
  54         DWORD m_Magic;
  55         DWORD m_Version;
  56         DWORD m_Action;
  57         DWORD m_Stage;
  58         DWORD m_ParentNo;
  59         DWORD m_Add;
  60         DWORD m_Del;
  61         DWORD m_FileNameSize;
  62         DWORD m_OldFileNameSize;
  63         TCHAR m_FileName[1];
  64 };
  65 # pragma pack ()
  66
  67 struct SLogCacheRevItemHeader
  68 {
  69         DWORD m_Magic;
  70         DWORD m_Version;
  71         DWORD m_FileCount;
  72 };
  73
  74 struct SLogCacheDataFileHeader
  75 {
  76         DWORD m_Magic;
  77         DWORD m_Version;
  78 };
  79
  80 class CGitHashMap:public std::map<CGitHash,GitRev>
  81 {
  82 public:
  83         bool IsExist(CGitHash &hash)
  84         {
  85                 return find(hash) != end();
  86         }
  87 };
  88
  89 #define INDEX_FILE_NAME _T("tortoisegit.index")
  90 #define DATA_FILE_NAME _T("tortoisegit.data")
  91 #define LOCK_FILE_NAME _T("tortoisegit.lock")
  92
  93
  94 class CLogCache
  95 {
  96 public:
  97
  98 protected:
  99         BOOL m_bEnabled;
 100
 101         HANDLE m_IndexFile;
 102         HANDLE m_IndexFileMap;
 103         SLogCacheIndexFile *m_pCacheIndex;
 104
 105
 106
 107         HANDLE m_DataFile;
 108         HANDLE m_DataFileMap;
 109         BYTE  *m_pCacheData;
 110         DWORD m_DataFileLength;
 111
 112         void CloseDataHandles();
 113         void CloseIndexHandles();
 114         void Sort();
 115
 116         BOOL CheckHeader(SLogCacheIndexHeader *header)
 117         {
 118                 __try
 119                 {
 120                         if(header->m_Magic != LOG_INDEX_MAGIC)
 121                                 return FALSE;
 122
 123                         if(header->m_Version != LOG_INDEX_VERSION)
 124                                 return FALSE;
 125
 126                         return TRUE;
 127                 }
 128                 __finally
 129                 {
 130                 }
 131                 return FALSE;
 132         }
 133
 134         BOOL CheckHeader(SLogCacheRevFileHeader *header)
 135         {
 136                 __try
 137                 {
 138                         if(header->m_Magic != LOG_DATA_FILE_MAGIC)
 139                                 return FALSE;
 140
 141                         if(header->m_Version != LOG_INDEX_VERSION)
 142                                 return FALSE;
 143
 144                         return TRUE;
 145                 }
 146                 __finally
 147                 {
 148                 }
 149                 return FALSE;
 150         }
 151
 152         BOOL CheckHeader(SLogCacheRevItemHeader *header)
 153         {
 154                 __try
 155                 {
 156                         if(header->m_Magic != LOG_DATA_ITEM_MAGIC)
 157                                 return FALSE;
 158
 159                         if(header->m_Version != LOG_INDEX_VERSION)
 160                                 return FALSE;
 161
 162                         return TRUE;
 163                 }
 164                 __finally
 165                 {
 166                 }
 167                 return FALSE;
 168         }
 169
 170         BOOL CheckHeader(SLogCacheDataFileHeader *header)
 171         {
 172                 __try
 173                 {
 174                         if(header->m_Magic != LOG_DATA_MAGIC)
 175                                 return FALSE;
 176
 177                         if(header->m_Version != LOG_INDEX_VERSION)
 178                                 return FALSE;
 179
 180                         return TRUE;
 181                 }
 182                 __finally
 183                 {
 184                 }
 185                 return FALSE;
 186         }
 187
 188         int SaveOneItem(GitRev &Rev, LONG offset);
 189
 190         CString m_GitDir;
 191         int RebuildCacheFile();
 192
 193 public:
 194         CLogCache();
 195         ~CLogCache();
 196         int FetchCacheIndex(CString GitDir);
 197         int LoadOneItem(GitRev &Rev,ULONGLONG offset);
 198         ULONGLONG GetOffset(CGitHash &hash, SLogCacheIndexFile *pData =NULL);
 199
 200         CGitHashMap m_HashMap;
 201
 202         GitRev * GetCacheData(CGitHash &Rev);
 203         int AddCacheEntry(GitRev &Rev);
 204         int SaveCache();
 205
 206         int ClearAllParent();
 207
 208 };