Add prototype and fix RtlpNtCreateKey.
[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 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     const 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 static const 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     /* @@ Wine registry key: HKCU\Software\Wine\shdocw */
322     r = RegOpenKeyW( HKEY_CURRENT_USER, szMozDlPath, &hkey );
323     if( r == ERROR_SUCCESS )
324     {
325         sz = MAX_PATH;
326         r = RegQueryValueExW( hkey, szUrlVal, NULL, &type, (LPBYTE)szUrl, &sz );
327         RegCloseKey( hkey );
328     }
329     if( r != ERROR_SUCCESS )
330         goto end;
331
332     /* built the path for the download */
333     p = strrchrW( szUrl, '/' );
334     if (!p)
335         goto end;
336     if (!GetTempPathW( MAX_PATH, path ))
337         goto end;
338     strcatW( path, p+1 );
339
340     /* download it */
341     dl = create_dl(info, &bCancelled);
342     r = URLDownloadToFileW( NULL, szUrl, path, 0, dl );
343     if( dl )
344         IBindStatusCallback_Release( dl );
345     if( (r != S_OK) || bCancelled )
346         goto end;
347
348     /* run it */
349     memset( &si, 0, sizeof si );
350     si.cb = sizeof si;
351     r = CreateProcessW( path, NULL, NULL, NULL, 0, 0, NULL, NULL, &si, &pi );
352     if( !r )
353         goto end;
354     WaitForSingleObject( pi.hProcess, INFINITE );
355
356 end:
357     EndDialog( hDlg, 0 );
358     return 0;
359 }
360
361 static INT_PTR CALLBACK
362 dlProc ( HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam)
363 {
364     HANDLE hThread;
365     DWORD ThreadId;
366     HWND hItem;
367
368     switch (uMsg)
369     {
370     case WM_INITDIALOG:
371         SetWindowLongPtrW( hwndDlg, GWLP_USERDATA, 0 );
372         hItem = GetDlgItem(hwndDlg, 1000);
373         if( hItem )
374         {
375             SendMessageW(hItem,PBM_SETRANGE,0,MAKELPARAM(0,100));
376             SendMessageW(hItem,PBM_SETPOS,0,0);
377         }
378         hThread = CreateThread(NULL,0,ThreadFunc,hwndDlg,0,&ThreadId);
379         if (!hThread)
380             return FALSE;
381         return TRUE;
382     case WM_COMMAND:
383         if( wParam == IDCANCEL )
384             SetWindowLongPtrW( hwndDlg, GWLP_USERDATA, 1 );
385         return FALSE;
386     default:
387         return FALSE;
388     }
389 }
390
391 static BOOL SHDOCVW_TryDownloadMozillaControl(void)
392 {
393     DWORD r;
394     WCHAR buf[0x100];
395     static const WCHAR szWine[] = { 'W','i','n','e',0 };
396     HANDLE hsem;
397
398     SetLastError( ERROR_SUCCESS );
399     hsem = CreateSemaphoreA( NULL, 0, 1, "mozctl_install_semaphore");
400     if( GetLastError() != ERROR_ALREADY_EXISTS )
401     {
402         LoadStringW( shdocvw_hinstance, 1001, buf, sizeof buf/sizeof(WCHAR) );
403         r = MessageBoxW(NULL, buf, szWine, MB_YESNO | MB_ICONQUESTION);
404         if( r != IDYES )
405             return FALSE;
406
407         DialogBoxW(shdocvw_hinstance, MAKEINTRESOURCEW(100), 0, dlProc);
408     }
409     else
410         WaitForSingleObject( hsem, INFINITE );
411     ReleaseSemaphore( hsem, 1, NULL );
412     CloseHandle( hsem );
413     
414     return TRUE;
415 }
416  
417 static BOOL SHDOCVW_TryLoadMozillaControl(void)
418 {
419     WCHAR szPath[MAX_PATH];
420     BOOL bTried = FALSE;
421
422     if( hMozCtl != (HMODULE)~0UL )
423         return hMozCtl ? TRUE : FALSE;
424
425     while( 1 )
426     {
427         if( SHDOCVW_GetMozctlPath( szPath, sizeof szPath ) )
428         {
429             hMozCtl = LoadLibraryExW(szPath, NULL, LOAD_WITH_ALTERED_SEARCH_PATH);
430             if( hMozCtl )
431                 return TRUE;
432         }
433         if( bTried )
434         {
435             MESSAGE("You need to install the Mozilla ActiveX control to\n");
436             MESSAGE("use Wine's builtin CLSID_WebBrowser from SHDOCVW.DLL\n");
437             return FALSE;
438         }
439         SHDOCVW_TryDownloadMozillaControl();
440         bTried = TRUE;
441     }
442 }
443
444 /*************************************************************************
445  *              DllGetClassObject (SHDOCVW.@)
446  */
447 HRESULT WINAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID *ppv)
448 {
449     TRACE("\n");
450
451     if( IsEqualGUID( &CLSID_WebBrowser, rclsid ) &&
452         SHDOCVW_TryLoadMozillaControl() )
453     {
454         HRESULT r;
455         fnGetClassObject pGetClassObject;
456
457         TRACE("WebBrowser class %s\n", debugstr_guid(rclsid) );
458
459         pGetClassObject = (fnGetClassObject)
460             GetProcAddress( hMozCtl, "DllGetClassObject" );
461
462         if( !pGetClassObject )
463             return CLASS_E_CLASSNOTAVAILABLE;
464         r = pGetClassObject( &CLSID_MozillaBrowser, riid, ppv );
465
466         TRACE("r = %08lx  *ppv = %p\n", r, *ppv );
467
468         return r;
469     }
470
471     if (IsEqualGUID(&IID_IClassFactory, riid))
472     {
473         /* Pass back our shdocvw class factory */
474         *ppv = (LPVOID)&SHDOCVW_ClassFactory;
475         IClassFactory_AddRef((IClassFactory*)&SHDOCVW_ClassFactory);
476
477         return S_OK;
478     }
479
480     return CLASS_E_CLASSNOTAVAILABLE;
481 }
482
483 /***********************************************************************
484  *              DllGetVersion (SHDOCVW.@)
485  */
486 HRESULT WINAPI DllGetVersion(DLLVERSIONINFO *info)
487 {
488     if (info->cbSize != sizeof(DLLVERSIONINFO)) FIXME("support DLLVERSIONINFO2\n");
489
490     /* this is what IE6 on Windows 98 reports */
491     info->dwMajorVersion = 6;
492     info->dwMinorVersion = 0;
493     info->dwBuildNumber = 2600;
494     info->dwPlatformID = DLLVER_PLATFORM_WINDOWS;
495
496     return NOERROR;
497 }
498
499 /*************************************************************************
500  *              DllInstall (SHDOCVW.@)
501  */
502 HRESULT WINAPI DllInstall(BOOL bInstall, LPCWSTR cmdline)
503 {
504    FIXME("(%s, %s): stub!\n", bInstall ? "TRUE":"FALSE", debugstr_w(cmdline));
505
506    return S_OK;
507 }
508
509 /*************************************************************************
510  * SHDOCVW_LoadShell32
511  *
512  * makes sure the handle to shell32 is valid
513  */
514  BOOL SHDOCVW_LoadShell32(void)
515 {
516      if (SHDOCVW_hshell32)
517        return TRUE;
518      return ((SHDOCVW_hshell32 = LoadLibraryA("shell32.dll")) != NULL);
519 }
520
521 /***********************************************************************
522  *              @ (SHDOCVW.110)
523  *
524  * Called by Win98 explorer.exe main binary, definitely has 0
525  * parameters.
526  */
527 DWORD WINAPI WinList_Init(void)
528 {
529     FIXME("(), stub!\n");
530     return 0x0deadfeed;
531 }
532
533 /***********************************************************************
534  *              @ (SHDOCVW.118)
535  *
536  * Called by Win98 explorer.exe main binary, definitely has only one
537  * parameter.
538  */
539 static BOOL (WINAPI *pShellDDEInit)(BOOL start) = NULL;
540
541 BOOL WINAPI ShellDDEInit(BOOL start)
542 {
543     TRACE("(%d)\n", start);
544
545     if (!pShellDDEInit)
546     {
547       if (!SHDOCVW_LoadShell32())
548         return FALSE;
549       pShellDDEInit = GetProcAddress(SHDOCVW_hshell32, (LPCSTR)188);
550     }
551
552     if (pShellDDEInit)
553       return pShellDDEInit(start);
554     else
555       return FALSE;
556 }
557
558 /***********************************************************************
559  *              @ (SHDOCVW.125)
560  *
561  * Called by Win98 explorer.exe main binary, definitely has 0
562  * parameters.
563  */
564 DWORD WINAPI RunInstallUninstallStubs(void)
565 {
566     FIXME("(), stub!\n");
567     return 0x0deadbee;
568 }
569
570 /***********************************************************************
571  *              SetQueryNetSessionCount (SHDOCVW.@)
572  */
573 DWORD WINAPI SetQueryNetSessionCount(DWORD arg)
574 {
575     FIXME("(%lu), stub!\n", arg);
576     return 0;
577 }