git.oblomov.eu Git - wine/blob - dlls/oledlg/pastespl.c

   1 /*
   2  * OleUIPasteSpecial implementation
   3  *
   4  * Copyright 2006 Huw Davies
   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 #define COM_NO_WINDOWS_H
  22 #define COBJMACROS
  23
  24 #include <stdarg.h>
  25
  26 #include "windef.h"
  27 #include "winbase.h"
  28 #include "winerror.h"
  29 #include "wingdi.h"
  30 #include "winuser.h"
  31 #include "winnls.h"
  32 #include "oledlg.h"
  33
  34 #include "oledlg_private.h"
  35
  36 #include "wine/debug.h"
  37
  38 WINE_DEFAULT_DEBUG_CHANNEL(ole);
  39
  40 typedef struct
  41 {
  42     OLEUIPASTESPECIALW *ps;
  43     DWORD flags;
  44 } ps_struct_t;
  45
  46 static const struct ps_flag
  47 {
  48     DWORD flag;
  49     const char *name;
  50 } ps_flags[] = {
  51 #define PS_FLAG_ENTRY(p) {p, #p}
  52     PS_FLAG_ENTRY(PSF_SHOWHELP),
  53     PS_FLAG_ENTRY(PSF_SELECTPASTE),
  54     PS_FLAG_ENTRY(PSF_SELECTPASTELINK),
  55     PS_FLAG_ENTRY(PSF_CHECKDISPLAYASICON),
  56     PS_FLAG_ENTRY(PSF_DISABLEDISPLAYASICON),
  57     PS_FLAG_ENTRY(PSF_HIDECHANGEICON),
  58     PS_FLAG_ENTRY(PSF_STAYONCLIPBOARDCHANGE),
  59     PS_FLAG_ENTRY(PSF_NOREFRESHDATAOBJECT),
  60     {-1, NULL}
  61 #undef PS_FLAG_ENTRY
  62 };
  63
  64 static void dump_ps_flags(DWORD flags)
  65 {
  66     char flagstr[1000] = "";
  67
  68     const struct ps_flag *flag = ps_flags;
  69     for( ; flag->name; flag++) {
  70         if(flags & flag->flag) {
  71             strcat(flagstr, flag->name);
  72             strcat(flagstr, "|");
  73         }
  74     }
  75     TRACE("flags %08x %s\n", flags, flagstr);
  76 }
  77
  78 static void dump_pastespecial(LPOLEUIPASTESPECIALW ps)
  79 {
  80     UINT i;
  81     dump_ps_flags(ps->dwFlags);
  82     TRACE("hwnd %p caption %s hook %p custdata %lx\n",
  83           ps->hWndOwner, debugstr_w(ps->lpszCaption), ps->lpfnHook, ps->lCustData);
  84     if(IS_INTRESOURCE(ps->lpszTemplate))
  85         TRACE("hinst %p template %04x hresource %p\n", ps->hInstance, (WORD)(ULONG_PTR)ps->lpszTemplate, ps->hResource);
  86     else
  87         TRACE("hinst %p template %s hresource %p\n", ps->hInstance, debugstr_w(ps->lpszTemplate), ps->hResource);
  88     TRACE("dataobj %p arrpasteent %p cpasteent %d arrlinktype %p clinktype %d\n",
  89           ps->lpSrcDataObj, ps->arrPasteEntries, ps->cPasteEntries,
  90           ps->arrLinkTypes, ps->cLinkTypes);
  91     TRACE("cclsidex %d lpclsidex %p nselect %d flink %d hmetapict %p size(%d,%d)\n",
  92           ps->cClsidExclude, ps->lpClsidExclude, ps->nSelectedIndex, ps->fLink,
  93           ps->hMetaPict, ps->sizel.cx, ps->sizel.cy);
  94     for(i = 0; i < ps->cPasteEntries; i++)
  95     {
  96         TRACE("arrPasteEntries[%d]: cFormat %08x pTargetDevice %p dwAspect %d lindex %d tymed %d\n",
  97               i, ps->arrPasteEntries[i].fmtetc.cfFormat, ps->arrPasteEntries[i].fmtetc.ptd,
  98               ps->arrPasteEntries[i].fmtetc.dwAspect, ps->arrPasteEntries[i].fmtetc.lindex,
  99               ps->arrPasteEntries[i].fmtetc.tymed);
 100         TRACE("\tformat name %s result text %s flags %04x\n", debugstr_w(ps->arrPasteEntries[i].lpstrFormatName),
 101               debugstr_w(ps->arrPasteEntries[i].lpstrResultText), ps->arrPasteEntries[i].dwFlags);
 102     }
 103     for(i = 0; i < ps->cLinkTypes; i++)
 104         TRACE("arrLinkTypes[%d] %08x\n", i, ps->arrLinkTypes[i]);
 105     for(i = 0; i < ps->cClsidExclude; i++)
 106         TRACE("lpClsidExclude[%d] %s\n", i, debugstr_guid(&ps->lpClsidExclude[i]));
 107
 108 }
 109
 110 static inline WCHAR *strdupAtoW(const char *str)
 111 {
 112     DWORD len;
 113     WCHAR *ret;
 114     if(!str) return NULL;
 115     len = MultiByteToWideChar(CP_ACP, 0, str, -1, NULL, 0);
 116     ret = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR));
 117     MultiByteToWideChar(CP_ACP, 0, str, -1, ret, len);
 118     return ret;
 119 }
 120
 121 static BOOL add_entry_to_lb(HWND hdlg, UINT id, OLEUIPASTEENTRYW *pe)
 122 {
 123     HWND hwnd = GetDlgItem(hdlg, id);
 124     BOOL ret = FALSE;
 125
 126     /* FIXME %s handling */
 127
 128     /* Note that this suffers from the same bug as native, in that if a new string
 129        is a substring of an already added string, then the FINDSTRING will succeed
 130        this is probably not what we want */
 131     if(SendMessageW(hwnd, LB_FINDSTRING, 0, (LPARAM)pe->lpstrFormatName) == -1)
 132     {
 133         LRESULT pos = SendMessageW(hwnd, LB_ADDSTRING, 0, (LPARAM)pe->lpstrFormatName);
 134         SendMessageW(hwnd, LB_SETITEMDATA, pos, (LPARAM)pe);
 135         ret = TRUE;
 136     }
 137     return ret;
 138 }
 139
 140 static DWORD init_pastelist(HWND hdlg, OLEUIPASTESPECIALW *ps)
 141 {
 142     IEnumFORMATETC *penum;
 143     HRESULT hr;
 144     FORMATETC fmts[20];
 145     DWORD fetched, items_added = 0;
 146
 147     hr = IDataObject_EnumFormatEtc(ps->lpSrcDataObj, DATADIR_GET, &penum);
 148     if(FAILED(hr))
 149     {
 150         WARN("Unable to create IEnumFORMATETC\n");
 151         return 0;
 152     }
 153
 154     /* The native version grabs only the first 20 fmts and we do the same */
 155     hr = IEnumFORMATETC_Next(penum, sizeof(fmts)/sizeof(fmts[0]), fmts, &fetched);
 156     TRACE("got %d formats hr %08x\n", fetched, hr);
 157
 158     if(SUCCEEDED(hr))
 159     {
 160         DWORD src_fmt, req_fmt;
 161         for(req_fmt = 0; req_fmt < ps->cPasteEntries; req_fmt++)
 162         {
 163             /* This is used by update_struct() to set nSelectedIndex on exit */
 164             ps->arrPasteEntries[req_fmt].dwScratchSpace = req_fmt;
 165             TRACE("req_fmt %x\n", ps->arrPasteEntries[req_fmt].fmtetc.cfFormat);
 166             for(src_fmt = 0; src_fmt < fetched; src_fmt++)
 167             {
 168                 TRACE("\tenum'ed fmt %x\n", fmts[src_fmt].cfFormat);
 169                 if(ps->arrPasteEntries[req_fmt].fmtetc.cfFormat == fmts[src_fmt].cfFormat)
 170                 {
 171                     add_entry_to_lb(hdlg, IDC_PS_PASTELIST, ps->arrPasteEntries + req_fmt);
 172                     items_added++;
 173                     break;
 174                 }
 175             }
 176         }
 177     }
 178
 179     IEnumFORMATETC_Release(penum);
 180     EnableWindow(GetDlgItem(hdlg, IDC_PS_PASTE), items_added ? TRUE : FALSE);
 181     return items_added;
 182 }
 183
 184 static DWORD init_linklist(HWND hdlg, OLEUIPASTESPECIALW *ps)
 185 {
 186     HRESULT hr;
 187     DWORD supported_mask = 0;
 188     DWORD items_added = 0;
 189     int link, req_fmt;
 190     FORMATETC fmt = {0, NULL, DVASPECT_CONTENT, -1, -1};
 191
 192     for(link = 0; link < ps->cLinkTypes && link < PS_MAXLINKTYPES; link++)
 193     {
 194         fmt.cfFormat = ps->arrLinkTypes[link];
 195         hr = IDataObject_QueryGetData(ps->lpSrcDataObj, &fmt);
 196         if(hr == S_OK)
 197             supported_mask |= 1 << link;
 198     }
 199     TRACE("supported_mask %02x\n", supported_mask);
 200     for(req_fmt = 0; req_fmt < ps->cPasteEntries; req_fmt++)
 201     {
 202         DWORD linktypes;
 203         if(ps->arrPasteEntries[req_fmt].dwFlags & OLEUIPASTE_LINKANYTYPE)
 204             linktypes = 0xff;
 205         else
 206             linktypes = ps->arrPasteEntries[req_fmt].dwFlags & 0xff;
 207
 208         if(linktypes & supported_mask)
 209         {
 210             add_entry_to_lb(hdlg, IDC_PS_PASTELINKLIST, ps->arrPasteEntries + req_fmt);
 211             items_added++;
 212         }
 213     }
 214
 215     EnableWindow(GetDlgItem(hdlg, IDC_PS_PASTELINK), items_added ? TRUE : FALSE);
 216     return items_added;
 217 }
 218
 219 static void init_lists(HWND hdlg, ps_struct_t *ps_struct)
 220 {
 221     DWORD pastes_added = init_pastelist(hdlg, ps_struct->ps);
 222     DWORD links_added = init_linklist(hdlg, ps_struct->ps);
 223     UINT check_id, list_id;
 224
 225     if((ps_struct->flags & (PSF_SELECTPASTE | PSF_SELECTPASTELINK)) == 0)
 226         ps_struct->flags |= PSF_SELECTPASTE;
 227
 228     if(!pastes_added && !links_added)
 229         ps_struct->flags &= ~(PSF_SELECTPASTE | PSF_SELECTPASTELINK);
 230     else if(!pastes_added && (ps_struct->flags & PSF_SELECTPASTE))
 231     {
 232         ps_struct->flags &= ~PSF_SELECTPASTE;
 233         ps_struct->flags |= PSF_SELECTPASTELINK;
 234     }
 235     else if(!links_added && (ps_struct->flags & PSF_SELECTPASTELINK))
 236     {
 237         ps_struct->flags &= ~PSF_SELECTPASTELINK;
 238         ps_struct->flags |= PSF_SELECTPASTE;
 239     }
 240
 241     check_id = 0;
 242     list_id = 0;
 243     if(ps_struct->flags & PSF_SELECTPASTE)
 244     {
 245         check_id = IDC_PS_PASTE;
 246         list_id = IDC_PS_PASTELIST;
 247     }
 248     else if(ps_struct->flags & PSF_SELECTPASTELINK)
 249     {
 250         check_id = IDC_PS_PASTELINK;
 251         list_id = IDC_PS_PASTELINKLIST;
 252     }
 253
 254     CheckRadioButton(hdlg, IDC_PS_PASTE, IDC_PS_PASTELINK, check_id);
 255 }
 256
 257 static void update_structure(HWND hdlg, ps_struct_t *ps_struct)
 258 {
 259     ps_struct->ps->dwFlags = ps_struct->flags;
 260     ps_struct->ps->fLink = (ps_struct->flags & PSF_SELECTPASTELINK) ? TRUE : FALSE;
 261 }
 262
 263 static void free_structure(ps_struct_t *ps_struct)
 264 {
 265     HeapFree(GetProcessHeap(), 0, ps_struct);
 266 }
 267
 268 static INT_PTR CALLBACK ps_dlg_proc(HWND hdlg, UINT msg, WPARAM wp, LPARAM lp)
 269 {
 270     /* native uses prop name "Structure", but we're not compatible
 271        with that so we'll prepend "Wine_". */
 272     static const WCHAR prop_name[] = {'W','i','n','e','_','S','t','r','u','c','t','u','r','e',0};
 273     ps_struct_t *ps_struct;
 274
 275     TRACE("(%p, %04x, %08x, %08lx)\n", hdlg, msg, wp, lp);
 276
 277     ps_struct = GetPropW(hdlg, prop_name);
 278
 279     if(msg != WM_INITDIALOG)
 280     {
 281         if(!ps_struct)
 282             return 0;
 283     }
 284
 285     switch(msg)
 286     {
 287     case WM_INITDIALOG:
 288     {
 289         ps_struct = HeapAlloc(GetProcessHeap(), 0, sizeof(*ps_struct));
 290         ps_struct->ps = (OLEUIPASTESPECIALW*)lp;
 291         ps_struct->flags = ps_struct->ps->dwFlags;
 292
 293         SetPropW(hdlg, prop_name, ps_struct);
 294
 295         if(ps_struct->ps->lpszCaption)
 296             SetWindowTextW(hdlg, ps_struct->ps->lpszCaption);
 297
 298         init_lists(hdlg, ps_struct);
 299
 300         return TRUE; /* use default focus */
 301     }
 302     case WM_COMMAND:
 303         switch(LOWORD(wp))
 304         {
 305         case IDOK:
 306         case IDCANCEL:
 307             if(wp == IDOK)
 308                 update_structure(hdlg, ps_struct);
 309             EndDialog(hdlg, wp);
 310             free_structure(ps_struct);
 311             return TRUE;
 312         }
 313         return FALSE;
 314     default:
 315         return FALSE;
 316     }
 317
 318 }
 319
 320 /***********************************************************************
 321  *           OleUIPasteSpecialA (OLEDLG.4)
 322  */
 323 UINT WINAPI OleUIPasteSpecialA(LPOLEUIPASTESPECIALA psA)
 324 {
 325     OLEUIPASTESPECIALW ps;
 326     UINT ret;
 327     TRACE("(%p)\n", psA);
 328
 329     memcpy(&ps, psA, psA->cbStruct);
 330
 331     ps.lpszCaption = strdupAtoW(psA->lpszCaption);
 332     if(!IS_INTRESOURCE(ps.lpszTemplate))
 333         ps.lpszTemplate = strdupAtoW(psA->lpszTemplate);
 334
 335     if(psA->cPasteEntries > 0)
 336     {
 337         DWORD size = psA->cPasteEntries * sizeof(ps.arrPasteEntries[0]);
 338         UINT i;
 339
 340         ps.arrPasteEntries = HeapAlloc(GetProcessHeap(), 0, size);
 341         memcpy(ps.arrPasteEntries, psA->arrPasteEntries, size);
 342         for(i = 0; i < psA->cPasteEntries; i++)
 343         {
 344             ps.arrPasteEntries[i].lpstrFormatName =
 345                 strdupAtoW(psA->arrPasteEntries[i].lpstrFormatName);
 346             ps.arrPasteEntries[i].lpstrResultText =
 347                 strdupAtoW(psA->arrPasteEntries[i].lpstrResultText);
 348         }
 349     }
 350
 351     ret = OleUIPasteSpecialW(&ps);
 352
 353     if(psA->cPasteEntries > 0)
 354     {
 355         UINT i;
 356         for(i = 0; i < psA->cPasteEntries; i++)
 357         {
 358             HeapFree(GetProcessHeap(), 0, (WCHAR*)ps.arrPasteEntries[i].lpstrFormatName);
 359             HeapFree(GetProcessHeap(), 0, (WCHAR*)ps.arrPasteEntries[i].lpstrResultText);
 360         }
 361         HeapFree(GetProcessHeap(), 0, ps.arrPasteEntries);
 362     }
 363     if(!IS_INTRESOURCE(ps.lpszTemplate))
 364         HeapFree(GetProcessHeap(), 0, (WCHAR*)ps.lpszTemplate);
 365     HeapFree(GetProcessHeap(), 0, (WCHAR*)ps.lpszCaption);
 366
 367     /* Copy back the output fields */
 368     psA->dwFlags = ps.dwFlags;
 369     psA->lpSrcDataObj = ps.lpSrcDataObj;
 370     psA->nSelectedIndex = ps.nSelectedIndex;
 371     psA->fLink = ps.fLink;
 372     psA->hMetaPict = ps.hMetaPict;
 373     psA->sizel = ps.sizel;
 374
 375     return ret;
 376 }
 377
 378 /***********************************************************************
 379  *           OleUIPasteSpecialW (OLEDLG.22)
 380  */
 381 UINT WINAPI OleUIPasteSpecialW(LPOLEUIPASTESPECIALW ps)
 382 {
 383     LPCDLGTEMPLATEW dlg_templ = (LPCDLGTEMPLATEW)ps->hResource;
 384
 385     TRACE("(%p)\n", ps);
 386
 387     if(TRACE_ON(ole)) dump_pastespecial(ps);
 388
 389     if(!ps->lpSrcDataObj)
 390         OleGetClipboard(&ps->lpSrcDataObj);
 391
 392     if(ps->hInstance || !ps->hResource)
 393     {
 394         HINSTANCE hInst = ps->hInstance ? ps->hInstance : OLEDLG_hInstance;
 395         const WCHAR *name = ps->hInstance ? ps->lpszTemplate : MAKEINTRESOURCEW(IDD_PASTESPECIAL4);
 396         HRSRC hrsrc;
 397
 398         if(name == NULL) return OLEUI_ERR_LPSZTEMPLATEINVALID;
 399         hrsrc = FindResourceW(hInst, name, MAKEINTRESOURCEW(RT_DIALOG));
 400         if(!hrsrc) return OLEUI_ERR_FINDTEMPLATEFAILURE;
 401         dlg_templ = LoadResource(hInst, hrsrc);
 402         if(!dlg_templ) return OLEUI_ERR_LOADTEMPLATEFAILURE;
 403     }
 404
 405     DialogBoxIndirectParamW(OLEDLG_hInstance, dlg_templ, ps->hWndOwner, ps_dlg_proc, (LPARAM)ps);
 406
 407     SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
 408     return OLEUI_FALSE;
 409 }