ext.c

   1 #include <shlobj.h>
   2 #include <stdarg.h>
   3 #include <tchar.h>
   4 #include "menuengine.h"
   5 #include "ext.h"
   6 #include "debug.h"
   7 #include "systeminfo.h"
   8
   9 DWORD object_count = 0;
  10 DWORD lock_count = 0;
  11
  12 /*
  13  * Standard methods for the IUnknown interface:
  14  * add_ref, release, query_interface and initialize.
  15  *
  16  * Both of our COM objects contain pointers to the git_data object.
  17  */
  18
  19 inline ULONG STDMETHODCALLTYPE add_ref_git_data(struct git_data *this_)
  20 {
  21         return ++(this_->count);
  22 }
  23
  24 inline ULONG STDMETHODCALLTYPE release_git_data(struct git_data *this_)
  25 {
  26         if (--(this_->count) == 0) {
  27                 GlobalFree(this_);
  28                 InterlockedDecrement(&object_count);
  29                 return 0;
  30         }
  31         return this_->count;
  32 }
  33
  34 inline STDMETHODIMP query_interface_git_data(struct git_data *this_,
  35                                              REFIID iid, LPVOID FAR *pointer)
  36 {
  37         if (IsEqualIID(iid, &IID_git_shell_ext) ||
  38                         IsEqualIID(iid, &IID_IShellExtInit) ||
  39                         IsEqualIID(iid, &IID_IUnknown)) {
  40                 *pointer = &this_->shell_ext;
  41         } else if (IsEqualIID(iid, &IID_git_menu) ||
  42                         IsEqualIID(iid, &IID_IContextMenu)) {
  43                 *pointer = &this_->menu;
  44         } else
  45                 return E_NOINTERFACE;
  46
  47         add_ref_git_data(this_);
  48         return NOERROR;
  49 }
  50
  51 inline STDMETHODIMP initialize_git_data(struct git_data *this_,
  52                                         LPCITEMIDLIST folder,
  53                                         LPDATAOBJECT data, HKEY id)
  54 {
  55         FORMATETC format
  56                 = {CF_HDROP, NULL, DVASPECT_CONTENT, -1, TYMED_HGLOBAL};
  57         STGMEDIUM stg = {TYMED_HGLOBAL};
  58         HDROP drop;
  59         UINT count;
  60         HRESULT result = S_OK;
  61
  62         /* if we can't find msysPath, don't even try to do anything else */
  63         if (!msys_path())
  64                 return E_NOTIMPL;
  65
  66         /* store the folder, if provided */
  67         if (folder)
  68                 SHGetPathFromIDList(folder, this_->name);
  69
  70         if (!data)
  71                 return S_OK;
  72
  73         if (FAILED(data->lpVtbl->GetData(data, &format, &stg)))
  74                 return E_INVALIDARG;
  75
  76         drop = (HDROP)GlobalLock(stg.hGlobal);
  77
  78         if (!drop)
  79                 return E_INVALIDARG;
  80
  81         count = DragQueryFile(drop, 0xFFFFFFFF, NULL, 0);
  82
  83         if (count == 0)
  84                 result = E_INVALIDARG;
  85         else if (!DragQueryFile(drop, 0, this_->name, sizeof(this_->name)))
  86                 result = E_INVALIDARG;
  87
  88         GlobalUnlock(stg.hGlobal);
  89         ReleaseStgMedium(&stg);
  90
  91         return result;
  92 }
  93
  94 /*
  95  * Define the shell extension.
  96  *
  97  * Its sole purpose is to be able to query_interface() the context menu.
  98  */
  99 DEFINE_STANDARD_METHODS(git_shell_ext)
 100
 101 struct git_shell_ext_virtual_table git_shell_ext_virtual_table = {
 102         query_interface_git_shell_ext,
 103         add_ref_git_shell_ext,
 104         release_git_shell_ext,
 105         initialize_git_shell_ext
 106 };