ext.c

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