1 /*
2  * Trash virtual folder support. The trashing engine is implemented in trash.c
3  *
4  * Copyright (C) 2006 Mikolaj Zalewski
5  * Copyright (C) 2009 Andrew Hill
6  * Copyright (C) 2017 Giannis Adamopoulos
7  *
8  * This library is free software; you can redistribute it and/or
9  * modify it under the terms of the GNU Lesser General Public
10  * License as published by the Free Software Foundation; either
11  * version 2.1 of the License, or (at your option) any later version.
12  *
13  * This library is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
16  * Lesser General Public License for more details.
17  *
18  * You should have received a copy of the GNU Lesser General Public
19  * License along with this library; if not, write to the Free Software
20  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
21  */
22 
23 #include <precomp.h>
24 
25 WINE_DEFAULT_DEBUG_CHANNEL (shell);
26 
27 class CRecyclerDropTarget :
28     public CComObjectRootEx<CComMultiThreadModelNoCS>,
29     public IDropTarget
30 {
31     private:
32         BOOL fAcceptFmt;       /* flag for pending Drop */
33         UINT cfShellIDList;
34 
35         static HRESULT _DoDeleteDataObject(IDataObject *pda, DWORD fMask)
36         {
37             HRESULT hr;
38             STGMEDIUM medium;
39             FORMATETC formatetc;
40             InitFormatEtc (formatetc, CF_HDROP, TYMED_HGLOBAL);
41             hr = pda->GetData(&formatetc, &medium);
42             if (FAILED_UNEXPECTEDLY(hr))
43                 return hr;
44 
45             LPDROPFILES lpdf = (LPDROPFILES) GlobalLock(medium.hGlobal);
46             if (!lpdf)
47             {
48                 ERR("Error locking global\n");
49                 return E_FAIL;
50             }
51 
52             /* Delete them */
53             SHFILEOPSTRUCTW FileOp;
54             ZeroMemory(&FileOp, sizeof(FileOp));
55             FileOp.wFunc = FO_DELETE;
56             FileOp.pFrom = (LPWSTR) (((byte*) lpdf) + lpdf->pFiles);;
57             if ((fMask & CMIC_MASK_SHIFT_DOWN) == 0)
58                 FileOp.fFlags = FOF_ALLOWUNDO;
59             TRACE("Deleting file (just the first) = %s, allowundo: %d\n", debugstr_w(FileOp.pFrom), (FileOp.fFlags == FOF_ALLOWUNDO));
60 
61             int res = SHFileOperationW(&FileOp);
62             if (res)
63             {
64                 ERR("SHFileOperation failed with 0x%x\n", res);
65                 hr = E_FAIL;
66             }
67 
68             ReleaseStgMedium(&medium);
69 
70             return hr;
71         }
72 
73         struct DeleteThreadData {
74             IStream *s;
75             DWORD fMask;
76         };
77 
78         static DWORD WINAPI _DoDeleteThreadProc(LPVOID lpParameter)
79         {
80             DeleteThreadData *data = static_cast<DeleteThreadData*>(lpParameter);
81             CoInitialize(NULL);
82             IDataObject *pDataObject;
83             HRESULT hr = CoGetInterfaceAndReleaseStream (data->s, IID_PPV_ARG(IDataObject, &pDataObject));
84             if (SUCCEEDED(hr))
85             {
86                 _DoDeleteDataObject(pDataObject, data->fMask);
87             }
88             pDataObject->Release();
89             CoUninitialize();
90             HeapFree(GetProcessHeap(), 0, data);
91             return 0;
92         }
93 
94         static void _DoDeleteAsync(IDataObject *pda, DWORD fMask)
95         {
96             DeleteThreadData *data = static_cast<DeleteThreadData*>(HeapAlloc(GetProcessHeap(), 0, sizeof(DeleteThreadData)));
97             data->fMask = fMask;
98             CoMarshalInterThreadInterfaceInStream(IID_IDataObject, pda, &data->s);
99             SHCreateThread(_DoDeleteThreadProc, data, NULL, NULL);
100         }
101 
102     public:
103 
104         CRecyclerDropTarget()
105         {
106             fAcceptFmt = FALSE;
107             cfShellIDList = RegisterClipboardFormatW(CFSTR_SHELLIDLIST);
108         }
109 
110         HRESULT WINAPI DragEnter(IDataObject *pDataObject,
111                                             DWORD dwKeyState, POINTL pt, DWORD *pdwEffect)
112         {
113             TRACE("Recycle bin drag over (%p)\n", this);
114             /* The recycle bin accepts pretty much everything, and sets a CSIDL flag. */
115             fAcceptFmt = TRUE;
116 
117             *pdwEffect = DROPEFFECT_MOVE;
118             return S_OK;
119         }
120 
121         HRESULT WINAPI DragOver(DWORD dwKeyState, POINTL pt, DWORD *pdwEffect)
122         {
123             TRACE("(%p)\n", this);
124 
125             if (!pdwEffect)
126                 return E_INVALIDARG;
127 
128             *pdwEffect = DROPEFFECT_MOVE;
129 
130             return S_OK;
131         }
132 
133         HRESULT WINAPI DragLeave()
134         {
135             TRACE("(%p)\n", this);
136 
137             fAcceptFmt = FALSE;
138 
139             return S_OK;
140         }
141 
142         HRESULT WINAPI Drop(IDataObject *pDataObject,
143                                        DWORD dwKeyState, POINTL pt, DWORD *pdwEffect)
144         {
145             TRACE("(%p) object dropped on recycle bin, effect %u\n", this, *pdwEffect);
146 
147             /* TODO: pdwEffect should be read and make the drop object be permanently deleted in the move case (shift held) */
148 
149             FORMATETC fmt;
150             TRACE("(%p)->(DataObject=%p)\n", this, pDataObject);
151             InitFormatEtc (fmt, cfShellIDList, TYMED_HGLOBAL);
152 
153             /* Handle cfShellIDList Drop objects here, otherwise send the appropriate message to other software */
154             if (SUCCEEDED(pDataObject->QueryGetData(&fmt)))
155             {
156                 DWORD fMask = 0;
157 
158                 if ((dwKeyState & MK_SHIFT) == MK_SHIFT)
159                     fMask |= CMIC_MASK_SHIFT_DOWN;
160 
161                 _DoDeleteAsync(pDataObject, fMask);
162             }
163             else
164             {
165                 /*
166                  * TODO call SetData on the data object with format CFSTR_TARGETCLSID
167                  * set to the Recycle Bin's class identifier CLSID_RecycleBin.
168                  */
169             }
170             return S_OK;
171         }
172 
173         DECLARE_NOT_AGGREGATABLE(CRecyclerDropTarget)
174 
175         DECLARE_PROTECT_FINAL_CONSTRUCT()
176 
177         BEGIN_COM_MAP(CRecyclerDropTarget)
178         COM_INTERFACE_ENTRY_IID(IID_IDropTarget, IDropTarget)
179         END_COM_MAP()
180 };
181 
182 HRESULT CRecyclerDropTarget_CreateInstance(REFIID riid, LPVOID * ppvOut)
183 {
184     return ShellObjectCreator<CRecyclerDropTarget>(riid, ppvOut);
185 }
186