src/TortoiseShell/ShellExt.cpp

   1 // TortoiseSVN - a Windows shell extension for easy version control
   2
   3 // Copyright (C) 2003-2008 - TortoiseSVN
   4
   5 // This program is free software; you can redistribute it and/or
   6 // modify it under the terms of the GNU General Public License
   7 // as published by the Free Software Foundation; either version 2
   8 // of the License, or (at your option) any later version.
   9
  10 // This program is distributed in the hope that it will be useful,
  11 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 // GNU General Public License for more details.
  14
  15 // You should have received a copy of the GNU General Public License
  16 // along with this program; if not, write to the Free Software Foundation,
  17 // 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  18 //
  19 #include "stdafx.h"
  20
  21 #pragma warning (disable : 4786)
  22
  23 // Initialize GUIDs (should be done only and at-least once per DLL/EXE)
  24 #include <initguid.h>
  25 #include "Guids.h"
  26 #include "git.h"
  27 #include "ShellExt.h"
  28 //#include "..\version.h"
  29 //#include "libintl.h"
  30 #undef swprintf
  31
  32 std::set<CShellExt *> g_exts;
  33
  34 // *********************** CShellExt *************************
  35 CShellExt::CShellExt(FileState state)
  36 {
  37         OSVERSIONINFOEX inf;
  38         SecureZeroMemory(&inf, sizeof(OSVERSIONINFOEX));
  39         inf.dwOSVersionInfoSize = sizeof(OSVERSIONINFOEX);
  40         GetVersionEx((OSVERSIONINFO *)&inf);
  41         fullver = MAKEWORD(inf.dwMinorVersion, inf.dwMajorVersion);
  42
  43     m_State = state;
  44
  45     m_cRef = 0L;
  46     g_cRefThisDll++;
  47
  48         g_exts.insert(this);
  49
  50     INITCOMMONCONTROLSEX used = {
  51         sizeof(INITCOMMONCONTROLSEX),
  52                         ICC_LISTVIEW_CLASSES | ICC_WIN95_CLASSES | ICC_BAR_CLASSES | ICC_USEREX_CLASSES
  53     };
  54     InitCommonControlsEx(&used);
  55         LoadLangDll();
  56
  57         if (fullver >= 0x0600)
  58         {
  59                 HMODULE hUxTheme = ::GetModuleHandle (_T("UXTHEME.DLL"));
  60
  61                 pfnGetBufferedPaintBits = (FN_GetBufferedPaintBits)::GetProcAddress(hUxTheme, "GetBufferedPaintBits");
  62                 pfnBeginBufferedPaint = (FN_BeginBufferedPaint)::GetProcAddress(hUxTheme, "BeginBufferedPaint");
  63                 pfnEndBufferedPaint = (FN_EndBufferedPaint)::GetProcAddress(hUxTheme, "EndBufferedPaint");
  64         }
  65 }
  66
  67 CShellExt::~CShellExt()
  68 {
  69         std::map<UINT, HBITMAP>::iterator it;
  70         for (it = bitmaps.begin(); it != bitmaps.end(); ++it)
  71         {
  72                 ::DeleteObject(it->second);
  73         }
  74         bitmaps.clear();
  75         g_cRefThisDll--;
  76         g_exts.erase(this);
  77 }
  78
  79 void LoadLangDll()
  80 {
  81         if ((g_langid != g_ShellCache.GetLangID())&&((g_langTimeout == 0)||(g_langTimeout < GetTickCount())))
  82         {
  83                 g_langid = g_ShellCache.GetLangID();
  84                 DWORD langId = g_langid;
  85                 TCHAR langDll[MAX_PATH*4];
  86                 HINSTANCE hInst = NULL;
  87                 TCHAR langdir[MAX_PATH] = {0};
  88                 char langdirA[MAX_PATH] = {0};
  89                 if (GetModuleFileName(g_hmodThisDll, langdir, MAX_PATH)==0)
  90                         return;
  91                 if (GetModuleFileNameA(g_hmodThisDll, langdirA, MAX_PATH)==0)
  92                         return;
  93                 TCHAR * dirpoint = _tcsrchr(langdir, '\\');
  94                 char * dirpointA = strrchr(langdirA, '\\');
  95                 if (dirpoint)
  96                         *dirpoint = 0;
  97                 if (dirpointA)
  98                         *dirpointA = 0;
  99                 dirpoint = _tcsrchr(langdir, '\\');
 100                 dirpointA = strrchr(langdirA, '\\');
 101                 if (dirpoint)
 102                         *dirpoint = 0;
 103                 if (dirpointA)
 104                         *dirpointA = 0;
 105                 strcat_s(langdirA, MAX_PATH, "\\Languages");
 106 //              bindtextdomain ("subversion", langdirA);
 107
 108                 do
 109                 {
 110                         _stprintf_s(langDll, MAX_PATH*4, _T("%s\\Languages\\TortoiseProc%d.dll"), langdir, langId);
 111                         BOOL versionmatch = TRUE;
 112
 113                         struct TRANSARRAY
 114                         {
 115                                 WORD wLanguageID;
 116                                 WORD wCharacterSet;
 117                         };
 118
 119                         DWORD dwReserved,dwBufferSize;
 120                         dwBufferSize = GetFileVersionInfoSize((LPTSTR)langDll,&dwReserved);
 121
 122                         if (dwBufferSize > 0)
 123                         {
 124                                 LPVOID pBuffer = (void*) malloc(dwBufferSize);
 125
 126                                 if (pBuffer != (void*) NULL)
 127                                 {
 128                                         UINT        nInfoSize = 0,
 129                                                 nFixedLength = 0;
 130                                         LPSTR       lpVersion = NULL;
 131                                         VOID*       lpFixedPointer;
 132                                         TRANSARRAY* lpTransArray;
 133                                         TCHAR       strLangProduktVersion[MAX_PATH];
 134
 135                                         if (GetFileVersionInfo((LPTSTR)langDll,
 136                                                 dwReserved,
 137                                                 dwBufferSize,
 138                                                 pBuffer))
 139                                         {
 140                                                 // Query the current language
 141                                                 if (VerQueryValue(      pBuffer,
 142                                                         _T("\\VarFileInfo\\Translation"),
 143                                                         &lpFixedPointer,
 144                                                         &nFixedLength))
 145                                                 {
 146                                                         lpTransArray = (TRANSARRAY*) lpFixedPointer;
 147
 148                                                         _stprintf_s(strLangProduktVersion, MAX_PATH, _T("\\StringFileInfo\\%04x%04x\\ProductVersion"),
 149                                                                 lpTransArray[0].wLanguageID, lpTransArray[0].wCharacterSet);
 150
 151                                                         if (VerQueryValue(pBuffer,
 152                                                                 (LPTSTR)strLangProduktVersion,
 153                                                                 (LPVOID *)&lpVersion,
 154                                                                 &nInfoSize))
 155                                                         {
 156 //                                                              versionmatch = (_tcscmp((LPCTSTR)lpVersion, _T(STRPRODUCTVER)) == 0);
 157                                                         }
 158
 159                                                 }
 160                                         }
 161                                         free(pBuffer);
 162                                 } // if (pBuffer != (void*) NULL)
 163                         } // if (dwBufferSize > 0)
 164                         else
 165                                 versionmatch = FALSE;
 166
 167                         if (versionmatch)
 168                                 hInst = LoadLibrary(langDll);
 169                         if (hInst != NULL)
 170                         {
 171                                 if (g_hResInst != g_hmodThisDll)
 172                                         FreeLibrary(g_hResInst);
 173                                 g_hResInst = hInst;
 174                         }
 175                         else
 176                         {
 177                                 DWORD lid = SUBLANGID(langId);
 178                                 lid--;
 179                                 if (lid > 0)
 180                                 {
 181                                         langId = MAKELANGID(PRIMARYLANGID(langId), lid);
 182                                 }
 183                                 else
 184                                         langId = 0;
 185                         }
 186                 } while ((hInst == NULL) && (langId != 0));
 187                 if (hInst == NULL)
 188                 {
 189                         // either the dll for the selected language is not present, or
 190                         // it is the wrong version.
 191                         // fall back to English and set a timeout so we don't retry
 192                         // to load the language dll too often
 193                         if (g_hResInst != g_hmodThisDll)
 194                                 FreeLibrary(g_hResInst);
 195                         g_hResInst = g_hmodThisDll;
 196                         g_langid = 1033;
 197                         // set a timeout of 10 seconds
 198                         if (g_ShellCache.GetLangID() != 1033)
 199                                 g_langTimeout = GetTickCount() + 10000;
 200                 }
 201                 else
 202                         g_langTimeout = 0;
 203         } // if (g_langid != g_ShellCache.GetLangID())
 204 }
 205
 206 STDMETHODIMP CShellExt::QueryInterface(REFIID riid, LPVOID FAR *ppv)
 207 {
 208     *ppv = NULL;
 209
 210     if (IsEqualIID(riid, IID_IShellExtInit) || IsEqualIID(riid, IID_IUnknown))
 211     {
 212         *ppv = (LPSHELLEXTINIT)this;
 213     }
 214     else if (IsEqualIID(riid, IID_IContextMenu))
 215     {
 216         *ppv = (LPCONTEXTMENU)this;
 217     }
 218     else if (IsEqualIID(riid, IID_IContextMenu2))
 219     {
 220         *ppv = (LPCONTEXTMENU2)this;
 221     }
 222     else if (IsEqualIID(riid, IID_IContextMenu3))
 223     {
 224         *ppv = (LPCONTEXTMENU3)this;
 225     }
 226     else if (IsEqualIID(riid, IID_IShellIconOverlayIdentifier))
 227     {
 228         *ppv = (IShellIconOverlayIdentifier*)this;
 229     }
 230     else if (IsEqualIID(riid, IID_IShellPropSheetExt))
 231     {
 232         *ppv = (LPSHELLPROPSHEETEXT)this;
 233     }
 234         else if (IsEqualIID(riid, IID_IColumnProvider))
 235         {
 236                 *ppv = (IColumnProvider *)this;
 237         }
 238         else if (IsEqualIID(riid, IID_IShellCopyHook))
 239         {
 240                 *ppv = (ICopyHook *)this;
 241         }
 242     if (*ppv)
 243     {
 244         AddRef();
 245
 246         return NOERROR;
 247     }
 248
 249     return E_NOINTERFACE;
 250 }
 251
 252 STDMETHODIMP_(ULONG) CShellExt::AddRef()
 253 {
 254     return ++m_cRef;
 255 }
 256
 257 STDMETHODIMP_(ULONG) CShellExt::Release()
 258 {
 259     if (--m_cRef)
 260         return m_cRef;
 261
 262     delete this;
 263
 264     return 0L;
 265 }
 266
 267 // IPersistFile members
 268 STDMETHODIMP CShellExt::GetClassID(CLSID *pclsid)
 269 {
 270     *pclsid = CLSID_Tortoisegit_UNCONTROLLED;
 271     return S_OK;
 272 }
 273
 274 STDMETHODIMP CShellExt::Load(LPCOLESTR /*pszFileName*/, DWORD /*dwMode*/)
 275 {
 276     return S_OK;
 277 }
 278
 279 // ICopyHook member
 280 UINT __stdcall CShellExt::CopyCallback(HWND /*hWnd*/, UINT wFunc, UINT /*wFlags*/, LPCTSTR pszSrcFile, DWORD /*dwSrcAttribs*/, LPCTSTR /*pszDestFile*/, DWORD /*dwDestAttribs*/)
 281 {
 282         if (wFunc == FO_COPY)
 283                 return IDYES;   // copying is not a problem for us
 284
 285         m_remoteCacheLink.ReleaseLockForPath(CTGitPath(pszSrcFile));
 286         // we could now wait a little bit to give the cache time to release the handles.
 287         // but the explorer/shell already retries any action for about two seconds
 288         // if it first fails. So if the cache hasn't released the handle yet, the explorer
 289         // will retry anyway, so we just leave here immediately.
 290         return IDYES;
 291 }
 292 // CShellExt member functions (needed for IShellPropSheetExt)
 293 STDMETHODIMP CShellExt::AddPages (LPFNADDPROPSHEETPAGE lpfnAddPage,
 294                                   LPARAM lParam)
 295 {
 296 #if 0
 297         for (std::vector<stdstring>::iterator I = files_.begin(); I != files_.end(); ++I)
 298         {
 299                 GitStatus svn = GitStatus();
 300                 if (svn.GetStatus(CTGitPath(I->c_str())) == (-2))
 301                         return NOERROR;                 // file/directory not under version control
 302
 303                 if (svn.status->entry == NULL)
 304                         return NOERROR;
 305         }
 306
 307         if (files_.size() == 0)
 308                 return NOERROR;
 309
 310         LoadLangDll();
 311     PROPSHEETPAGE psp;
 312         SecureZeroMemory(&psp, sizeof(PROPSHEETPAGE));
 313         HPROPSHEETPAGE hPage;
 314         CGitPropertyPage *sheetpage = new CGitPropertyPage(files_);
 315
 316     psp.dwSize = sizeof (psp);
 317     psp.dwFlags = PSP_USEREFPARENT | PSP_USETITLE | PSP_USEICONID | PSP_USECALLBACK;
 318         psp.hInstance = g_hResInst;
 319         psp.pszTemplate = MAKEINTRESOURCE(IDD_PROPPAGE);
 320     psp.pszIcon = MAKEINTRESOURCE(IDI_APPSMALL);
 321     psp.pszTitle = _T("Subversion");
 322     psp.pfnDlgProc = (DLGPROC) PageProc;
 323     psp.lParam = (LPARAM) sheetpage;
 324     psp.pfnCallback = PropPageCallbackProc;
 325     psp.pcRefParent = &g_cRefThisDll;
 326
 327     hPage = CreatePropertySheetPage (&psp);
 328
 329         if (hPage != NULL)
 330         {
 331         if (!lpfnAddPage (hPage, lParam))
 332         {
 333             delete sheetpage;
 334             DestroyPropertySheetPage (hPage);
 335         }
 336         }
 337 #endif
 338     return NOERROR;
 339 }
 340
 341
 342
 343 STDMETHODIMP CShellExt::ReplacePage (UINT /*uPageID*/, LPFNADDPROPSHEETPAGE /*lpfnReplaceWith*/, LPARAM /*lParam*/)
 344 {
 345     return E_FAIL;
 346 }
 347