Fix segmentation fault caused by incorrect referencing of client audio
[wine] / dlls / shdocvw / shdocvw_main.c
1 /*
2  * SHDOCVW - Internet Explorer Web Control
3  *
4  * Copyright 2001 John R. Sheets (for CodeWeavers)
5  * Copyright 2004 Mike McCormack (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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
20  */
21
22 #include "config.h"
23
24 #include <stdarg.h>
25 #include <stdio.h>
26 #include <string.h>
27
28 #define COBJMACROS
29 #define COM_NO_WINDOWS_H
30
31 #include "windef.h"
32 #include "winbase.h"
33 #include "winreg.h"
34 #include "winuser.h"
35 #include "winnls.h"
36 #include "ole2.h"
37 #include "shlwapi.h"
38
39 #include "shdocvw.h"
40 #include "uuids.h"
41 #include "urlmon.h"
42
43 #include "wine/unicode.h"
44 #include "wine/debug.h"
45
46 #include "initguid.h"
47
48 WINE_DEFAULT_DEBUG_CHANNEL(shdocvw);
49
50 LONG SHDOCVW_refCount = 0;
51
52 static const WCHAR szMozDlPath[] = {
53     'S','o','f','t','w','a','r','e','\\','W','i','n','e','\\',
54     's','h','d','o','c','v','w',0
55 };
56
57 DEFINE_GUID( CLSID_MozillaBrowser, 0x1339B54C,0x3453,0x11D2,0x93,0xB9,0x00,0x00,0x00,0x00,0x00,0x00);
58
59 typedef HRESULT (WINAPI *fnGetClassObject)(REFCLSID rclsid, REFIID iid, LPVOID *ppv);
60 typedef HRESULT (WINAPI *fnCanUnloadNow)(void);
61
62 HINSTANCE shdocvw_hinstance = 0;
63 static HMODULE SHDOCVW_hshell32 = 0;
64 static HMODULE hMozCtl = (HMODULE)~0UL;
65
66
67 /* convert a guid to a wide character string */
68 static void SHDOCVW_guid2wstr( const GUID *guid, LPWSTR wstr )
69 {
70     char str[40];
71
72     sprintf(str, "{%08lX-%04X-%04X-%02X%02X-%02X%02X%02X%02X%02X%02X}",
73            guid->Data1, guid->Data2, guid->Data3,
74            guid->Data4[0], guid->Data4[1], guid->Data4[2], guid->Data4[3],
75            guid->Data4[4], guid->Data4[5], guid->Data4[6], guid->Data4[7] );
76     MultiByteToWideChar( CP_ACP, 0, str, -1, wstr, 40 );
77 }
78
79 static BOOL SHDOCVW_GetMozctlPath( LPWSTR szPath, DWORD sz )
80 {
81     DWORD r, type;
82     BOOL ret = FALSE;
83     HKEY hkey;
84     static const WCHAR szPre[] = {
85         'S','o','f','t','w','a','r','e','\\',
86         'C','l','a','s','s','e','s','\\',
87         'C','L','S','I','D','\\',0 };
88     static const WCHAR szPost[] = {
89         '\\','I','n','p','r','o','c','S','e','r','v','e','r','3','2',0 };
90     WCHAR szRegPath[(sizeof(szPre)+sizeof(szPost))/sizeof(WCHAR)+40];
91
92     strcpyW( szRegPath, szPre );
93     SHDOCVW_guid2wstr( &CLSID_MozillaBrowser, &szRegPath[strlenW(szRegPath)] );
94     strcatW( szRegPath, szPost );
95
96     TRACE("key = %s\n", debugstr_w( szRegPath ) );
97
98     r = RegOpenKeyW( HKEY_LOCAL_MACHINE, szRegPath, &hkey );
99     if( r != ERROR_SUCCESS )
100         return FALSE;
101
102     r = RegQueryValueExW( hkey, NULL, NULL, &type, (LPBYTE)szPath, &sz );
103     ret = ( r == ERROR_SUCCESS ) && ( type == REG_SZ );
104     RegCloseKey( hkey );
105
106     return ret;
107 }
108
109 /*************************************************************************
110  * SHDOCVW DllMain
111  */
112 BOOL WINAPI DllMain(HINSTANCE hinst, DWORD fdwReason, LPVOID fImpLoad)
113 {
114         TRACE("%p 0x%lx %p\n", hinst, fdwReason, fImpLoad);
115         switch (fdwReason)
116         {
117           case DLL_PROCESS_ATTACH:
118             shdocvw_hinstance = hinst;
119             break;
120           case DLL_PROCESS_DETACH:
121             if (SHDOCVW_hshell32) FreeLibrary(SHDOCVW_hshell32);
122             if (hMozCtl && hMozCtl != (HMODULE)~0UL) FreeLibrary(hMozCtl);
123             break;
124         }
125         return TRUE;
126 }
127
128 /*************************************************************************
129  *              DllCanUnloadNow (SHDOCVW.@)
130  */
131 HRESULT WINAPI SHDOCVW_DllCanUnloadNow(void)
132 {
133     HRESULT moz_can_unload = S_OK;
134     fnCanUnloadNow pCanUnloadNow;
135
136     if (hMozCtl)
137     {
138         pCanUnloadNow = (fnCanUnloadNow)
139             GetProcAddress(hMozCtl, "DllCanUnloadNow");
140         if (pCanUnloadNow)
141             moz_can_unload = pCanUnloadNow();
142     }
143
144     if (moz_can_unload == S_OK && SHDOCVW_refCount == 0)
145         return S_OK;
146
147     return S_FALSE;
148 }
149
150 /*************************************************************************
151  *              SHDOCVW_TryDownloadMozillaControl
152  */
153 typedef struct _IBindStatusCallbackImpl {
154     IBindStatusCallbackVtbl *vtbl;
155     DWORD ref;
156     HWND hDialog;
157     BOOL *pbCancelled;
158 } IBindStatusCallbackImpl;
159
160 static HRESULT WINAPI
161 dlQueryInterface( IBindStatusCallback* This, REFIID riid, void** ppvObject )
162 {
163     if (ppvObject == NULL) return E_POINTER;
164     
165     if( IsEqualIID(riid, &IID_IUnknown) ||
166         IsEqualIID(riid, &IID_IBindStatusCallback))
167     {
168         IBindStatusCallback_AddRef( This );
169         *ppvObject = This;
170         return S_OK;
171     }
172     return E_NOINTERFACE;
173 }
174
175 static ULONG WINAPI dlAddRef( IBindStatusCallback* iface )
176 {
177     IBindStatusCallbackImpl *This = (IBindStatusCallbackImpl *) iface;
178
179     SHDOCVW_LockModule();
180     
181     return InterlockedIncrement( &This->ref );
182 }
183
184 static ULONG WINAPI dlRelease( IBindStatusCallback* iface )
185 {
186     IBindStatusCallbackImpl *This = (IBindStatusCallbackImpl *) iface;
187     DWORD ref = InterlockedDecrement( &This->ref );
188     
189     if( !ref )
190     {
191         DestroyWindow( This->hDialog );
192         HeapFree( GetProcessHeap(), 0, This );
193     }
194
195     SHDOCVW_UnlockModule();
196     
197     return ref;
198 }
199
200 static HRESULT WINAPI
201 dlOnStartBinding( IBindStatusCallback* iface, DWORD dwReserved, IBinding* pib)
202 {
203     ERR("\n");
204     return S_OK;
205 }
206
207 static HRESULT WINAPI
208 dlGetPriority( IBindStatusCallback* iface, LONG* pnPriority)
209 {
210     ERR("\n");
211     return S_OK;
212 }
213
214 static HRESULT WINAPI
215 dlOnLowResource( IBindStatusCallback* iface, DWORD reserved)
216 {
217     ERR("\n");
218     return S_OK;
219 }
220
221 static HRESULT WINAPI
222 dlOnProgress( IBindStatusCallback* iface, ULONG ulProgress,
223               ULONG ulProgressMax, ULONG ulStatusCode, LPCWSTR szStatusText)
224 {
225     IBindStatusCallbackImpl *This = (IBindStatusCallbackImpl *) iface;
226     HWND hItem;
227     LONG r;
228
229     hItem = GetDlgItem( This->hDialog, 1000 );
230     if( hItem && ulProgressMax )
231         SendMessageW(hItem,PBM_SETPOS,(ulProgress*100)/ulProgressMax,0);
232
233     hItem = GetDlgItem(This->hDialog, 104);
234     if( hItem )
235         SendMessageW(hItem,WM_SETTEXT, 0, (LPARAM) szStatusText);
236
237     SetLastError(0);
238     r = GetWindowLongPtrW( This->hDialog, GWLP_USERDATA );
239     if( r || GetLastError() )
240     {
241         *This->pbCancelled = TRUE;
242         ERR("Cancelled\n");
243         return E_ABORT;
244     }
245
246     return S_OK;
247 }
248
249 static HRESULT WINAPI
250 dlOnStopBinding( IBindStatusCallback* iface, HRESULT hresult, LPCWSTR szError)
251 {
252     ERR("\n");
253     return S_OK;
254 }
255
256 static HRESULT WINAPI
257 dlGetBindInfo( IBindStatusCallback* iface, DWORD* grfBINDF, BINDINFO* pbindinfo)
258 {
259     ERR("\n");
260     return S_OK;
261 }
262
263 static HRESULT WINAPI
264 dlOnDataAvailable( IBindStatusCallback* iface, DWORD grfBSCF,
265                    DWORD dwSize, FORMATETC* pformatetc, STGMEDIUM* pstgmed)
266 {
267     ERR("\n");
268     return S_OK;
269 }
270
271 static HRESULT WINAPI
272 dlOnObjectAvailable( IBindStatusCallback* iface, REFIID riid, IUnknown* punk)
273 {
274     ERR("\n");
275     return S_OK;
276 }
277
278 struct IBindStatusCallbackVtbl dlVtbl =
279 {
280     dlQueryInterface,
281     dlAddRef,
282     dlRelease,
283     dlOnStartBinding,
284     dlGetPriority,
285     dlOnLowResource,
286     dlOnProgress,
287     dlOnStopBinding,
288     dlGetBindInfo,
289     dlOnDataAvailable,
290     dlOnObjectAvailable
291 };
292
293 static IBindStatusCallback* create_dl(HWND dlg, BOOL *pbCancelled)
294 {
295     IBindStatusCallbackImpl *This;
296
297     This = HeapAlloc( GetProcessHeap(), 0, sizeof *This );
298     This->vtbl = &dlVtbl;
299     This->ref = 1;
300     This->hDialog = dlg;
301     This->pbCancelled = pbCancelled;
302
303     return (IBindStatusCallback*) This;
304 }
305
306 static DWORD WINAPI ThreadFunc( LPVOID info )
307 {
308     IBindStatusCallback *dl;
309     static const WCHAR szUrlVal[] = {'M','o','z','i','l','l','a','U','r','l',0};
310     WCHAR path[MAX_PATH], szUrl[MAX_PATH];
311     LPWSTR p;
312     STARTUPINFOW si;
313     PROCESS_INFORMATION pi;
314     HWND hDlg = info;
315     DWORD r, sz, type;
316     HKEY hkey;
317     BOOL bCancelled = FALSE;
318
319     /* find the name of the thing to download */
320     szUrl[0] = 0;
321     r = RegOpenKeyW( HKEY_LOCAL_MACHINE, szMozDlPath, &hkey );
322     if( r == ERROR_SUCCESS )
323     {
324         sz = MAX_PATH;
325         r = RegQueryValueExW( hkey, szUrlVal, NULL, &type, (LPBYTE)szUrl, &sz );
326         RegCloseKey( hkey );
327     }
328     if( r != ERROR_SUCCESS )
329         goto end;
330
331     /* built the path for the download */
332     p = strrchrW( szUrl, '/' );
333     if (!p)
334         goto end;
335     if (!GetTempPathW( MAX_PATH, path ))
336         goto end;
337     strcatW( path, p+1 );
338
339     /* download it */
340     dl = create_dl(info, &bCancelled);
341     r = URLDownloadToFileW( NULL, szUrl, path, 0, dl );
342     if( dl )
343         IBindStatusCallback_Release( dl );
344     if( (r != S_OK) || bCancelled )
345         goto end;
346
347     /* run it */
348     memset( &si, 0, sizeof si );
349     si.cb = sizeof si;
350     r = CreateProcessW( path, NULL, NULL, NULL, 0, 0, NULL, NULL, &si, &pi );
351     if( !r )
352         goto end;
353     WaitForSingleObject( pi.hProcess, INFINITE );
354
355 end:
356     EndDialog( hDlg, 0 );
357     return 0;
358 }
359
360 static INT_PTR CALLBACK
361 dlProc ( HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam)
362 {
363     HANDLE hThread;
364     DWORD ThreadId;
365     HWND hItem;
366
367     switch (uMsg)
368     {
369     case WM_INITDIALOG:
370         SetWindowLongPtrW( hwndDlg, GWLP_USERDATA, 0 );
371         hItem = GetDlgItem(hwndDlg, 1000);
372         if( hItem )
373         {
374             SendMessageW(hItem,PBM_SETRANGE,0,MAKELPARAM(0,100));
375             SendMessageW(hItem,PBM_SETPOS,0,0);
376         }
377         hThread = CreateThread(NULL,0,ThreadFunc,hwndDlg,0,&ThreadId);
378         if (!hThread)
379             return FALSE;
380         return TRUE;
381     case WM_COMMAND:
382         if( wParam == IDCANCEL )
383             SetWindowLongPtrW( hwndDlg, GWLP_USERDATA, 1 );
384         return FALSE;
385     default:
386         return FALSE;
387     }
388 }
389
390 static BOOL SHDOCVW_TryDownloadMozillaControl()
391 {
392     DWORD r;
393     WCHAR buf[0x100];
394     static const WCHAR szWine[] = { 'W','i','n','e',0 };
395     HANDLE hsem;
396
397     SetLastError( ERROR_SUCCESS );
398     hsem = CreateSemaphoreA( NULL, 0, 1, "mozctl_install_semaphore");
399     if( GetLastError() != ERROR_ALREADY_EXISTS )
400     {
401         LoadStringW( shdocvw_hinstance, 1001, buf, sizeof buf/sizeof(WCHAR) );
402         r = MessageBoxW(NULL, buf, szWine, MB_YESNO | MB_ICONQUESTION);
403         if( r != IDYES )
404             return FALSE;
405
406         DialogBoxW(shdocvw_hinstance, MAKEINTRESOURCEW(100), 0, dlProc);
407     }
408     else
409         WaitForSingleObject( hsem, INFINITE );
410     ReleaseSemaphore( hsem, 1, NULL );
411     CloseHandle( hsem );
412     
413     return TRUE;
414 }
415  
416 static BOOL SHDOCVW_TryLoadMozillaControl()
417 {
418     WCHAR szPath[MAX_PATH];
419     BOOL bTried = FALSE;
420
421     if( hMozCtl != (HMODULE)~0UL )
422         return hMozCtl ? TRUE : FALSE;
423
424     while( 1 )
425     {
426         if( SHDOCVW_GetMozctlPath( szPath, sizeof szPath ) )
427         {
428             hMozCtl = LoadLibraryExW(szPath, NULL, LOAD_WITH_ALTERED_SEARCH_PATH);
429             if( hMozCtl )
430                 return TRUE;
431         }
432         if( bTried )
433         {
434             MESSAGE("You need to install the Mozilla ActiveX control to\n");
435             MESSAGE("use Wine's builtin CLSID_WebBrowser from SHDOCVW.DLL\n");
436             return FALSE;
437         }
438         SHDOCVW_TryDownloadMozillaControl();
439         bTried = TRUE;
440     }
441 }
442
443 /*************************************************************************
444  *              DllGetClassObject (SHDOCVW.@)
445  */
446 HRESULT WINAPI SHDOCVW_DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID *ppv)
447 {
448     TRACE("\n");
449
450     if( IsEqualGUID( &CLSID_WebBrowser, rclsid ) &&
451         SHDOCVW_TryLoadMozillaControl() )
452     {
453         HRESULT r;
454         fnGetClassObject pGetClassObject;
455
456         TRACE("WebBrowser class %s\n", debugstr_guid(rclsid) );
457
458         pGetClassObject = (fnGetClassObject)
459             GetProcAddress( hMozCtl, "DllGetClassObject" );
460
461         if( !pGetClassObject )
462             return CLASS_E_CLASSNOTAVAILABLE;
463         r = pGetClassObject( &CLSID_MozillaBrowser, riid, ppv );
464
465         TRACE("r = %08lx  *ppv = %p\n", r, *ppv );
466
467         return r;
468     }
469
470     if (IsEqualGUID(&IID_IClassFactory, riid))
471     {
472         /* Pass back our shdocvw class factory */
473         *ppv = (LPVOID)&SHDOCVW_ClassFactory;
474         IClassFactory_AddRef((IClassFactory*)&SHDOCVW_ClassFactory);
475
476         return S_OK;
477     }
478
479     return CLASS_E_CLASSNOTAVAILABLE;
480 }
481
482 /***********************************************************************
483  *              DllGetVersion (SHDOCVW.@)
484  */
485 HRESULT WINAPI SHDOCVW_DllGetVersion(DLLVERSIONINFO *info)
486 {
487     if (info->cbSize != sizeof(DLLVERSIONINFO)) FIXME("support DLLVERSIONINFO2\n");
488
489     /* this is what IE6 on Windows 98 reports */
490     info->dwMajorVersion = 6;
491     info->dwMinorVersion = 0;
492     info->dwBuildNumber = 2600;
493     info->dwPlatformID = DLLVER_PLATFORM_WINDOWS;
494
495     return NOERROR;
496 }
497
498 /*************************************************************************
499  *              DllInstall (SHDOCVW.@)
500  */
501 HRESULT WINAPI SHDOCVW_DllInstall(BOOL bInstall, LPCWSTR cmdline)
502 {
503    FIXME("(%s, %s): stub!\n", bInstall ? "TRUE":"FALSE", debugstr_w(cmdline));
504
505    return S_OK;
506 }
507
508 /*************************************************************************
509  * SHDOCVW_LoadShell32
510  *
511  * makes sure the handle to shell32 is valid
512  */
513  BOOL SHDOCVW_LoadShell32(void)
514 {
515      if (SHDOCVW_hshell32)
516        return TRUE;
517      return ((SHDOCVW_hshell32 = LoadLibraryA("shell32.dll")) != NULL);
518 }
519
520 /***********************************************************************
521  *              @ (SHDOCVW.110)
522  *
523  * Called by Win98 explorer.exe main binary, definitely has 0
524  * parameters.
525  */
526 DWORD WINAPI WinList_Init(void)
527 {
528     FIXME("(), stub!\n");
529     return 0x0deadfeed;
530 }
531
532 /***********************************************************************
533  *              @ (SHDOCVW.118)
534  *
535  * Called by Win98 explorer.exe main binary, definitely has only one
536  * parameter.
537  */
538 static BOOL (WINAPI *pShellDDEInit)(BOOL start) = NULL;
539
540 BOOL WINAPI ShellDDEInit(BOOL start)
541 {
542     TRACE("(%d)\n", start);
543
544     if (!pShellDDEInit)
545     {
546       if (!SHDOCVW_LoadShell32())
547         return FALSE;
548       pShellDDEInit = GetProcAddress(SHDOCVW_hshell32, (LPCSTR)188);
549     }
550
551     if (pShellDDEInit)
552       return pShellDDEInit(start);
553     else
554       return FALSE;
555 }
556
557 /***********************************************************************
558  *              @ (SHDOCVW.125)
559  *
560  * Called by Win98 explorer.exe main binary, definitely has 0
561  * parameters.
562  */
563 DWORD WINAPI RunInstallUninstallStubs(void)
564 {
565     FIXME("(), stub!\n");
566     return 0x0deadbee;
567 }
568
569 /***********************************************************************
570  *              SetQueryNetSessionCount (SHDOCVW.@)
571  */
572 DWORD WINAPI SetQueryNetSessionCount(DWORD arg)
573 {
574     FIXME("(%lu), stub!\n", arg);
575     return 0;
576 }