msvcp: Prevent overflows while operating on string sizes.
[wine] / dlls / ieframe / iexplore.c
1 /*
2  * ieframe - Internet Explorer main frame window
3  *
4  * Copyright 2006 Mike McCormack (for CodeWeavers)
5  * Copyright 2006 Jacek Caban (for CodeWeavers)
6  *
7  * This library is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Lesser General Public
9  * License as published by the Free Software Foundation; either
10  * version 2.1 of the License, or (at your option) any later version.
11  *
12  * This library is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15  * Lesser General Public License for more details.
16  *
17  * You should have received a copy of the GNU Lesser General Public
18  * License along with this library; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
20  */
21
22 #define COBJMACROS
23
24 #include <stdarg.h>
25
26 #include "ieframe.h"
27 #include "resource.h"
28
29 #include "winuser.h"
30 #include "wingdi.h"
31 #include "winnls.h"
32 #include "ole2.h"
33 #include "exdisp.h"
34 #include "oleidl.h"
35
36 #include "mshtmcid.h"
37 #include "shellapi.h"
38 #include "winreg.h"
39 #include "shlwapi.h"
40 #include "intshcut.h"
41 #include "ddeml.h"
42
43 #include "wine/debug.h"
44
45 WINE_DEFAULT_DEBUG_CHANNEL(ieframe);
46
47 #define IDI_APPICON 1
48
49 #define WM_UPDATEADDRBAR    (WM_APP+1)
50
51 static const WCHAR szIEWinFrame[] = { 'I','E','F','r','a','m','e',0 };
52
53 /* Windows uses "Microsoft Internet Explorer" */
54 static const WCHAR wszWineInternetExplorer[] =
55         {'W','i','n','e',' ','I','n','t','e','r','n','e','t',' ','E','x','p','l','o','r','e','r',0};
56
57 static LONG obj_cnt;
58 static DWORD dde_inst;
59 static HSZ ddestr_iexplore, ddestr_openurl;
60 static struct list ie_list = LIST_INIT(ie_list);
61
62 HRESULT update_ie_statustext(InternetExplorer* This, LPCWSTR text)
63 {
64     if(!SendMessageW(This->status_hwnd, SB_SETTEXTW, MAKEWORD(SB_SIMPLEID, 0), (LPARAM)text))
65         return E_FAIL;
66
67     return S_OK;
68 }
69
70 static void adjust_ie_docobj_rect(HWND frame, RECT* rc)
71 {
72     HWND hwndRebar = GetDlgItem(frame, IDC_BROWSE_REBAR);
73     HWND hwndStatus = GetDlgItem(frame, IDC_BROWSE_STATUSBAR);
74     INT barHeight = SendMessageW(hwndRebar, RB_GETBARHEIGHT, 0, 0);
75
76     rc->top += barHeight;
77     rc->bottom -= barHeight;
78
79     if(IsWindowVisible(hwndStatus))
80     {
81         RECT statusrc;
82
83         GetClientRect(hwndStatus, &statusrc);
84         rc->bottom -= statusrc.bottom - statusrc.top;
85     }
86 }
87
88 static HMENU get_tb_menu(HMENU menu)
89 {
90     HMENU menu_view = GetSubMenu(menu, 1);
91
92     return GetSubMenu(menu_view, 0);
93 }
94
95 static HMENU get_fav_menu(HMENU menu)
96 {
97     return GetSubMenu(menu, 2);
98 }
99
100 static LPWSTR get_fav_url_from_id(HMENU menu, UINT id)
101 {
102     MENUITEMINFOW item;
103
104     item.cbSize = sizeof(item);
105     item.fMask = MIIM_DATA;
106
107     if(!GetMenuItemInfoW(menu, id, FALSE, &item))
108         return NULL;
109
110     return (LPWSTR)item.dwItemData;
111 }
112
113 static void free_fav_menu_data(HMENU menu)
114 {
115     LPWSTR url;
116     int i;
117
118     for(i = 0; (url = get_fav_url_from_id(menu, ID_BROWSE_GOTOFAV_FIRST + i)); i++)
119         heap_free( url );
120 }
121
122 static int get_menu_item_count(HMENU menu)
123 {
124     MENUITEMINFOW item;
125     int count = 0;
126     int i;
127
128     item.cbSize = sizeof(item);
129     item.fMask = MIIM_DATA | MIIM_SUBMENU;
130
131     for(i = 0; GetMenuItemInfoW(menu, i, TRUE, &item); i++)
132     {
133         if(item.hSubMenu)
134             count += get_menu_item_count(item.hSubMenu);
135         else
136             count++;
137     }
138
139     return count;
140 }
141
142 static void add_fav_to_menu(HMENU favmenu, HMENU menu, LPWSTR title, LPCWSTR url)
143 {
144     MENUITEMINFOW item;
145     /* Subtract the number of standard elements in the Favorites menu */
146     int favcount = get_menu_item_count(favmenu) - 2;
147     LPWSTR urlbuf;
148
149     if(favcount > (ID_BROWSE_GOTOFAV_MAX - ID_BROWSE_GOTOFAV_FIRST))
150     {
151         FIXME("Add support for more than %d Favorites\n", favcount);
152         return;
153     }
154
155     urlbuf = heap_alloc((lstrlenW(url) + 1) * sizeof(WCHAR));
156
157     if(!urlbuf)
158         return;
159
160     lstrcpyW(urlbuf, url);
161
162     item.cbSize = sizeof(item);
163     item.fMask = MIIM_FTYPE | MIIM_STRING | MIIM_DATA | MIIM_ID;
164     item.fType = MFT_STRING;
165     item.dwTypeData = title;
166     item.wID = ID_BROWSE_GOTOFAV_FIRST + favcount;
167     item.dwItemData = (ULONG_PTR)urlbuf;
168     InsertMenuItemW(menu, -1, TRUE, &item);
169 }
170
171 static void add_favs_to_menu(HMENU favmenu, HMENU menu, LPCWSTR dir)
172 {
173     WCHAR path[MAX_PATH*2];
174     const WCHAR search[] = {'*',0};
175     WCHAR* filename;
176     HANDLE findhandle;
177     WIN32_FIND_DATAW finddata;
178     IUniformResourceLocatorW* urlobj;
179     IPersistFile* urlfile = NULL;
180     HRESULT res;
181
182     lstrcpyW(path, dir);
183     PathAppendW(path, search);
184
185     findhandle = FindFirstFileW(path, &finddata);
186
187     if(findhandle == INVALID_HANDLE_VALUE)
188         return;
189
190     res = CoCreateInstance(&CLSID_InternetShortcut, NULL, CLSCTX_INPROC_SERVER, &IID_IUniformResourceLocatorW, (PVOID*)&urlobj);
191
192     if(SUCCEEDED(res))
193         res = IUnknown_QueryInterface((IUnknown*)urlobj, &IID_IPersistFile, (PVOID*)&urlfile);
194
195     if(SUCCEEDED(res))
196     {
197         filename = path + lstrlenW(path) - lstrlenW(search);
198
199         do
200         {
201             lstrcpyW(filename, finddata.cFileName);
202
203             if(finddata.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
204             {
205                 MENUITEMINFOW item;
206                 const WCHAR ignore1[] = {'.','.',0};
207                 const WCHAR ignore2[] = {'.',0};
208
209                 if(!lstrcmpW(filename, ignore1) || !lstrcmpW(filename, ignore2))
210                     continue;
211
212                 item.cbSize = sizeof(item);
213                 item.fMask = MIIM_STRING | MIIM_SUBMENU;
214                 item.dwTypeData = filename;
215                 item.hSubMenu = CreatePopupMenu();
216                 InsertMenuItemW(menu, -1, TRUE, &item);
217                 add_favs_to_menu(favmenu, item.hSubMenu, path);
218             } else
219             {
220                 WCHAR* fileext;
221                 WCHAR* url = NULL;
222                 const WCHAR urlext[] = {'.','u','r','l',0};
223
224                 if(lstrcmpiW(PathFindExtensionW(filename), urlext))
225                     continue;
226
227                 if(FAILED(IPersistFile_Load(urlfile, path, 0)))
228                     continue;
229
230                 urlobj->lpVtbl->GetURL(urlobj, &url);
231
232                 if(!url)
233                     continue;
234
235                 fileext = filename + lstrlenW(filename) - lstrlenW(urlext);
236                 *fileext = 0;
237                 add_fav_to_menu(favmenu, menu, filename, url);
238             }
239         } while(FindNextFileW(findhandle, &finddata));
240     }
241
242     if(urlfile)
243         IPersistFile_Release(urlfile);
244
245     if(urlobj)
246         IUnknown_Release((IUnknown*)urlobj);
247
248     FindClose(findhandle);
249 }
250
251 static void add_tbs_to_menu(HMENU menu)
252 {
253     HUSKEY toolbar_handle;
254     WCHAR toolbar_key[] = {'S','o','f','t','w','a','r','e','\\',
255                            'M','i','c','r','o','s','o','f','t','\\',
256                            'I','n','t','e','r','n','e','t',' ',
257                            'E','x','p','l','o','r','e','r','\\',
258                            'T','o','o','l','b','a','r',0};
259
260     if(SHRegOpenUSKeyW(toolbar_key, KEY_READ, NULL, &toolbar_handle, TRUE) == ERROR_SUCCESS)
261     {
262         HUSKEY classes_handle;
263         WCHAR classes_key[] = {'S','o','f','t','w','a','r','e','\\',
264                                'C','l','a','s','s','e','s','\\','C','L','S','I','D',0};
265         WCHAR guid[39];
266         DWORD value_len = sizeof(guid)/sizeof(guid[0]);
267         int i;
268
269         if(SHRegOpenUSKeyW(classes_key, KEY_READ, NULL, &classes_handle, TRUE) != ERROR_SUCCESS)
270         {
271             SHRegCloseUSKey(toolbar_handle);
272             ERR("Failed to open key %s\n", debugstr_w(classes_key));
273             return;
274         }
275
276         for(i = 0; SHRegEnumUSValueW(toolbar_handle, i, guid, &value_len, NULL, NULL, NULL, SHREGENUM_HKLM) == ERROR_SUCCESS; i++)
277         {
278             WCHAR tb_name[100];
279             DWORD tb_name_len = sizeof(tb_name)/sizeof(tb_name[0]);
280             HUSKEY tb_class_handle;
281             MENUITEMINFOW item;
282             LSTATUS ret;
283             value_len = sizeof(guid)/sizeof(guid[0]);
284
285             if(lstrlenW(guid) != 38)
286             {
287                 TRACE("Found invalid IE toolbar entry: %s\n", debugstr_w(guid));
288                 continue;
289             }
290
291             if(SHRegOpenUSKeyW(guid, KEY_READ, classes_handle, &tb_class_handle, TRUE) != ERROR_SUCCESS)
292             {
293                 ERR("Failed to get class info for %s\n", debugstr_w(guid));
294                 continue;
295             }
296
297             ret = SHRegQueryUSValueW(tb_class_handle, NULL, NULL, tb_name, &tb_name_len, TRUE, NULL, 0);
298
299             SHRegCloseUSKey(tb_class_handle);
300
301             if(ret != ERROR_SUCCESS)
302             {
303                 ERR("Failed to get toolbar name for %s\n", debugstr_w(guid));
304                 continue;
305             }
306
307             item.cbSize = sizeof(item);
308             item.fMask = MIIM_STRING;
309             item.dwTypeData = tb_name;
310             InsertMenuItemW(menu, GetMenuItemCount(menu), TRUE, &item);
311         }
312
313         SHRegCloseUSKey(classes_handle);
314         SHRegCloseUSKey(toolbar_handle);
315     }
316 }
317
318 static HMENU create_ie_menu(void)
319 {
320     HMENU menu = LoadMenuW(ieframe_instance, MAKEINTRESOURCEW(IDR_BROWSE_MAIN_MENU));
321     HMENU favmenu = get_fav_menu(menu);
322     WCHAR path[MAX_PATH];
323
324     add_tbs_to_menu(get_tb_menu(menu));
325
326     if(SHGetFolderPathW(NULL, CSIDL_COMMON_FAVORITES, NULL, SHGFP_TYPE_CURRENT, path) == S_OK)
327         add_favs_to_menu(favmenu, favmenu, path);
328
329     if(SHGetFolderPathW(NULL, CSIDL_FAVORITES, NULL, SHGFP_TYPE_CURRENT, path) == S_OK)
330         add_favs_to_menu(favmenu, favmenu, path);
331
332     return menu;
333 }
334
335 static void ie_navigate(InternetExplorer* This, LPCWSTR url)
336 {
337     VARIANT variant;
338
339     V_VT(&variant) = VT_BSTR;
340     V_BSTR(&variant) = SysAllocString(url);
341
342     IWebBrowser2_Navigate2(&This->IWebBrowser2_iface, &variant, NULL, NULL, NULL, NULL);
343
344     SysFreeString(V_BSTR(&variant));
345 }
346
347 static INT_PTR CALLBACK ie_dialog_open_proc(HWND hwnd, UINT msg, WPARAM wparam, LPARAM lparam)
348 {
349     static InternetExplorer* This;
350
351     switch(msg)
352     {
353         case WM_INITDIALOG:
354             This = (InternetExplorer*)lparam;
355             EnableWindow(GetDlgItem(hwnd, IDOK), FALSE);
356             return TRUE;
357
358         case WM_COMMAND:
359             switch(LOWORD(wparam))
360             {
361                 case IDC_BROWSE_OPEN_URL:
362                 {
363                     HWND hwndurl = GetDlgItem(hwnd, IDC_BROWSE_OPEN_URL);
364                     int len = GetWindowTextLengthW(hwndurl);
365
366                     EnableWindow(GetDlgItem(hwnd, IDOK), len != 0);
367                     break;
368                 }
369                 case IDOK:
370                 {
371                     HWND hwndurl = GetDlgItem(hwnd, IDC_BROWSE_OPEN_URL);
372                     int len = GetWindowTextLengthW(hwndurl);
373
374                     if(len)
375                     {
376                         VARIANT url;
377
378                         V_VT(&url) = VT_BSTR;
379                         V_BSTR(&url) = SysAllocStringLen(NULL, len);
380
381                         GetWindowTextW(hwndurl, V_BSTR(&url), len + 1);
382                         IWebBrowser2_Navigate2(&This->IWebBrowser2_iface, &url, NULL, NULL, NULL, NULL);
383
384                         SysFreeString(V_BSTR(&url));
385                     }
386                 }
387                 /* fall through */
388                 case IDCANCEL:
389                     EndDialog(hwnd, wparam);
390                     return TRUE;
391             }
392     }
393     return FALSE;
394 }
395
396 static void ie_dialog_about(HWND hwnd)
397 {
398     HICON icon = LoadImageW(GetModuleHandleW(0), MAKEINTRESOURCEW(IDI_APPICON), IMAGE_ICON, 48, 48, LR_SHARED);
399
400     ShellAboutW(hwnd, wszWineInternetExplorer, NULL, icon);
401
402     DestroyIcon(icon);
403 }
404
405 static void add_tb_separator(HWND hwnd)
406 {
407     TBBUTTON btn;
408
409     ZeroMemory(&btn, sizeof(btn));
410
411     btn.iBitmap = 3;
412     btn.fsStyle = BTNS_SEP;
413     SendMessageW(hwnd, TB_ADDBUTTONSW, 1, (LPARAM)&btn);
414 }
415
416 static void add_tb_button(HWND hwnd, int bmp, int cmd, int strId)
417 {
418     TBBUTTON btn;
419     WCHAR buf[30];
420
421     LoadStringW(ieframe_instance, strId, buf, sizeof(buf)/sizeof(buf[0]));
422
423     btn.iBitmap = bmp;
424     btn.idCommand = cmd;
425     btn.fsState = TBSTATE_ENABLED;
426     btn.fsStyle = BTNS_SHOWTEXT;
427     btn.dwData = 0;
428     btn.iString = (INT_PTR)buf;
429
430     SendMessageW(hwnd, TB_ADDBUTTONSW, 1, (LPARAM)&btn);
431 }
432
433 static void create_rebar(HWND hwnd)
434 {
435     HWND hwndRebar;
436     HWND hwndAddress;
437     HWND hwndToolbar;
438     REBARINFO rebarinf;
439     REBARBANDINFOW bandinf;
440     WCHAR addr[40];
441     HIMAGELIST imagelist;
442     WCHAR idb_ietoolbar[] = {'I','D','B','_','I','E','T','O','O','L','B','A','R',0};
443
444     LoadStringW(ieframe_instance, IDS_ADDRESS, addr, sizeof(addr)/sizeof(addr[0]));
445
446     hwndRebar = CreateWindowExW(WS_EX_TOOLWINDOW, REBARCLASSNAMEW, NULL,
447             WS_CHILD|WS_VISIBLE|WS_CLIPSIBLINGS|WS_CLIPCHILDREN|RBS_VARHEIGHT|CCS_TOP|CCS_NODIVIDER, 0, 0, 0, 0,
448             hwnd, (HMENU)IDC_BROWSE_REBAR, ieframe_instance, NULL);
449
450     rebarinf.cbSize = sizeof(rebarinf);
451     rebarinf.fMask = 0;
452     rebarinf.himl = NULL;
453     rebarinf.cbSize = sizeof(rebarinf);
454
455     SendMessageW(hwndRebar, RB_SETBARINFO, 0, (LPARAM)&rebarinf);
456
457     hwndToolbar = CreateWindowExW(TBSTYLE_EX_MIXEDBUTTONS, TOOLBARCLASSNAMEW, NULL, TBSTYLE_FLAT | WS_CHILD | WS_VISIBLE,
458             0, 0, 0, 0, hwndRebar, (HMENU)IDC_BROWSE_TOOLBAR, ieframe_instance, NULL);
459
460     imagelist = ImageList_LoadImageW(ieframe_instance, idb_ietoolbar, 32, 0, CLR_NONE, IMAGE_BITMAP, LR_CREATEDIBSECTION);
461
462     SendMessageW(hwndToolbar, TB_SETIMAGELIST, 0, (LPARAM)imagelist);
463     SendMessageW(hwndToolbar, TB_BUTTONSTRUCTSIZE, sizeof(TBBUTTON), 0);
464     add_tb_button(hwndToolbar, 0, ID_BROWSE_BACK, IDS_TB_BACK);
465     add_tb_button(hwndToolbar, 1, ID_BROWSE_FORWARD, IDS_TB_FORWARD);
466     add_tb_button(hwndToolbar, 2, ID_BROWSE_STOP, IDS_TB_STOP);
467     add_tb_button(hwndToolbar, 3, ID_BROWSE_REFRESH, IDS_TB_REFRESH);
468     add_tb_button(hwndToolbar, 4, ID_BROWSE_HOME, IDS_TB_HOME);
469     add_tb_separator(hwndToolbar);
470     add_tb_button(hwndToolbar, 5, ID_BROWSE_PRINT, IDS_TB_PRINT);
471     SendMessageW(hwndToolbar, TB_SETBUTTONSIZE, 0, MAKELPARAM(55,50));
472     SendMessageW(hwndToolbar, TB_AUTOSIZE, 0, 0);
473
474     bandinf.cbSize = sizeof(bandinf);
475     bandinf.fMask = RBBIM_STYLE | RBBIM_CHILD | RBBIM_CHILDSIZE | RBBIM_SIZE;
476     bandinf.fStyle = RBBS_CHILDEDGE;
477     bandinf.cx = 100;
478     bandinf.cyMinChild = 52;
479     bandinf.hwndChild = hwndToolbar;
480
481     SendMessageW(hwndRebar, RB_INSERTBANDW, -1, (LPARAM)&bandinf);
482
483     hwndAddress = CreateWindowExW(0, WC_COMBOBOXEXW, NULL, WS_BORDER|WS_CHILD|WS_VISIBLE|CBS_DROPDOWN,
484             0, 0, 100,20,hwndRebar, (HMENU)IDC_BROWSE_ADDRESSBAR, ieframe_instance, NULL);
485
486     bandinf.fMask |= RBBIM_TEXT;
487     bandinf.fStyle = RBBS_CHILDEDGE | RBBS_BREAK;
488     bandinf.lpText = addr;
489     bandinf.cyMinChild = 20;
490     bandinf.hwndChild = hwndAddress;
491
492     SendMessageW(hwndRebar, RB_INSERTBANDW, -1, (LPARAM)&bandinf);
493 }
494
495 static LRESULT iewnd_OnCreate(HWND hwnd, LPCREATESTRUCTW lpcs)
496 {
497     InternetExplorer* This = (InternetExplorer*)lpcs->lpCreateParams;
498     SetWindowLongPtrW(hwnd, 0, (LONG_PTR) lpcs->lpCreateParams);
499
500     This->menu = create_ie_menu();
501
502     This->status_hwnd = CreateStatusWindowW(WS_VISIBLE|WS_CHILD|SBT_NOBORDERS|CCS_NODIVIDER,
503                                             NULL, hwnd, IDC_BROWSE_STATUSBAR);
504     SendMessageW(This->status_hwnd, SB_SIMPLE, TRUE, 0);
505
506     create_rebar(hwnd);
507
508     return 0;
509 }
510
511 static LRESULT iewnd_OnSize(InternetExplorer *This, INT width, INT height)
512 {
513     HWND hwndRebar = GetDlgItem(This->frame_hwnd, IDC_BROWSE_REBAR);
514     INT barHeight = SendMessageW(hwndRebar, RB_GETBARHEIGHT, 0, 0);
515     RECT docarea = {0, 0, width, height};
516
517     SendMessageW(This->status_hwnd, WM_SIZE, 0, 0);
518
519     adjust_ie_docobj_rect(This->frame_hwnd, &docarea);
520
521     if(This->doc_host->doc_host.hwnd)
522         SetWindowPos(This->doc_host->doc_host.hwnd, NULL, docarea.left, docarea.top, docarea.right, docarea.bottom,
523                      SWP_NOZORDER | SWP_NOACTIVATE);
524
525     SetWindowPos(hwndRebar, NULL, 0, 0, width, barHeight, SWP_NOZORDER | SWP_NOACTIVATE);
526
527     return 0;
528 }
529
530 static LRESULT iewnd_OnNotify(InternetExplorer *This, WPARAM wparam, LPARAM lparam)
531 {
532     NMHDR* hdr = (NMHDR*)lparam;
533
534     if(hdr->idFrom == IDC_BROWSE_ADDRESSBAR && hdr->code == CBEN_ENDEDITW)
535     {
536         NMCBEENDEDITW* info = (NMCBEENDEDITW*)lparam;
537
538         if(info->fChanged && info->iWhy == CBENF_RETURN)
539         {
540             VARIANT vt;
541
542             V_VT(&vt) = VT_BSTR;
543             V_BSTR(&vt) = SysAllocString(info->szText);
544
545             IWebBrowser2_Navigate2(&This->IWebBrowser2_iface, &vt, NULL, NULL, NULL, NULL);
546
547             SysFreeString(V_BSTR(&vt));
548
549             return 0;
550         }
551     }
552
553     return 0;
554 }
555
556 static LRESULT iewnd_OnDestroy(InternetExplorer *This)
557 {
558     HWND hwndRebar = GetDlgItem(This->frame_hwnd, IDC_BROWSE_REBAR);
559     HWND hwndToolbar = GetDlgItem(hwndRebar, IDC_BROWSE_TOOLBAR);
560     HIMAGELIST list = (HIMAGELIST)SendMessageW(hwndToolbar, TB_GETIMAGELIST, 0, 0);
561
562     TRACE("%p\n", This);
563
564     free_fav_menu_data(get_fav_menu(This->menu));
565     ImageList_Destroy(list);
566     This->frame_hwnd = NULL;
567
568     return 0;
569 }
570
571 static LRESULT iewnd_OnCommand(InternetExplorer *This, HWND hwnd, UINT msg, WPARAM wparam, LPARAM lparam)
572 {
573     switch(LOWORD(wparam))
574     {
575         case ID_BROWSE_OPEN:
576             DialogBoxParamW(ieframe_instance, MAKEINTRESOURCEW(IDD_BROWSE_OPEN), hwnd, ie_dialog_open_proc, (LPARAM)This);
577             break;
578
579         case ID_BROWSE_PRINT:
580             if(This->doc_host->doc_host.document)
581             {
582                 IOleCommandTarget* target;
583
584                 if(FAILED(IUnknown_QueryInterface(This->doc_host->doc_host.document, &IID_IOleCommandTarget, (LPVOID*)&target)))
585                     break;
586
587                 IOleCommandTarget_Exec(target, &CGID_MSHTML, IDM_PRINT, OLECMDEXECOPT_DODEFAULT, NULL, NULL);
588
589                 IOleCommandTarget_Release(target);
590             }
591             break;
592
593         case ID_BROWSE_HOME:
594             IWebBrowser2_GoHome(&This->IWebBrowser2_iface);
595             break;
596
597         case ID_BROWSE_BACK:
598             IWebBrowser2_GoBack(&This->IWebBrowser2_iface);
599             break;
600
601         case ID_BROWSE_FORWARD:
602             IWebBrowser2_GoForward(&This->IWebBrowser2_iface);
603             break;
604
605         case ID_BROWSE_STOP:
606             IWebBrowser2_Stop(&This->IWebBrowser2_iface);
607             break;
608
609         case ID_BROWSE_REFRESH:
610             IWebBrowser2_Refresh(&This->IWebBrowser2_iface);
611             break;
612
613         case ID_BROWSE_ABOUT:
614             ie_dialog_about(hwnd);
615             break;
616
617         case ID_BROWSE_QUIT:
618             ShowWindow(hwnd, SW_HIDE);
619             break;
620
621         default:
622             if(LOWORD(wparam) >= ID_BROWSE_GOTOFAV_FIRST && LOWORD(wparam) <= ID_BROWSE_GOTOFAV_MAX)
623             {
624                 LPCWSTR url = get_fav_url_from_id(get_fav_menu(This->menu), LOWORD(wparam));
625
626                 if(url)
627                     ie_navigate(This, url);
628             }
629             return DefWindowProcW(hwnd, msg, wparam, lparam);
630     }
631     return 0;
632 }
633
634 static LRESULT update_addrbar(InternetExplorer *This, LPARAM lparam)
635 {
636     HWND hwndRebar = GetDlgItem(This->frame_hwnd, IDC_BROWSE_REBAR);
637     HWND hwndAddress = GetDlgItem(hwndRebar, IDC_BROWSE_ADDRESSBAR);
638     HWND hwndEdit = (HWND)SendMessageW(hwndAddress, CBEM_GETEDITCONTROL, 0, 0);
639     LPCWSTR url = (LPCWSTR)lparam;
640
641     SendMessageW(hwndEdit, WM_SETTEXT, 0, (LPARAM)url);
642
643     return 0;
644 }
645
646 static LRESULT WINAPI ie_window_proc(HWND hwnd, UINT msg, WPARAM wparam, LPARAM lparam)
647 {
648     InternetExplorer *This = (InternetExplorer*) GetWindowLongPtrW(hwnd, 0);
649
650     switch (msg)
651     {
652     case WM_CREATE:
653         return iewnd_OnCreate(hwnd, (LPCREATESTRUCTW)lparam);
654     case WM_CLOSE:
655         TRACE("WM_CLOSE\n");
656         ShowWindow(hwnd, SW_HIDE);
657         return 0;
658     case WM_SHOWWINDOW:
659         TRACE("WM_SHOWWINDOW %lx\n", wparam);
660         if(wparam)
661             IWebBrowser2_AddRef(&This->IWebBrowser2_iface);
662         else
663             IWebBrowser2_Release(&This->IWebBrowser2_iface);
664         break;
665     case WM_DESTROY:
666         return iewnd_OnDestroy(This);
667     case WM_SIZE:
668         return iewnd_OnSize(This, LOWORD(lparam), HIWORD(lparam));
669     case WM_COMMAND:
670         return iewnd_OnCommand(This, hwnd, msg, wparam, lparam);
671     case WM_NOTIFY:
672         return iewnd_OnNotify(This, wparam, lparam);
673     case WM_DOCHOSTTASK:
674         return process_dochost_tasks(&This->doc_host->doc_host);
675     case WM_UPDATEADDRBAR:
676         return update_addrbar(This, lparam);
677     }
678     return DefWindowProcW(hwnd, msg, wparam, lparam);
679 }
680
681 void register_iewindow_class(void)
682 {
683     WNDCLASSEXW wc;
684
685     memset(&wc, 0, sizeof wc);
686     wc.cbSize = sizeof(wc);
687     wc.style = 0;
688     wc.lpfnWndProc = ie_window_proc;
689     wc.cbClsExtra = 0;
690     wc.cbWndExtra = sizeof(InternetExplorer*);
691     wc.hInstance = ieframe_instance;
692     wc.hIcon = LoadIconW(GetModuleHandleW(0), MAKEINTRESOURCEW(IDI_APPICON));
693     wc.hIconSm = LoadImageW(GetModuleHandleW(0), MAKEINTRESOURCEW(IDI_APPICON), IMAGE_ICON,
694                             GetSystemMetrics(SM_CXSMICON), GetSystemMetrics(SM_CYSMICON), LR_SHARED);
695     wc.hCursor = LoadCursorW(0, MAKEINTRESOURCEW(IDC_ARROW));
696     wc.hbrBackground = 0;
697     wc.lpszClassName = szIEWinFrame;
698     wc.lpszMenuName = NULL;
699
700     RegisterClassExW(&wc);
701 }
702
703 void unregister_iewindow_class(void)
704 {
705     UnregisterClassW(szIEWinFrame, ieframe_instance);
706 }
707
708 static void create_frame_hwnd(InternetExplorer *This)
709 {
710     This->frame_hwnd = CreateWindowExW(
711             WS_EX_WINDOWEDGE,
712             szIEWinFrame, wszWineInternetExplorer,
713             WS_CLIPCHILDREN | WS_CAPTION | WS_SYSMENU | WS_THICKFRAME
714                 | WS_MINIMIZEBOX | WS_MAXIMIZEBOX,
715             CW_USEDEFAULT, CW_USEDEFAULT, CW_USEDEFAULT, CW_USEDEFAULT,
716             NULL, NULL /* FIXME */, ieframe_instance, This);
717
718     This->doc_host->doc_host.frame_hwnd = This->frame_hwnd;
719     create_doc_view_hwnd(&This->doc_host->doc_host);
720 }
721
722 static inline IEDocHost *impl_from_DocHost(DocHost *iface)
723 {
724     return CONTAINING_RECORD(iface, IEDocHost, doc_host);
725 }
726
727 static ULONG IEDocHost_addref(DocHost *iface)
728 {
729     IEDocHost *This = impl_from_DocHost(iface);
730     LONG ref = InterlockedIncrement(&This->ref);
731
732     TRACE("(%p) ref=%d\n", This, ref);
733
734     return ref;
735 }
736
737 static ULONG IEDocHost_release(DocHost *iface)
738 {
739     IEDocHost *This = impl_from_DocHost(iface);
740     LONG ref = InterlockedDecrement(&This->ref);
741
742     TRACE("(%p) ref=%d\n", This, ref);
743
744     if(!ref) {
745         if(This->ie)
746             ERR("This->ie is not NULL\n");
747         heap_free(This);
748     }
749
750     return ref;
751 }
752
753 static void WINAPI DocHostContainer_GetDocObjRect(DocHost* This, RECT* rc)
754 {
755     GetClientRect(This->frame_hwnd, rc);
756     adjust_ie_docobj_rect(This->frame_hwnd, rc);
757 }
758
759 static HRESULT WINAPI DocHostContainer_SetStatusText(DocHost *iface, LPCWSTR text)
760 {
761     IEDocHost *This = impl_from_DocHost(iface);
762     return update_ie_statustext(This->ie, text);
763 }
764
765 static void WINAPI DocHostContainer_SetURL(DocHost* iface, LPCWSTR url)
766 {
767     IEDocHost *This = impl_from_DocHost(iface);
768
769     if(!This->ie)
770         return;
771
772     This->ie->nohome = FALSE;
773     SendMessageW(This->ie->frame_hwnd, WM_UPDATEADDRBAR, 0, (LPARAM)url);
774 }
775
776 static HRESULT DocHostContainer_exec(DocHost* This, const GUID *cmd_group, DWORD cmdid, DWORD execopt, VARIANT *in,
777         VARIANT *out)
778 {
779     return S_OK;
780 }
781 static const IDocHostContainerVtbl DocHostContainerVtbl = {
782     IEDocHost_addref,
783     IEDocHost_release,
784     DocHostContainer_GetDocObjRect,
785     DocHostContainer_SetStatusText,
786     DocHostContainer_SetURL,
787     DocHostContainer_exec
788 };
789
790 static HRESULT create_ie(InternetExplorer **ret_obj)
791 {
792     InternetExplorer *ret;
793
794     ret = heap_alloc_zero(sizeof(InternetExplorer));
795     if(!ret)
796         return E_OUTOFMEMORY;
797
798     ret->doc_host = heap_alloc_zero(sizeof(IEDocHost));
799     if(!ret->doc_host) {
800         heap_free(ret);
801         return E_OUTOFMEMORY;
802     }
803
804     ret->ref = 1;
805     ret->doc_host->ref = 1;
806     ret->doc_host->ie = ret;
807
808     DocHost_Init(&ret->doc_host->doc_host, &ret->IWebBrowser2_iface, &DocHostContainerVtbl);
809
810     InternetExplorer_WebBrowser_Init(ret);
811
812     HlinkFrame_Init(&ret->hlink_frame, (IUnknown*)&ret->IWebBrowser2_iface, &ret->doc_host->doc_host);
813
814     create_frame_hwnd(ret);
815
816     InterlockedIncrement(&obj_cnt);
817     list_add_tail(&ie_list, &ret->entry);
818     *ret_obj = ret;
819     return S_OK;
820 }
821
822 HRESULT WINAPI InternetExplorer_Create(IClassFactory *iface, IUnknown *pOuter, REFIID riid, void **ppv)
823 {
824     InternetExplorer *ret;
825     HRESULT hres;
826
827     TRACE("(%p %s %p)\n", pOuter, debugstr_guid(riid), ppv);
828
829     hres = create_ie(&ret);
830     if(FAILED(hres))
831         return hres;
832
833     hres = IWebBrowser2_QueryInterface(&ret->IWebBrowser2_iface, riid, ppv);
834     IWebBrowser2_Release(&ret->IWebBrowser2_iface);
835     if(FAILED(hres))
836         return hres;
837
838     return S_OK;
839 }
840
841 void released_obj(void)
842 {
843     if(!InterlockedDecrement(&obj_cnt))
844         PostQuitMessage(0);
845 }
846
847 static BOOL create_ie_window(const WCHAR *cmdline)
848 {
849     InternetExplorer *ie;
850     HRESULT hres;
851
852     hres = create_ie(&ie);
853     if(FAILED(hres))
854         return FALSE;
855
856     IWebBrowser2_put_Visible(&ie->IWebBrowser2_iface, VARIANT_TRUE);
857     IWebBrowser2_put_MenuBar(&ie->IWebBrowser2_iface, VARIANT_TRUE);
858
859     if(!*cmdline) {
860         IWebBrowser2_GoHome(&ie->IWebBrowser2_iface);
861     }else {
862         VARIANT var_url;
863         int cmdlen;
864
865         static const WCHAR nohomeW[] = {'-','n','o','h','o','m','e'};
866
867         while(*cmdline == ' ' || *cmdline == '\t')
868             cmdline++;
869         cmdlen = strlenW(cmdline);
870         if(cmdlen > 2 && cmdline[0] == '"' && cmdline[cmdlen-1] == '"') {
871             cmdline++;
872             cmdlen -= 2;
873         }
874
875         if(cmdlen == sizeof(nohomeW)/sizeof(*nohomeW) && !memcmp(cmdline, nohomeW, sizeof(nohomeW))) {
876             ie->nohome = TRUE;
877         }else {
878             V_VT(&var_url) = VT_BSTR;
879
880             V_BSTR(&var_url) = SysAllocStringLen(cmdline, cmdlen);
881
882             /* navigate to the first page */
883             IWebBrowser2_Navigate2(&ie->IWebBrowser2_iface, &var_url, NULL, NULL, NULL, NULL);
884
885             SysFreeString(V_BSTR(&var_url));
886         }
887     }
888
889     IWebBrowser2_Release(&ie->IWebBrowser2_iface);
890     return TRUE;
891 }
892
893 static HDDEDATA open_dde_url(WCHAR *dde_url)
894 {
895     InternetExplorer *ie = NULL, *iter;
896     WCHAR *url, *url_end;
897     VARIANT urlv;
898     HRESULT hres;
899
900     TRACE("%s\n", debugstr_w(dde_url));
901
902     url = dde_url;
903     if(*url == '"') {
904         url++;
905         url_end = strchrW(url, '"');
906         if(!url_end) {
907             FIXME("missing string terminator\n");
908             return 0;
909         }
910         *url_end = 0;
911     }else {
912         url_end = strchrW(url, ',');
913         if(url_end)
914             *url_end = 0;
915         else
916             url_end = url + strlenW(url);
917     }
918
919     LIST_FOR_EACH_ENTRY(iter, &ie_list, InternetExplorer, entry) {
920         if(iter->nohome) {
921             IWebBrowser2_AddRef(&iter->IWebBrowser2_iface);
922             ie = iter;
923             break;
924         }
925     }
926
927     if(!ie) {
928         hres = create_ie(&ie);
929         if(FAILED(hres))
930             return 0;
931     }
932
933     IWebBrowser2_put_Visible(&ie->IWebBrowser2_iface, VARIANT_TRUE);
934     IWebBrowser2_put_MenuBar(&ie->IWebBrowser2_iface, VARIANT_TRUE);
935
936     V_VT(&urlv) = VT_BSTR;
937     V_BSTR(&urlv) = SysAllocStringLen(url, url_end-url);
938     if(!V_BSTR(&urlv)) {
939         IWebBrowser2_Release(&ie->IWebBrowser2_iface);
940         return 0;
941     }
942
943     hres = IWebBrowser2_Navigate2(&ie->IWebBrowser2_iface, &urlv, NULL, NULL, NULL, NULL);
944     if(FAILED(hres))
945         return 0;
946
947     IWebBrowser2_Release(&ie->IWebBrowser2_iface);
948     return ULongToHandle(DDE_FACK);
949 }
950
951 static HDDEDATA WINAPI dde_proc(UINT type, UINT uFmt, HCONV hConv, HSZ hsz1, HSZ hsz2, HDDEDATA data,
952         ULONG_PTR dwData1, ULONG_PTR dwData2)
953 {
954     switch(type) {
955     case XTYP_CONNECT:
956         TRACE("XTYP_CONNECT %p\n", hsz1);
957         return ULongToHandle(!DdeCmpStringHandles(hsz1, ddestr_openurl));
958
959     case XTYP_EXECUTE: {
960         WCHAR *url;
961         DWORD size;
962         HDDEDATA ret;
963
964         TRACE("XTYP_EXECUTE %p\n", data);
965
966         size = DdeGetData(data, NULL, 0, 0);
967         if(!size) {
968             WARN("size = 0\n");
969             break;
970         }
971
972         url = heap_alloc(size);
973         if(!url)
974             break;
975
976         if(DdeGetData(data, (BYTE*)url, size, 0) != size) {
977             ERR("error during read\n");
978             break;
979         }
980
981         ret = open_dde_url(url);
982
983         heap_free(url);
984         return ret;
985     }
986
987     case XTYP_REQUEST:
988         FIXME("XTYP_REQUEST\n");
989         break;
990
991     default:
992         TRACE("type %d\n", type);
993     }
994
995     return NULL;
996 }
997
998 static void init_dde(void)
999 {
1000     UINT res;
1001
1002     static const WCHAR iexploreW[] = {'I','E','x','p','l','o','r','e',0};
1003     static const WCHAR openurlW[] = {'W','W','W','_','O','p','e','n','U','R','L',0};
1004
1005     res = DdeInitializeW(&dde_inst, dde_proc, CBF_SKIP_ALLNOTIFICATIONS | CBF_FAIL_ADVISES | CBF_FAIL_POKES, 0);
1006     if(res != DMLERR_NO_ERROR) {
1007         WARN("DdeInitialize failed: %u\n", res);
1008         return;
1009     }
1010
1011     ddestr_iexplore = DdeCreateStringHandleW(dde_inst, iexploreW, CP_WINUNICODE);
1012     if(!ddestr_iexplore)
1013         WARN("Failed to create string handle: %u\n", DdeGetLastError(dde_inst));
1014
1015     ddestr_openurl = DdeCreateStringHandleW(dde_inst, openurlW, CP_WINUNICODE);
1016     if(!ddestr_openurl)
1017         WARN("Failed to create string handle: %u\n", DdeGetLastError(dde_inst));
1018
1019     res = HandleToULong(DdeNameService(dde_inst, ddestr_iexplore, 0, DNS_REGISTER));
1020     if(res != DMLERR_NO_ERROR)
1021         WARN("DdeNameService failed: %u\n", res);
1022 }
1023
1024 static void release_dde(void)
1025 {
1026     if(ddestr_iexplore)
1027         DdeNameService(dde_inst, ddestr_iexplore, 0, DNS_UNREGISTER);
1028     if(ddestr_openurl)
1029         DdeFreeStringHandle(dde_inst, ddestr_openurl);
1030     if(ddestr_iexplore)
1031         DdeFreeStringHandle(dde_inst, ddestr_iexplore);
1032     DdeUninitialize(dde_inst);
1033 }
1034
1035 /******************************************************************
1036  *              IEWinMain            (ieframe.101)
1037  *
1038  * Only returns on error.
1039  */
1040 DWORD WINAPI IEWinMain(const WCHAR *cmdline, int nShowWindow)
1041 {
1042     MSG msg;
1043     HRESULT hres;
1044
1045     static const WCHAR embeddingW[] = {'-','e','m','b','e','d','d','i','n','g',0};
1046
1047     TRACE("%s %d\n", debugstr_w(cmdline), nShowWindow);
1048
1049     CoInitialize(NULL);
1050
1051     hres = register_class_object(TRUE);
1052     if(FAILED(hres)) {
1053         CoUninitialize();
1054         ExitProcess(1);
1055     }
1056
1057     init_dde();
1058
1059     if(strcmpiW(cmdline, embeddingW)) {
1060         if(!create_ie_window(cmdline)) {
1061             CoUninitialize();
1062             ExitProcess(1);
1063         }
1064     }
1065
1066     /* run the message loop for this thread */
1067     while (GetMessageW(&msg, 0, 0, 0))
1068     {
1069         TranslateMessage(&msg);
1070         DispatchMessageW(&msg);
1071     }
1072
1073     register_class_object(FALSE);
1074     release_dde();
1075
1076     CoUninitialize();
1077
1078     ExitProcess(0);
1079     return 0;
1080 }