sync with trunk r46493
[reactos.git] / dll / win32 / riched20 / clipboard.c
1 /*
2 * Richedit clipboard handling
3 *
4 * Copyright (C) 2006 Kevin Koltzau
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 "editor.h"
22
23 WINE_DEFAULT_DEBUG_CHANNEL(richedit);
24
25 static UINT cfRTF = 0;
26
27 typedef struct DataObjectImpl {
28 const IDataObjectVtbl *lpVtbl;
29 LONG ref;
30
31 FORMATETC *fmtetc;
32 UINT fmtetc_cnt;
33
34 HANDLE unicode;
35 HANDLE rtf;
36 } DataObjectImpl;
37
38 typedef struct EnumFormatImpl {
39 const IEnumFORMATETCVtbl *lpVtbl;
40 LONG ref;
41
42 FORMATETC *fmtetc;
43 UINT fmtetc_cnt;
44
45 UINT cur;
46 } EnumFormatImpl;
47
48 static HRESULT EnumFormatImpl_Create(const FORMATETC *fmtetc, UINT size, LPENUMFORMATETC *lplpformatetc);
49
50 static HRESULT WINAPI EnumFormatImpl_QueryInterface(IEnumFORMATETC *iface, REFIID riid, LPVOID *ppvObj)
51 {
52 EnumFormatImpl *This = (EnumFormatImpl*)iface;
53 TRACE("%p %s\n", This, debugstr_guid(riid));
54
55 if (IsEqualGUID(riid, &IID_IUnknown) || IsEqualGUID(riid, &IID_IEnumFORMATETC)) {
56 IEnumFORMATETC_AddRef(iface);
57 *ppvObj = This;
58 return S_OK;
59 }
60 *ppvObj = NULL;
61 return E_NOINTERFACE;
62 }
63
64 static ULONG WINAPI EnumFormatImpl_AddRef(IEnumFORMATETC *iface)
65 {
66 EnumFormatImpl *This = (EnumFormatImpl*)iface;
67 LONG ref = InterlockedIncrement(&This->ref);
68 TRACE("(%p) ref=%d\n", This, ref);
69 return ref;
70 }
71
72 static ULONG WINAPI EnumFormatImpl_Release(IEnumFORMATETC *iface)
73 {
74 EnumFormatImpl *This = (EnumFormatImpl*)iface;
75 ULONG ref = InterlockedDecrement(&This->ref);
76 TRACE("(%p) ref=%d\n", This, ref);
77
78 if(!ref) {
79 GlobalFree(This->fmtetc);
80 heap_free(This);
81 }
82
83 return ref;
84 }
85
86 static HRESULT WINAPI EnumFormatImpl_Next(IEnumFORMATETC *iface, ULONG celt,
87 FORMATETC *rgelt, ULONG *pceltFetched)
88 {
89 EnumFormatImpl *This = (EnumFormatImpl*)iface;
90 ULONG count = 0;
91 TRACE("(%p)->(%d %p %p)\n", This, celt, rgelt, pceltFetched);
92
93 if(!rgelt)
94 return E_INVALIDARG;
95
96 count = min(celt, This->fmtetc_cnt-This->cur);
97 if(count > 0) {
98 memcpy(rgelt, This->fmtetc+This->cur, count*sizeof(FORMATETC));
99 This->cur += count;
100 }
101 if(pceltFetched)
102 *pceltFetched = count;
103 return count == celt ? S_OK : S_FALSE;
104 }
105
106 static HRESULT WINAPI EnumFormatImpl_Skip(IEnumFORMATETC *iface, ULONG celt)
107 {
108 EnumFormatImpl *This = (EnumFormatImpl*)iface;
109 ULONG count = 0;
110 TRACE("(%p)->(%d)\n", This, celt);
111
112 count = min(celt, This->fmtetc_cnt-This->cur);
113 This->cur += count;
114 return count == celt ? S_OK : S_FALSE;
115 }
116
117 static HRESULT WINAPI EnumFormatImpl_Reset(IEnumFORMATETC *iface)
118 {
119 EnumFormatImpl *This = (EnumFormatImpl*)iface;
120 TRACE("(%p)\n", This);
121
122 This->cur = 0;
123 return S_OK;
124 }
125
126 static HRESULT WINAPI EnumFormatImpl_Clone(IEnumFORMATETC *iface, IEnumFORMATETC **ppenum)
127 {
128 EnumFormatImpl *This = (EnumFormatImpl*)iface;
129 HRESULT hr;
130 TRACE("(%p)->(%p)\n", This, ppenum);
131
132 if(!ppenum)
133 return E_INVALIDARG;
134 hr = EnumFormatImpl_Create(This->fmtetc, This->fmtetc_cnt, ppenum);
135 if(SUCCEEDED(hr))
136 hr = IEnumFORMATETC_Skip(*ppenum, This->cur);
137 return hr;
138 }
139
140 static const IEnumFORMATETCVtbl VT_EnumFormatImpl = {
141 EnumFormatImpl_QueryInterface,
142 EnumFormatImpl_AddRef,
143 EnumFormatImpl_Release,
144 EnumFormatImpl_Next,
145 EnumFormatImpl_Skip,
146 EnumFormatImpl_Reset,
147 EnumFormatImpl_Clone
148 };
149
150 static HRESULT EnumFormatImpl_Create(const FORMATETC *fmtetc, UINT fmtetc_cnt, IEnumFORMATETC **lplpformatetc)
151 {
152 EnumFormatImpl *ret;
153 TRACE("\n");
154
155 ret = heap_alloc(sizeof(EnumFormatImpl));
156 ret->lpVtbl = &VT_EnumFormatImpl;
157 ret->ref = 1;
158 ret->cur = 0;
159 ret->fmtetc_cnt = fmtetc_cnt;
160 ret->fmtetc = GlobalAlloc(GMEM_ZEROINIT, fmtetc_cnt*sizeof(FORMATETC));
161 memcpy(ret->fmtetc, fmtetc, fmtetc_cnt*sizeof(FORMATETC));
162 *lplpformatetc = (LPENUMFORMATETC)ret;
163 return S_OK;
164 }
165
166 static HRESULT WINAPI DataObjectImpl_QueryInterface(IDataObject *iface, REFIID riid, LPVOID *ppvObj)
167 {
168 DataObjectImpl *This = (DataObjectImpl*)iface;
169 TRACE("(%p)->(%s)\n", This, debugstr_guid(riid));
170
171 if (IsEqualGUID(riid, &IID_IUnknown) || IsEqualGUID(riid, &IID_IDataObject)) {
172 IDataObject_AddRef(iface);
173 *ppvObj = This;
174 return S_OK;
175 }
176 *ppvObj = NULL;
177 return E_NOINTERFACE;
178 }
179
180 static ULONG WINAPI DataObjectImpl_AddRef(IDataObject* iface)
181 {
182 DataObjectImpl *This = (DataObjectImpl*)iface;
183 ULONG ref = InterlockedIncrement(&This->ref);
184 TRACE("(%p) ref=%d\n", This, ref);
185 return ref;
186 }
187
188 static ULONG WINAPI DataObjectImpl_Release(IDataObject* iface)
189 {
190 DataObjectImpl *This = (DataObjectImpl*)iface;
191 ULONG ref = InterlockedDecrement(&This->ref);
192 TRACE("(%p) ref=%d\n",This, ref);
193
194 if(!ref) {
195 if(This->unicode) GlobalFree(This->unicode);
196 if(This->rtf) GlobalFree(This->rtf);
197 if(This->fmtetc) GlobalFree(This->fmtetc);
198 heap_free(This);
199 }
200
201 return ref;
202 }
203
204 static HRESULT WINAPI DataObjectImpl_GetData(IDataObject* iface, FORMATETC *pformatetc, STGMEDIUM *pmedium)
205 {
206 DataObjectImpl *This = (DataObjectImpl*)iface;
207 TRACE("(%p)->(fmt=0x%08x tym=0x%08x)\n", This, pformatetc->cfFormat, pformatetc->tymed);
208
209 if(pformatetc->lindex != -1)
210 return DV_E_LINDEX;
211
212 if(!(pformatetc->tymed & TYMED_HGLOBAL))
213 return DV_E_TYMED;
214
215 if(This->unicode && pformatetc->cfFormat == CF_UNICODETEXT)
216 pmedium->u.hGlobal = This->unicode;
217 else if(This->rtf && pformatetc->cfFormat == cfRTF)
218 pmedium->u.hGlobal = This->rtf;
219 else
220 return DV_E_FORMATETC;
221
222 pmedium->tymed = TYMED_HGLOBAL;
223 pmedium->pUnkForRelease = (LPUNKNOWN)iface;
224 IUnknown_AddRef(pmedium->pUnkForRelease);
225 return S_OK;
226 }
227
228 static HRESULT WINAPI DataObjectImpl_GetDataHere(IDataObject* iface, FORMATETC *pformatetc, STGMEDIUM *pmedium)
229 {
230 DataObjectImpl *This = (DataObjectImpl*)iface;
231 FIXME("(%p): stub\n", This);
232 return E_NOTIMPL;
233 }
234
235 static HRESULT WINAPI DataObjectImpl_QueryGetData(IDataObject* iface, FORMATETC *pformatetc)
236 {
237 DataObjectImpl *This = (DataObjectImpl*)iface;
238 UINT i;
239 BOOL foundFormat = FALSE;
240 TRACE("(%p)->(fmt=0x%08x tym=0x%08x)\n", This, pformatetc->cfFormat, pformatetc->tymed);
241
242 if(pformatetc->lindex != -1)
243 return DV_E_LINDEX;
244
245 for(i=0; i<This->fmtetc_cnt; i++) {
246 if(This->fmtetc[i].cfFormat == pformatetc->cfFormat) {
247 foundFormat = TRUE;
248 if(This->fmtetc[i].tymed == pformatetc->tymed)
249 return S_OK;
250 }
251 }
252 return foundFormat?DV_E_FORMATETC:DV_E_TYMED;
253 }
254
255 static HRESULT WINAPI DataObjectImpl_GetCanonicalFormatEtc(IDataObject* iface, FORMATETC *pformatetcIn,
256 FORMATETC *pformatetcOut)
257 {
258 DataObjectImpl *This = (DataObjectImpl*)iface;
259 TRACE("(%p)->(%p,%p)\n", This, pformatetcIn, pformatetcOut);
260
261 if(pformatetcOut) {
262 *pformatetcOut = *pformatetcIn;
263 pformatetcOut->ptd = NULL;
264 }
265 return DATA_S_SAMEFORMATETC;
266 }
267
268 static HRESULT WINAPI DataObjectImpl_SetData(IDataObject* iface, FORMATETC *pformatetc,
269 STGMEDIUM *pmedium, BOOL fRelease)
270 {
271 DataObjectImpl *This = (DataObjectImpl*)iface;
272 FIXME("(%p): stub\n", This);
273 return E_NOTIMPL;
274 }
275
276 static HRESULT WINAPI DataObjectImpl_EnumFormatEtc(IDataObject* iface, DWORD dwDirection,
277 IEnumFORMATETC **ppenumFormatEtc)
278 {
279 DataObjectImpl *This = (DataObjectImpl*)iface;
280 TRACE("(%p)->(%d)\n", This, dwDirection);
281
282 if(dwDirection != DATADIR_GET) {
283 FIXME("Unsupported direction: %d\n", dwDirection);
284 /* WinXP riched20 also returns E_NOTIMPL in this case */
285 *ppenumFormatEtc = NULL;
286 return E_NOTIMPL;
287 }
288 return EnumFormatImpl_Create(This->fmtetc, This->fmtetc_cnt, ppenumFormatEtc);
289 }
290
291 static HRESULT WINAPI DataObjectImpl_DAdvise(IDataObject* iface, FORMATETC *pformatetc, DWORD advf,
292 IAdviseSink *pAdvSink, DWORD *pdwConnection)
293 {
294 DataObjectImpl *This = (DataObjectImpl*)iface;
295 FIXME("(%p): stub\n", This);
296 return E_NOTIMPL;
297 }
298
299 static HRESULT WINAPI DataObjectImpl_DUnadvise(IDataObject* iface, DWORD dwConnection)
300 {
301 DataObjectImpl *This = (DataObjectImpl*)iface;
302 FIXME("(%p): stub\n", This);
303 return E_NOTIMPL;
304 }
305
306 static HRESULT WINAPI DataObjectImpl_EnumDAdvise(IDataObject* iface, IEnumSTATDATA **ppenumAdvise)
307 {
308 DataObjectImpl *This = (DataObjectImpl*)iface;
309 FIXME("(%p): stub\n", This);
310 return E_NOTIMPL;
311 }
312
313 static const IDataObjectVtbl VT_DataObjectImpl =
314 {
315 DataObjectImpl_QueryInterface,
316 DataObjectImpl_AddRef,
317 DataObjectImpl_Release,
318 DataObjectImpl_GetData,
319 DataObjectImpl_GetDataHere,
320 DataObjectImpl_QueryGetData,
321 DataObjectImpl_GetCanonicalFormatEtc,
322 DataObjectImpl_SetData,
323 DataObjectImpl_EnumFormatEtc,
324 DataObjectImpl_DAdvise,
325 DataObjectImpl_DUnadvise,
326 DataObjectImpl_EnumDAdvise
327 };
328
329 static HGLOBAL get_unicode_text(ME_TextEditor *editor, const ME_Cursor *start, int nChars)
330 {
331 int pars = 0;
332 WCHAR *data;
333 HANDLE ret;
334 ME_DisplayItem *para;
335 int nEnd = ME_GetCursorOfs(start) + nChars;
336
337 /* count paragraphs in range */
338 para = start->pPara;
339 while((para = para->member.para.next_para) &&
340 para->member.para.nCharOfs <= nEnd)
341 pars++;
342
343 ret = GlobalAlloc(GMEM_MOVEABLE, sizeof(WCHAR) * (nChars + pars + 1));
344 data = GlobalLock(ret);
345 ME_GetTextW(editor, data, nChars + pars, start, nChars, TRUE);
346 GlobalUnlock(ret);
347 return ret;
348 }
349
350 typedef struct tagME_GlobalDestStruct
351 {
352 HGLOBAL hData;
353 int nLength;
354 } ME_GlobalDestStruct;
355
356 static DWORD CALLBACK ME_AppendToHGLOBAL(DWORD_PTR dwCookie, LPBYTE lpBuff, LONG cb, LONG *pcb)
357 {
358 ME_GlobalDestStruct *pData = (ME_GlobalDestStruct *)dwCookie;
359 int nMaxSize;
360 BYTE *pDest;
361
362 nMaxSize = GlobalSize(pData->hData);
363 if (pData->nLength+cb+1 >= cb) {
364 /* round up to 2^17 */
365 int nNewSize = (((nMaxSize+cb+1)|0x1FFFF)+1) & 0xFFFE0000;
366 pData->hData = GlobalReAlloc(pData->hData, nNewSize, 0);
367 }
368 pDest = GlobalLock(pData->hData);
369 memcpy(pDest + pData->nLength, lpBuff, cb);
370 pData->nLength += cb;
371 pDest[pData->nLength] = '\0';
372 GlobalUnlock(pData->hData);
373 *pcb = cb;
374
375 return 0;
376 }
377
378 static HGLOBAL get_rtf_text(ME_TextEditor *editor, const ME_Cursor *start, int nChars)
379 {
380 EDITSTREAM es;
381 ME_GlobalDestStruct gds;
382
383 gds.hData = GlobalAlloc(GMEM_MOVEABLE, 0);
384 gds.nLength = 0;
385 es.dwCookie = (DWORD_PTR)&gds;
386 es.pfnCallback = ME_AppendToHGLOBAL;
387 ME_StreamOutRange(editor, SF_RTF, start, nChars, &es);
388 GlobalReAlloc(gds.hData, gds.nLength+1, 0);
389 return gds.hData;
390 }
391
392 HRESULT ME_GetDataObject(ME_TextEditor *editor, const ME_Cursor *start,
393 int nChars, LPDATAOBJECT *lplpdataobj)
394 {
395 DataObjectImpl *obj;
396 TRACE("(%p,%d,%d)\n", editor, ME_GetCursorOfs(start), nChars);
397
398 obj = heap_alloc(sizeof(DataObjectImpl));
399 if(cfRTF == 0)
400 cfRTF = RegisterClipboardFormatA("Rich Text Format");
401
402 obj->lpVtbl = &VT_DataObjectImpl;
403 obj->ref = 1;
404 obj->unicode = get_unicode_text(editor, start, nChars);
405 obj->rtf = NULL;
406
407 obj->fmtetc_cnt = 1;
408 if(editor->mode & TM_RICHTEXT)
409 obj->fmtetc_cnt++;
410 obj->fmtetc = GlobalAlloc(GMEM_ZEROINIT, obj->fmtetc_cnt*sizeof(FORMATETC));
411 InitFormatEtc(obj->fmtetc[0], CF_UNICODETEXT, TYMED_HGLOBAL);
412 if(editor->mode & TM_RICHTEXT) {
413 obj->rtf = get_rtf_text(editor, start, nChars);
414 InitFormatEtc(obj->fmtetc[1], cfRTF, TYMED_HGLOBAL);
415 }
416
417 *lplpdataobj = (LPDATAOBJECT)obj;
418 return S_OK;
419 }