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