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