msi/tests: Skip the tests if we are unable to identify the user SID. This fixes crash...
[wine] / 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 COBJMACROS
22 #define NONAMELESSSTRUCT
23 #define NONAMELESSUNION
24
25 #include <stdarg.h>
26
27 #include "windef.h"
28 #include "winbase.h"
29 #include "winerror.h"
30 #include "wingdi.h"
31 #include "winuser.h"
32 #include "winnls.h"
33 #include "oledlg.h"
34
35 #include "oledlg_private.h"
36 #include "resource.h"
37
38 #include "wine/debug.h"
39 #include "wine/unicode.h"
40
41 WINE_DEFAULT_DEBUG_CHANNEL(ole);
42
43 typedef struct
44 {
45     OLEUIPASTESPECIALW *ps;
46     DWORD flags;
47     WCHAR *source_name;
48     WCHAR *link_source_name;
49     WCHAR *type_name;
50     WCHAR *link_type_name;
51     LPOLESTR app_name;
52 } ps_struct_t;
53
54 static const struct ps_flag
55 {
56     DWORD flag;
57     const char *name;
58 } ps_flags[] = {
59 #define PS_FLAG_ENTRY(p) {p, #p}
60     PS_FLAG_ENTRY(PSF_SHOWHELP),
61     PS_FLAG_ENTRY(PSF_SELECTPASTE),
62     PS_FLAG_ENTRY(PSF_SELECTPASTELINK),
63     PS_FLAG_ENTRY(PSF_CHECKDISPLAYASICON),
64     PS_FLAG_ENTRY(PSF_DISABLEDISPLAYASICON),
65     PS_FLAG_ENTRY(PSF_HIDECHANGEICON),
66     PS_FLAG_ENTRY(PSF_STAYONCLIPBOARDCHANGE),
67     PS_FLAG_ENTRY(PSF_NOREFRESHDATAOBJECT),
68     {-1, NULL}
69 #undef PS_FLAG_ENTRY
70 };
71
72 static void dump_ps_flags(DWORD flags)
73 {
74     char flagstr[1000] = "";
75
76     const struct ps_flag *flag = ps_flags;
77     for( ; flag->name; flag++) {
78         if(flags & flag->flag) {
79             strcat(flagstr, flag->name);
80             strcat(flagstr, "|");
81         }
82     }
83     TRACE("flags %08x %s\n", flags, flagstr);
84 }
85
86 static void dump_pastespecial(const OLEUIPASTESPECIALW *ps)
87 {
88     UINT i;
89     dump_ps_flags(ps->dwFlags);
90     TRACE("hwnd %p caption %s hook %p custdata %lx\n",
91           ps->hWndOwner, debugstr_w(ps->lpszCaption), ps->lpfnHook, ps->lCustData);
92     if(IS_INTRESOURCE(ps->lpszTemplate))
93         TRACE("hinst %p template %04x hresource %p\n", ps->hInstance, (WORD)(ULONG_PTR)ps->lpszTemplate, ps->hResource);
94     else
95         TRACE("hinst %p template %s hresource %p\n", ps->hInstance, debugstr_w(ps->lpszTemplate), ps->hResource);
96     TRACE("dataobj %p arrpasteent %p cpasteent %d arrlinktype %p clinktype %d\n",
97           ps->lpSrcDataObj, ps->arrPasteEntries, ps->cPasteEntries,
98           ps->arrLinkTypes, ps->cLinkTypes);
99     TRACE("cclsidex %d lpclsidex %p nselect %d flink %d hmetapict %p size(%d,%d)\n",
100           ps->cClsidExclude, ps->lpClsidExclude, ps->nSelectedIndex, ps->fLink,
101           ps->hMetaPict, ps->sizel.cx, ps->sizel.cy);
102     for(i = 0; i < ps->cPasteEntries; i++)
103     {
104         TRACE("arrPasteEntries[%d]: cFormat %08x pTargetDevice %p dwAspect %d lindex %d tymed %d\n",
105               i, ps->arrPasteEntries[i].fmtetc.cfFormat, ps->arrPasteEntries[i].fmtetc.ptd,
106               ps->arrPasteEntries[i].fmtetc.dwAspect, ps->arrPasteEntries[i].fmtetc.lindex,
107               ps->arrPasteEntries[i].fmtetc.tymed);
108         TRACE("\tformat name %s result text %s flags %04x\n", debugstr_w(ps->arrPasteEntries[i].lpstrFormatName),
109               debugstr_w(ps->arrPasteEntries[i].lpstrResultText), ps->arrPasteEntries[i].dwFlags);
110     }
111     for(i = 0; i < ps->cLinkTypes; i++)
112         TRACE("arrLinkTypes[%d] %08x\n", i, ps->arrLinkTypes[i]);
113     for(i = 0; i < ps->cClsidExclude; i++)
114         TRACE("lpClsidExclude[%d] %s\n", i, debugstr_guid(&ps->lpClsidExclude[i]));
115
116 }
117
118 static inline WCHAR *strdupAtoW(const char *str)
119 {
120     DWORD len;
121     WCHAR *ret;
122     if(!str) return NULL;
123     len = MultiByteToWideChar(CP_ACP, 0, str, -1, NULL, 0);
124     ret = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR));
125     MultiByteToWideChar(CP_ACP, 0, str, -1, ret, len);
126     return ret;
127 }
128
129 static inline WCHAR *strdupW(const WCHAR *str)
130 {
131     DWORD len;
132     WCHAR *ret;
133     if(!str) return NULL;
134     len = lstrlenW(str) + 1;
135     ret = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR));
136     memcpy(ret, str, len * sizeof(WCHAR));
137     return ret;
138 }
139
140 static void get_descriptors(HWND hdlg, ps_struct_t *ps_struct)
141 {
142     FORMATETC fmtetc;
143     STGMEDIUM stg;
144
145     fmtetc.tymed = TYMED_HGLOBAL;
146     fmtetc.dwAspect = DVASPECT_CONTENT;
147     fmtetc.ptd = NULL;
148     fmtetc.lindex = -1;
149
150     fmtetc.cfFormat = cf_object_descriptor;
151     if(IDataObject_GetData(ps_struct->ps->lpSrcDataObj, &fmtetc, &stg) == S_OK)
152     {
153         OBJECTDESCRIPTOR *obj_desc = GlobalLock(stg.u.hGlobal);
154         if(obj_desc->dwSrcOfCopy)
155             ps_struct->source_name = strdupW((WCHAR*)((char*)obj_desc + obj_desc->dwSrcOfCopy));
156         if(obj_desc->dwFullUserTypeName)
157             ps_struct->type_name = strdupW((WCHAR*)((char*)obj_desc + obj_desc->dwFullUserTypeName));
158         OleRegGetUserType(&obj_desc->clsid, USERCLASSTYPE_APPNAME, &ps_struct->app_name);
159         /* Get the icon here.  If dwDrawAspect & DVASCPECT_ICON call GetData(CF_METAFILEPICT), otherwise
160            native calls OleGetIconFromClass(obj_desc->clsid) */
161         GlobalUnlock(stg.u.hGlobal);
162         GlobalFree(stg.u.hGlobal);
163     }
164     else
165     {
166         /* Try to get some data using some of the other clipboard formats */
167     }
168
169     fmtetc.cfFormat = cf_link_src_descriptor;
170     if(IDataObject_GetData(ps_struct->ps->lpSrcDataObj, &fmtetc, &stg) == S_OK)
171     {
172         OBJECTDESCRIPTOR *obj_desc = GlobalLock(stg.u.hGlobal);
173         if(obj_desc->dwSrcOfCopy)
174             ps_struct->link_source_name = strdupW((WCHAR*)((char*)obj_desc + obj_desc->dwSrcOfCopy));
175         if(obj_desc->dwFullUserTypeName)
176             ps_struct->link_type_name = strdupW((WCHAR*)((char*)obj_desc + obj_desc->dwFullUserTypeName));
177         GlobalUnlock(stg.u.hGlobal);
178         GlobalFree(stg.u.hGlobal);
179     }
180
181     if(ps_struct->source_name == NULL && ps_struct->link_source_name == NULL)
182     {
183         WCHAR buf[200];
184         LoadStringW(OLEDLG_hInstance, IDS_PS_UNKNOWN_SRC, buf, sizeof(buf)/sizeof(WCHAR));
185         ps_struct->source_name = strdupW(buf);
186     }
187
188     if(ps_struct->type_name == NULL && ps_struct->link_type_name == NULL)
189     {
190         WCHAR buf[200];
191         LoadStringW(OLEDLG_hInstance, IDS_PS_UNKNOWN_TYPE, buf, sizeof(buf)/sizeof(WCHAR));
192         ps_struct->type_name = strdupW(buf);
193     }
194 }
195
196 static BOOL add_entry_to_lb(HWND hdlg, UINT id, OLEUIPASTEENTRYW *pe)
197 {
198     HWND hwnd = GetDlgItem(hdlg, id);
199     BOOL ret = FALSE;
200
201     /* FIXME %s handling */
202
203     /* Note that this suffers from the same bug as native, in that if a new string
204        is a substring of an already added string, then the FINDSTRING will succeed
205        this is probably not what we want */
206     if(SendMessageW(hwnd, LB_FINDSTRING, 0, (LPARAM)pe->lpstrFormatName) == -1)
207     {
208         LRESULT pos = SendMessageW(hwnd, LB_ADDSTRING, 0, (LPARAM)pe->lpstrFormatName);
209         SendMessageW(hwnd, LB_SETITEMDATA, pos, (LPARAM)pe);
210         ret = TRUE;
211     }
212     return ret;
213 }
214
215 static DWORD init_pastelist(HWND hdlg, OLEUIPASTESPECIALW *ps)
216 {
217     IEnumFORMATETC *penum;
218     HRESULT hr;
219     FORMATETC fmts[20];
220     DWORD fetched, items_added = 0;
221
222     hr = IDataObject_EnumFormatEtc(ps->lpSrcDataObj, DATADIR_GET, &penum);
223     if(FAILED(hr))
224     {
225         WARN("Unable to create IEnumFORMATETC\n");
226         return 0;
227     }
228
229     /* The native version grabs only the first 20 fmts and we do the same */
230     hr = IEnumFORMATETC_Next(penum, sizeof(fmts)/sizeof(fmts[0]), fmts, &fetched);
231     TRACE("got %d formats hr %08x\n", fetched, hr);
232
233     if(SUCCEEDED(hr))
234     {
235         DWORD src_fmt, req_fmt;
236         for(req_fmt = 0; req_fmt < ps->cPasteEntries; req_fmt++)
237         {
238             /* This is used by update_structure() to set nSelectedIndex on exit */
239             ps->arrPasteEntries[req_fmt].dwScratchSpace = req_fmt;
240             TRACE("req_fmt %x\n", ps->arrPasteEntries[req_fmt].fmtetc.cfFormat);
241             for(src_fmt = 0; src_fmt < fetched; src_fmt++)
242             {
243                 TRACE("\tenum'ed fmt %x\n", fmts[src_fmt].cfFormat);
244                 if(ps->arrPasteEntries[req_fmt].fmtetc.cfFormat == fmts[src_fmt].cfFormat)
245                 {
246                     add_entry_to_lb(hdlg, IDC_PS_PASTELIST, ps->arrPasteEntries + req_fmt);
247                     items_added++;
248                     break;
249                 }
250             }
251         }
252     }
253
254     IEnumFORMATETC_Release(penum);
255     EnableWindow(GetDlgItem(hdlg, IDC_PS_PASTE), items_added ? TRUE : FALSE);
256     return items_added;
257 }
258
259 static DWORD init_linklist(HWND hdlg, OLEUIPASTESPECIALW *ps)
260 {
261     HRESULT hr;
262     DWORD supported_mask = 0;
263     DWORD items_added = 0;
264     int link, req_fmt;
265     FORMATETC fmt = {0, NULL, DVASPECT_CONTENT, -1, -1};
266
267     for(link = 0; link < ps->cLinkTypes && link < PS_MAXLINKTYPES; link++)
268     {
269         fmt.cfFormat = ps->arrLinkTypes[link];
270         hr = IDataObject_QueryGetData(ps->lpSrcDataObj, &fmt);
271         if(hr == S_OK)
272             supported_mask |= 1 << link;
273     }
274     TRACE("supported_mask %02x\n", supported_mask);
275     for(req_fmt = 0; req_fmt < ps->cPasteEntries; req_fmt++)
276     {
277         DWORD linktypes;
278         if(ps->arrPasteEntries[req_fmt].dwFlags & OLEUIPASTE_LINKANYTYPE)
279             linktypes = 0xff;
280         else
281             linktypes = ps->arrPasteEntries[req_fmt].dwFlags & 0xff;
282
283         if(linktypes & supported_mask)
284         {
285             add_entry_to_lb(hdlg, IDC_PS_PASTELINKLIST, ps->arrPasteEntries + req_fmt);
286             items_added++;
287         }
288     }
289
290     EnableWindow(GetDlgItem(hdlg, IDC_PS_PASTELINK), items_added ? TRUE : FALSE);
291     return items_added;
292 }
293
294 /* copies src_list_id into the display list */
295 static void update_display_list(HWND hdlg, UINT src_list_id)
296 {
297     LONG count, i, old_pos;
298     WCHAR txt[256];
299     LONG item_data;
300     HWND display_list = GetDlgItem(hdlg, IDC_PS_DISPLAYLIST);
301     HWND list = GetDlgItem(hdlg, src_list_id);
302
303     old_pos = SendMessageW(display_list, LB_GETCURSEL, 0, 0);
304     if(old_pos == -1) old_pos = 0;
305
306     SendMessageW(display_list, WM_SETREDRAW, 0, 0);
307     SendMessageW(display_list, LB_RESETCONTENT, 0, 0);
308     count = SendMessageW(list, LB_GETCOUNT, 0, 0);
309     for(i = 0; i < count; i++)
310     {
311         SendMessageW(list, LB_GETTEXT, i, (LPARAM)txt);
312         item_data = SendMessageW(list, LB_GETITEMDATA, i, 0);
313         SendMessageW(display_list, LB_INSERTSTRING, i, (LPARAM)txt);
314         SendMessageW(display_list, LB_SETITEMDATA, i, item_data);
315     }
316     old_pos = max(old_pos, count);
317     SendMessageW(display_list, LB_SETCURSEL, 0, 0);
318     SendMessageW(display_list, WM_SETREDRAW, 1, 0);
319     if(GetForegroundWindow() == hdlg)
320         SetFocus(display_list);
321 }
322
323 static void init_lists(HWND hdlg, ps_struct_t *ps_struct)
324 {
325     DWORD pastes_added = init_pastelist(hdlg, ps_struct->ps);
326     DWORD links_added = init_linklist(hdlg, ps_struct->ps);
327     UINT check_id, list_id;
328
329     if((ps_struct->flags & (PSF_SELECTPASTE | PSF_SELECTPASTELINK)) == 0)
330         ps_struct->flags |= PSF_SELECTPASTE;
331
332     if(!pastes_added && !links_added)
333         ps_struct->flags &= ~(PSF_SELECTPASTE | PSF_SELECTPASTELINK);
334     else if(!pastes_added && (ps_struct->flags & PSF_SELECTPASTE))
335     {
336         ps_struct->flags &= ~PSF_SELECTPASTE;
337         ps_struct->flags |= PSF_SELECTPASTELINK;
338     }
339     else if(!links_added && (ps_struct->flags & PSF_SELECTPASTELINK))
340     {
341         ps_struct->flags &= ~PSF_SELECTPASTELINK;
342         ps_struct->flags |= PSF_SELECTPASTE;
343     }
344
345     check_id = 0;
346     list_id = 0;
347     if(ps_struct->flags & PSF_SELECTPASTE)
348     {
349         check_id = IDC_PS_PASTE;
350         list_id = IDC_PS_PASTELIST;
351     }
352     else if(ps_struct->flags & PSF_SELECTPASTELINK)
353     {
354         check_id = IDC_PS_PASTELINK;
355         list_id = IDC_PS_PASTELINKLIST;
356     }
357
358     CheckRadioButton(hdlg, IDC_PS_PASTE, IDC_PS_PASTELINK, check_id);
359
360     if(list_id)
361         update_display_list(hdlg, list_id);
362     else
363         EnableWindow(GetDlgItem(hdlg, IDOK), 0);
364 }
365
366 static void update_src_text(HWND hdlg, const ps_struct_t *ps_struct)
367 {
368     WCHAR *str;
369
370     if(ps_struct->flags & PSF_SELECTPASTE)
371     {
372         if(ps_struct->source_name)
373             str = ps_struct->source_name;
374         else
375             str = ps_struct->link_source_name;
376
377     }
378     else
379     {
380         if(ps_struct->link_source_name)
381             str = ps_struct->link_source_name;
382         else
383             str = ps_struct->source_name;
384
385     }
386     SetDlgItemTextW(hdlg, IDC_PS_SOURCETEXT, str);
387 }
388
389 static void update_as_icon(HWND hdlg, ps_struct_t *ps_struct)
390 {
391     HWND icon_display = GetDlgItem(hdlg, IDC_PS_ICONDISPLAY);
392     HWND display_as_icon = GetDlgItem(hdlg, IDC_PS_DISPLAYASICON);
393     HWND change_icon = GetDlgItem(hdlg, IDC_PS_CHANGEICON);
394
395     /* FIXME. No as icon handling */
396     ps_struct->flags &= ~PSF_CHECKDISPLAYASICON;
397
398     CheckDlgButton(hdlg, IDC_PS_DISPLAYASICON, ps_struct->flags & PSF_CHECKDISPLAYASICON);
399     EnableWindow(display_as_icon, 0);
400     ShowWindow(icon_display, SW_HIDE);
401     EnableWindow(icon_display, 0);
402     ShowWindow(change_icon, SW_HIDE);
403     EnableWindow(change_icon, 0);
404 }
405
406 static void update_result_text(HWND hdlg, const ps_struct_t *ps_struct)
407 {
408     WCHAR resource_txt[200];
409     UINT res_id;
410     OLEUIPASTEENTRYW *pent;
411     LONG cur_sel;
412     static const WCHAR percent_s[] = {'%','s',0};
413     WCHAR *result_txt, *ptr;
414
415     cur_sel = SendMessageW(GetDlgItem(hdlg, IDC_PS_DISPLAYLIST), LB_GETCURSEL, 0, 0);
416     if(cur_sel == -1) return;
417     pent = (OLEUIPASTEENTRYW*)SendMessageW(GetDlgItem(hdlg, IDC_PS_DISPLAYLIST), LB_GETITEMDATA, cur_sel, 0);
418
419     if(ps_struct->flags & PSF_SELECTPASTE)
420     {
421         if(ps_struct->flags & PSF_CHECKDISPLAYASICON)
422             res_id = IDS_PS_PASTE_OBJECT_AS_ICON;
423         else
424             res_id = IDS_PS_PASTE_DATA;
425     }
426     else
427     {
428         if(ps_struct->flags & PSF_CHECKDISPLAYASICON)
429             res_id = IDS_PS_PASTE_LINK_OBJECT_AS_ICON;
430         else
431             res_id = IDS_PS_PASTE_LINK_DATA;
432     }
433
434     LoadStringW(OLEDLG_hInstance, res_id, resource_txt, sizeof(resource_txt)/sizeof(WCHAR));
435     if((ptr = strstrW(resource_txt, percent_s)))
436     {
437         /* FIXME handle %s in ResultText. Sub appname if IDS_PS_PASTE_OBJECT{_AS_ICON}.  Else sub appropriate type name */
438         size_t result_txt_len = strlenW(pent->lpstrResultText);
439         ptrdiff_t offs = (char*)ptr - (char*)resource_txt;
440         result_txt = HeapAlloc(GetProcessHeap(), 0, (strlenW(resource_txt) + result_txt_len - 1) * sizeof(WCHAR));
441         memcpy(result_txt, resource_txt, offs);
442         memcpy((char*)result_txt + offs, pent->lpstrResultText, result_txt_len * sizeof(WCHAR));
443         memcpy((char*)result_txt + offs + result_txt_len * sizeof(WCHAR), ptr + 2, (strlenW(ptr + 2) + 1) * sizeof(WCHAR));
444     }
445     else
446         result_txt = resource_txt;
447
448     SetDlgItemTextW(hdlg, IDC_PS_RESULTTEXT, result_txt);
449
450     if(result_txt != resource_txt)
451         HeapFree(GetProcessHeap(), 0, result_txt);
452
453 }
454
455 static void selection_change(HWND hdlg, ps_struct_t *ps_struct)
456 {
457     update_as_icon(hdlg, ps_struct);
458     update_result_text(hdlg, ps_struct);
459 }
460
461 static void mode_change(HWND hdlg, ps_struct_t *ps_struct, UINT id)
462 {
463     if(id == IDC_PS_PASTE)
464     {
465         ps_struct->flags &= ~PSF_SELECTPASTELINK;
466         ps_struct->flags |= PSF_SELECTPASTE;
467     }
468     else
469     {
470         ps_struct->flags &= ~PSF_SELECTPASTE;
471         ps_struct->flags |= PSF_SELECTPASTELINK;
472     }
473
474     update_src_text(hdlg, ps_struct);
475     update_display_list(hdlg, id == IDC_PS_PASTE ? IDC_PS_PASTELIST : IDC_PS_PASTELINKLIST);
476     selection_change(hdlg, ps_struct);
477 }
478
479 static void post_help_msg(HWND hdlg, ps_struct_t *ps_struct)
480 {
481     PostMessageW(ps_struct->ps->hWndOwner, oleui_msg_help, (WPARAM)hdlg, IDD_PASTESPECIAL);
482 }
483
484 static void send_end_dialog_msg(HWND hdlg, ps_struct_t *ps_struct, UINT id)
485 {
486     SendMessageW(hdlg, oleui_msg_enddialog, id, 0);
487 }
488
489 static void update_structure(HWND hdlg, ps_struct_t *ps_struct)
490 {
491     LONG cur_sel = SendMessageW(GetDlgItem(hdlg, IDC_PS_DISPLAYLIST), LB_GETCURSEL, 0, 0);
492     if(cur_sel != -1)
493     {
494         OLEUIPASTEENTRYW *pent;
495         pent = (OLEUIPASTEENTRYW *)SendMessageW(GetDlgItem(hdlg, IDC_PS_DISPLAYLIST), LB_GETITEMDATA, cur_sel, 0);
496         ps_struct->ps->nSelectedIndex = pent->dwScratchSpace;
497     }
498     ps_struct->ps->dwFlags = ps_struct->flags;
499     ps_struct->ps->fLink = (ps_struct->flags & PSF_SELECTPASTELINK) ? TRUE : FALSE;
500 }
501
502 static void free_structure(ps_struct_t *ps_struct)
503 {
504     HeapFree(GetProcessHeap(), 0, ps_struct->type_name);
505     HeapFree(GetProcessHeap(), 0, ps_struct->source_name);
506     HeapFree(GetProcessHeap(), 0, ps_struct->link_type_name);
507     HeapFree(GetProcessHeap(), 0, ps_struct->link_source_name);
508     CoTaskMemFree(ps_struct->app_name);
509     HeapFree(GetProcessHeap(), 0, ps_struct);
510 }
511
512 static INT_PTR CALLBACK ps_dlg_proc(HWND hdlg, UINT msg, WPARAM wp, LPARAM lp)
513 {
514     /* native uses prop name "Structure", but we're not compatible
515        with that so we'll prepend "Wine_". */
516     static const WCHAR prop_name[] = {'W','i','n','e','_','S','t','r','u','c','t','u','r','e',0};
517     ps_struct_t *ps_struct;
518
519     TRACE("(%p, %04x, %08lx, %08lx)\n", hdlg, msg, wp, lp);
520
521     ps_struct = GetPropW(hdlg, prop_name);
522
523     if(msg != WM_INITDIALOG)
524     {
525         if(!ps_struct)
526             return 0;
527
528         if(ps_struct->ps->lpfnHook)
529         {
530             INT_PTR ret = ps_struct->ps->lpfnHook(hdlg, msg, wp, lp);
531             if(ret) return ret;
532         }
533     }
534
535     switch(msg)
536     {
537     case WM_INITDIALOG:
538     {
539         ps_struct = HeapAlloc(GetProcessHeap(), 0, sizeof(*ps_struct));
540         ps_struct->ps = (OLEUIPASTESPECIALW*)lp;
541         ps_struct->type_name = NULL;
542         ps_struct->source_name = NULL;
543         ps_struct->link_type_name = NULL;
544         ps_struct->link_source_name = NULL;
545         ps_struct->app_name = NULL;
546         ps_struct->flags = ps_struct->ps->dwFlags;
547
548         SetPropW(hdlg, prop_name, ps_struct);
549
550         if(!(ps_struct->ps->dwFlags & PSF_SHOWHELP))
551         {
552             ShowWindow(GetDlgItem(hdlg, IDC_OLEUIHELP), SW_HIDE);
553             EnableWindow(GetDlgItem(hdlg, IDC_OLEUIHELP), 0);
554         }
555
556         if(ps_struct->ps->lpszCaption)
557             SetWindowTextW(hdlg, ps_struct->ps->lpszCaption);
558
559         get_descriptors(hdlg, ps_struct);
560
561         init_lists(hdlg, ps_struct);
562
563         update_src_text(hdlg, ps_struct);
564
565         selection_change(hdlg, ps_struct);
566
567         SetFocus(GetDlgItem(hdlg, IDC_PS_DISPLAYLIST));
568
569         if(ps_struct->ps->lpfnHook)
570             ps_struct->ps->lpfnHook(hdlg, msg, 0, 0);
571         return FALSE; /* use new focus */
572     }
573     case WM_COMMAND:
574         switch(LOWORD(wp))
575         {
576         case IDC_PS_DISPLAYLIST:
577             switch(HIWORD(wp))
578             {
579             case LBN_SELCHANGE:
580                 selection_change(hdlg, ps_struct);
581                 return FALSE;
582             default:
583                 return FALSE;
584             }
585         case IDC_PS_PASTE:
586         case IDC_PS_PASTELINK:
587             switch(HIWORD(wp))
588             {
589             case BN_CLICKED:
590                 mode_change(hdlg, ps_struct, LOWORD(wp));
591                 return FALSE;
592
593             default:
594                 return FALSE;
595             }
596         case IDC_OLEUIHELP:
597             switch(HIWORD(wp))
598             {
599             case BN_CLICKED:
600                 post_help_msg(hdlg, ps_struct);
601                 return FALSE;
602             default:
603                 return FALSE;
604             }
605         case IDOK:
606         case IDCANCEL:
607             switch(HIWORD(wp))
608             {
609             case BN_CLICKED:
610                 send_end_dialog_msg(hdlg, ps_struct, LOWORD(wp));
611                 return FALSE;
612             default:
613                 return FALSE;
614             }
615         }
616         return FALSE;
617     default:
618         if(msg == oleui_msg_enddialog)
619         {
620             if(wp == IDOK)
621                 update_structure(hdlg, ps_struct);
622             EndDialog(hdlg, wp);
623             /* native does its cleanup in WM_DESTROY */
624             RemovePropW(hdlg, prop_name);
625             free_structure(ps_struct);
626             return TRUE;
627         }
628         return FALSE;
629     }
630
631 }
632
633 /***********************************************************************
634  *           OleUIPasteSpecialA (OLEDLG.4)
635  */
636 UINT WINAPI OleUIPasteSpecialA(LPOLEUIPASTESPECIALA psA)
637 {
638     OLEUIPASTESPECIALW ps;
639     UINT ret;
640     TRACE("(%p)\n", psA);
641
642     memcpy(&ps, psA, psA->cbStruct);
643
644     ps.lpszCaption = strdupAtoW(psA->lpszCaption);
645     if(!IS_INTRESOURCE(ps.lpszTemplate))
646         ps.lpszTemplate = strdupAtoW(psA->lpszTemplate);
647
648     if(psA->cPasteEntries > 0)
649     {
650         DWORD size = psA->cPasteEntries * sizeof(ps.arrPasteEntries[0]);
651         UINT i;
652
653         ps.arrPasteEntries = HeapAlloc(GetProcessHeap(), 0, size);
654         memcpy(ps.arrPasteEntries, psA->arrPasteEntries, size);
655         for(i = 0; i < psA->cPasteEntries; i++)
656         {
657             ps.arrPasteEntries[i].lpstrFormatName =
658                 strdupAtoW(psA->arrPasteEntries[i].lpstrFormatName);
659             ps.arrPasteEntries[i].lpstrResultText =
660                 strdupAtoW(psA->arrPasteEntries[i].lpstrResultText);
661         }
662     }
663
664     ret = OleUIPasteSpecialW(&ps);
665
666     if(psA->cPasteEntries > 0)
667     {
668         UINT i;
669         for(i = 0; i < psA->cPasteEntries; i++)
670         {
671             HeapFree(GetProcessHeap(), 0, (WCHAR*)ps.arrPasteEntries[i].lpstrFormatName);
672             HeapFree(GetProcessHeap(), 0, (WCHAR*)ps.arrPasteEntries[i].lpstrResultText);
673         }
674         HeapFree(GetProcessHeap(), 0, ps.arrPasteEntries);
675     }
676     if(!IS_INTRESOURCE(ps.lpszTemplate))
677         HeapFree(GetProcessHeap(), 0, (WCHAR*)ps.lpszTemplate);
678     HeapFree(GetProcessHeap(), 0, (WCHAR*)ps.lpszCaption);
679
680     /* Copy back the output fields */
681     psA->dwFlags = ps.dwFlags;
682     psA->lpSrcDataObj = ps.lpSrcDataObj;
683     psA->nSelectedIndex = ps.nSelectedIndex;
684     psA->fLink = ps.fLink;
685     psA->hMetaPict = ps.hMetaPict;
686     psA->sizel = ps.sizel;
687
688     return ret;
689 }
690
691 /***********************************************************************
692  *           OleUIPasteSpecialW (OLEDLG.22)
693  */
694 UINT WINAPI OleUIPasteSpecialW(LPOLEUIPASTESPECIALW ps)
695 {
696     LPCDLGTEMPLATEW dlg_templ = (LPCDLGTEMPLATEW)ps->hResource;
697     UINT ret;
698
699     TRACE("(%p)\n", ps);
700
701     if(TRACE_ON(ole)) dump_pastespecial(ps);
702
703     if(!ps->lpSrcDataObj)
704         OleGetClipboard(&ps->lpSrcDataObj);
705
706     if(ps->hInstance || !ps->hResource)
707     {
708         HINSTANCE hInst = ps->hInstance ? ps->hInstance : OLEDLG_hInstance;
709         const WCHAR *name = ps->hInstance ? ps->lpszTemplate : MAKEINTRESOURCEW(IDD_PASTESPECIAL4);
710         HRSRC hrsrc;
711
712         if(name == NULL) return OLEUI_ERR_LPSZTEMPLATEINVALID;
713         hrsrc = FindResourceW(hInst, name, MAKEINTRESOURCEW(RT_DIALOG));
714         if(!hrsrc) return OLEUI_ERR_FINDTEMPLATEFAILURE;
715         dlg_templ = LoadResource(hInst, hrsrc);
716         if(!dlg_templ) return OLEUI_ERR_LOADTEMPLATEFAILURE;
717     }
718
719     ret = DialogBoxIndirectParamW(OLEDLG_hInstance, dlg_templ, ps->hWndOwner, ps_dlg_proc, (LPARAM)ps);
720
721     return ret;
722 }