git.oblomov.eu Git - wine/blob - dlls/urlmon/urlmon_main.c

   1 /*
   2  * UrlMon
   3  *
   4  * Copyright (c) 2000 Patrik Stridvall
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  */
  20
  21 #include <stdarg.h>
  22
  23 #define COBJMACROS
  24
  25 #include "windef.h"
  26 #include "winbase.h"
  27 #include "winreg.h"
  28
  29 #define NO_SHLWAPI_REG
  30 #include "shlwapi.h"
  31 #include "wine/debug.h"
  32 #include "wine/unicode.h"
  33
  34 #include "winuser.h"
  35 #include "urlmon.h"
  36 #include "urlmon_main.h"
  37 #include "ole2.h"
  38
  39 WINE_DEFAULT_DEBUG_CHANNEL(urlmon);
  40
  41 LONG URLMON_refCount = 0;
  42
  43 HINSTANCE URLMON_hInstance = 0;
  44
  45 DWORD urlmon_tls = 0;
  46
  47 static void init_session(BOOL);
  48
  49 /***********************************************************************
  50  *              DllMain (URLMON.init)
  51  */
  52 BOOL WINAPI DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID fImpLoad)
  53 {
  54     TRACE("%p 0x%lx %p\n", hinstDLL, fdwReason, fImpLoad);
  55
  56     switch(fdwReason) {
  57     case DLL_PROCESS_ATTACH:
  58         DisableThreadLibraryCalls(hinstDLL);
  59         URLMON_hInstance = hinstDLL;
  60         init_session(TRUE);
  61         break;
  62
  63     case DLL_PROCESS_DETACH:
  64         if(urlmon_tls)
  65             TlsFree(urlmon_tls);
  66         init_session(FALSE);
  67         URLMON_hInstance = 0;
  68         break;
  69     }
  70     return TRUE;
  71 }
  72
  73
  74 /***********************************************************************
  75  *              DllInstall (URLMON.@)
  76  */
  77 HRESULT WINAPI DllInstall(BOOL bInstall, LPCWSTR cmdline)
  78 {
  79   FIXME("(%s, %s): stub\n", bInstall?"TRUE":"FALSE",
  80         debugstr_w(cmdline));
  81
  82   return S_OK;
  83 }
  84
  85 /***********************************************************************
  86  *              DllCanUnloadNow (URLMON.@)
  87  */
  88 HRESULT WINAPI DllCanUnloadNow(void)
  89 {
  90     return URLMON_refCount != 0 ? S_FALSE : S_OK;
  91 }
  92
  93
  94
  95 /******************************************************************************
  96  * Urlmon ClassFactory
  97  */
  98 typedef struct {
  99     const IClassFactoryVtbl *lpClassFactoryVtbl;
 100
 101     HRESULT (*pfnCreateInstance)(IUnknown *pUnkOuter, LPVOID *ppObj);
 102 } ClassFactory;
 103
 104 #define CLASSFACTORY(x)  ((IClassFactory*)  &(x)->lpClassFactoryVtbl)
 105
 106 static HRESULT WINAPI CF_QueryInterface(IClassFactory *iface, REFIID riid, LPVOID *ppv)
 107 {
 108     *ppv = NULL;
 109
 110     if(IsEqualGUID(riid, &IID_IUnknown)) {
 111         TRACE("(%p)->(IID_IUnknown %p)\n", iface, ppv);
 112         *ppv = iface;
 113     }else if(IsEqualGUID(riid, &IID_IClassFactory)) {
 114         TRACE("(%p)->(IID_IClassFactory %p)\n", iface, ppv);
 115         *ppv = iface;
 116     }
 117
 118     if(*ppv) {
 119         IUnknown_AddRef((IUnknown*)*ppv);
 120         return S_OK;
 121     }
 122
 123     WARN("(%p)->(%s,%p),not found\n", iface, debugstr_guid(riid), ppv);
 124     return E_NOINTERFACE;
 125 }
 126
 127 static ULONG WINAPI CF_AddRef(IClassFactory *iface)
 128 {
 129     URLMON_LockModule();
 130     return 2;
 131 }
 132
 133 static ULONG WINAPI CF_Release(IClassFactory *iface)
 134 {
 135     URLMON_UnlockModule();
 136     return 1;
 137 }
 138
 139
 140 static HRESULT WINAPI CF_CreateInstance(IClassFactory *iface, IUnknown *pOuter,
 141                                         REFIID riid, LPVOID *ppobj)
 142 {
 143     ClassFactory *This = (ClassFactory*)iface;
 144     HRESULT hres;
 145     LPUNKNOWN punk;
 146
 147     TRACE("(%p)->(%p,%s,%p)\n",This,pOuter,debugstr_guid(riid),ppobj);
 148
 149     *ppobj = NULL;
 150     if(SUCCEEDED(hres = This->pfnCreateInstance(pOuter, (LPVOID *) &punk))) {
 151         hres = IUnknown_QueryInterface(punk, riid, ppobj);
 152         IUnknown_Release(punk);
 153     }
 154     return hres;
 155 }
 156
 157 static HRESULT WINAPI CF_LockServer(LPCLASSFACTORY iface,BOOL dolock)
 158 {
 159     TRACE("(%d)\n", dolock);
 160
 161     if (dolock)
 162            URLMON_LockModule();
 163     else
 164            URLMON_UnlockModule();
 165
 166     return S_OK;
 167 }
 168
 169 static const IClassFactoryVtbl ClassFactoryVtbl =
 170 {
 171     CF_QueryInterface,
 172     CF_AddRef,
 173     CF_Release,
 174     CF_CreateInstance,
 175     CF_LockServer
 176 };
 177
 178 static const ClassFactory FileProtocolCF =
 179     { &ClassFactoryVtbl, FileProtocol_Construct};
 180 static const ClassFactory FtpProtocolCF =
 181     { &ClassFactoryVtbl, FtpProtocol_Construct};
 182 static const ClassFactory HttpProtocolCF =
 183     { &ClassFactoryVtbl, HttpProtocol_Construct};
 184 static const ClassFactory SecurityManagerCF =
 185     { &ClassFactoryVtbl, SecManagerImpl_Construct};
 186 static const ClassFactory ZoneManagerCF =
 187     { &ClassFactoryVtbl, ZoneMgrImpl_Construct};
 188
 189 struct object_creation_info
 190 {
 191     const CLSID *clsid;
 192     IClassFactory *cf;
 193     LPCWSTR protocol;
 194 };
 195
 196 static const WCHAR wszFile[] = {'f','i','l','e',0};
 197 static const WCHAR wszFtp[]  = {'f','t','p',0};
 198 static const WCHAR wszHttp[] = {'h','t','t','p',0};
 199
 200 static const struct object_creation_info object_creation[] =
 201 {
 202     { &CLSID_FileProtocol,            CLASSFACTORY(&FileProtocolCF),    wszFile },
 203     { &CLSID_FtpProtocol,             CLASSFACTORY(&FtpProtocolCF),     wszFtp  },
 204     { &CLSID_HttpProtocol,            CLASSFACTORY(&HttpProtocolCF),    wszHttp },
 205     { &CLSID_InternetSecurityManager, CLASSFACTORY(&SecurityManagerCF), NULL    },
 206     { &CLSID_InternetZoneManager,     CLASSFACTORY(&ZoneManagerCF),     NULL    }
 207 };
 208
 209 static void init_session(BOOL init)
 210 {
 211     IInternetSession *session;
 212     int i;
 213
 214     CoInternetGetSession(0, &session, 0);
 215
 216     for(i=0; i < sizeof(object_creation)/sizeof(object_creation[0]); i++) {
 217         if(object_creation[i].protocol) {
 218             if(init)
 219                 IInternetSession_RegisterNameSpace(session, object_creation[i].cf,
 220                         object_creation[i].clsid, object_creation[i].protocol, 0, NULL, 0);
 221             else
 222                 IInternetSession_UnregisterNameSpace(session, object_creation[i].cf,
 223                         object_creation[i].protocol);
 224         }
 225     }
 226
 227     IInternetSession_Release(session);
 228 }
 229
 230 /*******************************************************************************
 231  * DllGetClassObject [URLMON.@]
 232  * Retrieves class object from a DLL object
 233  *
 234  * NOTES
 235  *    Docs say returns STDAPI
 236  *
 237  * PARAMS
 238  *    rclsid [I] CLSID for the class object
 239  *    riid   [I] Reference to identifier of interface for class object
 240  *    ppv    [O] Address of variable to receive interface pointer for riid
 241  *
 242  * RETURNS
 243  *    Success: S_OK
 244  *    Failure: CLASS_E_CLASSNOTAVAILABLE, E_OUTOFMEMORY, E_INVALIDARG,
 245  *             E_UNEXPECTED
 246  */
 247
 248 HRESULT WINAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID *ppv)
 249 {
 250     int i;
 251
 252     TRACE("(%s,%s,%p)\n", debugstr_guid(rclsid), debugstr_guid(riid), ppv);
 253
 254     for (i=0; i < sizeof(object_creation)/sizeof(object_creation[0]); i++)
 255     {
 256         if (IsEqualGUID(object_creation[i].clsid, rclsid))
 257             return IClassFactory_QueryInterface(object_creation[i].cf, riid, ppv);
 258     }
 259
 260     FIXME("%s: no class found.\n", debugstr_guid(rclsid));
 261     return CLASS_E_CLASSNOTAVAILABLE;
 262 }
 263
 264
 265 /***********************************************************************
 266  *              DllRegisterServerEx (URLMON.@)
 267  */
 268 HRESULT WINAPI DllRegisterServerEx(void)
 269 {
 270     FIXME("(void): stub\n");
 271
 272     return E_FAIL;
 273 }
 274
 275 /**************************************************************************
 276  *                 UrlMkSetSessionOption (URLMON.@)
 277  */
 278 HRESULT WINAPI UrlMkSetSessionOption(DWORD dwOption, LPVOID pBuffer, DWORD dwBufferLength,
 279                                         DWORD Reserved)
 280 {
 281     FIXME("(%#lx, %p, %#lx): stub\n", dwOption, pBuffer, dwBufferLength);
 282
 283     return S_OK;
 284 }
 285
 286 static const CHAR Agent[] = "Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.0)";
 287
 288 /**************************************************************************
 289  *                 ObtainUserAgentString (URLMON.@)
 290  */
 291 HRESULT WINAPI ObtainUserAgentString(DWORD dwOption, LPSTR pcszUAOut, DWORD *cbSize)
 292 {
 293     FIXME("(%ld, %p, %p): stub\n", dwOption, pcszUAOut, cbSize);
 294
 295     if(dwOption) {
 296       ERR("dwOption: %ld, must be zero\n", dwOption);
 297     }
 298
 299     if (sizeof(Agent) < *cbSize)
 300         *cbSize = sizeof(Agent);
 301     lstrcpynA(pcszUAOut, Agent, *cbSize);
 302
 303     return S_OK;
 304 }
 305
 306 HRESULT WINAPI CoInternetCompareUrl(LPCWSTR pwzUrl1, LPCWSTR pwzUrl2, DWORD dwCompareFlags)
 307 {
 308     TRACE("(%s,%s,%08lx)\n", debugstr_w(pwzUrl1), debugstr_w(pwzUrl2), dwCompareFlags);
 309     return UrlCompareW(pwzUrl1, pwzUrl2, dwCompareFlags)==0?S_OK:S_FALSE;
 310 }
 311
 312 /**************************************************************************
 313  *                 IsValidURL (URLMON.@)
 314  *
 315  * Determines if a specified string is a valid URL.
 316  *
 317  * PARAMS
 318  *  pBC        [I] ignored, must be NULL.
 319  *  szURL      [I] string that represents the URL in question.
 320  *  dwReserved [I] reserved and must be zero.
 321  *
 322  * RETURNS
 323  *  Success: S_OK.
 324  *  Failure: S_FALSE.
 325  *  returns E_INVALIDARG if one or more of the args is invalid.
 326  *
 327  * TODO:
 328  *  test functionality against windows to see what a valid URL is.
 329  */
 330 HRESULT WINAPI IsValidURL(LPBC pBC, LPCWSTR szURL, DWORD dwReserved)
 331 {
 332     FIXME("(%p, %s, %ld): stub\n", pBC, debugstr_w(szURL), dwReserved);
 333
 334     if (pBC != NULL || dwReserved != 0)
 335         return E_INVALIDARG;
 336
 337     return S_OK;
 338 }
 339
 340 /**************************************************************************
 341  *                 FaultInIEFeature (URLMON.@)
 342  *
 343  *  Undocumented.  Appears to be used by native shdocvw.dll.
 344  */
 345 HRESULT WINAPI FaultInIEFeature( HWND hwnd, uCLSSPEC * pClassSpec,
 346                                  QUERYCONTEXT *pQuery, DWORD flags )
 347 {
 348     FIXME("%p %p %p %08lx\n", hwnd, pClassSpec, pQuery, flags);
 349     return E_NOTIMPL;
 350 }
 351
 352 /**************************************************************************
 353  *                 CoGetClassObjectFromURL (URLMON.@)
 354  */
 355 HRESULT WINAPI CoGetClassObjectFromURL( REFCLSID rclsid, LPCWSTR szCodeURL, DWORD dwFileVersionMS,
 356                                         DWORD dwFileVersionLS, LPCWSTR szContentType,
 357                                         LPBINDCTX pBindCtx, DWORD dwClsContext, LPVOID pvReserved,
 358                                         REFIID riid, LPVOID *ppv )
 359 {
 360     FIXME("(%s %s %ld %ld %s %p %ld %p %s %p) Stub!\n", debugstr_guid(rclsid), debugstr_w(szCodeURL),
 361         dwFileVersionMS, dwFileVersionLS, debugstr_w(szContentType), pBindCtx, dwClsContext, pvReserved,
 362         debugstr_guid(riid), ppv);
 363     return E_NOINTERFACE;
 364 }
 365
 366 /***********************************************************************
 367  *           ReleaseBindInfo (URLMON.@)
 368  *
 369  * Release the resources used by the specified BINDINFO structure.
 370  *
 371  * PARAMS
 372  *  pbindinfo [I] BINDINFO to release.
 373  *
 374  * RETURNS
 375  *  Nothing.
 376  */
 377 void WINAPI ReleaseBindInfo(BINDINFO* pbindinfo)
 378 {
 379     DWORD size;
 380
 381     TRACE("(%p)\n", pbindinfo);
 382
 383     if(!pbindinfo || !(size = pbindinfo->cbSize))
 384         return;
 385
 386     CoTaskMemFree(pbindinfo->szExtraInfo);
 387     ReleaseStgMedium(&pbindinfo->stgmedData);
 388
 389     if(offsetof(BINDINFO, szExtraInfo) < size)
 390         CoTaskMemFree(pbindinfo->szCustomVerb);
 391
 392
 393     if(pbindinfo->pUnk && offsetof(BINDINFO, pUnk) < size)
 394         IUnknown_Release(pbindinfo->pUnk);
 395
 396     memset(pbindinfo, 0, size);
 397     pbindinfo->cbSize = size;
 398 }
 399
 400 /***********************************************************************
 401  *           FindMimeFromData (URLMON.@)
 402  *
 403  * Determines the Multipurpose Internet Mail Extensions (MIME) type from the data provided.
 404  */
 405 static BOOL text_html_filter(LPVOID buf, DWORD size)
 406 {
 407     const char *b = buf;
 408     int i;
 409
 410     if(size < 5)
 411         return FALSE;
 412
 413     for(i=0; i < size-5; i++) {
 414         if(b[i] == '<'
 415            && (b[i+1] == 'h' || b[i+1] == 'H')
 416            && (b[i+2] == 't' || b[i+2] == 'T')
 417            && (b[i+3] == 'm' || b[i+3] == 'M')
 418            && (b[i+4] == 'l' || b[i+4] == 'L'))
 419             return TRUE;
 420     }
 421
 422     return FALSE;
 423 }
 424
 425 static BOOL image_gif_filter(LPVOID buf, DWORD size)
 426 {
 427     const BYTE const *b = buf;
 428
 429     return size >= 6
 430         && (b[0] == 'G' || b[0] == 'g')
 431         && (b[1] == 'I' || b[1] == 'i')
 432         && (b[2] == 'F' || b[2] == 'f')
 433         &&  b[3] == '8'
 434         && (b[4] == '7' || b[4] == '9')
 435         && (b[5] == 'A' || b[5] == 'a');
 436 }
 437
 438 static BOOL image_pjpeg_filter(LPVOID buf, DWORD size)
 439 {
 440     return size > 2 && *(BYTE*)buf == 0xff && *((BYTE*)buf+1) == 0xd8;
 441 }
 442
 443 static BOOL image_xpng_filter(LPVOID buf, DWORD size)
 444 {
 445     static const BYTE xpng_header[] = {0x89,'P','N','G',0x0d,0x0a,0x1a,0x0a};
 446     return size > sizeof(xpng_header) && !memcmp(buf, xpng_header, sizeof(xpng_header));
 447 }
 448
 449 static BOOL image_bmp_filter(LPVOID buf, DWORD size)
 450 {
 451     return size >= 14
 452         && *(BYTE*)buf == 0x42
 453         && *((BYTE*)buf+1) == 0x4d
 454         && *(DWORD*)((BYTE*)buf+6) == 0;
 455 }
 456
 457 static BOOL text_plain_filter(LPVOID buf, DWORD size)
 458 {
 459     UCHAR *ptr;
 460
 461     for(ptr = buf; ptr < (UCHAR*)buf+size-1; ptr++) {
 462         if(*ptr < 0x20 && *ptr != '\n' && *ptr != '\r' && *ptr != '\t')
 463             return FALSE;
 464     }
 465
 466     return TRUE;
 467 }
 468
 469 static BOOL application_octet_stream_filter(LPVOID buf, DWORD size)
 470 {
 471     return TRUE;
 472 }
 473
 474 HRESULT WINAPI FindMimeFromData(LPBC pBC, LPCWSTR pwzUrl, LPVOID pBuffer,
 475         DWORD cbSize, LPCWSTR pwzMimeProposed, DWORD dwMimeFlags,
 476         LPWSTR* ppwzMimeOut, DWORD dwReserved)
 477 {
 478     TRACE("(%p,%s,%p,%ld,%s,0x%lx,%p,0x%lx)\n", pBC, debugstr_w(pwzUrl), pBuffer, cbSize,
 479             debugstr_w(pwzMimeProposed), dwMimeFlags, ppwzMimeOut, dwReserved);
 480
 481     if(dwMimeFlags)
 482         WARN("dwMimeFlags=%08lx\n", dwMimeFlags);
 483     if(dwReserved)
 484         WARN("dwReserved=%ld\n", dwReserved);
 485
 486     /* pBC seams to not be used */
 487
 488     if(!ppwzMimeOut || (!pwzUrl && !pBuffer))
 489         return E_INVALIDARG;
 490
 491     if(pwzMimeProposed && (!pBuffer || (pBuffer && !cbSize))) {
 492         DWORD len;
 493
 494         if(!pwzMimeProposed)
 495             return E_FAIL;
 496
 497         len = strlenW(pwzMimeProposed)+1;
 498         *ppwzMimeOut = CoTaskMemAlloc(len*sizeof(WCHAR));
 499         memcpy(*ppwzMimeOut, pwzMimeProposed, len*sizeof(WCHAR));
 500         return S_OK;
 501     }
 502
 503     if(pBuffer) {
 504         DWORD len;
 505         LPCWSTR ret = NULL;
 506         int i;
 507
 508         static const WCHAR wszTextHtml[] = {'t','e','x','t','/','h','t','m','l',0};
 509         static const WCHAR wszImageGif[] = {'i','m','a','g','e','/','g','i','f',0};
 510         static const WCHAR wszImagePjpeg[] = {'i','m','a','g','e','/','p','j','p','e','g',0};
 511         static const WCHAR wszImageXPng[] = {'i','m','a','g','e','/','x','-','p','n','g',0};
 512         static const WCHAR wszImageBmp[] = {'i','m','a','g','e','/','b','m','p',0};
 513         static const WCHAR wszTextPlain[] = {'t','e','x','t','/','p','l','a','i','n','\0'};
 514         static const WCHAR wszAppOctetStream[] = {'a','p','p','l','i','c','a','t','i','o','n','/',
 515             'o','c','t','e','t','-','s','t','r','e','a','m','\0'};
 516
 517         static const struct {
 518             LPCWSTR mime;
 519             BOOL (*filter)(LPVOID,DWORD);
 520         } mime_filters[] = {
 521             {wszTextHtml,       text_html_filter},
 522             {wszImageGif,       image_gif_filter},
 523             {wszImagePjpeg,     image_pjpeg_filter},
 524             {wszImageXPng,      image_xpng_filter},
 525             {wszImageBmp,       image_bmp_filter},
 526             {wszTextPlain,      text_plain_filter},
 527             {wszAppOctetStream, application_octet_stream_filter}
 528         };
 529
 530         if(!cbSize)
 531             return E_FAIL;
 532
 533         if(pwzMimeProposed && strcmpW(pwzMimeProposed, wszAppOctetStream)) {
 534             for(i=0; i < sizeof(mime_filters)/sizeof(*mime_filters); i++) {
 535                 if(!strcmpW(pwzMimeProposed, mime_filters[i].mime))
 536                     break;
 537             }
 538
 539             if(i == sizeof(mime_filters)/sizeof(*mime_filters)
 540                     || mime_filters[i].filter(pBuffer, cbSize)) {
 541                 len = strlenW(pwzMimeProposed)+1;
 542                 *ppwzMimeOut = CoTaskMemAlloc(len*sizeof(WCHAR));
 543                 memcpy(*ppwzMimeOut, pwzMimeProposed, len*sizeof(WCHAR));
 544                 return S_OK;
 545             }
 546         }
 547
 548         i=0;
 549         while(!ret) {
 550             if(mime_filters[i].filter(pBuffer, cbSize))
 551                 ret = mime_filters[i].mime;
 552             i++;
 553         }
 554
 555         if(pwzMimeProposed) {
 556             if(i == sizeof(mime_filters)/sizeof(*mime_filters))
 557                 ret = pwzMimeProposed;
 558
 559             /* text/html is a special case */
 560            if(!strcmpW(pwzMimeProposed, wszTextHtml) && !strcmpW(ret, wszTextPlain))
 561                 ret = wszTextHtml;
 562         }
 563
 564         len = strlenW(ret)+1;
 565         *ppwzMimeOut = CoTaskMemAlloc(len*sizeof(WCHAR));
 566         memcpy(*ppwzMimeOut, ret, len*sizeof(WCHAR));
 567         return S_OK;
 568     }
 569
 570     if(pwzUrl) {
 571         HKEY hkey;
 572         DWORD res, size;
 573         LPCWSTR ptr;
 574         WCHAR mime[64];
 575
 576         static const WCHAR wszContentType[] =
 577                 {'C','o','n','t','e','n','t',' ','T','y','p','e','\0'};
 578
 579         ptr = strrchrW(pwzUrl, '.');
 580         if(!ptr)
 581             return E_FAIL;
 582
 583         res = RegOpenKeyW(HKEY_CLASSES_ROOT, ptr, &hkey);
 584         if(res != ERROR_SUCCESS)
 585             return E_FAIL;
 586
 587         size = sizeof(mime);
 588         res = RegQueryValueExW(hkey, wszContentType, NULL, NULL, (LPBYTE)mime, &size);
 589         RegCloseKey(hkey);
 590         if(res != ERROR_SUCCESS)
 591             return E_FAIL;
 592
 593         *ppwzMimeOut = CoTaskMemAlloc(size);
 594         memcpy(*ppwzMimeOut, mime, size);
 595         return S_OK;
 596     }
 597
 598     return E_FAIL;
 599 }