OSDN Git Service

Fix when setting ssh client is null. GIT_SSH environment variable is not clear
[tortoisegit/TortoiseGitJp.git] / src / Git / Git.cpp
index 87c5d89..2b7be0d 100644 (file)
 #include "Git.h"\r
 #include "atlconv.h"\r
 #include "GitRev.h"\r
+#include "registry.h"\r
+#include "GitConfig.h"\r
+\r
+\r
+static LPTSTR nextpath(LPCTSTR src, LPTSTR dst, UINT maxlen)\r
+{\r
+       LPCTSTR orgsrc;\r
+\r
+       while (*src == _T(';'))\r
+               src++;\r
+\r
+       orgsrc = src;\r
+\r
+       if (!--maxlen)\r
+               goto nullterm;\r
+\r
+       while (*src && *src != _T(';'))\r
+       {\r
+               if (*src != _T('"'))\r
+               {\r
+                       *dst++ = *src++;\r
+                       if (!--maxlen)\r
+                       {\r
+                               orgsrc = src;\r
+                               goto nullterm;\r
+                       }\r
+               }\r
+               else\r
+               {\r
+                       src++;\r
+                       while (*src && *src != _T('"'))\r
+                       {\r
+                               *dst++ = *src++;\r
+                               if (!--maxlen)\r
+                               {\r
+                                       orgsrc = src;\r
+                                       goto nullterm;\r
+                               }\r
+                       }\r
+\r
+                       if (*src)\r
+                               src++;\r
+               }\r
+       }\r
+\r
+       while (*src == _T(';'))\r
+               src++;\r
+\r
+nullterm:\r
+\r
+       *dst = 0;\r
+\r
+       return (orgsrc != src) ? (LPTSTR)src : NULL;\r
+}\r
+\r
+static inline BOOL FileExists(LPCTSTR lpszFileName)\r
+{\r
+       struct _stat st;\r
+       return _tstat(lpszFileName, &st) == 0;\r
+}\r
+\r
+static BOOL FindGitPath()\r
+{\r
+       size_t size;\r
+       _tgetenv_s(&size, NULL, 0, _T("PATH"));\r
+\r
+       if (!size)\r
+       {\r
+               return FALSE;\r
+       }\r
+\r
+       TCHAR *env = (TCHAR*)alloca(size * sizeof(TCHAR));\r
+       _tgetenv_s(&size, env, size, _T("PATH"));\r
+\r
+       TCHAR buf[_MAX_PATH];\r
+\r
+       // search in all paths defined in PATH\r
+       while ((env = nextpath(env, buf, _MAX_PATH-1)) && *buf)\r
+       {\r
+               TCHAR *pfin = buf + _tcslen(buf)-1;\r
+\r
+               // ensure trailing slash\r
+               if (*pfin != _T('/') && *pfin != _T('\\'))\r
+                       _tcscpy(++pfin, _T("\\"));\r
+\r
+               const int len = _tcslen(buf);\r
+\r
+               if ((len + 7) < _MAX_PATH)\r
+                       _tcscpy(pfin+1, _T("git.exe"));\r
+               else\r
+                       break;\r
+\r
+               if ( FileExists(buf) )\r
+               {\r
+                       // dir found\r
+                       return TRUE;\r
+               }\r
+       }\r
+\r
+       return FALSE;\r
+}\r
+\r
 \r
 #define MAX_DIRBUFFER 1000\r
+CString CGit::ms_LastMsysGitDir;\r
 CGit g_Git;\r
 CGit::CGit(void)\r
 {\r
        GetCurrentDirectory(MAX_DIRBUFFER,m_CurrentDir.GetBuffer(MAX_DIRBUFFER));\r
+\r
+       // make sure git/bin is in PATH before wingit.dll gets (delay) loaded by wgInit()\r
+       if ( !CheckMsysGitDir() )\r
+       {\r
+               // TODO\r
+       }\r
+\r
+       if ( !wgInit() )\r
+       {\r
+               // TODO\r
+       }\r
 }\r
 \r
 CGit::~CGit(void)\r
