ReactOS 0.4.15-dev-7681-g776c3a3
CSendToMenu.cpp
Go to the documentation of this file.
1/*
2 * provides SendTo shell item service
3 *
4 * Copyright 2019 Katayama Hirofumi MZ (katayama.hirofumi.mz@gmail.com)
5 *
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
10 *
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
15 *
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
19 */
20
21#include "precomp.h"
22
24
26 : m_hSubMenu(NULL)
27 , m_pItems(NULL)
28 , m_idCmdFirst(0)
29{
31 if (FAILED(hr))
32 {
33 ERR("SHGetDesktopFolder: %08lX\n", hr);
34 }
35
37}
38
40{
42
43 if (m_hSubMenu)
44 {
47 }
48}
49
51{
52 DWORD dwEffect = DROPEFFECT_MOVE | DROPEFFECT_COPY | DROPEFFECT_LINK;
53
54 BOOL bShift = (GetAsyncKeyState(VK_SHIFT) < 0);
55 BOOL bCtrl = (GetAsyncKeyState(VK_CONTROL) < 0);
56
57 // THIS CODE IS NOT HUMAN-FRIENDLY. SORRY.
58 // (We have to translate a SendTo action to a Drop action)
59 DWORD dwKeyState = MK_LBUTTON;
60 if (bShift && bCtrl)
61 dwKeyState |= MK_SHIFT | MK_CONTROL;
62 else if (!bShift)
63 dwKeyState |= MK_CONTROL;
64 if (bCtrl)
65 dwKeyState |= MK_SHIFT;
66
67 POINTL ptl = { 0, 0 };
68 HRESULT hr = pDropTarget->DragEnter(pDataObject, dwKeyState, ptl, &dwEffect);
70 {
71 pDropTarget->DragLeave();
72 return hr;
73 }
74
75 if (dwEffect == DROPEFFECT_NONE)
76 {
77 ERR("DROPEFFECT_NONE\n");
78 pDropTarget->DragLeave();
79 return E_FAIL;
80 }
81
82 // THIS CODE IS NOT HUMAN-FRIENDLY. SORRY.
83 // (We have to translate a SendTo action to a Drop action)
84 if (bShift && bCtrl)
85 dwEffect = DROPEFFECT_LINK;
86 else if (!bShift)
87 dwEffect = DROPEFFECT_MOVE;
88 else
89 dwEffect = DROPEFFECT_COPY;
90
91 hr = pDropTarget->Drop(pDataObject, dwKeyState, ptl, &dwEffect);
93 return hr;
94
95 return hr;
96}
97
98// get an IShellFolder from CSIDL
101 int csidl, PIDLIST_ABSOLUTE *ppidl)
102{
103 if (!ppFolder)
104 return E_POINTER;
105 *ppFolder = NULL;
106
107 if (ppidl)
108 *ppidl = NULL;
109
111 HRESULT hr = SHGetSpecialFolderLocation(hwnd, csidl, &pidl);
113 return hr;
114
115 IShellFolder *pFolder = NULL;
116 hr = m_pDesktop->BindToObject(pidl, NULL, IID_PPV_ARG(IShellFolder, &pFolder));
117
118 if (ppidl)
119 *ppidl = pidl.Detach();
120
122 return hr;
123
124 *ppFolder = pFolder;
125 return hr;
126}
127
128// get a UI object from PIDL
130 REFIID riid, LPVOID *ppvOut)
131{
132 *ppvOut = NULL;
133
134 PCITEMID_CHILD pidlLast;
135 CComPtr<IShellFolder> pFolder;
136 HRESULT hr = SHBindToParent(pidl, IID_PPV_ARG(IShellFolder, &pFolder), &pidlLast);
138 return hr;
139
140 hr = pFolder->GetUIObjectOf(hwnd, 1, &pidlLast, riid, NULL, ppvOut);
142 return hr;
143
144 return hr;
145}
146
148{
150 m_pItems = NULL;
151 while (pItems)
152 {
153 SENDTO_ITEM *pCurItem = pItems;
154 pItems = pItems->pNext;
155 delete pCurItem;
156 }
157}
158
160{
162
164
168 return hr;
169
170 CComPtr<IEnumIDList> pEnumIDList;
171 hr = m_pSendTo->EnumObjects(hwnd,
172 SHCONTF_FOLDERS | SHCONTF_NONFOLDERS,
173 &pEnumIDList);
175 return hr;
176
177 hr = S_OK;
179 while (pEnumIDList->Next(1, &child, NULL) == S_OK)
180 {
182
183 STRRET strret;
184 hr = m_pSendTo->GetDisplayNameOf(pidlChild, SHGDN_NORMAL, &strret);
186 continue;
187
188 CComHeapPtr<WCHAR> pszText;
189 hr = StrRetToStrW(&strret, pidlChild, &pszText);
191 continue;
192
194 pidlAbsolute.Attach(ILCombine(pidlSendTo, pidlChild));
195
196 SHFILEINFOW fi = { NULL };
199 SHGetFileInfoW(reinterpret_cast<LPWSTR>(static_cast<PIDLIST_ABSOLUTE>(pidlAbsolute)), 0,
200 &fi, sizeof(fi), uFlags);
201
202 SENDTO_ITEM *pNewItem =
203 new SENDTO_ITEM(pidlChild.Detach(), pszText.Detach(), fi.hIcon);
204 if (m_pItems)
205 {
206 pNewItem->pNext = m_pItems;
207 }
208 m_pItems = pNewItem;
209 }
210
211 return hr;
212}
213
215{
216 if (m_pItems == NULL)
217 {
220 return 0;
221 }
222
223 m_idCmdFirst = idCmdFirst;
224
225 UINT idCmd = idCmdFirst;
226 for (SENDTO_ITEM *pCurItem = m_pItems; pCurItem; pCurItem = pCurItem->pNext)
227 {
229 if (InsertMenuW(hMenu, Pos, uFlags, idCmd, pCurItem->pszText))
230 {
231 MENUITEMINFOW mii;
232 mii.cbSize = sizeof(mii);
234 mii.dwItemData = reinterpret_cast<ULONG_PTR>(pCurItem);
235 mii.hbmpItem = HBMMENU_CALLBACK;
236 SetMenuItemInfoW(hMenu, idCmd, FALSE, &mii);
237 ++idCmd;
238
239 // successful
240 }
241 }
242
243 if (idCmd == idCmdFirst)
244 {
246 AppendMenuW(hMenu, MF_GRAYED | MF_DISABLED | MF_STRING, idCmd, strNone);
247 ++idCmd;
248 }
249
250 return idCmd - idCmdFirst;
251}
252
254{
255 UINT idCmd = m_idCmdFirst + IdOffset;
256
257 MENUITEMINFOW mii = { sizeof(mii) };
258 mii.fMask = MIIM_DATA;
259 if (GetMenuItemInfoW(m_hSubMenu, idCmd, FALSE, &mii))
260 return reinterpret_cast<SENDTO_ITEM *>(mii.dwItemData);
261
262 ERR("GetMenuItemInfoW: %ld\n", GetLastError());
263 return NULL;
264}
265
267{
268 if (!m_pDataObject)
269 {
270 ERR("!m_pDataObject\n");
271 return E_FAIL;
272 }
273
274 HRESULT hr;
275 CComPtr<IDropTarget> pDropTarget;
276 hr = m_pSendTo->GetUIObjectOf(NULL, 1, &pItem->pidlChild, IID_IDropTarget,
277 NULL, (LPVOID *)&pDropTarget);
279 return hr;
280
281 hr = DoDrop(m_pDataObject, pDropTarget);
283 return hr;
284
285 return hr;
286}
287
290 UINT indexMenu,
291 UINT idCmdFirst,
292 UINT idCmdLast,
293 UINT uFlags)
294{
295 TRACE("%p %p %u %u %u %u\n", this,
296 hMenu, indexMenu, idCmdFirst, idCmdLast, uFlags);
297
298 if (uFlags & (CMF_NOVERBS | CMF_VERBSONLY))
299 return MAKE_HRESULT(SEVERITY_SUCCESS, 0, idCmdFirst);
300
301 HMENU hSubMenu = CreateMenu();
302 if (!hSubMenu)
303 {
304 ERR("CreateMenu: %ld\n", GetLastError());
305 return E_FAIL;
306 }
307
308 UINT cItems = InsertSendToItems(hSubMenu, idCmdFirst, 0);
309
311
312 MENUITEMINFOW mii = { sizeof(mii) };
314 mii.fType = MFT_STRING;
315 mii.wID = -1;
316 mii.dwTypeData = strSendTo.GetBuffer();
317 mii.cch = wcslen(mii.dwTypeData);
318 mii.fState = MFS_ENABLED;
319 mii.hSubMenu = hSubMenu;
320 if (!InsertMenuItemW(hMenu, indexMenu, TRUE, &mii))
321 {
322 ERR("InsertMenuItemW: %ld\n", GetLastError());
323 return E_FAIL;
324 }
325
326 HMENU hOldSubMenu = m_hSubMenu;
327 m_hSubMenu = hSubMenu;
328 DestroyMenu(hOldSubMenu);
329
330 return MAKE_HRESULT(SEVERITY_SUCCESS, 0, idCmdFirst + cItems);
331}
332
335{
336 HRESULT hr = E_FAIL;
337
338 WORD idCmd = LOWORD(lpici->lpVerb);
339 TRACE("idCmd: %d\n", idCmd);
340
341 SENDTO_ITEM *pItem = FindItemFromIdOffset(idCmd);
342 if (pItem)
343 {
344 hr = DoSendToItem(pItem, lpici);
345 }
346
347 TRACE("CSendToMenu::InvokeCommand %x\n", hr);
348 return hr;
349}
350
353 UINT uType,
354 UINT *pwReserved,
355 LPSTR pszName,
356 UINT cchMax)
357{
358 FIXME("%p %lu %u %p %p %u\n", this,
359 idCmd, uType, pwReserved, pszName, cchMax);
360
361 return E_NOTIMPL;
362}
363
366{
367 return S_OK;
368}
369
372 LRESULT *plResult)
373{
376
377 switch (uMsg)
378 {
379 case WM_MEASUREITEM:
380 {
381 MEASUREITEMSTRUCT* lpmis = reinterpret_cast<MEASUREITEMSTRUCT*>(lParam);
382 if (!lpmis || lpmis->CtlType != ODT_MENU)
383 break;
384
385 UINT cxMenuCheck = GetSystemMetrics(SM_CXMENUCHECK);
386 if (lpmis->itemWidth < cxMenuCheck)
387 lpmis->itemWidth = cxMenuCheck;
388 if (lpmis->itemHeight < cySmall)
389 lpmis->itemHeight = cySmall;
390
391 if (plResult)
392 *plResult = TRUE;
393 break;
394 }
395 case WM_DRAWITEM:
396 {
397 DRAWITEMSTRUCT* lpdis = reinterpret_cast<DRAWITEMSTRUCT*>(lParam);
398 if (!lpdis || lpdis->CtlType != ODT_MENU)
399 break;
400
401 SENDTO_ITEM *pItem = reinterpret_cast<SENDTO_ITEM *>(lpdis->itemData);
402 HICON hIcon = NULL;
403 if (pItem)
404 hIcon = pItem->hIcon;
405 if (!hIcon)
406 break;
407
408 const RECT& rcItem = lpdis->rcItem;
409 INT x = 4;
410 INT y = lpdis->rcItem.top;
411 y += (rcItem.bottom - rcItem.top - cySmall) / 2;
412 DrawIconEx(lpdis->hDC, x, y, hIcon, cxSmall, cySmall,
413 0, NULL, DI_NORMAL);
414
415 if (plResult)
416 *plResult = TRUE;
417 }
418 }
419
420 return S_OK;
421}
422
425 IDataObject *pdtobj, HKEY hkeyProgID)
426{
427 m_pDataObject = pdtobj;
428 return S_OK;
429}
HRESULT WINAPI SHGetDesktopFolder(IShellFolder **psf)
UINT cchMax
#define WINE_DEFAULT_DEBUG_CHANNEL(t)
Definition: precomp.h:23
#define IDS_NONE
Definition: resource.h:133
void shell(int argc, const char *argv[])
Definition: cmds.c:1231
#define STDMETHODIMP
Definition: basetyps.h:43
#define FIXME(fmt,...)
Definition: debug.h:111
#define ERR(fmt,...)
Definition: debug.h:110
void Release()
Definition: atlcomcli.h:170
T * Detach()
Definition: atlalloc.h:168
void Attach(T *lp)
Definition: atlalloc.h:162
SENDTO_ITEM * FindItemFromIdOffset(UINT IdOffset)
HRESULT GetUIObjectFromPidl(HWND hwnd, PIDLIST_ABSOLUTE pidl, REFIID riid, LPVOID *ppvOut)
SENDTO_ITEM * m_pItems
Definition: CSendToMenu.h:62
CComPtr< IShellFolder > m_pSendTo
Definition: CSendToMenu.h:66
STDMETHODIMP Initialize(PCIDLIST_ABSOLUTE pidlFolder, IDataObject *pdtobj, HKEY hkeyProgID)
UINT InsertSendToItems(HMENU hMenu, UINT idFirst, UINT idMenu)
HRESULT LoadAllItems(HWND hwnd)
CComPtr< IShellFolder > m_pDesktop
Definition: CSendToMenu.h:65
void UnloadAllItems()
HRESULT GetSpecialFolder(HWND hwnd, IShellFolder **ppFolder, int csidl, PIDLIST_ABSOLUTE *ppidl=NULL)
STDMETHODIMP QueryContextMenu(HMENU hMenu, UINT indexMenu, UINT idCmdFirst, UINT idCmdLast, UINT uFlags)
HMENU m_hSubMenu
Definition: CSendToMenu.h:61
STDMETHODIMP HandleMenuMsg(UINT uMsg, WPARAM wParam, LPARAM lParam)
STDMETHODIMP HandleMenuMsg2(UINT uMsg, WPARAM wParam, LPARAM lParam, LRESULT *plResult)
STDMETHODIMP GetCommandString(UINT_PTR idCommand, UINT uFlags, UINT *lpReserved, LPSTR lpszName, UINT uMaxNameLen)
HRESULT DoSendToItem(SENDTO_ITEM *pItem, LPCMINVOKECOMMANDINFO lpici)
HRESULT DoDrop(IDataObject *pDataObject, IDropTarget *pDropTarget)
Definition: CSendToMenu.cpp:50
CComPtr< IDataObject > m_pDataObject
Definition: CSendToMenu.h:67
STDMETHODIMP InvokeCommand(LPCMINVOKECOMMANDINFO lpcmi)
UINT m_idCmdFirst
Definition: CSendToMenu.h:63
WPARAM wParam
Definition: combotst.c:138
LPARAM lParam
Definition: combotst.c:139
#define E_NOTIMPL
Definition: ddrawi.h:99
#define E_FAIL
Definition: ddrawi.h:102
ush Pos
Definition: deflate.h:92
#define NULL
Definition: types.h:112
#define TRUE
Definition: types.h:120
#define FALSE
Definition: types.h:117
UINT uFlags
Definition: api.c:59
#define FAILED_UNEXPECTEDLY(hr)
Definition: precomp.h:121
HRESULT WINAPI SHGetSpecialFolderLocation(HWND hwndOwner, INT nFolder, LPITEMIDLIST *ppidl)
Definition: shellpath.c:3194
HRESULT WINAPI StrRetToStrW(LPSTRRET lpStrRet, const ITEMIDLIST *pidl, LPWSTR *ppszName)
Definition: string.c:1623
#define MAKE_HRESULT(sev, fac, code)
Definition: dmerror.h:30
unsigned int BOOL
Definition: ntddk_ex.h:94
unsigned long DWORD
Definition: ntddk_ex.h:95
unsigned short WORD
Definition: ntddk_ex.h:93
GLint GLint GLint GLint GLint x
Definition: gl.h:1548
GLint GLint GLint GLint GLint GLint y
Definition: gl.h:1548
_CRTIMP size_t __cdecl wcslen(_In_z_ const wchar_t *_Str)
REFIID riid
Definition: atlbase.h:39
HRESULT Drop([in, unique] IDataObject *pDataObj, [in] DWORD grfKeyState, [in] POINTL pt, [in, out] DWORD *pdwEffect)
HRESULT DragLeave()
HRESULT DragEnter([in, unique] IDataObject *pDataObj, [in] DWORD grfKeyState, [in] POINTL pt, [in, out] DWORD *pdwEffect)
#define S_OK
Definition: intsafe.h:52
#define FAILED(hr)
Definition: intsafe.h:51
static HICON
Definition: imagelist.c:84
static const CLSID *static CLSID *static const GUID VARIANT VARIANT *static IServiceProvider DWORD *static HMENU
Definition: ordinal.c:63
static HWND child
Definition: cursoricon.c:298
static int int const SCRIPT_CONTROL const SCRIPT_STATE SCRIPT_ITEM * pItems
Definition: usp10.c:62
HICON hIcon
Definition: msconfig.c:44
unsigned __int3264 UINT_PTR
Definition: mstsclib_h.h:274
unsigned int UINT
Definition: ndis.h:50
#define LOWORD(l)
Definition: pedump.c:82
LPITEMIDLIST WINAPI ILCombine(LPCITEMIDLIST pidl1, LPCITEMIDLIST pidl2)
Definition: pidl.c:703
HRESULT WINAPI SHBindToParent(LPCITEMIDLIST pidl, REFIID riid, LPVOID *ppv, LPCITEMIDLIST *ppidlLast)
Definition: pidl.c:1352
#define REFIID
Definition: guiddef.h:118
DWORD_PTR WINAPI SHGetFileInfoW(LPCWSTR path, DWORD dwFileAttributes, SHFILEINFOW *psfi, UINT sizeofpsfi, UINT flags)
Definition: shell32_main.c:415
#define SHGFI_ICON
Definition: shellapi.h:161
#define SHGFI_TYPENAME
Definition: shellapi.h:164
#define SHGFI_SMALLICON
Definition: shellapi.h:173
#define SHGFI_PIDL
Definition: shellapi.h:177
HRESULT hr
Definition: shlfolder.c:183
#define CSIDL_SENDTO
Definition: shlobj.h:2100
#define IDS_SENDTO_MENU
Definition: shresdef.h:247
#define TRACE(s)
Definition: solgame.cpp:4
PITEMID_CHILD pidlChild
Definition: CSendToMenu.h:36
HICON hIcon
Definition: shellapi.h:369
ULONG_PTR itemData
Definition: winuser.h:3092
ULONG_PTR dwItemData
Definition: winuser.h:3267
LPWSTR dwTypeData
Definition: winuser.h:3268
LONG bottom
Definition: windef.h:309
LONG top
Definition: windef.h:307
int32_t INT
Definition: typedefs.h:58
uint32_t ULONG_PTR
Definition: typedefs.h:65
DWORD WINAPI GetLastError(void)
Definition: except.c:1042
_In_ LONG _In_ HWND hwnd
Definition: winddi.h:4023
LONG_PTR LPARAM
Definition: windef.h:208
LONG_PTR LRESULT
Definition: windef.h:209
UINT_PTR WPARAM
Definition: windef.h:207
#define SEVERITY_SUCCESS
Definition: winerror.h:64
#define E_POINTER
Definition: winerror.h:2365
#define DI_NORMAL
Definition: wingdi.h:72
#define MK_SHIFT
Definition: winuser.h:2368
#define MIIM_ID
Definition: winuser.h:722
#define SM_CXMENUCHECK
Definition: winuser.h:1030
BOOL WINAPI InsertMenuW(_In_ HMENU, _In_ UINT, _In_ UINT, _In_ UINT_PTR, _In_opt_ LPCWSTR)
HMENU WINAPI CreateMenu(void)
Definition: menu.c:829
#define MF_STRING
Definition: winuser.h:138
#define VK_CONTROL
Definition: winuser.h:2202
#define SM_CYSMICON
Definition: winuser.h:1012
BOOL WINAPI SetMenuItemInfoW(_In_ HMENU, _In_ UINT, _In_ BOOL, _In_ LPCMENUITEMINFOW)
#define WM_DRAWITEM
Definition: winuser.h:1644
#define MIIM_STATE
Definition: winuser.h:721
#define SM_CXSMICON
Definition: winuser.h:1011
#define MIIM_SUBMENU
Definition: winuser.h:723
#define MF_ENABLED
Definition: winuser.h:128
#define MK_CONTROL
Definition: winuser.h:2369
#define MIIM_BITMAP
Definition: winuser.h:728
#define MF_BYPOSITION
Definition: winuser.h:203
BOOL WINAPI DrawIconEx(_In_ HDC, _In_ int, _In_ int, _In_ HICON, _In_ int, _In_ int, _In_ UINT, _In_opt_ HBRUSH, _In_ UINT)
Definition: cursoricon.c:2011
#define WM_MEASUREITEM
Definition: winuser.h:1645
#define MFS_ENABLED
Definition: winuser.h:750
BOOL WINAPI DestroyMenu(_In_ HMENU)
BOOL WINAPI GetMenuItemInfoW(_In_ HMENU, _In_ UINT, _In_ BOOL, _Inout_ LPMENUITEMINFOW)
#define MK_LBUTTON
Definition: winuser.h:2366
#define VK_SHIFT
Definition: winuser.h:2201
#define MFT_STRING
Definition: winuser.h:746
SHORT WINAPI GetAsyncKeyState(_In_ int)
#define MAKEINTRESOURCEW(i)
Definition: winuser.h:582
#define ODT_MENU
Definition: winuser.h:2536
#define HBMMENU_CALLBACK
Definition: winuser.h:2629
int WINAPI GetSystemMetrics(_In_ int)
#define MIIM_DATA
Definition: winuser.h:726
#define MIIM_TYPE
Definition: winuser.h:725
BOOL WINAPI InsertMenuItemW(_In_ HMENU, _In_ UINT, _In_ BOOL, _In_ LPCMENUITEMINFOW)
BOOL WINAPI AppendMenuW(_In_ HMENU, _In_ UINT, _In_ UINT_PTR, _In_opt_ LPCWSTR)
#define MF_GRAYED
Definition: winuser.h:129
#define MF_DISABLED
Definition: winuser.h:130
#define IID_PPV_ARG(Itype, ppType)
char * LPSTR
Definition: xmlstorage.h:182
WCHAR * LPWSTR
Definition: xmlstorage.h:184