explorer/dll.c

   1
   2 #include "../common/cache.h"
   3
   4 #include <shlobj.h>
   5 #include <stdio.h>
   6 #include "dll.h"
   7 #include "../common/menuengine.h"
   8 #include "ext.h"
   9 #include "factory.h"
  10 #include "../common/systeminfo.h"
  11 #include "registry.h"
  12
  13 const char *program_name = "Git-Cheetah";
  14 const char *program_version = "Git-Cheetah.Application.1";
  15 const char *program_id = "Git-Cheetah.Application";
  16
  17 /*
  18  * The following is just the necessary infrastructure for having a .dll
  19  * which can be registered as a COM object.
  20  */
  21 static HINSTANCE hInst;
  22
  23 HRESULT PASCAL DllGetClassObject(REFCLSID obj_guid, REFIID factory_guid,
  24                 void **factory_handle)
  25 {
  26         if (IsEqualCLSID(obj_guid, &CLSID_git_shell_ext) ||
  27                         IsEqualCLSID(obj_guid, &CLSID_git_menu))
  28                 return class_factory_query_interface(&factory,
  29                                 factory_guid, factory_handle);
  30
  31         *factory_handle = 0;
  32         return CLASS_E_CLASSNOTAVAILABLE;
  33 }
  34
  35 HRESULT PASCAL DllCanUnloadNow(void)
  36 {
  37    return (object_count || lock_count) ? S_FALSE : S_OK;
  38 }
  39
  40 BOOL WINAPI DllMain(HINSTANCE instance, DWORD reason, LPVOID reserved)
  41 {
  42         hInst = instance;
  43
  44         if (reason == DLL_PROCESS_ATTACH) {
  45                 object_count = lock_count = 0;
  46                 DisableThreadLibraryCalls(instance);
  47         }
  48
  49         return 1;
  50 }
  51
  52 /* replaces a substring pattern with a string replacement within a string
  53    the replacement occurs in-place, hence string must be large enough to
  54    hold the result
  55
  56    the function does not handle recursive replacements, e.g.
  57      strreplace ("foo", "bar", "another bar");
  58
  59    always returns *string
  60 */
  61 static char *strreplace(char *string, const size_t size,
  62                         const char *pattern, const char *replacement)
  63 {
  64         size_t len = strlen(string);
  65         const size_t pattern_len = strlen(pattern);
  66         const size_t replace_len = strlen(replacement);
  67
  68         char *found = strstr(string, pattern);
  69
  70         while (found) {
  71                 /* if the new len is greater than size, bail out */
  72                 if (len + replace_len - pattern_len >= size)
  73                         return string;
  74
  75                 if (pattern_len != replace_len)
  76                         memmove(found + replace_len,
  77                                 found + pattern_len,
  78                                 len - (found - string) - pattern_len + 1);
  79                 memcpy(found, replacement, replace_len);
  80                 len += replace_len - pattern_len;
  81
  82                 found = strstr(string, pattern);
  83         }
  84
  85         return string;
  86 }
  87
  88 /*
  89  * The following is the data for our minimal regedit engine,
  90  * required for registration/unregistration of the extension
  91  */
  92 #define CLASS_CHEETAH CLASSES_ROOT "CLSID\\@@CLSID@@"
  93 #define CONTEXTMENUHANDLER "shellex\\ContextMenuHandlers\\@@PROGRAM_NAME@@"
  94 #define COLUMNHANDLER "shellex\\ColumnHandlers"
  95
  96 static const char *get_module_filename() {
  97         static char module_filename[MAX_PATH] = { '\0' };
  98
  99         if (!*module_filename) {
 100                 DWORD module_size;
 101
 102                 module_size = GetModuleFileName(hInst,
 103                                 module_filename, MAX_PATH);
 104                 if (0 == module_size)
 105                         return NULL;
 106         }
 107
 108         return module_filename;
 109 }
 110
 111 /* as per "How to: Convert Between System::Guid and _GUID" */
 112 static const char *get_class_id()
 113 {
 114         static char class_id[MAX_REGISTRY_PATH] = { '\0' };
 115
 116         if (!*class_id) {
 117                 GUID guid = CLSID_git_shell_ext;
 118                 sprintf(class_id,
 119                         "{%08x-%04x-%04x-%02x%02x-%02x%02x%02x%02x%02x%02x}",
 120                         guid.Data1, guid.Data2, guid.Data3,
 121                         guid.Data4[0], guid.Data4[1], guid.Data4[2],
 122                         guid.Data4[3], guid.Data4[4], guid.Data4[5],
 123                         guid.Data4[6], guid.Data4[7]);
 124         }
 125
 126         return class_id;
 127 }
 128
 129 /*
 130  * Tries to find msysGit in the following order:
 131  * .. and ../.. (relative to the module)
 132  * %PATH%
 133  * as qgit (via InstallLocation of Git)
 134  SOFTWARE\Microsoft\Windows\CurrentVersion\Uninstall\Git_is1\InstallLocation
 135  */
 136 static const reg_value registry_info[] = {
 137         { CURRENT_WINDOWS APPROVED_EXT, "@@CLSID@@", "@@PROGRAM_NAME@@" },
 138         { CURRENT_WINDOWS APPROVED_EXT "\\@@CLSID@@",
 139                 NULL, NULL },
 140         { CURRENT_WINDOWS APPROVED_EXT "\\@@CLSID@@",
 141                 NULL,"@@PROGRAM_NAME@@" },
 142         { CLASS_CHEETAH, NULL, NULL },
 143         { CLASS_CHEETAH, NULL, "@@PROGRAM_NAME@@" },
 144         { CLASS_CHEETAH "\\InProcServer32", NULL, NULL },
 145         { CLASS_CHEETAH "\\InProcServer32", NULL, "@@PROGRAM_PATH@@"},
 146         { CLASS_CHEETAH "\\InProcServer32", "ThreadingModel", "Apartment" },
 147         { CLASSES_ROOT "*\\" CONTEXTMENUHANDLER, NULL, NULL },
 148         { CLASSES_ROOT "*\\" CONTEXTMENUHANDLER, NULL, "@@CLSID@@" },
 149         { CLASSES_ROOT "Directory\\" CONTEXTMENUHANDLER, NULL, NULL },
 150         { CLASSES_ROOT "Directory\\" CONTEXTMENUHANDLER, NULL, "@@CLSID@@" },
 151         { CLASSES_ROOT "Directory\\Background\\" CONTEXTMENUHANDLER,
 152                 NULL, NULL },
 153         { CLASSES_ROOT "Directory\\Background\\" CONTEXTMENUHANDLER,
 154                 NULL, "@@CLSID@@" },
 155         { CLASSES_ROOT "Drive\\" CONTEXTMENUHANDLER, NULL, NULL },
 156         { CLASSES_ROOT "Drive\\" CONTEXTMENUHANDLER, NULL, "@@CLSID@@"},
 157         { CLASSES_ROOT "Folder\\" CONTEXTMENUHANDLER, NULL, NULL },
 158         { CLASSES_ROOT "Folder\\" CONTEXTMENUHANDLER, NULL, "@@CLSID@@" },
 159         { CLASSES_ROOT "Folder\\" COLUMNHANDLER "\\@@CLSID@@", NULL, NULL },
 160         { CLASSES_ROOT "InternetShortcut\\" CONTEXTMENUHANDLER,
 161                 NULL, NULL },
 162         { CLASSES_ROOT "InternetShortcut\\" CONTEXTMENUHANDLER,
 163                 NULL, "@@CLSID@@" },
 164         { GIT_CHEETAH_REG_PATH, NULL, NULL },
 165         { GIT_CHEETAH_REG_PATH,
 166                 GIT_CHEETAH_REG_PATHTOMSYS, "@@MSYSGIT_PATH@@" },
 167         { NULL, NULL, NULL }
 168 };
 169
 170 static const reg_value debug_info[] = {
 171         { CURRENT_WINDOWS "Explorer", "DesktopProcess", "1" },
 172         { CURRENT_WINDOWS "Explorer\\AlwaysUnloadDll", NULL, NULL },
 173         { CURRENT_WINDOWS "Explorer\\Advanced", "SeparateProcess", "1" },
 174         { NULL, NULL, NULL }
 175 };
 176
 177 static char msysgit[MAX_PATH] = { '\0' };
 178
 179 static BOOL find_msysgit_in_path()
 180 {
 181         char *file; /* file part of the path to git.exe */
 182         DWORD dwFound; /* length of path to git.exe */
 183
 184         dwFound = SearchPath(NULL, "git.exe", NULL, MAX_PATH, msysgit, &file);
 185         /* if git is not in the PATH or its path is too long */
 186         if (0 == dwFound ||
 187                 dwFound > MAX_PATH)
 188                 return FALSE;
 189
 190         /*
 191          * git.exe is in "\bin\" from what we really need
 192          * the minimal case we can handle is c:\bin\git.exe
 193          * otherwise declare failure
 194          */
 195         if (file < msysgit + 7)
 196                 return FALSE;
 197         if (strnicmp(file - 5, "\\bin\\", 5))
 198                 return FALSE;
 199         file[-5] = '\0';
 200
 201         return TRUE;
 202 }
 203
 204 static BOOL find_msysgit_relative(const char *path)
 205 {
 206         char *c;
 207
 208         strcpy(msysgit, get_module_filename());
 209         c = strrchr(msysgit, '\\');
 210         c[1] = '\0';
 211         strcat(msysgit, path);
 212         strcat(msysgit, "\\bin\\git.exe");
 213         if (INVALID_FILE_ATTRIBUTES == GetFileAttributes(msysgit)) {
 214                 msysgit[0] = '\0'; /* restore the original result */
 215                 return FALSE;
 216         }
 217
 218         c[1] = '\0';
 219         strcat(msysgit, path);
 220         return TRUE;
 221 }
 222
 223 static BOOL find_msysgit_uninstall(HKEY root)
 224 {
 225         HKEY key;
 226         HRESULT result;
 227         DWORD valuelen = MAX_PATH;
 228
 229         result = RegOpenKeyEx(root,
 230                         CURRENT_WINDOWS "\\Uninstall\\Git_is1",
 231                         0, KEY_READ, &key);
 232         if (ERROR_SUCCESS != result)
 233                 return FALSE;
 234
 235         result = RegQueryValue(key, "InstallLocation",
 236                         (LPBYTE)msysgit, &valuelen);
 237         return ERROR_SUCCESS == result;
 238 }
 239
 240 static HKEY setup_root;
 241
 242 static const char *find_msysgit()
 243 {
 244         if ('\0' == msysgit[0]) {
 245                 if (find_msysgit_relative(".."))
 246                         return msysgit;
 247
 248                 if (find_msysgit_relative("..\\.."))
 249                         return msysgit;
 250
 251                 if (find_msysgit_in_path())
 252                         return msysgit;
 253
 254                 if (setup_root)
 255                         find_msysgit_uninstall(setup_root);
 256         }
 257
 258         return msysgit;
 259 }
 260
 261 /*
 262  * required by registry.c
 263  * supports @@PROGRAM_NAME@@, @@PROGRAM_PATH@@, @@CLSID@@ patterns
 264  */
 265 char *get_registry_path(const char *src, char dst[MAX_REGISTRY_PATH])
 266 {
 267         if (NULL == src)
 268                 return NULL;
 269
 270         strcpy(dst, src);
 271         strreplace(dst, MAX_REGISTRY_PATH,
 272                         "@@PROGRAM_NAME@@", program_name);
 273         strreplace(dst, MAX_REGISTRY_PATH,
 274                         "@@PROGRAM_PATH@@", get_module_filename());
 275         strreplace(dst, MAX_REGISTRY_PATH,
 276                         "@@CLSID@@", get_class_id());
 277         strreplace(dst, MAX_REGISTRY_PATH,
 278                         "@@MSYSGIT_PATH@@", find_msysgit());
 279
 280         return dst;
 281 }
 282
 283 HRESULT PASCAL DllRegisterServer(void)
 284 {
 285         HRESULT retval;
 286         setup_root = HKEY_CURRENT_USER;
 287         retval = create_reg_entries (setup_root, registry_info);
 288         SHChangeNotify(SHCNE_ASSOCCHANGED, SHCNF_IDLIST,NULL,NULL);
 289         return retval;
 290 }
 291
 292 HRESULT PASCAL DllUnregisterServer(void)
 293 {
 294         HRESULT retval;
 295         setup_root = HKEY_CURRENT_USER;
 296         retval = delete_reg_entries(setup_root, registry_info);
 297         SHChangeNotify(SHCNE_ASSOCCHANGED, SHCNF_IDLIST,NULL,NULL);
 298         return retval;
 299 }
 300
 301 /* provide means to create/delete keys:
 302    - described in Debugging with The Shell;
 303    - machine-wide registration and debugging info
 304
 305    possible combinations of regsvr32 command line options:
 306              -n  (absent)               (present)
 307    -i:
 308    (absent)      user reg               (invalid)
 309    debug         user reg+debug         user debug
 310    machine       user+machine reg       machine reg
 311    machinedebug  user+machine reg+debug machine reg+debug
 312
 313    Obviously missing option is "machine debug". To accomplish:
 314    - execute "regsvr32 -n -i:machinedebug" and then
 315    - regsvr32 -u -n -i:machine
 316 */
 317 HRESULT PASCAL DllInstall(BOOL bInstall, LPCWSTR pszCmdLine)
 318 {
 319         BOOL bDebug = NULL != wcsstr(pszCmdLine, L"debug");
 320         HRESULT result = ERROR_SUCCESS;
 321
 322         setup_root = wcsstr(pszCmdLine, L"machine") ?
 323                         HKEY_LOCAL_MACHINE : HKEY_CURRENT_USER;
 324
 325         if (bInstall) {
 326                 if (bDebug)
 327                         result = create_reg_entries(setup_root, debug_info);
 328
 329                 /* for user-only registration, use DllRegister */
 330                 if (ERROR_SUCCESS == result &&
 331                     HKEY_LOCAL_MACHINE == setup_root)
 332                         result = create_reg_entries(setup_root,
 333                                         registry_info);
 334         } else { /* uninstall */
 335                 if (bDebug)
 336                         result = delete_reg_entries(setup_root, debug_info);
 337
 338                 /* for user-only unregistration, use DllUnregister */
 339                 if (ERROR_SUCCESS == result &&
 340                     HKEY_LOCAL_MACHINE == setup_root)
 341                         result = delete_reg_entries(setup_root,
 342                                         registry_info);
 343         }
 344         SHChangeNotify(SHCNE_ASSOCCHANGED, SHCNF_IDLIST,NULL,NULL);
 345         return result;
 346 }