@@ -20,7 +134,7 @@ int CGit::RunAsync(CString cmd,PROCESS_INFORMATION *piOut,HANDLE *hReadOut,CStri
 {\r
        SECURITY_ATTRIBUTES sa;\r
        HANDLE hRead, hWrite;\r
-       HANDLE hStdioFile;\r
+       HANDLE hStdioFile = NULL;\r
 \r
        sa.nLength = sizeof(SECURITY_ATTRIBUTES);\r
        sa.lpSecurityDescriptor=NULL;\r
@@ -71,17 +185,32 @@ int CGit::RunAsync(CString cmd,PROCESS_INFORMATION *piOut,HANDLE *hReadOut,CStri
 }\r
 //Must use sperate function to convert ANSI str to union code string\r
 //Becuase A2W use stack as internal convert buffer. \r
-void CGit::StringAppend(CString *str,char *p)\r
+void CGit::StringAppend(CString *str,BYTE *p,int code,int length)\r
 {\r
-       USES_CONVERSION;\r
-       str->Append(A2W(p));\r
+     //USES_CONVERSION;\r
+        //str->Append(A2W_CP((LPCSTR)p,code));\r
+       WCHAR * buf;\r
 \r
+       int len ;\r
+       if(length<0)\r
+               len= strlen((const char*)p);\r
+       else\r
+               len=length;\r
+       //if (len==0)\r
+       //      return ;\r
+       //buf = new WCHAR[len*4 + 1];\r
+       buf = str->GetBuffer(len*4+1+str->GetLength())+str->GetLength();\r
+       SecureZeroMemory(buf, (len*4 + 1)*sizeof(WCHAR));\r
+       MultiByteToWideChar(code, 0, (LPCSTR)p, len, buf, len*4);\r
+       str->ReleaseBuffer();\r
+       //str->Append(buf);\r
+       //delete buf;\r
 }      \r
 BOOL CGit::IsInitRepos()\r
 {\r
        CString cmdout;\r
        cmdout.Empty();\r
-       if(g_Git.Run(_T("git.exe rev-parse --revs-only HEAD"),&cmdout))\r
+       if(g_Git.Run(_T("git.exe rev-parse --revs-only HEAD"),&cmdout,CP_UTF8))\r
        {\r
        //      CMessageBox::Show(NULL,cmdout,_T("TortoiseGit"),MB_OK);\r
                return TRUE;\r
@@ -91,7 +220,7 @@ BOOL CGit::IsInitRepos()
 \r
        return FALSE;\r
 }\r
-int CGit::Run(CString cmd, CString* output)\r
+int CGit::Run(CString cmd,BYTE_VECTOR *vector)\r
 {\r
        PROCESS_INFORMATION pi;\r
        HANDLE hRead;\r
@@ -99,10 +228,12 @@ int CGit::Run(CString cmd, CString* output)
                return GIT_ERROR_CREATE_PROCESS;\r
 \r
        DWORD readnumber;\r
-       while(ReadFile(hRead,g_Buffer,1023,&readnumber,NULL))\r
+       BYTE data;\r
+       while(ReadFile(hRead,&data,1,&readnumber,NULL))\r
        {\r
-               g_Buffer[readnumber]=0;\r
-               StringAppend(output,g_Buffer);\r
+               //g_Buffer[readnumber]=0;\r
+               vector->push_back(data);\r
+//             StringAppend(output,g_Buffer,codes);\r
        }\r
 \r
        \r
@@ -120,18 +251,30 @@ int CGit::Run(CString cmd, CString* output)
 \r
        CloseHandle(hRead);\r
        return exitcode;\r
+\r
+}\r
+int CGit::Run(CString cmd, CString* output,int code)\r
+{\r
+       BYTE_VECTOR vector;\r
+       int ret;\r
+       ret=Run(cmd,&vector);\r
+\r
+       vector.push_back(0);\r
+       \r
+       StringAppend(output,&(vector[0]),code);\r
+       return ret;\r
 }\r
 \r
 CString CGit::GetUserName(void)\r
 {\r
        CString UserName;\r
-       Run(_T("git.exe config user.name"),&UserName);\r
+       Run(_T("git.exe config user.name"),&UserName,CP_UTF8);\r
        return UserName;\r
 }\r
 CString CGit::GetUserEmail(void)\r
 {\r
        CString UserName;\r
-       Run(_T("git.exe config user.email"),&UserName);\r
+       Run(_T("git.exe config user.email"),&UserName,CP_UTF8);\r
        return UserName;\r
 }\r
 \r
@@ -140,7 +283,7 @@ CString CGit::GetCurrentBranch(void)
        CString output;\r
        //Run(_T("git.exe branch"),&branch);\r
 \r
-       int ret=g_Git.Run(_T("git.exe branch"),&output);\r
+       int ret=g_Git.Run(_T("git.exe branch"),&output,CP_UTF8);\r
        if(!ret)\r
        {               \r
                int pos=0;\r
@@ -160,59 +303,101 @@ CString CGit::GetCurrentBranch(void)
 int CGit::BuildOutputFormat(CString &format,bool IsFull)\r
 {\r
        CString log;\r
-       log.Format(_T("#<%c>%%n"),LOG_REV_ITEM_BEGIN);\r
+       log.Format(_T("#<%c>%%x00"),LOG_REV_ITEM_BEGIN);\r
        format += log;\r
        if(IsFull)\r
        {\r
-               log.Format(_T("#<%c>%%an%%n"),LOG_REV_AUTHOR_NAME);\r
+               log.Format(_T("#<%c>%%an%%x00"),LOG_REV_AUTHOR_NAME);\r
                format += log;\r
-               log.Format(_T("#<%c>%%ae%%n"),LOG_REV_AUTHOR_EMAIL);\r
+               log.Format(_T("#<%c>%%ae%%x00"),LOG_REV_AUTHOR_EMAIL);\r
                format += log;\r
-               log.Format(_T("#<%c>%%ai%%n"),LOG_REV_AUTHOR_DATE);\r
+               log.Format(_T("#<%c>%%ai%%x00"),LOG_REV_AUTHOR_DATE);\r
                format += log;\r
-               log.Format(_T("#<%c>%%cn%%n"),LOG_REV_COMMIT_NAME);\r
+               log.Format(_T("#<%c>%%cn%%x00"),LOG_REV_COMMIT_NAME);\r
                format += log;\r
-               log.Format(_T("#<%c>%%ce%%n"),LOG_REV_COMMIT_EMAIL);\r
+               log.Format(_T("#<%c>%%ce%%x00"),LOG_REV_COMMIT_EMAIL);\r
                format += log;\r
-               log.Format(_T("#<%c>%%ci%%n"),LOG_REV_COMMIT_DATE);\r
+               log.Format(_T("#<%c>%%ci%%x00"),LOG_REV_COMMIT_DATE);\r
                format += log;\r
-               log.Format(_T("#<%c>%%s%%n"),LOG_REV_COMMIT_SUBJECT);\r
+               log.Format(_T("#<%c>%%s%%x00"),LOG_REV_COMMIT_SUBJECT);\r
                format += log;\r
-               log.Format(_T("#<%c>%%b%%n"),LOG_REV_COMMIT_BODY);\r
+               log.Format(_T("#<%c>%%b%%x00"),LOG_REV_COMMIT_BODY);\r
                format += log;\r
        }\r
-       log.Format(_T("#<%c>%%m%%H%%n"),LOG_REV_COMMIT_HASH);\r
+       log.Format(_T("#<%c>%%m%%H%%x00"),LOG_REV_COMMIT_HASH);\r
        format += log;\r
-       log.Format(_T("#<%c>%%P%%n"),LOG_REV_COMMIT_PARENT);\r
+       log.Format(_T("#<%c>%%P%%x00"),LOG_REV_COMMIT_PARENT);\r
        format += log;\r
 \r
        if(IsFull)\r
        {\r
-               log.Format(_T("#<%c>%%n"),LOG_REV_COMMIT_FILE);\r
+               log.Format(_T("#<%c>%%x00"),LOG_REV_COMMIT_FILE);\r
                format += log;\r
        }\r
        return 0;\r
 }\r
 \r
-int CGit::GetLog(CString& logOut, CString &hash, int count)\r
+int CGit::GetLog(BYTE_VECTOR& logOut, CString &hash,  CTGitPath *path ,int count,int mask)\r
 {\r
 \r
        CString cmd;\r
        CString log;\r
        CString num;\r
        CString since;\r
+\r
+       CString file;\r
+\r
+       if(path)\r
+               file.Format(_T(" -- \"%s\""),path->GetGitPathString());\r
+       \r
        if(count>0)\r
                num.Format(_T("-n%d"),count);\r
 \r
-       cmd.Format(_T("git.exe log %s -C --numstat --raw --pretty=format:\""),\r
-                               num);\r
-       BuildOutputFormat(log);\r
+       CString param;\r
+\r
+       if(mask& LOG_INFO_STAT )\r
+               param += _T(" --numstat ");\r
+       if(mask& LOG_INFO_FILESTATE)\r
+               param += _T(" --raw ");\r
+\r
+       if(mask& LOG_INFO_FULLHISTORY)\r
+               param += _T(" --full-history ");\r
+\r
+       if(mask& LOG_INFO_BOUNDARY)\r
+               param += _T(" --left-right --boundary ");\r
+\r
+       if(mask& CGit::LOG_INFO_ALL_BRANCH)\r
+               param += _T(" --all ");\r
+\r
+       if(mask& CGit::LOG_INFO_DETECT_COPYRENAME)\r
+               param += _T(" -C ");\r
+       \r
+       if(mask& CGit::LOG_INFO_DETECT_RENAME )\r
+               param += _T(" -M ");\r
+\r
+       if(mask& CGit::LOG_INFO_FIRST_PARENT )\r
+               param += _T(" --first-parent ");\r
+       \r
+       if(mask& CGit::LOG_INFO_NO_MERGE )\r
+               param += _T(" --no-merges ");\r
+\r
+       if(mask& CGit::LOG_INFO_FOLLOW)\r
+               param += _T(" --follow ");\r
+\r
+       param+=hash;\r
+\r
+       cmd.Format(_T("git.exe log %s -z --topo-order %s --parents --pretty=format:\""),\r
+                               num,param);\r
+\r
+       BuildOutputFormat(log,!(mask&CGit::LOG_INFO_ONLY_HASH));\r
+\r
        cmd += log;\r
-       cmd += CString(_T("\"  "))+hash;\r
+       cmd += CString(_T("\"  "))+hash+file;\r
+\r
        return Run(cmd,&logOut);\r
 }\r
 \r
-\r
+#if 0\r
 int CGit::GetShortLog(CString &logOut,CTGitPath * path, int count)\r
 {\r
        CString cmd;\r
@@ -230,6 +415,7 @@ int CGit::GetShortLog(CString &logOut,CTGitPath * path, int count)
        //cmd += CString(_T("\" HEAD~40..HEAD"));\r
        return Run(cmd,&logOut);\r
 }\r
+#endif\r
 \r
 #define BUFSIZE 512\r
 void GetTempPath(CString &path)\r
@@ -277,8 +463,6 @@ CString GetTempFile()
 \r
 int CGit::RunLogFile(CString cmd,CString &filename)\r
 {\r
-       HANDLE hRead, hWrite;\r
-\r
        STARTUPINFO si;\r
        PROCESS_INFORMATION pi;\r
        si.cb=sizeof(STARTUPINFO);\r
@@ -311,15 +495,17 @@ int CGit::RunLogFile(CString cmd,CString &filename)
        CloseHandle(pi.hProcess);\r
        CloseHandle(houtfile);\r
        return GIT_SUCCESS;\r
-       return 0;\r
+//     return 0;\r
 }\r
 \r
 git_revnum_t CGit::GetHash(CString &friendname)\r
 {\r
+       // NOTE: could replace this with wgGetRevisionID call\r
+\r
        CString cmd;\r
        CString out;\r
        cmd.Format(_T("git.exe rev-parse %s" ),friendname);\r
-       Run(cmd,&out);\r
+       Run(cmd,&out,CP_UTF8);\r
        int pos=out.ReverseFind(_T('\n'));\r
        if(pos>0)\r
                return out.Left(pos);\r
@@ -332,7 +518,7 @@ int CGit::GetTagList(STRING_VECTOR &list)
        CString cmd,output;\r
        cmd=_T("git.exe tag -l");\r
        int i=0;\r
-       ret=g_Git.Run(cmd,&output);\r
+       ret=g_Git.Run(cmd,&output,CP_UTF8);\r
        if(!ret)\r
        {               \r
                int pos=0;\r
@@ -359,19 +545,19 @@ int CGit::GetBranchList(STRING_VECTOR &list,int *current,BRANCH_TYPE type)
                cmd+=_T(" -r");\r
 \r
        int i=0;\r
-       ret=g_Git.Run(cmd,&output);\r
+       ret=g_Git.Run(cmd,&output,CP_UTF8);\r
        if(!ret)\r
        {               \r
                int pos=0;\r
                CString one;\r
                while( pos>=0 )\r
                {\r
-                       i++;\r
                        one=output.Tokenize(_T("\n"),pos);\r
                        list.push_back(one.Right(one.GetLength()-2));\r
                        if(one[0] == _T('*'))\r
                                if(current)\r
                                        *current=i;\r
+                       i++;\r
                }\r
        }\r
        return ret;\r
@@ -382,7 +568,7 @@ int CGit::GetRemoteList(STRING_VECTOR &list)
        int ret;\r
        CString cmd,output;\r
        cmd=_T("git.exe config  --get-regexp remote.*.url");\r
-       ret=g_Git.Run(cmd,&output);\r
+       ret=g_Git.Run(cmd,&output,CP_UTF8);\r
        if(!ret)\r
        {\r
                int pos=0;\r
@@ -409,7 +595,7 @@ int CGit::GetMapHashToFriendName(MAP_HASH_NAME &map)
        int ret;\r
        CString cmd,output;\r
        cmd=_T("git show-ref -d");\r
-       ret=g_Git.Run(cmd,&output);\r
+       ret=g_Git.Run(cmd,&output,CP_UTF8);\r
        if(!ret)\r
        {\r
                int pos=0;\r
@@ -431,4 +617,88 @@ int CGit::GetMapHashToFriendName(MAP_HASH_NAME &map)
                }\r
        }\r
        return ret;\r
-}
\ No newline at end of file
+}\r
+\r
+BOOL CGit::CheckMsysGitDir()\r
+{\r
+       static BOOL bInitialized = FALSE;\r
+\r
+       if (bInitialized)\r
+       {\r
+               return TRUE;\r
+       }\r
+\r
+       TCHAR *oldpath,*home;\r
+       size_t size;\r
+\r
+       // set HOME if not set already\r
+       _tgetenv_s(&size, NULL, 0, _T("HOME"));\r
+       if (!size)\r
+       {\r
+               _tdupenv_s(&home,&size,_T("USERPROFILE")); \r
+               _tputenv_s(_T("HOME"),home);\r
+               free(home);\r
+       }\r
+\r
+       //setup ssh client\r
+       CRegString sshclient=CRegString(_T("Software\\TortoiseGit\\SSH"));\r
+       CString ssh=sshclient;\r
+\r
+       if(!ssh.IsEmpty())\r
+       {\r
+               _tputenv_s(_T("GIT_SSH"),ssh);\r
+       }else\r
+       {\r
+               _tputenv_s(_T("GIT_SSH"),_T(""));\r
+       }\r
+\r
+       // search PATH if git/bin directory is alredy present\r
+       if ( FindGitPath() )\r
+       {\r
+               bInitialized = TRUE;\r
+               return TRUE;\r
+       }\r
+\r
+       // add git/bin path to PATH\r
+\r
+       CRegString msysdir=CRegString(REG_MSYSGIT_PATH,_T(""),FALSE,HKEY_LOCAL_MACHINE);\r
+       CString str=msysdir;\r
+       if(str.IsEmpty())\r
+       {\r
+               CRegString msysinstalldir=CRegString(REG_MSYSGIT_INSTALL,_T(""),FALSE,HKEY_LOCAL_MACHINE);\r
+               str=msysinstalldir;\r
+               if ( !str.IsEmpty() )\r
+               {\r
+                       str += (str[str.GetLength()-1] != '\\') ? "\\bin" : "bin";\r
+                       msysdir=str;\r
+                       msysdir.write();\r
+               }\r
+               else\r
+               {\r
+                       return false;\r
+               }\r
+       }\r
+       //CGit::m_MsysGitPath=str;\r
+\r
+       //set path\r
+\r
+       _tdupenv_s(&oldpath,&size,_T("PATH")); \r
+\r
+       CString path;\r
+       path.Format(_T("%s;%s"),oldpath,str);\r
+\r
+       _tputenv_s(_T("PATH"),path);\r
+\r
+       free(oldpath);\r
+\r
+\r
+    if( !FindGitPath() )\r
+       {\r
+               return false;\r
+       }\r
+       else\r
+       {\r
+               bInitialized = TRUE;\r
+               return true;\r
+       }\r
+}\r