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     LONG 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\shdocvw */
 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 }